329 lines
14 KiB
C++
329 lines
14 KiB
C++
//===-- InstructionSimplify.h - Fold instrs into simpler forms --*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file declares routines for folding instructions into simpler forms
|
|
// that do not require creating new instructions. This does constant folding
|
|
// ("add i32 1, 1" -> "2") but can also handle non-constant operands, either
|
|
// returning a constant ("and i32 %x, 0" -> "0") or an already existing value
|
|
// ("and i32 %x, %x" -> "%x"). If the simplification is also an instruction
|
|
// then it dominates the original instruction.
|
|
//
|
|
// These routines implicitly resolve undef uses. The easiest way to be safe when
|
|
// using these routines to obtain simplified values for existing instructions is
|
|
// to always replace all uses of the instructions with the resulting simplified
|
|
// values. This will prevent other code from seeing the same undef uses and
|
|
// resolving them to different values.
|
|
//
|
|
// These routines are designed to tolerate moderately incomplete IR, such as
|
|
// instructions that are not connected to basic blocks yet. However, they do
|
|
// require that all the IR that they encounter be valid. In particular, they
|
|
// require that all non-constant values be defined in the same function, and the
|
|
// same call context of that function (and not split between caller and callee
|
|
// contexts of a directly recursive call, for example).
|
|
//
|
|
// Additionally, these routines can't simplify to the instructions that are not
|
|
// def-reachable, meaning we can't just scan the basic block for instructions
|
|
// to simplify to.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_ANALYSIS_INSTRUCTIONSIMPLIFY_H
|
|
#define LLVM_ANALYSIS_INSTRUCTIONSIMPLIFY_H
|
|
|
|
#include "llvm/IR/Instruction.h"
|
|
#include "llvm/IR/Operator.h"
|
|
|
|
namespace llvm {
|
|
|
|
template <typename T, typename... TArgs> class AnalysisManager;
|
|
template <class T> class ArrayRef;
|
|
class AssumptionCache;
|
|
class BinaryOperator;
|
|
class CallBase;
|
|
class DataLayout;
|
|
class DominatorTree;
|
|
class Function;
|
|
struct LoopStandardAnalysisResults;
|
|
class MDNode;
|
|
class OptimizationRemarkEmitter;
|
|
class Pass;
|
|
template <class T, unsigned n> class SmallSetVector;
|
|
class TargetLibraryInfo;
|
|
class Type;
|
|
class Value;
|
|
|
|
/// InstrInfoQuery provides an interface to query additional information for
|
|
/// instructions like metadata or keywords like nsw, which provides conservative
|
|
/// results if the users specified it is safe to use.
|
|
struct InstrInfoQuery {
|
|
InstrInfoQuery(bool UMD) : UseInstrInfo(UMD) {}
|
|
InstrInfoQuery() : UseInstrInfo(true) {}
|
|
bool UseInstrInfo = true;
|
|
|
|
MDNode *getMetadata(const Instruction *I, unsigned KindID) const {
|
|
if (UseInstrInfo)
|
|
return I->getMetadata(KindID);
|
|
return nullptr;
|
|
}
|
|
|
|
template <class InstT> bool hasNoUnsignedWrap(const InstT *Op) const {
|
|
if (UseInstrInfo)
|
|
return Op->hasNoUnsignedWrap();
|
|
return false;
|
|
}
|
|
|
|
template <class InstT> bool hasNoSignedWrap(const InstT *Op) const {
|
|
if (UseInstrInfo)
|
|
return Op->hasNoSignedWrap();
|
|
return false;
|
|
}
|
|
|
|
bool isExact(const BinaryOperator *Op) const {
|
|
if (UseInstrInfo && isa<PossiblyExactOperator>(Op))
|
|
return cast<PossiblyExactOperator>(Op)->isExact();
|
|
return false;
|
|
}
|
|
};
|
|
|
|
struct SimplifyQuery {
|
|
const DataLayout &DL;
|
|
const TargetLibraryInfo *TLI = nullptr;
|
|
const DominatorTree *DT = nullptr;
|
|
AssumptionCache *AC = nullptr;
|
|
const Instruction *CxtI = nullptr;
|
|
|
|
// Wrapper to query additional information for instructions like metadata or
|
|
// keywords like nsw, which provides conservative results if those cannot
|
|
// be safely used.
|
|
const InstrInfoQuery IIQ;
|
|
|
|
/// Controls whether simplifications are allowed to constrain the range of
|
|
/// possible values for uses of undef. If it is false, simplifications are not
|
|
/// allowed to assume a particular value for a use of undef for example.
|
|
bool CanUseUndef = true;
|
|
|
|
SimplifyQuery(const DataLayout &DL, const Instruction *CXTI = nullptr)
|
|
: DL(DL), CxtI(CXTI) {}
|
|
|
|
SimplifyQuery(const DataLayout &DL, const TargetLibraryInfo *TLI,
|
|
const DominatorTree *DT = nullptr,
|
|
AssumptionCache *AC = nullptr,
|
|
const Instruction *CXTI = nullptr, bool UseInstrInfo = true,
|
|
bool CanUseUndef = true)
|
|
: DL(DL), TLI(TLI), DT(DT), AC(AC), CxtI(CXTI), IIQ(UseInstrInfo),
|
|
CanUseUndef(CanUseUndef) {}
|
|
SimplifyQuery getWithInstruction(Instruction *I) const {
|
|
SimplifyQuery Copy(*this);
|
|
Copy.CxtI = I;
|
|
return Copy;
|
|
}
|
|
SimplifyQuery getWithoutUndef() const {
|
|
SimplifyQuery Copy(*this);
|
|
Copy.CanUseUndef = false;
|
|
return Copy;
|
|
}
|
|
|
|
/// If CanUseUndef is true, returns whether \p V is undef.
|
|
/// Otherwise always return false.
|
|
bool isUndefValue(Value *V) const {
|
|
if (!CanUseUndef)
|
|
return false;
|
|
return isa<UndefValue>(V);
|
|
}
|
|
};
|
|
|
|
// NOTE: the explicit multiple argument versions of these functions are
|
|
// deprecated.
|
|
// Please use the SimplifyQuery versions in new code.
|
|
|
|
/// Given operand for an FNeg, fold the result or return null.
|
|
Value *SimplifyFNegInst(Value *Op, FastMathFlags FMF,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an Add, fold the result or return null.
|
|
Value *SimplifyAddInst(Value *LHS, Value *RHS, bool isNSW, bool isNUW,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a Sub, fold the result or return null.
|
|
Value *SimplifySubInst(Value *LHS, Value *RHS, bool isNSW, bool isNUW,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an FAdd, fold the result or return null.
|
|
Value *SimplifyFAddInst(Value *LHS, Value *RHS, FastMathFlags FMF,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an FSub, fold the result or return null.
|
|
Value *SimplifyFSubInst(Value *LHS, Value *RHS, FastMathFlags FMF,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an FMul, fold the result or return null.
|
|
Value *SimplifyFMulInst(Value *LHS, Value *RHS, FastMathFlags FMF,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for the multiplication of a FMA, fold the result or return
|
|
/// null. In contrast to SimplifyFMulInst, this function will not perform
|
|
/// simplifications whose unrounded results differ when rounded to the argument
|
|
/// type.
|
|
Value *SimplifyFMAFMul(Value *LHS, Value *RHS, FastMathFlags FMF,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a Mul, fold the result or return null.
|
|
Value *SimplifyMulInst(Value *LHS, Value *RHS, const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an SDiv, fold the result or return null.
|
|
Value *SimplifySDivInst(Value *LHS, Value *RHS, const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a UDiv, fold the result or return null.
|
|
Value *SimplifyUDivInst(Value *LHS, Value *RHS, const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an FDiv, fold the result or return null.
|
|
Value *SimplifyFDivInst(Value *LHS, Value *RHS, FastMathFlags FMF,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an SRem, fold the result or return null.
|
|
Value *SimplifySRemInst(Value *LHS, Value *RHS, const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a URem, fold the result or return null.
|
|
Value *SimplifyURemInst(Value *LHS, Value *RHS, const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an FRem, fold the result or return null.
|
|
Value *SimplifyFRemInst(Value *LHS, Value *RHS, FastMathFlags FMF,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a Shl, fold the result or return null.
|
|
Value *SimplifyShlInst(Value *Op0, Value *Op1, bool isNSW, bool isNUW,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a LShr, fold the result or return null.
|
|
Value *SimplifyLShrInst(Value *Op0, Value *Op1, bool isExact,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a AShr, fold the result or return nulll.
|
|
Value *SimplifyAShrInst(Value *Op0, Value *Op1, bool isExact,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an And, fold the result or return null.
|
|
Value *SimplifyAndInst(Value *LHS, Value *RHS, const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an Or, fold the result or return null.
|
|
Value *SimplifyOrInst(Value *LHS, Value *RHS, const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an Xor, fold the result or return null.
|
|
Value *SimplifyXorInst(Value *LHS, Value *RHS, const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an ICmpInst, fold the result or return null.
|
|
Value *SimplifyICmpInst(unsigned Predicate, Value *LHS, Value *RHS,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an FCmpInst, fold the result or return null.
|
|
Value *SimplifyFCmpInst(unsigned Predicate, Value *LHS, Value *RHS,
|
|
FastMathFlags FMF, const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a SelectInst, fold the result or return null.
|
|
Value *SimplifySelectInst(Value *Cond, Value *TrueVal, Value *FalseVal,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a GetElementPtrInst, fold the result or return null.
|
|
Value *SimplifyGEPInst(Type *SrcTy, ArrayRef<Value *> Ops,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an InsertValueInst, fold the result or return null.
|
|
Value *SimplifyInsertValueInst(Value *Agg, Value *Val, ArrayRef<unsigned> Idxs,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an InsertElement, fold the result or return null.
|
|
Value *SimplifyInsertElementInst(Value *Vec, Value *Elt, Value *Idx,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an ExtractValueInst, fold the result or return null.
|
|
Value *SimplifyExtractValueInst(Value *Agg, ArrayRef<unsigned> Idxs,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for an ExtractElementInst, fold the result or return null.
|
|
Value *SimplifyExtractElementInst(Value *Vec, Value *Idx,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a CastInst, fold the result or return null.
|
|
Value *SimplifyCastInst(unsigned CastOpc, Value *Op, Type *Ty,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a ShuffleVectorInst, fold the result or return null.
|
|
/// See class ShuffleVectorInst for a description of the mask representation.
|
|
Value *SimplifyShuffleVectorInst(Value *Op0, Value *Op1, ArrayRef<int> Mask,
|
|
Type *RetTy, const SimplifyQuery &Q);
|
|
|
|
//=== Helper functions for higher up the class hierarchy.
|
|
|
|
/// Given operands for a CmpInst, fold the result or return null.
|
|
Value *SimplifyCmpInst(unsigned Predicate, Value *LHS, Value *RHS,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operand for a UnaryOperator, fold the result or return null.
|
|
Value *SimplifyUnOp(unsigned Opcode, Value *Op, const SimplifyQuery &Q);
|
|
|
|
/// Given operand for a UnaryOperator, fold the result or return null.
|
|
/// Try to use FastMathFlags when folding the result.
|
|
Value *SimplifyUnOp(unsigned Opcode, Value *Op, FastMathFlags FMF,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a BinaryOperator, fold the result or return null.
|
|
Value *SimplifyBinOp(unsigned Opcode, Value *LHS, Value *RHS,
|
|
const SimplifyQuery &Q);
|
|
|
|
/// Given operands for a BinaryOperator, fold the result or return null.
|
|
/// Try to use FastMathFlags when folding the result.
|
|
Value *SimplifyBinOp(unsigned Opcode, Value *LHS, Value *RHS,
|
|
FastMathFlags FMF, const SimplifyQuery &Q);
|
|
|
|
/// Given a callsite, fold the result or return null.
|
|
Value *SimplifyCall(CallBase *Call, const SimplifyQuery &Q);
|
|
|
|
/// Given an operand for a Freeze, see if we can fold the result.
|
|
/// If not, this returns null.
|
|
Value *SimplifyFreezeInst(Value *Op, const SimplifyQuery &Q);
|
|
|
|
/// See if we can compute a simplified version of this instruction. If not,
|
|
/// return null.
|
|
Value *SimplifyInstruction(Instruction *I, const SimplifyQuery &Q,
|
|
OptimizationRemarkEmitter *ORE = nullptr);
|
|
|
|
/// See if V simplifies when its operand Op is replaced with RepOp. If not,
|
|
/// return null.
|
|
/// AllowRefinement specifies whether the simplification can be a refinement,
|
|
/// or whether it needs to be strictly identical.
|
|
Value *SimplifyWithOpReplaced(Value *V, Value *Op, Value *RepOp,
|
|
const SimplifyQuery &Q, bool AllowRefinement);
|
|
|
|
/// Replace all uses of 'I' with 'SimpleV' and simplify the uses recursively.
|
|
///
|
|
/// This first performs a normal RAUW of I with SimpleV. It then recursively
|
|
/// attempts to simplify those users updated by the operation. The 'I'
|
|
/// instruction must not be equal to the simplified value 'SimpleV'.
|
|
/// If UnsimplifiedUsers is provided, instructions that could not be simplified
|
|
/// are added to it.
|
|
///
|
|
/// The function returns true if any simplifications were performed.
|
|
bool replaceAndRecursivelySimplify(
|
|
Instruction *I, Value *SimpleV, const TargetLibraryInfo *TLI = nullptr,
|
|
const DominatorTree *DT = nullptr, AssumptionCache *AC = nullptr,
|
|
SmallSetVector<Instruction *, 8> *UnsimplifiedUsers = nullptr);
|
|
|
|
// These helper functions return a SimplifyQuery structure that contains as
|
|
// many of the optional analysis we use as are currently valid. This is the
|
|
// strongly preferred way of constructing SimplifyQuery in passes.
|
|
const SimplifyQuery getBestSimplifyQuery(Pass &, Function &);
|
|
template <class T, class... TArgs>
|
|
const SimplifyQuery getBestSimplifyQuery(AnalysisManager<T, TArgs...> &,
|
|
Function &);
|
|
const SimplifyQuery getBestSimplifyQuery(LoopStandardAnalysisResults &,
|
|
const DataLayout &);
|
|
} // end namespace llvm
|
|
|
|
#endif
|
|
|