Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line | 
|---|---|---|---|
| 14 | pmbaty | 1 | //===- InstSimplifyFolder.h - InstSimplify folding helper --------*- C++-*-===// | 
| 2 | // | ||
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. | ||
| 4 | // See https://llvm.org/LICENSE.txt for license information. | ||
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
| 6 | // | ||
| 7 | //===----------------------------------------------------------------------===// | ||
| 8 | // | ||
| 9 | // This file defines the InstSimplifyFolder class, a helper for IRBuilder. | ||
| 10 | // It provides IRBuilder with a set of methods for folding operations to | ||
| 11 | // existing values using InstructionSimplify. At the moment, only a subset of | ||
| 12 | // the implementation uses InstructionSimplify. The rest of the implementation | ||
| 13 | // only folds constants. | ||
| 14 | // | ||
| 15 | // The folder also applies target-specific constant folding. | ||
| 16 | // | ||
| 17 | //===----------------------------------------------------------------------===// | ||
| 18 | |||
| 19 | #ifndef LLVM_ANALYSIS_INSTSIMPLIFYFOLDER_H | ||
| 20 | #define LLVM_ANALYSIS_INSTSIMPLIFYFOLDER_H | ||
| 21 | |||
| 22 | #include "llvm/ADT/ArrayRef.h" | ||
| 23 | #include "llvm/Analysis/InstructionSimplify.h" | ||
| 24 | #include "llvm/Analysis/TargetFolder.h" | ||
| 25 | #include "llvm/IR/IRBuilderFolder.h" | ||
| 26 | #include "llvm/IR/Instruction.h" | ||
| 27 | |||
| 28 | namespace llvm { | ||
| 29 | class Constant; | ||
| 30 | |||
| 31 | /// InstSimplifyFolder - Use InstructionSimplify to fold operations to existing | ||
| 32 | /// values. Also applies target-specific constant folding when not using | ||
| 33 | /// InstructionSimplify. | ||
| 34 | class InstSimplifyFolder final : public IRBuilderFolder { | ||
| 35 |   TargetFolder ConstFolder; | ||
| 36 |   SimplifyQuery SQ; | ||
| 37 | |||
| 38 | virtual void anchor(); | ||
| 39 | |||
| 40 | public: | ||
| 41 | InstSimplifyFolder(const DataLayout &DL) : ConstFolder(DL), SQ(DL) {} | ||
| 42 | |||
| 43 |   //===--------------------------------------------------------------------===// | ||
| 44 |   // Value-based folders. | ||
| 45 |   // | ||
| 46 |   // Return an existing value or a constant if the operation can be simplified. | ||
| 47 |   // Otherwise return nullptr. | ||
| 48 |   //===--------------------------------------------------------------------===// | ||
| 49 | |||
| 50 | Value *FoldBinOp(Instruction::BinaryOps Opc, Value *LHS, | ||
| 51 | Value *RHS) const override { | ||
| 52 | return simplifyBinOp(Opc, LHS, RHS, SQ); | ||
| 53 |   } | ||
| 54 | |||
| 55 | Value *FoldExactBinOp(Instruction::BinaryOps Opc, Value *LHS, Value *RHS, | ||
| 56 | bool IsExact) const override { | ||
| 57 | return simplifyBinOp(Opc, LHS, RHS, SQ); | ||
| 58 |   } | ||
| 59 | |||
| 60 | Value *FoldNoWrapBinOp(Instruction::BinaryOps Opc, Value *LHS, Value *RHS, | ||
| 61 | bool HasNUW, bool HasNSW) const override { | ||
| 62 | return simplifyBinOp(Opc, LHS, RHS, SQ); | ||
| 63 |   } | ||
| 64 | |||
| 65 | Value *FoldBinOpFMF(Instruction::BinaryOps Opc, Value *LHS, Value *RHS, | ||
| 66 | FastMathFlags FMF) const override { | ||
| 67 | return simplifyBinOp(Opc, LHS, RHS, FMF, SQ); | ||
| 68 |   } | ||
| 69 | |||
| 70 | Value *FoldUnOpFMF(Instruction::UnaryOps Opc, Value *V, | ||
| 71 | FastMathFlags FMF) const override { | ||
| 72 | return simplifyUnOp(Opc, V, FMF, SQ); | ||
| 73 |   } | ||
| 74 | |||
| 75 | Value *FoldICmp(CmpInst::Predicate P, Value *LHS, Value *RHS) const override { | ||
| 76 | return simplifyICmpInst(P, LHS, RHS, SQ); | ||
| 77 |   } | ||
| 78 | |||
| 79 | Value *FoldGEP(Type *Ty, Value *Ptr, ArrayRef<Value *> IdxList, | ||
| 80 | bool IsInBounds = false) const override { | ||
| 81 | return simplifyGEPInst(Ty, Ptr, IdxList, IsInBounds, SQ); | ||
| 82 |   } | ||
| 83 | |||
| 84 | Value *FoldSelect(Value *C, Value *True, Value *False) const override { | ||
| 85 | return simplifySelectInst(C, True, False, SQ); | ||
| 86 |   } | ||
| 87 | |||
| 88 | Value *FoldExtractValue(Value *Agg, | ||
| 89 | ArrayRef<unsigned> IdxList) const override { | ||
| 90 | return simplifyExtractValueInst(Agg, IdxList, SQ); | ||
| 91 | }; | ||
| 92 | |||
| 93 | Value *FoldInsertValue(Value *Agg, Value *Val, | ||
| 94 | ArrayRef<unsigned> IdxList) const override { | ||
| 95 | return simplifyInsertValueInst(Agg, Val, IdxList, SQ); | ||
| 96 |   } | ||
| 97 | |||
| 98 | Value *FoldExtractElement(Value *Vec, Value *Idx) const override { | ||
| 99 | return simplifyExtractElementInst(Vec, Idx, SQ); | ||
| 100 |   } | ||
| 101 | |||
| 102 | Value *FoldInsertElement(Value *Vec, Value *NewElt, | ||
| 103 | Value *Idx) const override { | ||
| 104 | return simplifyInsertElementInst(Vec, NewElt, Idx, SQ); | ||
| 105 |   } | ||
| 106 | |||
| 107 | Value *FoldShuffleVector(Value *V1, Value *V2, | ||
| 108 | ArrayRef<int> Mask) const override { | ||
| 109 | Type *RetTy = VectorType::get( | ||
| 110 | cast<VectorType>(V1->getType())->getElementType(), Mask.size(), | ||
| 111 | isa<ScalableVectorType>(V1->getType())); | ||
| 112 | return simplifyShuffleVectorInst(V1, V2, Mask, RetTy, SQ); | ||
| 113 |   } | ||
| 114 | |||
| 115 |   //===--------------------------------------------------------------------===// | ||
| 116 |   // Cast/Conversion Operators | ||
| 117 |   //===--------------------------------------------------------------------===// | ||
| 118 | |||
| 119 | Value *CreateCast(Instruction::CastOps Op, Constant *C, | ||
| 120 | Type *DestTy) const override { | ||
| 121 | if (C->getType() == DestTy) | ||
| 122 | return C; // avoid calling Fold | ||
| 123 | return ConstFolder.CreateCast(Op, C, DestTy); | ||
| 124 |   } | ||
| 125 | Value *CreateIntCast(Constant *C, Type *DestTy, | ||
| 126 | bool isSigned) const override { | ||
| 127 | if (C->getType() == DestTy) | ||
| 128 | return C; // avoid calling Fold | ||
| 129 | return ConstFolder.CreateIntCast(C, DestTy, isSigned); | ||
| 130 |   } | ||
| 131 | Value *CreatePointerCast(Constant *C, Type *DestTy) const override { | ||
| 132 | if (C->getType() == DestTy) | ||
| 133 | return C; // avoid calling Fold | ||
| 134 | return ConstFolder.CreatePointerCast(C, DestTy); | ||
| 135 |   } | ||
| 136 | Value *CreateFPCast(Constant *C, Type *DestTy) const override { | ||
| 137 | if (C->getType() == DestTy) | ||
| 138 | return C; // avoid calling Fold | ||
| 139 | return ConstFolder.CreateFPCast(C, DestTy); | ||
| 140 |   } | ||
| 141 | Value *CreateBitCast(Constant *C, Type *DestTy) const override { | ||
| 142 | return ConstFolder.CreateBitCast(C, DestTy); | ||
| 143 |   } | ||
| 144 | Value *CreateIntToPtr(Constant *C, Type *DestTy) const override { | ||
| 145 | return ConstFolder.CreateIntToPtr(C, DestTy); | ||
| 146 |   } | ||
| 147 | Value *CreatePtrToInt(Constant *C, Type *DestTy) const override { | ||
| 148 | return ConstFolder.CreatePtrToInt(C, DestTy); | ||
| 149 |   } | ||
| 150 | Value *CreateZExtOrBitCast(Constant *C, Type *DestTy) const override { | ||
| 151 | if (C->getType() == DestTy) | ||
| 152 | return C; // avoid calling Fold | ||
| 153 | return ConstFolder.CreateZExtOrBitCast(C, DestTy); | ||
| 154 |   } | ||
| 155 | Value *CreateSExtOrBitCast(Constant *C, Type *DestTy) const override { | ||
| 156 | if (C->getType() == DestTy) | ||
| 157 | return C; // avoid calling Fold | ||
| 158 | return ConstFolder.CreateSExtOrBitCast(C, DestTy); | ||
| 159 |   } | ||
| 160 | Value *CreateTruncOrBitCast(Constant *C, Type *DestTy) const override { | ||
| 161 | if (C->getType() == DestTy) | ||
| 162 | return C; // avoid calling Fold | ||
| 163 | return ConstFolder.CreateTruncOrBitCast(C, DestTy); | ||
| 164 |   } | ||
| 165 | |||
| 166 | Value *CreatePointerBitCastOrAddrSpaceCast(Constant *C, | ||
| 167 | Type *DestTy) const override { | ||
| 168 | if (C->getType() == DestTy) | ||
| 169 | return C; // avoid calling Fold | ||
| 170 | return ConstFolder.CreatePointerBitCastOrAddrSpaceCast(C, DestTy); | ||
| 171 |   } | ||
| 172 | |||
| 173 |   //===--------------------------------------------------------------------===// | ||
| 174 |   // Compare Instructions | ||
| 175 |   //===--------------------------------------------------------------------===// | ||
| 176 | |||
| 177 | Value *CreateFCmp(CmpInst::Predicate P, Constant *LHS, | ||
| 178 | Constant *RHS) const override { | ||
| 179 | return ConstFolder.CreateFCmp(P, LHS, RHS); | ||
| 180 |   } | ||
| 181 | }; | ||
| 182 | |||
| 183 | } // end namespace llvm | ||
| 184 | |||
| 185 | #endif // LLVM_ANALYSIS_INSTSIMPLIFYFOLDER_H |