Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1 | //===- InstructionCombining.cpp - Combine multiple instructions -----------===// |
2 | // | ||||
3 | // The LLVM Compiler Infrastructure | ||||
4 | // | ||||
Chris Lattner | 081ce94 | 2007-12-29 20:36:04 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
6 | // License. See LICENSE.TXT for details. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7 | // |
8 | //===----------------------------------------------------------------------===// | ||||
9 | // | ||||
10 | // InstructionCombining - Combine instructions to form fewer, simple | ||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 11 | // instructions. This pass does not modify the CFG. This pass is where |
12 | // algebraic simplification happens. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 13 | // |
14 | // This pass combines things like: | ||||
15 | // %Y = add i32 %X, 1 | ||||
16 | // %Z = add i32 %Y, 1 | ||||
17 | // into: | ||||
18 | // %Z = add i32 %X, 2 | ||||
19 | // | ||||
20 | // This is a simple worklist driven algorithm. | ||||
21 | // | ||||
22 | // This pass guarantees that the following canonicalizations are performed on | ||||
23 | // the program: | ||||
24 | // 1. If a binary operator has a constant operand, it is moved to the RHS | ||||
25 | // 2. Bitwise operators with constant operands are always grouped so that | ||||
26 | // shifts are performed first, then or's, then and's, then xor's. | ||||
27 | // 3. Compare instructions are converted from <,>,<=,>= to ==,!= if possible | ||||
28 | // 4. All cmp instructions on boolean values are replaced with logical ops | ||||
29 | // 5. add X, X is represented as (X*2) => (X << 1) | ||||
30 | // 6. Multiplies with a power-of-two constant argument are transformed into | ||||
31 | // shifts. | ||||
32 | // ... etc. | ||||
33 | // | ||||
34 | //===----------------------------------------------------------------------===// | ||||
35 | |||||
36 | #define DEBUG_TYPE "instcombine" | ||||
37 | #include "llvm/Transforms/Scalar.h" | ||||
38 | #include "llvm/IntrinsicInst.h" | ||||
39 | #include "llvm/Pass.h" | ||||
40 | #include "llvm/DerivedTypes.h" | ||||
41 | #include "llvm/GlobalVariable.h" | ||||
42 | #include "llvm/Analysis/ConstantFolding.h" | ||||
Chris Lattner | a432bc7 | 2008-06-02 01:18:21 +0000 | [diff] [blame] | 43 | #include "llvm/Analysis/ValueTracking.h" |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 44 | #include "llvm/Target/TargetData.h" |
45 | #include "llvm/Transforms/Utils/BasicBlockUtils.h" | ||||
46 | #include "llvm/Transforms/Utils/Local.h" | ||||
47 | #include "llvm/Support/CallSite.h" | ||||
Nick Lewycky | 0185bbf | 2008-02-03 16:33:09 +0000 | [diff] [blame] | 48 | #include "llvm/Support/ConstantRange.h" |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 49 | #include "llvm/Support/Debug.h" |
50 | #include "llvm/Support/GetElementPtrTypeIterator.h" | ||||
51 | #include "llvm/Support/InstVisitor.h" | ||||
52 | #include "llvm/Support/MathExtras.h" | ||||
53 | #include "llvm/Support/PatternMatch.h" | ||||
54 | #include "llvm/Support/Compiler.h" | ||||
55 | #include "llvm/ADT/DenseMap.h" | ||||
56 | #include "llvm/ADT/SmallVector.h" | ||||
57 | #include "llvm/ADT/SmallPtrSet.h" | ||||
58 | #include "llvm/ADT/Statistic.h" | ||||
59 | #include "llvm/ADT/STLExtras.h" | ||||
60 | #include <algorithm> | ||||
Edwin Török | a0e6fce | 2008-04-20 08:33:11 +0000 | [diff] [blame] | 61 | #include <climits> |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 62 | #include <sstream> |
63 | using namespace llvm; | ||||
64 | using namespace llvm::PatternMatch; | ||||
65 | |||||
66 | STATISTIC(NumCombined , "Number of insts combined"); | ||||
67 | STATISTIC(NumConstProp, "Number of constant folds"); | ||||
68 | STATISTIC(NumDeadInst , "Number of dead inst eliminated"); | ||||
69 | STATISTIC(NumDeadStore, "Number of dead stores eliminated"); | ||||
70 | STATISTIC(NumSunkInst , "Number of instructions sunk"); | ||||
71 | |||||
72 | namespace { | ||||
73 | class VISIBILITY_HIDDEN InstCombiner | ||||
74 | : public FunctionPass, | ||||
75 | public InstVisitor<InstCombiner, Instruction*> { | ||||
76 | // Worklist of all of the instructions that need to be simplified. | ||||
Chris Lattner | a06291a | 2008-08-15 04:03:01 +0000 | [diff] [blame] | 77 | SmallVector<Instruction*, 256> Worklist; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 78 | DenseMap<Instruction*, unsigned> WorklistMap; |
79 | TargetData *TD; | ||||
80 | bool MustPreserveLCSSA; | ||||
81 | public: | ||||
82 | static char ID; // Pass identification, replacement for typeid | ||||
Dan Gohman | 26f8c27 | 2008-09-04 17:05:41 +0000 | [diff] [blame] | 83 | InstCombiner() : FunctionPass(&ID) {} |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 84 | |
85 | /// AddToWorkList - Add the specified instruction to the worklist if it | ||||
86 | /// isn't already in it. | ||||
87 | void AddToWorkList(Instruction *I) { | ||||
Dan Gohman | 55d1966 | 2008-07-07 17:46:23 +0000 | [diff] [blame] | 88 | if (WorklistMap.insert(std::make_pair(I, Worklist.size())).second) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 89 | Worklist.push_back(I); |
90 | } | ||||
91 | |||||
92 | // RemoveFromWorkList - remove I from the worklist if it exists. | ||||
93 | void RemoveFromWorkList(Instruction *I) { | ||||
94 | DenseMap<Instruction*, unsigned>::iterator It = WorklistMap.find(I); | ||||
95 | if (It == WorklistMap.end()) return; // Not in worklist. | ||||
96 | |||||
97 | // Don't bother moving everything down, just null out the slot. | ||||
98 | Worklist[It->second] = 0; | ||||
99 | |||||
100 | WorklistMap.erase(It); | ||||
101 | } | ||||
102 | |||||
103 | Instruction *RemoveOneFromWorkList() { | ||||
104 | Instruction *I = Worklist.back(); | ||||
105 | Worklist.pop_back(); | ||||
106 | WorklistMap.erase(I); | ||||
107 | return I; | ||||
108 | } | ||||
109 | |||||
110 | |||||
111 | /// AddUsersToWorkList - When an instruction is simplified, add all users of | ||||
112 | /// the instruction to the work lists because they might get more simplified | ||||
113 | /// now. | ||||
114 | /// | ||||
115 | void AddUsersToWorkList(Value &I) { | ||||
116 | for (Value::use_iterator UI = I.use_begin(), UE = I.use_end(); | ||||
117 | UI != UE; ++UI) | ||||
118 | AddToWorkList(cast<Instruction>(*UI)); | ||||
119 | } | ||||
120 | |||||
121 | /// AddUsesToWorkList - When an instruction is simplified, add operands to | ||||
122 | /// the work lists because they might get more simplified now. | ||||
123 | /// | ||||
124 | void AddUsesToWorkList(Instruction &I) { | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 125 | for (User::op_iterator i = I.op_begin(), e = I.op_end(); i != e; ++i) |
126 | if (Instruction *Op = dyn_cast<Instruction>(*i)) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 127 | AddToWorkList(Op); |
128 | } | ||||
129 | |||||
130 | /// AddSoonDeadInstToWorklist - The specified instruction is about to become | ||||
131 | /// dead. Add all of its operands to the worklist, turning them into | ||||
132 | /// undef's to reduce the number of uses of those instructions. | ||||
133 | /// | ||||
134 | /// Return the specified operand before it is turned into an undef. | ||||
135 | /// | ||||
136 | Value *AddSoonDeadInstToWorklist(Instruction &I, unsigned op) { | ||||
137 | Value *R = I.getOperand(op); | ||||
138 | |||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 139 | for (User::op_iterator i = I.op_begin(), e = I.op_end(); i != e; ++i) |
140 | if (Instruction *Op = dyn_cast<Instruction>(*i)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 141 | AddToWorkList(Op); |
142 | // Set the operand to undef to drop the use. | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 143 | *i = UndefValue::get(Op->getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 144 | } |
145 | |||||
146 | return R; | ||||
147 | } | ||||
148 | |||||
149 | public: | ||||
150 | virtual bool runOnFunction(Function &F); | ||||
151 | |||||
152 | bool DoOneIteration(Function &F, unsigned ItNum); | ||||
153 | |||||
154 | virtual void getAnalysisUsage(AnalysisUsage &AU) const { | ||||
155 | AU.addRequired<TargetData>(); | ||||
156 | AU.addPreservedID(LCSSAID); | ||||
157 | AU.setPreservesCFG(); | ||||
158 | } | ||||
159 | |||||
160 | TargetData &getTargetData() const { return *TD; } | ||||
161 | |||||
162 | // Visitation implementation - Implement instruction combining for different | ||||
163 | // instruction types. The semantics are as follows: | ||||
164 | // Return Value: | ||||
165 | // null - No change was made | ||||
166 | // I - Change was made, I is still valid, I may be dead though | ||||
167 | // otherwise - Change was made, replace I with returned instruction | ||||
168 | // | ||||
169 | Instruction *visitAdd(BinaryOperator &I); | ||||
170 | Instruction *visitSub(BinaryOperator &I); | ||||
171 | Instruction *visitMul(BinaryOperator &I); | ||||
172 | Instruction *visitURem(BinaryOperator &I); | ||||
173 | Instruction *visitSRem(BinaryOperator &I); | ||||
174 | Instruction *visitFRem(BinaryOperator &I); | ||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 175 | bool SimplifyDivRemOfSelect(BinaryOperator &I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 176 | Instruction *commonRemTransforms(BinaryOperator &I); |
177 | Instruction *commonIRemTransforms(BinaryOperator &I); | ||||
178 | Instruction *commonDivTransforms(BinaryOperator &I); | ||||
179 | Instruction *commonIDivTransforms(BinaryOperator &I); | ||||
180 | Instruction *visitUDiv(BinaryOperator &I); | ||||
181 | Instruction *visitSDiv(BinaryOperator &I); | ||||
182 | Instruction *visitFDiv(BinaryOperator &I); | ||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 183 | Instruction *FoldAndOfICmps(Instruction &I, ICmpInst *LHS, ICmpInst *RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 184 | Instruction *visitAnd(BinaryOperator &I); |
Chris Lattner | 0c678e5 | 2008-11-16 05:20:07 +0000 | [diff] [blame] | 185 | Instruction *FoldOrOfICmps(Instruction &I, ICmpInst *LHS, ICmpInst *RHS); |
Bill Wendling | 9912f71 | 2008-12-01 08:32:40 +0000 | [diff] [blame] | 186 | Instruction *FoldOrWithConstants(BinaryOperator &I, Value *Op, |
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 187 | Value *A, Value *B, Value *C); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 188 | Instruction *visitOr (BinaryOperator &I); |
189 | Instruction *visitXor(BinaryOperator &I); | ||||
190 | Instruction *visitShl(BinaryOperator &I); | ||||
191 | Instruction *visitAShr(BinaryOperator &I); | ||||
192 | Instruction *visitLShr(BinaryOperator &I); | ||||
193 | Instruction *commonShiftTransforms(BinaryOperator &I); | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 194 | Instruction *FoldFCmp_IntToFP_Cst(FCmpInst &I, Instruction *LHSI, |
195 | Constant *RHSC); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 196 | Instruction *visitFCmpInst(FCmpInst &I); |
197 | Instruction *visitICmpInst(ICmpInst &I); | ||||
198 | Instruction *visitICmpInstWithCastAndCast(ICmpInst &ICI); | ||||
199 | Instruction *visitICmpInstWithInstAndIntCst(ICmpInst &ICI, | ||||
200 | Instruction *LHS, | ||||
201 | ConstantInt *RHS); | ||||
202 | Instruction *FoldICmpDivCst(ICmpInst &ICI, BinaryOperator *DivI, | ||||
203 | ConstantInt *DivRHS); | ||||
204 | |||||
205 | Instruction *FoldGEPICmp(User *GEPLHS, Value *RHS, | ||||
206 | ICmpInst::Predicate Cond, Instruction &I); | ||||
207 | Instruction *FoldShiftByConstant(Value *Op0, ConstantInt *Op1, | ||||
208 | BinaryOperator &I); | ||||
209 | Instruction *commonCastTransforms(CastInst &CI); | ||||
210 | Instruction *commonIntCastTransforms(CastInst &CI); | ||||
211 | Instruction *commonPointerCastTransforms(CastInst &CI); | ||||
212 | Instruction *visitTrunc(TruncInst &CI); | ||||
213 | Instruction *visitZExt(ZExtInst &CI); | ||||
214 | Instruction *visitSExt(SExtInst &CI); | ||||
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 215 | Instruction *visitFPTrunc(FPTruncInst &CI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 216 | Instruction *visitFPExt(CastInst &CI); |
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 217 | Instruction *visitFPToUI(FPToUIInst &FI); |
218 | Instruction *visitFPToSI(FPToSIInst &FI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 219 | Instruction *visitUIToFP(CastInst &CI); |
220 | Instruction *visitSIToFP(CastInst &CI); | ||||
221 | Instruction *visitPtrToInt(CastInst &CI); | ||||
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 222 | Instruction *visitIntToPtr(IntToPtrInst &CI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 223 | Instruction *visitBitCast(BitCastInst &CI); |
224 | Instruction *FoldSelectOpOp(SelectInst &SI, Instruction *TI, | ||||
225 | Instruction *FI); | ||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 226 | Instruction *visitSelectInst(SelectInst &SI); |
227 | Instruction *visitSelectInstWithICmp(SelectInst &SI, ICmpInst *ICI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 228 | Instruction *visitCallInst(CallInst &CI); |
229 | Instruction *visitInvokeInst(InvokeInst &II); | ||||
230 | Instruction *visitPHINode(PHINode &PN); | ||||
231 | Instruction *visitGetElementPtrInst(GetElementPtrInst &GEP); | ||||
232 | Instruction *visitAllocationInst(AllocationInst &AI); | ||||
233 | Instruction *visitFreeInst(FreeInst &FI); | ||||
234 | Instruction *visitLoadInst(LoadInst &LI); | ||||
235 | Instruction *visitStoreInst(StoreInst &SI); | ||||
236 | Instruction *visitBranchInst(BranchInst &BI); | ||||
237 | Instruction *visitSwitchInst(SwitchInst &SI); | ||||
238 | Instruction *visitInsertElementInst(InsertElementInst &IE); | ||||
239 | Instruction *visitExtractElementInst(ExtractElementInst &EI); | ||||
240 | Instruction *visitShuffleVectorInst(ShuffleVectorInst &SVI); | ||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 241 | Instruction *visitExtractValueInst(ExtractValueInst &EV); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 242 | |
243 | // visitInstruction - Specify what to return for unhandled instructions... | ||||
244 | Instruction *visitInstruction(Instruction &I) { return 0; } | ||||
245 | |||||
246 | private: | ||||
247 | Instruction *visitCallSite(CallSite CS); | ||||
248 | bool transformConstExprCastCall(CallSite CS); | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 249 | Instruction *transformCallThroughTrampoline(CallSite CS); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 250 | Instruction *transformZExtICmp(ICmpInst *ICI, Instruction &CI, |
251 | bool DoXform = true); | ||||
Chris Lattner | 3554f97 | 2008-05-20 05:46:13 +0000 | [diff] [blame] | 252 | bool WillNotOverflowSignedAdd(Value *LHS, Value *RHS); |
Dale Johannesen | 2c11fe2 | 2009-03-03 21:26:39 +0000 | [diff] [blame^] | 253 | DbgDeclareInst *hasOneUsePlusDeclare(Value *V); |
254 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 255 | |
256 | public: | ||||
257 | // InsertNewInstBefore - insert an instruction New before instruction Old | ||||
258 | // in the program. Add the new instruction to the worklist. | ||||
259 | // | ||||
260 | Instruction *InsertNewInstBefore(Instruction *New, Instruction &Old) { | ||||
261 | assert(New && New->getParent() == 0 && | ||||
262 | "New instruction already inserted into a basic block!"); | ||||
263 | BasicBlock *BB = Old.getParent(); | ||||
264 | BB->getInstList().insert(&Old, New); // Insert inst | ||||
265 | AddToWorkList(New); | ||||
266 | return New; | ||||
267 | } | ||||
268 | |||||
269 | /// InsertCastBefore - Insert a cast of V to TY before the instruction POS. | ||||
270 | /// This also adds the cast to the worklist. Finally, this returns the | ||||
271 | /// cast. | ||||
272 | Value *InsertCastBefore(Instruction::CastOps opc, Value *V, const Type *Ty, | ||||
273 | Instruction &Pos) { | ||||
274 | if (V->getType() == Ty) return V; | ||||
275 | |||||
276 | if (Constant *CV = dyn_cast<Constant>(V)) | ||||
277 | return ConstantExpr::getCast(opc, CV, Ty); | ||||
278 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 279 | Instruction *C = CastInst::Create(opc, V, Ty, V->getName(), &Pos); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 280 | AddToWorkList(C); |
281 | return C; | ||||
282 | } | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 283 | |
284 | Value *InsertBitCastBefore(Value *V, const Type *Ty, Instruction &Pos) { | ||||
285 | return InsertCastBefore(Instruction::BitCast, V, Ty, Pos); | ||||
286 | } | ||||
287 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 288 | |
289 | // ReplaceInstUsesWith - This method is to be used when an instruction is | ||||
290 | // found to be dead, replacable with another preexisting expression. Here | ||||
291 | // we add all uses of I to the worklist, replace all uses of I with the new | ||||
292 | // value, then return I, so that the inst combiner will know that I was | ||||
293 | // modified. | ||||
294 | // | ||||
295 | Instruction *ReplaceInstUsesWith(Instruction &I, Value *V) { | ||||
296 | AddUsersToWorkList(I); // Add all modified instrs to worklist | ||||
297 | if (&I != V) { | ||||
298 | I.replaceAllUsesWith(V); | ||||
299 | return &I; | ||||
300 | } else { | ||||
301 | // If we are replacing the instruction with itself, this must be in a | ||||
302 | // segment of unreachable code, so just clobber the instruction. | ||||
303 | I.replaceAllUsesWith(UndefValue::get(I.getType())); | ||||
304 | return &I; | ||||
305 | } | ||||
306 | } | ||||
307 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 308 | // EraseInstFromFunction - When dealing with an instruction that has side |
309 | // effects or produces a void value, we can't rely on DCE to delete the | ||||
310 | // instruction. Instead, visit methods should return the value returned by | ||||
311 | // this function. | ||||
312 | Instruction *EraseInstFromFunction(Instruction &I) { | ||||
313 | assert(I.use_empty() && "Cannot erase instruction that is used!"); | ||||
314 | AddUsesToWorkList(I); | ||||
315 | RemoveFromWorkList(&I); | ||||
316 | I.eraseFromParent(); | ||||
317 | return 0; // Don't do anything with FI | ||||
318 | } | ||||
Chris Lattner | a432bc7 | 2008-06-02 01:18:21 +0000 | [diff] [blame] | 319 | |
320 | void ComputeMaskedBits(Value *V, const APInt &Mask, APInt &KnownZero, | ||||
321 | APInt &KnownOne, unsigned Depth = 0) const { | ||||
322 | return llvm::ComputeMaskedBits(V, Mask, KnownZero, KnownOne, TD, Depth); | ||||
323 | } | ||||
324 | |||||
325 | bool MaskedValueIsZero(Value *V, const APInt &Mask, | ||||
326 | unsigned Depth = 0) const { | ||||
327 | return llvm::MaskedValueIsZero(V, Mask, TD, Depth); | ||||
328 | } | ||||
329 | unsigned ComputeNumSignBits(Value *Op, unsigned Depth = 0) const { | ||||
330 | return llvm::ComputeNumSignBits(Op, TD, Depth); | ||||
331 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 332 | |
333 | private: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 334 | |
335 | /// SimplifyCommutative - This performs a few simplifications for | ||||
336 | /// commutative operators. | ||||
337 | bool SimplifyCommutative(BinaryOperator &I); | ||||
338 | |||||
339 | /// SimplifyCompare - This reorders the operands of a CmpInst to get them in | ||||
340 | /// most-complex to least-complex order. | ||||
341 | bool SimplifyCompare(CmpInst &I); | ||||
342 | |||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 343 | /// SimplifyDemandedUseBits - Attempts to replace V with a simpler value |
344 | /// based on the demanded bits. | ||||
345 | Value *SimplifyDemandedUseBits(Value *V, APInt DemandedMask, | ||||
346 | APInt& KnownZero, APInt& KnownOne, | ||||
347 | unsigned Depth); | ||||
348 | bool SimplifyDemandedBits(Use &U, APInt DemandedMask, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 349 | APInt& KnownZero, APInt& KnownOne, |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 350 | unsigned Depth=0); |
351 | |||||
352 | /// SimplifyDemandedInstructionBits - Inst is an integer instruction that | ||||
353 | /// SimplifyDemandedBits knows about. See if the instruction has any | ||||
354 | /// properties that allow us to simplify its operands. | ||||
355 | bool SimplifyDemandedInstructionBits(Instruction &Inst); | ||||
356 | |||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 357 | Value *SimplifyDemandedVectorElts(Value *V, APInt DemandedElts, |
358 | APInt& UndefElts, unsigned Depth = 0); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 359 | |
360 | // FoldOpIntoPhi - Given a binary operator or cast instruction which has a | ||||
361 | // PHI node as operand #0, see if we can fold the instruction into the PHI | ||||
362 | // (which is only possible if all operands to the PHI are constants). | ||||
363 | Instruction *FoldOpIntoPhi(Instruction &I); | ||||
364 | |||||
365 | // FoldPHIArgOpIntoPHI - If all operands to a PHI node are the same "unary" | ||||
366 | // operator and they all are only used by the PHI, PHI together their | ||||
367 | // inputs, and do the operation once, to the result of the PHI. | ||||
368 | Instruction *FoldPHIArgOpIntoPHI(PHINode &PN); | ||||
369 | Instruction *FoldPHIArgBinOpIntoPHI(PHINode &PN); | ||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 370 | Instruction *FoldPHIArgGEPIntoPHI(PHINode &PN); |
371 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 372 | |
373 | Instruction *OptAndOp(Instruction *Op, ConstantInt *OpRHS, | ||||
374 | ConstantInt *AndRHS, BinaryOperator &TheAnd); | ||||
375 | |||||
376 | Value *FoldLogicalPlusAnd(Value *LHS, Value *RHS, ConstantInt *Mask, | ||||
377 | bool isSub, Instruction &I); | ||||
378 | Instruction *InsertRangeTest(Value *V, Constant *Lo, Constant *Hi, | ||||
379 | bool isSigned, bool Inside, Instruction &IB); | ||||
380 | Instruction *PromoteCastOfAllocation(BitCastInst &CI, AllocationInst &AI); | ||||
381 | Instruction *MatchBSwap(BinaryOperator &I); | ||||
382 | bool SimplifyStoreAtEndOfBlock(StoreInst &SI); | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 383 | Instruction *SimplifyMemTransfer(MemIntrinsic *MI); |
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 384 | Instruction *SimplifyMemSet(MemSetInst *MI); |
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 385 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 386 | |
387 | Value *EvaluateInDifferentType(Value *V, const Type *Ty, bool isSigned); | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 388 | |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 389 | bool CanEvaluateInDifferentType(Value *V, const IntegerType *Ty, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 390 | unsigned CastOpc, int &NumCastsRemoved); |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 391 | unsigned GetOrEnforceKnownAlignment(Value *V, |
392 | unsigned PrefAlign = 0); | ||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 393 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 394 | }; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 395 | } |
396 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 397 | char InstCombiner::ID = 0; |
398 | static RegisterPass<InstCombiner> | ||||
399 | X("instcombine", "Combine redundant instructions"); | ||||
400 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 401 | // getComplexity: Assign a complexity or rank value to LLVM Values... |
402 | // 0 -> undef, 1 -> Const, 2 -> Other, 3 -> Arg, 3 -> Unary, 4 -> OtherInst | ||||
403 | static unsigned getComplexity(Value *V) { | ||||
404 | if (isa<Instruction>(V)) { | ||||
405 | if (BinaryOperator::isNeg(V) || BinaryOperator::isNot(V)) | ||||
406 | return 3; | ||||
407 | return 4; | ||||
408 | } | ||||
409 | if (isa<Argument>(V)) return 3; | ||||
410 | return isa<Constant>(V) ? (isa<UndefValue>(V) ? 0 : 1) : 2; | ||||
411 | } | ||||
412 | |||||
413 | // isOnlyUse - Return true if this instruction will be deleted if we stop using | ||||
414 | // it. | ||||
415 | static bool isOnlyUse(Value *V) { | ||||
416 | return V->hasOneUse() || isa<Constant>(V); | ||||
417 | } | ||||
418 | |||||
419 | // getPromotedType - Return the specified type promoted as it would be to pass | ||||
420 | // though a va_arg area... | ||||
421 | static const Type *getPromotedType(const Type *Ty) { | ||||
422 | if (const IntegerType* ITy = dyn_cast<IntegerType>(Ty)) { | ||||
423 | if (ITy->getBitWidth() < 32) | ||||
424 | return Type::Int32Ty; | ||||
425 | } | ||||
426 | return Ty; | ||||
427 | } | ||||
428 | |||||
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 429 | /// getBitCastOperand - If the specified operand is a CastInst, a constant |
430 | /// expression bitcast, or a GetElementPtrInst with all zero indices, return the | ||||
431 | /// operand value, otherwise return null. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 432 | static Value *getBitCastOperand(Value *V) { |
433 | if (BitCastInst *I = dyn_cast<BitCastInst>(V)) | ||||
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 434 | // BitCastInst? |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 435 | return I->getOperand(0); |
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 436 | else if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(V)) { |
437 | // GetElementPtrInst? | ||||
438 | if (GEP->hasAllZeroIndices()) | ||||
439 | return GEP->getOperand(0); | ||||
440 | } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(V)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 441 | if (CE->getOpcode() == Instruction::BitCast) |
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 442 | // BitCast ConstantExp? |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 443 | return CE->getOperand(0); |
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 444 | else if (CE->getOpcode() == Instruction::GetElementPtr) { |
445 | // GetElementPtr ConstantExp? | ||||
446 | for (User::op_iterator I = CE->op_begin() + 1, E = CE->op_end(); | ||||
447 | I != E; ++I) { | ||||
448 | ConstantInt *CI = dyn_cast<ConstantInt>(I); | ||||
449 | if (!CI || !CI->isZero()) | ||||
450 | // Any non-zero indices? Not cast-like. | ||||
451 | return 0; | ||||
452 | } | ||||
453 | // All-zero indices? This is just like casting. | ||||
454 | return CE->getOperand(0); | ||||
455 | } | ||||
456 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 457 | return 0; |
458 | } | ||||
459 | |||||
460 | /// This function is a wrapper around CastInst::isEliminableCastPair. It | ||||
461 | /// simply extracts arguments and returns what that function returns. | ||||
462 | static Instruction::CastOps | ||||
463 | isEliminableCastPair( | ||||
464 | const CastInst *CI, ///< The first cast instruction | ||||
465 | unsigned opcode, ///< The opcode of the second cast instruction | ||||
466 | const Type *DstTy, ///< The target type for the second cast instruction | ||||
467 | TargetData *TD ///< The target data for pointer size | ||||
468 | ) { | ||||
469 | |||||
470 | const Type *SrcTy = CI->getOperand(0)->getType(); // A from above | ||||
471 | const Type *MidTy = CI->getType(); // B from above | ||||
472 | |||||
473 | // Get the opcodes of the two Cast instructions | ||||
474 | Instruction::CastOps firstOp = Instruction::CastOps(CI->getOpcode()); | ||||
475 | Instruction::CastOps secondOp = Instruction::CastOps(opcode); | ||||
476 | |||||
477 | return Instruction::CastOps( | ||||
478 | CastInst::isEliminableCastPair(firstOp, secondOp, SrcTy, MidTy, | ||||
479 | DstTy, TD->getIntPtrType())); | ||||
480 | } | ||||
481 | |||||
482 | /// ValueRequiresCast - Return true if the cast from "V to Ty" actually results | ||||
483 | /// in any code being generated. It does not require codegen if V is simple | ||||
484 | /// enough or if the cast can be folded into other casts. | ||||
485 | static bool ValueRequiresCast(Instruction::CastOps opcode, const Value *V, | ||||
486 | const Type *Ty, TargetData *TD) { | ||||
487 | if (V->getType() == Ty || isa<Constant>(V)) return false; | ||||
488 | |||||
489 | // If this is another cast that can be eliminated, it isn't codegen either. | ||||
490 | if (const CastInst *CI = dyn_cast<CastInst>(V)) | ||||
491 | if (isEliminableCastPair(CI, opcode, Ty, TD)) | ||||
492 | return false; | ||||
493 | return true; | ||||
494 | } | ||||
495 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 496 | // SimplifyCommutative - This performs a few simplifications for commutative |
497 | // operators: | ||||
498 | // | ||||
499 | // 1. Order operands such that they are listed from right (least complex) to | ||||
500 | // left (most complex). This puts constants before unary operators before | ||||
501 | // binary operators. | ||||
502 | // | ||||
503 | // 2. Transform: (op (op V, C1), C2) ==> (op V, (op C1, C2)) | ||||
504 | // 3. Transform: (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2)) | ||||
505 | // | ||||
506 | bool InstCombiner::SimplifyCommutative(BinaryOperator &I) { | ||||
507 | bool Changed = false; | ||||
508 | if (getComplexity(I.getOperand(0)) < getComplexity(I.getOperand(1))) | ||||
509 | Changed = !I.swapOperands(); | ||||
510 | |||||
511 | if (!I.isAssociative()) return Changed; | ||||
512 | Instruction::BinaryOps Opcode = I.getOpcode(); | ||||
513 | if (BinaryOperator *Op = dyn_cast<BinaryOperator>(I.getOperand(0))) | ||||
514 | if (Op->getOpcode() == Opcode && isa<Constant>(Op->getOperand(1))) { | ||||
515 | if (isa<Constant>(I.getOperand(1))) { | ||||
516 | Constant *Folded = ConstantExpr::get(I.getOpcode(), | ||||
517 | cast<Constant>(I.getOperand(1)), | ||||
518 | cast<Constant>(Op->getOperand(1))); | ||||
519 | I.setOperand(0, Op->getOperand(0)); | ||||
520 | I.setOperand(1, Folded); | ||||
521 | return true; | ||||
522 | } else if (BinaryOperator *Op1=dyn_cast<BinaryOperator>(I.getOperand(1))) | ||||
523 | if (Op1->getOpcode() == Opcode && isa<Constant>(Op1->getOperand(1)) && | ||||
524 | isOnlyUse(Op) && isOnlyUse(Op1)) { | ||||
525 | Constant *C1 = cast<Constant>(Op->getOperand(1)); | ||||
526 | Constant *C2 = cast<Constant>(Op1->getOperand(1)); | ||||
527 | |||||
528 | // Fold (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2)) | ||||
529 | Constant *Folded = ConstantExpr::get(I.getOpcode(), C1, C2); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 530 | Instruction *New = BinaryOperator::Create(Opcode, Op->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 531 | Op1->getOperand(0), |
532 | Op1->getName(), &I); | ||||
533 | AddToWorkList(New); | ||||
534 | I.setOperand(0, New); | ||||
535 | I.setOperand(1, Folded); | ||||
536 | return true; | ||||
537 | } | ||||
538 | } | ||||
539 | return Changed; | ||||
540 | } | ||||
541 | |||||
542 | /// SimplifyCompare - For a CmpInst this function just orders the operands | ||||
543 | /// so that theyare listed from right (least complex) to left (most complex). | ||||
544 | /// This puts constants before unary operators before binary operators. | ||||
545 | bool InstCombiner::SimplifyCompare(CmpInst &I) { | ||||
546 | if (getComplexity(I.getOperand(0)) >= getComplexity(I.getOperand(1))) | ||||
547 | return false; | ||||
548 | I.swapOperands(); | ||||
549 | // Compare instructions are not associative so there's nothing else we can do. | ||||
550 | return true; | ||||
551 | } | ||||
552 | |||||
553 | // dyn_castNegVal - Given a 'sub' instruction, return the RHS of the instruction | ||||
554 | // if the LHS is a constant zero (which is the 'negate' form). | ||||
555 | // | ||||
556 | static inline Value *dyn_castNegVal(Value *V) { | ||||
557 | if (BinaryOperator::isNeg(V)) | ||||
558 | return BinaryOperator::getNegArgument(V); | ||||
559 | |||||
560 | // Constants can be considered to be negated values if they can be folded. | ||||
561 | if (ConstantInt *C = dyn_cast<ConstantInt>(V)) | ||||
562 | return ConstantExpr::getNeg(C); | ||||
Nick Lewycky | 58867bc | 2008-05-23 04:54:45 +0000 | [diff] [blame] | 563 | |
564 | if (ConstantVector *C = dyn_cast<ConstantVector>(V)) | ||||
565 | if (C->getType()->getElementType()->isInteger()) | ||||
566 | return ConstantExpr::getNeg(C); | ||||
567 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 568 | return 0; |
569 | } | ||||
570 | |||||
571 | static inline Value *dyn_castNotVal(Value *V) { | ||||
572 | if (BinaryOperator::isNot(V)) | ||||
573 | return BinaryOperator::getNotArgument(V); | ||||
574 | |||||
575 | // Constants can be considered to be not'ed values... | ||||
576 | if (ConstantInt *C = dyn_cast<ConstantInt>(V)) | ||||
577 | return ConstantInt::get(~C->getValue()); | ||||
578 | return 0; | ||||
579 | } | ||||
580 | |||||
581 | // dyn_castFoldableMul - If this value is a multiply that can be folded into | ||||
582 | // other computations (because it has a constant operand), return the | ||||
583 | // non-constant operand of the multiply, and set CST to point to the multiplier. | ||||
584 | // Otherwise, return null. | ||||
585 | // | ||||
586 | static inline Value *dyn_castFoldableMul(Value *V, ConstantInt *&CST) { | ||||
587 | if (V->hasOneUse() && V->getType()->isInteger()) | ||||
588 | if (Instruction *I = dyn_cast<Instruction>(V)) { | ||||
589 | if (I->getOpcode() == Instruction::Mul) | ||||
590 | if ((CST = dyn_cast<ConstantInt>(I->getOperand(1)))) | ||||
591 | return I->getOperand(0); | ||||
592 | if (I->getOpcode() == Instruction::Shl) | ||||
593 | if ((CST = dyn_cast<ConstantInt>(I->getOperand(1)))) { | ||||
594 | // The multiplier is really 1 << CST. | ||||
595 | uint32_t BitWidth = cast<IntegerType>(V->getType())->getBitWidth(); | ||||
596 | uint32_t CSTVal = CST->getLimitedValue(BitWidth); | ||||
597 | CST = ConstantInt::get(APInt(BitWidth, 1).shl(CSTVal)); | ||||
598 | return I->getOperand(0); | ||||
599 | } | ||||
600 | } | ||||
601 | return 0; | ||||
602 | } | ||||
603 | |||||
604 | /// dyn_castGetElementPtr - If this is a getelementptr instruction or constant | ||||
605 | /// expression, return it. | ||||
606 | static User *dyn_castGetElementPtr(Value *V) { | ||||
607 | if (isa<GetElementPtrInst>(V)) return cast<User>(V); | ||||
608 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(V)) | ||||
609 | if (CE->getOpcode() == Instruction::GetElementPtr) | ||||
610 | return cast<User>(V); | ||||
611 | return false; | ||||
612 | } | ||||
613 | |||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 614 | /// getOpcode - If this is an Instruction or a ConstantExpr, return the |
615 | /// opcode value. Otherwise return UserOp1. | ||||
Dan Gohman | 8c39786 | 2008-05-29 19:53:46 +0000 | [diff] [blame] | 616 | static unsigned getOpcode(const Value *V) { |
617 | if (const Instruction *I = dyn_cast<Instruction>(V)) | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 618 | return I->getOpcode(); |
Dan Gohman | 8c39786 | 2008-05-29 19:53:46 +0000 | [diff] [blame] | 619 | if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(V)) |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 620 | return CE->getOpcode(); |
621 | // Use UserOp1 to mean there's no opcode. | ||||
622 | return Instruction::UserOp1; | ||||
623 | } | ||||
624 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 625 | /// AddOne - Add one to a ConstantInt |
626 | static ConstantInt *AddOne(ConstantInt *C) { | ||||
627 | APInt Val(C->getValue()); | ||||
628 | return ConstantInt::get(++Val); | ||||
629 | } | ||||
630 | /// SubOne - Subtract one from a ConstantInt | ||||
631 | static ConstantInt *SubOne(ConstantInt *C) { | ||||
632 | APInt Val(C->getValue()); | ||||
633 | return ConstantInt::get(--Val); | ||||
634 | } | ||||
635 | /// Add - Add two ConstantInts together | ||||
636 | static ConstantInt *Add(ConstantInt *C1, ConstantInt *C2) { | ||||
637 | return ConstantInt::get(C1->getValue() + C2->getValue()); | ||||
638 | } | ||||
639 | /// And - Bitwise AND two ConstantInts together | ||||
640 | static ConstantInt *And(ConstantInt *C1, ConstantInt *C2) { | ||||
641 | return ConstantInt::get(C1->getValue() & C2->getValue()); | ||||
642 | } | ||||
643 | /// Subtract - Subtract one ConstantInt from another | ||||
644 | static ConstantInt *Subtract(ConstantInt *C1, ConstantInt *C2) { | ||||
645 | return ConstantInt::get(C1->getValue() - C2->getValue()); | ||||
646 | } | ||||
647 | /// Multiply - Multiply two ConstantInts together | ||||
648 | static ConstantInt *Multiply(ConstantInt *C1, ConstantInt *C2) { | ||||
649 | return ConstantInt::get(C1->getValue() * C2->getValue()); | ||||
650 | } | ||||
Nick Lewycky | 9d798f9 | 2008-02-18 22:48:05 +0000 | [diff] [blame] | 651 | /// MultiplyOverflows - True if the multiply can not be expressed in an int |
652 | /// this size. | ||||
653 | static bool MultiplyOverflows(ConstantInt *C1, ConstantInt *C2, bool sign) { | ||||
654 | uint32_t W = C1->getBitWidth(); | ||||
655 | APInt LHSExt = C1->getValue(), RHSExt = C2->getValue(); | ||||
656 | if (sign) { | ||||
657 | LHSExt.sext(W * 2); | ||||
658 | RHSExt.sext(W * 2); | ||||
659 | } else { | ||||
660 | LHSExt.zext(W * 2); | ||||
661 | RHSExt.zext(W * 2); | ||||
662 | } | ||||
663 | |||||
664 | APInt MulExt = LHSExt * RHSExt; | ||||
665 | |||||
666 | if (sign) { | ||||
667 | APInt Min = APInt::getSignedMinValue(W).sext(W * 2); | ||||
668 | APInt Max = APInt::getSignedMaxValue(W).sext(W * 2); | ||||
669 | return MulExt.slt(Min) || MulExt.sgt(Max); | ||||
670 | } else | ||||
671 | return MulExt.ugt(APInt::getLowBitsSet(W * 2, W)); | ||||
672 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 673 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 674 | |
675 | /// ShrinkDemandedConstant - Check to see if the specified operand of the | ||||
676 | /// specified instruction is a constant integer. If so, check to see if there | ||||
677 | /// are any bits set in the constant that are not demanded. If so, shrink the | ||||
678 | /// constant and return true. | ||||
679 | static bool ShrinkDemandedConstant(Instruction *I, unsigned OpNo, | ||||
680 | APInt Demanded) { | ||||
681 | assert(I && "No instruction?"); | ||||
682 | assert(OpNo < I->getNumOperands() && "Operand index too large"); | ||||
683 | |||||
684 | // If the operand is not a constant integer, nothing to do. | ||||
685 | ConstantInt *OpC = dyn_cast<ConstantInt>(I->getOperand(OpNo)); | ||||
686 | if (!OpC) return false; | ||||
687 | |||||
688 | // If there are no bits set that aren't demanded, nothing to do. | ||||
689 | Demanded.zextOrTrunc(OpC->getValue().getBitWidth()); | ||||
690 | if ((~Demanded & OpC->getValue()) == 0) | ||||
691 | return false; | ||||
692 | |||||
693 | // This instruction is producing bits that are not demanded. Shrink the RHS. | ||||
694 | Demanded &= OpC->getValue(); | ||||
695 | I->setOperand(OpNo, ConstantInt::get(Demanded)); | ||||
696 | return true; | ||||
697 | } | ||||
698 | |||||
699 | // ComputeSignedMinMaxValuesFromKnownBits - Given a signed integer type and a | ||||
700 | // set of known zero and one bits, compute the maximum and minimum values that | ||||
701 | // could have the specified known zero and known one bits, returning them in | ||||
702 | // min/max. | ||||
703 | static void ComputeSignedMinMaxValuesFromKnownBits(const Type *Ty, | ||||
704 | const APInt& KnownZero, | ||||
705 | const APInt& KnownOne, | ||||
706 | APInt& Min, APInt& Max) { | ||||
707 | uint32_t BitWidth = cast<IntegerType>(Ty)->getBitWidth(); | ||||
708 | assert(KnownZero.getBitWidth() == BitWidth && | ||||
709 | KnownOne.getBitWidth() == BitWidth && | ||||
710 | Min.getBitWidth() == BitWidth && Max.getBitWidth() == BitWidth && | ||||
711 | "Ty, KnownZero, KnownOne and Min, Max must have equal bitwidth."); | ||||
712 | APInt UnknownBits = ~(KnownZero|KnownOne); | ||||
713 | |||||
714 | // The minimum value is when all unknown bits are zeros, EXCEPT for the sign | ||||
715 | // bit if it is unknown. | ||||
716 | Min = KnownOne; | ||||
717 | Max = KnownOne|UnknownBits; | ||||
718 | |||||
719 | if (UnknownBits[BitWidth-1]) { // Sign bit is unknown | ||||
720 | Min.set(BitWidth-1); | ||||
721 | Max.clear(BitWidth-1); | ||||
722 | } | ||||
723 | } | ||||
724 | |||||
725 | // ComputeUnsignedMinMaxValuesFromKnownBits - Given an unsigned integer type and | ||||
726 | // a set of known zero and one bits, compute the maximum and minimum values that | ||||
727 | // could have the specified known zero and known one bits, returning them in | ||||
728 | // min/max. | ||||
729 | static void ComputeUnsignedMinMaxValuesFromKnownBits(const Type *Ty, | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 730 | const APInt &KnownZero, |
731 | const APInt &KnownOne, | ||||
732 | APInt &Min, APInt &Max) { | ||||
733 | uint32_t BitWidth = cast<IntegerType>(Ty)->getBitWidth(); BitWidth = BitWidth; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 734 | assert(KnownZero.getBitWidth() == BitWidth && |
735 | KnownOne.getBitWidth() == BitWidth && | ||||
736 | Min.getBitWidth() == BitWidth && Max.getBitWidth() && | ||||
737 | "Ty, KnownZero, KnownOne and Min, Max must have equal bitwidth."); | ||||
738 | APInt UnknownBits = ~(KnownZero|KnownOne); | ||||
739 | |||||
740 | // The minimum value is when the unknown bits are all zeros. | ||||
741 | Min = KnownOne; | ||||
742 | // The maximum value is when the unknown bits are all ones. | ||||
743 | Max = KnownOne|UnknownBits; | ||||
744 | } | ||||
745 | |||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 746 | /// SimplifyDemandedInstructionBits - Inst is an integer instruction that |
747 | /// SimplifyDemandedBits knows about. See if the instruction has any | ||||
748 | /// properties that allow us to simplify its operands. | ||||
749 | bool InstCombiner::SimplifyDemandedInstructionBits(Instruction &Inst) { | ||||
750 | unsigned BitWidth = cast<IntegerType>(Inst.getType())->getBitWidth(); | ||||
751 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
752 | APInt DemandedMask(APInt::getAllOnesValue(BitWidth)); | ||||
753 | |||||
754 | Value *V = SimplifyDemandedUseBits(&Inst, DemandedMask, | ||||
755 | KnownZero, KnownOne, 0); | ||||
756 | if (V == 0) return false; | ||||
757 | if (V == &Inst) return true; | ||||
758 | ReplaceInstUsesWith(Inst, V); | ||||
759 | return true; | ||||
760 | } | ||||
761 | |||||
762 | /// SimplifyDemandedBits - This form of SimplifyDemandedBits simplifies the | ||||
763 | /// specified instruction operand if possible, updating it in place. It returns | ||||
764 | /// true if it made any change and false otherwise. | ||||
765 | bool InstCombiner::SimplifyDemandedBits(Use &U, APInt DemandedMask, | ||||
766 | APInt &KnownZero, APInt &KnownOne, | ||||
767 | unsigned Depth) { | ||||
768 | Value *NewVal = SimplifyDemandedUseBits(U.get(), DemandedMask, | ||||
769 | KnownZero, KnownOne, Depth); | ||||
770 | if (NewVal == 0) return false; | ||||
771 | U.set(NewVal); | ||||
772 | return true; | ||||
773 | } | ||||
774 | |||||
775 | |||||
776 | /// SimplifyDemandedUseBits - This function attempts to replace V with a simpler | ||||
777 | /// value based on the demanded bits. When this function is called, it is known | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 778 | /// that only the bits set in DemandedMask of the result of V are ever used |
779 | /// downstream. Consequently, depending on the mask and V, it may be possible | ||||
780 | /// to replace V with a constant or one of its operands. In such cases, this | ||||
781 | /// function does the replacement and returns true. In all other cases, it | ||||
782 | /// returns false after analyzing the expression and setting KnownOne and known | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 783 | /// to be one in the expression. KnownZero contains all the bits that are known |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 784 | /// to be zero in the expression. These are provided to potentially allow the |
785 | /// caller (which might recursively be SimplifyDemandedBits itself) to simplify | ||||
786 | /// the expression. KnownOne and KnownZero always follow the invariant that | ||||
787 | /// KnownOne & KnownZero == 0. That is, a bit can't be both 1 and 0. Note that | ||||
788 | /// the bits in KnownOne and KnownZero may only be accurate for those bits set | ||||
789 | /// in DemandedMask. Note also that the bitwidth of V, DemandedMask, KnownZero | ||||
790 | /// and KnownOne must all be the same. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 791 | /// |
792 | /// This returns null if it did not change anything and it permits no | ||||
793 | /// simplification. This returns V itself if it did some simplification of V's | ||||
794 | /// operands based on the information about what bits are demanded. This returns | ||||
795 | /// some other non-null value if it found out that V is equal to another value | ||||
796 | /// in the context where the specified bits are demanded, but not for all users. | ||||
797 | Value *InstCombiner::SimplifyDemandedUseBits(Value *V, APInt DemandedMask, | ||||
798 | APInt &KnownZero, APInt &KnownOne, | ||||
799 | unsigned Depth) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 800 | assert(V != 0 && "Null pointer of Value???"); |
801 | assert(Depth <= 6 && "Limit Search Depth"); | ||||
802 | uint32_t BitWidth = DemandedMask.getBitWidth(); | ||||
803 | const IntegerType *VTy = cast<IntegerType>(V->getType()); | ||||
804 | assert(VTy->getBitWidth() == BitWidth && | ||||
805 | KnownZero.getBitWidth() == BitWidth && | ||||
806 | KnownOne.getBitWidth() == BitWidth && | ||||
807 | "Value *V, DemandedMask, KnownZero and KnownOne \ | ||||
808 | must have same BitWidth"); | ||||
809 | if (ConstantInt *CI = dyn_cast<ConstantInt>(V)) { | ||||
810 | // We know all of the bits for a constant! | ||||
811 | KnownOne = CI->getValue() & DemandedMask; | ||||
812 | KnownZero = ~KnownOne & DemandedMask; | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 813 | return 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 814 | } |
815 | |||||
Chris Lattner | c5d7e4e | 2009-01-31 07:26:06 +0000 | [diff] [blame] | 816 | KnownZero.clear(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 817 | KnownOne.clear(); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 818 | if (DemandedMask == 0) { // Not demanding any bits from V. |
819 | if (isa<UndefValue>(V)) | ||||
820 | return 0; | ||||
821 | return UndefValue::get(VTy); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 822 | } |
823 | |||||
Chris Lattner | 0881733 | 2009-01-31 08:24:16 +0000 | [diff] [blame] | 824 | if (Depth == 6) // Limit search depth. |
825 | return 0; | ||||
826 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 827 | Instruction *I = dyn_cast<Instruction>(V); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 828 | if (!I) return 0; // Only analyze instructions. |
Chris Lattner | 0881733 | 2009-01-31 08:24:16 +0000 | [diff] [blame] | 829 | |
Chris Lattner | cd8d44c | 2009-01-31 08:40:03 +0000 | [diff] [blame] | 830 | APInt LHSKnownZero(BitWidth, 0), LHSKnownOne(BitWidth, 0); |
831 | APInt &RHSKnownZero = KnownZero, &RHSKnownOne = KnownOne; | ||||
832 | |||||
Chris Lattner | 0881733 | 2009-01-31 08:24:16 +0000 | [diff] [blame] | 833 | // If there are multiple uses of this value and we aren't at the root, then |
834 | // we can't do any simplifications of the operands, because DemandedMask | ||||
835 | // only reflects the bits demanded by *one* of the users. | ||||
836 | if (Depth != 0 && !I->hasOneUse()) { | ||||
Chris Lattner | cd8d44c | 2009-01-31 08:40:03 +0000 | [diff] [blame] | 837 | // Despite the fact that we can't simplify this instruction in all User's |
838 | // context, we can at least compute the knownzero/knownone bits, and we can | ||||
839 | // do simplifications that apply to *just* the one user if we know that | ||||
840 | // this instruction has a simpler value in that context. | ||||
841 | if (I->getOpcode() == Instruction::And) { | ||||
842 | // If either the LHS or the RHS are Zero, the result is zero. | ||||
843 | ComputeMaskedBits(I->getOperand(1), DemandedMask, | ||||
844 | RHSKnownZero, RHSKnownOne, Depth+1); | ||||
845 | ComputeMaskedBits(I->getOperand(0), DemandedMask & ~RHSKnownZero, | ||||
846 | LHSKnownZero, LHSKnownOne, Depth+1); | ||||
847 | |||||
848 | // If all of the demanded bits are known 1 on one side, return the other. | ||||
849 | // These bits cannot contribute to the result of the 'and' in this | ||||
850 | // context. | ||||
851 | if ((DemandedMask & ~LHSKnownZero & RHSKnownOne) == | ||||
852 | (DemandedMask & ~LHSKnownZero)) | ||||
853 | return I->getOperand(0); | ||||
854 | if ((DemandedMask & ~RHSKnownZero & LHSKnownOne) == | ||||
855 | (DemandedMask & ~RHSKnownZero)) | ||||
856 | return I->getOperand(1); | ||||
857 | |||||
858 | // If all of the demanded bits in the inputs are known zeros, return zero. | ||||
859 | if ((DemandedMask & (RHSKnownZero|LHSKnownZero)) == DemandedMask) | ||||
860 | return Constant::getNullValue(VTy); | ||||
861 | |||||
862 | } else if (I->getOpcode() == Instruction::Or) { | ||||
863 | // We can simplify (X|Y) -> X or Y in the user's context if we know that | ||||
864 | // only bits from X or Y are demanded. | ||||
865 | |||||
866 | // If either the LHS or the RHS are One, the result is One. | ||||
867 | ComputeMaskedBits(I->getOperand(1), DemandedMask, | ||||
868 | RHSKnownZero, RHSKnownOne, Depth+1); | ||||
869 | ComputeMaskedBits(I->getOperand(0), DemandedMask & ~RHSKnownOne, | ||||
870 | LHSKnownZero, LHSKnownOne, Depth+1); | ||||
871 | |||||
872 | // If all of the demanded bits are known zero on one side, return the | ||||
873 | // other. These bits cannot contribute to the result of the 'or' in this | ||||
874 | // context. | ||||
875 | if ((DemandedMask & ~LHSKnownOne & RHSKnownZero) == | ||||
876 | (DemandedMask & ~LHSKnownOne)) | ||||
877 | return I->getOperand(0); | ||||
878 | if ((DemandedMask & ~RHSKnownOne & LHSKnownZero) == | ||||
879 | (DemandedMask & ~RHSKnownOne)) | ||||
880 | return I->getOperand(1); | ||||
881 | |||||
882 | // If all of the potentially set bits on one side are known to be set on | ||||
883 | // the other side, just use the 'other' side. | ||||
884 | if ((DemandedMask & (~RHSKnownZero) & LHSKnownOne) == | ||||
885 | (DemandedMask & (~RHSKnownZero))) | ||||
886 | return I->getOperand(0); | ||||
887 | if ((DemandedMask & (~LHSKnownZero) & RHSKnownOne) == | ||||
888 | (DemandedMask & (~LHSKnownZero))) | ||||
889 | return I->getOperand(1); | ||||
890 | } | ||||
891 | |||||
Chris Lattner | 0881733 | 2009-01-31 08:24:16 +0000 | [diff] [blame] | 892 | // Compute the KnownZero/KnownOne bits to simplify things downstream. |
893 | ComputeMaskedBits(I, DemandedMask, KnownZero, KnownOne, Depth); | ||||
894 | return 0; | ||||
895 | } | ||||
896 | |||||
897 | // If this is the root being simplified, allow it to have multiple uses, | ||||
898 | // just set the DemandedMask to all bits so that we can try to simplify the | ||||
899 | // operands. This allows visitTruncInst (for example) to simplify the | ||||
900 | // operand of a trunc without duplicating all the logic below. | ||||
901 | if (Depth == 0 && !V->hasOneUse()) | ||||
902 | DemandedMask = APInt::getAllOnesValue(BitWidth); | ||||
903 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 904 | switch (I->getOpcode()) { |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 905 | default: |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 906 | ComputeMaskedBits(I, DemandedMask, RHSKnownZero, RHSKnownOne, Depth); |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 907 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 908 | case Instruction::And: |
909 | // If either the LHS or the RHS are Zero, the result is zero. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 910 | if (SimplifyDemandedBits(I->getOperandUse(1), DemandedMask, |
911 | RHSKnownZero, RHSKnownOne, Depth+1) || | ||||
912 | SimplifyDemandedBits(I->getOperandUse(0), DemandedMask & ~RHSKnownZero, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 913 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 914 | return I; |
915 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
916 | assert(!(LHSKnownZero & LHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 917 | |
918 | // If all of the demanded bits are known 1 on one side, return the other. | ||||
919 | // These bits cannot contribute to the result of the 'and'. | ||||
920 | if ((DemandedMask & ~LHSKnownZero & RHSKnownOne) == | ||||
921 | (DemandedMask & ~LHSKnownZero)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 922 | return I->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 923 | if ((DemandedMask & ~RHSKnownZero & LHSKnownOne) == |
924 | (DemandedMask & ~RHSKnownZero)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 925 | return I->getOperand(1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 926 | |
927 | // If all of the demanded bits in the inputs are known zeros, return zero. | ||||
928 | if ((DemandedMask & (RHSKnownZero|LHSKnownZero)) == DemandedMask) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 929 | return Constant::getNullValue(VTy); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 930 | |
931 | // If the RHS is a constant, see if we can simplify it. | ||||
932 | if (ShrinkDemandedConstant(I, 1, DemandedMask & ~LHSKnownZero)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 933 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 934 | |
935 | // Output known-1 bits are only known if set in both the LHS & RHS. | ||||
936 | RHSKnownOne &= LHSKnownOne; | ||||
937 | // Output known-0 are known to be clear if zero in either the LHS | RHS. | ||||
938 | RHSKnownZero |= LHSKnownZero; | ||||
939 | break; | ||||
940 | case Instruction::Or: | ||||
941 | // If either the LHS or the RHS are One, the result is One. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 942 | if (SimplifyDemandedBits(I->getOperandUse(1), DemandedMask, |
943 | RHSKnownZero, RHSKnownOne, Depth+1) || | ||||
944 | SimplifyDemandedBits(I->getOperandUse(0), DemandedMask & ~RHSKnownOne, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 945 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 946 | return I; |
947 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
948 | assert(!(LHSKnownZero & LHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 949 | |
950 | // If all of the demanded bits are known zero on one side, return the other. | ||||
951 | // These bits cannot contribute to the result of the 'or'. | ||||
952 | if ((DemandedMask & ~LHSKnownOne & RHSKnownZero) == | ||||
953 | (DemandedMask & ~LHSKnownOne)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 954 | return I->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 955 | if ((DemandedMask & ~RHSKnownOne & LHSKnownZero) == |
956 | (DemandedMask & ~RHSKnownOne)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 957 | return I->getOperand(1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 958 | |
959 | // If all of the potentially set bits on one side are known to be set on | ||||
960 | // the other side, just use the 'other' side. | ||||
961 | if ((DemandedMask & (~RHSKnownZero) & LHSKnownOne) == | ||||
962 | (DemandedMask & (~RHSKnownZero))) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 963 | return I->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 964 | if ((DemandedMask & (~LHSKnownZero) & RHSKnownOne) == |
965 | (DemandedMask & (~LHSKnownZero))) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 966 | return I->getOperand(1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 967 | |
968 | // If the RHS is a constant, see if we can simplify it. | ||||
969 | if (ShrinkDemandedConstant(I, 1, DemandedMask)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 970 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 971 | |
972 | // Output known-0 bits are only known if clear in both the LHS & RHS. | ||||
973 | RHSKnownZero &= LHSKnownZero; | ||||
974 | // Output known-1 are known to be set if set in either the LHS | RHS. | ||||
975 | RHSKnownOne |= LHSKnownOne; | ||||
976 | break; | ||||
977 | case Instruction::Xor: { | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 978 | if (SimplifyDemandedBits(I->getOperandUse(1), DemandedMask, |
979 | RHSKnownZero, RHSKnownOne, Depth+1) || | ||||
980 | SimplifyDemandedBits(I->getOperandUse(0), DemandedMask, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 981 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 982 | return I; |
983 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
984 | assert(!(LHSKnownZero & LHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 985 | |
986 | // If all of the demanded bits are known zero on one side, return the other. | ||||
987 | // These bits cannot contribute to the result of the 'xor'. | ||||
988 | if ((DemandedMask & RHSKnownZero) == DemandedMask) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 989 | return I->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 990 | if ((DemandedMask & LHSKnownZero) == DemandedMask) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 991 | return I->getOperand(1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 992 | |
993 | // Output known-0 bits are known if clear or set in both the LHS & RHS. | ||||
994 | APInt KnownZeroOut = (RHSKnownZero & LHSKnownZero) | | ||||
995 | (RHSKnownOne & LHSKnownOne); | ||||
996 | // Output known-1 are known to be set if set in only one of the LHS, RHS. | ||||
997 | APInt KnownOneOut = (RHSKnownZero & LHSKnownOne) | | ||||
998 | (RHSKnownOne & LHSKnownZero); | ||||
999 | |||||
1000 | // If all of the demanded bits are known to be zero on one side or the | ||||
1001 | // other, turn this into an *inclusive* or. | ||||
1002 | // e.g. (A & C1)^(B & C2) -> (A & C1)|(B & C2) iff C1&C2 == 0 | ||||
1003 | if ((DemandedMask & ~RHSKnownZero & ~LHSKnownZero) == 0) { | ||||
1004 | Instruction *Or = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1005 | BinaryOperator::CreateOr(I->getOperand(0), I->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1006 | I->getName()); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1007 | return InsertNewInstBefore(Or, *I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1008 | } |
1009 | |||||
1010 | // If all of the demanded bits on one side are known, and all of the set | ||||
1011 | // bits on that side are also known to be set on the other side, turn this | ||||
1012 | // into an AND, as we know the bits will be cleared. | ||||
1013 | // e.g. (X | C1) ^ C2 --> (X | C1) & ~C2 iff (C1&C2) == C2 | ||||
1014 | if ((DemandedMask & (RHSKnownZero|RHSKnownOne)) == DemandedMask) { | ||||
1015 | // all known | ||||
1016 | if ((RHSKnownOne & LHSKnownOne) == RHSKnownOne) { | ||||
1017 | Constant *AndC = ConstantInt::get(~RHSKnownOne & DemandedMask); | ||||
1018 | Instruction *And = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1019 | BinaryOperator::CreateAnd(I->getOperand(0), AndC, "tmp"); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1020 | return InsertNewInstBefore(And, *I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1021 | } |
1022 | } | ||||
1023 | |||||
1024 | // If the RHS is a constant, see if we can simplify it. | ||||
1025 | // FIXME: for XOR, we prefer to force bits to 1 if they will make a -1. | ||||
1026 | if (ShrinkDemandedConstant(I, 1, DemandedMask)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1027 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1028 | |
1029 | RHSKnownZero = KnownZeroOut; | ||||
1030 | RHSKnownOne = KnownOneOut; | ||||
1031 | break; | ||||
1032 | } | ||||
1033 | case Instruction::Select: | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1034 | if (SimplifyDemandedBits(I->getOperandUse(2), DemandedMask, |
1035 | RHSKnownZero, RHSKnownOne, Depth+1) || | ||||
1036 | SimplifyDemandedBits(I->getOperandUse(1), DemandedMask, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1037 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1038 | return I; |
1039 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
1040 | assert(!(LHSKnownZero & LHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1041 | |
1042 | // If the operands are constants, see if we can simplify them. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1043 | if (ShrinkDemandedConstant(I, 1, DemandedMask) || |
1044 | ShrinkDemandedConstant(I, 2, DemandedMask)) | ||||
1045 | return I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1046 | |
1047 | // Only known if known in both the LHS and RHS. | ||||
1048 | RHSKnownOne &= LHSKnownOne; | ||||
1049 | RHSKnownZero &= LHSKnownZero; | ||||
1050 | break; | ||||
1051 | case Instruction::Trunc: { | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1052 | unsigned truncBf = I->getOperand(0)->getType()->getPrimitiveSizeInBits(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1053 | DemandedMask.zext(truncBf); |
1054 | RHSKnownZero.zext(truncBf); | ||||
1055 | RHSKnownOne.zext(truncBf); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1056 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMask, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1057 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1058 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1059 | DemandedMask.trunc(BitWidth); |
1060 | RHSKnownZero.trunc(BitWidth); | ||||
1061 | RHSKnownOne.trunc(BitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1062 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1063 | break; |
1064 | } | ||||
1065 | case Instruction::BitCast: | ||||
1066 | if (!I->getOperand(0)->getType()->isInteger()) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1067 | return false; // vector->int or fp->int? |
1068 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMask, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1069 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1070 | return I; |
1071 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1072 | break; |
1073 | case Instruction::ZExt: { | ||||
1074 | // Compute the bits in the result that are not present in the input. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1075 | unsigned SrcBitWidth =I->getOperand(0)->getType()->getPrimitiveSizeInBits(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1076 | |
1077 | DemandedMask.trunc(SrcBitWidth); | ||||
1078 | RHSKnownZero.trunc(SrcBitWidth); | ||||
1079 | RHSKnownOne.trunc(SrcBitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1080 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMask, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1081 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1082 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1083 | DemandedMask.zext(BitWidth); |
1084 | RHSKnownZero.zext(BitWidth); | ||||
1085 | RHSKnownOne.zext(BitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1086 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1087 | // The top bits are known to be zero. |
1088 | RHSKnownZero |= APInt::getHighBitsSet(BitWidth, BitWidth - SrcBitWidth); | ||||
1089 | break; | ||||
1090 | } | ||||
1091 | case Instruction::SExt: { | ||||
1092 | // Compute the bits in the result that are not present in the input. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1093 | unsigned SrcBitWidth =I->getOperand(0)->getType()->getPrimitiveSizeInBits(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1094 | |
1095 | APInt InputDemandedBits = DemandedMask & | ||||
1096 | APInt::getLowBitsSet(BitWidth, SrcBitWidth); | ||||
1097 | |||||
1098 | APInt NewBits(APInt::getHighBitsSet(BitWidth, BitWidth - SrcBitWidth)); | ||||
1099 | // If any of the sign extended bits are demanded, we know that the sign | ||||
1100 | // bit is demanded. | ||||
1101 | if ((NewBits & DemandedMask) != 0) | ||||
1102 | InputDemandedBits.set(SrcBitWidth-1); | ||||
1103 | |||||
1104 | InputDemandedBits.trunc(SrcBitWidth); | ||||
1105 | RHSKnownZero.trunc(SrcBitWidth); | ||||
1106 | RHSKnownOne.trunc(SrcBitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1107 | if (SimplifyDemandedBits(I->getOperandUse(0), InputDemandedBits, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1108 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1109 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1110 | InputDemandedBits.zext(BitWidth); |
1111 | RHSKnownZero.zext(BitWidth); | ||||
1112 | RHSKnownOne.zext(BitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1113 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1114 | |
1115 | // If the sign bit of the input is known set or clear, then we know the | ||||
1116 | // top bits of the result. | ||||
1117 | |||||
1118 | // If the input sign bit is known zero, or if the NewBits are not demanded | ||||
1119 | // convert this into a zero extension. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1120 | if (RHSKnownZero[SrcBitWidth-1] || (NewBits & ~DemandedMask) == NewBits) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1121 | // Convert to ZExt cast |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1122 | CastInst *NewCast = new ZExtInst(I->getOperand(0), VTy, I->getName()); |
1123 | return InsertNewInstBefore(NewCast, *I); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1124 | } else if (RHSKnownOne[SrcBitWidth-1]) { // Input sign bit known set |
1125 | RHSKnownOne |= NewBits; | ||||
1126 | } | ||||
1127 | break; | ||||
1128 | } | ||||
1129 | case Instruction::Add: { | ||||
1130 | // Figure out what the input bits are. If the top bits of the and result | ||||
1131 | // are not demanded, then the add doesn't demand them from its input | ||||
1132 | // either. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1133 | unsigned NLZ = DemandedMask.countLeadingZeros(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1134 | |
1135 | // If there is a constant on the RHS, there are a variety of xformations | ||||
1136 | // we can do. | ||||
1137 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1138 | // If null, this should be simplified elsewhere. Some of the xforms here | ||||
1139 | // won't work if the RHS is zero. | ||||
1140 | if (RHS->isZero()) | ||||
1141 | break; | ||||
1142 | |||||
1143 | // If the top bit of the output is demanded, demand everything from the | ||||
1144 | // input. Otherwise, we demand all the input bits except NLZ top bits. | ||||
1145 | APInt InDemandedBits(APInt::getLowBitsSet(BitWidth, BitWidth - NLZ)); | ||||
1146 | |||||
1147 | // Find information about known zero/one bits in the input. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1148 | if (SimplifyDemandedBits(I->getOperandUse(0), InDemandedBits, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1149 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1150 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1151 | |
1152 | // If the RHS of the add has bits set that can't affect the input, reduce | ||||
1153 | // the constant. | ||||
1154 | if (ShrinkDemandedConstant(I, 1, InDemandedBits)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1155 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1156 | |
1157 | // Avoid excess work. | ||||
1158 | if (LHSKnownZero == 0 && LHSKnownOne == 0) | ||||
1159 | break; | ||||
1160 | |||||
1161 | // Turn it into OR if input bits are zero. | ||||
1162 | if ((LHSKnownZero & RHS->getValue()) == RHS->getValue()) { | ||||
1163 | Instruction *Or = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1164 | BinaryOperator::CreateOr(I->getOperand(0), I->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1165 | I->getName()); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1166 | return InsertNewInstBefore(Or, *I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1167 | } |
1168 | |||||
1169 | // We can say something about the output known-zero and known-one bits, | ||||
1170 | // depending on potential carries from the input constant and the | ||||
1171 | // unknowns. For example if the LHS is known to have at most the 0x0F0F0 | ||||
1172 | // bits set and the RHS constant is 0x01001, then we know we have a known | ||||
1173 | // one mask of 0x00001 and a known zero mask of 0xE0F0E. | ||||
1174 | |||||
1175 | // To compute this, we first compute the potential carry bits. These are | ||||
1176 | // the bits which may be modified. I'm not aware of a better way to do | ||||
1177 | // this scan. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1178 | const APInt &RHSVal = RHS->getValue(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1179 | APInt CarryBits((~LHSKnownZero + RHSVal) ^ (~LHSKnownZero ^ RHSVal)); |
1180 | |||||
1181 | // Now that we know which bits have carries, compute the known-1/0 sets. | ||||
1182 | |||||
1183 | // Bits are known one if they are known zero in one operand and one in the | ||||
1184 | // other, and there is no input carry. | ||||
1185 | RHSKnownOne = ((LHSKnownZero & RHSVal) | | ||||
1186 | (LHSKnownOne & ~RHSVal)) & ~CarryBits; | ||||
1187 | |||||
1188 | // Bits are known zero if they are known zero in both operands and there | ||||
1189 | // is no input carry. | ||||
1190 | RHSKnownZero = LHSKnownZero & ~RHSVal & ~CarryBits; | ||||
1191 | } else { | ||||
1192 | // If the high-bits of this ADD are not demanded, then it does not demand | ||||
1193 | // the high bits of its LHS or RHS. | ||||
1194 | if (DemandedMask[BitWidth-1] == 0) { | ||||
1195 | // Right fill the mask of bits for this ADD to demand the most | ||||
1196 | // significant bit and all those below it. | ||||
1197 | APInt DemandedFromOps(APInt::getLowBitsSet(BitWidth, BitWidth-NLZ)); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1198 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedFromOps, |
1199 | LHSKnownZero, LHSKnownOne, Depth+1) || | ||||
1200 | SimplifyDemandedBits(I->getOperandUse(1), DemandedFromOps, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1201 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1202 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1203 | } |
1204 | } | ||||
1205 | break; | ||||
1206 | } | ||||
1207 | case Instruction::Sub: | ||||
1208 | // If the high-bits of this SUB are not demanded, then it does not demand | ||||
1209 | // the high bits of its LHS or RHS. | ||||
1210 | if (DemandedMask[BitWidth-1] == 0) { | ||||
1211 | // Right fill the mask of bits for this SUB to demand the most | ||||
1212 | // significant bit and all those below it. | ||||
1213 | uint32_t NLZ = DemandedMask.countLeadingZeros(); | ||||
1214 | APInt DemandedFromOps(APInt::getLowBitsSet(BitWidth, BitWidth-NLZ)); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1215 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedFromOps, |
1216 | LHSKnownZero, LHSKnownOne, Depth+1) || | ||||
1217 | SimplifyDemandedBits(I->getOperandUse(1), DemandedFromOps, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1218 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1219 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1220 | } |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1221 | // Otherwise just hand the sub off to ComputeMaskedBits to fill in |
1222 | // the known zeros and ones. | ||||
1223 | ComputeMaskedBits(V, DemandedMask, RHSKnownZero, RHSKnownOne, Depth); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1224 | break; |
1225 | case Instruction::Shl: | ||||
1226 | if (ConstantInt *SA = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1227 | uint64_t ShiftAmt = SA->getLimitedValue(BitWidth); | ||||
1228 | APInt DemandedMaskIn(DemandedMask.lshr(ShiftAmt)); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1229 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMaskIn, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1230 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1231 | return I; |
1232 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1233 | RHSKnownZero <<= ShiftAmt; |
1234 | RHSKnownOne <<= ShiftAmt; | ||||
1235 | // low bits known zero. | ||||
1236 | if (ShiftAmt) | ||||
1237 | RHSKnownZero |= APInt::getLowBitsSet(BitWidth, ShiftAmt); | ||||
1238 | } | ||||
1239 | break; | ||||
1240 | case Instruction::LShr: | ||||
1241 | // For a logical shift right | ||||
1242 | if (ConstantInt *SA = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1243 | uint64_t ShiftAmt = SA->getLimitedValue(BitWidth); | ||||
1244 | |||||
1245 | // Unsigned shift right. | ||||
1246 | APInt DemandedMaskIn(DemandedMask.shl(ShiftAmt)); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1247 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMaskIn, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1248 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1249 | return I; |
1250 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1251 | RHSKnownZero = APIntOps::lshr(RHSKnownZero, ShiftAmt); |
1252 | RHSKnownOne = APIntOps::lshr(RHSKnownOne, ShiftAmt); | ||||
1253 | if (ShiftAmt) { | ||||
1254 | // Compute the new bits that are at the top now. | ||||
1255 | APInt HighBits(APInt::getHighBitsSet(BitWidth, ShiftAmt)); | ||||
1256 | RHSKnownZero |= HighBits; // high bits known zero. | ||||
1257 | } | ||||
1258 | } | ||||
1259 | break; | ||||
1260 | case Instruction::AShr: | ||||
1261 | // If this is an arithmetic shift right and only the low-bit is set, we can | ||||
1262 | // always convert this into a logical shr, even if the shift amount is | ||||
1263 | // variable. The low bit of the shift cannot be an input sign bit unless | ||||
1264 | // the shift amount is >= the size of the datatype, which is undefined. | ||||
1265 | if (DemandedMask == 1) { | ||||
1266 | // Perform the logical shift right. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1267 | Instruction *NewVal = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1268 | I->getOperand(0), I->getOperand(1), I->getName()); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1269 | return InsertNewInstBefore(NewVal, *I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1270 | } |
1271 | |||||
1272 | // If the sign bit is the only bit demanded by this ashr, then there is no | ||||
1273 | // need to do it, the shift doesn't change the high bit. | ||||
1274 | if (DemandedMask.isSignBit()) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1275 | return I->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1276 | |
1277 | if (ConstantInt *SA = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1278 | uint32_t ShiftAmt = SA->getLimitedValue(BitWidth); | ||||
1279 | |||||
1280 | // Signed shift right. | ||||
1281 | APInt DemandedMaskIn(DemandedMask.shl(ShiftAmt)); | ||||
1282 | // If any of the "high bits" are demanded, we should set the sign bit as | ||||
1283 | // demanded. | ||||
1284 | if (DemandedMask.countLeadingZeros() <= ShiftAmt) | ||||
1285 | DemandedMaskIn.set(BitWidth-1); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1286 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMaskIn, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1287 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1288 | return I; |
1289 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1290 | // Compute the new bits that are at the top now. |
1291 | APInt HighBits(APInt::getHighBitsSet(BitWidth, ShiftAmt)); | ||||
1292 | RHSKnownZero = APIntOps::lshr(RHSKnownZero, ShiftAmt); | ||||
1293 | RHSKnownOne = APIntOps::lshr(RHSKnownOne, ShiftAmt); | ||||
1294 | |||||
1295 | // Handle the sign bits. | ||||
1296 | APInt SignBit(APInt::getSignBit(BitWidth)); | ||||
1297 | // Adjust to where it is now in the mask. | ||||
1298 | SignBit = APIntOps::lshr(SignBit, ShiftAmt); | ||||
1299 | |||||
1300 | // If the input sign bit is known to be zero, or if none of the top bits | ||||
1301 | // are demanded, turn this into an unsigned shift right. | ||||
Zhou Sheng | 533604e | 2008-06-06 08:32:05 +0000 | [diff] [blame] | 1302 | if (BitWidth <= ShiftAmt || RHSKnownZero[BitWidth-ShiftAmt-1] || |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1303 | (HighBits & ~DemandedMask) == HighBits) { |
1304 | // Perform the logical shift right. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1305 | Instruction *NewVal = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1306 | I->getOperand(0), SA, I->getName()); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1307 | return InsertNewInstBefore(NewVal, *I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1308 | } else if ((RHSKnownOne & SignBit) != 0) { // New bits are known one. |
1309 | RHSKnownOne |= HighBits; | ||||
1310 | } | ||||
1311 | } | ||||
1312 | break; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1313 | case Instruction::SRem: |
1314 | if (ConstantInt *Rem = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
Nick Lewycky | cfaaece | 2008-11-02 02:41:50 +0000 | [diff] [blame] | 1315 | APInt RA = Rem->getValue().abs(); |
1316 | if (RA.isPowerOf2()) { | ||||
Nick Lewycky | 245de42 | 2008-07-12 05:04:38 +0000 | [diff] [blame] | 1317 | if (DemandedMask.ule(RA)) // srem won't affect demanded bits |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1318 | return I->getOperand(0); |
Nick Lewycky | 245de42 | 2008-07-12 05:04:38 +0000 | [diff] [blame] | 1319 | |
Nick Lewycky | cfaaece | 2008-11-02 02:41:50 +0000 | [diff] [blame] | 1320 | APInt LowBits = RA - 1; |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1321 | APInt Mask2 = LowBits | APInt::getSignBit(BitWidth); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1322 | if (SimplifyDemandedBits(I->getOperandUse(0), Mask2, |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1323 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1324 | return I; |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1325 | |
1326 | if (LHSKnownZero[BitWidth-1] || ((LHSKnownZero & LowBits) == LowBits)) | ||||
1327 | LHSKnownZero |= ~LowBits; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1328 | |
1329 | KnownZero |= LHSKnownZero & DemandedMask; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1330 | |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1331 | assert(!(KnownZero & KnownOne) && "Bits known to be one AND zero?"); |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1332 | } |
1333 | } | ||||
1334 | break; | ||||
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1335 | case Instruction::URem: { |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1336 | APInt KnownZero2(BitWidth, 0), KnownOne2(BitWidth, 0); |
1337 | APInt AllOnes = APInt::getAllOnesValue(BitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1338 | if (SimplifyDemandedBits(I->getOperandUse(0), AllOnes, |
1339 | KnownZero2, KnownOne2, Depth+1) || | ||||
1340 | SimplifyDemandedBits(I->getOperandUse(1), AllOnes, | ||||
Dan Gohman | 23ea06d | 2008-05-01 19:13:24 +0000 | [diff] [blame] | 1341 | KnownZero2, KnownOne2, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1342 | return I; |
Dan Gohman | 23ea06d | 2008-05-01 19:13:24 +0000 | [diff] [blame] | 1343 | |
Chris Lattner | ee5417c | 2009-01-21 18:09:24 +0000 | [diff] [blame] | 1344 | unsigned Leaders = KnownZero2.countLeadingOnes(); |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1345 | Leaders = std::max(Leaders, |
1346 | KnownZero2.countLeadingOnes()); | ||||
1347 | KnownZero = APInt::getHighBitsSet(BitWidth, Leaders) & DemandedMask; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1348 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1349 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 1350 | case Instruction::Call: |
1351 | if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(I)) { | ||||
1352 | switch (II->getIntrinsicID()) { | ||||
1353 | default: break; | ||||
1354 | case Intrinsic::bswap: { | ||||
1355 | // If the only bits demanded come from one byte of the bswap result, | ||||
1356 | // just shift the input byte into position to eliminate the bswap. | ||||
1357 | unsigned NLZ = DemandedMask.countLeadingZeros(); | ||||
1358 | unsigned NTZ = DemandedMask.countTrailingZeros(); | ||||
1359 | |||||
1360 | // Round NTZ down to the next byte. If we have 11 trailing zeros, then | ||||
1361 | // we need all the bits down to bit 8. Likewise, round NLZ. If we | ||||
1362 | // have 14 leading zeros, round to 8. | ||||
1363 | NLZ &= ~7; | ||||
1364 | NTZ &= ~7; | ||||
1365 | // If we need exactly one byte, we can do this transformation. | ||||
1366 | if (BitWidth-NLZ-NTZ == 8) { | ||||
1367 | unsigned ResultBit = NTZ; | ||||
1368 | unsigned InputBit = BitWidth-NTZ-8; | ||||
1369 | |||||
1370 | // Replace this with either a left or right shift to get the byte into | ||||
1371 | // the right place. | ||||
1372 | Instruction *NewVal; | ||||
1373 | if (InputBit > ResultBit) | ||||
1374 | NewVal = BinaryOperator::CreateLShr(I->getOperand(1), | ||||
1375 | ConstantInt::get(I->getType(), InputBit-ResultBit)); | ||||
1376 | else | ||||
1377 | NewVal = BinaryOperator::CreateShl(I->getOperand(1), | ||||
1378 | ConstantInt::get(I->getType(), ResultBit-InputBit)); | ||||
1379 | NewVal->takeName(I); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1380 | return InsertNewInstBefore(NewVal, *I); |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 1381 | } |
1382 | |||||
1383 | // TODO: Could compute known zero/one bits based on the input. | ||||
1384 | break; | ||||
1385 | } | ||||
1386 | } | ||||
1387 | } | ||||
Chris Lattner | 4946e22 | 2008-06-18 18:11:55 +0000 | [diff] [blame] | 1388 | ComputeMaskedBits(V, DemandedMask, RHSKnownZero, RHSKnownOne, Depth); |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 1389 | break; |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1390 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1391 | |
1392 | // If the client is only demanding bits that we know, return the known | ||||
1393 | // constant. | ||||
1394 | if ((DemandedMask & (RHSKnownZero|RHSKnownOne)) == DemandedMask) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1395 | return ConstantInt::get(RHSKnownOne); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1396 | return false; |
1397 | } | ||||
1398 | |||||
1399 | |||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1400 | /// SimplifyDemandedVectorElts - The specified value produces a vector with |
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1401 | /// any number of elements. DemandedElts contains the set of elements that are |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1402 | /// actually used by the caller. This method analyzes which elements of the |
1403 | /// operand are undef and returns that information in UndefElts. | ||||
1404 | /// | ||||
1405 | /// If the information about demanded elements can be used to simplify the | ||||
1406 | /// operation, the operation is simplified, then the resultant value is | ||||
1407 | /// returned. This returns null if no change was made. | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1408 | Value *InstCombiner::SimplifyDemandedVectorElts(Value *V, APInt DemandedElts, |
1409 | APInt& UndefElts, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1410 | unsigned Depth) { |
1411 | unsigned VWidth = cast<VectorType>(V->getType())->getNumElements(); | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1412 | APInt EltMask(APInt::getAllOnesValue(VWidth)); |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1413 | assert((DemandedElts & ~EltMask) == 0 && "Invalid DemandedElts!"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1414 | |
1415 | if (isa<UndefValue>(V)) { | ||||
1416 | // If the entire vector is undefined, just return this info. | ||||
1417 | UndefElts = EltMask; | ||||
1418 | return 0; | ||||
1419 | } else if (DemandedElts == 0) { // If nothing is demanded, provide undef. | ||||
1420 | UndefElts = EltMask; | ||||
1421 | return UndefValue::get(V->getType()); | ||||
1422 | } | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1423 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1424 | UndefElts = 0; |
1425 | if (ConstantVector *CP = dyn_cast<ConstantVector>(V)) { | ||||
1426 | const Type *EltTy = cast<VectorType>(V->getType())->getElementType(); | ||||
1427 | Constant *Undef = UndefValue::get(EltTy); | ||||
1428 | |||||
1429 | std::vector<Constant*> Elts; | ||||
1430 | for (unsigned i = 0; i != VWidth; ++i) | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1431 | if (!DemandedElts[i]) { // If not demanded, set to undef. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1432 | Elts.push_back(Undef); |
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1433 | UndefElts.set(i); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1434 | } else if (isa<UndefValue>(CP->getOperand(i))) { // Already undef. |
1435 | Elts.push_back(Undef); | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1436 | UndefElts.set(i); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1437 | } else { // Otherwise, defined. |
1438 | Elts.push_back(CP->getOperand(i)); | ||||
1439 | } | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1440 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1441 | // If we changed the constant, return it. |
1442 | Constant *NewCP = ConstantVector::get(Elts); | ||||
1443 | return NewCP != CP ? NewCP : 0; | ||||
1444 | } else if (isa<ConstantAggregateZero>(V)) { | ||||
1445 | // Simplify the CAZ to a ConstantVector where the non-demanded elements are | ||||
1446 | // set to undef. | ||||
Mon P Wang | 927daf5 | 2008-11-06 22:52:21 +0000 | [diff] [blame] | 1447 | |
1448 | // Check if this is identity. If so, return 0 since we are not simplifying | ||||
1449 | // anything. | ||||
1450 | if (DemandedElts == ((1ULL << VWidth) -1)) | ||||
1451 | return 0; | ||||
1452 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1453 | const Type *EltTy = cast<VectorType>(V->getType())->getElementType(); |
1454 | Constant *Zero = Constant::getNullValue(EltTy); | ||||
1455 | Constant *Undef = UndefValue::get(EltTy); | ||||
1456 | std::vector<Constant*> Elts; | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1457 | for (unsigned i = 0; i != VWidth; ++i) { |
1458 | Constant *Elt = DemandedElts[i] ? Zero : Undef; | ||||
1459 | Elts.push_back(Elt); | ||||
1460 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1461 | UndefElts = DemandedElts ^ EltMask; |
1462 | return ConstantVector::get(Elts); | ||||
1463 | } | ||||
1464 | |||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1465 | // Limit search depth. |
1466 | if (Depth == 10) | ||||
1467 | return false; | ||||
1468 | |||||
1469 | // If multiple users are using the root value, procede with | ||||
1470 | // simplification conservatively assuming that all elements | ||||
1471 | // are needed. | ||||
1472 | if (!V->hasOneUse()) { | ||||
1473 | // Quit if we find multiple users of a non-root value though. | ||||
1474 | // They'll be handled when it's their turn to be visited by | ||||
1475 | // the main instcombine process. | ||||
1476 | if (Depth != 0) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1477 | // TODO: Just compute the UndefElts information recursively. |
1478 | return false; | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1479 | |
1480 | // Conservatively assume that all elements are needed. | ||||
1481 | DemandedElts = EltMask; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1482 | } |
1483 | |||||
1484 | Instruction *I = dyn_cast<Instruction>(V); | ||||
1485 | if (!I) return false; // Only analyze instructions. | ||||
1486 | |||||
1487 | bool MadeChange = false; | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1488 | APInt UndefElts2(VWidth, 0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1489 | Value *TmpV; |
1490 | switch (I->getOpcode()) { | ||||
1491 | default: break; | ||||
1492 | |||||
1493 | case Instruction::InsertElement: { | ||||
1494 | // If this is a variable index, we don't know which element it overwrites. | ||||
1495 | // demand exactly the same input as we produce. | ||||
1496 | ConstantInt *Idx = dyn_cast<ConstantInt>(I->getOperand(2)); | ||||
1497 | if (Idx == 0) { | ||||
1498 | // Note that we can't propagate undef elt info, because we don't know | ||||
1499 | // which elt is getting updated. | ||||
1500 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts, | ||||
1501 | UndefElts2, Depth+1); | ||||
1502 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1503 | break; | ||||
1504 | } | ||||
1505 | |||||
1506 | // If this is inserting an element that isn't demanded, remove this | ||||
1507 | // insertelement. | ||||
1508 | unsigned IdxNo = Idx->getZExtValue(); | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1509 | if (IdxNo >= VWidth || !DemandedElts[IdxNo]) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1510 | return AddSoonDeadInstToWorklist(*I, 0); |
1511 | |||||
1512 | // Otherwise, the element inserted overwrites whatever was there, so the | ||||
1513 | // input demanded set is simpler than the output set. | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1514 | APInt DemandedElts2 = DemandedElts; |
1515 | DemandedElts2.clear(IdxNo); | ||||
1516 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts2, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1517 | UndefElts, Depth+1); |
1518 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1519 | |||||
1520 | // The inserted element is defined. | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1521 | UndefElts.clear(IdxNo); |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1522 | break; |
1523 | } | ||||
1524 | case Instruction::ShuffleVector: { | ||||
1525 | ShuffleVectorInst *Shuffle = cast<ShuffleVectorInst>(I); | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1526 | uint64_t LHSVWidth = |
1527 | cast<VectorType>(Shuffle->getOperand(0)->getType())->getNumElements(); | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1528 | APInt LeftDemanded(LHSVWidth, 0), RightDemanded(LHSVWidth, 0); |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1529 | for (unsigned i = 0; i < VWidth; i++) { |
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1530 | if (DemandedElts[i]) { |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1531 | unsigned MaskVal = Shuffle->getMaskValue(i); |
1532 | if (MaskVal != -1u) { | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1533 | assert(MaskVal < LHSVWidth * 2 && |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1534 | "shufflevector mask index out of range!"); |
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1535 | if (MaskVal < LHSVWidth) |
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1536 | LeftDemanded.set(MaskVal); |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1537 | else |
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1538 | RightDemanded.set(MaskVal - LHSVWidth); |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1539 | } |
1540 | } | ||||
1541 | } | ||||
1542 | |||||
Nate Begeman | b4d176f | 2009-02-11 22:36:25 +0000 | [diff] [blame] | 1543 | APInt UndefElts4(LHSVWidth, 0); |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1544 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), LeftDemanded, |
Nate Begeman | b4d176f | 2009-02-11 22:36:25 +0000 | [diff] [blame] | 1545 | UndefElts4, Depth+1); |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1546 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } |
1547 | |||||
Nate Begeman | b4d176f | 2009-02-11 22:36:25 +0000 | [diff] [blame] | 1548 | APInt UndefElts3(LHSVWidth, 0); |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1549 | TmpV = SimplifyDemandedVectorElts(I->getOperand(1), RightDemanded, |
1550 | UndefElts3, Depth+1); | ||||
1551 | if (TmpV) { I->setOperand(1, TmpV); MadeChange = true; } | ||||
1552 | |||||
1553 | bool NewUndefElts = false; | ||||
1554 | for (unsigned i = 0; i < VWidth; i++) { | ||||
1555 | unsigned MaskVal = Shuffle->getMaskValue(i); | ||||
Dan Gohman | 24f6ee2 | 2008-09-10 01:09:32 +0000 | [diff] [blame] | 1556 | if (MaskVal == -1u) { |
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1557 | UndefElts.set(i); |
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1558 | } else if (MaskVal < LHSVWidth) { |
Nate Begeman | b4d176f | 2009-02-11 22:36:25 +0000 | [diff] [blame] | 1559 | if (UndefElts4[MaskVal]) { |
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1560 | NewUndefElts = true; |
1561 | UndefElts.set(i); | ||||
1562 | } | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1563 | } else { |
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1564 | if (UndefElts3[MaskVal - LHSVWidth]) { |
1565 | NewUndefElts = true; | ||||
1566 | UndefElts.set(i); | ||||
1567 | } | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1568 | } |
1569 | } | ||||
1570 | |||||
1571 | if (NewUndefElts) { | ||||
1572 | // Add additional discovered undefs. | ||||
1573 | std::vector<Constant*> Elts; | ||||
1574 | for (unsigned i = 0; i < VWidth; ++i) { | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1575 | if (UndefElts[i]) |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1576 | Elts.push_back(UndefValue::get(Type::Int32Ty)); |
1577 | else | ||||
1578 | Elts.push_back(ConstantInt::get(Type::Int32Ty, | ||||
1579 | Shuffle->getMaskValue(i))); | ||||
1580 | } | ||||
1581 | I->setOperand(2, ConstantVector::get(Elts)); | ||||
1582 | MadeChange = true; | ||||
1583 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1584 | break; |
1585 | } | ||||
1586 | case Instruction::BitCast: { | ||||
1587 | // Vector->vector casts only. | ||||
1588 | const VectorType *VTy = dyn_cast<VectorType>(I->getOperand(0)->getType()); | ||||
1589 | if (!VTy) break; | ||||
1590 | unsigned InVWidth = VTy->getNumElements(); | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1591 | APInt InputDemandedElts(InVWidth, 0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1592 | unsigned Ratio; |
1593 | |||||
1594 | if (VWidth == InVWidth) { | ||||
1595 | // If we are converting from <4 x i32> -> <4 x f32>, we demand the same | ||||
1596 | // elements as are demanded of us. | ||||
1597 | Ratio = 1; | ||||
1598 | InputDemandedElts = DemandedElts; | ||||
1599 | } else if (VWidth > InVWidth) { | ||||
1600 | // Untested so far. | ||||
1601 | break; | ||||
1602 | |||||
1603 | // If there are more elements in the result than there are in the source, | ||||
1604 | // then an input element is live if any of the corresponding output | ||||
1605 | // elements are live. | ||||
1606 | Ratio = VWidth/InVWidth; | ||||
1607 | for (unsigned OutIdx = 0; OutIdx != VWidth; ++OutIdx) { | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1608 | if (DemandedElts[OutIdx]) |
1609 | InputDemandedElts.set(OutIdx/Ratio); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1610 | } |
1611 | } else { | ||||
1612 | // Untested so far. | ||||
1613 | break; | ||||
1614 | |||||
1615 | // If there are more elements in the source than there are in the result, | ||||
1616 | // then an input element is live if the corresponding output element is | ||||
1617 | // live. | ||||
1618 | Ratio = InVWidth/VWidth; | ||||
1619 | for (unsigned InIdx = 0; InIdx != InVWidth; ++InIdx) | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1620 | if (DemandedElts[InIdx/Ratio]) |
1621 | InputDemandedElts.set(InIdx); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1622 | } |
1623 | |||||
1624 | // div/rem demand all inputs, because they don't want divide by zero. | ||||
1625 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), InputDemandedElts, | ||||
1626 | UndefElts2, Depth+1); | ||||
1627 | if (TmpV) { | ||||
1628 | I->setOperand(0, TmpV); | ||||
1629 | MadeChange = true; | ||||
1630 | } | ||||
1631 | |||||
1632 | UndefElts = UndefElts2; | ||||
1633 | if (VWidth > InVWidth) { | ||||
1634 | assert(0 && "Unimp"); | ||||
1635 | // If there are more elements in the result than there are in the source, | ||||
1636 | // then an output element is undef if the corresponding input element is | ||||
1637 | // undef. | ||||
1638 | for (unsigned OutIdx = 0; OutIdx != VWidth; ++OutIdx) | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1639 | if (UndefElts2[OutIdx/Ratio]) |
1640 | UndefElts.set(OutIdx); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1641 | } else if (VWidth < InVWidth) { |
1642 | assert(0 && "Unimp"); | ||||
1643 | // If there are more elements in the source than there are in the result, | ||||
1644 | // then a result element is undef if all of the corresponding input | ||||
1645 | // elements are undef. | ||||
1646 | UndefElts = ~0ULL >> (64-VWidth); // Start out all undef. | ||||
1647 | for (unsigned InIdx = 0; InIdx != InVWidth; ++InIdx) | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 1648 | if (!UndefElts2[InIdx]) // Not undef? |
1649 | UndefElts.clear(InIdx/Ratio); // Clear undef bit. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1650 | } |
1651 | break; | ||||
1652 | } | ||||
1653 | case Instruction::And: | ||||
1654 | case Instruction::Or: | ||||
1655 | case Instruction::Xor: | ||||
1656 | case Instruction::Add: | ||||
1657 | case Instruction::Sub: | ||||
1658 | case Instruction::Mul: | ||||
1659 | // div/rem demand all inputs, because they don't want divide by zero. | ||||
1660 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts, | ||||
1661 | UndefElts, Depth+1); | ||||
1662 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1663 | TmpV = SimplifyDemandedVectorElts(I->getOperand(1), DemandedElts, | ||||
1664 | UndefElts2, Depth+1); | ||||
1665 | if (TmpV) { I->setOperand(1, TmpV); MadeChange = true; } | ||||
1666 | |||||
1667 | // Output elements are undefined if both are undefined. Consider things | ||||
1668 | // like undef&0. The result is known zero, not undef. | ||||
1669 | UndefElts &= UndefElts2; | ||||
1670 | break; | ||||
1671 | |||||
1672 | case Instruction::Call: { | ||||
1673 | IntrinsicInst *II = dyn_cast<IntrinsicInst>(I); | ||||
1674 | if (!II) break; | ||||
1675 | switch (II->getIntrinsicID()) { | ||||
1676 | default: break; | ||||
1677 | |||||
1678 | // Binary vector operations that work column-wise. A dest element is a | ||||
1679 | // function of the corresponding input elements from the two inputs. | ||||
1680 | case Intrinsic::x86_sse_sub_ss: | ||||
1681 | case Intrinsic::x86_sse_mul_ss: | ||||
1682 | case Intrinsic::x86_sse_min_ss: | ||||
1683 | case Intrinsic::x86_sse_max_ss: | ||||
1684 | case Intrinsic::x86_sse2_sub_sd: | ||||
1685 | case Intrinsic::x86_sse2_mul_sd: | ||||
1686 | case Intrinsic::x86_sse2_min_sd: | ||||
1687 | case Intrinsic::x86_sse2_max_sd: | ||||
1688 | TmpV = SimplifyDemandedVectorElts(II->getOperand(1), DemandedElts, | ||||
1689 | UndefElts, Depth+1); | ||||
1690 | if (TmpV) { II->setOperand(1, TmpV); MadeChange = true; } | ||||
1691 | TmpV = SimplifyDemandedVectorElts(II->getOperand(2), DemandedElts, | ||||
1692 | UndefElts2, Depth+1); | ||||
1693 | if (TmpV) { II->setOperand(2, TmpV); MadeChange = true; } | ||||
1694 | |||||
1695 | // If only the low elt is demanded and this is a scalarizable intrinsic, | ||||
1696 | // scalarize it now. | ||||
1697 | if (DemandedElts == 1) { | ||||
1698 | switch (II->getIntrinsicID()) { | ||||
1699 | default: break; | ||||
1700 | case Intrinsic::x86_sse_sub_ss: | ||||
1701 | case Intrinsic::x86_sse_mul_ss: | ||||
1702 | case Intrinsic::x86_sse2_sub_sd: | ||||
1703 | case Intrinsic::x86_sse2_mul_sd: | ||||
1704 | // TODO: Lower MIN/MAX/ABS/etc | ||||
1705 | Value *LHS = II->getOperand(1); | ||||
1706 | Value *RHS = II->getOperand(2); | ||||
1707 | // Extract the element as scalars. | ||||
1708 | LHS = InsertNewInstBefore(new ExtractElementInst(LHS, 0U,"tmp"), *II); | ||||
1709 | RHS = InsertNewInstBefore(new ExtractElementInst(RHS, 0U,"tmp"), *II); | ||||
1710 | |||||
1711 | switch (II->getIntrinsicID()) { | ||||
1712 | default: assert(0 && "Case stmts out of sync!"); | ||||
1713 | case Intrinsic::x86_sse_sub_ss: | ||||
1714 | case Intrinsic::x86_sse2_sub_sd: | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1715 | TmpV = InsertNewInstBefore(BinaryOperator::CreateSub(LHS, RHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1716 | II->getName()), *II); |
1717 | break; | ||||
1718 | case Intrinsic::x86_sse_mul_ss: | ||||
1719 | case Intrinsic::x86_sse2_mul_sd: | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1720 | TmpV = InsertNewInstBefore(BinaryOperator::CreateMul(LHS, RHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1721 | II->getName()), *II); |
1722 | break; | ||||
1723 | } | ||||
1724 | |||||
1725 | Instruction *New = | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 1726 | InsertElementInst::Create(UndefValue::get(II->getType()), TmpV, 0U, |
1727 | II->getName()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1728 | InsertNewInstBefore(New, *II); |
1729 | AddSoonDeadInstToWorklist(*II, 0); | ||||
1730 | return New; | ||||
1731 | } | ||||
1732 | } | ||||
1733 | |||||
1734 | // Output elements are undefined if both are undefined. Consider things | ||||
1735 | // like undef&0. The result is known zero, not undef. | ||||
1736 | UndefElts &= UndefElts2; | ||||
1737 | break; | ||||
1738 | } | ||||
1739 | break; | ||||
1740 | } | ||||
1741 | } | ||||
1742 | return MadeChange ? I : 0; | ||||
1743 | } | ||||
1744 | |||||
Dan Gohman | 5d56fd4 | 2008-05-19 22:14:15 +0000 | [diff] [blame] | 1745 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1746 | /// AssociativeOpt - Perform an optimization on an associative operator. This |
1747 | /// function is designed to check a chain of associative operators for a | ||||
1748 | /// potential to apply a certain optimization. Since the optimization may be | ||||
1749 | /// applicable if the expression was reassociated, this checks the chain, then | ||||
1750 | /// reassociates the expression as necessary to expose the optimization | ||||
1751 | /// opportunity. This makes use of a special Functor, which must define | ||||
1752 | /// 'shouldApply' and 'apply' methods. | ||||
1753 | /// | ||||
1754 | template<typename Functor> | ||||
Dan Gohman | d8bcf5b | 2008-05-20 01:14:05 +0000 | [diff] [blame] | 1755 | static Instruction *AssociativeOpt(BinaryOperator &Root, const Functor &F) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1756 | unsigned Opcode = Root.getOpcode(); |
1757 | Value *LHS = Root.getOperand(0); | ||||
1758 | |||||
1759 | // Quick check, see if the immediate LHS matches... | ||||
1760 | if (F.shouldApply(LHS)) | ||||
1761 | return F.apply(Root); | ||||
1762 | |||||
1763 | // Otherwise, if the LHS is not of the same opcode as the root, return. | ||||
1764 | Instruction *LHSI = dyn_cast<Instruction>(LHS); | ||||
1765 | while (LHSI && LHSI->getOpcode() == Opcode && LHSI->hasOneUse()) { | ||||
1766 | // Should we apply this transform to the RHS? | ||||
1767 | bool ShouldApply = F.shouldApply(LHSI->getOperand(1)); | ||||
1768 | |||||
1769 | // If not to the RHS, check to see if we should apply to the LHS... | ||||
1770 | if (!ShouldApply && F.shouldApply(LHSI->getOperand(0))) { | ||||
1771 | cast<BinaryOperator>(LHSI)->swapOperands(); // Make the LHS the RHS | ||||
1772 | ShouldApply = true; | ||||
1773 | } | ||||
1774 | |||||
1775 | // If the functor wants to apply the optimization to the RHS of LHSI, | ||||
1776 | // reassociate the expression from ((? op A) op B) to (? op (A op B)) | ||||
1777 | if (ShouldApply) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1778 | // Now all of the instructions are in the current basic block, go ahead |
1779 | // and perform the reassociation. | ||||
1780 | Instruction *TmpLHSI = cast<Instruction>(Root.getOperand(0)); | ||||
1781 | |||||
1782 | // First move the selected RHS to the LHS of the root... | ||||
1783 | Root.setOperand(0, LHSI->getOperand(1)); | ||||
1784 | |||||
1785 | // Make what used to be the LHS of the root be the user of the root... | ||||
1786 | Value *ExtraOperand = TmpLHSI->getOperand(1); | ||||
1787 | if (&Root == TmpLHSI) { | ||||
1788 | Root.replaceAllUsesWith(Constant::getNullValue(TmpLHSI->getType())); | ||||
1789 | return 0; | ||||
1790 | } | ||||
1791 | Root.replaceAllUsesWith(TmpLHSI); // Users now use TmpLHSI | ||||
1792 | TmpLHSI->setOperand(1, &Root); // TmpLHSI now uses the root | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1793 | BasicBlock::iterator ARI = &Root; ++ARI; |
Dan Gohman | 0bb9a3d | 2008-06-19 17:47:47 +0000 | [diff] [blame] | 1794 | TmpLHSI->moveBefore(ARI); // Move TmpLHSI to after Root |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1795 | ARI = Root; |
1796 | |||||
1797 | // Now propagate the ExtraOperand down the chain of instructions until we | ||||
1798 | // get to LHSI. | ||||
1799 | while (TmpLHSI != LHSI) { | ||||
1800 | Instruction *NextLHSI = cast<Instruction>(TmpLHSI->getOperand(0)); | ||||
1801 | // Move the instruction to immediately before the chain we are | ||||
1802 | // constructing to avoid breaking dominance properties. | ||||
Dan Gohman | 0bb9a3d | 2008-06-19 17:47:47 +0000 | [diff] [blame] | 1803 | NextLHSI->moveBefore(ARI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1804 | ARI = NextLHSI; |
1805 | |||||
1806 | Value *NextOp = NextLHSI->getOperand(1); | ||||
1807 | NextLHSI->setOperand(1, ExtraOperand); | ||||
1808 | TmpLHSI = NextLHSI; | ||||
1809 | ExtraOperand = NextOp; | ||||
1810 | } | ||||
1811 | |||||
1812 | // Now that the instructions are reassociated, have the functor perform | ||||
1813 | // the transformation... | ||||
1814 | return F.apply(Root); | ||||
1815 | } | ||||
1816 | |||||
1817 | LHSI = dyn_cast<Instruction>(LHSI->getOperand(0)); | ||||
1818 | } | ||||
1819 | return 0; | ||||
1820 | } | ||||
1821 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 1822 | namespace { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1823 | |
Nick Lewycky | 27f6c13 | 2008-05-23 04:34:58 +0000 | [diff] [blame] | 1824 | // AddRHS - Implements: X + X --> X << 1 |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1825 | struct AddRHS { |
1826 | Value *RHS; | ||||
1827 | AddRHS(Value *rhs) : RHS(rhs) {} | ||||
1828 | bool shouldApply(Value *LHS) const { return LHS == RHS; } | ||||
1829 | Instruction *apply(BinaryOperator &Add) const { | ||||
Nick Lewycky | 27f6c13 | 2008-05-23 04:34:58 +0000 | [diff] [blame] | 1830 | return BinaryOperator::CreateShl(Add.getOperand(0), |
1831 | ConstantInt::get(Add.getType(), 1)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1832 | } |
1833 | }; | ||||
1834 | |||||
1835 | // AddMaskingAnd - Implements (A & C1)+(B & C2) --> (A & C1)|(B & C2) | ||||
1836 | // iff C1&C2 == 0 | ||||
1837 | struct AddMaskingAnd { | ||||
1838 | Constant *C2; | ||||
1839 | AddMaskingAnd(Constant *c) : C2(c) {} | ||||
1840 | bool shouldApply(Value *LHS) const { | ||||
1841 | ConstantInt *C1; | ||||
1842 | return match(LHS, m_And(m_Value(), m_ConstantInt(C1))) && | ||||
1843 | ConstantExpr::getAnd(C1, C2)->isNullValue(); | ||||
1844 | } | ||||
1845 | Instruction *apply(BinaryOperator &Add) const { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1846 | return BinaryOperator::CreateOr(Add.getOperand(0), Add.getOperand(1)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1847 | } |
1848 | }; | ||||
1849 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 1850 | } |
1851 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1852 | static Value *FoldOperationIntoSelectOperand(Instruction &I, Value *SO, |
1853 | InstCombiner *IC) { | ||||
1854 | if (CastInst *CI = dyn_cast<CastInst>(&I)) { | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 1855 | return IC->InsertCastBefore(CI->getOpcode(), SO, I.getType(), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1856 | } |
1857 | |||||
1858 | // Figure out if the constant is the left or the right argument. | ||||
1859 | bool ConstIsRHS = isa<Constant>(I.getOperand(1)); | ||||
1860 | Constant *ConstOperand = cast<Constant>(I.getOperand(ConstIsRHS)); | ||||
1861 | |||||
1862 | if (Constant *SOC = dyn_cast<Constant>(SO)) { | ||||
1863 | if (ConstIsRHS) | ||||
1864 | return ConstantExpr::get(I.getOpcode(), SOC, ConstOperand); | ||||
1865 | return ConstantExpr::get(I.getOpcode(), ConstOperand, SOC); | ||||
1866 | } | ||||
1867 | |||||
1868 | Value *Op0 = SO, *Op1 = ConstOperand; | ||||
1869 | if (!ConstIsRHS) | ||||
1870 | std::swap(Op0, Op1); | ||||
1871 | Instruction *New; | ||||
1872 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(&I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1873 | New = BinaryOperator::Create(BO->getOpcode(), Op0, Op1,SO->getName()+".op"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1874 | else if (CmpInst *CI = dyn_cast<CmpInst>(&I)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1875 | New = CmpInst::Create(CI->getOpcode(), CI->getPredicate(), Op0, Op1, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1876 | SO->getName()+".cmp"); |
1877 | else { | ||||
1878 | assert(0 && "Unknown binary instruction type!"); | ||||
1879 | abort(); | ||||
1880 | } | ||||
1881 | return IC->InsertNewInstBefore(New, I); | ||||
1882 | } | ||||
1883 | |||||
1884 | // FoldOpIntoSelect - Given an instruction with a select as one operand and a | ||||
1885 | // constant as the other operand, try to fold the binary operator into the | ||||
1886 | // select arguments. This also works for Cast instructions, which obviously do | ||||
1887 | // not have a second operand. | ||||
1888 | static Instruction *FoldOpIntoSelect(Instruction &Op, SelectInst *SI, | ||||
1889 | InstCombiner *IC) { | ||||
1890 | // Don't modify shared select instructions | ||||
1891 | if (!SI->hasOneUse()) return 0; | ||||
1892 | Value *TV = SI->getOperand(1); | ||||
1893 | Value *FV = SI->getOperand(2); | ||||
1894 | |||||
1895 | if (isa<Constant>(TV) || isa<Constant>(FV)) { | ||||
1896 | // Bool selects with constant operands can be folded to logical ops. | ||||
1897 | if (SI->getType() == Type::Int1Ty) return 0; | ||||
1898 | |||||
1899 | Value *SelectTrueVal = FoldOperationIntoSelectOperand(Op, TV, IC); | ||||
1900 | Value *SelectFalseVal = FoldOperationIntoSelectOperand(Op, FV, IC); | ||||
1901 | |||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 1902 | return SelectInst::Create(SI->getCondition(), SelectTrueVal, |
1903 | SelectFalseVal); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1904 | } |
1905 | return 0; | ||||
1906 | } | ||||
1907 | |||||
1908 | |||||
1909 | /// FoldOpIntoPhi - Given a binary operator or cast instruction which has a PHI | ||||
1910 | /// node as operand #0, see if we can fold the instruction into the PHI (which | ||||
1911 | /// is only possible if all operands to the PHI are constants). | ||||
1912 | Instruction *InstCombiner::FoldOpIntoPhi(Instruction &I) { | ||||
1913 | PHINode *PN = cast<PHINode>(I.getOperand(0)); | ||||
1914 | unsigned NumPHIValues = PN->getNumIncomingValues(); | ||||
1915 | if (!PN->hasOneUse() || NumPHIValues == 0) return 0; | ||||
1916 | |||||
1917 | // Check to see if all of the operands of the PHI are constants. If there is | ||||
1918 | // one non-constant value, remember the BB it is. If there is more than one | ||||
1919 | // or if *it* is a PHI, bail out. | ||||
1920 | BasicBlock *NonConstBB = 0; | ||||
1921 | for (unsigned i = 0; i != NumPHIValues; ++i) | ||||
1922 | if (!isa<Constant>(PN->getIncomingValue(i))) { | ||||
1923 | if (NonConstBB) return 0; // More than one non-const value. | ||||
1924 | if (isa<PHINode>(PN->getIncomingValue(i))) return 0; // Itself a phi. | ||||
1925 | NonConstBB = PN->getIncomingBlock(i); | ||||
1926 | |||||
1927 | // If the incoming non-constant value is in I's block, we have an infinite | ||||
1928 | // loop. | ||||
1929 | if (NonConstBB == I.getParent()) | ||||
1930 | return 0; | ||||
1931 | } | ||||
1932 | |||||
1933 | // If there is exactly one non-constant value, we can insert a copy of the | ||||
1934 | // operation in that block. However, if this is a critical edge, we would be | ||||
1935 | // inserting the computation one some other paths (e.g. inside a loop). Only | ||||
1936 | // do this if the pred block is unconditionally branching into the phi block. | ||||
1937 | if (NonConstBB) { | ||||
1938 | BranchInst *BI = dyn_cast<BranchInst>(NonConstBB->getTerminator()); | ||||
1939 | if (!BI || !BI->isUnconditional()) return 0; | ||||
1940 | } | ||||
1941 | |||||
1942 | // Okay, we can do the transformation: create the new PHI node. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 1943 | PHINode *NewPN = PHINode::Create(I.getType(), ""); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1944 | NewPN->reserveOperandSpace(PN->getNumOperands()/2); |
1945 | InsertNewInstBefore(NewPN, *PN); | ||||
1946 | NewPN->takeName(PN); | ||||
1947 | |||||
1948 | // Next, add all of the operands to the PHI. | ||||
1949 | if (I.getNumOperands() == 2) { | ||||
1950 | Constant *C = cast<Constant>(I.getOperand(1)); | ||||
1951 | for (unsigned i = 0; i != NumPHIValues; ++i) { | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 1952 | Value *InV = 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1953 | if (Constant *InC = dyn_cast<Constant>(PN->getIncomingValue(i))) { |
1954 | if (CmpInst *CI = dyn_cast<CmpInst>(&I)) | ||||
1955 | InV = ConstantExpr::getCompare(CI->getPredicate(), InC, C); | ||||
1956 | else | ||||
1957 | InV = ConstantExpr::get(I.getOpcode(), InC, C); | ||||
1958 | } else { | ||||
1959 | assert(PN->getIncomingBlock(i) == NonConstBB); | ||||
1960 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(&I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1961 | InV = BinaryOperator::Create(BO->getOpcode(), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1962 | PN->getIncomingValue(i), C, "phitmp", |
1963 | NonConstBB->getTerminator()); | ||||
1964 | else if (CmpInst *CI = dyn_cast<CmpInst>(&I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1965 | InV = CmpInst::Create(CI->getOpcode(), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1966 | CI->getPredicate(), |
1967 | PN->getIncomingValue(i), C, "phitmp", | ||||
1968 | NonConstBB->getTerminator()); | ||||
1969 | else | ||||
1970 | assert(0 && "Unknown binop!"); | ||||
1971 | |||||
1972 | AddToWorkList(cast<Instruction>(InV)); | ||||
1973 | } | ||||
1974 | NewPN->addIncoming(InV, PN->getIncomingBlock(i)); | ||||
1975 | } | ||||
1976 | } else { | ||||
1977 | CastInst *CI = cast<CastInst>(&I); | ||||
1978 | const Type *RetTy = CI->getType(); | ||||
1979 | for (unsigned i = 0; i != NumPHIValues; ++i) { | ||||
1980 | Value *InV; | ||||
1981 | if (Constant *InC = dyn_cast<Constant>(PN->getIncomingValue(i))) { | ||||
1982 | InV = ConstantExpr::getCast(CI->getOpcode(), InC, RetTy); | ||||
1983 | } else { | ||||
1984 | assert(PN->getIncomingBlock(i) == NonConstBB); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1985 | InV = CastInst::Create(CI->getOpcode(), PN->getIncomingValue(i), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1986 | I.getType(), "phitmp", |
1987 | NonConstBB->getTerminator()); | ||||
1988 | AddToWorkList(cast<Instruction>(InV)); | ||||
1989 | } | ||||
1990 | NewPN->addIncoming(InV, PN->getIncomingBlock(i)); | ||||
1991 | } | ||||
1992 | } | ||||
1993 | return ReplaceInstUsesWith(I, NewPN); | ||||
1994 | } | ||||
1995 | |||||
Chris Lattner | 5547616 | 2008-01-29 06:52:45 +0000 | [diff] [blame] | 1996 | |
Chris Lattner | 3554f97 | 2008-05-20 05:46:13 +0000 | [diff] [blame] | 1997 | /// WillNotOverflowSignedAdd - Return true if we can prove that: |
1998 | /// (sext (add LHS, RHS)) === (add (sext LHS), (sext RHS)) | ||||
1999 | /// This basically requires proving that the add in the original type would not | ||||
2000 | /// overflow to change the sign bit or have a carry out. | ||||
2001 | bool InstCombiner::WillNotOverflowSignedAdd(Value *LHS, Value *RHS) { | ||||
2002 | // There are different heuristics we can use for this. Here are some simple | ||||
2003 | // ones. | ||||
2004 | |||||
2005 | // Add has the property that adding any two 2's complement numbers can only | ||||
2006 | // have one carry bit which can change a sign. As such, if LHS and RHS each | ||||
2007 | // have at least two sign bits, we know that the addition of the two values will | ||||
2008 | // sign extend fine. | ||||
2009 | if (ComputeNumSignBits(LHS) > 1 && ComputeNumSignBits(RHS) > 1) | ||||
2010 | return true; | ||||
2011 | |||||
2012 | |||||
2013 | // If one of the operands only has one non-zero bit, and if the other operand | ||||
2014 | // has a known-zero bit in a more significant place than it (not including the | ||||
2015 | // sign bit) the ripple may go up to and fill the zero, but won't change the | ||||
2016 | // sign. For example, (X & ~4) + 1. | ||||
2017 | |||||
2018 | // TODO: Implement. | ||||
2019 | |||||
2020 | return false; | ||||
2021 | } | ||||
2022 | |||||
Chris Lattner | 5547616 | 2008-01-29 06:52:45 +0000 | [diff] [blame] | 2023 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2024 | Instruction *InstCombiner::visitAdd(BinaryOperator &I) { |
2025 | bool Changed = SimplifyCommutative(I); | ||||
2026 | Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); | ||||
2027 | |||||
2028 | if (Constant *RHSC = dyn_cast<Constant>(RHS)) { | ||||
2029 | // X + undef -> undef | ||||
2030 | if (isa<UndefValue>(RHS)) | ||||
2031 | return ReplaceInstUsesWith(I, RHS); | ||||
2032 | |||||
2033 | // X + 0 --> X | ||||
2034 | if (!I.getType()->isFPOrFPVector()) { // NOTE: -0 + +0 = +0. | ||||
2035 | if (RHSC->isNullValue()) | ||||
2036 | return ReplaceInstUsesWith(I, LHS); | ||||
2037 | } else if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHSC)) { | ||||
Dale Johannesen | 2fc2078 | 2007-09-14 22:26:36 +0000 | [diff] [blame] | 2038 | if (CFP->isExactlyValue(ConstantFP::getNegativeZero |
2039 | (I.getType())->getValueAPF())) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2040 | return ReplaceInstUsesWith(I, LHS); |
2041 | } | ||||
2042 | |||||
2043 | if (ConstantInt *CI = dyn_cast<ConstantInt>(RHSC)) { | ||||
2044 | // X + (signbit) --> X ^ signbit | ||||
2045 | const APInt& Val = CI->getValue(); | ||||
2046 | uint32_t BitWidth = Val.getBitWidth(); | ||||
2047 | if (Val == APInt::getSignBit(BitWidth)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2048 | return BinaryOperator::CreateXor(LHS, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2049 | |
2050 | // See if SimplifyDemandedBits can simplify this. This handles stuff like | ||||
2051 | // (X & 254)+1 -> (X&254)|1 | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 2052 | if (!isa<VectorType>(I.getType()) && SimplifyDemandedInstructionBits(I)) |
2053 | return &I; | ||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 2054 | |
2055 | // zext(i1) - 1 -> select i1, 0, -1 | ||||
2056 | if (ZExtInst *ZI = dyn_cast<ZExtInst>(LHS)) | ||||
2057 | if (CI->isAllOnesValue() && | ||||
2058 | ZI->getOperand(0)->getType() == Type::Int1Ty) | ||||
2059 | return SelectInst::Create(ZI->getOperand(0), | ||||
2060 | Constant::getNullValue(I.getType()), | ||||
2061 | ConstantInt::getAllOnesValue(I.getType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2062 | } |
2063 | |||||
2064 | if (isa<PHINode>(LHS)) | ||||
2065 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2066 | return NV; | ||||
2067 | |||||
2068 | ConstantInt *XorRHS = 0; | ||||
2069 | Value *XorLHS = 0; | ||||
2070 | if (isa<ConstantInt>(RHSC) && | ||||
2071 | match(LHS, m_Xor(m_Value(XorLHS), m_ConstantInt(XorRHS)))) { | ||||
2072 | uint32_t TySizeBits = I.getType()->getPrimitiveSizeInBits(); | ||||
2073 | const APInt& RHSVal = cast<ConstantInt>(RHSC)->getValue(); | ||||
2074 | |||||
2075 | uint32_t Size = TySizeBits / 2; | ||||
2076 | APInt C0080Val(APInt(TySizeBits, 1ULL).shl(Size - 1)); | ||||
2077 | APInt CFF80Val(-C0080Val); | ||||
2078 | do { | ||||
2079 | if (TySizeBits > Size) { | ||||
2080 | // If we have ADD(XOR(AND(X, 0xFF), 0x80), 0xF..F80), it's a sext. | ||||
2081 | // If we have ADD(XOR(AND(X, 0xFF), 0xF..F80), 0x80), it's a sext. | ||||
2082 | if ((RHSVal == CFF80Val && XorRHS->getValue() == C0080Val) || | ||||
2083 | (RHSVal == C0080Val && XorRHS->getValue() == CFF80Val)) { | ||||
2084 | // This is a sign extend if the top bits are known zero. | ||||
2085 | if (!MaskedValueIsZero(XorLHS, | ||||
2086 | APInt::getHighBitsSet(TySizeBits, TySizeBits - Size))) | ||||
2087 | Size = 0; // Not a sign ext, but can't be any others either. | ||||
2088 | break; | ||||
2089 | } | ||||
2090 | } | ||||
2091 | Size >>= 1; | ||||
2092 | C0080Val = APIntOps::lshr(C0080Val, Size); | ||||
2093 | CFF80Val = APIntOps::ashr(CFF80Val, Size); | ||||
2094 | } while (Size >= 1); | ||||
2095 | |||||
2096 | // FIXME: This shouldn't be necessary. When the backends can handle types | ||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 2097 | // with funny bit widths then this switch statement should be removed. It |
2098 | // is just here to get the size of the "middle" type back up to something | ||||
2099 | // that the back ends can handle. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2100 | const Type *MiddleType = 0; |
2101 | switch (Size) { | ||||
2102 | default: break; | ||||
2103 | case 32: MiddleType = Type::Int32Ty; break; | ||||
2104 | case 16: MiddleType = Type::Int16Ty; break; | ||||
2105 | case 8: MiddleType = Type::Int8Ty; break; | ||||
2106 | } | ||||
2107 | if (MiddleType) { | ||||
2108 | Instruction *NewTrunc = new TruncInst(XorLHS, MiddleType, "sext"); | ||||
2109 | InsertNewInstBefore(NewTrunc, I); | ||||
2110 | return new SExtInst(NewTrunc, I.getType(), I.getName()); | ||||
2111 | } | ||||
2112 | } | ||||
2113 | } | ||||
2114 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2115 | if (I.getType() == Type::Int1Ty) |
2116 | return BinaryOperator::CreateXor(LHS, RHS); | ||||
2117 | |||||
Nick Lewycky | 4d474cd | 2008-05-23 04:39:38 +0000 | [diff] [blame] | 2118 | // X + X --> X << 1 |
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2119 | if (I.getType()->isInteger()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2120 | if (Instruction *Result = AssociativeOpt(I, AddRHS(RHS))) return Result; |
2121 | |||||
2122 | if (Instruction *RHSI = dyn_cast<Instruction>(RHS)) { | ||||
2123 | if (RHSI->getOpcode() == Instruction::Sub) | ||||
2124 | if (LHS == RHSI->getOperand(1)) // A + (B - A) --> B | ||||
2125 | return ReplaceInstUsesWith(I, RHSI->getOperand(0)); | ||||
2126 | } | ||||
2127 | if (Instruction *LHSI = dyn_cast<Instruction>(LHS)) { | ||||
2128 | if (LHSI->getOpcode() == Instruction::Sub) | ||||
2129 | if (RHS == LHSI->getOperand(1)) // (B - A) + A --> B | ||||
2130 | return ReplaceInstUsesWith(I, LHSI->getOperand(0)); | ||||
2131 | } | ||||
2132 | } | ||||
2133 | |||||
2134 | // -A + B --> B - A | ||||
Chris Lattner | 53c9fbf | 2008-02-17 21:03:36 +0000 | [diff] [blame] | 2135 | // -A + -B --> -(A + B) |
2136 | if (Value *LHSV = dyn_castNegVal(LHS)) { | ||||
Chris Lattner | 322a919 | 2008-02-18 17:50:16 +0000 | [diff] [blame] | 2137 | if (LHS->getType()->isIntOrIntVector()) { |
2138 | if (Value *RHSV = dyn_castNegVal(RHS)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2139 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSV, RHSV, "sum"); |
Chris Lattner | 322a919 | 2008-02-18 17:50:16 +0000 | [diff] [blame] | 2140 | InsertNewInstBefore(NewAdd, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2141 | return BinaryOperator::CreateNeg(NewAdd); |
Chris Lattner | 322a919 | 2008-02-18 17:50:16 +0000 | [diff] [blame] | 2142 | } |
Chris Lattner | 53c9fbf | 2008-02-17 21:03:36 +0000 | [diff] [blame] | 2143 | } |
2144 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2145 | return BinaryOperator::CreateSub(RHS, LHSV); |
Chris Lattner | 53c9fbf | 2008-02-17 21:03:36 +0000 | [diff] [blame] | 2146 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2147 | |
2148 | // A + -B --> A - B | ||||
2149 | if (!isa<Constant>(RHS)) | ||||
2150 | if (Value *V = dyn_castNegVal(RHS)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2151 | return BinaryOperator::CreateSub(LHS, V); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2152 | |
2153 | |||||
2154 | ConstantInt *C2; | ||||
2155 | if (Value *X = dyn_castFoldableMul(LHS, C2)) { | ||||
2156 | if (X == RHS) // X*C + X --> X * (C+1) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2157 | return BinaryOperator::CreateMul(RHS, AddOne(C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2158 | |
2159 | // X*C1 + X*C2 --> X * (C1+C2) | ||||
2160 | ConstantInt *C1; | ||||
2161 | if (X == dyn_castFoldableMul(RHS, C1)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2162 | return BinaryOperator::CreateMul(X, Add(C1, C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2163 | } |
2164 | |||||
2165 | // X + X*C --> X * (C+1) | ||||
2166 | if (dyn_castFoldableMul(RHS, C2) == LHS) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2167 | return BinaryOperator::CreateMul(LHS, AddOne(C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2168 | |
2169 | // X + ~X --> -1 since ~X = -X-1 | ||||
2170 | if (dyn_castNotVal(LHS) == RHS || dyn_castNotVal(RHS) == LHS) | ||||
2171 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
2172 | |||||
2173 | |||||
2174 | // (A & C1)+(B & C2) --> (A & C1)|(B & C2) iff C1&C2 == 0 | ||||
2175 | if (match(RHS, m_And(m_Value(), m_ConstantInt(C2)))) | ||||
2176 | if (Instruction *R = AssociativeOpt(I, AddMaskingAnd(C2))) | ||||
2177 | return R; | ||||
Chris Lattner | c1575ce | 2008-05-19 20:01:56 +0000 | [diff] [blame] | 2178 | |
2179 | // A+B --> A|B iff A and B have no bits set in common. | ||||
2180 | if (const IntegerType *IT = dyn_cast<IntegerType>(I.getType())) { | ||||
2181 | APInt Mask = APInt::getAllOnesValue(IT->getBitWidth()); | ||||
2182 | APInt LHSKnownOne(IT->getBitWidth(), 0); | ||||
2183 | APInt LHSKnownZero(IT->getBitWidth(), 0); | ||||
2184 | ComputeMaskedBits(LHS, Mask, LHSKnownZero, LHSKnownOne); | ||||
2185 | if (LHSKnownZero != 0) { | ||||
2186 | APInt RHSKnownOne(IT->getBitWidth(), 0); | ||||
2187 | APInt RHSKnownZero(IT->getBitWidth(), 0); | ||||
2188 | ComputeMaskedBits(RHS, Mask, RHSKnownZero, RHSKnownOne); | ||||
2189 | |||||
2190 | // No bits in common -> bitwise or. | ||||
Chris Lattner | 130443c | 2008-05-19 20:03:53 +0000 | [diff] [blame] | 2191 | if ((LHSKnownZero|RHSKnownZero).isAllOnesValue()) |
Chris Lattner | c1575ce | 2008-05-19 20:01:56 +0000 | [diff] [blame] | 2192 | return BinaryOperator::CreateOr(LHS, RHS); |
Chris Lattner | c1575ce | 2008-05-19 20:01:56 +0000 | [diff] [blame] | 2193 | } |
2194 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2195 | |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2196 | // W*X + Y*Z --> W * (X+Z) iff W == Y |
Nick Lewycky | 5d03b51 | 2008-02-03 08:19:11 +0000 | [diff] [blame] | 2197 | if (I.getType()->isIntOrIntVector()) { |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2198 | Value *W, *X, *Y, *Z; |
2199 | if (match(LHS, m_Mul(m_Value(W), m_Value(X))) && | ||||
2200 | match(RHS, m_Mul(m_Value(Y), m_Value(Z)))) { | ||||
2201 | if (W != Y) { | ||||
2202 | if (W == Z) { | ||||
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 2203 | std::swap(Y, Z); |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2204 | } else if (Y == X) { |
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 2205 | std::swap(W, X); |
2206 | } else if (X == Z) { | ||||
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2207 | std::swap(Y, Z); |
2208 | std::swap(W, X); | ||||
2209 | } | ||||
2210 | } | ||||
2211 | |||||
2212 | if (W == Y) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2213 | Value *NewAdd = InsertNewInstBefore(BinaryOperator::CreateAdd(X, Z, |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2214 | LHS->getName()), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2215 | return BinaryOperator::CreateMul(W, NewAdd); |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2216 | } |
2217 | } | ||||
2218 | } | ||||
2219 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2220 | if (ConstantInt *CRHS = dyn_cast<ConstantInt>(RHS)) { |
2221 | Value *X = 0; | ||||
2222 | if (match(LHS, m_Not(m_Value(X)))) // ~X + C --> (C-1) - X | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2223 | return BinaryOperator::CreateSub(SubOne(CRHS), X); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2224 | |
2225 | // (X & FF00) + xx00 -> (X+xx00) & FF00 | ||||
2226 | if (LHS->hasOneUse() && match(LHS, m_And(m_Value(X), m_ConstantInt(C2)))) { | ||||
2227 | Constant *Anded = And(CRHS, C2); | ||||
2228 | if (Anded == CRHS) { | ||||
2229 | // See if all bits from the first bit set in the Add RHS up are included | ||||
2230 | // in the mask. First, get the rightmost bit. | ||||
2231 | const APInt& AddRHSV = CRHS->getValue(); | ||||
2232 | |||||
2233 | // Form a mask of all bits from the lowest bit added through the top. | ||||
2234 | APInt AddRHSHighBits(~((AddRHSV & -AddRHSV)-1)); | ||||
2235 | |||||
2236 | // See if the and mask includes all of these bits. | ||||
2237 | APInt AddRHSHighBitsAnd(AddRHSHighBits & C2->getValue()); | ||||
2238 | |||||
2239 | if (AddRHSHighBits == AddRHSHighBitsAnd) { | ||||
2240 | // Okay, the xform is safe. Insert the new add pronto. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2241 | Value *NewAdd = InsertNewInstBefore(BinaryOperator::CreateAdd(X, CRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2242 | LHS->getName()), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2243 | return BinaryOperator::CreateAnd(NewAdd, C2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2244 | } |
2245 | } | ||||
2246 | } | ||||
2247 | |||||
2248 | // Try to fold constant add into select arguments. | ||||
2249 | if (SelectInst *SI = dyn_cast<SelectInst>(LHS)) | ||||
2250 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2251 | return R; | ||||
2252 | } | ||||
2253 | |||||
2254 | // add (cast *A to intptrtype) B -> | ||||
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2255 | // cast (GEP (cast *A to sbyte*) B) --> intptrtype |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2256 | { |
2257 | CastInst *CI = dyn_cast<CastInst>(LHS); | ||||
2258 | Value *Other = RHS; | ||||
2259 | if (!CI) { | ||||
2260 | CI = dyn_cast<CastInst>(RHS); | ||||
2261 | Other = LHS; | ||||
2262 | } | ||||
2263 | if (CI && CI->getType()->isSized() && | ||||
2264 | (CI->getType()->getPrimitiveSizeInBits() == | ||||
2265 | TD->getIntPtrType()->getPrimitiveSizeInBits()) | ||||
2266 | && isa<PointerType>(CI->getOperand(0)->getType())) { | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 2267 | unsigned AS = |
2268 | cast<PointerType>(CI->getOperand(0)->getType())->getAddressSpace(); | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 2269 | Value *I2 = InsertBitCastBefore(CI->getOperand(0), |
2270 | PointerType::get(Type::Int8Ty, AS), I); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2271 | I2 = InsertNewInstBefore(GetElementPtrInst::Create(I2, Other, "ctg2"), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2272 | return new PtrToIntInst(I2, CI->getType()); |
2273 | } | ||||
2274 | } | ||||
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2275 | |
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2276 | // add (select X 0 (sub n A)) A --> select X A n |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2277 | { |
2278 | SelectInst *SI = dyn_cast<SelectInst>(LHS); | ||||
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 2279 | Value *A = RHS; |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2280 | if (!SI) { |
2281 | SI = dyn_cast<SelectInst>(RHS); | ||||
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 2282 | A = LHS; |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2283 | } |
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2284 | if (SI && SI->hasOneUse()) { |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2285 | Value *TV = SI->getTrueValue(); |
2286 | Value *FV = SI->getFalseValue(); | ||||
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 2287 | Value *N; |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2288 | |
2289 | // Can we fold the add into the argument of the select? | ||||
2290 | // We check both true and false select arguments for a matching subtract. | ||||
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 2291 | if (match(FV, m_Zero()) && match(TV, m_Sub(m_Value(N), m_Specific(A)))) |
2292 | // Fold the add into the true select value. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2293 | return SelectInst::Create(SI->getCondition(), N, A); |
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 2294 | if (match(TV, m_Zero()) && match(FV, m_Sub(m_Value(N), m_Specific(A)))) |
2295 | // Fold the add into the false select value. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2296 | return SelectInst::Create(SI->getCondition(), A, N); |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2297 | } |
2298 | } | ||||
Chris Lattner | 5547616 | 2008-01-29 06:52:45 +0000 | [diff] [blame] | 2299 | |
2300 | // Check for X+0.0. Simplify it to X if we know X is not -0.0. | ||||
2301 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHS)) | ||||
2302 | if (CFP->getValueAPF().isPosZero() && CannotBeNegativeZero(LHS)) | ||||
2303 | return ReplaceInstUsesWith(I, LHS); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2304 | |
Chris Lattner | 3554f97 | 2008-05-20 05:46:13 +0000 | [diff] [blame] | 2305 | // Check for (add (sext x), y), see if we can merge this into an |
2306 | // integer add followed by a sext. | ||||
2307 | if (SExtInst *LHSConv = dyn_cast<SExtInst>(LHS)) { | ||||
2308 | // (add (sext x), cst) --> (sext (add x, cst')) | ||||
2309 | if (ConstantInt *RHSC = dyn_cast<ConstantInt>(RHS)) { | ||||
2310 | Constant *CI = | ||||
2311 | ConstantExpr::getTrunc(RHSC, LHSConv->getOperand(0)->getType()); | ||||
2312 | if (LHSConv->hasOneUse() && | ||||
2313 | ConstantExpr::getSExt(CI, I.getType()) == RHSC && | ||||
2314 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), CI)) { | ||||
2315 | // Insert the new, smaller add. | ||||
2316 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2317 | CI, "addconv"); | ||||
2318 | InsertNewInstBefore(NewAdd, I); | ||||
2319 | return new SExtInst(NewAdd, I.getType()); | ||||
2320 | } | ||||
2321 | } | ||||
2322 | |||||
2323 | // (add (sext x), (sext y)) --> (sext (add int x, y)) | ||||
2324 | if (SExtInst *RHSConv = dyn_cast<SExtInst>(RHS)) { | ||||
2325 | // Only do this if x/y have the same type, if at last one of them has a | ||||
2326 | // single use (so we don't increase the number of sexts), and if the | ||||
2327 | // integer add will not overflow. | ||||
2328 | if (LHSConv->getOperand(0)->getType()==RHSConv->getOperand(0)->getType()&& | ||||
2329 | (LHSConv->hasOneUse() || RHSConv->hasOneUse()) && | ||||
2330 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), | ||||
2331 | RHSConv->getOperand(0))) { | ||||
2332 | // Insert the new integer add. | ||||
2333 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2334 | RHSConv->getOperand(0), | ||||
2335 | "addconv"); | ||||
2336 | InsertNewInstBefore(NewAdd, I); | ||||
2337 | return new SExtInst(NewAdd, I.getType()); | ||||
2338 | } | ||||
2339 | } | ||||
2340 | } | ||||
2341 | |||||
2342 | // Check for (add double (sitofp x), y), see if we can merge this into an | ||||
2343 | // integer add followed by a promotion. | ||||
2344 | if (SIToFPInst *LHSConv = dyn_cast<SIToFPInst>(LHS)) { | ||||
2345 | // (add double (sitofp x), fpcst) --> (sitofp (add int x, intcst)) | ||||
2346 | // ... if the constant fits in the integer value. This is useful for things | ||||
2347 | // like (double)(x & 1234) + 4.0 -> (double)((X & 1234)+4) which no longer | ||||
2348 | // requires a constant pool load, and generally allows the add to be better | ||||
2349 | // instcombined. | ||||
2350 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHS)) { | ||||
2351 | Constant *CI = | ||||
2352 | ConstantExpr::getFPToSI(CFP, LHSConv->getOperand(0)->getType()); | ||||
2353 | if (LHSConv->hasOneUse() && | ||||
2354 | ConstantExpr::getSIToFP(CI, I.getType()) == CFP && | ||||
2355 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), CI)) { | ||||
2356 | // Insert the new integer add. | ||||
2357 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2358 | CI, "addconv"); | ||||
2359 | InsertNewInstBefore(NewAdd, I); | ||||
2360 | return new SIToFPInst(NewAdd, I.getType()); | ||||
2361 | } | ||||
2362 | } | ||||
2363 | |||||
2364 | // (add double (sitofp x), (sitofp y)) --> (sitofp (add int x, y)) | ||||
2365 | if (SIToFPInst *RHSConv = dyn_cast<SIToFPInst>(RHS)) { | ||||
2366 | // Only do this if x/y have the same type, if at last one of them has a | ||||
2367 | // single use (so we don't increase the number of int->fp conversions), | ||||
2368 | // and if the integer add will not overflow. | ||||
2369 | if (LHSConv->getOperand(0)->getType()==RHSConv->getOperand(0)->getType()&& | ||||
2370 | (LHSConv->hasOneUse() || RHSConv->hasOneUse()) && | ||||
2371 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), | ||||
2372 | RHSConv->getOperand(0))) { | ||||
2373 | // Insert the new integer add. | ||||
2374 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2375 | RHSConv->getOperand(0), | ||||
2376 | "addconv"); | ||||
2377 | InsertNewInstBefore(NewAdd, I); | ||||
2378 | return new SIToFPInst(NewAdd, I.getType()); | ||||
2379 | } | ||||
2380 | } | ||||
2381 | } | ||||
2382 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2383 | return Changed ? &I : 0; |
2384 | } | ||||
2385 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2386 | Instruction *InstCombiner::visitSub(BinaryOperator &I) { |
2387 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2388 | |||||
Chris Lattner | 27fbef4 | 2008-07-17 06:07:20 +0000 | [diff] [blame] | 2389 | if (Op0 == Op1 && // sub X, X -> 0 |
2390 | !I.getType()->isFPOrFPVector()) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2391 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
2392 | |||||
2393 | // If this is a 'B = x-(-A)', change to B = x+A... | ||||
2394 | if (Value *V = dyn_castNegVal(Op1)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2395 | return BinaryOperator::CreateAdd(Op0, V); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2396 | |
2397 | if (isa<UndefValue>(Op0)) | ||||
2398 | return ReplaceInstUsesWith(I, Op0); // undef - X -> undef | ||||
2399 | if (isa<UndefValue>(Op1)) | ||||
2400 | return ReplaceInstUsesWith(I, Op1); // X - undef -> undef | ||||
2401 | |||||
2402 | if (ConstantInt *C = dyn_cast<ConstantInt>(Op0)) { | ||||
2403 | // Replace (-1 - A) with (~A)... | ||||
2404 | if (C->isAllOnesValue()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2405 | return BinaryOperator::CreateNot(Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2406 | |
2407 | // C - ~X == X + (1+C) | ||||
2408 | Value *X = 0; | ||||
2409 | if (match(Op1, m_Not(m_Value(X)))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2410 | return BinaryOperator::CreateAdd(X, AddOne(C)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2411 | |
2412 | // -(X >>u 31) -> (X >>s 31) | ||||
2413 | // -(X >>s 31) -> (X >>u 31) | ||||
2414 | if (C->isZero()) { | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2415 | if (BinaryOperator *SI = dyn_cast<BinaryOperator>(Op1)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2416 | if (SI->getOpcode() == Instruction::LShr) { |
2417 | if (ConstantInt *CU = dyn_cast<ConstantInt>(SI->getOperand(1))) { | ||||
2418 | // Check to see if we are shifting out everything but the sign bit. | ||||
2419 | if (CU->getLimitedValue(SI->getType()->getPrimitiveSizeInBits()) == | ||||
2420 | SI->getType()->getPrimitiveSizeInBits()-1) { | ||||
2421 | // Ok, the transformation is safe. Insert AShr. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2422 | return BinaryOperator::Create(Instruction::AShr, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2423 | SI->getOperand(0), CU, SI->getName()); |
2424 | } | ||||
2425 | } | ||||
2426 | } | ||||
2427 | else if (SI->getOpcode() == Instruction::AShr) { | ||||
2428 | if (ConstantInt *CU = dyn_cast<ConstantInt>(SI->getOperand(1))) { | ||||
2429 | // Check to see if we are shifting out everything but the sign bit. | ||||
2430 | if (CU->getLimitedValue(SI->getType()->getPrimitiveSizeInBits()) == | ||||
2431 | SI->getType()->getPrimitiveSizeInBits()-1) { | ||||
2432 | // Ok, the transformation is safe. Insert LShr. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2433 | return BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2434 | SI->getOperand(0), CU, SI->getName()); |
2435 | } | ||||
2436 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2437 | } |
2438 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2439 | } |
2440 | |||||
2441 | // Try to fold constant sub into select arguments. | ||||
2442 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) | ||||
2443 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2444 | return R; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2445 | } |
2446 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2447 | if (I.getType() == Type::Int1Ty) |
2448 | return BinaryOperator::CreateXor(Op0, Op1); | ||||
2449 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2450 | if (BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1)) { |
2451 | if (Op1I->getOpcode() == Instruction::Add && | ||||
2452 | !Op0->getType()->isFPOrFPVector()) { | ||||
2453 | if (Op1I->getOperand(0) == Op0) // X-(X+Y) == -Y | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2454 | return BinaryOperator::CreateNeg(Op1I->getOperand(1), I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2455 | else if (Op1I->getOperand(1) == Op0) // X-(Y+X) == -Y |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2456 | return BinaryOperator::CreateNeg(Op1I->getOperand(0), I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2457 | else if (ConstantInt *CI1 = dyn_cast<ConstantInt>(I.getOperand(0))) { |
2458 | if (ConstantInt *CI2 = dyn_cast<ConstantInt>(Op1I->getOperand(1))) | ||||
2459 | // C1-(X+C2) --> (C1-C2)-X | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2460 | return BinaryOperator::CreateSub(Subtract(CI1, CI2), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2461 | Op1I->getOperand(0)); |
2462 | } | ||||
2463 | } | ||||
2464 | |||||
2465 | if (Op1I->hasOneUse()) { | ||||
2466 | // Replace (x - (y - z)) with (x + (z - y)) if the (y - z) subexpression | ||||
2467 | // is not used by anyone else... | ||||
2468 | // | ||||
2469 | if (Op1I->getOpcode() == Instruction::Sub && | ||||
2470 | !Op1I->getType()->isFPOrFPVector()) { | ||||
2471 | // Swap the two operands of the subexpr... | ||||
2472 | Value *IIOp0 = Op1I->getOperand(0), *IIOp1 = Op1I->getOperand(1); | ||||
2473 | Op1I->setOperand(0, IIOp1); | ||||
2474 | Op1I->setOperand(1, IIOp0); | ||||
2475 | |||||
2476 | // Create the new top level add instruction... | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2477 | return BinaryOperator::CreateAdd(Op0, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2478 | } |
2479 | |||||
2480 | // Replace (A - (A & B)) with (A & ~B) if this is the only use of (A&B)... | ||||
2481 | // | ||||
2482 | if (Op1I->getOpcode() == Instruction::And && | ||||
2483 | (Op1I->getOperand(0) == Op0 || Op1I->getOperand(1) == Op0)) { | ||||
2484 | Value *OtherOp = Op1I->getOperand(Op1I->getOperand(0) == Op0); | ||||
2485 | |||||
2486 | Value *NewNot = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2487 | InsertNewInstBefore(BinaryOperator::CreateNot(OtherOp, "B.not"), I); |
2488 | return BinaryOperator::CreateAnd(Op0, NewNot); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2489 | } |
2490 | |||||
2491 | // 0 - (X sdiv C) -> (X sdiv -C) | ||||
2492 | if (Op1I->getOpcode() == Instruction::SDiv) | ||||
2493 | if (ConstantInt *CSI = dyn_cast<ConstantInt>(Op0)) | ||||
2494 | if (CSI->isZero()) | ||||
2495 | if (Constant *DivRHS = dyn_cast<Constant>(Op1I->getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2496 | return BinaryOperator::CreateSDiv(Op1I->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2497 | ConstantExpr::getNeg(DivRHS)); |
2498 | |||||
2499 | // X - X*C --> X * (1-C) | ||||
2500 | ConstantInt *C2 = 0; | ||||
2501 | if (dyn_castFoldableMul(Op1I, C2) == Op0) { | ||||
2502 | Constant *CP1 = Subtract(ConstantInt::get(I.getType(), 1), C2); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2503 | return BinaryOperator::CreateMul(Op0, CP1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2504 | } |
2505 | } | ||||
2506 | } | ||||
2507 | |||||
2508 | if (!Op0->getType()->isFPOrFPVector()) | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2509 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2510 | if (Op0I->getOpcode() == Instruction::Add) { |
2511 | if (Op0I->getOperand(0) == Op1) // (Y+X)-Y == X | ||||
2512 | return ReplaceInstUsesWith(I, Op0I->getOperand(1)); | ||||
2513 | else if (Op0I->getOperand(1) == Op1) // (X+Y)-Y == X | ||||
2514 | return ReplaceInstUsesWith(I, Op0I->getOperand(0)); | ||||
2515 | } else if (Op0I->getOpcode() == Instruction::Sub) { | ||||
2516 | if (Op0I->getOperand(0) == Op1) // (X-Y)-X == -Y | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2517 | return BinaryOperator::CreateNeg(Op0I->getOperand(1), I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2518 | } |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2519 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2520 | |
2521 | ConstantInt *C1; | ||||
2522 | if (Value *X = dyn_castFoldableMul(Op0, C1)) { | ||||
2523 | if (X == Op1) // X*C - X --> X * (C-1) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2524 | return BinaryOperator::CreateMul(Op1, SubOne(C1)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2525 | |
2526 | ConstantInt *C2; // X*C1 - X*C2 -> X * (C1-C2) | ||||
2527 | if (X == dyn_castFoldableMul(Op1, C2)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2528 | return BinaryOperator::CreateMul(X, Subtract(C1, C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2529 | } |
2530 | return 0; | ||||
2531 | } | ||||
2532 | |||||
2533 | /// isSignBitCheck - Given an exploded icmp instruction, return true if the | ||||
2534 | /// comparison only checks the sign bit. If it only checks the sign bit, set | ||||
2535 | /// TrueIfSigned if the result of the comparison is true when the input value is | ||||
2536 | /// signed. | ||||
2537 | static bool isSignBitCheck(ICmpInst::Predicate pred, ConstantInt *RHS, | ||||
2538 | bool &TrueIfSigned) { | ||||
2539 | switch (pred) { | ||||
2540 | case ICmpInst::ICMP_SLT: // True if LHS s< 0 | ||||
2541 | TrueIfSigned = true; | ||||
2542 | return RHS->isZero(); | ||||
2543 | case ICmpInst::ICMP_SLE: // True if LHS s<= RHS and RHS == -1 | ||||
2544 | TrueIfSigned = true; | ||||
2545 | return RHS->isAllOnesValue(); | ||||
2546 | case ICmpInst::ICMP_SGT: // True if LHS s> -1 | ||||
2547 | TrueIfSigned = false; | ||||
2548 | return RHS->isAllOnesValue(); | ||||
2549 | case ICmpInst::ICMP_UGT: | ||||
2550 | // True if LHS u> RHS and RHS == high-bit-mask - 1 | ||||
2551 | TrueIfSigned = true; | ||||
2552 | return RHS->getValue() == | ||||
2553 | APInt::getSignedMaxValue(RHS->getType()->getPrimitiveSizeInBits()); | ||||
2554 | case ICmpInst::ICMP_UGE: | ||||
2555 | // True if LHS u>= RHS and RHS == high-bit-mask (2^7, 2^15, 2^31, etc) | ||||
2556 | TrueIfSigned = true; | ||||
Chris Lattner | 60813c2 | 2008-06-02 01:29:46 +0000 | [diff] [blame] | 2557 | return RHS->getValue().isSignBit(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2558 | default: |
2559 | return false; | ||||
2560 | } | ||||
2561 | } | ||||
2562 | |||||
2563 | Instruction *InstCombiner::visitMul(BinaryOperator &I) { | ||||
2564 | bool Changed = SimplifyCommutative(I); | ||||
2565 | Value *Op0 = I.getOperand(0); | ||||
2566 | |||||
2567 | if (isa<UndefValue>(I.getOperand(1))) // undef * X -> 0 | ||||
2568 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2569 | |||||
2570 | // Simplify mul instructions with a constant RHS... | ||||
2571 | if (Constant *Op1 = dyn_cast<Constant>(I.getOperand(1))) { | ||||
2572 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) { | ||||
2573 | |||||
2574 | // ((X << C1)*C2) == (X * (C2 << C1)) | ||||
2575 | if (BinaryOperator *SI = dyn_cast<BinaryOperator>(Op0)) | ||||
2576 | if (SI->getOpcode() == Instruction::Shl) | ||||
2577 | if (Constant *ShOp = dyn_cast<Constant>(SI->getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2578 | return BinaryOperator::CreateMul(SI->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2579 | ConstantExpr::getShl(CI, ShOp)); |
2580 | |||||
2581 | if (CI->isZero()) | ||||
2582 | return ReplaceInstUsesWith(I, Op1); // X * 0 == 0 | ||||
2583 | if (CI->equalsInt(1)) // X * 1 == X | ||||
2584 | return ReplaceInstUsesWith(I, Op0); | ||||
2585 | if (CI->isAllOnesValue()) // X * -1 == 0 - X | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2586 | return BinaryOperator::CreateNeg(Op0, I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2587 | |
2588 | const APInt& Val = cast<ConstantInt>(CI)->getValue(); | ||||
2589 | if (Val.isPowerOf2()) { // Replace X*(2^C) with X << C | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2590 | return BinaryOperator::CreateShl(Op0, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2591 | ConstantInt::get(Op0->getType(), Val.logBase2())); |
2592 | } | ||||
2593 | } else if (ConstantFP *Op1F = dyn_cast<ConstantFP>(Op1)) { | ||||
2594 | if (Op1F->isNullValue()) | ||||
2595 | return ReplaceInstUsesWith(I, Op1); | ||||
2596 | |||||
2597 | // "In IEEE floating point, x*1 is not equivalent to x for nans. However, | ||||
2598 | // ANSI says we can drop signals, so we can do this anyway." (from GCC) | ||||
Chris Lattner | 6297fc7 | 2008-08-11 22:06:05 +0000 | [diff] [blame] | 2599 | if (Op1F->isExactlyValue(1.0)) |
2600 | return ReplaceInstUsesWith(I, Op0); // Eliminate 'mul double %X, 1.0' | ||||
2601 | } else if (isa<VectorType>(Op1->getType())) { | ||||
2602 | if (isa<ConstantAggregateZero>(Op1)) | ||||
2603 | return ReplaceInstUsesWith(I, Op1); | ||||
Nick Lewycky | 9441873 | 2008-11-27 20:21:08 +0000 | [diff] [blame] | 2604 | |
2605 | if (ConstantVector *Op1V = dyn_cast<ConstantVector>(Op1)) { | ||||
2606 | if (Op1V->isAllOnesValue()) // X * -1 == 0 - X | ||||
2607 | return BinaryOperator::CreateNeg(Op0, I.getName()); | ||||
2608 | |||||
2609 | // As above, vector X*splat(1.0) -> X in all defined cases. | ||||
2610 | if (Constant *Splat = Op1V->getSplatValue()) { | ||||
2611 | if (ConstantFP *F = dyn_cast<ConstantFP>(Splat)) | ||||
2612 | if (F->isExactlyValue(1.0)) | ||||
2613 | return ReplaceInstUsesWith(I, Op0); | ||||
2614 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Splat)) | ||||
2615 | if (CI->equalsInt(1)) | ||||
2616 | return ReplaceInstUsesWith(I, Op0); | ||||
2617 | } | ||||
2618 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2619 | } |
2620 | |||||
2621 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) | ||||
2622 | if (Op0I->getOpcode() == Instruction::Add && Op0I->hasOneUse() && | ||||
Chris Lattner | 5819408 | 2008-05-18 04:11:26 +0000 | [diff] [blame] | 2623 | isa<ConstantInt>(Op0I->getOperand(1)) && isa<ConstantInt>(Op1)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2624 | // Canonicalize (X+C1)*C2 -> X*C2+C1*C2. |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2625 | Instruction *Add = BinaryOperator::CreateMul(Op0I->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2626 | Op1, "tmp"); |
2627 | InsertNewInstBefore(Add, I); | ||||
2628 | Value *C1C2 = ConstantExpr::getMul(Op1, | ||||
2629 | cast<Constant>(Op0I->getOperand(1))); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2630 | return BinaryOperator::CreateAdd(Add, C1C2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2631 | |
2632 | } | ||||
2633 | |||||
2634 | // Try to fold constant mul into select arguments. | ||||
2635 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
2636 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2637 | return R; | ||||
2638 | |||||
2639 | if (isa<PHINode>(Op0)) | ||||
2640 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2641 | return NV; | ||||
2642 | } | ||||
2643 | |||||
2644 | if (Value *Op0v = dyn_castNegVal(Op0)) // -X * -Y = X*Y | ||||
2645 | if (Value *Op1v = dyn_castNegVal(I.getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2646 | return BinaryOperator::CreateMul(Op0v, Op1v); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2647 | |
Nick Lewycky | 1c24640 | 2008-11-21 07:33:58 +0000 | [diff] [blame] | 2648 | // (X / Y) * Y = X - (X % Y) |
2649 | // (X / Y) * -Y = (X % Y) - X | ||||
2650 | { | ||||
2651 | Value *Op1 = I.getOperand(1); | ||||
2652 | BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0); | ||||
2653 | if (!BO || | ||||
2654 | (BO->getOpcode() != Instruction::UDiv && | ||||
2655 | BO->getOpcode() != Instruction::SDiv)) { | ||||
2656 | Op1 = Op0; | ||||
2657 | BO = dyn_cast<BinaryOperator>(I.getOperand(1)); | ||||
2658 | } | ||||
2659 | Value *Neg = dyn_castNegVal(Op1); | ||||
2660 | if (BO && BO->hasOneUse() && | ||||
2661 | (BO->getOperand(1) == Op1 || BO->getOperand(1) == Neg) && | ||||
2662 | (BO->getOpcode() == Instruction::UDiv || | ||||
2663 | BO->getOpcode() == Instruction::SDiv)) { | ||||
2664 | Value *Op0BO = BO->getOperand(0), *Op1BO = BO->getOperand(1); | ||||
2665 | |||||
2666 | Instruction *Rem; | ||||
2667 | if (BO->getOpcode() == Instruction::UDiv) | ||||
2668 | Rem = BinaryOperator::CreateURem(Op0BO, Op1BO); | ||||
2669 | else | ||||
2670 | Rem = BinaryOperator::CreateSRem(Op0BO, Op1BO); | ||||
2671 | |||||
2672 | InsertNewInstBefore(Rem, I); | ||||
2673 | Rem->takeName(BO); | ||||
2674 | |||||
2675 | if (Op1BO == Op1) | ||||
2676 | return BinaryOperator::CreateSub(Op0BO, Rem); | ||||
2677 | else | ||||
2678 | return BinaryOperator::CreateSub(Rem, Op0BO); | ||||
2679 | } | ||||
2680 | } | ||||
2681 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2682 | if (I.getType() == Type::Int1Ty) |
2683 | return BinaryOperator::CreateAnd(Op0, I.getOperand(1)); | ||||
2684 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2685 | // If one of the operands of the multiply is a cast from a boolean value, then |
2686 | // we know the bool is either zero or one, so this is a 'masking' multiply. | ||||
2687 | // See if we can simplify things based on how the boolean was originally | ||||
2688 | // formed. | ||||
2689 | CastInst *BoolCast = 0; | ||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2690 | if (ZExtInst *CI = dyn_cast<ZExtInst>(Op0)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2691 | if (CI->getOperand(0)->getType() == Type::Int1Ty) |
2692 | BoolCast = CI; | ||||
2693 | if (!BoolCast) | ||||
2694 | if (ZExtInst *CI = dyn_cast<ZExtInst>(I.getOperand(1))) | ||||
2695 | if (CI->getOperand(0)->getType() == Type::Int1Ty) | ||||
2696 | BoolCast = CI; | ||||
2697 | if (BoolCast) { | ||||
2698 | if (ICmpInst *SCI = dyn_cast<ICmpInst>(BoolCast->getOperand(0))) { | ||||
2699 | Value *SCIOp0 = SCI->getOperand(0), *SCIOp1 = SCI->getOperand(1); | ||||
2700 | const Type *SCOpTy = SCIOp0->getType(); | ||||
2701 | bool TIS = false; | ||||
2702 | |||||
2703 | // If the icmp is true iff the sign bit of X is set, then convert this | ||||
2704 | // multiply into a shift/and combination. | ||||
2705 | if (isa<ConstantInt>(SCIOp1) && | ||||
2706 | isSignBitCheck(SCI->getPredicate(), cast<ConstantInt>(SCIOp1), TIS) && | ||||
2707 | TIS) { | ||||
2708 | // Shift the X value right to turn it into "all signbits". | ||||
2709 | Constant *Amt = ConstantInt::get(SCIOp0->getType(), | ||||
2710 | SCOpTy->getPrimitiveSizeInBits()-1); | ||||
2711 | Value *V = | ||||
2712 | InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2713 | BinaryOperator::Create(Instruction::AShr, SCIOp0, Amt, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2714 | BoolCast->getOperand(0)->getName()+ |
2715 | ".mask"), I); | ||||
2716 | |||||
2717 | // If the multiply type is not the same as the source type, sign extend | ||||
2718 | // or truncate to the multiply type. | ||||
2719 | if (I.getType() != V->getType()) { | ||||
2720 | uint32_t SrcBits = V->getType()->getPrimitiveSizeInBits(); | ||||
2721 | uint32_t DstBits = I.getType()->getPrimitiveSizeInBits(); | ||||
2722 | Instruction::CastOps opcode = | ||||
2723 | (SrcBits == DstBits ? Instruction::BitCast : | ||||
2724 | (SrcBits < DstBits ? Instruction::SExt : Instruction::Trunc)); | ||||
2725 | V = InsertCastBefore(opcode, V, I.getType(), I); | ||||
2726 | } | ||||
2727 | |||||
2728 | Value *OtherOp = Op0 == BoolCast ? I.getOperand(1) : Op0; | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2729 | return BinaryOperator::CreateAnd(V, OtherOp); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2730 | } |
2731 | } | ||||
2732 | } | ||||
2733 | |||||
2734 | return Changed ? &I : 0; | ||||
2735 | } | ||||
2736 | |||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 2737 | /// SimplifyDivRemOfSelect - Try to fold a divide or remainder of a select |
2738 | /// instruction. | ||||
2739 | bool InstCombiner::SimplifyDivRemOfSelect(BinaryOperator &I) { | ||||
2740 | SelectInst *SI = cast<SelectInst>(I.getOperand(1)); | ||||
2741 | |||||
2742 | // div/rem X, (Cond ? 0 : Y) -> div/rem X, Y | ||||
2743 | int NonNullOperand = -1; | ||||
2744 | if (Constant *ST = dyn_cast<Constant>(SI->getOperand(1))) | ||||
2745 | if (ST->isNullValue()) | ||||
2746 | NonNullOperand = 2; | ||||
2747 | // div/rem X, (Cond ? Y : 0) -> div/rem X, Y | ||||
2748 | if (Constant *ST = dyn_cast<Constant>(SI->getOperand(2))) | ||||
2749 | if (ST->isNullValue()) | ||||
2750 | NonNullOperand = 1; | ||||
2751 | |||||
2752 | if (NonNullOperand == -1) | ||||
2753 | return false; | ||||
2754 | |||||
2755 | Value *SelectCond = SI->getOperand(0); | ||||
2756 | |||||
2757 | // Change the div/rem to use 'Y' instead of the select. | ||||
2758 | I.setOperand(1, SI->getOperand(NonNullOperand)); | ||||
2759 | |||||
2760 | // Okay, we know we replace the operand of the div/rem with 'Y' with no | ||||
2761 | // problem. However, the select, or the condition of the select may have | ||||
2762 | // multiple uses. Based on our knowledge that the operand must be non-zero, | ||||
2763 | // propagate the known value for the select into other uses of it, and | ||||
2764 | // propagate a known value of the condition into its other users. | ||||
2765 | |||||
2766 | // If the select and condition only have a single use, don't bother with this, | ||||
2767 | // early exit. | ||||
2768 | if (SI->use_empty() && SelectCond->hasOneUse()) | ||||
2769 | return true; | ||||
2770 | |||||
2771 | // Scan the current block backward, looking for other uses of SI. | ||||
2772 | BasicBlock::iterator BBI = &I, BBFront = I.getParent()->begin(); | ||||
2773 | |||||
2774 | while (BBI != BBFront) { | ||||
2775 | --BBI; | ||||
2776 | // If we found a call to a function, we can't assume it will return, so | ||||
2777 | // information from below it cannot be propagated above it. | ||||
2778 | if (isa<CallInst>(BBI) && !isa<IntrinsicInst>(BBI)) | ||||
2779 | break; | ||||
2780 | |||||
2781 | // Replace uses of the select or its condition with the known values. | ||||
2782 | for (Instruction::op_iterator I = BBI->op_begin(), E = BBI->op_end(); | ||||
2783 | I != E; ++I) { | ||||
2784 | if (*I == SI) { | ||||
2785 | *I = SI->getOperand(NonNullOperand); | ||||
2786 | AddToWorkList(BBI); | ||||
2787 | } else if (*I == SelectCond) { | ||||
2788 | *I = NonNullOperand == 1 ? ConstantInt::getTrue() : | ||||
2789 | ConstantInt::getFalse(); | ||||
2790 | AddToWorkList(BBI); | ||||
2791 | } | ||||
2792 | } | ||||
2793 | |||||
2794 | // If we past the instruction, quit looking for it. | ||||
2795 | if (&*BBI == SI) | ||||
2796 | SI = 0; | ||||
2797 | if (&*BBI == SelectCond) | ||||
2798 | SelectCond = 0; | ||||
2799 | |||||
2800 | // If we ran out of things to eliminate, break out of the loop. | ||||
2801 | if (SelectCond == 0 && SI == 0) | ||||
2802 | break; | ||||
2803 | |||||
2804 | } | ||||
2805 | return true; | ||||
2806 | } | ||||
2807 | |||||
2808 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2809 | /// This function implements the transforms on div instructions that work |
2810 | /// regardless of the kind of div instruction it is (udiv, sdiv, or fdiv). It is | ||||
2811 | /// used by the visitors to those instructions. | ||||
2812 | /// @brief Transforms common to all three div instructions | ||||
2813 | Instruction *InstCombiner::commonDivTransforms(BinaryOperator &I) { | ||||
2814 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2815 | |||||
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 2816 | // undef / X -> 0 for integer. |
2817 | // undef / X -> undef for FP (the undef could be a snan). | ||||
2818 | if (isa<UndefValue>(Op0)) { | ||||
2819 | if (Op0->getType()->isFPOrFPVector()) | ||||
2820 | return ReplaceInstUsesWith(I, Op0); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2821 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 2822 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2823 | |
2824 | // X / undef -> undef | ||||
2825 | if (isa<UndefValue>(Op1)) | ||||
2826 | return ReplaceInstUsesWith(I, Op1); | ||||
2827 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2828 | return 0; |
2829 | } | ||||
2830 | |||||
2831 | /// This function implements the transforms common to both integer division | ||||
2832 | /// instructions (udiv and sdiv). It is called by the visitors to those integer | ||||
2833 | /// division instructions. | ||||
2834 | /// @brief Common integer divide transforms | ||||
2835 | Instruction *InstCombiner::commonIDivTransforms(BinaryOperator &I) { | ||||
2836 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2837 | |||||
Chris Lattner | cefb36c | 2008-05-16 02:59:42 +0000 | [diff] [blame] | 2838 | // (sdiv X, X) --> 1 (udiv X, X) --> 1 |
Nick Lewycky | 386c013 | 2008-05-23 03:26:47 +0000 | [diff] [blame] | 2839 | if (Op0 == Op1) { |
2840 | if (const VectorType *Ty = dyn_cast<VectorType>(I.getType())) { | ||||
2841 | ConstantInt *CI = ConstantInt::get(Ty->getElementType(), 1); | ||||
2842 | std::vector<Constant*> Elts(Ty->getNumElements(), CI); | ||||
2843 | return ReplaceInstUsesWith(I, ConstantVector::get(Elts)); | ||||
2844 | } | ||||
2845 | |||||
2846 | ConstantInt *CI = ConstantInt::get(I.getType(), 1); | ||||
2847 | return ReplaceInstUsesWith(I, CI); | ||||
2848 | } | ||||
Chris Lattner | cefb36c | 2008-05-16 02:59:42 +0000 | [diff] [blame] | 2849 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2850 | if (Instruction *Common = commonDivTransforms(I)) |
2851 | return Common; | ||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 2852 | |
2853 | // Handle cases involving: [su]div X, (select Cond, Y, Z) | ||||
2854 | // This does not apply for fdiv. | ||||
2855 | if (isa<SelectInst>(Op1) && SimplifyDivRemOfSelect(I)) | ||||
2856 | return &I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2857 | |
2858 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
2859 | // div X, 1 == X | ||||
2860 | if (RHS->equalsInt(1)) | ||||
2861 | return ReplaceInstUsesWith(I, Op0); | ||||
2862 | |||||
2863 | // (X / C1) / C2 -> X / (C1*C2) | ||||
2864 | if (Instruction *LHS = dyn_cast<Instruction>(Op0)) | ||||
2865 | if (Instruction::BinaryOps(LHS->getOpcode()) == I.getOpcode()) | ||||
2866 | if (ConstantInt *LHSRHS = dyn_cast<ConstantInt>(LHS->getOperand(1))) { | ||||
Nick Lewycky | 9d798f9 | 2008-02-18 22:48:05 +0000 | [diff] [blame] | 2867 | if (MultiplyOverflows(RHS, LHSRHS, I.getOpcode()==Instruction::SDiv)) |
2868 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2869 | else | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2870 | return BinaryOperator::Create(I.getOpcode(), LHS->getOperand(0), |
Nick Lewycky | 9d798f9 | 2008-02-18 22:48:05 +0000 | [diff] [blame] | 2871 | Multiply(RHS, LHSRHS)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2872 | } |
2873 | |||||
2874 | if (!RHS->isZero()) { // avoid X udiv 0 | ||||
2875 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
2876 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2877 | return R; | ||||
2878 | if (isa<PHINode>(Op0)) | ||||
2879 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2880 | return NV; | ||||
2881 | } | ||||
2882 | } | ||||
2883 | |||||
2884 | // 0 / X == 0, we don't need to preserve faults! | ||||
2885 | if (ConstantInt *LHS = dyn_cast<ConstantInt>(Op0)) | ||||
2886 | if (LHS->equalsInt(0)) | ||||
2887 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2888 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2889 | // It can't be division by zero, hence it must be division by one. |
2890 | if (I.getType() == Type::Int1Ty) | ||||
2891 | return ReplaceInstUsesWith(I, Op0); | ||||
2892 | |||||
Nick Lewycky | 9441873 | 2008-11-27 20:21:08 +0000 | [diff] [blame] | 2893 | if (ConstantVector *Op1V = dyn_cast<ConstantVector>(Op1)) { |
2894 | if (ConstantInt *X = cast_or_null<ConstantInt>(Op1V->getSplatValue())) | ||||
2895 | // div X, 1 == X | ||||
2896 | if (X->isOne()) | ||||
2897 | return ReplaceInstUsesWith(I, Op0); | ||||
2898 | } | ||||
2899 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2900 | return 0; |
2901 | } | ||||
2902 | |||||
2903 | Instruction *InstCombiner::visitUDiv(BinaryOperator &I) { | ||||
2904 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2905 | |||||
2906 | // Handle the integer div common cases | ||||
2907 | if (Instruction *Common = commonIDivTransforms(I)) | ||||
2908 | return Common; | ||||
2909 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2910 | if (ConstantInt *C = dyn_cast<ConstantInt>(Op1)) { |
Nick Lewycky | 240182a | 2008-11-27 22:41:10 +0000 | [diff] [blame] | 2911 | // X udiv C^2 -> X >> C |
2912 | // Check to see if this is an unsigned division with an exact power of 2, | ||||
2913 | // if so, convert to a right shift. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2914 | if (C->getValue().isPowerOf2()) // 0 not included in isPowerOf2 |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2915 | return BinaryOperator::CreateLShr(Op0, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2916 | ConstantInt::get(Op0->getType(), C->getValue().logBase2())); |
Nick Lewycky | 240182a | 2008-11-27 22:41:10 +0000 | [diff] [blame] | 2917 | |
2918 | // X udiv C, where C >= signbit | ||||
2919 | if (C->getValue().isNegative()) { | ||||
2920 | Value *IC = InsertNewInstBefore(new ICmpInst(ICmpInst::ICMP_ULT, Op0, C), | ||||
2921 | I); | ||||
2922 | return SelectInst::Create(IC, Constant::getNullValue(I.getType()), | ||||
2923 | ConstantInt::get(I.getType(), 1)); | ||||
2924 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2925 | } |
2926 | |||||
2927 | // X udiv (C1 << N), where C1 is "1<<C2" --> X >> (N+C2) | ||||
2928 | if (BinaryOperator *RHSI = dyn_cast<BinaryOperator>(I.getOperand(1))) { | ||||
2929 | if (RHSI->getOpcode() == Instruction::Shl && | ||||
2930 | isa<ConstantInt>(RHSI->getOperand(0))) { | ||||
2931 | const APInt& C1 = cast<ConstantInt>(RHSI->getOperand(0))->getValue(); | ||||
2932 | if (C1.isPowerOf2()) { | ||||
2933 | Value *N = RHSI->getOperand(1); | ||||
2934 | const Type *NTy = N->getType(); | ||||
2935 | if (uint32_t C2 = C1.logBase2()) { | ||||
2936 | Constant *C2V = ConstantInt::get(NTy, C2); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2937 | N = InsertNewInstBefore(BinaryOperator::CreateAdd(N, C2V, "tmp"), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2938 | } |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2939 | return BinaryOperator::CreateLShr(Op0, N); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2940 | } |
2941 | } | ||||
2942 | } | ||||
2943 | |||||
2944 | // udiv X, (Select Cond, C1, C2) --> Select Cond, (shr X, C1), (shr X, C2) | ||||
2945 | // where C1&C2 are powers of two. | ||||
2946 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) | ||||
2947 | if (ConstantInt *STO = dyn_cast<ConstantInt>(SI->getOperand(1))) | ||||
2948 | if (ConstantInt *SFO = dyn_cast<ConstantInt>(SI->getOperand(2))) { | ||||
2949 | const APInt &TVA = STO->getValue(), &FVA = SFO->getValue(); | ||||
2950 | if (TVA.isPowerOf2() && FVA.isPowerOf2()) { | ||||
2951 | // Compute the shift amounts | ||||
2952 | uint32_t TSA = TVA.logBase2(), FSA = FVA.logBase2(); | ||||
2953 | // Construct the "on true" case of the select | ||||
2954 | Constant *TC = ConstantInt::get(Op0->getType(), TSA); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2955 | Instruction *TSI = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2956 | Op0, TC, SI->getName()+".t"); |
2957 | TSI = InsertNewInstBefore(TSI, I); | ||||
2958 | |||||
2959 | // Construct the "on false" case of the select | ||||
2960 | Constant *FC = ConstantInt::get(Op0->getType(), FSA); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2961 | Instruction *FSI = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2962 | Op0, FC, SI->getName()+".f"); |
2963 | FSI = InsertNewInstBefore(FSI, I); | ||||
2964 | |||||
2965 | // construct the select instruction and return it. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2966 | return SelectInst::Create(SI->getOperand(0), TSI, FSI, SI->getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2967 | } |
2968 | } | ||||
2969 | return 0; | ||||
2970 | } | ||||
2971 | |||||
2972 | Instruction *InstCombiner::visitSDiv(BinaryOperator &I) { | ||||
2973 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2974 | |||||
2975 | // Handle the integer div common cases | ||||
2976 | if (Instruction *Common = commonIDivTransforms(I)) | ||||
2977 | return Common; | ||||
2978 | |||||
2979 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
2980 | // sdiv X, -1 == -X | ||||
2981 | if (RHS->isAllOnesValue()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2982 | return BinaryOperator::CreateNeg(Op0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2983 | } |
2984 | |||||
2985 | // If the sign bits of both operands are zero (i.e. we can prove they are | ||||
2986 | // unsigned inputs), turn this into a udiv. | ||||
2987 | if (I.getType()->isInteger()) { | ||||
2988 | APInt Mask(APInt::getSignBit(I.getType()->getPrimitiveSizeInBits())); | ||||
2989 | if (MaskedValueIsZero(Op1, Mask) && MaskedValueIsZero(Op0, Mask)) { | ||||
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 2990 | // X sdiv Y -> X udiv Y, iff X and Y don't have sign bit set |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2991 | return BinaryOperator::CreateUDiv(Op0, Op1, I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2992 | } |
2993 | } | ||||
2994 | |||||
2995 | return 0; | ||||
2996 | } | ||||
2997 | |||||
2998 | Instruction *InstCombiner::visitFDiv(BinaryOperator &I) { | ||||
2999 | return commonDivTransforms(I); | ||||
3000 | } | ||||
3001 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3002 | /// This function implements the transforms on rem instructions that work |
3003 | /// regardless of the kind of rem instruction it is (urem, srem, or frem). It | ||||
3004 | /// is used by the visitors to those instructions. | ||||
3005 | /// @brief Transforms common to all three rem instructions | ||||
3006 | Instruction *InstCombiner::commonRemTransforms(BinaryOperator &I) { | ||||
3007 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3008 | |||||
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 3009 | if (isa<UndefValue>(Op0)) { // undef % X -> 0 |
3010 | if (I.getType()->isFPOrFPVector()) | ||||
3011 | return ReplaceInstUsesWith(I, Op0); // X % undef -> undef (could be SNaN) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3012 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 3013 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3014 | if (isa<UndefValue>(Op1)) |
3015 | return ReplaceInstUsesWith(I, Op1); // X % undef -> undef | ||||
3016 | |||||
3017 | // Handle cases involving: rem X, (select Cond, Y, Z) | ||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 3018 | if (isa<SelectInst>(Op1) && SimplifyDivRemOfSelect(I)) |
3019 | return &I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3020 | |
3021 | return 0; | ||||
3022 | } | ||||
3023 | |||||
3024 | /// This function implements the transforms common to both integer remainder | ||||
3025 | /// instructions (urem and srem). It is called by the visitors to those integer | ||||
3026 | /// remainder instructions. | ||||
3027 | /// @brief Common integer remainder transforms | ||||
3028 | Instruction *InstCombiner::commonIRemTransforms(BinaryOperator &I) { | ||||
3029 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3030 | |||||
3031 | if (Instruction *common = commonRemTransforms(I)) | ||||
3032 | return common; | ||||
3033 | |||||
Dale Johannesen | a51f737 | 2009-01-21 00:35:19 +0000 | [diff] [blame] | 3034 | // 0 % X == 0 for integer, we don't need to preserve faults! |
3035 | if (Constant *LHS = dyn_cast<Constant>(Op0)) | ||||
3036 | if (LHS->isNullValue()) | ||||
3037 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
3038 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3039 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { |
3040 | // X % 0 == undef, we don't need to preserve faults! | ||||
3041 | if (RHS->equalsInt(0)) | ||||
3042 | return ReplaceInstUsesWith(I, UndefValue::get(I.getType())); | ||||
3043 | |||||
3044 | if (RHS->equalsInt(1)) // X % 1 == 0 | ||||
3045 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
3046 | |||||
3047 | if (Instruction *Op0I = dyn_cast<Instruction>(Op0)) { | ||||
3048 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0I)) { | ||||
3049 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
3050 | return R; | ||||
3051 | } else if (isa<PHINode>(Op0I)) { | ||||
3052 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
3053 | return NV; | ||||
3054 | } | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 3055 | |
3056 | // See if we can fold away this rem instruction. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 3057 | if (SimplifyDemandedInstructionBits(I)) |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 3058 | return &I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3059 | } |
3060 | } | ||||
3061 | |||||
3062 | return 0; | ||||
3063 | } | ||||
3064 | |||||
3065 | Instruction *InstCombiner::visitURem(BinaryOperator &I) { | ||||
3066 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3067 | |||||
3068 | if (Instruction *common = commonIRemTransforms(I)) | ||||
3069 | return common; | ||||
3070 | |||||
3071 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
3072 | // X urem C^2 -> X and C | ||||
3073 | // Check to see if this is an unsigned remainder with an exact power of 2, | ||||
3074 | // if so, convert to a bitwise and. | ||||
3075 | if (ConstantInt *C = dyn_cast<ConstantInt>(RHS)) | ||||
3076 | if (C->getValue().isPowerOf2()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3077 | return BinaryOperator::CreateAnd(Op0, SubOne(C)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3078 | } |
3079 | |||||
3080 | if (Instruction *RHSI = dyn_cast<Instruction>(I.getOperand(1))) { | ||||
3081 | // Turn A % (C << N), where C is 2^k, into A & ((C << N)-1) | ||||
3082 | if (RHSI->getOpcode() == Instruction::Shl && | ||||
3083 | isa<ConstantInt>(RHSI->getOperand(0))) { | ||||
3084 | if (cast<ConstantInt>(RHSI->getOperand(0))->getValue().isPowerOf2()) { | ||||
3085 | Constant *N1 = ConstantInt::getAllOnesValue(I.getType()); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3086 | Value *Add = InsertNewInstBefore(BinaryOperator::CreateAdd(RHSI, N1, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3087 | "tmp"), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3088 | return BinaryOperator::CreateAnd(Op0, Add); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3089 | } |
3090 | } | ||||
3091 | } | ||||
3092 | |||||
3093 | // urem X, (select Cond, 2^C1, 2^C2) --> select Cond, (and X, C1), (and X, C2) | ||||
3094 | // where C1&C2 are powers of two. | ||||
3095 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) { | ||||
3096 | if (ConstantInt *STO = dyn_cast<ConstantInt>(SI->getOperand(1))) | ||||
3097 | if (ConstantInt *SFO = dyn_cast<ConstantInt>(SI->getOperand(2))) { | ||||
3098 | // STO == 0 and SFO == 0 handled above. | ||||
3099 | if ((STO->getValue().isPowerOf2()) && | ||||
3100 | (SFO->getValue().isPowerOf2())) { | ||||
3101 | Value *TrueAnd = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3102 | BinaryOperator::CreateAnd(Op0, SubOne(STO), SI->getName()+".t"), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3103 | Value *FalseAnd = InsertNewInstBefore( |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3104 | BinaryOperator::CreateAnd(Op0, SubOne(SFO), SI->getName()+".f"), I); |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 3105 | return SelectInst::Create(SI->getOperand(0), TrueAnd, FalseAnd); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3106 | } |
3107 | } | ||||
3108 | } | ||||
3109 | |||||
3110 | return 0; | ||||
3111 | } | ||||
3112 | |||||
3113 | Instruction *InstCombiner::visitSRem(BinaryOperator &I) { | ||||
3114 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3115 | |||||
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3116 | // Handle the integer rem common cases |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3117 | if (Instruction *common = commonIRemTransforms(I)) |
3118 | return common; | ||||
3119 | |||||
3120 | if (Value *RHSNeg = dyn_castNegVal(Op1)) | ||||
Nick Lewycky | cfadfbd | 2008-09-03 06:24:21 +0000 | [diff] [blame] | 3121 | if (!isa<Constant>(RHSNeg) || |
3122 | (isa<ConstantInt>(RHSNeg) && | ||||
3123 | cast<ConstantInt>(RHSNeg)->getValue().isStrictlyPositive())) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3124 | // X % -Y -> X % Y |
3125 | AddUsesToWorkList(I); | ||||
3126 | I.setOperand(1, RHSNeg); | ||||
3127 | return &I; | ||||
3128 | } | ||||
Nick Lewycky | 5515c7a | 2008-09-30 06:08:34 +0000 | [diff] [blame] | 3129 | |
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3130 | // If the sign bits of both operands are zero (i.e. we can prove they are |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3131 | // unsigned inputs), turn this into a urem. |
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3132 | if (I.getType()->isInteger()) { |
3133 | APInt Mask(APInt::getSignBit(I.getType()->getPrimitiveSizeInBits())); | ||||
3134 | if (MaskedValueIsZero(Op1, Mask) && MaskedValueIsZero(Op0, Mask)) { | ||||
3135 | // X srem Y -> X urem Y, iff X and Y don't have sign bit set | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3136 | return BinaryOperator::CreateURem(Op0, Op1, I.getName()); |
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3137 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3138 | } |
3139 | |||||
Nick Lewycky | da9fa43 | 2008-12-18 06:31:11 +0000 | [diff] [blame] | 3140 | // If it's a constant vector, flip any negative values positive. |
Nick Lewycky | fd74683 | 2008-12-20 16:48:00 +0000 | [diff] [blame] | 3141 | if (ConstantVector *RHSV = dyn_cast<ConstantVector>(Op1)) { |
3142 | unsigned VWidth = RHSV->getNumOperands(); | ||||
Nick Lewycky | da9fa43 | 2008-12-18 06:31:11 +0000 | [diff] [blame] | 3143 | |
Nick Lewycky | fd74683 | 2008-12-20 16:48:00 +0000 | [diff] [blame] | 3144 | bool hasNegative = false; |
3145 | for (unsigned i = 0; !hasNegative && i != VWidth; ++i) | ||||
3146 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(RHSV->getOperand(i))) | ||||
3147 | if (RHS->getValue().isNegative()) | ||||
3148 | hasNegative = true; | ||||
3149 | |||||
3150 | if (hasNegative) { | ||||
3151 | std::vector<Constant *> Elts(VWidth); | ||||
Nick Lewycky | da9fa43 | 2008-12-18 06:31:11 +0000 | [diff] [blame] | 3152 | for (unsigned i = 0; i != VWidth; ++i) { |
3153 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(RHSV->getOperand(i))) { | ||||
3154 | if (RHS->getValue().isNegative()) | ||||
3155 | Elts[i] = cast<ConstantInt>(ConstantExpr::getNeg(RHS)); | ||||
3156 | else | ||||
3157 | Elts[i] = RHS; | ||||
3158 | } | ||||
3159 | } | ||||
3160 | |||||
3161 | Constant *NewRHSV = ConstantVector::get(Elts); | ||||
3162 | if (NewRHSV != RHSV) { | ||||
Nick Lewycky | 338ecd5 | 2008-12-18 06:42:28 +0000 | [diff] [blame] | 3163 | AddUsesToWorkList(I); |
Nick Lewycky | da9fa43 | 2008-12-18 06:31:11 +0000 | [diff] [blame] | 3164 | I.setOperand(1, NewRHSV); |
3165 | return &I; | ||||
3166 | } | ||||
3167 | } | ||||
3168 | } | ||||
3169 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3170 | return 0; |
3171 | } | ||||
3172 | |||||
3173 | Instruction *InstCombiner::visitFRem(BinaryOperator &I) { | ||||
3174 | return commonRemTransforms(I); | ||||
3175 | } | ||||
3176 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3177 | // isOneBitSet - Return true if there is exactly one bit set in the specified |
3178 | // constant. | ||||
3179 | static bool isOneBitSet(const ConstantInt *CI) { | ||||
3180 | return CI->getValue().isPowerOf2(); | ||||
3181 | } | ||||
3182 | |||||
3183 | // isHighOnes - Return true if the constant is of the form 1+0+. | ||||
3184 | // This is the same as lowones(~X). | ||||
3185 | static bool isHighOnes(const ConstantInt *CI) { | ||||
3186 | return (~CI->getValue() + 1).isPowerOf2(); | ||||
3187 | } | ||||
3188 | |||||
3189 | /// getICmpCode - Encode a icmp predicate into a three bit mask. These bits | ||||
3190 | /// are carefully arranged to allow folding of expressions such as: | ||||
3191 | /// | ||||
3192 | /// (A < B) | (A > B) --> (A != B) | ||||
3193 | /// | ||||
3194 | /// Note that this is only valid if the first and second predicates have the | ||||
3195 | /// same sign. Is illegal to do: (A u< B) | (A s> B) | ||||
3196 | /// | ||||
3197 | /// Three bits are used to represent the condition, as follows: | ||||
3198 | /// 0 A > B | ||||
3199 | /// 1 A == B | ||||
3200 | /// 2 A < B | ||||
3201 | /// | ||||
3202 | /// <=> Value Definition | ||||
3203 | /// 000 0 Always false | ||||
3204 | /// 001 1 A > B | ||||
3205 | /// 010 2 A == B | ||||
3206 | /// 011 3 A >= B | ||||
3207 | /// 100 4 A < B | ||||
3208 | /// 101 5 A != B | ||||
3209 | /// 110 6 A <= B | ||||
3210 | /// 111 7 Always true | ||||
3211 | /// | ||||
3212 | static unsigned getICmpCode(const ICmpInst *ICI) { | ||||
3213 | switch (ICI->getPredicate()) { | ||||
3214 | // False -> 0 | ||||
3215 | case ICmpInst::ICMP_UGT: return 1; // 001 | ||||
3216 | case ICmpInst::ICMP_SGT: return 1; // 001 | ||||
3217 | case ICmpInst::ICMP_EQ: return 2; // 010 | ||||
3218 | case ICmpInst::ICMP_UGE: return 3; // 011 | ||||
3219 | case ICmpInst::ICMP_SGE: return 3; // 011 | ||||
3220 | case ICmpInst::ICMP_ULT: return 4; // 100 | ||||
3221 | case ICmpInst::ICMP_SLT: return 4; // 100 | ||||
3222 | case ICmpInst::ICMP_NE: return 5; // 101 | ||||
3223 | case ICmpInst::ICMP_ULE: return 6; // 110 | ||||
3224 | case ICmpInst::ICMP_SLE: return 6; // 110 | ||||
3225 | // True -> 7 | ||||
3226 | default: | ||||
3227 | assert(0 && "Invalid ICmp predicate!"); | ||||
3228 | return 0; | ||||
3229 | } | ||||
3230 | } | ||||
3231 | |||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3232 | /// getFCmpCode - Similar to getICmpCode but for FCmpInst. This encodes a fcmp |
3233 | /// predicate into a three bit mask. It also returns whether it is an ordered | ||||
3234 | /// predicate by reference. | ||||
3235 | static unsigned getFCmpCode(FCmpInst::Predicate CC, bool &isOrdered) { | ||||
3236 | isOrdered = false; | ||||
3237 | switch (CC) { | ||||
3238 | case FCmpInst::FCMP_ORD: isOrdered = true; return 0; // 000 | ||||
3239 | case FCmpInst::FCMP_UNO: return 0; // 000 | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 3240 | case FCmpInst::FCMP_OGT: isOrdered = true; return 1; // 001 |
3241 | case FCmpInst::FCMP_UGT: return 1; // 001 | ||||
3242 | case FCmpInst::FCMP_OEQ: isOrdered = true; return 2; // 010 | ||||
3243 | case FCmpInst::FCMP_UEQ: return 2; // 010 | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3244 | case FCmpInst::FCMP_OGE: isOrdered = true; return 3; // 011 |
3245 | case FCmpInst::FCMP_UGE: return 3; // 011 | ||||
3246 | case FCmpInst::FCMP_OLT: isOrdered = true; return 4; // 100 | ||||
3247 | case FCmpInst::FCMP_ULT: return 4; // 100 | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 3248 | case FCmpInst::FCMP_ONE: isOrdered = true; return 5; // 101 |
3249 | case FCmpInst::FCMP_UNE: return 5; // 101 | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3250 | case FCmpInst::FCMP_OLE: isOrdered = true; return 6; // 110 |
3251 | case FCmpInst::FCMP_ULE: return 6; // 110 | ||||
Evan Cheng | 7298805 | 2008-10-14 18:44:08 +0000 | [diff] [blame] | 3252 | // True -> 7 |
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3253 | default: |
3254 | // Not expecting FCMP_FALSE and FCMP_TRUE; | ||||
3255 | assert(0 && "Unexpected FCmp predicate!"); | ||||
3256 | return 0; | ||||
3257 | } | ||||
3258 | } | ||||
3259 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3260 | /// getICmpValue - This is the complement of getICmpCode, which turns an |
3261 | /// opcode and two operands into either a constant true or false, or a brand | ||||
Dan Gohman | da33874 | 2007-09-17 17:31:57 +0000 | [diff] [blame] | 3262 | /// new ICmp instruction. The sign is passed in to determine which kind |
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3263 | /// of predicate to use in the new icmp instruction. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3264 | static Value *getICmpValue(bool sign, unsigned code, Value *LHS, Value *RHS) { |
3265 | switch (code) { | ||||
3266 | default: assert(0 && "Illegal ICmp code!"); | ||||
3267 | case 0: return ConstantInt::getFalse(); | ||||
3268 | case 1: | ||||
3269 | if (sign) | ||||
3270 | return new ICmpInst(ICmpInst::ICMP_SGT, LHS, RHS); | ||||
3271 | else | ||||
3272 | return new ICmpInst(ICmpInst::ICMP_UGT, LHS, RHS); | ||||
3273 | case 2: return new ICmpInst(ICmpInst::ICMP_EQ, LHS, RHS); | ||||
3274 | case 3: | ||||
3275 | if (sign) | ||||
3276 | return new ICmpInst(ICmpInst::ICMP_SGE, LHS, RHS); | ||||
3277 | else | ||||
3278 | return new ICmpInst(ICmpInst::ICMP_UGE, LHS, RHS); | ||||
3279 | case 4: | ||||
3280 | if (sign) | ||||
3281 | return new ICmpInst(ICmpInst::ICMP_SLT, LHS, RHS); | ||||
3282 | else | ||||
3283 | return new ICmpInst(ICmpInst::ICMP_ULT, LHS, RHS); | ||||
3284 | case 5: return new ICmpInst(ICmpInst::ICMP_NE, LHS, RHS); | ||||
3285 | case 6: | ||||
3286 | if (sign) | ||||
3287 | return new ICmpInst(ICmpInst::ICMP_SLE, LHS, RHS); | ||||
3288 | else | ||||
3289 | return new ICmpInst(ICmpInst::ICMP_ULE, LHS, RHS); | ||||
3290 | case 7: return ConstantInt::getTrue(); | ||||
3291 | } | ||||
3292 | } | ||||
3293 | |||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3294 | /// getFCmpValue - This is the complement of getFCmpCode, which turns an |
3295 | /// opcode and two operands into either a FCmp instruction. isordered is passed | ||||
3296 | /// in to determine which kind of predicate to use in the new fcmp instruction. | ||||
3297 | static Value *getFCmpValue(bool isordered, unsigned code, | ||||
3298 | Value *LHS, Value *RHS) { | ||||
3299 | switch (code) { | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 3300 | default: assert(0 && "Illegal FCmp code!"); |
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3301 | case 0: |
3302 | if (isordered) | ||||
3303 | return new FCmpInst(FCmpInst::FCMP_ORD, LHS, RHS); | ||||
3304 | else | ||||
3305 | return new FCmpInst(FCmpInst::FCMP_UNO, LHS, RHS); | ||||
3306 | case 1: | ||||
3307 | if (isordered) | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3308 | return new FCmpInst(FCmpInst::FCMP_OGT, LHS, RHS); |
3309 | else | ||||
3310 | return new FCmpInst(FCmpInst::FCMP_UGT, LHS, RHS); | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 3311 | case 2: |
3312 | if (isordered) | ||||
3313 | return new FCmpInst(FCmpInst::FCMP_OEQ, LHS, RHS); | ||||
3314 | else | ||||
3315 | return new FCmpInst(FCmpInst::FCMP_UEQ, LHS, RHS); | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3316 | case 3: |
3317 | if (isordered) | ||||
3318 | return new FCmpInst(FCmpInst::FCMP_OGE, LHS, RHS); | ||||
3319 | else | ||||
3320 | return new FCmpInst(FCmpInst::FCMP_UGE, LHS, RHS); | ||||
3321 | case 4: | ||||
3322 | if (isordered) | ||||
3323 | return new FCmpInst(FCmpInst::FCMP_OLT, LHS, RHS); | ||||
3324 | else | ||||
3325 | return new FCmpInst(FCmpInst::FCMP_ULT, LHS, RHS); | ||||
3326 | case 5: | ||||
3327 | if (isordered) | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 3328 | return new FCmpInst(FCmpInst::FCMP_ONE, LHS, RHS); |
3329 | else | ||||
3330 | return new FCmpInst(FCmpInst::FCMP_UNE, LHS, RHS); | ||||
3331 | case 6: | ||||
3332 | if (isordered) | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3333 | return new FCmpInst(FCmpInst::FCMP_OLE, LHS, RHS); |
3334 | else | ||||
3335 | return new FCmpInst(FCmpInst::FCMP_ULE, LHS, RHS); | ||||
Evan Cheng | 7298805 | 2008-10-14 18:44:08 +0000 | [diff] [blame] | 3336 | case 7: return ConstantInt::getTrue(); |
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3337 | } |
3338 | } | ||||
3339 | |||||
Chris Lattner | 2972b82 | 2008-11-16 04:55:20 +0000 | [diff] [blame] | 3340 | /// PredicatesFoldable - Return true if both predicates match sign or if at |
3341 | /// least one of them is an equality comparison (which is signless). | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3342 | static bool PredicatesFoldable(ICmpInst::Predicate p1, ICmpInst::Predicate p2) { |
3343 | return (ICmpInst::isSignedPredicate(p1) == ICmpInst::isSignedPredicate(p2)) || | ||||
Chris Lattner | 2972b82 | 2008-11-16 04:55:20 +0000 | [diff] [blame] | 3344 | (ICmpInst::isSignedPredicate(p1) && ICmpInst::isEquality(p2)) || |
3345 | (ICmpInst::isSignedPredicate(p2) && ICmpInst::isEquality(p1)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3346 | } |
3347 | |||||
3348 | namespace { | ||||
3349 | // FoldICmpLogical - Implements (icmp1 A, B) & (icmp2 A, B) --> (icmp3 A, B) | ||||
3350 | struct FoldICmpLogical { | ||||
3351 | InstCombiner &IC; | ||||
3352 | Value *LHS, *RHS; | ||||
3353 | ICmpInst::Predicate pred; | ||||
3354 | FoldICmpLogical(InstCombiner &ic, ICmpInst *ICI) | ||||
3355 | : IC(ic), LHS(ICI->getOperand(0)), RHS(ICI->getOperand(1)), | ||||
3356 | pred(ICI->getPredicate()) {} | ||||
3357 | bool shouldApply(Value *V) const { | ||||
3358 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(V)) | ||||
3359 | if (PredicatesFoldable(pred, ICI->getPredicate())) | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 3360 | return ((ICI->getOperand(0) == LHS && ICI->getOperand(1) == RHS) || |
3361 | (ICI->getOperand(0) == RHS && ICI->getOperand(1) == LHS)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3362 | return false; |
3363 | } | ||||
3364 | Instruction *apply(Instruction &Log) const { | ||||
3365 | ICmpInst *ICI = cast<ICmpInst>(Log.getOperand(0)); | ||||
3366 | if (ICI->getOperand(0) != LHS) { | ||||
3367 | assert(ICI->getOperand(1) == LHS); | ||||
3368 | ICI->swapOperands(); // Swap the LHS and RHS of the ICmp | ||||
3369 | } | ||||
3370 | |||||
3371 | ICmpInst *RHSICI = cast<ICmpInst>(Log.getOperand(1)); | ||||
3372 | unsigned LHSCode = getICmpCode(ICI); | ||||
3373 | unsigned RHSCode = getICmpCode(RHSICI); | ||||
3374 | unsigned Code; | ||||
3375 | switch (Log.getOpcode()) { | ||||
3376 | case Instruction::And: Code = LHSCode & RHSCode; break; | ||||
3377 | case Instruction::Or: Code = LHSCode | RHSCode; break; | ||||
3378 | case Instruction::Xor: Code = LHSCode ^ RHSCode; break; | ||||
3379 | default: assert(0 && "Illegal logical opcode!"); return 0; | ||||
3380 | } | ||||
3381 | |||||
3382 | bool isSigned = ICmpInst::isSignedPredicate(RHSICI->getPredicate()) || | ||||
3383 | ICmpInst::isSignedPredicate(ICI->getPredicate()); | ||||
3384 | |||||
3385 | Value *RV = getICmpValue(isSigned, Code, LHS, RHS); | ||||
3386 | if (Instruction *I = dyn_cast<Instruction>(RV)) | ||||
3387 | return I; | ||||
3388 | // Otherwise, it's a constant boolean value... | ||||
3389 | return IC.ReplaceInstUsesWith(Log, RV); | ||||
3390 | } | ||||
3391 | }; | ||||
3392 | } // end anonymous namespace | ||||
3393 | |||||
3394 | // OptAndOp - This handles expressions of the form ((val OP C1) & C2). Where | ||||
3395 | // the Op parameter is 'OP', OpRHS is 'C1', and AndRHS is 'C2'. Op is | ||||
3396 | // guaranteed to be a binary operator. | ||||
3397 | Instruction *InstCombiner::OptAndOp(Instruction *Op, | ||||
3398 | ConstantInt *OpRHS, | ||||
3399 | ConstantInt *AndRHS, | ||||
3400 | BinaryOperator &TheAnd) { | ||||
3401 | Value *X = Op->getOperand(0); | ||||
3402 | Constant *Together = 0; | ||||
3403 | if (!Op->isShift()) | ||||
3404 | Together = And(AndRHS, OpRHS); | ||||
3405 | |||||
3406 | switch (Op->getOpcode()) { | ||||
3407 | case Instruction::Xor: | ||||
3408 | if (Op->hasOneUse()) { | ||||
3409 | // (X ^ C1) & C2 --> (X & C2) ^ (C1&C2) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3410 | Instruction *And = BinaryOperator::CreateAnd(X, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3411 | InsertNewInstBefore(And, TheAnd); |
3412 | And->takeName(Op); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3413 | return BinaryOperator::CreateXor(And, Together); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3414 | } |
3415 | break; | ||||
3416 | case Instruction::Or: | ||||
3417 | if (Together == AndRHS) // (X | C) & C --> C | ||||
3418 | return ReplaceInstUsesWith(TheAnd, AndRHS); | ||||
3419 | |||||
3420 | if (Op->hasOneUse() && Together != OpRHS) { | ||||
3421 | // (X | C1) & C2 --> (X | (C1&C2)) & C2 | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3422 | Instruction *Or = BinaryOperator::CreateOr(X, Together); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3423 | InsertNewInstBefore(Or, TheAnd); |
3424 | Or->takeName(Op); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3425 | return BinaryOperator::CreateAnd(Or, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3426 | } |
3427 | break; | ||||
3428 | case Instruction::Add: | ||||
3429 | if (Op->hasOneUse()) { | ||||
3430 | // Adding a one to a single bit bit-field should be turned into an XOR | ||||
3431 | // of the bit. First thing to check is to see if this AND is with a | ||||
3432 | // single bit constant. | ||||
3433 | const APInt& AndRHSV = cast<ConstantInt>(AndRHS)->getValue(); | ||||
3434 | |||||
3435 | // If there is only one bit set... | ||||
3436 | if (isOneBitSet(cast<ConstantInt>(AndRHS))) { | ||||
3437 | // Ok, at this point, we know that we are masking the result of the | ||||
3438 | // ADD down to exactly one bit. If the constant we are adding has | ||||
3439 | // no bits set below this bit, then we can eliminate the ADD. | ||||
3440 | const APInt& AddRHS = cast<ConstantInt>(OpRHS)->getValue(); | ||||
3441 | |||||
3442 | // Check to see if any bits below the one bit set in AndRHSV are set. | ||||
3443 | if ((AddRHS & (AndRHSV-1)) == 0) { | ||||
3444 | // If not, the only thing that can effect the output of the AND is | ||||
3445 | // the bit specified by AndRHSV. If that bit is set, the effect of | ||||
3446 | // the XOR is to toggle the bit. If it is clear, then the ADD has | ||||
3447 | // no effect. | ||||
3448 | if ((AddRHS & AndRHSV) == 0) { // Bit is not set, noop | ||||
3449 | TheAnd.setOperand(0, X); | ||||
3450 | return &TheAnd; | ||||
3451 | } else { | ||||
3452 | // Pull the XOR out of the AND. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3453 | Instruction *NewAnd = BinaryOperator::CreateAnd(X, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3454 | InsertNewInstBefore(NewAnd, TheAnd); |
3455 | NewAnd->takeName(Op); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3456 | return BinaryOperator::CreateXor(NewAnd, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3457 | } |
3458 | } | ||||
3459 | } | ||||
3460 | } | ||||
3461 | break; | ||||
3462 | |||||
3463 | case Instruction::Shl: { | ||||
3464 | // We know that the AND will not produce any of the bits shifted in, so if | ||||
3465 | // the anded constant includes them, clear them now! | ||||
3466 | // | ||||
3467 | uint32_t BitWidth = AndRHS->getType()->getBitWidth(); | ||||
3468 | uint32_t OpRHSVal = OpRHS->getLimitedValue(BitWidth); | ||||
3469 | APInt ShlMask(APInt::getHighBitsSet(BitWidth, BitWidth-OpRHSVal)); | ||||
3470 | ConstantInt *CI = ConstantInt::get(AndRHS->getValue() & ShlMask); | ||||
3471 | |||||
3472 | if (CI->getValue() == ShlMask) { | ||||
3473 | // Masking out bits that the shift already masks | ||||
3474 | return ReplaceInstUsesWith(TheAnd, Op); // No need for the and. | ||||
3475 | } else if (CI != AndRHS) { // Reducing bits set in and. | ||||
3476 | TheAnd.setOperand(1, CI); | ||||
3477 | return &TheAnd; | ||||
3478 | } | ||||
3479 | break; | ||||
3480 | } | ||||
3481 | case Instruction::LShr: | ||||
3482 | { | ||||
3483 | // We know that the AND will not produce any of the bits shifted in, so if | ||||
3484 | // the anded constant includes them, clear them now! This only applies to | ||||
3485 | // unsigned shifts, because a signed shr may bring in set bits! | ||||
3486 | // | ||||
3487 | uint32_t BitWidth = AndRHS->getType()->getBitWidth(); | ||||
3488 | uint32_t OpRHSVal = OpRHS->getLimitedValue(BitWidth); | ||||
3489 | APInt ShrMask(APInt::getLowBitsSet(BitWidth, BitWidth - OpRHSVal)); | ||||
3490 | ConstantInt *CI = ConstantInt::get(AndRHS->getValue() & ShrMask); | ||||
3491 | |||||
3492 | if (CI->getValue() == ShrMask) { | ||||
3493 | // Masking out bits that the shift already masks. | ||||
3494 | return ReplaceInstUsesWith(TheAnd, Op); | ||||
3495 | } else if (CI != AndRHS) { | ||||
3496 | TheAnd.setOperand(1, CI); // Reduce bits set in and cst. | ||||
3497 | return &TheAnd; | ||||
3498 | } | ||||
3499 | break; | ||||
3500 | } | ||||
3501 | case Instruction::AShr: | ||||
3502 | // Signed shr. | ||||
3503 | // See if this is shifting in some sign extension, then masking it out | ||||
3504 | // with an and. | ||||
3505 | if (Op->hasOneUse()) { | ||||
3506 | uint32_t BitWidth = AndRHS->getType()->getBitWidth(); | ||||
3507 | uint32_t OpRHSVal = OpRHS->getLimitedValue(BitWidth); | ||||
3508 | APInt ShrMask(APInt::getLowBitsSet(BitWidth, BitWidth - OpRHSVal)); | ||||
3509 | Constant *C = ConstantInt::get(AndRHS->getValue() & ShrMask); | ||||
3510 | if (C == AndRHS) { // Masking out bits shifted in. | ||||
3511 | // (Val ashr C1) & C2 -> (Val lshr C1) & C2 | ||||
3512 | // Make the argument unsigned. | ||||
3513 | Value *ShVal = Op->getOperand(0); | ||||
3514 | ShVal = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3515 | BinaryOperator::CreateLShr(ShVal, OpRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3516 | Op->getName()), TheAnd); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3517 | return BinaryOperator::CreateAnd(ShVal, AndRHS, TheAnd.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3518 | } |
3519 | } | ||||
3520 | break; | ||||
3521 | } | ||||
3522 | return 0; | ||||
3523 | } | ||||
3524 | |||||
3525 | |||||
3526 | /// InsertRangeTest - Emit a computation of: (V >= Lo && V < Hi) if Inside is | ||||
3527 | /// true, otherwise (V < Lo || V >= Hi). In pratice, we emit the more efficient | ||||
3528 | /// (V-Lo) <u Hi-Lo. This method expects that Lo <= Hi. isSigned indicates | ||||
3529 | /// whether to treat the V, Lo and HI as signed or not. IB is the location to | ||||
3530 | /// insert new instructions. | ||||
3531 | Instruction *InstCombiner::InsertRangeTest(Value *V, Constant *Lo, Constant *Hi, | ||||
3532 | bool isSigned, bool Inside, | ||||
3533 | Instruction &IB) { | ||||
3534 | assert(cast<ConstantInt>(ConstantExpr::getICmp((isSigned ? | ||||
3535 | ICmpInst::ICMP_SLE:ICmpInst::ICMP_ULE), Lo, Hi))->getZExtValue() && | ||||
3536 | "Lo is not <= Hi in range emission code!"); | ||||
3537 | |||||
3538 | if (Inside) { | ||||
3539 | if (Lo == Hi) // Trivially false. | ||||
3540 | return new ICmpInst(ICmpInst::ICMP_NE, V, V); | ||||
3541 | |||||
3542 | // V >= Min && V < Hi --> V < Hi | ||||
3543 | if (cast<ConstantInt>(Lo)->isMinValue(isSigned)) { | ||||
3544 | ICmpInst::Predicate pred = (isSigned ? | ||||
3545 | ICmpInst::ICMP_SLT : ICmpInst::ICMP_ULT); | ||||
3546 | return new ICmpInst(pred, V, Hi); | ||||
3547 | } | ||||
3548 | |||||
3549 | // Emit V-Lo <u Hi-Lo | ||||
3550 | Constant *NegLo = ConstantExpr::getNeg(Lo); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3551 | Instruction *Add = BinaryOperator::CreateAdd(V, NegLo, V->getName()+".off"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3552 | InsertNewInstBefore(Add, IB); |
3553 | Constant *UpperBound = ConstantExpr::getAdd(NegLo, Hi); | ||||
3554 | return new ICmpInst(ICmpInst::ICMP_ULT, Add, UpperBound); | ||||
3555 | } | ||||
3556 | |||||
3557 | if (Lo == Hi) // Trivially true. | ||||
3558 | return new ICmpInst(ICmpInst::ICMP_EQ, V, V); | ||||
3559 | |||||
3560 | // V < Min || V >= Hi -> V > Hi-1 | ||||
3561 | Hi = SubOne(cast<ConstantInt>(Hi)); | ||||
3562 | if (cast<ConstantInt>(Lo)->isMinValue(isSigned)) { | ||||
3563 | ICmpInst::Predicate pred = (isSigned ? | ||||
3564 | ICmpInst::ICMP_SGT : ICmpInst::ICMP_UGT); | ||||
3565 | return new ICmpInst(pred, V, Hi); | ||||
3566 | } | ||||
3567 | |||||
3568 | // Emit V-Lo >u Hi-1-Lo | ||||
3569 | // Note that Hi has already had one subtracted from it, above. | ||||
3570 | ConstantInt *NegLo = cast<ConstantInt>(ConstantExpr::getNeg(Lo)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3571 | Instruction *Add = BinaryOperator::CreateAdd(V, NegLo, V->getName()+".off"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3572 | InsertNewInstBefore(Add, IB); |
3573 | Constant *LowerBound = ConstantExpr::getAdd(NegLo, Hi); | ||||
3574 | return new ICmpInst(ICmpInst::ICMP_UGT, Add, LowerBound); | ||||
3575 | } | ||||
3576 | |||||
3577 | // isRunOfOnes - Returns true iff Val consists of one contiguous run of 1s with | ||||
3578 | // any number of 0s on either side. The 1s are allowed to wrap from LSB to | ||||
3579 | // MSB, so 0x000FFF0, 0x0000FFFF, and 0xFF0000FF are all runs. 0x0F0F0000 is | ||||
3580 | // not, since all 1s are not contiguous. | ||||
3581 | static bool isRunOfOnes(ConstantInt *Val, uint32_t &MB, uint32_t &ME) { | ||||
3582 | const APInt& V = Val->getValue(); | ||||
3583 | uint32_t BitWidth = Val->getType()->getBitWidth(); | ||||
3584 | if (!APIntOps::isShiftedMask(BitWidth, V)) return false; | ||||
3585 | |||||
3586 | // look for the first zero bit after the run of ones | ||||
3587 | MB = BitWidth - ((V - 1) ^ V).countLeadingZeros(); | ||||
3588 | // look for the first non-zero bit | ||||
3589 | ME = V.getActiveBits(); | ||||
3590 | return true; | ||||
3591 | } | ||||
3592 | |||||
3593 | /// FoldLogicalPlusAnd - This is part of an expression (LHS +/- RHS) & Mask, | ||||
3594 | /// where isSub determines whether the operator is a sub. If we can fold one of | ||||
3595 | /// the following xforms: | ||||
3596 | /// | ||||
3597 | /// ((A & N) +/- B) & Mask -> (A +/- B) & Mask iff N&Mask == Mask | ||||
3598 | /// ((A | N) +/- B) & Mask -> (A +/- B) & Mask iff N&Mask == 0 | ||||
3599 | /// ((A ^ N) +/- B) & Mask -> (A +/- B) & Mask iff N&Mask == 0 | ||||
3600 | /// | ||||
3601 | /// return (A +/- B). | ||||
3602 | /// | ||||
3603 | Value *InstCombiner::FoldLogicalPlusAnd(Value *LHS, Value *RHS, | ||||
3604 | ConstantInt *Mask, bool isSub, | ||||
3605 | Instruction &I) { | ||||
3606 | Instruction *LHSI = dyn_cast<Instruction>(LHS); | ||||
3607 | if (!LHSI || LHSI->getNumOperands() != 2 || | ||||
3608 | !isa<ConstantInt>(LHSI->getOperand(1))) return 0; | ||||
3609 | |||||
3610 | ConstantInt *N = cast<ConstantInt>(LHSI->getOperand(1)); | ||||
3611 | |||||
3612 | switch (LHSI->getOpcode()) { | ||||
3613 | default: return 0; | ||||
3614 | case Instruction::And: | ||||
3615 | if (And(N, Mask) == Mask) { | ||||
3616 | // If the AndRHS is a power of two minus one (0+1+), this is simple. | ||||
3617 | if ((Mask->getValue().countLeadingZeros() + | ||||
3618 | Mask->getValue().countPopulation()) == | ||||
3619 | Mask->getValue().getBitWidth()) | ||||
3620 | break; | ||||
3621 | |||||
3622 | // Otherwise, if Mask is 0+1+0+, and if B is known to have the low 0+ | ||||
3623 | // part, we don't need any explicit masks to take them out of A. If that | ||||
3624 | // is all N is, ignore it. | ||||
3625 | uint32_t MB = 0, ME = 0; | ||||
3626 | if (isRunOfOnes(Mask, MB, ME)) { // begin/end bit of run, inclusive | ||||
3627 | uint32_t BitWidth = cast<IntegerType>(RHS->getType())->getBitWidth(); | ||||
3628 | APInt Mask(APInt::getLowBitsSet(BitWidth, MB-1)); | ||||
3629 | if (MaskedValueIsZero(RHS, Mask)) | ||||
3630 | break; | ||||
3631 | } | ||||
3632 | } | ||||
3633 | return 0; | ||||
3634 | case Instruction::Or: | ||||
3635 | case Instruction::Xor: | ||||
3636 | // If the AndRHS is a power of two minus one (0+1+), and N&Mask == 0 | ||||
3637 | if ((Mask->getValue().countLeadingZeros() + | ||||
3638 | Mask->getValue().countPopulation()) == Mask->getValue().getBitWidth() | ||||
3639 | && And(N, Mask)->isZero()) | ||||
3640 | break; | ||||
3641 | return 0; | ||||
3642 | } | ||||
3643 | |||||
3644 | Instruction *New; | ||||
3645 | if (isSub) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3646 | New = BinaryOperator::CreateSub(LHSI->getOperand(0), RHS, "fold"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3647 | else |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3648 | New = BinaryOperator::CreateAdd(LHSI->getOperand(0), RHS, "fold"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3649 | return InsertNewInstBefore(New, I); |
3650 | } | ||||
3651 | |||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3652 | /// FoldAndOfICmps - Fold (icmp)&(icmp) if possible. |
3653 | Instruction *InstCombiner::FoldAndOfICmps(Instruction &I, | ||||
3654 | ICmpInst *LHS, ICmpInst *RHS) { | ||||
Chris Lattner | f380348 | 2008-11-16 05:10:52 +0000 | [diff] [blame] | 3655 | Value *Val, *Val2; |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3656 | ConstantInt *LHSCst, *RHSCst; |
3657 | ICmpInst::Predicate LHSCC, RHSCC; | ||||
3658 | |||||
Chris Lattner | f380348 | 2008-11-16 05:10:52 +0000 | [diff] [blame] | 3659 | // This only handles icmp of constants: (icmp1 A, C1) & (icmp2 B, C2). |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3660 | if (!match(LHS, m_ICmp(LHSCC, m_Value(Val), m_ConstantInt(LHSCst))) || |
Chris Lattner | f380348 | 2008-11-16 05:10:52 +0000 | [diff] [blame] | 3661 | !match(RHS, m_ICmp(RHSCC, m_Value(Val2), m_ConstantInt(RHSCst)))) |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3662 | return 0; |
Chris Lattner | f380348 | 2008-11-16 05:10:52 +0000 | [diff] [blame] | 3663 | |
3664 | // (icmp ult A, C) & (icmp ult B, C) --> (icmp ult (A|B), C) | ||||
3665 | // where C is a power of 2 | ||||
3666 | if (LHSCst == RHSCst && LHSCC == RHSCC && LHSCC == ICmpInst::ICMP_ULT && | ||||
3667 | LHSCst->getValue().isPowerOf2()) { | ||||
3668 | Instruction *NewOr = BinaryOperator::CreateOr(Val, Val2); | ||||
3669 | InsertNewInstBefore(NewOr, I); | ||||
3670 | return new ICmpInst(LHSCC, NewOr, LHSCst); | ||||
3671 | } | ||||
3672 | |||||
3673 | // From here on, we only handle: | ||||
3674 | // (icmp1 A, C1) & (icmp2 A, C2) --> something simpler. | ||||
3675 | if (Val != Val2) return 0; | ||||
3676 | |||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3677 | // ICMP_[US][GL]E X, CST is folded to ICMP_[US][GL]T elsewhere. |
3678 | if (LHSCC == ICmpInst::ICMP_UGE || LHSCC == ICmpInst::ICMP_ULE || | ||||
3679 | RHSCC == ICmpInst::ICMP_UGE || RHSCC == ICmpInst::ICMP_ULE || | ||||
3680 | LHSCC == ICmpInst::ICMP_SGE || LHSCC == ICmpInst::ICMP_SLE || | ||||
3681 | RHSCC == ICmpInst::ICMP_SGE || RHSCC == ICmpInst::ICMP_SLE) | ||||
3682 | return 0; | ||||
3683 | |||||
3684 | // We can't fold (ugt x, C) & (sgt x, C2). | ||||
3685 | if (!PredicatesFoldable(LHSCC, RHSCC)) | ||||
3686 | return 0; | ||||
3687 | |||||
3688 | // Ensure that the larger constant is on the RHS. | ||||
Chris Lattner | 665298f | 2008-11-16 05:14:43 +0000 | [diff] [blame] | 3689 | bool ShouldSwap; |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3690 | if (ICmpInst::isSignedPredicate(LHSCC) || |
3691 | (ICmpInst::isEquality(LHSCC) && | ||||
3692 | ICmpInst::isSignedPredicate(RHSCC))) | ||||
Chris Lattner | 665298f | 2008-11-16 05:14:43 +0000 | [diff] [blame] | 3693 | ShouldSwap = LHSCst->getValue().sgt(RHSCst->getValue()); |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3694 | else |
Chris Lattner | 665298f | 2008-11-16 05:14:43 +0000 | [diff] [blame] | 3695 | ShouldSwap = LHSCst->getValue().ugt(RHSCst->getValue()); |
3696 | |||||
3697 | if (ShouldSwap) { | ||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3698 | std::swap(LHS, RHS); |
3699 | std::swap(LHSCst, RHSCst); | ||||
3700 | std::swap(LHSCC, RHSCC); | ||||
3701 | } | ||||
3702 | |||||
3703 | // At this point, we know we have have two icmp instructions | ||||
3704 | // comparing a value against two constants and and'ing the result | ||||
3705 | // together. Because of the above check, we know that we only have | ||||
3706 | // icmp eq, icmp ne, icmp [su]lt, and icmp [SU]gt here. We also know | ||||
3707 | // (from the FoldICmpLogical check above), that the two constants | ||||
3708 | // are not equal and that the larger constant is on the RHS | ||||
3709 | assert(LHSCst != RHSCst && "Compares not folded above?"); | ||||
3710 | |||||
3711 | switch (LHSCC) { | ||||
3712 | default: assert(0 && "Unknown integer condition code!"); | ||||
3713 | case ICmpInst::ICMP_EQ: | ||||
3714 | switch (RHSCC) { | ||||
3715 | default: assert(0 && "Unknown integer condition code!"); | ||||
3716 | case ICmpInst::ICMP_EQ: // (X == 13 & X == 15) -> false | ||||
3717 | case ICmpInst::ICMP_UGT: // (X == 13 & X > 15) -> false | ||||
3718 | case ICmpInst::ICMP_SGT: // (X == 13 & X > 15) -> false | ||||
3719 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
3720 | case ICmpInst::ICMP_NE: // (X == 13 & X != 15) -> X == 13 | ||||
3721 | case ICmpInst::ICMP_ULT: // (X == 13 & X < 15) -> X == 13 | ||||
3722 | case ICmpInst::ICMP_SLT: // (X == 13 & X < 15) -> X == 13 | ||||
3723 | return ReplaceInstUsesWith(I, LHS); | ||||
3724 | } | ||||
3725 | case ICmpInst::ICMP_NE: | ||||
3726 | switch (RHSCC) { | ||||
3727 | default: assert(0 && "Unknown integer condition code!"); | ||||
3728 | case ICmpInst::ICMP_ULT: | ||||
3729 | if (LHSCst == SubOne(RHSCst)) // (X != 13 & X u< 14) -> X < 13 | ||||
3730 | return new ICmpInst(ICmpInst::ICMP_ULT, Val, LHSCst); | ||||
3731 | break; // (X != 13 & X u< 15) -> no change | ||||
3732 | case ICmpInst::ICMP_SLT: | ||||
3733 | if (LHSCst == SubOne(RHSCst)) // (X != 13 & X s< 14) -> X < 13 | ||||
3734 | return new ICmpInst(ICmpInst::ICMP_SLT, Val, LHSCst); | ||||
3735 | break; // (X != 13 & X s< 15) -> no change | ||||
3736 | case ICmpInst::ICMP_EQ: // (X != 13 & X == 15) -> X == 15 | ||||
3737 | case ICmpInst::ICMP_UGT: // (X != 13 & X u> 15) -> X u> 15 | ||||
3738 | case ICmpInst::ICMP_SGT: // (X != 13 & X s> 15) -> X s> 15 | ||||
3739 | return ReplaceInstUsesWith(I, RHS); | ||||
3740 | case ICmpInst::ICMP_NE: | ||||
3741 | if (LHSCst == SubOne(RHSCst)){// (X != 13 & X != 14) -> X-13 >u 1 | ||||
3742 | Constant *AddCST = ConstantExpr::getNeg(LHSCst); | ||||
3743 | Instruction *Add = BinaryOperator::CreateAdd(Val, AddCST, | ||||
3744 | Val->getName()+".off"); | ||||
3745 | InsertNewInstBefore(Add, I); | ||||
3746 | return new ICmpInst(ICmpInst::ICMP_UGT, Add, | ||||
3747 | ConstantInt::get(Add->getType(), 1)); | ||||
3748 | } | ||||
3749 | break; // (X != 13 & X != 15) -> no change | ||||
3750 | } | ||||
3751 | break; | ||||
3752 | case ICmpInst::ICMP_ULT: | ||||
3753 | switch (RHSCC) { | ||||
3754 | default: assert(0 && "Unknown integer condition code!"); | ||||
3755 | case ICmpInst::ICMP_EQ: // (X u< 13 & X == 15) -> false | ||||
3756 | case ICmpInst::ICMP_UGT: // (X u< 13 & X u> 15) -> false | ||||
3757 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
3758 | case ICmpInst::ICMP_SGT: // (X u< 13 & X s> 15) -> no change | ||||
3759 | break; | ||||
3760 | case ICmpInst::ICMP_NE: // (X u< 13 & X != 15) -> X u< 13 | ||||
3761 | case ICmpInst::ICMP_ULT: // (X u< 13 & X u< 15) -> X u< 13 | ||||
3762 | return ReplaceInstUsesWith(I, LHS); | ||||
3763 | case ICmpInst::ICMP_SLT: // (X u< 13 & X s< 15) -> no change | ||||
3764 | break; | ||||
3765 | } | ||||
3766 | break; | ||||
3767 | case ICmpInst::ICMP_SLT: | ||||
3768 | switch (RHSCC) { | ||||
3769 | default: assert(0 && "Unknown integer condition code!"); | ||||
3770 | case ICmpInst::ICMP_EQ: // (X s< 13 & X == 15) -> false | ||||
3771 | case ICmpInst::ICMP_SGT: // (X s< 13 & X s> 15) -> false | ||||
3772 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
3773 | case ICmpInst::ICMP_UGT: // (X s< 13 & X u> 15) -> no change | ||||
3774 | break; | ||||
3775 | case ICmpInst::ICMP_NE: // (X s< 13 & X != 15) -> X < 13 | ||||
3776 | case ICmpInst::ICMP_SLT: // (X s< 13 & X s< 15) -> X < 13 | ||||
3777 | return ReplaceInstUsesWith(I, LHS); | ||||
3778 | case ICmpInst::ICMP_ULT: // (X s< 13 & X u< 15) -> no change | ||||
3779 | break; | ||||
3780 | } | ||||
3781 | break; | ||||
3782 | case ICmpInst::ICMP_UGT: | ||||
3783 | switch (RHSCC) { | ||||
3784 | default: assert(0 && "Unknown integer condition code!"); | ||||
3785 | case ICmpInst::ICMP_EQ: // (X u> 13 & X == 15) -> X == 15 | ||||
3786 | case ICmpInst::ICMP_UGT: // (X u> 13 & X u> 15) -> X u> 15 | ||||
3787 | return ReplaceInstUsesWith(I, RHS); | ||||
3788 | case ICmpInst::ICMP_SGT: // (X u> 13 & X s> 15) -> no change | ||||
3789 | break; | ||||
3790 | case ICmpInst::ICMP_NE: | ||||
3791 | if (RHSCst == AddOne(LHSCst)) // (X u> 13 & X != 14) -> X u> 14 | ||||
3792 | return new ICmpInst(LHSCC, Val, RHSCst); | ||||
3793 | break; // (X u> 13 & X != 15) -> no change | ||||
Chris Lattner | 0c678e5 | 2008-11-16 05:20:07 +0000 | [diff] [blame] | 3794 | case ICmpInst::ICMP_ULT: // (X u> 13 & X u< 15) -> (X-14) <u 1 |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3795 | return InsertRangeTest(Val, AddOne(LHSCst), RHSCst, false, true, I); |
3796 | case ICmpInst::ICMP_SLT: // (X u> 13 & X s< 15) -> no change | ||||
3797 | break; | ||||
3798 | } | ||||
3799 | break; | ||||
3800 | case ICmpInst::ICMP_SGT: | ||||
3801 | switch (RHSCC) { | ||||
3802 | default: assert(0 && "Unknown integer condition code!"); | ||||
3803 | case ICmpInst::ICMP_EQ: // (X s> 13 & X == 15) -> X == 15 | ||||
3804 | case ICmpInst::ICMP_SGT: // (X s> 13 & X s> 15) -> X s> 15 | ||||
3805 | return ReplaceInstUsesWith(I, RHS); | ||||
3806 | case ICmpInst::ICMP_UGT: // (X s> 13 & X u> 15) -> no change | ||||
3807 | break; | ||||
3808 | case ICmpInst::ICMP_NE: | ||||
3809 | if (RHSCst == AddOne(LHSCst)) // (X s> 13 & X != 14) -> X s> 14 | ||||
3810 | return new ICmpInst(LHSCC, Val, RHSCst); | ||||
3811 | break; // (X s> 13 & X != 15) -> no change | ||||
Chris Lattner | 0c678e5 | 2008-11-16 05:20:07 +0000 | [diff] [blame] | 3812 | case ICmpInst::ICMP_SLT: // (X s> 13 & X s< 15) -> (X-14) s< 1 |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3813 | return InsertRangeTest(Val, AddOne(LHSCst), RHSCst, true, true, I); |
3814 | case ICmpInst::ICMP_ULT: // (X s> 13 & X u< 15) -> no change | ||||
3815 | break; | ||||
3816 | } | ||||
3817 | break; | ||||
3818 | } | ||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3819 | |
3820 | return 0; | ||||
3821 | } | ||||
3822 | |||||
3823 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3824 | Instruction *InstCombiner::visitAnd(BinaryOperator &I) { |
3825 | bool Changed = SimplifyCommutative(I); | ||||
3826 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3827 | |||||
3828 | if (isa<UndefValue>(Op1)) // X & undef -> 0 | ||||
3829 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
3830 | |||||
3831 | // and X, X = X | ||||
3832 | if (Op0 == Op1) | ||||
3833 | return ReplaceInstUsesWith(I, Op1); | ||||
3834 | |||||
3835 | // See if we can simplify any instructions used by the instruction whose sole | ||||
3836 | // purpose is to compute bits we don't care about. | ||||
3837 | if (!isa<VectorType>(I.getType())) { | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 3838 | if (SimplifyDemandedInstructionBits(I)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3839 | return &I; |
3840 | } else { | ||||
3841 | if (ConstantVector *CP = dyn_cast<ConstantVector>(Op1)) { | ||||
3842 | if (CP->isAllOnesValue()) // X & <-1,-1> -> X | ||||
3843 | return ReplaceInstUsesWith(I, I.getOperand(0)); | ||||
3844 | } else if (isa<ConstantAggregateZero>(Op1)) { | ||||
3845 | return ReplaceInstUsesWith(I, Op1); // X & <0,0> -> <0,0> | ||||
3846 | } | ||||
3847 | } | ||||
3848 | |||||
3849 | if (ConstantInt *AndRHS = dyn_cast<ConstantInt>(Op1)) { | ||||
3850 | const APInt& AndRHSMask = AndRHS->getValue(); | ||||
3851 | APInt NotAndRHS(~AndRHSMask); | ||||
3852 | |||||
3853 | // Optimize a variety of ((val OP C1) & C2) combinations... | ||||
3854 | if (isa<BinaryOperator>(Op0)) { | ||||
3855 | Instruction *Op0I = cast<Instruction>(Op0); | ||||
3856 | Value *Op0LHS = Op0I->getOperand(0); | ||||
3857 | Value *Op0RHS = Op0I->getOperand(1); | ||||
3858 | switch (Op0I->getOpcode()) { | ||||
3859 | case Instruction::Xor: | ||||
3860 | case Instruction::Or: | ||||
3861 | // If the mask is only needed on one incoming arm, push it up. | ||||
3862 | if (Op0I->hasOneUse()) { | ||||
3863 | if (MaskedValueIsZero(Op0LHS, NotAndRHS)) { | ||||
3864 | // Not masking anything out for the LHS, move to RHS. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3865 | Instruction *NewRHS = BinaryOperator::CreateAnd(Op0RHS, AndRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3866 | Op0RHS->getName()+".masked"); |
3867 | InsertNewInstBefore(NewRHS, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3868 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3869 | cast<BinaryOperator>(Op0I)->getOpcode(), Op0LHS, NewRHS); |
3870 | } | ||||
3871 | if (!isa<Constant>(Op0RHS) && | ||||
3872 | MaskedValueIsZero(Op0RHS, NotAndRHS)) { | ||||
3873 | // Not masking anything out for the RHS, move to LHS. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3874 | Instruction *NewLHS = BinaryOperator::CreateAnd(Op0LHS, AndRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3875 | Op0LHS->getName()+".masked"); |
3876 | InsertNewInstBefore(NewLHS, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3877 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3878 | cast<BinaryOperator>(Op0I)->getOpcode(), NewLHS, Op0RHS); |
3879 | } | ||||
3880 | } | ||||
3881 | |||||
3882 | break; | ||||
3883 | case Instruction::Add: | ||||
3884 | // ((A & N) + B) & AndRHS -> (A + B) & AndRHS iff N&AndRHS == AndRHS. | ||||
3885 | // ((A | N) + B) & AndRHS -> (A + B) & AndRHS iff N&AndRHS == 0 | ||||
3886 | // ((A ^ N) + B) & AndRHS -> (A + B) & AndRHS iff N&AndRHS == 0 | ||||
3887 | if (Value *V = FoldLogicalPlusAnd(Op0LHS, Op0RHS, AndRHS, false, I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3888 | return BinaryOperator::CreateAnd(V, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3889 | if (Value *V = FoldLogicalPlusAnd(Op0RHS, Op0LHS, AndRHS, false, I)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3890 | return BinaryOperator::CreateAnd(V, AndRHS); // Add commutes |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3891 | break; |
3892 | |||||
3893 | case Instruction::Sub: | ||||
3894 | // ((A & N) - B) & AndRHS -> (A - B) & AndRHS iff N&AndRHS == AndRHS. | ||||
3895 | // ((A | N) - B) & AndRHS -> (A - B) & AndRHS iff N&AndRHS == 0 | ||||
3896 | // ((A ^ N) - B) & AndRHS -> (A - B) & AndRHS iff N&AndRHS == 0 | ||||
3897 | if (Value *V = FoldLogicalPlusAnd(Op0LHS, Op0RHS, AndRHS, true, I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3898 | return BinaryOperator::CreateAnd(V, AndRHS); |
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 3899 | |
Nick Lewycky | a349ba4 | 2008-07-10 05:51:40 +0000 | [diff] [blame] | 3900 | // (A - N) & AndRHS -> -N & AndRHS iff A&AndRHS==0 and AndRHS |
3901 | // has 1's for all bits that the subtraction with A might affect. | ||||
3902 | if (Op0I->hasOneUse()) { | ||||
3903 | uint32_t BitWidth = AndRHSMask.getBitWidth(); | ||||
3904 | uint32_t Zeros = AndRHSMask.countLeadingZeros(); | ||||
3905 | APInt Mask = APInt::getLowBitsSet(BitWidth, BitWidth - Zeros); | ||||
3906 | |||||
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 3907 | ConstantInt *A = dyn_cast<ConstantInt>(Op0LHS); |
Nick Lewycky | a349ba4 | 2008-07-10 05:51:40 +0000 | [diff] [blame] | 3908 | if (!(A && A->isZero()) && // avoid infinite recursion. |
3909 | MaskedValueIsZero(Op0LHS, Mask)) { | ||||
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 3910 | Instruction *NewNeg = BinaryOperator::CreateNeg(Op0RHS); |
3911 | InsertNewInstBefore(NewNeg, I); | ||||
3912 | return BinaryOperator::CreateAnd(NewNeg, AndRHS); | ||||
3913 | } | ||||
3914 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3915 | break; |
Nick Lewycky | 659ed4d | 2008-07-09 05:20:13 +0000 | [diff] [blame] | 3916 | |
3917 | case Instruction::Shl: | ||||
3918 | case Instruction::LShr: | ||||
3919 | // (1 << x) & 1 --> zext(x == 0) | ||||
3920 | // (1 >> x) & 1 --> zext(x == 0) | ||||
Nick Lewycky | f1b1222 | 2008-07-09 07:35:26 +0000 | [diff] [blame] | 3921 | if (AndRHSMask == 1 && Op0LHS == AndRHS) { |
Nick Lewycky | 659ed4d | 2008-07-09 05:20:13 +0000 | [diff] [blame] | 3922 | Instruction *NewICmp = new ICmpInst(ICmpInst::ICMP_EQ, Op0RHS, |
3923 | Constant::getNullValue(I.getType())); | ||||
3924 | InsertNewInstBefore(NewICmp, I); | ||||
3925 | return new ZExtInst(NewICmp, I.getType()); | ||||
3926 | } | ||||
3927 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3928 | } |
3929 | |||||
3930 | if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) | ||||
3931 | if (Instruction *Res = OptAndOp(Op0I, Op0CI, AndRHS, I)) | ||||
3932 | return Res; | ||||
3933 | } else if (CastInst *CI = dyn_cast<CastInst>(Op0)) { | ||||
3934 | // If this is an integer truncation or change from signed-to-unsigned, and | ||||
3935 | // if the source is an and/or with immediate, transform it. This | ||||
3936 | // frequently occurs for bitfield accesses. | ||||
3937 | if (Instruction *CastOp = dyn_cast<Instruction>(CI->getOperand(0))) { | ||||
3938 | if ((isa<TruncInst>(CI) || isa<BitCastInst>(CI)) && | ||||
3939 | CastOp->getNumOperands() == 2) | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 3940 | if (ConstantInt *AndCI = dyn_cast<ConstantInt>(CastOp->getOperand(1))) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3941 | if (CastOp->getOpcode() == Instruction::And) { |
3942 | // Change: and (cast (and X, C1) to T), C2 | ||||
3943 | // into : and (cast X to T), trunc_or_bitcast(C1)&C2 | ||||
3944 | // This will fold the two constants together, which may allow | ||||
3945 | // other simplifications. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3946 | Instruction *NewCast = CastInst::CreateTruncOrBitCast( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3947 | CastOp->getOperand(0), I.getType(), |
3948 | CastOp->getName()+".shrunk"); | ||||
3949 | NewCast = InsertNewInstBefore(NewCast, I); | ||||
3950 | // trunc_or_bitcast(C1)&C2 | ||||
3951 | Constant *C3 = ConstantExpr::getTruncOrBitCast(AndCI,I.getType()); | ||||
3952 | C3 = ConstantExpr::getAnd(C3, AndRHS); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3953 | return BinaryOperator::CreateAnd(NewCast, C3); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3954 | } else if (CastOp->getOpcode() == Instruction::Or) { |
3955 | // Change: and (cast (or X, C1) to T), C2 | ||||
3956 | // into : trunc(C1)&C2 iff trunc(C1)&C2 == C2 | ||||
3957 | Constant *C3 = ConstantExpr::getTruncOrBitCast(AndCI,I.getType()); | ||||
3958 | if (ConstantExpr::getAnd(C3, AndRHS) == AndRHS) // trunc(C1)&C2 | ||||
3959 | return ReplaceInstUsesWith(I, AndRHS); | ||||
3960 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 3961 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3962 | } |
3963 | } | ||||
3964 | |||||
3965 | // Try to fold constant and into select arguments. | ||||
3966 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
3967 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
3968 | return R; | ||||
3969 | if (isa<PHINode>(Op0)) | ||||
3970 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
3971 | return NV; | ||||
3972 | } | ||||
3973 | |||||
3974 | Value *Op0NotVal = dyn_castNotVal(Op0); | ||||
3975 | Value *Op1NotVal = dyn_castNotVal(Op1); | ||||
3976 | |||||
3977 | if (Op0NotVal == Op1 || Op1NotVal == Op0) // A & ~A == ~A & A == 0 | ||||
3978 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
3979 | |||||
3980 | // (~A & ~B) == (~(A | B)) - De Morgan's Law | ||||
3981 | if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3982 | Instruction *Or = BinaryOperator::CreateOr(Op0NotVal, Op1NotVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3983 | I.getName()+".demorgan"); |
3984 | InsertNewInstBefore(Or, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3985 | return BinaryOperator::CreateNot(Or); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3986 | } |
3987 | |||||
3988 | { | ||||
3989 | Value *A = 0, *B = 0, *C = 0, *D = 0; | ||||
3990 | if (match(Op0, m_Or(m_Value(A), m_Value(B)))) { | ||||
3991 | if (A == Op1 || B == Op1) // (A | ?) & A --> A | ||||
3992 | return ReplaceInstUsesWith(I, Op1); | ||||
3993 | |||||
3994 | // (A|B) & ~(A&B) -> A^B | ||||
3995 | if (match(Op1, m_Not(m_And(m_Value(C), m_Value(D))))) { | ||||
3996 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3997 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3998 | } |
3999 | } | ||||
4000 | |||||
4001 | if (match(Op1, m_Or(m_Value(A), m_Value(B)))) { | ||||
4002 | if (A == Op0 || B == Op0) // A & (A | ?) --> A | ||||
4003 | return ReplaceInstUsesWith(I, Op0); | ||||
4004 | |||||
4005 | // ~(A&B) & (A|B) -> A^B | ||||
4006 | if (match(Op0, m_Not(m_And(m_Value(C), m_Value(D))))) { | ||||
4007 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4008 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4009 | } |
4010 | } | ||||
4011 | |||||
4012 | if (Op0->hasOneUse() && | ||||
4013 | match(Op0, m_Xor(m_Value(A), m_Value(B)))) { | ||||
4014 | if (A == Op1) { // (A^B)&A -> A&(A^B) | ||||
4015 | I.swapOperands(); // Simplify below | ||||
4016 | std::swap(Op0, Op1); | ||||
4017 | } else if (B == Op1) { // (A^B)&B -> B&(B^A) | ||||
4018 | cast<BinaryOperator>(Op0)->swapOperands(); | ||||
4019 | I.swapOperands(); // Simplify below | ||||
4020 | std::swap(Op0, Op1); | ||||
4021 | } | ||||
4022 | } | ||||
Bill Wendling | ce5e0af | 2008-11-30 13:08:13 +0000 | [diff] [blame] | 4023 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4024 | if (Op1->hasOneUse() && |
4025 | match(Op1, m_Xor(m_Value(A), m_Value(B)))) { | ||||
4026 | if (B == Op0) { // B&(A^B) -> B&(B^A) | ||||
4027 | cast<BinaryOperator>(Op1)->swapOperands(); | ||||
4028 | std::swap(A, B); | ||||
4029 | } | ||||
4030 | if (A == Op0) { // A&(A^B) -> A & ~B | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4031 | Instruction *NotB = BinaryOperator::CreateNot(B, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4032 | InsertNewInstBefore(NotB, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4033 | return BinaryOperator::CreateAnd(A, NotB); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4034 | } |
4035 | } | ||||
Bill Wendling | ce5e0af | 2008-11-30 13:08:13 +0000 | [diff] [blame] | 4036 | |
4037 | // (A&((~A)|B)) -> A&B | ||||
Chris Lattner | 9db479f | 2008-12-01 05:16:26 +0000 | [diff] [blame] | 4038 | if (match(Op0, m_Or(m_Not(m_Specific(Op1)), m_Value(A))) || |
4039 | match(Op0, m_Or(m_Value(A), m_Not(m_Specific(Op1))))) | ||||
4040 | return BinaryOperator::CreateAnd(A, Op1); | ||||
4041 | if (match(Op1, m_Or(m_Not(m_Specific(Op0)), m_Value(A))) || | ||||
4042 | match(Op1, m_Or(m_Value(A), m_Not(m_Specific(Op0))))) | ||||
4043 | return BinaryOperator::CreateAnd(A, Op0); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4044 | } |
4045 | |||||
4046 | if (ICmpInst *RHS = dyn_cast<ICmpInst>(Op1)) { | ||||
4047 | // (icmp1 A, B) & (icmp2 A, B) --> (icmp3 A, B) | ||||
4048 | if (Instruction *R = AssociativeOpt(I, FoldICmpLogical(*this, RHS))) | ||||
4049 | return R; | ||||
4050 | |||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 4051 | if (ICmpInst *LHS = dyn_cast<ICmpInst>(Op0)) |
4052 | if (Instruction *Res = FoldAndOfICmps(I, LHS, RHS)) | ||||
4053 | return Res; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4054 | } |
4055 | |||||
4056 | // fold (and (cast A), (cast B)) -> (cast (and A, B)) | ||||
4057 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) | ||||
4058 | if (CastInst *Op1C = dyn_cast<CastInst>(Op1)) | ||||
4059 | if (Op0C->getOpcode() == Op1C->getOpcode()) { // same cast kind ? | ||||
4060 | const Type *SrcTy = Op0C->getOperand(0)->getType(); | ||||
4061 | if (SrcTy == Op1C->getOperand(0)->getType() && SrcTy->isInteger() && | ||||
4062 | // Only do this if the casts both really cause code to be generated. | ||||
4063 | ValueRequiresCast(Op0C->getOpcode(), Op0C->getOperand(0), | ||||
4064 | I.getType(), TD) && | ||||
4065 | ValueRequiresCast(Op1C->getOpcode(), Op1C->getOperand(0), | ||||
4066 | I.getType(), TD)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4067 | Instruction *NewOp = BinaryOperator::CreateAnd(Op0C->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4068 | Op1C->getOperand(0), |
4069 | I.getName()); | ||||
4070 | InsertNewInstBefore(NewOp, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4071 | return CastInst::Create(Op0C->getOpcode(), NewOp, I.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4072 | } |
4073 | } | ||||
4074 | |||||
4075 | // (X >> Z) & (Y >> Z) -> (X&Y) >> Z for all shifts. | ||||
4076 | if (BinaryOperator *SI1 = dyn_cast<BinaryOperator>(Op1)) { | ||||
4077 | if (BinaryOperator *SI0 = dyn_cast<BinaryOperator>(Op0)) | ||||
4078 | if (SI0->isShift() && SI0->getOpcode() == SI1->getOpcode() && | ||||
4079 | SI0->getOperand(1) == SI1->getOperand(1) && | ||||
4080 | (SI0->hasOneUse() || SI1->hasOneUse())) { | ||||
4081 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4082 | InsertNewInstBefore(BinaryOperator::CreateAnd(SI0->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4083 | SI1->getOperand(0), |
4084 | SI0->getName()), I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4085 | return BinaryOperator::Create(SI1->getOpcode(), NewOp, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4086 | SI1->getOperand(1)); |
4087 | } | ||||
4088 | } | ||||
4089 | |||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 4090 | // If and'ing two fcmp, try combine them into one. |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4091 | if (FCmpInst *LHS = dyn_cast<FCmpInst>(I.getOperand(0))) { |
4092 | if (FCmpInst *RHS = dyn_cast<FCmpInst>(I.getOperand(1))) { | ||||
4093 | if (LHS->getPredicate() == FCmpInst::FCMP_ORD && | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 4094 | RHS->getPredicate() == FCmpInst::FCMP_ORD) { |
4095 | // (fcmp ord x, c) & (fcmp ord y, c) -> (fcmp ord x, y) | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4096 | if (ConstantFP *LHSC = dyn_cast<ConstantFP>(LHS->getOperand(1))) |
4097 | if (ConstantFP *RHSC = dyn_cast<ConstantFP>(RHS->getOperand(1))) { | ||||
4098 | // If either of the constants are nans, then the whole thing returns | ||||
4099 | // false. | ||||
Chris Lattner | a6c7dce | 2007-10-24 18:54:45 +0000 | [diff] [blame] | 4100 | if (LHSC->getValueAPF().isNaN() || RHSC->getValueAPF().isNaN()) |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4101 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
4102 | return new FCmpInst(FCmpInst::FCMP_ORD, LHS->getOperand(0), | ||||
4103 | RHS->getOperand(0)); | ||||
4104 | } | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 4105 | } else { |
4106 | Value *Op0LHS, *Op0RHS, *Op1LHS, *Op1RHS; | ||||
4107 | FCmpInst::Predicate Op0CC, Op1CC; | ||||
4108 | if (match(Op0, m_FCmp(Op0CC, m_Value(Op0LHS), m_Value(Op0RHS))) && | ||||
4109 | match(Op1, m_FCmp(Op1CC, m_Value(Op1LHS), m_Value(Op1RHS)))) { | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 4110 | if (Op0LHS == Op1RHS && Op0RHS == Op1LHS) { |
4111 | // Swap RHS operands to match LHS. | ||||
4112 | Op1CC = FCmpInst::getSwappedPredicate(Op1CC); | ||||
4113 | std::swap(Op1LHS, Op1RHS); | ||||
4114 | } | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 4115 | if (Op0LHS == Op1LHS && Op0RHS == Op1RHS) { |
4116 | // Simplify (fcmp cc0 x, y) & (fcmp cc1 x, y). | ||||
4117 | if (Op0CC == Op1CC) | ||||
4118 | return new FCmpInst((FCmpInst::Predicate)Op0CC, Op0LHS, Op0RHS); | ||||
4119 | else if (Op0CC == FCmpInst::FCMP_FALSE || | ||||
4120 | Op1CC == FCmpInst::FCMP_FALSE) | ||||
4121 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
4122 | else if (Op0CC == FCmpInst::FCMP_TRUE) | ||||
4123 | return ReplaceInstUsesWith(I, Op1); | ||||
4124 | else if (Op1CC == FCmpInst::FCMP_TRUE) | ||||
4125 | return ReplaceInstUsesWith(I, Op0); | ||||
4126 | bool Op0Ordered; | ||||
4127 | bool Op1Ordered; | ||||
4128 | unsigned Op0Pred = getFCmpCode(Op0CC, Op0Ordered); | ||||
4129 | unsigned Op1Pred = getFCmpCode(Op1CC, Op1Ordered); | ||||
4130 | if (Op1Pred == 0) { | ||||
4131 | std::swap(Op0, Op1); | ||||
4132 | std::swap(Op0Pred, Op1Pred); | ||||
4133 | std::swap(Op0Ordered, Op1Ordered); | ||||
4134 | } | ||||
4135 | if (Op0Pred == 0) { | ||||
4136 | // uno && ueq -> uno && (uno || eq) -> ueq | ||||
4137 | // ord && olt -> ord && (ord && lt) -> olt | ||||
4138 | if (Op0Ordered == Op1Ordered) | ||||
4139 | return ReplaceInstUsesWith(I, Op1); | ||||
4140 | // uno && oeq -> uno && (ord && eq) -> false | ||||
4141 | // uno && ord -> false | ||||
4142 | if (!Op0Ordered) | ||||
4143 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
4144 | // ord && ueq -> ord && (uno || eq) -> oeq | ||||
4145 | return cast<Instruction>(getFCmpValue(true, Op1Pred, | ||||
4146 | Op0LHS, Op0RHS)); | ||||
4147 | } | ||||
4148 | } | ||||
4149 | } | ||||
4150 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4151 | } |
4152 | } | ||||
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 4153 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4154 | return Changed ? &I : 0; |
4155 | } | ||||
4156 | |||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4157 | /// CollectBSwapParts - Analyze the specified subexpression and see if it is |
4158 | /// capable of providing pieces of a bswap. The subexpression provides pieces | ||||
4159 | /// of a bswap if it is proven that each of the non-zero bytes in the output of | ||||
4160 | /// the expression came from the corresponding "byte swapped" byte in some other | ||||
4161 | /// value. For example, if the current subexpression is "(shl i32 %X, 24)" then | ||||
4162 | /// we know that the expression deposits the low byte of %X into the high byte | ||||
4163 | /// of the bswap result and that all other bytes are zero. This expression is | ||||
4164 | /// accepted, the high byte of ByteValues is set to X to indicate a correct | ||||
4165 | /// match. | ||||
4166 | /// | ||||
4167 | /// This function returns true if the match was unsuccessful and false if so. | ||||
4168 | /// On entry to the function the "OverallLeftShift" is a signed integer value | ||||
4169 | /// indicating the number of bytes that the subexpression is later shifted. For | ||||
4170 | /// example, if the expression is later right shifted by 16 bits, the | ||||
4171 | /// OverallLeftShift value would be -2 on entry. This is used to specify which | ||||
4172 | /// byte of ByteValues is actually being set. | ||||
4173 | /// | ||||
4174 | /// Similarly, ByteMask is a bitmask where a bit is clear if its corresponding | ||||
4175 | /// byte is masked to zero by a user. For example, in (X & 255), X will be | ||||
4176 | /// processed with a bytemask of 1. Because bytemask is 32-bits, this limits | ||||
4177 | /// this function to working on up to 32-byte (256 bit) values. ByteMask is | ||||
4178 | /// always in the local (OverallLeftShift) coordinate space. | ||||
4179 | /// | ||||
4180 | static bool CollectBSwapParts(Value *V, int OverallLeftShift, uint32_t ByteMask, | ||||
4181 | SmallVector<Value*, 8> &ByteValues) { | ||||
4182 | if (Instruction *I = dyn_cast<Instruction>(V)) { | ||||
4183 | // If this is an or instruction, it may be an inner node of the bswap. | ||||
4184 | if (I->getOpcode() == Instruction::Or) { | ||||
4185 | return CollectBSwapParts(I->getOperand(0), OverallLeftShift, ByteMask, | ||||
4186 | ByteValues) || | ||||
4187 | CollectBSwapParts(I->getOperand(1), OverallLeftShift, ByteMask, | ||||
4188 | ByteValues); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4189 | } |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4190 | |
4191 | // If this is a logical shift by a constant multiple of 8, recurse with | ||||
4192 | // OverallLeftShift and ByteMask adjusted. | ||||
4193 | if (I->isLogicalShift() && isa<ConstantInt>(I->getOperand(1))) { | ||||
4194 | unsigned ShAmt = | ||||
4195 | cast<ConstantInt>(I->getOperand(1))->getLimitedValue(~0U); | ||||
4196 | // Ensure the shift amount is defined and of a byte value. | ||||
4197 | if ((ShAmt & 7) || (ShAmt > 8*ByteValues.size())) | ||||
4198 | return true; | ||||
4199 | |||||
4200 | unsigned ByteShift = ShAmt >> 3; | ||||
4201 | if (I->getOpcode() == Instruction::Shl) { | ||||
4202 | // X << 2 -> collect(X, +2) | ||||
4203 | OverallLeftShift += ByteShift; | ||||
4204 | ByteMask >>= ByteShift; | ||||
4205 | } else { | ||||
4206 | // X >>u 2 -> collect(X, -2) | ||||
4207 | OverallLeftShift -= ByteShift; | ||||
4208 | ByteMask <<= ByteShift; | ||||
Chris Lattner | 4444859 | 2008-10-08 06:42:28 +0000 | [diff] [blame] | 4209 | ByteMask &= (~0U >> (32-ByteValues.size())); |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4210 | } |
4211 | |||||
4212 | if (OverallLeftShift >= (int)ByteValues.size()) return true; | ||||
4213 | if (OverallLeftShift <= -(int)ByteValues.size()) return true; | ||||
4214 | |||||
4215 | return CollectBSwapParts(I->getOperand(0), OverallLeftShift, ByteMask, | ||||
4216 | ByteValues); | ||||
4217 | } | ||||
4218 | |||||
4219 | // If this is a logical 'and' with a mask that clears bytes, clear the | ||||
4220 | // corresponding bytes in ByteMask. | ||||
4221 | if (I->getOpcode() == Instruction::And && | ||||
4222 | isa<ConstantInt>(I->getOperand(1))) { | ||||
4223 | // Scan every byte of the and mask, seeing if the byte is either 0 or 255. | ||||
4224 | unsigned NumBytes = ByteValues.size(); | ||||
4225 | APInt Byte(I->getType()->getPrimitiveSizeInBits(), 255); | ||||
4226 | const APInt &AndMask = cast<ConstantInt>(I->getOperand(1))->getValue(); | ||||
4227 | |||||
4228 | for (unsigned i = 0; i != NumBytes; ++i, Byte <<= 8) { | ||||
4229 | // If this byte is masked out by a later operation, we don't care what | ||||
4230 | // the and mask is. | ||||
4231 | if ((ByteMask & (1 << i)) == 0) | ||||
4232 | continue; | ||||
4233 | |||||
4234 | // If the AndMask is all zeros for this byte, clear the bit. | ||||
4235 | APInt MaskB = AndMask & Byte; | ||||
4236 | if (MaskB == 0) { | ||||
4237 | ByteMask &= ~(1U << i); | ||||
4238 | continue; | ||||
4239 | } | ||||
4240 | |||||
4241 | // If the AndMask is not all ones for this byte, it's not a bytezap. | ||||
4242 | if (MaskB != Byte) | ||||
4243 | return true; | ||||
4244 | |||||
4245 | // Otherwise, this byte is kept. | ||||
4246 | } | ||||
4247 | |||||
4248 | return CollectBSwapParts(I->getOperand(0), OverallLeftShift, ByteMask, | ||||
4249 | ByteValues); | ||||
4250 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4251 | } |
4252 | |||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4253 | // Okay, we got to something that isn't a shift, 'or' or 'and'. This must be |
4254 | // the input value to the bswap. Some observations: 1) if more than one byte | ||||
4255 | // is demanded from this input, then it could not be successfully assembled | ||||
4256 | // into a byteswap. At least one of the two bytes would not be aligned with | ||||
4257 | // their ultimate destination. | ||||
4258 | if (!isPowerOf2_32(ByteMask)) return true; | ||||
4259 | unsigned InputByteNo = CountTrailingZeros_32(ByteMask); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4260 | |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4261 | // 2) The input and ultimate destinations must line up: if byte 3 of an i32 |
4262 | // is demanded, it needs to go into byte 0 of the result. This means that the | ||||
4263 | // byte needs to be shifted until it lands in the right byte bucket. The | ||||
4264 | // shift amount depends on the position: if the byte is coming from the high | ||||
4265 | // part of the value (e.g. byte 3) then it must be shifted right. If from the | ||||
4266 | // low part, it must be shifted left. | ||||
4267 | unsigned DestByteNo = InputByteNo + OverallLeftShift; | ||||
4268 | if (InputByteNo < ByteValues.size()/2) { | ||||
4269 | if (ByteValues.size()-1-DestByteNo != InputByteNo) | ||||
4270 | return true; | ||||
4271 | } else { | ||||
4272 | if (ByteValues.size()-1-DestByteNo != InputByteNo) | ||||
4273 | return true; | ||||
4274 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4275 | |
4276 | // If the destination byte value is already defined, the values are or'd | ||||
4277 | // together, which isn't a bswap (unless it's an or of the same bits). | ||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4278 | if (ByteValues[DestByteNo] && ByteValues[DestByteNo] != V) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4279 | return true; |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4280 | ByteValues[DestByteNo] = V; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4281 | return false; |
4282 | } | ||||
4283 | |||||
4284 | /// MatchBSwap - Given an OR instruction, check to see if this is a bswap idiom. | ||||
4285 | /// If so, insert the new bswap intrinsic and return it. | ||||
4286 | Instruction *InstCombiner::MatchBSwap(BinaryOperator &I) { | ||||
4287 | const IntegerType *ITy = dyn_cast<IntegerType>(I.getType()); | ||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4288 | if (!ITy || ITy->getBitWidth() % 16 || |
4289 | // ByteMask only allows up to 32-byte values. | ||||
4290 | ITy->getBitWidth() > 32*8) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4291 | return 0; // Can only bswap pairs of bytes. Can't do vectors. |
4292 | |||||
4293 | /// ByteValues - For each byte of the result, we keep track of which value | ||||
4294 | /// defines each byte. | ||||
4295 | SmallVector<Value*, 8> ByteValues; | ||||
4296 | ByteValues.resize(ITy->getBitWidth()/8); | ||||
4297 | |||||
4298 | // Try to find all the pieces corresponding to the bswap. | ||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4299 | uint32_t ByteMask = ~0U >> (32-ByteValues.size()); |
4300 | if (CollectBSwapParts(&I, 0, ByteMask, ByteValues)) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4301 | return 0; |
4302 | |||||
4303 | // Check to see if all of the bytes come from the same value. | ||||
4304 | Value *V = ByteValues[0]; | ||||
4305 | if (V == 0) return 0; // Didn't find a byte? Must be zero. | ||||
4306 | |||||
4307 | // Check to make sure that all of the bytes come from the same value. | ||||
4308 | for (unsigned i = 1, e = ByteValues.size(); i != e; ++i) | ||||
4309 | if (ByteValues[i] != V) | ||||
4310 | return 0; | ||||
Chandler Carruth | a228e39 | 2007-08-04 01:51:18 +0000 | [diff] [blame] | 4311 | const Type *Tys[] = { ITy }; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4312 | Module *M = I.getParent()->getParent()->getParent(); |
Chandler Carruth | a228e39 | 2007-08-04 01:51:18 +0000 | [diff] [blame] | 4313 | Function *F = Intrinsic::getDeclaration(M, Intrinsic::bswap, Tys, 1); |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 4314 | return CallInst::Create(F, V); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4315 | } |
4316 | |||||
Chris Lattner | dd7772b | 2008-11-16 04:24:12 +0000 | [diff] [blame] | 4317 | /// MatchSelectFromAndOr - We have an expression of the form (A&C)|(B&D). Check |
4318 | /// If A is (cond?-1:0) and either B or D is ~(cond?-1,0) or (cond?0,-1), then | ||||
4319 | /// we can simplify this expression to "cond ? C : D or B". | ||||
4320 | static Instruction *MatchSelectFromAndOr(Value *A, Value *B, | ||||
4321 | Value *C, Value *D) { | ||||
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4322 | // If A is not a select of -1/0, this cannot match. |
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 4323 | Value *Cond = 0; |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 4324 | if (!match(A, m_SelectCst<-1, 0>(m_Value(Cond)))) |
Chris Lattner | dd7772b | 2008-11-16 04:24:12 +0000 | [diff] [blame] | 4325 | return 0; |
4326 | |||||
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4327 | // ((cond?-1:0)&C) | (B&(cond?0:-1)) -> cond ? C : B. |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 4328 | if (match(D, m_SelectCst<0, -1>(m_Specific(Cond)))) |
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4329 | return SelectInst::Create(Cond, C, B); |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 4330 | if (match(D, m_Not(m_SelectCst<-1, 0>(m_Specific(Cond))))) |
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4331 | return SelectInst::Create(Cond, C, B); |
4332 | // ((cond?-1:0)&C) | ((cond?0:-1)&D) -> cond ? C : D. | ||||
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 4333 | if (match(B, m_SelectCst<0, -1>(m_Specific(Cond)))) |
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4334 | return SelectInst::Create(Cond, C, D); |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 4335 | if (match(B, m_Not(m_SelectCst<-1, 0>(m_Specific(Cond))))) |
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4336 | return SelectInst::Create(Cond, C, D); |
Chris Lattner | dd7772b | 2008-11-16 04:24:12 +0000 | [diff] [blame] | 4337 | return 0; |
4338 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4339 | |
Chris Lattner | 0c678e5 | 2008-11-16 05:20:07 +0000 | [diff] [blame] | 4340 | /// FoldOrOfICmps - Fold (icmp)|(icmp) if possible. |
4341 | Instruction *InstCombiner::FoldOrOfICmps(Instruction &I, | ||||
4342 | ICmpInst *LHS, ICmpInst *RHS) { | ||||
4343 | Value *Val, *Val2; | ||||
4344 | ConstantInt *LHSCst, *RHSCst; | ||||
4345 | ICmpInst::Predicate LHSCC, RHSCC; | ||||
4346 | |||||
4347 | // This only handles icmp of constants: (icmp1 A, C1) | (icmp2 B, C2). | ||||
4348 | if (!match(LHS, m_ICmp(LHSCC, m_Value(Val), m_ConstantInt(LHSCst))) || | ||||
4349 | !match(RHS, m_ICmp(RHSCC, m_Value(Val2), m_ConstantInt(RHSCst)))) | ||||
4350 | return 0; | ||||
4351 | |||||
4352 | // From here on, we only handle: | ||||
4353 | // (icmp1 A, C1) | (icmp2 A, C2) --> something simpler. | ||||
4354 | if (Val != Val2) return 0; | ||||
4355 | |||||
4356 | // ICMP_[US][GL]E X, CST is folded to ICMP_[US][GL]T elsewhere. | ||||
4357 | if (LHSCC == ICmpInst::ICMP_UGE || LHSCC == ICmpInst::ICMP_ULE || | ||||
4358 | RHSCC == ICmpInst::ICMP_UGE || RHSCC == ICmpInst::ICMP_ULE || | ||||
4359 | LHSCC == ICmpInst::ICMP_SGE || LHSCC == ICmpInst::ICMP_SLE || | ||||
4360 | RHSCC == ICmpInst::ICMP_SGE || RHSCC == ICmpInst::ICMP_SLE) | ||||
4361 | return 0; | ||||
4362 | |||||
4363 | // We can't fold (ugt x, C) | (sgt x, C2). | ||||
4364 | if (!PredicatesFoldable(LHSCC, RHSCC)) | ||||
4365 | return 0; | ||||
4366 | |||||
4367 | // Ensure that the larger constant is on the RHS. | ||||
4368 | bool ShouldSwap; | ||||
4369 | if (ICmpInst::isSignedPredicate(LHSCC) || | ||||
4370 | (ICmpInst::isEquality(LHSCC) && | ||||
4371 | ICmpInst::isSignedPredicate(RHSCC))) | ||||
4372 | ShouldSwap = LHSCst->getValue().sgt(RHSCst->getValue()); | ||||
4373 | else | ||||
4374 | ShouldSwap = LHSCst->getValue().ugt(RHSCst->getValue()); | ||||
4375 | |||||
4376 | if (ShouldSwap) { | ||||
4377 | std::swap(LHS, RHS); | ||||
4378 | std::swap(LHSCst, RHSCst); | ||||
4379 | std::swap(LHSCC, RHSCC); | ||||
4380 | } | ||||
4381 | |||||
4382 | // At this point, we know we have have two icmp instructions | ||||
4383 | // comparing a value against two constants and or'ing the result | ||||
4384 | // together. Because of the above check, we know that we only have | ||||
4385 | // ICMP_EQ, ICMP_NE, ICMP_LT, and ICMP_GT here. We also know (from the | ||||
4386 | // FoldICmpLogical check above), that the two constants are not | ||||
4387 | // equal. | ||||
4388 | assert(LHSCst != RHSCst && "Compares not folded above?"); | ||||
4389 | |||||
4390 | switch (LHSCC) { | ||||
4391 | default: assert(0 && "Unknown integer condition code!"); | ||||
4392 | case ICmpInst::ICMP_EQ: | ||||
4393 | switch (RHSCC) { | ||||
4394 | default: assert(0 && "Unknown integer condition code!"); | ||||
4395 | case ICmpInst::ICMP_EQ: | ||||
4396 | if (LHSCst == SubOne(RHSCst)) { // (X == 13 | X == 14) -> X-13 <u 2 | ||||
4397 | Constant *AddCST = ConstantExpr::getNeg(LHSCst); | ||||
4398 | Instruction *Add = BinaryOperator::CreateAdd(Val, AddCST, | ||||
4399 | Val->getName()+".off"); | ||||
4400 | InsertNewInstBefore(Add, I); | ||||
4401 | AddCST = Subtract(AddOne(RHSCst), LHSCst); | ||||
4402 | return new ICmpInst(ICmpInst::ICMP_ULT, Add, AddCST); | ||||
4403 | } | ||||
4404 | break; // (X == 13 | X == 15) -> no change | ||||
4405 | case ICmpInst::ICMP_UGT: // (X == 13 | X u> 14) -> no change | ||||
4406 | case ICmpInst::ICMP_SGT: // (X == 13 | X s> 14) -> no change | ||||
4407 | break; | ||||
4408 | case ICmpInst::ICMP_NE: // (X == 13 | X != 15) -> X != 15 | ||||
4409 | case ICmpInst::ICMP_ULT: // (X == 13 | X u< 15) -> X u< 15 | ||||
4410 | case ICmpInst::ICMP_SLT: // (X == 13 | X s< 15) -> X s< 15 | ||||
4411 | return ReplaceInstUsesWith(I, RHS); | ||||
4412 | } | ||||
4413 | break; | ||||
4414 | case ICmpInst::ICMP_NE: | ||||
4415 | switch (RHSCC) { | ||||
4416 | default: assert(0 && "Unknown integer condition code!"); | ||||
4417 | case ICmpInst::ICMP_EQ: // (X != 13 | X == 15) -> X != 13 | ||||
4418 | case ICmpInst::ICMP_UGT: // (X != 13 | X u> 15) -> X != 13 | ||||
4419 | case ICmpInst::ICMP_SGT: // (X != 13 | X s> 15) -> X != 13 | ||||
4420 | return ReplaceInstUsesWith(I, LHS); | ||||
4421 | case ICmpInst::ICMP_NE: // (X != 13 | X != 15) -> true | ||||
4422 | case ICmpInst::ICMP_ULT: // (X != 13 | X u< 15) -> true | ||||
4423 | case ICmpInst::ICMP_SLT: // (X != 13 | X s< 15) -> true | ||||
4424 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4425 | } | ||||
4426 | break; | ||||
4427 | case ICmpInst::ICMP_ULT: | ||||
4428 | switch (RHSCC) { | ||||
4429 | default: assert(0 && "Unknown integer condition code!"); | ||||
4430 | case ICmpInst::ICMP_EQ: // (X u< 13 | X == 14) -> no change | ||||
4431 | break; | ||||
4432 | case ICmpInst::ICMP_UGT: // (X u< 13 | X u> 15) -> (X-13) u> 2 | ||||
4433 | // If RHSCst is [us]MAXINT, it is always false. Not handling | ||||
4434 | // this can cause overflow. | ||||
4435 | if (RHSCst->isMaxValue(false)) | ||||
4436 | return ReplaceInstUsesWith(I, LHS); | ||||
4437 | return InsertRangeTest(Val, LHSCst, AddOne(RHSCst), false, false, I); | ||||
4438 | case ICmpInst::ICMP_SGT: // (X u< 13 | X s> 15) -> no change | ||||
4439 | break; | ||||
4440 | case ICmpInst::ICMP_NE: // (X u< 13 | X != 15) -> X != 15 | ||||
4441 | case ICmpInst::ICMP_ULT: // (X u< 13 | X u< 15) -> X u< 15 | ||||
4442 | return ReplaceInstUsesWith(I, RHS); | ||||
4443 | case ICmpInst::ICMP_SLT: // (X u< 13 | X s< 15) -> no change | ||||
4444 | break; | ||||
4445 | } | ||||
4446 | break; | ||||
4447 | case ICmpInst::ICMP_SLT: | ||||
4448 | switch (RHSCC) { | ||||
4449 | default: assert(0 && "Unknown integer condition code!"); | ||||
4450 | case ICmpInst::ICMP_EQ: // (X s< 13 | X == 14) -> no change | ||||
4451 | break; | ||||
4452 | case ICmpInst::ICMP_SGT: // (X s< 13 | X s> 15) -> (X-13) s> 2 | ||||
4453 | // If RHSCst is [us]MAXINT, it is always false. Not handling | ||||
4454 | // this can cause overflow. | ||||
4455 | if (RHSCst->isMaxValue(true)) | ||||
4456 | return ReplaceInstUsesWith(I, LHS); | ||||
4457 | return InsertRangeTest(Val, LHSCst, AddOne(RHSCst), true, false, I); | ||||
4458 | case ICmpInst::ICMP_UGT: // (X s< 13 | X u> 15) -> no change | ||||
4459 | break; | ||||
4460 | case ICmpInst::ICMP_NE: // (X s< 13 | X != 15) -> X != 15 | ||||
4461 | case ICmpInst::ICMP_SLT: // (X s< 13 | X s< 15) -> X s< 15 | ||||
4462 | return ReplaceInstUsesWith(I, RHS); | ||||
4463 | case ICmpInst::ICMP_ULT: // (X s< 13 | X u< 15) -> no change | ||||
4464 | break; | ||||
4465 | } | ||||
4466 | break; | ||||
4467 | case ICmpInst::ICMP_UGT: | ||||
4468 | switch (RHSCC) { | ||||
4469 | default: assert(0 && "Unknown integer condition code!"); | ||||
4470 | case ICmpInst::ICMP_EQ: // (X u> 13 | X == 15) -> X u> 13 | ||||
4471 | case ICmpInst::ICMP_UGT: // (X u> 13 | X u> 15) -> X u> 13 | ||||
4472 | return ReplaceInstUsesWith(I, LHS); | ||||
4473 | case ICmpInst::ICMP_SGT: // (X u> 13 | X s> 15) -> no change | ||||
4474 | break; | ||||
4475 | case ICmpInst::ICMP_NE: // (X u> 13 | X != 15) -> true | ||||
4476 | case ICmpInst::ICMP_ULT: // (X u> 13 | X u< 15) -> true | ||||
4477 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4478 | case ICmpInst::ICMP_SLT: // (X u> 13 | X s< 15) -> no change | ||||
4479 | break; | ||||
4480 | } | ||||
4481 | break; | ||||
4482 | case ICmpInst::ICMP_SGT: | ||||
4483 | switch (RHSCC) { | ||||
4484 | default: assert(0 && "Unknown integer condition code!"); | ||||
4485 | case ICmpInst::ICMP_EQ: // (X s> 13 | X == 15) -> X > 13 | ||||
4486 | case ICmpInst::ICMP_SGT: // (X s> 13 | X s> 15) -> X > 13 | ||||
4487 | return ReplaceInstUsesWith(I, LHS); | ||||
4488 | case ICmpInst::ICMP_UGT: // (X s> 13 | X u> 15) -> no change | ||||
4489 | break; | ||||
4490 | case ICmpInst::ICMP_NE: // (X s> 13 | X != 15) -> true | ||||
4491 | case ICmpInst::ICMP_SLT: // (X s> 13 | X s< 15) -> true | ||||
4492 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4493 | case ICmpInst::ICMP_ULT: // (X s> 13 | X u< 15) -> no change | ||||
4494 | break; | ||||
4495 | } | ||||
4496 | break; | ||||
4497 | } | ||||
4498 | return 0; | ||||
4499 | } | ||||
4500 | |||||
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4501 | /// FoldOrWithConstants - This helper function folds: |
4502 | /// | ||||
Bill Wendling | 236a119 | 2008-12-02 05:09:00 +0000 | [diff] [blame] | 4503 | /// ((A | B) & C1) | (B & C2) |
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4504 | /// |
4505 | /// into: | ||||
4506 | /// | ||||
Bill Wendling | 236a119 | 2008-12-02 05:09:00 +0000 | [diff] [blame] | 4507 | /// (A & C1) | B |
Bill Wendling | 9912f71 | 2008-12-01 08:32:40 +0000 | [diff] [blame] | 4508 | /// |
Bill Wendling | 236a119 | 2008-12-02 05:09:00 +0000 | [diff] [blame] | 4509 | /// when the XOR of the two constants is "all ones" (-1). |
Bill Wendling | 9912f71 | 2008-12-01 08:32:40 +0000 | [diff] [blame] | 4510 | Instruction *InstCombiner::FoldOrWithConstants(BinaryOperator &I, Value *Op, |
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4511 | Value *A, Value *B, Value *C) { |
Bill Wendling | fc5b8e6 | 2008-12-02 05:06:43 +0000 | [diff] [blame] | 4512 | ConstantInt *CI1 = dyn_cast<ConstantInt>(C); |
4513 | if (!CI1) return 0; | ||||
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4514 | |
Bill Wendling | 0a0dcaf | 2008-12-02 06:24:20 +0000 | [diff] [blame] | 4515 | Value *V1 = 0; |
4516 | ConstantInt *CI2 = 0; | ||||
4517 | if (!match(Op, m_And(m_Value(V1), m_ConstantInt(CI2)))) return 0; | ||||
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4518 | |
Bill Wendling | 86ee316 | 2008-12-02 06:18:11 +0000 | [diff] [blame] | 4519 | APInt Xor = CI1->getValue() ^ CI2->getValue(); |
4520 | if (!Xor.isAllOnesValue()) return 0; | ||||
4521 | |||||
Bill Wendling | 0a0dcaf | 2008-12-02 06:24:20 +0000 | [diff] [blame] | 4522 | if (V1 == A || V1 == B) { |
Bill Wendling | 86ee316 | 2008-12-02 06:18:11 +0000 | [diff] [blame] | 4523 | Instruction *NewOp = |
Bill Wendling | 6c8ecbb | 2008-12-02 06:22:04 +0000 | [diff] [blame] | 4524 | InsertNewInstBefore(BinaryOperator::CreateAnd((V1 == A) ? B : A, CI1), I); |
4525 | return BinaryOperator::CreateOr(NewOp, V1); | ||||
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4526 | } |
4527 | |||||
4528 | return 0; | ||||
4529 | } | ||||
4530 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4531 | Instruction *InstCombiner::visitOr(BinaryOperator &I) { |
4532 | bool Changed = SimplifyCommutative(I); | ||||
4533 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
4534 | |||||
4535 | if (isa<UndefValue>(Op1)) // X | undef -> -1 | ||||
4536 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4537 | |||||
4538 | // or X, X = X | ||||
4539 | if (Op0 == Op1) | ||||
4540 | return ReplaceInstUsesWith(I, Op0); | ||||
4541 | |||||
4542 | // See if we can simplify any instructions used by the instruction whose sole | ||||
4543 | // purpose is to compute bits we don't care about. | ||||
4544 | if (!isa<VectorType>(I.getType())) { | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 4545 | if (SimplifyDemandedInstructionBits(I)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4546 | return &I; |
4547 | } else if (isa<ConstantAggregateZero>(Op1)) { | ||||
4548 | return ReplaceInstUsesWith(I, Op0); // X | <0,0> -> X | ||||
4549 | } else if (ConstantVector *CP = dyn_cast<ConstantVector>(Op1)) { | ||||
4550 | if (CP->isAllOnesValue()) // X | <-1,-1> -> <-1,-1> | ||||
4551 | return ReplaceInstUsesWith(I, I.getOperand(1)); | ||||
4552 | } | ||||
4553 | |||||
4554 | |||||
4555 | |||||
4556 | // or X, -1 == -1 | ||||
4557 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
4558 | ConstantInt *C1 = 0; Value *X = 0; | ||||
4559 | // (X & C1) | C2 --> (X | C2) & (C1|C2) | ||||
4560 | if (match(Op0, m_And(m_Value(X), m_ConstantInt(C1))) && isOnlyUse(Op0)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4561 | Instruction *Or = BinaryOperator::CreateOr(X, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4562 | InsertNewInstBefore(Or, I); |
4563 | Or->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4564 | return BinaryOperator::CreateAnd(Or, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4565 | ConstantInt::get(RHS->getValue() | C1->getValue())); |
4566 | } | ||||
4567 | |||||
4568 | // (X ^ C1) | C2 --> (X | C2) ^ (C1&~C2) | ||||
4569 | if (match(Op0, m_Xor(m_Value(X), m_ConstantInt(C1))) && isOnlyUse(Op0)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4570 | Instruction *Or = BinaryOperator::CreateOr(X, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4571 | InsertNewInstBefore(Or, I); |
4572 | Or->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4573 | return BinaryOperator::CreateXor(Or, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4574 | ConstantInt::get(C1->getValue() & ~RHS->getValue())); |
4575 | } | ||||
4576 | |||||
4577 | // Try to fold constant and into select arguments. | ||||
4578 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
4579 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
4580 | return R; | ||||
4581 | if (isa<PHINode>(Op0)) | ||||
4582 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
4583 | return NV; | ||||
4584 | } | ||||
4585 | |||||
4586 | Value *A = 0, *B = 0; | ||||
4587 | ConstantInt *C1 = 0, *C2 = 0; | ||||
4588 | |||||
4589 | if (match(Op0, m_And(m_Value(A), m_Value(B)))) | ||||
4590 | if (A == Op1 || B == Op1) // (A & ?) | A --> A | ||||
4591 | return ReplaceInstUsesWith(I, Op1); | ||||
4592 | if (match(Op1, m_And(m_Value(A), m_Value(B)))) | ||||
4593 | if (A == Op0 || B == Op0) // A | (A & ?) --> A | ||||
4594 | return ReplaceInstUsesWith(I, Op0); | ||||
4595 | |||||
4596 | // (A | B) | C and A | (B | C) -> bswap if possible. | ||||
4597 | // (A >> B) | (C << D) and (A << B) | (B >> C) -> bswap if possible. | ||||
4598 | if (match(Op0, m_Or(m_Value(), m_Value())) || | ||||
4599 | match(Op1, m_Or(m_Value(), m_Value())) || | ||||
4600 | (match(Op0, m_Shift(m_Value(), m_Value())) && | ||||
4601 | match(Op1, m_Shift(m_Value(), m_Value())))) { | ||||
4602 | if (Instruction *BSwap = MatchBSwap(I)) | ||||
4603 | return BSwap; | ||||
4604 | } | ||||
4605 | |||||
4606 | // (X^C)|Y -> (X|Y)^C iff Y&C == 0 | ||||
4607 | if (Op0->hasOneUse() && match(Op0, m_Xor(m_Value(A), m_ConstantInt(C1))) && | ||||
4608 | MaskedValueIsZero(Op1, C1->getValue())) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4609 | Instruction *NOr = BinaryOperator::CreateOr(A, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4610 | InsertNewInstBefore(NOr, I); |
4611 | NOr->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4612 | return BinaryOperator::CreateXor(NOr, C1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4613 | } |
4614 | |||||
4615 | // Y|(X^C) -> (X|Y)^C iff Y&C == 0 | ||||
4616 | if (Op1->hasOneUse() && match(Op1, m_Xor(m_Value(A), m_ConstantInt(C1))) && | ||||
4617 | MaskedValueIsZero(Op0, C1->getValue())) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4618 | Instruction *NOr = BinaryOperator::CreateOr(A, Op0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4619 | InsertNewInstBefore(NOr, I); |
4620 | NOr->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4621 | return BinaryOperator::CreateXor(NOr, C1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4622 | } |
4623 | |||||
4624 | // (A & C)|(B & D) | ||||
4625 | Value *C = 0, *D = 0; | ||||
4626 | if (match(Op0, m_And(m_Value(A), m_Value(C))) && | ||||
4627 | match(Op1, m_And(m_Value(B), m_Value(D)))) { | ||||
4628 | Value *V1 = 0, *V2 = 0, *V3 = 0; | ||||
4629 | C1 = dyn_cast<ConstantInt>(C); | ||||
4630 | C2 = dyn_cast<ConstantInt>(D); | ||||
4631 | if (C1 && C2) { // (A & C1)|(B & C2) | ||||
4632 | // If we have: ((V + N) & C1) | (V & C2) | ||||
4633 | // .. and C2 = ~C1 and C2 is 0+1+ and (N & C2) == 0 | ||||
4634 | // replace with V+N. | ||||
4635 | if (C1->getValue() == ~C2->getValue()) { | ||||
4636 | if ((C2->getValue() & (C2->getValue()+1)) == 0 && // C2 == 0+1+ | ||||
4637 | match(A, m_Add(m_Value(V1), m_Value(V2)))) { | ||||
4638 | // Add commutes, try both ways. | ||||
4639 | if (V1 == B && MaskedValueIsZero(V2, C2->getValue())) | ||||
4640 | return ReplaceInstUsesWith(I, A); | ||||
4641 | if (V2 == B && MaskedValueIsZero(V1, C2->getValue())) | ||||
4642 | return ReplaceInstUsesWith(I, A); | ||||
4643 | } | ||||
4644 | // Or commutes, try both ways. | ||||
4645 | if ((C1->getValue() & (C1->getValue()+1)) == 0 && | ||||
4646 | match(B, m_Add(m_Value(V1), m_Value(V2)))) { | ||||
4647 | // Add commutes, try both ways. | ||||
4648 | if (V1 == A && MaskedValueIsZero(V2, C1->getValue())) | ||||
4649 | return ReplaceInstUsesWith(I, B); | ||||
4650 | if (V2 == A && MaskedValueIsZero(V1, C1->getValue())) | ||||
4651 | return ReplaceInstUsesWith(I, B); | ||||
4652 | } | ||||
4653 | } | ||||
4654 | V1 = 0; V2 = 0; V3 = 0; | ||||
4655 | } | ||||
4656 | |||||
4657 | // Check to see if we have any common things being and'ed. If so, find the | ||||
4658 | // terms for V1 & (V2|V3). | ||||
4659 | if (isOnlyUse(Op0) || isOnlyUse(Op1)) { | ||||
4660 | if (A == B) // (A & C)|(A & D) == A & (C|D) | ||||
4661 | V1 = A, V2 = C, V3 = D; | ||||
4662 | else if (A == D) // (A & C)|(B & A) == A & (B|C) | ||||
4663 | V1 = A, V2 = B, V3 = C; | ||||
4664 | else if (C == B) // (A & C)|(C & D) == C & (A|D) | ||||
4665 | V1 = C, V2 = A, V3 = D; | ||||
4666 | else if (C == D) // (A & C)|(B & C) == C & (A|B) | ||||
4667 | V1 = C, V2 = A, V3 = B; | ||||
4668 | |||||
4669 | if (V1) { | ||||
4670 | Value *Or = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4671 | InsertNewInstBefore(BinaryOperator::CreateOr(V2, V3, "tmp"), I); |
4672 | return BinaryOperator::CreateAnd(V1, Or); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4673 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4674 | } |
Dan Gohman | 279952c | 2008-10-28 22:38:57 +0000 | [diff] [blame] | 4675 | |
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 4676 | // (A & (C0?-1:0)) | (B & ~(C0?-1:0)) -> C0 ? A : B, and commuted variants |
Chris Lattner | dd7772b | 2008-11-16 04:24:12 +0000 | [diff] [blame] | 4677 | if (Instruction *Match = MatchSelectFromAndOr(A, B, C, D)) |
4678 | return Match; | ||||
4679 | if (Instruction *Match = MatchSelectFromAndOr(B, A, D, C)) | ||||
4680 | return Match; | ||||
4681 | if (Instruction *Match = MatchSelectFromAndOr(C, B, A, D)) | ||||
4682 | return Match; | ||||
4683 | if (Instruction *Match = MatchSelectFromAndOr(D, A, B, C)) | ||||
4684 | return Match; | ||||
Bill Wendling | 22ca835 | 2008-11-30 13:52:49 +0000 | [diff] [blame] | 4685 | |
Bill Wendling | 22ca835 | 2008-11-30 13:52:49 +0000 | [diff] [blame] | 4686 | // ((A&~B)|(~A&B)) -> A^B |
Bill Wendling | c1f3113 | 2008-12-01 08:09:47 +0000 | [diff] [blame] | 4687 | if ((match(C, m_Not(m_Specific(D))) && |
4688 | match(B, m_Not(m_Specific(A))))) | ||||
4689 | return BinaryOperator::CreateXor(A, D); | ||||
Bill Wendling | 22ca835 | 2008-11-30 13:52:49 +0000 | [diff] [blame] | 4690 | // ((~B&A)|(~A&B)) -> A^B |
Bill Wendling | c1f3113 | 2008-12-01 08:09:47 +0000 | [diff] [blame] | 4691 | if ((match(A, m_Not(m_Specific(D))) && |
4692 | match(B, m_Not(m_Specific(C))))) | ||||
4693 | return BinaryOperator::CreateXor(C, D); | ||||
Bill Wendling | 22ca835 | 2008-11-30 13:52:49 +0000 | [diff] [blame] | 4694 | // ((A&~B)|(B&~A)) -> A^B |
Bill Wendling | c1f3113 | 2008-12-01 08:09:47 +0000 | [diff] [blame] | 4695 | if ((match(C, m_Not(m_Specific(B))) && |
4696 | match(D, m_Not(m_Specific(A))))) | ||||
4697 | return BinaryOperator::CreateXor(A, B); | ||||
Bill Wendling | 22ca835 | 2008-11-30 13:52:49 +0000 | [diff] [blame] | 4698 | // ((~B&A)|(B&~A)) -> A^B |
Bill Wendling | c1f3113 | 2008-12-01 08:09:47 +0000 | [diff] [blame] | 4699 | if ((match(A, m_Not(m_Specific(B))) && |
4700 | match(D, m_Not(m_Specific(C))))) | ||||
4701 | return BinaryOperator::CreateXor(C, B); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4702 | } |
4703 | |||||
4704 | // (X >> Z) | (Y >> Z) -> (X|Y) >> Z for all shifts. | ||||
4705 | if (BinaryOperator *SI1 = dyn_cast<BinaryOperator>(Op1)) { | ||||
4706 | if (BinaryOperator *SI0 = dyn_cast<BinaryOperator>(Op0)) | ||||
4707 | if (SI0->isShift() && SI0->getOpcode() == SI1->getOpcode() && | ||||
4708 | SI0->getOperand(1) == SI1->getOperand(1) && | ||||
4709 | (SI0->hasOneUse() || SI1->hasOneUse())) { | ||||
4710 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4711 | InsertNewInstBefore(BinaryOperator::CreateOr(SI0->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4712 | SI1->getOperand(0), |
4713 | SI0->getName()), I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4714 | return BinaryOperator::Create(SI1->getOpcode(), NewOp, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4715 | SI1->getOperand(1)); |
4716 | } | ||||
4717 | } | ||||
4718 | |||||
Bill Wendling | d8ce237 | 2008-12-01 01:07:11 +0000 | [diff] [blame] | 4719 | // ((A|B)&1)|(B&-2) -> (A&1) | B |
4720 | if (match(Op0, m_And(m_Or(m_Value(A), m_Value(B)), m_Value(C))) || | ||||
4721 | match(Op0, m_And(m_Value(C), m_Or(m_Value(A), m_Value(B))))) { | ||||
Bill Wendling | 9912f71 | 2008-12-01 08:32:40 +0000 | [diff] [blame] | 4722 | Instruction *Ret = FoldOrWithConstants(I, Op1, A, B, C); |
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4723 | if (Ret) return Ret; |
Bill Wendling | d8ce237 | 2008-12-01 01:07:11 +0000 | [diff] [blame] | 4724 | } |
4725 | // (B&-2)|((A|B)&1) -> (A&1) | B | ||||
4726 | if (match(Op1, m_And(m_Or(m_Value(A), m_Value(B)), m_Value(C))) || | ||||
4727 | match(Op1, m_And(m_Value(C), m_Or(m_Value(A), m_Value(B))))) { | ||||
Bill Wendling | 9912f71 | 2008-12-01 08:32:40 +0000 | [diff] [blame] | 4728 | Instruction *Ret = FoldOrWithConstants(I, Op0, A, B, C); |
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4729 | if (Ret) return Ret; |
Bill Wendling | d8ce237 | 2008-12-01 01:07:11 +0000 | [diff] [blame] | 4730 | } |
4731 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4732 | if (match(Op0, m_Not(m_Value(A)))) { // ~A | Op1 |
4733 | if (A == Op1) // ~A | A == -1 | ||||
4734 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4735 | } else { | ||||
4736 | A = 0; | ||||
4737 | } | ||||
4738 | // Note, A is still live here! | ||||
4739 | if (match(Op1, m_Not(m_Value(B)))) { // Op0 | ~B | ||||
4740 | if (Op0 == B) | ||||
4741 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4742 | |||||
4743 | // (~A | ~B) == (~(A & B)) - De Morgan's Law | ||||
4744 | if (A && isOnlyUse(Op0) && isOnlyUse(Op1)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4745 | Value *And = InsertNewInstBefore(BinaryOperator::CreateAnd(A, B, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4746 | I.getName()+".demorgan"), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4747 | return BinaryOperator::CreateNot(And); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4748 | } |
4749 | } | ||||
4750 | |||||
4751 | // (icmp1 A, B) | (icmp2 A, B) --> (icmp3 A, B) | ||||
4752 | if (ICmpInst *RHS = dyn_cast<ICmpInst>(I.getOperand(1))) { | ||||
4753 | if (Instruction *R = AssociativeOpt(I, FoldICmpLogical(*this, RHS))) | ||||
4754 | return R; | ||||
4755 | |||||
Chris Lattner | 0c678e5 | 2008-11-16 05:20:07 +0000 | [diff] [blame] | 4756 | if (ICmpInst *LHS = dyn_cast<ICmpInst>(I.getOperand(0))) |
4757 | if (Instruction *Res = FoldOrOfICmps(I, LHS, RHS)) | ||||
4758 | return Res; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4759 | } |
4760 | |||||
4761 | // fold (or (cast A), (cast B)) -> (cast (or A, B)) | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4762 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4763 | if (CastInst *Op1C = dyn_cast<CastInst>(Op1)) |
4764 | if (Op0C->getOpcode() == Op1C->getOpcode()) {// same cast kind ? | ||||
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 4765 | if (!isa<ICmpInst>(Op0C->getOperand(0)) || |
4766 | !isa<ICmpInst>(Op1C->getOperand(0))) { | ||||
4767 | const Type *SrcTy = Op0C->getOperand(0)->getType(); | ||||
4768 | if (SrcTy == Op1C->getOperand(0)->getType() && SrcTy->isInteger() && | ||||
4769 | // Only do this if the casts both really cause code to be | ||||
4770 | // generated. | ||||
4771 | ValueRequiresCast(Op0C->getOpcode(), Op0C->getOperand(0), | ||||
4772 | I.getType(), TD) && | ||||
4773 | ValueRequiresCast(Op1C->getOpcode(), Op1C->getOperand(0), | ||||
4774 | I.getType(), TD)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4775 | Instruction *NewOp = BinaryOperator::CreateOr(Op0C->getOperand(0), |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 4776 | Op1C->getOperand(0), |
4777 | I.getName()); | ||||
4778 | InsertNewInstBefore(NewOp, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4779 | return CastInst::Create(Op0C->getOpcode(), NewOp, I.getType()); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 4780 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4781 | } |
4782 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4783 | } |
4784 | |||||
4785 | |||||
4786 | // (fcmp uno x, c) | (fcmp uno y, c) -> (fcmp uno x, y) | ||||
4787 | if (FCmpInst *LHS = dyn_cast<FCmpInst>(I.getOperand(0))) { | ||||
4788 | if (FCmpInst *RHS = dyn_cast<FCmpInst>(I.getOperand(1))) { | ||||
4789 | if (LHS->getPredicate() == FCmpInst::FCMP_UNO && | ||||
Chris Lattner | be9e63e | 2008-02-29 06:09:11 +0000 | [diff] [blame] | 4790 | RHS->getPredicate() == FCmpInst::FCMP_UNO && |
Evan Cheng | 7298805 | 2008-10-14 18:44:08 +0000 | [diff] [blame] | 4791 | LHS->getOperand(0)->getType() == RHS->getOperand(0)->getType()) { |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4792 | if (ConstantFP *LHSC = dyn_cast<ConstantFP>(LHS->getOperand(1))) |
4793 | if (ConstantFP *RHSC = dyn_cast<ConstantFP>(RHS->getOperand(1))) { | ||||
4794 | // If either of the constants are nans, then the whole thing returns | ||||
4795 | // true. | ||||
Chris Lattner | a6c7dce | 2007-10-24 18:54:45 +0000 | [diff] [blame] | 4796 | if (LHSC->getValueAPF().isNaN() || RHSC->getValueAPF().isNaN()) |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4797 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
4798 | |||||
4799 | // Otherwise, no need to compare the two constants, compare the | ||||
4800 | // rest. | ||||
4801 | return new FCmpInst(FCmpInst::FCMP_UNO, LHS->getOperand(0), | ||||
4802 | RHS->getOperand(0)); | ||||
4803 | } | ||||
Evan Cheng | 7298805 | 2008-10-14 18:44:08 +0000 | [diff] [blame] | 4804 | } else { |
4805 | Value *Op0LHS, *Op0RHS, *Op1LHS, *Op1RHS; | ||||
4806 | FCmpInst::Predicate Op0CC, Op1CC; | ||||
4807 | if (match(Op0, m_FCmp(Op0CC, m_Value(Op0LHS), m_Value(Op0RHS))) && | ||||
4808 | match(Op1, m_FCmp(Op1CC, m_Value(Op1LHS), m_Value(Op1RHS)))) { | ||||
4809 | if (Op0LHS == Op1RHS && Op0RHS == Op1LHS) { | ||||
4810 | // Swap RHS operands to match LHS. | ||||
4811 | Op1CC = FCmpInst::getSwappedPredicate(Op1CC); | ||||
4812 | std::swap(Op1LHS, Op1RHS); | ||||
4813 | } | ||||
4814 | if (Op0LHS == Op1LHS && Op0RHS == Op1RHS) { | ||||
4815 | // Simplify (fcmp cc0 x, y) | (fcmp cc1 x, y). | ||||
4816 | if (Op0CC == Op1CC) | ||||
4817 | return new FCmpInst((FCmpInst::Predicate)Op0CC, Op0LHS, Op0RHS); | ||||
4818 | else if (Op0CC == FCmpInst::FCMP_TRUE || | ||||
4819 | Op1CC == FCmpInst::FCMP_TRUE) | ||||
4820 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4821 | else if (Op0CC == FCmpInst::FCMP_FALSE) | ||||
4822 | return ReplaceInstUsesWith(I, Op1); | ||||
4823 | else if (Op1CC == FCmpInst::FCMP_FALSE) | ||||
4824 | return ReplaceInstUsesWith(I, Op0); | ||||
4825 | bool Op0Ordered; | ||||
4826 | bool Op1Ordered; | ||||
4827 | unsigned Op0Pred = getFCmpCode(Op0CC, Op0Ordered); | ||||
4828 | unsigned Op1Pred = getFCmpCode(Op1CC, Op1Ordered); | ||||
4829 | if (Op0Ordered == Op1Ordered) { | ||||
4830 | // If both are ordered or unordered, return a new fcmp with | ||||
4831 | // or'ed predicates. | ||||
4832 | Value *RV = getFCmpValue(Op0Ordered, Op0Pred|Op1Pred, | ||||
4833 | Op0LHS, Op0RHS); | ||||
4834 | if (Instruction *I = dyn_cast<Instruction>(RV)) | ||||
4835 | return I; | ||||
4836 | // Otherwise, it's a constant boolean value... | ||||
4837 | return ReplaceInstUsesWith(I, RV); | ||||
4838 | } | ||||
4839 | } | ||||
4840 | } | ||||
4841 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4842 | } |
4843 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4844 | |
4845 | return Changed ? &I : 0; | ||||
4846 | } | ||||
4847 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 4848 | namespace { |
4849 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4850 | // XorSelf - Implements: X ^ X --> 0 |
4851 | struct XorSelf { | ||||
4852 | Value *RHS; | ||||
4853 | XorSelf(Value *rhs) : RHS(rhs) {} | ||||
4854 | bool shouldApply(Value *LHS) const { return LHS == RHS; } | ||||
4855 | Instruction *apply(BinaryOperator &Xor) const { | ||||
4856 | return &Xor; | ||||
4857 | } | ||||
4858 | }; | ||||
4859 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 4860 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4861 | |
4862 | Instruction *InstCombiner::visitXor(BinaryOperator &I) { | ||||
4863 | bool Changed = SimplifyCommutative(I); | ||||
4864 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
4865 | |||||
Evan Cheng | e5cd803 | 2008-03-25 20:07:13 +0000 | [diff] [blame] | 4866 | if (isa<UndefValue>(Op1)) { |
4867 | if (isa<UndefValue>(Op0)) | ||||
4868 | // Handle undef ^ undef -> 0 special case. This is a common | ||||
4869 | // idiom (misuse). | ||||
4870 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4871 | return ReplaceInstUsesWith(I, Op1); // X ^ undef -> undef |
Evan Cheng | e5cd803 | 2008-03-25 20:07:13 +0000 | [diff] [blame] | 4872 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4873 | |
4874 | // xor X, X = 0, even if X is nested in a sequence of Xor's. | ||||
4875 | if (Instruction *Result = AssociativeOpt(I, XorSelf(Op1))) { | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 4876 | assert(Result == &I && "AssociativeOpt didn't work?"); Result=Result; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4877 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
4878 | } | ||||
4879 | |||||
4880 | // See if we can simplify any instructions used by the instruction whose sole | ||||
4881 | // purpose is to compute bits we don't care about. | ||||
4882 | if (!isa<VectorType>(I.getType())) { | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 4883 | if (SimplifyDemandedInstructionBits(I)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4884 | return &I; |
4885 | } else if (isa<ConstantAggregateZero>(Op1)) { | ||||
4886 | return ReplaceInstUsesWith(I, Op0); // X ^ <0,0> -> X | ||||
4887 | } | ||||
4888 | |||||
4889 | // Is this a ~ operation? | ||||
4890 | if (Value *NotOp = dyn_castNotVal(&I)) { | ||||
4891 | // ~(~X & Y) --> (X | ~Y) - De Morgan's Law | ||||
4892 | // ~(~X | Y) === (X & ~Y) - De Morgan's Law | ||||
4893 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(NotOp)) { | ||||
4894 | if (Op0I->getOpcode() == Instruction::And || | ||||
4895 | Op0I->getOpcode() == Instruction::Or) { | ||||
4896 | if (dyn_castNotVal(Op0I->getOperand(1))) Op0I->swapOperands(); | ||||
4897 | if (Value *Op0NotVal = dyn_castNotVal(Op0I->getOperand(0))) { | ||||
4898 | Instruction *NotY = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4899 | BinaryOperator::CreateNot(Op0I->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4900 | Op0I->getOperand(1)->getName()+".not"); |
4901 | InsertNewInstBefore(NotY, I); | ||||
4902 | if (Op0I->getOpcode() == Instruction::And) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4903 | return BinaryOperator::CreateOr(Op0NotVal, NotY); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4904 | else |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4905 | return BinaryOperator::CreateAnd(Op0NotVal, NotY); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4906 | } |
4907 | } | ||||
4908 | } | ||||
4909 | } | ||||
4910 | |||||
4911 | |||||
4912 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
Nick Lewycky | 1405e92 | 2007-08-06 20:04:16 +0000 | [diff] [blame] | 4913 | if (RHS == ConstantInt::getTrue() && Op0->hasOneUse()) { |
Bill Wendling | 6174195 | 2009-01-01 01:18:23 +0000 | [diff] [blame] | 4914 | // xor (cmp A, B), true = not (cmp A, B) = !cmp A, B |
Nick Lewycky | 1405e92 | 2007-08-06 20:04:16 +0000 | [diff] [blame] | 4915 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(Op0)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4916 | return new ICmpInst(ICI->getInversePredicate(), |
4917 | ICI->getOperand(0), ICI->getOperand(1)); | ||||
4918 | |||||
Nick Lewycky | 1405e92 | 2007-08-06 20:04:16 +0000 | [diff] [blame] | 4919 | if (FCmpInst *FCI = dyn_cast<FCmpInst>(Op0)) |
4920 | return new FCmpInst(FCI->getInversePredicate(), | ||||
4921 | FCI->getOperand(0), FCI->getOperand(1)); | ||||
4922 | } | ||||
4923 | |||||
Nick Lewycky | 0aa63aa | 2008-05-31 19:01:33 +0000 | [diff] [blame] | 4924 | // fold (xor(zext(cmp)), 1) and (xor(sext(cmp)), -1) to ext(!cmp). |
4925 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) { | ||||
4926 | if (CmpInst *CI = dyn_cast<CmpInst>(Op0C->getOperand(0))) { | ||||
4927 | if (CI->hasOneUse() && Op0C->hasOneUse()) { | ||||
4928 | Instruction::CastOps Opcode = Op0C->getOpcode(); | ||||
4929 | if (Opcode == Instruction::ZExt || Opcode == Instruction::SExt) { | ||||
4930 | if (RHS == ConstantExpr::getCast(Opcode, ConstantInt::getTrue(), | ||||
4931 | Op0C->getDestTy())) { | ||||
4932 | Instruction *NewCI = InsertNewInstBefore(CmpInst::Create( | ||||
4933 | CI->getOpcode(), CI->getInversePredicate(), | ||||
4934 | CI->getOperand(0), CI->getOperand(1)), I); | ||||
4935 | NewCI->takeName(CI); | ||||
4936 | return CastInst::Create(Opcode, NewCI, Op0C->getType()); | ||||
4937 | } | ||||
4938 | } | ||||
4939 | } | ||||
4940 | } | ||||
4941 | } | ||||
4942 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4943 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) { |
4944 | // ~(c-X) == X-c-1 == X+(-c-1) | ||||
4945 | if (Op0I->getOpcode() == Instruction::Sub && RHS->isAllOnesValue()) | ||||
4946 | if (Constant *Op0I0C = dyn_cast<Constant>(Op0I->getOperand(0))) { | ||||
4947 | Constant *NegOp0I0C = ConstantExpr::getNeg(Op0I0C); | ||||
4948 | Constant *ConstantRHS = ConstantExpr::getSub(NegOp0I0C, | ||||
4949 | ConstantInt::get(I.getType(), 1)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4950 | return BinaryOperator::CreateAdd(Op0I->getOperand(1), ConstantRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4951 | } |
4952 | |||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 4953 | if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4954 | if (Op0I->getOpcode() == Instruction::Add) { |
4955 | // ~(X-c) --> (-c-1)-X | ||||
4956 | if (RHS->isAllOnesValue()) { | ||||
4957 | Constant *NegOp0CI = ConstantExpr::getNeg(Op0CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4958 | return BinaryOperator::CreateSub( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4959 | ConstantExpr::getSub(NegOp0CI, |
4960 | ConstantInt::get(I.getType(), 1)), | ||||
4961 | Op0I->getOperand(0)); | ||||
4962 | } else if (RHS->getValue().isSignBit()) { | ||||
4963 | // (X + C) ^ signbit -> (X + C + signbit) | ||||
4964 | Constant *C = ConstantInt::get(RHS->getValue() + Op0CI->getValue()); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4965 | return BinaryOperator::CreateAdd(Op0I->getOperand(0), C); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4966 | |
4967 | } | ||||
4968 | } else if (Op0I->getOpcode() == Instruction::Or) { | ||||
4969 | // (X|C1)^C2 -> X^(C1|C2) iff X&~C1 == 0 | ||||
4970 | if (MaskedValueIsZero(Op0I->getOperand(0), Op0CI->getValue())) { | ||||
4971 | Constant *NewRHS = ConstantExpr::getOr(Op0CI, RHS); | ||||
4972 | // Anything in both C1 and C2 is known to be zero, remove it from | ||||
4973 | // NewRHS. | ||||
4974 | Constant *CommonBits = And(Op0CI, RHS); | ||||
4975 | NewRHS = ConstantExpr::getAnd(NewRHS, | ||||
4976 | ConstantExpr::getNot(CommonBits)); | ||||
4977 | AddToWorkList(Op0I); | ||||
4978 | I.setOperand(0, Op0I->getOperand(0)); | ||||
4979 | I.setOperand(1, NewRHS); | ||||
4980 | return &I; | ||||
4981 | } | ||||
4982 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 4983 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4984 | } |
4985 | |||||
4986 | // Try to fold constant and into select arguments. | ||||
4987 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
4988 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
4989 | return R; | ||||
4990 | if (isa<PHINode>(Op0)) | ||||
4991 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
4992 | return NV; | ||||
4993 | } | ||||
4994 | |||||
4995 | if (Value *X = dyn_castNotVal(Op0)) // ~A ^ A == -1 | ||||
4996 | if (X == Op1) | ||||
4997 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4998 | |||||
4999 | if (Value *X = dyn_castNotVal(Op1)) // A ^ ~A == -1 | ||||
5000 | if (X == Op0) | ||||
5001 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
5002 | |||||
5003 | |||||
5004 | BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1); | ||||
5005 | if (Op1I) { | ||||
5006 | Value *A, *B; | ||||
5007 | if (match(Op1I, m_Or(m_Value(A), m_Value(B)))) { | ||||
5008 | if (A == Op0) { // B^(B|A) == (A|B)^B | ||||
5009 | Op1I->swapOperands(); | ||||
5010 | I.swapOperands(); | ||||
5011 | std::swap(Op0, Op1); | ||||
5012 | } else if (B == Op0) { // B^(A|B) == (A|B)^B | ||||
5013 | I.swapOperands(); // Simplified below. | ||||
5014 | std::swap(Op0, Op1); | ||||
5015 | } | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 5016 | } else if (match(Op1I, m_Xor(m_Specific(Op0), m_Value(B)))) { |
5017 | return ReplaceInstUsesWith(I, B); // A^(A^B) == B | ||||
5018 | } else if (match(Op1I, m_Xor(m_Value(A), m_Specific(Op0)))) { | ||||
5019 | return ReplaceInstUsesWith(I, A); // A^(B^A) == B | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5020 | } else if (match(Op1I, m_And(m_Value(A), m_Value(B))) && Op1I->hasOneUse()){ |
5021 | if (A == Op0) { // A^(A&B) -> A^(B&A) | ||||
5022 | Op1I->swapOperands(); | ||||
5023 | std::swap(A, B); | ||||
5024 | } | ||||
5025 | if (B == Op0) { // A^(B&A) -> (B&A)^A | ||||
5026 | I.swapOperands(); // Simplified below. | ||||
5027 | std::swap(Op0, Op1); | ||||
5028 | } | ||||
5029 | } | ||||
5030 | } | ||||
5031 | |||||
5032 | BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0); | ||||
5033 | if (Op0I) { | ||||
5034 | Value *A, *B; | ||||
5035 | if (match(Op0I, m_Or(m_Value(A), m_Value(B))) && Op0I->hasOneUse()) { | ||||
5036 | if (A == Op1) // (B|A)^B == (A|B)^B | ||||
5037 | std::swap(A, B); | ||||
5038 | if (B == Op1) { // (A|B)^B == A & ~B | ||||
5039 | Instruction *NotB = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5040 | InsertNewInstBefore(BinaryOperator::CreateNot(Op1, "tmp"), I); |
5041 | return BinaryOperator::CreateAnd(A, NotB); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5042 | } |
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 5043 | } else if (match(Op0I, m_Xor(m_Specific(Op1), m_Value(B)))) { |
5044 | return ReplaceInstUsesWith(I, B); // (A^B)^A == B | ||||
5045 | } else if (match(Op0I, m_Xor(m_Value(A), m_Specific(Op1)))) { | ||||
5046 | return ReplaceInstUsesWith(I, A); // (B^A)^A == B | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5047 | } else if (match(Op0I, m_And(m_Value(A), m_Value(B))) && Op0I->hasOneUse()){ |
5048 | if (A == Op1) // (A&B)^A -> (B&A)^A | ||||
5049 | std::swap(A, B); | ||||
5050 | if (B == Op1 && // (B&A)^A == ~B & A | ||||
5051 | !isa<ConstantInt>(Op1)) { // Canonical form is (B&C)^C | ||||
5052 | Instruction *N = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5053 | InsertNewInstBefore(BinaryOperator::CreateNot(A, "tmp"), I); |
5054 | return BinaryOperator::CreateAnd(N, Op1); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5055 | } |
5056 | } | ||||
5057 | } | ||||
5058 | |||||
5059 | // (X >> Z) ^ (Y >> Z) -> (X^Y) >> Z for all shifts. | ||||
5060 | if (Op0I && Op1I && Op0I->isShift() && | ||||
5061 | Op0I->getOpcode() == Op1I->getOpcode() && | ||||
5062 | Op0I->getOperand(1) == Op1I->getOperand(1) && | ||||
5063 | (Op1I->hasOneUse() || Op1I->hasOneUse())) { | ||||
5064 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5065 | InsertNewInstBefore(BinaryOperator::CreateXor(Op0I->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5066 | Op1I->getOperand(0), |
5067 | Op0I->getName()), I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5068 | return BinaryOperator::Create(Op1I->getOpcode(), NewOp, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5069 | Op1I->getOperand(1)); |
5070 | } | ||||
5071 | |||||
5072 | if (Op0I && Op1I) { | ||||
5073 | Value *A, *B, *C, *D; | ||||
5074 | // (A & B)^(A | B) -> A ^ B | ||||
5075 | if (match(Op0I, m_And(m_Value(A), m_Value(B))) && | ||||
5076 | match(Op1I, m_Or(m_Value(C), m_Value(D)))) { | ||||
5077 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5078 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5079 | } |
5080 | // (A | B)^(A & B) -> A ^ B | ||||
5081 | if (match(Op0I, m_Or(m_Value(A), m_Value(B))) && | ||||
5082 | match(Op1I, m_And(m_Value(C), m_Value(D)))) { | ||||
5083 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5084 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5085 | } |
5086 | |||||
5087 | // (A & B)^(C & D) | ||||
5088 | if ((Op0I->hasOneUse() || Op1I->hasOneUse()) && | ||||
5089 | match(Op0I, m_And(m_Value(A), m_Value(B))) && | ||||
5090 | match(Op1I, m_And(m_Value(C), m_Value(D)))) { | ||||
5091 | // (X & Y)^(X & Y) -> (Y^Z) & X | ||||
5092 | Value *X = 0, *Y = 0, *Z = 0; | ||||
5093 | if (A == C) | ||||
5094 | X = A, Y = B, Z = D; | ||||
5095 | else if (A == D) | ||||
5096 | X = A, Y = B, Z = C; | ||||
5097 | else if (B == C) | ||||
5098 | X = B, Y = A, Z = D; | ||||
5099 | else if (B == D) | ||||
5100 | X = B, Y = A, Z = C; | ||||
5101 | |||||
5102 | if (X) { | ||||
5103 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5104 | InsertNewInstBefore(BinaryOperator::CreateXor(Y, Z, Op0->getName()), I); |
5105 | return BinaryOperator::CreateAnd(NewOp, X); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5106 | } |
5107 | } | ||||
5108 | } | ||||
5109 | |||||
5110 | // (icmp1 A, B) ^ (icmp2 A, B) --> (icmp3 A, B) | ||||
5111 | if (ICmpInst *RHS = dyn_cast<ICmpInst>(I.getOperand(1))) | ||||
5112 | if (Instruction *R = AssociativeOpt(I, FoldICmpLogical(*this, RHS))) | ||||
5113 | return R; | ||||
5114 | |||||
5115 | // fold (xor (cast A), (cast B)) -> (cast (xor A, B)) | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 5116 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5117 | if (CastInst *Op1C = dyn_cast<CastInst>(Op1)) |
5118 | if (Op0C->getOpcode() == Op1C->getOpcode()) { // same cast kind? | ||||
5119 | const Type *SrcTy = Op0C->getOperand(0)->getType(); | ||||
5120 | if (SrcTy == Op1C->getOperand(0)->getType() && SrcTy->isInteger() && | ||||
5121 | // Only do this if the casts both really cause code to be generated. | ||||
5122 | ValueRequiresCast(Op0C->getOpcode(), Op0C->getOperand(0), | ||||
5123 | I.getType(), TD) && | ||||
5124 | ValueRequiresCast(Op1C->getOpcode(), Op1C->getOperand(0), | ||||
5125 | I.getType(), TD)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5126 | Instruction *NewOp = BinaryOperator::CreateXor(Op0C->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5127 | Op1C->getOperand(0), |
5128 | I.getName()); | ||||
5129 | InsertNewInstBefore(NewOp, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5130 | return CastInst::Create(Op0C->getOpcode(), NewOp, I.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5131 | } |
5132 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 5133 | } |
Nick Lewycky | 0aa63aa | 2008-05-31 19:01:33 +0000 | [diff] [blame] | 5134 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5135 | return Changed ? &I : 0; |
5136 | } | ||||
5137 | |||||
5138 | /// AddWithOverflow - Compute Result = In1+In2, returning true if the result | ||||
5139 | /// overflowed for this type. | ||||
5140 | static bool AddWithOverflow(ConstantInt *&Result, ConstantInt *In1, | ||||
5141 | ConstantInt *In2, bool IsSigned = false) { | ||||
5142 | Result = cast<ConstantInt>(Add(In1, In2)); | ||||
5143 | |||||
5144 | if (IsSigned) | ||||
5145 | if (In2->getValue().isNegative()) | ||||
5146 | return Result->getValue().sgt(In1->getValue()); | ||||
5147 | else | ||||
5148 | return Result->getValue().slt(In1->getValue()); | ||||
5149 | else | ||||
5150 | return Result->getValue().ult(In1->getValue()); | ||||
5151 | } | ||||
5152 | |||||
Dan Gohman | b80d561 | 2008-09-10 23:30:57 +0000 | [diff] [blame] | 5153 | /// SubWithOverflow - Compute Result = In1-In2, returning true if the result |
5154 | /// overflowed for this type. | ||||
5155 | static bool SubWithOverflow(ConstantInt *&Result, ConstantInt *In1, | ||||
5156 | ConstantInt *In2, bool IsSigned = false) { | ||||
Dan Gohman | 2c3b489 | 2008-09-11 18:53:02 +0000 | [diff] [blame] | 5157 | Result = cast<ConstantInt>(Subtract(In1, In2)); |
Dan Gohman | b80d561 | 2008-09-10 23:30:57 +0000 | [diff] [blame] | 5158 | |
5159 | if (IsSigned) | ||||
5160 | if (In2->getValue().isNegative()) | ||||
5161 | return Result->getValue().slt(In1->getValue()); | ||||
5162 | else | ||||
5163 | return Result->getValue().sgt(In1->getValue()); | ||||
5164 | else | ||||
5165 | return Result->getValue().ugt(In1->getValue()); | ||||
5166 | } | ||||
5167 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5168 | /// EmitGEPOffset - Given a getelementptr instruction/constantexpr, emit the |
5169 | /// code necessary to compute the offset from the base pointer (without adding | ||||
5170 | /// in the base pointer). Return the result as a signed integer of intptr size. | ||||
5171 | static Value *EmitGEPOffset(User *GEP, Instruction &I, InstCombiner &IC) { | ||||
5172 | TargetData &TD = IC.getTargetData(); | ||||
5173 | gep_type_iterator GTI = gep_type_begin(GEP); | ||||
5174 | const Type *IntPtrTy = TD.getIntPtrType(); | ||||
5175 | Value *Result = Constant::getNullValue(IntPtrTy); | ||||
5176 | |||||
5177 | // Build a mask for high order bits. | ||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5178 | unsigned IntPtrWidth = TD.getPointerSizeInBits(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5179 | uint64_t PtrSizeMask = ~0ULL >> (64-IntPtrWidth); |
5180 | |||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 5181 | for (User::op_iterator i = GEP->op_begin() + 1, e = GEP->op_end(); i != e; |
5182 | ++i, ++GTI) { | ||||
5183 | Value *Op = *i; | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 5184 | uint64_t Size = TD.getTypePaddedSize(GTI.getIndexedType()) & PtrSizeMask; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5185 | if (ConstantInt *OpC = dyn_cast<ConstantInt>(Op)) { |
5186 | if (OpC->isZero()) continue; | ||||
5187 | |||||
5188 | // Handle a struct index, which adds its field offset to the pointer. | ||||
5189 | if (const StructType *STy = dyn_cast<StructType>(*GTI)) { | ||||
5190 | Size = TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue()); | ||||
5191 | |||||
5192 | if (ConstantInt *RC = dyn_cast<ConstantInt>(Result)) | ||||
5193 | Result = ConstantInt::get(RC->getValue() + APInt(IntPtrWidth, Size)); | ||||
5194 | else | ||||
5195 | Result = IC.InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5196 | BinaryOperator::CreateAdd(Result, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5197 | ConstantInt::get(IntPtrTy, Size), |
5198 | GEP->getName()+".offs"), I); | ||||
5199 | continue; | ||||
5200 | } | ||||
5201 | |||||
5202 | Constant *Scale = ConstantInt::get(IntPtrTy, Size); | ||||
5203 | Constant *OC = ConstantExpr::getIntegerCast(OpC, IntPtrTy, true /*SExt*/); | ||||
5204 | Scale = ConstantExpr::getMul(OC, Scale); | ||||
5205 | if (Constant *RC = dyn_cast<Constant>(Result)) | ||||
5206 | Result = ConstantExpr::getAdd(RC, Scale); | ||||
5207 | else { | ||||
5208 | // Emit an add instruction. | ||||
5209 | Result = IC.InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5210 | BinaryOperator::CreateAdd(Result, Scale, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5211 | GEP->getName()+".offs"), I); |
5212 | } | ||||
5213 | continue; | ||||
5214 | } | ||||
5215 | // Convert to correct type. | ||||
5216 | if (Op->getType() != IntPtrTy) { | ||||
5217 | if (Constant *OpC = dyn_cast<Constant>(Op)) | ||||
5218 | Op = ConstantExpr::getSExt(OpC, IntPtrTy); | ||||
5219 | else | ||||
5220 | Op = IC.InsertNewInstBefore(new SExtInst(Op, IntPtrTy, | ||||
5221 | Op->getName()+".c"), I); | ||||
5222 | } | ||||
5223 | if (Size != 1) { | ||||
5224 | Constant *Scale = ConstantInt::get(IntPtrTy, Size); | ||||
5225 | if (Constant *OpC = dyn_cast<Constant>(Op)) | ||||
5226 | Op = ConstantExpr::getMul(OpC, Scale); | ||||
5227 | else // We'll let instcombine(mul) convert this to a shl if possible. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5228 | Op = IC.InsertNewInstBefore(BinaryOperator::CreateMul(Op, Scale, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5229 | GEP->getName()+".idx"), I); |
5230 | } | ||||
5231 | |||||
5232 | // Emit an add instruction. | ||||
5233 | if (isa<Constant>(Op) && isa<Constant>(Result)) | ||||
5234 | Result = ConstantExpr::getAdd(cast<Constant>(Op), | ||||
5235 | cast<Constant>(Result)); | ||||
5236 | else | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5237 | Result = IC.InsertNewInstBefore(BinaryOperator::CreateAdd(Op, Result, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5238 | GEP->getName()+".offs"), I); |
5239 | } | ||||
5240 | return Result; | ||||
5241 | } | ||||
5242 | |||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5243 | |
5244 | /// EvaluateGEPOffsetExpression - Return an value that can be used to compare of | ||||
5245 | /// the *offset* implied by GEP to zero. For example, if we have &A[i], we want | ||||
5246 | /// to return 'i' for "icmp ne i, 0". Note that, in general, indices can be | ||||
5247 | /// complex, and scales are involved. The above expression would also be legal | ||||
5248 | /// to codegen as "icmp ne (i*4), 0" (assuming A is a pointer to i32). This | ||||
5249 | /// later form is less amenable to optimization though, and we are allowed to | ||||
5250 | /// generate the first by knowing that pointer arithmetic doesn't overflow. | ||||
5251 | /// | ||||
5252 | /// If we can't emit an optimized form for this expression, this returns null. | ||||
5253 | /// | ||||
5254 | static Value *EvaluateGEPOffsetExpression(User *GEP, Instruction &I, | ||||
5255 | InstCombiner &IC) { | ||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5256 | TargetData &TD = IC.getTargetData(); |
5257 | gep_type_iterator GTI = gep_type_begin(GEP); | ||||
5258 | |||||
5259 | // Check to see if this gep only has a single variable index. If so, and if | ||||
5260 | // any constant indices are a multiple of its scale, then we can compute this | ||||
5261 | // in terms of the scale of the variable index. For example, if the GEP | ||||
5262 | // implies an offset of "12 + i*4", then we can codegen this as "3 + i", | ||||
5263 | // because the expression will cross zero at the same point. | ||||
5264 | unsigned i, e = GEP->getNumOperands(); | ||||
5265 | int64_t Offset = 0; | ||||
5266 | for (i = 1; i != e; ++i, ++GTI) { | ||||
5267 | if (ConstantInt *CI = dyn_cast<ConstantInt>(GEP->getOperand(i))) { | ||||
5268 | // Compute the aggregate offset of constant indices. | ||||
5269 | if (CI->isZero()) continue; | ||||
5270 | |||||
5271 | // Handle a struct index, which adds its field offset to the pointer. | ||||
5272 | if (const StructType *STy = dyn_cast<StructType>(*GTI)) { | ||||
5273 | Offset += TD.getStructLayout(STy)->getElementOffset(CI->getZExtValue()); | ||||
5274 | } else { | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 5275 | uint64_t Size = TD.getTypePaddedSize(GTI.getIndexedType()); |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5276 | Offset += Size*CI->getSExtValue(); |
5277 | } | ||||
5278 | } else { | ||||
5279 | // Found our variable index. | ||||
5280 | break; | ||||
5281 | } | ||||
5282 | } | ||||
5283 | |||||
5284 | // If there are no variable indices, we must have a constant offset, just | ||||
5285 | // evaluate it the general way. | ||||
5286 | if (i == e) return 0; | ||||
5287 | |||||
5288 | Value *VariableIdx = GEP->getOperand(i); | ||||
5289 | // Determine the scale factor of the variable element. For example, this is | ||||
5290 | // 4 if the variable index is into an array of i32. | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 5291 | uint64_t VariableScale = TD.getTypePaddedSize(GTI.getIndexedType()); |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5292 | |
5293 | // Verify that there are no other variable indices. If so, emit the hard way. | ||||
5294 | for (++i, ++GTI; i != e; ++i, ++GTI) { | ||||
5295 | ConstantInt *CI = dyn_cast<ConstantInt>(GEP->getOperand(i)); | ||||
5296 | if (!CI) return 0; | ||||
5297 | |||||
5298 | // Compute the aggregate offset of constant indices. | ||||
5299 | if (CI->isZero()) continue; | ||||
5300 | |||||
5301 | // Handle a struct index, which adds its field offset to the pointer. | ||||
5302 | if (const StructType *STy = dyn_cast<StructType>(*GTI)) { | ||||
5303 | Offset += TD.getStructLayout(STy)->getElementOffset(CI->getZExtValue()); | ||||
5304 | } else { | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 5305 | uint64_t Size = TD.getTypePaddedSize(GTI.getIndexedType()); |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5306 | Offset += Size*CI->getSExtValue(); |
5307 | } | ||||
5308 | } | ||||
5309 | |||||
5310 | // Okay, we know we have a single variable index, which must be a | ||||
5311 | // pointer/array/vector index. If there is no offset, life is simple, return | ||||
5312 | // the index. | ||||
5313 | unsigned IntPtrWidth = TD.getPointerSizeInBits(); | ||||
5314 | if (Offset == 0) { | ||||
5315 | // Cast to intptrty in case a truncation occurs. If an extension is needed, | ||||
5316 | // we don't need to bother extending: the extension won't affect where the | ||||
5317 | // computation crosses zero. | ||||
5318 | if (VariableIdx->getType()->getPrimitiveSizeInBits() > IntPtrWidth) | ||||
5319 | VariableIdx = new TruncInst(VariableIdx, TD.getIntPtrType(), | ||||
5320 | VariableIdx->getNameStart(), &I); | ||||
5321 | return VariableIdx; | ||||
5322 | } | ||||
5323 | |||||
5324 | // Otherwise, there is an index. The computation we will do will be modulo | ||||
5325 | // the pointer size, so get it. | ||||
5326 | uint64_t PtrSizeMask = ~0ULL >> (64-IntPtrWidth); | ||||
5327 | |||||
5328 | Offset &= PtrSizeMask; | ||||
5329 | VariableScale &= PtrSizeMask; | ||||
5330 | |||||
5331 | // To do this transformation, any constant index must be a multiple of the | ||||
5332 | // variable scale factor. For example, we can evaluate "12 + 4*i" as "3 + i", | ||||
5333 | // but we can't evaluate "10 + 3*i" in terms of i. Check that the offset is a | ||||
5334 | // multiple of the variable scale. | ||||
5335 | int64_t NewOffs = Offset / (int64_t)VariableScale; | ||||
5336 | if (Offset != NewOffs*(int64_t)VariableScale) | ||||
5337 | return 0; | ||||
5338 | |||||
5339 | // Okay, we can do this evaluation. Start by converting the index to intptr. | ||||
5340 | const Type *IntPtrTy = TD.getIntPtrType(); | ||||
5341 | if (VariableIdx->getType() != IntPtrTy) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5342 | VariableIdx = CastInst::CreateIntegerCast(VariableIdx, IntPtrTy, |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5343 | true /*SExt*/, |
5344 | VariableIdx->getNameStart(), &I); | ||||
5345 | Constant *OffsetVal = ConstantInt::get(IntPtrTy, NewOffs); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5346 | return BinaryOperator::CreateAdd(VariableIdx, OffsetVal, "offset", &I); |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5347 | } |
5348 | |||||
5349 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5350 | /// FoldGEPICmp - Fold comparisons between a GEP instruction and something |
5351 | /// else. At this point we know that the GEP is on the LHS of the comparison. | ||||
5352 | Instruction *InstCombiner::FoldGEPICmp(User *GEPLHS, Value *RHS, | ||||
5353 | ICmpInst::Predicate Cond, | ||||
5354 | Instruction &I) { | ||||
5355 | assert(dyn_castGetElementPtr(GEPLHS) && "LHS is not a getelementptr!"); | ||||
5356 | |||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5357 | // Look through bitcasts. |
5358 | if (BitCastInst *BCI = dyn_cast<BitCastInst>(RHS)) | ||||
5359 | RHS = BCI->getOperand(0); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5360 | |
5361 | Value *PtrBase = GEPLHS->getOperand(0); | ||||
5362 | if (PtrBase == RHS) { | ||||
Chris Lattner | af97d02 | 2008-02-05 04:45:32 +0000 | [diff] [blame] | 5363 | // ((gep Ptr, OFFSET) cmp Ptr) ---> (OFFSET cmp 0). |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5364 | // This transformation (ignoring the base and scales) is valid because we |
5365 | // know pointers can't overflow. See if we can output an optimized form. | ||||
5366 | Value *Offset = EvaluateGEPOffsetExpression(GEPLHS, I, *this); | ||||
5367 | |||||
5368 | // If not, synthesize the offset the hard way. | ||||
5369 | if (Offset == 0) | ||||
5370 | Offset = EmitGEPOffset(GEPLHS, I, *this); | ||||
Chris Lattner | af97d02 | 2008-02-05 04:45:32 +0000 | [diff] [blame] | 5371 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), Offset, |
5372 | Constant::getNullValue(Offset->getType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5373 | } else if (User *GEPRHS = dyn_castGetElementPtr(RHS)) { |
5374 | // If the base pointers are different, but the indices are the same, just | ||||
5375 | // compare the base pointer. | ||||
5376 | if (PtrBase != GEPRHS->getOperand(0)) { | ||||
5377 | bool IndicesTheSame = GEPLHS->getNumOperands()==GEPRHS->getNumOperands(); | ||||
5378 | IndicesTheSame &= GEPLHS->getOperand(0)->getType() == | ||||
5379 | GEPRHS->getOperand(0)->getType(); | ||||
5380 | if (IndicesTheSame) | ||||
5381 | for (unsigned i = 1, e = GEPLHS->getNumOperands(); i != e; ++i) | ||||
5382 | if (GEPLHS->getOperand(i) != GEPRHS->getOperand(i)) { | ||||
5383 | IndicesTheSame = false; | ||||
5384 | break; | ||||
5385 | } | ||||
5386 | |||||
5387 | // If all indices are the same, just compare the base pointers. | ||||
5388 | if (IndicesTheSame) | ||||
5389 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), | ||||
5390 | GEPLHS->getOperand(0), GEPRHS->getOperand(0)); | ||||
5391 | |||||
5392 | // Otherwise, the base pointers are different and the indices are | ||||
5393 | // different, bail out. | ||||
5394 | return 0; | ||||
5395 | } | ||||
5396 | |||||
5397 | // If one of the GEPs has all zero indices, recurse. | ||||
5398 | bool AllZeros = true; | ||||
5399 | for (unsigned i = 1, e = GEPLHS->getNumOperands(); i != e; ++i) | ||||
5400 | if (!isa<Constant>(GEPLHS->getOperand(i)) || | ||||
5401 | !cast<Constant>(GEPLHS->getOperand(i))->isNullValue()) { | ||||
5402 | AllZeros = false; | ||||
5403 | break; | ||||
5404 | } | ||||
5405 | if (AllZeros) | ||||
5406 | return FoldGEPICmp(GEPRHS, GEPLHS->getOperand(0), | ||||
5407 | ICmpInst::getSwappedPredicate(Cond), I); | ||||
5408 | |||||
5409 | // If the other GEP has all zero indices, recurse. | ||||
5410 | AllZeros = true; | ||||
5411 | for (unsigned i = 1, e = GEPRHS->getNumOperands(); i != e; ++i) | ||||
5412 | if (!isa<Constant>(GEPRHS->getOperand(i)) || | ||||
5413 | !cast<Constant>(GEPRHS->getOperand(i))->isNullValue()) { | ||||
5414 | AllZeros = false; | ||||
5415 | break; | ||||
5416 | } | ||||
5417 | if (AllZeros) | ||||
5418 | return FoldGEPICmp(GEPLHS, GEPRHS->getOperand(0), Cond, I); | ||||
5419 | |||||
5420 | if (GEPLHS->getNumOperands() == GEPRHS->getNumOperands()) { | ||||
5421 | // If the GEPs only differ by one index, compare it. | ||||
5422 | unsigned NumDifferences = 0; // Keep track of # differences. | ||||
5423 | unsigned DiffOperand = 0; // The operand that differs. | ||||
5424 | for (unsigned i = 1, e = GEPRHS->getNumOperands(); i != e; ++i) | ||||
5425 | if (GEPLHS->getOperand(i) != GEPRHS->getOperand(i)) { | ||||
5426 | if (GEPLHS->getOperand(i)->getType()->getPrimitiveSizeInBits() != | ||||
5427 | GEPRHS->getOperand(i)->getType()->getPrimitiveSizeInBits()) { | ||||
5428 | // Irreconcilable differences. | ||||
5429 | NumDifferences = 2; | ||||
5430 | break; | ||||
5431 | } else { | ||||
5432 | if (NumDifferences++) break; | ||||
5433 | DiffOperand = i; | ||||
5434 | } | ||||
5435 | } | ||||
5436 | |||||
5437 | if (NumDifferences == 0) // SAME GEP? | ||||
5438 | return ReplaceInstUsesWith(I, // No comparison is needed here. | ||||
Nick Lewycky | 2de09a9 | 2007-09-06 02:40:25 +0000 | [diff] [blame] | 5439 | ConstantInt::get(Type::Int1Ty, |
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 5440 | ICmpInst::isTrueWhenEqual(Cond))); |
Nick Lewycky | 2de09a9 | 2007-09-06 02:40:25 +0000 | [diff] [blame] | 5441 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5442 | else if (NumDifferences == 1) { |
5443 | Value *LHSV = GEPLHS->getOperand(DiffOperand); | ||||
5444 | Value *RHSV = GEPRHS->getOperand(DiffOperand); | ||||
5445 | // Make sure we do a signed comparison here. | ||||
5446 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), LHSV, RHSV); | ||||
5447 | } | ||||
5448 | } | ||||
5449 | |||||
5450 | // Only lower this if the icmp is the only user of the GEP or if we expect | ||||
5451 | // the result to fold to a constant! | ||||
5452 | if ((isa<ConstantExpr>(GEPLHS) || GEPLHS->hasOneUse()) && | ||||
5453 | (isa<ConstantExpr>(GEPRHS) || GEPRHS->hasOneUse())) { | ||||
5454 | // ((gep Ptr, OFFSET1) cmp (gep Ptr, OFFSET2) ---> (OFFSET1 cmp OFFSET2) | ||||
5455 | Value *L = EmitGEPOffset(GEPLHS, I, *this); | ||||
5456 | Value *R = EmitGEPOffset(GEPRHS, I, *this); | ||||
5457 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), L, R); | ||||
5458 | } | ||||
5459 | } | ||||
5460 | return 0; | ||||
5461 | } | ||||
5462 | |||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5463 | /// FoldFCmp_IntToFP_Cst - Fold fcmp ([us]itofp x, cst) if possible. |
5464 | /// | ||||
5465 | Instruction *InstCombiner::FoldFCmp_IntToFP_Cst(FCmpInst &I, | ||||
5466 | Instruction *LHSI, | ||||
5467 | Constant *RHSC) { | ||||
5468 | if (!isa<ConstantFP>(RHSC)) return 0; | ||||
5469 | const APFloat &RHS = cast<ConstantFP>(RHSC)->getValueAPF(); | ||||
5470 | |||||
5471 | // Get the width of the mantissa. We don't want to hack on conversions that | ||||
5472 | // might lose information from the integer, e.g. "i64 -> float" | ||||
Chris Lattner | 9ce836b | 2008-05-19 21:17:23 +0000 | [diff] [blame] | 5473 | int MantissaWidth = LHSI->getType()->getFPMantissaWidth(); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5474 | if (MantissaWidth == -1) return 0; // Unknown. |
5475 | |||||
5476 | // Check to see that the input is converted from an integer type that is small | ||||
5477 | // enough that preserves all bits. TODO: check here for "known" sign bits. | ||||
5478 | // This would allow us to handle (fptosi (x >>s 62) to float) if x is i64 f.e. | ||||
5479 | unsigned InputSize = LHSI->getOperand(0)->getType()->getPrimitiveSizeInBits(); | ||||
5480 | |||||
5481 | // If this is a uitofp instruction, we need an extra bit to hold the sign. | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5482 | bool LHSUnsigned = isa<UIToFPInst>(LHSI); |
5483 | if (LHSUnsigned) | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5484 | ++InputSize; |
5485 | |||||
5486 | // If the conversion would lose info, don't hack on this. | ||||
5487 | if ((int)InputSize > MantissaWidth) | ||||
5488 | return 0; | ||||
5489 | |||||
5490 | // Otherwise, we can potentially simplify the comparison. We know that it | ||||
5491 | // will always come through as an integer value and we know the constant is | ||||
5492 | // not a NAN (it would have been previously simplified). | ||||
5493 | assert(!RHS.isNaN() && "NaN comparison not already folded!"); | ||||
5494 | |||||
5495 | ICmpInst::Predicate Pred; | ||||
5496 | switch (I.getPredicate()) { | ||||
5497 | default: assert(0 && "Unexpected predicate!"); | ||||
5498 | case FCmpInst::FCMP_UEQ: | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5499 | case FCmpInst::FCMP_OEQ: |
5500 | Pred = ICmpInst::ICMP_EQ; | ||||
5501 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5502 | case FCmpInst::FCMP_UGT: |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5503 | case FCmpInst::FCMP_OGT: |
5504 | Pred = LHSUnsigned ? ICmpInst::ICMP_UGT : ICmpInst::ICMP_SGT; | ||||
5505 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5506 | case FCmpInst::FCMP_UGE: |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5507 | case FCmpInst::FCMP_OGE: |
5508 | Pred = LHSUnsigned ? ICmpInst::ICMP_UGE : ICmpInst::ICMP_SGE; | ||||
5509 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5510 | case FCmpInst::FCMP_ULT: |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5511 | case FCmpInst::FCMP_OLT: |
5512 | Pred = LHSUnsigned ? ICmpInst::ICMP_ULT : ICmpInst::ICMP_SLT; | ||||
5513 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5514 | case FCmpInst::FCMP_ULE: |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5515 | case FCmpInst::FCMP_OLE: |
5516 | Pred = LHSUnsigned ? ICmpInst::ICMP_ULE : ICmpInst::ICMP_SLE; | ||||
5517 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5518 | case FCmpInst::FCMP_UNE: |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5519 | case FCmpInst::FCMP_ONE: |
5520 | Pred = ICmpInst::ICMP_NE; | ||||
5521 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5522 | case FCmpInst::FCMP_ORD: |
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5523 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5524 | case FCmpInst::FCMP_UNO: |
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5525 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5526 | } |
5527 | |||||
5528 | const IntegerType *IntTy = cast<IntegerType>(LHSI->getOperand(0)->getType()); | ||||
5529 | |||||
5530 | // Now we know that the APFloat is a normal number, zero or inf. | ||||
5531 | |||||
Chris Lattner | f13ff49 | 2008-05-20 03:50:52 +0000 | [diff] [blame] | 5532 | // See if the FP constant is too large for the integer. For example, |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5533 | // comparing an i8 to 300.0. |
5534 | unsigned IntWidth = IntTy->getPrimitiveSizeInBits(); | ||||
5535 | |||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5536 | if (!LHSUnsigned) { |
5537 | // If the RHS value is > SignedMax, fold the comparison. This handles +INF | ||||
5538 | // and large values. | ||||
5539 | APFloat SMax(RHS.getSemantics(), APFloat::fcZero, false); | ||||
5540 | SMax.convertFromAPInt(APInt::getSignedMaxValue(IntWidth), true, | ||||
5541 | APFloat::rmNearestTiesToEven); | ||||
5542 | if (SMax.compare(RHS) == APFloat::cmpLessThan) { // smax < 13123.0 | ||||
5543 | if (Pred == ICmpInst::ICMP_NE || Pred == ICmpInst::ICMP_SLT || | ||||
5544 | Pred == ICmpInst::ICMP_SLE) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5545 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
5546 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5547 | } |
5548 | } else { | ||||
5549 | // If the RHS value is > UnsignedMax, fold the comparison. This handles | ||||
5550 | // +INF and large values. | ||||
5551 | APFloat UMax(RHS.getSemantics(), APFloat::fcZero, false); | ||||
5552 | UMax.convertFromAPInt(APInt::getMaxValue(IntWidth), false, | ||||
5553 | APFloat::rmNearestTiesToEven); | ||||
5554 | if (UMax.compare(RHS) == APFloat::cmpLessThan) { // umax < 13123.0 | ||||
5555 | if (Pred == ICmpInst::ICMP_NE || Pred == ICmpInst::ICMP_ULT || | ||||
5556 | Pred == ICmpInst::ICMP_ULE) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5557 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
5558 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5559 | } |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5560 | } |
5561 | |||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5562 | if (!LHSUnsigned) { |
5563 | // See if the RHS value is < SignedMin. | ||||
5564 | APFloat SMin(RHS.getSemantics(), APFloat::fcZero, false); | ||||
5565 | SMin.convertFromAPInt(APInt::getSignedMinValue(IntWidth), true, | ||||
5566 | APFloat::rmNearestTiesToEven); | ||||
5567 | if (SMin.compare(RHS) == APFloat::cmpGreaterThan) { // smin > 12312.0 | ||||
5568 | if (Pred == ICmpInst::ICMP_NE || Pred == ICmpInst::ICMP_SGT || | ||||
5569 | Pred == ICmpInst::ICMP_SGE) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5570 | return ReplaceInstUsesWith(I,ConstantInt::getTrue()); |
5571 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5572 | } |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5573 | } |
5574 | |||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5575 | // Okay, now we know that the FP constant fits in the range [SMIN, SMAX] or |
5576 | // [0, UMAX], but it may still be fractional. See if it is fractional by | ||||
5577 | // casting the FP value to the integer value and back, checking for equality. | ||||
5578 | // Don't do this for zero, because -0.0 is not fractional. | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5579 | Constant *RHSInt = ConstantExpr::getFPToSI(RHSC, IntTy); |
5580 | if (!RHS.isZero() && | ||||
5581 | ConstantExpr::getSIToFP(RHSInt, RHSC->getType()) != RHSC) { | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5582 | // If we had a comparison against a fractional value, we have to adjust the |
5583 | // compare predicate and sometimes the value. RHSC is rounded towards zero | ||||
5584 | // at this point. | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5585 | switch (Pred) { |
5586 | default: assert(0 && "Unexpected integer comparison!"); | ||||
5587 | case ICmpInst::ICMP_NE: // (float)int != 4.4 --> true | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5588 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5589 | case ICmpInst::ICMP_EQ: // (float)int == 4.4 --> false |
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5590 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5591 | case ICmpInst::ICMP_ULE: |
5592 | // (float)int <= 4.4 --> int <= 4 | ||||
5593 | // (float)int <= -4.4 --> false | ||||
5594 | if (RHS.isNegative()) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5595 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5596 | break; |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5597 | case ICmpInst::ICMP_SLE: |
5598 | // (float)int <= 4.4 --> int <= 4 | ||||
5599 | // (float)int <= -4.4 --> int < -4 | ||||
5600 | if (RHS.isNegative()) | ||||
5601 | Pred = ICmpInst::ICMP_SLT; | ||||
5602 | break; | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5603 | case ICmpInst::ICMP_ULT: |
5604 | // (float)int < -4.4 --> false | ||||
5605 | // (float)int < 4.4 --> int <= 4 | ||||
5606 | if (RHS.isNegative()) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5607 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5608 | Pred = ICmpInst::ICMP_ULE; |
5609 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5610 | case ICmpInst::ICMP_SLT: |
5611 | // (float)int < -4.4 --> int < -4 | ||||
5612 | // (float)int < 4.4 --> int <= 4 | ||||
5613 | if (!RHS.isNegative()) | ||||
5614 | Pred = ICmpInst::ICMP_SLE; | ||||
5615 | break; | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5616 | case ICmpInst::ICMP_UGT: |
5617 | // (float)int > 4.4 --> int > 4 | ||||
5618 | // (float)int > -4.4 --> true | ||||
5619 | if (RHS.isNegative()) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5620 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5621 | break; |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5622 | case ICmpInst::ICMP_SGT: |
5623 | // (float)int > 4.4 --> int > 4 | ||||
5624 | // (float)int > -4.4 --> int >= -4 | ||||
5625 | if (RHS.isNegative()) | ||||
5626 | Pred = ICmpInst::ICMP_SGE; | ||||
5627 | break; | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5628 | case ICmpInst::ICMP_UGE: |
5629 | // (float)int >= -4.4 --> true | ||||
5630 | // (float)int >= 4.4 --> int > 4 | ||||
5631 | if (!RHS.isNegative()) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5632 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5633 | Pred = ICmpInst::ICMP_UGT; |
5634 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5635 | case ICmpInst::ICMP_SGE: |
5636 | // (float)int >= -4.4 --> int >= -4 | ||||
5637 | // (float)int >= 4.4 --> int > 4 | ||||
5638 | if (!RHS.isNegative()) | ||||
5639 | Pred = ICmpInst::ICMP_SGT; | ||||
5640 | break; | ||||
5641 | } | ||||
5642 | } | ||||
5643 | |||||
5644 | // Lower this FP comparison into an appropriate integer version of the | ||||
5645 | // comparison. | ||||
5646 | return new ICmpInst(Pred, LHSI->getOperand(0), RHSInt); | ||||
5647 | } | ||||
5648 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5649 | Instruction *InstCombiner::visitFCmpInst(FCmpInst &I) { |
5650 | bool Changed = SimplifyCompare(I); | ||||
5651 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
5652 | |||||
5653 | // Fold trivial predicates. | ||||
5654 | if (I.getPredicate() == FCmpInst::FCMP_FALSE) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5655 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5656 | if (I.getPredicate() == FCmpInst::FCMP_TRUE) |
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5657 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5658 | |
5659 | // Simplify 'fcmp pred X, X' | ||||
5660 | if (Op0 == Op1) { | ||||
5661 | switch (I.getPredicate()) { | ||||
5662 | default: assert(0 && "Unknown predicate!"); | ||||
5663 | case FCmpInst::FCMP_UEQ: // True if unordered or equal | ||||
5664 | case FCmpInst::FCMP_UGE: // True if unordered, greater than, or equal | ||||
5665 | case FCmpInst::FCMP_ULE: // True if unordered, less than, or equal | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5666 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5667 | case FCmpInst::FCMP_OGT: // True if ordered and greater than |
5668 | case FCmpInst::FCMP_OLT: // True if ordered and less than | ||||
5669 | case FCmpInst::FCMP_ONE: // True if ordered and operands are unequal | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5670 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5671 | |
5672 | case FCmpInst::FCMP_UNO: // True if unordered: isnan(X) | isnan(Y) | ||||
5673 | case FCmpInst::FCMP_ULT: // True if unordered or less than | ||||
5674 | case FCmpInst::FCMP_UGT: // True if unordered or greater than | ||||
5675 | case FCmpInst::FCMP_UNE: // True if unordered or not equal | ||||
5676 | // Canonicalize these to be 'fcmp uno %X, 0.0'. | ||||
5677 | I.setPredicate(FCmpInst::FCMP_UNO); | ||||
5678 | I.setOperand(1, Constant::getNullValue(Op0->getType())); | ||||
5679 | return &I; | ||||
5680 | |||||
5681 | case FCmpInst::FCMP_ORD: // True if ordered (no nans) | ||||
5682 | case FCmpInst::FCMP_OEQ: // True if ordered and equal | ||||
5683 | case FCmpInst::FCMP_OGE: // True if ordered and greater than or equal | ||||
5684 | case FCmpInst::FCMP_OLE: // True if ordered and less than or equal | ||||
5685 | // Canonicalize these to be 'fcmp ord %X, 0.0'. | ||||
5686 | I.setPredicate(FCmpInst::FCMP_ORD); | ||||
5687 | I.setOperand(1, Constant::getNullValue(Op0->getType())); | ||||
5688 | return &I; | ||||
5689 | } | ||||
5690 | } | ||||
5691 | |||||
5692 | if (isa<UndefValue>(Op1)) // fcmp pred X, undef -> undef | ||||
5693 | return ReplaceInstUsesWith(I, UndefValue::get(Type::Int1Ty)); | ||||
5694 | |||||
5695 | // Handle fcmp with constant RHS | ||||
5696 | if (Constant *RHSC = dyn_cast<Constant>(Op1)) { | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5697 | // If the constant is a nan, see if we can fold the comparison based on it. |
5698 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHSC)) { | ||||
5699 | if (CFP->getValueAPF().isNaN()) { | ||||
5700 | if (FCmpInst::isOrdered(I.getPredicate())) // True if ordered and... | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5701 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | f13ff49 | 2008-05-20 03:50:52 +0000 | [diff] [blame] | 5702 | assert(FCmpInst::isUnordered(I.getPredicate()) && |
5703 | "Comparison must be either ordered or unordered!"); | ||||
5704 | // True if unordered. | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5705 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5706 | } |
5707 | } | ||||
5708 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5709 | if (Instruction *LHSI = dyn_cast<Instruction>(Op0)) |
5710 | switch (LHSI->getOpcode()) { | ||||
5711 | case Instruction::PHI: | ||||
Chris Lattner | a2417ba | 2008-06-08 20:52:11 +0000 | [diff] [blame] | 5712 | // Only fold fcmp into the PHI if the phi and fcmp are in the same |
5713 | // block. If in the same block, we're encouraging jump threading. If | ||||
5714 | // not, we are just pessimizing the code by making an i1 phi. | ||||
5715 | if (LHSI->getParent() == I.getParent()) | ||||
5716 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
5717 | return NV; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5718 | break; |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5719 | case Instruction::SIToFP: |
5720 | case Instruction::UIToFP: | ||||
5721 | if (Instruction *NV = FoldFCmp_IntToFP_Cst(I, LHSI, RHSC)) | ||||
5722 | return NV; | ||||
5723 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5724 | case Instruction::Select: |
5725 | // If either operand of the select is a constant, we can fold the | ||||
5726 | // comparison into the select arms, which will cause one to be | ||||
5727 | // constant folded and the select turned into a bitwise or. | ||||
5728 | Value *Op1 = 0, *Op2 = 0; | ||||
5729 | if (LHSI->hasOneUse()) { | ||||
5730 | if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(1))) { | ||||
5731 | // Fold the known value into the constant operand. | ||||
5732 | Op1 = ConstantExpr::getCompare(I.getPredicate(), C, RHSC); | ||||
5733 | // Insert a new FCmp of the other select operand. | ||||
5734 | Op2 = InsertNewInstBefore(new FCmpInst(I.getPredicate(), | ||||
5735 | LHSI->getOperand(2), RHSC, | ||||
5736 | I.getName()), I); | ||||
5737 | } else if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(2))) { | ||||
5738 | // Fold the known value into the constant operand. | ||||
5739 | Op2 = ConstantExpr::getCompare(I.getPredicate(), C, RHSC); | ||||
5740 | // Insert a new FCmp of the other select operand. | ||||
5741 | Op1 = InsertNewInstBefore(new FCmpInst(I.getPredicate(), | ||||
5742 | LHSI->getOperand(1), RHSC, | ||||
5743 | I.getName()), I); | ||||
5744 | } | ||||
5745 | } | ||||
5746 | |||||
5747 | if (Op1) | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 5748 | return SelectInst::Create(LHSI->getOperand(0), Op1, Op2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5749 | break; |
5750 | } | ||||
5751 | } | ||||
5752 | |||||
5753 | return Changed ? &I : 0; | ||||
5754 | } | ||||
5755 | |||||
5756 | Instruction *InstCombiner::visitICmpInst(ICmpInst &I) { | ||||
5757 | bool Changed = SimplifyCompare(I); | ||||
5758 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
5759 | const Type *Ty = Op0->getType(); | ||||
5760 | |||||
5761 | // icmp X, X | ||||
5762 | if (Op0 == Op1) | ||||
5763 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, | ||||
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 5764 | I.isTrueWhenEqual())); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5765 | |
5766 | if (isa<UndefValue>(Op1)) // X icmp undef -> undef | ||||
5767 | return ReplaceInstUsesWith(I, UndefValue::get(Type::Int1Ty)); | ||||
Christopher Lamb | f78cd32 | 2007-12-18 21:32:20 +0000 | [diff] [blame] | 5768 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5769 | // icmp <global/alloca*/null>, <global/alloca*/null> - Global/Stack value |
5770 | // addresses never equal each other! We already know that Op0 != Op1. | ||||
5771 | if ((isa<GlobalValue>(Op0) || isa<AllocaInst>(Op0) || | ||||
5772 | isa<ConstantPointerNull>(Op0)) && | ||||
5773 | (isa<GlobalValue>(Op1) || isa<AllocaInst>(Op1) || | ||||
5774 | isa<ConstantPointerNull>(Op1))) | ||||
5775 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, | ||||
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 5776 | !I.isTrueWhenEqual())); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5777 | |
5778 | // icmp's with boolean values can always be turned into bitwise operations | ||||
5779 | if (Ty == Type::Int1Ty) { | ||||
5780 | switch (I.getPredicate()) { | ||||
5781 | default: assert(0 && "Invalid icmp instruction!"); | ||||
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5782 | case ICmpInst::ICMP_EQ: { // icmp eq i1 A, B -> ~(A^B) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5783 | Instruction *Xor = BinaryOperator::CreateXor(Op0, Op1, I.getName()+"tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5784 | InsertNewInstBefore(Xor, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5785 | return BinaryOperator::CreateNot(Xor); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5786 | } |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5787 | case ICmpInst::ICMP_NE: // icmp eq i1 A, B -> A^B |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5788 | return BinaryOperator::CreateXor(Op0, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5789 | |
5790 | case ICmpInst::ICMP_UGT: | ||||
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5791 | std::swap(Op0, Op1); // Change icmp ugt -> icmp ult |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5792 | // FALL THROUGH |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5793 | case ICmpInst::ICMP_ULT:{ // icmp ult i1 A, B -> ~A & B |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5794 | Instruction *Not = BinaryOperator::CreateNot(Op0, I.getName()+"tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5795 | InsertNewInstBefore(Not, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5796 | return BinaryOperator::CreateAnd(Not, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5797 | } |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5798 | case ICmpInst::ICMP_SGT: |
5799 | std::swap(Op0, Op1); // Change icmp sgt -> icmp slt | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5800 | // FALL THROUGH |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5801 | case ICmpInst::ICMP_SLT: { // icmp slt i1 A, B -> A & ~B |
5802 | Instruction *Not = BinaryOperator::CreateNot(Op1, I.getName()+"tmp"); | ||||
5803 | InsertNewInstBefore(Not, I); | ||||
5804 | return BinaryOperator::CreateAnd(Not, Op0); | ||||
5805 | } | ||||
5806 | case ICmpInst::ICMP_UGE: | ||||
5807 | std::swap(Op0, Op1); // Change icmp uge -> icmp ule | ||||
5808 | // FALL THROUGH | ||||
5809 | case ICmpInst::ICMP_ULE: { // icmp ule i1 A, B -> ~A | B | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5810 | Instruction *Not = BinaryOperator::CreateNot(Op0, I.getName()+"tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5811 | InsertNewInstBefore(Not, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5812 | return BinaryOperator::CreateOr(Not, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5813 | } |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5814 | case ICmpInst::ICMP_SGE: |
5815 | std::swap(Op0, Op1); // Change icmp sge -> icmp sle | ||||
5816 | // FALL THROUGH | ||||
5817 | case ICmpInst::ICMP_SLE: { // icmp sle i1 A, B -> A | ~B | ||||
5818 | Instruction *Not = BinaryOperator::CreateNot(Op1, I.getName()+"tmp"); | ||||
5819 | InsertNewInstBefore(Not, I); | ||||
5820 | return BinaryOperator::CreateOr(Not, Op0); | ||||
5821 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5822 | } |
5823 | } | ||||
5824 | |||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 5825 | // See if we are doing a comparison with a constant. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5826 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) { |
Nick Lewycky | 7c5c237 | 2009-02-27 06:37:39 +0000 | [diff] [blame] | 5827 | Value *A = 0, *B = 0; |
Christopher Lamb | fa6b310 | 2007-12-20 07:21:11 +0000 | [diff] [blame] | 5828 | |
Chris Lattner | be6c54a | 2008-01-05 01:18:20 +0000 | [diff] [blame] | 5829 | // (icmp ne/eq (sub A B) 0) -> (icmp ne/eq A, B) |
5830 | if (I.isEquality() && CI->isNullValue() && | ||||
5831 | match(Op0, m_Sub(m_Value(A), m_Value(B)))) { | ||||
5832 | // (icmp cond A B) if cond is equality | ||||
5833 | return new ICmpInst(I.getPredicate(), A, B); | ||||
Owen Anderson | 42f61ed | 2007-12-28 07:42:12 +0000 | [diff] [blame] | 5834 | } |
Christopher Lamb | fa6b310 | 2007-12-20 07:21:11 +0000 | [diff] [blame] | 5835 | |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 5836 | // If we have an icmp le or icmp ge instruction, turn it into the |
5837 | // appropriate icmp lt or icmp gt instruction. This allows us to rely on | ||||
5838 | // them being folded in the code below. | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5839 | switch (I.getPredicate()) { |
5840 | default: break; | ||||
5841 | case ICmpInst::ICMP_ULE: | ||||
5842 | if (CI->isMaxValue(false)) // A <=u MAX -> TRUE | ||||
5843 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5844 | return new ICmpInst(ICmpInst::ICMP_ULT, Op0, AddOne(CI)); | ||||
5845 | case ICmpInst::ICMP_SLE: | ||||
5846 | if (CI->isMaxValue(true)) // A <=s MAX -> TRUE | ||||
5847 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5848 | return new ICmpInst(ICmpInst::ICMP_SLT, Op0, AddOne(CI)); | ||||
5849 | case ICmpInst::ICMP_UGE: | ||||
5850 | if (CI->isMinValue(false)) // A >=u MIN -> TRUE | ||||
5851 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5852 | return new ICmpInst( ICmpInst::ICMP_UGT, Op0, SubOne(CI)); | ||||
5853 | case ICmpInst::ICMP_SGE: | ||||
5854 | if (CI->isMinValue(true)) // A >=s MIN -> TRUE | ||||
5855 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5856 | return new ICmpInst(ICmpInst::ICMP_SGT, Op0, SubOne(CI)); | ||||
5857 | } | ||||
5858 | |||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5859 | // See if we can fold the comparison based on range information we can get |
5860 | // by checking whether bits are known to be zero or one in the input. | ||||
5861 | uint32_t BitWidth = cast<IntegerType>(Ty)->getBitWidth(); | ||||
5862 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
5863 | |||||
5864 | // If this comparison is a normal comparison, it demands all | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5865 | // bits, if it is a sign bit comparison, it only demands the sign bit. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5866 | bool UnusedBit; |
5867 | bool isSignBit = isSignBitCheck(I.getPredicate(), CI, UnusedBit); | ||||
5868 | |||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 5869 | if (SimplifyDemandedBits(I.getOperandUse(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5870 | isSignBit ? APInt::getSignBit(BitWidth) |
5871 | : APInt::getAllOnesValue(BitWidth), | ||||
5872 | KnownZero, KnownOne, 0)) | ||||
5873 | return &I; | ||||
5874 | |||||
5875 | // Given the known and unknown bits, compute a range that the LHS could be | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5876 | // in. Compute the Min, Max and RHS values based on the known bits. For the |
5877 | // EQ and NE we use unsigned values. | ||||
5878 | APInt Min(BitWidth, 0), Max(BitWidth, 0); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5879 | if (ICmpInst::isSignedPredicate(I.getPredicate())) |
5880 | ComputeSignedMinMaxValuesFromKnownBits(Ty, KnownZero, KnownOne, Min, Max); | ||||
5881 | else | ||||
5882 | ComputeUnsignedMinMaxValuesFromKnownBits(Ty, KnownZero, KnownOne,Min,Max); | ||||
5883 | |||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5884 | // If Min and Max are known to be the same, then SimplifyDemandedBits |
5885 | // figured out that the LHS is a constant. Just constant fold this now so | ||||
5886 | // that code below can assume that Min != Max. | ||||
5887 | if (Min == Max) | ||||
5888 | return ReplaceInstUsesWith(I, ConstantExpr::getICmp(I.getPredicate(), | ||||
5889 | ConstantInt::get(Min), | ||||
5890 | CI)); | ||||
5891 | |||||
5892 | // Based on the range information we know about the LHS, see if we can | ||||
5893 | // simplify this comparison. For example, (x&4) < 8 is always true. | ||||
5894 | const APInt &RHSVal = CI->getValue(); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5895 | switch (I.getPredicate()) { // LE/GE have been folded already. |
5896 | default: assert(0 && "Unknown icmp opcode!"); | ||||
5897 | case ICmpInst::ICMP_EQ: | ||||
5898 | if (Max.ult(RHSVal) || Min.ugt(RHSVal)) | ||||
5899 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
5900 | break; | ||||
5901 | case ICmpInst::ICMP_NE: | ||||
5902 | if (Max.ult(RHSVal) || Min.ugt(RHSVal)) | ||||
5903 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5904 | break; | ||||
5905 | case ICmpInst::ICMP_ULT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5906 | if (Max.ult(RHSVal)) // A <u C -> true iff max(A) < C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5907 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5908 | if (Min.uge(RHSVal)) // A <u C -> false iff min(A) >= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5909 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5910 | if (RHSVal == Max) // A <u MAX -> A != MAX |
5911 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
5912 | if (RHSVal == Min+1) // A <u MIN+1 -> A == MIN | ||||
5913 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, SubOne(CI)); | ||||
5914 | |||||
5915 | // (x <u 2147483648) -> (x >s -1) -> true if sign bit clear | ||||
5916 | if (CI->isMinValue(true)) | ||||
5917 | return new ICmpInst(ICmpInst::ICMP_SGT, Op0, | ||||
5918 | ConstantInt::getAllOnesValue(Op0->getType())); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5919 | break; |
5920 | case ICmpInst::ICMP_UGT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5921 | if (Min.ugt(RHSVal)) // A >u C -> true iff min(A) > C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5922 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5923 | if (Max.ule(RHSVal)) // A >u C -> false iff max(A) <= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5924 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5925 | |
5926 | if (RHSVal == Min) // A >u MIN -> A != MIN | ||||
5927 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
5928 | if (RHSVal == Max-1) // A >u MAX-1 -> A == MAX | ||||
5929 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, AddOne(CI)); | ||||
5930 | |||||
5931 | // (x >u 2147483647) -> (x <s 0) -> true if sign bit set | ||||
5932 | if (CI->isMaxValue(true)) | ||||
5933 | return new ICmpInst(ICmpInst::ICMP_SLT, Op0, | ||||
5934 | ConstantInt::getNullValue(Op0->getType())); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5935 | break; |
5936 | case ICmpInst::ICMP_SLT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5937 | if (Max.slt(RHSVal)) // A <s C -> true iff max(A) < C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5938 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | 611b43e | 2008-07-11 06:40:29 +0000 | [diff] [blame] | 5939 | if (Min.sge(RHSVal)) // A <s C -> false iff min(A) >= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5940 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5941 | if (RHSVal == Max) // A <s MAX -> A != MAX |
5942 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
Chris Lattner | 3496f3e | 2008-07-11 06:36:01 +0000 | [diff] [blame] | 5943 | if (RHSVal == Min+1) // A <s MIN+1 -> A == MIN |
Chris Lattner | 55ab315 | 2008-07-11 06:38:16 +0000 | [diff] [blame] | 5944 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, SubOne(CI)); |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5945 | break; |
5946 | case ICmpInst::ICMP_SGT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5947 | if (Min.sgt(RHSVal)) // A >s C -> true iff min(A) > C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5948 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5949 | if (Max.sle(RHSVal)) // A >s C -> false iff max(A) <= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5950 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5951 | |
5952 | if (RHSVal == Min) // A >s MIN -> A != MIN | ||||
5953 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
5954 | if (RHSVal == Max-1) // A >s MAX-1 -> A == MAX | ||||
5955 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, AddOne(CI)); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5956 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5957 | } |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 5958 | } |
5959 | |||||
5960 | // Test if the ICmpInst instruction is used exclusively by a select as | ||||
5961 | // part of a minimum or maximum operation. If so, refrain from doing | ||||
5962 | // any other folding. This helps out other analyses which understand | ||||
5963 | // non-obfuscated minimum and maximum idioms, such as ScalarEvolution | ||||
5964 | // and CodeGen. And in this case, at least one of the comparison | ||||
5965 | // operands has at least one user besides the compare (the select), | ||||
5966 | // which would often largely negate the benefit of folding anyway. | ||||
5967 | if (I.hasOneUse()) | ||||
5968 | if (SelectInst *SI = dyn_cast<SelectInst>(*I.use_begin())) | ||||
5969 | if ((SI->getOperand(1) == Op0 && SI->getOperand(2) == Op1) || | ||||
5970 | (SI->getOperand(2) == Op0 && SI->getOperand(1) == Op1)) | ||||
5971 | return 0; | ||||
5972 | |||||
5973 | // See if we are doing a comparison between a constant and an instruction that | ||||
5974 | // can be folded into the comparison. | ||||
5975 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5976 | // Since the RHS is a ConstantInt (CI), if the left hand side is an |
5977 | // instruction, see if that instruction also has constants so that the | ||||
5978 | // instruction can be folded into the icmp | ||||
5979 | if (Instruction *LHSI = dyn_cast<Instruction>(Op0)) | ||||
5980 | if (Instruction *Res = visitICmpInstWithInstAndIntCst(I, LHSI, CI)) | ||||
5981 | return Res; | ||||
5982 | } | ||||
5983 | |||||
5984 | // Handle icmp with constant (but not simple integer constant) RHS | ||||
5985 | if (Constant *RHSC = dyn_cast<Constant>(Op1)) { | ||||
5986 | if (Instruction *LHSI = dyn_cast<Instruction>(Op0)) | ||||
5987 | switch (LHSI->getOpcode()) { | ||||
5988 | case Instruction::GetElementPtr: | ||||
5989 | if (RHSC->isNullValue()) { | ||||
5990 | // icmp pred GEP (P, int 0, int 0, int 0), null -> icmp pred P, null | ||||
5991 | bool isAllZeros = true; | ||||
5992 | for (unsigned i = 1, e = LHSI->getNumOperands(); i != e; ++i) | ||||
5993 | if (!isa<Constant>(LHSI->getOperand(i)) || | ||||
5994 | !cast<Constant>(LHSI->getOperand(i))->isNullValue()) { | ||||
5995 | isAllZeros = false; | ||||
5996 | break; | ||||
5997 | } | ||||
5998 | if (isAllZeros) | ||||
5999 | return new ICmpInst(I.getPredicate(), LHSI->getOperand(0), | ||||
6000 | Constant::getNullValue(LHSI->getOperand(0)->getType())); | ||||
6001 | } | ||||
6002 | break; | ||||
6003 | |||||
6004 | case Instruction::PHI: | ||||
Chris Lattner | a2417ba | 2008-06-08 20:52:11 +0000 | [diff] [blame] | 6005 | // Only fold icmp into the PHI if the phi and fcmp are in the same |
6006 | // block. If in the same block, we're encouraging jump threading. If | ||||
6007 | // not, we are just pessimizing the code by making an i1 phi. | ||||
6008 | if (LHSI->getParent() == I.getParent()) | ||||
6009 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
6010 | return NV; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6011 | break; |
6012 | case Instruction::Select: { | ||||
6013 | // If either operand of the select is a constant, we can fold the | ||||
6014 | // comparison into the select arms, which will cause one to be | ||||
6015 | // constant folded and the select turned into a bitwise or. | ||||
6016 | Value *Op1 = 0, *Op2 = 0; | ||||
6017 | if (LHSI->hasOneUse()) { | ||||
6018 | if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(1))) { | ||||
6019 | // Fold the known value into the constant operand. | ||||
6020 | Op1 = ConstantExpr::getICmp(I.getPredicate(), C, RHSC); | ||||
6021 | // Insert a new ICmp of the other select operand. | ||||
6022 | Op2 = InsertNewInstBefore(new ICmpInst(I.getPredicate(), | ||||
6023 | LHSI->getOperand(2), RHSC, | ||||
6024 | I.getName()), I); | ||||
6025 | } else if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(2))) { | ||||
6026 | // Fold the known value into the constant operand. | ||||
6027 | Op2 = ConstantExpr::getICmp(I.getPredicate(), C, RHSC); | ||||
6028 | // Insert a new ICmp of the other select operand. | ||||
6029 | Op1 = InsertNewInstBefore(new ICmpInst(I.getPredicate(), | ||||
6030 | LHSI->getOperand(1), RHSC, | ||||
6031 | I.getName()), I); | ||||
6032 | } | ||||
6033 | } | ||||
6034 | |||||
6035 | if (Op1) | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 6036 | return SelectInst::Create(LHSI->getOperand(0), Op1, Op2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6037 | break; |
6038 | } | ||||
6039 | case Instruction::Malloc: | ||||
6040 | // If we have (malloc != null), and if the malloc has a single use, we | ||||
6041 | // can assume it is successful and remove the malloc. | ||||
6042 | if (LHSI->hasOneUse() && isa<ConstantPointerNull>(RHSC)) { | ||||
6043 | AddToWorkList(LHSI); | ||||
6044 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, | ||||
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 6045 | !I.isTrueWhenEqual())); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6046 | } |
6047 | break; | ||||
6048 | } | ||||
6049 | } | ||||
6050 | |||||
6051 | // If we can optimize a 'icmp GEP, P' or 'icmp P, GEP', do so now. | ||||
6052 | if (User *GEP = dyn_castGetElementPtr(Op0)) | ||||
6053 | if (Instruction *NI = FoldGEPICmp(GEP, Op1, I.getPredicate(), I)) | ||||
6054 | return NI; | ||||
6055 | if (User *GEP = dyn_castGetElementPtr(Op1)) | ||||
6056 | if (Instruction *NI = FoldGEPICmp(GEP, Op0, | ||||
6057 | ICmpInst::getSwappedPredicate(I.getPredicate()), I)) | ||||
6058 | return NI; | ||||
6059 | |||||
6060 | // Test to see if the operands of the icmp are casted versions of other | ||||
6061 | // values. If the ptr->ptr cast can be stripped off both arguments, we do so | ||||
6062 | // now. | ||||
6063 | if (BitCastInst *CI = dyn_cast<BitCastInst>(Op0)) { | ||||
6064 | if (isa<PointerType>(Op0->getType()) && | ||||
6065 | (isa<Constant>(Op1) || isa<BitCastInst>(Op1))) { | ||||
6066 | // We keep moving the cast from the left operand over to the right | ||||
6067 | // operand, where it can often be eliminated completely. | ||||
6068 | Op0 = CI->getOperand(0); | ||||
6069 | |||||
6070 | // If operand #1 is a bitcast instruction, it must also be a ptr->ptr cast | ||||
6071 | // so eliminate it as well. | ||||
6072 | if (BitCastInst *CI2 = dyn_cast<BitCastInst>(Op1)) | ||||
6073 | Op1 = CI2->getOperand(0); | ||||
6074 | |||||
6075 | // If Op1 is a constant, we can fold the cast into the constant. | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 6076 | if (Op0->getType() != Op1->getType()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6077 | if (Constant *Op1C = dyn_cast<Constant>(Op1)) { |
6078 | Op1 = ConstantExpr::getBitCast(Op1C, Op0->getType()); | ||||
6079 | } else { | ||||
6080 | // Otherwise, cast the RHS right before the icmp | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 6081 | Op1 = InsertBitCastBefore(Op1, Op0->getType(), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6082 | } |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 6083 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6084 | return new ICmpInst(I.getPredicate(), Op0, Op1); |
6085 | } | ||||
6086 | } | ||||
6087 | |||||
6088 | if (isa<CastInst>(Op0)) { | ||||
6089 | // Handle the special case of: icmp (cast bool to X), <cst> | ||||
6090 | // This comes up when you have code like | ||||
6091 | // int X = A < B; | ||||
6092 | // if (X) ... | ||||
6093 | // For generality, we handle any zero-extension of any operand comparison | ||||
6094 | // with a constant or another cast from the same type. | ||||
6095 | if (isa<ConstantInt>(Op1) || isa<CastInst>(Op1)) | ||||
6096 | if (Instruction *R = visitICmpInstWithCastAndCast(I)) | ||||
6097 | return R; | ||||
6098 | } | ||||
6099 | |||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 6100 | // See if it's the same type of instruction on the left and right. |
6101 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) { | ||||
6102 | if (BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1)) { | ||||
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 6103 | if (Op0I->getOpcode() == Op1I->getOpcode() && Op0I->hasOneUse() && |
Nick Lewycky | dac8433 | 2009-01-31 21:30:05 +0000 | [diff] [blame] | 6104 | Op1I->hasOneUse() && Op0I->getOperand(1) == Op1I->getOperand(1)) { |
Nick Lewycky | cfadfbd | 2008-09-03 06:24:21 +0000 | [diff] [blame] | 6105 | switch (Op0I->getOpcode()) { |
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 6106 | default: break; |
6107 | case Instruction::Add: | ||||
6108 | case Instruction::Sub: | ||||
6109 | case Instruction::Xor: | ||||
Chris Lattner | f3b445e | 2009-02-02 07:15:30 +0000 | [diff] [blame] | 6110 | if (I.isEquality()) // a+x icmp eq/ne b+x --> a icmp b |
Nick Lewycky | dac8433 | 2009-01-31 21:30:05 +0000 | [diff] [blame] | 6111 | return new ICmpInst(I.getPredicate(), Op0I->getOperand(0), |
6112 | Op1I->getOperand(0)); | ||||
Chris Lattner | f3b445e | 2009-02-02 07:15:30 +0000 | [diff] [blame] | 6113 | // icmp u/s (a ^ signbit), (b ^ signbit) --> icmp s/u a, b |
6114 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) { | ||||
6115 | if (CI->getValue().isSignBit()) { | ||||
6116 | ICmpInst::Predicate Pred = I.isSignedPredicate() | ||||
6117 | ? I.getUnsignedPredicate() | ||||
6118 | : I.getSignedPredicate(); | ||||
6119 | return new ICmpInst(Pred, Op0I->getOperand(0), | ||||
6120 | Op1I->getOperand(0)); | ||||
6121 | } | ||||
6122 | |||||
6123 | if (CI->getValue().isMaxSignedValue()) { | ||||
6124 | ICmpInst::Predicate Pred = I.isSignedPredicate() | ||||
6125 | ? I.getUnsignedPredicate() | ||||
6126 | : I.getSignedPredicate(); | ||||
6127 | Pred = I.getSwappedPredicate(Pred); | ||||
6128 | return new ICmpInst(Pred, Op0I->getOperand(0), | ||||
6129 | Op1I->getOperand(0)); | ||||
Nick Lewycky | dac8433 | 2009-01-31 21:30:05 +0000 | [diff] [blame] | 6130 | } |
6131 | } | ||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 6132 | break; |
6133 | case Instruction::Mul: | ||||
Nick Lewycky | dac8433 | 2009-01-31 21:30:05 +0000 | [diff] [blame] | 6134 | if (!I.isEquality()) |
6135 | break; | ||||
6136 | |||||
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 6137 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) { |
6138 | // a * Cst icmp eq/ne b * Cst --> a & Mask icmp b & Mask | ||||
6139 | // Mask = -1 >> count-trailing-zeros(Cst). | ||||
6140 | if (!CI->isZero() && !CI->isOne()) { | ||||
6141 | const APInt &AP = CI->getValue(); | ||||
6142 | ConstantInt *Mask = ConstantInt::get( | ||||
6143 | APInt::getLowBitsSet(AP.getBitWidth(), | ||||
6144 | AP.getBitWidth() - | ||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 6145 | AP.countTrailingZeros())); |
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 6146 | Instruction *And1 = BinaryOperator::CreateAnd(Op0I->getOperand(0), |
6147 | Mask); | ||||
6148 | Instruction *And2 = BinaryOperator::CreateAnd(Op1I->getOperand(0), | ||||
6149 | Mask); | ||||
6150 | InsertNewInstBefore(And1, I); | ||||
6151 | InsertNewInstBefore(And2, I); | ||||
6152 | return new ICmpInst(I.getPredicate(), And1, And2); | ||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 6153 | } |
6154 | } | ||||
6155 | break; | ||||
6156 | } | ||||
6157 | } | ||||
6158 | } | ||||
6159 | } | ||||
6160 | |||||
Chris Lattner | a4e1eef | 2008-05-09 05:19:28 +0000 | [diff] [blame] | 6161 | // ~x < ~y --> y < x |
6162 | { Value *A, *B; | ||||
6163 | if (match(Op0, m_Not(m_Value(A))) && | ||||
6164 | match(Op1, m_Not(m_Value(B)))) | ||||
6165 | return new ICmpInst(I.getPredicate(), B, A); | ||||
6166 | } | ||||
6167 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6168 | if (I.isEquality()) { |
6169 | Value *A, *B, *C, *D; | ||||
Chris Lattner | a4e1eef | 2008-05-09 05:19:28 +0000 | [diff] [blame] | 6170 | |
6171 | // -x == -y --> x == y | ||||
6172 | if (match(Op0, m_Neg(m_Value(A))) && | ||||
6173 | match(Op1, m_Neg(m_Value(B)))) | ||||
6174 | return new ICmpInst(I.getPredicate(), A, B); | ||||
6175 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6176 | if (match(Op0, m_Xor(m_Value(A), m_Value(B)))) { |
6177 | if (A == Op1 || B == Op1) { // (A^B) == A -> B == 0 | ||||
6178 | Value *OtherVal = A == Op1 ? B : A; | ||||
6179 | return new ICmpInst(I.getPredicate(), OtherVal, | ||||
6180 | Constant::getNullValue(A->getType())); | ||||
6181 | } | ||||
6182 | |||||
6183 | if (match(Op1, m_Xor(m_Value(C), m_Value(D)))) { | ||||
6184 | // A^c1 == C^c2 --> A == C^(c1^c2) | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 6185 | ConstantInt *C1, *C2; |
6186 | if (match(B, m_ConstantInt(C1)) && | ||||
6187 | match(D, m_ConstantInt(C2)) && Op1->hasOneUse()) { | ||||
6188 | Constant *NC = ConstantInt::get(C1->getValue() ^ C2->getValue()); | ||||
6189 | Instruction *Xor = BinaryOperator::CreateXor(C, NC, "tmp"); | ||||
6190 | return new ICmpInst(I.getPredicate(), A, | ||||
6191 | InsertNewInstBefore(Xor, I)); | ||||
6192 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6193 | |
6194 | // A^B == A^D -> B == D | ||||
6195 | if (A == C) return new ICmpInst(I.getPredicate(), B, D); | ||||
6196 | if (A == D) return new ICmpInst(I.getPredicate(), B, C); | ||||
6197 | if (B == C) return new ICmpInst(I.getPredicate(), A, D); | ||||
6198 | if (B == D) return new ICmpInst(I.getPredicate(), A, C); | ||||
6199 | } | ||||
6200 | } | ||||
6201 | |||||
6202 | if (match(Op1, m_Xor(m_Value(A), m_Value(B))) && | ||||
6203 | (A == Op0 || B == Op0)) { | ||||
6204 | // A == (A^B) -> B == 0 | ||||
6205 | Value *OtherVal = A == Op0 ? B : A; | ||||
6206 | return new ICmpInst(I.getPredicate(), OtherVal, | ||||
6207 | Constant::getNullValue(A->getType())); | ||||
6208 | } | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 6209 | |
6210 | // (A-B) == A -> B == 0 | ||||
6211 | if (match(Op0, m_Sub(m_Specific(Op1), m_Value(B)))) | ||||
6212 | return new ICmpInst(I.getPredicate(), B, | ||||
6213 | Constant::getNullValue(B->getType())); | ||||
6214 | |||||
6215 | // A == (A-B) -> B == 0 | ||||
6216 | if (match(Op1, m_Sub(m_Specific(Op0), m_Value(B)))) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6217 | return new ICmpInst(I.getPredicate(), B, |
6218 | Constant::getNullValue(B->getType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6219 | |
6220 | // (X&Z) == (Y&Z) -> (X^Y) & Z == 0 | ||||
6221 | if (Op0->hasOneUse() && Op1->hasOneUse() && | ||||
6222 | match(Op0, m_And(m_Value(A), m_Value(B))) && | ||||
6223 | match(Op1, m_And(m_Value(C), m_Value(D)))) { | ||||
6224 | Value *X = 0, *Y = 0, *Z = 0; | ||||
6225 | |||||
6226 | if (A == C) { | ||||
6227 | X = B; Y = D; Z = A; | ||||
6228 | } else if (A == D) { | ||||
6229 | X = B; Y = C; Z = A; | ||||
6230 | } else if (B == C) { | ||||
6231 | X = A; Y = D; Z = B; | ||||
6232 | } else if (B == D) { | ||||
6233 | X = A; Y = C; Z = B; | ||||
6234 | } | ||||
6235 | |||||
6236 | if (X) { // Build (X^Y) & Z | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6237 | Op1 = InsertNewInstBefore(BinaryOperator::CreateXor(X, Y, "tmp"), I); |
6238 | Op1 = InsertNewInstBefore(BinaryOperator::CreateAnd(Op1, Z, "tmp"), I); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6239 | I.setOperand(0, Op1); |
6240 | I.setOperand(1, Constant::getNullValue(Op1->getType())); | ||||
6241 | return &I; | ||||
6242 | } | ||||
6243 | } | ||||
6244 | } | ||||
6245 | return Changed ? &I : 0; | ||||
6246 | } | ||||
6247 | |||||
6248 | |||||
6249 | /// FoldICmpDivCst - Fold "icmp pred, ([su]div X, DivRHS), CmpRHS" where DivRHS | ||||
6250 | /// and CmpRHS are both known to be integer constants. | ||||
6251 | Instruction *InstCombiner::FoldICmpDivCst(ICmpInst &ICI, BinaryOperator *DivI, | ||||
6252 | ConstantInt *DivRHS) { | ||||
6253 | ConstantInt *CmpRHS = cast<ConstantInt>(ICI.getOperand(1)); | ||||
6254 | const APInt &CmpRHSV = CmpRHS->getValue(); | ||||
6255 | |||||
6256 | // FIXME: If the operand types don't match the type of the divide | ||||
6257 | // then don't attempt this transform. The code below doesn't have the | ||||
6258 | // logic to deal with a signed divide and an unsigned compare (and | ||||
6259 | // vice versa). This is because (x /s C1) <s C2 produces different | ||||
6260 | // results than (x /s C1) <u C2 or (x /u C1) <s C2 or even | ||||
6261 | // (x /u C1) <u C2. Simply casting the operands and result won't | ||||
6262 | // work. :( The if statement below tests that condition and bails | ||||
6263 | // if it finds it. | ||||
6264 | bool DivIsSigned = DivI->getOpcode() == Instruction::SDiv; | ||||
6265 | if (!ICI.isEquality() && DivIsSigned != ICI.isSignedPredicate()) | ||||
6266 | return 0; | ||||
6267 | if (DivRHS->isZero()) | ||||
6268 | return 0; // The ProdOV computation fails on divide by zero. | ||||
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 6269 | if (DivIsSigned && DivRHS->isAllOnesValue()) |
6270 | return 0; // The overflow computation also screws up here | ||||
6271 | if (DivRHS->isOne()) | ||||
6272 | return 0; // Not worth bothering, and eliminates some funny cases | ||||
6273 | // with INT_MIN. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6274 | |
6275 | // Compute Prod = CI * DivRHS. We are essentially solving an equation | ||||
6276 | // of form X/C1=C2. We solve for X by multiplying C1 (DivRHS) and | ||||
6277 | // C2 (CI). By solving for X we can turn this into a range check | ||||
6278 | // instead of computing a divide. | ||||
6279 | ConstantInt *Prod = Multiply(CmpRHS, DivRHS); | ||||
6280 | |||||
6281 | // Determine if the product overflows by seeing if the product is | ||||
6282 | // not equal to the divide. Make sure we do the same kind of divide | ||||
6283 | // as in the LHS instruction that we're folding. | ||||
6284 | bool ProdOV = (DivIsSigned ? ConstantExpr::getSDiv(Prod, DivRHS) : | ||||
6285 | ConstantExpr::getUDiv(Prod, DivRHS)) != CmpRHS; | ||||
6286 | |||||
6287 | // Get the ICmp opcode | ||||
6288 | ICmpInst::Predicate Pred = ICI.getPredicate(); | ||||
6289 | |||||
6290 | // Figure out the interval that is being checked. For example, a comparison | ||||
6291 | // like "X /u 5 == 0" is really checking that X is in the interval [0, 5). | ||||
6292 | // Compute this interval based on the constants involved and the signedness of | ||||
6293 | // the compare/divide. This computes a half-open interval, keeping track of | ||||
6294 | // whether either value in the interval overflows. After analysis each | ||||
6295 | // overflow variable is set to 0 if it's corresponding bound variable is valid | ||||
6296 | // -1 if overflowed off the bottom end, or +1 if overflowed off the top end. | ||||
6297 | int LoOverflow = 0, HiOverflow = 0; | ||||
6298 | ConstantInt *LoBound = 0, *HiBound = 0; | ||||
6299 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6300 | if (!DivIsSigned) { // udiv |
6301 | // e.g. X/5 op 3 --> [15, 20) | ||||
6302 | LoBound = Prod; | ||||
6303 | HiOverflow = LoOverflow = ProdOV; | ||||
6304 | if (!HiOverflow) | ||||
6305 | HiOverflow = AddWithOverflow(HiBound, LoBound, DivRHS, false); | ||||
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 6306 | } else if (DivRHS->getValue().isStrictlyPositive()) { // Divisor is > 0. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6307 | if (CmpRHSV == 0) { // (X / pos) op 0 |
6308 | // Can't overflow. e.g. X/2 op 0 --> [-1, 2) | ||||
6309 | LoBound = cast<ConstantInt>(ConstantExpr::getNeg(SubOne(DivRHS))); | ||||
6310 | HiBound = DivRHS; | ||||
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 6311 | } else if (CmpRHSV.isStrictlyPositive()) { // (X / pos) op pos |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6312 | LoBound = Prod; // e.g. X/5 op 3 --> [15, 20) |
6313 | HiOverflow = LoOverflow = ProdOV; | ||||
6314 | if (!HiOverflow) | ||||
6315 | HiOverflow = AddWithOverflow(HiBound, Prod, DivRHS, true); | ||||
6316 | } else { // (X / pos) op neg | ||||
6317 | // e.g. X/5 op -3 --> [-15-4, -15+1) --> [-19, -14) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6318 | HiBound = AddOne(Prod); |
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 6319 | LoOverflow = HiOverflow = ProdOV ? -1 : 0; |
6320 | if (!LoOverflow) { | ||||
6321 | ConstantInt* DivNeg = cast<ConstantInt>(ConstantExpr::getNeg(DivRHS)); | ||||
6322 | LoOverflow = AddWithOverflow(LoBound, HiBound, DivNeg, | ||||
6323 | true) ? -1 : 0; | ||||
6324 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6325 | } |
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 6326 | } else if (DivRHS->getValue().isNegative()) { // Divisor is < 0. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6327 | if (CmpRHSV == 0) { // (X / neg) op 0 |
6328 | // e.g. X/-5 op 0 --> [-4, 5) | ||||
6329 | LoBound = AddOne(DivRHS); | ||||
6330 | HiBound = cast<ConstantInt>(ConstantExpr::getNeg(DivRHS)); | ||||
6331 | if (HiBound == DivRHS) { // -INTMIN = INTMIN | ||||
6332 | HiOverflow = 1; // [INTMIN+1, overflow) | ||||
6333 | HiBound = 0; // e.g. X/INTMIN = 0 --> X > INTMIN | ||||
6334 | } | ||||
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 6335 | } else if (CmpRHSV.isStrictlyPositive()) { // (X / neg) op pos |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6336 | // e.g. X/-5 op 3 --> [-19, -14) |
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 6337 | HiBound = AddOne(Prod); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6338 | HiOverflow = LoOverflow = ProdOV ? -1 : 0; |
6339 | if (!LoOverflow) | ||||
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 6340 | LoOverflow = AddWithOverflow(LoBound, HiBound, DivRHS, true) ? -1 : 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6341 | } else { // (X / neg) op neg |
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 6342 | LoBound = Prod; // e.g. X/-5 op -3 --> [15, 20) |
6343 | LoOverflow = HiOverflow = ProdOV; | ||||
Dan Gohman | 45408ea | 2008-09-11 00:25:00 +0000 | [diff] [blame] | 6344 | if (!HiOverflow) |
6345 | HiOverflow = SubWithOverflow(HiBound, Prod, DivRHS, true); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6346 | } |
6347 | |||||
6348 | // Dividing by a negative swaps the condition. LT <-> GT | ||||
6349 | Pred = ICmpInst::getSwappedPredicate(Pred); | ||||
6350 | } | ||||
6351 | |||||
6352 | Value *X = DivI->getOperand(0); | ||||
6353 | switch (Pred) { | ||||
6354 | default: assert(0 && "Unhandled icmp opcode!"); | ||||
6355 | case ICmpInst::ICMP_EQ: | ||||
6356 | if (LoOverflow && HiOverflow) | ||||
6357 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6358 | else if (HiOverflow) | ||||
6359 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SGE : | ||||
6360 | ICmpInst::ICMP_UGE, X, LoBound); | ||||
6361 | else if (LoOverflow) | ||||
6362 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SLT : | ||||
6363 | ICmpInst::ICMP_ULT, X, HiBound); | ||||
6364 | else | ||||
6365 | return InsertRangeTest(X, LoBound, HiBound, DivIsSigned, true, ICI); | ||||
6366 | case ICmpInst::ICMP_NE: | ||||
6367 | if (LoOverflow && HiOverflow) | ||||
6368 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6369 | else if (HiOverflow) | ||||
6370 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SLT : | ||||
6371 | ICmpInst::ICMP_ULT, X, LoBound); | ||||
6372 | else if (LoOverflow) | ||||
6373 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SGE : | ||||
6374 | ICmpInst::ICMP_UGE, X, HiBound); | ||||
6375 | else | ||||
6376 | return InsertRangeTest(X, LoBound, HiBound, DivIsSigned, false, ICI); | ||||
6377 | case ICmpInst::ICMP_ULT: | ||||
6378 | case ICmpInst::ICMP_SLT: | ||||
6379 | if (LoOverflow == +1) // Low bound is greater than input range. | ||||
6380 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6381 | if (LoOverflow == -1) // Low bound is less than input range. | ||||
6382 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6383 | return new ICmpInst(Pred, X, LoBound); | ||||
6384 | case ICmpInst::ICMP_UGT: | ||||
6385 | case ICmpInst::ICMP_SGT: | ||||
6386 | if (HiOverflow == +1) // High bound greater than input range. | ||||
6387 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6388 | else if (HiOverflow == -1) // High bound less than input range. | ||||
6389 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6390 | if (Pred == ICmpInst::ICMP_UGT) | ||||
6391 | return new ICmpInst(ICmpInst::ICMP_UGE, X, HiBound); | ||||
6392 | else | ||||
6393 | return new ICmpInst(ICmpInst::ICMP_SGE, X, HiBound); | ||||
6394 | } | ||||
6395 | } | ||||
6396 | |||||
6397 | |||||
6398 | /// visitICmpInstWithInstAndIntCst - Handle "icmp (instr, intcst)". | ||||
6399 | /// | ||||
6400 | Instruction *InstCombiner::visitICmpInstWithInstAndIntCst(ICmpInst &ICI, | ||||
6401 | Instruction *LHSI, | ||||
6402 | ConstantInt *RHS) { | ||||
6403 | const APInt &RHSV = RHS->getValue(); | ||||
6404 | |||||
6405 | switch (LHSI->getOpcode()) { | ||||
Chris Lattner | 56be123 | 2009-01-09 07:47:06 +0000 | [diff] [blame] | 6406 | case Instruction::Trunc: |
6407 | if (ICI.isEquality() && LHSI->hasOneUse()) { | ||||
6408 | // Simplify icmp eq (trunc x to i8), 42 -> icmp eq x, 42|highbits if all | ||||
6409 | // of the high bits truncated out of x are known. | ||||
6410 | unsigned DstBits = LHSI->getType()->getPrimitiveSizeInBits(), | ||||
6411 | SrcBits = LHSI->getOperand(0)->getType()->getPrimitiveSizeInBits(); | ||||
6412 | APInt Mask(APInt::getHighBitsSet(SrcBits, SrcBits-DstBits)); | ||||
6413 | APInt KnownZero(SrcBits, 0), KnownOne(SrcBits, 0); | ||||
6414 | ComputeMaskedBits(LHSI->getOperand(0), Mask, KnownZero, KnownOne); | ||||
6415 | |||||
6416 | // If all the high bits are known, we can do this xform. | ||||
6417 | if ((KnownZero|KnownOne).countLeadingOnes() >= SrcBits-DstBits) { | ||||
6418 | // Pull in the high bits from known-ones set. | ||||
6419 | APInt NewRHS(RHS->getValue()); | ||||
6420 | NewRHS.zext(SrcBits); | ||||
6421 | NewRHS |= KnownOne; | ||||
6422 | return new ICmpInst(ICI.getPredicate(), LHSI->getOperand(0), | ||||
6423 | ConstantInt::get(NewRHS)); | ||||
6424 | } | ||||
6425 | } | ||||
6426 | break; | ||||
6427 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6428 | case Instruction::Xor: // (icmp pred (xor X, XorCST), CI) |
6429 | if (ConstantInt *XorCST = dyn_cast<ConstantInt>(LHSI->getOperand(1))) { | ||||
6430 | // If this is a comparison that tests the signbit (X < 0) or (x > -1), | ||||
6431 | // fold the xor. | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 6432 | if ((ICI.getPredicate() == ICmpInst::ICMP_SLT && RHSV == 0) || |
6433 | (ICI.getPredicate() == ICmpInst::ICMP_SGT && RHSV.isAllOnesValue())) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6434 | Value *CompareVal = LHSI->getOperand(0); |
6435 | |||||
6436 | // If the sign bit of the XorCST is not set, there is no change to | ||||
6437 | // the operation, just stop using the Xor. | ||||
6438 | if (!XorCST->getValue().isNegative()) { | ||||
6439 | ICI.setOperand(0, CompareVal); | ||||
6440 | AddToWorkList(LHSI); | ||||
6441 | return &ICI; | ||||
6442 | } | ||||
6443 | |||||
6444 | // Was the old condition true if the operand is positive? | ||||
6445 | bool isTrueIfPositive = ICI.getPredicate() == ICmpInst::ICMP_SGT; | ||||
6446 | |||||
6447 | // If so, the new one isn't. | ||||
6448 | isTrueIfPositive ^= true; | ||||
6449 | |||||
6450 | if (isTrueIfPositive) | ||||
6451 | return new ICmpInst(ICmpInst::ICMP_SGT, CompareVal, SubOne(RHS)); | ||||
6452 | else | ||||
6453 | return new ICmpInst(ICmpInst::ICMP_SLT, CompareVal, AddOne(RHS)); | ||||
6454 | } | ||||
Nick Lewycky | dac8433 | 2009-01-31 21:30:05 +0000 | [diff] [blame] | 6455 | |
6456 | if (LHSI->hasOneUse()) { | ||||
6457 | // (icmp u/s (xor A SignBit), C) -> (icmp s/u A, (xor C SignBit)) | ||||
6458 | if (!ICI.isEquality() && XorCST->getValue().isSignBit()) { | ||||
6459 | const APInt &SignBit = XorCST->getValue(); | ||||
6460 | ICmpInst::Predicate Pred = ICI.isSignedPredicate() | ||||
6461 | ? ICI.getUnsignedPredicate() | ||||
6462 | : ICI.getSignedPredicate(); | ||||
6463 | return new ICmpInst(Pred, LHSI->getOperand(0), | ||||
6464 | ConstantInt::get(RHSV ^ SignBit)); | ||||
6465 | } | ||||
6466 | |||||
6467 | // (icmp u/s (xor A ~SignBit), C) -> (icmp s/u (xor C ~SignBit), A) | ||||
Chris Lattner | f3b445e | 2009-02-02 07:15:30 +0000 | [diff] [blame] | 6468 | if (!ICI.isEquality() && XorCST->getValue().isMaxSignedValue()) { |
Nick Lewycky | dac8433 | 2009-01-31 21:30:05 +0000 | [diff] [blame] | 6469 | const APInt &NotSignBit = XorCST->getValue(); |
6470 | ICmpInst::Predicate Pred = ICI.isSignedPredicate() | ||||
6471 | ? ICI.getUnsignedPredicate() | ||||
6472 | : ICI.getSignedPredicate(); | ||||
6473 | Pred = ICI.getSwappedPredicate(Pred); | ||||
6474 | return new ICmpInst(Pred, LHSI->getOperand(0), | ||||
6475 | ConstantInt::get(RHSV ^ NotSignBit)); | ||||
6476 | } | ||||
6477 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6478 | } |
6479 | break; | ||||
6480 | case Instruction::And: // (icmp pred (and X, AndCST), RHS) | ||||
6481 | if (LHSI->hasOneUse() && isa<ConstantInt>(LHSI->getOperand(1)) && | ||||
6482 | LHSI->getOperand(0)->hasOneUse()) { | ||||
6483 | ConstantInt *AndCST = cast<ConstantInt>(LHSI->getOperand(1)); | ||||
6484 | |||||
6485 | // If the LHS is an AND of a truncating cast, we can widen the | ||||
6486 | // and/compare to be the input width without changing the value | ||||
6487 | // produced, eliminating a cast. | ||||
6488 | if (TruncInst *Cast = dyn_cast<TruncInst>(LHSI->getOperand(0))) { | ||||
6489 | // We can do this transformation if either the AND constant does not | ||||
6490 | // have its sign bit set or if it is an equality comparison. | ||||
6491 | // Extending a relational comparison when we're checking the sign | ||||
6492 | // bit would not work. | ||||
6493 | if (Cast->hasOneUse() && | ||||
Anton Korobeynikov | 6a4a933 | 2008-02-20 12:07:57 +0000 | [diff] [blame] | 6494 | (ICI.isEquality() || |
6495 | (AndCST->getValue().isNonNegative() && RHSV.isNonNegative()))) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6496 | uint32_t BitWidth = |
6497 | cast<IntegerType>(Cast->getOperand(0)->getType())->getBitWidth(); | ||||
6498 | APInt NewCST = AndCST->getValue(); | ||||
6499 | NewCST.zext(BitWidth); | ||||
6500 | APInt NewCI = RHSV; | ||||
6501 | NewCI.zext(BitWidth); | ||||
6502 | Instruction *NewAnd = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6503 | BinaryOperator::CreateAnd(Cast->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6504 | ConstantInt::get(NewCST),LHSI->getName()); |
6505 | InsertNewInstBefore(NewAnd, ICI); | ||||
6506 | return new ICmpInst(ICI.getPredicate(), NewAnd, | ||||
6507 | ConstantInt::get(NewCI)); | ||||
6508 | } | ||||
6509 | } | ||||
6510 | |||||
6511 | // If this is: (X >> C1) & C2 != C3 (where any shift and any compare | ||||
6512 | // could exist), turn it into (X & (C2 << C1)) != (C3 << C1). This | ||||
6513 | // happens a LOT in code produced by the C front-end, for bitfield | ||||
6514 | // access. | ||||
6515 | BinaryOperator *Shift = dyn_cast<BinaryOperator>(LHSI->getOperand(0)); | ||||
6516 | if (Shift && !Shift->isShift()) | ||||
6517 | Shift = 0; | ||||
6518 | |||||
6519 | ConstantInt *ShAmt; | ||||
6520 | ShAmt = Shift ? dyn_cast<ConstantInt>(Shift->getOperand(1)) : 0; | ||||
6521 | const Type *Ty = Shift ? Shift->getType() : 0; // Type of the shift. | ||||
6522 | const Type *AndTy = AndCST->getType(); // Type of the and. | ||||
6523 | |||||
6524 | // We can fold this as long as we can't shift unknown bits | ||||
6525 | // into the mask. This can only happen with signed shift | ||||
6526 | // rights, as they sign-extend. | ||||
6527 | if (ShAmt) { | ||||
6528 | bool CanFold = Shift->isLogicalShift(); | ||||
6529 | if (!CanFold) { | ||||
6530 | // To test for the bad case of the signed shr, see if any | ||||
6531 | // of the bits shifted in could be tested after the mask. | ||||
6532 | uint32_t TyBits = Ty->getPrimitiveSizeInBits(); | ||||
6533 | int ShAmtVal = TyBits - ShAmt->getLimitedValue(TyBits); | ||||
6534 | |||||
6535 | uint32_t BitWidth = AndTy->getPrimitiveSizeInBits(); | ||||
6536 | if ((APInt::getHighBitsSet(BitWidth, BitWidth-ShAmtVal) & | ||||
6537 | AndCST->getValue()) == 0) | ||||
6538 | CanFold = true; | ||||
6539 | } | ||||
6540 | |||||
6541 | if (CanFold) { | ||||
6542 | Constant *NewCst; | ||||
6543 | if (Shift->getOpcode() == Instruction::Shl) | ||||
6544 | NewCst = ConstantExpr::getLShr(RHS, ShAmt); | ||||
6545 | else | ||||
6546 | NewCst = ConstantExpr::getShl(RHS, ShAmt); | ||||
6547 | |||||
6548 | // Check to see if we are shifting out any of the bits being | ||||
6549 | // compared. | ||||
6550 | if (ConstantExpr::get(Shift->getOpcode(), NewCst, ShAmt) != RHS) { | ||||
6551 | // If we shifted bits out, the fold is not going to work out. | ||||
6552 | // As a special case, check to see if this means that the | ||||
6553 | // result is always true or false now. | ||||
6554 | if (ICI.getPredicate() == ICmpInst::ICMP_EQ) | ||||
6555 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6556 | if (ICI.getPredicate() == ICmpInst::ICMP_NE) | ||||
6557 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6558 | } else { | ||||
6559 | ICI.setOperand(1, NewCst); | ||||
6560 | Constant *NewAndCST; | ||||
6561 | if (Shift->getOpcode() == Instruction::Shl) | ||||
6562 | NewAndCST = ConstantExpr::getLShr(AndCST, ShAmt); | ||||
6563 | else | ||||
6564 | NewAndCST = ConstantExpr::getShl(AndCST, ShAmt); | ||||
6565 | LHSI->setOperand(1, NewAndCST); | ||||
6566 | LHSI->setOperand(0, Shift->getOperand(0)); | ||||
6567 | AddToWorkList(Shift); // Shift is dead. | ||||
6568 | AddUsesToWorkList(ICI); | ||||
6569 | return &ICI; | ||||
6570 | } | ||||
6571 | } | ||||
6572 | } | ||||
6573 | |||||
6574 | // Turn ((X >> Y) & C) == 0 into (X & (C << Y)) == 0. The later is | ||||
6575 | // preferable because it allows the C<<Y expression to be hoisted out | ||||
6576 | // of a loop if Y is invariant and X is not. | ||||
6577 | if (Shift && Shift->hasOneUse() && RHSV == 0 && | ||||
6578 | ICI.isEquality() && !Shift->isArithmeticShift() && | ||||
6579 | isa<Instruction>(Shift->getOperand(0))) { | ||||
6580 | // Compute C << Y. | ||||
6581 | Value *NS; | ||||
6582 | if (Shift->getOpcode() == Instruction::LShr) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6583 | NS = BinaryOperator::CreateShl(AndCST, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6584 | Shift->getOperand(1), "tmp"); |
6585 | } else { | ||||
6586 | // Insert a logical shift. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6587 | NS = BinaryOperator::CreateLShr(AndCST, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6588 | Shift->getOperand(1), "tmp"); |
6589 | } | ||||
6590 | InsertNewInstBefore(cast<Instruction>(NS), ICI); | ||||
6591 | |||||
6592 | // Compute X & (C << Y). | ||||
6593 | Instruction *NewAnd = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6594 | BinaryOperator::CreateAnd(Shift->getOperand(0), NS, LHSI->getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6595 | InsertNewInstBefore(NewAnd, ICI); |
6596 | |||||
6597 | ICI.setOperand(0, NewAnd); | ||||
6598 | return &ICI; | ||||
6599 | } | ||||
6600 | } | ||||
6601 | break; | ||||
6602 | |||||
6603 | case Instruction::Shl: { // (icmp pred (shl X, ShAmt), CI) | ||||
6604 | ConstantInt *ShAmt = dyn_cast<ConstantInt>(LHSI->getOperand(1)); | ||||
6605 | if (!ShAmt) break; | ||||
6606 | |||||
6607 | uint32_t TypeBits = RHSV.getBitWidth(); | ||||
6608 | |||||
6609 | // Check that the shift amount is in range. If not, don't perform | ||||
6610 | // undefined shifts. When the shift is visited it will be | ||||
6611 | // simplified. | ||||
6612 | if (ShAmt->uge(TypeBits)) | ||||
6613 | break; | ||||
6614 | |||||
6615 | if (ICI.isEquality()) { | ||||
6616 | // If we are comparing against bits always shifted out, the | ||||
6617 | // comparison cannot succeed. | ||||
6618 | Constant *Comp = | ||||
6619 | ConstantExpr::getShl(ConstantExpr::getLShr(RHS, ShAmt), ShAmt); | ||||
6620 | if (Comp != RHS) {// Comparing against a bit that we know is zero. | ||||
6621 | bool IsICMP_NE = ICI.getPredicate() == ICmpInst::ICMP_NE; | ||||
6622 | Constant *Cst = ConstantInt::get(Type::Int1Ty, IsICMP_NE); | ||||
6623 | return ReplaceInstUsesWith(ICI, Cst); | ||||
6624 | } | ||||
6625 | |||||
6626 | if (LHSI->hasOneUse()) { | ||||
6627 | // Otherwise strength reduce the shift into an and. | ||||
6628 | uint32_t ShAmtVal = (uint32_t)ShAmt->getLimitedValue(TypeBits); | ||||
6629 | Constant *Mask = | ||||
6630 | ConstantInt::get(APInt::getLowBitsSet(TypeBits, TypeBits-ShAmtVal)); | ||||
6631 | |||||
6632 | Instruction *AndI = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6633 | BinaryOperator::CreateAnd(LHSI->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6634 | Mask, LHSI->getName()+".mask"); |
6635 | Value *And = InsertNewInstBefore(AndI, ICI); | ||||
6636 | return new ICmpInst(ICI.getPredicate(), And, | ||||
6637 | ConstantInt::get(RHSV.lshr(ShAmtVal))); | ||||
6638 | } | ||||
6639 | } | ||||
6640 | |||||
6641 | // Otherwise, if this is a comparison of the sign bit, simplify to and/test. | ||||
6642 | bool TrueIfSigned = false; | ||||
6643 | if (LHSI->hasOneUse() && | ||||
6644 | isSignBitCheck(ICI.getPredicate(), RHS, TrueIfSigned)) { | ||||
6645 | // (X << 31) <s 0 --> (X&1) != 0 | ||||
6646 | Constant *Mask = ConstantInt::get(APInt(TypeBits, 1) << | ||||
6647 | (TypeBits-ShAmt->getZExtValue()-1)); | ||||
6648 | Instruction *AndI = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6649 | BinaryOperator::CreateAnd(LHSI->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6650 | Mask, LHSI->getName()+".mask"); |
6651 | Value *And = InsertNewInstBefore(AndI, ICI); | ||||
6652 | |||||
6653 | return new ICmpInst(TrueIfSigned ? ICmpInst::ICMP_NE : ICmpInst::ICMP_EQ, | ||||
6654 | And, Constant::getNullValue(And->getType())); | ||||
6655 | } | ||||
6656 | break; | ||||
6657 | } | ||||
6658 | |||||
6659 | case Instruction::LShr: // (icmp pred (shr X, ShAmt), CI) | ||||
6660 | case Instruction::AShr: { | ||||
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6661 | // Only handle equality comparisons of shift-by-constant. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6662 | ConstantInt *ShAmt = dyn_cast<ConstantInt>(LHSI->getOperand(1)); |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6663 | if (!ShAmt || !ICI.isEquality()) break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6664 | |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6665 | // Check that the shift amount is in range. If not, don't perform |
6666 | // undefined shifts. When the shift is visited it will be | ||||
6667 | // simplified. | ||||
6668 | uint32_t TypeBits = RHSV.getBitWidth(); | ||||
6669 | if (ShAmt->uge(TypeBits)) | ||||
6670 | break; | ||||
6671 | |||||
6672 | uint32_t ShAmtVal = (uint32_t)ShAmt->getLimitedValue(TypeBits); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6673 | |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6674 | // If we are comparing against bits always shifted out, the |
6675 | // comparison cannot succeed. | ||||
6676 | APInt Comp = RHSV << ShAmtVal; | ||||
6677 | if (LHSI->getOpcode() == Instruction::LShr) | ||||
6678 | Comp = Comp.lshr(ShAmtVal); | ||||
6679 | else | ||||
6680 | Comp = Comp.ashr(ShAmtVal); | ||||
6681 | |||||
6682 | if (Comp != RHSV) { // Comparing against a bit that we know is zero. | ||||
6683 | bool IsICMP_NE = ICI.getPredicate() == ICmpInst::ICMP_NE; | ||||
6684 | Constant *Cst = ConstantInt::get(Type::Int1Ty, IsICMP_NE); | ||||
6685 | return ReplaceInstUsesWith(ICI, Cst); | ||||
6686 | } | ||||
6687 | |||||
6688 | // Otherwise, check to see if the bits shifted out are known to be zero. | ||||
6689 | // If so, we can compare against the unshifted value: | ||||
6690 | // (X & 4) >> 1 == 2 --> (X & 4) == 4. | ||||
Evan Cheng | fb9292a | 2008-04-23 00:38:06 +0000 | [diff] [blame] | 6691 | if (LHSI->hasOneUse() && |
6692 | MaskedValueIsZero(LHSI->getOperand(0), | ||||
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6693 | APInt::getLowBitsSet(Comp.getBitWidth(), ShAmtVal))) { |
6694 | return new ICmpInst(ICI.getPredicate(), LHSI->getOperand(0), | ||||
6695 | ConstantExpr::getShl(RHS, ShAmt)); | ||||
6696 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6697 | |
Evan Cheng | fb9292a | 2008-04-23 00:38:06 +0000 | [diff] [blame] | 6698 | if (LHSI->hasOneUse()) { |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6699 | // Otherwise strength reduce the shift into an and. |
6700 | APInt Val(APInt::getHighBitsSet(TypeBits, TypeBits - ShAmtVal)); | ||||
6701 | Constant *Mask = ConstantInt::get(Val); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6702 | |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6703 | Instruction *AndI = |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6704 | BinaryOperator::CreateAnd(LHSI->getOperand(0), |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6705 | Mask, LHSI->getName()+".mask"); |
6706 | Value *And = InsertNewInstBefore(AndI, ICI); | ||||
6707 | return new ICmpInst(ICI.getPredicate(), And, | ||||
6708 | ConstantExpr::getShl(RHS, ShAmt)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6709 | } |
6710 | break; | ||||
6711 | } | ||||
6712 | |||||
6713 | case Instruction::SDiv: | ||||
6714 | case Instruction::UDiv: | ||||
6715 | // Fold: icmp pred ([us]div X, C1), C2 -> range test | ||||
6716 | // Fold this div into the comparison, producing a range check. | ||||
6717 | // Determine, based on the divide type, what the range is being | ||||
6718 | // checked. If there is an overflow on the low or high side, remember | ||||
6719 | // it, otherwise compute the range [low, hi) bounding the new value. | ||||
6720 | // See: InsertRangeTest above for the kinds of replacements possible. | ||||
6721 | if (ConstantInt *DivRHS = dyn_cast<ConstantInt>(LHSI->getOperand(1))) | ||||
6722 | if (Instruction *R = FoldICmpDivCst(ICI, cast<BinaryOperator>(LHSI), | ||||
6723 | DivRHS)) | ||||
6724 | return R; | ||||
6725 | break; | ||||
Nick Lewycky | 0185bbf | 2008-02-03 16:33:09 +0000 | [diff] [blame] | 6726 | |
6727 | case Instruction::Add: | ||||
6728 | // Fold: icmp pred (add, X, C1), C2 | ||||
6729 | |||||
6730 | if (!ICI.isEquality()) { | ||||
6731 | ConstantInt *LHSC = dyn_cast<ConstantInt>(LHSI->getOperand(1)); | ||||
6732 | if (!LHSC) break; | ||||
6733 | const APInt &LHSV = LHSC->getValue(); | ||||
6734 | |||||
6735 | ConstantRange CR = ICI.makeConstantRange(ICI.getPredicate(), RHSV) | ||||
6736 | .subtract(LHSV); | ||||
6737 | |||||
6738 | if (ICI.isSignedPredicate()) { | ||||
6739 | if (CR.getLower().isSignBit()) { | ||||
6740 | return new ICmpInst(ICmpInst::ICMP_SLT, LHSI->getOperand(0), | ||||
6741 | ConstantInt::get(CR.getUpper())); | ||||
6742 | } else if (CR.getUpper().isSignBit()) { | ||||
6743 | return new ICmpInst(ICmpInst::ICMP_SGE, LHSI->getOperand(0), | ||||
6744 | ConstantInt::get(CR.getLower())); | ||||
6745 | } | ||||
6746 | } else { | ||||
6747 | if (CR.getLower().isMinValue()) { | ||||
6748 | return new ICmpInst(ICmpInst::ICMP_ULT, LHSI->getOperand(0), | ||||
6749 | ConstantInt::get(CR.getUpper())); | ||||
6750 | } else if (CR.getUpper().isMinValue()) { | ||||
6751 | return new ICmpInst(ICmpInst::ICMP_UGE, LHSI->getOperand(0), | ||||
6752 | ConstantInt::get(CR.getLower())); | ||||
6753 | } | ||||
6754 | } | ||||
6755 | } | ||||
6756 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6757 | } |
6758 | |||||
6759 | // Simplify icmp_eq and icmp_ne instructions with integer constant RHS. | ||||
6760 | if (ICI.isEquality()) { | ||||
6761 | bool isICMP_NE = ICI.getPredicate() == ICmpInst::ICMP_NE; | ||||
6762 | |||||
6763 | // If the first operand is (add|sub|and|or|xor|rem) with a constant, and | ||||
6764 | // the second operand is a constant, simplify a bit. | ||||
6765 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(LHSI)) { | ||||
6766 | switch (BO->getOpcode()) { | ||||
6767 | case Instruction::SRem: | ||||
6768 | // If we have a signed (X % (2^c)) == 0, turn it into an unsigned one. | ||||
6769 | if (RHSV == 0 && isa<ConstantInt>(BO->getOperand(1)) &&BO->hasOneUse()){ | ||||
6770 | const APInt &V = cast<ConstantInt>(BO->getOperand(1))->getValue(); | ||||
6771 | if (V.sgt(APInt(V.getBitWidth(), 1)) && V.isPowerOf2()) { | ||||
6772 | Instruction *NewRem = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6773 | BinaryOperator::CreateURem(BO->getOperand(0), BO->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6774 | BO->getName()); |
6775 | InsertNewInstBefore(NewRem, ICI); | ||||
6776 | return new ICmpInst(ICI.getPredicate(), NewRem, | ||||
6777 | Constant::getNullValue(BO->getType())); | ||||
6778 | } | ||||
6779 | } | ||||
6780 | break; | ||||
6781 | case Instruction::Add: | ||||
6782 | // Replace ((add A, B) != C) with (A != C-B) if B & C are constants. | ||||
6783 | if (ConstantInt *BOp1C = dyn_cast<ConstantInt>(BO->getOperand(1))) { | ||||
6784 | if (BO->hasOneUse()) | ||||
6785 | return new ICmpInst(ICI.getPredicate(), BO->getOperand(0), | ||||
6786 | Subtract(RHS, BOp1C)); | ||||
6787 | } else if (RHSV == 0) { | ||||
6788 | // Replace ((add A, B) != 0) with (A != -B) if A or B is | ||||
6789 | // efficiently invertible, or if the add has just this one use. | ||||
6790 | Value *BOp0 = BO->getOperand(0), *BOp1 = BO->getOperand(1); | ||||
6791 | |||||
6792 | if (Value *NegVal = dyn_castNegVal(BOp1)) | ||||
6793 | return new ICmpInst(ICI.getPredicate(), BOp0, NegVal); | ||||
6794 | else if (Value *NegVal = dyn_castNegVal(BOp0)) | ||||
6795 | return new ICmpInst(ICI.getPredicate(), NegVal, BOp1); | ||||
6796 | else if (BO->hasOneUse()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6797 | Instruction *Neg = BinaryOperator::CreateNeg(BOp1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6798 | InsertNewInstBefore(Neg, ICI); |
6799 | Neg->takeName(BO); | ||||
6800 | return new ICmpInst(ICI.getPredicate(), BOp0, Neg); | ||||
6801 | } | ||||
6802 | } | ||||
6803 | break; | ||||
6804 | case Instruction::Xor: | ||||
6805 | // For the xor case, we can xor two constants together, eliminating | ||||
6806 | // the explicit xor. | ||||
6807 | if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1))) | ||||
6808 | return new ICmpInst(ICI.getPredicate(), BO->getOperand(0), | ||||
6809 | ConstantExpr::getXor(RHS, BOC)); | ||||
6810 | |||||
6811 | // FALLTHROUGH | ||||
6812 | case Instruction::Sub: | ||||
6813 | // Replace (([sub|xor] A, B) != 0) with (A != B) | ||||
6814 | if (RHSV == 0) | ||||
6815 | return new ICmpInst(ICI.getPredicate(), BO->getOperand(0), | ||||
6816 | BO->getOperand(1)); | ||||
6817 | break; | ||||
6818 | |||||
6819 | case Instruction::Or: | ||||
6820 | // If bits are being or'd in that are not present in the constant we | ||||
6821 | // are comparing against, then the comparison could never succeed! | ||||
6822 | if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1))) { | ||||
6823 | Constant *NotCI = ConstantExpr::getNot(RHS); | ||||
6824 | if (!ConstantExpr::getAnd(BOC, NotCI)->isNullValue()) | ||||
6825 | return ReplaceInstUsesWith(ICI, ConstantInt::get(Type::Int1Ty, | ||||
6826 | isICMP_NE)); | ||||
6827 | } | ||||
6828 | break; | ||||
6829 | |||||
6830 | case Instruction::And: | ||||
6831 | if (ConstantInt *BOC = dyn_cast<ConstantInt>(BO->getOperand(1))) { | ||||
6832 | // If bits are being compared against that are and'd out, then the | ||||
6833 | // comparison can never succeed! | ||||
6834 | if ((RHSV & ~BOC->getValue()) != 0) | ||||
6835 | return ReplaceInstUsesWith(ICI, ConstantInt::get(Type::Int1Ty, | ||||
6836 | isICMP_NE)); | ||||
6837 | |||||
6838 | // If we have ((X & C) == C), turn it into ((X & C) != 0). | ||||
6839 | if (RHS == BOC && RHSV.isPowerOf2()) | ||||
6840 | return new ICmpInst(isICMP_NE ? ICmpInst::ICMP_EQ : | ||||
6841 | ICmpInst::ICMP_NE, LHSI, | ||||
6842 | Constant::getNullValue(RHS->getType())); | ||||
6843 | |||||
6844 | // Replace (and X, (1 << size(X)-1) != 0) with x s< 0 | ||||
Chris Lattner | 60813c2 | 2008-06-02 01:29:46 +0000 | [diff] [blame] | 6845 | if (BOC->getValue().isSignBit()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6846 | Value *X = BO->getOperand(0); |
6847 | Constant *Zero = Constant::getNullValue(X->getType()); | ||||
6848 | ICmpInst::Predicate pred = isICMP_NE ? | ||||
6849 | ICmpInst::ICMP_SLT : ICmpInst::ICMP_SGE; | ||||
6850 | return new ICmpInst(pred, X, Zero); | ||||
6851 | } | ||||
6852 | |||||
6853 | // ((X & ~7) == 0) --> X < 8 | ||||
6854 | if (RHSV == 0 && isHighOnes(BOC)) { | ||||
6855 | Value *X = BO->getOperand(0); | ||||
6856 | Constant *NegX = ConstantExpr::getNeg(BOC); | ||||
6857 | ICmpInst::Predicate pred = isICMP_NE ? | ||||
6858 | ICmpInst::ICMP_UGE : ICmpInst::ICMP_ULT; | ||||
6859 | return new ICmpInst(pred, X, NegX); | ||||
6860 | } | ||||
6861 | } | ||||
6862 | default: break; | ||||
6863 | } | ||||
6864 | } else if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(LHSI)) { | ||||
6865 | // Handle icmp {eq|ne} <intrinsic>, intcst. | ||||
6866 | if (II->getIntrinsicID() == Intrinsic::bswap) { | ||||
6867 | AddToWorkList(II); | ||||
6868 | ICI.setOperand(0, II->getOperand(1)); | ||||
6869 | ICI.setOperand(1, ConstantInt::get(RHSV.byteSwap())); | ||||
6870 | return &ICI; | ||||
6871 | } | ||||
6872 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6873 | } |
6874 | return 0; | ||||
6875 | } | ||||
6876 | |||||
6877 | /// visitICmpInstWithCastAndCast - Handle icmp (cast x to y), (cast/cst). | ||||
6878 | /// We only handle extending casts so far. | ||||
6879 | /// | ||||
6880 | Instruction *InstCombiner::visitICmpInstWithCastAndCast(ICmpInst &ICI) { | ||||
6881 | const CastInst *LHSCI = cast<CastInst>(ICI.getOperand(0)); | ||||
6882 | Value *LHSCIOp = LHSCI->getOperand(0); | ||||
6883 | const Type *SrcTy = LHSCIOp->getType(); | ||||
6884 | const Type *DestTy = LHSCI->getType(); | ||||
6885 | Value *RHSCIOp; | ||||
6886 | |||||
6887 | // Turn icmp (ptrtoint x), (ptrtoint/c) into a compare of the input if the | ||||
6888 | // integer type is the same size as the pointer type. | ||||
6889 | if (LHSCI->getOpcode() == Instruction::PtrToInt && | ||||
6890 | getTargetData().getPointerSizeInBits() == | ||||
6891 | cast<IntegerType>(DestTy)->getBitWidth()) { | ||||
6892 | Value *RHSOp = 0; | ||||
6893 | if (Constant *RHSC = dyn_cast<Constant>(ICI.getOperand(1))) { | ||||
6894 | RHSOp = ConstantExpr::getIntToPtr(RHSC, SrcTy); | ||||
6895 | } else if (PtrToIntInst *RHSC = dyn_cast<PtrToIntInst>(ICI.getOperand(1))) { | ||||
6896 | RHSOp = RHSC->getOperand(0); | ||||
6897 | // If the pointer types don't match, insert a bitcast. | ||||
6898 | if (LHSCIOp->getType() != RHSOp->getType()) | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 6899 | RHSOp = InsertBitCastBefore(RHSOp, LHSCIOp->getType(), ICI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6900 | } |
6901 | |||||
6902 | if (RHSOp) | ||||
6903 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, RHSOp); | ||||
6904 | } | ||||
6905 | |||||
6906 | // The code below only handles extension cast instructions, so far. | ||||
6907 | // Enforce this. | ||||
6908 | if (LHSCI->getOpcode() != Instruction::ZExt && | ||||
6909 | LHSCI->getOpcode() != Instruction::SExt) | ||||
6910 | return 0; | ||||
6911 | |||||
6912 | bool isSignedExt = LHSCI->getOpcode() == Instruction::SExt; | ||||
6913 | bool isSignedCmp = ICI.isSignedPredicate(); | ||||
6914 | |||||
6915 | if (CastInst *CI = dyn_cast<CastInst>(ICI.getOperand(1))) { | ||||
6916 | // Not an extension from the same type? | ||||
6917 | RHSCIOp = CI->getOperand(0); | ||||
6918 | if (RHSCIOp->getType() != LHSCIOp->getType()) | ||||
6919 | return 0; | ||||
6920 | |||||
Nick Lewycky | d4264dc | 2008-01-28 03:48:02 +0000 | [diff] [blame] | 6921 | // If the signedness of the two casts doesn't agree (i.e. one is a sext |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6922 | // and the other is a zext), then we can't handle this. |
6923 | if (CI->getOpcode() != LHSCI->getOpcode()) | ||||
6924 | return 0; | ||||
6925 | |||||
Nick Lewycky | d4264dc | 2008-01-28 03:48:02 +0000 | [diff] [blame] | 6926 | // Deal with equality cases early. |
6927 | if (ICI.isEquality()) | ||||
6928 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, RHSCIOp); | ||||
6929 | |||||
6930 | // A signed comparison of sign extended values simplifies into a | ||||
6931 | // signed comparison. | ||||
6932 | if (isSignedCmp && isSignedExt) | ||||
6933 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, RHSCIOp); | ||||
6934 | |||||
6935 | // The other three cases all fold into an unsigned comparison. | ||||
6936 | return new ICmpInst(ICI.getUnsignedPredicate(), LHSCIOp, RHSCIOp); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6937 | } |
6938 | |||||
6939 | // If we aren't dealing with a constant on the RHS, exit early | ||||
6940 | ConstantInt *CI = dyn_cast<ConstantInt>(ICI.getOperand(1)); | ||||
6941 | if (!CI) | ||||
6942 | return 0; | ||||
6943 | |||||
6944 | // Compute the constant that would happen if we truncated to SrcTy then | ||||
6945 | // reextended to DestTy. | ||||
6946 | Constant *Res1 = ConstantExpr::getTrunc(CI, SrcTy); | ||||
6947 | Constant *Res2 = ConstantExpr::getCast(LHSCI->getOpcode(), Res1, DestTy); | ||||
6948 | |||||
6949 | // If the re-extended constant didn't change... | ||||
6950 | if (Res2 == CI) { | ||||
6951 | // Make sure that sign of the Cmp and the sign of the Cast are the same. | ||||
6952 | // For example, we might have: | ||||
6953 | // %A = sext short %X to uint | ||||
6954 | // %B = icmp ugt uint %A, 1330 | ||||
6955 | // It is incorrect to transform this into | ||||
6956 | // %B = icmp ugt short %X, 1330 | ||||
6957 | // because %A may have negative value. | ||||
6958 | // | ||||
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 6959 | // However, we allow this when the compare is EQ/NE, because they are |
6960 | // signless. | ||||
6961 | if (isSignedExt == isSignedCmp || ICI.isEquality()) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6962 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, Res1); |
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 6963 | return 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6964 | } |
6965 | |||||
6966 | // The re-extended constant changed so the constant cannot be represented | ||||
6967 | // in the shorter type. Consequently, we cannot emit a simple comparison. | ||||
6968 | |||||
6969 | // First, handle some easy cases. We know the result cannot be equal at this | ||||
6970 | // point so handle the ICI.isEquality() cases | ||||
6971 | if (ICI.getPredicate() == ICmpInst::ICMP_EQ) | ||||
6972 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6973 | if (ICI.getPredicate() == ICmpInst::ICMP_NE) | ||||
6974 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6975 | |||||
6976 | // Evaluate the comparison for LT (we invert for GT below). LE and GE cases | ||||
6977 | // should have been folded away previously and not enter in here. | ||||
6978 | Value *Result; | ||||
6979 | if (isSignedCmp) { | ||||
6980 | // We're performing a signed comparison. | ||||
6981 | if (cast<ConstantInt>(CI)->getValue().isNegative()) | ||||
6982 | Result = ConstantInt::getFalse(); // X < (small) --> false | ||||
6983 | else | ||||
6984 | Result = ConstantInt::getTrue(); // X < (large) --> true | ||||
6985 | } else { | ||||
6986 | // We're performing an unsigned comparison. | ||||
6987 | if (isSignedExt) { | ||||
6988 | // We're performing an unsigned comp with a sign extended value. | ||||
6989 | // This is true if the input is >= 0. [aka >s -1] | ||||
6990 | Constant *NegOne = ConstantInt::getAllOnesValue(SrcTy); | ||||
6991 | Result = InsertNewInstBefore(new ICmpInst(ICmpInst::ICMP_SGT, LHSCIOp, | ||||
6992 | NegOne, ICI.getName()), ICI); | ||||
6993 | } else { | ||||
6994 | // Unsigned extend & unsigned compare -> always true. | ||||
6995 | Result = ConstantInt::getTrue(); | ||||
6996 | } | ||||
6997 | } | ||||
6998 | |||||
6999 | // Finally, return the value computed. | ||||
7000 | if (ICI.getPredicate() == ICmpInst::ICMP_ULT || | ||||
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 7001 | ICI.getPredicate() == ICmpInst::ICMP_SLT) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7002 | return ReplaceInstUsesWith(ICI, Result); |
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 7003 | |
7004 | assert((ICI.getPredicate()==ICmpInst::ICMP_UGT || | ||||
7005 | ICI.getPredicate()==ICmpInst::ICMP_SGT) && | ||||
7006 | "ICmp should be folded!"); | ||||
7007 | if (Constant *CI = dyn_cast<Constant>(Result)) | ||||
7008 | return ReplaceInstUsesWith(ICI, ConstantExpr::getNot(CI)); | ||||
7009 | return BinaryOperator::CreateNot(Result); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7010 | } |
7011 | |||||
7012 | Instruction *InstCombiner::visitShl(BinaryOperator &I) { | ||||
7013 | return commonShiftTransforms(I); | ||||
7014 | } | ||||
7015 | |||||
7016 | Instruction *InstCombiner::visitLShr(BinaryOperator &I) { | ||||
7017 | return commonShiftTransforms(I); | ||||
7018 | } | ||||
7019 | |||||
7020 | Instruction *InstCombiner::visitAShr(BinaryOperator &I) { | ||||
Chris Lattner | e3c504f | 2007-12-06 01:59:46 +0000 | [diff] [blame] | 7021 | if (Instruction *R = commonShiftTransforms(I)) |
7022 | return R; | ||||
7023 | |||||
7024 | Value *Op0 = I.getOperand(0); | ||||
7025 | |||||
7026 | // ashr int -1, X = -1 (for any arithmetic shift rights of ~0) | ||||
7027 | if (ConstantInt *CSI = dyn_cast<ConstantInt>(Op0)) | ||||
7028 | if (CSI->isAllOnesValue()) | ||||
7029 | return ReplaceInstUsesWith(I, CSI); | ||||
7030 | |||||
7031 | // See if we can turn a signed shr into an unsigned shr. | ||||
Nate Begeman | bb1ce94 | 2008-07-29 15:49:41 +0000 | [diff] [blame] | 7032 | if (!isa<VectorType>(I.getType()) && |
7033 | MaskedValueIsZero(Op0, | ||||
Chris Lattner | e3c504f | 2007-12-06 01:59:46 +0000 | [diff] [blame] | 7034 | APInt::getSignBit(I.getType()->getPrimitiveSizeInBits()))) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7035 | return BinaryOperator::CreateLShr(Op0, I.getOperand(1)); |
Dan Gohman | 843649e | 2009-02-24 02:00:40 +0000 | [diff] [blame] | 7036 | |
7037 | // Arithmetic shifting an all-sign-bit value is a no-op. | ||||
7038 | unsigned NumSignBits = ComputeNumSignBits(Op0); | ||||
7039 | if (NumSignBits == Op0->getType()->getPrimitiveSizeInBits()) | ||||
7040 | return ReplaceInstUsesWith(I, Op0); | ||||
7041 | |||||
Chris Lattner | e3c504f | 2007-12-06 01:59:46 +0000 | [diff] [blame] | 7042 | return 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7043 | } |
7044 | |||||
7045 | Instruction *InstCombiner::commonShiftTransforms(BinaryOperator &I) { | ||||
7046 | assert(I.getOperand(1)->getType() == I.getOperand(0)->getType()); | ||||
7047 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
7048 | |||||
7049 | // shl X, 0 == X and shr X, 0 == X | ||||
7050 | // shl 0, X == 0 and shr 0, X == 0 | ||||
7051 | if (Op1 == Constant::getNullValue(Op1->getType()) || | ||||
7052 | Op0 == Constant::getNullValue(Op0->getType())) | ||||
7053 | return ReplaceInstUsesWith(I, Op0); | ||||
7054 | |||||
7055 | if (isa<UndefValue>(Op0)) { | ||||
7056 | if (I.getOpcode() == Instruction::AShr) // undef >>s X -> undef | ||||
7057 | return ReplaceInstUsesWith(I, Op0); | ||||
7058 | else // undef << X -> 0, undef >>u X -> 0 | ||||
7059 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
7060 | } | ||||
7061 | if (isa<UndefValue>(Op1)) { | ||||
7062 | if (I.getOpcode() == Instruction::AShr) // X >>s undef -> X | ||||
7063 | return ReplaceInstUsesWith(I, Op0); | ||||
7064 | else // X << undef, X >>u undef -> 0 | ||||
7065 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
7066 | } | ||||
7067 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7068 | // Try to fold constant and into select arguments. |
7069 | if (isa<Constant>(Op0)) | ||||
7070 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) | ||||
7071 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
7072 | return R; | ||||
7073 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7074 | if (ConstantInt *CUI = dyn_cast<ConstantInt>(Op1)) |
7075 | if (Instruction *Res = FoldShiftByConstant(Op0, CUI, I)) | ||||
7076 | return Res; | ||||
7077 | return 0; | ||||
7078 | } | ||||
7079 | |||||
7080 | Instruction *InstCombiner::FoldShiftByConstant(Value *Op0, ConstantInt *Op1, | ||||
7081 | BinaryOperator &I) { | ||||
Chris Lattner | 0881733 | 2009-01-31 08:24:16 +0000 | [diff] [blame] | 7082 | bool isLeftShift = I.getOpcode() == Instruction::Shl; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7083 | |
7084 | // See if we can simplify any instructions used by the instruction whose sole | ||||
7085 | // purpose is to compute bits we don't care about. | ||||
7086 | uint32_t TypeBits = Op0->getType()->getPrimitiveSizeInBits(); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 7087 | if (SimplifyDemandedInstructionBits(I)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7088 | return &I; |
7089 | |||||
7090 | // shl uint X, 32 = 0 and shr ubyte Y, 9 = 0, ... just don't eliminate shr | ||||
7091 | // of a signed value. | ||||
7092 | // | ||||
7093 | if (Op1->uge(TypeBits)) { | ||||
7094 | if (I.getOpcode() != Instruction::AShr) | ||||
7095 | return ReplaceInstUsesWith(I, Constant::getNullValue(Op0->getType())); | ||||
7096 | else { | ||||
7097 | I.setOperand(1, ConstantInt::get(I.getType(), TypeBits-1)); | ||||
7098 | return &I; | ||||
7099 | } | ||||
7100 | } | ||||
7101 | |||||
7102 | // ((X*C1) << C2) == (X * (C1 << C2)) | ||||
7103 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0)) | ||||
7104 | if (BO->getOpcode() == Instruction::Mul && isLeftShift) | ||||
7105 | if (Constant *BOOp = dyn_cast<Constant>(BO->getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7106 | return BinaryOperator::CreateMul(BO->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7107 | ConstantExpr::getShl(BOOp, Op1)); |
7108 | |||||
7109 | // Try to fold constant and into select arguments. | ||||
7110 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
7111 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
7112 | return R; | ||||
7113 | if (isa<PHINode>(Op0)) | ||||
7114 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
7115 | return NV; | ||||
7116 | |||||
Chris Lattner | c6d1f64 | 2007-12-22 09:07:47 +0000 | [diff] [blame] | 7117 | // Fold shift2(trunc(shift1(x,c1)), c2) -> trunc(shift2(shift1(x,c1),c2)) |
7118 | if (TruncInst *TI = dyn_cast<TruncInst>(Op0)) { | ||||
7119 | Instruction *TrOp = dyn_cast<Instruction>(TI->getOperand(0)); | ||||
7120 | // If 'shift2' is an ashr, we would have to get the sign bit into a funny | ||||
7121 | // place. Don't try to do this transformation in this case. Also, we | ||||
7122 | // require that the input operand is a shift-by-constant so that we have | ||||
7123 | // confidence that the shifts will get folded together. We could do this | ||||
7124 | // xform in more cases, but it is unlikely to be profitable. | ||||
7125 | if (TrOp && I.isLogicalShift() && TrOp->isShift() && | ||||
7126 | isa<ConstantInt>(TrOp->getOperand(1))) { | ||||
7127 | // Okay, we'll do this xform. Make the shift of shift. | ||||
7128 | Constant *ShAmt = ConstantExpr::getZExt(Op1, TrOp->getType()); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7129 | Instruction *NSh = BinaryOperator::Create(I.getOpcode(), TrOp, ShAmt, |
Chris Lattner | c6d1f64 | 2007-12-22 09:07:47 +0000 | [diff] [blame] | 7130 | I.getName()); |
7131 | InsertNewInstBefore(NSh, I); // (shift2 (shift1 & 0x00FF), c2) | ||||
7132 | |||||
7133 | // For logical shifts, the truncation has the effect of making the high | ||||
7134 | // part of the register be zeros. Emulate this by inserting an AND to | ||||
7135 | // clear the top bits as needed. This 'and' will usually be zapped by | ||||
7136 | // other xforms later if dead. | ||||
7137 | unsigned SrcSize = TrOp->getType()->getPrimitiveSizeInBits(); | ||||
7138 | unsigned DstSize = TI->getType()->getPrimitiveSizeInBits(); | ||||
7139 | APInt MaskV(APInt::getLowBitsSet(SrcSize, DstSize)); | ||||
7140 | |||||
7141 | // The mask we constructed says what the trunc would do if occurring | ||||
7142 | // between the shifts. We want to know the effect *after* the second | ||||
7143 | // shift. We know that it is a logical shift by a constant, so adjust the | ||||
7144 | // mask as appropriate. | ||||
7145 | if (I.getOpcode() == Instruction::Shl) | ||||
7146 | MaskV <<= Op1->getZExtValue(); | ||||
7147 | else { | ||||
7148 | assert(I.getOpcode() == Instruction::LShr && "Unknown logical shift"); | ||||
7149 | MaskV = MaskV.lshr(Op1->getZExtValue()); | ||||
7150 | } | ||||
7151 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7152 | Instruction *And = BinaryOperator::CreateAnd(NSh, ConstantInt::get(MaskV), |
Chris Lattner | c6d1f64 | 2007-12-22 09:07:47 +0000 | [diff] [blame] | 7153 | TI->getName()); |
7154 | InsertNewInstBefore(And, I); // shift1 & 0x00FF | ||||
7155 | |||||
7156 | // Return the value truncated to the interesting size. | ||||
7157 | return new TruncInst(And, I.getType()); | ||||
7158 | } | ||||
7159 | } | ||||
7160 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7161 | if (Op0->hasOneUse()) { |
7162 | if (BinaryOperator *Op0BO = dyn_cast<BinaryOperator>(Op0)) { | ||||
7163 | // Turn ((X >> C) + Y) << C -> (X + (Y << C)) & (~0 << C) | ||||
7164 | Value *V1, *V2; | ||||
7165 | ConstantInt *CC; | ||||
7166 | switch (Op0BO->getOpcode()) { | ||||
7167 | default: break; | ||||
7168 | case Instruction::Add: | ||||
7169 | case Instruction::And: | ||||
7170 | case Instruction::Or: | ||||
7171 | case Instruction::Xor: { | ||||
7172 | // These operators commute. | ||||
7173 | // Turn (Y + (X >> C)) << C -> (X + (Y << C)) & (~0 << C) | ||||
7174 | if (isLeftShift && Op0BO->getOperand(1)->hasOneUse() && | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 7175 | match(Op0BO->getOperand(1), m_Shr(m_Value(V1), m_Specific(Op1)))){ |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7176 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7177 | Op0BO->getOperand(0), Op1, |
7178 | Op0BO->getName()); | ||||
7179 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
7180 | Instruction *X = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7181 | BinaryOperator::Create(Op0BO->getOpcode(), YS, V1, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7182 | Op0BO->getOperand(1)->getName()); |
7183 | InsertNewInstBefore(X, I); // (X + (Y << C)) | ||||
7184 | uint32_t Op1Val = Op1->getLimitedValue(TypeBits); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7185 | return BinaryOperator::CreateAnd(X, ConstantInt::get( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7186 | APInt::getHighBitsSet(TypeBits, TypeBits-Op1Val))); |
7187 | } | ||||
7188 | |||||
7189 | // Turn (Y + ((X >> C) & CC)) << C -> ((X & (CC << C)) + (Y << C)) | ||||
7190 | Value *Op0BOOp1 = Op0BO->getOperand(1); | ||||
7191 | if (isLeftShift && Op0BOOp1->hasOneUse() && | ||||
7192 | match(Op0BOOp1, | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 7193 | m_And(m_Shr(m_Value(V1), m_Specific(Op1)), |
7194 | m_ConstantInt(CC))) && | ||||
7195 | cast<BinaryOperator>(Op0BOOp1)->getOperand(0)->hasOneUse()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7196 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7197 | Op0BO->getOperand(0), Op1, |
7198 | Op0BO->getName()); | ||||
7199 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
7200 | Instruction *XM = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7201 | BinaryOperator::CreateAnd(V1, ConstantExpr::getShl(CC, Op1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7202 | V1->getName()+".mask"); |
7203 | InsertNewInstBefore(XM, I); // X & (CC << C) | ||||
7204 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7205 | return BinaryOperator::Create(Op0BO->getOpcode(), YS, XM); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7206 | } |
7207 | } | ||||
7208 | |||||
7209 | // FALL THROUGH. | ||||
7210 | case Instruction::Sub: { | ||||
7211 | // Turn ((X >> C) + Y) << C -> (X + (Y << C)) & (~0 << C) | ||||
7212 | if (isLeftShift && Op0BO->getOperand(0)->hasOneUse() && | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 7213 | match(Op0BO->getOperand(0), m_Shr(m_Value(V1), m_Specific(Op1)))){ |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7214 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7215 | Op0BO->getOperand(1), Op1, |
7216 | Op0BO->getName()); | ||||
7217 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
7218 | Instruction *X = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7219 | BinaryOperator::Create(Op0BO->getOpcode(), V1, YS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7220 | Op0BO->getOperand(0)->getName()); |
7221 | InsertNewInstBefore(X, I); // (X + (Y << C)) | ||||
7222 | uint32_t Op1Val = Op1->getLimitedValue(TypeBits); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7223 | return BinaryOperator::CreateAnd(X, ConstantInt::get( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7224 | APInt::getHighBitsSet(TypeBits, TypeBits-Op1Val))); |
7225 | } | ||||
7226 | |||||
7227 | // Turn (((X >> C)&CC) + Y) << C -> (X + (Y << C)) & (CC << C) | ||||
7228 | if (isLeftShift && Op0BO->getOperand(0)->hasOneUse() && | ||||
7229 | match(Op0BO->getOperand(0), | ||||
7230 | m_And(m_Shr(m_Value(V1), m_Value(V2)), | ||||
7231 | m_ConstantInt(CC))) && V2 == Op1 && | ||||
7232 | cast<BinaryOperator>(Op0BO->getOperand(0)) | ||||
7233 | ->getOperand(0)->hasOneUse()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7234 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7235 | Op0BO->getOperand(1), Op1, |
7236 | Op0BO->getName()); | ||||
7237 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
7238 | Instruction *XM = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7239 | BinaryOperator::CreateAnd(V1, ConstantExpr::getShl(CC, Op1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7240 | V1->getName()+".mask"); |
7241 | InsertNewInstBefore(XM, I); // X & (CC << C) | ||||
7242 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7243 | return BinaryOperator::Create(Op0BO->getOpcode(), XM, YS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7244 | } |
7245 | |||||
7246 | break; | ||||
7247 | } | ||||
7248 | } | ||||
7249 | |||||
7250 | |||||
7251 | // If the operand is an bitwise operator with a constant RHS, and the | ||||
7252 | // shift is the only use, we can pull it out of the shift. | ||||
7253 | if (ConstantInt *Op0C = dyn_cast<ConstantInt>(Op0BO->getOperand(1))) { | ||||
7254 | bool isValid = true; // Valid only for And, Or, Xor | ||||
7255 | bool highBitSet = false; // Transform if high bit of constant set? | ||||
7256 | |||||
7257 | switch (Op0BO->getOpcode()) { | ||||
7258 | default: isValid = false; break; // Do not perform transform! | ||||
7259 | case Instruction::Add: | ||||
7260 | isValid = isLeftShift; | ||||
7261 | break; | ||||
7262 | case Instruction::Or: | ||||
7263 | case Instruction::Xor: | ||||
7264 | highBitSet = false; | ||||
7265 | break; | ||||
7266 | case Instruction::And: | ||||
7267 | highBitSet = true; | ||||
7268 | break; | ||||
7269 | } | ||||
7270 | |||||
7271 | // If this is a signed shift right, and the high bit is modified | ||||
7272 | // by the logical operation, do not perform the transformation. | ||||
7273 | // The highBitSet boolean indicates the value of the high bit of | ||||
7274 | // the constant which would cause it to be modified for this | ||||
7275 | // operation. | ||||
7276 | // | ||||
Chris Lattner | 15b76e3 | 2007-12-06 06:25:04 +0000 | [diff] [blame] | 7277 | if (isValid && I.getOpcode() == Instruction::AShr) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7278 | isValid = Op0C->getValue()[TypeBits-1] == highBitSet; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7279 | |
7280 | if (isValid) { | ||||
7281 | Constant *NewRHS = ConstantExpr::get(I.getOpcode(), Op0C, Op1); | ||||
7282 | |||||
7283 | Instruction *NewShift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7284 | BinaryOperator::Create(I.getOpcode(), Op0BO->getOperand(0), Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7285 | InsertNewInstBefore(NewShift, I); |
7286 | NewShift->takeName(Op0BO); | ||||
7287 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7288 | return BinaryOperator::Create(Op0BO->getOpcode(), NewShift, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7289 | NewRHS); |
7290 | } | ||||
7291 | } | ||||
7292 | } | ||||
7293 | } | ||||
7294 | |||||
7295 | // Find out if this is a shift of a shift by a constant. | ||||
7296 | BinaryOperator *ShiftOp = dyn_cast<BinaryOperator>(Op0); | ||||
7297 | if (ShiftOp && !ShiftOp->isShift()) | ||||
7298 | ShiftOp = 0; | ||||
7299 | |||||
7300 | if (ShiftOp && isa<ConstantInt>(ShiftOp->getOperand(1))) { | ||||
7301 | ConstantInt *ShiftAmt1C = cast<ConstantInt>(ShiftOp->getOperand(1)); | ||||
7302 | uint32_t ShiftAmt1 = ShiftAmt1C->getLimitedValue(TypeBits); | ||||
7303 | uint32_t ShiftAmt2 = Op1->getLimitedValue(TypeBits); | ||||
7304 | assert(ShiftAmt2 != 0 && "Should have been simplified earlier"); | ||||
7305 | if (ShiftAmt1 == 0) return 0; // Will be simplified in the future. | ||||
7306 | Value *X = ShiftOp->getOperand(0); | ||||
7307 | |||||
7308 | uint32_t AmtSum = ShiftAmt1+ShiftAmt2; // Fold into one big shift. | ||||
7309 | if (AmtSum > TypeBits) | ||||
7310 | AmtSum = TypeBits; | ||||
7311 | |||||
7312 | const IntegerType *Ty = cast<IntegerType>(I.getType()); | ||||
7313 | |||||
7314 | // Check for (X << c1) << c2 and (X >> c1) >> c2 | ||||
7315 | if (I.getOpcode() == ShiftOp->getOpcode()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7316 | return BinaryOperator::Create(I.getOpcode(), X, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7317 | ConstantInt::get(Ty, AmtSum)); |
7318 | } else if (ShiftOp->getOpcode() == Instruction::LShr && | ||||
7319 | I.getOpcode() == Instruction::AShr) { | ||||
7320 | // ((X >>u C1) >>s C2) -> (X >>u (C1+C2)) since C1 != 0. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7321 | return BinaryOperator::CreateLShr(X, ConstantInt::get(Ty, AmtSum)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7322 | } else if (ShiftOp->getOpcode() == Instruction::AShr && |
7323 | I.getOpcode() == Instruction::LShr) { | ||||
7324 | // ((X >>s C1) >>u C2) -> ((X >>s (C1+C2)) & mask) since C1 != 0. | ||||
7325 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7326 | BinaryOperator::CreateAShr(X, ConstantInt::get(Ty, AmtSum)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7327 | InsertNewInstBefore(Shift, I); |
7328 | |||||
7329 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7330 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7331 | } |
7332 | |||||
7333 | // Okay, if we get here, one shift must be left, and the other shift must be | ||||
7334 | // right. See if the amounts are equal. | ||||
7335 | if (ShiftAmt1 == ShiftAmt2) { | ||||
7336 | // If we have ((X >>? C) << C), turn this into X & (-1 << C). | ||||
7337 | if (I.getOpcode() == Instruction::Shl) { | ||||
7338 | APInt Mask(APInt::getHighBitsSet(TypeBits, TypeBits - ShiftAmt1)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7339 | return BinaryOperator::CreateAnd(X, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7340 | } |
7341 | // If we have ((X << C) >>u C), turn this into X & (-1 >>u C). | ||||
7342 | if (I.getOpcode() == Instruction::LShr) { | ||||
7343 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt1)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7344 | return BinaryOperator::CreateAnd(X, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7345 | } |
7346 | // We can simplify ((X << C) >>s C) into a trunc + sext. | ||||
7347 | // NOTE: we could do this for any C, but that would make 'unusual' integer | ||||
7348 | // types. For now, just stick to ones well-supported by the code | ||||
7349 | // generators. | ||||
7350 | const Type *SExtType = 0; | ||||
7351 | switch (Ty->getBitWidth() - ShiftAmt1) { | ||||
7352 | case 1 : | ||||
7353 | case 8 : | ||||
7354 | case 16 : | ||||
7355 | case 32 : | ||||
7356 | case 64 : | ||||
7357 | case 128: | ||||
7358 | SExtType = IntegerType::get(Ty->getBitWidth() - ShiftAmt1); | ||||
7359 | break; | ||||
7360 | default: break; | ||||
7361 | } | ||||
7362 | if (SExtType) { | ||||
7363 | Instruction *NewTrunc = new TruncInst(X, SExtType, "sext"); | ||||
7364 | InsertNewInstBefore(NewTrunc, I); | ||||
7365 | return new SExtInst(NewTrunc, Ty); | ||||
7366 | } | ||||
7367 | // Otherwise, we can't handle it yet. | ||||
7368 | } else if (ShiftAmt1 < ShiftAmt2) { | ||||
7369 | uint32_t ShiftDiff = ShiftAmt2-ShiftAmt1; | ||||
7370 | |||||
7371 | // (X >>? C1) << C2 --> X << (C2-C1) & (-1 << C2) | ||||
7372 | if (I.getOpcode() == Instruction::Shl) { | ||||
7373 | assert(ShiftOp->getOpcode() == Instruction::LShr || | ||||
7374 | ShiftOp->getOpcode() == Instruction::AShr); | ||||
7375 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7376 | BinaryOperator::CreateShl(X, ConstantInt::get(Ty, ShiftDiff)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7377 | InsertNewInstBefore(Shift, I); |
7378 | |||||
7379 | APInt Mask(APInt::getHighBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7380 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7381 | } |
7382 | |||||
7383 | // (X << C1) >>u C2 --> X >>u (C2-C1) & (-1 >> C2) | ||||
7384 | if (I.getOpcode() == Instruction::LShr) { | ||||
7385 | assert(ShiftOp->getOpcode() == Instruction::Shl); | ||||
7386 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7387 | BinaryOperator::CreateLShr(X, ConstantInt::get(Ty, ShiftDiff)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7388 | InsertNewInstBefore(Shift, I); |
7389 | |||||
7390 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7391 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7392 | } |
7393 | |||||
7394 | // We can't handle (X << C1) >>s C2, it shifts arbitrary bits in. | ||||
7395 | } else { | ||||
7396 | assert(ShiftAmt2 < ShiftAmt1); | ||||
7397 | uint32_t ShiftDiff = ShiftAmt1-ShiftAmt2; | ||||
7398 | |||||
7399 | // (X >>? C1) << C2 --> X >>? (C1-C2) & (-1 << C2) | ||||
7400 | if (I.getOpcode() == Instruction::Shl) { | ||||
7401 | assert(ShiftOp->getOpcode() == Instruction::LShr || | ||||
7402 | ShiftOp->getOpcode() == Instruction::AShr); | ||||
7403 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7404 | BinaryOperator::Create(ShiftOp->getOpcode(), X, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7405 | ConstantInt::get(Ty, ShiftDiff)); |
7406 | InsertNewInstBefore(Shift, I); | ||||
7407 | |||||
7408 | APInt Mask(APInt::getHighBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7409 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7410 | } |
7411 | |||||
7412 | // (X << C1) >>u C2 --> X << (C1-C2) & (-1 >> C2) | ||||
7413 | if (I.getOpcode() == Instruction::LShr) { | ||||
7414 | assert(ShiftOp->getOpcode() == Instruction::Shl); | ||||
7415 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7416 | BinaryOperator::CreateShl(X, ConstantInt::get(Ty, ShiftDiff)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7417 | InsertNewInstBefore(Shift, I); |
7418 | |||||
7419 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7420 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7421 | } |
7422 | |||||
7423 | // We can't handle (X << C1) >>a C2, it shifts arbitrary bits in. | ||||
7424 | } | ||||
7425 | } | ||||
7426 | return 0; | ||||
7427 | } | ||||
7428 | |||||
7429 | |||||
7430 | /// DecomposeSimpleLinearExpr - Analyze 'Val', seeing if it is a simple linear | ||||
7431 | /// expression. If so, decompose it, returning some value X, such that Val is | ||||
7432 | /// X*Scale+Offset. | ||||
7433 | /// | ||||
7434 | static Value *DecomposeSimpleLinearExpr(Value *Val, unsigned &Scale, | ||||
7435 | int &Offset) { | ||||
7436 | assert(Val->getType() == Type::Int32Ty && "Unexpected allocation size type!"); | ||||
7437 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Val)) { | ||||
7438 | Offset = CI->getZExtValue(); | ||||
Chris Lattner | c59171a | 2007-10-12 05:30:59 +0000 | [diff] [blame] | 7439 | Scale = 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7440 | return ConstantInt::get(Type::Int32Ty, 0); |
Chris Lattner | c59171a | 2007-10-12 05:30:59 +0000 | [diff] [blame] | 7441 | } else if (BinaryOperator *I = dyn_cast<BinaryOperator>(Val)) { |
7442 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
7443 | if (I->getOpcode() == Instruction::Shl) { | ||||
7444 | // This is a value scaled by '1 << the shift amt'. | ||||
7445 | Scale = 1U << RHS->getZExtValue(); | ||||
7446 | Offset = 0; | ||||
7447 | return I->getOperand(0); | ||||
7448 | } else if (I->getOpcode() == Instruction::Mul) { | ||||
7449 | // This value is scaled by 'RHS'. | ||||
7450 | Scale = RHS->getZExtValue(); | ||||
7451 | Offset = 0; | ||||
7452 | return I->getOperand(0); | ||||
7453 | } else if (I->getOpcode() == Instruction::Add) { | ||||
7454 | // We have X+C. Check to see if we really have (X*C2)+C1, | ||||
7455 | // where C1 is divisible by C2. | ||||
7456 | unsigned SubScale; | ||||
7457 | Value *SubVal = | ||||
7458 | DecomposeSimpleLinearExpr(I->getOperand(0), SubScale, Offset); | ||||
7459 | Offset += RHS->getZExtValue(); | ||||
7460 | Scale = SubScale; | ||||
7461 | return SubVal; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7462 | } |
7463 | } | ||||
7464 | } | ||||
7465 | |||||
7466 | // Otherwise, we can't look past this. | ||||
7467 | Scale = 1; | ||||
7468 | Offset = 0; | ||||
7469 | return Val; | ||||
7470 | } | ||||
7471 | |||||
7472 | |||||
7473 | /// PromoteCastOfAllocation - If we find a cast of an allocation instruction, | ||||
7474 | /// try to eliminate the cast by moving the type information into the alloc. | ||||
7475 | Instruction *InstCombiner::PromoteCastOfAllocation(BitCastInst &CI, | ||||
7476 | AllocationInst &AI) { | ||||
7477 | const PointerType *PTy = cast<PointerType>(CI.getType()); | ||||
7478 | |||||
7479 | // Remove any uses of AI that are dead. | ||||
7480 | assert(!CI.use_empty() && "Dead instructions should be removed earlier!"); | ||||
7481 | |||||
7482 | for (Value::use_iterator UI = AI.use_begin(), E = AI.use_end(); UI != E; ) { | ||||
7483 | Instruction *User = cast<Instruction>(*UI++); | ||||
7484 | if (isInstructionTriviallyDead(User)) { | ||||
7485 | while (UI != E && *UI == User) | ||||
7486 | ++UI; // If this instruction uses AI more than once, don't break UI. | ||||
7487 | |||||
7488 | ++NumDeadInst; | ||||
7489 | DOUT << "IC: DCE: " << *User; | ||||
7490 | EraseInstFromFunction(*User); | ||||
7491 | } | ||||
7492 | } | ||||
7493 | |||||
7494 | // Get the type really allocated and the type casted to. | ||||
7495 | const Type *AllocElTy = AI.getAllocatedType(); | ||||
7496 | const Type *CastElTy = PTy->getElementType(); | ||||
7497 | if (!AllocElTy->isSized() || !CastElTy->isSized()) return 0; | ||||
7498 | |||||
7499 | unsigned AllocElTyAlign = TD->getABITypeAlignment(AllocElTy); | ||||
7500 | unsigned CastElTyAlign = TD->getABITypeAlignment(CastElTy); | ||||
7501 | if (CastElTyAlign < AllocElTyAlign) return 0; | ||||
7502 | |||||
7503 | // If the allocation has multiple uses, only promote it if we are strictly | ||||
7504 | // increasing the alignment of the resultant allocation. If we keep it the | ||||
7505 | // same, we open the door to infinite loops of various kinds. | ||||
7506 | if (!AI.hasOneUse() && CastElTyAlign == AllocElTyAlign) return 0; | ||||
7507 | |||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 7508 | uint64_t AllocElTySize = TD->getTypePaddedSize(AllocElTy); |
7509 | uint64_t CastElTySize = TD->getTypePaddedSize(CastElTy); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7510 | if (CastElTySize == 0 || AllocElTySize == 0) return 0; |
7511 | |||||
7512 | // See if we can satisfy the modulus by pulling a scale out of the array | ||||
7513 | // size argument. | ||||
7514 | unsigned ArraySizeScale; | ||||
7515 | int ArrayOffset; | ||||
7516 | Value *NumElements = // See if the array size is a decomposable linear expr. | ||||
7517 | DecomposeSimpleLinearExpr(AI.getOperand(0), ArraySizeScale, ArrayOffset); | ||||
7518 | |||||
7519 | // If we can now satisfy the modulus, by using a non-1 scale, we really can | ||||
7520 | // do the xform. | ||||
7521 | if ((AllocElTySize*ArraySizeScale) % CastElTySize != 0 || | ||||
7522 | (AllocElTySize*ArrayOffset ) % CastElTySize != 0) return 0; | ||||
7523 | |||||
7524 | unsigned Scale = (AllocElTySize*ArraySizeScale)/CastElTySize; | ||||
7525 | Value *Amt = 0; | ||||
7526 | if (Scale == 1) { | ||||
7527 | Amt = NumElements; | ||||
7528 | } else { | ||||
7529 | // If the allocation size is constant, form a constant mul expression | ||||
7530 | Amt = ConstantInt::get(Type::Int32Ty, Scale); | ||||
7531 | if (isa<ConstantInt>(NumElements)) | ||||
7532 | Amt = Multiply(cast<ConstantInt>(NumElements), cast<ConstantInt>(Amt)); | ||||
7533 | // otherwise multiply the amount and the number of elements | ||||
7534 | else if (Scale != 1) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7535 | Instruction *Tmp = BinaryOperator::CreateMul(Amt, NumElements, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7536 | Amt = InsertNewInstBefore(Tmp, AI); |
7537 | } | ||||
7538 | } | ||||
7539 | |||||
7540 | if (int Offset = (AllocElTySize*ArrayOffset)/CastElTySize) { | ||||
7541 | Value *Off = ConstantInt::get(Type::Int32Ty, Offset, true); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7542 | Instruction *Tmp = BinaryOperator::CreateAdd(Amt, Off, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7543 | Amt = InsertNewInstBefore(Tmp, AI); |
7544 | } | ||||
7545 | |||||
7546 | AllocationInst *New; | ||||
7547 | if (isa<MallocInst>(AI)) | ||||
7548 | New = new MallocInst(CastElTy, Amt, AI.getAlignment()); | ||||
7549 | else | ||||
7550 | New = new AllocaInst(CastElTy, Amt, AI.getAlignment()); | ||||
7551 | InsertNewInstBefore(New, AI); | ||||
7552 | New->takeName(&AI); | ||||
7553 | |||||
7554 | // If the allocation has multiple uses, insert a cast and change all things | ||||
7555 | // that used it to use the new cast. This will also hack on CI, but it will | ||||
7556 | // die soon. | ||||
7557 | if (!AI.hasOneUse()) { | ||||
7558 | AddUsesToWorkList(AI); | ||||
7559 | // New is the allocation instruction, pointer typed. AI is the original | ||||
7560 | // allocation instruction, also pointer typed. Thus, cast to use is BitCast. | ||||
7561 | CastInst *NewCast = new BitCastInst(New, AI.getType(), "tmpcast"); | ||||
7562 | InsertNewInstBefore(NewCast, AI); | ||||
7563 | AI.replaceAllUsesWith(NewCast); | ||||
7564 | } | ||||
7565 | return ReplaceInstUsesWith(CI, New); | ||||
7566 | } | ||||
7567 | |||||
7568 | /// CanEvaluateInDifferentType - Return true if we can take the specified value | ||||
7569 | /// and return it as type Ty without inserting any new casts and without | ||||
7570 | /// changing the computed value. This is used by code that tries to decide | ||||
7571 | /// whether promoting or shrinking integer operations to wider or smaller types | ||||
7572 | /// will allow us to eliminate a truncate or extend. | ||||
7573 | /// | ||||
7574 | /// This is a truncation operation if Ty is smaller than V->getType(), or an | ||||
7575 | /// extension operation if Ty is larger. | ||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7576 | /// |
7577 | /// If CastOpc is a truncation, then Ty will be a type smaller than V. We | ||||
7578 | /// should return true if trunc(V) can be computed by computing V in the smaller | ||||
7579 | /// type. If V is an instruction, then trunc(inst(x,y)) can be computed as | ||||
7580 | /// inst(trunc(x),trunc(y)), which only makes sense if x and y can be | ||||
7581 | /// efficiently truncated. | ||||
7582 | /// | ||||
7583 | /// If CastOpc is a sext or zext, we are asking if the low bits of the value can | ||||
7584 | /// bit computed in a larger type, which is then and'd or sext_in_reg'd to get | ||||
7585 | /// the final result. | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7586 | bool InstCombiner::CanEvaluateInDifferentType(Value *V, const IntegerType *Ty, |
7587 | unsigned CastOpc, | ||||
7588 | int &NumCastsRemoved){ | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7589 | // We can always evaluate constants in another type. |
7590 | if (isa<ConstantInt>(V)) | ||||
7591 | return true; | ||||
7592 | |||||
7593 | Instruction *I = dyn_cast<Instruction>(V); | ||||
7594 | if (!I) return false; | ||||
7595 | |||||
7596 | const IntegerType *OrigTy = cast<IntegerType>(V->getType()); | ||||
7597 | |||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7598 | // If this is an extension or truncate, we can often eliminate it. |
7599 | if (isa<TruncInst>(I) || isa<ZExtInst>(I) || isa<SExtInst>(I)) { | ||||
7600 | // If this is a cast from the destination type, we can trivially eliminate | ||||
7601 | // it, and this will remove a cast overall. | ||||
7602 | if (I->getOperand(0)->getType() == Ty) { | ||||
7603 | // If the first operand is itself a cast, and is eliminable, do not count | ||||
7604 | // this as an eliminable cast. We would prefer to eliminate those two | ||||
7605 | // casts first. | ||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7606 | if (!isa<CastInst>(I->getOperand(0)) && I->hasOneUse()) |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7607 | ++NumCastsRemoved; |
7608 | return true; | ||||
7609 | } | ||||
7610 | } | ||||
7611 | |||||
7612 | // We can't extend or shrink something that has multiple uses: doing so would | ||||
7613 | // require duplicating the instruction in general, which isn't profitable. | ||||
7614 | if (!I->hasOneUse()) return false; | ||||
7615 | |||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7616 | unsigned Opc = I->getOpcode(); |
7617 | switch (Opc) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7618 | case Instruction::Add: |
7619 | case Instruction::Sub: | ||||
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7620 | case Instruction::Mul: |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7621 | case Instruction::And: |
7622 | case Instruction::Or: | ||||
7623 | case Instruction::Xor: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7624 | // These operators can all arbitrarily be extended or truncated. |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7625 | return CanEvaluateInDifferentType(I->getOperand(0), Ty, CastOpc, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7626 | NumCastsRemoved) && |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7627 | CanEvaluateInDifferentType(I->getOperand(1), Ty, CastOpc, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7628 | NumCastsRemoved); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7629 | |
7630 | case Instruction::Shl: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7631 | // If we are truncating the result of this SHL, and if it's a shift of a |
7632 | // constant amount, we can always perform a SHL in a smaller type. | ||||
7633 | if (ConstantInt *CI = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
7634 | uint32_t BitWidth = Ty->getBitWidth(); | ||||
7635 | if (BitWidth < OrigTy->getBitWidth() && | ||||
7636 | CI->getLimitedValue(BitWidth) < BitWidth) | ||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7637 | return CanEvaluateInDifferentType(I->getOperand(0), Ty, CastOpc, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7638 | NumCastsRemoved); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7639 | } |
7640 | break; | ||||
7641 | case Instruction::LShr: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7642 | // If this is a truncate of a logical shr, we can truncate it to a smaller |
7643 | // lshr iff we know that the bits we would otherwise be shifting in are | ||||
7644 | // already zeros. | ||||
7645 | if (ConstantInt *CI = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
7646 | uint32_t OrigBitWidth = OrigTy->getBitWidth(); | ||||
7647 | uint32_t BitWidth = Ty->getBitWidth(); | ||||
7648 | if (BitWidth < OrigBitWidth && | ||||
7649 | MaskedValueIsZero(I->getOperand(0), | ||||
7650 | APInt::getHighBitsSet(OrigBitWidth, OrigBitWidth-BitWidth)) && | ||||
7651 | CI->getLimitedValue(BitWidth) < BitWidth) { | ||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7652 | return CanEvaluateInDifferentType(I->getOperand(0), Ty, CastOpc, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7653 | NumCastsRemoved); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7654 | } |
7655 | } | ||||
7656 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7657 | case Instruction::ZExt: |
7658 | case Instruction::SExt: | ||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7659 | case Instruction::Trunc: |
7660 | // If this is the same kind of case as our original (e.g. zext+zext), we | ||||
Chris Lattner | 9c909d2 | 2007-08-02 17:23:38 +0000 | [diff] [blame] | 7661 | // can safely replace it. Note that replacing it does not reduce the number |
7662 | // of casts in the input. | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7663 | if (Opc == CastOpc) |
7664 | return true; | ||||
7665 | |||||
7666 | // sext (zext ty1), ty2 -> zext ty2 | ||||
Evan Cheng | 7bb0d95 | 2009-01-15 17:09:07 +0000 | [diff] [blame] | 7667 | if (CastOpc == Instruction::SExt && Opc == Instruction::ZExt) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7668 | return true; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7669 | break; |
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7670 | case Instruction::Select: { |
7671 | SelectInst *SI = cast<SelectInst>(I); | ||||
7672 | return CanEvaluateInDifferentType(SI->getTrueValue(), Ty, CastOpc, | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7673 | NumCastsRemoved) && |
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7674 | CanEvaluateInDifferentType(SI->getFalseValue(), Ty, CastOpc, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7675 | NumCastsRemoved); |
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7676 | } |
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7677 | case Instruction::PHI: { |
7678 | // We can change a phi if we can change all operands. | ||||
7679 | PHINode *PN = cast<PHINode>(I); | ||||
7680 | for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) | ||||
7681 | if (!CanEvaluateInDifferentType(PN->getIncomingValue(i), Ty, CastOpc, | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7682 | NumCastsRemoved)) |
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7683 | return false; |
7684 | return true; | ||||
7685 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7686 | default: |
7687 | // TODO: Can handle more cases here. | ||||
7688 | break; | ||||
7689 | } | ||||
7690 | |||||
7691 | return false; | ||||
7692 | } | ||||
7693 | |||||
7694 | /// EvaluateInDifferentType - Given an expression that | ||||
7695 | /// CanEvaluateInDifferentType returns true for, actually insert the code to | ||||
7696 | /// evaluate the expression. | ||||
7697 | Value *InstCombiner::EvaluateInDifferentType(Value *V, const Type *Ty, | ||||
7698 | bool isSigned) { | ||||
7699 | if (Constant *C = dyn_cast<Constant>(V)) | ||||
7700 | return ConstantExpr::getIntegerCast(C, Ty, isSigned /*Sext or ZExt*/); | ||||
7701 | |||||
7702 | // Otherwise, it must be an instruction. | ||||
7703 | Instruction *I = cast<Instruction>(V); | ||||
7704 | Instruction *Res = 0; | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7705 | unsigned Opc = I->getOpcode(); |
7706 | switch (Opc) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7707 | case Instruction::Add: |
7708 | case Instruction::Sub: | ||||
Nick Lewycky | c52646a | 2008-01-22 05:08:48 +0000 | [diff] [blame] | 7709 | case Instruction::Mul: |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7710 | case Instruction::And: |
7711 | case Instruction::Or: | ||||
7712 | case Instruction::Xor: | ||||
7713 | case Instruction::AShr: | ||||
7714 | case Instruction::LShr: | ||||
7715 | case Instruction::Shl: { | ||||
7716 | Value *LHS = EvaluateInDifferentType(I->getOperand(0), Ty, isSigned); | ||||
7717 | Value *RHS = EvaluateInDifferentType(I->getOperand(1), Ty, isSigned); | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7718 | Res = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7719 | break; |
7720 | } | ||||
7721 | case Instruction::Trunc: | ||||
7722 | case Instruction::ZExt: | ||||
7723 | case Instruction::SExt: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7724 | // If the source type of the cast is the type we're trying for then we can |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7725 | // just return the source. There's no need to insert it because it is not |
7726 | // new. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7727 | if (I->getOperand(0)->getType() == Ty) |
7728 | return I->getOperand(0); | ||||
7729 | |||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7730 | // Otherwise, must be the same type of cast, so just reinsert a new one. |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7731 | Res = CastInst::Create(cast<CastInst>(I)->getOpcode(), I->getOperand(0), |
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7732 | Ty); |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7733 | break; |
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7734 | case Instruction::Select: { |
7735 | Value *True = EvaluateInDifferentType(I->getOperand(1), Ty, isSigned); | ||||
7736 | Value *False = EvaluateInDifferentType(I->getOperand(2), Ty, isSigned); | ||||
7737 | Res = SelectInst::Create(I->getOperand(0), True, False); | ||||
7738 | break; | ||||
7739 | } | ||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7740 | case Instruction::PHI: { |
7741 | PHINode *OPN = cast<PHINode>(I); | ||||
7742 | PHINode *NPN = PHINode::Create(Ty); | ||||
7743 | for (unsigned i = 0, e = OPN->getNumIncomingValues(); i != e; ++i) { | ||||
7744 | Value *V =EvaluateInDifferentType(OPN->getIncomingValue(i), Ty, isSigned); | ||||
7745 | NPN->addIncoming(V, OPN->getIncomingBlock(i)); | ||||
7746 | } | ||||
7747 | Res = NPN; | ||||
7748 | break; | ||||
7749 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7750 | default: |
7751 | // TODO: Can handle more cases here. | ||||
7752 | assert(0 && "Unreachable!"); | ||||
7753 | break; | ||||
7754 | } | ||||
7755 | |||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7756 | Res->takeName(I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7757 | return InsertNewInstBefore(Res, *I); |
7758 | } | ||||
7759 | |||||
7760 | /// @brief Implement the transforms common to all CastInst visitors. | ||||
7761 | Instruction *InstCombiner::commonCastTransforms(CastInst &CI) { | ||||
7762 | Value *Src = CI.getOperand(0); | ||||
7763 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7764 | // Many cases of "cast of a cast" are eliminable. If it's eliminable we just |
7765 | // eliminate it now. | ||||
7766 | if (CastInst *CSrc = dyn_cast<CastInst>(Src)) { // A->B->C cast | ||||
7767 | if (Instruction::CastOps opc = | ||||
7768 | isEliminableCastPair(CSrc, CI.getOpcode(), CI.getType(), TD)) { | ||||
7769 | // The first cast (CSrc) is eliminable so we need to fix up or replace | ||||
7770 | // the second cast (CI). CSrc will then have a good chance of being dead. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7771 | return CastInst::Create(opc, CSrc->getOperand(0), CI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7772 | } |
7773 | } | ||||
7774 | |||||
7775 | // If we are casting a select then fold the cast into the select | ||||
7776 | if (SelectInst *SI = dyn_cast<SelectInst>(Src)) | ||||
7777 | if (Instruction *NV = FoldOpIntoSelect(CI, SI, this)) | ||||
7778 | return NV; | ||||
7779 | |||||
7780 | // If we are casting a PHI then fold the cast into the PHI | ||||
7781 | if (isa<PHINode>(Src)) | ||||
7782 | if (Instruction *NV = FoldOpIntoPhi(CI)) | ||||
7783 | return NV; | ||||
7784 | |||||
7785 | return 0; | ||||
7786 | } | ||||
7787 | |||||
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7788 | /// FindElementAtOffset - Given a type and a constant offset, determine whether |
7789 | /// or not there is a sequence of GEP indices into the type that will land us at | ||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 7790 | /// the specified offset. If so, fill them into NewIndices and return the |
7791 | /// resultant element type, otherwise return null. | ||||
7792 | static const Type *FindElementAtOffset(const Type *Ty, int64_t Offset, | ||||
7793 | SmallVectorImpl<Value*> &NewIndices, | ||||
7794 | const TargetData *TD) { | ||||
7795 | if (!Ty->isSized()) return 0; | ||||
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7796 | |
7797 | // Start with the index over the outer type. Note that the type size | ||||
7798 | // might be zero (even if the offset isn't zero) if the indexed type | ||||
7799 | // is something like [0 x {int, int}] | ||||
7800 | const Type *IntPtrTy = TD->getIntPtrType(); | ||||
7801 | int64_t FirstIdx = 0; | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 7802 | if (int64_t TySize = TD->getTypePaddedSize(Ty)) { |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7803 | FirstIdx = Offset/TySize; |
Chris Lattner | 0bd6f2b | 2009-01-11 20:41:36 +0000 | [diff] [blame] | 7804 | Offset -= FirstIdx*TySize; |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7805 | |
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7806 | // Handle hosts where % returns negative instead of values [0..TySize). |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7807 | if (Offset < 0) { |
7808 | --FirstIdx; | ||||
7809 | Offset += TySize; | ||||
7810 | assert(Offset >= 0); | ||||
7811 | } | ||||
7812 | assert((uint64_t)Offset < (uint64_t)TySize && "Out of range offset"); | ||||
7813 | } | ||||
7814 | |||||
7815 | NewIndices.push_back(ConstantInt::get(IntPtrTy, FirstIdx)); | ||||
7816 | |||||
7817 | // Index into the types. If we fail, set OrigBase to null. | ||||
7818 | while (Offset) { | ||||
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7819 | // Indexing into tail padding between struct/array elements. |
7820 | if (uint64_t(Offset*8) >= TD->getTypeSizeInBits(Ty)) | ||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 7821 | return 0; |
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7822 | |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7823 | if (const StructType *STy = dyn_cast<StructType>(Ty)) { |
7824 | const StructLayout *SL = TD->getStructLayout(STy); | ||||
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7825 | assert(Offset < (int64_t)SL->getSizeInBytes() && |
7826 | "Offset must stay within the indexed type"); | ||||
7827 | |||||
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7828 | unsigned Elt = SL->getElementContainingOffset(Offset); |
7829 | NewIndices.push_back(ConstantInt::get(Type::Int32Ty, Elt)); | ||||
7830 | |||||
7831 | Offset -= SL->getElementOffset(Elt); | ||||
7832 | Ty = STy->getElementType(Elt); | ||||
Chris Lattner | d35ce6a | 2009-01-11 20:23:52 +0000 | [diff] [blame] | 7833 | } else if (const ArrayType *AT = dyn_cast<ArrayType>(Ty)) { |
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 7834 | uint64_t EltSize = TD->getTypePaddedSize(AT->getElementType()); |
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7835 | assert(EltSize && "Cannot index into a zero-sized array"); |
7836 | NewIndices.push_back(ConstantInt::get(IntPtrTy,Offset/EltSize)); | ||||
7837 | Offset %= EltSize; | ||||
Chris Lattner | d35ce6a | 2009-01-11 20:23:52 +0000 | [diff] [blame] | 7838 | Ty = AT->getElementType(); |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7839 | } else { |
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7840 | // Otherwise, we can't index into the middle of this atomic type, bail. |
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 7841 | return 0; |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7842 | } |
7843 | } | ||||
7844 | |||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 7845 | return Ty; |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7846 | } |
7847 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7848 | /// @brief Implement the transforms for cast of pointer (bitcast/ptrtoint) |
7849 | Instruction *InstCombiner::commonPointerCastTransforms(CastInst &CI) { | ||||
7850 | Value *Src = CI.getOperand(0); | ||||
7851 | |||||
7852 | if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Src)) { | ||||
7853 | // If casting the result of a getelementptr instruction with no offset, turn | ||||
7854 | // this into a cast of the original pointer! | ||||
7855 | if (GEP->hasAllZeroIndices()) { | ||||
7856 | // Changing the cast operand is usually not a good idea but it is safe | ||||
7857 | // here because the pointer operand is being replaced with another | ||||
7858 | // pointer operand so the opcode doesn't need to change. | ||||
7859 | AddToWorkList(GEP); | ||||
7860 | CI.setOperand(0, GEP->getOperand(0)); | ||||
7861 | return &CI; | ||||
7862 | } | ||||
7863 | |||||
7864 | // If the GEP has a single use, and the base pointer is a bitcast, and the | ||||
7865 | // GEP computes a constant offset, see if we can convert these three | ||||
7866 | // instructions into fewer. This typically happens with unions and other | ||||
7867 | // non-type-safe code. | ||||
7868 | if (GEP->hasOneUse() && isa<BitCastInst>(GEP->getOperand(0))) { | ||||
7869 | if (GEP->hasAllConstantIndices()) { | ||||
7870 | // We are guaranteed to get a constant from EmitGEPOffset. | ||||
7871 | ConstantInt *OffsetV = cast<ConstantInt>(EmitGEPOffset(GEP, CI, *this)); | ||||
7872 | int64_t Offset = OffsetV->getSExtValue(); | ||||
7873 | |||||
7874 | // Get the base pointer input of the bitcast, and the type it points to. | ||||
7875 | Value *OrigBase = cast<BitCastInst>(GEP->getOperand(0))->getOperand(0); | ||||
7876 | const Type *GEPIdxTy = | ||||
7877 | cast<PointerType>(OrigBase->getType())->getElementType(); | ||||
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7878 | SmallVector<Value*, 8> NewIndices; |
7879 | if (FindElementAtOffset(GEPIdxTy, Offset, NewIndices, TD)) { | ||||
7880 | // If we were able to index down into an element, create the GEP | ||||
7881 | // and bitcast the result. This eliminates one bitcast, potentially | ||||
7882 | // two. | ||||
7883 | Instruction *NGEP = GetElementPtrInst::Create(OrigBase, | ||||
7884 | NewIndices.begin(), | ||||
7885 | NewIndices.end(), ""); | ||||
7886 | InsertNewInstBefore(NGEP, CI); | ||||
7887 | NGEP->takeName(GEP); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7888 | |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7889 | if (isa<BitCastInst>(CI)) |
7890 | return new BitCastInst(NGEP, CI.getType()); | ||||
7891 | assert(isa<PtrToIntInst>(CI)); | ||||
7892 | return new PtrToIntInst(NGEP, CI.getType()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7893 | } |
7894 | } | ||||
7895 | } | ||||
7896 | } | ||||
7897 | |||||
7898 | return commonCastTransforms(CI); | ||||
7899 | } | ||||
7900 | |||||
7901 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7902 | /// Only the TRUNC, ZEXT, SEXT, and BITCAST can both operand and result as |
7903 | /// integer types. This function implements the common transforms for all those | ||||
7904 | /// cases. | ||||
7905 | /// @brief Implement the transforms common to CastInst with integer operands | ||||
7906 | Instruction *InstCombiner::commonIntCastTransforms(CastInst &CI) { | ||||
7907 | if (Instruction *Result = commonCastTransforms(CI)) | ||||
7908 | return Result; | ||||
7909 | |||||
7910 | Value *Src = CI.getOperand(0); | ||||
7911 | const Type *SrcTy = Src->getType(); | ||||
7912 | const Type *DestTy = CI.getType(); | ||||
7913 | uint32_t SrcBitSize = SrcTy->getPrimitiveSizeInBits(); | ||||
7914 | uint32_t DestBitSize = DestTy->getPrimitiveSizeInBits(); | ||||
7915 | |||||
7916 | // See if we can simplify any instructions used by the LHS whose sole | ||||
7917 | // purpose is to compute bits we don't care about. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 7918 | if (SimplifyDemandedInstructionBits(CI)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7919 | return &CI; |
7920 | |||||
7921 | // If the source isn't an instruction or has more than one use then we | ||||
7922 | // can't do anything more. | ||||
7923 | Instruction *SrcI = dyn_cast<Instruction>(Src); | ||||
7924 | if (!SrcI || !Src->hasOneUse()) | ||||
7925 | return 0; | ||||
7926 | |||||
7927 | // Attempt to propagate the cast into the instruction for int->int casts. | ||||
7928 | int NumCastsRemoved = 0; | ||||
7929 | if (!isa<BitCastInst>(CI) && | ||||
7930 | CanEvaluateInDifferentType(SrcI, cast<IntegerType>(DestTy), | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7931 | CI.getOpcode(), NumCastsRemoved)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7932 | // If this cast is a truncate, evaluting in a different type always |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7933 | // eliminates the cast, so it is always a win. If this is a zero-extension, |
7934 | // we need to do an AND to maintain the clear top-part of the computation, | ||||
7935 | // so we require that the input have eliminated at least one cast. If this | ||||
7936 | // is a sign extension, we insert two new casts (to do the extension) so we | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7937 | // require that two casts have been eliminated. |
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7938 | bool DoXForm = false; |
7939 | bool JustReplace = false; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7940 | switch (CI.getOpcode()) { |
7941 | default: | ||||
7942 | // All the others use floating point so we shouldn't actually | ||||
7943 | // get here because of the check above. | ||||
7944 | assert(0 && "Unknown cast type"); | ||||
7945 | case Instruction::Trunc: | ||||
7946 | DoXForm = true; | ||||
7947 | break; | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7948 | case Instruction::ZExt: { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7949 | DoXForm = NumCastsRemoved >= 1; |
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7950 | if (!DoXForm && 0) { |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7951 | // If it's unnecessary to issue an AND to clear the high bits, it's |
7952 | // always profitable to do this xform. | ||||
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7953 | Value *TryRes = EvaluateInDifferentType(SrcI, DestTy, false); |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7954 | APInt Mask(APInt::getBitsSet(DestBitSize, SrcBitSize, DestBitSize)); |
7955 | if (MaskedValueIsZero(TryRes, Mask)) | ||||
7956 | return ReplaceInstUsesWith(CI, TryRes); | ||||
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7957 | |
7958 | if (Instruction *TryI = dyn_cast<Instruction>(TryRes)) | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7959 | if (TryI->use_empty()) |
7960 | EraseInstFromFunction(*TryI); | ||||
7961 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7962 | break; |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7963 | } |
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7964 | case Instruction::SExt: { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7965 | DoXForm = NumCastsRemoved >= 2; |
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7966 | if (!DoXForm && !isa<TruncInst>(SrcI) && 0) { |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7967 | // If we do not have to emit the truncate + sext pair, then it's always |
7968 | // profitable to do this xform. | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7969 | // |
7970 | // It's not safe to eliminate the trunc + sext pair if one of the | ||||
7971 | // eliminated cast is a truncate. e.g. | ||||
7972 | // t2 = trunc i32 t1 to i16 | ||||
7973 | // t3 = sext i16 t2 to i32 | ||||
7974 | // != | ||||
7975 | // i32 t1 | ||||
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7976 | Value *TryRes = EvaluateInDifferentType(SrcI, DestTy, true); |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7977 | unsigned NumSignBits = ComputeNumSignBits(TryRes); |
7978 | if (NumSignBits > (DestBitSize - SrcBitSize)) | ||||
7979 | return ReplaceInstUsesWith(CI, TryRes); | ||||
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7980 | |
7981 | if (Instruction *TryI = dyn_cast<Instruction>(TryRes)) | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7982 | if (TryI->use_empty()) |
7983 | EraseInstFromFunction(*TryI); | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7984 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7985 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7986 | } |
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7987 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7988 | |
7989 | if (DoXForm) { | ||||
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7990 | DOUT << "ICE: EvaluateInDifferentType converting expression type to avoid" |
7991 | << " cast: " << CI; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7992 | Value *Res = EvaluateInDifferentType(SrcI, DestTy, |
7993 | CI.getOpcode() == Instruction::SExt); | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7994 | if (JustReplace) |
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7995 | // Just replace this cast with the result. |
7996 | return ReplaceInstUsesWith(CI, Res); | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7997 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7998 | assert(Res->getType() == DestTy); |
7999 | switch (CI.getOpcode()) { | ||||
8000 | default: assert(0 && "Unknown cast type!"); | ||||
8001 | case Instruction::Trunc: | ||||
8002 | case Instruction::BitCast: | ||||
8003 | // Just replace this cast with the result. | ||||
8004 | return ReplaceInstUsesWith(CI, Res); | ||||
8005 | case Instruction::ZExt: { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8006 | assert(SrcBitSize < DestBitSize && "Not a zext?"); |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 8007 | |
8008 | // If the high bits are already zero, just replace this cast with the | ||||
8009 | // result. | ||||
8010 | APInt Mask(APInt::getBitsSet(DestBitSize, SrcBitSize, DestBitSize)); | ||||
8011 | if (MaskedValueIsZero(Res, Mask)) | ||||
8012 | return ReplaceInstUsesWith(CI, Res); | ||||
8013 | |||||
8014 | // We need to emit an AND to clear the high bits. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8015 | Constant *C = ConstantInt::get(APInt::getLowBitsSet(DestBitSize, |
8016 | SrcBitSize)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8017 | return BinaryOperator::CreateAnd(Res, C); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8018 | } |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 8019 | case Instruction::SExt: { |
8020 | // If the high bits are already filled with sign bit, just replace this | ||||
8021 | // cast with the result. | ||||
8022 | unsigned NumSignBits = ComputeNumSignBits(Res); | ||||
8023 | if (NumSignBits > (DestBitSize - SrcBitSize)) | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 8024 | return ReplaceInstUsesWith(CI, Res); |
8025 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8026 | // We need to emit a cast to truncate, then a cast to sext. |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8027 | return CastInst::Create(Instruction::SExt, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8028 | InsertCastBefore(Instruction::Trunc, Res, Src->getType(), |
8029 | CI), DestTy); | ||||
8030 | } | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 8031 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8032 | } |
8033 | } | ||||
8034 | |||||
8035 | Value *Op0 = SrcI->getNumOperands() > 0 ? SrcI->getOperand(0) : 0; | ||||
8036 | Value *Op1 = SrcI->getNumOperands() > 1 ? SrcI->getOperand(1) : 0; | ||||
8037 | |||||
8038 | switch (SrcI->getOpcode()) { | ||||
8039 | case Instruction::Add: | ||||
8040 | case Instruction::Mul: | ||||
8041 | case Instruction::And: | ||||
8042 | case Instruction::Or: | ||||
8043 | case Instruction::Xor: | ||||
8044 | // If we are discarding information, rewrite. | ||||
8045 | if (DestBitSize <= SrcBitSize && DestBitSize != 1) { | ||||
8046 | // Don't insert two casts if they cannot be eliminated. We allow | ||||
8047 | // two casts to be inserted if the sizes are the same. This could | ||||
8048 | // only be converting signedness, which is a noop. | ||||
8049 | if (DestBitSize == SrcBitSize || | ||||
8050 | !ValueRequiresCast(CI.getOpcode(), Op1, DestTy,TD) || | ||||
8051 | !ValueRequiresCast(CI.getOpcode(), Op0, DestTy, TD)) { | ||||
8052 | Instruction::CastOps opcode = CI.getOpcode(); | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 8053 | Value *Op0c = InsertCastBefore(opcode, Op0, DestTy, *SrcI); |
8054 | Value *Op1c = InsertCastBefore(opcode, Op1, DestTy, *SrcI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8055 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8056 | cast<BinaryOperator>(SrcI)->getOpcode(), Op0c, Op1c); |
8057 | } | ||||
8058 | } | ||||
8059 | |||||
8060 | // cast (xor bool X, true) to int --> xor (cast bool X to int), 1 | ||||
8061 | if (isa<ZExtInst>(CI) && SrcBitSize == 1 && | ||||
8062 | SrcI->getOpcode() == Instruction::Xor && | ||||
8063 | Op1 == ConstantInt::getTrue() && | ||||
8064 | (!Op0->hasOneUse() || !isa<CmpInst>(Op0))) { | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 8065 | Value *New = InsertCastBefore(Instruction::ZExt, Op0, DestTy, CI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8066 | return BinaryOperator::CreateXor(New, ConstantInt::get(CI.getType(), 1)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8067 | } |
8068 | break; | ||||
8069 | case Instruction::SDiv: | ||||
8070 | case Instruction::UDiv: | ||||
8071 | case Instruction::SRem: | ||||
8072 | case Instruction::URem: | ||||
8073 | // If we are just changing the sign, rewrite. | ||||
8074 | if (DestBitSize == SrcBitSize) { | ||||
8075 | // Don't insert two casts if they cannot be eliminated. We allow | ||||
8076 | // two casts to be inserted if the sizes are the same. This could | ||||
8077 | // only be converting signedness, which is a noop. | ||||
8078 | if (!ValueRequiresCast(CI.getOpcode(), Op1, DestTy, TD) || | ||||
8079 | !ValueRequiresCast(CI.getOpcode(), Op0, DestTy, TD)) { | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 8080 | Value *Op0c = InsertCastBefore(Instruction::BitCast, |
8081 | Op0, DestTy, *SrcI); | ||||
8082 | Value *Op1c = InsertCastBefore(Instruction::BitCast, | ||||
8083 | Op1, DestTy, *SrcI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8084 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8085 | cast<BinaryOperator>(SrcI)->getOpcode(), Op0c, Op1c); |
8086 | } | ||||
8087 | } | ||||
8088 | break; | ||||
8089 | |||||
8090 | case Instruction::Shl: | ||||
8091 | // Allow changing the sign of the source operand. Do not allow | ||||
8092 | // changing the size of the shift, UNLESS the shift amount is a | ||||
8093 | // constant. We must not change variable sized shifts to a smaller | ||||
8094 | // size, because it is undefined to shift more bits out than exist | ||||
8095 | // in the value. | ||||
8096 | if (DestBitSize == SrcBitSize || | ||||
8097 | (DestBitSize < SrcBitSize && isa<Constant>(Op1))) { | ||||
8098 | Instruction::CastOps opcode = (DestBitSize == SrcBitSize ? | ||||
8099 | Instruction::BitCast : Instruction::Trunc); | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 8100 | Value *Op0c = InsertCastBefore(opcode, Op0, DestTy, *SrcI); |
8101 | Value *Op1c = InsertCastBefore(opcode, Op1, DestTy, *SrcI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8102 | return BinaryOperator::CreateShl(Op0c, Op1c); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8103 | } |
8104 | break; | ||||
8105 | case Instruction::AShr: | ||||
8106 | // If this is a signed shr, and if all bits shifted in are about to be | ||||
8107 | // truncated off, turn it into an unsigned shr to allow greater | ||||
8108 | // simplifications. | ||||
8109 | if (DestBitSize < SrcBitSize && | ||||
8110 | isa<ConstantInt>(Op1)) { | ||||
8111 | uint32_t ShiftAmt = cast<ConstantInt>(Op1)->getLimitedValue(SrcBitSize); | ||||
8112 | if (SrcBitSize > ShiftAmt && SrcBitSize-ShiftAmt >= DestBitSize) { | ||||
8113 | // Insert the new logical shift right. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8114 | return BinaryOperator::CreateLShr(Op0, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8115 | } |
8116 | } | ||||
8117 | break; | ||||
8118 | } | ||||
8119 | return 0; | ||||
8120 | } | ||||
8121 | |||||
8122 | Instruction *InstCombiner::visitTrunc(TruncInst &CI) { | ||||
8123 | if (Instruction *Result = commonIntCastTransforms(CI)) | ||||
8124 | return Result; | ||||
8125 | |||||
8126 | Value *Src = CI.getOperand(0); | ||||
8127 | const Type *Ty = CI.getType(); | ||||
8128 | uint32_t DestBitWidth = Ty->getPrimitiveSizeInBits(); | ||||
8129 | uint32_t SrcBitWidth = cast<IntegerType>(Src->getType())->getBitWidth(); | ||||
8130 | |||||
8131 | if (Instruction *SrcI = dyn_cast<Instruction>(Src)) { | ||||
8132 | switch (SrcI->getOpcode()) { | ||||
8133 | default: break; | ||||
8134 | case Instruction::LShr: | ||||
8135 | // We can shrink lshr to something smaller if we know the bits shifted in | ||||
8136 | // are already zeros. | ||||
8137 | if (ConstantInt *ShAmtV = dyn_cast<ConstantInt>(SrcI->getOperand(1))) { | ||||
8138 | uint32_t ShAmt = ShAmtV->getLimitedValue(SrcBitWidth); | ||||
8139 | |||||
8140 | // Get a mask for the bits shifting in. | ||||
8141 | APInt Mask(APInt::getLowBitsSet(SrcBitWidth, ShAmt).shl(DestBitWidth)); | ||||
8142 | Value* SrcIOp0 = SrcI->getOperand(0); | ||||
8143 | if (SrcI->hasOneUse() && MaskedValueIsZero(SrcIOp0, Mask)) { | ||||
8144 | if (ShAmt >= DestBitWidth) // All zeros. | ||||
8145 | return ReplaceInstUsesWith(CI, Constant::getNullValue(Ty)); | ||||
8146 | |||||
8147 | // Okay, we can shrink this. Truncate the input, then return a new | ||||
8148 | // shift. | ||||
8149 | Value *V1 = InsertCastBefore(Instruction::Trunc, SrcIOp0, Ty, CI); | ||||
8150 | Value *V2 = InsertCastBefore(Instruction::Trunc, SrcI->getOperand(1), | ||||
8151 | Ty, CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8152 | return BinaryOperator::CreateLShr(V1, V2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8153 | } |
8154 | } else { // This is a variable shr. | ||||
8155 | |||||
8156 | // Turn 'trunc (lshr X, Y) to bool' into '(X & (1 << Y)) != 0'. This is | ||||
8157 | // more LLVM instructions, but allows '1 << Y' to be hoisted if | ||||
8158 | // loop-invariant and CSE'd. | ||||
8159 | if (CI.getType() == Type::Int1Ty && SrcI->hasOneUse()) { | ||||
8160 | Value *One = ConstantInt::get(SrcI->getType(), 1); | ||||
8161 | |||||
8162 | Value *V = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8163 | BinaryOperator::CreateShl(One, SrcI->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8164 | "tmp"), CI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8165 | V = InsertNewInstBefore(BinaryOperator::CreateAnd(V, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8166 | SrcI->getOperand(0), |
8167 | "tmp"), CI); | ||||
8168 | Value *Zero = Constant::getNullValue(V->getType()); | ||||
8169 | return new ICmpInst(ICmpInst::ICMP_NE, V, Zero); | ||||
8170 | } | ||||
8171 | } | ||||
8172 | break; | ||||
8173 | } | ||||
8174 | } | ||||
8175 | |||||
8176 | return 0; | ||||
8177 | } | ||||
8178 | |||||
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8179 | /// transformZExtICmp - Transform (zext icmp) to bitwise / integer operations |
8180 | /// in order to eliminate the icmp. | ||||
8181 | Instruction *InstCombiner::transformZExtICmp(ICmpInst *ICI, Instruction &CI, | ||||
8182 | bool DoXform) { | ||||
8183 | // If we are just checking for a icmp eq of a single bit and zext'ing it | ||||
8184 | // to an integer, then shift the bit to the appropriate place and then | ||||
8185 | // cast to integer to avoid the comparison. | ||||
8186 | if (ConstantInt *Op1C = dyn_cast<ConstantInt>(ICI->getOperand(1))) { | ||||
8187 | const APInt &Op1CV = Op1C->getValue(); | ||||
8188 | |||||
8189 | // zext (x <s 0) to i32 --> x>>u31 true if signbit set. | ||||
8190 | // zext (x >s -1) to i32 --> (x>>u31)^1 true if signbit clear. | ||||
8191 | if ((ICI->getPredicate() == ICmpInst::ICMP_SLT && Op1CV == 0) || | ||||
8192 | (ICI->getPredicate() == ICmpInst::ICMP_SGT &&Op1CV.isAllOnesValue())) { | ||||
8193 | if (!DoXform) return ICI; | ||||
8194 | |||||
8195 | Value *In = ICI->getOperand(0); | ||||
8196 | Value *Sh = ConstantInt::get(In->getType(), | ||||
8197 | In->getType()->getPrimitiveSizeInBits()-1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8198 | In = InsertNewInstBefore(BinaryOperator::CreateLShr(In, Sh, |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8199 | In->getName()+".lobit"), |
8200 | CI); | ||||
8201 | if (In->getType() != CI.getType()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8202 | In = CastInst::CreateIntegerCast(In, CI.getType(), |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8203 | false/*ZExt*/, "tmp", &CI); |
8204 | |||||
8205 | if (ICI->getPredicate() == ICmpInst::ICMP_SGT) { | ||||
8206 | Constant *One = ConstantInt::get(In->getType(), 1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8207 | In = InsertNewInstBefore(BinaryOperator::CreateXor(In, One, |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8208 | In->getName()+".not"), |
8209 | CI); | ||||
8210 | } | ||||
8211 | |||||
8212 | return ReplaceInstUsesWith(CI, In); | ||||
8213 | } | ||||
8214 | |||||
8215 | |||||
8216 | |||||
8217 | // zext (X == 0) to i32 --> X^1 iff X has only the low bit set. | ||||
8218 | // zext (X == 0) to i32 --> (X>>1)^1 iff X has only the 2nd bit set. | ||||
8219 | // zext (X == 1) to i32 --> X iff X has only the low bit set. | ||||
8220 | // zext (X == 2) to i32 --> X>>1 iff X has only the 2nd bit set. | ||||
8221 | // zext (X != 0) to i32 --> X iff X has only the low bit set. | ||||
8222 | // zext (X != 0) to i32 --> X>>1 iff X has only the 2nd bit set. | ||||
8223 | // zext (X != 1) to i32 --> X^1 iff X has only the low bit set. | ||||
8224 | // zext (X != 2) to i32 --> (X>>1)^1 iff X has only the 2nd bit set. | ||||
8225 | if ((Op1CV == 0 || Op1CV.isPowerOf2()) && | ||||
8226 | // This only works for EQ and NE | ||||
8227 | ICI->isEquality()) { | ||||
8228 | // If Op1C some other power of two, convert: | ||||
8229 | uint32_t BitWidth = Op1C->getType()->getBitWidth(); | ||||
8230 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
8231 | APInt TypeMask(APInt::getAllOnesValue(BitWidth)); | ||||
8232 | ComputeMaskedBits(ICI->getOperand(0), TypeMask, KnownZero, KnownOne); | ||||
8233 | |||||
8234 | APInt KnownZeroMask(~KnownZero); | ||||
8235 | if (KnownZeroMask.isPowerOf2()) { // Exactly 1 possible 1? | ||||
8236 | if (!DoXform) return ICI; | ||||
8237 | |||||
8238 | bool isNE = ICI->getPredicate() == ICmpInst::ICMP_NE; | ||||
8239 | if (Op1CV != 0 && (Op1CV != KnownZeroMask)) { | ||||
8240 | // (X&4) == 2 --> false | ||||
8241 | // (X&4) != 2 --> true | ||||
8242 | Constant *Res = ConstantInt::get(Type::Int1Ty, isNE); | ||||
8243 | Res = ConstantExpr::getZExt(Res, CI.getType()); | ||||
8244 | return ReplaceInstUsesWith(CI, Res); | ||||
8245 | } | ||||
8246 | |||||
8247 | uint32_t ShiftAmt = KnownZeroMask.logBase2(); | ||||
8248 | Value *In = ICI->getOperand(0); | ||||
8249 | if (ShiftAmt) { | ||||
8250 | // Perform a logical shr by shiftamt. | ||||
8251 | // Insert the shift to put the result in the low bit. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8252 | In = InsertNewInstBefore(BinaryOperator::CreateLShr(In, |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8253 | ConstantInt::get(In->getType(), ShiftAmt), |
8254 | In->getName()+".lobit"), CI); | ||||
8255 | } | ||||
8256 | |||||
8257 | if ((Op1CV != 0) == isNE) { // Toggle the low bit. | ||||
8258 | Constant *One = ConstantInt::get(In->getType(), 1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8259 | In = BinaryOperator::CreateXor(In, One, "tmp"); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8260 | InsertNewInstBefore(cast<Instruction>(In), CI); |
8261 | } | ||||
8262 | |||||
8263 | if (CI.getType() == In->getType()) | ||||
8264 | return ReplaceInstUsesWith(CI, In); | ||||
8265 | else | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8266 | return CastInst::CreateIntegerCast(In, CI.getType(), false/*ZExt*/); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8267 | } |
8268 | } | ||||
8269 | } | ||||
8270 | |||||
8271 | return 0; | ||||
8272 | } | ||||
8273 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8274 | Instruction *InstCombiner::visitZExt(ZExtInst &CI) { |
8275 | // If one of the common conversion will work .. | ||||
8276 | if (Instruction *Result = commonIntCastTransforms(CI)) | ||||
8277 | return Result; | ||||
8278 | |||||
8279 | Value *Src = CI.getOperand(0); | ||||
8280 | |||||
Chris Lattner | 215d56e | 2009-02-17 20:47:23 +0000 | [diff] [blame] | 8281 | // If this is a TRUNC followed by a ZEXT then we are dealing with integral |
8282 | // types and if the sizes are just right we can convert this into a logical | ||||
8283 | // 'and' which will be much cheaper than the pair of casts. | ||||
8284 | if (TruncInst *CSrc = dyn_cast<TruncInst>(Src)) { // A->B->C cast | ||||
8285 | // Get the sizes of the types involved. We know that the intermediate type | ||||
8286 | // will be smaller than A or C, but don't know the relation between A and C. | ||||
8287 | Value *A = CSrc->getOperand(0); | ||||
8288 | unsigned SrcSize = A->getType()->getPrimitiveSizeInBits(); | ||||
8289 | unsigned MidSize = CSrc->getType()->getPrimitiveSizeInBits(); | ||||
8290 | unsigned DstSize = CI.getType()->getPrimitiveSizeInBits(); | ||||
8291 | // If we're actually extending zero bits, then if | ||||
8292 | // SrcSize < DstSize: zext(a & mask) | ||||
8293 | // SrcSize == DstSize: a & mask | ||||
8294 | // SrcSize > DstSize: trunc(a) & mask | ||||
8295 | if (SrcSize < DstSize) { | ||||
8296 | APInt AndValue(APInt::getLowBitsSet(SrcSize, MidSize)); | ||||
8297 | Constant *AndConst = ConstantInt::get(AndValue); | ||||
8298 | Instruction *And = | ||||
8299 | BinaryOperator::CreateAnd(A, AndConst, CSrc->getName()+".mask"); | ||||
8300 | InsertNewInstBefore(And, CI); | ||||
8301 | return new ZExtInst(And, CI.getType()); | ||||
8302 | } else if (SrcSize == DstSize) { | ||||
8303 | APInt AndValue(APInt::getLowBitsSet(SrcSize, MidSize)); | ||||
8304 | return BinaryOperator::CreateAnd(A, ConstantInt::get(AndValue)); | ||||
8305 | } else if (SrcSize > DstSize) { | ||||
8306 | Instruction *Trunc = new TruncInst(A, CI.getType(), "tmp"); | ||||
8307 | InsertNewInstBefore(Trunc, CI); | ||||
8308 | APInt AndValue(APInt::getLowBitsSet(DstSize, MidSize)); | ||||
8309 | return BinaryOperator::CreateAnd(Trunc, ConstantInt::get(AndValue)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8310 | } |
8311 | } | ||||
8312 | |||||
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8313 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(Src)) |
8314 | return transformZExtICmp(ICI, CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8315 | |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8316 | BinaryOperator *SrcI = dyn_cast<BinaryOperator>(Src); |
8317 | if (SrcI && SrcI->getOpcode() == Instruction::Or) { | ||||
8318 | // zext (or icmp, icmp) --> or (zext icmp), (zext icmp) if at least one | ||||
8319 | // of the (zext icmp) will be transformed. | ||||
8320 | ICmpInst *LHS = dyn_cast<ICmpInst>(SrcI->getOperand(0)); | ||||
8321 | ICmpInst *RHS = dyn_cast<ICmpInst>(SrcI->getOperand(1)); | ||||
8322 | if (LHS && RHS && LHS->hasOneUse() && RHS->hasOneUse() && | ||||
8323 | (transformZExtICmp(LHS, CI, false) || | ||||
8324 | transformZExtICmp(RHS, CI, false))) { | ||||
8325 | Value *LCast = InsertCastBefore(Instruction::ZExt, LHS, CI.getType(), CI); | ||||
8326 | Value *RCast = InsertCastBefore(Instruction::ZExt, RHS, CI.getType(), CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8327 | return BinaryOperator::Create(Instruction::Or, LCast, RCast); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8328 | } |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8329 | } |
8330 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8331 | return 0; |
8332 | } | ||||
8333 | |||||
8334 | Instruction *InstCombiner::visitSExt(SExtInst &CI) { | ||||
8335 | if (Instruction *I = commonIntCastTransforms(CI)) | ||||
8336 | return I; | ||||
8337 | |||||
8338 | Value *Src = CI.getOperand(0); | ||||
8339 | |||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8340 | // Canonicalize sign-extend from i1 to a select. |
8341 | if (Src->getType() == Type::Int1Ty) | ||||
8342 | return SelectInst::Create(Src, | ||||
8343 | ConstantInt::getAllOnesValue(CI.getType()), | ||||
8344 | Constant::getNullValue(CI.getType())); | ||||
Dan Gohman | f0f1202 | 2008-05-20 21:01:12 +0000 | [diff] [blame] | 8345 | |
8346 | // See if the value being truncated is already sign extended. If so, just | ||||
8347 | // eliminate the trunc/sext pair. | ||||
8348 | if (getOpcode(Src) == Instruction::Trunc) { | ||||
8349 | Value *Op = cast<User>(Src)->getOperand(0); | ||||
8350 | unsigned OpBits = cast<IntegerType>(Op->getType())->getBitWidth(); | ||||
8351 | unsigned MidBits = cast<IntegerType>(Src->getType())->getBitWidth(); | ||||
8352 | unsigned DestBits = cast<IntegerType>(CI.getType())->getBitWidth(); | ||||
8353 | unsigned NumSignBits = ComputeNumSignBits(Op); | ||||
8354 | |||||
8355 | if (OpBits == DestBits) { | ||||
8356 | // Op is i32, Mid is i8, and Dest is i32. If Op has more than 24 sign | ||||
8357 | // bits, it is already ready. | ||||
8358 | if (NumSignBits > DestBits-MidBits) | ||||
8359 | return ReplaceInstUsesWith(CI, Op); | ||||
8360 | } else if (OpBits < DestBits) { | ||||
8361 | // Op is i32, Mid is i8, and Dest is i64. If Op has more than 24 sign | ||||
8362 | // bits, just sext from i32. | ||||
8363 | if (NumSignBits > OpBits-MidBits) | ||||
8364 | return new SExtInst(Op, CI.getType(), "tmp"); | ||||
8365 | } else { | ||||
8366 | // Op is i64, Mid is i8, and Dest is i32. If Op has more than 56 sign | ||||
8367 | // bits, just truncate to i32. | ||||
8368 | if (NumSignBits > OpBits-MidBits) | ||||
8369 | return new TruncInst(Op, CI.getType(), "tmp"); | ||||
8370 | } | ||||
8371 | } | ||||
Chris Lattner | 8a2d059 | 2008-08-06 07:35:52 +0000 | [diff] [blame] | 8372 | |
8373 | // If the input is a shl/ashr pair of a same constant, then this is a sign | ||||
8374 | // extension from a smaller value. If we could trust arbitrary bitwidth | ||||
8375 | // integers, we could turn this into a truncate to the smaller bit and then | ||||
8376 | // use a sext for the whole extension. Since we don't, look deeper and check | ||||
8377 | // for a truncate. If the source and dest are the same type, eliminate the | ||||
8378 | // trunc and extend and just do shifts. For example, turn: | ||||
8379 | // %a = trunc i32 %i to i8 | ||||
8380 | // %b = shl i8 %a, 6 | ||||
8381 | // %c = ashr i8 %b, 6 | ||||
8382 | // %d = sext i8 %c to i32 | ||||
8383 | // into: | ||||
8384 | // %a = shl i32 %i, 30 | ||||
8385 | // %d = ashr i32 %a, 30 | ||||
8386 | Value *A = 0; | ||||
8387 | ConstantInt *BA = 0, *CA = 0; | ||||
8388 | if (match(Src, m_AShr(m_Shl(m_Value(A), m_ConstantInt(BA)), | ||||
8389 | m_ConstantInt(CA))) && | ||||
8390 | BA == CA && isa<TruncInst>(A)) { | ||||
8391 | Value *I = cast<TruncInst>(A)->getOperand(0); | ||||
8392 | if (I->getType() == CI.getType()) { | ||||
8393 | unsigned MidSize = Src->getType()->getPrimitiveSizeInBits(); | ||||
8394 | unsigned SrcDstSize = CI.getType()->getPrimitiveSizeInBits(); | ||||
8395 | unsigned ShAmt = CA->getZExtValue()+SrcDstSize-MidSize; | ||||
8396 | Constant *ShAmtV = ConstantInt::get(CI.getType(), ShAmt); | ||||
8397 | I = InsertNewInstBefore(BinaryOperator::CreateShl(I, ShAmtV, | ||||
8398 | CI.getName()), CI); | ||||
8399 | return BinaryOperator::CreateAShr(I, ShAmtV); | ||||
8400 | } | ||||
8401 | } | ||||
8402 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8403 | return 0; |
8404 | } | ||||
8405 | |||||
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8406 | /// FitsInFPType - Return a Constant* for the specified FP constant if it fits |
8407 | /// in the specified FP type without changing its value. | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 8408 | static Constant *FitsInFPType(ConstantFP *CFP, const fltSemantics &Sem) { |
Dale Johannesen | 6e547b4 | 2008-10-09 23:00:39 +0000 | [diff] [blame] | 8409 | bool losesInfo; |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8410 | APFloat F = CFP->getValueAPF(); |
Dale Johannesen | 6e547b4 | 2008-10-09 23:00:39 +0000 | [diff] [blame] | 8411 | (void)F.convert(Sem, APFloat::rmNearestTiesToEven, &losesInfo); |
8412 | if (!losesInfo) | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 8413 | return ConstantFP::get(F); |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8414 | return 0; |
8415 | } | ||||
8416 | |||||
8417 | /// LookThroughFPExtensions - If this is an fp extension instruction, look | ||||
8418 | /// through it until we get the source value. | ||||
8419 | static Value *LookThroughFPExtensions(Value *V) { | ||||
8420 | if (Instruction *I = dyn_cast<Instruction>(V)) | ||||
8421 | if (I->getOpcode() == Instruction::FPExt) | ||||
8422 | return LookThroughFPExtensions(I->getOperand(0)); | ||||
8423 | |||||
8424 | // If this value is a constant, return the constant in the smallest FP type | ||||
8425 | // that can accurately represent it. This allows us to turn | ||||
8426 | // (float)((double)X+2.0) into x+2.0f. | ||||
8427 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(V)) { | ||||
8428 | if (CFP->getType() == Type::PPC_FP128Ty) | ||||
8429 | return V; // No constant folding of this. | ||||
8430 | // See if the value can be truncated to float and then reextended. | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 8431 | if (Value *V = FitsInFPType(CFP, APFloat::IEEEsingle)) |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8432 | return V; |
8433 | if (CFP->getType() == Type::DoubleTy) | ||||
8434 | return V; // Won't shrink. | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 8435 | if (Value *V = FitsInFPType(CFP, APFloat::IEEEdouble)) |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8436 | return V; |
8437 | // Don't try to shrink to various long double types. | ||||
8438 | } | ||||
8439 | |||||
8440 | return V; | ||||
8441 | } | ||||
8442 | |||||
8443 | Instruction *InstCombiner::visitFPTrunc(FPTruncInst &CI) { | ||||
8444 | if (Instruction *I = commonCastTransforms(CI)) | ||||
8445 | return I; | ||||
8446 | |||||
8447 | // If we have fptrunc(add (fpextend x), (fpextend y)), where x and y are | ||||
8448 | // smaller than the destination type, we can eliminate the truncate by doing | ||||
8449 | // the add as the smaller type. This applies to add/sub/mul/div as well as | ||||
8450 | // many builtins (sqrt, etc). | ||||
8451 | BinaryOperator *OpI = dyn_cast<BinaryOperator>(CI.getOperand(0)); | ||||
8452 | if (OpI && OpI->hasOneUse()) { | ||||
8453 | switch (OpI->getOpcode()) { | ||||
8454 | default: break; | ||||
8455 | case Instruction::Add: | ||||
8456 | case Instruction::Sub: | ||||
8457 | case Instruction::Mul: | ||||
8458 | case Instruction::FDiv: | ||||
8459 | case Instruction::FRem: | ||||
8460 | const Type *SrcTy = OpI->getType(); | ||||
8461 | Value *LHSTrunc = LookThroughFPExtensions(OpI->getOperand(0)); | ||||
8462 | Value *RHSTrunc = LookThroughFPExtensions(OpI->getOperand(1)); | ||||
8463 | if (LHSTrunc->getType() != SrcTy && | ||||
8464 | RHSTrunc->getType() != SrcTy) { | ||||
8465 | unsigned DstSize = CI.getType()->getPrimitiveSizeInBits(); | ||||
8466 | // If the source types were both smaller than the destination type of | ||||
8467 | // the cast, do this xform. | ||||
8468 | if (LHSTrunc->getType()->getPrimitiveSizeInBits() <= DstSize && | ||||
8469 | RHSTrunc->getType()->getPrimitiveSizeInBits() <= DstSize) { | ||||
8470 | LHSTrunc = InsertCastBefore(Instruction::FPExt, LHSTrunc, | ||||
8471 | CI.getType(), CI); | ||||
8472 | RHSTrunc = InsertCastBefore(Instruction::FPExt, RHSTrunc, | ||||
8473 | CI.getType(), CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8474 | return BinaryOperator::Create(OpI->getOpcode(), LHSTrunc, RHSTrunc); |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8475 | } |
8476 | } | ||||
8477 | break; | ||||
8478 | } | ||||
8479 | } | ||||
8480 | return 0; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8481 | } |
8482 | |||||
8483 | Instruction *InstCombiner::visitFPExt(CastInst &CI) { | ||||
8484 | return commonCastTransforms(CI); | ||||
8485 | } | ||||
8486 | |||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8487 | Instruction *InstCombiner::visitFPToUI(FPToUIInst &FI) { |
Chris Lattner | 5f4d691 | 2008-08-06 05:13:06 +0000 | [diff] [blame] | 8488 | Instruction *OpI = dyn_cast<Instruction>(FI.getOperand(0)); |
8489 | if (OpI == 0) | ||||
8490 | return commonCastTransforms(FI); | ||||
8491 | |||||
8492 | // fptoui(uitofp(X)) --> X | ||||
8493 | // fptoui(sitofp(X)) --> X | ||||
8494 | // This is safe if the intermediate type has enough bits in its mantissa to | ||||
8495 | // accurately represent all values of X. For example, do not do this with | ||||
8496 | // i64->float->i64. This is also safe for sitofp case, because any negative | ||||
8497 | // 'X' value would cause an undefined result for the fptoui. | ||||
8498 | if ((isa<UIToFPInst>(OpI) || isa<SIToFPInst>(OpI)) && | ||||
8499 | OpI->getOperand(0)->getType() == FI.getType() && | ||||
8500 | (int)FI.getType()->getPrimitiveSizeInBits() < /*extra bit for sign */ | ||||
8501 | OpI->getType()->getFPMantissaWidth()) | ||||
8502 | return ReplaceInstUsesWith(FI, OpI->getOperand(0)); | ||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8503 | |
8504 | return commonCastTransforms(FI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8505 | } |
8506 | |||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8507 | Instruction *InstCombiner::visitFPToSI(FPToSIInst &FI) { |
Chris Lattner | 5f4d691 | 2008-08-06 05:13:06 +0000 | [diff] [blame] | 8508 | Instruction *OpI = dyn_cast<Instruction>(FI.getOperand(0)); |
8509 | if (OpI == 0) | ||||
8510 | return commonCastTransforms(FI); | ||||
8511 | |||||
8512 | // fptosi(sitofp(X)) --> X | ||||
8513 | // fptosi(uitofp(X)) --> X | ||||
8514 | // This is safe if the intermediate type has enough bits in its mantissa to | ||||
8515 | // accurately represent all values of X. For example, do not do this with | ||||
8516 | // i64->float->i64. This is also safe for sitofp case, because any negative | ||||
8517 | // 'X' value would cause an undefined result for the fptoui. | ||||
8518 | if ((isa<UIToFPInst>(OpI) || isa<SIToFPInst>(OpI)) && | ||||
8519 | OpI->getOperand(0)->getType() == FI.getType() && | ||||
8520 | (int)FI.getType()->getPrimitiveSizeInBits() <= | ||||
8521 | OpI->getType()->getFPMantissaWidth()) | ||||
8522 | return ReplaceInstUsesWith(FI, OpI->getOperand(0)); | ||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8523 | |
8524 | return commonCastTransforms(FI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8525 | } |
8526 | |||||
8527 | Instruction *InstCombiner::visitUIToFP(CastInst &CI) { | ||||
8528 | return commonCastTransforms(CI); | ||||
8529 | } | ||||
8530 | |||||
8531 | Instruction *InstCombiner::visitSIToFP(CastInst &CI) { | ||||
8532 | return commonCastTransforms(CI); | ||||
8533 | } | ||||
8534 | |||||
8535 | Instruction *InstCombiner::visitPtrToInt(CastInst &CI) { | ||||
8536 | return commonPointerCastTransforms(CI); | ||||
8537 | } | ||||
8538 | |||||
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8539 | Instruction *InstCombiner::visitIntToPtr(IntToPtrInst &CI) { |
8540 | if (Instruction *I = commonCastTransforms(CI)) | ||||
8541 | return I; | ||||
8542 | |||||
8543 | const Type *DestPointee = cast<PointerType>(CI.getType())->getElementType(); | ||||
8544 | if (!DestPointee->isSized()) return 0; | ||||
8545 | |||||
8546 | // If this is inttoptr(add (ptrtoint x), cst), try to turn this into a GEP. | ||||
8547 | ConstantInt *Cst; | ||||
8548 | Value *X; | ||||
8549 | if (match(CI.getOperand(0), m_Add(m_Cast<PtrToIntInst>(m_Value(X)), | ||||
8550 | m_ConstantInt(Cst)))) { | ||||
8551 | // If the source and destination operands have the same type, see if this | ||||
8552 | // is a single-index GEP. | ||||
8553 | if (X->getType() == CI.getType()) { | ||||
8554 | // Get the size of the pointee type. | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 8555 | uint64_t Size = TD->getTypePaddedSize(DestPointee); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8556 | |
8557 | // Convert the constant to intptr type. | ||||
8558 | APInt Offset = Cst->getValue(); | ||||
8559 | Offset.sextOrTrunc(TD->getPointerSizeInBits()); | ||||
8560 | |||||
8561 | // If Offset is evenly divisible by Size, we can do this xform. | ||||
8562 | if (Size && !APIntOps::srem(Offset, APInt(Offset.getBitWidth(), Size))){ | ||||
8563 | Offset = APIntOps::sdiv(Offset, APInt(Offset.getBitWidth(), Size)); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8564 | return GetElementPtrInst::Create(X, ConstantInt::get(Offset)); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8565 | } |
8566 | } | ||||
8567 | // TODO: Could handle other cases, e.g. where add is indexing into field of | ||||
8568 | // struct etc. | ||||
8569 | } else if (CI.getOperand(0)->hasOneUse() && | ||||
8570 | match(CI.getOperand(0), m_Add(m_Value(X), m_ConstantInt(Cst)))) { | ||||
8571 | // Otherwise, if this is inttoptr(add x, cst), try to turn this into an | ||||
8572 | // "inttoptr+GEP" instead of "add+intptr". | ||||
8573 | |||||
8574 | // Get the size of the pointee type. | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 8575 | uint64_t Size = TD->getTypePaddedSize(DestPointee); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8576 | |
8577 | // Convert the constant to intptr type. | ||||
8578 | APInt Offset = Cst->getValue(); | ||||
8579 | Offset.sextOrTrunc(TD->getPointerSizeInBits()); | ||||
8580 | |||||
8581 | // If Offset is evenly divisible by Size, we can do this xform. | ||||
8582 | if (Size && !APIntOps::srem(Offset, APInt(Offset.getBitWidth(), Size))){ | ||||
8583 | Offset = APIntOps::sdiv(Offset, APInt(Offset.getBitWidth(), Size)); | ||||
8584 | |||||
8585 | Instruction *P = InsertNewInstBefore(new IntToPtrInst(X, CI.getType(), | ||||
8586 | "tmp"), CI); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8587 | return GetElementPtrInst::Create(P, ConstantInt::get(Offset), "tmp"); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8588 | } |
8589 | } | ||||
8590 | return 0; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8591 | } |
8592 | |||||
8593 | Instruction *InstCombiner::visitBitCast(BitCastInst &CI) { | ||||
8594 | // If the operands are integer typed then apply the integer transforms, | ||||
8595 | // otherwise just apply the common ones. | ||||
8596 | Value *Src = CI.getOperand(0); | ||||
8597 | const Type *SrcTy = Src->getType(); | ||||
8598 | const Type *DestTy = CI.getType(); | ||||
8599 | |||||
8600 | if (SrcTy->isInteger() && DestTy->isInteger()) { | ||||
8601 | if (Instruction *Result = commonIntCastTransforms(CI)) | ||||
8602 | return Result; | ||||
8603 | } else if (isa<PointerType>(SrcTy)) { | ||||
8604 | if (Instruction *I = commonPointerCastTransforms(CI)) | ||||
8605 | return I; | ||||
8606 | } else { | ||||
8607 | if (Instruction *Result = commonCastTransforms(CI)) | ||||
8608 | return Result; | ||||
8609 | } | ||||
8610 | |||||
8611 | |||||
8612 | // Get rid of casts from one type to the same type. These are useless and can | ||||
8613 | // be replaced by the operand. | ||||
8614 | if (DestTy == Src->getType()) | ||||
8615 | return ReplaceInstUsesWith(CI, Src); | ||||
8616 | |||||
8617 | if (const PointerType *DstPTy = dyn_cast<PointerType>(DestTy)) { | ||||
8618 | const PointerType *SrcPTy = cast<PointerType>(SrcTy); | ||||
8619 | const Type *DstElTy = DstPTy->getElementType(); | ||||
8620 | const Type *SrcElTy = SrcPTy->getElementType(); | ||||
8621 | |||||
Nate Begeman | df5b361 | 2008-03-31 00:22:16 +0000 | [diff] [blame] | 8622 | // If the address spaces don't match, don't eliminate the bitcast, which is |
8623 | // required for changing types. | ||||
8624 | if (SrcPTy->getAddressSpace() != DstPTy->getAddressSpace()) | ||||
8625 | return 0; | ||||
8626 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8627 | // If we are casting a malloc or alloca to a pointer to a type of the same |
8628 | // size, rewrite the allocation instruction to allocate the "right" type. | ||||
8629 | if (AllocationInst *AI = dyn_cast<AllocationInst>(Src)) | ||||
8630 | if (Instruction *V = PromoteCastOfAllocation(CI, *AI)) | ||||
8631 | return V; | ||||
8632 | |||||
8633 | // If the source and destination are pointers, and this cast is equivalent | ||||
8634 | // to a getelementptr X, 0, 0, 0... turn it into the appropriate gep. | ||||
8635 | // This can enhance SROA and other transforms that want type-safe pointers. | ||||
8636 | Constant *ZeroUInt = Constant::getNullValue(Type::Int32Ty); | ||||
8637 | unsigned NumZeros = 0; | ||||
8638 | while (SrcElTy != DstElTy && | ||||
8639 | isa<CompositeType>(SrcElTy) && !isa<PointerType>(SrcElTy) && | ||||
8640 | SrcElTy->getNumContainedTypes() /* not "{}" */) { | ||||
8641 | SrcElTy = cast<CompositeType>(SrcElTy)->getTypeAtIndex(ZeroUInt); | ||||
8642 | ++NumZeros; | ||||
8643 | } | ||||
8644 | |||||
8645 | // If we found a path from the src to dest, create the getelementptr now. | ||||
8646 | if (SrcElTy == DstElTy) { | ||||
8647 | SmallVector<Value*, 8> Idxs(NumZeros+1, ZeroUInt); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8648 | return GetElementPtrInst::Create(Src, Idxs.begin(), Idxs.end(), "", |
8649 | ((Instruction*) NULL)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8650 | } |
8651 | } | ||||
8652 | |||||
8653 | if (ShuffleVectorInst *SVI = dyn_cast<ShuffleVectorInst>(Src)) { | ||||
8654 | if (SVI->hasOneUse()) { | ||||
8655 | // Okay, we have (bitconvert (shuffle ..)). Check to see if this is | ||||
8656 | // a bitconvert to a vector with the same # elts. | ||||
8657 | if (isa<VectorType>(DestTy) && | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 8658 | cast<VectorType>(DestTy)->getNumElements() == |
8659 | SVI->getType()->getNumElements() && | ||||
8660 | SVI->getType()->getNumElements() == | ||||
8661 | cast<VectorType>(SVI->getOperand(0)->getType())->getNumElements()) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8662 | CastInst *Tmp; |
8663 | // If either of the operands is a cast from CI.getType(), then | ||||
8664 | // evaluating the shuffle in the casted destination's type will allow | ||||
8665 | // us to eliminate at least one cast. | ||||
8666 | if (((Tmp = dyn_cast<CastInst>(SVI->getOperand(0))) && | ||||
8667 | Tmp->getOperand(0)->getType() == DestTy) || | ||||
8668 | ((Tmp = dyn_cast<CastInst>(SVI->getOperand(1))) && | ||||
8669 | Tmp->getOperand(0)->getType() == DestTy)) { | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 8670 | Value *LHS = InsertCastBefore(Instruction::BitCast, |
8671 | SVI->getOperand(0), DestTy, CI); | ||||
8672 | Value *RHS = InsertCastBefore(Instruction::BitCast, | ||||
8673 | SVI->getOperand(1), DestTy, CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8674 | // Return a new shuffle vector. Use the same element ID's, as we |
8675 | // know the vector types match #elts. | ||||
8676 | return new ShuffleVectorInst(LHS, RHS, SVI->getOperand(2)); | ||||
8677 | } | ||||
8678 | } | ||||
8679 | } | ||||
8680 | } | ||||
8681 | return 0; | ||||
8682 | } | ||||
8683 | |||||
8684 | /// GetSelectFoldableOperands - We want to turn code that looks like this: | ||||
8685 | /// %C = or %A, %B | ||||
8686 | /// %D = select %cond, %C, %A | ||||
8687 | /// into: | ||||
8688 | /// %C = select %cond, %B, 0 | ||||
8689 | /// %D = or %A, %C | ||||
8690 | /// | ||||
8691 | /// Assuming that the specified instruction is an operand to the select, return | ||||
8692 | /// a bitmask indicating which operands of this instruction are foldable if they | ||||
8693 | /// equal the other incoming value of the select. | ||||
8694 | /// | ||||
8695 | static unsigned GetSelectFoldableOperands(Instruction *I) { | ||||
8696 | switch (I->getOpcode()) { | ||||
8697 | case Instruction::Add: | ||||
8698 | case Instruction::Mul: | ||||
8699 | case Instruction::And: | ||||
8700 | case Instruction::Or: | ||||
8701 | case Instruction::Xor: | ||||
8702 | return 3; // Can fold through either operand. | ||||
8703 | case Instruction::Sub: // Can only fold on the amount subtracted. | ||||
8704 | case Instruction::Shl: // Can only fold on the shift amount. | ||||
8705 | case Instruction::LShr: | ||||
8706 | case Instruction::AShr: | ||||
8707 | return 1; | ||||
8708 | default: | ||||
8709 | return 0; // Cannot fold | ||||
8710 | } | ||||
8711 | } | ||||
8712 | |||||
8713 | /// GetSelectFoldableConstant - For the same transformation as the previous | ||||
8714 | /// function, return the identity constant that goes into the select. | ||||
8715 | static Constant *GetSelectFoldableConstant(Instruction *I) { | ||||
8716 | switch (I->getOpcode()) { | ||||
8717 | default: assert(0 && "This cannot happen!"); abort(); | ||||
8718 | case Instruction::Add: | ||||
8719 | case Instruction::Sub: | ||||
8720 | case Instruction::Or: | ||||
8721 | case Instruction::Xor: | ||||
8722 | case Instruction::Shl: | ||||
8723 | case Instruction::LShr: | ||||
8724 | case Instruction::AShr: | ||||
8725 | return Constant::getNullValue(I->getType()); | ||||
8726 | case Instruction::And: | ||||
8727 | return Constant::getAllOnesValue(I->getType()); | ||||
8728 | case Instruction::Mul: | ||||
8729 | return ConstantInt::get(I->getType(), 1); | ||||
8730 | } | ||||
8731 | } | ||||
8732 | |||||
8733 | /// FoldSelectOpOp - Here we have (select c, TI, FI), and we know that TI and FI | ||||
8734 | /// have the same opcode and only one use each. Try to simplify this. | ||||
8735 | Instruction *InstCombiner::FoldSelectOpOp(SelectInst &SI, Instruction *TI, | ||||
8736 | Instruction *FI) { | ||||
8737 | if (TI->getNumOperands() == 1) { | ||||
8738 | // If this is a non-volatile load or a cast from the same type, | ||||
8739 | // merge. | ||||
8740 | if (TI->isCast()) { | ||||
8741 | if (TI->getOperand(0)->getType() != FI->getOperand(0)->getType()) | ||||
8742 | return 0; | ||||
8743 | } else { | ||||
8744 | return 0; // unknown unary op. | ||||
8745 | } | ||||
8746 | |||||
8747 | // Fold this by inserting a select from the input values. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8748 | SelectInst *NewSI = SelectInst::Create(SI.getCondition(), TI->getOperand(0), |
8749 | FI->getOperand(0), SI.getName()+".v"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8750 | InsertNewInstBefore(NewSI, SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8751 | return CastInst::Create(Instruction::CastOps(TI->getOpcode()), NewSI, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8752 | TI->getType()); |
8753 | } | ||||
8754 | |||||
8755 | // Only handle binary operators here. | ||||
8756 | if (!isa<BinaryOperator>(TI)) | ||||
8757 | return 0; | ||||
8758 | |||||
8759 | // Figure out if the operations have any operands in common. | ||||
8760 | Value *MatchOp, *OtherOpT, *OtherOpF; | ||||
8761 | bool MatchIsOpZero; | ||||
8762 | if (TI->getOperand(0) == FI->getOperand(0)) { | ||||
8763 | MatchOp = TI->getOperand(0); | ||||
8764 | OtherOpT = TI->getOperand(1); | ||||
8765 | OtherOpF = FI->getOperand(1); | ||||
8766 | MatchIsOpZero = true; | ||||
8767 | } else if (TI->getOperand(1) == FI->getOperand(1)) { | ||||
8768 | MatchOp = TI->getOperand(1); | ||||
8769 | OtherOpT = TI->getOperand(0); | ||||
8770 | OtherOpF = FI->getOperand(0); | ||||
8771 | MatchIsOpZero = false; | ||||
8772 | } else if (!TI->isCommutative()) { | ||||
8773 | return 0; | ||||
8774 | } else if (TI->getOperand(0) == FI->getOperand(1)) { | ||||
8775 | MatchOp = TI->getOperand(0); | ||||
8776 | OtherOpT = TI->getOperand(1); | ||||
8777 | OtherOpF = FI->getOperand(0); | ||||
8778 | MatchIsOpZero = true; | ||||
8779 | } else if (TI->getOperand(1) == FI->getOperand(0)) { | ||||
8780 | MatchOp = TI->getOperand(1); | ||||
8781 | OtherOpT = TI->getOperand(0); | ||||
8782 | OtherOpF = FI->getOperand(1); | ||||
8783 | MatchIsOpZero = true; | ||||
8784 | } else { | ||||
8785 | return 0; | ||||
8786 | } | ||||
8787 | |||||
8788 | // If we reach here, they do have operations in common. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8789 | SelectInst *NewSI = SelectInst::Create(SI.getCondition(), OtherOpT, |
8790 | OtherOpF, SI.getName()+".v"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8791 | InsertNewInstBefore(NewSI, SI); |
8792 | |||||
8793 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(TI)) { | ||||
8794 | if (MatchIsOpZero) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8795 | return BinaryOperator::Create(BO->getOpcode(), MatchOp, NewSI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8796 | else |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8797 | return BinaryOperator::Create(BO->getOpcode(), NewSI, MatchOp); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8798 | } |
8799 | assert(0 && "Shouldn't get here"); | ||||
8800 | return 0; | ||||
8801 | } | ||||
8802 | |||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 8803 | /// visitSelectInstWithICmp - Visit a SelectInst that has an |
8804 | /// ICmpInst as its first operand. | ||||
8805 | /// | ||||
8806 | Instruction *InstCombiner::visitSelectInstWithICmp(SelectInst &SI, | ||||
8807 | ICmpInst *ICI) { | ||||
8808 | bool Changed = false; | ||||
8809 | ICmpInst::Predicate Pred = ICI->getPredicate(); | ||||
8810 | Value *CmpLHS = ICI->getOperand(0); | ||||
8811 | Value *CmpRHS = ICI->getOperand(1); | ||||
8812 | Value *TrueVal = SI.getTrueValue(); | ||||
8813 | Value *FalseVal = SI.getFalseValue(); | ||||
8814 | |||||
8815 | // Check cases where the comparison is with a constant that | ||||
8816 | // can be adjusted to fit the min/max idiom. We may edit ICI in | ||||
8817 | // place here, so make sure the select is the only user. | ||||
8818 | if (ICI->hasOneUse()) | ||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8819 | if (ConstantInt *CI = dyn_cast<ConstantInt>(CmpRHS)) { |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 8820 | switch (Pred) { |
8821 | default: break; | ||||
8822 | case ICmpInst::ICMP_ULT: | ||||
8823 | case ICmpInst::ICMP_SLT: { | ||||
8824 | // X < MIN ? T : F --> F | ||||
8825 | if (CI->isMinValue(Pred == ICmpInst::ICMP_SLT)) | ||||
8826 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8827 | // X < C ? X : C-1 --> X > C-1 ? C-1 : X | ||||
8828 | Constant *AdjustedRHS = SubOne(CI); | ||||
8829 | if ((CmpLHS == TrueVal && AdjustedRHS == FalseVal) || | ||||
8830 | (CmpLHS == FalseVal && AdjustedRHS == TrueVal)) { | ||||
8831 | Pred = ICmpInst::getSwappedPredicate(Pred); | ||||
8832 | CmpRHS = AdjustedRHS; | ||||
8833 | std::swap(FalseVal, TrueVal); | ||||
8834 | ICI->setPredicate(Pred); | ||||
8835 | ICI->setOperand(1, CmpRHS); | ||||
8836 | SI.setOperand(1, TrueVal); | ||||
8837 | SI.setOperand(2, FalseVal); | ||||
8838 | Changed = true; | ||||
8839 | } | ||||
8840 | break; | ||||
8841 | } | ||||
8842 | case ICmpInst::ICMP_UGT: | ||||
8843 | case ICmpInst::ICMP_SGT: { | ||||
8844 | // X > MAX ? T : F --> F | ||||
8845 | if (CI->isMaxValue(Pred == ICmpInst::ICMP_SGT)) | ||||
8846 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8847 | // X > C ? X : C+1 --> X < C+1 ? C+1 : X | ||||
8848 | Constant *AdjustedRHS = AddOne(CI); | ||||
8849 | if ((CmpLHS == TrueVal && AdjustedRHS == FalseVal) || | ||||
8850 | (CmpLHS == FalseVal && AdjustedRHS == TrueVal)) { | ||||
8851 | Pred = ICmpInst::getSwappedPredicate(Pred); | ||||
8852 | CmpRHS = AdjustedRHS; | ||||
8853 | std::swap(FalseVal, TrueVal); | ||||
8854 | ICI->setPredicate(Pred); | ||||
8855 | ICI->setOperand(1, CmpRHS); | ||||
8856 | SI.setOperand(1, TrueVal); | ||||
8857 | SI.setOperand(2, FalseVal); | ||||
8858 | Changed = true; | ||||
8859 | } | ||||
8860 | break; | ||||
8861 | } | ||||
8862 | } | ||||
8863 | |||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8864 | // (x <s 0) ? -1 : 0 -> ashr x, 31 -> all ones if signed |
8865 | // (x >s -1) ? -1 : 0 -> ashr x, 31 -> all ones if not signed | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 8866 | CmpInst::Predicate Pred = CmpInst::BAD_ICMP_PREDICATE; |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 8867 | if (match(TrueVal, m_ConstantInt<-1>()) && |
8868 | match(FalseVal, m_ConstantInt<0>())) | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 8869 | Pred = ICI->getPredicate(); |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 8870 | else if (match(TrueVal, m_ConstantInt<0>()) && |
8871 | match(FalseVal, m_ConstantInt<-1>())) | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 8872 | Pred = CmpInst::getInversePredicate(ICI->getPredicate()); |
8873 | |||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8874 | if (Pred != CmpInst::BAD_ICMP_PREDICATE) { |
8875 | // If we are just checking for a icmp eq of a single bit and zext'ing it | ||||
8876 | // to an integer, then shift the bit to the appropriate place and then | ||||
8877 | // cast to integer to avoid the comparison. | ||||
8878 | const APInt &Op1CV = CI->getValue(); | ||||
8879 | |||||
8880 | // sext (x <s 0) to i32 --> x>>s31 true if signbit set. | ||||
8881 | // sext (x >s -1) to i32 --> (x>>s31)^-1 true if signbit clear. | ||||
8882 | if ((Pred == ICmpInst::ICMP_SLT && Op1CV == 0) || | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 8883 | (Pred == ICmpInst::ICMP_SGT && Op1CV.isAllOnesValue())) { |
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8884 | Value *In = ICI->getOperand(0); |
8885 | Value *Sh = ConstantInt::get(In->getType(), | ||||
8886 | In->getType()->getPrimitiveSizeInBits()-1); | ||||
8887 | In = InsertNewInstBefore(BinaryOperator::CreateAShr(In, Sh, | ||||
8888 | In->getName()+".lobit"), | ||||
8889 | *ICI); | ||||
Dan Gohman | 47a6077 | 2008-11-02 00:17:33 +0000 | [diff] [blame] | 8890 | if (In->getType() != SI.getType()) |
8891 | In = CastInst::CreateIntegerCast(In, SI.getType(), | ||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8892 | true/*SExt*/, "tmp", ICI); |
8893 | |||||
8894 | if (Pred == ICmpInst::ICMP_SGT) | ||||
8895 | In = InsertNewInstBefore(BinaryOperator::CreateNot(In, | ||||
8896 | In->getName()+".not"), *ICI); | ||||
8897 | |||||
8898 | return ReplaceInstUsesWith(SI, In); | ||||
8899 | } | ||||
8900 | } | ||||
8901 | } | ||||
8902 | |||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 8903 | if (CmpLHS == TrueVal && CmpRHS == FalseVal) { |
8904 | // Transform (X == Y) ? X : Y -> Y | ||||
8905 | if (Pred == ICmpInst::ICMP_EQ) | ||||
8906 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8907 | // Transform (X != Y) ? X : Y -> X | ||||
8908 | if (Pred == ICmpInst::ICMP_NE) | ||||
8909 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8910 | /// NOTE: if we wanted to, this is where to detect integer MIN/MAX | ||||
8911 | |||||
8912 | } else if (CmpLHS == FalseVal && CmpRHS == TrueVal) { | ||||
8913 | // Transform (X == Y) ? Y : X -> X | ||||
8914 | if (Pred == ICmpInst::ICMP_EQ) | ||||
8915 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8916 | // Transform (X != Y) ? Y : X -> Y | ||||
8917 | if (Pred == ICmpInst::ICMP_NE) | ||||
8918 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8919 | /// NOTE: if we wanted to, this is where to detect integer MIN/MAX | ||||
8920 | } | ||||
8921 | |||||
8922 | /// NOTE: if we wanted to, this is where to detect integer ABS | ||||
8923 | |||||
8924 | return Changed ? &SI : 0; | ||||
8925 | } | ||||
8926 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8927 | Instruction *InstCombiner::visitSelectInst(SelectInst &SI) { |
8928 | Value *CondVal = SI.getCondition(); | ||||
8929 | Value *TrueVal = SI.getTrueValue(); | ||||
8930 | Value *FalseVal = SI.getFalseValue(); | ||||
8931 | |||||
8932 | // select true, X, Y -> X | ||||
8933 | // select false, X, Y -> Y | ||||
8934 | if (ConstantInt *C = dyn_cast<ConstantInt>(CondVal)) | ||||
8935 | return ReplaceInstUsesWith(SI, C->getZExtValue() ? TrueVal : FalseVal); | ||||
8936 | |||||
8937 | // select C, X, X -> X | ||||
8938 | if (TrueVal == FalseVal) | ||||
8939 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8940 | |||||
8941 | if (isa<UndefValue>(TrueVal)) // select C, undef, X -> X | ||||
8942 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8943 | if (isa<UndefValue>(FalseVal)) // select C, X, undef -> X | ||||
8944 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8945 | if (isa<UndefValue>(CondVal)) { // select undef, X, Y -> X or Y | ||||
8946 | if (isa<Constant>(TrueVal)) | ||||
8947 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8948 | else | ||||
8949 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8950 | } | ||||
8951 | |||||
8952 | if (SI.getType() == Type::Int1Ty) { | ||||
8953 | if (ConstantInt *C = dyn_cast<ConstantInt>(TrueVal)) { | ||||
8954 | if (C->getZExtValue()) { | ||||
8955 | // Change: A = select B, true, C --> A = or B, C | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8956 | return BinaryOperator::CreateOr(CondVal, FalseVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8957 | } else { |
8958 | // Change: A = select B, false, C --> A = and !B, C | ||||
8959 | Value *NotCond = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8960 | InsertNewInstBefore(BinaryOperator::CreateNot(CondVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8961 | "not."+CondVal->getName()), SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8962 | return BinaryOperator::CreateAnd(NotCond, FalseVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8963 | } |
8964 | } else if (ConstantInt *C = dyn_cast<ConstantInt>(FalseVal)) { | ||||
8965 | if (C->getZExtValue() == false) { | ||||
8966 | // Change: A = select B, C, false --> A = and B, C | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8967 | return BinaryOperator::CreateAnd(CondVal, TrueVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8968 | } else { |
8969 | // Change: A = select B, C, true --> A = or !B, C | ||||
8970 | Value *NotCond = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8971 | InsertNewInstBefore(BinaryOperator::CreateNot(CondVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8972 | "not."+CondVal->getName()), SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8973 | return BinaryOperator::CreateOr(NotCond, TrueVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8974 | } |
8975 | } | ||||
Chris Lattner | 53f85a7 | 2007-11-25 21:27:53 +0000 | [diff] [blame] | 8976 | |
8977 | // select a, b, a -> a&b | ||||
8978 | // select a, a, b -> a|b | ||||
8979 | if (CondVal == TrueVal) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8980 | return BinaryOperator::CreateOr(CondVal, FalseVal); |
Chris Lattner | 53f85a7 | 2007-11-25 21:27:53 +0000 | [diff] [blame] | 8981 | else if (CondVal == FalseVal) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8982 | return BinaryOperator::CreateAnd(CondVal, TrueVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8983 | } |
8984 | |||||
8985 | // Selecting between two integer constants? | ||||
8986 | if (ConstantInt *TrueValC = dyn_cast<ConstantInt>(TrueVal)) | ||||
8987 | if (ConstantInt *FalseValC = dyn_cast<ConstantInt>(FalseVal)) { | ||||
8988 | // select C, 1, 0 -> zext C to int | ||||
8989 | if (FalseValC->isZero() && TrueValC->getValue() == 1) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8990 | return CastInst::Create(Instruction::ZExt, CondVal, SI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8991 | } else if (TrueValC->isZero() && FalseValC->getValue() == 1) { |
8992 | // select C, 0, 1 -> zext !C to int | ||||
8993 | Value *NotCond = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8994 | InsertNewInstBefore(BinaryOperator::CreateNot(CondVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8995 | "not."+CondVal->getName()), SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8996 | return CastInst::Create(Instruction::ZExt, NotCond, SI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8997 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8998 | |
8999 | if (ICmpInst *IC = dyn_cast<ICmpInst>(SI.getCondition())) { | ||||
9000 | |||||
9001 | // (x <s 0) ? -1 : 0 -> ashr x, 31 | ||||
9002 | if (TrueValC->isAllOnesValue() && FalseValC->isZero()) | ||||
9003 | if (ConstantInt *CmpCst = dyn_cast<ConstantInt>(IC->getOperand(1))) { | ||||
9004 | if (IC->getPredicate() == ICmpInst::ICMP_SLT && CmpCst->isZero()) { | ||||
9005 | // The comparison constant and the result are not neccessarily the | ||||
9006 | // same width. Make an all-ones value by inserting a AShr. | ||||
9007 | Value *X = IC->getOperand(0); | ||||
9008 | uint32_t Bits = X->getType()->getPrimitiveSizeInBits(); | ||||
9009 | Constant *ShAmt = ConstantInt::get(X->getType(), Bits-1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9010 | Instruction *SRA = BinaryOperator::Create(Instruction::AShr, X, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9011 | ShAmt, "ones"); |
9012 | InsertNewInstBefore(SRA, SI); | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 9013 | |
9014 | // Then cast to the appropriate width. | ||||
9015 | return CastInst::CreateIntegerCast(SRA, SI.getType(), true); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9016 | } |
9017 | } | ||||
9018 | |||||
9019 | |||||
9020 | // If one of the constants is zero (we know they can't both be) and we | ||||
9021 | // have an icmp instruction with zero, and we have an 'and' with the | ||||
9022 | // non-constant value, eliminate this whole mess. This corresponds to | ||||
9023 | // cases like this: ((X & 27) ? 27 : 0) | ||||
9024 | if (TrueValC->isZero() || FalseValC->isZero()) | ||||
9025 | if (IC->isEquality() && isa<ConstantInt>(IC->getOperand(1)) && | ||||
9026 | cast<Constant>(IC->getOperand(1))->isNullValue()) | ||||
9027 | if (Instruction *ICA = dyn_cast<Instruction>(IC->getOperand(0))) | ||||
9028 | if (ICA->getOpcode() == Instruction::And && | ||||
9029 | isa<ConstantInt>(ICA->getOperand(1)) && | ||||
9030 | (ICA->getOperand(1) == TrueValC || | ||||
9031 | ICA->getOperand(1) == FalseValC) && | ||||
9032 | isOneBitSet(cast<ConstantInt>(ICA->getOperand(1)))) { | ||||
9033 | // Okay, now we know that everything is set up, we just don't | ||||
9034 | // know whether we have a icmp_ne or icmp_eq and whether the | ||||
9035 | // true or false val is the zero. | ||||
9036 | bool ShouldNotVal = !TrueValC->isZero(); | ||||
9037 | ShouldNotVal ^= IC->getPredicate() == ICmpInst::ICMP_NE; | ||||
9038 | Value *V = ICA; | ||||
9039 | if (ShouldNotVal) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9040 | V = InsertNewInstBefore(BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9041 | Instruction::Xor, V, ICA->getOperand(1)), SI); |
9042 | return ReplaceInstUsesWith(SI, V); | ||||
9043 | } | ||||
9044 | } | ||||
9045 | } | ||||
9046 | |||||
9047 | // See if we are selecting two values based on a comparison of the two values. | ||||
9048 | if (FCmpInst *FCI = dyn_cast<FCmpInst>(CondVal)) { | ||||
9049 | if (FCI->getOperand(0) == TrueVal && FCI->getOperand(1) == FalseVal) { | ||||
9050 | // Transform (X == Y) ? X : Y -> Y | ||||
Dale Johannesen | 2e1b769 | 2007-10-03 17:45:27 +0000 | [diff] [blame] | 9051 | if (FCI->getPredicate() == FCmpInst::FCMP_OEQ) { |
9052 | // This is not safe in general for floating point: | ||||
9053 | // consider X== -0, Y== +0. | ||||
9054 | // It becomes safe if either operand is a nonzero constant. | ||||
9055 | ConstantFP *CFPt, *CFPf; | ||||
9056 | if (((CFPt = dyn_cast<ConstantFP>(TrueVal)) && | ||||
9057 | !CFPt->getValueAPF().isZero()) || | ||||
9058 | ((CFPf = dyn_cast<ConstantFP>(FalseVal)) && | ||||
9059 | !CFPf->getValueAPF().isZero())) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9060 | return ReplaceInstUsesWith(SI, FalseVal); |
Dale Johannesen | 2e1b769 | 2007-10-03 17:45:27 +0000 | [diff] [blame] | 9061 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9062 | // Transform (X != Y) ? X : Y -> X |
9063 | if (FCI->getPredicate() == FCmpInst::FCMP_ONE) | ||||
9064 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 9065 | // NOTE: if we wanted to, this is where to detect MIN/MAX |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9066 | |
9067 | } else if (FCI->getOperand(0) == FalseVal && FCI->getOperand(1) == TrueVal){ | ||||
9068 | // Transform (X == Y) ? Y : X -> X | ||||
Dale Johannesen | 2e1b769 | 2007-10-03 17:45:27 +0000 | [diff] [blame] | 9069 | if (FCI->getPredicate() == FCmpInst::FCMP_OEQ) { |
9070 | // This is not safe in general for floating point: | ||||
9071 | // consider X== -0, Y== +0. | ||||
9072 | // It becomes safe if either operand is a nonzero constant. | ||||
9073 | ConstantFP *CFPt, *CFPf; | ||||
9074 | if (((CFPt = dyn_cast<ConstantFP>(TrueVal)) && | ||||
9075 | !CFPt->getValueAPF().isZero()) || | ||||
9076 | ((CFPf = dyn_cast<ConstantFP>(FalseVal)) && | ||||
9077 | !CFPf->getValueAPF().isZero())) | ||||
9078 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
9079 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9080 | // Transform (X != Y) ? Y : X -> Y |
9081 | if (FCI->getPredicate() == FCmpInst::FCMP_ONE) | ||||
9082 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 9083 | // NOTE: if we wanted to, this is where to detect MIN/MAX |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9084 | } |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 9085 | // NOTE: if we wanted to, this is where to detect ABS |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9086 | } |
9087 | |||||
9088 | // See if we are selecting two values based on a comparison of the two values. | ||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 9089 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(CondVal)) |
9090 | if (Instruction *Result = visitSelectInstWithICmp(SI, ICI)) | ||||
9091 | return Result; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9092 | |
9093 | if (Instruction *TI = dyn_cast<Instruction>(TrueVal)) | ||||
9094 | if (Instruction *FI = dyn_cast<Instruction>(FalseVal)) | ||||
9095 | if (TI->hasOneUse() && FI->hasOneUse()) { | ||||
9096 | Instruction *AddOp = 0, *SubOp = 0; | ||||
9097 | |||||
9098 | // Turn (select C, (op X, Y), (op X, Z)) -> (op X, (select C, Y, Z)) | ||||
9099 | if (TI->getOpcode() == FI->getOpcode()) | ||||
9100 | if (Instruction *IV = FoldSelectOpOp(SI, TI, FI)) | ||||
9101 | return IV; | ||||
9102 | |||||
9103 | // Turn select C, (X+Y), (X-Y) --> (X+(select C, Y, (-Y))). This is | ||||
9104 | // even legal for FP. | ||||
9105 | if (TI->getOpcode() == Instruction::Sub && | ||||
9106 | FI->getOpcode() == Instruction::Add) { | ||||
9107 | AddOp = FI; SubOp = TI; | ||||
9108 | } else if (FI->getOpcode() == Instruction::Sub && | ||||
9109 | TI->getOpcode() == Instruction::Add) { | ||||
9110 | AddOp = TI; SubOp = FI; | ||||
9111 | } | ||||
9112 | |||||
9113 | if (AddOp) { | ||||
9114 | Value *OtherAddOp = 0; | ||||
9115 | if (SubOp->getOperand(0) == AddOp->getOperand(0)) { | ||||
9116 | OtherAddOp = AddOp->getOperand(1); | ||||
9117 | } else if (SubOp->getOperand(0) == AddOp->getOperand(1)) { | ||||
9118 | OtherAddOp = AddOp->getOperand(0); | ||||
9119 | } | ||||
9120 | |||||
9121 | if (OtherAddOp) { | ||||
9122 | // So at this point we know we have (Y -> OtherAddOp): | ||||
9123 | // select C, (add X, Y), (sub X, Z) | ||||
9124 | Value *NegVal; // Compute -Z | ||||
9125 | if (Constant *C = dyn_cast<Constant>(SubOp->getOperand(1))) { | ||||
9126 | NegVal = ConstantExpr::getNeg(C); | ||||
9127 | } else { | ||||
9128 | NegVal = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9129 | BinaryOperator::CreateNeg(SubOp->getOperand(1), "tmp"), SI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9130 | } |
9131 | |||||
9132 | Value *NewTrueOp = OtherAddOp; | ||||
9133 | Value *NewFalseOp = NegVal; | ||||
9134 | if (AddOp != TI) | ||||
9135 | std::swap(NewTrueOp, NewFalseOp); | ||||
9136 | Instruction *NewSel = | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9137 | SelectInst::Create(CondVal, NewTrueOp, |
9138 | NewFalseOp, SI.getName() + ".p"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9139 | |
9140 | NewSel = InsertNewInstBefore(NewSel, SI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9141 | return BinaryOperator::CreateAdd(SubOp->getOperand(0), NewSel); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9142 | } |
9143 | } | ||||
9144 | } | ||||
9145 | |||||
9146 | // See if we can fold the select into one of our operands. | ||||
9147 | if (SI.getType()->isInteger()) { | ||||
9148 | // See the comment above GetSelectFoldableOperands for a description of the | ||||
9149 | // transformation we are doing here. | ||||
9150 | if (Instruction *TVI = dyn_cast<Instruction>(TrueVal)) | ||||
9151 | if (TVI->hasOneUse() && TVI->getNumOperands() == 2 && | ||||
9152 | !isa<Constant>(FalseVal)) | ||||
9153 | if (unsigned SFO = GetSelectFoldableOperands(TVI)) { | ||||
9154 | unsigned OpToFold = 0; | ||||
9155 | if ((SFO & 1) && FalseVal == TVI->getOperand(0)) { | ||||
9156 | OpToFold = 1; | ||||
9157 | } else if ((SFO & 2) && FalseVal == TVI->getOperand(1)) { | ||||
9158 | OpToFold = 2; | ||||
9159 | } | ||||
9160 | |||||
9161 | if (OpToFold) { | ||||
9162 | Constant *C = GetSelectFoldableConstant(TVI); | ||||
9163 | Instruction *NewSel = | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9164 | SelectInst::Create(SI.getCondition(), |
9165 | TVI->getOperand(2-OpToFold), C); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9166 | InsertNewInstBefore(NewSel, SI); |
9167 | NewSel->takeName(TVI); | ||||
9168 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(TVI)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9169 | return BinaryOperator::Create(BO->getOpcode(), FalseVal, NewSel); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9170 | else { |
9171 | assert(0 && "Unknown instruction!!"); | ||||
9172 | } | ||||
9173 | } | ||||
9174 | } | ||||
9175 | |||||
9176 | if (Instruction *FVI = dyn_cast<Instruction>(FalseVal)) | ||||
9177 | if (FVI->hasOneUse() && FVI->getNumOperands() == 2 && | ||||
9178 | !isa<Constant>(TrueVal)) | ||||
9179 | if (unsigned SFO = GetSelectFoldableOperands(FVI)) { | ||||
9180 | unsigned OpToFold = 0; | ||||
9181 | if ((SFO & 1) && TrueVal == FVI->getOperand(0)) { | ||||
9182 | OpToFold = 1; | ||||
9183 | } else if ((SFO & 2) && TrueVal == FVI->getOperand(1)) { | ||||
9184 | OpToFold = 2; | ||||
9185 | } | ||||
9186 | |||||
9187 | if (OpToFold) { | ||||
9188 | Constant *C = GetSelectFoldableConstant(FVI); | ||||
9189 | Instruction *NewSel = | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9190 | SelectInst::Create(SI.getCondition(), C, |
9191 | FVI->getOperand(2-OpToFold)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9192 | InsertNewInstBefore(NewSel, SI); |
9193 | NewSel->takeName(FVI); | ||||
9194 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(FVI)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9195 | return BinaryOperator::Create(BO->getOpcode(), TrueVal, NewSel); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9196 | else |
9197 | assert(0 && "Unknown instruction!!"); | ||||
9198 | } | ||||
9199 | } | ||||
9200 | } | ||||
9201 | |||||
9202 | if (BinaryOperator::isNot(CondVal)) { | ||||
9203 | SI.setOperand(0, BinaryOperator::getNotArgument(CondVal)); | ||||
9204 | SI.setOperand(1, FalseVal); | ||||
9205 | SI.setOperand(2, TrueVal); | ||||
9206 | return &SI; | ||||
9207 | } | ||||
9208 | |||||
9209 | return 0; | ||||
9210 | } | ||||
9211 | |||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9212 | /// EnforceKnownAlignment - If the specified pointer points to an object that |
9213 | /// we control, modify the object's alignment to PrefAlign. This isn't | ||||
9214 | /// often possible though. If alignment is important, a more reliable approach | ||||
9215 | /// is to simply align all global variables and allocation instructions to | ||||
9216 | /// their preferred alignment from the beginning. | ||||
9217 | /// | ||||
9218 | static unsigned EnforceKnownAlignment(Value *V, | ||||
9219 | unsigned Align, unsigned PrefAlign) { | ||||
Chris Lattner | 47cf345 | 2007-08-09 19:05:49 +0000 | [diff] [blame] | 9220 | |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9221 | User *U = dyn_cast<User>(V); |
9222 | if (!U) return Align; | ||||
9223 | |||||
9224 | switch (getOpcode(U)) { | ||||
9225 | default: break; | ||||
9226 | case Instruction::BitCast: | ||||
9227 | return EnforceKnownAlignment(U->getOperand(0), Align, PrefAlign); | ||||
9228 | case Instruction::GetElementPtr: { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9229 | // If all indexes are zero, it is just the alignment of the base pointer. |
9230 | bool AllZeroOperands = true; | ||||
Gabor Greif | e92fbe2 | 2008-06-12 21:51:29 +0000 | [diff] [blame] | 9231 | for (User::op_iterator i = U->op_begin() + 1, e = U->op_end(); i != e; ++i) |
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 9232 | if (!isa<Constant>(*i) || |
9233 | !cast<Constant>(*i)->isNullValue()) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9234 | AllZeroOperands = false; |
9235 | break; | ||||
9236 | } | ||||
Chris Lattner | 47cf345 | 2007-08-09 19:05:49 +0000 | [diff] [blame] | 9237 | |
9238 | if (AllZeroOperands) { | ||||
9239 | // Treat this like a bitcast. | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9240 | return EnforceKnownAlignment(U->getOperand(0), Align, PrefAlign); |
Chris Lattner | 47cf345 | 2007-08-09 19:05:49 +0000 | [diff] [blame] | 9241 | } |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9242 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9243 | } |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9244 | } |
9245 | |||||
9246 | if (GlobalValue *GV = dyn_cast<GlobalValue>(V)) { | ||||
9247 | // If there is a large requested alignment and we can, bump up the alignment | ||||
9248 | // of the global. | ||||
9249 | if (!GV->isDeclaration()) { | ||||
Dan Gohman | f6fe71e | 2009-02-16 23:02:21 +0000 | [diff] [blame] | 9250 | if (GV->getAlignment() >= PrefAlign) |
9251 | Align = GV->getAlignment(); | ||||
9252 | else { | ||||
9253 | GV->setAlignment(PrefAlign); | ||||
9254 | Align = PrefAlign; | ||||
9255 | } | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9256 | } |
9257 | } else if (AllocationInst *AI = dyn_cast<AllocationInst>(V)) { | ||||
9258 | // If there is a requested alignment and if this is an alloca, round up. We | ||||
9259 | // don't do this for malloc, because some systems can't respect the request. | ||||
9260 | if (isa<AllocaInst>(AI)) { | ||||
Dan Gohman | f6fe71e | 2009-02-16 23:02:21 +0000 | [diff] [blame] | 9261 | if (AI->getAlignment() >= PrefAlign) |
9262 | Align = AI->getAlignment(); | ||||
9263 | else { | ||||
9264 | AI->setAlignment(PrefAlign); | ||||
9265 | Align = PrefAlign; | ||||
9266 | } | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9267 | } |
9268 | } | ||||
9269 | |||||
9270 | return Align; | ||||
9271 | } | ||||
9272 | |||||
9273 | /// GetOrEnforceKnownAlignment - If the specified pointer has an alignment that | ||||
9274 | /// we can determine, return it, otherwise return 0. If PrefAlign is specified, | ||||
9275 | /// and it is more than the alignment of the ultimate object, see if we can | ||||
9276 | /// increase the alignment of the ultimate object, making this check succeed. | ||||
9277 | unsigned InstCombiner::GetOrEnforceKnownAlignment(Value *V, | ||||
9278 | unsigned PrefAlign) { | ||||
9279 | unsigned BitWidth = TD ? TD->getTypeSizeInBits(V->getType()) : | ||||
9280 | sizeof(PrefAlign) * CHAR_BIT; | ||||
9281 | APInt Mask = APInt::getAllOnesValue(BitWidth); | ||||
9282 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
9283 | ComputeMaskedBits(V, Mask, KnownZero, KnownOne); | ||||
9284 | unsigned TrailZ = KnownZero.countTrailingOnes(); | ||||
9285 | unsigned Align = 1u << std::min(BitWidth - 1, TrailZ); | ||||
9286 | |||||
9287 | if (PrefAlign > Align) | ||||
9288 | Align = EnforceKnownAlignment(V, Align, PrefAlign); | ||||
9289 | |||||
9290 | // We don't need to make any adjustment. | ||||
9291 | return Align; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9292 | } |
9293 | |||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9294 | Instruction *InstCombiner::SimplifyMemTransfer(MemIntrinsic *MI) { |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9295 | unsigned DstAlign = GetOrEnforceKnownAlignment(MI->getOperand(1)); |
Dan Gohman | eb25491 | 2009-02-22 18:06:32 +0000 | [diff] [blame] | 9296 | unsigned SrcAlign = GetOrEnforceKnownAlignment(MI->getOperand(2)); |
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9297 | unsigned MinAlign = std::min(DstAlign, SrcAlign); |
9298 | unsigned CopyAlign = MI->getAlignment()->getZExtValue(); | ||||
9299 | |||||
9300 | if (CopyAlign < MinAlign) { | ||||
9301 | MI->setAlignment(ConstantInt::get(Type::Int32Ty, MinAlign)); | ||||
9302 | return MI; | ||||
9303 | } | ||||
9304 | |||||
9305 | // If MemCpyInst length is 1/2/4/8 bytes then replace memcpy with | ||||
9306 | // load/store. | ||||
9307 | ConstantInt *MemOpLength = dyn_cast<ConstantInt>(MI->getOperand(3)); | ||||
9308 | if (MemOpLength == 0) return 0; | ||||
9309 | |||||
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9310 | // Source and destination pointer types are always "i8*" for intrinsic. See |
9311 | // if the size is something we can handle with a single primitive load/store. | ||||
9312 | // A single load+store correctly handles overlapping memory in the memmove | ||||
9313 | // case. | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9314 | unsigned Size = MemOpLength->getZExtValue(); |
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 9315 | if (Size == 0) return MI; // Delete this mem transfer. |
9316 | |||||
9317 | if (Size > 8 || (Size&(Size-1))) | ||||
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9318 | return 0; // If not 1/2/4/8 bytes, exit. |
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9319 | |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9320 | // Use an integer load+store unless we can find something better. |
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9321 | Type *NewPtrTy = PointerType::getUnqual(IntegerType::get(Size<<3)); |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9322 | |
9323 | // Memcpy forces the use of i8* for the source and destination. That means | ||||
9324 | // that if you're using memcpy to move one double around, you'll get a cast | ||||
9325 | // from double* to i8*. We'd much rather use a double load+store rather than | ||||
9326 | // an i64 load+store, here because this improves the odds that the source or | ||||
9327 | // dest address will be promotable. See if we can find a better type than the | ||||
9328 | // integer datatype. | ||||
9329 | if (Value *Op = getBitCastOperand(MI->getOperand(1))) { | ||||
9330 | const Type *SrcETy = cast<PointerType>(Op->getType())->getElementType(); | ||||
9331 | if (SrcETy->isSized() && TD->getTypeStoreSize(SrcETy) == Size) { | ||||
9332 | // The SrcETy might be something like {{{double}}} or [1 x double]. Rip | ||||
9333 | // down through these levels if so. | ||||
Dan Gohman | b8e94f6 | 2008-05-23 01:52:21 +0000 | [diff] [blame] | 9334 | while (!SrcETy->isSingleValueType()) { |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9335 | if (const StructType *STy = dyn_cast<StructType>(SrcETy)) { |
9336 | if (STy->getNumElements() == 1) | ||||
9337 | SrcETy = STy->getElementType(0); | ||||
9338 | else | ||||
9339 | break; | ||||
9340 | } else if (const ArrayType *ATy = dyn_cast<ArrayType>(SrcETy)) { | ||||
9341 | if (ATy->getNumElements() == 1) | ||||
9342 | SrcETy = ATy->getElementType(); | ||||
9343 | else | ||||
9344 | break; | ||||
9345 | } else | ||||
9346 | break; | ||||
9347 | } | ||||
9348 | |||||
Dan Gohman | b8e94f6 | 2008-05-23 01:52:21 +0000 | [diff] [blame] | 9349 | if (SrcETy->isSingleValueType()) |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9350 | NewPtrTy = PointerType::getUnqual(SrcETy); |
9351 | } | ||||
9352 | } | ||||
9353 | |||||
9354 | |||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9355 | // If the memcpy/memmove provides better alignment info than we can |
9356 | // infer, use it. | ||||
9357 | SrcAlign = std::max(SrcAlign, CopyAlign); | ||||
9358 | DstAlign = std::max(DstAlign, CopyAlign); | ||||
9359 | |||||
9360 | Value *Src = InsertBitCastBefore(MI->getOperand(2), NewPtrTy, *MI); | ||||
9361 | Value *Dest = InsertBitCastBefore(MI->getOperand(1), NewPtrTy, *MI); | ||||
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9362 | Instruction *L = new LoadInst(Src, "tmp", false, SrcAlign); |
9363 | InsertNewInstBefore(L, *MI); | ||||
9364 | InsertNewInstBefore(new StoreInst(L, Dest, false, DstAlign), *MI); | ||||
9365 | |||||
9366 | // Set the size of the copy to 0, it will be deleted on the next iteration. | ||||
9367 | MI->setOperand(3, Constant::getNullValue(MemOpLength->getType())); | ||||
9368 | return MI; | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9369 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9370 | |
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 9371 | Instruction *InstCombiner::SimplifyMemSet(MemSetInst *MI) { |
9372 | unsigned Alignment = GetOrEnforceKnownAlignment(MI->getDest()); | ||||
9373 | if (MI->getAlignment()->getZExtValue() < Alignment) { | ||||
9374 | MI->setAlignment(ConstantInt::get(Type::Int32Ty, Alignment)); | ||||
9375 | return MI; | ||||
9376 | } | ||||
9377 | |||||
9378 | // Extract the length and alignment and fill if they are constant. | ||||
9379 | ConstantInt *LenC = dyn_cast<ConstantInt>(MI->getLength()); | ||||
9380 | ConstantInt *FillC = dyn_cast<ConstantInt>(MI->getValue()); | ||||
9381 | if (!LenC || !FillC || FillC->getType() != Type::Int8Ty) | ||||
9382 | return 0; | ||||
9383 | uint64_t Len = LenC->getZExtValue(); | ||||
9384 | Alignment = MI->getAlignment()->getZExtValue(); | ||||
9385 | |||||
9386 | // If the length is zero, this is a no-op | ||||
9387 | if (Len == 0) return MI; // memset(d,c,0,a) -> noop | ||||
9388 | |||||
9389 | // memset(s,c,n) -> store s, c (for n=1,2,4,8) | ||||
9390 | if (Len <= 8 && isPowerOf2_32((uint32_t)Len)) { | ||||
9391 | const Type *ITy = IntegerType::get(Len*8); // n=1 -> i8. | ||||
9392 | |||||
9393 | Value *Dest = MI->getDest(); | ||||
9394 | Dest = InsertBitCastBefore(Dest, PointerType::getUnqual(ITy), *MI); | ||||
9395 | |||||
9396 | // Alignment 0 is identity for alignment 1 for memset, but not store. | ||||
9397 | if (Alignment == 0) Alignment = 1; | ||||
9398 | |||||
9399 | // Extract the fill value and store. | ||||
9400 | uint64_t Fill = FillC->getZExtValue()*0x0101010101010101ULL; | ||||
9401 | InsertNewInstBefore(new StoreInst(ConstantInt::get(ITy, Fill), Dest, false, | ||||
9402 | Alignment), *MI); | ||||
9403 | |||||
9404 | // Set the size of the copy to 0, it will be deleted on the next iteration. | ||||
9405 | MI->setLength(Constant::getNullValue(LenC->getType())); | ||||
9406 | return MI; | ||||
9407 | } | ||||
9408 | |||||
9409 | return 0; | ||||
9410 | } | ||||
9411 | |||||
9412 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9413 | /// visitCallInst - CallInst simplification. This mostly only handles folding |
9414 | /// of intrinsic instructions. For normal calls, it allows visitCallSite to do | ||||
9415 | /// the heavy lifting. | ||||
9416 | /// | ||||
9417 | Instruction *InstCombiner::visitCallInst(CallInst &CI) { | ||||
9418 | IntrinsicInst *II = dyn_cast<IntrinsicInst>(&CI); | ||||
9419 | if (!II) return visitCallSite(&CI); | ||||
9420 | |||||
9421 | // Intrinsics cannot occur in an invoke, so handle them here instead of in | ||||
9422 | // visitCallSite. | ||||
9423 | if (MemIntrinsic *MI = dyn_cast<MemIntrinsic>(II)) { | ||||
9424 | bool Changed = false; | ||||
9425 | |||||
9426 | // memmove/cpy/set of zero bytes is a noop. | ||||
9427 | if (Constant *NumBytes = dyn_cast<Constant>(MI->getLength())) { | ||||
9428 | if (NumBytes->isNullValue()) return EraseInstFromFunction(CI); | ||||
9429 | |||||
9430 | if (ConstantInt *CI = dyn_cast<ConstantInt>(NumBytes)) | ||||
9431 | if (CI->getZExtValue() == 1) { | ||||
9432 | // Replace the instruction with just byte operations. We would | ||||
9433 | // transform other cases to loads/stores, but we don't know if | ||||
9434 | // alignment is sufficient. | ||||
9435 | } | ||||
9436 | } | ||||
9437 | |||||
9438 | // If we have a memmove and the source operation is a constant global, | ||||
9439 | // then the source and dest pointers can't alias, so we can change this | ||||
9440 | // into a call to memcpy. | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9441 | if (MemMoveInst *MMI = dyn_cast<MemMoveInst>(MI)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9442 | if (GlobalVariable *GVSrc = dyn_cast<GlobalVariable>(MMI->getSource())) |
9443 | if (GVSrc->isConstant()) { | ||||
9444 | Module *M = CI.getParent()->getParent()->getParent(); | ||||
Chris Lattner | 82c2e43 | 2008-11-21 16:42:48 +0000 | [diff] [blame] | 9445 | Intrinsic::ID MemCpyID = Intrinsic::memcpy; |
9446 | const Type *Tys[1]; | ||||
9447 | Tys[0] = CI.getOperand(3)->getType(); | ||||
9448 | CI.setOperand(0, | ||||
9449 | Intrinsic::getDeclaration(M, MemCpyID, Tys, 1)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9450 | Changed = true; |
9451 | } | ||||
Chris Lattner | 59b27d9 | 2008-05-28 05:30:41 +0000 | [diff] [blame] | 9452 | |
9453 | // memmove(x,x,size) -> noop. | ||||
9454 | if (MMI->getSource() == MMI->getDest()) | ||||
9455 | return EraseInstFromFunction(CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9456 | } |
9457 | |||||
9458 | // If we can determine a pointer alignment that is bigger than currently | ||||
9459 | // set, update the alignment. | ||||
9460 | if (isa<MemCpyInst>(MI) || isa<MemMoveInst>(MI)) { | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9461 | if (Instruction *I = SimplifyMemTransfer(MI)) |
9462 | return I; | ||||
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 9463 | } else if (MemSetInst *MSI = dyn_cast<MemSetInst>(MI)) { |
9464 | if (Instruction *I = SimplifyMemSet(MSI)) | ||||
9465 | return I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9466 | } |
9467 | |||||
9468 | if (Changed) return II; | ||||
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9469 | } |
9470 | |||||
9471 | switch (II->getIntrinsicID()) { | ||||
9472 | default: break; | ||||
9473 | case Intrinsic::bswap: | ||||
9474 | // bswap(bswap(x)) -> x | ||||
9475 | if (IntrinsicInst *Operand = dyn_cast<IntrinsicInst>(II->getOperand(1))) | ||||
9476 | if (Operand->getIntrinsicID() == Intrinsic::bswap) | ||||
9477 | return ReplaceInstUsesWith(CI, Operand->getOperand(1)); | ||||
9478 | break; | ||||
9479 | case Intrinsic::ppc_altivec_lvx: | ||||
9480 | case Intrinsic::ppc_altivec_lvxl: | ||||
9481 | case Intrinsic::x86_sse_loadu_ps: | ||||
9482 | case Intrinsic::x86_sse2_loadu_pd: | ||||
9483 | case Intrinsic::x86_sse2_loadu_dq: | ||||
9484 | // Turn PPC lvx -> load if the pointer is known aligned. | ||||
9485 | // Turn X86 loadups -> load if the pointer is known aligned. | ||||
9486 | if (GetOrEnforceKnownAlignment(II->getOperand(1), 16) >= 16) { | ||||
9487 | Value *Ptr = InsertBitCastBefore(II->getOperand(1), | ||||
9488 | PointerType::getUnqual(II->getType()), | ||||
9489 | CI); | ||||
9490 | return new LoadInst(Ptr); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9491 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9492 | break; |
9493 | case Intrinsic::ppc_altivec_stvx: | ||||
9494 | case Intrinsic::ppc_altivec_stvxl: | ||||
9495 | // Turn stvx -> store if the pointer is known aligned. | ||||
9496 | if (GetOrEnforceKnownAlignment(II->getOperand(2), 16) >= 16) { | ||||
9497 | const Type *OpPtrTy = | ||||
9498 | PointerType::getUnqual(II->getOperand(1)->getType()); | ||||
9499 | Value *Ptr = InsertBitCastBefore(II->getOperand(2), OpPtrTy, CI); | ||||
9500 | return new StoreInst(II->getOperand(1), Ptr); | ||||
9501 | } | ||||
9502 | break; | ||||
9503 | case Intrinsic::x86_sse_storeu_ps: | ||||
9504 | case Intrinsic::x86_sse2_storeu_pd: | ||||
9505 | case Intrinsic::x86_sse2_storeu_dq: | ||||
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9506 | // Turn X86 storeu -> store if the pointer is known aligned. |
9507 | if (GetOrEnforceKnownAlignment(II->getOperand(1), 16) >= 16) { | ||||
9508 | const Type *OpPtrTy = | ||||
9509 | PointerType::getUnqual(II->getOperand(2)->getType()); | ||||
9510 | Value *Ptr = InsertBitCastBefore(II->getOperand(1), OpPtrTy, CI); | ||||
9511 | return new StoreInst(II->getOperand(2), Ptr); | ||||
9512 | } | ||||
9513 | break; | ||||
9514 | |||||
9515 | case Intrinsic::x86_sse_cvttss2si: { | ||||
9516 | // These intrinsics only demands the 0th element of its input vector. If | ||||
9517 | // we can simplify the input based on that, do so now. | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 9518 | unsigned VWidth = |
9519 | cast<VectorType>(II->getOperand(1)->getType())->getNumElements(); | ||||
9520 | APInt DemandedElts(VWidth, 1); | ||||
9521 | APInt UndefElts(VWidth, 0); | ||||
9522 | if (Value *V = SimplifyDemandedVectorElts(II->getOperand(1), DemandedElts, | ||||
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9523 | UndefElts)) { |
9524 | II->setOperand(1, V); | ||||
9525 | return II; | ||||
9526 | } | ||||
9527 | break; | ||||
9528 | } | ||||
9529 | |||||
9530 | case Intrinsic::ppc_altivec_vperm: | ||||
9531 | // Turn vperm(V1,V2,mask) -> shuffle(V1,V2,mask) if mask is a constant. | ||||
9532 | if (ConstantVector *Mask = dyn_cast<ConstantVector>(II->getOperand(3))) { | ||||
9533 | assert(Mask->getNumOperands() == 16 && "Bad type for intrinsic!"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9534 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9535 | // Check that all of the elements are integer constants or undefs. |
9536 | bool AllEltsOk = true; | ||||
9537 | for (unsigned i = 0; i != 16; ++i) { | ||||
9538 | if (!isa<ConstantInt>(Mask->getOperand(i)) && | ||||
9539 | !isa<UndefValue>(Mask->getOperand(i))) { | ||||
9540 | AllEltsOk = false; | ||||
9541 | break; | ||||
9542 | } | ||||
9543 | } | ||||
9544 | |||||
9545 | if (AllEltsOk) { | ||||
9546 | // Cast the input vectors to byte vectors. | ||||
9547 | Value *Op0 =InsertBitCastBefore(II->getOperand(1),Mask->getType(),CI); | ||||
9548 | Value *Op1 =InsertBitCastBefore(II->getOperand(2),Mask->getType(),CI); | ||||
9549 | Value *Result = UndefValue::get(Op0->getType()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9550 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9551 | // Only extract each element once. |
9552 | Value *ExtractedElts[32]; | ||||
9553 | memset(ExtractedElts, 0, sizeof(ExtractedElts)); | ||||
9554 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9555 | for (unsigned i = 0; i != 16; ++i) { |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9556 | if (isa<UndefValue>(Mask->getOperand(i))) |
9557 | continue; | ||||
9558 | unsigned Idx=cast<ConstantInt>(Mask->getOperand(i))->getZExtValue(); | ||||
9559 | Idx &= 31; // Match the hardware behavior. | ||||
9560 | |||||
9561 | if (ExtractedElts[Idx] == 0) { | ||||
9562 | Instruction *Elt = | ||||
9563 | new ExtractElementInst(Idx < 16 ? Op0 : Op1, Idx&15, "tmp"); | ||||
9564 | InsertNewInstBefore(Elt, CI); | ||||
9565 | ExtractedElts[Idx] = Elt; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9566 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9567 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9568 | // Insert this value into the result vector. |
9569 | Result = InsertElementInst::Create(Result, ExtractedElts[Idx], | ||||
9570 | i, "tmp"); | ||||
9571 | InsertNewInstBefore(cast<Instruction>(Result), CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9572 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9573 | return CastInst::Create(Instruction::BitCast, Result, CI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9574 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9575 | } |
9576 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9577 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9578 | case Intrinsic::stackrestore: { |
9579 | // If the save is right next to the restore, remove the restore. This can | ||||
9580 | // happen when variable allocas are DCE'd. | ||||
9581 | if (IntrinsicInst *SS = dyn_cast<IntrinsicInst>(II->getOperand(1))) { | ||||
9582 | if (SS->getIntrinsicID() == Intrinsic::stacksave) { | ||||
9583 | BasicBlock::iterator BI = SS; | ||||
9584 | if (&*++BI == II) | ||||
9585 | return EraseInstFromFunction(CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9586 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9587 | } |
9588 | |||||
9589 | // Scan down this block to see if there is another stack restore in the | ||||
9590 | // same block without an intervening call/alloca. | ||||
9591 | BasicBlock::iterator BI = II; | ||||
9592 | TerminatorInst *TI = II->getParent()->getTerminator(); | ||||
9593 | bool CannotRemove = false; | ||||
9594 | for (++BI; &*BI != TI; ++BI) { | ||||
9595 | if (isa<AllocaInst>(BI)) { | ||||
9596 | CannotRemove = true; | ||||
9597 | break; | ||||
9598 | } | ||||
Chris Lattner | a6b477c | 2008-06-25 05:59:28 +0000 | [diff] [blame] | 9599 | if (CallInst *BCI = dyn_cast<CallInst>(BI)) { |
9600 | if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(BCI)) { | ||||
9601 | // If there is a stackrestore below this one, remove this one. | ||||
9602 | if (II->getIntrinsicID() == Intrinsic::stackrestore) | ||||
9603 | return EraseInstFromFunction(CI); | ||||
9604 | // Otherwise, ignore the intrinsic. | ||||
9605 | } else { | ||||
9606 | // If we found a non-intrinsic call, we can't remove the stack | ||||
9607 | // restore. | ||||
Chris Lattner | 416d91c | 2008-02-18 06:12:38 +0000 | [diff] [blame] | 9608 | CannotRemove = true; |
9609 | break; | ||||
9610 | } | ||||
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9611 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9612 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9613 | |
9614 | // If the stack restore is in a return/unwind block and if there are no | ||||
9615 | // allocas or calls between the restore and the return, nuke the restore. | ||||
9616 | if (!CannotRemove && (isa<ReturnInst>(TI) || isa<UnwindInst>(TI))) | ||||
9617 | return EraseInstFromFunction(CI); | ||||
9618 | break; | ||||
9619 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9620 | } |
9621 | |||||
9622 | return visitCallSite(II); | ||||
9623 | } | ||||
9624 | |||||
9625 | // InvokeInst simplification | ||||
9626 | // | ||||
9627 | Instruction *InstCombiner::visitInvokeInst(InvokeInst &II) { | ||||
9628 | return visitCallSite(&II); | ||||
9629 | } | ||||
9630 | |||||
Dale Johannesen | 9602183 | 2008-04-25 21:16:07 +0000 | [diff] [blame] | 9631 | /// isSafeToEliminateVarargsCast - If this cast does not affect the value |
9632 | /// passed through the varargs area, we can eliminate the use of the cast. | ||||
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9633 | static bool isSafeToEliminateVarargsCast(const CallSite CS, |
9634 | const CastInst * const CI, | ||||
9635 | const TargetData * const TD, | ||||
9636 | const int ix) { | ||||
9637 | if (!CI->isLosslessCast()) | ||||
9638 | return false; | ||||
9639 | |||||
9640 | // The size of ByVal arguments is derived from the type, so we | ||||
9641 | // can't change to a type with a different size. If the size were | ||||
9642 | // passed explicitly we could avoid this check. | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9643 | if (!CS.paramHasAttr(ix, Attribute::ByVal)) |
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9644 | return true; |
9645 | |||||
9646 | const Type* SrcTy = | ||||
9647 | cast<PointerType>(CI->getOperand(0)->getType())->getElementType(); | ||||
9648 | const Type* DstTy = cast<PointerType>(CI->getType())->getElementType(); | ||||
9649 | if (!SrcTy->isSized() || !DstTy->isSized()) | ||||
9650 | return false; | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 9651 | if (TD->getTypePaddedSize(SrcTy) != TD->getTypePaddedSize(DstTy)) |
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9652 | return false; |
9653 | return true; | ||||
9654 | } | ||||
9655 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9656 | // visitCallSite - Improvements for call and invoke instructions. |
9657 | // | ||||
9658 | Instruction *InstCombiner::visitCallSite(CallSite CS) { | ||||
9659 | bool Changed = false; | ||||
9660 | |||||
9661 | // If the callee is a constexpr cast of a function, attempt to move the cast | ||||
9662 | // to the arguments of the call/invoke. | ||||
9663 | if (transformConstExprCastCall(CS)) return 0; | ||||
9664 | |||||
9665 | Value *Callee = CS.getCalledValue(); | ||||
9666 | |||||
9667 | if (Function *CalleeF = dyn_cast<Function>(Callee)) | ||||
9668 | if (CalleeF->getCallingConv() != CS.getCallingConv()) { | ||||
9669 | Instruction *OldCall = CS.getInstruction(); | ||||
9670 | // If the call and callee calling conventions don't match, this call must | ||||
9671 | // be unreachable, as the call is undefined. | ||||
9672 | new StoreInst(ConstantInt::getTrue(), | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 9673 | UndefValue::get(PointerType::getUnqual(Type::Int1Ty)), |
9674 | OldCall); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9675 | if (!OldCall->use_empty()) |
9676 | OldCall->replaceAllUsesWith(UndefValue::get(OldCall->getType())); | ||||
9677 | if (isa<CallInst>(OldCall)) // Not worth removing an invoke here. | ||||
9678 | return EraseInstFromFunction(*OldCall); | ||||
9679 | return 0; | ||||
9680 | } | ||||
9681 | |||||
9682 | if (isa<ConstantPointerNull>(Callee) || isa<UndefValue>(Callee)) { | ||||
9683 | // This instruction is not reachable, just remove it. We insert a store to | ||||
9684 | // undef so that we know that this code is not reachable, despite the fact | ||||
9685 | // that we can't modify the CFG here. | ||||
9686 | new StoreInst(ConstantInt::getTrue(), | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 9687 | UndefValue::get(PointerType::getUnqual(Type::Int1Ty)), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9688 | CS.getInstruction()); |
9689 | |||||
9690 | if (!CS.getInstruction()->use_empty()) | ||||
9691 | CS.getInstruction()-> | ||||
9692 | replaceAllUsesWith(UndefValue::get(CS.getInstruction()->getType())); | ||||
9693 | |||||
9694 | if (InvokeInst *II = dyn_cast<InvokeInst>(CS.getInstruction())) { | ||||
9695 | // Don't break the CFG, insert a dummy cond branch. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9696 | BranchInst::Create(II->getNormalDest(), II->getUnwindDest(), |
9697 | ConstantInt::getTrue(), II); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9698 | } |
9699 | return EraseInstFromFunction(*CS.getInstruction()); | ||||
9700 | } | ||||
9701 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9702 | if (BitCastInst *BC = dyn_cast<BitCastInst>(Callee)) |
9703 | if (IntrinsicInst *In = dyn_cast<IntrinsicInst>(BC->getOperand(0))) | ||||
9704 | if (In->getIntrinsicID() == Intrinsic::init_trampoline) | ||||
9705 | return transformCallThroughTrampoline(CS); | ||||
9706 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9707 | const PointerType *PTy = cast<PointerType>(Callee->getType()); |
9708 | const FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); | ||||
9709 | if (FTy->isVarArg()) { | ||||
Dale Johannesen | 502336c | 2008-04-23 01:03:05 +0000 | [diff] [blame] | 9710 | int ix = FTy->getNumParams() + (isa<InvokeInst>(Callee) ? 3 : 1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9711 | // See if we can optimize any arguments passed through the varargs area of |
9712 | // the call. | ||||
9713 | for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(), | ||||
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9714 | E = CS.arg_end(); I != E; ++I, ++ix) { |
9715 | CastInst *CI = dyn_cast<CastInst>(*I); | ||||
9716 | if (CI && isSafeToEliminateVarargsCast(CS, CI, TD, ix)) { | ||||
9717 | *I = CI->getOperand(0); | ||||
9718 | Changed = true; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9719 | } |
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9720 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9721 | } |
9722 | |||||
Duncan Sands | 2937e35 | 2007-12-19 21:13:37 +0000 | [diff] [blame] | 9723 | if (isa<InlineAsm>(Callee) && !CS.doesNotThrow()) { |
Duncan Sands | 7868f3c | 2007-12-16 15:51:49 +0000 | [diff] [blame] | 9724 | // Inline asm calls cannot throw - mark them 'nounwind'. |
Duncan Sands | 2937e35 | 2007-12-19 21:13:37 +0000 | [diff] [blame] | 9725 | CS.setDoesNotThrow(); |
Duncan Sands | 7868f3c | 2007-12-16 15:51:49 +0000 | [diff] [blame] | 9726 | Changed = true; |
9727 | } | ||||
9728 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9729 | return Changed ? CS.getInstruction() : 0; |
9730 | } | ||||
9731 | |||||
9732 | // transformConstExprCastCall - If the callee is a constexpr cast of a function, | ||||
9733 | // attempt to move the cast to the arguments of the call/invoke. | ||||
9734 | // | ||||
9735 | bool InstCombiner::transformConstExprCastCall(CallSite CS) { | ||||
9736 | if (!isa<ConstantExpr>(CS.getCalledValue())) return false; | ||||
9737 | ConstantExpr *CE = cast<ConstantExpr>(CS.getCalledValue()); | ||||
9738 | if (CE->getOpcode() != Instruction::BitCast || | ||||
9739 | !isa<Function>(CE->getOperand(0))) | ||||
9740 | return false; | ||||
9741 | Function *Callee = cast<Function>(CE->getOperand(0)); | ||||
9742 | Instruction *Caller = CS.getInstruction(); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9743 | const AttrListPtr &CallerPAL = CS.getAttributes(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9744 | |
9745 | // Okay, this is a cast from a function to a different type. Unless doing so | ||||
9746 | // would cause a type conversion of one of our arguments, change this call to | ||||
9747 | // be a direct call with arguments casted to the appropriate types. | ||||
9748 | // | ||||
9749 | const FunctionType *FT = Callee->getFunctionType(); | ||||
9750 | const Type *OldRetTy = Caller->getType(); | ||||
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9751 | const Type *NewRetTy = FT->getReturnType(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9752 | |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9753 | if (isa<StructType>(NewRetTy)) |
Devang Patel | d091d32 | 2008-03-11 18:04:06 +0000 | [diff] [blame] | 9754 | return false; // TODO: Handle multiple return values. |
9755 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9756 | // Check to see if we are changing the return type... |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9757 | if (OldRetTy != NewRetTy) { |
Bill Wendling | d9644a4 | 2008-05-14 22:45:20 +0000 | [diff] [blame] | 9758 | if (Callee->isDeclaration() && |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9759 | // Conversion is ok if changing from one pointer type to another or from |
9760 | // a pointer to an integer of the same size. | ||||
9761 | !((isa<PointerType>(OldRetTy) || OldRetTy == TD->getIntPtrType()) && | ||||
Duncan Sands | 886cadb | 2008-06-17 15:55:30 +0000 | [diff] [blame] | 9762 | (isa<PointerType>(NewRetTy) || NewRetTy == TD->getIntPtrType()))) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9763 | return false; // Cannot transform this return value. |
9764 | |||||
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9765 | if (!Caller->use_empty() && |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9766 | // void -> non-void is handled specially |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9767 | NewRetTy != Type::VoidTy && !CastInst::isCastable(NewRetTy, OldRetTy)) |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9768 | return false; // Cannot transform this return value. |
9769 | |||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9770 | if (!CallerPAL.isEmpty() && !Caller->use_empty()) { |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9771 | Attributes RAttrs = CallerPAL.getRetAttributes(); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9772 | if (RAttrs & Attribute::typeIncompatible(NewRetTy)) |
Duncan Sands | dbe97dc | 2008-01-07 17:16:06 +0000 | [diff] [blame] | 9773 | return false; // Attribute not compatible with transformed value. |
9774 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9775 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9776 | // If the callsite is an invoke instruction, and the return value is used by |
9777 | // a PHI node in a successor, we cannot change the return type of the call | ||||
9778 | // because there is no place to put the cast instruction (without breaking | ||||
9779 | // the critical edge). Bail out in this case. | ||||
9780 | if (!Caller->use_empty()) | ||||
9781 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) | ||||
9782 | for (Value::use_iterator UI = II->use_begin(), E = II->use_end(); | ||||
9783 | UI != E; ++UI) | ||||
9784 | if (PHINode *PN = dyn_cast<PHINode>(*UI)) | ||||
9785 | if (PN->getParent() == II->getNormalDest() || | ||||
9786 | PN->getParent() == II->getUnwindDest()) | ||||
9787 | return false; | ||||
9788 | } | ||||
9789 | |||||
9790 | unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin()); | ||||
9791 | unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs); | ||||
9792 | |||||
9793 | CallSite::arg_iterator AI = CS.arg_begin(); | ||||
9794 | for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) { | ||||
9795 | const Type *ParamTy = FT->getParamType(i); | ||||
9796 | const Type *ActTy = (*AI)->getType(); | ||||
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9797 | |
9798 | if (!CastInst::isCastable(ActTy, ParamTy)) | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9799 | return false; // Cannot transform this parameter value. |
9800 | |||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9801 | if (CallerPAL.getParamAttributes(i + 1) |
9802 | & Attribute::typeIncompatible(ParamTy)) | ||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9803 | return false; // Attribute not compatible with transformed value. |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9804 | |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9805 | // Converting from one pointer type to another or between a pointer and an |
9806 | // integer of the same size is safe even if we do not have a body. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9807 | bool isConvertible = ActTy == ParamTy || |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9808 | ((isa<PointerType>(ParamTy) || ParamTy == TD->getIntPtrType()) && |
9809 | (isa<PointerType>(ActTy) || ActTy == TD->getIntPtrType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9810 | if (Callee->isDeclaration() && !isConvertible) return false; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9811 | } |
9812 | |||||
9813 | if (FT->getNumParams() < NumActualArgs && !FT->isVarArg() && | ||||
9814 | Callee->isDeclaration()) | ||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9815 | return false; // Do not delete arguments unless we have a function body. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9816 | |
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9817 | if (FT->getNumParams() < NumActualArgs && FT->isVarArg() && |
9818 | !CallerPAL.isEmpty()) | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9819 | // In this case we have more arguments than the new function type, but we |
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9820 | // won't be dropping them. Check that these extra arguments have attributes |
9821 | // that are compatible with being a vararg call argument. | ||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9822 | for (unsigned i = CallerPAL.getNumSlots(); i; --i) { |
9823 | if (CallerPAL.getSlot(i - 1).Index <= FT->getNumParams()) | ||||
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9824 | break; |
Devang Patel | e480dfa | 2008-09-23 23:03:40 +0000 | [diff] [blame] | 9825 | Attributes PAttrs = CallerPAL.getSlot(i - 1).Attrs; |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9826 | if (PAttrs & Attribute::VarArgsIncompatible) |
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9827 | return false; |
9828 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9829 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9830 | // Okay, we decided that this is a safe thing to do: go ahead and start |
9831 | // inserting cast instructions as necessary... | ||||
9832 | std::vector<Value*> Args; | ||||
9833 | Args.reserve(NumActualArgs); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9834 | SmallVector<AttributeWithIndex, 8> attrVec; |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9835 | attrVec.reserve(NumCommonArgs); |
9836 | |||||
9837 | // Get any return attributes. | ||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9838 | Attributes RAttrs = CallerPAL.getRetAttributes(); |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9839 | |
9840 | // If the return value is not being used, the type may not be compatible | ||||
9841 | // with the existing attributes. Wipe out any problematic attributes. | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9842 | RAttrs &= ~Attribute::typeIncompatible(NewRetTy); |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9843 | |
9844 | // Add the new return attributes. | ||||
9845 | if (RAttrs) | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9846 | attrVec.push_back(AttributeWithIndex::get(0, RAttrs)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9847 | |
9848 | AI = CS.arg_begin(); | ||||
9849 | for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) { | ||||
9850 | const Type *ParamTy = FT->getParamType(i); | ||||
9851 | if ((*AI)->getType() == ParamTy) { | ||||
9852 | Args.push_back(*AI); | ||||
9853 | } else { | ||||
9854 | Instruction::CastOps opcode = CastInst::getCastOpcode(*AI, | ||||
9855 | false, ParamTy, false); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9856 | CastInst *NewCast = CastInst::Create(opcode, *AI, ParamTy, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9857 | Args.push_back(InsertNewInstBefore(NewCast, *Caller)); |
9858 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9859 | |
9860 | // Add any parameter attributes. | ||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9861 | if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9862 | attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9863 | } |
9864 | |||||
9865 | // If the function takes more arguments than the call was taking, add them | ||||
9866 | // now... | ||||
9867 | for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i) | ||||
9868 | Args.push_back(Constant::getNullValue(FT->getParamType(i))); | ||||
9869 | |||||
9870 | // If we are removing arguments to the function, emit an obnoxious warning... | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 9871 | if (FT->getNumParams() < NumActualArgs) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9872 | if (!FT->isVarArg()) { |
9873 | cerr << "WARNING: While resolving call to function '" | ||||
9874 | << Callee->getName() << "' arguments were dropped!\n"; | ||||
9875 | } else { | ||||
9876 | // Add all of the arguments in their promoted form to the arg list... | ||||
9877 | for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) { | ||||
9878 | const Type *PTy = getPromotedType((*AI)->getType()); | ||||
9879 | if (PTy != (*AI)->getType()) { | ||||
9880 | // Must promote to pass through va_arg area! | ||||
9881 | Instruction::CastOps opcode = CastInst::getCastOpcode(*AI, false, | ||||
9882 | PTy, false); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9883 | Instruction *Cast = CastInst::Create(opcode, *AI, PTy, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9884 | InsertNewInstBefore(Cast, *Caller); |
9885 | Args.push_back(Cast); | ||||
9886 | } else { | ||||
9887 | Args.push_back(*AI); | ||||
9888 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9889 | |
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9890 | // Add any parameter attributes. |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9891 | if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9892 | attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); |
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9893 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9894 | } |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 9895 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9896 | |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9897 | if (Attributes FnAttrs = CallerPAL.getFnAttributes()) |
9898 | attrVec.push_back(AttributeWithIndex::get(~0, FnAttrs)); | ||||
9899 | |||||
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9900 | if (NewRetTy == Type::VoidTy) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9901 | Caller->setName(""); // Void type should not have a name. |
9902 | |||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9903 | const AttrListPtr &NewCallerPAL = AttrListPtr::get(attrVec.begin(),attrVec.end()); |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9904 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9905 | Instruction *NC; |
9906 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9907 | NC = InvokeInst::Create(Callee, II->getNormalDest(), II->getUnwindDest(), |
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9908 | Args.begin(), Args.end(), |
9909 | Caller->getName(), Caller); | ||||
Reid Spencer | 6b0b09a | 2007-07-30 19:53:57 +0000 | [diff] [blame] | 9910 | cast<InvokeInst>(NC)->setCallingConv(II->getCallingConv()); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9911 | cast<InvokeInst>(NC)->setAttributes(NewCallerPAL); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9912 | } else { |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9913 | NC = CallInst::Create(Callee, Args.begin(), Args.end(), |
9914 | Caller->getName(), Caller); | ||||
Duncan Sands | f5588dc | 2007-11-27 13:23:08 +0000 | [diff] [blame] | 9915 | CallInst *CI = cast<CallInst>(Caller); |
9916 | if (CI->isTailCall()) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9917 | cast<CallInst>(NC)->setTailCall(); |
Duncan Sands | f5588dc | 2007-11-27 13:23:08 +0000 | [diff] [blame] | 9918 | cast<CallInst>(NC)->setCallingConv(CI->getCallingConv()); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9919 | cast<CallInst>(NC)->setAttributes(NewCallerPAL); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9920 | } |
9921 | |||||
9922 | // Insert a cast of the return type as necessary. | ||||
9923 | Value *NV = NC; | ||||
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9924 | if (OldRetTy != NV->getType() && !Caller->use_empty()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9925 | if (NV->getType() != Type::VoidTy) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9926 | Instruction::CastOps opcode = CastInst::getCastOpcode(NC, false, |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9927 | OldRetTy, false); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9928 | NV = NC = CastInst::Create(opcode, NC, OldRetTy, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9929 | |
9930 | // If this is an invoke instruction, we should insert it after the first | ||||
9931 | // non-phi, instruction in the normal successor block. | ||||
9932 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { | ||||
Dan Gohman | 514277c | 2008-05-23 21:05:58 +0000 | [diff] [blame] | 9933 | BasicBlock::iterator I = II->getNormalDest()->getFirstNonPHI(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9934 | InsertNewInstBefore(NC, *I); |
9935 | } else { | ||||
9936 | // Otherwise, it's a call, just insert cast right after the call instr | ||||
9937 | InsertNewInstBefore(NC, *Caller); | ||||
9938 | } | ||||
9939 | AddUsersToWorkList(*Caller); | ||||
9940 | } else { | ||||
9941 | NV = UndefValue::get(Caller->getType()); | ||||
9942 | } | ||||
9943 | } | ||||
9944 | |||||
9945 | if (Caller->getType() != Type::VoidTy && !Caller->use_empty()) | ||||
9946 | Caller->replaceAllUsesWith(NV); | ||||
9947 | Caller->eraseFromParent(); | ||||
9948 | RemoveFromWorkList(Caller); | ||||
9949 | return true; | ||||
9950 | } | ||||
9951 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9952 | // transformCallThroughTrampoline - Turn a call to a function created by the |
9953 | // init_trampoline intrinsic into a direct call to the underlying function. | ||||
9954 | // | ||||
9955 | Instruction *InstCombiner::transformCallThroughTrampoline(CallSite CS) { | ||||
9956 | Value *Callee = CS.getCalledValue(); | ||||
9957 | const PointerType *PTy = cast<PointerType>(Callee->getType()); | ||||
9958 | const FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9959 | const AttrListPtr &Attrs = CS.getAttributes(); |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9960 | |
9961 | // If the call already has the 'nest' attribute somewhere then give up - | ||||
9962 | // otherwise 'nest' would occur twice after splicing in the chain. | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9963 | if (Attrs.hasAttrSomewhere(Attribute::Nest)) |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9964 | return 0; |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9965 | |
9966 | IntrinsicInst *Tramp = | ||||
9967 | cast<IntrinsicInst>(cast<BitCastInst>(Callee)->getOperand(0)); | ||||
9968 | |||||
Anton Korobeynikov | 48fc88f | 2008-05-07 22:54:15 +0000 | [diff] [blame] | 9969 | Function *NestF = cast<Function>(Tramp->getOperand(2)->stripPointerCasts()); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9970 | const PointerType *NestFPTy = cast<PointerType>(NestF->getType()); |
9971 | const FunctionType *NestFTy = cast<FunctionType>(NestFPTy->getElementType()); | ||||
9972 | |||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9973 | const AttrListPtr &NestAttrs = NestF->getAttributes(); |
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9974 | if (!NestAttrs.isEmpty()) { |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9975 | unsigned NestIdx = 1; |
9976 | const Type *NestTy = 0; | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9977 | Attributes NestAttr = Attribute::None; |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9978 | |
9979 | // Look for a parameter marked with the 'nest' attribute. | ||||
9980 | for (FunctionType::param_iterator I = NestFTy->param_begin(), | ||||
9981 | E = NestFTy->param_end(); I != E; ++NestIdx, ++I) | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9982 | if (NestAttrs.paramHasAttr(NestIdx, Attribute::Nest)) { |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9983 | // Record the parameter type and any other attributes. |
9984 | NestTy = *I; | ||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9985 | NestAttr = NestAttrs.getParamAttributes(NestIdx); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9986 | break; |
9987 | } | ||||
9988 | |||||
9989 | if (NestTy) { | ||||
9990 | Instruction *Caller = CS.getInstruction(); | ||||
9991 | std::vector<Value*> NewArgs; | ||||
9992 | NewArgs.reserve(unsigned(CS.arg_end()-CS.arg_begin())+1); | ||||
9993 | |||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9994 | SmallVector<AttributeWithIndex, 8> NewAttrs; |
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9995 | NewAttrs.reserve(Attrs.getNumSlots() + 1); |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9996 | |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9997 | // Insert the nest argument into the call argument list, which may |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9998 | // mean appending it. Likewise for attributes. |
9999 | |||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 10000 | // Add any result attributes. |
10001 | if (Attributes Attr = Attrs.getRetAttributes()) | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 10002 | NewAttrs.push_back(AttributeWithIndex::get(0, Attr)); |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 10003 | |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10004 | { |
10005 | unsigned Idx = 1; | ||||
10006 | CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); | ||||
10007 | do { | ||||
10008 | if (Idx == NestIdx) { | ||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 10009 | // Add the chain argument and attributes. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10010 | Value *NestVal = Tramp->getOperand(3); |
10011 | if (NestVal->getType() != NestTy) | ||||
10012 | NestVal = new BitCastInst(NestVal, NestTy, "nest", Caller); | ||||
10013 | NewArgs.push_back(NestVal); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 10014 | NewAttrs.push_back(AttributeWithIndex::get(NestIdx, NestAttr)); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10015 | } |
10016 | |||||
10017 | if (I == E) | ||||
10018 | break; | ||||
10019 | |||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 10020 | // Add the original argument and attributes. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10021 | NewArgs.push_back(*I); |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 10022 | if (Attributes Attr = Attrs.getParamAttributes(Idx)) |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 10023 | NewAttrs.push_back |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 10024 | (AttributeWithIndex::get(Idx + (Idx >= NestIdx), Attr)); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10025 | |
10026 | ++Idx, ++I; | ||||
10027 | } while (1); | ||||
10028 | } | ||||
10029 | |||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 10030 | // Add any function attributes. |
10031 | if (Attributes Attr = Attrs.getFnAttributes()) | ||||
10032 | NewAttrs.push_back(AttributeWithIndex::get(~0, Attr)); | ||||
10033 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10034 | // The trampoline may have been bitcast to a bogus type (FTy). |
10035 | // Handle this by synthesizing a new function type, equal to FTy | ||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 10036 | // with the chain parameter inserted. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10037 | |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10038 | std::vector<const Type*> NewTypes; |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10039 | NewTypes.reserve(FTy->getNumParams()+1); |
10040 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10041 | // Insert the chain's type into the list of parameter types, which may |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 10042 | // mean appending it. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10043 | { |
10044 | unsigned Idx = 1; | ||||
10045 | FunctionType::param_iterator I = FTy->param_begin(), | ||||
10046 | E = FTy->param_end(); | ||||
10047 | |||||
10048 | do { | ||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 10049 | if (Idx == NestIdx) |
10050 | // Add the chain's type. | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10051 | NewTypes.push_back(NestTy); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10052 | |
10053 | if (I == E) | ||||
10054 | break; | ||||
10055 | |||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 10056 | // Add the original type. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10057 | NewTypes.push_back(*I); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10058 | |
10059 | ++Idx, ++I; | ||||
10060 | } while (1); | ||||
10061 | } | ||||
10062 | |||||
10063 | // Replace the trampoline call with a direct call. Let the generic | ||||
10064 | // code sort out any function type mismatches. | ||||
10065 | FunctionType *NewFTy = | ||||
Duncan Sands | f5588dc | 2007-11-27 13:23:08 +0000 | [diff] [blame] | 10066 | FunctionType::get(FTy->getReturnType(), NewTypes, FTy->isVarArg()); |
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 10067 | Constant *NewCallee = NestF->getType() == PointerType::getUnqual(NewFTy) ? |
10068 | NestF : ConstantExpr::getBitCast(NestF, PointerType::getUnqual(NewFTy)); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 10069 | const AttrListPtr &NewPAL = AttrListPtr::get(NewAttrs.begin(),NewAttrs.end()); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10070 | |
10071 | Instruction *NewCaller; | ||||
10072 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10073 | NewCaller = InvokeInst::Create(NewCallee, |
10074 | II->getNormalDest(), II->getUnwindDest(), | ||||
10075 | NewArgs.begin(), NewArgs.end(), | ||||
10076 | Caller->getName(), Caller); | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10077 | cast<InvokeInst>(NewCaller)->setCallingConv(II->getCallingConv()); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 10078 | cast<InvokeInst>(NewCaller)->setAttributes(NewPAL); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10079 | } else { |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10080 | NewCaller = CallInst::Create(NewCallee, NewArgs.begin(), NewArgs.end(), |
10081 | Caller->getName(), Caller); | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10082 | if (cast<CallInst>(Caller)->isTailCall()) |
10083 | cast<CallInst>(NewCaller)->setTailCall(); | ||||
10084 | cast<CallInst>(NewCaller)-> | ||||
10085 | setCallingConv(cast<CallInst>(Caller)->getCallingConv()); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 10086 | cast<CallInst>(NewCaller)->setAttributes(NewPAL); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10087 | } |
10088 | if (Caller->getType() != Type::VoidTy && !Caller->use_empty()) | ||||
10089 | Caller->replaceAllUsesWith(NewCaller); | ||||
10090 | Caller->eraseFromParent(); | ||||
10091 | RemoveFromWorkList(Caller); | ||||
10092 | return 0; | ||||
10093 | } | ||||
10094 | } | ||||
10095 | |||||
10096 | // Replace the trampoline call with a direct call. Since there is no 'nest' | ||||
10097 | // parameter, there is no need to adjust the argument list. Let the generic | ||||
10098 | // code sort out any function type mismatches. | ||||
10099 | Constant *NewCallee = | ||||
10100 | NestF->getType() == PTy ? NestF : ConstantExpr::getBitCast(NestF, PTy); | ||||
10101 | CS.setCalledFunction(NewCallee); | ||||
10102 | return CS.getInstruction(); | ||||
10103 | } | ||||
10104 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10105 | /// FoldPHIArgBinOpIntoPHI - If we have something like phi [add (a,b), add(c,d)] |
10106 | /// and if a/b/c/d and the add's all have a single use, turn this into two phi's | ||||
10107 | /// and a single binop. | ||||
10108 | Instruction *InstCombiner::FoldPHIArgBinOpIntoPHI(PHINode &PN) { | ||||
10109 | Instruction *FirstInst = cast<Instruction>(PN.getIncomingValue(0)); | ||||
Chris Lattner | 3007801 | 2008-12-01 03:42:51 +0000 | [diff] [blame] | 10110 | assert(isa<BinaryOperator>(FirstInst) || isa<CmpInst>(FirstInst)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10111 | unsigned Opc = FirstInst->getOpcode(); |
10112 | Value *LHSVal = FirstInst->getOperand(0); | ||||
10113 | Value *RHSVal = FirstInst->getOperand(1); | ||||
10114 | |||||
10115 | const Type *LHSType = LHSVal->getType(); | ||||
10116 | const Type *RHSType = RHSVal->getType(); | ||||
10117 | |||||
10118 | // Scan to see if all operands are the same opcode, all have one use, and all | ||||
10119 | // kill their operands (i.e. the operands have one use). | ||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10120 | for (unsigned i = 1; i != PN.getNumIncomingValues(); ++i) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10121 | Instruction *I = dyn_cast<Instruction>(PN.getIncomingValue(i)); |
10122 | if (!I || I->getOpcode() != Opc || !I->hasOneUse() || | ||||
10123 | // Verify type of the LHS matches so we don't fold cmp's of different | ||||
10124 | // types or GEP's with different index types. | ||||
10125 | I->getOperand(0)->getType() != LHSType || | ||||
10126 | I->getOperand(1)->getType() != RHSType) | ||||
10127 | return 0; | ||||
10128 | |||||
10129 | // If they are CmpInst instructions, check their predicates | ||||
10130 | if (Opc == Instruction::ICmp || Opc == Instruction::FCmp) | ||||
10131 | if (cast<CmpInst>(I)->getPredicate() != | ||||
10132 | cast<CmpInst>(FirstInst)->getPredicate()) | ||||
10133 | return 0; | ||||
10134 | |||||
10135 | // Keep track of which operand needs a phi node. | ||||
10136 | if (I->getOperand(0) != LHSVal) LHSVal = 0; | ||||
10137 | if (I->getOperand(1) != RHSVal) RHSVal = 0; | ||||
10138 | } | ||||
10139 | |||||
Chris Lattner | 3007801 | 2008-12-01 03:42:51 +0000 | [diff] [blame] | 10140 | // Otherwise, this is safe to transform! |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10141 | |
10142 | Value *InLHS = FirstInst->getOperand(0); | ||||
10143 | Value *InRHS = FirstInst->getOperand(1); | ||||
10144 | PHINode *NewLHS = 0, *NewRHS = 0; | ||||
10145 | if (LHSVal == 0) { | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 10146 | NewLHS = PHINode::Create(LHSType, |
10147 | FirstInst->getOperand(0)->getName() + ".pn"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10148 | NewLHS->reserveOperandSpace(PN.getNumOperands()/2); |
10149 | NewLHS->addIncoming(InLHS, PN.getIncomingBlock(0)); | ||||
10150 | InsertNewInstBefore(NewLHS, PN); | ||||
10151 | LHSVal = NewLHS; | ||||
10152 | } | ||||
10153 | |||||
10154 | if (RHSVal == 0) { | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 10155 | NewRHS = PHINode::Create(RHSType, |
10156 | FirstInst->getOperand(1)->getName() + ".pn"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10157 | NewRHS->reserveOperandSpace(PN.getNumOperands()/2); |
10158 | NewRHS->addIncoming(InRHS, PN.getIncomingBlock(0)); | ||||
10159 | InsertNewInstBefore(NewRHS, PN); | ||||
10160 | RHSVal = NewRHS; | ||||
10161 | } | ||||
10162 | |||||
10163 | // Add all operands to the new PHIs. | ||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10164 | if (NewLHS || NewRHS) { |
10165 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
10166 | Instruction *InInst = cast<Instruction>(PN.getIncomingValue(i)); | ||||
10167 | if (NewLHS) { | ||||
10168 | Value *NewInLHS = InInst->getOperand(0); | ||||
10169 | NewLHS->addIncoming(NewInLHS, PN.getIncomingBlock(i)); | ||||
10170 | } | ||||
10171 | if (NewRHS) { | ||||
10172 | Value *NewInRHS = InInst->getOperand(1); | ||||
10173 | NewRHS->addIncoming(NewInRHS, PN.getIncomingBlock(i)); | ||||
10174 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10175 | } |
10176 | } | ||||
10177 | |||||
10178 | if (BinaryOperator *BinOp = dyn_cast<BinaryOperator>(FirstInst)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10179 | return BinaryOperator::Create(BinOp->getOpcode(), LHSVal, RHSVal); |
Chris Lattner | 3007801 | 2008-12-01 03:42:51 +0000 | [diff] [blame] | 10180 | CmpInst *CIOp = cast<CmpInst>(FirstInst); |
10181 | return CmpInst::Create(CIOp->getOpcode(), CIOp->getPredicate(), LHSVal, | ||||
10182 | RHSVal); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10183 | } |
10184 | |||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10185 | Instruction *InstCombiner::FoldPHIArgGEPIntoPHI(PHINode &PN) { |
10186 | GetElementPtrInst *FirstInst =cast<GetElementPtrInst>(PN.getIncomingValue(0)); | ||||
10187 | |||||
10188 | SmallVector<Value*, 16> FixedOperands(FirstInst->op_begin(), | ||||
10189 | FirstInst->op_end()); | ||||
Chris Lattner | adf354b | 2009-02-21 00:46:50 +0000 | [diff] [blame] | 10190 | // This is true if all GEP bases are allocas and if all indices into them are |
10191 | // constants. | ||||
10192 | bool AllBasePointersAreAllocas = true; | ||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10193 | |
10194 | // Scan to see if all operands are the same opcode, all have one use, and all | ||||
10195 | // kill their operands (i.e. the operands have one use). | ||||
10196 | for (unsigned i = 1; i != PN.getNumIncomingValues(); ++i) { | ||||
10197 | GetElementPtrInst *GEP= dyn_cast<GetElementPtrInst>(PN.getIncomingValue(i)); | ||||
10198 | if (!GEP || !GEP->hasOneUse() || GEP->getType() != FirstInst->getType() || | ||||
10199 | GEP->getNumOperands() != FirstInst->getNumOperands()) | ||||
10200 | return 0; | ||||
10201 | |||||
Chris Lattner | adf354b | 2009-02-21 00:46:50 +0000 | [diff] [blame] | 10202 | // Keep track of whether or not all GEPs are of alloca pointers. |
10203 | if (AllBasePointersAreAllocas && | ||||
10204 | (!isa<AllocaInst>(GEP->getOperand(0)) || | ||||
10205 | !GEP->hasAllConstantIndices())) | ||||
10206 | AllBasePointersAreAllocas = false; | ||||
10207 | |||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10208 | // Compare the operand lists. |
10209 | for (unsigned op = 0, e = FirstInst->getNumOperands(); op != e; ++op) { | ||||
10210 | if (FirstInst->getOperand(op) == GEP->getOperand(op)) | ||||
10211 | continue; | ||||
10212 | |||||
10213 | // Don't merge two GEPs when two operands differ (introducing phi nodes) | ||||
10214 | // if one of the PHIs has a constant for the index. The index may be | ||||
10215 | // substantially cheaper to compute for the constants, so making it a | ||||
10216 | // variable index could pessimize the path. This also handles the case | ||||
10217 | // for struct indices, which must always be constant. | ||||
10218 | if (isa<ConstantInt>(FirstInst->getOperand(op)) || | ||||
10219 | isa<ConstantInt>(GEP->getOperand(op))) | ||||
10220 | return 0; | ||||
10221 | |||||
10222 | if (FirstInst->getOperand(op)->getType() !=GEP->getOperand(op)->getType()) | ||||
10223 | return 0; | ||||
10224 | FixedOperands[op] = 0; // Needs a PHI. | ||||
10225 | } | ||||
10226 | } | ||||
10227 | |||||
Chris Lattner | adf354b | 2009-02-21 00:46:50 +0000 | [diff] [blame] | 10228 | // If all of the base pointers of the PHI'd GEPs are from allocas, don't |
Chris Lattner | f1e30c8 | 2009-02-23 05:56:17 +0000 | [diff] [blame] | 10229 | // bother doing this transformation. At best, this will just save a bit of |
Chris Lattner | adf354b | 2009-02-21 00:46:50 +0000 | [diff] [blame] | 10230 | // offset calculation, but all the predecessors will have to materialize the |
10231 | // stack address into a register anyway. We'd actually rather *clone* the | ||||
10232 | // load up into the predecessors so that we have a load of a gep of an alloca, | ||||
10233 | // which can usually all be folded into the load. | ||||
10234 | if (AllBasePointersAreAllocas) | ||||
10235 | return 0; | ||||
10236 | |||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10237 | // Otherwise, this is safe to transform. Insert PHI nodes for each operand |
10238 | // that is variable. | ||||
10239 | SmallVector<PHINode*, 16> OperandPhis(FixedOperands.size()); | ||||
10240 | |||||
10241 | bool HasAnyPHIs = false; | ||||
10242 | for (unsigned i = 0, e = FixedOperands.size(); i != e; ++i) { | ||||
10243 | if (FixedOperands[i]) continue; // operand doesn't need a phi. | ||||
10244 | Value *FirstOp = FirstInst->getOperand(i); | ||||
10245 | PHINode *NewPN = PHINode::Create(FirstOp->getType(), | ||||
10246 | FirstOp->getName()+".pn"); | ||||
10247 | InsertNewInstBefore(NewPN, PN); | ||||
10248 | |||||
10249 | NewPN->reserveOperandSpace(e); | ||||
10250 | NewPN->addIncoming(FirstOp, PN.getIncomingBlock(0)); | ||||
10251 | OperandPhis[i] = NewPN; | ||||
10252 | FixedOperands[i] = NewPN; | ||||
10253 | HasAnyPHIs = true; | ||||
10254 | } | ||||
10255 | |||||
10256 | |||||
10257 | // Add all operands to the new PHIs. | ||||
10258 | if (HasAnyPHIs) { | ||||
10259 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
10260 | GetElementPtrInst *InGEP =cast<GetElementPtrInst>(PN.getIncomingValue(i)); | ||||
10261 | BasicBlock *InBB = PN.getIncomingBlock(i); | ||||
10262 | |||||
10263 | for (unsigned op = 0, e = OperandPhis.size(); op != e; ++op) | ||||
10264 | if (PHINode *OpPhi = OperandPhis[op]) | ||||
10265 | OpPhi->addIncoming(InGEP->getOperand(op), InBB); | ||||
10266 | } | ||||
10267 | } | ||||
10268 | |||||
10269 | Value *Base = FixedOperands[0]; | ||||
10270 | return GetElementPtrInst::Create(Base, FixedOperands.begin()+1, | ||||
10271 | FixedOperands.end()); | ||||
10272 | } | ||||
10273 | |||||
10274 | |||||
Chris Lattner | f1e30c8 | 2009-02-23 05:56:17 +0000 | [diff] [blame] | 10275 | /// isSafeAndProfitableToSinkLoad - Return true if we know that it is safe to |
10276 | /// sink the load out of the block that defines it. This means that it must be | ||||
Chris Lattner | adf354b | 2009-02-21 00:46:50 +0000 | [diff] [blame] | 10277 | /// obvious the value of the load is not changed from the point of the load to |
10278 | /// the end of the block it is in. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10279 | /// |
10280 | /// Finally, it is safe, but not profitable, to sink a load targetting a | ||||
10281 | /// non-address-taken alloca. Doing so will cause us to not promote the alloca | ||||
10282 | /// to a register. | ||||
Chris Lattner | adf354b | 2009-02-21 00:46:50 +0000 | [diff] [blame] | 10283 | static bool isSafeAndProfitableToSinkLoad(LoadInst *L) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10284 | BasicBlock::iterator BBI = L, E = L->getParent()->end(); |
10285 | |||||
10286 | for (++BBI; BBI != E; ++BBI) | ||||
10287 | if (BBI->mayWriteToMemory()) | ||||
10288 | return false; | ||||
10289 | |||||
10290 | // Check for non-address taken alloca. If not address-taken already, it isn't | ||||
10291 | // profitable to do this xform. | ||||
10292 | if (AllocaInst *AI = dyn_cast<AllocaInst>(L->getOperand(0))) { | ||||
10293 | bool isAddressTaken = false; | ||||
10294 | for (Value::use_iterator UI = AI->use_begin(), E = AI->use_end(); | ||||
10295 | UI != E; ++UI) { | ||||
10296 | if (isa<LoadInst>(UI)) continue; | ||||
10297 | if (StoreInst *SI = dyn_cast<StoreInst>(*UI)) { | ||||
10298 | // If storing TO the alloca, then the address isn't taken. | ||||
10299 | if (SI->getOperand(1) == AI) continue; | ||||
10300 | } | ||||
10301 | isAddressTaken = true; | ||||
10302 | break; | ||||
10303 | } | ||||
10304 | |||||
Chris Lattner | adf354b | 2009-02-21 00:46:50 +0000 | [diff] [blame] | 10305 | if (!isAddressTaken && AI->isStaticAlloca()) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10306 | return false; |
10307 | } | ||||
10308 | |||||
Chris Lattner | adf354b | 2009-02-21 00:46:50 +0000 | [diff] [blame] | 10309 | // If this load is a load from a GEP with a constant offset from an alloca, |
10310 | // then we don't want to sink it. In its present form, it will be | ||||
10311 | // load [constant stack offset]. Sinking it will cause us to have to | ||||
10312 | // materialize the stack addresses in each predecessor in a register only to | ||||
10313 | // do a shared load from register in the successor. | ||||
10314 | if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(L->getOperand(0))) | ||||
10315 | if (AllocaInst *AI = dyn_cast<AllocaInst>(GEP->getOperand(0))) | ||||
10316 | if (AI->isStaticAlloca() && GEP->hasAllConstantIndices()) | ||||
10317 | return false; | ||||
10318 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10319 | return true; |
10320 | } | ||||
10321 | |||||
10322 | |||||
10323 | // FoldPHIArgOpIntoPHI - If all operands to a PHI node are the same "unary" | ||||
10324 | // operator and they all are only used by the PHI, PHI together their | ||||
10325 | // inputs, and do the operation once, to the result of the PHI. | ||||
10326 | Instruction *InstCombiner::FoldPHIArgOpIntoPHI(PHINode &PN) { | ||||
10327 | Instruction *FirstInst = cast<Instruction>(PN.getIncomingValue(0)); | ||||
10328 | |||||
10329 | // Scan the instruction, looking for input operations that can be folded away. | ||||
10330 | // If all input operands to the phi are the same instruction (e.g. a cast from | ||||
10331 | // the same type or "+42") we can pull the operation through the PHI, reducing | ||||
10332 | // code size and simplifying code. | ||||
10333 | Constant *ConstantOp = 0; | ||||
10334 | const Type *CastSrcTy = 0; | ||||
10335 | bool isVolatile = false; | ||||
10336 | if (isa<CastInst>(FirstInst)) { | ||||
10337 | CastSrcTy = FirstInst->getOperand(0)->getType(); | ||||
10338 | } else if (isa<BinaryOperator>(FirstInst) || isa<CmpInst>(FirstInst)) { | ||||
10339 | // Can fold binop, compare or shift here if the RHS is a constant, | ||||
10340 | // otherwise call FoldPHIArgBinOpIntoPHI. | ||||
10341 | ConstantOp = dyn_cast<Constant>(FirstInst->getOperand(1)); | ||||
10342 | if (ConstantOp == 0) | ||||
10343 | return FoldPHIArgBinOpIntoPHI(PN); | ||||
10344 | } else if (LoadInst *LI = dyn_cast<LoadInst>(FirstInst)) { | ||||
10345 | isVolatile = LI->isVolatile(); | ||||
10346 | // We can't sink the load if the loaded value could be modified between the | ||||
10347 | // load and the PHI. | ||||
10348 | if (LI->getParent() != PN.getIncomingBlock(0) || | ||||
Chris Lattner | adf354b | 2009-02-21 00:46:50 +0000 | [diff] [blame] | 10349 | !isSafeAndProfitableToSinkLoad(LI)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10350 | return 0; |
Chris Lattner | 2d9fdd8 | 2008-07-08 17:18:32 +0000 | [diff] [blame] | 10351 | |
10352 | // If the PHI is of volatile loads and the load block has multiple | ||||
10353 | // successors, sinking it would remove a load of the volatile value from | ||||
10354 | // the path through the other successor. | ||||
10355 | if (isVolatile && | ||||
10356 | LI->getParent()->getTerminator()->getNumSuccessors() != 1) | ||||
10357 | return 0; | ||||
10358 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10359 | } else if (isa<GetElementPtrInst>(FirstInst)) { |
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10360 | return FoldPHIArgGEPIntoPHI(PN); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10361 | } else { |
10362 | return 0; // Cannot fold this operation. | ||||
10363 | } | ||||
10364 | |||||
10365 | // Check to see if all arguments are the same operation. | ||||
10366 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
10367 | if (!isa<Instruction>(PN.getIncomingValue(i))) return 0; | ||||
10368 | Instruction *I = cast<Instruction>(PN.getIncomingValue(i)); | ||||
10369 | if (!I->hasOneUse() || !I->isSameOperationAs(FirstInst)) | ||||
10370 | return 0; | ||||
10371 | if (CastSrcTy) { | ||||
10372 | if (I->getOperand(0)->getType() != CastSrcTy) | ||||
10373 | return 0; // Cast operation must match. | ||||
10374 | } else if (LoadInst *LI = dyn_cast<LoadInst>(I)) { | ||||
10375 | // We can't sink the load if the loaded value could be modified between | ||||
10376 | // the load and the PHI. | ||||
10377 | if (LI->isVolatile() != isVolatile || | ||||
10378 | LI->getParent() != PN.getIncomingBlock(i) || | ||||
Chris Lattner | adf354b | 2009-02-21 00:46:50 +0000 | [diff] [blame] | 10379 | !isSafeAndProfitableToSinkLoad(LI)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10380 | return 0; |
Chris Lattner | f786701 | 2008-04-29 17:28:22 +0000 | [diff] [blame] | 10381 | |
Chris Lattner | 2d9fdd8 | 2008-07-08 17:18:32 +0000 | [diff] [blame] | 10382 | // If the PHI is of volatile loads and the load block has multiple |
10383 | // successors, sinking it would remove a load of the volatile value from | ||||
10384 | // the path through the other successor. | ||||
Chris Lattner | f786701 | 2008-04-29 17:28:22 +0000 | [diff] [blame] | 10385 | if (isVolatile && |
10386 | LI->getParent()->getTerminator()->getNumSuccessors() != 1) | ||||
10387 | return 0; | ||||
Chris Lattner | f786701 | 2008-04-29 17:28:22 +0000 | [diff] [blame] | 10388 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10389 | } else if (I->getOperand(1) != ConstantOp) { |
10390 | return 0; | ||||
10391 | } | ||||
10392 | } | ||||
10393 | |||||
10394 | // Okay, they are all the same operation. Create a new PHI node of the | ||||
10395 | // correct type, and PHI together all of the LHS's of the instructions. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10396 | PHINode *NewPN = PHINode::Create(FirstInst->getOperand(0)->getType(), |
10397 | PN.getName()+".in"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10398 | NewPN->reserveOperandSpace(PN.getNumOperands()/2); |
10399 | |||||
10400 | Value *InVal = FirstInst->getOperand(0); | ||||
10401 | NewPN->addIncoming(InVal, PN.getIncomingBlock(0)); | ||||
10402 | |||||
10403 | // Add all operands to the new PHI. | ||||
10404 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
10405 | Value *NewInVal = cast<Instruction>(PN.getIncomingValue(i))->getOperand(0); | ||||
10406 | if (NewInVal != InVal) | ||||
10407 | InVal = 0; | ||||
10408 | NewPN->addIncoming(NewInVal, PN.getIncomingBlock(i)); | ||||
10409 | } | ||||
10410 | |||||
10411 | Value *PhiVal; | ||||
10412 | if (InVal) { | ||||
10413 | // The new PHI unions all of the same values together. This is really | ||||
10414 | // common, so we handle it intelligently here for compile-time speed. | ||||
10415 | PhiVal = InVal; | ||||
10416 | delete NewPN; | ||||
10417 | } else { | ||||
10418 | InsertNewInstBefore(NewPN, PN); | ||||
10419 | PhiVal = NewPN; | ||||
10420 | } | ||||
10421 | |||||
10422 | // Insert and return the new operation. | ||||
10423 | if (CastInst* FirstCI = dyn_cast<CastInst>(FirstInst)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10424 | return CastInst::Create(FirstCI->getOpcode(), PhiVal, PN.getType()); |
Chris Lattner | fc984e9 | 2008-04-29 17:13:43 +0000 | [diff] [blame] | 10425 | if (BinaryOperator *BinOp = dyn_cast<BinaryOperator>(FirstInst)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10426 | return BinaryOperator::Create(BinOp->getOpcode(), PhiVal, ConstantOp); |
Chris Lattner | fc984e9 | 2008-04-29 17:13:43 +0000 | [diff] [blame] | 10427 | if (CmpInst *CIOp = dyn_cast<CmpInst>(FirstInst)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10428 | return CmpInst::Create(CIOp->getOpcode(), CIOp->getPredicate(), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10429 | PhiVal, ConstantOp); |
Chris Lattner | fc984e9 | 2008-04-29 17:13:43 +0000 | [diff] [blame] | 10430 | assert(isa<LoadInst>(FirstInst) && "Unknown operation"); |
10431 | |||||
10432 | // If this was a volatile load that we are merging, make sure to loop through | ||||
10433 | // and mark all the input loads as non-volatile. If we don't do this, we will | ||||
10434 | // insert a new volatile load and the old ones will not be deletable. | ||||
10435 | if (isVolatile) | ||||
10436 | for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) | ||||
10437 | cast<LoadInst>(PN.getIncomingValue(i))->setVolatile(false); | ||||
10438 | |||||
10439 | return new LoadInst(PhiVal, "", isVolatile); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10440 | } |
10441 | |||||
10442 | /// DeadPHICycle - Return true if this PHI node is only used by a PHI node cycle | ||||
10443 | /// that is dead. | ||||
10444 | static bool DeadPHICycle(PHINode *PN, | ||||
10445 | SmallPtrSet<PHINode*, 16> &PotentiallyDeadPHIs) { | ||||
10446 | if (PN->use_empty()) return true; | ||||
10447 | if (!PN->hasOneUse()) return false; | ||||
10448 | |||||
10449 | // Remember this node, and if we find the cycle, return. | ||||
10450 | if (!PotentiallyDeadPHIs.insert(PN)) | ||||
10451 | return true; | ||||
Chris Lattner | adf2e34 | 2007-08-28 04:23:55 +0000 | [diff] [blame] | 10452 | |
10453 | // Don't scan crazily complex things. | ||||
10454 | if (PotentiallyDeadPHIs.size() == 16) | ||||
10455 | return false; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10456 | |
10457 | if (PHINode *PU = dyn_cast<PHINode>(PN->use_back())) | ||||
10458 | return DeadPHICycle(PU, PotentiallyDeadPHIs); | ||||
10459 | |||||
10460 | return false; | ||||
10461 | } | ||||
10462 | |||||
Chris Lattner | 27b695d | 2007-11-06 21:52:06 +0000 | [diff] [blame] | 10463 | /// PHIsEqualValue - Return true if this phi node is always equal to |
10464 | /// NonPhiInVal. This happens with mutually cyclic phi nodes like: | ||||
10465 | /// z = some value; x = phi (y, z); y = phi (x, z) | ||||
10466 | static bool PHIsEqualValue(PHINode *PN, Value *NonPhiInVal, | ||||
10467 | SmallPtrSet<PHINode*, 16> &ValueEqualPHIs) { | ||||
10468 | // See if we already saw this PHI node. | ||||
10469 | if (!ValueEqualPHIs.insert(PN)) | ||||
10470 | return true; | ||||
10471 | |||||
10472 | // Don't scan crazily complex things. | ||||
10473 | if (ValueEqualPHIs.size() == 16) | ||||
10474 | return false; | ||||
10475 | |||||
10476 | // Scan the operands to see if they are either phi nodes or are equal to | ||||
10477 | // the value. | ||||
10478 | for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) { | ||||
10479 | Value *Op = PN->getIncomingValue(i); | ||||
10480 | if (PHINode *OpPN = dyn_cast<PHINode>(Op)) { | ||||
10481 | if (!PHIsEqualValue(OpPN, NonPhiInVal, ValueEqualPHIs)) | ||||
10482 | return false; | ||||
10483 | } else if (Op != NonPhiInVal) | ||||
10484 | return false; | ||||
10485 | } | ||||
10486 | |||||
10487 | return true; | ||||
10488 | } | ||||
10489 | |||||
10490 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10491 | // PHINode simplification |
10492 | // | ||||
10493 | Instruction *InstCombiner::visitPHINode(PHINode &PN) { | ||||
10494 | // If LCSSA is around, don't mess with Phi nodes | ||||
10495 | if (MustPreserveLCSSA) return 0; | ||||
10496 | |||||
10497 | if (Value *V = PN.hasConstantValue()) | ||||
10498 | return ReplaceInstUsesWith(PN, V); | ||||
10499 | |||||
10500 | // If all PHI operands are the same operation, pull them through the PHI, | ||||
10501 | // reducing code size. | ||||
10502 | if (isa<Instruction>(PN.getIncomingValue(0)) && | ||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10503 | isa<Instruction>(PN.getIncomingValue(1)) && |
10504 | cast<Instruction>(PN.getIncomingValue(0))->getOpcode() == | ||||
10505 | cast<Instruction>(PN.getIncomingValue(1))->getOpcode() && | ||||
10506 | // FIXME: The hasOneUse check will fail for PHIs that use the value more | ||||
10507 | // than themselves more than once. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10508 | PN.getIncomingValue(0)->hasOneUse()) |
10509 | if (Instruction *Result = FoldPHIArgOpIntoPHI(PN)) | ||||
10510 | return Result; | ||||
10511 | |||||
10512 | // If this is a trivial cycle in the PHI node graph, remove it. Basically, if | ||||
10513 | // this PHI only has a single use (a PHI), and if that PHI only has one use (a | ||||
10514 | // PHI)... break the cycle. | ||||
10515 | if (PN.hasOneUse()) { | ||||
10516 | Instruction *PHIUser = cast<Instruction>(PN.use_back()); | ||||
10517 | if (PHINode *PU = dyn_cast<PHINode>(PHIUser)) { | ||||
10518 | SmallPtrSet<PHINode*, 16> PotentiallyDeadPHIs; | ||||
10519 | PotentiallyDeadPHIs.insert(&PN); | ||||
10520 | if (DeadPHICycle(PU, PotentiallyDeadPHIs)) | ||||
10521 | return ReplaceInstUsesWith(PN, UndefValue::get(PN.getType())); | ||||
10522 | } | ||||
10523 | |||||
10524 | // If this phi has a single use, and if that use just computes a value for | ||||
10525 | // the next iteration of a loop, delete the phi. This occurs with unused | ||||
10526 | // induction variables, e.g. "for (int j = 0; ; ++j);". Detecting this | ||||
10527 | // common case here is good because the only other things that catch this | ||||
10528 | // are induction variable analysis (sometimes) and ADCE, which is only run | ||||
10529 | // late. | ||||
10530 | if (PHIUser->hasOneUse() && | ||||
10531 | (isa<BinaryOperator>(PHIUser) || isa<GetElementPtrInst>(PHIUser)) && | ||||
10532 | PHIUser->use_back() == &PN) { | ||||
10533 | return ReplaceInstUsesWith(PN, UndefValue::get(PN.getType())); | ||||
10534 | } | ||||
10535 | } | ||||
10536 | |||||
Chris Lattner | 27b695d | 2007-11-06 21:52:06 +0000 | [diff] [blame] | 10537 | // We sometimes end up with phi cycles that non-obviously end up being the |
10538 | // same value, for example: | ||||
10539 | // z = some value; x = phi (y, z); y = phi (x, z) | ||||
10540 | // where the phi nodes don't necessarily need to be in the same block. Do a | ||||
10541 | // quick check to see if the PHI node only contains a single non-phi value, if | ||||
10542 | // so, scan to see if the phi cycle is actually equal to that value. | ||||
10543 | { | ||||
10544 | unsigned InValNo = 0, NumOperandVals = PN.getNumIncomingValues(); | ||||
10545 | // Scan for the first non-phi operand. | ||||
10546 | while (InValNo != NumOperandVals && | ||||
10547 | isa<PHINode>(PN.getIncomingValue(InValNo))) | ||||
10548 | ++InValNo; | ||||
10549 | |||||
10550 | if (InValNo != NumOperandVals) { | ||||
10551 | Value *NonPhiInVal = PN.getOperand(InValNo); | ||||
10552 | |||||
10553 | // Scan the rest of the operands to see if there are any conflicts, if so | ||||
10554 | // there is no need to recursively scan other phis. | ||||
10555 | for (++InValNo; InValNo != NumOperandVals; ++InValNo) { | ||||
10556 | Value *OpVal = PN.getIncomingValue(InValNo); | ||||
10557 | if (OpVal != NonPhiInVal && !isa<PHINode>(OpVal)) | ||||
10558 | break; | ||||
10559 | } | ||||
10560 | |||||
10561 | // If we scanned over all operands, then we have one unique value plus | ||||
10562 | // phi values. Scan PHI nodes to see if they all merge in each other or | ||||
10563 | // the value. | ||||
10564 | if (InValNo == NumOperandVals) { | ||||
10565 | SmallPtrSet<PHINode*, 16> ValueEqualPHIs; | ||||
10566 | if (PHIsEqualValue(&PN, NonPhiInVal, ValueEqualPHIs)) | ||||
10567 | return ReplaceInstUsesWith(PN, NonPhiInVal); | ||||
10568 | } | ||||
10569 | } | ||||
10570 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10571 | return 0; |
10572 | } | ||||
10573 | |||||
10574 | static Value *InsertCastToIntPtrTy(Value *V, const Type *DTy, | ||||
10575 | Instruction *InsertPoint, | ||||
10576 | InstCombiner *IC) { | ||||
10577 | unsigned PtrSize = DTy->getPrimitiveSizeInBits(); | ||||
10578 | unsigned VTySize = V->getType()->getPrimitiveSizeInBits(); | ||||
10579 | // We must cast correctly to the pointer type. Ensure that we | ||||
10580 | // sign extend the integer value if it is smaller as this is | ||||
10581 | // used for address computation. | ||||
10582 | Instruction::CastOps opcode = | ||||
10583 | (VTySize < PtrSize ? Instruction::SExt : | ||||
10584 | (VTySize == PtrSize ? Instruction::BitCast : Instruction::Trunc)); | ||||
10585 | return IC->InsertCastBefore(opcode, V, DTy, *InsertPoint); | ||||
10586 | } | ||||
10587 | |||||
10588 | |||||
10589 | Instruction *InstCombiner::visitGetElementPtrInst(GetElementPtrInst &GEP) { | ||||
10590 | Value *PtrOp = GEP.getOperand(0); | ||||
10591 | // Is it 'getelementptr %P, i32 0' or 'getelementptr %P' | ||||
10592 | // If so, eliminate the noop. | ||||
10593 | if (GEP.getNumOperands() == 1) | ||||
10594 | return ReplaceInstUsesWith(GEP, PtrOp); | ||||
10595 | |||||
10596 | if (isa<UndefValue>(GEP.getOperand(0))) | ||||
10597 | return ReplaceInstUsesWith(GEP, UndefValue::get(GEP.getType())); | ||||
10598 | |||||
10599 | bool HasZeroPointerIndex = false; | ||||
10600 | if (Constant *C = dyn_cast<Constant>(GEP.getOperand(1))) | ||||
10601 | HasZeroPointerIndex = C->isNullValue(); | ||||
10602 | |||||
10603 | if (GEP.getNumOperands() == 2 && HasZeroPointerIndex) | ||||
10604 | return ReplaceInstUsesWith(GEP, PtrOp); | ||||
10605 | |||||
10606 | // Eliminate unneeded casts for indices. | ||||
10607 | bool MadeChange = false; | ||||
10608 | |||||
10609 | gep_type_iterator GTI = gep_type_begin(GEP); | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10610 | for (User::op_iterator i = GEP.op_begin() + 1, e = GEP.op_end(); |
10611 | i != e; ++i, ++GTI) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10612 | if (isa<SequentialType>(*GTI)) { |
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10613 | if (CastInst *CI = dyn_cast<CastInst>(*i)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10614 | if (CI->getOpcode() == Instruction::ZExt || |
10615 | CI->getOpcode() == Instruction::SExt) { | ||||
10616 | const Type *SrcTy = CI->getOperand(0)->getType(); | ||||
10617 | // We can eliminate a cast from i32 to i64 iff the target | ||||
10618 | // is a 32-bit pointer target. | ||||
10619 | if (SrcTy->getPrimitiveSizeInBits() >= TD->getPointerSizeInBits()) { | ||||
10620 | MadeChange = true; | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10621 | *i = CI->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10622 | } |
10623 | } | ||||
10624 | } | ||||
10625 | // If we are using a wider index than needed for this platform, shrink it | ||||
Dan Gohman | 5d639ed | 2008-09-11 23:06:38 +0000 | [diff] [blame] | 10626 | // to what we need. If narrower, sign-extend it to what we need. |
10627 | // If the incoming value needs a cast instruction, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10628 | // insert it. This explicit cast can make subsequent optimizations more |
10629 | // obvious. | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10630 | Value *Op = *i; |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10631 | if (TD->getTypeSizeInBits(Op->getType()) > TD->getPointerSizeInBits()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10632 | if (Constant *C = dyn_cast<Constant>(Op)) { |
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10633 | *i = ConstantExpr::getTrunc(C, TD->getIntPtrType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10634 | MadeChange = true; |
10635 | } else { | ||||
10636 | Op = InsertCastBefore(Instruction::Trunc, Op, TD->getIntPtrType(), | ||||
10637 | GEP); | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10638 | *i = Op; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10639 | MadeChange = true; |
10640 | } | ||||
Dan Gohman | 5d639ed | 2008-09-11 23:06:38 +0000 | [diff] [blame] | 10641 | } else if (TD->getTypeSizeInBits(Op->getType()) < TD->getPointerSizeInBits()) { |
10642 | if (Constant *C = dyn_cast<Constant>(Op)) { | ||||
10643 | *i = ConstantExpr::getSExt(C, TD->getIntPtrType()); | ||||
10644 | MadeChange = true; | ||||
10645 | } else { | ||||
10646 | Op = InsertCastBefore(Instruction::SExt, Op, TD->getIntPtrType(), | ||||
10647 | GEP); | ||||
10648 | *i = Op; | ||||
10649 | MadeChange = true; | ||||
10650 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10651 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10652 | } |
10653 | } | ||||
10654 | if (MadeChange) return &GEP; | ||||
10655 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10656 | // Combine Indices - If the source pointer to this getelementptr instruction |
10657 | // is a getelementptr instruction, combine the indices of the two | ||||
10658 | // getelementptr instructions into a single instruction. | ||||
10659 | // | ||||
10660 | SmallVector<Value*, 8> SrcGEPOperands; | ||||
10661 | if (User *Src = dyn_castGetElementPtr(PtrOp)) | ||||
10662 | SrcGEPOperands.append(Src->op_begin(), Src->op_end()); | ||||
10663 | |||||
10664 | if (!SrcGEPOperands.empty()) { | ||||
10665 | // Note that if our source is a gep chain itself that we wait for that | ||||
10666 | // chain to be resolved before we perform this transformation. This | ||||
10667 | // avoids us creating a TON of code in some cases. | ||||
10668 | // | ||||
10669 | if (isa<GetElementPtrInst>(SrcGEPOperands[0]) && | ||||
10670 | cast<Instruction>(SrcGEPOperands[0])->getNumOperands() == 2) | ||||
10671 | return 0; // Wait until our source is folded to completion. | ||||
10672 | |||||
10673 | SmallVector<Value*, 8> Indices; | ||||
10674 | |||||
10675 | // Find out whether the last index in the source GEP is a sequential idx. | ||||
10676 | bool EndsWithSequential = false; | ||||
10677 | for (gep_type_iterator I = gep_type_begin(*cast<User>(PtrOp)), | ||||
10678 | E = gep_type_end(*cast<User>(PtrOp)); I != E; ++I) | ||||
10679 | EndsWithSequential = !isa<StructType>(*I); | ||||
10680 | |||||
10681 | // Can we combine the two pointer arithmetics offsets? | ||||
10682 | if (EndsWithSequential) { | ||||
10683 | // Replace: gep (gep %P, long B), long A, ... | ||||
10684 | // With: T = long A+B; gep %P, T, ... | ||||
10685 | // | ||||
10686 | Value *Sum, *SO1 = SrcGEPOperands.back(), *GO1 = GEP.getOperand(1); | ||||
10687 | if (SO1 == Constant::getNullValue(SO1->getType())) { | ||||
10688 | Sum = GO1; | ||||
10689 | } else if (GO1 == Constant::getNullValue(GO1->getType())) { | ||||
10690 | Sum = SO1; | ||||
10691 | } else { | ||||
10692 | // If they aren't the same type, convert both to an integer of the | ||||
10693 | // target's pointer size. | ||||
10694 | if (SO1->getType() != GO1->getType()) { | ||||
10695 | if (Constant *SO1C = dyn_cast<Constant>(SO1)) { | ||||
10696 | SO1 = ConstantExpr::getIntegerCast(SO1C, GO1->getType(), true); | ||||
10697 | } else if (Constant *GO1C = dyn_cast<Constant>(GO1)) { | ||||
10698 | GO1 = ConstantExpr::getIntegerCast(GO1C, SO1->getType(), true); | ||||
10699 | } else { | ||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 10700 | unsigned PS = TD->getPointerSizeInBits(); |
10701 | if (TD->getTypeSizeInBits(SO1->getType()) == PS) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10702 | // Convert GO1 to SO1's type. |
10703 | GO1 = InsertCastToIntPtrTy(GO1, SO1->getType(), &GEP, this); | ||||
10704 | |||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 10705 | } else if (TD->getTypeSizeInBits(GO1->getType()) == PS) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10706 | // Convert SO1 to GO1's type. |
10707 | SO1 = InsertCastToIntPtrTy(SO1, GO1->getType(), &GEP, this); | ||||
10708 | } else { | ||||
10709 | const Type *PT = TD->getIntPtrType(); | ||||
10710 | SO1 = InsertCastToIntPtrTy(SO1, PT, &GEP, this); | ||||
10711 | GO1 = InsertCastToIntPtrTy(GO1, PT, &GEP, this); | ||||
10712 | } | ||||
10713 | } | ||||
10714 | } | ||||
10715 | if (isa<Constant>(SO1) && isa<Constant>(GO1)) | ||||
10716 | Sum = ConstantExpr::getAdd(cast<Constant>(SO1), cast<Constant>(GO1)); | ||||
10717 | else { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10718 | Sum = BinaryOperator::CreateAdd(SO1, GO1, PtrOp->getName()+".sum"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10719 | InsertNewInstBefore(cast<Instruction>(Sum), GEP); |
10720 | } | ||||
10721 | } | ||||
10722 | |||||
10723 | // Recycle the GEP we already have if possible. | ||||
10724 | if (SrcGEPOperands.size() == 2) { | ||||
10725 | GEP.setOperand(0, SrcGEPOperands[0]); | ||||
10726 | GEP.setOperand(1, Sum); | ||||
10727 | return &GEP; | ||||
10728 | } else { | ||||
10729 | Indices.insert(Indices.end(), SrcGEPOperands.begin()+1, | ||||
10730 | SrcGEPOperands.end()-1); | ||||
10731 | Indices.push_back(Sum); | ||||
10732 | Indices.insert(Indices.end(), GEP.op_begin()+2, GEP.op_end()); | ||||
10733 | } | ||||
10734 | } else if (isa<Constant>(*GEP.idx_begin()) && | ||||
10735 | cast<Constant>(*GEP.idx_begin())->isNullValue() && | ||||
10736 | SrcGEPOperands.size() != 1) { | ||||
10737 | // Otherwise we can do the fold if the first index of the GEP is a zero | ||||
10738 | Indices.insert(Indices.end(), SrcGEPOperands.begin()+1, | ||||
10739 | SrcGEPOperands.end()); | ||||
10740 | Indices.insert(Indices.end(), GEP.idx_begin()+1, GEP.idx_end()); | ||||
10741 | } | ||||
10742 | |||||
10743 | if (!Indices.empty()) | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10744 | return GetElementPtrInst::Create(SrcGEPOperands[0], Indices.begin(), |
10745 | Indices.end(), GEP.getName()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10746 | |
10747 | } else if (GlobalValue *GV = dyn_cast<GlobalValue>(PtrOp)) { | ||||
10748 | // GEP of global variable. If all of the indices for this GEP are | ||||
10749 | // constants, we can promote this to a constexpr instead of an instruction. | ||||
10750 | |||||
10751 | // Scan for nonconstants... | ||||
10752 | SmallVector<Constant*, 8> Indices; | ||||
10753 | User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end(); | ||||
10754 | for (; I != E && isa<Constant>(*I); ++I) | ||||
10755 | Indices.push_back(cast<Constant>(*I)); | ||||
10756 | |||||
10757 | if (I == E) { // If they are all constants... | ||||
10758 | Constant *CE = ConstantExpr::getGetElementPtr(GV, | ||||
10759 | &Indices[0],Indices.size()); | ||||
10760 | |||||
10761 | // Replace all uses of the GEP with the new constexpr... | ||||
10762 | return ReplaceInstUsesWith(GEP, CE); | ||||
10763 | } | ||||
10764 | } else if (Value *X = getBitCastOperand(PtrOp)) { // Is the operand a cast? | ||||
10765 | if (!isa<PointerType>(X->getType())) { | ||||
10766 | // Not interesting. Source pointer must be a cast from pointer. | ||||
10767 | } else if (HasZeroPointerIndex) { | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10768 | // transform: GEP (bitcast [10 x i8]* X to [0 x i8]*), i32 0, ... |
10769 | // into : GEP [10 x i8]* X, i32 0, ... | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10770 | // |
Duncan Sands | cf866e6 | 2009-03-02 09:18:21 +0000 | [diff] [blame] | 10771 | // Likewise, transform: GEP (bitcast i8* X to [0 x i8]*), i32 0, ... |
10772 | // into : GEP i8* X, ... | ||||
10773 | // | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10774 | // This occurs when the program declares an array extern like "int X[];" |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10775 | const PointerType *CPTy = cast<PointerType>(PtrOp->getType()); |
10776 | const PointerType *XTy = cast<PointerType>(X->getType()); | ||||
Duncan Sands | cf866e6 | 2009-03-02 09:18:21 +0000 | [diff] [blame] | 10777 | if (const ArrayType *CATy = |
10778 | dyn_cast<ArrayType>(CPTy->getElementType())) { | ||||
10779 | // GEP (bitcast i8* X to [0 x i8]*), i32 0, ... ? | ||||
10780 | if (CATy->getElementType() == XTy->getElementType()) { | ||||
10781 | // -> GEP i8* X, ... | ||||
10782 | SmallVector<Value*, 8> Indices(GEP.idx_begin()+1, GEP.idx_end()); | ||||
10783 | return GetElementPtrInst::Create(X, Indices.begin(), Indices.end(), | ||||
10784 | GEP.getName()); | ||||
10785 | } else if (const ArrayType *XATy = | ||||
10786 | dyn_cast<ArrayType>(XTy->getElementType())) { | ||||
10787 | // GEP (bitcast [10 x i8]* X to [0 x i8]*), i32 0, ... ? | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10788 | if (CATy->getElementType() == XATy->getElementType()) { |
Duncan Sands | cf866e6 | 2009-03-02 09:18:21 +0000 | [diff] [blame] | 10789 | // -> GEP [10 x i8]* X, i32 0, ... |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10790 | // At this point, we know that the cast source type is a pointer |
10791 | // to an array of the same type as the destination pointer | ||||
10792 | // array. Because the array type is never stepped over (there | ||||
10793 | // is a leading zero) we can fold the cast into this GEP. | ||||
10794 | GEP.setOperand(0, X); | ||||
10795 | return &GEP; | ||||
10796 | } | ||||
Duncan Sands | cf866e6 | 2009-03-02 09:18:21 +0000 | [diff] [blame] | 10797 | } |
10798 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10799 | } else if (GEP.getNumOperands() == 2) { |
10800 | // Transform things like: | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10801 | // %t = getelementptr i32* bitcast ([2 x i32]* %str to i32*), i32 %V |
10802 | // into: %t1 = getelementptr [2 x i32]* %str, i32 0, i32 %V; bitcast | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10803 | const Type *SrcElTy = cast<PointerType>(X->getType())->getElementType(); |
10804 | const Type *ResElTy=cast<PointerType>(PtrOp->getType())->getElementType(); | ||||
10805 | if (isa<ArrayType>(SrcElTy) && | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 10806 | TD->getTypePaddedSize(cast<ArrayType>(SrcElTy)->getElementType()) == |
10807 | TD->getTypePaddedSize(ResElTy)) { | ||||
David Greene | 393be88 | 2007-09-04 15:46:09 +0000 | [diff] [blame] | 10808 | Value *Idx[2]; |
10809 | Idx[0] = Constant::getNullValue(Type::Int32Ty); | ||||
10810 | Idx[1] = GEP.getOperand(1); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10811 | Value *V = InsertNewInstBefore( |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10812 | GetElementPtrInst::Create(X, Idx, Idx + 2, GEP.getName()), GEP); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10813 | // V and GEP are both pointer types --> BitCast |
10814 | return new BitCastInst(V, GEP.getType()); | ||||
10815 | } | ||||
10816 | |||||
10817 | // Transform things like: | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10818 | // getelementptr i8* bitcast ([100 x double]* X to i8*), i32 %tmp |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10819 | // (where tmp = 8*tmp2) into: |
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10820 | // getelementptr [100 x double]* %arr, i32 0, i32 %tmp2; bitcast |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10821 | |
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10822 | if (isa<ArrayType>(SrcElTy) && ResElTy == Type::Int8Ty) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10823 | uint64_t ArrayEltSize = |
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 10824 | TD->getTypePaddedSize(cast<ArrayType>(SrcElTy)->getElementType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10825 | |
10826 | // Check to see if "tmp" is a scale by a multiple of ArrayEltSize. We | ||||
10827 | // allow either a mul, shift, or constant here. | ||||
10828 | Value *NewIdx = 0; | ||||
10829 | ConstantInt *Scale = 0; | ||||
10830 | if (ArrayEltSize == 1) { | ||||
10831 | NewIdx = GEP.getOperand(1); | ||||
10832 | Scale = ConstantInt::get(NewIdx->getType(), 1); | ||||
10833 | } else if (ConstantInt *CI = dyn_cast<ConstantInt>(GEP.getOperand(1))) { | ||||
10834 | NewIdx = ConstantInt::get(CI->getType(), 1); | ||||
10835 | Scale = CI; | ||||
10836 | } else if (Instruction *Inst =dyn_cast<Instruction>(GEP.getOperand(1))){ | ||||
10837 | if (Inst->getOpcode() == Instruction::Shl && | ||||
10838 | isa<ConstantInt>(Inst->getOperand(1))) { | ||||
10839 | ConstantInt *ShAmt = cast<ConstantInt>(Inst->getOperand(1)); | ||||
10840 | uint32_t ShAmtVal = ShAmt->getLimitedValue(64); | ||||
10841 | Scale = ConstantInt::get(Inst->getType(), 1ULL << ShAmtVal); | ||||
10842 | NewIdx = Inst->getOperand(0); | ||||
10843 | } else if (Inst->getOpcode() == Instruction::Mul && | ||||
10844 | isa<ConstantInt>(Inst->getOperand(1))) { | ||||
10845 | Scale = cast<ConstantInt>(Inst->getOperand(1)); | ||||
10846 | NewIdx = Inst->getOperand(0); | ||||
10847 | } | ||||
10848 | } | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10849 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10850 | // If the index will be to exactly the right offset with the scale taken |
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10851 | // out, perform the transformation. Note, we don't know whether Scale is |
10852 | // signed or not. We'll use unsigned version of division/modulo | ||||
10853 | // operation after making sure Scale doesn't have the sign bit set. | ||||
Chris Lattner | 0296271 | 2009-02-25 18:20:01 +0000 | [diff] [blame] | 10854 | if (ArrayEltSize && Scale && Scale->getSExtValue() >= 0LL && |
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10855 | Scale->getZExtValue() % ArrayEltSize == 0) { |
10856 | Scale = ConstantInt::get(Scale->getType(), | ||||
10857 | Scale->getZExtValue() / ArrayEltSize); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10858 | if (Scale->getZExtValue() != 1) { |
10859 | Constant *C = ConstantExpr::getIntegerCast(Scale, NewIdx->getType(), | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10860 | false /*ZExt*/); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10861 | Instruction *Sc = BinaryOperator::CreateMul(NewIdx, C, "idxscale"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10862 | NewIdx = InsertNewInstBefore(Sc, GEP); |
10863 | } | ||||
10864 | |||||
10865 | // Insert the new GEP instruction. | ||||
David Greene | 393be88 | 2007-09-04 15:46:09 +0000 | [diff] [blame] | 10866 | Value *Idx[2]; |
10867 | Idx[0] = Constant::getNullValue(Type::Int32Ty); | ||||
10868 | Idx[1] = NewIdx; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10869 | Instruction *NewGEP = |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10870 | GetElementPtrInst::Create(X, Idx, Idx + 2, GEP.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10871 | NewGEP = InsertNewInstBefore(NewGEP, GEP); |
10872 | // The NewGEP must be pointer typed, so must the old one -> BitCast | ||||
10873 | return new BitCastInst(NewGEP, GEP.getType()); | ||||
10874 | } | ||||
10875 | } | ||||
10876 | } | ||||
10877 | } | ||||
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10878 | |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 10879 | /// See if we can simplify: |
10880 | /// X = bitcast A to B* | ||||
10881 | /// Y = gep X, <...constant indices...> | ||||
10882 | /// into a gep of the original struct. This is important for SROA and alias | ||||
10883 | /// analysis of unions. If "A" is also a bitcast, wait for A/X to be merged. | ||||
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10884 | if (BitCastInst *BCI = dyn_cast<BitCastInst>(PtrOp)) { |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 10885 | if (!isa<BitCastInst>(BCI->getOperand(0)) && GEP.hasAllConstantIndices()) { |
10886 | // Determine how much the GEP moves the pointer. We are guaranteed to get | ||||
10887 | // a constant back from EmitGEPOffset. | ||||
10888 | ConstantInt *OffsetV = cast<ConstantInt>(EmitGEPOffset(&GEP, GEP, *this)); | ||||
10889 | int64_t Offset = OffsetV->getSExtValue(); | ||||
10890 | |||||
10891 | // If this GEP instruction doesn't move the pointer, just replace the GEP | ||||
10892 | // with a bitcast of the real input to the dest type. | ||||
10893 | if (Offset == 0) { | ||||
10894 | // If the bitcast is of an allocation, and the allocation will be | ||||
10895 | // converted to match the type of the cast, don't touch this. | ||||
10896 | if (isa<AllocationInst>(BCI->getOperand(0))) { | ||||
10897 | // See if the bitcast simplifies, if so, don't nuke this GEP yet. | ||||
10898 | if (Instruction *I = visitBitCast(*BCI)) { | ||||
10899 | if (I != BCI) { | ||||
10900 | I->takeName(BCI); | ||||
10901 | BCI->getParent()->getInstList().insert(BCI, I); | ||||
10902 | ReplaceInstUsesWith(*BCI, I); | ||||
10903 | } | ||||
10904 | return &GEP; | ||||
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10905 | } |
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10906 | } |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 10907 | return new BitCastInst(BCI->getOperand(0), GEP.getType()); |
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10908 | } |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 10909 | |
10910 | // Otherwise, if the offset is non-zero, we need to find out if there is a | ||||
10911 | // field at Offset in 'A's type. If so, we can pull the cast through the | ||||
10912 | // GEP. | ||||
10913 | SmallVector<Value*, 8> NewIndices; | ||||
10914 | const Type *InTy = | ||||
10915 | cast<PointerType>(BCI->getOperand(0)->getType())->getElementType(); | ||||
10916 | if (FindElementAtOffset(InTy, Offset, NewIndices, TD)) { | ||||
10917 | Instruction *NGEP = | ||||
10918 | GetElementPtrInst::Create(BCI->getOperand(0), NewIndices.begin(), | ||||
10919 | NewIndices.end()); | ||||
10920 | if (NGEP->getType() == GEP.getType()) return NGEP; | ||||
10921 | InsertNewInstBefore(NGEP, GEP); | ||||
10922 | NGEP->takeName(&GEP); | ||||
10923 | return new BitCastInst(NGEP, GEP.getType()); | ||||
10924 | } | ||||
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10925 | } |
10926 | } | ||||
10927 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10928 | return 0; |
10929 | } | ||||
10930 | |||||
10931 | Instruction *InstCombiner::visitAllocationInst(AllocationInst &AI) { | ||||
10932 | // Convert: malloc Ty, C - where C is a constant != 1 into: malloc [C x Ty], 1 | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10933 | if (AI.isArrayAllocation()) { // Check C != 1 |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10934 | if (const ConstantInt *C = dyn_cast<ConstantInt>(AI.getArraySize())) { |
10935 | const Type *NewTy = | ||||
10936 | ArrayType::get(AI.getAllocatedType(), C->getZExtValue()); | ||||
10937 | AllocationInst *New = 0; | ||||
10938 | |||||
10939 | // Create and insert the replacement instruction... | ||||
10940 | if (isa<MallocInst>(AI)) | ||||
10941 | New = new MallocInst(NewTy, 0, AI.getAlignment(), AI.getName()); | ||||
10942 | else { | ||||
10943 | assert(isa<AllocaInst>(AI) && "Unknown type of allocation inst!"); | ||||
10944 | New = new AllocaInst(NewTy, 0, AI.getAlignment(), AI.getName()); | ||||
10945 | } | ||||
10946 | |||||
10947 | InsertNewInstBefore(New, AI); | ||||
10948 | |||||
10949 | // Scan to the end of the allocation instructions, to skip over a block of | ||||
10950 | // allocas if possible... | ||||
10951 | // | ||||
10952 | BasicBlock::iterator It = New; | ||||
10953 | while (isa<AllocationInst>(*It)) ++It; | ||||
10954 | |||||
10955 | // Now that I is pointing to the first non-allocation-inst in the block, | ||||
10956 | // insert our getelementptr instruction... | ||||
10957 | // | ||||
10958 | Value *NullIdx = Constant::getNullValue(Type::Int32Ty); | ||||
David Greene | 393be88 | 2007-09-04 15:46:09 +0000 | [diff] [blame] | 10959 | Value *Idx[2]; |
10960 | Idx[0] = NullIdx; | ||||
10961 | Idx[1] = NullIdx; | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10962 | Value *V = GetElementPtrInst::Create(New, Idx, Idx + 2, |
10963 | New->getName()+".sub", It); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10964 | |
10965 | // Now make everything use the getelementptr instead of the original | ||||
10966 | // allocation. | ||||
10967 | return ReplaceInstUsesWith(AI, V); | ||||
10968 | } else if (isa<UndefValue>(AI.getArraySize())) { | ||||
10969 | return ReplaceInstUsesWith(AI, Constant::getNullValue(AI.getType())); | ||||
10970 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10971 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10972 | |
Dan Gohman | 28e78f0 | 2009-01-13 20:18:38 +0000 | [diff] [blame] | 10973 | if (isa<AllocaInst>(AI) && AI.getAllocatedType()->isSized()) { |
10974 | // If alloca'ing a zero byte object, replace the alloca with a null pointer. | ||||
10975 | // Note that we only do this for alloca's, because malloc should allocate and | ||||
10976 | // return a unique pointer, even for a zero byte allocation. | ||||
10977 | if (TD->getTypePaddedSize(AI.getAllocatedType()) == 0) | ||||
10978 | return ReplaceInstUsesWith(AI, Constant::getNullValue(AI.getType())); | ||||
10979 | |||||
10980 | // If the alignment is 0 (unspecified), assign it the preferred alignment. | ||||
10981 | if (AI.getAlignment() == 0) | ||||
10982 | AI.setAlignment(TD->getPrefTypeAlignment(AI.getAllocatedType())); | ||||
10983 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10984 | |
10985 | return 0; | ||||
10986 | } | ||||
10987 | |||||
10988 | Instruction *InstCombiner::visitFreeInst(FreeInst &FI) { | ||||
10989 | Value *Op = FI.getOperand(0); | ||||
10990 | |||||
10991 | // free undef -> unreachable. | ||||
10992 | if (isa<UndefValue>(Op)) { | ||||
10993 | // Insert a new store to null because we cannot modify the CFG here. | ||||
10994 | new StoreInst(ConstantInt::getTrue(), | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 10995 | UndefValue::get(PointerType::getUnqual(Type::Int1Ty)), &FI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10996 | return EraseInstFromFunction(FI); |
10997 | } | ||||
10998 | |||||
10999 | // If we have 'free null' delete the instruction. This can happen in stl code | ||||
11000 | // when lots of inlining happens. | ||||
11001 | if (isa<ConstantPointerNull>(Op)) | ||||
11002 | return EraseInstFromFunction(FI); | ||||
11003 | |||||
11004 | // Change free <ty>* (cast <ty2>* X to <ty>*) into free <ty2>* X | ||||
11005 | if (BitCastInst *CI = dyn_cast<BitCastInst>(Op)) { | ||||
11006 | FI.setOperand(0, CI->getOperand(0)); | ||||
11007 | return &FI; | ||||
11008 | } | ||||
11009 | |||||
11010 | // Change free (gep X, 0,0,0,0) into free(X) | ||||
11011 | if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(Op)) { | ||||
11012 | if (GEPI->hasAllZeroIndices()) { | ||||
11013 | AddToWorkList(GEPI); | ||||
11014 | FI.setOperand(0, GEPI->getOperand(0)); | ||||
11015 | return &FI; | ||||
11016 | } | ||||
11017 | } | ||||
11018 | |||||
11019 | // Change free(malloc) into nothing, if the malloc has a single use. | ||||
11020 | if (MallocInst *MI = dyn_cast<MallocInst>(Op)) | ||||
11021 | if (MI->hasOneUse()) { | ||||
11022 | EraseInstFromFunction(FI); | ||||
11023 | return EraseInstFromFunction(*MI); | ||||
11024 | } | ||||
11025 | |||||
11026 | return 0; | ||||
11027 | } | ||||
11028 | |||||
11029 | |||||
11030 | /// InstCombineLoadCast - Fold 'load (cast P)' -> cast (load P)' when possible. | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 11031 | static Instruction *InstCombineLoadCast(InstCombiner &IC, LoadInst &LI, |
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 11032 | const TargetData *TD) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11033 | User *CI = cast<User>(LI.getOperand(0)); |
11034 | Value *CastOp = CI->getOperand(0); | ||||
11035 | |||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 11036 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(CI)) { |
11037 | // Instead of loading constant c string, use corresponding integer value | ||||
11038 | // directly if string length is small enough. | ||||
Evan Cheng | 833501d | 2008-06-30 07:31:25 +0000 | [diff] [blame] | 11039 | std::string Str; |
11040 | if (GetConstantStringInfo(CE->getOperand(0), Str) && !Str.empty()) { | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 11041 | unsigned len = Str.length(); |
11042 | const Type *Ty = cast<PointerType>(CE->getType())->getElementType(); | ||||
11043 | unsigned numBits = Ty->getPrimitiveSizeInBits(); | ||||
11044 | // Replace LI with immediate integer store. | ||||
11045 | if ((numBits >> 3) == len + 1) { | ||||
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 11046 | APInt StrVal(numBits, 0); |
11047 | APInt SingleChar(numBits, 0); | ||||
11048 | if (TD->isLittleEndian()) { | ||||
11049 | for (signed i = len-1; i >= 0; i--) { | ||||
Nick Lewycky | 1803a02 | 2009-02-21 20:50:42 +0000 | [diff] [blame] | 11050 | SingleChar = (uint64_t) Str[i] & UCHAR_MAX; |
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 11051 | StrVal = (StrVal << 8) | SingleChar; |
11052 | } | ||||
11053 | } else { | ||||
11054 | for (unsigned i = 0; i < len; i++) { | ||||
Nick Lewycky | 1803a02 | 2009-02-21 20:50:42 +0000 | [diff] [blame] | 11055 | SingleChar = (uint64_t) Str[i] & UCHAR_MAX; |
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 11056 | StrVal = (StrVal << 8) | SingleChar; |
11057 | } | ||||
11058 | // Append NULL at the end. | ||||
11059 | SingleChar = 0; | ||||
11060 | StrVal = (StrVal << 8) | SingleChar; | ||||
11061 | } | ||||
11062 | Value *NL = ConstantInt::get(StrVal); | ||||
11063 | return IC.ReplaceInstUsesWith(LI, NL); | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 11064 | } |
11065 | } | ||||
11066 | } | ||||
11067 | |||||
Mon P Wang | bd05ed8 | 2009-02-07 22:19:29 +0000 | [diff] [blame] | 11068 | const PointerType *DestTy = cast<PointerType>(CI->getType()); |
11069 | const Type *DestPTy = DestTy->getElementType(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11070 | if (const PointerType *SrcTy = dyn_cast<PointerType>(CastOp->getType())) { |
Mon P Wang | bd05ed8 | 2009-02-07 22:19:29 +0000 | [diff] [blame] | 11071 | |
11072 | // If the address spaces don't match, don't eliminate the cast. | ||||
11073 | if (DestTy->getAddressSpace() != SrcTy->getAddressSpace()) | ||||
11074 | return 0; | ||||
11075 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11076 | const Type *SrcPTy = SrcTy->getElementType(); |
11077 | |||||
11078 | if (DestPTy->isInteger() || isa<PointerType>(DestPTy) || | ||||
11079 | isa<VectorType>(DestPTy)) { | ||||
11080 | // If the source is an array, the code below will not succeed. Check to | ||||
11081 | // see if a trivial 'gep P, 0, 0' will help matters. Only do this for | ||||
11082 | // constants. | ||||
11083 | if (const ArrayType *ASrcTy = dyn_cast<ArrayType>(SrcPTy)) | ||||
11084 | if (Constant *CSrc = dyn_cast<Constant>(CastOp)) | ||||
11085 | if (ASrcTy->getNumElements() != 0) { | ||||
11086 | Value *Idxs[2]; | ||||
11087 | Idxs[0] = Idxs[1] = Constant::getNullValue(Type::Int32Ty); | ||||
11088 | CastOp = ConstantExpr::getGetElementPtr(CSrc, Idxs, 2); | ||||
11089 | SrcTy = cast<PointerType>(CastOp->getType()); | ||||
11090 | SrcPTy = SrcTy->getElementType(); | ||||
11091 | } | ||||
11092 | |||||
11093 | if ((SrcPTy->isInteger() || isa<PointerType>(SrcPTy) || | ||||
11094 | isa<VectorType>(SrcPTy)) && | ||||
11095 | // Do not allow turning this into a load of an integer, which is then | ||||
11096 | // casted to a pointer, this pessimizes pointer analysis a lot. | ||||
11097 | (isa<PointerType>(SrcPTy) == isa<PointerType>(LI.getType())) && | ||||
11098 | IC.getTargetData().getTypeSizeInBits(SrcPTy) == | ||||
11099 | IC.getTargetData().getTypeSizeInBits(DestPTy)) { | ||||
11100 | |||||
11101 | // Okay, we are casting from one integer or pointer type to another of | ||||
11102 | // the same size. Instead of casting the pointer before the load, cast | ||||
11103 | // the result of the loaded value. | ||||
11104 | Value *NewLoad = IC.InsertNewInstBefore(new LoadInst(CastOp, | ||||
11105 | CI->getName(), | ||||
11106 | LI.isVolatile()),LI); | ||||
11107 | // Now cast the result of the load. | ||||
11108 | return new BitCastInst(NewLoad, LI.getType()); | ||||
11109 | } | ||||
11110 | } | ||||
11111 | } | ||||
11112 | return 0; | ||||
11113 | } | ||||
11114 | |||||
11115 | /// isSafeToLoadUnconditionally - Return true if we know that executing a load | ||||
11116 | /// from this value cannot trap. If it is not obviously safe to load from the | ||||
11117 | /// specified pointer, we do a quick local scan of the basic block containing | ||||
11118 | /// ScanFrom, to determine if the address is already accessed. | ||||
11119 | static bool isSafeToLoadUnconditionally(Value *V, Instruction *ScanFrom) { | ||||
Duncan Sands | 9b27dbe | 2007-09-19 10:10:31 +0000 | [diff] [blame] | 11120 | // If it is an alloca it is always safe to load from. |
11121 | if (isa<AllocaInst>(V)) return true; | ||||
11122 | |||||
Duncan Sands | e40a94a | 2007-09-19 10:25:38 +0000 | [diff] [blame] | 11123 | // If it is a global variable it is mostly safe to load from. |
Duncan Sands | 9b27dbe | 2007-09-19 10:10:31 +0000 | [diff] [blame] | 11124 | if (const GlobalValue *GV = dyn_cast<GlobalVariable>(V)) |
Duncan Sands | e40a94a | 2007-09-19 10:25:38 +0000 | [diff] [blame] | 11125 | // Don't try to evaluate aliases. External weak GV can be null. |
Duncan Sands | 9b27dbe | 2007-09-19 10:10:31 +0000 | [diff] [blame] | 11126 | return !isa<GlobalAlias>(GV) && !GV->hasExternalWeakLinkage(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11127 | |
11128 | // Otherwise, be a little bit agressive by scanning the local block where we | ||||
11129 | // want to check to see if the pointer is already being loaded or stored | ||||
11130 | // from/to. If so, the previous load or store would have already trapped, | ||||
11131 | // so there is no harm doing an extra load (also, CSE will later eliminate | ||||
11132 | // the load entirely). | ||||
11133 | BasicBlock::iterator BBI = ScanFrom, E = ScanFrom->getParent()->begin(); | ||||
11134 | |||||
11135 | while (BBI != E) { | ||||
11136 | --BBI; | ||||
11137 | |||||
Chris Lattner | 476983a | 2008-06-20 05:12:56 +0000 | [diff] [blame] | 11138 | // If we see a free or a call (which might do a free) the pointer could be |
11139 | // marked invalid. | ||||
11140 | if (isa<FreeInst>(BBI) || isa<CallInst>(BBI)) | ||||
11141 | return false; | ||||
11142 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11143 | if (LoadInst *LI = dyn_cast<LoadInst>(BBI)) { |
11144 | if (LI->getOperand(0) == V) return true; | ||||
Chris Lattner | 476983a | 2008-06-20 05:12:56 +0000 | [diff] [blame] | 11145 | } else if (StoreInst *SI = dyn_cast<StoreInst>(BBI)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11146 | if (SI->getOperand(1) == V) return true; |
Chris Lattner | 476983a | 2008-06-20 05:12:56 +0000 | [diff] [blame] | 11147 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11148 | |
11149 | } | ||||
11150 | return false; | ||||
11151 | } | ||||
11152 | |||||
11153 | Instruction *InstCombiner::visitLoadInst(LoadInst &LI) { | ||||
11154 | Value *Op = LI.getOperand(0); | ||||
11155 | |||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 11156 | // Attempt to improve the alignment. |
Dan Gohman | 3719257 | 2009-02-16 00:44:23 +0000 | [diff] [blame] | 11157 | unsigned KnownAlign = |
11158 | GetOrEnforceKnownAlignment(Op, TD->getPrefTypeAlignment(LI.getType())); | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 11159 | if (KnownAlign > |
11160 | (LI.getAlignment() == 0 ? TD->getABITypeAlignment(LI.getType()) : | ||||
11161 | LI.getAlignment())) | ||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 11162 | LI.setAlignment(KnownAlign); |
11163 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11164 | // load (cast X) --> cast (load X) iff safe |
11165 | if (isa<CastInst>(Op)) | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 11166 | if (Instruction *Res = InstCombineLoadCast(*this, LI, TD)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11167 | return Res; |
11168 | |||||
11169 | // None of the following transforms are legal for volatile loads. | ||||
11170 | if (LI.isVolatile()) return 0; | ||||
11171 | |||||
Dan Gohman | 0ff5a1f | 2008-10-15 23:19:35 +0000 | [diff] [blame] | 11172 | // Do really simple store-to-load forwarding and load CSE, to catch cases |
11173 | // where there are several consequtive memory accesses to the same location, | ||||
11174 | // separated by a few arithmetic operations. | ||||
11175 | BasicBlock::iterator BBI = &LI; | ||||
Chris Lattner | 6fd8c80 | 2008-11-27 08:56:30 +0000 | [diff] [blame] | 11176 | if (Value *AvailableVal = FindAvailableLoadedValue(Op, LI.getParent(), BBI,6)) |
11177 | return ReplaceInstUsesWith(LI, AvailableVal); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11178 | |
Christopher Lamb | 2c17539 | 2007-12-29 07:56:53 +0000 | [diff] [blame] | 11179 | if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(Op)) { |
11180 | const Value *GEPI0 = GEPI->getOperand(0); | ||||
11181 | // TODO: Consider a target hook for valid address spaces for this xform. | ||||
11182 | if (isa<ConstantPointerNull>(GEPI0) && | ||||
11183 | cast<PointerType>(GEPI0->getType())->getAddressSpace() == 0) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11184 | // Insert a new store to null instruction before the load to indicate |
11185 | // that this code is not reachable. We do this instead of inserting | ||||
11186 | // an unreachable instruction directly because we cannot modify the | ||||
11187 | // CFG. | ||||
11188 | new StoreInst(UndefValue::get(LI.getType()), | ||||
11189 | Constant::getNullValue(Op->getType()), &LI); | ||||
11190 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
11191 | } | ||||
Christopher Lamb | 2c17539 | 2007-12-29 07:56:53 +0000 | [diff] [blame] | 11192 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11193 | |
11194 | if (Constant *C = dyn_cast<Constant>(Op)) { | ||||
11195 | // load null/undef -> undef | ||||
Christopher Lamb | 2c17539 | 2007-12-29 07:56:53 +0000 | [diff] [blame] | 11196 | // TODO: Consider a target hook for valid address spaces for this xform. |
11197 | if (isa<UndefValue>(C) || (C->isNullValue() && | ||||
11198 | cast<PointerType>(Op->getType())->getAddressSpace() == 0)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11199 | // Insert a new store to null instruction before the load to indicate that |
11200 | // this code is not reachable. We do this instead of inserting an | ||||
11201 | // unreachable instruction directly because we cannot modify the CFG. | ||||
11202 | new StoreInst(UndefValue::get(LI.getType()), | ||||
11203 | Constant::getNullValue(Op->getType()), &LI); | ||||
11204 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
11205 | } | ||||
11206 | |||||
11207 | // Instcombine load (constant global) into the value loaded. | ||||
11208 | if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op)) | ||||
11209 | if (GV->isConstant() && !GV->isDeclaration()) | ||||
11210 | return ReplaceInstUsesWith(LI, GV->getInitializer()); | ||||
11211 | |||||
11212 | // Instcombine load (constantexpr_GEP global, 0, ...) into the value loaded. | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 11213 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Op)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11214 | if (CE->getOpcode() == Instruction::GetElementPtr) { |
11215 | if (GlobalVariable *GV = dyn_cast<GlobalVariable>(CE->getOperand(0))) | ||||
11216 | if (GV->isConstant() && !GV->isDeclaration()) | ||||
11217 | if (Constant *V = | ||||
11218 | ConstantFoldLoadThroughGEPConstantExpr(GV->getInitializer(), CE)) | ||||
11219 | return ReplaceInstUsesWith(LI, V); | ||||
11220 | if (CE->getOperand(0)->isNullValue()) { | ||||
11221 | // Insert a new store to null instruction before the load to indicate | ||||
11222 | // that this code is not reachable. We do this instead of inserting | ||||
11223 | // an unreachable instruction directly because we cannot modify the | ||||
11224 | // CFG. | ||||
11225 | new StoreInst(UndefValue::get(LI.getType()), | ||||
11226 | Constant::getNullValue(Op->getType()), &LI); | ||||
11227 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
11228 | } | ||||
11229 | |||||
11230 | } else if (CE->isCast()) { | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 11231 | if (Instruction *Res = InstCombineLoadCast(*this, LI, TD)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11232 | return Res; |
11233 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 11234 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11235 | } |
Chris Lattner | 0270a11 | 2007-08-11 18:48:48 +0000 | [diff] [blame] | 11236 | |
11237 | // If this load comes from anywhere in a constant global, and if the global | ||||
11238 | // is all undef or zero, we know what it loads. | ||||
Duncan Sands | 52fb873 | 2008-10-01 15:25:41 +0000 | [diff] [blame] | 11239 | if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op->getUnderlyingObject())){ |
Chris Lattner | 0270a11 | 2007-08-11 18:48:48 +0000 | [diff] [blame] | 11240 | if (GV->isConstant() && GV->hasInitializer()) { |
11241 | if (GV->getInitializer()->isNullValue()) | ||||
11242 | return ReplaceInstUsesWith(LI, Constant::getNullValue(LI.getType())); | ||||
11243 | else if (isa<UndefValue>(GV->getInitializer())) | ||||
11244 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
11245 | } | ||||
11246 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11247 | |
11248 | if (Op->hasOneUse()) { | ||||
11249 | // Change select and PHI nodes to select values instead of addresses: this | ||||
11250 | // helps alias analysis out a lot, allows many others simplifications, and | ||||
11251 | // exposes redundancy in the code. | ||||
11252 | // | ||||
11253 | // Note that we cannot do the transformation unless we know that the | ||||
11254 | // introduced loads cannot trap! Something like this is valid as long as | ||||
11255 | // the condition is always false: load (select bool %C, int* null, int* %G), | ||||
11256 | // but it would not be valid if we transformed it to load from null | ||||
11257 | // unconditionally. | ||||
11258 | // | ||||
11259 | if (SelectInst *SI = dyn_cast<SelectInst>(Op)) { | ||||
11260 | // load (select (Cond, &V1, &V2)) --> select(Cond, load &V1, load &V2). | ||||
11261 | if (isSafeToLoadUnconditionally(SI->getOperand(1), SI) && | ||||
11262 | isSafeToLoadUnconditionally(SI->getOperand(2), SI)) { | ||||
11263 | Value *V1 = InsertNewInstBefore(new LoadInst(SI->getOperand(1), | ||||
11264 | SI->getOperand(1)->getName()+".val"), LI); | ||||
11265 | Value *V2 = InsertNewInstBefore(new LoadInst(SI->getOperand(2), | ||||
11266 | SI->getOperand(2)->getName()+".val"), LI); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 11267 | return SelectInst::Create(SI->getCondition(), V1, V2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11268 | } |
11269 | |||||
11270 | // load (select (cond, null, P)) -> load P | ||||
11271 | if (Constant *C = dyn_cast<Constant>(SI->getOperand(1))) | ||||
11272 | if (C->isNullValue()) { | ||||
11273 | LI.setOperand(0, SI->getOperand(2)); | ||||
11274 | return &LI; | ||||
11275 | } | ||||
11276 | |||||
11277 | // load (select (cond, P, null)) -> load P | ||||
11278 | if (Constant *C = dyn_cast<Constant>(SI->getOperand(2))) | ||||
11279 | if (C->isNullValue()) { | ||||
11280 | LI.setOperand(0, SI->getOperand(1)); | ||||
11281 | return &LI; | ||||
11282 | } | ||||
11283 | } | ||||
11284 | } | ||||
11285 | return 0; | ||||
11286 | } | ||||
11287 | |||||
11288 | /// InstCombineStoreToCast - Fold store V, (cast P) -> store (cast V), P | ||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11289 | /// when possible. This makes it generally easy to do alias analysis and/or |
11290 | /// SROA/mem2reg of the memory object. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11291 | static Instruction *InstCombineStoreToCast(InstCombiner &IC, StoreInst &SI) { |
11292 | User *CI = cast<User>(SI.getOperand(1)); | ||||
11293 | Value *CastOp = CI->getOperand(0); | ||||
11294 | |||||
11295 | const Type *DestPTy = cast<PointerType>(CI->getType())->getElementType(); | ||||
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11296 | const PointerType *SrcTy = dyn_cast<PointerType>(CastOp->getType()); |
11297 | if (SrcTy == 0) return 0; | ||||
11298 | |||||
11299 | const Type *SrcPTy = SrcTy->getElementType(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11300 | |
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11301 | if (!DestPTy->isInteger() && !isa<PointerType>(DestPTy)) |
11302 | return 0; | ||||
11303 | |||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11304 | /// NewGEPIndices - If SrcPTy is an aggregate type, we can emit a "noop gep" |
11305 | /// to its first element. This allows us to handle things like: | ||||
11306 | /// store i32 xxx, (bitcast {foo*, float}* %P to i32*) | ||||
11307 | /// on 32-bit hosts. | ||||
11308 | SmallVector<Value*, 4> NewGEPIndices; | ||||
11309 | |||||
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11310 | // If the source is an array, the code below will not succeed. Check to |
11311 | // see if a trivial 'gep P, 0, 0' will help matters. Only do this for | ||||
11312 | // constants. | ||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11313 | if (isa<ArrayType>(SrcPTy) || isa<StructType>(SrcPTy)) { |
11314 | // Index through pointer. | ||||
11315 | Constant *Zero = Constant::getNullValue(Type::Int32Ty); | ||||
11316 | NewGEPIndices.push_back(Zero); | ||||
11317 | |||||
11318 | while (1) { | ||||
11319 | if (const StructType *STy = dyn_cast<StructType>(SrcPTy)) { | ||||
edwin | 7dc0aa3 | 2009-01-24 17:16:04 +0000 | [diff] [blame] | 11320 | if (!STy->getNumElements()) /* Struct can be empty {} */ |
edwin | 07d74e7 | 2009-01-24 11:30:49 +0000 | [diff] [blame] | 11321 | break; |
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11322 | NewGEPIndices.push_back(Zero); |
11323 | SrcPTy = STy->getElementType(0); | ||||
11324 | } else if (const ArrayType *ATy = dyn_cast<ArrayType>(SrcPTy)) { | ||||
11325 | NewGEPIndices.push_back(Zero); | ||||
11326 | SrcPTy = ATy->getElementType(); | ||||
11327 | } else { | ||||
11328 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11329 | } |
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11330 | } |
11331 | |||||
11332 | SrcTy = PointerType::get(SrcPTy, SrcTy->getAddressSpace()); | ||||
11333 | } | ||||
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11334 | |
11335 | if (!SrcPTy->isInteger() && !isa<PointerType>(SrcPTy)) | ||||
11336 | return 0; | ||||
11337 | |||||
Chris Lattner | c73a0d1 | 2009-01-16 20:12:52 +0000 | [diff] [blame] | 11338 | // If the pointers point into different address spaces or if they point to |
11339 | // values with different sizes, we can't do the transformation. | ||||
11340 | if (SrcTy->getAddressSpace() != | ||||
11341 | cast<PointerType>(CI->getType())->getAddressSpace() || | ||||
11342 | IC.getTargetData().getTypeSizeInBits(SrcPTy) != | ||||
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11343 | IC.getTargetData().getTypeSizeInBits(DestPTy)) |
11344 | return 0; | ||||
11345 | |||||
11346 | // Okay, we are casting from one integer or pointer type to another of | ||||
11347 | // the same size. Instead of casting the pointer before | ||||
11348 | // the store, cast the value to be stored. | ||||
11349 | Value *NewCast; | ||||
11350 | Value *SIOp0 = SI.getOperand(0); | ||||
11351 | Instruction::CastOps opcode = Instruction::BitCast; | ||||
11352 | const Type* CastSrcTy = SIOp0->getType(); | ||||
11353 | const Type* CastDstTy = SrcPTy; | ||||
11354 | if (isa<PointerType>(CastDstTy)) { | ||||
11355 | if (CastSrcTy->isInteger()) | ||||
11356 | opcode = Instruction::IntToPtr; | ||||
11357 | } else if (isa<IntegerType>(CastDstTy)) { | ||||
11358 | if (isa<PointerType>(SIOp0->getType())) | ||||
11359 | opcode = Instruction::PtrToInt; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11360 | } |
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11361 | |
11362 | // SIOp0 is a pointer to aggregate and this is a store to the first field, | ||||
11363 | // emit a GEP to index into its first field. | ||||
11364 | if (!NewGEPIndices.empty()) { | ||||
11365 | if (Constant *C = dyn_cast<Constant>(CastOp)) | ||||
11366 | CastOp = ConstantExpr::getGetElementPtr(C, &NewGEPIndices[0], | ||||
11367 | NewGEPIndices.size()); | ||||
11368 | else | ||||
11369 | CastOp = IC.InsertNewInstBefore( | ||||
11370 | GetElementPtrInst::Create(CastOp, NewGEPIndices.begin(), | ||||
11371 | NewGEPIndices.end()), SI); | ||||
11372 | } | ||||
11373 | |||||
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11374 | if (Constant *C = dyn_cast<Constant>(SIOp0)) |
11375 | NewCast = ConstantExpr::getCast(opcode, C, CastDstTy); | ||||
11376 | else | ||||
11377 | NewCast = IC.InsertNewInstBefore( | ||||
11378 | CastInst::Create(opcode, SIOp0, CastDstTy, SIOp0->getName()+".c"), | ||||
11379 | SI); | ||||
11380 | return new StoreInst(NewCast, CastOp); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11381 | } |
11382 | |||||
Chris Lattner | 6fd8c80 | 2008-11-27 08:56:30 +0000 | [diff] [blame] | 11383 | /// equivalentAddressValues - Test if A and B will obviously have the same |
11384 | /// value. This includes recognizing that %t0 and %t1 will have the same | ||||
11385 | /// value in code like this: | ||||
Dan Gohman | 8387bb3 | 2009-03-03 02:55:14 +0000 | [diff] [blame] | 11386 | /// %t0 = getelementptr \@a, 0, 3 |
Chris Lattner | 6fd8c80 | 2008-11-27 08:56:30 +0000 | [diff] [blame] | 11387 | /// store i32 0, i32* %t0 |
Dan Gohman | 8387bb3 | 2009-03-03 02:55:14 +0000 | [diff] [blame] | 11388 | /// %t1 = getelementptr \@a, 0, 3 |
Chris Lattner | 6fd8c80 | 2008-11-27 08:56:30 +0000 | [diff] [blame] | 11389 | /// %t2 = load i32* %t1 |
11390 | /// | ||||
11391 | static bool equivalentAddressValues(Value *A, Value *B) { | ||||
11392 | // Test if the values are trivially equivalent. | ||||
11393 | if (A == B) return true; | ||||
11394 | |||||
11395 | // Test if the values come form identical arithmetic instructions. | ||||
11396 | if (isa<BinaryOperator>(A) || | ||||
11397 | isa<CastInst>(A) || | ||||
11398 | isa<PHINode>(A) || | ||||
11399 | isa<GetElementPtrInst>(A)) | ||||
11400 | if (Instruction *BI = dyn_cast<Instruction>(B)) | ||||
11401 | if (cast<Instruction>(A)->isIdenticalTo(BI)) | ||||
11402 | return true; | ||||
11403 | |||||
11404 | // Otherwise they may not be equivalent. | ||||
11405 | return false; | ||||
11406 | } | ||||
11407 | |||||
Dale Johannesen | 2c11fe2 | 2009-03-03 21:26:39 +0000 | [diff] [blame^] | 11408 | // If this instruction has two uses, one of which is a llvm.dbg.declare, |
11409 | // return the llvm.dbg.declare. | ||||
11410 | DbgDeclareInst *InstCombiner::hasOneUsePlusDeclare(Value *V) { | ||||
11411 | if (!V->hasNUses(2)) | ||||
11412 | return 0; | ||||
11413 | for (Value::use_iterator UI = V->use_begin(), E = V->use_end(); | ||||
11414 | UI != E; ++UI) { | ||||
11415 | if (DbgDeclareInst *DI = dyn_cast<DbgDeclareInst>(UI)) | ||||
11416 | return DI; | ||||
11417 | if (isa<BitCastInst>(UI) && UI->hasOneUse()) { | ||||
11418 | if (DbgDeclareInst *DI = dyn_cast<DbgDeclareInst>(UI->use_begin())) | ||||
11419 | return DI; | ||||
11420 | } | ||||
11421 | } | ||||
11422 | return 0; | ||||
11423 | } | ||||
11424 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11425 | Instruction *InstCombiner::visitStoreInst(StoreInst &SI) { |
11426 | Value *Val = SI.getOperand(0); | ||||
11427 | Value *Ptr = SI.getOperand(1); | ||||
11428 | |||||
11429 | if (isa<UndefValue>(Ptr)) { // store X, undef -> noop (even if volatile) | ||||
11430 | EraseInstFromFunction(SI); | ||||
11431 | ++NumCombined; | ||||
11432 | return 0; | ||||
11433 | } | ||||
11434 | |||||
11435 | // If the RHS is an alloca with a single use, zapify the store, making the | ||||
11436 | // alloca dead. | ||||
Dale Johannesen | 2c11fe2 | 2009-03-03 21:26:39 +0000 | [diff] [blame^] | 11437 | // If the RHS is an alloca with a two uses, the other one being a |
11438 | // llvm.dbg.declare, zapify the store and the declare, making the | ||||
11439 | // alloca dead. We must do this to prevent declare's from affecting | ||||
11440 | // codegen. | ||||
11441 | if (!SI.isVolatile()) { | ||||
11442 | if (Ptr->hasOneUse()) { | ||||
11443 | if (isa<AllocaInst>(Ptr)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11444 | EraseInstFromFunction(SI); |
11445 | ++NumCombined; | ||||
11446 | return 0; | ||||
11447 | } | ||||
Dale Johannesen | 2c11fe2 | 2009-03-03 21:26:39 +0000 | [diff] [blame^] | 11448 | if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Ptr)) { |
11449 | if (isa<AllocaInst>(GEP->getOperand(0))) { | ||||
11450 | if (GEP->getOperand(0)->hasOneUse()) { | ||||
11451 | EraseInstFromFunction(SI); | ||||
11452 | ++NumCombined; | ||||
11453 | return 0; | ||||
11454 | } | ||||
11455 | if (DbgDeclareInst *DI = hasOneUsePlusDeclare(GEP->getOperand(0))) { | ||||
11456 | EraseInstFromFunction(*DI); | ||||
11457 | EraseInstFromFunction(SI); | ||||
11458 | ++NumCombined; | ||||
11459 | return 0; | ||||
11460 | } | ||||
11461 | } | ||||
11462 | } | ||||
11463 | } | ||||
11464 | if (DbgDeclareInst *DI = hasOneUsePlusDeclare(Ptr)) { | ||||
11465 | EraseInstFromFunction(*DI); | ||||
11466 | EraseInstFromFunction(SI); | ||||
11467 | ++NumCombined; | ||||
11468 | return 0; | ||||
11469 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11470 | } |
11471 | |||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 11472 | // Attempt to improve the alignment. |
Dan Gohman | 3719257 | 2009-02-16 00:44:23 +0000 | [diff] [blame] | 11473 | unsigned KnownAlign = |
11474 | GetOrEnforceKnownAlignment(Ptr, TD->getPrefTypeAlignment(Val->getType())); | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 11475 | if (KnownAlign > |
11476 | (SI.getAlignment() == 0 ? TD->getABITypeAlignment(Val->getType()) : | ||||
11477 | SI.getAlignment())) | ||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 11478 | SI.setAlignment(KnownAlign); |
11479 | |||||
Dale Johannesen | 2bf6a6b | 2009-03-03 01:43:03 +0000 | [diff] [blame] | 11480 | // Do really simple DSE, to catch cases where there are several consecutive |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11481 | // stores to the same location, separated by a few arithmetic operations. This |
11482 | // situation often occurs with bitfield accesses. | ||||
11483 | BasicBlock::iterator BBI = &SI; | ||||
11484 | for (unsigned ScanInsts = 6; BBI != SI.getParent()->begin() && ScanInsts; | ||||
11485 | --ScanInsts) { | ||||
Dale Johannesen | 2bf6a6b | 2009-03-03 01:43:03 +0000 | [diff] [blame] | 11486 | // Don't count debug info directives, lest they affect codegen. |
11487 | if (isa<DbgInfoIntrinsic>(BBI)) { | ||||
11488 | ScanInsts++; | ||||
11489 | --BBI; | ||||
11490 | continue; | ||||
11491 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11492 | --BBI; |
11493 | |||||
11494 | if (StoreInst *PrevSI = dyn_cast<StoreInst>(BBI)) { | ||||
11495 | // Prev store isn't volatile, and stores to the same location? | ||||
Chris Lattner | 6fd8c80 | 2008-11-27 08:56:30 +0000 | [diff] [blame] | 11496 | if (!PrevSI->isVolatile() &&equivalentAddressValues(PrevSI->getOperand(1), |
11497 | SI.getOperand(1))) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11498 | ++NumDeadStore; |
11499 | ++BBI; | ||||
11500 | EraseInstFromFunction(*PrevSI); | ||||
11501 | continue; | ||||
11502 | } | ||||
11503 | break; | ||||
11504 | } | ||||
11505 | |||||
11506 | // If this is a load, we have to stop. However, if the loaded value is from | ||||
11507 | // the pointer we're loading and is producing the pointer we're storing, | ||||
11508 | // then *this* store is dead (X = load P; store X -> P). | ||||
11509 | if (LoadInst *LI = dyn_cast<LoadInst>(BBI)) { | ||||
Dan Gohman | 0ff5a1f | 2008-10-15 23:19:35 +0000 | [diff] [blame] | 11510 | if (LI == Val && equivalentAddressValues(LI->getOperand(0), Ptr) && |
11511 | !SI.isVolatile()) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11512 | EraseInstFromFunction(SI); |
11513 | ++NumCombined; | ||||
11514 | return 0; | ||||
11515 | } | ||||
11516 | // Otherwise, this is a load from some other location. Stores before it | ||||
11517 | // may not be dead. | ||||
11518 | break; | ||||
11519 | } | ||||
11520 | |||||
11521 | // Don't skip over loads or things that can modify memory. | ||||
Chris Lattner | 8450428 | 2008-05-08 17:20:30 +0000 | [diff] [blame] | 11522 | if (BBI->mayWriteToMemory() || BBI->mayReadFromMemory()) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11523 | break; |
11524 | } | ||||
11525 | |||||
11526 | |||||
11527 | if (SI.isVolatile()) return 0; // Don't hack volatile stores. | ||||
11528 | |||||
11529 | // store X, null -> turns into 'unreachable' in SimplifyCFG | ||||
11530 | if (isa<ConstantPointerNull>(Ptr)) { | ||||
11531 | if (!isa<UndefValue>(Val)) { | ||||
11532 | SI.setOperand(0, UndefValue::get(Val->getType())); | ||||
11533 | if (Instruction *U = dyn_cast<Instruction>(Val)) | ||||
11534 | AddToWorkList(U); // Dropped a use. | ||||
11535 | ++NumCombined; | ||||
11536 | } | ||||
11537 | return 0; // Do not modify these! | ||||
11538 | } | ||||
11539 | |||||
11540 | // store undef, Ptr -> noop | ||||
11541 | if (isa<UndefValue>(Val)) { | ||||
11542 | EraseInstFromFunction(SI); | ||||
11543 | ++NumCombined; | ||||
11544 | return 0; | ||||
11545 | } | ||||
11546 | |||||
11547 | // If the pointer destination is a cast, see if we can fold the cast into the | ||||
11548 | // source instead. | ||||
11549 | if (isa<CastInst>(Ptr)) | ||||
11550 | if (Instruction *Res = InstCombineStoreToCast(*this, SI)) | ||||
11551 | return Res; | ||||
11552 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr)) | ||||
11553 | if (CE->isCast()) | ||||
11554 | if (Instruction *Res = InstCombineStoreToCast(*this, SI)) | ||||
11555 | return Res; | ||||
11556 | |||||
11557 | |||||
11558 | // If this store is the last instruction in the basic block, and if the block | ||||
11559 | // ends with an unconditional branch, try to move it to the successor block. | ||||
11560 | BBI = &SI; ++BBI; | ||||
11561 | if (BranchInst *BI = dyn_cast<BranchInst>(BBI)) | ||||
11562 | if (BI->isUnconditional()) | ||||
11563 | if (SimplifyStoreAtEndOfBlock(SI)) | ||||
11564 | return 0; // xform done! | ||||
11565 | |||||
11566 | return 0; | ||||
11567 | } | ||||
11568 | |||||
11569 | /// SimplifyStoreAtEndOfBlock - Turn things like: | ||||
11570 | /// if () { *P = v1; } else { *P = v2 } | ||||
11571 | /// into a phi node with a store in the successor. | ||||
11572 | /// | ||||
11573 | /// Simplify things like: | ||||
11574 | /// *P = v1; if () { *P = v2; } | ||||
11575 | /// into a phi node with a store in the successor. | ||||
11576 | /// | ||||
11577 | bool InstCombiner::SimplifyStoreAtEndOfBlock(StoreInst &SI) { | ||||
11578 | BasicBlock *StoreBB = SI.getParent(); | ||||
11579 | |||||
11580 | // Check to see if the successor block has exactly two incoming edges. If | ||||
11581 | // so, see if the other predecessor contains a store to the same location. | ||||
11582 | // if so, insert a PHI node (if needed) and move the stores down. | ||||
11583 | BasicBlock *DestBB = StoreBB->getTerminator()->getSuccessor(0); | ||||
11584 | |||||
11585 | // Determine whether Dest has exactly two predecessors and, if so, compute | ||||
11586 | // the other predecessor. | ||||
11587 | pred_iterator PI = pred_begin(DestBB); | ||||
11588 | BasicBlock *OtherBB = 0; | ||||
11589 | if (*PI != StoreBB) | ||||
11590 | OtherBB = *PI; | ||||
11591 | ++PI; | ||||
11592 | if (PI == pred_end(DestBB)) | ||||
11593 | return false; | ||||
11594 | |||||
11595 | if (*PI != StoreBB) { | ||||
11596 | if (OtherBB) | ||||
11597 | return false; | ||||
11598 | OtherBB = *PI; | ||||
11599 | } | ||||
11600 | if (++PI != pred_end(DestBB)) | ||||
11601 | return false; | ||||
Eli Friedman | ab39f9a | 2008-06-13 21:17:49 +0000 | [diff] [blame] | 11602 | |
11603 | // Bail out if all the relevant blocks aren't distinct (this can happen, | ||||
11604 | // for example, if SI is in an infinite loop) | ||||
11605 | if (StoreBB == DestBB || OtherBB == DestBB) | ||||
11606 | return false; | ||||
11607 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11608 | // Verify that the other block ends in a branch and is not otherwise empty. |
11609 | BasicBlock::iterator BBI = OtherBB->getTerminator(); | ||||
11610 | BranchInst *OtherBr = dyn_cast<BranchInst>(BBI); | ||||
11611 | if (!OtherBr || BBI == OtherBB->begin()) | ||||
11612 | return false; | ||||
11613 | |||||
11614 | // If the other block ends in an unconditional branch, check for the 'if then | ||||
11615 | // else' case. there is an instruction before the branch. | ||||
11616 | StoreInst *OtherStore = 0; | ||||
11617 | if (OtherBr->isUnconditional()) { | ||||
11618 | // If this isn't a store, or isn't a store to the same location, bail out. | ||||
11619 | --BBI; | ||||
11620 | OtherStore = dyn_cast<StoreInst>(BBI); | ||||
11621 | if (!OtherStore || OtherStore->getOperand(1) != SI.getOperand(1)) | ||||
11622 | return false; | ||||
11623 | } else { | ||||
11624 | // Otherwise, the other block ended with a conditional branch. If one of the | ||||
11625 | // destinations is StoreBB, then we have the if/then case. | ||||
11626 | if (OtherBr->getSuccessor(0) != StoreBB && | ||||
11627 | OtherBr->getSuccessor(1) != StoreBB) | ||||
11628 | return false; | ||||
11629 | |||||
11630 | // Okay, we know that OtherBr now goes to Dest and StoreBB, so this is an | ||||
11631 | // if/then triangle. See if there is a store to the same ptr as SI that | ||||
11632 | // lives in OtherBB. | ||||
11633 | for (;; --BBI) { | ||||
11634 | // Check to see if we find the matching store. | ||||
11635 | if ((OtherStore = dyn_cast<StoreInst>(BBI))) { | ||||
11636 | if (OtherStore->getOperand(1) != SI.getOperand(1)) | ||||
11637 | return false; | ||||
11638 | break; | ||||
11639 | } | ||||
Eli Friedman | 3a311d5 | 2008-06-13 22:02:12 +0000 | [diff] [blame] | 11640 | // If we find something that may be using or overwriting the stored |
11641 | // value, or if we run out of instructions, we can't do the xform. | ||||
11642 | if (BBI->mayReadFromMemory() || BBI->mayWriteToMemory() || | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11643 | BBI == OtherBB->begin()) |
11644 | return false; | ||||
11645 | } | ||||
11646 | |||||
11647 | // In order to eliminate the store in OtherBr, we have to | ||||
Eli Friedman | 3a311d5 | 2008-06-13 22:02:12 +0000 | [diff] [blame] | 11648 | // make sure nothing reads or overwrites the stored value in |
11649 | // StoreBB. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11650 | for (BasicBlock::iterator I = StoreBB->begin(); &*I != &SI; ++I) { |
11651 | // FIXME: This should really be AA driven. | ||||
Eli Friedman | 3a311d5 | 2008-06-13 22:02:12 +0000 | [diff] [blame] | 11652 | if (I->mayReadFromMemory() || I->mayWriteToMemory()) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11653 | return false; |
11654 | } | ||||
11655 | } | ||||
11656 | |||||
11657 | // Insert a PHI node now if we need it. | ||||
11658 | Value *MergedVal = OtherStore->getOperand(0); | ||||
11659 | if (MergedVal != SI.getOperand(0)) { | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 11660 | PHINode *PN = PHINode::Create(MergedVal->getType(), "storemerge"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11661 | PN->reserveOperandSpace(2); |
11662 | PN->addIncoming(SI.getOperand(0), SI.getParent()); | ||||
11663 | PN->addIncoming(OtherStore->getOperand(0), OtherBB); | ||||
11664 | MergedVal = InsertNewInstBefore(PN, DestBB->front()); | ||||
11665 | } | ||||
11666 | |||||
11667 | // Advance to a place where it is safe to insert the new store and | ||||
11668 | // insert it. | ||||
Dan Gohman | 514277c | 2008-05-23 21:05:58 +0000 | [diff] [blame] | 11669 | BBI = DestBB->getFirstNonPHI(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11670 | InsertNewInstBefore(new StoreInst(MergedVal, SI.getOperand(1), |
11671 | OtherStore->isVolatile()), *BBI); | ||||
11672 | |||||
11673 | // Nuke the old stores. | ||||
11674 | EraseInstFromFunction(SI); | ||||
11675 | EraseInstFromFunction(*OtherStore); | ||||
11676 | ++NumCombined; | ||||
11677 | return true; | ||||
11678 | } | ||||
11679 | |||||
11680 | |||||
11681 | Instruction *InstCombiner::visitBranchInst(BranchInst &BI) { | ||||
11682 | // Change br (not X), label True, label False to: br X, label False, True | ||||
11683 | Value *X = 0; | ||||
11684 | BasicBlock *TrueDest; | ||||
11685 | BasicBlock *FalseDest; | ||||
11686 | if (match(&BI, m_Br(m_Not(m_Value(X)), TrueDest, FalseDest)) && | ||||
11687 | !isa<Constant>(X)) { | ||||
11688 | // Swap Destinations and condition... | ||||
11689 | BI.setCondition(X); | ||||
11690 | BI.setSuccessor(0, FalseDest); | ||||
11691 | BI.setSuccessor(1, TrueDest); | ||||
11692 | return &BI; | ||||
11693 | } | ||||
11694 | |||||
11695 | // Cannonicalize fcmp_one -> fcmp_oeq | ||||
11696 | FCmpInst::Predicate FPred; Value *Y; | ||||
11697 | if (match(&BI, m_Br(m_FCmp(FPred, m_Value(X), m_Value(Y)), | ||||
11698 | TrueDest, FalseDest))) | ||||
11699 | if ((FPred == FCmpInst::FCMP_ONE || FPred == FCmpInst::FCMP_OLE || | ||||
11700 | FPred == FCmpInst::FCMP_OGE) && BI.getCondition()->hasOneUse()) { | ||||
11701 | FCmpInst *I = cast<FCmpInst>(BI.getCondition()); | ||||
11702 | FCmpInst::Predicate NewPred = FCmpInst::getInversePredicate(FPred); | ||||
11703 | Instruction *NewSCC = new FCmpInst(NewPred, X, Y, "", I); | ||||
11704 | NewSCC->takeName(I); | ||||
11705 | // Swap Destinations and condition... | ||||
11706 | BI.setCondition(NewSCC); | ||||
11707 | BI.setSuccessor(0, FalseDest); | ||||
11708 | BI.setSuccessor(1, TrueDest); | ||||
11709 | RemoveFromWorkList(I); | ||||
11710 | I->eraseFromParent(); | ||||
11711 | AddToWorkList(NewSCC); | ||||
11712 | return &BI; | ||||
11713 | } | ||||
11714 | |||||
11715 | // Cannonicalize icmp_ne -> icmp_eq | ||||
11716 | ICmpInst::Predicate IPred; | ||||
11717 | if (match(&BI, m_Br(m_ICmp(IPred, m_Value(X), m_Value(Y)), | ||||
11718 | TrueDest, FalseDest))) | ||||
11719 | if ((IPred == ICmpInst::ICMP_NE || IPred == ICmpInst::ICMP_ULE || | ||||
11720 | IPred == ICmpInst::ICMP_SLE || IPred == ICmpInst::ICMP_UGE || | ||||
11721 | IPred == ICmpInst::ICMP_SGE) && BI.getCondition()->hasOneUse()) { | ||||
11722 | ICmpInst *I = cast<ICmpInst>(BI.getCondition()); | ||||
11723 | ICmpInst::Predicate NewPred = ICmpInst::getInversePredicate(IPred); | ||||
11724 | Instruction *NewSCC = new ICmpInst(NewPred, X, Y, "", I); | ||||
11725 | NewSCC->takeName(I); | ||||
11726 | // Swap Destinations and condition... | ||||
11727 | BI.setCondition(NewSCC); | ||||
11728 | BI.setSuccessor(0, FalseDest); | ||||
11729 | BI.setSuccessor(1, TrueDest); | ||||
11730 | RemoveFromWorkList(I); | ||||
11731 | I->eraseFromParent();; | ||||
11732 | AddToWorkList(NewSCC); | ||||
11733 | return &BI; | ||||
11734 | } | ||||
11735 | |||||
11736 | return 0; | ||||
11737 | } | ||||
11738 | |||||
11739 | Instruction *InstCombiner::visitSwitchInst(SwitchInst &SI) { | ||||
11740 | Value *Cond = SI.getCondition(); | ||||
11741 | if (Instruction *I = dyn_cast<Instruction>(Cond)) { | ||||
11742 | if (I->getOpcode() == Instruction::Add) | ||||
11743 | if (ConstantInt *AddRHS = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
11744 | // change 'switch (X+4) case 1:' into 'switch (X) case -3' | ||||
11745 | for (unsigned i = 2, e = SI.getNumOperands(); i != e; i += 2) | ||||
11746 | SI.setOperand(i,ConstantExpr::getSub(cast<Constant>(SI.getOperand(i)), | ||||
11747 | AddRHS)); | ||||
11748 | SI.setOperand(0, I->getOperand(0)); | ||||
11749 | AddToWorkList(I); | ||||
11750 | return &SI; | ||||
11751 | } | ||||
11752 | } | ||||
11753 | return 0; | ||||
11754 | } | ||||
11755 | |||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 11756 | Instruction *InstCombiner::visitExtractValueInst(ExtractValueInst &EV) { |
Matthijs Kooijman | 45e8eb4 | 2008-07-16 12:55:45 +0000 | [diff] [blame] | 11757 | Value *Agg = EV.getAggregateOperand(); |
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 11758 | |
Matthijs Kooijman | 45e8eb4 | 2008-07-16 12:55:45 +0000 | [diff] [blame] | 11759 | if (!EV.hasIndices()) |
11760 | return ReplaceInstUsesWith(EV, Agg); | ||||
11761 | |||||
11762 | if (Constant *C = dyn_cast<Constant>(Agg)) { | ||||
11763 | if (isa<UndefValue>(C)) | ||||
11764 | return ReplaceInstUsesWith(EV, UndefValue::get(EV.getType())); | ||||
11765 | |||||
11766 | if (isa<ConstantAggregateZero>(C)) | ||||
11767 | return ReplaceInstUsesWith(EV, Constant::getNullValue(EV.getType())); | ||||
11768 | |||||
11769 | if (isa<ConstantArray>(C) || isa<ConstantStruct>(C)) { | ||||
11770 | // Extract the element indexed by the first index out of the constant | ||||
11771 | Value *V = C->getOperand(*EV.idx_begin()); | ||||
11772 | if (EV.getNumIndices() > 1) | ||||
11773 | // Extract the remaining indices out of the constant indexed by the | ||||
11774 | // first index | ||||
11775 | return ExtractValueInst::Create(V, EV.idx_begin() + 1, EV.idx_end()); | ||||
11776 | else | ||||
11777 | return ReplaceInstUsesWith(EV, V); | ||||
11778 | } | ||||
11779 | return 0; // Can't handle other constants | ||||
11780 | } | ||||
11781 | if (InsertValueInst *IV = dyn_cast<InsertValueInst>(Agg)) { | ||||
11782 | // We're extracting from an insertvalue instruction, compare the indices | ||||
11783 | const unsigned *exti, *exte, *insi, *inse; | ||||
11784 | for (exti = EV.idx_begin(), insi = IV->idx_begin(), | ||||
11785 | exte = EV.idx_end(), inse = IV->idx_end(); | ||||
11786 | exti != exte && insi != inse; | ||||
11787 | ++exti, ++insi) { | ||||
11788 | if (*insi != *exti) | ||||
11789 | // The insert and extract both reference distinctly different elements. | ||||
11790 | // This means the extract is not influenced by the insert, and we can | ||||
11791 | // replace the aggregate operand of the extract with the aggregate | ||||
11792 | // operand of the insert. i.e., replace | ||||
11793 | // %I = insertvalue { i32, { i32 } } %A, { i32 } { i32 42 }, 1 | ||||
11794 | // %E = extractvalue { i32, { i32 } } %I, 0 | ||||
11795 | // with | ||||
11796 | // %E = extractvalue { i32, { i32 } } %A, 0 | ||||
11797 | return ExtractValueInst::Create(IV->getAggregateOperand(), | ||||
11798 | EV.idx_begin(), EV.idx_end()); | ||||
11799 | } | ||||
11800 | if (exti == exte && insi == inse) | ||||
11801 | // Both iterators are at the end: Index lists are identical. Replace | ||||
11802 | // %B = insertvalue { i32, { i32 } } %A, i32 42, 1, 0 | ||||
11803 | // %C = extractvalue { i32, { i32 } } %B, 1, 0 | ||||
11804 | // with "i32 42" | ||||
11805 | return ReplaceInstUsesWith(EV, IV->getInsertedValueOperand()); | ||||
11806 | if (exti == exte) { | ||||
11807 | // The extract list is a prefix of the insert list. i.e. replace | ||||
11808 | // %I = insertvalue { i32, { i32 } } %A, i32 42, 1, 0 | ||||
11809 | // %E = extractvalue { i32, { i32 } } %I, 1 | ||||
11810 | // with | ||||
11811 | // %X = extractvalue { i32, { i32 } } %A, 1 | ||||
11812 | // %E = insertvalue { i32 } %X, i32 42, 0 | ||||
11813 | // by switching the order of the insert and extract (though the | ||||
11814 | // insertvalue should be left in, since it may have other uses). | ||||
11815 | Value *NewEV = InsertNewInstBefore( | ||||
11816 | ExtractValueInst::Create(IV->getAggregateOperand(), | ||||
11817 | EV.idx_begin(), EV.idx_end()), | ||||
11818 | EV); | ||||
11819 | return InsertValueInst::Create(NewEV, IV->getInsertedValueOperand(), | ||||
11820 | insi, inse); | ||||
11821 | } | ||||
11822 | if (insi == inse) | ||||
11823 | // The insert list is a prefix of the extract list | ||||
11824 | // We can simply remove the common indices from the extract and make it | ||||
11825 | // operate on the inserted value instead of the insertvalue result. | ||||
11826 | // i.e., replace | ||||
11827 | // %I = insertvalue { i32, { i32 } } %A, { i32 } { i32 42 }, 1 | ||||
11828 | // %E = extractvalue { i32, { i32 } } %I, 1, 0 | ||||
11829 | // with | ||||
11830 | // %E extractvalue { i32 } { i32 42 }, 0 | ||||
11831 | return ExtractValueInst::Create(IV->getInsertedValueOperand(), | ||||
11832 | exti, exte); | ||||
11833 | } | ||||
11834 | // Can't simplify extracts from other values. Note that nested extracts are | ||||
11835 | // already simplified implicitely by the above (extract ( extract (insert) ) | ||||
11836 | // will be translated into extract ( insert ( extract ) ) first and then just | ||||
11837 | // the value inserted, if appropriate). | ||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 11838 | return 0; |
11839 | } | ||||
11840 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11841 | /// CheapToScalarize - Return true if the value is cheaper to scalarize than it |
11842 | /// is to leave as a vector operation. | ||||
11843 | static bool CheapToScalarize(Value *V, bool isConstant) { | ||||
11844 | if (isa<ConstantAggregateZero>(V)) | ||||
11845 | return true; | ||||
11846 | if (ConstantVector *C = dyn_cast<ConstantVector>(V)) { | ||||
11847 | if (isConstant) return true; | ||||
11848 | // If all elts are the same, we can extract. | ||||
11849 | Constant *Op0 = C->getOperand(0); | ||||
11850 | for (unsigned i = 1; i < C->getNumOperands(); ++i) | ||||
11851 | if (C->getOperand(i) != Op0) | ||||
11852 | return false; | ||||
11853 | return true; | ||||
11854 | } | ||||
11855 | Instruction *I = dyn_cast<Instruction>(V); | ||||
11856 | if (!I) return false; | ||||
11857 | |||||
11858 | // Insert element gets simplified to the inserted element or is deleted if | ||||
11859 | // this is constant idx extract element and its a constant idx insertelt. | ||||
11860 | if (I->getOpcode() == Instruction::InsertElement && isConstant && | ||||
11861 | isa<ConstantInt>(I->getOperand(2))) | ||||
11862 | return true; | ||||
11863 | if (I->getOpcode() == Instruction::Load && I->hasOneUse()) | ||||
11864 | return true; | ||||
11865 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(I)) | ||||
11866 | if (BO->hasOneUse() && | ||||
11867 | (CheapToScalarize(BO->getOperand(0), isConstant) || | ||||
11868 | CheapToScalarize(BO->getOperand(1), isConstant))) | ||||
11869 | return true; | ||||
11870 | if (CmpInst *CI = dyn_cast<CmpInst>(I)) | ||||
11871 | if (CI->hasOneUse() && | ||||
11872 | (CheapToScalarize(CI->getOperand(0), isConstant) || | ||||
11873 | CheapToScalarize(CI->getOperand(1), isConstant))) | ||||
11874 | return true; | ||||
11875 | |||||
11876 | return false; | ||||
11877 | } | ||||
11878 | |||||
11879 | /// Read and decode a shufflevector mask. | ||||
11880 | /// | ||||
11881 | /// It turns undef elements into values that are larger than the number of | ||||
11882 | /// elements in the input. | ||||
11883 | static std::vector<unsigned> getShuffleMask(const ShuffleVectorInst *SVI) { | ||||
11884 | unsigned NElts = SVI->getType()->getNumElements(); | ||||
11885 | if (isa<ConstantAggregateZero>(SVI->getOperand(2))) | ||||
11886 | return std::vector<unsigned>(NElts, 0); | ||||
11887 | if (isa<UndefValue>(SVI->getOperand(2))) | ||||
11888 | return std::vector<unsigned>(NElts, 2*NElts); | ||||
11889 | |||||
11890 | std::vector<unsigned> Result; | ||||
11891 | const ConstantVector *CP = cast<ConstantVector>(SVI->getOperand(2)); | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 11892 | for (User::const_op_iterator i = CP->op_begin(), e = CP->op_end(); i!=e; ++i) |
11893 | if (isa<UndefValue>(*i)) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11894 | Result.push_back(NElts*2); // undef -> 8 |
11895 | else | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 11896 | Result.push_back(cast<ConstantInt>(*i)->getZExtValue()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11897 | return Result; |
11898 | } | ||||
11899 | |||||
11900 | /// FindScalarElement - Given a vector and an element number, see if the scalar | ||||
11901 | /// value is already around as a register, for example if it were inserted then | ||||
11902 | /// extracted from the vector. | ||||
11903 | static Value *FindScalarElement(Value *V, unsigned EltNo) { | ||||
11904 | assert(isa<VectorType>(V->getType()) && "Not looking at a vector?"); | ||||
11905 | const VectorType *PTy = cast<VectorType>(V->getType()); | ||||
11906 | unsigned Width = PTy->getNumElements(); | ||||
11907 | if (EltNo >= Width) // Out of range access. | ||||
11908 | return UndefValue::get(PTy->getElementType()); | ||||
11909 | |||||
11910 | if (isa<UndefValue>(V)) | ||||
11911 | return UndefValue::get(PTy->getElementType()); | ||||
11912 | else if (isa<ConstantAggregateZero>(V)) | ||||
11913 | return Constant::getNullValue(PTy->getElementType()); | ||||
11914 | else if (ConstantVector *CP = dyn_cast<ConstantVector>(V)) | ||||
11915 | return CP->getOperand(EltNo); | ||||
11916 | else if (InsertElementInst *III = dyn_cast<InsertElementInst>(V)) { | ||||
11917 | // If this is an insert to a variable element, we don't know what it is. | ||||
11918 | if (!isa<ConstantInt>(III->getOperand(2))) | ||||
11919 | return 0; | ||||
11920 | unsigned IIElt = cast<ConstantInt>(III->getOperand(2))->getZExtValue(); | ||||
11921 | |||||
11922 | // If this is an insert to the element we are looking for, return the | ||||
11923 | // inserted value. | ||||
11924 | if (EltNo == IIElt) | ||||
11925 | return III->getOperand(1); | ||||
11926 | |||||
11927 | // Otherwise, the insertelement doesn't modify the value, recurse on its | ||||
11928 | // vector input. | ||||
11929 | return FindScalarElement(III->getOperand(0), EltNo); | ||||
11930 | } else if (ShuffleVectorInst *SVI = dyn_cast<ShuffleVectorInst>(V)) { | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 11931 | unsigned LHSWidth = |
11932 | cast<VectorType>(SVI->getOperand(0)->getType())->getNumElements(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11933 | unsigned InEl = getShuffleMask(SVI)[EltNo]; |
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 11934 | if (InEl < LHSWidth) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11935 | return FindScalarElement(SVI->getOperand(0), InEl); |
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 11936 | else if (InEl < LHSWidth*2) |
11937 | return FindScalarElement(SVI->getOperand(1), InEl - LHSWidth); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11938 | else |
11939 | return UndefValue::get(PTy->getElementType()); | ||||
11940 | } | ||||
11941 | |||||
11942 | // Otherwise, we don't know. | ||||
11943 | return 0; | ||||
11944 | } | ||||
11945 | |||||
11946 | Instruction *InstCombiner::visitExtractElementInst(ExtractElementInst &EI) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11947 | // If vector val is undef, replace extract with scalar undef. |
11948 | if (isa<UndefValue>(EI.getOperand(0))) | ||||
11949 | return ReplaceInstUsesWith(EI, UndefValue::get(EI.getType())); | ||||
11950 | |||||
11951 | // If vector val is constant 0, replace extract with scalar 0. | ||||
11952 | if (isa<ConstantAggregateZero>(EI.getOperand(0))) | ||||
11953 | return ReplaceInstUsesWith(EI, Constant::getNullValue(EI.getType())); | ||||
11954 | |||||
11955 | if (ConstantVector *C = dyn_cast<ConstantVector>(EI.getOperand(0))) { | ||||
Matthijs Kooijman | dd3425f | 2008-06-11 09:00:12 +0000 | [diff] [blame] | 11956 | // If vector val is constant with all elements the same, replace EI with |
11957 | // that element. When the elements are not identical, we cannot replace yet | ||||
11958 | // (we do that below, but only when the index is constant). | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11959 | Constant *op0 = C->getOperand(0); |
11960 | for (unsigned i = 1; i < C->getNumOperands(); ++i) | ||||
11961 | if (C->getOperand(i) != op0) { | ||||
11962 | op0 = 0; | ||||
11963 | break; | ||||
11964 | } | ||||
11965 | if (op0) | ||||
11966 | return ReplaceInstUsesWith(EI, op0); | ||||
11967 | } | ||||
11968 | |||||
11969 | // If extracting a specified index from the vector, see if we can recursively | ||||
11970 | // find a previously computed scalar that was inserted into the vector. | ||||
11971 | if (ConstantInt *IdxC = dyn_cast<ConstantInt>(EI.getOperand(1))) { | ||||
11972 | unsigned IndexVal = IdxC->getZExtValue(); | ||||
11973 | unsigned VectorWidth = | ||||
11974 | cast<VectorType>(EI.getOperand(0)->getType())->getNumElements(); | ||||
11975 | |||||
11976 | // If this is extracting an invalid index, turn this into undef, to avoid | ||||
11977 | // crashing the code below. | ||||
11978 | if (IndexVal >= VectorWidth) | ||||
11979 | return ReplaceInstUsesWith(EI, UndefValue::get(EI.getType())); | ||||
11980 | |||||
11981 | // This instruction only demands the single element from the input vector. | ||||
11982 | // If the input vector has a single use, simplify it based on this use | ||||
11983 | // property. | ||||
11984 | if (EI.getOperand(0)->hasOneUse() && VectorWidth != 1) { | ||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 11985 | APInt UndefElts(VectorWidth, 0); |
11986 | APInt DemandedMask(VectorWidth, 1 << IndexVal); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11987 | if (Value *V = SimplifyDemandedVectorElts(EI.getOperand(0), |
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 11988 | DemandedMask, UndefElts)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11989 | EI.setOperand(0, V); |
11990 | return &EI; | ||||
11991 | } | ||||
11992 | } | ||||
11993 | |||||
11994 | if (Value *Elt = FindScalarElement(EI.getOperand(0), IndexVal)) | ||||
11995 | return ReplaceInstUsesWith(EI, Elt); | ||||
11996 | |||||
11997 | // If the this extractelement is directly using a bitcast from a vector of | ||||
11998 | // the same number of elements, see if we can find the source element from | ||||
11999 | // it. In this case, we will end up needing to bitcast the scalars. | ||||
12000 | if (BitCastInst *BCI = dyn_cast<BitCastInst>(EI.getOperand(0))) { | ||||
12001 | if (const VectorType *VT = | ||||
12002 | dyn_cast<VectorType>(BCI->getOperand(0)->getType())) | ||||
12003 | if (VT->getNumElements() == VectorWidth) | ||||
12004 | if (Value *Elt = FindScalarElement(BCI->getOperand(0), IndexVal)) | ||||
12005 | return new BitCastInst(Elt, EI.getType()); | ||||
12006 | } | ||||
12007 | } | ||||
12008 | |||||
12009 | if (Instruction *I = dyn_cast<Instruction>(EI.getOperand(0))) { | ||||
12010 | if (I->hasOneUse()) { | ||||
12011 | // Push extractelement into predecessor operation if legal and | ||||
12012 | // profitable to do so | ||||
12013 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(I)) { | ||||
12014 | bool isConstantElt = isa<ConstantInt>(EI.getOperand(1)); | ||||
12015 | if (CheapToScalarize(BO, isConstantElt)) { | ||||
12016 | ExtractElementInst *newEI0 = | ||||
12017 | new ExtractElementInst(BO->getOperand(0), EI.getOperand(1), | ||||
12018 | EI.getName()+".lhs"); | ||||
12019 | ExtractElementInst *newEI1 = | ||||
12020 | new ExtractElementInst(BO->getOperand(1), EI.getOperand(1), | ||||
12021 | EI.getName()+".rhs"); | ||||
12022 | InsertNewInstBefore(newEI0, EI); | ||||
12023 | InsertNewInstBefore(newEI1, EI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 12024 | return BinaryOperator::Create(BO->getOpcode(), newEI0, newEI1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12025 | } |
12026 | } else if (isa<LoadInst>(I)) { | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 12027 | unsigned AS = |
12028 | cast<PointerType>(I->getOperand(0)->getType())->getAddressSpace(); | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 12029 | Value *Ptr = InsertBitCastBefore(I->getOperand(0), |
12030 | PointerType::get(EI.getType(), AS),EI); | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 12031 | GetElementPtrInst *GEP = |
12032 | GetElementPtrInst::Create(Ptr, EI.getOperand(1), I->getName()+".gep"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12033 | InsertNewInstBefore(GEP, EI); |
12034 | return new LoadInst(GEP); | ||||
12035 | } | ||||
12036 | } | ||||
12037 | if (InsertElementInst *IE = dyn_cast<InsertElementInst>(I)) { | ||||
12038 | // Extracting the inserted element? | ||||
12039 | if (IE->getOperand(2) == EI.getOperand(1)) | ||||
12040 | return ReplaceInstUsesWith(EI, IE->getOperand(1)); | ||||
12041 | // If the inserted and extracted elements are constants, they must not | ||||
12042 | // be the same value, extract from the pre-inserted value instead. | ||||
12043 | if (isa<Constant>(IE->getOperand(2)) && | ||||
12044 | isa<Constant>(EI.getOperand(1))) { | ||||
12045 | AddUsesToWorkList(EI); | ||||
12046 | EI.setOperand(0, IE->getOperand(0)); | ||||
12047 | return &EI; | ||||
12048 | } | ||||
12049 | } else if (ShuffleVectorInst *SVI = dyn_cast<ShuffleVectorInst>(I)) { | ||||
12050 | // If this is extracting an element from a shufflevector, figure out where | ||||
12051 | // it came from and extract from the appropriate input element instead. | ||||
12052 | if (ConstantInt *Elt = dyn_cast<ConstantInt>(EI.getOperand(1))) { | ||||
12053 | unsigned SrcIdx = getShuffleMask(SVI)[Elt->getZExtValue()]; | ||||
12054 | Value *Src; | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 12055 | unsigned LHSWidth = |
12056 | cast<VectorType>(SVI->getOperand(0)->getType())->getNumElements(); | ||||
12057 | |||||
12058 | if (SrcIdx < LHSWidth) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12059 | Src = SVI->getOperand(0); |
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 12060 | else if (SrcIdx < LHSWidth*2) { |
12061 | SrcIdx -= LHSWidth; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12062 | Src = SVI->getOperand(1); |
12063 | } else { | ||||
12064 | return ReplaceInstUsesWith(EI, UndefValue::get(EI.getType())); | ||||
12065 | } | ||||
12066 | return new ExtractElementInst(Src, SrcIdx); | ||||
12067 | } | ||||
12068 | } | ||||
12069 | } | ||||
12070 | return 0; | ||||
12071 | } | ||||
12072 | |||||
12073 | /// CollectSingleShuffleElements - If V is a shuffle of values that ONLY returns | ||||
12074 | /// elements from either LHS or RHS, return the shuffle mask and true. | ||||
12075 | /// Otherwise, return false. | ||||
12076 | static bool CollectSingleShuffleElements(Value *V, Value *LHS, Value *RHS, | ||||
12077 | std::vector<Constant*> &Mask) { | ||||
12078 | assert(V->getType() == LHS->getType() && V->getType() == RHS->getType() && | ||||
12079 | "Invalid CollectSingleShuffleElements"); | ||||
12080 | unsigned NumElts = cast<VectorType>(V->getType())->getNumElements(); | ||||
12081 | |||||
12082 | if (isa<UndefValue>(V)) { | ||||
12083 | Mask.assign(NumElts, UndefValue::get(Type::Int32Ty)); | ||||
12084 | return true; | ||||
12085 | } else if (V == LHS) { | ||||
12086 | for (unsigned i = 0; i != NumElts; ++i) | ||||
12087 | Mask.push_back(ConstantInt::get(Type::Int32Ty, i)); | ||||
12088 | return true; | ||||
12089 | } else if (V == RHS) { | ||||
12090 | for (unsigned i = 0; i != NumElts; ++i) | ||||
12091 | Mask.push_back(ConstantInt::get(Type::Int32Ty, i+NumElts)); | ||||
12092 | return true; | ||||
12093 | } else if (InsertElementInst *IEI = dyn_cast<InsertElementInst>(V)) { | ||||
12094 | // If this is an insert of an extract from some other vector, include it. | ||||
12095 | Value *VecOp = IEI->getOperand(0); | ||||
12096 | Value *ScalarOp = IEI->getOperand(1); | ||||
12097 | Value *IdxOp = IEI->getOperand(2); | ||||
12098 | |||||
12099 | if (!isa<ConstantInt>(IdxOp)) | ||||
12100 | return false; | ||||
12101 | unsigned InsertedIdx = cast<ConstantInt>(IdxOp)->getZExtValue(); | ||||
12102 | |||||
12103 | if (isa<UndefValue>(ScalarOp)) { // inserting undef into vector. | ||||
12104 | // Okay, we can handle this if the vector we are insertinting into is | ||||
12105 | // transitively ok. | ||||
12106 | if (CollectSingleShuffleElements(VecOp, LHS, RHS, Mask)) { | ||||
12107 | // If so, update the mask to reflect the inserted undef. | ||||
12108 | Mask[InsertedIdx] = UndefValue::get(Type::Int32Ty); | ||||
12109 | return true; | ||||
12110 | } | ||||
12111 | } else if (ExtractElementInst *EI = dyn_cast<ExtractElementInst>(ScalarOp)){ | ||||
12112 | if (isa<ConstantInt>(EI->getOperand(1)) && | ||||
12113 | EI->getOperand(0)->getType() == V->getType()) { | ||||
12114 | unsigned ExtractedIdx = | ||||
12115 | cast<ConstantInt>(EI->getOperand(1))->getZExtValue(); | ||||
12116 | |||||
12117 | // This must be extracting from either LHS or RHS. | ||||
12118 | if (EI->getOperand(0) == LHS || EI->getOperand(0) == RHS) { | ||||
12119 | // Okay, we can handle this if the vector we are insertinting into is | ||||
12120 | // transitively ok. | ||||
12121 | if (CollectSingleShuffleElements(VecOp, LHS, RHS, Mask)) { | ||||
12122 | // If so, update the mask to reflect the inserted value. | ||||
12123 | if (EI->getOperand(0) == LHS) { | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 12124 | Mask[InsertedIdx % NumElts] = |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12125 | ConstantInt::get(Type::Int32Ty, ExtractedIdx); |
12126 | } else { | ||||
12127 | assert(EI->getOperand(0) == RHS); | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 12128 | Mask[InsertedIdx % NumElts] = |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12129 | ConstantInt::get(Type::Int32Ty, ExtractedIdx+NumElts); |
12130 | |||||
12131 | } | ||||
12132 | return true; | ||||
12133 | } | ||||
12134 | } | ||||
12135 | } | ||||
12136 | } | ||||
12137 | } | ||||
12138 | // TODO: Handle shufflevector here! | ||||
12139 | |||||
12140 | return false; | ||||
12141 | } | ||||
12142 | |||||
12143 | /// CollectShuffleElements - We are building a shuffle of V, using RHS as the | ||||
12144 | /// RHS of the shuffle instruction, if it is not null. Return a shuffle mask | ||||
12145 | /// that computes V and the LHS value of the shuffle. | ||||
12146 | static Value *CollectShuffleElements(Value *V, std::vector<Constant*> &Mask, | ||||
12147 | Value *&RHS) { | ||||
12148 | assert(isa<VectorType>(V->getType()) && | ||||
12149 | (RHS == 0 || V->getType() == RHS->getType()) && | ||||
12150 | "Invalid shuffle!"); | ||||
12151 | unsigned NumElts = cast<VectorType>(V->getType())->getNumElements(); | ||||
12152 | |||||
12153 | if (isa<UndefValue>(V)) { | ||||
12154 | Mask.assign(NumElts, UndefValue::get(Type::Int32Ty)); | ||||
12155 | return V; | ||||
12156 | } else if (isa<ConstantAggregateZero>(V)) { | ||||
12157 | Mask.assign(NumElts, ConstantInt::get(Type::Int32Ty, 0)); | ||||
12158 | return V; | ||||
12159 | } else if (InsertElementInst *IEI = dyn_cast<InsertElementInst>(V)) { | ||||
12160 | // If this is an insert of an extract from some other vector, include it. | ||||
12161 | Value *VecOp = IEI->getOperand(0); | ||||
12162 | Value *ScalarOp = IEI->getOperand(1); | ||||
12163 | Value *IdxOp = IEI->getOperand(2); | ||||
12164 | |||||
12165 | if (ExtractElementInst *EI = dyn_cast<ExtractElementInst>(ScalarOp)) { | ||||
12166 | if (isa<ConstantInt>(EI->getOperand(1)) && isa<ConstantInt>(IdxOp) && | ||||
12167 | EI->getOperand(0)->getType() == V->getType()) { | ||||
12168 | unsigned ExtractedIdx = | ||||
12169 | cast<ConstantInt>(EI->getOperand(1))->getZExtValue(); | ||||
12170 | unsigned InsertedIdx = cast<ConstantInt>(IdxOp)->getZExtValue(); | ||||
12171 | |||||
12172 | // Either the extracted from or inserted into vector must be RHSVec, | ||||
12173 | // otherwise we'd end up with a shuffle of three inputs. | ||||
12174 | if (EI->getOperand(0) == RHS || RHS == 0) { | ||||
12175 | RHS = EI->getOperand(0); | ||||
12176 | Value *V = CollectShuffleElements(VecOp, Mask, RHS); | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 12177 | Mask[InsertedIdx % NumElts] = |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12178 | ConstantInt::get(Type::Int32Ty, NumElts+ExtractedIdx); |
12179 | return V; | ||||
12180 | } | ||||
12181 | |||||
12182 | if (VecOp == RHS) { | ||||
12183 | Value *V = CollectShuffleElements(EI->getOperand(0), Mask, RHS); | ||||
12184 | // Everything but the extracted element is replaced with the RHS. | ||||
12185 | for (unsigned i = 0; i != NumElts; ++i) { | ||||
12186 | if (i != InsertedIdx) | ||||
12187 | Mask[i] = ConstantInt::get(Type::Int32Ty, NumElts+i); | ||||
12188 | } | ||||
12189 | return V; | ||||
12190 | } | ||||
12191 | |||||
12192 | // If this insertelement is a chain that comes from exactly these two | ||||
12193 | // vectors, return the vector and the effective shuffle. | ||||
12194 | if (CollectSingleShuffleElements(IEI, EI->getOperand(0), RHS, Mask)) | ||||
12195 | return EI->getOperand(0); | ||||
12196 | |||||
12197 | } | ||||
12198 | } | ||||
12199 | } | ||||
12200 | // TODO: Handle shufflevector here! | ||||
12201 | |||||
12202 | // Otherwise, can't do anything fancy. Return an identity vector. | ||||
12203 | for (unsigned i = 0; i != NumElts; ++i) | ||||
12204 | Mask.push_back(ConstantInt::get(Type::Int32Ty, i)); | ||||
12205 | return V; | ||||
12206 | } | ||||
12207 | |||||
12208 | Instruction *InstCombiner::visitInsertElementInst(InsertElementInst &IE) { | ||||
12209 | Value *VecOp = IE.getOperand(0); | ||||
12210 | Value *ScalarOp = IE.getOperand(1); | ||||
12211 | Value *IdxOp = IE.getOperand(2); | ||||
12212 | |||||
12213 | // Inserting an undef or into an undefined place, remove this. | ||||
12214 | if (isa<UndefValue>(ScalarOp) || isa<UndefValue>(IdxOp)) | ||||
12215 | ReplaceInstUsesWith(IE, VecOp); | ||||
12216 | |||||
12217 | // If the inserted element was extracted from some other vector, and if the | ||||
12218 | // indexes are constant, try to turn this into a shufflevector operation. | ||||
12219 | if (ExtractElementInst *EI = dyn_cast<ExtractElementInst>(ScalarOp)) { | ||||
12220 | if (isa<ConstantInt>(EI->getOperand(1)) && isa<ConstantInt>(IdxOp) && | ||||
12221 | EI->getOperand(0)->getType() == IE.getType()) { | ||||
12222 | unsigned NumVectorElts = IE.getType()->getNumElements(); | ||||
12223 | unsigned ExtractedIdx = | ||||
12224 | cast<ConstantInt>(EI->getOperand(1))->getZExtValue(); | ||||
12225 | unsigned InsertedIdx = cast<ConstantInt>(IdxOp)->getZExtValue(); | ||||
12226 | |||||
12227 | if (ExtractedIdx >= NumVectorElts) // Out of range extract. | ||||
12228 | return ReplaceInstUsesWith(IE, VecOp); | ||||
12229 | |||||
12230 | if (InsertedIdx >= NumVectorElts) // Out of range insert. | ||||
12231 | return ReplaceInstUsesWith(IE, UndefValue::get(IE.getType())); | ||||
12232 | |||||
12233 | // If we are extracting a value from a vector, then inserting it right | ||||
12234 | // back into the same place, just use the input vector. | ||||
12235 | if (EI->getOperand(0) == VecOp && ExtractedIdx == InsertedIdx) | ||||
12236 | return ReplaceInstUsesWith(IE, VecOp); | ||||
12237 | |||||
12238 | // We could theoretically do this for ANY input. However, doing so could | ||||
12239 | // turn chains of insertelement instructions into a chain of shufflevector | ||||
12240 | // instructions, and right now we do not merge shufflevectors. As such, | ||||
12241 | // only do this in a situation where it is clear that there is benefit. | ||||
12242 | if (isa<UndefValue>(VecOp) || isa<ConstantAggregateZero>(VecOp)) { | ||||
12243 | // Turn this into shuffle(EIOp0, VecOp, Mask). The result has all of | ||||
12244 | // the values of VecOp, except then one read from EIOp0. | ||||
12245 | // Build a new shuffle mask. | ||||
12246 | std::vector<Constant*> Mask; | ||||
12247 | if (isa<UndefValue>(VecOp)) | ||||
12248 | Mask.assign(NumVectorElts, UndefValue::get(Type::Int32Ty)); | ||||
12249 | else { | ||||
12250 | assert(isa<ConstantAggregateZero>(VecOp) && "Unknown thing"); | ||||
12251 | Mask.assign(NumVectorElts, ConstantInt::get(Type::Int32Ty, | ||||
12252 | NumVectorElts)); | ||||
12253 | } | ||||
12254 | Mask[InsertedIdx] = ConstantInt::get(Type::Int32Ty, ExtractedIdx); | ||||
12255 | return new ShuffleVectorInst(EI->getOperand(0), VecOp, | ||||
12256 | ConstantVector::get(Mask)); | ||||
12257 | } | ||||
12258 | |||||
12259 | // If this insertelement isn't used by some other insertelement, turn it | ||||
12260 | // (and any insertelements it points to), into one big shuffle. | ||||
12261 | if (!IE.hasOneUse() || !isa<InsertElementInst>(IE.use_back())) { | ||||
12262 | std::vector<Constant*> Mask; | ||||
12263 | Value *RHS = 0; | ||||
12264 | Value *LHS = CollectShuffleElements(&IE, Mask, RHS); | ||||
12265 | if (RHS == 0) RHS = UndefValue::get(LHS->getType()); | ||||
12266 | // We now have a shuffle of LHS, RHS, Mask. | ||||
12267 | return new ShuffleVectorInst(LHS, RHS, ConstantVector::get(Mask)); | ||||
12268 | } | ||||
12269 | } | ||||
12270 | } | ||||
12271 | |||||
12272 | return 0; | ||||
12273 | } | ||||
12274 | |||||
12275 | |||||
12276 | Instruction *InstCombiner::visitShuffleVectorInst(ShuffleVectorInst &SVI) { | ||||
12277 | Value *LHS = SVI.getOperand(0); | ||||
12278 | Value *RHS = SVI.getOperand(1); | ||||
12279 | std::vector<unsigned> Mask = getShuffleMask(&SVI); | ||||
12280 | |||||
12281 | bool MadeChange = false; | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 12282 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12283 | // Undefined shuffle mask -> undefined value. |
12284 | if (isa<UndefValue>(SVI.getOperand(2))) | ||||
12285 | return ReplaceInstUsesWith(SVI, UndefValue::get(SVI.getType())); | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 12286 | |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 12287 | unsigned VWidth = cast<VectorType>(SVI.getType())->getNumElements(); |
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 12288 | |
12289 | if (VWidth != cast<VectorType>(LHS->getType())->getNumElements()) | ||||
12290 | return 0; | ||||
12291 | |||||
Evan Cheng | 63295ab | 2009-02-03 10:05:09 +0000 | [diff] [blame] | 12292 | APInt UndefElts(VWidth, 0); |
12293 | APInt AllOnesEltMask(APInt::getAllOnesValue(VWidth)); | ||||
12294 | if (SimplifyDemandedVectorElts(&SVI, AllOnesEltMask, UndefElts)) { | ||||
Dan Gohman | 83b702d | 2008-09-11 22:47:57 +0000 | [diff] [blame] | 12295 | LHS = SVI.getOperand(0); |
12296 | RHS = SVI.getOperand(1); | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 12297 | MadeChange = true; |
Dan Gohman | 83b702d | 2008-09-11 22:47:57 +0000 | [diff] [blame] | 12298 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12299 | |
12300 | // Canonicalize shuffle(x ,x,mask) -> shuffle(x, undef,mask') | ||||
12301 | // Canonicalize shuffle(undef,x,mask) -> shuffle(x, undef,mask'). | ||||
12302 | if (LHS == RHS || isa<UndefValue>(LHS)) { | ||||
12303 | if (isa<UndefValue>(LHS) && LHS == RHS) { | ||||
12304 | // shuffle(undef,undef,mask) -> undef. | ||||
12305 | return ReplaceInstUsesWith(SVI, LHS); | ||||
12306 | } | ||||
12307 | |||||
12308 | // Remap any references to RHS to use LHS. | ||||
12309 | std::vector<Constant*> Elts; | ||||
12310 | for (unsigned i = 0, e = Mask.size(); i != e; ++i) { | ||||
12311 | if (Mask[i] >= 2*e) | ||||
12312 | Elts.push_back(UndefValue::get(Type::Int32Ty)); | ||||
12313 | else { | ||||
12314 | if ((Mask[i] >= e && isa<UndefValue>(RHS)) || | ||||
Dan Gohman | bba96b9 | 2008-08-06 18:17:32 +0000 | [diff] [blame] | 12315 | (Mask[i] < e && isa<UndefValue>(LHS))) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12316 | Mask[i] = 2*e; // Turn into undef. |
Dan Gohman | bba96b9 | 2008-08-06 18:17:32 +0000 | [diff] [blame] | 12317 | Elts.push_back(UndefValue::get(Type::Int32Ty)); |
12318 | } else { | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 12319 | Mask[i] = Mask[i] % e; // Force to LHS. |
Dan Gohman | bba96b9 | 2008-08-06 18:17:32 +0000 | [diff] [blame] | 12320 | Elts.push_back(ConstantInt::get(Type::Int32Ty, Mask[i])); |
12321 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12322 | } |
12323 | } | ||||
12324 | SVI.setOperand(0, SVI.getOperand(1)); | ||||
12325 | SVI.setOperand(1, UndefValue::get(RHS->getType())); | ||||
12326 | SVI.setOperand(2, ConstantVector::get(Elts)); | ||||
12327 | LHS = SVI.getOperand(0); | ||||
12328 | RHS = SVI.getOperand(1); | ||||
12329 | MadeChange = true; | ||||
12330 | } | ||||
12331 | |||||
12332 | // Analyze the shuffle, are the LHS or RHS and identity shuffles? | ||||
12333 | bool isLHSID = true, isRHSID = true; | ||||
12334 | |||||
12335 | for (unsigned i = 0, e = Mask.size(); i != e; ++i) { | ||||
12336 | if (Mask[i] >= e*2) continue; // Ignore undef values. | ||||
12337 | // Is this an identity shuffle of the LHS value? | ||||
12338 | isLHSID &= (Mask[i] == i); | ||||
12339 | |||||
12340 | // Is this an identity shuffle of the RHS value? | ||||
12341 | isRHSID &= (Mask[i]-e == i); | ||||
12342 | } | ||||
12343 | |||||
12344 | // Eliminate identity shuffles. | ||||
12345 | if (isLHSID) return ReplaceInstUsesWith(SVI, LHS); | ||||
12346 | if (isRHSID) return ReplaceInstUsesWith(SVI, RHS); | ||||
12347 | |||||
12348 | // If the LHS is a shufflevector itself, see if we can combine it with this | ||||
12349 | // one without producing an unusual shuffle. Here we are really conservative: | ||||
12350 | // we are absolutely afraid of producing a shuffle mask not in the input | ||||
12351 | // program, because the code gen may not be smart enough to turn a merged | ||||
12352 | // shuffle into two specific shuffles: it may produce worse code. As such, | ||||
12353 | // we only merge two shuffles if the result is one of the two input shuffle | ||||
12354 | // masks. In this case, merging the shuffles just removes one instruction, | ||||
12355 | // which we know is safe. This is good for things like turning: | ||||
12356 | // (splat(splat)) -> splat. | ||||
12357 | if (ShuffleVectorInst *LHSSVI = dyn_cast<ShuffleVectorInst>(LHS)) { | ||||
12358 | if (isa<UndefValue>(RHS)) { | ||||
12359 | std::vector<unsigned> LHSMask = getShuffleMask(LHSSVI); | ||||
12360 | |||||
12361 | std::vector<unsigned> NewMask; | ||||
12362 | for (unsigned i = 0, e = Mask.size(); i != e; ++i) | ||||
12363 | if (Mask[i] >= 2*e) | ||||
12364 | NewMask.push_back(2*e); | ||||
12365 | else | ||||
12366 | NewMask.push_back(LHSMask[Mask[i]]); | ||||
12367 | |||||
12368 | // If the result mask is equal to the src shuffle or this shuffle mask, do | ||||
12369 | // the replacement. | ||||
12370 | if (NewMask == LHSMask || NewMask == Mask) { | ||||
wangmp | 496a76d | 2009-01-26 04:39:00 +0000 | [diff] [blame] | 12371 | unsigned LHSInNElts = |
12372 | cast<VectorType>(LHSSVI->getOperand(0)->getType())->getNumElements(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12373 | std::vector<Constant*> Elts; |
12374 | for (unsigned i = 0, e = NewMask.size(); i != e; ++i) { | ||||
wangmp | 496a76d | 2009-01-26 04:39:00 +0000 | [diff] [blame] | 12375 | if (NewMask[i] >= LHSInNElts*2) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12376 | Elts.push_back(UndefValue::get(Type::Int32Ty)); |
12377 | } else { | ||||
12378 | Elts.push_back(ConstantInt::get(Type::Int32Ty, NewMask[i])); | ||||
12379 | } | ||||
12380 | } | ||||
12381 | return new ShuffleVectorInst(LHSSVI->getOperand(0), | ||||
12382 | LHSSVI->getOperand(1), | ||||
12383 | ConstantVector::get(Elts)); | ||||
12384 | } | ||||
12385 | } | ||||
12386 | } | ||||
12387 | |||||
12388 | return MadeChange ? &SVI : 0; | ||||
12389 | } | ||||
12390 | |||||
12391 | |||||
12392 | |||||
12393 | |||||
12394 | /// TryToSinkInstruction - Try to move the specified instruction from its | ||||
12395 | /// current block into the beginning of DestBlock, which can only happen if it's | ||||
12396 | /// safe to move the instruction past all of the instructions between it and the | ||||
12397 | /// end of its block. | ||||
12398 | static bool TryToSinkInstruction(Instruction *I, BasicBlock *DestBlock) { | ||||
12399 | assert(I->hasOneUse() && "Invariants didn't hold!"); | ||||
12400 | |||||
12401 | // Cannot move control-flow-involving, volatile loads, vaarg, etc. | ||||
Chris Lattner | cb19a1c | 2008-05-09 15:07:33 +0000 | [diff] [blame] | 12402 | if (isa<PHINode>(I) || I->mayWriteToMemory() || isa<TerminatorInst>(I)) |
12403 | return false; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12404 | |
12405 | // Do not sink alloca instructions out of the entry block. | ||||
12406 | if (isa<AllocaInst>(I) && I->getParent() == | ||||
12407 | &DestBlock->getParent()->getEntryBlock()) | ||||
12408 | return false; | ||||
12409 | |||||
12410 | // We can only sink load instructions if there is nothing between the load and | ||||
12411 | // the end of block that could change the value. | ||||
Chris Lattner | 0db40a6 | 2008-05-08 17:37:37 +0000 | [diff] [blame] | 12412 | if (I->mayReadFromMemory()) { |
12413 | for (BasicBlock::iterator Scan = I, E = I->getParent()->end(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12414 | Scan != E; ++Scan) |
12415 | if (Scan->mayWriteToMemory()) | ||||
12416 | return false; | ||||
12417 | } | ||||
12418 | |||||
Dan Gohman | 514277c | 2008-05-23 21:05:58 +0000 | [diff] [blame] | 12419 | BasicBlock::iterator InsertPos = DestBlock->getFirstNonPHI(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12420 | |
Dale Johannesen | 24339f1 | 2009-03-03 01:09:07 +0000 | [diff] [blame] | 12421 | CopyPrecedingStopPoint(I, InsertPos); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12422 | I->moveBefore(InsertPos); |
12423 | ++NumSunkInst; | ||||
12424 | return true; | ||||
12425 | } | ||||
12426 | |||||
12427 | |||||
12428 | /// AddReachableCodeToWorklist - Walk the function in depth-first order, adding | ||||
12429 | /// all reachable code to the worklist. | ||||
12430 | /// | ||||
12431 | /// This has a couple of tricks to make the code faster and more powerful. In | ||||
12432 | /// particular, we constant fold and DCE instructions as we go, to avoid adding | ||||
12433 | /// them to the worklist (this significantly speeds up instcombine on code where | ||||
12434 | /// many instructions are dead or constant). Additionally, if we find a branch | ||||
12435 | /// whose condition is a known constant, we only visit the reachable successors. | ||||
12436 | /// | ||||
12437 | static void AddReachableCodeToWorklist(BasicBlock *BB, | ||||
12438 | SmallPtrSet<BasicBlock*, 64> &Visited, | ||||
12439 | InstCombiner &IC, | ||||
12440 | const TargetData *TD) { | ||||
Chris Lattner | a06291a | 2008-08-15 04:03:01 +0000 | [diff] [blame] | 12441 | SmallVector<BasicBlock*, 256> Worklist; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12442 | Worklist.push_back(BB); |
12443 | |||||
12444 | while (!Worklist.empty()) { | ||||
12445 | BB = Worklist.back(); | ||||
12446 | Worklist.pop_back(); | ||||
12447 | |||||
12448 | // We have now visited this block! If we've already been here, ignore it. | ||||
12449 | if (!Visited.insert(BB)) continue; | ||||
Devang Patel | 794140c | 2008-11-19 18:56:50 +0000 | [diff] [blame] | 12450 | |
12451 | DbgInfoIntrinsic *DBI_Prev = NULL; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12452 | for (BasicBlock::iterator BBI = BB->begin(), E = BB->end(); BBI != E; ) { |
12453 | Instruction *Inst = BBI++; | ||||
12454 | |||||
12455 | // DCE instruction if trivially dead. | ||||
12456 | if (isInstructionTriviallyDead(Inst)) { | ||||
12457 | ++NumDeadInst; | ||||
12458 | DOUT << "IC: DCE: " << *Inst; | ||||
12459 | Inst->eraseFromParent(); | ||||
12460 | continue; | ||||
12461 | } | ||||
12462 | |||||
12463 | // ConstantProp instruction if trivially constant. | ||||
12464 | if (Constant *C = ConstantFoldInstruction(Inst, TD)) { | ||||
12465 | DOUT << "IC: ConstFold to: " << *C << " from: " << *Inst; | ||||
12466 | Inst->replaceAllUsesWith(C); | ||||
12467 | ++NumConstProp; | ||||
12468 | Inst->eraseFromParent(); | ||||
12469 | continue; | ||||
12470 | } | ||||
Chris Lattner | e0f462d | 2007-07-20 22:06:41 +0000 | [diff] [blame] | 12471 | |
Devang Patel | 794140c | 2008-11-19 18:56:50 +0000 | [diff] [blame] | 12472 | // If there are two consecutive llvm.dbg.stoppoint calls then |
12473 | // it is likely that the optimizer deleted code in between these | ||||
12474 | // two intrinsics. | ||||
12475 | DbgInfoIntrinsic *DBI_Next = dyn_cast<DbgInfoIntrinsic>(Inst); | ||||
12476 | if (DBI_Next) { | ||||
12477 | if (DBI_Prev | ||||
12478 | && DBI_Prev->getIntrinsicID() == llvm::Intrinsic::dbg_stoppoint | ||||
12479 | && DBI_Next->getIntrinsicID() == llvm::Intrinsic::dbg_stoppoint) { | ||||
12480 | IC.RemoveFromWorkList(DBI_Prev); | ||||
12481 | DBI_Prev->eraseFromParent(); | ||||
12482 | } | ||||
12483 | DBI_Prev = DBI_Next; | ||||
Zhou Sheng | 77e03b9 | 2009-02-23 10:14:11 +0000 | [diff] [blame] | 12484 | } else { |
12485 | DBI_Prev = 0; | ||||
Devang Patel | 794140c | 2008-11-19 18:56:50 +0000 | [diff] [blame] | 12486 | } |
12487 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12488 | IC.AddToWorkList(Inst); |
12489 | } | ||||
12490 | |||||
12491 | // Recursively visit successors. If this is a branch or switch on a | ||||
12492 | // constant, only visit the reachable successor. | ||||
12493 | TerminatorInst *TI = BB->getTerminator(); | ||||
12494 | if (BranchInst *BI = dyn_cast<BranchInst>(TI)) { | ||||
12495 | if (BI->isConditional() && isa<ConstantInt>(BI->getCondition())) { | ||||
12496 | bool CondVal = cast<ConstantInt>(BI->getCondition())->getZExtValue(); | ||||
Nick Lewycky | d551cf1 | 2008-03-09 08:50:23 +0000 | [diff] [blame] | 12497 | BasicBlock *ReachableBB = BI->getSuccessor(!CondVal); |
Nick Lewycky | d8aa33a | 2008-04-25 16:53:59 +0000 | [diff] [blame] | 12498 | Worklist.push_back(ReachableBB); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12499 | continue; |
12500 | } | ||||
12501 | } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) { | ||||
12502 | if (ConstantInt *Cond = dyn_cast<ConstantInt>(SI->getCondition())) { | ||||
12503 | // See if this is an explicit destination. | ||||
12504 | for (unsigned i = 1, e = SI->getNumSuccessors(); i != e; ++i) | ||||
12505 | if (SI->getCaseValue(i) == Cond) { | ||||
Nick Lewycky | d551cf1 | 2008-03-09 08:50:23 +0000 | [diff] [blame] | 12506 | BasicBlock *ReachableBB = SI->getSuccessor(i); |
Nick Lewycky | d8aa33a | 2008-04-25 16:53:59 +0000 | [diff] [blame] | 12507 | Worklist.push_back(ReachableBB); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12508 | continue; |
12509 | } | ||||
12510 | |||||
12511 | // Otherwise it is the default destination. | ||||
12512 | Worklist.push_back(SI->getSuccessor(0)); | ||||
12513 | continue; | ||||
12514 | } | ||||
12515 | } | ||||
12516 | |||||
12517 | for (unsigned i = 0, e = TI->getNumSuccessors(); i != e; ++i) | ||||
12518 | Worklist.push_back(TI->getSuccessor(i)); | ||||
12519 | } | ||||
12520 | } | ||||
12521 | |||||
12522 | bool InstCombiner::DoOneIteration(Function &F, unsigned Iteration) { | ||||
12523 | bool Changed = false; | ||||
12524 | TD = &getAnalysis<TargetData>(); | ||||
12525 | |||||
12526 | DEBUG(DOUT << "\n\nINSTCOMBINE ITERATION #" << Iteration << " on " | ||||
12527 | << F.getNameStr() << "\n"); | ||||
12528 | |||||
12529 | { | ||||
12530 | // Do a depth-first traversal of the function, populate the worklist with | ||||
12531 | // the reachable instructions. Ignore blocks that are not reachable. Keep | ||||
12532 | // track of which blocks we visit. | ||||
12533 | SmallPtrSet<BasicBlock*, 64> Visited; | ||||
12534 | AddReachableCodeToWorklist(F.begin(), Visited, *this, TD); | ||||
12535 | |||||
12536 | // Do a quick scan over the function. If we find any blocks that are | ||||
12537 | // unreachable, remove any instructions inside of them. This prevents | ||||
12538 | // the instcombine code from having to deal with some bad special cases. | ||||
12539 | for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) | ||||
12540 | if (!Visited.count(BB)) { | ||||
12541 | Instruction *Term = BB->getTerminator(); | ||||
12542 | while (Term != BB->begin()) { // Remove instrs bottom-up | ||||
12543 | BasicBlock::iterator I = Term; --I; | ||||
12544 | |||||
12545 | DOUT << "IC: DCE: " << *I; | ||||
12546 | ++NumDeadInst; | ||||
12547 | |||||
12548 | if (!I->use_empty()) | ||||
12549 | I->replaceAllUsesWith(UndefValue::get(I->getType())); | ||||
12550 | I->eraseFromParent(); | ||||
Chris Lattner | f6d5886 | 2009-01-31 07:04:22 +0000 | [diff] [blame] | 12551 | Changed = true; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12552 | } |
12553 | } | ||||
12554 | } | ||||
12555 | |||||
12556 | while (!Worklist.empty()) { | ||||
12557 | Instruction *I = RemoveOneFromWorkList(); | ||||
12558 | if (I == 0) continue; // skip null values. | ||||
12559 | |||||
12560 | // Check to see if we can DCE the instruction. | ||||
12561 | if (isInstructionTriviallyDead(I)) { | ||||
12562 | // Add operands to the worklist. | ||||
12563 | if (I->getNumOperands() < 4) | ||||
12564 | AddUsesToWorkList(*I); | ||||
12565 | ++NumDeadInst; | ||||
12566 | |||||
12567 | DOUT << "IC: DCE: " << *I; | ||||
12568 | |||||
12569 | I->eraseFromParent(); | ||||
12570 | RemoveFromWorkList(I); | ||||
Chris Lattner | f6d5886 | 2009-01-31 07:04:22 +0000 | [diff] [blame] | 12571 | Changed = true; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12572 | continue; |
12573 | } | ||||
12574 | |||||
12575 | // Instruction isn't dead, see if we can constant propagate it. | ||||
12576 | if (Constant *C = ConstantFoldInstruction(I, TD)) { | ||||
12577 | DOUT << "IC: ConstFold to: " << *C << " from: " << *I; | ||||
12578 | |||||
12579 | // Add operands to the worklist. | ||||
12580 | AddUsesToWorkList(*I); | ||||
12581 | ReplaceInstUsesWith(*I, C); | ||||
12582 | |||||
12583 | ++NumConstProp; | ||||
12584 | I->eraseFromParent(); | ||||
12585 | RemoveFromWorkList(I); | ||||
Chris Lattner | f6d5886 | 2009-01-31 07:04:22 +0000 | [diff] [blame] | 12586 | Changed = true; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12587 | continue; |
12588 | } | ||||
12589 | |||||
Nick Lewycky | adb6792 | 2008-05-25 20:56:15 +0000 | [diff] [blame] | 12590 | if (TD && I->getType()->getTypeID() == Type::VoidTyID) { |
12591 | // See if we can constant fold its operands. | ||||
Chris Lattner | f6d5886 | 2009-01-31 07:04:22 +0000 | [diff] [blame] | 12592 | for (User::op_iterator i = I->op_begin(), e = I->op_end(); i != e; ++i) |
12593 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(i)) | ||||
Nick Lewycky | adb6792 | 2008-05-25 20:56:15 +0000 | [diff] [blame] | 12594 | if (Constant *NewC = ConstantFoldConstantExpression(CE, TD)) |
Chris Lattner | f6d5886 | 2009-01-31 07:04:22 +0000 | [diff] [blame] | 12595 | if (NewC != CE) { |
12596 | i->set(NewC); | ||||
12597 | Changed = true; | ||||
12598 | } | ||||
Nick Lewycky | adb6792 | 2008-05-25 20:56:15 +0000 | [diff] [blame] | 12599 | } |
12600 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12601 | // See if we can trivially sink this instruction to a successor basic block. |
Dan Gohman | 29474e9 | 2008-07-23 00:34:11 +0000 | [diff] [blame] | 12602 | if (I->hasOneUse()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12603 | BasicBlock *BB = I->getParent(); |
12604 | BasicBlock *UserParent = cast<Instruction>(I->use_back())->getParent(); | ||||
12605 | if (UserParent != BB) { | ||||
12606 | bool UserIsSuccessor = false; | ||||
12607 | // See if the user is one of our successors. | ||||
12608 | for (succ_iterator SI = succ_begin(BB), E = succ_end(BB); SI != E; ++SI) | ||||
12609 | if (*SI == UserParent) { | ||||
12610 | UserIsSuccessor = true; | ||||
12611 | break; | ||||
12612 | } | ||||
12613 | |||||
12614 | // If the user is one of our immediate successors, and if that successor | ||||
12615 | // only has us as a predecessors (we'd have to split the critical edge | ||||
12616 | // otherwise), we can keep going. | ||||
12617 | if (UserIsSuccessor && !isa<PHINode>(I->use_back()) && | ||||
12618 | next(pred_begin(UserParent)) == pred_end(UserParent)) | ||||
12619 | // Okay, the CFG is simple enough, try to sink this instruction. | ||||
12620 | Changed |= TryToSinkInstruction(I, UserParent); | ||||
12621 | } | ||||
12622 | } | ||||
12623 | |||||
12624 | // Now that we have an instruction, try combining it to simplify it... | ||||
12625 | #ifndef NDEBUG | ||||
12626 | std::string OrigI; | ||||
12627 | #endif | ||||
12628 | DEBUG(std::ostringstream SS; I->print(SS); OrigI = SS.str();); | ||||
12629 | if (Instruction *Result = visit(*I)) { | ||||
12630 | ++NumCombined; | ||||
12631 | // Should we replace the old instruction with a new one? | ||||
12632 | if (Result != I) { | ||||
12633 | DOUT << "IC: Old = " << *I | ||||
12634 | << " New = " << *Result; | ||||
12635 | |||||
12636 | // Everything uses the new instruction now. | ||||
12637 | I->replaceAllUsesWith(Result); | ||||
12638 | |||||
12639 | // Push the new instruction and any users onto the worklist. | ||||
12640 | AddToWorkList(Result); | ||||
12641 | AddUsersToWorkList(*Result); | ||||
12642 | |||||
12643 | // Move the name to the new instruction first. | ||||
12644 | Result->takeName(I); | ||||
12645 | |||||
12646 | // Insert the new instruction into the basic block... | ||||
12647 | BasicBlock *InstParent = I->getParent(); | ||||
12648 | BasicBlock::iterator InsertPos = I; | ||||
12649 | |||||
12650 | if (!isa<PHINode>(Result)) // If combining a PHI, don't insert | ||||
12651 | while (isa<PHINode>(InsertPos)) // middle of a block of PHIs. | ||||
12652 | ++InsertPos; | ||||
12653 | |||||
12654 | InstParent->getInstList().insert(InsertPos, Result); | ||||
12655 | |||||
12656 | // Make sure that we reprocess all operands now that we reduced their | ||||
12657 | // use counts. | ||||
12658 | AddUsesToWorkList(*I); | ||||
12659 | |||||
12660 | // Instructions can end up on the worklist more than once. Make sure | ||||
12661 | // we do not process an instruction that has been deleted. | ||||
12662 | RemoveFromWorkList(I); | ||||
12663 | |||||
12664 | // Erase the old instruction. | ||||
12665 | InstParent->getInstList().erase(I); | ||||
12666 | } else { | ||||
12667 | #ifndef NDEBUG | ||||
12668 | DOUT << "IC: Mod = " << OrigI | ||||
12669 | << " New = " << *I; | ||||
12670 | #endif | ||||
12671 | |||||
12672 | // If the instruction was modified, it's possible that it is now dead. | ||||
12673 | // if so, remove it. | ||||
12674 | if (isInstructionTriviallyDead(I)) { | ||||
12675 | // Make sure we process all operands now that we are reducing their | ||||
12676 | // use counts. | ||||
12677 | AddUsesToWorkList(*I); | ||||
12678 | |||||
12679 | // Instructions may end up in the worklist more than once. Erase all | ||||
12680 | // occurrences of this instruction. | ||||
12681 | RemoveFromWorkList(I); | ||||
12682 | I->eraseFromParent(); | ||||
12683 | } else { | ||||
12684 | AddToWorkList(I); | ||||
12685 | AddUsersToWorkList(*I); | ||||
12686 | } | ||||
12687 | } | ||||
12688 | Changed = true; | ||||
12689 | } | ||||
12690 | } | ||||
12691 | |||||
12692 | assert(WorklistMap.empty() && "Worklist empty, but map not?"); | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 12693 | |
12694 | // Do an explicit clear, this shrinks the map if needed. | ||||
12695 | WorklistMap.clear(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12696 | return Changed; |
12697 | } | ||||
12698 | |||||
12699 | |||||
12700 | bool InstCombiner::runOnFunction(Function &F) { | ||||
12701 | MustPreserveLCSSA = mustPreserveAnalysisID(LCSSAID); | ||||
12702 | |||||
12703 | bool EverMadeChange = false; | ||||
12704 | |||||
12705 | // Iterate while there is work to do. | ||||
12706 | unsigned Iteration = 0; | ||||
Bill Wendling | d9644a4 | 2008-05-14 22:45:20 +0000 | [diff] [blame] | 12707 | while (DoOneIteration(F, Iteration++)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12708 | EverMadeChange = true; |
12709 | return EverMadeChange; | ||||
12710 | } | ||||
12711 | |||||
12712 | FunctionPass *llvm::createInstructionCombiningPass() { | ||||
12713 | return new InstCombiner(); | ||||
12714 | } |