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); | ||||
183 | Instruction *visitAnd(BinaryOperator &I); | ||||
184 | Instruction *visitOr (BinaryOperator &I); | ||||
185 | Instruction *visitXor(BinaryOperator &I); | ||||
186 | Instruction *visitShl(BinaryOperator &I); | ||||
187 | Instruction *visitAShr(BinaryOperator &I); | ||||
188 | Instruction *visitLShr(BinaryOperator &I); | ||||
189 | Instruction *commonShiftTransforms(BinaryOperator &I); | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 190 | Instruction *FoldFCmp_IntToFP_Cst(FCmpInst &I, Instruction *LHSI, |
191 | Constant *RHSC); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 192 | Instruction *visitFCmpInst(FCmpInst &I); |
193 | Instruction *visitICmpInst(ICmpInst &I); | ||||
194 | Instruction *visitICmpInstWithCastAndCast(ICmpInst &ICI); | ||||
195 | Instruction *visitICmpInstWithInstAndIntCst(ICmpInst &ICI, | ||||
196 | Instruction *LHS, | ||||
197 | ConstantInt *RHS); | ||||
198 | Instruction *FoldICmpDivCst(ICmpInst &ICI, BinaryOperator *DivI, | ||||
199 | ConstantInt *DivRHS); | ||||
200 | |||||
201 | Instruction *FoldGEPICmp(User *GEPLHS, Value *RHS, | ||||
202 | ICmpInst::Predicate Cond, Instruction &I); | ||||
203 | Instruction *FoldShiftByConstant(Value *Op0, ConstantInt *Op1, | ||||
204 | BinaryOperator &I); | ||||
205 | Instruction *commonCastTransforms(CastInst &CI); | ||||
206 | Instruction *commonIntCastTransforms(CastInst &CI); | ||||
207 | Instruction *commonPointerCastTransforms(CastInst &CI); | ||||
208 | Instruction *visitTrunc(TruncInst &CI); | ||||
209 | Instruction *visitZExt(ZExtInst &CI); | ||||
210 | Instruction *visitSExt(SExtInst &CI); | ||||
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 211 | Instruction *visitFPTrunc(FPTruncInst &CI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 212 | Instruction *visitFPExt(CastInst &CI); |
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 213 | Instruction *visitFPToUI(FPToUIInst &FI); |
214 | Instruction *visitFPToSI(FPToSIInst &FI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 215 | Instruction *visitUIToFP(CastInst &CI); |
216 | Instruction *visitSIToFP(CastInst &CI); | ||||
217 | Instruction *visitPtrToInt(CastInst &CI); | ||||
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 218 | Instruction *visitIntToPtr(IntToPtrInst &CI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 219 | Instruction *visitBitCast(BitCastInst &CI); |
220 | Instruction *FoldSelectOpOp(SelectInst &SI, Instruction *TI, | ||||
221 | Instruction *FI); | ||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 222 | Instruction *visitSelectInst(SelectInst &SI); |
223 | Instruction *visitSelectInstWithICmp(SelectInst &SI, ICmpInst *ICI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 224 | Instruction *visitCallInst(CallInst &CI); |
225 | Instruction *visitInvokeInst(InvokeInst &II); | ||||
226 | Instruction *visitPHINode(PHINode &PN); | ||||
227 | Instruction *visitGetElementPtrInst(GetElementPtrInst &GEP); | ||||
228 | Instruction *visitAllocationInst(AllocationInst &AI); | ||||
229 | Instruction *visitFreeInst(FreeInst &FI); | ||||
230 | Instruction *visitLoadInst(LoadInst &LI); | ||||
231 | Instruction *visitStoreInst(StoreInst &SI); | ||||
232 | Instruction *visitBranchInst(BranchInst &BI); | ||||
233 | Instruction *visitSwitchInst(SwitchInst &SI); | ||||
234 | Instruction *visitInsertElementInst(InsertElementInst &IE); | ||||
235 | Instruction *visitExtractElementInst(ExtractElementInst &EI); | ||||
236 | Instruction *visitShuffleVectorInst(ShuffleVectorInst &SVI); | ||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 237 | Instruction *visitExtractValueInst(ExtractValueInst &EV); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 238 | |
239 | // visitInstruction - Specify what to return for unhandled instructions... | ||||
240 | Instruction *visitInstruction(Instruction &I) { return 0; } | ||||
241 | |||||
242 | private: | ||||
243 | Instruction *visitCallSite(CallSite CS); | ||||
244 | bool transformConstExprCastCall(CallSite CS); | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 245 | Instruction *transformCallThroughTrampoline(CallSite CS); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 246 | Instruction *transformZExtICmp(ICmpInst *ICI, Instruction &CI, |
247 | bool DoXform = true); | ||||
Chris Lattner | 3554f97 | 2008-05-20 05:46:13 +0000 | [diff] [blame] | 248 | bool WillNotOverflowSignedAdd(Value *LHS, Value *RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 249 | |
250 | public: | ||||
251 | // InsertNewInstBefore - insert an instruction New before instruction Old | ||||
252 | // in the program. Add the new instruction to the worklist. | ||||
253 | // | ||||
254 | Instruction *InsertNewInstBefore(Instruction *New, Instruction &Old) { | ||||
255 | assert(New && New->getParent() == 0 && | ||||
256 | "New instruction already inserted into a basic block!"); | ||||
257 | BasicBlock *BB = Old.getParent(); | ||||
258 | BB->getInstList().insert(&Old, New); // Insert inst | ||||
259 | AddToWorkList(New); | ||||
260 | return New; | ||||
261 | } | ||||
262 | |||||
263 | /// InsertCastBefore - Insert a cast of V to TY before the instruction POS. | ||||
264 | /// This also adds the cast to the worklist. Finally, this returns the | ||||
265 | /// cast. | ||||
266 | Value *InsertCastBefore(Instruction::CastOps opc, Value *V, const Type *Ty, | ||||
267 | Instruction &Pos) { | ||||
268 | if (V->getType() == Ty) return V; | ||||
269 | |||||
270 | if (Constant *CV = dyn_cast<Constant>(V)) | ||||
271 | return ConstantExpr::getCast(opc, CV, Ty); | ||||
272 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 273 | Instruction *C = CastInst::Create(opc, V, Ty, V->getName(), &Pos); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 274 | AddToWorkList(C); |
275 | return C; | ||||
276 | } | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 277 | |
278 | Value *InsertBitCastBefore(Value *V, const Type *Ty, Instruction &Pos) { | ||||
279 | return InsertCastBefore(Instruction::BitCast, V, Ty, Pos); | ||||
280 | } | ||||
281 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 282 | |
283 | // ReplaceInstUsesWith - This method is to be used when an instruction is | ||||
284 | // found to be dead, replacable with another preexisting expression. Here | ||||
285 | // we add all uses of I to the worklist, replace all uses of I with the new | ||||
286 | // value, then return I, so that the inst combiner will know that I was | ||||
287 | // modified. | ||||
288 | // | ||||
289 | Instruction *ReplaceInstUsesWith(Instruction &I, Value *V) { | ||||
290 | AddUsersToWorkList(I); // Add all modified instrs to worklist | ||||
291 | if (&I != V) { | ||||
292 | I.replaceAllUsesWith(V); | ||||
293 | return &I; | ||||
294 | } else { | ||||
295 | // If we are replacing the instruction with itself, this must be in a | ||||
296 | // segment of unreachable code, so just clobber the instruction. | ||||
297 | I.replaceAllUsesWith(UndefValue::get(I.getType())); | ||||
298 | return &I; | ||||
299 | } | ||||
300 | } | ||||
301 | |||||
302 | // UpdateValueUsesWith - This method is to be used when an value is | ||||
303 | // found to be replacable with another preexisting expression or was | ||||
304 | // updated. Here we add all uses of I to the worklist, replace all uses of | ||||
305 | // I with the new value (unless the instruction was just updated), then | ||||
306 | // return true, so that the inst combiner will know that I was modified. | ||||
307 | // | ||||
308 | bool UpdateValueUsesWith(Value *Old, Value *New) { | ||||
309 | AddUsersToWorkList(*Old); // Add all modified instrs to worklist | ||||
310 | if (Old != New) | ||||
311 | Old->replaceAllUsesWith(New); | ||||
312 | if (Instruction *I = dyn_cast<Instruction>(Old)) | ||||
313 | AddToWorkList(I); | ||||
314 | if (Instruction *I = dyn_cast<Instruction>(New)) | ||||
315 | AddToWorkList(I); | ||||
316 | return true; | ||||
317 | } | ||||
318 | |||||
319 | // EraseInstFromFunction - When dealing with an instruction that has side | ||||
320 | // effects or produces a void value, we can't rely on DCE to delete the | ||||
321 | // instruction. Instead, visit methods should return the value returned by | ||||
322 | // this function. | ||||
323 | Instruction *EraseInstFromFunction(Instruction &I) { | ||||
324 | assert(I.use_empty() && "Cannot erase instruction that is used!"); | ||||
325 | AddUsesToWorkList(I); | ||||
326 | RemoveFromWorkList(&I); | ||||
327 | I.eraseFromParent(); | ||||
328 | return 0; // Don't do anything with FI | ||||
329 | } | ||||
Chris Lattner | a432bc7 | 2008-06-02 01:18:21 +0000 | [diff] [blame] | 330 | |
331 | void ComputeMaskedBits(Value *V, const APInt &Mask, APInt &KnownZero, | ||||
332 | APInt &KnownOne, unsigned Depth = 0) const { | ||||
333 | return llvm::ComputeMaskedBits(V, Mask, KnownZero, KnownOne, TD, Depth); | ||||
334 | } | ||||
335 | |||||
336 | bool MaskedValueIsZero(Value *V, const APInt &Mask, | ||||
337 | unsigned Depth = 0) const { | ||||
338 | return llvm::MaskedValueIsZero(V, Mask, TD, Depth); | ||||
339 | } | ||||
340 | unsigned ComputeNumSignBits(Value *Op, unsigned Depth = 0) const { | ||||
341 | return llvm::ComputeNumSignBits(Op, TD, Depth); | ||||
342 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 343 | |
344 | private: | ||||
345 | /// InsertOperandCastBefore - This inserts a cast of V to DestTy before the | ||||
346 | /// InsertBefore instruction. This is specialized a bit to avoid inserting | ||||
347 | /// casts that are known to not do anything... | ||||
348 | /// | ||||
349 | Value *InsertOperandCastBefore(Instruction::CastOps opcode, | ||||
350 | Value *V, const Type *DestTy, | ||||
351 | Instruction *InsertBefore); | ||||
352 | |||||
353 | /// SimplifyCommutative - This performs a few simplifications for | ||||
354 | /// commutative operators. | ||||
355 | bool SimplifyCommutative(BinaryOperator &I); | ||||
356 | |||||
357 | /// SimplifyCompare - This reorders the operands of a CmpInst to get them in | ||||
358 | /// most-complex to least-complex order. | ||||
359 | bool SimplifyCompare(CmpInst &I); | ||||
360 | |||||
361 | /// SimplifyDemandedBits - Attempts to replace V with a simpler value based | ||||
362 | /// on the demanded bits. | ||||
363 | bool SimplifyDemandedBits(Value *V, APInt DemandedMask, | ||||
364 | APInt& KnownZero, APInt& KnownOne, | ||||
365 | unsigned Depth = 0); | ||||
366 | |||||
367 | Value *SimplifyDemandedVectorElts(Value *V, uint64_t DemandedElts, | ||||
368 | uint64_t &UndefElts, unsigned Depth = 0); | ||||
369 | |||||
370 | // FoldOpIntoPhi - Given a binary operator or cast instruction which has a | ||||
371 | // PHI node as operand #0, see if we can fold the instruction into the PHI | ||||
372 | // (which is only possible if all operands to the PHI are constants). | ||||
373 | Instruction *FoldOpIntoPhi(Instruction &I); | ||||
374 | |||||
375 | // FoldPHIArgOpIntoPHI - If all operands to a PHI node are the same "unary" | ||||
376 | // operator and they all are only used by the PHI, PHI together their | ||||
377 | // inputs, and do the operation once, to the result of the PHI. | ||||
378 | Instruction *FoldPHIArgOpIntoPHI(PHINode &PN); | ||||
379 | Instruction *FoldPHIArgBinOpIntoPHI(PHINode &PN); | ||||
380 | |||||
381 | |||||
382 | Instruction *OptAndOp(Instruction *Op, ConstantInt *OpRHS, | ||||
383 | ConstantInt *AndRHS, BinaryOperator &TheAnd); | ||||
384 | |||||
385 | Value *FoldLogicalPlusAnd(Value *LHS, Value *RHS, ConstantInt *Mask, | ||||
386 | bool isSub, Instruction &I); | ||||
387 | Instruction *InsertRangeTest(Value *V, Constant *Lo, Constant *Hi, | ||||
388 | bool isSigned, bool Inside, Instruction &IB); | ||||
389 | Instruction *PromoteCastOfAllocation(BitCastInst &CI, AllocationInst &AI); | ||||
390 | Instruction *MatchBSwap(BinaryOperator &I); | ||||
391 | bool SimplifyStoreAtEndOfBlock(StoreInst &SI); | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 392 | Instruction *SimplifyMemTransfer(MemIntrinsic *MI); |
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 393 | Instruction *SimplifyMemSet(MemSetInst *MI); |
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 394 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 395 | |
396 | Value *EvaluateInDifferentType(Value *V, const Type *Ty, bool isSigned); | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 397 | |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 398 | bool CanEvaluateInDifferentType(Value *V, const IntegerType *Ty, |
399 | unsigned CastOpc, | ||||
400 | int &NumCastsRemoved); | ||||
401 | unsigned GetOrEnforceKnownAlignment(Value *V, | ||||
402 | unsigned PrefAlign = 0); | ||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 403 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 404 | }; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 405 | } |
406 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 407 | char InstCombiner::ID = 0; |
408 | static RegisterPass<InstCombiner> | ||||
409 | X("instcombine", "Combine redundant instructions"); | ||||
410 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 411 | // getComplexity: Assign a complexity or rank value to LLVM Values... |
412 | // 0 -> undef, 1 -> Const, 2 -> Other, 3 -> Arg, 3 -> Unary, 4 -> OtherInst | ||||
413 | static unsigned getComplexity(Value *V) { | ||||
414 | if (isa<Instruction>(V)) { | ||||
415 | if (BinaryOperator::isNeg(V) || BinaryOperator::isNot(V)) | ||||
416 | return 3; | ||||
417 | return 4; | ||||
418 | } | ||||
419 | if (isa<Argument>(V)) return 3; | ||||
420 | return isa<Constant>(V) ? (isa<UndefValue>(V) ? 0 : 1) : 2; | ||||
421 | } | ||||
422 | |||||
423 | // isOnlyUse - Return true if this instruction will be deleted if we stop using | ||||
424 | // it. | ||||
425 | static bool isOnlyUse(Value *V) { | ||||
426 | return V->hasOneUse() || isa<Constant>(V); | ||||
427 | } | ||||
428 | |||||
429 | // getPromotedType - Return the specified type promoted as it would be to pass | ||||
430 | // though a va_arg area... | ||||
431 | static const Type *getPromotedType(const Type *Ty) { | ||||
432 | if (const IntegerType* ITy = dyn_cast<IntegerType>(Ty)) { | ||||
433 | if (ITy->getBitWidth() < 32) | ||||
434 | return Type::Int32Ty; | ||||
435 | } | ||||
436 | return Ty; | ||||
437 | } | ||||
438 | |||||
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 439 | /// getBitCastOperand - If the specified operand is a CastInst, a constant |
440 | /// expression bitcast, or a GetElementPtrInst with all zero indices, return the | ||||
441 | /// operand value, otherwise return null. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 442 | static Value *getBitCastOperand(Value *V) { |
443 | if (BitCastInst *I = dyn_cast<BitCastInst>(V)) | ||||
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 444 | // BitCastInst? |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 445 | return I->getOperand(0); |
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 446 | else if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(V)) { |
447 | // GetElementPtrInst? | ||||
448 | if (GEP->hasAllZeroIndices()) | ||||
449 | return GEP->getOperand(0); | ||||
450 | } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(V)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 451 | if (CE->getOpcode() == Instruction::BitCast) |
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 452 | // BitCast ConstantExp? |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 453 | return CE->getOperand(0); |
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 454 | else if (CE->getOpcode() == Instruction::GetElementPtr) { |
455 | // GetElementPtr ConstantExp? | ||||
456 | for (User::op_iterator I = CE->op_begin() + 1, E = CE->op_end(); | ||||
457 | I != E; ++I) { | ||||
458 | ConstantInt *CI = dyn_cast<ConstantInt>(I); | ||||
459 | if (!CI || !CI->isZero()) | ||||
460 | // Any non-zero indices? Not cast-like. | ||||
461 | return 0; | ||||
462 | } | ||||
463 | // All-zero indices? This is just like casting. | ||||
464 | return CE->getOperand(0); | ||||
465 | } | ||||
466 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 467 | return 0; |
468 | } | ||||
469 | |||||
470 | /// This function is a wrapper around CastInst::isEliminableCastPair. It | ||||
471 | /// simply extracts arguments and returns what that function returns. | ||||
472 | static Instruction::CastOps | ||||
473 | isEliminableCastPair( | ||||
474 | const CastInst *CI, ///< The first cast instruction | ||||
475 | unsigned opcode, ///< The opcode of the second cast instruction | ||||
476 | const Type *DstTy, ///< The target type for the second cast instruction | ||||
477 | TargetData *TD ///< The target data for pointer size | ||||
478 | ) { | ||||
479 | |||||
480 | const Type *SrcTy = CI->getOperand(0)->getType(); // A from above | ||||
481 | const Type *MidTy = CI->getType(); // B from above | ||||
482 | |||||
483 | // Get the opcodes of the two Cast instructions | ||||
484 | Instruction::CastOps firstOp = Instruction::CastOps(CI->getOpcode()); | ||||
485 | Instruction::CastOps secondOp = Instruction::CastOps(opcode); | ||||
486 | |||||
487 | return Instruction::CastOps( | ||||
488 | CastInst::isEliminableCastPair(firstOp, secondOp, SrcTy, MidTy, | ||||
489 | DstTy, TD->getIntPtrType())); | ||||
490 | } | ||||
491 | |||||
492 | /// ValueRequiresCast - Return true if the cast from "V to Ty" actually results | ||||
493 | /// in any code being generated. It does not require codegen if V is simple | ||||
494 | /// enough or if the cast can be folded into other casts. | ||||
495 | static bool ValueRequiresCast(Instruction::CastOps opcode, const Value *V, | ||||
496 | const Type *Ty, TargetData *TD) { | ||||
497 | if (V->getType() == Ty || isa<Constant>(V)) return false; | ||||
498 | |||||
499 | // If this is another cast that can be eliminated, it isn't codegen either. | ||||
500 | if (const CastInst *CI = dyn_cast<CastInst>(V)) | ||||
501 | if (isEliminableCastPair(CI, opcode, Ty, TD)) | ||||
502 | return false; | ||||
503 | return true; | ||||
504 | } | ||||
505 | |||||
506 | /// InsertOperandCastBefore - This inserts a cast of V to DestTy before the | ||||
507 | /// InsertBefore instruction. This is specialized a bit to avoid inserting | ||||
508 | /// casts that are known to not do anything... | ||||
509 | /// | ||||
510 | Value *InstCombiner::InsertOperandCastBefore(Instruction::CastOps opcode, | ||||
511 | Value *V, const Type *DestTy, | ||||
512 | Instruction *InsertBefore) { | ||||
513 | if (V->getType() == DestTy) return V; | ||||
514 | if (Constant *C = dyn_cast<Constant>(V)) | ||||
515 | return ConstantExpr::getCast(opcode, C, DestTy); | ||||
516 | |||||
517 | return InsertCastBefore(opcode, V, DestTy, *InsertBefore); | ||||
518 | } | ||||
519 | |||||
520 | // SimplifyCommutative - This performs a few simplifications for commutative | ||||
521 | // operators: | ||||
522 | // | ||||
523 | // 1. Order operands such that they are listed from right (least complex) to | ||||
524 | // left (most complex). This puts constants before unary operators before | ||||
525 | // binary operators. | ||||
526 | // | ||||
527 | // 2. Transform: (op (op V, C1), C2) ==> (op V, (op C1, C2)) | ||||
528 | // 3. Transform: (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2)) | ||||
529 | // | ||||
530 | bool InstCombiner::SimplifyCommutative(BinaryOperator &I) { | ||||
531 | bool Changed = false; | ||||
532 | if (getComplexity(I.getOperand(0)) < getComplexity(I.getOperand(1))) | ||||
533 | Changed = !I.swapOperands(); | ||||
534 | |||||
535 | if (!I.isAssociative()) return Changed; | ||||
536 | Instruction::BinaryOps Opcode = I.getOpcode(); | ||||
537 | if (BinaryOperator *Op = dyn_cast<BinaryOperator>(I.getOperand(0))) | ||||
538 | if (Op->getOpcode() == Opcode && isa<Constant>(Op->getOperand(1))) { | ||||
539 | if (isa<Constant>(I.getOperand(1))) { | ||||
540 | Constant *Folded = ConstantExpr::get(I.getOpcode(), | ||||
541 | cast<Constant>(I.getOperand(1)), | ||||
542 | cast<Constant>(Op->getOperand(1))); | ||||
543 | I.setOperand(0, Op->getOperand(0)); | ||||
544 | I.setOperand(1, Folded); | ||||
545 | return true; | ||||
546 | } else if (BinaryOperator *Op1=dyn_cast<BinaryOperator>(I.getOperand(1))) | ||||
547 | if (Op1->getOpcode() == Opcode && isa<Constant>(Op1->getOperand(1)) && | ||||
548 | isOnlyUse(Op) && isOnlyUse(Op1)) { | ||||
549 | Constant *C1 = cast<Constant>(Op->getOperand(1)); | ||||
550 | Constant *C2 = cast<Constant>(Op1->getOperand(1)); | ||||
551 | |||||
552 | // Fold (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2)) | ||||
553 | Constant *Folded = ConstantExpr::get(I.getOpcode(), C1, C2); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 554 | Instruction *New = BinaryOperator::Create(Opcode, Op->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 555 | Op1->getOperand(0), |
556 | Op1->getName(), &I); | ||||
557 | AddToWorkList(New); | ||||
558 | I.setOperand(0, New); | ||||
559 | I.setOperand(1, Folded); | ||||
560 | return true; | ||||
561 | } | ||||
562 | } | ||||
563 | return Changed; | ||||
564 | } | ||||
565 | |||||
566 | /// SimplifyCompare - For a CmpInst this function just orders the operands | ||||
567 | /// so that theyare listed from right (least complex) to left (most complex). | ||||
568 | /// This puts constants before unary operators before binary operators. | ||||
569 | bool InstCombiner::SimplifyCompare(CmpInst &I) { | ||||
570 | if (getComplexity(I.getOperand(0)) >= getComplexity(I.getOperand(1))) | ||||
571 | return false; | ||||
572 | I.swapOperands(); | ||||
573 | // Compare instructions are not associative so there's nothing else we can do. | ||||
574 | return true; | ||||
575 | } | ||||
576 | |||||
577 | // dyn_castNegVal - Given a 'sub' instruction, return the RHS of the instruction | ||||
578 | // if the LHS is a constant zero (which is the 'negate' form). | ||||
579 | // | ||||
580 | static inline Value *dyn_castNegVal(Value *V) { | ||||
581 | if (BinaryOperator::isNeg(V)) | ||||
582 | return BinaryOperator::getNegArgument(V); | ||||
583 | |||||
584 | // Constants can be considered to be negated values if they can be folded. | ||||
585 | if (ConstantInt *C = dyn_cast<ConstantInt>(V)) | ||||
586 | return ConstantExpr::getNeg(C); | ||||
Nick Lewycky | 58867bc | 2008-05-23 04:54:45 +0000 | [diff] [blame] | 587 | |
588 | if (ConstantVector *C = dyn_cast<ConstantVector>(V)) | ||||
589 | if (C->getType()->getElementType()->isInteger()) | ||||
590 | return ConstantExpr::getNeg(C); | ||||
591 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 592 | return 0; |
593 | } | ||||
594 | |||||
595 | static inline Value *dyn_castNotVal(Value *V) { | ||||
596 | if (BinaryOperator::isNot(V)) | ||||
597 | return BinaryOperator::getNotArgument(V); | ||||
598 | |||||
599 | // Constants can be considered to be not'ed values... | ||||
600 | if (ConstantInt *C = dyn_cast<ConstantInt>(V)) | ||||
601 | return ConstantInt::get(~C->getValue()); | ||||
602 | return 0; | ||||
603 | } | ||||
604 | |||||
605 | // dyn_castFoldableMul - If this value is a multiply that can be folded into | ||||
606 | // other computations (because it has a constant operand), return the | ||||
607 | // non-constant operand of the multiply, and set CST to point to the multiplier. | ||||
608 | // Otherwise, return null. | ||||
609 | // | ||||
610 | static inline Value *dyn_castFoldableMul(Value *V, ConstantInt *&CST) { | ||||
611 | if (V->hasOneUse() && V->getType()->isInteger()) | ||||
612 | if (Instruction *I = dyn_cast<Instruction>(V)) { | ||||
613 | if (I->getOpcode() == Instruction::Mul) | ||||
614 | if ((CST = dyn_cast<ConstantInt>(I->getOperand(1)))) | ||||
615 | return I->getOperand(0); | ||||
616 | if (I->getOpcode() == Instruction::Shl) | ||||
617 | if ((CST = dyn_cast<ConstantInt>(I->getOperand(1)))) { | ||||
618 | // The multiplier is really 1 << CST. | ||||
619 | uint32_t BitWidth = cast<IntegerType>(V->getType())->getBitWidth(); | ||||
620 | uint32_t CSTVal = CST->getLimitedValue(BitWidth); | ||||
621 | CST = ConstantInt::get(APInt(BitWidth, 1).shl(CSTVal)); | ||||
622 | return I->getOperand(0); | ||||
623 | } | ||||
624 | } | ||||
625 | return 0; | ||||
626 | } | ||||
627 | |||||
628 | /// dyn_castGetElementPtr - If this is a getelementptr instruction or constant | ||||
629 | /// expression, return it. | ||||
630 | static User *dyn_castGetElementPtr(Value *V) { | ||||
631 | if (isa<GetElementPtrInst>(V)) return cast<User>(V); | ||||
632 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(V)) | ||||
633 | if (CE->getOpcode() == Instruction::GetElementPtr) | ||||
634 | return cast<User>(V); | ||||
635 | return false; | ||||
636 | } | ||||
637 | |||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 638 | /// getOpcode - If this is an Instruction or a ConstantExpr, return the |
639 | /// opcode value. Otherwise return UserOp1. | ||||
Dan Gohman | 8c39786 | 2008-05-29 19:53:46 +0000 | [diff] [blame] | 640 | static unsigned getOpcode(const Value *V) { |
641 | if (const Instruction *I = dyn_cast<Instruction>(V)) | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 642 | return I->getOpcode(); |
Dan Gohman | 8c39786 | 2008-05-29 19:53:46 +0000 | [diff] [blame] | 643 | if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(V)) |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 644 | return CE->getOpcode(); |
645 | // Use UserOp1 to mean there's no opcode. | ||||
646 | return Instruction::UserOp1; | ||||
647 | } | ||||
648 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 649 | /// AddOne - Add one to a ConstantInt |
650 | static ConstantInt *AddOne(ConstantInt *C) { | ||||
651 | APInt Val(C->getValue()); | ||||
652 | return ConstantInt::get(++Val); | ||||
653 | } | ||||
654 | /// SubOne - Subtract one from a ConstantInt | ||||
655 | static ConstantInt *SubOne(ConstantInt *C) { | ||||
656 | APInt Val(C->getValue()); | ||||
657 | return ConstantInt::get(--Val); | ||||
658 | } | ||||
659 | /// Add - Add two ConstantInts together | ||||
660 | static ConstantInt *Add(ConstantInt *C1, ConstantInt *C2) { | ||||
661 | return ConstantInt::get(C1->getValue() + C2->getValue()); | ||||
662 | } | ||||
663 | /// And - Bitwise AND two ConstantInts together | ||||
664 | static ConstantInt *And(ConstantInt *C1, ConstantInt *C2) { | ||||
665 | return ConstantInt::get(C1->getValue() & C2->getValue()); | ||||
666 | } | ||||
667 | /// Subtract - Subtract one ConstantInt from another | ||||
668 | static ConstantInt *Subtract(ConstantInt *C1, ConstantInt *C2) { | ||||
669 | return ConstantInt::get(C1->getValue() - C2->getValue()); | ||||
670 | } | ||||
671 | /// Multiply - Multiply two ConstantInts together | ||||
672 | static ConstantInt *Multiply(ConstantInt *C1, ConstantInt *C2) { | ||||
673 | return ConstantInt::get(C1->getValue() * C2->getValue()); | ||||
674 | } | ||||
Nick Lewycky | 9d798f9 | 2008-02-18 22:48:05 +0000 | [diff] [blame] | 675 | /// MultiplyOverflows - True if the multiply can not be expressed in an int |
676 | /// this size. | ||||
677 | static bool MultiplyOverflows(ConstantInt *C1, ConstantInt *C2, bool sign) { | ||||
678 | uint32_t W = C1->getBitWidth(); | ||||
679 | APInt LHSExt = C1->getValue(), RHSExt = C2->getValue(); | ||||
680 | if (sign) { | ||||
681 | LHSExt.sext(W * 2); | ||||
682 | RHSExt.sext(W * 2); | ||||
683 | } else { | ||||
684 | LHSExt.zext(W * 2); | ||||
685 | RHSExt.zext(W * 2); | ||||
686 | } | ||||
687 | |||||
688 | APInt MulExt = LHSExt * RHSExt; | ||||
689 | |||||
690 | if (sign) { | ||||
691 | APInt Min = APInt::getSignedMinValue(W).sext(W * 2); | ||||
692 | APInt Max = APInt::getSignedMaxValue(W).sext(W * 2); | ||||
693 | return MulExt.slt(Min) || MulExt.sgt(Max); | ||||
694 | } else | ||||
695 | return MulExt.ugt(APInt::getLowBitsSet(W * 2, W)); | ||||
696 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 697 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 698 | |
699 | /// ShrinkDemandedConstant - Check to see if the specified operand of the | ||||
700 | /// specified instruction is a constant integer. If so, check to see if there | ||||
701 | /// are any bits set in the constant that are not demanded. If so, shrink the | ||||
702 | /// constant and return true. | ||||
703 | static bool ShrinkDemandedConstant(Instruction *I, unsigned OpNo, | ||||
704 | APInt Demanded) { | ||||
705 | assert(I && "No instruction?"); | ||||
706 | assert(OpNo < I->getNumOperands() && "Operand index too large"); | ||||
707 | |||||
708 | // If the operand is not a constant integer, nothing to do. | ||||
709 | ConstantInt *OpC = dyn_cast<ConstantInt>(I->getOperand(OpNo)); | ||||
710 | if (!OpC) return false; | ||||
711 | |||||
712 | // If there are no bits set that aren't demanded, nothing to do. | ||||
713 | Demanded.zextOrTrunc(OpC->getValue().getBitWidth()); | ||||
714 | if ((~Demanded & OpC->getValue()) == 0) | ||||
715 | return false; | ||||
716 | |||||
717 | // This instruction is producing bits that are not demanded. Shrink the RHS. | ||||
718 | Demanded &= OpC->getValue(); | ||||
719 | I->setOperand(OpNo, ConstantInt::get(Demanded)); | ||||
720 | return true; | ||||
721 | } | ||||
722 | |||||
723 | // ComputeSignedMinMaxValuesFromKnownBits - Given a signed integer type and a | ||||
724 | // set of known zero and one bits, compute the maximum and minimum values that | ||||
725 | // could have the specified known zero and known one bits, returning them in | ||||
726 | // min/max. | ||||
727 | static void ComputeSignedMinMaxValuesFromKnownBits(const Type *Ty, | ||||
728 | const APInt& KnownZero, | ||||
729 | const APInt& KnownOne, | ||||
730 | APInt& Min, APInt& Max) { | ||||
731 | uint32_t BitWidth = cast<IntegerType>(Ty)->getBitWidth(); | ||||
732 | assert(KnownZero.getBitWidth() == BitWidth && | ||||
733 | KnownOne.getBitWidth() == BitWidth && | ||||
734 | Min.getBitWidth() == BitWidth && Max.getBitWidth() == BitWidth && | ||||
735 | "Ty, KnownZero, KnownOne and Min, Max must have equal bitwidth."); | ||||
736 | APInt UnknownBits = ~(KnownZero|KnownOne); | ||||
737 | |||||
738 | // The minimum value is when all unknown bits are zeros, EXCEPT for the sign | ||||
739 | // bit if it is unknown. | ||||
740 | Min = KnownOne; | ||||
741 | Max = KnownOne|UnknownBits; | ||||
742 | |||||
743 | if (UnknownBits[BitWidth-1]) { // Sign bit is unknown | ||||
744 | Min.set(BitWidth-1); | ||||
745 | Max.clear(BitWidth-1); | ||||
746 | } | ||||
747 | } | ||||
748 | |||||
749 | // ComputeUnsignedMinMaxValuesFromKnownBits - Given an unsigned integer type and | ||||
750 | // a set of known zero and one bits, compute the maximum and minimum values that | ||||
751 | // could have the specified known zero and known one bits, returning them in | ||||
752 | // min/max. | ||||
753 | static void ComputeUnsignedMinMaxValuesFromKnownBits(const Type *Ty, | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 754 | const APInt &KnownZero, |
755 | const APInt &KnownOne, | ||||
756 | APInt &Min, APInt &Max) { | ||||
757 | uint32_t BitWidth = cast<IntegerType>(Ty)->getBitWidth(); BitWidth = BitWidth; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 758 | assert(KnownZero.getBitWidth() == BitWidth && |
759 | KnownOne.getBitWidth() == BitWidth && | ||||
760 | Min.getBitWidth() == BitWidth && Max.getBitWidth() && | ||||
761 | "Ty, KnownZero, KnownOne and Min, Max must have equal bitwidth."); | ||||
762 | APInt UnknownBits = ~(KnownZero|KnownOne); | ||||
763 | |||||
764 | // The minimum value is when the unknown bits are all zeros. | ||||
765 | Min = KnownOne; | ||||
766 | // The maximum value is when the unknown bits are all ones. | ||||
767 | Max = KnownOne|UnknownBits; | ||||
768 | } | ||||
769 | |||||
770 | /// SimplifyDemandedBits - This function attempts to replace V with a simpler | ||||
771 | /// value based on the demanded bits. When this function is called, it is known | ||||
772 | /// that only the bits set in DemandedMask of the result of V are ever used | ||||
773 | /// downstream. Consequently, depending on the mask and V, it may be possible | ||||
774 | /// to replace V with a constant or one of its operands. In such cases, this | ||||
775 | /// function does the replacement and returns true. In all other cases, it | ||||
776 | /// returns false after analyzing the expression and setting KnownOne and known | ||||
777 | /// to be one in the expression. KnownZero contains all the bits that are known | ||||
778 | /// to be zero in the expression. These are provided to potentially allow the | ||||
779 | /// caller (which might recursively be SimplifyDemandedBits itself) to simplify | ||||
780 | /// the expression. KnownOne and KnownZero always follow the invariant that | ||||
781 | /// KnownOne & KnownZero == 0. That is, a bit can't be both 1 and 0. Note that | ||||
782 | /// the bits in KnownOne and KnownZero may only be accurate for those bits set | ||||
783 | /// in DemandedMask. Note also that the bitwidth of V, DemandedMask, KnownZero | ||||
784 | /// and KnownOne must all be the same. | ||||
785 | bool InstCombiner::SimplifyDemandedBits(Value *V, APInt DemandedMask, | ||||
786 | APInt& KnownZero, APInt& KnownOne, | ||||
787 | unsigned Depth) { | ||||
788 | assert(V != 0 && "Null pointer of Value???"); | ||||
789 | assert(Depth <= 6 && "Limit Search Depth"); | ||||
790 | uint32_t BitWidth = DemandedMask.getBitWidth(); | ||||
791 | const IntegerType *VTy = cast<IntegerType>(V->getType()); | ||||
792 | assert(VTy->getBitWidth() == BitWidth && | ||||
793 | KnownZero.getBitWidth() == BitWidth && | ||||
794 | KnownOne.getBitWidth() == BitWidth && | ||||
795 | "Value *V, DemandedMask, KnownZero and KnownOne \ | ||||
796 | must have same BitWidth"); | ||||
797 | if (ConstantInt *CI = dyn_cast<ConstantInt>(V)) { | ||||
798 | // We know all of the bits for a constant! | ||||
799 | KnownOne = CI->getValue() & DemandedMask; | ||||
800 | KnownZero = ~KnownOne & DemandedMask; | ||||
801 | return false; | ||||
802 | } | ||||
803 | |||||
804 | KnownZero.clear(); | ||||
805 | KnownOne.clear(); | ||||
806 | if (!V->hasOneUse()) { // Other users may use these bits. | ||||
807 | if (Depth != 0) { // Not at the root. | ||||
808 | // Just compute the KnownZero/KnownOne bits to simplify things downstream. | ||||
809 | ComputeMaskedBits(V, DemandedMask, KnownZero, KnownOne, Depth); | ||||
810 | return false; | ||||
811 | } | ||||
812 | // If this is the root being simplified, allow it to have multiple uses, | ||||
813 | // just set the DemandedMask to all bits. | ||||
814 | DemandedMask = APInt::getAllOnesValue(BitWidth); | ||||
815 | } else if (DemandedMask == 0) { // Not demanding any bits from V. | ||||
816 | if (V != UndefValue::get(VTy)) | ||||
817 | return UpdateValueUsesWith(V, UndefValue::get(VTy)); | ||||
818 | return false; | ||||
819 | } else if (Depth == 6) { // Limit search depth. | ||||
820 | return false; | ||||
821 | } | ||||
822 | |||||
823 | Instruction *I = dyn_cast<Instruction>(V); | ||||
824 | if (!I) return false; // Only analyze instructions. | ||||
825 | |||||
826 | APInt LHSKnownZero(BitWidth, 0), LHSKnownOne(BitWidth, 0); | ||||
827 | APInt &RHSKnownZero = KnownZero, &RHSKnownOne = KnownOne; | ||||
828 | switch (I->getOpcode()) { | ||||
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 829 | default: |
830 | ComputeMaskedBits(V, DemandedMask, RHSKnownZero, RHSKnownOne, Depth); | ||||
831 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 832 | case Instruction::And: |
833 | // If either the LHS or the RHS are Zero, the result is zero. | ||||
834 | if (SimplifyDemandedBits(I->getOperand(1), DemandedMask, | ||||
835 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
836 | return true; | ||||
837 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
838 | "Bits known to be one AND zero?"); | ||||
839 | |||||
840 | // If something is known zero on the RHS, the bits aren't demanded on the | ||||
841 | // LHS. | ||||
842 | if (SimplifyDemandedBits(I->getOperand(0), DemandedMask & ~RHSKnownZero, | ||||
843 | LHSKnownZero, LHSKnownOne, Depth+1)) | ||||
844 | return true; | ||||
845 | assert((LHSKnownZero & LHSKnownOne) == 0 && | ||||
846 | "Bits known to be one AND zero?"); | ||||
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'. | ||||
850 | if ((DemandedMask & ~LHSKnownZero & RHSKnownOne) == | ||||
851 | (DemandedMask & ~LHSKnownZero)) | ||||
852 | return UpdateValueUsesWith(I, I->getOperand(0)); | ||||
853 | if ((DemandedMask & ~RHSKnownZero & LHSKnownOne) == | ||||
854 | (DemandedMask & ~RHSKnownZero)) | ||||
855 | return UpdateValueUsesWith(I, I->getOperand(1)); | ||||
856 | |||||
857 | // If all of the demanded bits in the inputs are known zeros, return zero. | ||||
858 | if ((DemandedMask & (RHSKnownZero|LHSKnownZero)) == DemandedMask) | ||||
859 | return UpdateValueUsesWith(I, Constant::getNullValue(VTy)); | ||||
860 | |||||
861 | // If the RHS is a constant, see if we can simplify it. | ||||
862 | if (ShrinkDemandedConstant(I, 1, DemandedMask & ~LHSKnownZero)) | ||||
863 | return UpdateValueUsesWith(I, I); | ||||
864 | |||||
865 | // Output known-1 bits are only known if set in both the LHS & RHS. | ||||
866 | RHSKnownOne &= LHSKnownOne; | ||||
867 | // Output known-0 are known to be clear if zero in either the LHS | RHS. | ||||
868 | RHSKnownZero |= LHSKnownZero; | ||||
869 | break; | ||||
870 | case Instruction::Or: | ||||
871 | // If either the LHS or the RHS are One, the result is One. | ||||
872 | if (SimplifyDemandedBits(I->getOperand(1), DemandedMask, | ||||
873 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
874 | return true; | ||||
875 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
876 | "Bits known to be one AND zero?"); | ||||
877 | // If something is known one on the RHS, the bits aren't demanded on the | ||||
878 | // LHS. | ||||
879 | if (SimplifyDemandedBits(I->getOperand(0), DemandedMask & ~RHSKnownOne, | ||||
880 | LHSKnownZero, LHSKnownOne, Depth+1)) | ||||
881 | return true; | ||||
882 | assert((LHSKnownZero & LHSKnownOne) == 0 && | ||||
883 | "Bits known to be one AND zero?"); | ||||
884 | |||||
885 | // If all of the demanded bits are known zero on one side, return the other. | ||||
886 | // These bits cannot contribute to the result of the 'or'. | ||||
887 | if ((DemandedMask & ~LHSKnownOne & RHSKnownZero) == | ||||
888 | (DemandedMask & ~LHSKnownOne)) | ||||
889 | return UpdateValueUsesWith(I, I->getOperand(0)); | ||||
890 | if ((DemandedMask & ~RHSKnownOne & LHSKnownZero) == | ||||
891 | (DemandedMask & ~RHSKnownOne)) | ||||
892 | return UpdateValueUsesWith(I, I->getOperand(1)); | ||||
893 | |||||
894 | // If all of the potentially set bits on one side are known to be set on | ||||
895 | // the other side, just use the 'other' side. | ||||
896 | if ((DemandedMask & (~RHSKnownZero) & LHSKnownOne) == | ||||
897 | (DemandedMask & (~RHSKnownZero))) | ||||
898 | return UpdateValueUsesWith(I, I->getOperand(0)); | ||||
899 | if ((DemandedMask & (~LHSKnownZero) & RHSKnownOne) == | ||||
900 | (DemandedMask & (~LHSKnownZero))) | ||||
901 | return UpdateValueUsesWith(I, I->getOperand(1)); | ||||
902 | |||||
903 | // If the RHS is a constant, see if we can simplify it. | ||||
904 | if (ShrinkDemandedConstant(I, 1, DemandedMask)) | ||||
905 | return UpdateValueUsesWith(I, I); | ||||
906 | |||||
907 | // Output known-0 bits are only known if clear in both the LHS & RHS. | ||||
908 | RHSKnownZero &= LHSKnownZero; | ||||
909 | // Output known-1 are known to be set if set in either the LHS | RHS. | ||||
910 | RHSKnownOne |= LHSKnownOne; | ||||
911 | break; | ||||
912 | case Instruction::Xor: { | ||||
913 | if (SimplifyDemandedBits(I->getOperand(1), DemandedMask, | ||||
914 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
915 | return true; | ||||
916 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
917 | "Bits known to be one AND zero?"); | ||||
918 | if (SimplifyDemandedBits(I->getOperand(0), DemandedMask, | ||||
919 | LHSKnownZero, LHSKnownOne, Depth+1)) | ||||
920 | return true; | ||||
921 | assert((LHSKnownZero & LHSKnownOne) == 0 && | ||||
922 | "Bits known to be one AND zero?"); | ||||
923 | |||||
924 | // If all of the demanded bits are known zero on one side, return the other. | ||||
925 | // These bits cannot contribute to the result of the 'xor'. | ||||
926 | if ((DemandedMask & RHSKnownZero) == DemandedMask) | ||||
927 | return UpdateValueUsesWith(I, I->getOperand(0)); | ||||
928 | if ((DemandedMask & LHSKnownZero) == DemandedMask) | ||||
929 | return UpdateValueUsesWith(I, I->getOperand(1)); | ||||
930 | |||||
931 | // Output known-0 bits are known if clear or set in both the LHS & RHS. | ||||
932 | APInt KnownZeroOut = (RHSKnownZero & LHSKnownZero) | | ||||
933 | (RHSKnownOne & LHSKnownOne); | ||||
934 | // Output known-1 are known to be set if set in only one of the LHS, RHS. | ||||
935 | APInt KnownOneOut = (RHSKnownZero & LHSKnownOne) | | ||||
936 | (RHSKnownOne & LHSKnownZero); | ||||
937 | |||||
938 | // If all of the demanded bits are known to be zero on one side or the | ||||
939 | // other, turn this into an *inclusive* or. | ||||
940 | // e.g. (A & C1)^(B & C2) -> (A & C1)|(B & C2) iff C1&C2 == 0 | ||||
941 | if ((DemandedMask & ~RHSKnownZero & ~LHSKnownZero) == 0) { | ||||
942 | Instruction *Or = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 943 | BinaryOperator::CreateOr(I->getOperand(0), I->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 944 | I->getName()); |
945 | InsertNewInstBefore(Or, *I); | ||||
946 | return UpdateValueUsesWith(I, Or); | ||||
947 | } | ||||
948 | |||||
949 | // If all of the demanded bits on one side are known, and all of the set | ||||
950 | // bits on that side are also known to be set on the other side, turn this | ||||
951 | // into an AND, as we know the bits will be cleared. | ||||
952 | // e.g. (X | C1) ^ C2 --> (X | C1) & ~C2 iff (C1&C2) == C2 | ||||
953 | if ((DemandedMask & (RHSKnownZero|RHSKnownOne)) == DemandedMask) { | ||||
954 | // all known | ||||
955 | if ((RHSKnownOne & LHSKnownOne) == RHSKnownOne) { | ||||
956 | Constant *AndC = ConstantInt::get(~RHSKnownOne & DemandedMask); | ||||
957 | Instruction *And = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 958 | BinaryOperator::CreateAnd(I->getOperand(0), AndC, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 959 | InsertNewInstBefore(And, *I); |
960 | return UpdateValueUsesWith(I, And); | ||||
961 | } | ||||
962 | } | ||||
963 | |||||
964 | // If the RHS is a constant, see if we can simplify it. | ||||
965 | // FIXME: for XOR, we prefer to force bits to 1 if they will make a -1. | ||||
966 | if (ShrinkDemandedConstant(I, 1, DemandedMask)) | ||||
967 | return UpdateValueUsesWith(I, I); | ||||
968 | |||||
969 | RHSKnownZero = KnownZeroOut; | ||||
970 | RHSKnownOne = KnownOneOut; | ||||
971 | break; | ||||
972 | } | ||||
973 | case Instruction::Select: | ||||
974 | if (SimplifyDemandedBits(I->getOperand(2), DemandedMask, | ||||
975 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
976 | return true; | ||||
977 | if (SimplifyDemandedBits(I->getOperand(1), DemandedMask, | ||||
978 | LHSKnownZero, LHSKnownOne, Depth+1)) | ||||
979 | return true; | ||||
980 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
981 | "Bits known to be one AND zero?"); | ||||
982 | assert((LHSKnownZero & LHSKnownOne) == 0 && | ||||
983 | "Bits known to be one AND zero?"); | ||||
984 | |||||
985 | // If the operands are constants, see if we can simplify them. | ||||
986 | if (ShrinkDemandedConstant(I, 1, DemandedMask)) | ||||
987 | return UpdateValueUsesWith(I, I); | ||||
988 | if (ShrinkDemandedConstant(I, 2, DemandedMask)) | ||||
989 | return UpdateValueUsesWith(I, I); | ||||
990 | |||||
991 | // Only known if known in both the LHS and RHS. | ||||
992 | RHSKnownOne &= LHSKnownOne; | ||||
993 | RHSKnownZero &= LHSKnownZero; | ||||
994 | break; | ||||
995 | case Instruction::Trunc: { | ||||
996 | uint32_t truncBf = | ||||
997 | cast<IntegerType>(I->getOperand(0)->getType())->getBitWidth(); | ||||
998 | DemandedMask.zext(truncBf); | ||||
999 | RHSKnownZero.zext(truncBf); | ||||
1000 | RHSKnownOne.zext(truncBf); | ||||
1001 | if (SimplifyDemandedBits(I->getOperand(0), DemandedMask, | ||||
1002 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
1003 | return true; | ||||
1004 | DemandedMask.trunc(BitWidth); | ||||
1005 | RHSKnownZero.trunc(BitWidth); | ||||
1006 | RHSKnownOne.trunc(BitWidth); | ||||
1007 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
1008 | "Bits known to be one AND zero?"); | ||||
1009 | break; | ||||
1010 | } | ||||
1011 | case Instruction::BitCast: | ||||
1012 | if (!I->getOperand(0)->getType()->isInteger()) | ||||
1013 | return false; | ||||
1014 | |||||
1015 | if (SimplifyDemandedBits(I->getOperand(0), DemandedMask, | ||||
1016 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
1017 | return true; | ||||
1018 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
1019 | "Bits known to be one AND zero?"); | ||||
1020 | break; | ||||
1021 | case Instruction::ZExt: { | ||||
1022 | // Compute the bits in the result that are not present in the input. | ||||
1023 | const IntegerType *SrcTy = cast<IntegerType>(I->getOperand(0)->getType()); | ||||
1024 | uint32_t SrcBitWidth = SrcTy->getBitWidth(); | ||||
1025 | |||||
1026 | DemandedMask.trunc(SrcBitWidth); | ||||
1027 | RHSKnownZero.trunc(SrcBitWidth); | ||||
1028 | RHSKnownOne.trunc(SrcBitWidth); | ||||
1029 | if (SimplifyDemandedBits(I->getOperand(0), DemandedMask, | ||||
1030 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
1031 | return true; | ||||
1032 | DemandedMask.zext(BitWidth); | ||||
1033 | RHSKnownZero.zext(BitWidth); | ||||
1034 | RHSKnownOne.zext(BitWidth); | ||||
1035 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
1036 | "Bits known to be one AND zero?"); | ||||
1037 | // The top bits are known to be zero. | ||||
1038 | RHSKnownZero |= APInt::getHighBitsSet(BitWidth, BitWidth - SrcBitWidth); | ||||
1039 | break; | ||||
1040 | } | ||||
1041 | case Instruction::SExt: { | ||||
1042 | // Compute the bits in the result that are not present in the input. | ||||
1043 | const IntegerType *SrcTy = cast<IntegerType>(I->getOperand(0)->getType()); | ||||
1044 | uint32_t SrcBitWidth = SrcTy->getBitWidth(); | ||||
1045 | |||||
1046 | APInt InputDemandedBits = DemandedMask & | ||||
1047 | APInt::getLowBitsSet(BitWidth, SrcBitWidth); | ||||
1048 | |||||
1049 | APInt NewBits(APInt::getHighBitsSet(BitWidth, BitWidth - SrcBitWidth)); | ||||
1050 | // If any of the sign extended bits are demanded, we know that the sign | ||||
1051 | // bit is demanded. | ||||
1052 | if ((NewBits & DemandedMask) != 0) | ||||
1053 | InputDemandedBits.set(SrcBitWidth-1); | ||||
1054 | |||||
1055 | InputDemandedBits.trunc(SrcBitWidth); | ||||
1056 | RHSKnownZero.trunc(SrcBitWidth); | ||||
1057 | RHSKnownOne.trunc(SrcBitWidth); | ||||
1058 | if (SimplifyDemandedBits(I->getOperand(0), InputDemandedBits, | ||||
1059 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
1060 | return true; | ||||
1061 | InputDemandedBits.zext(BitWidth); | ||||
1062 | RHSKnownZero.zext(BitWidth); | ||||
1063 | RHSKnownOne.zext(BitWidth); | ||||
1064 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
1065 | "Bits known to be one AND zero?"); | ||||
1066 | |||||
1067 | // If the sign bit of the input is known set or clear, then we know the | ||||
1068 | // top bits of the result. | ||||
1069 | |||||
1070 | // If the input sign bit is known zero, or if the NewBits are not demanded | ||||
1071 | // convert this into a zero extension. | ||||
1072 | if (RHSKnownZero[SrcBitWidth-1] || (NewBits & ~DemandedMask) == NewBits) | ||||
1073 | { | ||||
1074 | // Convert to ZExt cast | ||||
1075 | CastInst *NewCast = new ZExtInst(I->getOperand(0), VTy, I->getName(), I); | ||||
1076 | return UpdateValueUsesWith(I, NewCast); | ||||
1077 | } else if (RHSKnownOne[SrcBitWidth-1]) { // Input sign bit known set | ||||
1078 | RHSKnownOne |= NewBits; | ||||
1079 | } | ||||
1080 | break; | ||||
1081 | } | ||||
1082 | case Instruction::Add: { | ||||
1083 | // Figure out what the input bits are. If the top bits of the and result | ||||
1084 | // are not demanded, then the add doesn't demand them from its input | ||||
1085 | // either. | ||||
1086 | uint32_t NLZ = DemandedMask.countLeadingZeros(); | ||||
1087 | |||||
1088 | // If there is a constant on the RHS, there are a variety of xformations | ||||
1089 | // we can do. | ||||
1090 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1091 | // If null, this should be simplified elsewhere. Some of the xforms here | ||||
1092 | // won't work if the RHS is zero. | ||||
1093 | if (RHS->isZero()) | ||||
1094 | break; | ||||
1095 | |||||
1096 | // If the top bit of the output is demanded, demand everything from the | ||||
1097 | // input. Otherwise, we demand all the input bits except NLZ top bits. | ||||
1098 | APInt InDemandedBits(APInt::getLowBitsSet(BitWidth, BitWidth - NLZ)); | ||||
1099 | |||||
1100 | // Find information about known zero/one bits in the input. | ||||
1101 | if (SimplifyDemandedBits(I->getOperand(0), InDemandedBits, | ||||
1102 | LHSKnownZero, LHSKnownOne, Depth+1)) | ||||
1103 | return true; | ||||
1104 | |||||
1105 | // If the RHS of the add has bits set that can't affect the input, reduce | ||||
1106 | // the constant. | ||||
1107 | if (ShrinkDemandedConstant(I, 1, InDemandedBits)) | ||||
1108 | return UpdateValueUsesWith(I, I); | ||||
1109 | |||||
1110 | // Avoid excess work. | ||||
1111 | if (LHSKnownZero == 0 && LHSKnownOne == 0) | ||||
1112 | break; | ||||
1113 | |||||
1114 | // Turn it into OR if input bits are zero. | ||||
1115 | if ((LHSKnownZero & RHS->getValue()) == RHS->getValue()) { | ||||
1116 | Instruction *Or = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1117 | BinaryOperator::CreateOr(I->getOperand(0), I->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1118 | I->getName()); |
1119 | InsertNewInstBefore(Or, *I); | ||||
1120 | return UpdateValueUsesWith(I, Or); | ||||
1121 | } | ||||
1122 | |||||
1123 | // We can say something about the output known-zero and known-one bits, | ||||
1124 | // depending on potential carries from the input constant and the | ||||
1125 | // unknowns. For example if the LHS is known to have at most the 0x0F0F0 | ||||
1126 | // bits set and the RHS constant is 0x01001, then we know we have a known | ||||
1127 | // one mask of 0x00001 and a known zero mask of 0xE0F0E. | ||||
1128 | |||||
1129 | // To compute this, we first compute the potential carry bits. These are | ||||
1130 | // the bits which may be modified. I'm not aware of a better way to do | ||||
1131 | // this scan. | ||||
1132 | const APInt& RHSVal = RHS->getValue(); | ||||
1133 | APInt CarryBits((~LHSKnownZero + RHSVal) ^ (~LHSKnownZero ^ RHSVal)); | ||||
1134 | |||||
1135 | // Now that we know which bits have carries, compute the known-1/0 sets. | ||||
1136 | |||||
1137 | // Bits are known one if they are known zero in one operand and one in the | ||||
1138 | // other, and there is no input carry. | ||||
1139 | RHSKnownOne = ((LHSKnownZero & RHSVal) | | ||||
1140 | (LHSKnownOne & ~RHSVal)) & ~CarryBits; | ||||
1141 | |||||
1142 | // Bits are known zero if they are known zero in both operands and there | ||||
1143 | // is no input carry. | ||||
1144 | RHSKnownZero = LHSKnownZero & ~RHSVal & ~CarryBits; | ||||
1145 | } else { | ||||
1146 | // If the high-bits of this ADD are not demanded, then it does not demand | ||||
1147 | // the high bits of its LHS or RHS. | ||||
1148 | if (DemandedMask[BitWidth-1] == 0) { | ||||
1149 | // Right fill the mask of bits for this ADD to demand the most | ||||
1150 | // significant bit and all those below it. | ||||
1151 | APInt DemandedFromOps(APInt::getLowBitsSet(BitWidth, BitWidth-NLZ)); | ||||
1152 | if (SimplifyDemandedBits(I->getOperand(0), DemandedFromOps, | ||||
1153 | LHSKnownZero, LHSKnownOne, Depth+1)) | ||||
1154 | return true; | ||||
1155 | if (SimplifyDemandedBits(I->getOperand(1), DemandedFromOps, | ||||
1156 | LHSKnownZero, LHSKnownOne, Depth+1)) | ||||
1157 | return true; | ||||
1158 | } | ||||
1159 | } | ||||
1160 | break; | ||||
1161 | } | ||||
1162 | case Instruction::Sub: | ||||
1163 | // If the high-bits of this SUB are not demanded, then it does not demand | ||||
1164 | // the high bits of its LHS or RHS. | ||||
1165 | if (DemandedMask[BitWidth-1] == 0) { | ||||
1166 | // Right fill the mask of bits for this SUB to demand the most | ||||
1167 | // significant bit and all those below it. | ||||
1168 | uint32_t NLZ = DemandedMask.countLeadingZeros(); | ||||
1169 | APInt DemandedFromOps(APInt::getLowBitsSet(BitWidth, BitWidth-NLZ)); | ||||
1170 | if (SimplifyDemandedBits(I->getOperand(0), DemandedFromOps, | ||||
1171 | LHSKnownZero, LHSKnownOne, Depth+1)) | ||||
1172 | return true; | ||||
1173 | if (SimplifyDemandedBits(I->getOperand(1), DemandedFromOps, | ||||
1174 | LHSKnownZero, LHSKnownOne, Depth+1)) | ||||
1175 | return true; | ||||
1176 | } | ||||
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1177 | // Otherwise just hand the sub off to ComputeMaskedBits to fill in |
1178 | // the known zeros and ones. | ||||
1179 | ComputeMaskedBits(V, DemandedMask, RHSKnownZero, RHSKnownOne, Depth); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1180 | break; |
1181 | case Instruction::Shl: | ||||
1182 | if (ConstantInt *SA = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1183 | uint64_t ShiftAmt = SA->getLimitedValue(BitWidth); | ||||
1184 | APInt DemandedMaskIn(DemandedMask.lshr(ShiftAmt)); | ||||
1185 | if (SimplifyDemandedBits(I->getOperand(0), DemandedMaskIn, | ||||
1186 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
1187 | return true; | ||||
1188 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
1189 | "Bits known to be one AND zero?"); | ||||
1190 | RHSKnownZero <<= ShiftAmt; | ||||
1191 | RHSKnownOne <<= ShiftAmt; | ||||
1192 | // low bits known zero. | ||||
1193 | if (ShiftAmt) | ||||
1194 | RHSKnownZero |= APInt::getLowBitsSet(BitWidth, ShiftAmt); | ||||
1195 | } | ||||
1196 | break; | ||||
1197 | case Instruction::LShr: | ||||
1198 | // For a logical shift right | ||||
1199 | if (ConstantInt *SA = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1200 | uint64_t ShiftAmt = SA->getLimitedValue(BitWidth); | ||||
1201 | |||||
1202 | // Unsigned shift right. | ||||
1203 | APInt DemandedMaskIn(DemandedMask.shl(ShiftAmt)); | ||||
1204 | if (SimplifyDemandedBits(I->getOperand(0), DemandedMaskIn, | ||||
1205 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
1206 | return true; | ||||
1207 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
1208 | "Bits known to be one AND zero?"); | ||||
1209 | RHSKnownZero = APIntOps::lshr(RHSKnownZero, ShiftAmt); | ||||
1210 | RHSKnownOne = APIntOps::lshr(RHSKnownOne, ShiftAmt); | ||||
1211 | if (ShiftAmt) { | ||||
1212 | // Compute the new bits that are at the top now. | ||||
1213 | APInt HighBits(APInt::getHighBitsSet(BitWidth, ShiftAmt)); | ||||
1214 | RHSKnownZero |= HighBits; // high bits known zero. | ||||
1215 | } | ||||
1216 | } | ||||
1217 | break; | ||||
1218 | case Instruction::AShr: | ||||
1219 | // If this is an arithmetic shift right and only the low-bit is set, we can | ||||
1220 | // always convert this into a logical shr, even if the shift amount is | ||||
1221 | // variable. The low bit of the shift cannot be an input sign bit unless | ||||
1222 | // the shift amount is >= the size of the datatype, which is undefined. | ||||
1223 | if (DemandedMask == 1) { | ||||
1224 | // Perform the logical shift right. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1225 | Value *NewVal = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1226 | I->getOperand(0), I->getOperand(1), I->getName()); |
1227 | InsertNewInstBefore(cast<Instruction>(NewVal), *I); | ||||
1228 | return UpdateValueUsesWith(I, NewVal); | ||||
1229 | } | ||||
1230 | |||||
1231 | // If the sign bit is the only bit demanded by this ashr, then there is no | ||||
1232 | // need to do it, the shift doesn't change the high bit. | ||||
1233 | if (DemandedMask.isSignBit()) | ||||
1234 | return UpdateValueUsesWith(I, I->getOperand(0)); | ||||
1235 | |||||
1236 | if (ConstantInt *SA = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1237 | uint32_t ShiftAmt = SA->getLimitedValue(BitWidth); | ||||
1238 | |||||
1239 | // Signed shift right. | ||||
1240 | APInt DemandedMaskIn(DemandedMask.shl(ShiftAmt)); | ||||
1241 | // If any of the "high bits" are demanded, we should set the sign bit as | ||||
1242 | // demanded. | ||||
1243 | if (DemandedMask.countLeadingZeros() <= ShiftAmt) | ||||
1244 | DemandedMaskIn.set(BitWidth-1); | ||||
1245 | if (SimplifyDemandedBits(I->getOperand(0), | ||||
1246 | DemandedMaskIn, | ||||
1247 | RHSKnownZero, RHSKnownOne, Depth+1)) | ||||
1248 | return true; | ||||
1249 | assert((RHSKnownZero & RHSKnownOne) == 0 && | ||||
1250 | "Bits known to be one AND zero?"); | ||||
1251 | // Compute the new bits that are at the top now. | ||||
1252 | APInt HighBits(APInt::getHighBitsSet(BitWidth, ShiftAmt)); | ||||
1253 | RHSKnownZero = APIntOps::lshr(RHSKnownZero, ShiftAmt); | ||||
1254 | RHSKnownOne = APIntOps::lshr(RHSKnownOne, ShiftAmt); | ||||
1255 | |||||
1256 | // Handle the sign bits. | ||||
1257 | APInt SignBit(APInt::getSignBit(BitWidth)); | ||||
1258 | // Adjust to where it is now in the mask. | ||||
1259 | SignBit = APIntOps::lshr(SignBit, ShiftAmt); | ||||
1260 | |||||
1261 | // If the input sign bit is known to be zero, or if none of the top bits | ||||
1262 | // are demanded, turn this into an unsigned shift right. | ||||
Zhou Sheng | 533604e | 2008-06-06 08:32:05 +0000 | [diff] [blame] | 1263 | if (BitWidth <= ShiftAmt || RHSKnownZero[BitWidth-ShiftAmt-1] || |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1264 | (HighBits & ~DemandedMask) == HighBits) { |
1265 | // Perform the logical shift right. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1266 | Value *NewVal = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1267 | I->getOperand(0), SA, I->getName()); |
1268 | InsertNewInstBefore(cast<Instruction>(NewVal), *I); | ||||
1269 | return UpdateValueUsesWith(I, NewVal); | ||||
1270 | } else if ((RHSKnownOne & SignBit) != 0) { // New bits are known one. | ||||
1271 | RHSKnownOne |= HighBits; | ||||
1272 | } | ||||
1273 | } | ||||
1274 | break; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1275 | case Instruction::SRem: |
1276 | if (ConstantInt *Rem = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1277 | APInt RA = Rem->getValue(); | ||||
1278 | if (RA.isPowerOf2() || (-RA).isPowerOf2()) { | ||||
Nick Lewycky | 245de42 | 2008-07-12 05:04:38 +0000 | [diff] [blame] | 1279 | if (DemandedMask.ule(RA)) // srem won't affect demanded bits |
1280 | return UpdateValueUsesWith(I, I->getOperand(0)); | ||||
1281 | |||||
Dan Gohman | 5a154a1 | 2008-05-06 00:51:48 +0000 | [diff] [blame] | 1282 | APInt LowBits = RA.isStrictlyPositive() ? (RA - 1) : ~RA; |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1283 | APInt Mask2 = LowBits | APInt::getSignBit(BitWidth); |
1284 | if (SimplifyDemandedBits(I->getOperand(0), Mask2, | ||||
1285 | LHSKnownZero, LHSKnownOne, Depth+1)) | ||||
1286 | return true; | ||||
1287 | |||||
1288 | if (LHSKnownZero[BitWidth-1] || ((LHSKnownZero & LowBits) == LowBits)) | ||||
1289 | LHSKnownZero |= ~LowBits; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1290 | |
1291 | KnownZero |= LHSKnownZero & DemandedMask; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1292 | |
1293 | assert((KnownZero & KnownOne) == 0&&"Bits known to be one AND zero?"); | ||||
1294 | } | ||||
1295 | } | ||||
1296 | break; | ||||
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1297 | case Instruction::URem: { |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1298 | APInt KnownZero2(BitWidth, 0), KnownOne2(BitWidth, 0); |
1299 | APInt AllOnes = APInt::getAllOnesValue(BitWidth); | ||||
Dan Gohman | 23ea06d | 2008-05-01 19:13:24 +0000 | [diff] [blame] | 1300 | if (SimplifyDemandedBits(I->getOperand(0), AllOnes, |
1301 | KnownZero2, KnownOne2, Depth+1)) | ||||
1302 | return true; | ||||
1303 | |||||
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1304 | uint32_t Leaders = KnownZero2.countLeadingOnes(); |
Dan Gohman | 23ea06d | 2008-05-01 19:13:24 +0000 | [diff] [blame] | 1305 | if (SimplifyDemandedBits(I->getOperand(1), AllOnes, |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1306 | KnownZero2, KnownOne2, Depth+1)) |
1307 | return true; | ||||
1308 | |||||
1309 | Leaders = std::max(Leaders, | ||||
1310 | KnownZero2.countLeadingOnes()); | ||||
1311 | KnownZero = APInt::getHighBitsSet(BitWidth, Leaders) & DemandedMask; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1312 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1313 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 1314 | case Instruction::Call: |
1315 | if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(I)) { | ||||
1316 | switch (II->getIntrinsicID()) { | ||||
1317 | default: break; | ||||
1318 | case Intrinsic::bswap: { | ||||
1319 | // If the only bits demanded come from one byte of the bswap result, | ||||
1320 | // just shift the input byte into position to eliminate the bswap. | ||||
1321 | unsigned NLZ = DemandedMask.countLeadingZeros(); | ||||
1322 | unsigned NTZ = DemandedMask.countTrailingZeros(); | ||||
1323 | |||||
1324 | // Round NTZ down to the next byte. If we have 11 trailing zeros, then | ||||
1325 | // we need all the bits down to bit 8. Likewise, round NLZ. If we | ||||
1326 | // have 14 leading zeros, round to 8. | ||||
1327 | NLZ &= ~7; | ||||
1328 | NTZ &= ~7; | ||||
1329 | // If we need exactly one byte, we can do this transformation. | ||||
1330 | if (BitWidth-NLZ-NTZ == 8) { | ||||
1331 | unsigned ResultBit = NTZ; | ||||
1332 | unsigned InputBit = BitWidth-NTZ-8; | ||||
1333 | |||||
1334 | // Replace this with either a left or right shift to get the byte into | ||||
1335 | // the right place. | ||||
1336 | Instruction *NewVal; | ||||
1337 | if (InputBit > ResultBit) | ||||
1338 | NewVal = BinaryOperator::CreateLShr(I->getOperand(1), | ||||
1339 | ConstantInt::get(I->getType(), InputBit-ResultBit)); | ||||
1340 | else | ||||
1341 | NewVal = BinaryOperator::CreateShl(I->getOperand(1), | ||||
1342 | ConstantInt::get(I->getType(), ResultBit-InputBit)); | ||||
1343 | NewVal->takeName(I); | ||||
1344 | InsertNewInstBefore(NewVal, *I); | ||||
1345 | return UpdateValueUsesWith(I, NewVal); | ||||
1346 | } | ||||
1347 | |||||
1348 | // TODO: Could compute known zero/one bits based on the input. | ||||
1349 | break; | ||||
1350 | } | ||||
1351 | } | ||||
1352 | } | ||||
Chris Lattner | 4946e22 | 2008-06-18 18:11:55 +0000 | [diff] [blame] | 1353 | ComputeMaskedBits(V, DemandedMask, RHSKnownZero, RHSKnownOne, Depth); |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 1354 | break; |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1355 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1356 | |
1357 | // If the client is only demanding bits that we know, return the known | ||||
1358 | // constant. | ||||
1359 | if ((DemandedMask & (RHSKnownZero|RHSKnownOne)) == DemandedMask) | ||||
1360 | return UpdateValueUsesWith(I, ConstantInt::get(RHSKnownOne)); | ||||
1361 | return false; | ||||
1362 | } | ||||
1363 | |||||
1364 | |||||
1365 | /// SimplifyDemandedVectorElts - The specified value producecs a vector with | ||||
1366 | /// 64 or fewer elements. DemandedElts contains the set of elements that are | ||||
1367 | /// actually used by the caller. This method analyzes which elements of the | ||||
1368 | /// operand are undef and returns that information in UndefElts. | ||||
1369 | /// | ||||
1370 | /// If the information about demanded elements can be used to simplify the | ||||
1371 | /// operation, the operation is simplified, then the resultant value is | ||||
1372 | /// returned. This returns null if no change was made. | ||||
1373 | Value *InstCombiner::SimplifyDemandedVectorElts(Value *V, uint64_t DemandedElts, | ||||
1374 | uint64_t &UndefElts, | ||||
1375 | unsigned Depth) { | ||||
1376 | unsigned VWidth = cast<VectorType>(V->getType())->getNumElements(); | ||||
1377 | assert(VWidth <= 64 && "Vector too wide to analyze!"); | ||||
1378 | uint64_t EltMask = ~0ULL >> (64-VWidth); | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1379 | assert((DemandedElts & ~EltMask) == 0 && "Invalid DemandedElts!"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1380 | |
1381 | if (isa<UndefValue>(V)) { | ||||
1382 | // If the entire vector is undefined, just return this info. | ||||
1383 | UndefElts = EltMask; | ||||
1384 | return 0; | ||||
1385 | } else if (DemandedElts == 0) { // If nothing is demanded, provide undef. | ||||
1386 | UndefElts = EltMask; | ||||
1387 | return UndefValue::get(V->getType()); | ||||
1388 | } | ||||
1389 | |||||
1390 | UndefElts = 0; | ||||
1391 | if (ConstantVector *CP = dyn_cast<ConstantVector>(V)) { | ||||
1392 | const Type *EltTy = cast<VectorType>(V->getType())->getElementType(); | ||||
1393 | Constant *Undef = UndefValue::get(EltTy); | ||||
1394 | |||||
1395 | std::vector<Constant*> Elts; | ||||
1396 | for (unsigned i = 0; i != VWidth; ++i) | ||||
1397 | if (!(DemandedElts & (1ULL << i))) { // If not demanded, set to undef. | ||||
1398 | Elts.push_back(Undef); | ||||
1399 | UndefElts |= (1ULL << i); | ||||
1400 | } else if (isa<UndefValue>(CP->getOperand(i))) { // Already undef. | ||||
1401 | Elts.push_back(Undef); | ||||
1402 | UndefElts |= (1ULL << i); | ||||
1403 | } else { // Otherwise, defined. | ||||
1404 | Elts.push_back(CP->getOperand(i)); | ||||
1405 | } | ||||
1406 | |||||
1407 | // If we changed the constant, return it. | ||||
1408 | Constant *NewCP = ConstantVector::get(Elts); | ||||
1409 | return NewCP != CP ? NewCP : 0; | ||||
1410 | } else if (isa<ConstantAggregateZero>(V)) { | ||||
1411 | // Simplify the CAZ to a ConstantVector where the non-demanded elements are | ||||
1412 | // set to undef. | ||||
1413 | const Type *EltTy = cast<VectorType>(V->getType())->getElementType(); | ||||
1414 | Constant *Zero = Constant::getNullValue(EltTy); | ||||
1415 | Constant *Undef = UndefValue::get(EltTy); | ||||
1416 | std::vector<Constant*> Elts; | ||||
1417 | for (unsigned i = 0; i != VWidth; ++i) | ||||
1418 | Elts.push_back((DemandedElts & (1ULL << i)) ? Zero : Undef); | ||||
1419 | UndefElts = DemandedElts ^ EltMask; | ||||
1420 | return ConstantVector::get(Elts); | ||||
1421 | } | ||||
1422 | |||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1423 | // Limit search depth. |
1424 | if (Depth == 10) | ||||
1425 | return false; | ||||
1426 | |||||
1427 | // If multiple users are using the root value, procede with | ||||
1428 | // simplification conservatively assuming that all elements | ||||
1429 | // are needed. | ||||
1430 | if (!V->hasOneUse()) { | ||||
1431 | // Quit if we find multiple users of a non-root value though. | ||||
1432 | // They'll be handled when it's their turn to be visited by | ||||
1433 | // the main instcombine process. | ||||
1434 | if (Depth != 0) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1435 | // TODO: Just compute the UndefElts information recursively. |
1436 | return false; | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1437 | |
1438 | // Conservatively assume that all elements are needed. | ||||
1439 | DemandedElts = EltMask; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1440 | } |
1441 | |||||
1442 | Instruction *I = dyn_cast<Instruction>(V); | ||||
1443 | if (!I) return false; // Only analyze instructions. | ||||
1444 | |||||
1445 | bool MadeChange = false; | ||||
1446 | uint64_t UndefElts2; | ||||
1447 | Value *TmpV; | ||||
1448 | switch (I->getOpcode()) { | ||||
1449 | default: break; | ||||
1450 | |||||
1451 | case Instruction::InsertElement: { | ||||
1452 | // If this is a variable index, we don't know which element it overwrites. | ||||
1453 | // demand exactly the same input as we produce. | ||||
1454 | ConstantInt *Idx = dyn_cast<ConstantInt>(I->getOperand(2)); | ||||
1455 | if (Idx == 0) { | ||||
1456 | // Note that we can't propagate undef elt info, because we don't know | ||||
1457 | // which elt is getting updated. | ||||
1458 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts, | ||||
1459 | UndefElts2, Depth+1); | ||||
1460 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1461 | break; | ||||
1462 | } | ||||
1463 | |||||
1464 | // If this is inserting an element that isn't demanded, remove this | ||||
1465 | // insertelement. | ||||
1466 | unsigned IdxNo = Idx->getZExtValue(); | ||||
1467 | if (IdxNo >= VWidth || (DemandedElts & (1ULL << IdxNo)) == 0) | ||||
1468 | return AddSoonDeadInstToWorklist(*I, 0); | ||||
1469 | |||||
1470 | // Otherwise, the element inserted overwrites whatever was there, so the | ||||
1471 | // input demanded set is simpler than the output set. | ||||
1472 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), | ||||
1473 | DemandedElts & ~(1ULL << IdxNo), | ||||
1474 | UndefElts, Depth+1); | ||||
1475 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1476 | |||||
1477 | // The inserted element is defined. | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1478 | UndefElts &= ~(1ULL << IdxNo); |
1479 | break; | ||||
1480 | } | ||||
1481 | case Instruction::ShuffleVector: { | ||||
1482 | ShuffleVectorInst *Shuffle = cast<ShuffleVectorInst>(I); | ||||
1483 | uint64_t LeftDemanded = 0, RightDemanded = 0; | ||||
1484 | for (unsigned i = 0; i < VWidth; i++) { | ||||
1485 | if (DemandedElts & (1ULL << i)) { | ||||
1486 | unsigned MaskVal = Shuffle->getMaskValue(i); | ||||
1487 | if (MaskVal != -1u) { | ||||
1488 | assert(MaskVal < VWidth * 2 && | ||||
1489 | "shufflevector mask index out of range!"); | ||||
1490 | if (MaskVal < VWidth) | ||||
1491 | LeftDemanded |= 1ULL << MaskVal; | ||||
1492 | else | ||||
1493 | RightDemanded |= 1ULL << (MaskVal - VWidth); | ||||
1494 | } | ||||
1495 | } | ||||
1496 | } | ||||
1497 | |||||
1498 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), LeftDemanded, | ||||
1499 | UndefElts2, Depth+1); | ||||
1500 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1501 | |||||
1502 | uint64_t UndefElts3; | ||||
1503 | TmpV = SimplifyDemandedVectorElts(I->getOperand(1), RightDemanded, | ||||
1504 | UndefElts3, Depth+1); | ||||
1505 | if (TmpV) { I->setOperand(1, TmpV); MadeChange = true; } | ||||
1506 | |||||
1507 | bool NewUndefElts = false; | ||||
1508 | for (unsigned i = 0; i < VWidth; i++) { | ||||
1509 | unsigned MaskVal = Shuffle->getMaskValue(i); | ||||
Dan Gohman | 24f6ee2 | 2008-09-10 01:09:32 +0000 | [diff] [blame] | 1510 | if (MaskVal == -1u) { |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1511 | uint64_t NewBit = 1ULL << i; |
1512 | UndefElts |= NewBit; | ||||
1513 | } else if (MaskVal < VWidth) { | ||||
1514 | uint64_t NewBit = ((UndefElts2 >> MaskVal) & 1) << i; | ||||
1515 | NewUndefElts |= NewBit; | ||||
1516 | UndefElts |= NewBit; | ||||
1517 | } else { | ||||
1518 | uint64_t NewBit = ((UndefElts3 >> (MaskVal - VWidth)) & 1) << i; | ||||
1519 | NewUndefElts |= NewBit; | ||||
1520 | UndefElts |= NewBit; | ||||
1521 | } | ||||
1522 | } | ||||
1523 | |||||
1524 | if (NewUndefElts) { | ||||
1525 | // Add additional discovered undefs. | ||||
1526 | std::vector<Constant*> Elts; | ||||
1527 | for (unsigned i = 0; i < VWidth; ++i) { | ||||
1528 | if (UndefElts & (1ULL << i)) | ||||
1529 | Elts.push_back(UndefValue::get(Type::Int32Ty)); | ||||
1530 | else | ||||
1531 | Elts.push_back(ConstantInt::get(Type::Int32Ty, | ||||
1532 | Shuffle->getMaskValue(i))); | ||||
1533 | } | ||||
1534 | I->setOperand(2, ConstantVector::get(Elts)); | ||||
1535 | MadeChange = true; | ||||
1536 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1537 | break; |
1538 | } | ||||
1539 | case Instruction::BitCast: { | ||||
1540 | // Vector->vector casts only. | ||||
1541 | const VectorType *VTy = dyn_cast<VectorType>(I->getOperand(0)->getType()); | ||||
1542 | if (!VTy) break; | ||||
1543 | unsigned InVWidth = VTy->getNumElements(); | ||||
1544 | uint64_t InputDemandedElts = 0; | ||||
1545 | unsigned Ratio; | ||||
1546 | |||||
1547 | if (VWidth == InVWidth) { | ||||
1548 | // If we are converting from <4 x i32> -> <4 x f32>, we demand the same | ||||
1549 | // elements as are demanded of us. | ||||
1550 | Ratio = 1; | ||||
1551 | InputDemandedElts = DemandedElts; | ||||
1552 | } else if (VWidth > InVWidth) { | ||||
1553 | // Untested so far. | ||||
1554 | break; | ||||
1555 | |||||
1556 | // If there are more elements in the result than there are in the source, | ||||
1557 | // then an input element is live if any of the corresponding output | ||||
1558 | // elements are live. | ||||
1559 | Ratio = VWidth/InVWidth; | ||||
1560 | for (unsigned OutIdx = 0; OutIdx != VWidth; ++OutIdx) { | ||||
1561 | if (DemandedElts & (1ULL << OutIdx)) | ||||
1562 | InputDemandedElts |= 1ULL << (OutIdx/Ratio); | ||||
1563 | } | ||||
1564 | } else { | ||||
1565 | // Untested so far. | ||||
1566 | break; | ||||
1567 | |||||
1568 | // If there are more elements in the source than there are in the result, | ||||
1569 | // then an input element is live if the corresponding output element is | ||||
1570 | // live. | ||||
1571 | Ratio = InVWidth/VWidth; | ||||
1572 | for (unsigned InIdx = 0; InIdx != InVWidth; ++InIdx) | ||||
1573 | if (DemandedElts & (1ULL << InIdx/Ratio)) | ||||
1574 | InputDemandedElts |= 1ULL << InIdx; | ||||
1575 | } | ||||
1576 | |||||
1577 | // div/rem demand all inputs, because they don't want divide by zero. | ||||
1578 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), InputDemandedElts, | ||||
1579 | UndefElts2, Depth+1); | ||||
1580 | if (TmpV) { | ||||
1581 | I->setOperand(0, TmpV); | ||||
1582 | MadeChange = true; | ||||
1583 | } | ||||
1584 | |||||
1585 | UndefElts = UndefElts2; | ||||
1586 | if (VWidth > InVWidth) { | ||||
1587 | assert(0 && "Unimp"); | ||||
1588 | // If there are more elements in the result than there are in the source, | ||||
1589 | // then an output element is undef if the corresponding input element is | ||||
1590 | // undef. | ||||
1591 | for (unsigned OutIdx = 0; OutIdx != VWidth; ++OutIdx) | ||||
1592 | if (UndefElts2 & (1ULL << (OutIdx/Ratio))) | ||||
1593 | UndefElts |= 1ULL << OutIdx; | ||||
1594 | } else if (VWidth < InVWidth) { | ||||
1595 | assert(0 && "Unimp"); | ||||
1596 | // If there are more elements in the source than there are in the result, | ||||
1597 | // then a result element is undef if all of the corresponding input | ||||
1598 | // elements are undef. | ||||
1599 | UndefElts = ~0ULL >> (64-VWidth); // Start out all undef. | ||||
1600 | for (unsigned InIdx = 0; InIdx != InVWidth; ++InIdx) | ||||
1601 | if ((UndefElts2 & (1ULL << InIdx)) == 0) // Not undef? | ||||
1602 | UndefElts &= ~(1ULL << (InIdx/Ratio)); // Clear undef bit. | ||||
1603 | } | ||||
1604 | break; | ||||
1605 | } | ||||
1606 | case Instruction::And: | ||||
1607 | case Instruction::Or: | ||||
1608 | case Instruction::Xor: | ||||
1609 | case Instruction::Add: | ||||
1610 | case Instruction::Sub: | ||||
1611 | case Instruction::Mul: | ||||
1612 | // div/rem demand all inputs, because they don't want divide by zero. | ||||
1613 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts, | ||||
1614 | UndefElts, Depth+1); | ||||
1615 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1616 | TmpV = SimplifyDemandedVectorElts(I->getOperand(1), DemandedElts, | ||||
1617 | UndefElts2, Depth+1); | ||||
1618 | if (TmpV) { I->setOperand(1, TmpV); MadeChange = true; } | ||||
1619 | |||||
1620 | // Output elements are undefined if both are undefined. Consider things | ||||
1621 | // like undef&0. The result is known zero, not undef. | ||||
1622 | UndefElts &= UndefElts2; | ||||
1623 | break; | ||||
1624 | |||||
1625 | case Instruction::Call: { | ||||
1626 | IntrinsicInst *II = dyn_cast<IntrinsicInst>(I); | ||||
1627 | if (!II) break; | ||||
1628 | switch (II->getIntrinsicID()) { | ||||
1629 | default: break; | ||||
1630 | |||||
1631 | // Binary vector operations that work column-wise. A dest element is a | ||||
1632 | // function of the corresponding input elements from the two inputs. | ||||
1633 | case Intrinsic::x86_sse_sub_ss: | ||||
1634 | case Intrinsic::x86_sse_mul_ss: | ||||
1635 | case Intrinsic::x86_sse_min_ss: | ||||
1636 | case Intrinsic::x86_sse_max_ss: | ||||
1637 | case Intrinsic::x86_sse2_sub_sd: | ||||
1638 | case Intrinsic::x86_sse2_mul_sd: | ||||
1639 | case Intrinsic::x86_sse2_min_sd: | ||||
1640 | case Intrinsic::x86_sse2_max_sd: | ||||
1641 | TmpV = SimplifyDemandedVectorElts(II->getOperand(1), DemandedElts, | ||||
1642 | UndefElts, Depth+1); | ||||
1643 | if (TmpV) { II->setOperand(1, TmpV); MadeChange = true; } | ||||
1644 | TmpV = SimplifyDemandedVectorElts(II->getOperand(2), DemandedElts, | ||||
1645 | UndefElts2, Depth+1); | ||||
1646 | if (TmpV) { II->setOperand(2, TmpV); MadeChange = true; } | ||||
1647 | |||||
1648 | // If only the low elt is demanded and this is a scalarizable intrinsic, | ||||
1649 | // scalarize it now. | ||||
1650 | if (DemandedElts == 1) { | ||||
1651 | switch (II->getIntrinsicID()) { | ||||
1652 | default: break; | ||||
1653 | case Intrinsic::x86_sse_sub_ss: | ||||
1654 | case Intrinsic::x86_sse_mul_ss: | ||||
1655 | case Intrinsic::x86_sse2_sub_sd: | ||||
1656 | case Intrinsic::x86_sse2_mul_sd: | ||||
1657 | // TODO: Lower MIN/MAX/ABS/etc | ||||
1658 | Value *LHS = II->getOperand(1); | ||||
1659 | Value *RHS = II->getOperand(2); | ||||
1660 | // Extract the element as scalars. | ||||
1661 | LHS = InsertNewInstBefore(new ExtractElementInst(LHS, 0U,"tmp"), *II); | ||||
1662 | RHS = InsertNewInstBefore(new ExtractElementInst(RHS, 0U,"tmp"), *II); | ||||
1663 | |||||
1664 | switch (II->getIntrinsicID()) { | ||||
1665 | default: assert(0 && "Case stmts out of sync!"); | ||||
1666 | case Intrinsic::x86_sse_sub_ss: | ||||
1667 | case Intrinsic::x86_sse2_sub_sd: | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1668 | TmpV = InsertNewInstBefore(BinaryOperator::CreateSub(LHS, RHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1669 | II->getName()), *II); |
1670 | break; | ||||
1671 | case Intrinsic::x86_sse_mul_ss: | ||||
1672 | case Intrinsic::x86_sse2_mul_sd: | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1673 | TmpV = InsertNewInstBefore(BinaryOperator::CreateMul(LHS, RHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1674 | II->getName()), *II); |
1675 | break; | ||||
1676 | } | ||||
1677 | |||||
1678 | Instruction *New = | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 1679 | InsertElementInst::Create(UndefValue::get(II->getType()), TmpV, 0U, |
1680 | II->getName()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1681 | InsertNewInstBefore(New, *II); |
1682 | AddSoonDeadInstToWorklist(*II, 0); | ||||
1683 | return New; | ||||
1684 | } | ||||
1685 | } | ||||
1686 | |||||
1687 | // Output elements are undefined if both are undefined. Consider things | ||||
1688 | // like undef&0. The result is known zero, not undef. | ||||
1689 | UndefElts &= UndefElts2; | ||||
1690 | break; | ||||
1691 | } | ||||
1692 | break; | ||||
1693 | } | ||||
1694 | } | ||||
1695 | return MadeChange ? I : 0; | ||||
1696 | } | ||||
1697 | |||||
Dan Gohman | 5d56fd4 | 2008-05-19 22:14:15 +0000 | [diff] [blame] | 1698 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1699 | /// AssociativeOpt - Perform an optimization on an associative operator. This |
1700 | /// function is designed to check a chain of associative operators for a | ||||
1701 | /// potential to apply a certain optimization. Since the optimization may be | ||||
1702 | /// applicable if the expression was reassociated, this checks the chain, then | ||||
1703 | /// reassociates the expression as necessary to expose the optimization | ||||
1704 | /// opportunity. This makes use of a special Functor, which must define | ||||
1705 | /// 'shouldApply' and 'apply' methods. | ||||
1706 | /// | ||||
1707 | template<typename Functor> | ||||
Dan Gohman | d8bcf5b | 2008-05-20 01:14:05 +0000 | [diff] [blame] | 1708 | static Instruction *AssociativeOpt(BinaryOperator &Root, const Functor &F) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1709 | unsigned Opcode = Root.getOpcode(); |
1710 | Value *LHS = Root.getOperand(0); | ||||
1711 | |||||
1712 | // Quick check, see if the immediate LHS matches... | ||||
1713 | if (F.shouldApply(LHS)) | ||||
1714 | return F.apply(Root); | ||||
1715 | |||||
1716 | // Otherwise, if the LHS is not of the same opcode as the root, return. | ||||
1717 | Instruction *LHSI = dyn_cast<Instruction>(LHS); | ||||
1718 | while (LHSI && LHSI->getOpcode() == Opcode && LHSI->hasOneUse()) { | ||||
1719 | // Should we apply this transform to the RHS? | ||||
1720 | bool ShouldApply = F.shouldApply(LHSI->getOperand(1)); | ||||
1721 | |||||
1722 | // If not to the RHS, check to see if we should apply to the LHS... | ||||
1723 | if (!ShouldApply && F.shouldApply(LHSI->getOperand(0))) { | ||||
1724 | cast<BinaryOperator>(LHSI)->swapOperands(); // Make the LHS the RHS | ||||
1725 | ShouldApply = true; | ||||
1726 | } | ||||
1727 | |||||
1728 | // If the functor wants to apply the optimization to the RHS of LHSI, | ||||
1729 | // reassociate the expression from ((? op A) op B) to (? op (A op B)) | ||||
1730 | if (ShouldApply) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1731 | // Now all of the instructions are in the current basic block, go ahead |
1732 | // and perform the reassociation. | ||||
1733 | Instruction *TmpLHSI = cast<Instruction>(Root.getOperand(0)); | ||||
1734 | |||||
1735 | // First move the selected RHS to the LHS of the root... | ||||
1736 | Root.setOperand(0, LHSI->getOperand(1)); | ||||
1737 | |||||
1738 | // Make what used to be the LHS of the root be the user of the root... | ||||
1739 | Value *ExtraOperand = TmpLHSI->getOperand(1); | ||||
1740 | if (&Root == TmpLHSI) { | ||||
1741 | Root.replaceAllUsesWith(Constant::getNullValue(TmpLHSI->getType())); | ||||
1742 | return 0; | ||||
1743 | } | ||||
1744 | Root.replaceAllUsesWith(TmpLHSI); // Users now use TmpLHSI | ||||
1745 | TmpLHSI->setOperand(1, &Root); // TmpLHSI now uses the root | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1746 | BasicBlock::iterator ARI = &Root; ++ARI; |
Dan Gohman | 0bb9a3d | 2008-06-19 17:47:47 +0000 | [diff] [blame] | 1747 | TmpLHSI->moveBefore(ARI); // Move TmpLHSI to after Root |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1748 | ARI = Root; |
1749 | |||||
1750 | // Now propagate the ExtraOperand down the chain of instructions until we | ||||
1751 | // get to LHSI. | ||||
1752 | while (TmpLHSI != LHSI) { | ||||
1753 | Instruction *NextLHSI = cast<Instruction>(TmpLHSI->getOperand(0)); | ||||
1754 | // Move the instruction to immediately before the chain we are | ||||
1755 | // constructing to avoid breaking dominance properties. | ||||
Dan Gohman | 0bb9a3d | 2008-06-19 17:47:47 +0000 | [diff] [blame] | 1756 | NextLHSI->moveBefore(ARI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1757 | ARI = NextLHSI; |
1758 | |||||
1759 | Value *NextOp = NextLHSI->getOperand(1); | ||||
1760 | NextLHSI->setOperand(1, ExtraOperand); | ||||
1761 | TmpLHSI = NextLHSI; | ||||
1762 | ExtraOperand = NextOp; | ||||
1763 | } | ||||
1764 | |||||
1765 | // Now that the instructions are reassociated, have the functor perform | ||||
1766 | // the transformation... | ||||
1767 | return F.apply(Root); | ||||
1768 | } | ||||
1769 | |||||
1770 | LHSI = dyn_cast<Instruction>(LHSI->getOperand(0)); | ||||
1771 | } | ||||
1772 | return 0; | ||||
1773 | } | ||||
1774 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 1775 | namespace { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1776 | |
Nick Lewycky | 27f6c13 | 2008-05-23 04:34:58 +0000 | [diff] [blame] | 1777 | // AddRHS - Implements: X + X --> X << 1 |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1778 | struct AddRHS { |
1779 | Value *RHS; | ||||
1780 | AddRHS(Value *rhs) : RHS(rhs) {} | ||||
1781 | bool shouldApply(Value *LHS) const { return LHS == RHS; } | ||||
1782 | Instruction *apply(BinaryOperator &Add) const { | ||||
Nick Lewycky | 27f6c13 | 2008-05-23 04:34:58 +0000 | [diff] [blame] | 1783 | return BinaryOperator::CreateShl(Add.getOperand(0), |
1784 | ConstantInt::get(Add.getType(), 1)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1785 | } |
1786 | }; | ||||
1787 | |||||
1788 | // AddMaskingAnd - Implements (A & C1)+(B & C2) --> (A & C1)|(B & C2) | ||||
1789 | // iff C1&C2 == 0 | ||||
1790 | struct AddMaskingAnd { | ||||
1791 | Constant *C2; | ||||
1792 | AddMaskingAnd(Constant *c) : C2(c) {} | ||||
1793 | bool shouldApply(Value *LHS) const { | ||||
1794 | ConstantInt *C1; | ||||
1795 | return match(LHS, m_And(m_Value(), m_ConstantInt(C1))) && | ||||
1796 | ConstantExpr::getAnd(C1, C2)->isNullValue(); | ||||
1797 | } | ||||
1798 | Instruction *apply(BinaryOperator &Add) const { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1799 | return BinaryOperator::CreateOr(Add.getOperand(0), Add.getOperand(1)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1800 | } |
1801 | }; | ||||
1802 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 1803 | } |
1804 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1805 | static Value *FoldOperationIntoSelectOperand(Instruction &I, Value *SO, |
1806 | InstCombiner *IC) { | ||||
1807 | if (CastInst *CI = dyn_cast<CastInst>(&I)) { | ||||
1808 | if (Constant *SOC = dyn_cast<Constant>(SO)) | ||||
1809 | return ConstantExpr::getCast(CI->getOpcode(), SOC, I.getType()); | ||||
1810 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1811 | return IC->InsertNewInstBefore(CastInst::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1812 | CI->getOpcode(), SO, I.getType(), SO->getName() + ".cast"), I); |
1813 | } | ||||
1814 | |||||
1815 | // Figure out if the constant is the left or the right argument. | ||||
1816 | bool ConstIsRHS = isa<Constant>(I.getOperand(1)); | ||||
1817 | Constant *ConstOperand = cast<Constant>(I.getOperand(ConstIsRHS)); | ||||
1818 | |||||
1819 | if (Constant *SOC = dyn_cast<Constant>(SO)) { | ||||
1820 | if (ConstIsRHS) | ||||
1821 | return ConstantExpr::get(I.getOpcode(), SOC, ConstOperand); | ||||
1822 | return ConstantExpr::get(I.getOpcode(), ConstOperand, SOC); | ||||
1823 | } | ||||
1824 | |||||
1825 | Value *Op0 = SO, *Op1 = ConstOperand; | ||||
1826 | if (!ConstIsRHS) | ||||
1827 | std::swap(Op0, Op1); | ||||
1828 | Instruction *New; | ||||
1829 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(&I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1830 | New = BinaryOperator::Create(BO->getOpcode(), Op0, Op1,SO->getName()+".op"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1831 | else if (CmpInst *CI = dyn_cast<CmpInst>(&I)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1832 | New = CmpInst::Create(CI->getOpcode(), CI->getPredicate(), Op0, Op1, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1833 | SO->getName()+".cmp"); |
1834 | else { | ||||
1835 | assert(0 && "Unknown binary instruction type!"); | ||||
1836 | abort(); | ||||
1837 | } | ||||
1838 | return IC->InsertNewInstBefore(New, I); | ||||
1839 | } | ||||
1840 | |||||
1841 | // FoldOpIntoSelect - Given an instruction with a select as one operand and a | ||||
1842 | // constant as the other operand, try to fold the binary operator into the | ||||
1843 | // select arguments. This also works for Cast instructions, which obviously do | ||||
1844 | // not have a second operand. | ||||
1845 | static Instruction *FoldOpIntoSelect(Instruction &Op, SelectInst *SI, | ||||
1846 | InstCombiner *IC) { | ||||
1847 | // Don't modify shared select instructions | ||||
1848 | if (!SI->hasOneUse()) return 0; | ||||
1849 | Value *TV = SI->getOperand(1); | ||||
1850 | Value *FV = SI->getOperand(2); | ||||
1851 | |||||
1852 | if (isa<Constant>(TV) || isa<Constant>(FV)) { | ||||
1853 | // Bool selects with constant operands can be folded to logical ops. | ||||
1854 | if (SI->getType() == Type::Int1Ty) return 0; | ||||
1855 | |||||
1856 | Value *SelectTrueVal = FoldOperationIntoSelectOperand(Op, TV, IC); | ||||
1857 | Value *SelectFalseVal = FoldOperationIntoSelectOperand(Op, FV, IC); | ||||
1858 | |||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 1859 | return SelectInst::Create(SI->getCondition(), SelectTrueVal, |
1860 | SelectFalseVal); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1861 | } |
1862 | return 0; | ||||
1863 | } | ||||
1864 | |||||
1865 | |||||
1866 | /// FoldOpIntoPhi - Given a binary operator or cast instruction which has a PHI | ||||
1867 | /// node as operand #0, see if we can fold the instruction into the PHI (which | ||||
1868 | /// is only possible if all operands to the PHI are constants). | ||||
1869 | Instruction *InstCombiner::FoldOpIntoPhi(Instruction &I) { | ||||
1870 | PHINode *PN = cast<PHINode>(I.getOperand(0)); | ||||
1871 | unsigned NumPHIValues = PN->getNumIncomingValues(); | ||||
1872 | if (!PN->hasOneUse() || NumPHIValues == 0) return 0; | ||||
1873 | |||||
1874 | // Check to see if all of the operands of the PHI are constants. If there is | ||||
1875 | // one non-constant value, remember the BB it is. If there is more than one | ||||
1876 | // or if *it* is a PHI, bail out. | ||||
1877 | BasicBlock *NonConstBB = 0; | ||||
1878 | for (unsigned i = 0; i != NumPHIValues; ++i) | ||||
1879 | if (!isa<Constant>(PN->getIncomingValue(i))) { | ||||
1880 | if (NonConstBB) return 0; // More than one non-const value. | ||||
1881 | if (isa<PHINode>(PN->getIncomingValue(i))) return 0; // Itself a phi. | ||||
1882 | NonConstBB = PN->getIncomingBlock(i); | ||||
1883 | |||||
1884 | // If the incoming non-constant value is in I's block, we have an infinite | ||||
1885 | // loop. | ||||
1886 | if (NonConstBB == I.getParent()) | ||||
1887 | return 0; | ||||
1888 | } | ||||
1889 | |||||
1890 | // If there is exactly one non-constant value, we can insert a copy of the | ||||
1891 | // operation in that block. However, if this is a critical edge, we would be | ||||
1892 | // inserting the computation one some other paths (e.g. inside a loop). Only | ||||
1893 | // do this if the pred block is unconditionally branching into the phi block. | ||||
1894 | if (NonConstBB) { | ||||
1895 | BranchInst *BI = dyn_cast<BranchInst>(NonConstBB->getTerminator()); | ||||
1896 | if (!BI || !BI->isUnconditional()) return 0; | ||||
1897 | } | ||||
1898 | |||||
1899 | // Okay, we can do the transformation: create the new PHI node. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 1900 | PHINode *NewPN = PHINode::Create(I.getType(), ""); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1901 | NewPN->reserveOperandSpace(PN->getNumOperands()/2); |
1902 | InsertNewInstBefore(NewPN, *PN); | ||||
1903 | NewPN->takeName(PN); | ||||
1904 | |||||
1905 | // Next, add all of the operands to the PHI. | ||||
1906 | if (I.getNumOperands() == 2) { | ||||
1907 | Constant *C = cast<Constant>(I.getOperand(1)); | ||||
1908 | for (unsigned i = 0; i != NumPHIValues; ++i) { | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 1909 | Value *InV = 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1910 | if (Constant *InC = dyn_cast<Constant>(PN->getIncomingValue(i))) { |
1911 | if (CmpInst *CI = dyn_cast<CmpInst>(&I)) | ||||
1912 | InV = ConstantExpr::getCompare(CI->getPredicate(), InC, C); | ||||
1913 | else | ||||
1914 | InV = ConstantExpr::get(I.getOpcode(), InC, C); | ||||
1915 | } else { | ||||
1916 | assert(PN->getIncomingBlock(i) == NonConstBB); | ||||
1917 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(&I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1918 | InV = BinaryOperator::Create(BO->getOpcode(), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1919 | PN->getIncomingValue(i), C, "phitmp", |
1920 | NonConstBB->getTerminator()); | ||||
1921 | else if (CmpInst *CI = dyn_cast<CmpInst>(&I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1922 | InV = CmpInst::Create(CI->getOpcode(), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1923 | CI->getPredicate(), |
1924 | PN->getIncomingValue(i), C, "phitmp", | ||||
1925 | NonConstBB->getTerminator()); | ||||
1926 | else | ||||
1927 | assert(0 && "Unknown binop!"); | ||||
1928 | |||||
1929 | AddToWorkList(cast<Instruction>(InV)); | ||||
1930 | } | ||||
1931 | NewPN->addIncoming(InV, PN->getIncomingBlock(i)); | ||||
1932 | } | ||||
1933 | } else { | ||||
1934 | CastInst *CI = cast<CastInst>(&I); | ||||
1935 | const Type *RetTy = CI->getType(); | ||||
1936 | for (unsigned i = 0; i != NumPHIValues; ++i) { | ||||
1937 | Value *InV; | ||||
1938 | if (Constant *InC = dyn_cast<Constant>(PN->getIncomingValue(i))) { | ||||
1939 | InV = ConstantExpr::getCast(CI->getOpcode(), InC, RetTy); | ||||
1940 | } else { | ||||
1941 | assert(PN->getIncomingBlock(i) == NonConstBB); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1942 | InV = CastInst::Create(CI->getOpcode(), PN->getIncomingValue(i), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1943 | I.getType(), "phitmp", |
1944 | NonConstBB->getTerminator()); | ||||
1945 | AddToWorkList(cast<Instruction>(InV)); | ||||
1946 | } | ||||
1947 | NewPN->addIncoming(InV, PN->getIncomingBlock(i)); | ||||
1948 | } | ||||
1949 | } | ||||
1950 | return ReplaceInstUsesWith(I, NewPN); | ||||
1951 | } | ||||
1952 | |||||
Chris Lattner | 5547616 | 2008-01-29 06:52:45 +0000 | [diff] [blame] | 1953 | |
Chris Lattner | 3554f97 | 2008-05-20 05:46:13 +0000 | [diff] [blame] | 1954 | /// WillNotOverflowSignedAdd - Return true if we can prove that: |
1955 | /// (sext (add LHS, RHS)) === (add (sext LHS), (sext RHS)) | ||||
1956 | /// This basically requires proving that the add in the original type would not | ||||
1957 | /// overflow to change the sign bit or have a carry out. | ||||
1958 | bool InstCombiner::WillNotOverflowSignedAdd(Value *LHS, Value *RHS) { | ||||
1959 | // There are different heuristics we can use for this. Here are some simple | ||||
1960 | // ones. | ||||
1961 | |||||
1962 | // Add has the property that adding any two 2's complement numbers can only | ||||
1963 | // have one carry bit which can change a sign. As such, if LHS and RHS each | ||||
1964 | // have at least two sign bits, we know that the addition of the two values will | ||||
1965 | // sign extend fine. | ||||
1966 | if (ComputeNumSignBits(LHS) > 1 && ComputeNumSignBits(RHS) > 1) | ||||
1967 | return true; | ||||
1968 | |||||
1969 | |||||
1970 | // If one of the operands only has one non-zero bit, and if the other operand | ||||
1971 | // has a known-zero bit in a more significant place than it (not including the | ||||
1972 | // sign bit) the ripple may go up to and fill the zero, but won't change the | ||||
1973 | // sign. For example, (X & ~4) + 1. | ||||
1974 | |||||
1975 | // TODO: Implement. | ||||
1976 | |||||
1977 | return false; | ||||
1978 | } | ||||
1979 | |||||
Chris Lattner | 5547616 | 2008-01-29 06:52:45 +0000 | [diff] [blame] | 1980 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1981 | Instruction *InstCombiner::visitAdd(BinaryOperator &I) { |
1982 | bool Changed = SimplifyCommutative(I); | ||||
1983 | Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); | ||||
1984 | |||||
1985 | if (Constant *RHSC = dyn_cast<Constant>(RHS)) { | ||||
1986 | // X + undef -> undef | ||||
1987 | if (isa<UndefValue>(RHS)) | ||||
1988 | return ReplaceInstUsesWith(I, RHS); | ||||
1989 | |||||
1990 | // X + 0 --> X | ||||
1991 | if (!I.getType()->isFPOrFPVector()) { // NOTE: -0 + +0 = +0. | ||||
1992 | if (RHSC->isNullValue()) | ||||
1993 | return ReplaceInstUsesWith(I, LHS); | ||||
1994 | } else if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHSC)) { | ||||
Dale Johannesen | 2fc2078 | 2007-09-14 22:26:36 +0000 | [diff] [blame] | 1995 | if (CFP->isExactlyValue(ConstantFP::getNegativeZero |
1996 | (I.getType())->getValueAPF())) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1997 | return ReplaceInstUsesWith(I, LHS); |
1998 | } | ||||
1999 | |||||
2000 | if (ConstantInt *CI = dyn_cast<ConstantInt>(RHSC)) { | ||||
2001 | // X + (signbit) --> X ^ signbit | ||||
2002 | const APInt& Val = CI->getValue(); | ||||
2003 | uint32_t BitWidth = Val.getBitWidth(); | ||||
2004 | if (Val == APInt::getSignBit(BitWidth)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2005 | return BinaryOperator::CreateXor(LHS, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2006 | |
2007 | // See if SimplifyDemandedBits can simplify this. This handles stuff like | ||||
2008 | // (X & 254)+1 -> (X&254)|1 | ||||
2009 | if (!isa<VectorType>(I.getType())) { | ||||
2010 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
2011 | if (SimplifyDemandedBits(&I, APInt::getAllOnesValue(BitWidth), | ||||
2012 | KnownZero, KnownOne)) | ||||
2013 | return &I; | ||||
2014 | } | ||||
2015 | } | ||||
2016 | |||||
2017 | if (isa<PHINode>(LHS)) | ||||
2018 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2019 | return NV; | ||||
2020 | |||||
2021 | ConstantInt *XorRHS = 0; | ||||
2022 | Value *XorLHS = 0; | ||||
2023 | if (isa<ConstantInt>(RHSC) && | ||||
2024 | match(LHS, m_Xor(m_Value(XorLHS), m_ConstantInt(XorRHS)))) { | ||||
2025 | uint32_t TySizeBits = I.getType()->getPrimitiveSizeInBits(); | ||||
2026 | const APInt& RHSVal = cast<ConstantInt>(RHSC)->getValue(); | ||||
2027 | |||||
2028 | uint32_t Size = TySizeBits / 2; | ||||
2029 | APInt C0080Val(APInt(TySizeBits, 1ULL).shl(Size - 1)); | ||||
2030 | APInt CFF80Val(-C0080Val); | ||||
2031 | do { | ||||
2032 | if (TySizeBits > Size) { | ||||
2033 | // If we have ADD(XOR(AND(X, 0xFF), 0x80), 0xF..F80), it's a sext. | ||||
2034 | // If we have ADD(XOR(AND(X, 0xFF), 0xF..F80), 0x80), it's a sext. | ||||
2035 | if ((RHSVal == CFF80Val && XorRHS->getValue() == C0080Val) || | ||||
2036 | (RHSVal == C0080Val && XorRHS->getValue() == CFF80Val)) { | ||||
2037 | // This is a sign extend if the top bits are known zero. | ||||
2038 | if (!MaskedValueIsZero(XorLHS, | ||||
2039 | APInt::getHighBitsSet(TySizeBits, TySizeBits - Size))) | ||||
2040 | Size = 0; // Not a sign ext, but can't be any others either. | ||||
2041 | break; | ||||
2042 | } | ||||
2043 | } | ||||
2044 | Size >>= 1; | ||||
2045 | C0080Val = APIntOps::lshr(C0080Val, Size); | ||||
2046 | CFF80Val = APIntOps::ashr(CFF80Val, Size); | ||||
2047 | } while (Size >= 1); | ||||
2048 | |||||
2049 | // FIXME: This shouldn't be necessary. When the backends can handle types | ||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 2050 | // with funny bit widths then this switch statement should be removed. It |
2051 | // is just here to get the size of the "middle" type back up to something | ||||
2052 | // that the back ends can handle. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2053 | const Type *MiddleType = 0; |
2054 | switch (Size) { | ||||
2055 | default: break; | ||||
2056 | case 32: MiddleType = Type::Int32Ty; break; | ||||
2057 | case 16: MiddleType = Type::Int16Ty; break; | ||||
2058 | case 8: MiddleType = Type::Int8Ty; break; | ||||
2059 | } | ||||
2060 | if (MiddleType) { | ||||
2061 | Instruction *NewTrunc = new TruncInst(XorLHS, MiddleType, "sext"); | ||||
2062 | InsertNewInstBefore(NewTrunc, I); | ||||
2063 | return new SExtInst(NewTrunc, I.getType(), I.getName()); | ||||
2064 | } | ||||
2065 | } | ||||
2066 | } | ||||
2067 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2068 | if (I.getType() == Type::Int1Ty) |
2069 | return BinaryOperator::CreateXor(LHS, RHS); | ||||
2070 | |||||
Nick Lewycky | 4d474cd | 2008-05-23 04:39:38 +0000 | [diff] [blame] | 2071 | // X + X --> X << 1 |
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2072 | if (I.getType()->isInteger()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2073 | if (Instruction *Result = AssociativeOpt(I, AddRHS(RHS))) return Result; |
2074 | |||||
2075 | if (Instruction *RHSI = dyn_cast<Instruction>(RHS)) { | ||||
2076 | if (RHSI->getOpcode() == Instruction::Sub) | ||||
2077 | if (LHS == RHSI->getOperand(1)) // A + (B - A) --> B | ||||
2078 | return ReplaceInstUsesWith(I, RHSI->getOperand(0)); | ||||
2079 | } | ||||
2080 | if (Instruction *LHSI = dyn_cast<Instruction>(LHS)) { | ||||
2081 | if (LHSI->getOpcode() == Instruction::Sub) | ||||
2082 | if (RHS == LHSI->getOperand(1)) // (B - A) + A --> B | ||||
2083 | return ReplaceInstUsesWith(I, LHSI->getOperand(0)); | ||||
2084 | } | ||||
2085 | } | ||||
2086 | |||||
2087 | // -A + B --> B - A | ||||
Chris Lattner | 53c9fbf | 2008-02-17 21:03:36 +0000 | [diff] [blame] | 2088 | // -A + -B --> -(A + B) |
2089 | if (Value *LHSV = dyn_castNegVal(LHS)) { | ||||
Chris Lattner | 322a919 | 2008-02-18 17:50:16 +0000 | [diff] [blame] | 2090 | if (LHS->getType()->isIntOrIntVector()) { |
2091 | if (Value *RHSV = dyn_castNegVal(RHS)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2092 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSV, RHSV, "sum"); |
Chris Lattner | 322a919 | 2008-02-18 17:50:16 +0000 | [diff] [blame] | 2093 | InsertNewInstBefore(NewAdd, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2094 | return BinaryOperator::CreateNeg(NewAdd); |
Chris Lattner | 322a919 | 2008-02-18 17:50:16 +0000 | [diff] [blame] | 2095 | } |
Chris Lattner | 53c9fbf | 2008-02-17 21:03:36 +0000 | [diff] [blame] | 2096 | } |
2097 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2098 | return BinaryOperator::CreateSub(RHS, LHSV); |
Chris Lattner | 53c9fbf | 2008-02-17 21:03:36 +0000 | [diff] [blame] | 2099 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2100 | |
2101 | // A + -B --> A - B | ||||
2102 | if (!isa<Constant>(RHS)) | ||||
2103 | if (Value *V = dyn_castNegVal(RHS)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2104 | return BinaryOperator::CreateSub(LHS, V); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2105 | |
2106 | |||||
2107 | ConstantInt *C2; | ||||
2108 | if (Value *X = dyn_castFoldableMul(LHS, C2)) { | ||||
2109 | if (X == RHS) // X*C + X --> X * (C+1) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2110 | return BinaryOperator::CreateMul(RHS, AddOne(C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2111 | |
2112 | // X*C1 + X*C2 --> X * (C1+C2) | ||||
2113 | ConstantInt *C1; | ||||
2114 | if (X == dyn_castFoldableMul(RHS, C1)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2115 | return BinaryOperator::CreateMul(X, Add(C1, C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2116 | } |
2117 | |||||
2118 | // X + X*C --> X * (C+1) | ||||
2119 | if (dyn_castFoldableMul(RHS, C2) == LHS) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2120 | return BinaryOperator::CreateMul(LHS, AddOne(C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2121 | |
2122 | // X + ~X --> -1 since ~X = -X-1 | ||||
2123 | if (dyn_castNotVal(LHS) == RHS || dyn_castNotVal(RHS) == LHS) | ||||
2124 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
2125 | |||||
2126 | |||||
2127 | // (A & C1)+(B & C2) --> (A & C1)|(B & C2) iff C1&C2 == 0 | ||||
2128 | if (match(RHS, m_And(m_Value(), m_ConstantInt(C2)))) | ||||
2129 | if (Instruction *R = AssociativeOpt(I, AddMaskingAnd(C2))) | ||||
2130 | return R; | ||||
Chris Lattner | c1575ce | 2008-05-19 20:01:56 +0000 | [diff] [blame] | 2131 | |
2132 | // A+B --> A|B iff A and B have no bits set in common. | ||||
2133 | if (const IntegerType *IT = dyn_cast<IntegerType>(I.getType())) { | ||||
2134 | APInt Mask = APInt::getAllOnesValue(IT->getBitWidth()); | ||||
2135 | APInt LHSKnownOne(IT->getBitWidth(), 0); | ||||
2136 | APInt LHSKnownZero(IT->getBitWidth(), 0); | ||||
2137 | ComputeMaskedBits(LHS, Mask, LHSKnownZero, LHSKnownOne); | ||||
2138 | if (LHSKnownZero != 0) { | ||||
2139 | APInt RHSKnownOne(IT->getBitWidth(), 0); | ||||
2140 | APInt RHSKnownZero(IT->getBitWidth(), 0); | ||||
2141 | ComputeMaskedBits(RHS, Mask, RHSKnownZero, RHSKnownOne); | ||||
2142 | |||||
2143 | // No bits in common -> bitwise or. | ||||
Chris Lattner | 130443c | 2008-05-19 20:03:53 +0000 | [diff] [blame] | 2144 | if ((LHSKnownZero|RHSKnownZero).isAllOnesValue()) |
Chris Lattner | c1575ce | 2008-05-19 20:01:56 +0000 | [diff] [blame] | 2145 | return BinaryOperator::CreateOr(LHS, RHS); |
Chris Lattner | c1575ce | 2008-05-19 20:01:56 +0000 | [diff] [blame] | 2146 | } |
2147 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2148 | |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2149 | // W*X + Y*Z --> W * (X+Z) iff W == Y |
Nick Lewycky | 5d03b51 | 2008-02-03 08:19:11 +0000 | [diff] [blame] | 2150 | if (I.getType()->isIntOrIntVector()) { |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2151 | Value *W, *X, *Y, *Z; |
2152 | if (match(LHS, m_Mul(m_Value(W), m_Value(X))) && | ||||
2153 | match(RHS, m_Mul(m_Value(Y), m_Value(Z)))) { | ||||
2154 | if (W != Y) { | ||||
2155 | if (W == Z) { | ||||
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 2156 | std::swap(Y, Z); |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2157 | } else if (Y == X) { |
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 2158 | std::swap(W, X); |
2159 | } else if (X == Z) { | ||||
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2160 | std::swap(Y, Z); |
2161 | std::swap(W, X); | ||||
2162 | } | ||||
2163 | } | ||||
2164 | |||||
2165 | if (W == Y) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2166 | Value *NewAdd = InsertNewInstBefore(BinaryOperator::CreateAdd(X, Z, |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2167 | LHS->getName()), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2168 | return BinaryOperator::CreateMul(W, NewAdd); |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2169 | } |
2170 | } | ||||
2171 | } | ||||
2172 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2173 | if (ConstantInt *CRHS = dyn_cast<ConstantInt>(RHS)) { |
2174 | Value *X = 0; | ||||
2175 | 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] | 2176 | return BinaryOperator::CreateSub(SubOne(CRHS), X); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2177 | |
2178 | // (X & FF00) + xx00 -> (X+xx00) & FF00 | ||||
2179 | if (LHS->hasOneUse() && match(LHS, m_And(m_Value(X), m_ConstantInt(C2)))) { | ||||
2180 | Constant *Anded = And(CRHS, C2); | ||||
2181 | if (Anded == CRHS) { | ||||
2182 | // See if all bits from the first bit set in the Add RHS up are included | ||||
2183 | // in the mask. First, get the rightmost bit. | ||||
2184 | const APInt& AddRHSV = CRHS->getValue(); | ||||
2185 | |||||
2186 | // Form a mask of all bits from the lowest bit added through the top. | ||||
2187 | APInt AddRHSHighBits(~((AddRHSV & -AddRHSV)-1)); | ||||
2188 | |||||
2189 | // See if the and mask includes all of these bits. | ||||
2190 | APInt AddRHSHighBitsAnd(AddRHSHighBits & C2->getValue()); | ||||
2191 | |||||
2192 | if (AddRHSHighBits == AddRHSHighBitsAnd) { | ||||
2193 | // Okay, the xform is safe. Insert the new add pronto. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2194 | Value *NewAdd = InsertNewInstBefore(BinaryOperator::CreateAdd(X, CRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2195 | LHS->getName()), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2196 | return BinaryOperator::CreateAnd(NewAdd, C2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2197 | } |
2198 | } | ||||
2199 | } | ||||
2200 | |||||
2201 | // Try to fold constant add into select arguments. | ||||
2202 | if (SelectInst *SI = dyn_cast<SelectInst>(LHS)) | ||||
2203 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2204 | return R; | ||||
2205 | } | ||||
2206 | |||||
2207 | // add (cast *A to intptrtype) B -> | ||||
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2208 | // cast (GEP (cast *A to sbyte*) B) --> intptrtype |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2209 | { |
2210 | CastInst *CI = dyn_cast<CastInst>(LHS); | ||||
2211 | Value *Other = RHS; | ||||
2212 | if (!CI) { | ||||
2213 | CI = dyn_cast<CastInst>(RHS); | ||||
2214 | Other = LHS; | ||||
2215 | } | ||||
2216 | if (CI && CI->getType()->isSized() && | ||||
2217 | (CI->getType()->getPrimitiveSizeInBits() == | ||||
2218 | TD->getIntPtrType()->getPrimitiveSizeInBits()) | ||||
2219 | && isa<PointerType>(CI->getOperand(0)->getType())) { | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 2220 | unsigned AS = |
2221 | cast<PointerType>(CI->getOperand(0)->getType())->getAddressSpace(); | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 2222 | Value *I2 = InsertBitCastBefore(CI->getOperand(0), |
2223 | PointerType::get(Type::Int8Ty, AS), I); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2224 | I2 = InsertNewInstBefore(GetElementPtrInst::Create(I2, Other, "ctg2"), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2225 | return new PtrToIntInst(I2, CI->getType()); |
2226 | } | ||||
2227 | } | ||||
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2228 | |
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2229 | // add (select X 0 (sub n A)) A --> select X A n |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2230 | { |
2231 | SelectInst *SI = dyn_cast<SelectInst>(LHS); | ||||
2232 | Value *Other = RHS; | ||||
2233 | if (!SI) { | ||||
2234 | SI = dyn_cast<SelectInst>(RHS); | ||||
2235 | Other = LHS; | ||||
2236 | } | ||||
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2237 | if (SI && SI->hasOneUse()) { |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2238 | Value *TV = SI->getTrueValue(); |
2239 | Value *FV = SI->getFalseValue(); | ||||
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2240 | Value *A, *N; |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2241 | |
2242 | // Can we fold the add into the argument of the select? | ||||
2243 | // We check both true and false select arguments for a matching subtract. | ||||
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2244 | if (match(FV, m_Zero()) && match(TV, m_Sub(m_Value(N), m_Value(A))) && |
2245 | A == Other) // Fold the add into the true select value. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2246 | return SelectInst::Create(SI->getCondition(), N, A); |
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2247 | if (match(TV, m_Zero()) && match(FV, m_Sub(m_Value(N), m_Value(A))) && |
2248 | A == Other) // Fold the add into the false select value. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2249 | return SelectInst::Create(SI->getCondition(), A, N); |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2250 | } |
2251 | } | ||||
Chris Lattner | 5547616 | 2008-01-29 06:52:45 +0000 | [diff] [blame] | 2252 | |
2253 | // Check for X+0.0. Simplify it to X if we know X is not -0.0. | ||||
2254 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHS)) | ||||
2255 | if (CFP->getValueAPF().isPosZero() && CannotBeNegativeZero(LHS)) | ||||
2256 | return ReplaceInstUsesWith(I, LHS); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2257 | |
Chris Lattner | 3554f97 | 2008-05-20 05:46:13 +0000 | [diff] [blame] | 2258 | // Check for (add (sext x), y), see if we can merge this into an |
2259 | // integer add followed by a sext. | ||||
2260 | if (SExtInst *LHSConv = dyn_cast<SExtInst>(LHS)) { | ||||
2261 | // (add (sext x), cst) --> (sext (add x, cst')) | ||||
2262 | if (ConstantInt *RHSC = dyn_cast<ConstantInt>(RHS)) { | ||||
2263 | Constant *CI = | ||||
2264 | ConstantExpr::getTrunc(RHSC, LHSConv->getOperand(0)->getType()); | ||||
2265 | if (LHSConv->hasOneUse() && | ||||
2266 | ConstantExpr::getSExt(CI, I.getType()) == RHSC && | ||||
2267 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), CI)) { | ||||
2268 | // Insert the new, smaller add. | ||||
2269 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2270 | CI, "addconv"); | ||||
2271 | InsertNewInstBefore(NewAdd, I); | ||||
2272 | return new SExtInst(NewAdd, I.getType()); | ||||
2273 | } | ||||
2274 | } | ||||
2275 | |||||
2276 | // (add (sext x), (sext y)) --> (sext (add int x, y)) | ||||
2277 | if (SExtInst *RHSConv = dyn_cast<SExtInst>(RHS)) { | ||||
2278 | // Only do this if x/y have the same type, if at last one of them has a | ||||
2279 | // single use (so we don't increase the number of sexts), and if the | ||||
2280 | // integer add will not overflow. | ||||
2281 | if (LHSConv->getOperand(0)->getType()==RHSConv->getOperand(0)->getType()&& | ||||
2282 | (LHSConv->hasOneUse() || RHSConv->hasOneUse()) && | ||||
2283 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), | ||||
2284 | RHSConv->getOperand(0))) { | ||||
2285 | // Insert the new integer add. | ||||
2286 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2287 | RHSConv->getOperand(0), | ||||
2288 | "addconv"); | ||||
2289 | InsertNewInstBefore(NewAdd, I); | ||||
2290 | return new SExtInst(NewAdd, I.getType()); | ||||
2291 | } | ||||
2292 | } | ||||
2293 | } | ||||
2294 | |||||
2295 | // Check for (add double (sitofp x), y), see if we can merge this into an | ||||
2296 | // integer add followed by a promotion. | ||||
2297 | if (SIToFPInst *LHSConv = dyn_cast<SIToFPInst>(LHS)) { | ||||
2298 | // (add double (sitofp x), fpcst) --> (sitofp (add int x, intcst)) | ||||
2299 | // ... if the constant fits in the integer value. This is useful for things | ||||
2300 | // like (double)(x & 1234) + 4.0 -> (double)((X & 1234)+4) which no longer | ||||
2301 | // requires a constant pool load, and generally allows the add to be better | ||||
2302 | // instcombined. | ||||
2303 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHS)) { | ||||
2304 | Constant *CI = | ||||
2305 | ConstantExpr::getFPToSI(CFP, LHSConv->getOperand(0)->getType()); | ||||
2306 | if (LHSConv->hasOneUse() && | ||||
2307 | ConstantExpr::getSIToFP(CI, I.getType()) == CFP && | ||||
2308 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), CI)) { | ||||
2309 | // Insert the new integer add. | ||||
2310 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2311 | CI, "addconv"); | ||||
2312 | InsertNewInstBefore(NewAdd, I); | ||||
2313 | return new SIToFPInst(NewAdd, I.getType()); | ||||
2314 | } | ||||
2315 | } | ||||
2316 | |||||
2317 | // (add double (sitofp x), (sitofp y)) --> (sitofp (add int x, y)) | ||||
2318 | if (SIToFPInst *RHSConv = dyn_cast<SIToFPInst>(RHS)) { | ||||
2319 | // Only do this if x/y have the same type, if at last one of them has a | ||||
2320 | // single use (so we don't increase the number of int->fp conversions), | ||||
2321 | // and if the integer add will not overflow. | ||||
2322 | if (LHSConv->getOperand(0)->getType()==RHSConv->getOperand(0)->getType()&& | ||||
2323 | (LHSConv->hasOneUse() || RHSConv->hasOneUse()) && | ||||
2324 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), | ||||
2325 | RHSConv->getOperand(0))) { | ||||
2326 | // Insert the new integer add. | ||||
2327 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2328 | RHSConv->getOperand(0), | ||||
2329 | "addconv"); | ||||
2330 | InsertNewInstBefore(NewAdd, I); | ||||
2331 | return new SIToFPInst(NewAdd, I.getType()); | ||||
2332 | } | ||||
2333 | } | ||||
2334 | } | ||||
2335 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2336 | return Changed ? &I : 0; |
2337 | } | ||||
2338 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2339 | Instruction *InstCombiner::visitSub(BinaryOperator &I) { |
2340 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2341 | |||||
Chris Lattner | 27fbef4 | 2008-07-17 06:07:20 +0000 | [diff] [blame] | 2342 | if (Op0 == Op1 && // sub X, X -> 0 |
2343 | !I.getType()->isFPOrFPVector()) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2344 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
2345 | |||||
2346 | // If this is a 'B = x-(-A)', change to B = x+A... | ||||
2347 | if (Value *V = dyn_castNegVal(Op1)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2348 | return BinaryOperator::CreateAdd(Op0, V); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2349 | |
2350 | if (isa<UndefValue>(Op0)) | ||||
2351 | return ReplaceInstUsesWith(I, Op0); // undef - X -> undef | ||||
2352 | if (isa<UndefValue>(Op1)) | ||||
2353 | return ReplaceInstUsesWith(I, Op1); // X - undef -> undef | ||||
2354 | |||||
2355 | if (ConstantInt *C = dyn_cast<ConstantInt>(Op0)) { | ||||
2356 | // Replace (-1 - A) with (~A)... | ||||
2357 | if (C->isAllOnesValue()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2358 | return BinaryOperator::CreateNot(Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2359 | |
2360 | // C - ~X == X + (1+C) | ||||
2361 | Value *X = 0; | ||||
2362 | if (match(Op1, m_Not(m_Value(X)))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2363 | return BinaryOperator::CreateAdd(X, AddOne(C)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2364 | |
2365 | // -(X >>u 31) -> (X >>s 31) | ||||
2366 | // -(X >>s 31) -> (X >>u 31) | ||||
2367 | if (C->isZero()) { | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2368 | if (BinaryOperator *SI = dyn_cast<BinaryOperator>(Op1)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2369 | if (SI->getOpcode() == Instruction::LShr) { |
2370 | if (ConstantInt *CU = dyn_cast<ConstantInt>(SI->getOperand(1))) { | ||||
2371 | // Check to see if we are shifting out everything but the sign bit. | ||||
2372 | if (CU->getLimitedValue(SI->getType()->getPrimitiveSizeInBits()) == | ||||
2373 | SI->getType()->getPrimitiveSizeInBits()-1) { | ||||
2374 | // Ok, the transformation is safe. Insert AShr. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2375 | return BinaryOperator::Create(Instruction::AShr, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2376 | SI->getOperand(0), CU, SI->getName()); |
2377 | } | ||||
2378 | } | ||||
2379 | } | ||||
2380 | else if (SI->getOpcode() == Instruction::AShr) { | ||||
2381 | if (ConstantInt *CU = dyn_cast<ConstantInt>(SI->getOperand(1))) { | ||||
2382 | // Check to see if we are shifting out everything but the sign bit. | ||||
2383 | if (CU->getLimitedValue(SI->getType()->getPrimitiveSizeInBits()) == | ||||
2384 | SI->getType()->getPrimitiveSizeInBits()-1) { | ||||
2385 | // Ok, the transformation is safe. Insert LShr. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2386 | return BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2387 | SI->getOperand(0), CU, SI->getName()); |
2388 | } | ||||
2389 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2390 | } |
2391 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2392 | } |
2393 | |||||
2394 | // Try to fold constant sub into select arguments. | ||||
2395 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) | ||||
2396 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2397 | return R; | ||||
2398 | |||||
2399 | if (isa<PHINode>(Op0)) | ||||
2400 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2401 | return NV; | ||||
2402 | } | ||||
2403 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2404 | if (I.getType() == Type::Int1Ty) |
2405 | return BinaryOperator::CreateXor(Op0, Op1); | ||||
2406 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2407 | if (BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1)) { |
2408 | if (Op1I->getOpcode() == Instruction::Add && | ||||
2409 | !Op0->getType()->isFPOrFPVector()) { | ||||
2410 | if (Op1I->getOperand(0) == Op0) // X-(X+Y) == -Y | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2411 | return BinaryOperator::CreateNeg(Op1I->getOperand(1), I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2412 | else if (Op1I->getOperand(1) == Op0) // X-(Y+X) == -Y |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2413 | return BinaryOperator::CreateNeg(Op1I->getOperand(0), I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2414 | else if (ConstantInt *CI1 = dyn_cast<ConstantInt>(I.getOperand(0))) { |
2415 | if (ConstantInt *CI2 = dyn_cast<ConstantInt>(Op1I->getOperand(1))) | ||||
2416 | // C1-(X+C2) --> (C1-C2)-X | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2417 | return BinaryOperator::CreateSub(Subtract(CI1, CI2), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2418 | Op1I->getOperand(0)); |
2419 | } | ||||
2420 | } | ||||
2421 | |||||
2422 | if (Op1I->hasOneUse()) { | ||||
2423 | // Replace (x - (y - z)) with (x + (z - y)) if the (y - z) subexpression | ||||
2424 | // is not used by anyone else... | ||||
2425 | // | ||||
2426 | if (Op1I->getOpcode() == Instruction::Sub && | ||||
2427 | !Op1I->getType()->isFPOrFPVector()) { | ||||
2428 | // Swap the two operands of the subexpr... | ||||
2429 | Value *IIOp0 = Op1I->getOperand(0), *IIOp1 = Op1I->getOperand(1); | ||||
2430 | Op1I->setOperand(0, IIOp1); | ||||
2431 | Op1I->setOperand(1, IIOp0); | ||||
2432 | |||||
2433 | // Create the new top level add instruction... | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2434 | return BinaryOperator::CreateAdd(Op0, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2435 | } |
2436 | |||||
2437 | // Replace (A - (A & B)) with (A & ~B) if this is the only use of (A&B)... | ||||
2438 | // | ||||
2439 | if (Op1I->getOpcode() == Instruction::And && | ||||
2440 | (Op1I->getOperand(0) == Op0 || Op1I->getOperand(1) == Op0)) { | ||||
2441 | Value *OtherOp = Op1I->getOperand(Op1I->getOperand(0) == Op0); | ||||
2442 | |||||
2443 | Value *NewNot = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2444 | InsertNewInstBefore(BinaryOperator::CreateNot(OtherOp, "B.not"), I); |
2445 | return BinaryOperator::CreateAnd(Op0, NewNot); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2446 | } |
2447 | |||||
2448 | // 0 - (X sdiv C) -> (X sdiv -C) | ||||
2449 | if (Op1I->getOpcode() == Instruction::SDiv) | ||||
2450 | if (ConstantInt *CSI = dyn_cast<ConstantInt>(Op0)) | ||||
2451 | if (CSI->isZero()) | ||||
2452 | if (Constant *DivRHS = dyn_cast<Constant>(Op1I->getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2453 | return BinaryOperator::CreateSDiv(Op1I->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2454 | ConstantExpr::getNeg(DivRHS)); |
2455 | |||||
2456 | // X - X*C --> X * (1-C) | ||||
2457 | ConstantInt *C2 = 0; | ||||
2458 | if (dyn_castFoldableMul(Op1I, C2) == Op0) { | ||||
2459 | Constant *CP1 = Subtract(ConstantInt::get(I.getType(), 1), C2); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2460 | return BinaryOperator::CreateMul(Op0, CP1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2461 | } |
Dan Gohman | da33874 | 2007-09-17 17:31:57 +0000 | [diff] [blame] | 2462 | |
2463 | // X - ((X / Y) * Y) --> X % Y | ||||
2464 | if (Op1I->getOpcode() == Instruction::Mul) | ||||
2465 | if (Instruction *I = dyn_cast<Instruction>(Op1I->getOperand(0))) | ||||
2466 | if (Op0 == I->getOperand(0) && | ||||
2467 | Op1I->getOperand(1) == I->getOperand(1)) { | ||||
2468 | if (I->getOpcode() == Instruction::SDiv) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2469 | return BinaryOperator::CreateSRem(Op0, Op1I->getOperand(1)); |
Dan Gohman | da33874 | 2007-09-17 17:31:57 +0000 | [diff] [blame] | 2470 | if (I->getOpcode() == Instruction::UDiv) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2471 | return BinaryOperator::CreateURem(Op0, Op1I->getOperand(1)); |
Dan Gohman | da33874 | 2007-09-17 17:31:57 +0000 | [diff] [blame] | 2472 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2473 | } |
2474 | } | ||||
2475 | |||||
2476 | if (!Op0->getType()->isFPOrFPVector()) | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2477 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2478 | if (Op0I->getOpcode() == Instruction::Add) { |
2479 | if (Op0I->getOperand(0) == Op1) // (Y+X)-Y == X | ||||
2480 | return ReplaceInstUsesWith(I, Op0I->getOperand(1)); | ||||
2481 | else if (Op0I->getOperand(1) == Op1) // (X+Y)-Y == X | ||||
2482 | return ReplaceInstUsesWith(I, Op0I->getOperand(0)); | ||||
2483 | } else if (Op0I->getOpcode() == Instruction::Sub) { | ||||
2484 | if (Op0I->getOperand(0) == Op1) // (X-Y)-X == -Y | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2485 | return BinaryOperator::CreateNeg(Op0I->getOperand(1), I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2486 | } |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2487 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2488 | |
2489 | ConstantInt *C1; | ||||
2490 | if (Value *X = dyn_castFoldableMul(Op0, C1)) { | ||||
2491 | if (X == Op1) // X*C - X --> X * (C-1) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2492 | return BinaryOperator::CreateMul(Op1, SubOne(C1)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2493 | |
2494 | ConstantInt *C2; // X*C1 - X*C2 -> X * (C1-C2) | ||||
2495 | if (X == dyn_castFoldableMul(Op1, C2)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2496 | return BinaryOperator::CreateMul(X, Subtract(C1, C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2497 | } |
2498 | return 0; | ||||
2499 | } | ||||
2500 | |||||
2501 | /// isSignBitCheck - Given an exploded icmp instruction, return true if the | ||||
2502 | /// comparison only checks the sign bit. If it only checks the sign bit, set | ||||
2503 | /// TrueIfSigned if the result of the comparison is true when the input value is | ||||
2504 | /// signed. | ||||
2505 | static bool isSignBitCheck(ICmpInst::Predicate pred, ConstantInt *RHS, | ||||
2506 | bool &TrueIfSigned) { | ||||
2507 | switch (pred) { | ||||
2508 | case ICmpInst::ICMP_SLT: // True if LHS s< 0 | ||||
2509 | TrueIfSigned = true; | ||||
2510 | return RHS->isZero(); | ||||
2511 | case ICmpInst::ICMP_SLE: // True if LHS s<= RHS and RHS == -1 | ||||
2512 | TrueIfSigned = true; | ||||
2513 | return RHS->isAllOnesValue(); | ||||
2514 | case ICmpInst::ICMP_SGT: // True if LHS s> -1 | ||||
2515 | TrueIfSigned = false; | ||||
2516 | return RHS->isAllOnesValue(); | ||||
2517 | case ICmpInst::ICMP_UGT: | ||||
2518 | // True if LHS u> RHS and RHS == high-bit-mask - 1 | ||||
2519 | TrueIfSigned = true; | ||||
2520 | return RHS->getValue() == | ||||
2521 | APInt::getSignedMaxValue(RHS->getType()->getPrimitiveSizeInBits()); | ||||
2522 | case ICmpInst::ICMP_UGE: | ||||
2523 | // True if LHS u>= RHS and RHS == high-bit-mask (2^7, 2^15, 2^31, etc) | ||||
2524 | TrueIfSigned = true; | ||||
Chris Lattner | 60813c2 | 2008-06-02 01:29:46 +0000 | [diff] [blame] | 2525 | return RHS->getValue().isSignBit(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2526 | default: |
2527 | return false; | ||||
2528 | } | ||||
2529 | } | ||||
2530 | |||||
2531 | Instruction *InstCombiner::visitMul(BinaryOperator &I) { | ||||
2532 | bool Changed = SimplifyCommutative(I); | ||||
2533 | Value *Op0 = I.getOperand(0); | ||||
2534 | |||||
2535 | if (isa<UndefValue>(I.getOperand(1))) // undef * X -> 0 | ||||
2536 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2537 | |||||
2538 | // Simplify mul instructions with a constant RHS... | ||||
2539 | if (Constant *Op1 = dyn_cast<Constant>(I.getOperand(1))) { | ||||
2540 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) { | ||||
2541 | |||||
2542 | // ((X << C1)*C2) == (X * (C2 << C1)) | ||||
2543 | if (BinaryOperator *SI = dyn_cast<BinaryOperator>(Op0)) | ||||
2544 | if (SI->getOpcode() == Instruction::Shl) | ||||
2545 | if (Constant *ShOp = dyn_cast<Constant>(SI->getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2546 | return BinaryOperator::CreateMul(SI->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2547 | ConstantExpr::getShl(CI, ShOp)); |
2548 | |||||
2549 | if (CI->isZero()) | ||||
2550 | return ReplaceInstUsesWith(I, Op1); // X * 0 == 0 | ||||
2551 | if (CI->equalsInt(1)) // X * 1 == X | ||||
2552 | return ReplaceInstUsesWith(I, Op0); | ||||
2553 | if (CI->isAllOnesValue()) // X * -1 == 0 - X | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2554 | return BinaryOperator::CreateNeg(Op0, I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2555 | |
2556 | const APInt& Val = cast<ConstantInt>(CI)->getValue(); | ||||
2557 | if (Val.isPowerOf2()) { // Replace X*(2^C) with X << C | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2558 | return BinaryOperator::CreateShl(Op0, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2559 | ConstantInt::get(Op0->getType(), Val.logBase2())); |
2560 | } | ||||
2561 | } else if (ConstantFP *Op1F = dyn_cast<ConstantFP>(Op1)) { | ||||
2562 | if (Op1F->isNullValue()) | ||||
2563 | return ReplaceInstUsesWith(I, Op1); | ||||
2564 | |||||
2565 | // "In IEEE floating point, x*1 is not equivalent to x for nans. However, | ||||
2566 | // 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] | 2567 | if (Op1F->isExactlyValue(1.0)) |
2568 | return ReplaceInstUsesWith(I, Op0); // Eliminate 'mul double %X, 1.0' | ||||
2569 | } else if (isa<VectorType>(Op1->getType())) { | ||||
2570 | if (isa<ConstantAggregateZero>(Op1)) | ||||
2571 | return ReplaceInstUsesWith(I, Op1); | ||||
2572 | |||||
2573 | // As above, vector X*splat(1.0) -> X in all defined cases. | ||||
2574 | if (ConstantVector *Op1V = dyn_cast<ConstantVector>(Op1)) | ||||
2575 | if (ConstantFP *F = dyn_cast_or_null<ConstantFP>(Op1V->getSplatValue())) | ||||
2576 | if (F->isExactlyValue(1.0)) | ||||
2577 | return ReplaceInstUsesWith(I, Op0); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2578 | } |
2579 | |||||
2580 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) | ||||
2581 | if (Op0I->getOpcode() == Instruction::Add && Op0I->hasOneUse() && | ||||
Chris Lattner | 5819408 | 2008-05-18 04:11:26 +0000 | [diff] [blame] | 2582 | isa<ConstantInt>(Op0I->getOperand(1)) && isa<ConstantInt>(Op1)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2583 | // Canonicalize (X+C1)*C2 -> X*C2+C1*C2. |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2584 | Instruction *Add = BinaryOperator::CreateMul(Op0I->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2585 | Op1, "tmp"); |
2586 | InsertNewInstBefore(Add, I); | ||||
2587 | Value *C1C2 = ConstantExpr::getMul(Op1, | ||||
2588 | cast<Constant>(Op0I->getOperand(1))); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2589 | return BinaryOperator::CreateAdd(Add, C1C2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2590 | |
2591 | } | ||||
2592 | |||||
2593 | // Try to fold constant mul into select arguments. | ||||
2594 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
2595 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2596 | return R; | ||||
2597 | |||||
2598 | if (isa<PHINode>(Op0)) | ||||
2599 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2600 | return NV; | ||||
2601 | } | ||||
2602 | |||||
2603 | if (Value *Op0v = dyn_castNegVal(Op0)) // -X * -Y = X*Y | ||||
2604 | if (Value *Op1v = dyn_castNegVal(I.getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2605 | return BinaryOperator::CreateMul(Op0v, Op1v); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2606 | |
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2607 | if (I.getType() == Type::Int1Ty) |
2608 | return BinaryOperator::CreateAnd(Op0, I.getOperand(1)); | ||||
2609 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2610 | // If one of the operands of the multiply is a cast from a boolean value, then |
2611 | // we know the bool is either zero or one, so this is a 'masking' multiply. | ||||
2612 | // See if we can simplify things based on how the boolean was originally | ||||
2613 | // formed. | ||||
2614 | CastInst *BoolCast = 0; | ||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2615 | if (ZExtInst *CI = dyn_cast<ZExtInst>(Op0)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2616 | if (CI->getOperand(0)->getType() == Type::Int1Ty) |
2617 | BoolCast = CI; | ||||
2618 | if (!BoolCast) | ||||
2619 | if (ZExtInst *CI = dyn_cast<ZExtInst>(I.getOperand(1))) | ||||
2620 | if (CI->getOperand(0)->getType() == Type::Int1Ty) | ||||
2621 | BoolCast = CI; | ||||
2622 | if (BoolCast) { | ||||
2623 | if (ICmpInst *SCI = dyn_cast<ICmpInst>(BoolCast->getOperand(0))) { | ||||
2624 | Value *SCIOp0 = SCI->getOperand(0), *SCIOp1 = SCI->getOperand(1); | ||||
2625 | const Type *SCOpTy = SCIOp0->getType(); | ||||
2626 | bool TIS = false; | ||||
2627 | |||||
2628 | // If the icmp is true iff the sign bit of X is set, then convert this | ||||
2629 | // multiply into a shift/and combination. | ||||
2630 | if (isa<ConstantInt>(SCIOp1) && | ||||
2631 | isSignBitCheck(SCI->getPredicate(), cast<ConstantInt>(SCIOp1), TIS) && | ||||
2632 | TIS) { | ||||
2633 | // Shift the X value right to turn it into "all signbits". | ||||
2634 | Constant *Amt = ConstantInt::get(SCIOp0->getType(), | ||||
2635 | SCOpTy->getPrimitiveSizeInBits()-1); | ||||
2636 | Value *V = | ||||
2637 | InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2638 | BinaryOperator::Create(Instruction::AShr, SCIOp0, Amt, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2639 | BoolCast->getOperand(0)->getName()+ |
2640 | ".mask"), I); | ||||
2641 | |||||
2642 | // If the multiply type is not the same as the source type, sign extend | ||||
2643 | // or truncate to the multiply type. | ||||
2644 | if (I.getType() != V->getType()) { | ||||
2645 | uint32_t SrcBits = V->getType()->getPrimitiveSizeInBits(); | ||||
2646 | uint32_t DstBits = I.getType()->getPrimitiveSizeInBits(); | ||||
2647 | Instruction::CastOps opcode = | ||||
2648 | (SrcBits == DstBits ? Instruction::BitCast : | ||||
2649 | (SrcBits < DstBits ? Instruction::SExt : Instruction::Trunc)); | ||||
2650 | V = InsertCastBefore(opcode, V, I.getType(), I); | ||||
2651 | } | ||||
2652 | |||||
2653 | Value *OtherOp = Op0 == BoolCast ? I.getOperand(1) : Op0; | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2654 | return BinaryOperator::CreateAnd(V, OtherOp); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2655 | } |
2656 | } | ||||
2657 | } | ||||
2658 | |||||
2659 | return Changed ? &I : 0; | ||||
2660 | } | ||||
2661 | |||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 2662 | /// SimplifyDivRemOfSelect - Try to fold a divide or remainder of a select |
2663 | /// instruction. | ||||
2664 | bool InstCombiner::SimplifyDivRemOfSelect(BinaryOperator &I) { | ||||
2665 | SelectInst *SI = cast<SelectInst>(I.getOperand(1)); | ||||
2666 | |||||
2667 | // div/rem X, (Cond ? 0 : Y) -> div/rem X, Y | ||||
2668 | int NonNullOperand = -1; | ||||
2669 | if (Constant *ST = dyn_cast<Constant>(SI->getOperand(1))) | ||||
2670 | if (ST->isNullValue()) | ||||
2671 | NonNullOperand = 2; | ||||
2672 | // div/rem X, (Cond ? Y : 0) -> div/rem X, Y | ||||
2673 | if (Constant *ST = dyn_cast<Constant>(SI->getOperand(2))) | ||||
2674 | if (ST->isNullValue()) | ||||
2675 | NonNullOperand = 1; | ||||
2676 | |||||
2677 | if (NonNullOperand == -1) | ||||
2678 | return false; | ||||
2679 | |||||
2680 | Value *SelectCond = SI->getOperand(0); | ||||
2681 | |||||
2682 | // Change the div/rem to use 'Y' instead of the select. | ||||
2683 | I.setOperand(1, SI->getOperand(NonNullOperand)); | ||||
2684 | |||||
2685 | // Okay, we know we replace the operand of the div/rem with 'Y' with no | ||||
2686 | // problem. However, the select, or the condition of the select may have | ||||
2687 | // multiple uses. Based on our knowledge that the operand must be non-zero, | ||||
2688 | // propagate the known value for the select into other uses of it, and | ||||
2689 | // propagate a known value of the condition into its other users. | ||||
2690 | |||||
2691 | // If the select and condition only have a single use, don't bother with this, | ||||
2692 | // early exit. | ||||
2693 | if (SI->use_empty() && SelectCond->hasOneUse()) | ||||
2694 | return true; | ||||
2695 | |||||
2696 | // Scan the current block backward, looking for other uses of SI. | ||||
2697 | BasicBlock::iterator BBI = &I, BBFront = I.getParent()->begin(); | ||||
2698 | |||||
2699 | while (BBI != BBFront) { | ||||
2700 | --BBI; | ||||
2701 | // If we found a call to a function, we can't assume it will return, so | ||||
2702 | // information from below it cannot be propagated above it. | ||||
2703 | if (isa<CallInst>(BBI) && !isa<IntrinsicInst>(BBI)) | ||||
2704 | break; | ||||
2705 | |||||
2706 | // Replace uses of the select or its condition with the known values. | ||||
2707 | for (Instruction::op_iterator I = BBI->op_begin(), E = BBI->op_end(); | ||||
2708 | I != E; ++I) { | ||||
2709 | if (*I == SI) { | ||||
2710 | *I = SI->getOperand(NonNullOperand); | ||||
2711 | AddToWorkList(BBI); | ||||
2712 | } else if (*I == SelectCond) { | ||||
2713 | *I = NonNullOperand == 1 ? ConstantInt::getTrue() : | ||||
2714 | ConstantInt::getFalse(); | ||||
2715 | AddToWorkList(BBI); | ||||
2716 | } | ||||
2717 | } | ||||
2718 | |||||
2719 | // If we past the instruction, quit looking for it. | ||||
2720 | if (&*BBI == SI) | ||||
2721 | SI = 0; | ||||
2722 | if (&*BBI == SelectCond) | ||||
2723 | SelectCond = 0; | ||||
2724 | |||||
2725 | // If we ran out of things to eliminate, break out of the loop. | ||||
2726 | if (SelectCond == 0 && SI == 0) | ||||
2727 | break; | ||||
2728 | |||||
2729 | } | ||||
2730 | return true; | ||||
2731 | } | ||||
2732 | |||||
2733 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2734 | /// This function implements the transforms on div instructions that work |
2735 | /// regardless of the kind of div instruction it is (udiv, sdiv, or fdiv). It is | ||||
2736 | /// used by the visitors to those instructions. | ||||
2737 | /// @brief Transforms common to all three div instructions | ||||
2738 | Instruction *InstCombiner::commonDivTransforms(BinaryOperator &I) { | ||||
2739 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2740 | |||||
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 2741 | // undef / X -> 0 for integer. |
2742 | // undef / X -> undef for FP (the undef could be a snan). | ||||
2743 | if (isa<UndefValue>(Op0)) { | ||||
2744 | if (Op0->getType()->isFPOrFPVector()) | ||||
2745 | return ReplaceInstUsesWith(I, Op0); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2746 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 2747 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2748 | |
2749 | // X / undef -> undef | ||||
2750 | if (isa<UndefValue>(Op1)) | ||||
2751 | return ReplaceInstUsesWith(I, Op1); | ||||
2752 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2753 | return 0; |
2754 | } | ||||
2755 | |||||
2756 | /// This function implements the transforms common to both integer division | ||||
2757 | /// instructions (udiv and sdiv). It is called by the visitors to those integer | ||||
2758 | /// division instructions. | ||||
2759 | /// @brief Common integer divide transforms | ||||
2760 | Instruction *InstCombiner::commonIDivTransforms(BinaryOperator &I) { | ||||
2761 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2762 | |||||
Chris Lattner | cefb36c | 2008-05-16 02:59:42 +0000 | [diff] [blame] | 2763 | // (sdiv X, X) --> 1 (udiv X, X) --> 1 |
Nick Lewycky | 386c013 | 2008-05-23 03:26:47 +0000 | [diff] [blame] | 2764 | if (Op0 == Op1) { |
2765 | if (const VectorType *Ty = dyn_cast<VectorType>(I.getType())) { | ||||
2766 | ConstantInt *CI = ConstantInt::get(Ty->getElementType(), 1); | ||||
2767 | std::vector<Constant*> Elts(Ty->getNumElements(), CI); | ||||
2768 | return ReplaceInstUsesWith(I, ConstantVector::get(Elts)); | ||||
2769 | } | ||||
2770 | |||||
2771 | ConstantInt *CI = ConstantInt::get(I.getType(), 1); | ||||
2772 | return ReplaceInstUsesWith(I, CI); | ||||
2773 | } | ||||
Chris Lattner | cefb36c | 2008-05-16 02:59:42 +0000 | [diff] [blame] | 2774 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2775 | if (Instruction *Common = commonDivTransforms(I)) |
2776 | return Common; | ||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 2777 | |
2778 | // Handle cases involving: [su]div X, (select Cond, Y, Z) | ||||
2779 | // This does not apply for fdiv. | ||||
2780 | if (isa<SelectInst>(Op1) && SimplifyDivRemOfSelect(I)) | ||||
2781 | return &I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2782 | |
2783 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
2784 | // div X, 1 == X | ||||
2785 | if (RHS->equalsInt(1)) | ||||
2786 | return ReplaceInstUsesWith(I, Op0); | ||||
2787 | |||||
2788 | // (X / C1) / C2 -> X / (C1*C2) | ||||
2789 | if (Instruction *LHS = dyn_cast<Instruction>(Op0)) | ||||
2790 | if (Instruction::BinaryOps(LHS->getOpcode()) == I.getOpcode()) | ||||
2791 | if (ConstantInt *LHSRHS = dyn_cast<ConstantInt>(LHS->getOperand(1))) { | ||||
Nick Lewycky | 9d798f9 | 2008-02-18 22:48:05 +0000 | [diff] [blame] | 2792 | if (MultiplyOverflows(RHS, LHSRHS, I.getOpcode()==Instruction::SDiv)) |
2793 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2794 | else | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2795 | return BinaryOperator::Create(I.getOpcode(), LHS->getOperand(0), |
Nick Lewycky | 9d798f9 | 2008-02-18 22:48:05 +0000 | [diff] [blame] | 2796 | Multiply(RHS, LHSRHS)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2797 | } |
2798 | |||||
2799 | if (!RHS->isZero()) { // avoid X udiv 0 | ||||
2800 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
2801 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2802 | return R; | ||||
2803 | if (isa<PHINode>(Op0)) | ||||
2804 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2805 | return NV; | ||||
2806 | } | ||||
2807 | } | ||||
2808 | |||||
2809 | // 0 / X == 0, we don't need to preserve faults! | ||||
2810 | if (ConstantInt *LHS = dyn_cast<ConstantInt>(Op0)) | ||||
2811 | if (LHS->equalsInt(0)) | ||||
2812 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2813 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2814 | // It can't be division by zero, hence it must be division by one. |
2815 | if (I.getType() == Type::Int1Ty) | ||||
2816 | return ReplaceInstUsesWith(I, Op0); | ||||
2817 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2818 | return 0; |
2819 | } | ||||
2820 | |||||
2821 | Instruction *InstCombiner::visitUDiv(BinaryOperator &I) { | ||||
2822 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2823 | |||||
2824 | // Handle the integer div common cases | ||||
2825 | if (Instruction *Common = commonIDivTransforms(I)) | ||||
2826 | return Common; | ||||
2827 | |||||
2828 | // X udiv C^2 -> X >> C | ||||
2829 | // Check to see if this is an unsigned division with an exact power of 2, | ||||
2830 | // if so, convert to a right shift. | ||||
2831 | if (ConstantInt *C = dyn_cast<ConstantInt>(Op1)) { | ||||
2832 | if (C->getValue().isPowerOf2()) // 0 not included in isPowerOf2 | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2833 | return BinaryOperator::CreateLShr(Op0, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2834 | ConstantInt::get(Op0->getType(), C->getValue().logBase2())); |
2835 | } | ||||
2836 | |||||
2837 | // X udiv (C1 << N), where C1 is "1<<C2" --> X >> (N+C2) | ||||
2838 | if (BinaryOperator *RHSI = dyn_cast<BinaryOperator>(I.getOperand(1))) { | ||||
2839 | if (RHSI->getOpcode() == Instruction::Shl && | ||||
2840 | isa<ConstantInt>(RHSI->getOperand(0))) { | ||||
2841 | const APInt& C1 = cast<ConstantInt>(RHSI->getOperand(0))->getValue(); | ||||
2842 | if (C1.isPowerOf2()) { | ||||
2843 | Value *N = RHSI->getOperand(1); | ||||
2844 | const Type *NTy = N->getType(); | ||||
2845 | if (uint32_t C2 = C1.logBase2()) { | ||||
2846 | Constant *C2V = ConstantInt::get(NTy, C2); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2847 | N = InsertNewInstBefore(BinaryOperator::CreateAdd(N, C2V, "tmp"), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2848 | } |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2849 | return BinaryOperator::CreateLShr(Op0, N); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2850 | } |
2851 | } | ||||
2852 | } | ||||
2853 | |||||
2854 | // udiv X, (Select Cond, C1, C2) --> Select Cond, (shr X, C1), (shr X, C2) | ||||
2855 | // where C1&C2 are powers of two. | ||||
2856 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) | ||||
2857 | if (ConstantInt *STO = dyn_cast<ConstantInt>(SI->getOperand(1))) | ||||
2858 | if (ConstantInt *SFO = dyn_cast<ConstantInt>(SI->getOperand(2))) { | ||||
2859 | const APInt &TVA = STO->getValue(), &FVA = SFO->getValue(); | ||||
2860 | if (TVA.isPowerOf2() && FVA.isPowerOf2()) { | ||||
2861 | // Compute the shift amounts | ||||
2862 | uint32_t TSA = TVA.logBase2(), FSA = FVA.logBase2(); | ||||
2863 | // Construct the "on true" case of the select | ||||
2864 | Constant *TC = ConstantInt::get(Op0->getType(), TSA); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2865 | Instruction *TSI = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2866 | Op0, TC, SI->getName()+".t"); |
2867 | TSI = InsertNewInstBefore(TSI, I); | ||||
2868 | |||||
2869 | // Construct the "on false" case of the select | ||||
2870 | Constant *FC = ConstantInt::get(Op0->getType(), FSA); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2871 | Instruction *FSI = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2872 | Op0, FC, SI->getName()+".f"); |
2873 | FSI = InsertNewInstBefore(FSI, I); | ||||
2874 | |||||
2875 | // construct the select instruction and return it. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2876 | return SelectInst::Create(SI->getOperand(0), TSI, FSI, SI->getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2877 | } |
2878 | } | ||||
2879 | return 0; | ||||
2880 | } | ||||
2881 | |||||
2882 | Instruction *InstCombiner::visitSDiv(BinaryOperator &I) { | ||||
2883 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2884 | |||||
2885 | // Handle the integer div common cases | ||||
2886 | if (Instruction *Common = commonIDivTransforms(I)) | ||||
2887 | return Common; | ||||
2888 | |||||
2889 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
2890 | // sdiv X, -1 == -X | ||||
2891 | if (RHS->isAllOnesValue()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2892 | return BinaryOperator::CreateNeg(Op0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2893 | |
2894 | // -X/C -> X/-C | ||||
2895 | if (Value *LHSNeg = dyn_castNegVal(Op0)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2896 | return BinaryOperator::CreateSDiv(LHSNeg, ConstantExpr::getNeg(RHS)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2897 | } |
2898 | |||||
2899 | // If the sign bits of both operands are zero (i.e. we can prove they are | ||||
2900 | // unsigned inputs), turn this into a udiv. | ||||
2901 | if (I.getType()->isInteger()) { | ||||
2902 | APInt Mask(APInt::getSignBit(I.getType()->getPrimitiveSizeInBits())); | ||||
2903 | if (MaskedValueIsZero(Op1, Mask) && MaskedValueIsZero(Op0, Mask)) { | ||||
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 2904 | // 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] | 2905 | return BinaryOperator::CreateUDiv(Op0, Op1, I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2906 | } |
2907 | } | ||||
2908 | |||||
2909 | return 0; | ||||
2910 | } | ||||
2911 | |||||
2912 | Instruction *InstCombiner::visitFDiv(BinaryOperator &I) { | ||||
2913 | return commonDivTransforms(I); | ||||
2914 | } | ||||
2915 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2916 | /// This function implements the transforms on rem instructions that work |
2917 | /// regardless of the kind of rem instruction it is (urem, srem, or frem). It | ||||
2918 | /// is used by the visitors to those instructions. | ||||
2919 | /// @brief Transforms common to all three rem instructions | ||||
2920 | Instruction *InstCombiner::commonRemTransforms(BinaryOperator &I) { | ||||
2921 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2922 | |||||
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 2923 | // 0 % X == 0 for integer, we don't need to preserve faults! |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2924 | if (Constant *LHS = dyn_cast<Constant>(Op0)) |
2925 | if (LHS->isNullValue()) | ||||
2926 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2927 | |||||
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 2928 | if (isa<UndefValue>(Op0)) { // undef % X -> 0 |
2929 | if (I.getType()->isFPOrFPVector()) | ||||
2930 | return ReplaceInstUsesWith(I, Op0); // X % undef -> undef (could be SNaN) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2931 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 2932 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2933 | if (isa<UndefValue>(Op1)) |
2934 | return ReplaceInstUsesWith(I, Op1); // X % undef -> undef | ||||
2935 | |||||
2936 | // Handle cases involving: rem X, (select Cond, Y, Z) | ||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 2937 | if (isa<SelectInst>(Op1) && SimplifyDivRemOfSelect(I)) |
2938 | return &I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2939 | |
2940 | return 0; | ||||
2941 | } | ||||
2942 | |||||
2943 | /// This function implements the transforms common to both integer remainder | ||||
2944 | /// instructions (urem and srem). It is called by the visitors to those integer | ||||
2945 | /// remainder instructions. | ||||
2946 | /// @brief Common integer remainder transforms | ||||
2947 | Instruction *InstCombiner::commonIRemTransforms(BinaryOperator &I) { | ||||
2948 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2949 | |||||
2950 | if (Instruction *common = commonRemTransforms(I)) | ||||
2951 | return common; | ||||
2952 | |||||
2953 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
2954 | // X % 0 == undef, we don't need to preserve faults! | ||||
2955 | if (RHS->equalsInt(0)) | ||||
2956 | return ReplaceInstUsesWith(I, UndefValue::get(I.getType())); | ||||
2957 | |||||
2958 | if (RHS->equalsInt(1)) // X % 1 == 0 | ||||
2959 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2960 | |||||
2961 | if (Instruction *Op0I = dyn_cast<Instruction>(Op0)) { | ||||
2962 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0I)) { | ||||
2963 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2964 | return R; | ||||
2965 | } else if (isa<PHINode>(Op0I)) { | ||||
2966 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2967 | return NV; | ||||
2968 | } | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 2969 | |
2970 | // See if we can fold away this rem instruction. | ||||
2971 | uint32_t BitWidth = cast<IntegerType>(I.getType())->getBitWidth(); | ||||
2972 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
2973 | if (SimplifyDemandedBits(&I, APInt::getAllOnesValue(BitWidth), | ||||
2974 | KnownZero, KnownOne)) | ||||
2975 | return &I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2976 | } |
2977 | } | ||||
2978 | |||||
2979 | return 0; | ||||
2980 | } | ||||
2981 | |||||
2982 | Instruction *InstCombiner::visitURem(BinaryOperator &I) { | ||||
2983 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2984 | |||||
2985 | if (Instruction *common = commonIRemTransforms(I)) | ||||
2986 | return common; | ||||
2987 | |||||
2988 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
2989 | // X urem C^2 -> X and C | ||||
2990 | // Check to see if this is an unsigned remainder with an exact power of 2, | ||||
2991 | // if so, convert to a bitwise and. | ||||
2992 | if (ConstantInt *C = dyn_cast<ConstantInt>(RHS)) | ||||
2993 | if (C->getValue().isPowerOf2()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2994 | return BinaryOperator::CreateAnd(Op0, SubOne(C)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2995 | } |
2996 | |||||
2997 | if (Instruction *RHSI = dyn_cast<Instruction>(I.getOperand(1))) { | ||||
2998 | // Turn A % (C << N), where C is 2^k, into A & ((C << N)-1) | ||||
2999 | if (RHSI->getOpcode() == Instruction::Shl && | ||||
3000 | isa<ConstantInt>(RHSI->getOperand(0))) { | ||||
3001 | if (cast<ConstantInt>(RHSI->getOperand(0))->getValue().isPowerOf2()) { | ||||
3002 | Constant *N1 = ConstantInt::getAllOnesValue(I.getType()); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3003 | Value *Add = InsertNewInstBefore(BinaryOperator::CreateAdd(RHSI, N1, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3004 | "tmp"), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3005 | return BinaryOperator::CreateAnd(Op0, Add); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3006 | } |
3007 | } | ||||
3008 | } | ||||
3009 | |||||
3010 | // urem X, (select Cond, 2^C1, 2^C2) --> select Cond, (and X, C1), (and X, C2) | ||||
3011 | // where C1&C2 are powers of two. | ||||
3012 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) { | ||||
3013 | if (ConstantInt *STO = dyn_cast<ConstantInt>(SI->getOperand(1))) | ||||
3014 | if (ConstantInt *SFO = dyn_cast<ConstantInt>(SI->getOperand(2))) { | ||||
3015 | // STO == 0 and SFO == 0 handled above. | ||||
3016 | if ((STO->getValue().isPowerOf2()) && | ||||
3017 | (SFO->getValue().isPowerOf2())) { | ||||
3018 | Value *TrueAnd = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3019 | BinaryOperator::CreateAnd(Op0, SubOne(STO), SI->getName()+".t"), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3020 | Value *FalseAnd = InsertNewInstBefore( |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3021 | BinaryOperator::CreateAnd(Op0, SubOne(SFO), SI->getName()+".f"), I); |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 3022 | return SelectInst::Create(SI->getOperand(0), TrueAnd, FalseAnd); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3023 | } |
3024 | } | ||||
3025 | } | ||||
3026 | |||||
3027 | return 0; | ||||
3028 | } | ||||
3029 | |||||
3030 | Instruction *InstCombiner::visitSRem(BinaryOperator &I) { | ||||
3031 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3032 | |||||
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3033 | // Handle the integer rem common cases |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3034 | if (Instruction *common = commonIRemTransforms(I)) |
3035 | return common; | ||||
3036 | |||||
3037 | if (Value *RHSNeg = dyn_castNegVal(Op1)) | ||||
Nick Lewycky | cfadfbd | 2008-09-03 06:24:21 +0000 | [diff] [blame] | 3038 | if (!isa<Constant>(RHSNeg) || |
3039 | (isa<ConstantInt>(RHSNeg) && | ||||
3040 | cast<ConstantInt>(RHSNeg)->getValue().isStrictlyPositive())) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3041 | // X % -Y -> X % Y |
3042 | AddUsesToWorkList(I); | ||||
3043 | I.setOperand(1, RHSNeg); | ||||
3044 | return &I; | ||||
3045 | } | ||||
Nick Lewycky | 5515c7a | 2008-09-30 06:08:34 +0000 | [diff] [blame] | 3046 | |
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3047 | // 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] | 3048 | // unsigned inputs), turn this into a urem. |
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3049 | if (I.getType()->isInteger()) { |
3050 | APInt Mask(APInt::getSignBit(I.getType()->getPrimitiveSizeInBits())); | ||||
3051 | if (MaskedValueIsZero(Op1, Mask) && MaskedValueIsZero(Op0, Mask)) { | ||||
3052 | // 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] | 3053 | return BinaryOperator::CreateURem(Op0, Op1, I.getName()); |
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3054 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3055 | } |
3056 | |||||
3057 | return 0; | ||||
3058 | } | ||||
3059 | |||||
3060 | Instruction *InstCombiner::visitFRem(BinaryOperator &I) { | ||||
3061 | return commonRemTransforms(I); | ||||
3062 | } | ||||
3063 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3064 | // isOneBitSet - Return true if there is exactly one bit set in the specified |
3065 | // constant. | ||||
3066 | static bool isOneBitSet(const ConstantInt *CI) { | ||||
3067 | return CI->getValue().isPowerOf2(); | ||||
3068 | } | ||||
3069 | |||||
3070 | // isHighOnes - Return true if the constant is of the form 1+0+. | ||||
3071 | // This is the same as lowones(~X). | ||||
3072 | static bool isHighOnes(const ConstantInt *CI) { | ||||
3073 | return (~CI->getValue() + 1).isPowerOf2(); | ||||
3074 | } | ||||
3075 | |||||
3076 | /// getICmpCode - Encode a icmp predicate into a three bit mask. These bits | ||||
3077 | /// are carefully arranged to allow folding of expressions such as: | ||||
3078 | /// | ||||
3079 | /// (A < B) | (A > B) --> (A != B) | ||||
3080 | /// | ||||
3081 | /// Note that this is only valid if the first and second predicates have the | ||||
3082 | /// same sign. Is illegal to do: (A u< B) | (A s> B) | ||||
3083 | /// | ||||
3084 | /// Three bits are used to represent the condition, as follows: | ||||
3085 | /// 0 A > B | ||||
3086 | /// 1 A == B | ||||
3087 | /// 2 A < B | ||||
3088 | /// | ||||
3089 | /// <=> Value Definition | ||||
3090 | /// 000 0 Always false | ||||
3091 | /// 001 1 A > B | ||||
3092 | /// 010 2 A == B | ||||
3093 | /// 011 3 A >= B | ||||
3094 | /// 100 4 A < B | ||||
3095 | /// 101 5 A != B | ||||
3096 | /// 110 6 A <= B | ||||
3097 | /// 111 7 Always true | ||||
3098 | /// | ||||
3099 | static unsigned getICmpCode(const ICmpInst *ICI) { | ||||
3100 | switch (ICI->getPredicate()) { | ||||
3101 | // False -> 0 | ||||
3102 | case ICmpInst::ICMP_UGT: return 1; // 001 | ||||
3103 | case ICmpInst::ICMP_SGT: return 1; // 001 | ||||
3104 | case ICmpInst::ICMP_EQ: return 2; // 010 | ||||
3105 | case ICmpInst::ICMP_UGE: return 3; // 011 | ||||
3106 | case ICmpInst::ICMP_SGE: return 3; // 011 | ||||
3107 | case ICmpInst::ICMP_ULT: return 4; // 100 | ||||
3108 | case ICmpInst::ICMP_SLT: return 4; // 100 | ||||
3109 | case ICmpInst::ICMP_NE: return 5; // 101 | ||||
3110 | case ICmpInst::ICMP_ULE: return 6; // 110 | ||||
3111 | case ICmpInst::ICMP_SLE: return 6; // 110 | ||||
3112 | // True -> 7 | ||||
3113 | default: | ||||
3114 | assert(0 && "Invalid ICmp predicate!"); | ||||
3115 | return 0; | ||||
3116 | } | ||||
3117 | } | ||||
3118 | |||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame^] | 3119 | /// getFCmpCode - Similar to getICmpCode but for FCmpInst. This encodes a fcmp |
3120 | /// predicate into a three bit mask. It also returns whether it is an ordered | ||||
3121 | /// predicate by reference. | ||||
3122 | static unsigned getFCmpCode(FCmpInst::Predicate CC, bool &isOrdered) { | ||||
3123 | isOrdered = false; | ||||
3124 | switch (CC) { | ||||
3125 | case FCmpInst::FCMP_ORD: isOrdered = true; return 0; // 000 | ||||
3126 | case FCmpInst::FCMP_UNO: return 0; // 000 | ||||
3127 | case FCmpInst::FCMP_OEQ: isOrdered = true; return 1; // 001 | ||||
3128 | case FCmpInst::FCMP_UEQ: return 1; // 001 | ||||
3129 | case FCmpInst::FCMP_OGT: isOrdered = true; return 2; // 010 | ||||
3130 | case FCmpInst::FCMP_UGT: return 2; // 010 | ||||
3131 | case FCmpInst::FCMP_OGE: isOrdered = true; return 3; // 011 | ||||
3132 | case FCmpInst::FCMP_UGE: return 3; // 011 | ||||
3133 | case FCmpInst::FCMP_OLT: isOrdered = true; return 4; // 100 | ||||
3134 | case FCmpInst::FCMP_ULT: return 4; // 100 | ||||
3135 | case FCmpInst::FCMP_OLE: isOrdered = true; return 6; // 110 | ||||
3136 | case FCmpInst::FCMP_ULE: return 6; // 110 | ||||
3137 | default: | ||||
3138 | // Not expecting FCMP_FALSE and FCMP_TRUE; | ||||
3139 | assert(0 && "Unexpected FCmp predicate!"); | ||||
3140 | return 0; | ||||
3141 | } | ||||
3142 | } | ||||
3143 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3144 | /// getICmpValue - This is the complement of getICmpCode, which turns an |
3145 | /// 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] | 3146 | /// new ICmp instruction. The sign is passed in to determine which kind |
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame^] | 3147 | /// of predicate to use in the new icmp instruction. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3148 | static Value *getICmpValue(bool sign, unsigned code, Value *LHS, Value *RHS) { |
3149 | switch (code) { | ||||
3150 | default: assert(0 && "Illegal ICmp code!"); | ||||
3151 | case 0: return ConstantInt::getFalse(); | ||||
3152 | case 1: | ||||
3153 | if (sign) | ||||
3154 | return new ICmpInst(ICmpInst::ICMP_SGT, LHS, RHS); | ||||
3155 | else | ||||
3156 | return new ICmpInst(ICmpInst::ICMP_UGT, LHS, RHS); | ||||
3157 | case 2: return new ICmpInst(ICmpInst::ICMP_EQ, LHS, RHS); | ||||
3158 | case 3: | ||||
3159 | if (sign) | ||||
3160 | return new ICmpInst(ICmpInst::ICMP_SGE, LHS, RHS); | ||||
3161 | else | ||||
3162 | return new ICmpInst(ICmpInst::ICMP_UGE, LHS, RHS); | ||||
3163 | case 4: | ||||
3164 | if (sign) | ||||
3165 | return new ICmpInst(ICmpInst::ICMP_SLT, LHS, RHS); | ||||
3166 | else | ||||
3167 | return new ICmpInst(ICmpInst::ICMP_ULT, LHS, RHS); | ||||
3168 | case 5: return new ICmpInst(ICmpInst::ICMP_NE, LHS, RHS); | ||||
3169 | case 6: | ||||
3170 | if (sign) | ||||
3171 | return new ICmpInst(ICmpInst::ICMP_SLE, LHS, RHS); | ||||
3172 | else | ||||
3173 | return new ICmpInst(ICmpInst::ICMP_ULE, LHS, RHS); | ||||
3174 | case 7: return ConstantInt::getTrue(); | ||||
3175 | } | ||||
3176 | } | ||||
3177 | |||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame^] | 3178 | /// getFCmpValue - This is the complement of getFCmpCode, which turns an |
3179 | /// opcode and two operands into either a FCmp instruction. isordered is passed | ||||
3180 | /// in to determine which kind of predicate to use in the new fcmp instruction. | ||||
3181 | static Value *getFCmpValue(bool isordered, unsigned code, | ||||
3182 | Value *LHS, Value *RHS) { | ||||
3183 | switch (code) { | ||||
3184 | default: assert(0 && "Illegal ICmp code!"); | ||||
3185 | case 0: | ||||
3186 | if (isordered) | ||||
3187 | return new FCmpInst(FCmpInst::FCMP_ORD, LHS, RHS); | ||||
3188 | else | ||||
3189 | return new FCmpInst(FCmpInst::FCMP_UNO, LHS, RHS); | ||||
3190 | case 1: | ||||
3191 | if (isordered) | ||||
3192 | return new FCmpInst(FCmpInst::FCMP_OEQ, LHS, RHS); | ||||
3193 | else | ||||
3194 | return new FCmpInst(FCmpInst::FCMP_UEQ, LHS, RHS); | ||||
3195 | case 2: | ||||
3196 | if (isordered) | ||||
3197 | return new FCmpInst(FCmpInst::FCMP_OGT, LHS, RHS); | ||||
3198 | else | ||||
3199 | return new FCmpInst(FCmpInst::FCMP_UGT, LHS, RHS); | ||||
3200 | case 3: | ||||
3201 | if (isordered) | ||||
3202 | return new FCmpInst(FCmpInst::FCMP_OGE, LHS, RHS); | ||||
3203 | else | ||||
3204 | return new FCmpInst(FCmpInst::FCMP_UGE, LHS, RHS); | ||||
3205 | case 4: | ||||
3206 | if (isordered) | ||||
3207 | return new FCmpInst(FCmpInst::FCMP_OLT, LHS, RHS); | ||||
3208 | else | ||||
3209 | return new FCmpInst(FCmpInst::FCMP_ULT, LHS, RHS); | ||||
3210 | case 5: | ||||
3211 | if (isordered) | ||||
3212 | return new FCmpInst(FCmpInst::FCMP_OLE, LHS, RHS); | ||||
3213 | else | ||||
3214 | return new FCmpInst(FCmpInst::FCMP_ULE, LHS, RHS); | ||||
3215 | } | ||||
3216 | } | ||||
3217 | |||||
3218 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3219 | static bool PredicatesFoldable(ICmpInst::Predicate p1, ICmpInst::Predicate p2) { |
3220 | return (ICmpInst::isSignedPredicate(p1) == ICmpInst::isSignedPredicate(p2)) || | ||||
3221 | (ICmpInst::isSignedPredicate(p1) && | ||||
3222 | (p2 == ICmpInst::ICMP_EQ || p2 == ICmpInst::ICMP_NE)) || | ||||
3223 | (ICmpInst::isSignedPredicate(p2) && | ||||
3224 | (p1 == ICmpInst::ICMP_EQ || p1 == ICmpInst::ICMP_NE)); | ||||
3225 | } | ||||
3226 | |||||
3227 | namespace { | ||||
3228 | // FoldICmpLogical - Implements (icmp1 A, B) & (icmp2 A, B) --> (icmp3 A, B) | ||||
3229 | struct FoldICmpLogical { | ||||
3230 | InstCombiner &IC; | ||||
3231 | Value *LHS, *RHS; | ||||
3232 | ICmpInst::Predicate pred; | ||||
3233 | FoldICmpLogical(InstCombiner &ic, ICmpInst *ICI) | ||||
3234 | : IC(ic), LHS(ICI->getOperand(0)), RHS(ICI->getOperand(1)), | ||||
3235 | pred(ICI->getPredicate()) {} | ||||
3236 | bool shouldApply(Value *V) const { | ||||
3237 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(V)) | ||||
3238 | if (PredicatesFoldable(pred, ICI->getPredicate())) | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 3239 | return ((ICI->getOperand(0) == LHS && ICI->getOperand(1) == RHS) || |
3240 | (ICI->getOperand(0) == RHS && ICI->getOperand(1) == LHS)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3241 | return false; |
3242 | } | ||||
3243 | Instruction *apply(Instruction &Log) const { | ||||
3244 | ICmpInst *ICI = cast<ICmpInst>(Log.getOperand(0)); | ||||
3245 | if (ICI->getOperand(0) != LHS) { | ||||
3246 | assert(ICI->getOperand(1) == LHS); | ||||
3247 | ICI->swapOperands(); // Swap the LHS and RHS of the ICmp | ||||
3248 | } | ||||
3249 | |||||
3250 | ICmpInst *RHSICI = cast<ICmpInst>(Log.getOperand(1)); | ||||
3251 | unsigned LHSCode = getICmpCode(ICI); | ||||
3252 | unsigned RHSCode = getICmpCode(RHSICI); | ||||
3253 | unsigned Code; | ||||
3254 | switch (Log.getOpcode()) { | ||||
3255 | case Instruction::And: Code = LHSCode & RHSCode; break; | ||||
3256 | case Instruction::Or: Code = LHSCode | RHSCode; break; | ||||
3257 | case Instruction::Xor: Code = LHSCode ^ RHSCode; break; | ||||
3258 | default: assert(0 && "Illegal logical opcode!"); return 0; | ||||
3259 | } | ||||
3260 | |||||
3261 | bool isSigned = ICmpInst::isSignedPredicate(RHSICI->getPredicate()) || | ||||
3262 | ICmpInst::isSignedPredicate(ICI->getPredicate()); | ||||
3263 | |||||
3264 | Value *RV = getICmpValue(isSigned, Code, LHS, RHS); | ||||
3265 | if (Instruction *I = dyn_cast<Instruction>(RV)) | ||||
3266 | return I; | ||||
3267 | // Otherwise, it's a constant boolean value... | ||||
3268 | return IC.ReplaceInstUsesWith(Log, RV); | ||||
3269 | } | ||||
3270 | }; | ||||
3271 | } // end anonymous namespace | ||||
3272 | |||||
3273 | // OptAndOp - This handles expressions of the form ((val OP C1) & C2). Where | ||||
3274 | // the Op parameter is 'OP', OpRHS is 'C1', and AndRHS is 'C2'. Op is | ||||
3275 | // guaranteed to be a binary operator. | ||||
3276 | Instruction *InstCombiner::OptAndOp(Instruction *Op, | ||||
3277 | ConstantInt *OpRHS, | ||||
3278 | ConstantInt *AndRHS, | ||||
3279 | BinaryOperator &TheAnd) { | ||||
3280 | Value *X = Op->getOperand(0); | ||||
3281 | Constant *Together = 0; | ||||
3282 | if (!Op->isShift()) | ||||
3283 | Together = And(AndRHS, OpRHS); | ||||
3284 | |||||
3285 | switch (Op->getOpcode()) { | ||||
3286 | case Instruction::Xor: | ||||
3287 | if (Op->hasOneUse()) { | ||||
3288 | // (X ^ C1) & C2 --> (X & C2) ^ (C1&C2) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3289 | Instruction *And = BinaryOperator::CreateAnd(X, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3290 | InsertNewInstBefore(And, TheAnd); |
3291 | And->takeName(Op); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3292 | return BinaryOperator::CreateXor(And, Together); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3293 | } |
3294 | break; | ||||
3295 | case Instruction::Or: | ||||
3296 | if (Together == AndRHS) // (X | C) & C --> C | ||||
3297 | return ReplaceInstUsesWith(TheAnd, AndRHS); | ||||
3298 | |||||
3299 | if (Op->hasOneUse() && Together != OpRHS) { | ||||
3300 | // (X | C1) & C2 --> (X | (C1&C2)) & C2 | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3301 | Instruction *Or = BinaryOperator::CreateOr(X, Together); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3302 | InsertNewInstBefore(Or, TheAnd); |
3303 | Or->takeName(Op); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3304 | return BinaryOperator::CreateAnd(Or, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3305 | } |
3306 | break; | ||||
3307 | case Instruction::Add: | ||||
3308 | if (Op->hasOneUse()) { | ||||
3309 | // Adding a one to a single bit bit-field should be turned into an XOR | ||||
3310 | // of the bit. First thing to check is to see if this AND is with a | ||||
3311 | // single bit constant. | ||||
3312 | const APInt& AndRHSV = cast<ConstantInt>(AndRHS)->getValue(); | ||||
3313 | |||||
3314 | // If there is only one bit set... | ||||
3315 | if (isOneBitSet(cast<ConstantInt>(AndRHS))) { | ||||
3316 | // Ok, at this point, we know that we are masking the result of the | ||||
3317 | // ADD down to exactly one bit. If the constant we are adding has | ||||
3318 | // no bits set below this bit, then we can eliminate the ADD. | ||||
3319 | const APInt& AddRHS = cast<ConstantInt>(OpRHS)->getValue(); | ||||
3320 | |||||
3321 | // Check to see if any bits below the one bit set in AndRHSV are set. | ||||
3322 | if ((AddRHS & (AndRHSV-1)) == 0) { | ||||
3323 | // If not, the only thing that can effect the output of the AND is | ||||
3324 | // the bit specified by AndRHSV. If that bit is set, the effect of | ||||
3325 | // the XOR is to toggle the bit. If it is clear, then the ADD has | ||||
3326 | // no effect. | ||||
3327 | if ((AddRHS & AndRHSV) == 0) { // Bit is not set, noop | ||||
3328 | TheAnd.setOperand(0, X); | ||||
3329 | return &TheAnd; | ||||
3330 | } else { | ||||
3331 | // Pull the XOR out of the AND. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3332 | Instruction *NewAnd = BinaryOperator::CreateAnd(X, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3333 | InsertNewInstBefore(NewAnd, TheAnd); |
3334 | NewAnd->takeName(Op); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3335 | return BinaryOperator::CreateXor(NewAnd, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3336 | } |
3337 | } | ||||
3338 | } | ||||
3339 | } | ||||
3340 | break; | ||||
3341 | |||||
3342 | case Instruction::Shl: { | ||||
3343 | // We know that the AND will not produce any of the bits shifted in, so if | ||||
3344 | // the anded constant includes them, clear them now! | ||||
3345 | // | ||||
3346 | uint32_t BitWidth = AndRHS->getType()->getBitWidth(); | ||||
3347 | uint32_t OpRHSVal = OpRHS->getLimitedValue(BitWidth); | ||||
3348 | APInt ShlMask(APInt::getHighBitsSet(BitWidth, BitWidth-OpRHSVal)); | ||||
3349 | ConstantInt *CI = ConstantInt::get(AndRHS->getValue() & ShlMask); | ||||
3350 | |||||
3351 | if (CI->getValue() == ShlMask) { | ||||
3352 | // Masking out bits that the shift already masks | ||||
3353 | return ReplaceInstUsesWith(TheAnd, Op); // No need for the and. | ||||
3354 | } else if (CI != AndRHS) { // Reducing bits set in and. | ||||
3355 | TheAnd.setOperand(1, CI); | ||||
3356 | return &TheAnd; | ||||
3357 | } | ||||
3358 | break; | ||||
3359 | } | ||||
3360 | case Instruction::LShr: | ||||
3361 | { | ||||
3362 | // We know that the AND will not produce any of the bits shifted in, so if | ||||
3363 | // the anded constant includes them, clear them now! This only applies to | ||||
3364 | // unsigned shifts, because a signed shr may bring in set bits! | ||||
3365 | // | ||||
3366 | uint32_t BitWidth = AndRHS->getType()->getBitWidth(); | ||||
3367 | uint32_t OpRHSVal = OpRHS->getLimitedValue(BitWidth); | ||||
3368 | APInt ShrMask(APInt::getLowBitsSet(BitWidth, BitWidth - OpRHSVal)); | ||||
3369 | ConstantInt *CI = ConstantInt::get(AndRHS->getValue() & ShrMask); | ||||
3370 | |||||
3371 | if (CI->getValue() == ShrMask) { | ||||
3372 | // Masking out bits that the shift already masks. | ||||
3373 | return ReplaceInstUsesWith(TheAnd, Op); | ||||
3374 | } else if (CI != AndRHS) { | ||||
3375 | TheAnd.setOperand(1, CI); // Reduce bits set in and cst. | ||||
3376 | return &TheAnd; | ||||
3377 | } | ||||
3378 | break; | ||||
3379 | } | ||||
3380 | case Instruction::AShr: | ||||
3381 | // Signed shr. | ||||
3382 | // See if this is shifting in some sign extension, then masking it out | ||||
3383 | // with an and. | ||||
3384 | if (Op->hasOneUse()) { | ||||
3385 | uint32_t BitWidth = AndRHS->getType()->getBitWidth(); | ||||
3386 | uint32_t OpRHSVal = OpRHS->getLimitedValue(BitWidth); | ||||
3387 | APInt ShrMask(APInt::getLowBitsSet(BitWidth, BitWidth - OpRHSVal)); | ||||
3388 | Constant *C = ConstantInt::get(AndRHS->getValue() & ShrMask); | ||||
3389 | if (C == AndRHS) { // Masking out bits shifted in. | ||||
3390 | // (Val ashr C1) & C2 -> (Val lshr C1) & C2 | ||||
3391 | // Make the argument unsigned. | ||||
3392 | Value *ShVal = Op->getOperand(0); | ||||
3393 | ShVal = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3394 | BinaryOperator::CreateLShr(ShVal, OpRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3395 | Op->getName()), TheAnd); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3396 | return BinaryOperator::CreateAnd(ShVal, AndRHS, TheAnd.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3397 | } |
3398 | } | ||||
3399 | break; | ||||
3400 | } | ||||
3401 | return 0; | ||||
3402 | } | ||||
3403 | |||||
3404 | |||||
3405 | /// InsertRangeTest - Emit a computation of: (V >= Lo && V < Hi) if Inside is | ||||
3406 | /// true, otherwise (V < Lo || V >= Hi). In pratice, we emit the more efficient | ||||
3407 | /// (V-Lo) <u Hi-Lo. This method expects that Lo <= Hi. isSigned indicates | ||||
3408 | /// whether to treat the V, Lo and HI as signed or not. IB is the location to | ||||
3409 | /// insert new instructions. | ||||
3410 | Instruction *InstCombiner::InsertRangeTest(Value *V, Constant *Lo, Constant *Hi, | ||||
3411 | bool isSigned, bool Inside, | ||||
3412 | Instruction &IB) { | ||||
3413 | assert(cast<ConstantInt>(ConstantExpr::getICmp((isSigned ? | ||||
3414 | ICmpInst::ICMP_SLE:ICmpInst::ICMP_ULE), Lo, Hi))->getZExtValue() && | ||||
3415 | "Lo is not <= Hi in range emission code!"); | ||||
3416 | |||||
3417 | if (Inside) { | ||||
3418 | if (Lo == Hi) // Trivially false. | ||||
3419 | return new ICmpInst(ICmpInst::ICMP_NE, V, V); | ||||
3420 | |||||
3421 | // V >= Min && V < Hi --> V < Hi | ||||
3422 | if (cast<ConstantInt>(Lo)->isMinValue(isSigned)) { | ||||
3423 | ICmpInst::Predicate pred = (isSigned ? | ||||
3424 | ICmpInst::ICMP_SLT : ICmpInst::ICMP_ULT); | ||||
3425 | return new ICmpInst(pred, V, Hi); | ||||
3426 | } | ||||
3427 | |||||
3428 | // Emit V-Lo <u Hi-Lo | ||||
3429 | Constant *NegLo = ConstantExpr::getNeg(Lo); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3430 | Instruction *Add = BinaryOperator::CreateAdd(V, NegLo, V->getName()+".off"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3431 | InsertNewInstBefore(Add, IB); |
3432 | Constant *UpperBound = ConstantExpr::getAdd(NegLo, Hi); | ||||
3433 | return new ICmpInst(ICmpInst::ICMP_ULT, Add, UpperBound); | ||||
3434 | } | ||||
3435 | |||||
3436 | if (Lo == Hi) // Trivially true. | ||||
3437 | return new ICmpInst(ICmpInst::ICMP_EQ, V, V); | ||||
3438 | |||||
3439 | // V < Min || V >= Hi -> V > Hi-1 | ||||
3440 | Hi = SubOne(cast<ConstantInt>(Hi)); | ||||
3441 | if (cast<ConstantInt>(Lo)->isMinValue(isSigned)) { | ||||
3442 | ICmpInst::Predicate pred = (isSigned ? | ||||
3443 | ICmpInst::ICMP_SGT : ICmpInst::ICMP_UGT); | ||||
3444 | return new ICmpInst(pred, V, Hi); | ||||
3445 | } | ||||
3446 | |||||
3447 | // Emit V-Lo >u Hi-1-Lo | ||||
3448 | // Note that Hi has already had one subtracted from it, above. | ||||
3449 | ConstantInt *NegLo = cast<ConstantInt>(ConstantExpr::getNeg(Lo)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3450 | Instruction *Add = BinaryOperator::CreateAdd(V, NegLo, V->getName()+".off"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3451 | InsertNewInstBefore(Add, IB); |
3452 | Constant *LowerBound = ConstantExpr::getAdd(NegLo, Hi); | ||||
3453 | return new ICmpInst(ICmpInst::ICMP_UGT, Add, LowerBound); | ||||
3454 | } | ||||
3455 | |||||
3456 | // isRunOfOnes - Returns true iff Val consists of one contiguous run of 1s with | ||||
3457 | // any number of 0s on either side. The 1s are allowed to wrap from LSB to | ||||
3458 | // MSB, so 0x000FFF0, 0x0000FFFF, and 0xFF0000FF are all runs. 0x0F0F0000 is | ||||
3459 | // not, since all 1s are not contiguous. | ||||
3460 | static bool isRunOfOnes(ConstantInt *Val, uint32_t &MB, uint32_t &ME) { | ||||
3461 | const APInt& V = Val->getValue(); | ||||
3462 | uint32_t BitWidth = Val->getType()->getBitWidth(); | ||||
3463 | if (!APIntOps::isShiftedMask(BitWidth, V)) return false; | ||||
3464 | |||||
3465 | // look for the first zero bit after the run of ones | ||||
3466 | MB = BitWidth - ((V - 1) ^ V).countLeadingZeros(); | ||||
3467 | // look for the first non-zero bit | ||||
3468 | ME = V.getActiveBits(); | ||||
3469 | return true; | ||||
3470 | } | ||||
3471 | |||||
3472 | /// FoldLogicalPlusAnd - This is part of an expression (LHS +/- RHS) & Mask, | ||||
3473 | /// where isSub determines whether the operator is a sub. If we can fold one of | ||||
3474 | /// the following xforms: | ||||
3475 | /// | ||||
3476 | /// ((A & N) +/- B) & Mask -> (A +/- B) & Mask iff N&Mask == Mask | ||||
3477 | /// ((A | N) +/- B) & Mask -> (A +/- B) & Mask iff N&Mask == 0 | ||||
3478 | /// ((A ^ N) +/- B) & Mask -> (A +/- B) & Mask iff N&Mask == 0 | ||||
3479 | /// | ||||
3480 | /// return (A +/- B). | ||||
3481 | /// | ||||
3482 | Value *InstCombiner::FoldLogicalPlusAnd(Value *LHS, Value *RHS, | ||||
3483 | ConstantInt *Mask, bool isSub, | ||||
3484 | Instruction &I) { | ||||
3485 | Instruction *LHSI = dyn_cast<Instruction>(LHS); | ||||
3486 | if (!LHSI || LHSI->getNumOperands() != 2 || | ||||
3487 | !isa<ConstantInt>(LHSI->getOperand(1))) return 0; | ||||
3488 | |||||
3489 | ConstantInt *N = cast<ConstantInt>(LHSI->getOperand(1)); | ||||
3490 | |||||
3491 | switch (LHSI->getOpcode()) { | ||||
3492 | default: return 0; | ||||
3493 | case Instruction::And: | ||||
3494 | if (And(N, Mask) == Mask) { | ||||
3495 | // If the AndRHS is a power of two minus one (0+1+), this is simple. | ||||
3496 | if ((Mask->getValue().countLeadingZeros() + | ||||
3497 | Mask->getValue().countPopulation()) == | ||||
3498 | Mask->getValue().getBitWidth()) | ||||
3499 | break; | ||||
3500 | |||||
3501 | // Otherwise, if Mask is 0+1+0+, and if B is known to have the low 0+ | ||||
3502 | // part, we don't need any explicit masks to take them out of A. If that | ||||
3503 | // is all N is, ignore it. | ||||
3504 | uint32_t MB = 0, ME = 0; | ||||
3505 | if (isRunOfOnes(Mask, MB, ME)) { // begin/end bit of run, inclusive | ||||
3506 | uint32_t BitWidth = cast<IntegerType>(RHS->getType())->getBitWidth(); | ||||
3507 | APInt Mask(APInt::getLowBitsSet(BitWidth, MB-1)); | ||||
3508 | if (MaskedValueIsZero(RHS, Mask)) | ||||
3509 | break; | ||||
3510 | } | ||||
3511 | } | ||||
3512 | return 0; | ||||
3513 | case Instruction::Or: | ||||
3514 | case Instruction::Xor: | ||||
3515 | // If the AndRHS is a power of two minus one (0+1+), and N&Mask == 0 | ||||
3516 | if ((Mask->getValue().countLeadingZeros() + | ||||
3517 | Mask->getValue().countPopulation()) == Mask->getValue().getBitWidth() | ||||
3518 | && And(N, Mask)->isZero()) | ||||
3519 | break; | ||||
3520 | return 0; | ||||
3521 | } | ||||
3522 | |||||
3523 | Instruction *New; | ||||
3524 | if (isSub) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3525 | New = BinaryOperator::CreateSub(LHSI->getOperand(0), RHS, "fold"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3526 | else |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3527 | New = BinaryOperator::CreateAdd(LHSI->getOperand(0), RHS, "fold"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3528 | return InsertNewInstBefore(New, I); |
3529 | } | ||||
3530 | |||||
3531 | Instruction *InstCombiner::visitAnd(BinaryOperator &I) { | ||||
3532 | bool Changed = SimplifyCommutative(I); | ||||
3533 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3534 | |||||
3535 | if (isa<UndefValue>(Op1)) // X & undef -> 0 | ||||
3536 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
3537 | |||||
3538 | // and X, X = X | ||||
3539 | if (Op0 == Op1) | ||||
3540 | return ReplaceInstUsesWith(I, Op1); | ||||
3541 | |||||
3542 | // See if we can simplify any instructions used by the instruction whose sole | ||||
3543 | // purpose is to compute bits we don't care about. | ||||
3544 | if (!isa<VectorType>(I.getType())) { | ||||
3545 | uint32_t BitWidth = cast<IntegerType>(I.getType())->getBitWidth(); | ||||
3546 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
3547 | if (SimplifyDemandedBits(&I, APInt::getAllOnesValue(BitWidth), | ||||
3548 | KnownZero, KnownOne)) | ||||
3549 | return &I; | ||||
3550 | } else { | ||||
3551 | if (ConstantVector *CP = dyn_cast<ConstantVector>(Op1)) { | ||||
3552 | if (CP->isAllOnesValue()) // X & <-1,-1> -> X | ||||
3553 | return ReplaceInstUsesWith(I, I.getOperand(0)); | ||||
3554 | } else if (isa<ConstantAggregateZero>(Op1)) { | ||||
3555 | return ReplaceInstUsesWith(I, Op1); // X & <0,0> -> <0,0> | ||||
3556 | } | ||||
3557 | } | ||||
3558 | |||||
3559 | if (ConstantInt *AndRHS = dyn_cast<ConstantInt>(Op1)) { | ||||
3560 | const APInt& AndRHSMask = AndRHS->getValue(); | ||||
3561 | APInt NotAndRHS(~AndRHSMask); | ||||
3562 | |||||
3563 | // Optimize a variety of ((val OP C1) & C2) combinations... | ||||
3564 | if (isa<BinaryOperator>(Op0)) { | ||||
3565 | Instruction *Op0I = cast<Instruction>(Op0); | ||||
3566 | Value *Op0LHS = Op0I->getOperand(0); | ||||
3567 | Value *Op0RHS = Op0I->getOperand(1); | ||||
3568 | switch (Op0I->getOpcode()) { | ||||
3569 | case Instruction::Xor: | ||||
3570 | case Instruction::Or: | ||||
3571 | // If the mask is only needed on one incoming arm, push it up. | ||||
3572 | if (Op0I->hasOneUse()) { | ||||
3573 | if (MaskedValueIsZero(Op0LHS, NotAndRHS)) { | ||||
3574 | // Not masking anything out for the LHS, move to RHS. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3575 | Instruction *NewRHS = BinaryOperator::CreateAnd(Op0RHS, AndRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3576 | Op0RHS->getName()+".masked"); |
3577 | InsertNewInstBefore(NewRHS, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3578 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3579 | cast<BinaryOperator>(Op0I)->getOpcode(), Op0LHS, NewRHS); |
3580 | } | ||||
3581 | if (!isa<Constant>(Op0RHS) && | ||||
3582 | MaskedValueIsZero(Op0RHS, NotAndRHS)) { | ||||
3583 | // Not masking anything out for the RHS, move to LHS. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3584 | Instruction *NewLHS = BinaryOperator::CreateAnd(Op0LHS, AndRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3585 | Op0LHS->getName()+".masked"); |
3586 | InsertNewInstBefore(NewLHS, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3587 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3588 | cast<BinaryOperator>(Op0I)->getOpcode(), NewLHS, Op0RHS); |
3589 | } | ||||
3590 | } | ||||
3591 | |||||
3592 | break; | ||||
3593 | case Instruction::Add: | ||||
3594 | // ((A & N) + B) & AndRHS -> (A + B) & AndRHS iff N&AndRHS == AndRHS. | ||||
3595 | // ((A | N) + B) & AndRHS -> (A + B) & AndRHS iff N&AndRHS == 0 | ||||
3596 | // ((A ^ N) + B) & AndRHS -> (A + B) & AndRHS iff N&AndRHS == 0 | ||||
3597 | if (Value *V = FoldLogicalPlusAnd(Op0LHS, Op0RHS, AndRHS, false, I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3598 | return BinaryOperator::CreateAnd(V, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3599 | if (Value *V = FoldLogicalPlusAnd(Op0RHS, Op0LHS, AndRHS, false, I)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3600 | return BinaryOperator::CreateAnd(V, AndRHS); // Add commutes |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3601 | break; |
3602 | |||||
3603 | case Instruction::Sub: | ||||
3604 | // ((A & N) - B) & AndRHS -> (A - B) & AndRHS iff N&AndRHS == AndRHS. | ||||
3605 | // ((A | N) - B) & AndRHS -> (A - B) & AndRHS iff N&AndRHS == 0 | ||||
3606 | // ((A ^ N) - B) & AndRHS -> (A - B) & AndRHS iff N&AndRHS == 0 | ||||
3607 | if (Value *V = FoldLogicalPlusAnd(Op0LHS, Op0RHS, AndRHS, true, I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3608 | return BinaryOperator::CreateAnd(V, AndRHS); |
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 3609 | |
Nick Lewycky | a349ba4 | 2008-07-10 05:51:40 +0000 | [diff] [blame] | 3610 | // (A - N) & AndRHS -> -N & AndRHS iff A&AndRHS==0 and AndRHS |
3611 | // has 1's for all bits that the subtraction with A might affect. | ||||
3612 | if (Op0I->hasOneUse()) { | ||||
3613 | uint32_t BitWidth = AndRHSMask.getBitWidth(); | ||||
3614 | uint32_t Zeros = AndRHSMask.countLeadingZeros(); | ||||
3615 | APInt Mask = APInt::getLowBitsSet(BitWidth, BitWidth - Zeros); | ||||
3616 | |||||
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 3617 | ConstantInt *A = dyn_cast<ConstantInt>(Op0LHS); |
Nick Lewycky | a349ba4 | 2008-07-10 05:51:40 +0000 | [diff] [blame] | 3618 | if (!(A && A->isZero()) && // avoid infinite recursion. |
3619 | MaskedValueIsZero(Op0LHS, Mask)) { | ||||
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 3620 | Instruction *NewNeg = BinaryOperator::CreateNeg(Op0RHS); |
3621 | InsertNewInstBefore(NewNeg, I); | ||||
3622 | return BinaryOperator::CreateAnd(NewNeg, AndRHS); | ||||
3623 | } | ||||
3624 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3625 | break; |
Nick Lewycky | 659ed4d | 2008-07-09 05:20:13 +0000 | [diff] [blame] | 3626 | |
3627 | case Instruction::Shl: | ||||
3628 | case Instruction::LShr: | ||||
3629 | // (1 << x) & 1 --> zext(x == 0) | ||||
3630 | // (1 >> x) & 1 --> zext(x == 0) | ||||
Nick Lewycky | f1b1222 | 2008-07-09 07:35:26 +0000 | [diff] [blame] | 3631 | if (AndRHSMask == 1 && Op0LHS == AndRHS) { |
Nick Lewycky | 659ed4d | 2008-07-09 05:20:13 +0000 | [diff] [blame] | 3632 | Instruction *NewICmp = new ICmpInst(ICmpInst::ICMP_EQ, Op0RHS, |
3633 | Constant::getNullValue(I.getType())); | ||||
3634 | InsertNewInstBefore(NewICmp, I); | ||||
3635 | return new ZExtInst(NewICmp, I.getType()); | ||||
3636 | } | ||||
3637 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3638 | } |
3639 | |||||
3640 | if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) | ||||
3641 | if (Instruction *Res = OptAndOp(Op0I, Op0CI, AndRHS, I)) | ||||
3642 | return Res; | ||||
3643 | } else if (CastInst *CI = dyn_cast<CastInst>(Op0)) { | ||||
3644 | // If this is an integer truncation or change from signed-to-unsigned, and | ||||
3645 | // if the source is an and/or with immediate, transform it. This | ||||
3646 | // frequently occurs for bitfield accesses. | ||||
3647 | if (Instruction *CastOp = dyn_cast<Instruction>(CI->getOperand(0))) { | ||||
3648 | if ((isa<TruncInst>(CI) || isa<BitCastInst>(CI)) && | ||||
3649 | CastOp->getNumOperands() == 2) | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 3650 | if (ConstantInt *AndCI = dyn_cast<ConstantInt>(CastOp->getOperand(1))) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3651 | if (CastOp->getOpcode() == Instruction::And) { |
3652 | // Change: and (cast (and X, C1) to T), C2 | ||||
3653 | // into : and (cast X to T), trunc_or_bitcast(C1)&C2 | ||||
3654 | // This will fold the two constants together, which may allow | ||||
3655 | // other simplifications. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3656 | Instruction *NewCast = CastInst::CreateTruncOrBitCast( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3657 | CastOp->getOperand(0), I.getType(), |
3658 | CastOp->getName()+".shrunk"); | ||||
3659 | NewCast = InsertNewInstBefore(NewCast, I); | ||||
3660 | // trunc_or_bitcast(C1)&C2 | ||||
3661 | Constant *C3 = ConstantExpr::getTruncOrBitCast(AndCI,I.getType()); | ||||
3662 | C3 = ConstantExpr::getAnd(C3, AndRHS); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3663 | return BinaryOperator::CreateAnd(NewCast, C3); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3664 | } else if (CastOp->getOpcode() == Instruction::Or) { |
3665 | // Change: and (cast (or X, C1) to T), C2 | ||||
3666 | // into : trunc(C1)&C2 iff trunc(C1)&C2 == C2 | ||||
3667 | Constant *C3 = ConstantExpr::getTruncOrBitCast(AndCI,I.getType()); | ||||
3668 | if (ConstantExpr::getAnd(C3, AndRHS) == AndRHS) // trunc(C1)&C2 | ||||
3669 | return ReplaceInstUsesWith(I, AndRHS); | ||||
3670 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 3671 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3672 | } |
3673 | } | ||||
3674 | |||||
3675 | // Try to fold constant and into select arguments. | ||||
3676 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
3677 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
3678 | return R; | ||||
3679 | if (isa<PHINode>(Op0)) | ||||
3680 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
3681 | return NV; | ||||
3682 | } | ||||
3683 | |||||
3684 | Value *Op0NotVal = dyn_castNotVal(Op0); | ||||
3685 | Value *Op1NotVal = dyn_castNotVal(Op1); | ||||
3686 | |||||
3687 | if (Op0NotVal == Op1 || Op1NotVal == Op0) // A & ~A == ~A & A == 0 | ||||
3688 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
3689 | |||||
3690 | // (~A & ~B) == (~(A | B)) - De Morgan's Law | ||||
3691 | if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3692 | Instruction *Or = BinaryOperator::CreateOr(Op0NotVal, Op1NotVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3693 | I.getName()+".demorgan"); |
3694 | InsertNewInstBefore(Or, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3695 | return BinaryOperator::CreateNot(Or); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3696 | } |
3697 | |||||
3698 | { | ||||
3699 | Value *A = 0, *B = 0, *C = 0, *D = 0; | ||||
3700 | if (match(Op0, m_Or(m_Value(A), m_Value(B)))) { | ||||
3701 | if (A == Op1 || B == Op1) // (A | ?) & A --> A | ||||
3702 | return ReplaceInstUsesWith(I, Op1); | ||||
3703 | |||||
3704 | // (A|B) & ~(A&B) -> A^B | ||||
3705 | if (match(Op1, m_Not(m_And(m_Value(C), m_Value(D))))) { | ||||
3706 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3707 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3708 | } |
3709 | } | ||||
3710 | |||||
3711 | if (match(Op1, m_Or(m_Value(A), m_Value(B)))) { | ||||
3712 | if (A == Op0 || B == Op0) // A & (A | ?) --> A | ||||
3713 | return ReplaceInstUsesWith(I, Op0); | ||||
3714 | |||||
3715 | // ~(A&B) & (A|B) -> A^B | ||||
3716 | if (match(Op0, m_Not(m_And(m_Value(C), m_Value(D))))) { | ||||
3717 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3718 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3719 | } |
3720 | } | ||||
3721 | |||||
3722 | if (Op0->hasOneUse() && | ||||
3723 | match(Op0, m_Xor(m_Value(A), m_Value(B)))) { | ||||
3724 | if (A == Op1) { // (A^B)&A -> A&(A^B) | ||||
3725 | I.swapOperands(); // Simplify below | ||||
3726 | std::swap(Op0, Op1); | ||||
3727 | } else if (B == Op1) { // (A^B)&B -> B&(B^A) | ||||
3728 | cast<BinaryOperator>(Op0)->swapOperands(); | ||||
3729 | I.swapOperands(); // Simplify below | ||||
3730 | std::swap(Op0, Op1); | ||||
3731 | } | ||||
3732 | } | ||||
3733 | if (Op1->hasOneUse() && | ||||
3734 | match(Op1, m_Xor(m_Value(A), m_Value(B)))) { | ||||
3735 | if (B == Op0) { // B&(A^B) -> B&(B^A) | ||||
3736 | cast<BinaryOperator>(Op1)->swapOperands(); | ||||
3737 | std::swap(A, B); | ||||
3738 | } | ||||
3739 | if (A == Op0) { // A&(A^B) -> A & ~B | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3740 | Instruction *NotB = BinaryOperator::CreateNot(B, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3741 | InsertNewInstBefore(NotB, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3742 | return BinaryOperator::CreateAnd(A, NotB); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3743 | } |
3744 | } | ||||
3745 | } | ||||
3746 | |||||
Nick Lewycky | 771d605 | 2008-08-06 04:54:03 +0000 | [diff] [blame] | 3747 | { // (icmp ult A, C) & (icmp ult B, C) --> (icmp ult (A|B), C) |
3748 | // where C is a power of 2 | ||||
3749 | Value *A, *B; | ||||
3750 | ConstantInt *C1, *C2; | ||||
Evan Cheng | 94fd974 | 2008-08-20 23:36:48 +0000 | [diff] [blame] | 3751 | ICmpInst::Predicate LHSCC = ICmpInst::BAD_ICMP_PREDICATE; |
3752 | ICmpInst::Predicate RHSCC = ICmpInst::BAD_ICMP_PREDICATE; | ||||
Nick Lewycky | 771d605 | 2008-08-06 04:54:03 +0000 | [diff] [blame] | 3753 | if (match(&I, m_And(m_ICmp(LHSCC, m_Value(A), m_ConstantInt(C1)), |
3754 | m_ICmp(RHSCC, m_Value(B), m_ConstantInt(C2))))) | ||||
3755 | if (C1 == C2 && LHSCC == RHSCC && LHSCC == ICmpInst::ICMP_ULT && | ||||
3756 | C1->getValue().isPowerOf2()) { | ||||
3757 | Instruction *NewOr = BinaryOperator::CreateOr(A, B); | ||||
3758 | InsertNewInstBefore(NewOr, I); | ||||
3759 | return new ICmpInst(LHSCC, NewOr, C1); | ||||
3760 | } | ||||
3761 | } | ||||
3762 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3763 | if (ICmpInst *RHS = dyn_cast<ICmpInst>(Op1)) { |
3764 | // (icmp1 A, B) & (icmp2 A, B) --> (icmp3 A, B) | ||||
3765 | if (Instruction *R = AssociativeOpt(I, FoldICmpLogical(*this, RHS))) | ||||
3766 | return R; | ||||
3767 | |||||
3768 | Value *LHSVal, *RHSVal; | ||||
3769 | ConstantInt *LHSCst, *RHSCst; | ||||
3770 | ICmpInst::Predicate LHSCC, RHSCC; | ||||
3771 | if (match(Op0, m_ICmp(LHSCC, m_Value(LHSVal), m_ConstantInt(LHSCst)))) | ||||
3772 | if (match(RHS, m_ICmp(RHSCC, m_Value(RHSVal), m_ConstantInt(RHSCst)))) | ||||
3773 | if (LHSVal == RHSVal && // Found (X icmp C1) & (X icmp C2) | ||||
3774 | // ICMP_[GL]E X, CST is folded to ICMP_[GL]T elsewhere. | ||||
3775 | LHSCC != ICmpInst::ICMP_UGE && LHSCC != ICmpInst::ICMP_ULE && | ||||
3776 | RHSCC != ICmpInst::ICMP_UGE && RHSCC != ICmpInst::ICMP_ULE && | ||||
3777 | LHSCC != ICmpInst::ICMP_SGE && LHSCC != ICmpInst::ICMP_SLE && | ||||
Chris Lattner | 205ad1d | 2007-11-22 23:47:13 +0000 | [diff] [blame] | 3778 | RHSCC != ICmpInst::ICMP_SGE && RHSCC != ICmpInst::ICMP_SLE && |
3779 | |||||
3780 | // Don't try to fold ICMP_SLT + ICMP_ULT. | ||||
3781 | (ICmpInst::isEquality(LHSCC) || ICmpInst::isEquality(RHSCC) || | ||||
3782 | ICmpInst::isSignedPredicate(LHSCC) == | ||||
3783 | ICmpInst::isSignedPredicate(RHSCC))) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3784 | // Ensure that the larger constant is on the RHS. |
Chris Lattner | da628ca | 2008-01-13 20:59:02 +0000 | [diff] [blame] | 3785 | ICmpInst::Predicate GT; |
3786 | if (ICmpInst::isSignedPredicate(LHSCC) || | ||||
3787 | (ICmpInst::isEquality(LHSCC) && | ||||
3788 | ICmpInst::isSignedPredicate(RHSCC))) | ||||
3789 | GT = ICmpInst::ICMP_SGT; | ||||
3790 | else | ||||
3791 | GT = ICmpInst::ICMP_UGT; | ||||
3792 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3793 | Constant *Cmp = ConstantExpr::getICmp(GT, LHSCst, RHSCst); |
3794 | ICmpInst *LHS = cast<ICmpInst>(Op0); | ||||
3795 | if (cast<ConstantInt>(Cmp)->getZExtValue()) { | ||||
3796 | std::swap(LHS, RHS); | ||||
3797 | std::swap(LHSCst, RHSCst); | ||||
3798 | std::swap(LHSCC, RHSCC); | ||||
3799 | } | ||||
3800 | |||||
3801 | // At this point, we know we have have two icmp instructions | ||||
3802 | // comparing a value against two constants and and'ing the result | ||||
3803 | // together. Because of the above check, we know that we only have | ||||
3804 | // icmp eq, icmp ne, icmp [su]lt, and icmp [SU]gt here. We also know | ||||
3805 | // (from the FoldICmpLogical check above), that the two constants | ||||
3806 | // are not equal and that the larger constant is on the RHS | ||||
3807 | assert(LHSCst != RHSCst && "Compares not folded above?"); | ||||
3808 | |||||
3809 | switch (LHSCC) { | ||||
3810 | default: assert(0 && "Unknown integer condition code!"); | ||||
3811 | case ICmpInst::ICMP_EQ: | ||||
3812 | switch (RHSCC) { | ||||
3813 | default: assert(0 && "Unknown integer condition code!"); | ||||
3814 | case ICmpInst::ICMP_EQ: // (X == 13 & X == 15) -> false | ||||
3815 | case ICmpInst::ICMP_UGT: // (X == 13 & X > 15) -> false | ||||
3816 | case ICmpInst::ICMP_SGT: // (X == 13 & X > 15) -> false | ||||
3817 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
3818 | case ICmpInst::ICMP_NE: // (X == 13 & X != 15) -> X == 13 | ||||
3819 | case ICmpInst::ICMP_ULT: // (X == 13 & X < 15) -> X == 13 | ||||
3820 | case ICmpInst::ICMP_SLT: // (X == 13 & X < 15) -> X == 13 | ||||
3821 | return ReplaceInstUsesWith(I, LHS); | ||||
3822 | } | ||||
3823 | case ICmpInst::ICMP_NE: | ||||
3824 | switch (RHSCC) { | ||||
3825 | default: assert(0 && "Unknown integer condition code!"); | ||||
3826 | case ICmpInst::ICMP_ULT: | ||||
3827 | if (LHSCst == SubOne(RHSCst)) // (X != 13 & X u< 14) -> X < 13 | ||||
3828 | return new ICmpInst(ICmpInst::ICMP_ULT, LHSVal, LHSCst); | ||||
3829 | break; // (X != 13 & X u< 15) -> no change | ||||
3830 | case ICmpInst::ICMP_SLT: | ||||
3831 | if (LHSCst == SubOne(RHSCst)) // (X != 13 & X s< 14) -> X < 13 | ||||
3832 | return new ICmpInst(ICmpInst::ICMP_SLT, LHSVal, LHSCst); | ||||
3833 | break; // (X != 13 & X s< 15) -> no change | ||||
3834 | case ICmpInst::ICMP_EQ: // (X != 13 & X == 15) -> X == 15 | ||||
3835 | case ICmpInst::ICMP_UGT: // (X != 13 & X u> 15) -> X u> 15 | ||||
3836 | case ICmpInst::ICMP_SGT: // (X != 13 & X s> 15) -> X s> 15 | ||||
3837 | return ReplaceInstUsesWith(I, RHS); | ||||
3838 | case ICmpInst::ICMP_NE: | ||||
3839 | if (LHSCst == SubOne(RHSCst)){// (X != 13 & X != 14) -> X-13 >u 1 | ||||
3840 | Constant *AddCST = ConstantExpr::getNeg(LHSCst); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3841 | Instruction *Add = BinaryOperator::CreateAdd(LHSVal, AddCST, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3842 | LHSVal->getName()+".off"); |
3843 | InsertNewInstBefore(Add, I); | ||||
3844 | return new ICmpInst(ICmpInst::ICMP_UGT, Add, | ||||
3845 | ConstantInt::get(Add->getType(), 1)); | ||||
3846 | } | ||||
3847 | break; // (X != 13 & X != 15) -> no change | ||||
3848 | } | ||||
3849 | break; | ||||
3850 | case ICmpInst::ICMP_ULT: | ||||
3851 | switch (RHSCC) { | ||||
3852 | default: assert(0 && "Unknown integer condition code!"); | ||||
3853 | case ICmpInst::ICMP_EQ: // (X u< 13 & X == 15) -> false | ||||
3854 | case ICmpInst::ICMP_UGT: // (X u< 13 & X u> 15) -> false | ||||
3855 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
3856 | case ICmpInst::ICMP_SGT: // (X u< 13 & X s> 15) -> no change | ||||
3857 | break; | ||||
3858 | case ICmpInst::ICMP_NE: // (X u< 13 & X != 15) -> X u< 13 | ||||
3859 | case ICmpInst::ICMP_ULT: // (X u< 13 & X u< 15) -> X u< 13 | ||||
3860 | return ReplaceInstUsesWith(I, LHS); | ||||
3861 | case ICmpInst::ICMP_SLT: // (X u< 13 & X s< 15) -> no change | ||||
3862 | break; | ||||
3863 | } | ||||
3864 | break; | ||||
3865 | case ICmpInst::ICMP_SLT: | ||||
3866 | switch (RHSCC) { | ||||
3867 | default: assert(0 && "Unknown integer condition code!"); | ||||
3868 | case ICmpInst::ICMP_EQ: // (X s< 13 & X == 15) -> false | ||||
3869 | case ICmpInst::ICMP_SGT: // (X s< 13 & X s> 15) -> false | ||||
3870 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
3871 | case ICmpInst::ICMP_UGT: // (X s< 13 & X u> 15) -> no change | ||||
3872 | break; | ||||
3873 | case ICmpInst::ICMP_NE: // (X s< 13 & X != 15) -> X < 13 | ||||
3874 | case ICmpInst::ICMP_SLT: // (X s< 13 & X s< 15) -> X < 13 | ||||
3875 | return ReplaceInstUsesWith(I, LHS); | ||||
3876 | case ICmpInst::ICMP_ULT: // (X s< 13 & X u< 15) -> no change | ||||
3877 | break; | ||||
3878 | } | ||||
3879 | break; | ||||
3880 | case ICmpInst::ICMP_UGT: | ||||
3881 | switch (RHSCC) { | ||||
3882 | default: assert(0 && "Unknown integer condition code!"); | ||||
Eli Friedman | 22b8562 | 2008-06-21 23:36:13 +0000 | [diff] [blame] | 3883 | case ICmpInst::ICMP_EQ: // (X u> 13 & X == 15) -> X == 15 |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3884 | case ICmpInst::ICMP_UGT: // (X u> 13 & X u> 15) -> X u> 15 |
3885 | return ReplaceInstUsesWith(I, RHS); | ||||
3886 | case ICmpInst::ICMP_SGT: // (X u> 13 & X s> 15) -> no change | ||||
3887 | break; | ||||
3888 | case ICmpInst::ICMP_NE: | ||||
3889 | if (RHSCst == AddOne(LHSCst)) // (X u> 13 & X != 14) -> X u> 14 | ||||
3890 | return new ICmpInst(LHSCC, LHSVal, RHSCst); | ||||
3891 | break; // (X u> 13 & X != 15) -> no change | ||||
3892 | case ICmpInst::ICMP_ULT: // (X u> 13 & X u< 15) ->(X-14) <u 1 | ||||
3893 | return InsertRangeTest(LHSVal, AddOne(LHSCst), RHSCst, false, | ||||
3894 | true, I); | ||||
3895 | case ICmpInst::ICMP_SLT: // (X u> 13 & X s< 15) -> no change | ||||
3896 | break; | ||||
3897 | } | ||||
3898 | break; | ||||
3899 | case ICmpInst::ICMP_SGT: | ||||
3900 | switch (RHSCC) { | ||||
3901 | default: assert(0 && "Unknown integer condition code!"); | ||||
Chris Lattner | ab0fc25 | 2007-11-16 06:04:17 +0000 | [diff] [blame] | 3902 | case ICmpInst::ICMP_EQ: // (X s> 13 & X == 15) -> X == 15 |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3903 | case ICmpInst::ICMP_SGT: // (X s> 13 & X s> 15) -> X s> 15 |
3904 | return ReplaceInstUsesWith(I, RHS); | ||||
3905 | case ICmpInst::ICMP_UGT: // (X s> 13 & X u> 15) -> no change | ||||
3906 | break; | ||||
3907 | case ICmpInst::ICMP_NE: | ||||
3908 | if (RHSCst == AddOne(LHSCst)) // (X s> 13 & X != 14) -> X s> 14 | ||||
3909 | return new ICmpInst(LHSCC, LHSVal, RHSCst); | ||||
3910 | break; // (X s> 13 & X != 15) -> no change | ||||
3911 | case ICmpInst::ICMP_SLT: // (X s> 13 & X s< 15) ->(X-14) s< 1 | ||||
3912 | return InsertRangeTest(LHSVal, AddOne(LHSCst), RHSCst, true, | ||||
3913 | true, I); | ||||
3914 | case ICmpInst::ICMP_ULT: // (X s> 13 & X u< 15) -> no change | ||||
3915 | break; | ||||
3916 | } | ||||
3917 | break; | ||||
3918 | } | ||||
3919 | } | ||||
3920 | } | ||||
3921 | |||||
3922 | // fold (and (cast A), (cast B)) -> (cast (and A, B)) | ||||
3923 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) | ||||
3924 | if (CastInst *Op1C = dyn_cast<CastInst>(Op1)) | ||||
3925 | if (Op0C->getOpcode() == Op1C->getOpcode()) { // same cast kind ? | ||||
3926 | const Type *SrcTy = Op0C->getOperand(0)->getType(); | ||||
3927 | if (SrcTy == Op1C->getOperand(0)->getType() && SrcTy->isInteger() && | ||||
3928 | // Only do this if the casts both really cause code to be generated. | ||||
3929 | ValueRequiresCast(Op0C->getOpcode(), Op0C->getOperand(0), | ||||
3930 | I.getType(), TD) && | ||||
3931 | ValueRequiresCast(Op1C->getOpcode(), Op1C->getOperand(0), | ||||
3932 | I.getType(), TD)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3933 | Instruction *NewOp = BinaryOperator::CreateAnd(Op0C->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3934 | Op1C->getOperand(0), |
3935 | I.getName()); | ||||
3936 | InsertNewInstBefore(NewOp, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3937 | return CastInst::Create(Op0C->getOpcode(), NewOp, I.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3938 | } |
3939 | } | ||||
3940 | |||||
3941 | // (X >> Z) & (Y >> Z) -> (X&Y) >> Z for all shifts. | ||||
3942 | if (BinaryOperator *SI1 = dyn_cast<BinaryOperator>(Op1)) { | ||||
3943 | if (BinaryOperator *SI0 = dyn_cast<BinaryOperator>(Op0)) | ||||
3944 | if (SI0->isShift() && SI0->getOpcode() == SI1->getOpcode() && | ||||
3945 | SI0->getOperand(1) == SI1->getOperand(1) && | ||||
3946 | (SI0->hasOneUse() || SI1->hasOneUse())) { | ||||
3947 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3948 | InsertNewInstBefore(BinaryOperator::CreateAnd(SI0->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3949 | SI1->getOperand(0), |
3950 | SI0->getName()), I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3951 | return BinaryOperator::Create(SI1->getOpcode(), NewOp, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3952 | SI1->getOperand(1)); |
3953 | } | ||||
3954 | } | ||||
3955 | |||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame^] | 3956 | // If and'ing two fcmp, try combine them into one. |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 3957 | if (FCmpInst *LHS = dyn_cast<FCmpInst>(I.getOperand(0))) { |
3958 | if (FCmpInst *RHS = dyn_cast<FCmpInst>(I.getOperand(1))) { | ||||
3959 | if (LHS->getPredicate() == FCmpInst::FCMP_ORD && | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame^] | 3960 | RHS->getPredicate() == FCmpInst::FCMP_ORD) { |
3961 | // (fcmp ord x, c) & (fcmp ord y, c) -> (fcmp ord x, y) | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 3962 | if (ConstantFP *LHSC = dyn_cast<ConstantFP>(LHS->getOperand(1))) |
3963 | if (ConstantFP *RHSC = dyn_cast<ConstantFP>(RHS->getOperand(1))) { | ||||
3964 | // If either of the constants are nans, then the whole thing returns | ||||
3965 | // false. | ||||
Chris Lattner | a6c7dce | 2007-10-24 18:54:45 +0000 | [diff] [blame] | 3966 | if (LHSC->getValueAPF().isNaN() || RHSC->getValueAPF().isNaN()) |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 3967 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
3968 | return new FCmpInst(FCmpInst::FCMP_ORD, LHS->getOperand(0), | ||||
3969 | RHS->getOperand(0)); | ||||
3970 | } | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame^] | 3971 | } else { |
3972 | Value *Op0LHS, *Op0RHS, *Op1LHS, *Op1RHS; | ||||
3973 | FCmpInst::Predicate Op0CC, Op1CC; | ||||
3974 | if (match(Op0, m_FCmp(Op0CC, m_Value(Op0LHS), m_Value(Op0RHS))) && | ||||
3975 | match(Op1, m_FCmp(Op1CC, m_Value(Op1LHS), m_Value(Op1RHS)))) { | ||||
3976 | if (Op0LHS == Op1LHS && Op0RHS == Op1RHS) { | ||||
3977 | // Simplify (fcmp cc0 x, y) & (fcmp cc1 x, y). | ||||
3978 | if (Op0CC == Op1CC) | ||||
3979 | return new FCmpInst((FCmpInst::Predicate)Op0CC, Op0LHS, Op0RHS); | ||||
3980 | else if (Op0CC == FCmpInst::FCMP_FALSE || | ||||
3981 | Op1CC == FCmpInst::FCMP_FALSE) | ||||
3982 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
3983 | else if (Op0CC == FCmpInst::FCMP_TRUE) | ||||
3984 | return ReplaceInstUsesWith(I, Op1); | ||||
3985 | else if (Op1CC == FCmpInst::FCMP_TRUE) | ||||
3986 | return ReplaceInstUsesWith(I, Op0); | ||||
3987 | bool Op0Ordered; | ||||
3988 | bool Op1Ordered; | ||||
3989 | unsigned Op0Pred = getFCmpCode(Op0CC, Op0Ordered); | ||||
3990 | unsigned Op1Pred = getFCmpCode(Op1CC, Op1Ordered); | ||||
3991 | if (Op1Pred == 0) { | ||||
3992 | std::swap(Op0, Op1); | ||||
3993 | std::swap(Op0Pred, Op1Pred); | ||||
3994 | std::swap(Op0Ordered, Op1Ordered); | ||||
3995 | } | ||||
3996 | if (Op0Pred == 0) { | ||||
3997 | // uno && ueq -> uno && (uno || eq) -> ueq | ||||
3998 | // ord && olt -> ord && (ord && lt) -> olt | ||||
3999 | if (Op0Ordered == Op1Ordered) | ||||
4000 | return ReplaceInstUsesWith(I, Op1); | ||||
4001 | // uno && oeq -> uno && (ord && eq) -> false | ||||
4002 | // uno && ord -> false | ||||
4003 | if (!Op0Ordered) | ||||
4004 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
4005 | // ord && ueq -> ord && (uno || eq) -> oeq | ||||
4006 | return cast<Instruction>(getFCmpValue(true, Op1Pred, | ||||
4007 | Op0LHS, Op0RHS)); | ||||
4008 | } | ||||
4009 | } | ||||
4010 | } | ||||
4011 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4012 | } |
4013 | } | ||||
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 4014 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4015 | return Changed ? &I : 0; |
4016 | } | ||||
4017 | |||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4018 | /// CollectBSwapParts - Analyze the specified subexpression and see if it is |
4019 | /// capable of providing pieces of a bswap. The subexpression provides pieces | ||||
4020 | /// of a bswap if it is proven that each of the non-zero bytes in the output of | ||||
4021 | /// the expression came from the corresponding "byte swapped" byte in some other | ||||
4022 | /// value. For example, if the current subexpression is "(shl i32 %X, 24)" then | ||||
4023 | /// we know that the expression deposits the low byte of %X into the high byte | ||||
4024 | /// of the bswap result and that all other bytes are zero. This expression is | ||||
4025 | /// accepted, the high byte of ByteValues is set to X to indicate a correct | ||||
4026 | /// match. | ||||
4027 | /// | ||||
4028 | /// This function returns true if the match was unsuccessful and false if so. | ||||
4029 | /// On entry to the function the "OverallLeftShift" is a signed integer value | ||||
4030 | /// indicating the number of bytes that the subexpression is later shifted. For | ||||
4031 | /// example, if the expression is later right shifted by 16 bits, the | ||||
4032 | /// OverallLeftShift value would be -2 on entry. This is used to specify which | ||||
4033 | /// byte of ByteValues is actually being set. | ||||
4034 | /// | ||||
4035 | /// Similarly, ByteMask is a bitmask where a bit is clear if its corresponding | ||||
4036 | /// byte is masked to zero by a user. For example, in (X & 255), X will be | ||||
4037 | /// processed with a bytemask of 1. Because bytemask is 32-bits, this limits | ||||
4038 | /// this function to working on up to 32-byte (256 bit) values. ByteMask is | ||||
4039 | /// always in the local (OverallLeftShift) coordinate space. | ||||
4040 | /// | ||||
4041 | static bool CollectBSwapParts(Value *V, int OverallLeftShift, uint32_t ByteMask, | ||||
4042 | SmallVector<Value*, 8> &ByteValues) { | ||||
4043 | if (Instruction *I = dyn_cast<Instruction>(V)) { | ||||
4044 | // If this is an or instruction, it may be an inner node of the bswap. | ||||
4045 | if (I->getOpcode() == Instruction::Or) { | ||||
4046 | return CollectBSwapParts(I->getOperand(0), OverallLeftShift, ByteMask, | ||||
4047 | ByteValues) || | ||||
4048 | CollectBSwapParts(I->getOperand(1), OverallLeftShift, ByteMask, | ||||
4049 | ByteValues); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4050 | } |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4051 | |
4052 | // If this is a logical shift by a constant multiple of 8, recurse with | ||||
4053 | // OverallLeftShift and ByteMask adjusted. | ||||
4054 | if (I->isLogicalShift() && isa<ConstantInt>(I->getOperand(1))) { | ||||
4055 | unsigned ShAmt = | ||||
4056 | cast<ConstantInt>(I->getOperand(1))->getLimitedValue(~0U); | ||||
4057 | // Ensure the shift amount is defined and of a byte value. | ||||
4058 | if ((ShAmt & 7) || (ShAmt > 8*ByteValues.size())) | ||||
4059 | return true; | ||||
4060 | |||||
4061 | unsigned ByteShift = ShAmt >> 3; | ||||
4062 | if (I->getOpcode() == Instruction::Shl) { | ||||
4063 | // X << 2 -> collect(X, +2) | ||||
4064 | OverallLeftShift += ByteShift; | ||||
4065 | ByteMask >>= ByteShift; | ||||
4066 | } else { | ||||
4067 | // X >>u 2 -> collect(X, -2) | ||||
4068 | OverallLeftShift -= ByteShift; | ||||
4069 | ByteMask <<= ByteShift; | ||||
Chris Lattner | 4444859 | 2008-10-08 06:42:28 +0000 | [diff] [blame] | 4070 | ByteMask &= (~0U >> (32-ByteValues.size())); |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4071 | } |
4072 | |||||
4073 | if (OverallLeftShift >= (int)ByteValues.size()) return true; | ||||
4074 | if (OverallLeftShift <= -(int)ByteValues.size()) return true; | ||||
4075 | |||||
4076 | return CollectBSwapParts(I->getOperand(0), OverallLeftShift, ByteMask, | ||||
4077 | ByteValues); | ||||
4078 | } | ||||
4079 | |||||
4080 | // If this is a logical 'and' with a mask that clears bytes, clear the | ||||
4081 | // corresponding bytes in ByteMask. | ||||
4082 | if (I->getOpcode() == Instruction::And && | ||||
4083 | isa<ConstantInt>(I->getOperand(1))) { | ||||
4084 | // Scan every byte of the and mask, seeing if the byte is either 0 or 255. | ||||
4085 | unsigned NumBytes = ByteValues.size(); | ||||
4086 | APInt Byte(I->getType()->getPrimitiveSizeInBits(), 255); | ||||
4087 | const APInt &AndMask = cast<ConstantInt>(I->getOperand(1))->getValue(); | ||||
4088 | |||||
4089 | for (unsigned i = 0; i != NumBytes; ++i, Byte <<= 8) { | ||||
4090 | // If this byte is masked out by a later operation, we don't care what | ||||
4091 | // the and mask is. | ||||
4092 | if ((ByteMask & (1 << i)) == 0) | ||||
4093 | continue; | ||||
4094 | |||||
4095 | // If the AndMask is all zeros for this byte, clear the bit. | ||||
4096 | APInt MaskB = AndMask & Byte; | ||||
4097 | if (MaskB == 0) { | ||||
4098 | ByteMask &= ~(1U << i); | ||||
4099 | continue; | ||||
4100 | } | ||||
4101 | |||||
4102 | // If the AndMask is not all ones for this byte, it's not a bytezap. | ||||
4103 | if (MaskB != Byte) | ||||
4104 | return true; | ||||
4105 | |||||
4106 | // Otherwise, this byte is kept. | ||||
4107 | } | ||||
4108 | |||||
4109 | return CollectBSwapParts(I->getOperand(0), OverallLeftShift, ByteMask, | ||||
4110 | ByteValues); | ||||
4111 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4112 | } |
4113 | |||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4114 | // Okay, we got to something that isn't a shift, 'or' or 'and'. This must be |
4115 | // the input value to the bswap. Some observations: 1) if more than one byte | ||||
4116 | // is demanded from this input, then it could not be successfully assembled | ||||
4117 | // into a byteswap. At least one of the two bytes would not be aligned with | ||||
4118 | // their ultimate destination. | ||||
4119 | if (!isPowerOf2_32(ByteMask)) return true; | ||||
4120 | unsigned InputByteNo = CountTrailingZeros_32(ByteMask); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4121 | |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4122 | // 2) The input and ultimate destinations must line up: if byte 3 of an i32 |
4123 | // is demanded, it needs to go into byte 0 of the result. This means that the | ||||
4124 | // byte needs to be shifted until it lands in the right byte bucket. The | ||||
4125 | // shift amount depends on the position: if the byte is coming from the high | ||||
4126 | // part of the value (e.g. byte 3) then it must be shifted right. If from the | ||||
4127 | // low part, it must be shifted left. | ||||
4128 | unsigned DestByteNo = InputByteNo + OverallLeftShift; | ||||
4129 | if (InputByteNo < ByteValues.size()/2) { | ||||
4130 | if (ByteValues.size()-1-DestByteNo != InputByteNo) | ||||
4131 | return true; | ||||
4132 | } else { | ||||
4133 | if (ByteValues.size()-1-DestByteNo != InputByteNo) | ||||
4134 | return true; | ||||
4135 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4136 | |
4137 | // If the destination byte value is already defined, the values are or'd | ||||
4138 | // 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] | 4139 | if (ByteValues[DestByteNo] && ByteValues[DestByteNo] != V) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4140 | return true; |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4141 | ByteValues[DestByteNo] = V; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4142 | return false; |
4143 | } | ||||
4144 | |||||
4145 | /// MatchBSwap - Given an OR instruction, check to see if this is a bswap idiom. | ||||
4146 | /// If so, insert the new bswap intrinsic and return it. | ||||
4147 | Instruction *InstCombiner::MatchBSwap(BinaryOperator &I) { | ||||
4148 | const IntegerType *ITy = dyn_cast<IntegerType>(I.getType()); | ||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4149 | if (!ITy || ITy->getBitWidth() % 16 || |
4150 | // ByteMask only allows up to 32-byte values. | ||||
4151 | ITy->getBitWidth() > 32*8) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4152 | return 0; // Can only bswap pairs of bytes. Can't do vectors. |
4153 | |||||
4154 | /// ByteValues - For each byte of the result, we keep track of which value | ||||
4155 | /// defines each byte. | ||||
4156 | SmallVector<Value*, 8> ByteValues; | ||||
4157 | ByteValues.resize(ITy->getBitWidth()/8); | ||||
4158 | |||||
4159 | // Try to find all the pieces corresponding to the bswap. | ||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4160 | uint32_t ByteMask = ~0U >> (32-ByteValues.size()); |
4161 | if (CollectBSwapParts(&I, 0, ByteMask, ByteValues)) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4162 | return 0; |
4163 | |||||
4164 | // Check to see if all of the bytes come from the same value. | ||||
4165 | Value *V = ByteValues[0]; | ||||
4166 | if (V == 0) return 0; // Didn't find a byte? Must be zero. | ||||
4167 | |||||
4168 | // Check to make sure that all of the bytes come from the same value. | ||||
4169 | for (unsigned i = 1, e = ByteValues.size(); i != e; ++i) | ||||
4170 | if (ByteValues[i] != V) | ||||
4171 | return 0; | ||||
Chandler Carruth | a228e39 | 2007-08-04 01:51:18 +0000 | [diff] [blame] | 4172 | const Type *Tys[] = { ITy }; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4173 | Module *M = I.getParent()->getParent()->getParent(); |
Chandler Carruth | a228e39 | 2007-08-04 01:51:18 +0000 | [diff] [blame] | 4174 | Function *F = Intrinsic::getDeclaration(M, Intrinsic::bswap, Tys, 1); |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 4175 | return CallInst::Create(F, V); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4176 | } |
4177 | |||||
4178 | |||||
4179 | Instruction *InstCombiner::visitOr(BinaryOperator &I) { | ||||
4180 | bool Changed = SimplifyCommutative(I); | ||||
4181 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
4182 | |||||
4183 | if (isa<UndefValue>(Op1)) // X | undef -> -1 | ||||
4184 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4185 | |||||
4186 | // or X, X = X | ||||
4187 | if (Op0 == Op1) | ||||
4188 | return ReplaceInstUsesWith(I, Op0); | ||||
4189 | |||||
4190 | // See if we can simplify any instructions used by the instruction whose sole | ||||
4191 | // purpose is to compute bits we don't care about. | ||||
4192 | if (!isa<VectorType>(I.getType())) { | ||||
4193 | uint32_t BitWidth = cast<IntegerType>(I.getType())->getBitWidth(); | ||||
4194 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
4195 | if (SimplifyDemandedBits(&I, APInt::getAllOnesValue(BitWidth), | ||||
4196 | KnownZero, KnownOne)) | ||||
4197 | return &I; | ||||
4198 | } else if (isa<ConstantAggregateZero>(Op1)) { | ||||
4199 | return ReplaceInstUsesWith(I, Op0); // X | <0,0> -> X | ||||
4200 | } else if (ConstantVector *CP = dyn_cast<ConstantVector>(Op1)) { | ||||
4201 | if (CP->isAllOnesValue()) // X | <-1,-1> -> <-1,-1> | ||||
4202 | return ReplaceInstUsesWith(I, I.getOperand(1)); | ||||
4203 | } | ||||
4204 | |||||
4205 | |||||
4206 | |||||
4207 | // or X, -1 == -1 | ||||
4208 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
4209 | ConstantInt *C1 = 0; Value *X = 0; | ||||
4210 | // (X & C1) | C2 --> (X | C2) & (C1|C2) | ||||
4211 | 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] | 4212 | Instruction *Or = BinaryOperator::CreateOr(X, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4213 | InsertNewInstBefore(Or, I); |
4214 | Or->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4215 | return BinaryOperator::CreateAnd(Or, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4216 | ConstantInt::get(RHS->getValue() | C1->getValue())); |
4217 | } | ||||
4218 | |||||
4219 | // (X ^ C1) | C2 --> (X | C2) ^ (C1&~C2) | ||||
4220 | 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] | 4221 | Instruction *Or = BinaryOperator::CreateOr(X, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4222 | InsertNewInstBefore(Or, I); |
4223 | Or->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4224 | return BinaryOperator::CreateXor(Or, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4225 | ConstantInt::get(C1->getValue() & ~RHS->getValue())); |
4226 | } | ||||
4227 | |||||
4228 | // Try to fold constant and into select arguments. | ||||
4229 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
4230 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
4231 | return R; | ||||
4232 | if (isa<PHINode>(Op0)) | ||||
4233 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
4234 | return NV; | ||||
4235 | } | ||||
4236 | |||||
4237 | Value *A = 0, *B = 0; | ||||
4238 | ConstantInt *C1 = 0, *C2 = 0; | ||||
4239 | |||||
4240 | if (match(Op0, m_And(m_Value(A), m_Value(B)))) | ||||
4241 | if (A == Op1 || B == Op1) // (A & ?) | A --> A | ||||
4242 | return ReplaceInstUsesWith(I, Op1); | ||||
4243 | if (match(Op1, m_And(m_Value(A), m_Value(B)))) | ||||
4244 | if (A == Op0 || B == Op0) // A | (A & ?) --> A | ||||
4245 | return ReplaceInstUsesWith(I, Op0); | ||||
4246 | |||||
4247 | // (A | B) | C and A | (B | C) -> bswap if possible. | ||||
4248 | // (A >> B) | (C << D) and (A << B) | (B >> C) -> bswap if possible. | ||||
4249 | if (match(Op0, m_Or(m_Value(), m_Value())) || | ||||
4250 | match(Op1, m_Or(m_Value(), m_Value())) || | ||||
4251 | (match(Op0, m_Shift(m_Value(), m_Value())) && | ||||
4252 | match(Op1, m_Shift(m_Value(), m_Value())))) { | ||||
4253 | if (Instruction *BSwap = MatchBSwap(I)) | ||||
4254 | return BSwap; | ||||
4255 | } | ||||
4256 | |||||
4257 | // (X^C)|Y -> (X|Y)^C iff Y&C == 0 | ||||
4258 | if (Op0->hasOneUse() && match(Op0, m_Xor(m_Value(A), m_ConstantInt(C1))) && | ||||
4259 | MaskedValueIsZero(Op1, C1->getValue())) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4260 | Instruction *NOr = BinaryOperator::CreateOr(A, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4261 | InsertNewInstBefore(NOr, I); |
4262 | NOr->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4263 | return BinaryOperator::CreateXor(NOr, C1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4264 | } |
4265 | |||||
4266 | // Y|(X^C) -> (X|Y)^C iff Y&C == 0 | ||||
4267 | if (Op1->hasOneUse() && match(Op1, m_Xor(m_Value(A), m_ConstantInt(C1))) && | ||||
4268 | MaskedValueIsZero(Op0, C1->getValue())) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4269 | Instruction *NOr = BinaryOperator::CreateOr(A, Op0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4270 | InsertNewInstBefore(NOr, I); |
4271 | NOr->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4272 | return BinaryOperator::CreateXor(NOr, C1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4273 | } |
4274 | |||||
4275 | // (A & C)|(B & D) | ||||
4276 | Value *C = 0, *D = 0; | ||||
4277 | if (match(Op0, m_And(m_Value(A), m_Value(C))) && | ||||
4278 | match(Op1, m_And(m_Value(B), m_Value(D)))) { | ||||
4279 | Value *V1 = 0, *V2 = 0, *V3 = 0; | ||||
4280 | C1 = dyn_cast<ConstantInt>(C); | ||||
4281 | C2 = dyn_cast<ConstantInt>(D); | ||||
4282 | if (C1 && C2) { // (A & C1)|(B & C2) | ||||
4283 | // If we have: ((V + N) & C1) | (V & C2) | ||||
4284 | // .. and C2 = ~C1 and C2 is 0+1+ and (N & C2) == 0 | ||||
4285 | // replace with V+N. | ||||
4286 | if (C1->getValue() == ~C2->getValue()) { | ||||
4287 | if ((C2->getValue() & (C2->getValue()+1)) == 0 && // C2 == 0+1+ | ||||
4288 | match(A, m_Add(m_Value(V1), m_Value(V2)))) { | ||||
4289 | // Add commutes, try both ways. | ||||
4290 | if (V1 == B && MaskedValueIsZero(V2, C2->getValue())) | ||||
4291 | return ReplaceInstUsesWith(I, A); | ||||
4292 | if (V2 == B && MaskedValueIsZero(V1, C2->getValue())) | ||||
4293 | return ReplaceInstUsesWith(I, A); | ||||
4294 | } | ||||
4295 | // Or commutes, try both ways. | ||||
4296 | if ((C1->getValue() & (C1->getValue()+1)) == 0 && | ||||
4297 | match(B, m_Add(m_Value(V1), m_Value(V2)))) { | ||||
4298 | // Add commutes, try both ways. | ||||
4299 | if (V1 == A && MaskedValueIsZero(V2, C1->getValue())) | ||||
4300 | return ReplaceInstUsesWith(I, B); | ||||
4301 | if (V2 == A && MaskedValueIsZero(V1, C1->getValue())) | ||||
4302 | return ReplaceInstUsesWith(I, B); | ||||
4303 | } | ||||
4304 | } | ||||
4305 | V1 = 0; V2 = 0; V3 = 0; | ||||
4306 | } | ||||
4307 | |||||
4308 | // Check to see if we have any common things being and'ed. If so, find the | ||||
4309 | // terms for V1 & (V2|V3). | ||||
4310 | if (isOnlyUse(Op0) || isOnlyUse(Op1)) { | ||||
4311 | if (A == B) // (A & C)|(A & D) == A & (C|D) | ||||
4312 | V1 = A, V2 = C, V3 = D; | ||||
4313 | else if (A == D) // (A & C)|(B & A) == A & (B|C) | ||||
4314 | V1 = A, V2 = B, V3 = C; | ||||
4315 | else if (C == B) // (A & C)|(C & D) == C & (A|D) | ||||
4316 | V1 = C, V2 = A, V3 = D; | ||||
4317 | else if (C == D) // (A & C)|(B & C) == C & (A|B) | ||||
4318 | V1 = C, V2 = A, V3 = B; | ||||
4319 | |||||
4320 | if (V1) { | ||||
4321 | Value *Or = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4322 | InsertNewInstBefore(BinaryOperator::CreateOr(V2, V3, "tmp"), I); |
4323 | return BinaryOperator::CreateAnd(V1, Or); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4324 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4325 | } |
4326 | } | ||||
4327 | |||||
4328 | // (X >> Z) | (Y >> Z) -> (X|Y) >> Z for all shifts. | ||||
4329 | if (BinaryOperator *SI1 = dyn_cast<BinaryOperator>(Op1)) { | ||||
4330 | if (BinaryOperator *SI0 = dyn_cast<BinaryOperator>(Op0)) | ||||
4331 | if (SI0->isShift() && SI0->getOpcode() == SI1->getOpcode() && | ||||
4332 | SI0->getOperand(1) == SI1->getOperand(1) && | ||||
4333 | (SI0->hasOneUse() || SI1->hasOneUse())) { | ||||
4334 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4335 | InsertNewInstBefore(BinaryOperator::CreateOr(SI0->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4336 | SI1->getOperand(0), |
4337 | SI0->getName()), I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4338 | return BinaryOperator::Create(SI1->getOpcode(), NewOp, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4339 | SI1->getOperand(1)); |
4340 | } | ||||
4341 | } | ||||
4342 | |||||
4343 | if (match(Op0, m_Not(m_Value(A)))) { // ~A | Op1 | ||||
4344 | if (A == Op1) // ~A | A == -1 | ||||
4345 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4346 | } else { | ||||
4347 | A = 0; | ||||
4348 | } | ||||
4349 | // Note, A is still live here! | ||||
4350 | if (match(Op1, m_Not(m_Value(B)))) { // Op0 | ~B | ||||
4351 | if (Op0 == B) | ||||
4352 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4353 | |||||
4354 | // (~A | ~B) == (~(A & B)) - De Morgan's Law | ||||
4355 | if (A && isOnlyUse(Op0) && isOnlyUse(Op1)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4356 | Value *And = InsertNewInstBefore(BinaryOperator::CreateAnd(A, B, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4357 | I.getName()+".demorgan"), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4358 | return BinaryOperator::CreateNot(And); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4359 | } |
4360 | } | ||||
4361 | |||||
4362 | // (icmp1 A, B) | (icmp2 A, B) --> (icmp3 A, B) | ||||
4363 | if (ICmpInst *RHS = dyn_cast<ICmpInst>(I.getOperand(1))) { | ||||
4364 | if (Instruction *R = AssociativeOpt(I, FoldICmpLogical(*this, RHS))) | ||||
4365 | return R; | ||||
4366 | |||||
4367 | Value *LHSVal, *RHSVal; | ||||
4368 | ConstantInt *LHSCst, *RHSCst; | ||||
4369 | ICmpInst::Predicate LHSCC, RHSCC; | ||||
4370 | if (match(Op0, m_ICmp(LHSCC, m_Value(LHSVal), m_ConstantInt(LHSCst)))) | ||||
4371 | if (match(RHS, m_ICmp(RHSCC, m_Value(RHSVal), m_ConstantInt(RHSCst)))) | ||||
4372 | if (LHSVal == RHSVal && // Found (X icmp C1) | (X icmp C2) | ||||
4373 | // icmp [us][gl]e x, cst is folded to icmp [us][gl]t elsewhere. | ||||
4374 | LHSCC != ICmpInst::ICMP_UGE && LHSCC != ICmpInst::ICMP_ULE && | ||||
4375 | RHSCC != ICmpInst::ICMP_UGE && RHSCC != ICmpInst::ICMP_ULE && | ||||
4376 | LHSCC != ICmpInst::ICMP_SGE && LHSCC != ICmpInst::ICMP_SLE && | ||||
4377 | RHSCC != ICmpInst::ICMP_SGE && RHSCC != ICmpInst::ICMP_SLE && | ||||
4378 | // We can't fold (ugt x, C) | (sgt x, C2). | ||||
4379 | PredicatesFoldable(LHSCC, RHSCC)) { | ||||
4380 | // Ensure that the larger constant is on the RHS. | ||||
4381 | ICmpInst *LHS = cast<ICmpInst>(Op0); | ||||
4382 | bool NeedsSwap; | ||||
Nick Lewycky | 5515c7a | 2008-09-30 06:08:34 +0000 | [diff] [blame] | 4383 | if (ICmpInst::isEquality(LHSCC) ? ICmpInst::isSignedPredicate(RHSCC) |
4384 | : ICmpInst::isSignedPredicate(LHSCC)) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4385 | NeedsSwap = LHSCst->getValue().sgt(RHSCst->getValue()); |
4386 | else | ||||
4387 | NeedsSwap = LHSCst->getValue().ugt(RHSCst->getValue()); | ||||
4388 | |||||
4389 | if (NeedsSwap) { | ||||
4390 | std::swap(LHS, RHS); | ||||
4391 | std::swap(LHSCst, RHSCst); | ||||
4392 | std::swap(LHSCC, RHSCC); | ||||
4393 | } | ||||
4394 | |||||
4395 | // At this point, we know we have have two icmp instructions | ||||
4396 | // comparing a value against two constants and or'ing the result | ||||
4397 | // together. Because of the above check, we know that we only have | ||||
4398 | // ICMP_EQ, ICMP_NE, ICMP_LT, and ICMP_GT here. We also know (from the | ||||
4399 | // FoldICmpLogical check above), that the two constants are not | ||||
4400 | // equal. | ||||
4401 | assert(LHSCst != RHSCst && "Compares not folded above?"); | ||||
4402 | |||||
4403 | switch (LHSCC) { | ||||
4404 | default: assert(0 && "Unknown integer condition code!"); | ||||
4405 | case ICmpInst::ICMP_EQ: | ||||
4406 | switch (RHSCC) { | ||||
4407 | default: assert(0 && "Unknown integer condition code!"); | ||||
4408 | case ICmpInst::ICMP_EQ: | ||||
4409 | if (LHSCst == SubOne(RHSCst)) {// (X == 13 | X == 14) -> X-13 <u 2 | ||||
4410 | Constant *AddCST = ConstantExpr::getNeg(LHSCst); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4411 | Instruction *Add = BinaryOperator::CreateAdd(LHSVal, AddCST, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4412 | LHSVal->getName()+".off"); |
4413 | InsertNewInstBefore(Add, I); | ||||
4414 | AddCST = Subtract(AddOne(RHSCst), LHSCst); | ||||
4415 | return new ICmpInst(ICmpInst::ICMP_ULT, Add, AddCST); | ||||
4416 | } | ||||
4417 | break; // (X == 13 | X == 15) -> no change | ||||
4418 | case ICmpInst::ICMP_UGT: // (X == 13 | X u> 14) -> no change | ||||
4419 | case ICmpInst::ICMP_SGT: // (X == 13 | X s> 14) -> no change | ||||
4420 | break; | ||||
4421 | case ICmpInst::ICMP_NE: // (X == 13 | X != 15) -> X != 15 | ||||
4422 | case ICmpInst::ICMP_ULT: // (X == 13 | X u< 15) -> X u< 15 | ||||
4423 | case ICmpInst::ICMP_SLT: // (X == 13 | X s< 15) -> X s< 15 | ||||
4424 | return ReplaceInstUsesWith(I, RHS); | ||||
4425 | } | ||||
4426 | break; | ||||
4427 | case ICmpInst::ICMP_NE: | ||||
4428 | switch (RHSCC) { | ||||
4429 | default: assert(0 && "Unknown integer condition code!"); | ||||
4430 | case ICmpInst::ICMP_EQ: // (X != 13 | X == 15) -> X != 13 | ||||
4431 | case ICmpInst::ICMP_UGT: // (X != 13 | X u> 15) -> X != 13 | ||||
4432 | case ICmpInst::ICMP_SGT: // (X != 13 | X s> 15) -> X != 13 | ||||
4433 | return ReplaceInstUsesWith(I, LHS); | ||||
4434 | case ICmpInst::ICMP_NE: // (X != 13 | X != 15) -> true | ||||
4435 | case ICmpInst::ICMP_ULT: // (X != 13 | X u< 15) -> true | ||||
4436 | case ICmpInst::ICMP_SLT: // (X != 13 | X s< 15) -> true | ||||
4437 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4438 | } | ||||
4439 | break; | ||||
4440 | case ICmpInst::ICMP_ULT: | ||||
4441 | switch (RHSCC) { | ||||
4442 | default: assert(0 && "Unknown integer condition code!"); | ||||
4443 | case ICmpInst::ICMP_EQ: // (X u< 13 | X == 14) -> no change | ||||
4444 | break; | ||||
4445 | case ICmpInst::ICMP_UGT: // (X u< 13 | X u> 15) ->(X-13) u> 2 | ||||
Chris Lattner | 2637686 | 2007-11-01 02:18:41 +0000 | [diff] [blame] | 4446 | // If RHSCst is [us]MAXINT, it is always false. Not handling |
4447 | // this can cause overflow. | ||||
4448 | if (RHSCst->isMaxValue(false)) | ||||
4449 | return ReplaceInstUsesWith(I, LHS); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4450 | return InsertRangeTest(LHSVal, LHSCst, AddOne(RHSCst), false, |
4451 | false, I); | ||||
4452 | case ICmpInst::ICMP_SGT: // (X u< 13 | X s> 15) -> no change | ||||
4453 | break; | ||||
4454 | case ICmpInst::ICMP_NE: // (X u< 13 | X != 15) -> X != 15 | ||||
4455 | case ICmpInst::ICMP_ULT: // (X u< 13 | X u< 15) -> X u< 15 | ||||
4456 | return ReplaceInstUsesWith(I, RHS); | ||||
4457 | case ICmpInst::ICMP_SLT: // (X u< 13 | X s< 15) -> no change | ||||
4458 | break; | ||||
4459 | } | ||||
4460 | break; | ||||
4461 | case ICmpInst::ICMP_SLT: | ||||
4462 | switch (RHSCC) { | ||||
4463 | default: assert(0 && "Unknown integer condition code!"); | ||||
4464 | case ICmpInst::ICMP_EQ: // (X s< 13 | X == 14) -> no change | ||||
4465 | break; | ||||
4466 | case ICmpInst::ICMP_SGT: // (X s< 13 | X s> 15) ->(X-13) s> 2 | ||||
Chris Lattner | 2637686 | 2007-11-01 02:18:41 +0000 | [diff] [blame] | 4467 | // If RHSCst is [us]MAXINT, it is always false. Not handling |
4468 | // this can cause overflow. | ||||
4469 | if (RHSCst->isMaxValue(true)) | ||||
4470 | return ReplaceInstUsesWith(I, LHS); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4471 | return InsertRangeTest(LHSVal, LHSCst, AddOne(RHSCst), true, |
4472 | false, I); | ||||
4473 | case ICmpInst::ICMP_UGT: // (X s< 13 | X u> 15) -> no change | ||||
4474 | break; | ||||
4475 | case ICmpInst::ICMP_NE: // (X s< 13 | X != 15) -> X != 15 | ||||
4476 | case ICmpInst::ICMP_SLT: // (X s< 13 | X s< 15) -> X s< 15 | ||||
4477 | return ReplaceInstUsesWith(I, RHS); | ||||
4478 | case ICmpInst::ICMP_ULT: // (X s< 13 | X u< 15) -> no change | ||||
4479 | break; | ||||
4480 | } | ||||
4481 | break; | ||||
4482 | case ICmpInst::ICMP_UGT: | ||||
4483 | switch (RHSCC) { | ||||
4484 | default: assert(0 && "Unknown integer condition code!"); | ||||
4485 | case ICmpInst::ICMP_EQ: // (X u> 13 | X == 15) -> X u> 13 | ||||
4486 | case ICmpInst::ICMP_UGT: // (X u> 13 | X u> 15) -> X u> 13 | ||||
4487 | return ReplaceInstUsesWith(I, LHS); | ||||
4488 | case ICmpInst::ICMP_SGT: // (X u> 13 | X s> 15) -> no change | ||||
4489 | break; | ||||
4490 | case ICmpInst::ICMP_NE: // (X u> 13 | X != 15) -> true | ||||
4491 | case ICmpInst::ICMP_ULT: // (X u> 13 | X u< 15) -> true | ||||
4492 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4493 | case ICmpInst::ICMP_SLT: // (X u> 13 | X s< 15) -> no change | ||||
4494 | break; | ||||
4495 | } | ||||
4496 | break; | ||||
4497 | case ICmpInst::ICMP_SGT: | ||||
4498 | switch (RHSCC) { | ||||
4499 | default: assert(0 && "Unknown integer condition code!"); | ||||
4500 | case ICmpInst::ICMP_EQ: // (X s> 13 | X == 15) -> X > 13 | ||||
4501 | case ICmpInst::ICMP_SGT: // (X s> 13 | X s> 15) -> X > 13 | ||||
4502 | return ReplaceInstUsesWith(I, LHS); | ||||
4503 | case ICmpInst::ICMP_UGT: // (X s> 13 | X u> 15) -> no change | ||||
4504 | break; | ||||
4505 | case ICmpInst::ICMP_NE: // (X s> 13 | X != 15) -> true | ||||
4506 | case ICmpInst::ICMP_SLT: // (X s> 13 | X s< 15) -> true | ||||
4507 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4508 | case ICmpInst::ICMP_ULT: // (X s> 13 | X u< 15) -> no change | ||||
4509 | break; | ||||
4510 | } | ||||
4511 | break; | ||||
4512 | } | ||||
4513 | } | ||||
4514 | } | ||||
4515 | |||||
4516 | // fold (or (cast A), (cast B)) -> (cast (or A, B)) | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4517 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4518 | if (CastInst *Op1C = dyn_cast<CastInst>(Op1)) |
4519 | if (Op0C->getOpcode() == Op1C->getOpcode()) {// same cast kind ? | ||||
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 4520 | if (!isa<ICmpInst>(Op0C->getOperand(0)) || |
4521 | !isa<ICmpInst>(Op1C->getOperand(0))) { | ||||
4522 | const Type *SrcTy = Op0C->getOperand(0)->getType(); | ||||
4523 | if (SrcTy == Op1C->getOperand(0)->getType() && SrcTy->isInteger() && | ||||
4524 | // Only do this if the casts both really cause code to be | ||||
4525 | // generated. | ||||
4526 | ValueRequiresCast(Op0C->getOpcode(), Op0C->getOperand(0), | ||||
4527 | I.getType(), TD) && | ||||
4528 | ValueRequiresCast(Op1C->getOpcode(), Op1C->getOperand(0), | ||||
4529 | I.getType(), TD)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4530 | Instruction *NewOp = BinaryOperator::CreateOr(Op0C->getOperand(0), |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 4531 | Op1C->getOperand(0), |
4532 | I.getName()); | ||||
4533 | InsertNewInstBefore(NewOp, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4534 | return CastInst::Create(Op0C->getOpcode(), NewOp, I.getType()); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 4535 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4536 | } |
4537 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4538 | } |
4539 | |||||
4540 | |||||
4541 | // (fcmp uno x, c) | (fcmp uno y, c) -> (fcmp uno x, y) | ||||
4542 | if (FCmpInst *LHS = dyn_cast<FCmpInst>(I.getOperand(0))) { | ||||
4543 | if (FCmpInst *RHS = dyn_cast<FCmpInst>(I.getOperand(1))) { | ||||
4544 | if (LHS->getPredicate() == FCmpInst::FCMP_UNO && | ||||
Chris Lattner | be9e63e | 2008-02-29 06:09:11 +0000 | [diff] [blame] | 4545 | RHS->getPredicate() == FCmpInst::FCMP_UNO && |
4546 | LHS->getOperand(0)->getType() == RHS->getOperand(0)->getType()) | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4547 | if (ConstantFP *LHSC = dyn_cast<ConstantFP>(LHS->getOperand(1))) |
4548 | if (ConstantFP *RHSC = dyn_cast<ConstantFP>(RHS->getOperand(1))) { | ||||
4549 | // If either of the constants are nans, then the whole thing returns | ||||
4550 | // true. | ||||
Chris Lattner | a6c7dce | 2007-10-24 18:54:45 +0000 | [diff] [blame] | 4551 | if (LHSC->getValueAPF().isNaN() || RHSC->getValueAPF().isNaN()) |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4552 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
4553 | |||||
4554 | // Otherwise, no need to compare the two constants, compare the | ||||
4555 | // rest. | ||||
4556 | return new FCmpInst(FCmpInst::FCMP_UNO, LHS->getOperand(0), | ||||
4557 | RHS->getOperand(0)); | ||||
4558 | } | ||||
4559 | } | ||||
4560 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4561 | |
4562 | return Changed ? &I : 0; | ||||
4563 | } | ||||
4564 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 4565 | namespace { |
4566 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4567 | // XorSelf - Implements: X ^ X --> 0 |
4568 | struct XorSelf { | ||||
4569 | Value *RHS; | ||||
4570 | XorSelf(Value *rhs) : RHS(rhs) {} | ||||
4571 | bool shouldApply(Value *LHS) const { return LHS == RHS; } | ||||
4572 | Instruction *apply(BinaryOperator &Xor) const { | ||||
4573 | return &Xor; | ||||
4574 | } | ||||
4575 | }; | ||||
4576 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 4577 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4578 | |
4579 | Instruction *InstCombiner::visitXor(BinaryOperator &I) { | ||||
4580 | bool Changed = SimplifyCommutative(I); | ||||
4581 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
4582 | |||||
Evan Cheng | e5cd803 | 2008-03-25 20:07:13 +0000 | [diff] [blame] | 4583 | if (isa<UndefValue>(Op1)) { |
4584 | if (isa<UndefValue>(Op0)) | ||||
4585 | // Handle undef ^ undef -> 0 special case. This is a common | ||||
4586 | // idiom (misuse). | ||||
4587 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4588 | return ReplaceInstUsesWith(I, Op1); // X ^ undef -> undef |
Evan Cheng | e5cd803 | 2008-03-25 20:07:13 +0000 | [diff] [blame] | 4589 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4590 | |
4591 | // xor X, X = 0, even if X is nested in a sequence of Xor's. | ||||
4592 | if (Instruction *Result = AssociativeOpt(I, XorSelf(Op1))) { | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 4593 | assert(Result == &I && "AssociativeOpt didn't work?"); Result=Result; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4594 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
4595 | } | ||||
4596 | |||||
4597 | // See if we can simplify any instructions used by the instruction whose sole | ||||
4598 | // purpose is to compute bits we don't care about. | ||||
4599 | if (!isa<VectorType>(I.getType())) { | ||||
4600 | uint32_t BitWidth = cast<IntegerType>(I.getType())->getBitWidth(); | ||||
4601 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
4602 | if (SimplifyDemandedBits(&I, APInt::getAllOnesValue(BitWidth), | ||||
4603 | KnownZero, KnownOne)) | ||||
4604 | return &I; | ||||
4605 | } else if (isa<ConstantAggregateZero>(Op1)) { | ||||
4606 | return ReplaceInstUsesWith(I, Op0); // X ^ <0,0> -> X | ||||
4607 | } | ||||
4608 | |||||
4609 | // Is this a ~ operation? | ||||
4610 | if (Value *NotOp = dyn_castNotVal(&I)) { | ||||
4611 | // ~(~X & Y) --> (X | ~Y) - De Morgan's Law | ||||
4612 | // ~(~X | Y) === (X & ~Y) - De Morgan's Law | ||||
4613 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(NotOp)) { | ||||
4614 | if (Op0I->getOpcode() == Instruction::And || | ||||
4615 | Op0I->getOpcode() == Instruction::Or) { | ||||
4616 | if (dyn_castNotVal(Op0I->getOperand(1))) Op0I->swapOperands(); | ||||
4617 | if (Value *Op0NotVal = dyn_castNotVal(Op0I->getOperand(0))) { | ||||
4618 | Instruction *NotY = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4619 | BinaryOperator::CreateNot(Op0I->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4620 | Op0I->getOperand(1)->getName()+".not"); |
4621 | InsertNewInstBefore(NotY, I); | ||||
4622 | if (Op0I->getOpcode() == Instruction::And) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4623 | return BinaryOperator::CreateOr(Op0NotVal, NotY); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4624 | else |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4625 | return BinaryOperator::CreateAnd(Op0NotVal, NotY); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4626 | } |
4627 | } | ||||
4628 | } | ||||
4629 | } | ||||
4630 | |||||
4631 | |||||
4632 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
Nick Lewycky | 1405e92 | 2007-08-06 20:04:16 +0000 | [diff] [blame] | 4633 | // xor (cmp A, B), true = not (cmp A, B) = !cmp A, B |
4634 | if (RHS == ConstantInt::getTrue() && Op0->hasOneUse()) { | ||||
4635 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(Op0)) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4636 | return new ICmpInst(ICI->getInversePredicate(), |
4637 | ICI->getOperand(0), ICI->getOperand(1)); | ||||
4638 | |||||
Nick Lewycky | 1405e92 | 2007-08-06 20:04:16 +0000 | [diff] [blame] | 4639 | if (FCmpInst *FCI = dyn_cast<FCmpInst>(Op0)) |
4640 | return new FCmpInst(FCI->getInversePredicate(), | ||||
4641 | FCI->getOperand(0), FCI->getOperand(1)); | ||||
4642 | } | ||||
4643 | |||||
Nick Lewycky | 0aa63aa | 2008-05-31 19:01:33 +0000 | [diff] [blame] | 4644 | // fold (xor(zext(cmp)), 1) and (xor(sext(cmp)), -1) to ext(!cmp). |
4645 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) { | ||||
4646 | if (CmpInst *CI = dyn_cast<CmpInst>(Op0C->getOperand(0))) { | ||||
4647 | if (CI->hasOneUse() && Op0C->hasOneUse()) { | ||||
4648 | Instruction::CastOps Opcode = Op0C->getOpcode(); | ||||
4649 | if (Opcode == Instruction::ZExt || Opcode == Instruction::SExt) { | ||||
4650 | if (RHS == ConstantExpr::getCast(Opcode, ConstantInt::getTrue(), | ||||
4651 | Op0C->getDestTy())) { | ||||
4652 | Instruction *NewCI = InsertNewInstBefore(CmpInst::Create( | ||||
4653 | CI->getOpcode(), CI->getInversePredicate(), | ||||
4654 | CI->getOperand(0), CI->getOperand(1)), I); | ||||
4655 | NewCI->takeName(CI); | ||||
4656 | return CastInst::Create(Opcode, NewCI, Op0C->getType()); | ||||
4657 | } | ||||
4658 | } | ||||
4659 | } | ||||
4660 | } | ||||
4661 | } | ||||
4662 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4663 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) { |
4664 | // ~(c-X) == X-c-1 == X+(-c-1) | ||||
4665 | if (Op0I->getOpcode() == Instruction::Sub && RHS->isAllOnesValue()) | ||||
4666 | if (Constant *Op0I0C = dyn_cast<Constant>(Op0I->getOperand(0))) { | ||||
4667 | Constant *NegOp0I0C = ConstantExpr::getNeg(Op0I0C); | ||||
4668 | Constant *ConstantRHS = ConstantExpr::getSub(NegOp0I0C, | ||||
4669 | ConstantInt::get(I.getType(), 1)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4670 | return BinaryOperator::CreateAdd(Op0I->getOperand(1), ConstantRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4671 | } |
4672 | |||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 4673 | if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4674 | if (Op0I->getOpcode() == Instruction::Add) { |
4675 | // ~(X-c) --> (-c-1)-X | ||||
4676 | if (RHS->isAllOnesValue()) { | ||||
4677 | Constant *NegOp0CI = ConstantExpr::getNeg(Op0CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4678 | return BinaryOperator::CreateSub( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4679 | ConstantExpr::getSub(NegOp0CI, |
4680 | ConstantInt::get(I.getType(), 1)), | ||||
4681 | Op0I->getOperand(0)); | ||||
4682 | } else if (RHS->getValue().isSignBit()) { | ||||
4683 | // (X + C) ^ signbit -> (X + C + signbit) | ||||
4684 | Constant *C = ConstantInt::get(RHS->getValue() + Op0CI->getValue()); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4685 | return BinaryOperator::CreateAdd(Op0I->getOperand(0), C); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4686 | |
4687 | } | ||||
4688 | } else if (Op0I->getOpcode() == Instruction::Or) { | ||||
4689 | // (X|C1)^C2 -> X^(C1|C2) iff X&~C1 == 0 | ||||
4690 | if (MaskedValueIsZero(Op0I->getOperand(0), Op0CI->getValue())) { | ||||
4691 | Constant *NewRHS = ConstantExpr::getOr(Op0CI, RHS); | ||||
4692 | // Anything in both C1 and C2 is known to be zero, remove it from | ||||
4693 | // NewRHS. | ||||
4694 | Constant *CommonBits = And(Op0CI, RHS); | ||||
4695 | NewRHS = ConstantExpr::getAnd(NewRHS, | ||||
4696 | ConstantExpr::getNot(CommonBits)); | ||||
4697 | AddToWorkList(Op0I); | ||||
4698 | I.setOperand(0, Op0I->getOperand(0)); | ||||
4699 | I.setOperand(1, NewRHS); | ||||
4700 | return &I; | ||||
4701 | } | ||||
4702 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 4703 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4704 | } |
4705 | |||||
4706 | // Try to fold constant and into select arguments. | ||||
4707 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
4708 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
4709 | return R; | ||||
4710 | if (isa<PHINode>(Op0)) | ||||
4711 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
4712 | return NV; | ||||
4713 | } | ||||
4714 | |||||
4715 | if (Value *X = dyn_castNotVal(Op0)) // ~A ^ A == -1 | ||||
4716 | if (X == Op1) | ||||
4717 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4718 | |||||
4719 | if (Value *X = dyn_castNotVal(Op1)) // A ^ ~A == -1 | ||||
4720 | if (X == Op0) | ||||
4721 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4722 | |||||
4723 | |||||
4724 | BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1); | ||||
4725 | if (Op1I) { | ||||
4726 | Value *A, *B; | ||||
4727 | if (match(Op1I, m_Or(m_Value(A), m_Value(B)))) { | ||||
4728 | if (A == Op0) { // B^(B|A) == (A|B)^B | ||||
4729 | Op1I->swapOperands(); | ||||
4730 | I.swapOperands(); | ||||
4731 | std::swap(Op0, Op1); | ||||
4732 | } else if (B == Op0) { // B^(A|B) == (A|B)^B | ||||
4733 | I.swapOperands(); // Simplified below. | ||||
4734 | std::swap(Op0, Op1); | ||||
4735 | } | ||||
4736 | } else if (match(Op1I, m_Xor(m_Value(A), m_Value(B)))) { | ||||
4737 | if (Op0 == A) // A^(A^B) == B | ||||
4738 | return ReplaceInstUsesWith(I, B); | ||||
4739 | else if (Op0 == B) // A^(B^A) == B | ||||
4740 | return ReplaceInstUsesWith(I, A); | ||||
4741 | } else if (match(Op1I, m_And(m_Value(A), m_Value(B))) && Op1I->hasOneUse()){ | ||||
4742 | if (A == Op0) { // A^(A&B) -> A^(B&A) | ||||
4743 | Op1I->swapOperands(); | ||||
4744 | std::swap(A, B); | ||||
4745 | } | ||||
4746 | if (B == Op0) { // A^(B&A) -> (B&A)^A | ||||
4747 | I.swapOperands(); // Simplified below. | ||||
4748 | std::swap(Op0, Op1); | ||||
4749 | } | ||||
4750 | } | ||||
4751 | } | ||||
4752 | |||||
4753 | BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0); | ||||
4754 | if (Op0I) { | ||||
4755 | Value *A, *B; | ||||
4756 | if (match(Op0I, m_Or(m_Value(A), m_Value(B))) && Op0I->hasOneUse()) { | ||||
4757 | if (A == Op1) // (B|A)^B == (A|B)^B | ||||
4758 | std::swap(A, B); | ||||
4759 | if (B == Op1) { // (A|B)^B == A & ~B | ||||
4760 | Instruction *NotB = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4761 | InsertNewInstBefore(BinaryOperator::CreateNot(Op1, "tmp"), I); |
4762 | return BinaryOperator::CreateAnd(A, NotB); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4763 | } |
4764 | } else if (match(Op0I, m_Xor(m_Value(A), m_Value(B)))) { | ||||
4765 | if (Op1 == A) // (A^B)^A == B | ||||
4766 | return ReplaceInstUsesWith(I, B); | ||||
4767 | else if (Op1 == B) // (B^A)^A == B | ||||
4768 | return ReplaceInstUsesWith(I, A); | ||||
4769 | } else if (match(Op0I, m_And(m_Value(A), m_Value(B))) && Op0I->hasOneUse()){ | ||||
4770 | if (A == Op1) // (A&B)^A -> (B&A)^A | ||||
4771 | std::swap(A, B); | ||||
4772 | if (B == Op1 && // (B&A)^A == ~B & A | ||||
4773 | !isa<ConstantInt>(Op1)) { // Canonical form is (B&C)^C | ||||
4774 | Instruction *N = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4775 | InsertNewInstBefore(BinaryOperator::CreateNot(A, "tmp"), I); |
4776 | return BinaryOperator::CreateAnd(N, Op1); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4777 | } |
4778 | } | ||||
4779 | } | ||||
4780 | |||||
4781 | // (X >> Z) ^ (Y >> Z) -> (X^Y) >> Z for all shifts. | ||||
4782 | if (Op0I && Op1I && Op0I->isShift() && | ||||
4783 | Op0I->getOpcode() == Op1I->getOpcode() && | ||||
4784 | Op0I->getOperand(1) == Op1I->getOperand(1) && | ||||
4785 | (Op1I->hasOneUse() || Op1I->hasOneUse())) { | ||||
4786 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4787 | InsertNewInstBefore(BinaryOperator::CreateXor(Op0I->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4788 | Op1I->getOperand(0), |
4789 | Op0I->getName()), I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4790 | return BinaryOperator::Create(Op1I->getOpcode(), NewOp, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4791 | Op1I->getOperand(1)); |
4792 | } | ||||
4793 | |||||
4794 | if (Op0I && Op1I) { | ||||
4795 | Value *A, *B, *C, *D; | ||||
4796 | // (A & B)^(A | B) -> A ^ B | ||||
4797 | if (match(Op0I, m_And(m_Value(A), m_Value(B))) && | ||||
4798 | match(Op1I, m_Or(m_Value(C), m_Value(D)))) { | ||||
4799 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4800 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4801 | } |
4802 | // (A | B)^(A & B) -> A ^ B | ||||
4803 | if (match(Op0I, m_Or(m_Value(A), m_Value(B))) && | ||||
4804 | match(Op1I, m_And(m_Value(C), m_Value(D)))) { | ||||
4805 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4806 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4807 | } |
4808 | |||||
4809 | // (A & B)^(C & D) | ||||
4810 | if ((Op0I->hasOneUse() || Op1I->hasOneUse()) && | ||||
4811 | match(Op0I, m_And(m_Value(A), m_Value(B))) && | ||||
4812 | match(Op1I, m_And(m_Value(C), m_Value(D)))) { | ||||
4813 | // (X & Y)^(X & Y) -> (Y^Z) & X | ||||
4814 | Value *X = 0, *Y = 0, *Z = 0; | ||||
4815 | if (A == C) | ||||
4816 | X = A, Y = B, Z = D; | ||||
4817 | else if (A == D) | ||||
4818 | X = A, Y = B, Z = C; | ||||
4819 | else if (B == C) | ||||
4820 | X = B, Y = A, Z = D; | ||||
4821 | else if (B == D) | ||||
4822 | X = B, Y = A, Z = C; | ||||
4823 | |||||
4824 | if (X) { | ||||
4825 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4826 | InsertNewInstBefore(BinaryOperator::CreateXor(Y, Z, Op0->getName()), I); |
4827 | return BinaryOperator::CreateAnd(NewOp, X); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4828 | } |
4829 | } | ||||
4830 | } | ||||
4831 | |||||
4832 | // (icmp1 A, B) ^ (icmp2 A, B) --> (icmp3 A, B) | ||||
4833 | if (ICmpInst *RHS = dyn_cast<ICmpInst>(I.getOperand(1))) | ||||
4834 | if (Instruction *R = AssociativeOpt(I, FoldICmpLogical(*this, RHS))) | ||||
4835 | return R; | ||||
4836 | |||||
4837 | // fold (xor (cast A), (cast B)) -> (cast (xor A, B)) | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4838 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4839 | if (CastInst *Op1C = dyn_cast<CastInst>(Op1)) |
4840 | if (Op0C->getOpcode() == Op1C->getOpcode()) { // same cast kind? | ||||
4841 | const Type *SrcTy = Op0C->getOperand(0)->getType(); | ||||
4842 | if (SrcTy == Op1C->getOperand(0)->getType() && SrcTy->isInteger() && | ||||
4843 | // Only do this if the casts both really cause code to be generated. | ||||
4844 | ValueRequiresCast(Op0C->getOpcode(), Op0C->getOperand(0), | ||||
4845 | I.getType(), TD) && | ||||
4846 | ValueRequiresCast(Op1C->getOpcode(), Op1C->getOperand(0), | ||||
4847 | I.getType(), TD)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4848 | Instruction *NewOp = BinaryOperator::CreateXor(Op0C->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4849 | Op1C->getOperand(0), |
4850 | I.getName()); | ||||
4851 | InsertNewInstBefore(NewOp, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4852 | return CastInst::Create(Op0C->getOpcode(), NewOp, I.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4853 | } |
4854 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4855 | } |
Nick Lewycky | 0aa63aa | 2008-05-31 19:01:33 +0000 | [diff] [blame] | 4856 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4857 | return Changed ? &I : 0; |
4858 | } | ||||
4859 | |||||
4860 | /// AddWithOverflow - Compute Result = In1+In2, returning true if the result | ||||
4861 | /// overflowed for this type. | ||||
4862 | static bool AddWithOverflow(ConstantInt *&Result, ConstantInt *In1, | ||||
4863 | ConstantInt *In2, bool IsSigned = false) { | ||||
4864 | Result = cast<ConstantInt>(Add(In1, In2)); | ||||
4865 | |||||
4866 | if (IsSigned) | ||||
4867 | if (In2->getValue().isNegative()) | ||||
4868 | return Result->getValue().sgt(In1->getValue()); | ||||
4869 | else | ||||
4870 | return Result->getValue().slt(In1->getValue()); | ||||
4871 | else | ||||
4872 | return Result->getValue().ult(In1->getValue()); | ||||
4873 | } | ||||
4874 | |||||
Dan Gohman | b80d561 | 2008-09-10 23:30:57 +0000 | [diff] [blame] | 4875 | /// SubWithOverflow - Compute Result = In1-In2, returning true if the result |
4876 | /// overflowed for this type. | ||||
4877 | static bool SubWithOverflow(ConstantInt *&Result, ConstantInt *In1, | ||||
4878 | ConstantInt *In2, bool IsSigned = false) { | ||||
Dan Gohman | 2c3b489 | 2008-09-11 18:53:02 +0000 | [diff] [blame] | 4879 | Result = cast<ConstantInt>(Subtract(In1, In2)); |
Dan Gohman | b80d561 | 2008-09-10 23:30:57 +0000 | [diff] [blame] | 4880 | |
4881 | if (IsSigned) | ||||
4882 | if (In2->getValue().isNegative()) | ||||
4883 | return Result->getValue().slt(In1->getValue()); | ||||
4884 | else | ||||
4885 | return Result->getValue().sgt(In1->getValue()); | ||||
4886 | else | ||||
4887 | return Result->getValue().ugt(In1->getValue()); | ||||
4888 | } | ||||
4889 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4890 | /// EmitGEPOffset - Given a getelementptr instruction/constantexpr, emit the |
4891 | /// code necessary to compute the offset from the base pointer (without adding | ||||
4892 | /// in the base pointer). Return the result as a signed integer of intptr size. | ||||
4893 | static Value *EmitGEPOffset(User *GEP, Instruction &I, InstCombiner &IC) { | ||||
4894 | TargetData &TD = IC.getTargetData(); | ||||
4895 | gep_type_iterator GTI = gep_type_begin(GEP); | ||||
4896 | const Type *IntPtrTy = TD.getIntPtrType(); | ||||
4897 | Value *Result = Constant::getNullValue(IntPtrTy); | ||||
4898 | |||||
4899 | // Build a mask for high order bits. | ||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 4900 | unsigned IntPtrWidth = TD.getPointerSizeInBits(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4901 | uint64_t PtrSizeMask = ~0ULL >> (64-IntPtrWidth); |
4902 | |||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 4903 | for (User::op_iterator i = GEP->op_begin() + 1, e = GEP->op_end(); i != e; |
4904 | ++i, ++GTI) { | ||||
4905 | Value *Op = *i; | ||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 4906 | uint64_t Size = TD.getABITypeSize(GTI.getIndexedType()) & PtrSizeMask; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4907 | if (ConstantInt *OpC = dyn_cast<ConstantInt>(Op)) { |
4908 | if (OpC->isZero()) continue; | ||||
4909 | |||||
4910 | // Handle a struct index, which adds its field offset to the pointer. | ||||
4911 | if (const StructType *STy = dyn_cast<StructType>(*GTI)) { | ||||
4912 | Size = TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue()); | ||||
4913 | |||||
4914 | if (ConstantInt *RC = dyn_cast<ConstantInt>(Result)) | ||||
4915 | Result = ConstantInt::get(RC->getValue() + APInt(IntPtrWidth, Size)); | ||||
4916 | else | ||||
4917 | Result = IC.InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4918 | BinaryOperator::CreateAdd(Result, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4919 | ConstantInt::get(IntPtrTy, Size), |
4920 | GEP->getName()+".offs"), I); | ||||
4921 | continue; | ||||
4922 | } | ||||
4923 | |||||
4924 | Constant *Scale = ConstantInt::get(IntPtrTy, Size); | ||||
4925 | Constant *OC = ConstantExpr::getIntegerCast(OpC, IntPtrTy, true /*SExt*/); | ||||
4926 | Scale = ConstantExpr::getMul(OC, Scale); | ||||
4927 | if (Constant *RC = dyn_cast<Constant>(Result)) | ||||
4928 | Result = ConstantExpr::getAdd(RC, Scale); | ||||
4929 | else { | ||||
4930 | // Emit an add instruction. | ||||
4931 | Result = IC.InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4932 | BinaryOperator::CreateAdd(Result, Scale, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4933 | GEP->getName()+".offs"), I); |
4934 | } | ||||
4935 | continue; | ||||
4936 | } | ||||
4937 | // Convert to correct type. | ||||
4938 | if (Op->getType() != IntPtrTy) { | ||||
4939 | if (Constant *OpC = dyn_cast<Constant>(Op)) | ||||
4940 | Op = ConstantExpr::getSExt(OpC, IntPtrTy); | ||||
4941 | else | ||||
4942 | Op = IC.InsertNewInstBefore(new SExtInst(Op, IntPtrTy, | ||||
4943 | Op->getName()+".c"), I); | ||||
4944 | } | ||||
4945 | if (Size != 1) { | ||||
4946 | Constant *Scale = ConstantInt::get(IntPtrTy, Size); | ||||
4947 | if (Constant *OpC = dyn_cast<Constant>(Op)) | ||||
4948 | Op = ConstantExpr::getMul(OpC, Scale); | ||||
4949 | 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] | 4950 | Op = IC.InsertNewInstBefore(BinaryOperator::CreateMul(Op, Scale, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4951 | GEP->getName()+".idx"), I); |
4952 | } | ||||
4953 | |||||
4954 | // Emit an add instruction. | ||||
4955 | if (isa<Constant>(Op) && isa<Constant>(Result)) | ||||
4956 | Result = ConstantExpr::getAdd(cast<Constant>(Op), | ||||
4957 | cast<Constant>(Result)); | ||||
4958 | else | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4959 | Result = IC.InsertNewInstBefore(BinaryOperator::CreateAdd(Op, Result, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4960 | GEP->getName()+".offs"), I); |
4961 | } | ||||
4962 | return Result; | ||||
4963 | } | ||||
4964 | |||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 4965 | |
4966 | /// EvaluateGEPOffsetExpression - Return an value that can be used to compare of | ||||
4967 | /// the *offset* implied by GEP to zero. For example, if we have &A[i], we want | ||||
4968 | /// to return 'i' for "icmp ne i, 0". Note that, in general, indices can be | ||||
4969 | /// complex, and scales are involved. The above expression would also be legal | ||||
4970 | /// to codegen as "icmp ne (i*4), 0" (assuming A is a pointer to i32). This | ||||
4971 | /// later form is less amenable to optimization though, and we are allowed to | ||||
4972 | /// generate the first by knowing that pointer arithmetic doesn't overflow. | ||||
4973 | /// | ||||
4974 | /// If we can't emit an optimized form for this expression, this returns null. | ||||
4975 | /// | ||||
4976 | static Value *EvaluateGEPOffsetExpression(User *GEP, Instruction &I, | ||||
4977 | InstCombiner &IC) { | ||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 4978 | TargetData &TD = IC.getTargetData(); |
4979 | gep_type_iterator GTI = gep_type_begin(GEP); | ||||
4980 | |||||
4981 | // Check to see if this gep only has a single variable index. If so, and if | ||||
4982 | // any constant indices are a multiple of its scale, then we can compute this | ||||
4983 | // in terms of the scale of the variable index. For example, if the GEP | ||||
4984 | // implies an offset of "12 + i*4", then we can codegen this as "3 + i", | ||||
4985 | // because the expression will cross zero at the same point. | ||||
4986 | unsigned i, e = GEP->getNumOperands(); | ||||
4987 | int64_t Offset = 0; | ||||
4988 | for (i = 1; i != e; ++i, ++GTI) { | ||||
4989 | if (ConstantInt *CI = dyn_cast<ConstantInt>(GEP->getOperand(i))) { | ||||
4990 | // Compute the aggregate offset of constant indices. | ||||
4991 | if (CI->isZero()) continue; | ||||
4992 | |||||
4993 | // Handle a struct index, which adds its field offset to the pointer. | ||||
4994 | if (const StructType *STy = dyn_cast<StructType>(*GTI)) { | ||||
4995 | Offset += TD.getStructLayout(STy)->getElementOffset(CI->getZExtValue()); | ||||
4996 | } else { | ||||
4997 | uint64_t Size = TD.getABITypeSize(GTI.getIndexedType()); | ||||
4998 | Offset += Size*CI->getSExtValue(); | ||||
4999 | } | ||||
5000 | } else { | ||||
5001 | // Found our variable index. | ||||
5002 | break; | ||||
5003 | } | ||||
5004 | } | ||||
5005 | |||||
5006 | // If there are no variable indices, we must have a constant offset, just | ||||
5007 | // evaluate it the general way. | ||||
5008 | if (i == e) return 0; | ||||
5009 | |||||
5010 | Value *VariableIdx = GEP->getOperand(i); | ||||
5011 | // Determine the scale factor of the variable element. For example, this is | ||||
5012 | // 4 if the variable index is into an array of i32. | ||||
5013 | uint64_t VariableScale = TD.getABITypeSize(GTI.getIndexedType()); | ||||
5014 | |||||
5015 | // Verify that there are no other variable indices. If so, emit the hard way. | ||||
5016 | for (++i, ++GTI; i != e; ++i, ++GTI) { | ||||
5017 | ConstantInt *CI = dyn_cast<ConstantInt>(GEP->getOperand(i)); | ||||
5018 | if (!CI) return 0; | ||||
5019 | |||||
5020 | // Compute the aggregate offset of constant indices. | ||||
5021 | if (CI->isZero()) continue; | ||||
5022 | |||||
5023 | // Handle a struct index, which adds its field offset to the pointer. | ||||
5024 | if (const StructType *STy = dyn_cast<StructType>(*GTI)) { | ||||
5025 | Offset += TD.getStructLayout(STy)->getElementOffset(CI->getZExtValue()); | ||||
5026 | } else { | ||||
5027 | uint64_t Size = TD.getABITypeSize(GTI.getIndexedType()); | ||||
5028 | Offset += Size*CI->getSExtValue(); | ||||
5029 | } | ||||
5030 | } | ||||
5031 | |||||
5032 | // Okay, we know we have a single variable index, which must be a | ||||
5033 | // pointer/array/vector index. If there is no offset, life is simple, return | ||||
5034 | // the index. | ||||
5035 | unsigned IntPtrWidth = TD.getPointerSizeInBits(); | ||||
5036 | if (Offset == 0) { | ||||
5037 | // Cast to intptrty in case a truncation occurs. If an extension is needed, | ||||
5038 | // we don't need to bother extending: the extension won't affect where the | ||||
5039 | // computation crosses zero. | ||||
5040 | if (VariableIdx->getType()->getPrimitiveSizeInBits() > IntPtrWidth) | ||||
5041 | VariableIdx = new TruncInst(VariableIdx, TD.getIntPtrType(), | ||||
5042 | VariableIdx->getNameStart(), &I); | ||||
5043 | return VariableIdx; | ||||
5044 | } | ||||
5045 | |||||
5046 | // Otherwise, there is an index. The computation we will do will be modulo | ||||
5047 | // the pointer size, so get it. | ||||
5048 | uint64_t PtrSizeMask = ~0ULL >> (64-IntPtrWidth); | ||||
5049 | |||||
5050 | Offset &= PtrSizeMask; | ||||
5051 | VariableScale &= PtrSizeMask; | ||||
5052 | |||||
5053 | // To do this transformation, any constant index must be a multiple of the | ||||
5054 | // variable scale factor. For example, we can evaluate "12 + 4*i" as "3 + i", | ||||
5055 | // but we can't evaluate "10 + 3*i" in terms of i. Check that the offset is a | ||||
5056 | // multiple of the variable scale. | ||||
5057 | int64_t NewOffs = Offset / (int64_t)VariableScale; | ||||
5058 | if (Offset != NewOffs*(int64_t)VariableScale) | ||||
5059 | return 0; | ||||
5060 | |||||
5061 | // Okay, we can do this evaluation. Start by converting the index to intptr. | ||||
5062 | const Type *IntPtrTy = TD.getIntPtrType(); | ||||
5063 | if (VariableIdx->getType() != IntPtrTy) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5064 | VariableIdx = CastInst::CreateIntegerCast(VariableIdx, IntPtrTy, |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5065 | true /*SExt*/, |
5066 | VariableIdx->getNameStart(), &I); | ||||
5067 | Constant *OffsetVal = ConstantInt::get(IntPtrTy, NewOffs); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5068 | return BinaryOperator::CreateAdd(VariableIdx, OffsetVal, "offset", &I); |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5069 | } |
5070 | |||||
5071 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5072 | /// FoldGEPICmp - Fold comparisons between a GEP instruction and something |
5073 | /// else. At this point we know that the GEP is on the LHS of the comparison. | ||||
5074 | Instruction *InstCombiner::FoldGEPICmp(User *GEPLHS, Value *RHS, | ||||
5075 | ICmpInst::Predicate Cond, | ||||
5076 | Instruction &I) { | ||||
5077 | assert(dyn_castGetElementPtr(GEPLHS) && "LHS is not a getelementptr!"); | ||||
5078 | |||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5079 | // Look through bitcasts. |
5080 | if (BitCastInst *BCI = dyn_cast<BitCastInst>(RHS)) | ||||
5081 | RHS = BCI->getOperand(0); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5082 | |
5083 | Value *PtrBase = GEPLHS->getOperand(0); | ||||
5084 | if (PtrBase == RHS) { | ||||
Chris Lattner | af97d02 | 2008-02-05 04:45:32 +0000 | [diff] [blame] | 5085 | // ((gep Ptr, OFFSET) cmp Ptr) ---> (OFFSET cmp 0). |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5086 | // This transformation (ignoring the base and scales) is valid because we |
5087 | // know pointers can't overflow. See if we can output an optimized form. | ||||
5088 | Value *Offset = EvaluateGEPOffsetExpression(GEPLHS, I, *this); | ||||
5089 | |||||
5090 | // If not, synthesize the offset the hard way. | ||||
5091 | if (Offset == 0) | ||||
5092 | Offset = EmitGEPOffset(GEPLHS, I, *this); | ||||
Chris Lattner | af97d02 | 2008-02-05 04:45:32 +0000 | [diff] [blame] | 5093 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), Offset, |
5094 | Constant::getNullValue(Offset->getType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5095 | } else if (User *GEPRHS = dyn_castGetElementPtr(RHS)) { |
5096 | // If the base pointers are different, but the indices are the same, just | ||||
5097 | // compare the base pointer. | ||||
5098 | if (PtrBase != GEPRHS->getOperand(0)) { | ||||
5099 | bool IndicesTheSame = GEPLHS->getNumOperands()==GEPRHS->getNumOperands(); | ||||
5100 | IndicesTheSame &= GEPLHS->getOperand(0)->getType() == | ||||
5101 | GEPRHS->getOperand(0)->getType(); | ||||
5102 | if (IndicesTheSame) | ||||
5103 | for (unsigned i = 1, e = GEPLHS->getNumOperands(); i != e; ++i) | ||||
5104 | if (GEPLHS->getOperand(i) != GEPRHS->getOperand(i)) { | ||||
5105 | IndicesTheSame = false; | ||||
5106 | break; | ||||
5107 | } | ||||
5108 | |||||
5109 | // If all indices are the same, just compare the base pointers. | ||||
5110 | if (IndicesTheSame) | ||||
5111 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), | ||||
5112 | GEPLHS->getOperand(0), GEPRHS->getOperand(0)); | ||||
5113 | |||||
5114 | // Otherwise, the base pointers are different and the indices are | ||||
5115 | // different, bail out. | ||||
5116 | return 0; | ||||
5117 | } | ||||
5118 | |||||
5119 | // If one of the GEPs has all zero indices, recurse. | ||||
5120 | bool AllZeros = true; | ||||
5121 | for (unsigned i = 1, e = GEPLHS->getNumOperands(); i != e; ++i) | ||||
5122 | if (!isa<Constant>(GEPLHS->getOperand(i)) || | ||||
5123 | !cast<Constant>(GEPLHS->getOperand(i))->isNullValue()) { | ||||
5124 | AllZeros = false; | ||||
5125 | break; | ||||
5126 | } | ||||
5127 | if (AllZeros) | ||||
5128 | return FoldGEPICmp(GEPRHS, GEPLHS->getOperand(0), | ||||
5129 | ICmpInst::getSwappedPredicate(Cond), I); | ||||
5130 | |||||
5131 | // If the other GEP has all zero indices, recurse. | ||||
5132 | AllZeros = true; | ||||
5133 | for (unsigned i = 1, e = GEPRHS->getNumOperands(); i != e; ++i) | ||||
5134 | if (!isa<Constant>(GEPRHS->getOperand(i)) || | ||||
5135 | !cast<Constant>(GEPRHS->getOperand(i))->isNullValue()) { | ||||
5136 | AllZeros = false; | ||||
5137 | break; | ||||
5138 | } | ||||
5139 | if (AllZeros) | ||||
5140 | return FoldGEPICmp(GEPLHS, GEPRHS->getOperand(0), Cond, I); | ||||
5141 | |||||
5142 | if (GEPLHS->getNumOperands() == GEPRHS->getNumOperands()) { | ||||
5143 | // If the GEPs only differ by one index, compare it. | ||||
5144 | unsigned NumDifferences = 0; // Keep track of # differences. | ||||
5145 | unsigned DiffOperand = 0; // The operand that differs. | ||||
5146 | for (unsigned i = 1, e = GEPRHS->getNumOperands(); i != e; ++i) | ||||
5147 | if (GEPLHS->getOperand(i) != GEPRHS->getOperand(i)) { | ||||
5148 | if (GEPLHS->getOperand(i)->getType()->getPrimitiveSizeInBits() != | ||||
5149 | GEPRHS->getOperand(i)->getType()->getPrimitiveSizeInBits()) { | ||||
5150 | // Irreconcilable differences. | ||||
5151 | NumDifferences = 2; | ||||
5152 | break; | ||||
5153 | } else { | ||||
5154 | if (NumDifferences++) break; | ||||
5155 | DiffOperand = i; | ||||
5156 | } | ||||
5157 | } | ||||
5158 | |||||
5159 | if (NumDifferences == 0) // SAME GEP? | ||||
5160 | return ReplaceInstUsesWith(I, // No comparison is needed here. | ||||
Nick Lewycky | 2de09a9 | 2007-09-06 02:40:25 +0000 | [diff] [blame] | 5161 | ConstantInt::get(Type::Int1Ty, |
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 5162 | ICmpInst::isTrueWhenEqual(Cond))); |
Nick Lewycky | 2de09a9 | 2007-09-06 02:40:25 +0000 | [diff] [blame] | 5163 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5164 | else if (NumDifferences == 1) { |
5165 | Value *LHSV = GEPLHS->getOperand(DiffOperand); | ||||
5166 | Value *RHSV = GEPRHS->getOperand(DiffOperand); | ||||
5167 | // Make sure we do a signed comparison here. | ||||
5168 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), LHSV, RHSV); | ||||
5169 | } | ||||
5170 | } | ||||
5171 | |||||
5172 | // Only lower this if the icmp is the only user of the GEP or if we expect | ||||
5173 | // the result to fold to a constant! | ||||
5174 | if ((isa<ConstantExpr>(GEPLHS) || GEPLHS->hasOneUse()) && | ||||
5175 | (isa<ConstantExpr>(GEPRHS) || GEPRHS->hasOneUse())) { | ||||
5176 | // ((gep Ptr, OFFSET1) cmp (gep Ptr, OFFSET2) ---> (OFFSET1 cmp OFFSET2) | ||||
5177 | Value *L = EmitGEPOffset(GEPLHS, I, *this); | ||||
5178 | Value *R = EmitGEPOffset(GEPRHS, I, *this); | ||||
5179 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), L, R); | ||||
5180 | } | ||||
5181 | } | ||||
5182 | return 0; | ||||
5183 | } | ||||
5184 | |||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5185 | /// FoldFCmp_IntToFP_Cst - Fold fcmp ([us]itofp x, cst) if possible. |
5186 | /// | ||||
5187 | Instruction *InstCombiner::FoldFCmp_IntToFP_Cst(FCmpInst &I, | ||||
5188 | Instruction *LHSI, | ||||
5189 | Constant *RHSC) { | ||||
5190 | if (!isa<ConstantFP>(RHSC)) return 0; | ||||
5191 | const APFloat &RHS = cast<ConstantFP>(RHSC)->getValueAPF(); | ||||
5192 | |||||
5193 | // Get the width of the mantissa. We don't want to hack on conversions that | ||||
5194 | // might lose information from the integer, e.g. "i64 -> float" | ||||
Chris Lattner | 9ce836b | 2008-05-19 21:17:23 +0000 | [diff] [blame] | 5195 | int MantissaWidth = LHSI->getType()->getFPMantissaWidth(); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5196 | if (MantissaWidth == -1) return 0; // Unknown. |
5197 | |||||
5198 | // Check to see that the input is converted from an integer type that is small | ||||
5199 | // enough that preserves all bits. TODO: check here for "known" sign bits. | ||||
5200 | // This would allow us to handle (fptosi (x >>s 62) to float) if x is i64 f.e. | ||||
5201 | unsigned InputSize = LHSI->getOperand(0)->getType()->getPrimitiveSizeInBits(); | ||||
5202 | |||||
5203 | // If this is a uitofp instruction, we need an extra bit to hold the sign. | ||||
5204 | if (isa<UIToFPInst>(LHSI)) | ||||
5205 | ++InputSize; | ||||
5206 | |||||
5207 | // If the conversion would lose info, don't hack on this. | ||||
5208 | if ((int)InputSize > MantissaWidth) | ||||
5209 | return 0; | ||||
5210 | |||||
5211 | // Otherwise, we can potentially simplify the comparison. We know that it | ||||
5212 | // will always come through as an integer value and we know the constant is | ||||
5213 | // not a NAN (it would have been previously simplified). | ||||
5214 | assert(!RHS.isNaN() && "NaN comparison not already folded!"); | ||||
5215 | |||||
5216 | ICmpInst::Predicate Pred; | ||||
5217 | switch (I.getPredicate()) { | ||||
5218 | default: assert(0 && "Unexpected predicate!"); | ||||
5219 | case FCmpInst::FCMP_UEQ: | ||||
5220 | case FCmpInst::FCMP_OEQ: Pred = ICmpInst::ICMP_EQ; break; | ||||
5221 | case FCmpInst::FCMP_UGT: | ||||
5222 | case FCmpInst::FCMP_OGT: Pred = ICmpInst::ICMP_SGT; break; | ||||
5223 | case FCmpInst::FCMP_UGE: | ||||
5224 | case FCmpInst::FCMP_OGE: Pred = ICmpInst::ICMP_SGE; break; | ||||
5225 | case FCmpInst::FCMP_ULT: | ||||
5226 | case FCmpInst::FCMP_OLT: Pred = ICmpInst::ICMP_SLT; break; | ||||
5227 | case FCmpInst::FCMP_ULE: | ||||
5228 | case FCmpInst::FCMP_OLE: Pred = ICmpInst::ICMP_SLE; break; | ||||
5229 | case FCmpInst::FCMP_UNE: | ||||
5230 | case FCmpInst::FCMP_ONE: Pred = ICmpInst::ICMP_NE; break; | ||||
5231 | case FCmpInst::FCMP_ORD: | ||||
5232 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 1)); | ||||
5233 | case FCmpInst::FCMP_UNO: | ||||
5234 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 0)); | ||||
5235 | } | ||||
5236 | |||||
5237 | const IntegerType *IntTy = cast<IntegerType>(LHSI->getOperand(0)->getType()); | ||||
5238 | |||||
5239 | // Now we know that the APFloat is a normal number, zero or inf. | ||||
5240 | |||||
Chris Lattner | f13ff49 | 2008-05-20 03:50:52 +0000 | [diff] [blame] | 5241 | // 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] | 5242 | // comparing an i8 to 300.0. |
5243 | unsigned IntWidth = IntTy->getPrimitiveSizeInBits(); | ||||
5244 | |||||
5245 | // If the RHS value is > SignedMax, fold the comparison. This handles +INF | ||||
5246 | // and large values. | ||||
5247 | APFloat SMax(RHS.getSemantics(), APFloat::fcZero, false); | ||||
5248 | SMax.convertFromAPInt(APInt::getSignedMaxValue(IntWidth), true, | ||||
5249 | APFloat::rmNearestTiesToEven); | ||||
5250 | if (SMax.compare(RHS) == APFloat::cmpLessThan) { // smax < 13123.0 | ||||
Chris Lattner | 82a8000 | 2008-05-24 04:06:28 +0000 | [diff] [blame] | 5251 | if (Pred == ICmpInst::ICMP_NE || Pred == ICmpInst::ICMP_SLT || |
5252 | Pred == ICmpInst::ICMP_SLE) | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5253 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 1)); |
5254 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 0)); | ||||
5255 | } | ||||
5256 | |||||
5257 | // See if the RHS value is < SignedMin. | ||||
5258 | APFloat SMin(RHS.getSemantics(), APFloat::fcZero, false); | ||||
5259 | SMin.convertFromAPInt(APInt::getSignedMinValue(IntWidth), true, | ||||
5260 | APFloat::rmNearestTiesToEven); | ||||
5261 | if (SMin.compare(RHS) == APFloat::cmpGreaterThan) { // smin > 12312.0 | ||||
Chris Lattner | 82a8000 | 2008-05-24 04:06:28 +0000 | [diff] [blame] | 5262 | if (Pred == ICmpInst::ICMP_NE || Pred == ICmpInst::ICMP_SGT || |
5263 | Pred == ICmpInst::ICMP_SGE) | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5264 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 1)); |
5265 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 0)); | ||||
5266 | } | ||||
5267 | |||||
5268 | // Okay, now we know that the FP constant fits in the range [SMIN, SMAX] but | ||||
5269 | // it may still be fractional. See if it is fractional by casting the FP | ||||
5270 | // value to the integer value and back, checking for equality. Don't do this | ||||
5271 | // for zero, because -0.0 is not fractional. | ||||
5272 | Constant *RHSInt = ConstantExpr::getFPToSI(RHSC, IntTy); | ||||
5273 | if (!RHS.isZero() && | ||||
5274 | ConstantExpr::getSIToFP(RHSInt, RHSC->getType()) != RHSC) { | ||||
5275 | // If we had a comparison against a fractional value, we have to adjust | ||||
5276 | // the compare predicate and sometimes the value. RHSC is rounded towards | ||||
5277 | // zero at this point. | ||||
5278 | switch (Pred) { | ||||
5279 | default: assert(0 && "Unexpected integer comparison!"); | ||||
5280 | case ICmpInst::ICMP_NE: // (float)int != 4.4 --> true | ||||
5281 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 1)); | ||||
5282 | case ICmpInst::ICMP_EQ: // (float)int == 4.4 --> false | ||||
5283 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 0)); | ||||
5284 | case ICmpInst::ICMP_SLE: | ||||
5285 | // (float)int <= 4.4 --> int <= 4 | ||||
5286 | // (float)int <= -4.4 --> int < -4 | ||||
5287 | if (RHS.isNegative()) | ||||
5288 | Pred = ICmpInst::ICMP_SLT; | ||||
5289 | break; | ||||
5290 | case ICmpInst::ICMP_SLT: | ||||
5291 | // (float)int < -4.4 --> int < -4 | ||||
5292 | // (float)int < 4.4 --> int <= 4 | ||||
5293 | if (!RHS.isNegative()) | ||||
5294 | Pred = ICmpInst::ICMP_SLE; | ||||
5295 | break; | ||||
5296 | case ICmpInst::ICMP_SGT: | ||||
5297 | // (float)int > 4.4 --> int > 4 | ||||
5298 | // (float)int > -4.4 --> int >= -4 | ||||
5299 | if (RHS.isNegative()) | ||||
5300 | Pred = ICmpInst::ICMP_SGE; | ||||
5301 | break; | ||||
5302 | case ICmpInst::ICMP_SGE: | ||||
5303 | // (float)int >= -4.4 --> int >= -4 | ||||
5304 | // (float)int >= 4.4 --> int > 4 | ||||
5305 | if (!RHS.isNegative()) | ||||
5306 | Pred = ICmpInst::ICMP_SGT; | ||||
5307 | break; | ||||
5308 | } | ||||
5309 | } | ||||
5310 | |||||
5311 | // Lower this FP comparison into an appropriate integer version of the | ||||
5312 | // comparison. | ||||
5313 | return new ICmpInst(Pred, LHSI->getOperand(0), RHSInt); | ||||
5314 | } | ||||
5315 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5316 | Instruction *InstCombiner::visitFCmpInst(FCmpInst &I) { |
5317 | bool Changed = SimplifyCompare(I); | ||||
5318 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
5319 | |||||
5320 | // Fold trivial predicates. | ||||
5321 | if (I.getPredicate() == FCmpInst::FCMP_FALSE) | ||||
5322 | return ReplaceInstUsesWith(I, Constant::getNullValue(Type::Int1Ty)); | ||||
5323 | if (I.getPredicate() == FCmpInst::FCMP_TRUE) | ||||
5324 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 1)); | ||||
5325 | |||||
5326 | // Simplify 'fcmp pred X, X' | ||||
5327 | if (Op0 == Op1) { | ||||
5328 | switch (I.getPredicate()) { | ||||
5329 | default: assert(0 && "Unknown predicate!"); | ||||
5330 | case FCmpInst::FCMP_UEQ: // True if unordered or equal | ||||
5331 | case FCmpInst::FCMP_UGE: // True if unordered, greater than, or equal | ||||
5332 | case FCmpInst::FCMP_ULE: // True if unordered, less than, or equal | ||||
5333 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 1)); | ||||
5334 | case FCmpInst::FCMP_OGT: // True if ordered and greater than | ||||
5335 | case FCmpInst::FCMP_OLT: // True if ordered and less than | ||||
5336 | case FCmpInst::FCMP_ONE: // True if ordered and operands are unequal | ||||
5337 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 0)); | ||||
5338 | |||||
5339 | case FCmpInst::FCMP_UNO: // True if unordered: isnan(X) | isnan(Y) | ||||
5340 | case FCmpInst::FCMP_ULT: // True if unordered or less than | ||||
5341 | case FCmpInst::FCMP_UGT: // True if unordered or greater than | ||||
5342 | case FCmpInst::FCMP_UNE: // True if unordered or not equal | ||||
5343 | // Canonicalize these to be 'fcmp uno %X, 0.0'. | ||||
5344 | I.setPredicate(FCmpInst::FCMP_UNO); | ||||
5345 | I.setOperand(1, Constant::getNullValue(Op0->getType())); | ||||
5346 | return &I; | ||||
5347 | |||||
5348 | case FCmpInst::FCMP_ORD: // True if ordered (no nans) | ||||
5349 | case FCmpInst::FCMP_OEQ: // True if ordered and equal | ||||
5350 | case FCmpInst::FCMP_OGE: // True if ordered and greater than or equal | ||||
5351 | case FCmpInst::FCMP_OLE: // True if ordered and less than or equal | ||||
5352 | // Canonicalize these to be 'fcmp ord %X, 0.0'. | ||||
5353 | I.setPredicate(FCmpInst::FCMP_ORD); | ||||
5354 | I.setOperand(1, Constant::getNullValue(Op0->getType())); | ||||
5355 | return &I; | ||||
5356 | } | ||||
5357 | } | ||||
5358 | |||||
5359 | if (isa<UndefValue>(Op1)) // fcmp pred X, undef -> undef | ||||
5360 | return ReplaceInstUsesWith(I, UndefValue::get(Type::Int1Ty)); | ||||
5361 | |||||
5362 | // Handle fcmp with constant RHS | ||||
5363 | if (Constant *RHSC = dyn_cast<Constant>(Op1)) { | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5364 | // If the constant is a nan, see if we can fold the comparison based on it. |
5365 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHSC)) { | ||||
5366 | if (CFP->getValueAPF().isNaN()) { | ||||
5367 | if (FCmpInst::isOrdered(I.getPredicate())) // True if ordered and... | ||||
5368 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 0)); | ||||
Chris Lattner | f13ff49 | 2008-05-20 03:50:52 +0000 | [diff] [blame] | 5369 | assert(FCmpInst::isUnordered(I.getPredicate()) && |
5370 | "Comparison must be either ordered or unordered!"); | ||||
5371 | // True if unordered. | ||||
5372 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, 1)); | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5373 | } |
5374 | } | ||||
5375 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5376 | if (Instruction *LHSI = dyn_cast<Instruction>(Op0)) |
5377 | switch (LHSI->getOpcode()) { | ||||
5378 | case Instruction::PHI: | ||||
Chris Lattner | a2417ba | 2008-06-08 20:52:11 +0000 | [diff] [blame] | 5379 | // Only fold fcmp into the PHI if the phi and fcmp are in the same |
5380 | // block. If in the same block, we're encouraging jump threading. If | ||||
5381 | // not, we are just pessimizing the code by making an i1 phi. | ||||
5382 | if (LHSI->getParent() == I.getParent()) | ||||
5383 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
5384 | return NV; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5385 | break; |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5386 | case Instruction::SIToFP: |
5387 | case Instruction::UIToFP: | ||||
5388 | if (Instruction *NV = FoldFCmp_IntToFP_Cst(I, LHSI, RHSC)) | ||||
5389 | return NV; | ||||
5390 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5391 | case Instruction::Select: |
5392 | // If either operand of the select is a constant, we can fold the | ||||
5393 | // comparison into the select arms, which will cause one to be | ||||
5394 | // constant folded and the select turned into a bitwise or. | ||||
5395 | Value *Op1 = 0, *Op2 = 0; | ||||
5396 | if (LHSI->hasOneUse()) { | ||||
5397 | if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(1))) { | ||||
5398 | // Fold the known value into the constant operand. | ||||
5399 | Op1 = ConstantExpr::getCompare(I.getPredicate(), C, RHSC); | ||||
5400 | // Insert a new FCmp of the other select operand. | ||||
5401 | Op2 = InsertNewInstBefore(new FCmpInst(I.getPredicate(), | ||||
5402 | LHSI->getOperand(2), RHSC, | ||||
5403 | I.getName()), I); | ||||
5404 | } else if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(2))) { | ||||
5405 | // Fold the known value into the constant operand. | ||||
5406 | Op2 = ConstantExpr::getCompare(I.getPredicate(), C, RHSC); | ||||
5407 | // Insert a new FCmp of the other select operand. | ||||
5408 | Op1 = InsertNewInstBefore(new FCmpInst(I.getPredicate(), | ||||
5409 | LHSI->getOperand(1), RHSC, | ||||
5410 | I.getName()), I); | ||||
5411 | } | ||||
5412 | } | ||||
5413 | |||||
5414 | if (Op1) | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 5415 | return SelectInst::Create(LHSI->getOperand(0), Op1, Op2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5416 | break; |
5417 | } | ||||
5418 | } | ||||
5419 | |||||
5420 | return Changed ? &I : 0; | ||||
5421 | } | ||||
5422 | |||||
5423 | Instruction *InstCombiner::visitICmpInst(ICmpInst &I) { | ||||
5424 | bool Changed = SimplifyCompare(I); | ||||
5425 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
5426 | const Type *Ty = Op0->getType(); | ||||
5427 | |||||
5428 | // icmp X, X | ||||
5429 | if (Op0 == Op1) | ||||
5430 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, | ||||
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 5431 | I.isTrueWhenEqual())); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5432 | |
5433 | if (isa<UndefValue>(Op1)) // X icmp undef -> undef | ||||
5434 | return ReplaceInstUsesWith(I, UndefValue::get(Type::Int1Ty)); | ||||
Christopher Lamb | f78cd32 | 2007-12-18 21:32:20 +0000 | [diff] [blame] | 5435 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5436 | // icmp <global/alloca*/null>, <global/alloca*/null> - Global/Stack value |
5437 | // addresses never equal each other! We already know that Op0 != Op1. | ||||
5438 | if ((isa<GlobalValue>(Op0) || isa<AllocaInst>(Op0) || | ||||
5439 | isa<ConstantPointerNull>(Op0)) && | ||||
5440 | (isa<GlobalValue>(Op1) || isa<AllocaInst>(Op1) || | ||||
5441 | isa<ConstantPointerNull>(Op1))) | ||||
5442 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, | ||||
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 5443 | !I.isTrueWhenEqual())); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5444 | |
5445 | // icmp's with boolean values can always be turned into bitwise operations | ||||
5446 | if (Ty == Type::Int1Ty) { | ||||
5447 | switch (I.getPredicate()) { | ||||
5448 | default: assert(0 && "Invalid icmp instruction!"); | ||||
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5449 | case ICmpInst::ICMP_EQ: { // icmp eq i1 A, B -> ~(A^B) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5450 | Instruction *Xor = BinaryOperator::CreateXor(Op0, Op1, I.getName()+"tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5451 | InsertNewInstBefore(Xor, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5452 | return BinaryOperator::CreateNot(Xor); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5453 | } |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5454 | case ICmpInst::ICMP_NE: // icmp eq i1 A, B -> A^B |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5455 | return BinaryOperator::CreateXor(Op0, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5456 | |
5457 | case ICmpInst::ICMP_UGT: | ||||
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5458 | std::swap(Op0, Op1); // Change icmp ugt -> icmp ult |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5459 | // FALL THROUGH |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5460 | case ICmpInst::ICMP_ULT:{ // icmp ult i1 A, B -> ~A & B |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5461 | Instruction *Not = BinaryOperator::CreateNot(Op0, I.getName()+"tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5462 | InsertNewInstBefore(Not, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5463 | return BinaryOperator::CreateAnd(Not, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5464 | } |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5465 | case ICmpInst::ICMP_SGT: |
5466 | std::swap(Op0, Op1); // Change icmp sgt -> icmp slt | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5467 | // FALL THROUGH |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5468 | case ICmpInst::ICMP_SLT: { // icmp slt i1 A, B -> A & ~B |
5469 | Instruction *Not = BinaryOperator::CreateNot(Op1, I.getName()+"tmp"); | ||||
5470 | InsertNewInstBefore(Not, I); | ||||
5471 | return BinaryOperator::CreateAnd(Not, Op0); | ||||
5472 | } | ||||
5473 | case ICmpInst::ICMP_UGE: | ||||
5474 | std::swap(Op0, Op1); // Change icmp uge -> icmp ule | ||||
5475 | // FALL THROUGH | ||||
5476 | case ICmpInst::ICMP_ULE: { // icmp ule i1 A, B -> ~A | B | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5477 | Instruction *Not = BinaryOperator::CreateNot(Op0, I.getName()+"tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5478 | InsertNewInstBefore(Not, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5479 | return BinaryOperator::CreateOr(Not, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5480 | } |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5481 | case ICmpInst::ICMP_SGE: |
5482 | std::swap(Op0, Op1); // Change icmp sge -> icmp sle | ||||
5483 | // FALL THROUGH | ||||
5484 | case ICmpInst::ICMP_SLE: { // icmp sle i1 A, B -> A | ~B | ||||
5485 | Instruction *Not = BinaryOperator::CreateNot(Op1, I.getName()+"tmp"); | ||||
5486 | InsertNewInstBefore(Not, I); | ||||
5487 | return BinaryOperator::CreateOr(Not, Op0); | ||||
5488 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5489 | } |
5490 | } | ||||
5491 | |||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 5492 | // See if we are doing a comparison with a constant. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5493 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) { |
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 5494 | Value *A, *B; |
Christopher Lamb | fa6b310 | 2007-12-20 07:21:11 +0000 | [diff] [blame] | 5495 | |
Chris Lattner | be6c54a | 2008-01-05 01:18:20 +0000 | [diff] [blame] | 5496 | // (icmp ne/eq (sub A B) 0) -> (icmp ne/eq A, B) |
5497 | if (I.isEquality() && CI->isNullValue() && | ||||
5498 | match(Op0, m_Sub(m_Value(A), m_Value(B)))) { | ||||
5499 | // (icmp cond A B) if cond is equality | ||||
5500 | return new ICmpInst(I.getPredicate(), A, B); | ||||
Owen Anderson | 42f61ed | 2007-12-28 07:42:12 +0000 | [diff] [blame] | 5501 | } |
Christopher Lamb | fa6b310 | 2007-12-20 07:21:11 +0000 | [diff] [blame] | 5502 | |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 5503 | // If we have an icmp le or icmp ge instruction, turn it into the |
5504 | // appropriate icmp lt or icmp gt instruction. This allows us to rely on | ||||
5505 | // them being folded in the code below. | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5506 | switch (I.getPredicate()) { |
5507 | default: break; | ||||
5508 | case ICmpInst::ICMP_ULE: | ||||
5509 | if (CI->isMaxValue(false)) // A <=u MAX -> TRUE | ||||
5510 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5511 | return new ICmpInst(ICmpInst::ICMP_ULT, Op0, AddOne(CI)); | ||||
5512 | case ICmpInst::ICMP_SLE: | ||||
5513 | if (CI->isMaxValue(true)) // A <=s MAX -> TRUE | ||||
5514 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5515 | return new ICmpInst(ICmpInst::ICMP_SLT, Op0, AddOne(CI)); | ||||
5516 | case ICmpInst::ICMP_UGE: | ||||
5517 | if (CI->isMinValue(false)) // A >=u MIN -> TRUE | ||||
5518 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5519 | return new ICmpInst( ICmpInst::ICMP_UGT, Op0, SubOne(CI)); | ||||
5520 | case ICmpInst::ICMP_SGE: | ||||
5521 | if (CI->isMinValue(true)) // A >=s MIN -> TRUE | ||||
5522 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5523 | return new ICmpInst(ICmpInst::ICMP_SGT, Op0, SubOne(CI)); | ||||
5524 | } | ||||
5525 | |||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5526 | // See if we can fold the comparison based on range information we can get |
5527 | // by checking whether bits are known to be zero or one in the input. | ||||
5528 | uint32_t BitWidth = cast<IntegerType>(Ty)->getBitWidth(); | ||||
5529 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
5530 | |||||
5531 | // If this comparison is a normal comparison, it demands all | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5532 | // 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] | 5533 | bool UnusedBit; |
5534 | bool isSignBit = isSignBitCheck(I.getPredicate(), CI, UnusedBit); | ||||
5535 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5536 | if (SimplifyDemandedBits(Op0, |
5537 | isSignBit ? APInt::getSignBit(BitWidth) | ||||
5538 | : APInt::getAllOnesValue(BitWidth), | ||||
5539 | KnownZero, KnownOne, 0)) | ||||
5540 | return &I; | ||||
5541 | |||||
5542 | // 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] | 5543 | // in. Compute the Min, Max and RHS values based on the known bits. For the |
5544 | // EQ and NE we use unsigned values. | ||||
5545 | APInt Min(BitWidth, 0), Max(BitWidth, 0); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5546 | if (ICmpInst::isSignedPredicate(I.getPredicate())) |
5547 | ComputeSignedMinMaxValuesFromKnownBits(Ty, KnownZero, KnownOne, Min, Max); | ||||
5548 | else | ||||
5549 | ComputeUnsignedMinMaxValuesFromKnownBits(Ty, KnownZero, KnownOne,Min,Max); | ||||
5550 | |||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5551 | // If Min and Max are known to be the same, then SimplifyDemandedBits |
5552 | // figured out that the LHS is a constant. Just constant fold this now so | ||||
5553 | // that code below can assume that Min != Max. | ||||
5554 | if (Min == Max) | ||||
5555 | return ReplaceInstUsesWith(I, ConstantExpr::getICmp(I.getPredicate(), | ||||
5556 | ConstantInt::get(Min), | ||||
5557 | CI)); | ||||
5558 | |||||
5559 | // Based on the range information we know about the LHS, see if we can | ||||
5560 | // simplify this comparison. For example, (x&4) < 8 is always true. | ||||
5561 | const APInt &RHSVal = CI->getValue(); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5562 | switch (I.getPredicate()) { // LE/GE have been folded already. |
5563 | default: assert(0 && "Unknown icmp opcode!"); | ||||
5564 | case ICmpInst::ICMP_EQ: | ||||
5565 | if (Max.ult(RHSVal) || Min.ugt(RHSVal)) | ||||
5566 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
5567 | break; | ||||
5568 | case ICmpInst::ICMP_NE: | ||||
5569 | if (Max.ult(RHSVal) || Min.ugt(RHSVal)) | ||||
5570 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5571 | break; | ||||
5572 | case ICmpInst::ICMP_ULT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5573 | if (Max.ult(RHSVal)) // A <u C -> true iff max(A) < C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5574 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5575 | if (Min.uge(RHSVal)) // A <u C -> false iff min(A) >= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5576 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5577 | if (RHSVal == Max) // A <u MAX -> A != MAX |
5578 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
5579 | if (RHSVal == Min+1) // A <u MIN+1 -> A == MIN | ||||
5580 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, SubOne(CI)); | ||||
5581 | |||||
5582 | // (x <u 2147483648) -> (x >s -1) -> true if sign bit clear | ||||
5583 | if (CI->isMinValue(true)) | ||||
5584 | return new ICmpInst(ICmpInst::ICMP_SGT, Op0, | ||||
5585 | ConstantInt::getAllOnesValue(Op0->getType())); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5586 | break; |
5587 | case ICmpInst::ICMP_UGT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5588 | if (Min.ugt(RHSVal)) // A >u C -> true iff min(A) > C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5589 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5590 | if (Max.ule(RHSVal)) // A >u C -> false iff max(A) <= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5591 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5592 | |
5593 | if (RHSVal == Min) // A >u MIN -> A != MIN | ||||
5594 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
5595 | if (RHSVal == Max-1) // A >u MAX-1 -> A == MAX | ||||
5596 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, AddOne(CI)); | ||||
5597 | |||||
5598 | // (x >u 2147483647) -> (x <s 0) -> true if sign bit set | ||||
5599 | if (CI->isMaxValue(true)) | ||||
5600 | return new ICmpInst(ICmpInst::ICMP_SLT, Op0, | ||||
5601 | ConstantInt::getNullValue(Op0->getType())); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5602 | break; |
5603 | case ICmpInst::ICMP_SLT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5604 | if (Max.slt(RHSVal)) // A <s C -> true iff max(A) < C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5605 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | 611b43e | 2008-07-11 06:40:29 +0000 | [diff] [blame] | 5606 | if (Min.sge(RHSVal)) // A <s C -> false iff min(A) >= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5607 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5608 | if (RHSVal == Max) // A <s MAX -> A != MAX |
5609 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
Chris Lattner | 3496f3e | 2008-07-11 06:36:01 +0000 | [diff] [blame] | 5610 | if (RHSVal == Min+1) // A <s MIN+1 -> A == MIN |
Chris Lattner | 55ab315 | 2008-07-11 06:38:16 +0000 | [diff] [blame] | 5611 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, SubOne(CI)); |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5612 | break; |
5613 | case ICmpInst::ICMP_SGT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5614 | if (Min.sgt(RHSVal)) // A >s C -> true iff min(A) > C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5615 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5616 | if (Max.sle(RHSVal)) // A >s C -> false iff max(A) <= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5617 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5618 | |
5619 | if (RHSVal == Min) // A >s MIN -> A != MIN | ||||
5620 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
5621 | if (RHSVal == Max-1) // A >s MAX-1 -> A == MAX | ||||
5622 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, AddOne(CI)); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5623 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5624 | } |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 5625 | } |
5626 | |||||
5627 | // Test if the ICmpInst instruction is used exclusively by a select as | ||||
5628 | // part of a minimum or maximum operation. If so, refrain from doing | ||||
5629 | // any other folding. This helps out other analyses which understand | ||||
5630 | // non-obfuscated minimum and maximum idioms, such as ScalarEvolution | ||||
5631 | // and CodeGen. And in this case, at least one of the comparison | ||||
5632 | // operands has at least one user besides the compare (the select), | ||||
5633 | // which would often largely negate the benefit of folding anyway. | ||||
5634 | if (I.hasOneUse()) | ||||
5635 | if (SelectInst *SI = dyn_cast<SelectInst>(*I.use_begin())) | ||||
5636 | if ((SI->getOperand(1) == Op0 && SI->getOperand(2) == Op1) || | ||||
5637 | (SI->getOperand(2) == Op0 && SI->getOperand(1) == Op1)) | ||||
5638 | return 0; | ||||
5639 | |||||
5640 | // See if we are doing a comparison between a constant and an instruction that | ||||
5641 | // can be folded into the comparison. | ||||
5642 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5643 | // Since the RHS is a ConstantInt (CI), if the left hand side is an |
5644 | // instruction, see if that instruction also has constants so that the | ||||
5645 | // instruction can be folded into the icmp | ||||
5646 | if (Instruction *LHSI = dyn_cast<Instruction>(Op0)) | ||||
5647 | if (Instruction *Res = visitICmpInstWithInstAndIntCst(I, LHSI, CI)) | ||||
5648 | return Res; | ||||
5649 | } | ||||
5650 | |||||
5651 | // Handle icmp with constant (but not simple integer constant) RHS | ||||
5652 | if (Constant *RHSC = dyn_cast<Constant>(Op1)) { | ||||
5653 | if (Instruction *LHSI = dyn_cast<Instruction>(Op0)) | ||||
5654 | switch (LHSI->getOpcode()) { | ||||
5655 | case Instruction::GetElementPtr: | ||||
5656 | if (RHSC->isNullValue()) { | ||||
5657 | // icmp pred GEP (P, int 0, int 0, int 0), null -> icmp pred P, null | ||||
5658 | bool isAllZeros = true; | ||||
5659 | for (unsigned i = 1, e = LHSI->getNumOperands(); i != e; ++i) | ||||
5660 | if (!isa<Constant>(LHSI->getOperand(i)) || | ||||
5661 | !cast<Constant>(LHSI->getOperand(i))->isNullValue()) { | ||||
5662 | isAllZeros = false; | ||||
5663 | break; | ||||
5664 | } | ||||
5665 | if (isAllZeros) | ||||
5666 | return new ICmpInst(I.getPredicate(), LHSI->getOperand(0), | ||||
5667 | Constant::getNullValue(LHSI->getOperand(0)->getType())); | ||||
5668 | } | ||||
5669 | break; | ||||
5670 | |||||
5671 | case Instruction::PHI: | ||||
Chris Lattner | a2417ba | 2008-06-08 20:52:11 +0000 | [diff] [blame] | 5672 | // Only fold icmp into the PHI if the phi and fcmp are in the same |
5673 | // block. If in the same block, we're encouraging jump threading. If | ||||
5674 | // not, we are just pessimizing the code by making an i1 phi. | ||||
5675 | if (LHSI->getParent() == I.getParent()) | ||||
5676 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
5677 | return NV; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5678 | break; |
5679 | case Instruction::Select: { | ||||
5680 | // If either operand of the select is a constant, we can fold the | ||||
5681 | // comparison into the select arms, which will cause one to be | ||||
5682 | // constant folded and the select turned into a bitwise or. | ||||
5683 | Value *Op1 = 0, *Op2 = 0; | ||||
5684 | if (LHSI->hasOneUse()) { | ||||
5685 | if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(1))) { | ||||
5686 | // Fold the known value into the constant operand. | ||||
5687 | Op1 = ConstantExpr::getICmp(I.getPredicate(), C, RHSC); | ||||
5688 | // Insert a new ICmp of the other select operand. | ||||
5689 | Op2 = InsertNewInstBefore(new ICmpInst(I.getPredicate(), | ||||
5690 | LHSI->getOperand(2), RHSC, | ||||
5691 | I.getName()), I); | ||||
5692 | } else if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(2))) { | ||||
5693 | // Fold the known value into the constant operand. | ||||
5694 | Op2 = ConstantExpr::getICmp(I.getPredicate(), C, RHSC); | ||||
5695 | // Insert a new ICmp of the other select operand. | ||||
5696 | Op1 = InsertNewInstBefore(new ICmpInst(I.getPredicate(), | ||||
5697 | LHSI->getOperand(1), RHSC, | ||||
5698 | I.getName()), I); | ||||
5699 | } | ||||
5700 | } | ||||
5701 | |||||
5702 | if (Op1) | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 5703 | return SelectInst::Create(LHSI->getOperand(0), Op1, Op2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5704 | break; |
5705 | } | ||||
5706 | case Instruction::Malloc: | ||||
5707 | // If we have (malloc != null), and if the malloc has a single use, we | ||||
5708 | // can assume it is successful and remove the malloc. | ||||
5709 | if (LHSI->hasOneUse() && isa<ConstantPointerNull>(RHSC)) { | ||||
5710 | AddToWorkList(LHSI); | ||||
5711 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, | ||||
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 5712 | !I.isTrueWhenEqual())); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5713 | } |
5714 | break; | ||||
5715 | } | ||||
5716 | } | ||||
5717 | |||||
5718 | // If we can optimize a 'icmp GEP, P' or 'icmp P, GEP', do so now. | ||||
5719 | if (User *GEP = dyn_castGetElementPtr(Op0)) | ||||
5720 | if (Instruction *NI = FoldGEPICmp(GEP, Op1, I.getPredicate(), I)) | ||||
5721 | return NI; | ||||
5722 | if (User *GEP = dyn_castGetElementPtr(Op1)) | ||||
5723 | if (Instruction *NI = FoldGEPICmp(GEP, Op0, | ||||
5724 | ICmpInst::getSwappedPredicate(I.getPredicate()), I)) | ||||
5725 | return NI; | ||||
5726 | |||||
5727 | // Test to see if the operands of the icmp are casted versions of other | ||||
5728 | // values. If the ptr->ptr cast can be stripped off both arguments, we do so | ||||
5729 | // now. | ||||
5730 | if (BitCastInst *CI = dyn_cast<BitCastInst>(Op0)) { | ||||
5731 | if (isa<PointerType>(Op0->getType()) && | ||||
5732 | (isa<Constant>(Op1) || isa<BitCastInst>(Op1))) { | ||||
5733 | // We keep moving the cast from the left operand over to the right | ||||
5734 | // operand, where it can often be eliminated completely. | ||||
5735 | Op0 = CI->getOperand(0); | ||||
5736 | |||||
5737 | // If operand #1 is a bitcast instruction, it must also be a ptr->ptr cast | ||||
5738 | // so eliminate it as well. | ||||
5739 | if (BitCastInst *CI2 = dyn_cast<BitCastInst>(Op1)) | ||||
5740 | Op1 = CI2->getOperand(0); | ||||
5741 | |||||
5742 | // 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] | 5743 | if (Op0->getType() != Op1->getType()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5744 | if (Constant *Op1C = dyn_cast<Constant>(Op1)) { |
5745 | Op1 = ConstantExpr::getBitCast(Op1C, Op0->getType()); | ||||
5746 | } else { | ||||
5747 | // Otherwise, cast the RHS right before the icmp | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 5748 | Op1 = InsertBitCastBefore(Op1, Op0->getType(), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5749 | } |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 5750 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5751 | return new ICmpInst(I.getPredicate(), Op0, Op1); |
5752 | } | ||||
5753 | } | ||||
5754 | |||||
5755 | if (isa<CastInst>(Op0)) { | ||||
5756 | // Handle the special case of: icmp (cast bool to X), <cst> | ||||
5757 | // This comes up when you have code like | ||||
5758 | // int X = A < B; | ||||
5759 | // if (X) ... | ||||
5760 | // For generality, we handle any zero-extension of any operand comparison | ||||
5761 | // with a constant or another cast from the same type. | ||||
5762 | if (isa<ConstantInt>(Op1) || isa<CastInst>(Op1)) | ||||
5763 | if (Instruction *R = visitICmpInstWithCastAndCast(I)) | ||||
5764 | return R; | ||||
5765 | } | ||||
5766 | |||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 5767 | // See if it's the same type of instruction on the left and right. |
5768 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) { | ||||
5769 | if (BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1)) { | ||||
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 5770 | if (Op0I->getOpcode() == Op1I->getOpcode() && Op0I->hasOneUse() && |
5771 | Op1I->hasOneUse() && Op0I->getOperand(1) == Op1I->getOperand(1) && | ||||
5772 | I.isEquality()) { | ||||
Nick Lewycky | cfadfbd | 2008-09-03 06:24:21 +0000 | [diff] [blame] | 5773 | switch (Op0I->getOpcode()) { |
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 5774 | default: break; |
5775 | case Instruction::Add: | ||||
5776 | case Instruction::Sub: | ||||
5777 | case Instruction::Xor: | ||||
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 5778 | // a+x icmp eq/ne b+x --> a icmp b |
5779 | return new ICmpInst(I.getPredicate(), Op0I->getOperand(0), | ||||
5780 | Op1I->getOperand(0)); | ||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 5781 | break; |
5782 | case Instruction::Mul: | ||||
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 5783 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) { |
5784 | // a * Cst icmp eq/ne b * Cst --> a & Mask icmp b & Mask | ||||
5785 | // Mask = -1 >> count-trailing-zeros(Cst). | ||||
5786 | if (!CI->isZero() && !CI->isOne()) { | ||||
5787 | const APInt &AP = CI->getValue(); | ||||
5788 | ConstantInt *Mask = ConstantInt::get( | ||||
5789 | APInt::getLowBitsSet(AP.getBitWidth(), | ||||
5790 | AP.getBitWidth() - | ||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 5791 | AP.countTrailingZeros())); |
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 5792 | Instruction *And1 = BinaryOperator::CreateAnd(Op0I->getOperand(0), |
5793 | Mask); | ||||
5794 | Instruction *And2 = BinaryOperator::CreateAnd(Op1I->getOperand(0), | ||||
5795 | Mask); | ||||
5796 | InsertNewInstBefore(And1, I); | ||||
5797 | InsertNewInstBefore(And2, I); | ||||
5798 | return new ICmpInst(I.getPredicate(), And1, And2); | ||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 5799 | } |
5800 | } | ||||
5801 | break; | ||||
5802 | } | ||||
5803 | } | ||||
5804 | } | ||||
5805 | } | ||||
5806 | |||||
Chris Lattner | a4e1eef | 2008-05-09 05:19:28 +0000 | [diff] [blame] | 5807 | // ~x < ~y --> y < x |
5808 | { Value *A, *B; | ||||
5809 | if (match(Op0, m_Not(m_Value(A))) && | ||||
5810 | match(Op1, m_Not(m_Value(B)))) | ||||
5811 | return new ICmpInst(I.getPredicate(), B, A); | ||||
5812 | } | ||||
5813 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5814 | if (I.isEquality()) { |
5815 | Value *A, *B, *C, *D; | ||||
Chris Lattner | a4e1eef | 2008-05-09 05:19:28 +0000 | [diff] [blame] | 5816 | |
5817 | // -x == -y --> x == y | ||||
5818 | if (match(Op0, m_Neg(m_Value(A))) && | ||||
5819 | match(Op1, m_Neg(m_Value(B)))) | ||||
5820 | return new ICmpInst(I.getPredicate(), A, B); | ||||
5821 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5822 | if (match(Op0, m_Xor(m_Value(A), m_Value(B)))) { |
5823 | if (A == Op1 || B == Op1) { // (A^B) == A -> B == 0 | ||||
5824 | Value *OtherVal = A == Op1 ? B : A; | ||||
5825 | return new ICmpInst(I.getPredicate(), OtherVal, | ||||
5826 | Constant::getNullValue(A->getType())); | ||||
5827 | } | ||||
5828 | |||||
5829 | if (match(Op1, m_Xor(m_Value(C), m_Value(D)))) { | ||||
5830 | // A^c1 == C^c2 --> A == C^(c1^c2) | ||||
5831 | if (ConstantInt *C1 = dyn_cast<ConstantInt>(B)) | ||||
5832 | if (ConstantInt *C2 = dyn_cast<ConstantInt>(D)) | ||||
5833 | if (Op1->hasOneUse()) { | ||||
5834 | Constant *NC = ConstantInt::get(C1->getValue() ^ C2->getValue()); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5835 | Instruction *Xor = BinaryOperator::CreateXor(C, NC, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5836 | return new ICmpInst(I.getPredicate(), A, |
5837 | InsertNewInstBefore(Xor, I)); | ||||
5838 | } | ||||
5839 | |||||
5840 | // A^B == A^D -> B == D | ||||
5841 | if (A == C) return new ICmpInst(I.getPredicate(), B, D); | ||||
5842 | if (A == D) return new ICmpInst(I.getPredicate(), B, C); | ||||
5843 | if (B == C) return new ICmpInst(I.getPredicate(), A, D); | ||||
5844 | if (B == D) return new ICmpInst(I.getPredicate(), A, C); | ||||
5845 | } | ||||
5846 | } | ||||
5847 | |||||
5848 | if (match(Op1, m_Xor(m_Value(A), m_Value(B))) && | ||||
5849 | (A == Op0 || B == Op0)) { | ||||
5850 | // A == (A^B) -> B == 0 | ||||
5851 | Value *OtherVal = A == Op0 ? B : A; | ||||
5852 | return new ICmpInst(I.getPredicate(), OtherVal, | ||||
5853 | Constant::getNullValue(A->getType())); | ||||
5854 | } | ||||
5855 | if (match(Op0, m_Sub(m_Value(A), m_Value(B))) && A == Op1) { | ||||
5856 | // (A-B) == A -> B == 0 | ||||
5857 | return new ICmpInst(I.getPredicate(), B, | ||||
5858 | Constant::getNullValue(B->getType())); | ||||
5859 | } | ||||
5860 | if (match(Op1, m_Sub(m_Value(A), m_Value(B))) && A == Op0) { | ||||
5861 | // A == (A-B) -> B == 0 | ||||
5862 | return new ICmpInst(I.getPredicate(), B, | ||||
5863 | Constant::getNullValue(B->getType())); | ||||
5864 | } | ||||
5865 | |||||
5866 | // (X&Z) == (Y&Z) -> (X^Y) & Z == 0 | ||||
5867 | if (Op0->hasOneUse() && Op1->hasOneUse() && | ||||
5868 | match(Op0, m_And(m_Value(A), m_Value(B))) && | ||||
5869 | match(Op1, m_And(m_Value(C), m_Value(D)))) { | ||||
5870 | Value *X = 0, *Y = 0, *Z = 0; | ||||
5871 | |||||
5872 | if (A == C) { | ||||
5873 | X = B; Y = D; Z = A; | ||||
5874 | } else if (A == D) { | ||||
5875 | X = B; Y = C; Z = A; | ||||
5876 | } else if (B == C) { | ||||
5877 | X = A; Y = D; Z = B; | ||||
5878 | } else if (B == D) { | ||||
5879 | X = A; Y = C; Z = B; | ||||
5880 | } | ||||
5881 | |||||
5882 | if (X) { // Build (X^Y) & Z | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5883 | Op1 = InsertNewInstBefore(BinaryOperator::CreateXor(X, Y, "tmp"), I); |
5884 | Op1 = InsertNewInstBefore(BinaryOperator::CreateAnd(Op1, Z, "tmp"), I); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5885 | I.setOperand(0, Op1); |
5886 | I.setOperand(1, Constant::getNullValue(Op1->getType())); | ||||
5887 | return &I; | ||||
5888 | } | ||||
5889 | } | ||||
5890 | } | ||||
5891 | return Changed ? &I : 0; | ||||
5892 | } | ||||
5893 | |||||
5894 | |||||
5895 | /// FoldICmpDivCst - Fold "icmp pred, ([su]div X, DivRHS), CmpRHS" where DivRHS | ||||
5896 | /// and CmpRHS are both known to be integer constants. | ||||
5897 | Instruction *InstCombiner::FoldICmpDivCst(ICmpInst &ICI, BinaryOperator *DivI, | ||||
5898 | ConstantInt *DivRHS) { | ||||
5899 | ConstantInt *CmpRHS = cast<ConstantInt>(ICI.getOperand(1)); | ||||
5900 | const APInt &CmpRHSV = CmpRHS->getValue(); | ||||
5901 | |||||
5902 | // FIXME: If the operand types don't match the type of the divide | ||||
5903 | // then don't attempt this transform. The code below doesn't have the | ||||
5904 | // logic to deal with a signed divide and an unsigned compare (and | ||||
5905 | // vice versa). This is because (x /s C1) <s C2 produces different | ||||
5906 | // results than (x /s C1) <u C2 or (x /u C1) <s C2 or even | ||||
5907 | // (x /u C1) <u C2. Simply casting the operands and result won't | ||||
5908 | // work. :( The if statement below tests that condition and bails | ||||
5909 | // if it finds it. | ||||
5910 | bool DivIsSigned = DivI->getOpcode() == Instruction::SDiv; | ||||
5911 | if (!ICI.isEquality() && DivIsSigned != ICI.isSignedPredicate()) | ||||
5912 | return 0; | ||||
5913 | if (DivRHS->isZero()) | ||||
5914 | return 0; // The ProdOV computation fails on divide by zero. | ||||
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 5915 | if (DivIsSigned && DivRHS->isAllOnesValue()) |
5916 | return 0; // The overflow computation also screws up here | ||||
5917 | if (DivRHS->isOne()) | ||||
5918 | return 0; // Not worth bothering, and eliminates some funny cases | ||||
5919 | // with INT_MIN. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5920 | |
5921 | // Compute Prod = CI * DivRHS. We are essentially solving an equation | ||||
5922 | // of form X/C1=C2. We solve for X by multiplying C1 (DivRHS) and | ||||
5923 | // C2 (CI). By solving for X we can turn this into a range check | ||||
5924 | // instead of computing a divide. | ||||
5925 | ConstantInt *Prod = Multiply(CmpRHS, DivRHS); | ||||
5926 | |||||
5927 | // Determine if the product overflows by seeing if the product is | ||||
5928 | // not equal to the divide. Make sure we do the same kind of divide | ||||
5929 | // as in the LHS instruction that we're folding. | ||||
5930 | bool ProdOV = (DivIsSigned ? ConstantExpr::getSDiv(Prod, DivRHS) : | ||||
5931 | ConstantExpr::getUDiv(Prod, DivRHS)) != CmpRHS; | ||||
5932 | |||||
5933 | // Get the ICmp opcode | ||||
5934 | ICmpInst::Predicate Pred = ICI.getPredicate(); | ||||
5935 | |||||
5936 | // Figure out the interval that is being checked. For example, a comparison | ||||
5937 | // like "X /u 5 == 0" is really checking that X is in the interval [0, 5). | ||||
5938 | // Compute this interval based on the constants involved and the signedness of | ||||
5939 | // the compare/divide. This computes a half-open interval, keeping track of | ||||
5940 | // whether either value in the interval overflows. After analysis each | ||||
5941 | // overflow variable is set to 0 if it's corresponding bound variable is valid | ||||
5942 | // -1 if overflowed off the bottom end, or +1 if overflowed off the top end. | ||||
5943 | int LoOverflow = 0, HiOverflow = 0; | ||||
5944 | ConstantInt *LoBound = 0, *HiBound = 0; | ||||
5945 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5946 | if (!DivIsSigned) { // udiv |
5947 | // e.g. X/5 op 3 --> [15, 20) | ||||
5948 | LoBound = Prod; | ||||
5949 | HiOverflow = LoOverflow = ProdOV; | ||||
5950 | if (!HiOverflow) | ||||
5951 | HiOverflow = AddWithOverflow(HiBound, LoBound, DivRHS, false); | ||||
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 5952 | } else if (DivRHS->getValue().isStrictlyPositive()) { // Divisor is > 0. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5953 | if (CmpRHSV == 0) { // (X / pos) op 0 |
5954 | // Can't overflow. e.g. X/2 op 0 --> [-1, 2) | ||||
5955 | LoBound = cast<ConstantInt>(ConstantExpr::getNeg(SubOne(DivRHS))); | ||||
5956 | HiBound = DivRHS; | ||||
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 5957 | } else if (CmpRHSV.isStrictlyPositive()) { // (X / pos) op pos |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5958 | LoBound = Prod; // e.g. X/5 op 3 --> [15, 20) |
5959 | HiOverflow = LoOverflow = ProdOV; | ||||
5960 | if (!HiOverflow) | ||||
5961 | HiOverflow = AddWithOverflow(HiBound, Prod, DivRHS, true); | ||||
5962 | } else { // (X / pos) op neg | ||||
5963 | // e.g. X/5 op -3 --> [-15-4, -15+1) --> [-19, -14) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5964 | HiBound = AddOne(Prod); |
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 5965 | LoOverflow = HiOverflow = ProdOV ? -1 : 0; |
5966 | if (!LoOverflow) { | ||||
5967 | ConstantInt* DivNeg = cast<ConstantInt>(ConstantExpr::getNeg(DivRHS)); | ||||
5968 | LoOverflow = AddWithOverflow(LoBound, HiBound, DivNeg, | ||||
5969 | true) ? -1 : 0; | ||||
5970 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5971 | } |
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 5972 | } else if (DivRHS->getValue().isNegative()) { // Divisor is < 0. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5973 | if (CmpRHSV == 0) { // (X / neg) op 0 |
5974 | // e.g. X/-5 op 0 --> [-4, 5) | ||||
5975 | LoBound = AddOne(DivRHS); | ||||
5976 | HiBound = cast<ConstantInt>(ConstantExpr::getNeg(DivRHS)); | ||||
5977 | if (HiBound == DivRHS) { // -INTMIN = INTMIN | ||||
5978 | HiOverflow = 1; // [INTMIN+1, overflow) | ||||
5979 | HiBound = 0; // e.g. X/INTMIN = 0 --> X > INTMIN | ||||
5980 | } | ||||
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 5981 | } else if (CmpRHSV.isStrictlyPositive()) { // (X / neg) op pos |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5982 | // e.g. X/-5 op 3 --> [-19, -14) |
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 5983 | HiBound = AddOne(Prod); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5984 | HiOverflow = LoOverflow = ProdOV ? -1 : 0; |
5985 | if (!LoOverflow) | ||||
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 5986 | LoOverflow = AddWithOverflow(LoBound, HiBound, DivRHS, true) ? -1 : 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5987 | } else { // (X / neg) op neg |
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 5988 | LoBound = Prod; // e.g. X/-5 op -3 --> [15, 20) |
5989 | LoOverflow = HiOverflow = ProdOV; | ||||
Dan Gohman | 45408ea | 2008-09-11 00:25:00 +0000 | [diff] [blame] | 5990 | if (!HiOverflow) |
5991 | HiOverflow = SubWithOverflow(HiBound, Prod, DivRHS, true); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5992 | } |
5993 | |||||
5994 | // Dividing by a negative swaps the condition. LT <-> GT | ||||
5995 | Pred = ICmpInst::getSwappedPredicate(Pred); | ||||
5996 | } | ||||
5997 | |||||
5998 | Value *X = DivI->getOperand(0); | ||||
5999 | switch (Pred) { | ||||
6000 | default: assert(0 && "Unhandled icmp opcode!"); | ||||
6001 | case ICmpInst::ICMP_EQ: | ||||
6002 | if (LoOverflow && HiOverflow) | ||||
6003 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6004 | else if (HiOverflow) | ||||
6005 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SGE : | ||||
6006 | ICmpInst::ICMP_UGE, X, LoBound); | ||||
6007 | else if (LoOverflow) | ||||
6008 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SLT : | ||||
6009 | ICmpInst::ICMP_ULT, X, HiBound); | ||||
6010 | else | ||||
6011 | return InsertRangeTest(X, LoBound, HiBound, DivIsSigned, true, ICI); | ||||
6012 | case ICmpInst::ICMP_NE: | ||||
6013 | if (LoOverflow && HiOverflow) | ||||
6014 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6015 | else if (HiOverflow) | ||||
6016 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SLT : | ||||
6017 | ICmpInst::ICMP_ULT, X, LoBound); | ||||
6018 | else if (LoOverflow) | ||||
6019 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SGE : | ||||
6020 | ICmpInst::ICMP_UGE, X, HiBound); | ||||
6021 | else | ||||
6022 | return InsertRangeTest(X, LoBound, HiBound, DivIsSigned, false, ICI); | ||||
6023 | case ICmpInst::ICMP_ULT: | ||||
6024 | case ICmpInst::ICMP_SLT: | ||||
6025 | if (LoOverflow == +1) // Low bound is greater than input range. | ||||
6026 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6027 | if (LoOverflow == -1) // Low bound is less than input range. | ||||
6028 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6029 | return new ICmpInst(Pred, X, LoBound); | ||||
6030 | case ICmpInst::ICMP_UGT: | ||||
6031 | case ICmpInst::ICMP_SGT: | ||||
6032 | if (HiOverflow == +1) // High bound greater than input range. | ||||
6033 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6034 | else if (HiOverflow == -1) // High bound less than input range. | ||||
6035 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6036 | if (Pred == ICmpInst::ICMP_UGT) | ||||
6037 | return new ICmpInst(ICmpInst::ICMP_UGE, X, HiBound); | ||||
6038 | else | ||||
6039 | return new ICmpInst(ICmpInst::ICMP_SGE, X, HiBound); | ||||
6040 | } | ||||
6041 | } | ||||
6042 | |||||
6043 | |||||
6044 | /// visitICmpInstWithInstAndIntCst - Handle "icmp (instr, intcst)". | ||||
6045 | /// | ||||
6046 | Instruction *InstCombiner::visitICmpInstWithInstAndIntCst(ICmpInst &ICI, | ||||
6047 | Instruction *LHSI, | ||||
6048 | ConstantInt *RHS) { | ||||
6049 | const APInt &RHSV = RHS->getValue(); | ||||
6050 | |||||
6051 | switch (LHSI->getOpcode()) { | ||||
6052 | case Instruction::Xor: // (icmp pred (xor X, XorCST), CI) | ||||
6053 | if (ConstantInt *XorCST = dyn_cast<ConstantInt>(LHSI->getOperand(1))) { | ||||
6054 | // If this is a comparison that tests the signbit (X < 0) or (x > -1), | ||||
6055 | // fold the xor. | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 6056 | if ((ICI.getPredicate() == ICmpInst::ICMP_SLT && RHSV == 0) || |
6057 | (ICI.getPredicate() == ICmpInst::ICMP_SGT && RHSV.isAllOnesValue())) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6058 | Value *CompareVal = LHSI->getOperand(0); |
6059 | |||||
6060 | // If the sign bit of the XorCST is not set, there is no change to | ||||
6061 | // the operation, just stop using the Xor. | ||||
6062 | if (!XorCST->getValue().isNegative()) { | ||||
6063 | ICI.setOperand(0, CompareVal); | ||||
6064 | AddToWorkList(LHSI); | ||||
6065 | return &ICI; | ||||
6066 | } | ||||
6067 | |||||
6068 | // Was the old condition true if the operand is positive? | ||||
6069 | bool isTrueIfPositive = ICI.getPredicate() == ICmpInst::ICMP_SGT; | ||||
6070 | |||||
6071 | // If so, the new one isn't. | ||||
6072 | isTrueIfPositive ^= true; | ||||
6073 | |||||
6074 | if (isTrueIfPositive) | ||||
6075 | return new ICmpInst(ICmpInst::ICMP_SGT, CompareVal, SubOne(RHS)); | ||||
6076 | else | ||||
6077 | return new ICmpInst(ICmpInst::ICMP_SLT, CompareVal, AddOne(RHS)); | ||||
6078 | } | ||||
6079 | } | ||||
6080 | break; | ||||
6081 | case Instruction::And: // (icmp pred (and X, AndCST), RHS) | ||||
6082 | if (LHSI->hasOneUse() && isa<ConstantInt>(LHSI->getOperand(1)) && | ||||
6083 | LHSI->getOperand(0)->hasOneUse()) { | ||||
6084 | ConstantInt *AndCST = cast<ConstantInt>(LHSI->getOperand(1)); | ||||
6085 | |||||
6086 | // If the LHS is an AND of a truncating cast, we can widen the | ||||
6087 | // and/compare to be the input width without changing the value | ||||
6088 | // produced, eliminating a cast. | ||||
6089 | if (TruncInst *Cast = dyn_cast<TruncInst>(LHSI->getOperand(0))) { | ||||
6090 | // We can do this transformation if either the AND constant does not | ||||
6091 | // have its sign bit set or if it is an equality comparison. | ||||
6092 | // Extending a relational comparison when we're checking the sign | ||||
6093 | // bit would not work. | ||||
6094 | if (Cast->hasOneUse() && | ||||
Anton Korobeynikov | 6a4a933 | 2008-02-20 12:07:57 +0000 | [diff] [blame] | 6095 | (ICI.isEquality() || |
6096 | (AndCST->getValue().isNonNegative() && RHSV.isNonNegative()))) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6097 | uint32_t BitWidth = |
6098 | cast<IntegerType>(Cast->getOperand(0)->getType())->getBitWidth(); | ||||
6099 | APInt NewCST = AndCST->getValue(); | ||||
6100 | NewCST.zext(BitWidth); | ||||
6101 | APInt NewCI = RHSV; | ||||
6102 | NewCI.zext(BitWidth); | ||||
6103 | Instruction *NewAnd = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6104 | BinaryOperator::CreateAnd(Cast->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6105 | ConstantInt::get(NewCST),LHSI->getName()); |
6106 | InsertNewInstBefore(NewAnd, ICI); | ||||
6107 | return new ICmpInst(ICI.getPredicate(), NewAnd, | ||||
6108 | ConstantInt::get(NewCI)); | ||||
6109 | } | ||||
6110 | } | ||||
6111 | |||||
6112 | // If this is: (X >> C1) & C2 != C3 (where any shift and any compare | ||||
6113 | // could exist), turn it into (X & (C2 << C1)) != (C3 << C1). This | ||||
6114 | // happens a LOT in code produced by the C front-end, for bitfield | ||||
6115 | // access. | ||||
6116 | BinaryOperator *Shift = dyn_cast<BinaryOperator>(LHSI->getOperand(0)); | ||||
6117 | if (Shift && !Shift->isShift()) | ||||
6118 | Shift = 0; | ||||
6119 | |||||
6120 | ConstantInt *ShAmt; | ||||
6121 | ShAmt = Shift ? dyn_cast<ConstantInt>(Shift->getOperand(1)) : 0; | ||||
6122 | const Type *Ty = Shift ? Shift->getType() : 0; // Type of the shift. | ||||
6123 | const Type *AndTy = AndCST->getType(); // Type of the and. | ||||
6124 | |||||
6125 | // We can fold this as long as we can't shift unknown bits | ||||
6126 | // into the mask. This can only happen with signed shift | ||||
6127 | // rights, as they sign-extend. | ||||
6128 | if (ShAmt) { | ||||
6129 | bool CanFold = Shift->isLogicalShift(); | ||||
6130 | if (!CanFold) { | ||||
6131 | // To test for the bad case of the signed shr, see if any | ||||
6132 | // of the bits shifted in could be tested after the mask. | ||||
6133 | uint32_t TyBits = Ty->getPrimitiveSizeInBits(); | ||||
6134 | int ShAmtVal = TyBits - ShAmt->getLimitedValue(TyBits); | ||||
6135 | |||||
6136 | uint32_t BitWidth = AndTy->getPrimitiveSizeInBits(); | ||||
6137 | if ((APInt::getHighBitsSet(BitWidth, BitWidth-ShAmtVal) & | ||||
6138 | AndCST->getValue()) == 0) | ||||
6139 | CanFold = true; | ||||
6140 | } | ||||
6141 | |||||
6142 | if (CanFold) { | ||||
6143 | Constant *NewCst; | ||||
6144 | if (Shift->getOpcode() == Instruction::Shl) | ||||
6145 | NewCst = ConstantExpr::getLShr(RHS, ShAmt); | ||||
6146 | else | ||||
6147 | NewCst = ConstantExpr::getShl(RHS, ShAmt); | ||||
6148 | |||||
6149 | // Check to see if we are shifting out any of the bits being | ||||
6150 | // compared. | ||||
6151 | if (ConstantExpr::get(Shift->getOpcode(), NewCst, ShAmt) != RHS) { | ||||
6152 | // If we shifted bits out, the fold is not going to work out. | ||||
6153 | // As a special case, check to see if this means that the | ||||
6154 | // result is always true or false now. | ||||
6155 | if (ICI.getPredicate() == ICmpInst::ICMP_EQ) | ||||
6156 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6157 | if (ICI.getPredicate() == ICmpInst::ICMP_NE) | ||||
6158 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6159 | } else { | ||||
6160 | ICI.setOperand(1, NewCst); | ||||
6161 | Constant *NewAndCST; | ||||
6162 | if (Shift->getOpcode() == Instruction::Shl) | ||||
6163 | NewAndCST = ConstantExpr::getLShr(AndCST, ShAmt); | ||||
6164 | else | ||||
6165 | NewAndCST = ConstantExpr::getShl(AndCST, ShAmt); | ||||
6166 | LHSI->setOperand(1, NewAndCST); | ||||
6167 | LHSI->setOperand(0, Shift->getOperand(0)); | ||||
6168 | AddToWorkList(Shift); // Shift is dead. | ||||
6169 | AddUsesToWorkList(ICI); | ||||
6170 | return &ICI; | ||||
6171 | } | ||||
6172 | } | ||||
6173 | } | ||||
6174 | |||||
6175 | // Turn ((X >> Y) & C) == 0 into (X & (C << Y)) == 0. The later is | ||||
6176 | // preferable because it allows the C<<Y expression to be hoisted out | ||||
6177 | // of a loop if Y is invariant and X is not. | ||||
6178 | if (Shift && Shift->hasOneUse() && RHSV == 0 && | ||||
6179 | ICI.isEquality() && !Shift->isArithmeticShift() && | ||||
6180 | isa<Instruction>(Shift->getOperand(0))) { | ||||
6181 | // Compute C << Y. | ||||
6182 | Value *NS; | ||||
6183 | if (Shift->getOpcode() == Instruction::LShr) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6184 | NS = BinaryOperator::CreateShl(AndCST, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6185 | Shift->getOperand(1), "tmp"); |
6186 | } else { | ||||
6187 | // Insert a logical shift. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6188 | NS = BinaryOperator::CreateLShr(AndCST, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6189 | Shift->getOperand(1), "tmp"); |
6190 | } | ||||
6191 | InsertNewInstBefore(cast<Instruction>(NS), ICI); | ||||
6192 | |||||
6193 | // Compute X & (C << Y). | ||||
6194 | Instruction *NewAnd = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6195 | BinaryOperator::CreateAnd(Shift->getOperand(0), NS, LHSI->getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6196 | InsertNewInstBefore(NewAnd, ICI); |
6197 | |||||
6198 | ICI.setOperand(0, NewAnd); | ||||
6199 | return &ICI; | ||||
6200 | } | ||||
6201 | } | ||||
6202 | break; | ||||
6203 | |||||
6204 | case Instruction::Shl: { // (icmp pred (shl X, ShAmt), CI) | ||||
6205 | ConstantInt *ShAmt = dyn_cast<ConstantInt>(LHSI->getOperand(1)); | ||||
6206 | if (!ShAmt) break; | ||||
6207 | |||||
6208 | uint32_t TypeBits = RHSV.getBitWidth(); | ||||
6209 | |||||
6210 | // Check that the shift amount is in range. If not, don't perform | ||||
6211 | // undefined shifts. When the shift is visited it will be | ||||
6212 | // simplified. | ||||
6213 | if (ShAmt->uge(TypeBits)) | ||||
6214 | break; | ||||
6215 | |||||
6216 | if (ICI.isEquality()) { | ||||
6217 | // If we are comparing against bits always shifted out, the | ||||
6218 | // comparison cannot succeed. | ||||
6219 | Constant *Comp = | ||||
6220 | ConstantExpr::getShl(ConstantExpr::getLShr(RHS, ShAmt), ShAmt); | ||||
6221 | if (Comp != RHS) {// Comparing against a bit that we know is zero. | ||||
6222 | bool IsICMP_NE = ICI.getPredicate() == ICmpInst::ICMP_NE; | ||||
6223 | Constant *Cst = ConstantInt::get(Type::Int1Ty, IsICMP_NE); | ||||
6224 | return ReplaceInstUsesWith(ICI, Cst); | ||||
6225 | } | ||||
6226 | |||||
6227 | if (LHSI->hasOneUse()) { | ||||
6228 | // Otherwise strength reduce the shift into an and. | ||||
6229 | uint32_t ShAmtVal = (uint32_t)ShAmt->getLimitedValue(TypeBits); | ||||
6230 | Constant *Mask = | ||||
6231 | ConstantInt::get(APInt::getLowBitsSet(TypeBits, TypeBits-ShAmtVal)); | ||||
6232 | |||||
6233 | Instruction *AndI = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6234 | BinaryOperator::CreateAnd(LHSI->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6235 | Mask, LHSI->getName()+".mask"); |
6236 | Value *And = InsertNewInstBefore(AndI, ICI); | ||||
6237 | return new ICmpInst(ICI.getPredicate(), And, | ||||
6238 | ConstantInt::get(RHSV.lshr(ShAmtVal))); | ||||
6239 | } | ||||
6240 | } | ||||
6241 | |||||
6242 | // Otherwise, if this is a comparison of the sign bit, simplify to and/test. | ||||
6243 | bool TrueIfSigned = false; | ||||
6244 | if (LHSI->hasOneUse() && | ||||
6245 | isSignBitCheck(ICI.getPredicate(), RHS, TrueIfSigned)) { | ||||
6246 | // (X << 31) <s 0 --> (X&1) != 0 | ||||
6247 | Constant *Mask = ConstantInt::get(APInt(TypeBits, 1) << | ||||
6248 | (TypeBits-ShAmt->getZExtValue()-1)); | ||||
6249 | Instruction *AndI = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6250 | BinaryOperator::CreateAnd(LHSI->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6251 | Mask, LHSI->getName()+".mask"); |
6252 | Value *And = InsertNewInstBefore(AndI, ICI); | ||||
6253 | |||||
6254 | return new ICmpInst(TrueIfSigned ? ICmpInst::ICMP_NE : ICmpInst::ICMP_EQ, | ||||
6255 | And, Constant::getNullValue(And->getType())); | ||||
6256 | } | ||||
6257 | break; | ||||
6258 | } | ||||
6259 | |||||
6260 | case Instruction::LShr: // (icmp pred (shr X, ShAmt), CI) | ||||
6261 | case Instruction::AShr: { | ||||
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6262 | // Only handle equality comparisons of shift-by-constant. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6263 | ConstantInt *ShAmt = dyn_cast<ConstantInt>(LHSI->getOperand(1)); |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6264 | if (!ShAmt || !ICI.isEquality()) break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6265 | |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6266 | // Check that the shift amount is in range. If not, don't perform |
6267 | // undefined shifts. When the shift is visited it will be | ||||
6268 | // simplified. | ||||
6269 | uint32_t TypeBits = RHSV.getBitWidth(); | ||||
6270 | if (ShAmt->uge(TypeBits)) | ||||
6271 | break; | ||||
6272 | |||||
6273 | uint32_t ShAmtVal = (uint32_t)ShAmt->getLimitedValue(TypeBits); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6274 | |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6275 | // If we are comparing against bits always shifted out, the |
6276 | // comparison cannot succeed. | ||||
6277 | APInt Comp = RHSV << ShAmtVal; | ||||
6278 | if (LHSI->getOpcode() == Instruction::LShr) | ||||
6279 | Comp = Comp.lshr(ShAmtVal); | ||||
6280 | else | ||||
6281 | Comp = Comp.ashr(ShAmtVal); | ||||
6282 | |||||
6283 | if (Comp != RHSV) { // Comparing against a bit that we know is zero. | ||||
6284 | bool IsICMP_NE = ICI.getPredicate() == ICmpInst::ICMP_NE; | ||||
6285 | Constant *Cst = ConstantInt::get(Type::Int1Ty, IsICMP_NE); | ||||
6286 | return ReplaceInstUsesWith(ICI, Cst); | ||||
6287 | } | ||||
6288 | |||||
6289 | // Otherwise, check to see if the bits shifted out are known to be zero. | ||||
6290 | // If so, we can compare against the unshifted value: | ||||
6291 | // (X & 4) >> 1 == 2 --> (X & 4) == 4. | ||||
Evan Cheng | fb9292a | 2008-04-23 00:38:06 +0000 | [diff] [blame] | 6292 | if (LHSI->hasOneUse() && |
6293 | MaskedValueIsZero(LHSI->getOperand(0), | ||||
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6294 | APInt::getLowBitsSet(Comp.getBitWidth(), ShAmtVal))) { |
6295 | return new ICmpInst(ICI.getPredicate(), LHSI->getOperand(0), | ||||
6296 | ConstantExpr::getShl(RHS, ShAmt)); | ||||
6297 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6298 | |
Evan Cheng | fb9292a | 2008-04-23 00:38:06 +0000 | [diff] [blame] | 6299 | if (LHSI->hasOneUse()) { |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6300 | // Otherwise strength reduce the shift into an and. |
6301 | APInt Val(APInt::getHighBitsSet(TypeBits, TypeBits - ShAmtVal)); | ||||
6302 | Constant *Mask = ConstantInt::get(Val); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6303 | |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6304 | Instruction *AndI = |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6305 | BinaryOperator::CreateAnd(LHSI->getOperand(0), |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6306 | Mask, LHSI->getName()+".mask"); |
6307 | Value *And = InsertNewInstBefore(AndI, ICI); | ||||
6308 | return new ICmpInst(ICI.getPredicate(), And, | ||||
6309 | ConstantExpr::getShl(RHS, ShAmt)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6310 | } |
6311 | break; | ||||
6312 | } | ||||
6313 | |||||
6314 | case Instruction::SDiv: | ||||
6315 | case Instruction::UDiv: | ||||
6316 | // Fold: icmp pred ([us]div X, C1), C2 -> range test | ||||
6317 | // Fold this div into the comparison, producing a range check. | ||||
6318 | // Determine, based on the divide type, what the range is being | ||||
6319 | // checked. If there is an overflow on the low or high side, remember | ||||
6320 | // it, otherwise compute the range [low, hi) bounding the new value. | ||||
6321 | // See: InsertRangeTest above for the kinds of replacements possible. | ||||
6322 | if (ConstantInt *DivRHS = dyn_cast<ConstantInt>(LHSI->getOperand(1))) | ||||
6323 | if (Instruction *R = FoldICmpDivCst(ICI, cast<BinaryOperator>(LHSI), | ||||
6324 | DivRHS)) | ||||
6325 | return R; | ||||
6326 | break; | ||||
Nick Lewycky | 0185bbf | 2008-02-03 16:33:09 +0000 | [diff] [blame] | 6327 | |
6328 | case Instruction::Add: | ||||
6329 | // Fold: icmp pred (add, X, C1), C2 | ||||
6330 | |||||
6331 | if (!ICI.isEquality()) { | ||||
6332 | ConstantInt *LHSC = dyn_cast<ConstantInt>(LHSI->getOperand(1)); | ||||
6333 | if (!LHSC) break; | ||||
6334 | const APInt &LHSV = LHSC->getValue(); | ||||
6335 | |||||
6336 | ConstantRange CR = ICI.makeConstantRange(ICI.getPredicate(), RHSV) | ||||
6337 | .subtract(LHSV); | ||||
6338 | |||||
6339 | if (ICI.isSignedPredicate()) { | ||||
6340 | if (CR.getLower().isSignBit()) { | ||||
6341 | return new ICmpInst(ICmpInst::ICMP_SLT, LHSI->getOperand(0), | ||||
6342 | ConstantInt::get(CR.getUpper())); | ||||
6343 | } else if (CR.getUpper().isSignBit()) { | ||||
6344 | return new ICmpInst(ICmpInst::ICMP_SGE, LHSI->getOperand(0), | ||||
6345 | ConstantInt::get(CR.getLower())); | ||||
6346 | } | ||||
6347 | } else { | ||||
6348 | if (CR.getLower().isMinValue()) { | ||||
6349 | return new ICmpInst(ICmpInst::ICMP_ULT, LHSI->getOperand(0), | ||||
6350 | ConstantInt::get(CR.getUpper())); | ||||
6351 | } else if (CR.getUpper().isMinValue()) { | ||||
6352 | return new ICmpInst(ICmpInst::ICMP_UGE, LHSI->getOperand(0), | ||||
6353 | ConstantInt::get(CR.getLower())); | ||||
6354 | } | ||||
6355 | } | ||||
6356 | } | ||||
6357 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6358 | } |
6359 | |||||
6360 | // Simplify icmp_eq and icmp_ne instructions with integer constant RHS. | ||||
6361 | if (ICI.isEquality()) { | ||||
6362 | bool isICMP_NE = ICI.getPredicate() == ICmpInst::ICMP_NE; | ||||
6363 | |||||
6364 | // If the first operand is (add|sub|and|or|xor|rem) with a constant, and | ||||
6365 | // the second operand is a constant, simplify a bit. | ||||
6366 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(LHSI)) { | ||||
6367 | switch (BO->getOpcode()) { | ||||
6368 | case Instruction::SRem: | ||||
6369 | // If we have a signed (X % (2^c)) == 0, turn it into an unsigned one. | ||||
6370 | if (RHSV == 0 && isa<ConstantInt>(BO->getOperand(1)) &&BO->hasOneUse()){ | ||||
6371 | const APInt &V = cast<ConstantInt>(BO->getOperand(1))->getValue(); | ||||
6372 | if (V.sgt(APInt(V.getBitWidth(), 1)) && V.isPowerOf2()) { | ||||
6373 | Instruction *NewRem = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6374 | BinaryOperator::CreateURem(BO->getOperand(0), BO->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6375 | BO->getName()); |
6376 | InsertNewInstBefore(NewRem, ICI); | ||||
6377 | return new ICmpInst(ICI.getPredicate(), NewRem, | ||||
6378 | Constant::getNullValue(BO->getType())); | ||||
6379 | } | ||||
6380 | } | ||||
6381 | break; | ||||
6382 | case Instruction::Add: | ||||
6383 | // Replace ((add A, B) != C) with (A != C-B) if B & C are constants. | ||||
6384 | if (ConstantInt *BOp1C = dyn_cast<ConstantInt>(BO->getOperand(1))) { | ||||
6385 | if (BO->hasOneUse()) | ||||
6386 | return new ICmpInst(ICI.getPredicate(), BO->getOperand(0), | ||||
6387 | Subtract(RHS, BOp1C)); | ||||
6388 | } else if (RHSV == 0) { | ||||
6389 | // Replace ((add A, B) != 0) with (A != -B) if A or B is | ||||
6390 | // efficiently invertible, or if the add has just this one use. | ||||
6391 | Value *BOp0 = BO->getOperand(0), *BOp1 = BO->getOperand(1); | ||||
6392 | |||||
6393 | if (Value *NegVal = dyn_castNegVal(BOp1)) | ||||
6394 | return new ICmpInst(ICI.getPredicate(), BOp0, NegVal); | ||||
6395 | else if (Value *NegVal = dyn_castNegVal(BOp0)) | ||||
6396 | return new ICmpInst(ICI.getPredicate(), NegVal, BOp1); | ||||
6397 | else if (BO->hasOneUse()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6398 | Instruction *Neg = BinaryOperator::CreateNeg(BOp1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6399 | InsertNewInstBefore(Neg, ICI); |
6400 | Neg->takeName(BO); | ||||
6401 | return new ICmpInst(ICI.getPredicate(), BOp0, Neg); | ||||
6402 | } | ||||
6403 | } | ||||
6404 | break; | ||||
6405 | case Instruction::Xor: | ||||
6406 | // For the xor case, we can xor two constants together, eliminating | ||||
6407 | // the explicit xor. | ||||
6408 | if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1))) | ||||
6409 | return new ICmpInst(ICI.getPredicate(), BO->getOperand(0), | ||||
6410 | ConstantExpr::getXor(RHS, BOC)); | ||||
6411 | |||||
6412 | // FALLTHROUGH | ||||
6413 | case Instruction::Sub: | ||||
6414 | // Replace (([sub|xor] A, B) != 0) with (A != B) | ||||
6415 | if (RHSV == 0) | ||||
6416 | return new ICmpInst(ICI.getPredicate(), BO->getOperand(0), | ||||
6417 | BO->getOperand(1)); | ||||
6418 | break; | ||||
6419 | |||||
6420 | case Instruction::Or: | ||||
6421 | // If bits are being or'd in that are not present in the constant we | ||||
6422 | // are comparing against, then the comparison could never succeed! | ||||
6423 | if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1))) { | ||||
6424 | Constant *NotCI = ConstantExpr::getNot(RHS); | ||||
6425 | if (!ConstantExpr::getAnd(BOC, NotCI)->isNullValue()) | ||||
6426 | return ReplaceInstUsesWith(ICI, ConstantInt::get(Type::Int1Ty, | ||||
6427 | isICMP_NE)); | ||||
6428 | } | ||||
6429 | break; | ||||
6430 | |||||
6431 | case Instruction::And: | ||||
6432 | if (ConstantInt *BOC = dyn_cast<ConstantInt>(BO->getOperand(1))) { | ||||
6433 | // If bits are being compared against that are and'd out, then the | ||||
6434 | // comparison can never succeed! | ||||
6435 | if ((RHSV & ~BOC->getValue()) != 0) | ||||
6436 | return ReplaceInstUsesWith(ICI, ConstantInt::get(Type::Int1Ty, | ||||
6437 | isICMP_NE)); | ||||
6438 | |||||
6439 | // If we have ((X & C) == C), turn it into ((X & C) != 0). | ||||
6440 | if (RHS == BOC && RHSV.isPowerOf2()) | ||||
6441 | return new ICmpInst(isICMP_NE ? ICmpInst::ICMP_EQ : | ||||
6442 | ICmpInst::ICMP_NE, LHSI, | ||||
6443 | Constant::getNullValue(RHS->getType())); | ||||
6444 | |||||
6445 | // Replace (and X, (1 << size(X)-1) != 0) with x s< 0 | ||||
Chris Lattner | 60813c2 | 2008-06-02 01:29:46 +0000 | [diff] [blame] | 6446 | if (BOC->getValue().isSignBit()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6447 | Value *X = BO->getOperand(0); |
6448 | Constant *Zero = Constant::getNullValue(X->getType()); | ||||
6449 | ICmpInst::Predicate pred = isICMP_NE ? | ||||
6450 | ICmpInst::ICMP_SLT : ICmpInst::ICMP_SGE; | ||||
6451 | return new ICmpInst(pred, X, Zero); | ||||
6452 | } | ||||
6453 | |||||
6454 | // ((X & ~7) == 0) --> X < 8 | ||||
6455 | if (RHSV == 0 && isHighOnes(BOC)) { | ||||
6456 | Value *X = BO->getOperand(0); | ||||
6457 | Constant *NegX = ConstantExpr::getNeg(BOC); | ||||
6458 | ICmpInst::Predicate pred = isICMP_NE ? | ||||
6459 | ICmpInst::ICMP_UGE : ICmpInst::ICMP_ULT; | ||||
6460 | return new ICmpInst(pred, X, NegX); | ||||
6461 | } | ||||
6462 | } | ||||
6463 | default: break; | ||||
6464 | } | ||||
6465 | } else if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(LHSI)) { | ||||
6466 | // Handle icmp {eq|ne} <intrinsic>, intcst. | ||||
6467 | if (II->getIntrinsicID() == Intrinsic::bswap) { | ||||
6468 | AddToWorkList(II); | ||||
6469 | ICI.setOperand(0, II->getOperand(1)); | ||||
6470 | ICI.setOperand(1, ConstantInt::get(RHSV.byteSwap())); | ||||
6471 | return &ICI; | ||||
6472 | } | ||||
6473 | } | ||||
6474 | } else { // Not a ICMP_EQ/ICMP_NE | ||||
6475 | // If the LHS is a cast from an integral value of the same size, | ||||
6476 | // then since we know the RHS is a constant, try to simlify. | ||||
6477 | if (CastInst *Cast = dyn_cast<CastInst>(LHSI)) { | ||||
6478 | Value *CastOp = Cast->getOperand(0); | ||||
6479 | const Type *SrcTy = CastOp->getType(); | ||||
6480 | uint32_t SrcTySize = SrcTy->getPrimitiveSizeInBits(); | ||||
6481 | if (SrcTy->isInteger() && | ||||
6482 | SrcTySize == Cast->getType()->getPrimitiveSizeInBits()) { | ||||
6483 | // If this is an unsigned comparison, try to make the comparison use | ||||
6484 | // smaller constant values. | ||||
6485 | if (ICI.getPredicate() == ICmpInst::ICMP_ULT && RHSV.isSignBit()) { | ||||
6486 | // X u< 128 => X s> -1 | ||||
6487 | return new ICmpInst(ICmpInst::ICMP_SGT, CastOp, | ||||
6488 | ConstantInt::get(APInt::getAllOnesValue(SrcTySize))); | ||||
6489 | } else if (ICI.getPredicate() == ICmpInst::ICMP_UGT && | ||||
6490 | RHSV == APInt::getSignedMaxValue(SrcTySize)) { | ||||
6491 | // X u> 127 => X s< 0 | ||||
6492 | return new ICmpInst(ICmpInst::ICMP_SLT, CastOp, | ||||
6493 | Constant::getNullValue(SrcTy)); | ||||
6494 | } | ||||
6495 | } | ||||
6496 | } | ||||
6497 | } | ||||
6498 | return 0; | ||||
6499 | } | ||||
6500 | |||||
6501 | /// visitICmpInstWithCastAndCast - Handle icmp (cast x to y), (cast/cst). | ||||
6502 | /// We only handle extending casts so far. | ||||
6503 | /// | ||||
6504 | Instruction *InstCombiner::visitICmpInstWithCastAndCast(ICmpInst &ICI) { | ||||
6505 | const CastInst *LHSCI = cast<CastInst>(ICI.getOperand(0)); | ||||
6506 | Value *LHSCIOp = LHSCI->getOperand(0); | ||||
6507 | const Type *SrcTy = LHSCIOp->getType(); | ||||
6508 | const Type *DestTy = LHSCI->getType(); | ||||
6509 | Value *RHSCIOp; | ||||
6510 | |||||
6511 | // Turn icmp (ptrtoint x), (ptrtoint/c) into a compare of the input if the | ||||
6512 | // integer type is the same size as the pointer type. | ||||
6513 | if (LHSCI->getOpcode() == Instruction::PtrToInt && | ||||
6514 | getTargetData().getPointerSizeInBits() == | ||||
6515 | cast<IntegerType>(DestTy)->getBitWidth()) { | ||||
6516 | Value *RHSOp = 0; | ||||
6517 | if (Constant *RHSC = dyn_cast<Constant>(ICI.getOperand(1))) { | ||||
6518 | RHSOp = ConstantExpr::getIntToPtr(RHSC, SrcTy); | ||||
6519 | } else if (PtrToIntInst *RHSC = dyn_cast<PtrToIntInst>(ICI.getOperand(1))) { | ||||
6520 | RHSOp = RHSC->getOperand(0); | ||||
6521 | // If the pointer types don't match, insert a bitcast. | ||||
6522 | if (LHSCIOp->getType() != RHSOp->getType()) | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 6523 | RHSOp = InsertBitCastBefore(RHSOp, LHSCIOp->getType(), ICI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6524 | } |
6525 | |||||
6526 | if (RHSOp) | ||||
6527 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, RHSOp); | ||||
6528 | } | ||||
6529 | |||||
6530 | // The code below only handles extension cast instructions, so far. | ||||
6531 | // Enforce this. | ||||
6532 | if (LHSCI->getOpcode() != Instruction::ZExt && | ||||
6533 | LHSCI->getOpcode() != Instruction::SExt) | ||||
6534 | return 0; | ||||
6535 | |||||
6536 | bool isSignedExt = LHSCI->getOpcode() == Instruction::SExt; | ||||
6537 | bool isSignedCmp = ICI.isSignedPredicate(); | ||||
6538 | |||||
6539 | if (CastInst *CI = dyn_cast<CastInst>(ICI.getOperand(1))) { | ||||
6540 | // Not an extension from the same type? | ||||
6541 | RHSCIOp = CI->getOperand(0); | ||||
6542 | if (RHSCIOp->getType() != LHSCIOp->getType()) | ||||
6543 | return 0; | ||||
6544 | |||||
Nick Lewycky | d4264dc | 2008-01-28 03:48:02 +0000 | [diff] [blame] | 6545 | // 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] | 6546 | // and the other is a zext), then we can't handle this. |
6547 | if (CI->getOpcode() != LHSCI->getOpcode()) | ||||
6548 | return 0; | ||||
6549 | |||||
Nick Lewycky | d4264dc | 2008-01-28 03:48:02 +0000 | [diff] [blame] | 6550 | // Deal with equality cases early. |
6551 | if (ICI.isEquality()) | ||||
6552 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, RHSCIOp); | ||||
6553 | |||||
6554 | // A signed comparison of sign extended values simplifies into a | ||||
6555 | // signed comparison. | ||||
6556 | if (isSignedCmp && isSignedExt) | ||||
6557 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, RHSCIOp); | ||||
6558 | |||||
6559 | // The other three cases all fold into an unsigned comparison. | ||||
6560 | return new ICmpInst(ICI.getUnsignedPredicate(), LHSCIOp, RHSCIOp); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6561 | } |
6562 | |||||
6563 | // If we aren't dealing with a constant on the RHS, exit early | ||||
6564 | ConstantInt *CI = dyn_cast<ConstantInt>(ICI.getOperand(1)); | ||||
6565 | if (!CI) | ||||
6566 | return 0; | ||||
6567 | |||||
6568 | // Compute the constant that would happen if we truncated to SrcTy then | ||||
6569 | // reextended to DestTy. | ||||
6570 | Constant *Res1 = ConstantExpr::getTrunc(CI, SrcTy); | ||||
6571 | Constant *Res2 = ConstantExpr::getCast(LHSCI->getOpcode(), Res1, DestTy); | ||||
6572 | |||||
6573 | // If the re-extended constant didn't change... | ||||
6574 | if (Res2 == CI) { | ||||
6575 | // Make sure that sign of the Cmp and the sign of the Cast are the same. | ||||
6576 | // For example, we might have: | ||||
6577 | // %A = sext short %X to uint | ||||
6578 | // %B = icmp ugt uint %A, 1330 | ||||
6579 | // It is incorrect to transform this into | ||||
6580 | // %B = icmp ugt short %X, 1330 | ||||
6581 | // because %A may have negative value. | ||||
6582 | // | ||||
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 6583 | // However, we allow this when the compare is EQ/NE, because they are |
6584 | // signless. | ||||
6585 | if (isSignedExt == isSignedCmp || ICI.isEquality()) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6586 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, Res1); |
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 6587 | return 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6588 | } |
6589 | |||||
6590 | // The re-extended constant changed so the constant cannot be represented | ||||
6591 | // in the shorter type. Consequently, we cannot emit a simple comparison. | ||||
6592 | |||||
6593 | // First, handle some easy cases. We know the result cannot be equal at this | ||||
6594 | // point so handle the ICI.isEquality() cases | ||||
6595 | if (ICI.getPredicate() == ICmpInst::ICMP_EQ) | ||||
6596 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6597 | if (ICI.getPredicate() == ICmpInst::ICMP_NE) | ||||
6598 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6599 | |||||
6600 | // Evaluate the comparison for LT (we invert for GT below). LE and GE cases | ||||
6601 | // should have been folded away previously and not enter in here. | ||||
6602 | Value *Result; | ||||
6603 | if (isSignedCmp) { | ||||
6604 | // We're performing a signed comparison. | ||||
6605 | if (cast<ConstantInt>(CI)->getValue().isNegative()) | ||||
6606 | Result = ConstantInt::getFalse(); // X < (small) --> false | ||||
6607 | else | ||||
6608 | Result = ConstantInt::getTrue(); // X < (large) --> true | ||||
6609 | } else { | ||||
6610 | // We're performing an unsigned comparison. | ||||
6611 | if (isSignedExt) { | ||||
6612 | // We're performing an unsigned comp with a sign extended value. | ||||
6613 | // This is true if the input is >= 0. [aka >s -1] | ||||
6614 | Constant *NegOne = ConstantInt::getAllOnesValue(SrcTy); | ||||
6615 | Result = InsertNewInstBefore(new ICmpInst(ICmpInst::ICMP_SGT, LHSCIOp, | ||||
6616 | NegOne, ICI.getName()), ICI); | ||||
6617 | } else { | ||||
6618 | // Unsigned extend & unsigned compare -> always true. | ||||
6619 | Result = ConstantInt::getTrue(); | ||||
6620 | } | ||||
6621 | } | ||||
6622 | |||||
6623 | // Finally, return the value computed. | ||||
6624 | if (ICI.getPredicate() == ICmpInst::ICMP_ULT || | ||||
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 6625 | ICI.getPredicate() == ICmpInst::ICMP_SLT) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6626 | return ReplaceInstUsesWith(ICI, Result); |
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 6627 | |
6628 | assert((ICI.getPredicate()==ICmpInst::ICMP_UGT || | ||||
6629 | ICI.getPredicate()==ICmpInst::ICMP_SGT) && | ||||
6630 | "ICmp should be folded!"); | ||||
6631 | if (Constant *CI = dyn_cast<Constant>(Result)) | ||||
6632 | return ReplaceInstUsesWith(ICI, ConstantExpr::getNot(CI)); | ||||
6633 | return BinaryOperator::CreateNot(Result); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6634 | } |
6635 | |||||
6636 | Instruction *InstCombiner::visitShl(BinaryOperator &I) { | ||||
6637 | return commonShiftTransforms(I); | ||||
6638 | } | ||||
6639 | |||||
6640 | Instruction *InstCombiner::visitLShr(BinaryOperator &I) { | ||||
6641 | return commonShiftTransforms(I); | ||||
6642 | } | ||||
6643 | |||||
6644 | Instruction *InstCombiner::visitAShr(BinaryOperator &I) { | ||||
Chris Lattner | e3c504f | 2007-12-06 01:59:46 +0000 | [diff] [blame] | 6645 | if (Instruction *R = commonShiftTransforms(I)) |
6646 | return R; | ||||
6647 | |||||
6648 | Value *Op0 = I.getOperand(0); | ||||
6649 | |||||
6650 | // ashr int -1, X = -1 (for any arithmetic shift rights of ~0) | ||||
6651 | if (ConstantInt *CSI = dyn_cast<ConstantInt>(Op0)) | ||||
6652 | if (CSI->isAllOnesValue()) | ||||
6653 | return ReplaceInstUsesWith(I, CSI); | ||||
6654 | |||||
6655 | // See if we can turn a signed shr into an unsigned shr. | ||||
Nate Begeman | bb1ce94 | 2008-07-29 15:49:41 +0000 | [diff] [blame] | 6656 | if (!isa<VectorType>(I.getType()) && |
6657 | MaskedValueIsZero(Op0, | ||||
Chris Lattner | e3c504f | 2007-12-06 01:59:46 +0000 | [diff] [blame] | 6658 | APInt::getSignBit(I.getType()->getPrimitiveSizeInBits()))) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6659 | return BinaryOperator::CreateLShr(Op0, I.getOperand(1)); |
Chris Lattner | e3c504f | 2007-12-06 01:59:46 +0000 | [diff] [blame] | 6660 | |
6661 | return 0; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6662 | } |
6663 | |||||
6664 | Instruction *InstCombiner::commonShiftTransforms(BinaryOperator &I) { | ||||
6665 | assert(I.getOperand(1)->getType() == I.getOperand(0)->getType()); | ||||
6666 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
6667 | |||||
6668 | // shl X, 0 == X and shr X, 0 == X | ||||
6669 | // shl 0, X == 0 and shr 0, X == 0 | ||||
6670 | if (Op1 == Constant::getNullValue(Op1->getType()) || | ||||
6671 | Op0 == Constant::getNullValue(Op0->getType())) | ||||
6672 | return ReplaceInstUsesWith(I, Op0); | ||||
6673 | |||||
6674 | if (isa<UndefValue>(Op0)) { | ||||
6675 | if (I.getOpcode() == Instruction::AShr) // undef >>s X -> undef | ||||
6676 | return ReplaceInstUsesWith(I, Op0); | ||||
6677 | else // undef << X -> 0, undef >>u X -> 0 | ||||
6678 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
6679 | } | ||||
6680 | if (isa<UndefValue>(Op1)) { | ||||
6681 | if (I.getOpcode() == Instruction::AShr) // X >>s undef -> X | ||||
6682 | return ReplaceInstUsesWith(I, Op0); | ||||
6683 | else // X << undef, X >>u undef -> 0 | ||||
6684 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
6685 | } | ||||
6686 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6687 | // Try to fold constant and into select arguments. |
6688 | if (isa<Constant>(Op0)) | ||||
6689 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) | ||||
6690 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
6691 | return R; | ||||
6692 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6693 | if (ConstantInt *CUI = dyn_cast<ConstantInt>(Op1)) |
6694 | if (Instruction *Res = FoldShiftByConstant(Op0, CUI, I)) | ||||
6695 | return Res; | ||||
6696 | return 0; | ||||
6697 | } | ||||
6698 | |||||
6699 | Instruction *InstCombiner::FoldShiftByConstant(Value *Op0, ConstantInt *Op1, | ||||
6700 | BinaryOperator &I) { | ||||
6701 | bool isLeftShift = I.getOpcode() == Instruction::Shl; | ||||
6702 | |||||
6703 | // See if we can simplify any instructions used by the instruction whose sole | ||||
6704 | // purpose is to compute bits we don't care about. | ||||
6705 | uint32_t TypeBits = Op0->getType()->getPrimitiveSizeInBits(); | ||||
6706 | APInt KnownZero(TypeBits, 0), KnownOne(TypeBits, 0); | ||||
6707 | if (SimplifyDemandedBits(&I, APInt::getAllOnesValue(TypeBits), | ||||
6708 | KnownZero, KnownOne)) | ||||
6709 | return &I; | ||||
6710 | |||||
6711 | // shl uint X, 32 = 0 and shr ubyte Y, 9 = 0, ... just don't eliminate shr | ||||
6712 | // of a signed value. | ||||
6713 | // | ||||
6714 | if (Op1->uge(TypeBits)) { | ||||
6715 | if (I.getOpcode() != Instruction::AShr) | ||||
6716 | return ReplaceInstUsesWith(I, Constant::getNullValue(Op0->getType())); | ||||
6717 | else { | ||||
6718 | I.setOperand(1, ConstantInt::get(I.getType(), TypeBits-1)); | ||||
6719 | return &I; | ||||
6720 | } | ||||
6721 | } | ||||
6722 | |||||
6723 | // ((X*C1) << C2) == (X * (C1 << C2)) | ||||
6724 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0)) | ||||
6725 | if (BO->getOpcode() == Instruction::Mul && isLeftShift) | ||||
6726 | if (Constant *BOOp = dyn_cast<Constant>(BO->getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6727 | return BinaryOperator::CreateMul(BO->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6728 | ConstantExpr::getShl(BOOp, Op1)); |
6729 | |||||
6730 | // Try to fold constant and into select arguments. | ||||
6731 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
6732 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
6733 | return R; | ||||
6734 | if (isa<PHINode>(Op0)) | ||||
6735 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
6736 | return NV; | ||||
6737 | |||||
Chris Lattner | c6d1f64 | 2007-12-22 09:07:47 +0000 | [diff] [blame] | 6738 | // Fold shift2(trunc(shift1(x,c1)), c2) -> trunc(shift2(shift1(x,c1),c2)) |
6739 | if (TruncInst *TI = dyn_cast<TruncInst>(Op0)) { | ||||
6740 | Instruction *TrOp = dyn_cast<Instruction>(TI->getOperand(0)); | ||||
6741 | // If 'shift2' is an ashr, we would have to get the sign bit into a funny | ||||
6742 | // place. Don't try to do this transformation in this case. Also, we | ||||
6743 | // require that the input operand is a shift-by-constant so that we have | ||||
6744 | // confidence that the shifts will get folded together. We could do this | ||||
6745 | // xform in more cases, but it is unlikely to be profitable. | ||||
6746 | if (TrOp && I.isLogicalShift() && TrOp->isShift() && | ||||
6747 | isa<ConstantInt>(TrOp->getOperand(1))) { | ||||
6748 | // Okay, we'll do this xform. Make the shift of shift. | ||||
6749 | Constant *ShAmt = ConstantExpr::getZExt(Op1, TrOp->getType()); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6750 | Instruction *NSh = BinaryOperator::Create(I.getOpcode(), TrOp, ShAmt, |
Chris Lattner | c6d1f64 | 2007-12-22 09:07:47 +0000 | [diff] [blame] | 6751 | I.getName()); |
6752 | InsertNewInstBefore(NSh, I); // (shift2 (shift1 & 0x00FF), c2) | ||||
6753 | |||||
6754 | // For logical shifts, the truncation has the effect of making the high | ||||
6755 | // part of the register be zeros. Emulate this by inserting an AND to | ||||
6756 | // clear the top bits as needed. This 'and' will usually be zapped by | ||||
6757 | // other xforms later if dead. | ||||
6758 | unsigned SrcSize = TrOp->getType()->getPrimitiveSizeInBits(); | ||||
6759 | unsigned DstSize = TI->getType()->getPrimitiveSizeInBits(); | ||||
6760 | APInt MaskV(APInt::getLowBitsSet(SrcSize, DstSize)); | ||||
6761 | |||||
6762 | // The mask we constructed says what the trunc would do if occurring | ||||
6763 | // between the shifts. We want to know the effect *after* the second | ||||
6764 | // shift. We know that it is a logical shift by a constant, so adjust the | ||||
6765 | // mask as appropriate. | ||||
6766 | if (I.getOpcode() == Instruction::Shl) | ||||
6767 | MaskV <<= Op1->getZExtValue(); | ||||
6768 | else { | ||||
6769 | assert(I.getOpcode() == Instruction::LShr && "Unknown logical shift"); | ||||
6770 | MaskV = MaskV.lshr(Op1->getZExtValue()); | ||||
6771 | } | ||||
6772 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6773 | Instruction *And = BinaryOperator::CreateAnd(NSh, ConstantInt::get(MaskV), |
Chris Lattner | c6d1f64 | 2007-12-22 09:07:47 +0000 | [diff] [blame] | 6774 | TI->getName()); |
6775 | InsertNewInstBefore(And, I); // shift1 & 0x00FF | ||||
6776 | |||||
6777 | // Return the value truncated to the interesting size. | ||||
6778 | return new TruncInst(And, I.getType()); | ||||
6779 | } | ||||
6780 | } | ||||
6781 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6782 | if (Op0->hasOneUse()) { |
6783 | if (BinaryOperator *Op0BO = dyn_cast<BinaryOperator>(Op0)) { | ||||
6784 | // Turn ((X >> C) + Y) << C -> (X + (Y << C)) & (~0 << C) | ||||
6785 | Value *V1, *V2; | ||||
6786 | ConstantInt *CC; | ||||
6787 | switch (Op0BO->getOpcode()) { | ||||
6788 | default: break; | ||||
6789 | case Instruction::Add: | ||||
6790 | case Instruction::And: | ||||
6791 | case Instruction::Or: | ||||
6792 | case Instruction::Xor: { | ||||
6793 | // These operators commute. | ||||
6794 | // Turn (Y + (X >> C)) << C -> (X + (Y << C)) & (~0 << C) | ||||
6795 | if (isLeftShift && Op0BO->getOperand(1)->hasOneUse() && | ||||
6796 | match(Op0BO->getOperand(1), | ||||
6797 | m_Shr(m_Value(V1), m_ConstantInt(CC))) && CC == Op1) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6798 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6799 | Op0BO->getOperand(0), Op1, |
6800 | Op0BO->getName()); | ||||
6801 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
6802 | Instruction *X = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6803 | BinaryOperator::Create(Op0BO->getOpcode(), YS, V1, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6804 | Op0BO->getOperand(1)->getName()); |
6805 | InsertNewInstBefore(X, I); // (X + (Y << C)) | ||||
6806 | uint32_t Op1Val = Op1->getLimitedValue(TypeBits); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6807 | return BinaryOperator::CreateAnd(X, ConstantInt::get( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6808 | APInt::getHighBitsSet(TypeBits, TypeBits-Op1Val))); |
6809 | } | ||||
6810 | |||||
6811 | // Turn (Y + ((X >> C) & CC)) << C -> ((X & (CC << C)) + (Y << C)) | ||||
6812 | Value *Op0BOOp1 = Op0BO->getOperand(1); | ||||
6813 | if (isLeftShift && Op0BOOp1->hasOneUse() && | ||||
6814 | match(Op0BOOp1, | ||||
6815 | m_And(m_Shr(m_Value(V1), m_Value(V2)),m_ConstantInt(CC))) && | ||||
6816 | cast<BinaryOperator>(Op0BOOp1)->getOperand(0)->hasOneUse() && | ||||
6817 | V2 == Op1) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6818 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6819 | Op0BO->getOperand(0), Op1, |
6820 | Op0BO->getName()); | ||||
6821 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
6822 | Instruction *XM = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6823 | BinaryOperator::CreateAnd(V1, ConstantExpr::getShl(CC, Op1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6824 | V1->getName()+".mask"); |
6825 | InsertNewInstBefore(XM, I); // X & (CC << C) | ||||
6826 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6827 | return BinaryOperator::Create(Op0BO->getOpcode(), YS, XM); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6828 | } |
6829 | } | ||||
6830 | |||||
6831 | // FALL THROUGH. | ||||
6832 | case Instruction::Sub: { | ||||
6833 | // Turn ((X >> C) + Y) << C -> (X + (Y << C)) & (~0 << C) | ||||
6834 | if (isLeftShift && Op0BO->getOperand(0)->hasOneUse() && | ||||
6835 | match(Op0BO->getOperand(0), | ||||
6836 | m_Shr(m_Value(V1), m_ConstantInt(CC))) && CC == Op1) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6837 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6838 | Op0BO->getOperand(1), Op1, |
6839 | Op0BO->getName()); | ||||
6840 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
6841 | Instruction *X = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6842 | BinaryOperator::Create(Op0BO->getOpcode(), V1, YS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6843 | Op0BO->getOperand(0)->getName()); |
6844 | InsertNewInstBefore(X, I); // (X + (Y << C)) | ||||
6845 | uint32_t Op1Val = Op1->getLimitedValue(TypeBits); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6846 | return BinaryOperator::CreateAnd(X, ConstantInt::get( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6847 | APInt::getHighBitsSet(TypeBits, TypeBits-Op1Val))); |
6848 | } | ||||
6849 | |||||
6850 | // Turn (((X >> C)&CC) + Y) << C -> (X + (Y << C)) & (CC << C) | ||||
6851 | if (isLeftShift && Op0BO->getOperand(0)->hasOneUse() && | ||||
6852 | match(Op0BO->getOperand(0), | ||||
6853 | m_And(m_Shr(m_Value(V1), m_Value(V2)), | ||||
6854 | m_ConstantInt(CC))) && V2 == Op1 && | ||||
6855 | cast<BinaryOperator>(Op0BO->getOperand(0)) | ||||
6856 | ->getOperand(0)->hasOneUse()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6857 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6858 | Op0BO->getOperand(1), Op1, |
6859 | Op0BO->getName()); | ||||
6860 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
6861 | Instruction *XM = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6862 | BinaryOperator::CreateAnd(V1, ConstantExpr::getShl(CC, Op1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6863 | V1->getName()+".mask"); |
6864 | InsertNewInstBefore(XM, I); // X & (CC << C) | ||||
6865 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6866 | return BinaryOperator::Create(Op0BO->getOpcode(), XM, YS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6867 | } |
6868 | |||||
6869 | break; | ||||
6870 | } | ||||
6871 | } | ||||
6872 | |||||
6873 | |||||
6874 | // If the operand is an bitwise operator with a constant RHS, and the | ||||
6875 | // shift is the only use, we can pull it out of the shift. | ||||
6876 | if (ConstantInt *Op0C = dyn_cast<ConstantInt>(Op0BO->getOperand(1))) { | ||||
6877 | bool isValid = true; // Valid only for And, Or, Xor | ||||
6878 | bool highBitSet = false; // Transform if high bit of constant set? | ||||
6879 | |||||
6880 | switch (Op0BO->getOpcode()) { | ||||
6881 | default: isValid = false; break; // Do not perform transform! | ||||
6882 | case Instruction::Add: | ||||
6883 | isValid = isLeftShift; | ||||
6884 | break; | ||||
6885 | case Instruction::Or: | ||||
6886 | case Instruction::Xor: | ||||
6887 | highBitSet = false; | ||||
6888 | break; | ||||
6889 | case Instruction::And: | ||||
6890 | highBitSet = true; | ||||
6891 | break; | ||||
6892 | } | ||||
6893 | |||||
6894 | // If this is a signed shift right, and the high bit is modified | ||||
6895 | // by the logical operation, do not perform the transformation. | ||||
6896 | // The highBitSet boolean indicates the value of the high bit of | ||||
6897 | // the constant which would cause it to be modified for this | ||||
6898 | // operation. | ||||
6899 | // | ||||
Chris Lattner | 15b76e3 | 2007-12-06 06:25:04 +0000 | [diff] [blame] | 6900 | if (isValid && I.getOpcode() == Instruction::AShr) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6901 | isValid = Op0C->getValue()[TypeBits-1] == highBitSet; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6902 | |
6903 | if (isValid) { | ||||
6904 | Constant *NewRHS = ConstantExpr::get(I.getOpcode(), Op0C, Op1); | ||||
6905 | |||||
6906 | Instruction *NewShift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6907 | BinaryOperator::Create(I.getOpcode(), Op0BO->getOperand(0), Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6908 | InsertNewInstBefore(NewShift, I); |
6909 | NewShift->takeName(Op0BO); | ||||
6910 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6911 | return BinaryOperator::Create(Op0BO->getOpcode(), NewShift, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6912 | NewRHS); |
6913 | } | ||||
6914 | } | ||||
6915 | } | ||||
6916 | } | ||||
6917 | |||||
6918 | // Find out if this is a shift of a shift by a constant. | ||||
6919 | BinaryOperator *ShiftOp = dyn_cast<BinaryOperator>(Op0); | ||||
6920 | if (ShiftOp && !ShiftOp->isShift()) | ||||
6921 | ShiftOp = 0; | ||||
6922 | |||||
6923 | if (ShiftOp && isa<ConstantInt>(ShiftOp->getOperand(1))) { | ||||
6924 | ConstantInt *ShiftAmt1C = cast<ConstantInt>(ShiftOp->getOperand(1)); | ||||
6925 | uint32_t ShiftAmt1 = ShiftAmt1C->getLimitedValue(TypeBits); | ||||
6926 | uint32_t ShiftAmt2 = Op1->getLimitedValue(TypeBits); | ||||
6927 | assert(ShiftAmt2 != 0 && "Should have been simplified earlier"); | ||||
6928 | if (ShiftAmt1 == 0) return 0; // Will be simplified in the future. | ||||
6929 | Value *X = ShiftOp->getOperand(0); | ||||
6930 | |||||
6931 | uint32_t AmtSum = ShiftAmt1+ShiftAmt2; // Fold into one big shift. | ||||
6932 | if (AmtSum > TypeBits) | ||||
6933 | AmtSum = TypeBits; | ||||
6934 | |||||
6935 | const IntegerType *Ty = cast<IntegerType>(I.getType()); | ||||
6936 | |||||
6937 | // Check for (X << c1) << c2 and (X >> c1) >> c2 | ||||
6938 | if (I.getOpcode() == ShiftOp->getOpcode()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6939 | return BinaryOperator::Create(I.getOpcode(), X, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6940 | ConstantInt::get(Ty, AmtSum)); |
6941 | } else if (ShiftOp->getOpcode() == Instruction::LShr && | ||||
6942 | I.getOpcode() == Instruction::AShr) { | ||||
6943 | // ((X >>u C1) >>s C2) -> (X >>u (C1+C2)) since C1 != 0. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6944 | return BinaryOperator::CreateLShr(X, ConstantInt::get(Ty, AmtSum)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6945 | } else if (ShiftOp->getOpcode() == Instruction::AShr && |
6946 | I.getOpcode() == Instruction::LShr) { | ||||
6947 | // ((X >>s C1) >>u C2) -> ((X >>s (C1+C2)) & mask) since C1 != 0. | ||||
6948 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6949 | BinaryOperator::CreateAShr(X, ConstantInt::get(Ty, AmtSum)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6950 | InsertNewInstBefore(Shift, I); |
6951 | |||||
6952 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6953 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6954 | } |
6955 | |||||
6956 | // Okay, if we get here, one shift must be left, and the other shift must be | ||||
6957 | // right. See if the amounts are equal. | ||||
6958 | if (ShiftAmt1 == ShiftAmt2) { | ||||
6959 | // If we have ((X >>? C) << C), turn this into X & (-1 << C). | ||||
6960 | if (I.getOpcode() == Instruction::Shl) { | ||||
6961 | APInt Mask(APInt::getHighBitsSet(TypeBits, TypeBits - ShiftAmt1)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6962 | return BinaryOperator::CreateAnd(X, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6963 | } |
6964 | // If we have ((X << C) >>u C), turn this into X & (-1 >>u C). | ||||
6965 | if (I.getOpcode() == Instruction::LShr) { | ||||
6966 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt1)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6967 | return BinaryOperator::CreateAnd(X, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6968 | } |
6969 | // We can simplify ((X << C) >>s C) into a trunc + sext. | ||||
6970 | // NOTE: we could do this for any C, but that would make 'unusual' integer | ||||
6971 | // types. For now, just stick to ones well-supported by the code | ||||
6972 | // generators. | ||||
6973 | const Type *SExtType = 0; | ||||
6974 | switch (Ty->getBitWidth() - ShiftAmt1) { | ||||
6975 | case 1 : | ||||
6976 | case 8 : | ||||
6977 | case 16 : | ||||
6978 | case 32 : | ||||
6979 | case 64 : | ||||
6980 | case 128: | ||||
6981 | SExtType = IntegerType::get(Ty->getBitWidth() - ShiftAmt1); | ||||
6982 | break; | ||||
6983 | default: break; | ||||
6984 | } | ||||
6985 | if (SExtType) { | ||||
6986 | Instruction *NewTrunc = new TruncInst(X, SExtType, "sext"); | ||||
6987 | InsertNewInstBefore(NewTrunc, I); | ||||
6988 | return new SExtInst(NewTrunc, Ty); | ||||
6989 | } | ||||
6990 | // Otherwise, we can't handle it yet. | ||||
6991 | } else if (ShiftAmt1 < ShiftAmt2) { | ||||
6992 | uint32_t ShiftDiff = ShiftAmt2-ShiftAmt1; | ||||
6993 | |||||
6994 | // (X >>? C1) << C2 --> X << (C2-C1) & (-1 << C2) | ||||
6995 | if (I.getOpcode() == Instruction::Shl) { | ||||
6996 | assert(ShiftOp->getOpcode() == Instruction::LShr || | ||||
6997 | ShiftOp->getOpcode() == Instruction::AShr); | ||||
6998 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6999 | BinaryOperator::CreateShl(X, ConstantInt::get(Ty, ShiftDiff)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7000 | InsertNewInstBefore(Shift, I); |
7001 | |||||
7002 | APInt Mask(APInt::getHighBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7003 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7004 | } |
7005 | |||||
7006 | // (X << C1) >>u C2 --> X >>u (C2-C1) & (-1 >> C2) | ||||
7007 | if (I.getOpcode() == Instruction::LShr) { | ||||
7008 | assert(ShiftOp->getOpcode() == Instruction::Shl); | ||||
7009 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7010 | BinaryOperator::CreateLShr(X, ConstantInt::get(Ty, ShiftDiff)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7011 | InsertNewInstBefore(Shift, I); |
7012 | |||||
7013 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7014 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7015 | } |
7016 | |||||
7017 | // We can't handle (X << C1) >>s C2, it shifts arbitrary bits in. | ||||
7018 | } else { | ||||
7019 | assert(ShiftAmt2 < ShiftAmt1); | ||||
7020 | uint32_t ShiftDiff = ShiftAmt1-ShiftAmt2; | ||||
7021 | |||||
7022 | // (X >>? C1) << C2 --> X >>? (C1-C2) & (-1 << C2) | ||||
7023 | if (I.getOpcode() == Instruction::Shl) { | ||||
7024 | assert(ShiftOp->getOpcode() == Instruction::LShr || | ||||
7025 | ShiftOp->getOpcode() == Instruction::AShr); | ||||
7026 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7027 | BinaryOperator::Create(ShiftOp->getOpcode(), X, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7028 | ConstantInt::get(Ty, ShiftDiff)); |
7029 | InsertNewInstBefore(Shift, I); | ||||
7030 | |||||
7031 | APInt Mask(APInt::getHighBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7032 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7033 | } |
7034 | |||||
7035 | // (X << C1) >>u C2 --> X << (C1-C2) & (-1 >> C2) | ||||
7036 | if (I.getOpcode() == Instruction::LShr) { | ||||
7037 | assert(ShiftOp->getOpcode() == Instruction::Shl); | ||||
7038 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7039 | BinaryOperator::CreateShl(X, ConstantInt::get(Ty, ShiftDiff)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7040 | InsertNewInstBefore(Shift, I); |
7041 | |||||
7042 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7043 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7044 | } |
7045 | |||||
7046 | // We can't handle (X << C1) >>a C2, it shifts arbitrary bits in. | ||||
7047 | } | ||||
7048 | } | ||||
7049 | return 0; | ||||
7050 | } | ||||
7051 | |||||
7052 | |||||
7053 | /// DecomposeSimpleLinearExpr - Analyze 'Val', seeing if it is a simple linear | ||||
7054 | /// expression. If so, decompose it, returning some value X, such that Val is | ||||
7055 | /// X*Scale+Offset. | ||||
7056 | /// | ||||
7057 | static Value *DecomposeSimpleLinearExpr(Value *Val, unsigned &Scale, | ||||
7058 | int &Offset) { | ||||
7059 | assert(Val->getType() == Type::Int32Ty && "Unexpected allocation size type!"); | ||||
7060 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Val)) { | ||||
7061 | Offset = CI->getZExtValue(); | ||||
Chris Lattner | c59171a | 2007-10-12 05:30:59 +0000 | [diff] [blame] | 7062 | Scale = 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7063 | return ConstantInt::get(Type::Int32Ty, 0); |
Chris Lattner | c59171a | 2007-10-12 05:30:59 +0000 | [diff] [blame] | 7064 | } else if (BinaryOperator *I = dyn_cast<BinaryOperator>(Val)) { |
7065 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
7066 | if (I->getOpcode() == Instruction::Shl) { | ||||
7067 | // This is a value scaled by '1 << the shift amt'. | ||||
7068 | Scale = 1U << RHS->getZExtValue(); | ||||
7069 | Offset = 0; | ||||
7070 | return I->getOperand(0); | ||||
7071 | } else if (I->getOpcode() == Instruction::Mul) { | ||||
7072 | // This value is scaled by 'RHS'. | ||||
7073 | Scale = RHS->getZExtValue(); | ||||
7074 | Offset = 0; | ||||
7075 | return I->getOperand(0); | ||||
7076 | } else if (I->getOpcode() == Instruction::Add) { | ||||
7077 | // We have X+C. Check to see if we really have (X*C2)+C1, | ||||
7078 | // where C1 is divisible by C2. | ||||
7079 | unsigned SubScale; | ||||
7080 | Value *SubVal = | ||||
7081 | DecomposeSimpleLinearExpr(I->getOperand(0), SubScale, Offset); | ||||
7082 | Offset += RHS->getZExtValue(); | ||||
7083 | Scale = SubScale; | ||||
7084 | return SubVal; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7085 | } |
7086 | } | ||||
7087 | } | ||||
7088 | |||||
7089 | // Otherwise, we can't look past this. | ||||
7090 | Scale = 1; | ||||
7091 | Offset = 0; | ||||
7092 | return Val; | ||||
7093 | } | ||||
7094 | |||||
7095 | |||||
7096 | /// PromoteCastOfAllocation - If we find a cast of an allocation instruction, | ||||
7097 | /// try to eliminate the cast by moving the type information into the alloc. | ||||
7098 | Instruction *InstCombiner::PromoteCastOfAllocation(BitCastInst &CI, | ||||
7099 | AllocationInst &AI) { | ||||
7100 | const PointerType *PTy = cast<PointerType>(CI.getType()); | ||||
7101 | |||||
7102 | // Remove any uses of AI that are dead. | ||||
7103 | assert(!CI.use_empty() && "Dead instructions should be removed earlier!"); | ||||
7104 | |||||
7105 | for (Value::use_iterator UI = AI.use_begin(), E = AI.use_end(); UI != E; ) { | ||||
7106 | Instruction *User = cast<Instruction>(*UI++); | ||||
7107 | if (isInstructionTriviallyDead(User)) { | ||||
7108 | while (UI != E && *UI == User) | ||||
7109 | ++UI; // If this instruction uses AI more than once, don't break UI. | ||||
7110 | |||||
7111 | ++NumDeadInst; | ||||
7112 | DOUT << "IC: DCE: " << *User; | ||||
7113 | EraseInstFromFunction(*User); | ||||
7114 | } | ||||
7115 | } | ||||
7116 | |||||
7117 | // Get the type really allocated and the type casted to. | ||||
7118 | const Type *AllocElTy = AI.getAllocatedType(); | ||||
7119 | const Type *CastElTy = PTy->getElementType(); | ||||
7120 | if (!AllocElTy->isSized() || !CastElTy->isSized()) return 0; | ||||
7121 | |||||
7122 | unsigned AllocElTyAlign = TD->getABITypeAlignment(AllocElTy); | ||||
7123 | unsigned CastElTyAlign = TD->getABITypeAlignment(CastElTy); | ||||
7124 | if (CastElTyAlign < AllocElTyAlign) return 0; | ||||
7125 | |||||
7126 | // If the allocation has multiple uses, only promote it if we are strictly | ||||
7127 | // increasing the alignment of the resultant allocation. If we keep it the | ||||
7128 | // same, we open the door to infinite loops of various kinds. | ||||
7129 | if (!AI.hasOneUse() && CastElTyAlign == AllocElTyAlign) return 0; | ||||
7130 | |||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 7131 | uint64_t AllocElTySize = TD->getABITypeSize(AllocElTy); |
7132 | uint64_t CastElTySize = TD->getABITypeSize(CastElTy); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7133 | if (CastElTySize == 0 || AllocElTySize == 0) return 0; |
7134 | |||||
7135 | // See if we can satisfy the modulus by pulling a scale out of the array | ||||
7136 | // size argument. | ||||
7137 | unsigned ArraySizeScale; | ||||
7138 | int ArrayOffset; | ||||
7139 | Value *NumElements = // See if the array size is a decomposable linear expr. | ||||
7140 | DecomposeSimpleLinearExpr(AI.getOperand(0), ArraySizeScale, ArrayOffset); | ||||
7141 | |||||
7142 | // If we can now satisfy the modulus, by using a non-1 scale, we really can | ||||
7143 | // do the xform. | ||||
7144 | if ((AllocElTySize*ArraySizeScale) % CastElTySize != 0 || | ||||
7145 | (AllocElTySize*ArrayOffset ) % CastElTySize != 0) return 0; | ||||
7146 | |||||
7147 | unsigned Scale = (AllocElTySize*ArraySizeScale)/CastElTySize; | ||||
7148 | Value *Amt = 0; | ||||
7149 | if (Scale == 1) { | ||||
7150 | Amt = NumElements; | ||||
7151 | } else { | ||||
7152 | // If the allocation size is constant, form a constant mul expression | ||||
7153 | Amt = ConstantInt::get(Type::Int32Ty, Scale); | ||||
7154 | if (isa<ConstantInt>(NumElements)) | ||||
7155 | Amt = Multiply(cast<ConstantInt>(NumElements), cast<ConstantInt>(Amt)); | ||||
7156 | // otherwise multiply the amount and the number of elements | ||||
7157 | else if (Scale != 1) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7158 | Instruction *Tmp = BinaryOperator::CreateMul(Amt, NumElements, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7159 | Amt = InsertNewInstBefore(Tmp, AI); |
7160 | } | ||||
7161 | } | ||||
7162 | |||||
7163 | if (int Offset = (AllocElTySize*ArrayOffset)/CastElTySize) { | ||||
7164 | Value *Off = ConstantInt::get(Type::Int32Ty, Offset, true); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7165 | Instruction *Tmp = BinaryOperator::CreateAdd(Amt, Off, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7166 | Amt = InsertNewInstBefore(Tmp, AI); |
7167 | } | ||||
7168 | |||||
7169 | AllocationInst *New; | ||||
7170 | if (isa<MallocInst>(AI)) | ||||
7171 | New = new MallocInst(CastElTy, Amt, AI.getAlignment()); | ||||
7172 | else | ||||
7173 | New = new AllocaInst(CastElTy, Amt, AI.getAlignment()); | ||||
7174 | InsertNewInstBefore(New, AI); | ||||
7175 | New->takeName(&AI); | ||||
7176 | |||||
7177 | // If the allocation has multiple uses, insert a cast and change all things | ||||
7178 | // that used it to use the new cast. This will also hack on CI, but it will | ||||
7179 | // die soon. | ||||
7180 | if (!AI.hasOneUse()) { | ||||
7181 | AddUsesToWorkList(AI); | ||||
7182 | // New is the allocation instruction, pointer typed. AI is the original | ||||
7183 | // allocation instruction, also pointer typed. Thus, cast to use is BitCast. | ||||
7184 | CastInst *NewCast = new BitCastInst(New, AI.getType(), "tmpcast"); | ||||
7185 | InsertNewInstBefore(NewCast, AI); | ||||
7186 | AI.replaceAllUsesWith(NewCast); | ||||
7187 | } | ||||
7188 | return ReplaceInstUsesWith(CI, New); | ||||
7189 | } | ||||
7190 | |||||
7191 | /// CanEvaluateInDifferentType - Return true if we can take the specified value | ||||
7192 | /// and return it as type Ty without inserting any new casts and without | ||||
7193 | /// changing the computed value. This is used by code that tries to decide | ||||
7194 | /// whether promoting or shrinking integer operations to wider or smaller types | ||||
7195 | /// will allow us to eliminate a truncate or extend. | ||||
7196 | /// | ||||
7197 | /// This is a truncation operation if Ty is smaller than V->getType(), or an | ||||
7198 | /// extension operation if Ty is larger. | ||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7199 | /// |
7200 | /// If CastOpc is a truncation, then Ty will be a type smaller than V. We | ||||
7201 | /// should return true if trunc(V) can be computed by computing V in the smaller | ||||
7202 | /// type. If V is an instruction, then trunc(inst(x,y)) can be computed as | ||||
7203 | /// inst(trunc(x),trunc(y)), which only makes sense if x and y can be | ||||
7204 | /// efficiently truncated. | ||||
7205 | /// | ||||
7206 | /// If CastOpc is a sext or zext, we are asking if the low bits of the value can | ||||
7207 | /// bit computed in a larger type, which is then and'd or sext_in_reg'd to get | ||||
7208 | /// the final result. | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 7209 | bool InstCombiner::CanEvaluateInDifferentType(Value *V, const IntegerType *Ty, |
7210 | unsigned CastOpc, | ||||
7211 | int &NumCastsRemoved) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7212 | // We can always evaluate constants in another type. |
7213 | if (isa<ConstantInt>(V)) | ||||
7214 | return true; | ||||
7215 | |||||
7216 | Instruction *I = dyn_cast<Instruction>(V); | ||||
7217 | if (!I) return false; | ||||
7218 | |||||
7219 | const IntegerType *OrigTy = cast<IntegerType>(V->getType()); | ||||
7220 | |||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7221 | // If this is an extension or truncate, we can often eliminate it. |
7222 | if (isa<TruncInst>(I) || isa<ZExtInst>(I) || isa<SExtInst>(I)) { | ||||
7223 | // If this is a cast from the destination type, we can trivially eliminate | ||||
7224 | // it, and this will remove a cast overall. | ||||
7225 | if (I->getOperand(0)->getType() == Ty) { | ||||
7226 | // If the first operand is itself a cast, and is eliminable, do not count | ||||
7227 | // this as an eliminable cast. We would prefer to eliminate those two | ||||
7228 | // casts first. | ||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7229 | if (!isa<CastInst>(I->getOperand(0)) && I->hasOneUse()) |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7230 | ++NumCastsRemoved; |
7231 | return true; | ||||
7232 | } | ||||
7233 | } | ||||
7234 | |||||
7235 | // We can't extend or shrink something that has multiple uses: doing so would | ||||
7236 | // require duplicating the instruction in general, which isn't profitable. | ||||
7237 | if (!I->hasOneUse()) return false; | ||||
7238 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7239 | switch (I->getOpcode()) { |
7240 | case Instruction::Add: | ||||
7241 | case Instruction::Sub: | ||||
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7242 | case Instruction::Mul: |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7243 | case Instruction::And: |
7244 | case Instruction::Or: | ||||
7245 | case Instruction::Xor: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7246 | // These operators can all arbitrarily be extended or truncated. |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7247 | return CanEvaluateInDifferentType(I->getOperand(0), Ty, CastOpc, |
7248 | NumCastsRemoved) && | ||||
7249 | CanEvaluateInDifferentType(I->getOperand(1), Ty, CastOpc, | ||||
7250 | NumCastsRemoved); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7251 | |
7252 | case Instruction::Shl: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7253 | // If we are truncating the result of this SHL, and if it's a shift of a |
7254 | // constant amount, we can always perform a SHL in a smaller type. | ||||
7255 | if (ConstantInt *CI = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
7256 | uint32_t BitWidth = Ty->getBitWidth(); | ||||
7257 | if (BitWidth < OrigTy->getBitWidth() && | ||||
7258 | CI->getLimitedValue(BitWidth) < BitWidth) | ||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7259 | return CanEvaluateInDifferentType(I->getOperand(0), Ty, CastOpc, |
7260 | NumCastsRemoved); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7261 | } |
7262 | break; | ||||
7263 | case Instruction::LShr: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7264 | // If this is a truncate of a logical shr, we can truncate it to a smaller |
7265 | // lshr iff we know that the bits we would otherwise be shifting in are | ||||
7266 | // already zeros. | ||||
7267 | if (ConstantInt *CI = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
7268 | uint32_t OrigBitWidth = OrigTy->getBitWidth(); | ||||
7269 | uint32_t BitWidth = Ty->getBitWidth(); | ||||
7270 | if (BitWidth < OrigBitWidth && | ||||
7271 | MaskedValueIsZero(I->getOperand(0), | ||||
7272 | APInt::getHighBitsSet(OrigBitWidth, OrigBitWidth-BitWidth)) && | ||||
7273 | CI->getLimitedValue(BitWidth) < BitWidth) { | ||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7274 | return CanEvaluateInDifferentType(I->getOperand(0), Ty, CastOpc, |
7275 | NumCastsRemoved); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7276 | } |
7277 | } | ||||
7278 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7279 | case Instruction::ZExt: |
7280 | case Instruction::SExt: | ||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7281 | case Instruction::Trunc: |
7282 | // 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] | 7283 | // can safely replace it. Note that replacing it does not reduce the number |
7284 | // of casts in the input. | ||||
7285 | if (I->getOpcode() == CastOpc) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7286 | return true; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7287 | break; |
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7288 | case Instruction::Select: { |
7289 | SelectInst *SI = cast<SelectInst>(I); | ||||
7290 | return CanEvaluateInDifferentType(SI->getTrueValue(), Ty, CastOpc, | ||||
7291 | NumCastsRemoved) && | ||||
7292 | CanEvaluateInDifferentType(SI->getFalseValue(), Ty, CastOpc, | ||||
7293 | NumCastsRemoved); | ||||
7294 | } | ||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7295 | case Instruction::PHI: { |
7296 | // We can change a phi if we can change all operands. | ||||
7297 | PHINode *PN = cast<PHINode>(I); | ||||
7298 | for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) | ||||
7299 | if (!CanEvaluateInDifferentType(PN->getIncomingValue(i), Ty, CastOpc, | ||||
7300 | NumCastsRemoved)) | ||||
7301 | return false; | ||||
7302 | return true; | ||||
7303 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7304 | default: |
7305 | // TODO: Can handle more cases here. | ||||
7306 | break; | ||||
7307 | } | ||||
7308 | |||||
7309 | return false; | ||||
7310 | } | ||||
7311 | |||||
7312 | /// EvaluateInDifferentType - Given an expression that | ||||
7313 | /// CanEvaluateInDifferentType returns true for, actually insert the code to | ||||
7314 | /// evaluate the expression. | ||||
7315 | Value *InstCombiner::EvaluateInDifferentType(Value *V, const Type *Ty, | ||||
7316 | bool isSigned) { | ||||
7317 | if (Constant *C = dyn_cast<Constant>(V)) | ||||
7318 | return ConstantExpr::getIntegerCast(C, Ty, isSigned /*Sext or ZExt*/); | ||||
7319 | |||||
7320 | // Otherwise, it must be an instruction. | ||||
7321 | Instruction *I = cast<Instruction>(V); | ||||
7322 | Instruction *Res = 0; | ||||
7323 | switch (I->getOpcode()) { | ||||
7324 | case Instruction::Add: | ||||
7325 | case Instruction::Sub: | ||||
Nick Lewycky | c52646a | 2008-01-22 05:08:48 +0000 | [diff] [blame] | 7326 | case Instruction::Mul: |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7327 | case Instruction::And: |
7328 | case Instruction::Or: | ||||
7329 | case Instruction::Xor: | ||||
7330 | case Instruction::AShr: | ||||
7331 | case Instruction::LShr: | ||||
7332 | case Instruction::Shl: { | ||||
7333 | Value *LHS = EvaluateInDifferentType(I->getOperand(0), Ty, isSigned); | ||||
7334 | Value *RHS = EvaluateInDifferentType(I->getOperand(1), Ty, isSigned); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7335 | Res = BinaryOperator::Create((Instruction::BinaryOps)I->getOpcode(), |
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7336 | LHS, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7337 | break; |
7338 | } | ||||
7339 | case Instruction::Trunc: | ||||
7340 | case Instruction::ZExt: | ||||
7341 | case Instruction::SExt: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7342 | // 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] | 7343 | // just return the source. There's no need to insert it because it is not |
7344 | // new. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7345 | if (I->getOperand(0)->getType() == Ty) |
7346 | return I->getOperand(0); | ||||
7347 | |||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7348 | // 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] | 7349 | Res = CastInst::Create(cast<CastInst>(I)->getOpcode(), I->getOperand(0), |
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7350 | Ty); |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7351 | break; |
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7352 | case Instruction::Select: { |
7353 | Value *True = EvaluateInDifferentType(I->getOperand(1), Ty, isSigned); | ||||
7354 | Value *False = EvaluateInDifferentType(I->getOperand(2), Ty, isSigned); | ||||
7355 | Res = SelectInst::Create(I->getOperand(0), True, False); | ||||
7356 | break; | ||||
7357 | } | ||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7358 | case Instruction::PHI: { |
7359 | PHINode *OPN = cast<PHINode>(I); | ||||
7360 | PHINode *NPN = PHINode::Create(Ty); | ||||
7361 | for (unsigned i = 0, e = OPN->getNumIncomingValues(); i != e; ++i) { | ||||
7362 | Value *V =EvaluateInDifferentType(OPN->getIncomingValue(i), Ty, isSigned); | ||||
7363 | NPN->addIncoming(V, OPN->getIncomingBlock(i)); | ||||
7364 | } | ||||
7365 | Res = NPN; | ||||
7366 | break; | ||||
7367 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7368 | default: |
7369 | // TODO: Can handle more cases here. | ||||
7370 | assert(0 && "Unreachable!"); | ||||
7371 | break; | ||||
7372 | } | ||||
7373 | |||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7374 | Res->takeName(I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7375 | return InsertNewInstBefore(Res, *I); |
7376 | } | ||||
7377 | |||||
7378 | /// @brief Implement the transforms common to all CastInst visitors. | ||||
7379 | Instruction *InstCombiner::commonCastTransforms(CastInst &CI) { | ||||
7380 | Value *Src = CI.getOperand(0); | ||||
7381 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7382 | // Many cases of "cast of a cast" are eliminable. If it's eliminable we just |
7383 | // eliminate it now. | ||||
7384 | if (CastInst *CSrc = dyn_cast<CastInst>(Src)) { // A->B->C cast | ||||
7385 | if (Instruction::CastOps opc = | ||||
7386 | isEliminableCastPair(CSrc, CI.getOpcode(), CI.getType(), TD)) { | ||||
7387 | // The first cast (CSrc) is eliminable so we need to fix up or replace | ||||
7388 | // 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] | 7389 | return CastInst::Create(opc, CSrc->getOperand(0), CI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7390 | } |
7391 | } | ||||
7392 | |||||
7393 | // If we are casting a select then fold the cast into the select | ||||
7394 | if (SelectInst *SI = dyn_cast<SelectInst>(Src)) | ||||
7395 | if (Instruction *NV = FoldOpIntoSelect(CI, SI, this)) | ||||
7396 | return NV; | ||||
7397 | |||||
7398 | // If we are casting a PHI then fold the cast into the PHI | ||||
7399 | if (isa<PHINode>(Src)) | ||||
7400 | if (Instruction *NV = FoldOpIntoPhi(CI)) | ||||
7401 | return NV; | ||||
7402 | |||||
7403 | return 0; | ||||
7404 | } | ||||
7405 | |||||
7406 | /// @brief Implement the transforms for cast of pointer (bitcast/ptrtoint) | ||||
7407 | Instruction *InstCombiner::commonPointerCastTransforms(CastInst &CI) { | ||||
7408 | Value *Src = CI.getOperand(0); | ||||
7409 | |||||
7410 | if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Src)) { | ||||
7411 | // If casting the result of a getelementptr instruction with no offset, turn | ||||
7412 | // this into a cast of the original pointer! | ||||
7413 | if (GEP->hasAllZeroIndices()) { | ||||
7414 | // Changing the cast operand is usually not a good idea but it is safe | ||||
7415 | // here because the pointer operand is being replaced with another | ||||
7416 | // pointer operand so the opcode doesn't need to change. | ||||
7417 | AddToWorkList(GEP); | ||||
7418 | CI.setOperand(0, GEP->getOperand(0)); | ||||
7419 | return &CI; | ||||
7420 | } | ||||
7421 | |||||
7422 | // If the GEP has a single use, and the base pointer is a bitcast, and the | ||||
7423 | // GEP computes a constant offset, see if we can convert these three | ||||
7424 | // instructions into fewer. This typically happens with unions and other | ||||
7425 | // non-type-safe code. | ||||
7426 | if (GEP->hasOneUse() && isa<BitCastInst>(GEP->getOperand(0))) { | ||||
7427 | if (GEP->hasAllConstantIndices()) { | ||||
7428 | // We are guaranteed to get a constant from EmitGEPOffset. | ||||
7429 | ConstantInt *OffsetV = cast<ConstantInt>(EmitGEPOffset(GEP, CI, *this)); | ||||
7430 | int64_t Offset = OffsetV->getSExtValue(); | ||||
7431 | |||||
7432 | // Get the base pointer input of the bitcast, and the type it points to. | ||||
7433 | Value *OrigBase = cast<BitCastInst>(GEP->getOperand(0))->getOperand(0); | ||||
7434 | const Type *GEPIdxTy = | ||||
7435 | cast<PointerType>(OrigBase->getType())->getElementType(); | ||||
7436 | if (GEPIdxTy->isSized()) { | ||||
7437 | SmallVector<Value*, 8> NewIndices; | ||||
7438 | |||||
7439 | // Start with the index over the outer type. Note that the type size | ||||
7440 | // might be zero (even if the offset isn't zero) if the indexed type | ||||
7441 | // is something like [0 x {int, int}] | ||||
7442 | const Type *IntPtrTy = TD->getIntPtrType(); | ||||
7443 | int64_t FirstIdx = 0; | ||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 7444 | if (int64_t TySize = TD->getABITypeSize(GEPIdxTy)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7445 | FirstIdx = Offset/TySize; |
7446 | Offset %= TySize; | ||||
7447 | |||||
7448 | // Handle silly modulus not returning values values [0..TySize). | ||||
7449 | if (Offset < 0) { | ||||
7450 | --FirstIdx; | ||||
7451 | Offset += TySize; | ||||
7452 | assert(Offset >= 0); | ||||
7453 | } | ||||
7454 | assert((uint64_t)Offset < (uint64_t)TySize &&"Out of range offset"); | ||||
7455 | } | ||||
7456 | |||||
7457 | NewIndices.push_back(ConstantInt::get(IntPtrTy, FirstIdx)); | ||||
7458 | |||||
7459 | // Index into the types. If we fail, set OrigBase to null. | ||||
7460 | while (Offset) { | ||||
7461 | if (const StructType *STy = dyn_cast<StructType>(GEPIdxTy)) { | ||||
7462 | const StructLayout *SL = TD->getStructLayout(STy); | ||||
7463 | if (Offset < (int64_t)SL->getSizeInBytes()) { | ||||
7464 | unsigned Elt = SL->getElementContainingOffset(Offset); | ||||
7465 | NewIndices.push_back(ConstantInt::get(Type::Int32Ty, Elt)); | ||||
7466 | |||||
7467 | Offset -= SL->getElementOffset(Elt); | ||||
7468 | GEPIdxTy = STy->getElementType(Elt); | ||||
7469 | } else { | ||||
7470 | // Otherwise, we can't index into this, bail out. | ||||
7471 | Offset = 0; | ||||
7472 | OrigBase = 0; | ||||
7473 | } | ||||
7474 | } else if (isa<ArrayType>(GEPIdxTy) || isa<VectorType>(GEPIdxTy)) { | ||||
7475 | const SequentialType *STy = cast<SequentialType>(GEPIdxTy); | ||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 7476 | if (uint64_t EltSize = TD->getABITypeSize(STy->getElementType())){ |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7477 | NewIndices.push_back(ConstantInt::get(IntPtrTy,Offset/EltSize)); |
7478 | Offset %= EltSize; | ||||
7479 | } else { | ||||
7480 | NewIndices.push_back(ConstantInt::get(IntPtrTy, 0)); | ||||
7481 | } | ||||
7482 | GEPIdxTy = STy->getElementType(); | ||||
7483 | } else { | ||||
7484 | // Otherwise, we can't index into this, bail out. | ||||
7485 | Offset = 0; | ||||
7486 | OrigBase = 0; | ||||
7487 | } | ||||
7488 | } | ||||
7489 | if (OrigBase) { | ||||
7490 | // If we were able to index down into an element, create the GEP | ||||
7491 | // and bitcast the result. This eliminates one bitcast, potentially | ||||
7492 | // two. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 7493 | Instruction *NGEP = GetElementPtrInst::Create(OrigBase, |
7494 | NewIndices.begin(), | ||||
7495 | NewIndices.end(), ""); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7496 | InsertNewInstBefore(NGEP, CI); |
7497 | NGEP->takeName(GEP); | ||||
7498 | |||||
7499 | if (isa<BitCastInst>(CI)) | ||||
7500 | return new BitCastInst(NGEP, CI.getType()); | ||||
7501 | assert(isa<PtrToIntInst>(CI)); | ||||
7502 | return new PtrToIntInst(NGEP, CI.getType()); | ||||
7503 | } | ||||
7504 | } | ||||
7505 | } | ||||
7506 | } | ||||
7507 | } | ||||
7508 | |||||
7509 | return commonCastTransforms(CI); | ||||
7510 | } | ||||
7511 | |||||
7512 | |||||
7513 | |||||
7514 | /// Only the TRUNC, ZEXT, SEXT, and BITCAST can both operand and result as | ||||
7515 | /// integer types. This function implements the common transforms for all those | ||||
7516 | /// cases. | ||||
7517 | /// @brief Implement the transforms common to CastInst with integer operands | ||||
7518 | Instruction *InstCombiner::commonIntCastTransforms(CastInst &CI) { | ||||
7519 | if (Instruction *Result = commonCastTransforms(CI)) | ||||
7520 | return Result; | ||||
7521 | |||||
7522 | Value *Src = CI.getOperand(0); | ||||
7523 | const Type *SrcTy = Src->getType(); | ||||
7524 | const Type *DestTy = CI.getType(); | ||||
7525 | uint32_t SrcBitSize = SrcTy->getPrimitiveSizeInBits(); | ||||
7526 | uint32_t DestBitSize = DestTy->getPrimitiveSizeInBits(); | ||||
7527 | |||||
7528 | // See if we can simplify any instructions used by the LHS whose sole | ||||
7529 | // purpose is to compute bits we don't care about. | ||||
7530 | APInt KnownZero(DestBitSize, 0), KnownOne(DestBitSize, 0); | ||||
7531 | if (SimplifyDemandedBits(&CI, APInt::getAllOnesValue(DestBitSize), | ||||
7532 | KnownZero, KnownOne)) | ||||
7533 | return &CI; | ||||
7534 | |||||
7535 | // If the source isn't an instruction or has more than one use then we | ||||
7536 | // can't do anything more. | ||||
7537 | Instruction *SrcI = dyn_cast<Instruction>(Src); | ||||
7538 | if (!SrcI || !Src->hasOneUse()) | ||||
7539 | return 0; | ||||
7540 | |||||
7541 | // Attempt to propagate the cast into the instruction for int->int casts. | ||||
7542 | int NumCastsRemoved = 0; | ||||
7543 | if (!isa<BitCastInst>(CI) && | ||||
7544 | CanEvaluateInDifferentType(SrcI, cast<IntegerType>(DestTy), | ||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7545 | CI.getOpcode(), NumCastsRemoved)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7546 | // If this cast is a truncate, evaluting in a different type always |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7547 | // eliminates the cast, so it is always a win. If this is a zero-extension, |
7548 | // we need to do an AND to maintain the clear top-part of the computation, | ||||
7549 | // so we require that the input have eliminated at least one cast. If this | ||||
7550 | // 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] | 7551 | // require that two casts have been eliminated. |
7552 | bool DoXForm; | ||||
7553 | switch (CI.getOpcode()) { | ||||
7554 | default: | ||||
7555 | // All the others use floating point so we shouldn't actually | ||||
7556 | // get here because of the check above. | ||||
7557 | assert(0 && "Unknown cast type"); | ||||
7558 | case Instruction::Trunc: | ||||
7559 | DoXForm = true; | ||||
7560 | break; | ||||
7561 | case Instruction::ZExt: | ||||
7562 | DoXForm = NumCastsRemoved >= 1; | ||||
7563 | break; | ||||
7564 | case Instruction::SExt: | ||||
7565 | DoXForm = NumCastsRemoved >= 2; | ||||
7566 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7567 | } |
7568 | |||||
7569 | if (DoXForm) { | ||||
7570 | Value *Res = EvaluateInDifferentType(SrcI, DestTy, | ||||
7571 | CI.getOpcode() == Instruction::SExt); | ||||
7572 | assert(Res->getType() == DestTy); | ||||
7573 | switch (CI.getOpcode()) { | ||||
7574 | default: assert(0 && "Unknown cast type!"); | ||||
7575 | case Instruction::Trunc: | ||||
7576 | case Instruction::BitCast: | ||||
7577 | // Just replace this cast with the result. | ||||
7578 | return ReplaceInstUsesWith(CI, Res); | ||||
7579 | case Instruction::ZExt: { | ||||
7580 | // We need to emit an AND to clear the high bits. | ||||
7581 | assert(SrcBitSize < DestBitSize && "Not a zext?"); | ||||
7582 | Constant *C = ConstantInt::get(APInt::getLowBitsSet(DestBitSize, | ||||
7583 | SrcBitSize)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7584 | return BinaryOperator::CreateAnd(Res, C); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7585 | } |
7586 | case Instruction::SExt: | ||||
7587 | // 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] | 7588 | return CastInst::Create(Instruction::SExt, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7589 | InsertCastBefore(Instruction::Trunc, Res, Src->getType(), |
7590 | CI), DestTy); | ||||
7591 | } | ||||
7592 | } | ||||
7593 | } | ||||
7594 | |||||
7595 | Value *Op0 = SrcI->getNumOperands() > 0 ? SrcI->getOperand(0) : 0; | ||||
7596 | Value *Op1 = SrcI->getNumOperands() > 1 ? SrcI->getOperand(1) : 0; | ||||
7597 | |||||
7598 | switch (SrcI->getOpcode()) { | ||||
7599 | case Instruction::Add: | ||||
7600 | case Instruction::Mul: | ||||
7601 | case Instruction::And: | ||||
7602 | case Instruction::Or: | ||||
7603 | case Instruction::Xor: | ||||
7604 | // If we are discarding information, rewrite. | ||||
7605 | if (DestBitSize <= SrcBitSize && DestBitSize != 1) { | ||||
7606 | // Don't insert two casts if they cannot be eliminated. We allow | ||||
7607 | // two casts to be inserted if the sizes are the same. This could | ||||
7608 | // only be converting signedness, which is a noop. | ||||
7609 | if (DestBitSize == SrcBitSize || | ||||
7610 | !ValueRequiresCast(CI.getOpcode(), Op1, DestTy,TD) || | ||||
7611 | !ValueRequiresCast(CI.getOpcode(), Op0, DestTy, TD)) { | ||||
7612 | Instruction::CastOps opcode = CI.getOpcode(); | ||||
7613 | Value *Op0c = InsertOperandCastBefore(opcode, Op0, DestTy, SrcI); | ||||
7614 | Value *Op1c = InsertOperandCastBefore(opcode, Op1, DestTy, SrcI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7615 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7616 | cast<BinaryOperator>(SrcI)->getOpcode(), Op0c, Op1c); |
7617 | } | ||||
7618 | } | ||||
7619 | |||||
7620 | // cast (xor bool X, true) to int --> xor (cast bool X to int), 1 | ||||
7621 | if (isa<ZExtInst>(CI) && SrcBitSize == 1 && | ||||
7622 | SrcI->getOpcode() == Instruction::Xor && | ||||
7623 | Op1 == ConstantInt::getTrue() && | ||||
7624 | (!Op0->hasOneUse() || !isa<CmpInst>(Op0))) { | ||||
7625 | Value *New = InsertOperandCastBefore(Instruction::ZExt, Op0, DestTy, &CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7626 | return BinaryOperator::CreateXor(New, ConstantInt::get(CI.getType(), 1)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7627 | } |
7628 | break; | ||||
7629 | case Instruction::SDiv: | ||||
7630 | case Instruction::UDiv: | ||||
7631 | case Instruction::SRem: | ||||
7632 | case Instruction::URem: | ||||
7633 | // If we are just changing the sign, rewrite. | ||||
7634 | if (DestBitSize == SrcBitSize) { | ||||
7635 | // Don't insert two casts if they cannot be eliminated. We allow | ||||
7636 | // two casts to be inserted if the sizes are the same. This could | ||||
7637 | // only be converting signedness, which is a noop. | ||||
7638 | if (!ValueRequiresCast(CI.getOpcode(), Op1, DestTy, TD) || | ||||
7639 | !ValueRequiresCast(CI.getOpcode(), Op0, DestTy, TD)) { | ||||
7640 | Value *Op0c = InsertOperandCastBefore(Instruction::BitCast, | ||||
7641 | Op0, DestTy, SrcI); | ||||
7642 | Value *Op1c = InsertOperandCastBefore(Instruction::BitCast, | ||||
7643 | Op1, DestTy, SrcI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7644 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7645 | cast<BinaryOperator>(SrcI)->getOpcode(), Op0c, Op1c); |
7646 | } | ||||
7647 | } | ||||
7648 | break; | ||||
7649 | |||||
7650 | case Instruction::Shl: | ||||
7651 | // Allow changing the sign of the source operand. Do not allow | ||||
7652 | // changing the size of the shift, UNLESS the shift amount is a | ||||
7653 | // constant. We must not change variable sized shifts to a smaller | ||||
7654 | // size, because it is undefined to shift more bits out than exist | ||||
7655 | // in the value. | ||||
7656 | if (DestBitSize == SrcBitSize || | ||||
7657 | (DestBitSize < SrcBitSize && isa<Constant>(Op1))) { | ||||
7658 | Instruction::CastOps opcode = (DestBitSize == SrcBitSize ? | ||||
7659 | Instruction::BitCast : Instruction::Trunc); | ||||
7660 | Value *Op0c = InsertOperandCastBefore(opcode, Op0, DestTy, SrcI); | ||||
7661 | Value *Op1c = InsertOperandCastBefore(opcode, Op1, DestTy, SrcI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7662 | return BinaryOperator::CreateShl(Op0c, Op1c); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7663 | } |
7664 | break; | ||||
7665 | case Instruction::AShr: | ||||
7666 | // If this is a signed shr, and if all bits shifted in are about to be | ||||
7667 | // truncated off, turn it into an unsigned shr to allow greater | ||||
7668 | // simplifications. | ||||
7669 | if (DestBitSize < SrcBitSize && | ||||
7670 | isa<ConstantInt>(Op1)) { | ||||
7671 | uint32_t ShiftAmt = cast<ConstantInt>(Op1)->getLimitedValue(SrcBitSize); | ||||
7672 | if (SrcBitSize > ShiftAmt && SrcBitSize-ShiftAmt >= DestBitSize) { | ||||
7673 | // Insert the new logical shift right. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7674 | return BinaryOperator::CreateLShr(Op0, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7675 | } |
7676 | } | ||||
7677 | break; | ||||
7678 | } | ||||
7679 | return 0; | ||||
7680 | } | ||||
7681 | |||||
7682 | Instruction *InstCombiner::visitTrunc(TruncInst &CI) { | ||||
7683 | if (Instruction *Result = commonIntCastTransforms(CI)) | ||||
7684 | return Result; | ||||
7685 | |||||
7686 | Value *Src = CI.getOperand(0); | ||||
7687 | const Type *Ty = CI.getType(); | ||||
7688 | uint32_t DestBitWidth = Ty->getPrimitiveSizeInBits(); | ||||
7689 | uint32_t SrcBitWidth = cast<IntegerType>(Src->getType())->getBitWidth(); | ||||
7690 | |||||
7691 | if (Instruction *SrcI = dyn_cast<Instruction>(Src)) { | ||||
7692 | switch (SrcI->getOpcode()) { | ||||
7693 | default: break; | ||||
7694 | case Instruction::LShr: | ||||
7695 | // We can shrink lshr to something smaller if we know the bits shifted in | ||||
7696 | // are already zeros. | ||||
7697 | if (ConstantInt *ShAmtV = dyn_cast<ConstantInt>(SrcI->getOperand(1))) { | ||||
7698 | uint32_t ShAmt = ShAmtV->getLimitedValue(SrcBitWidth); | ||||
7699 | |||||
7700 | // Get a mask for the bits shifting in. | ||||
7701 | APInt Mask(APInt::getLowBitsSet(SrcBitWidth, ShAmt).shl(DestBitWidth)); | ||||
7702 | Value* SrcIOp0 = SrcI->getOperand(0); | ||||
7703 | if (SrcI->hasOneUse() && MaskedValueIsZero(SrcIOp0, Mask)) { | ||||
7704 | if (ShAmt >= DestBitWidth) // All zeros. | ||||
7705 | return ReplaceInstUsesWith(CI, Constant::getNullValue(Ty)); | ||||
7706 | |||||
7707 | // Okay, we can shrink this. Truncate the input, then return a new | ||||
7708 | // shift. | ||||
7709 | Value *V1 = InsertCastBefore(Instruction::Trunc, SrcIOp0, Ty, CI); | ||||
7710 | Value *V2 = InsertCastBefore(Instruction::Trunc, SrcI->getOperand(1), | ||||
7711 | Ty, CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7712 | return BinaryOperator::CreateLShr(V1, V2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7713 | } |
7714 | } else { // This is a variable shr. | ||||
7715 | |||||
7716 | // Turn 'trunc (lshr X, Y) to bool' into '(X & (1 << Y)) != 0'. This is | ||||
7717 | // more LLVM instructions, but allows '1 << Y' to be hoisted if | ||||
7718 | // loop-invariant and CSE'd. | ||||
7719 | if (CI.getType() == Type::Int1Ty && SrcI->hasOneUse()) { | ||||
7720 | Value *One = ConstantInt::get(SrcI->getType(), 1); | ||||
7721 | |||||
7722 | Value *V = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7723 | BinaryOperator::CreateShl(One, SrcI->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7724 | "tmp"), CI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7725 | V = InsertNewInstBefore(BinaryOperator::CreateAnd(V, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7726 | SrcI->getOperand(0), |
7727 | "tmp"), CI); | ||||
7728 | Value *Zero = Constant::getNullValue(V->getType()); | ||||
7729 | return new ICmpInst(ICmpInst::ICMP_NE, V, Zero); | ||||
7730 | } | ||||
7731 | } | ||||
7732 | break; | ||||
7733 | } | ||||
7734 | } | ||||
7735 | |||||
7736 | return 0; | ||||
7737 | } | ||||
7738 | |||||
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 7739 | /// transformZExtICmp - Transform (zext icmp) to bitwise / integer operations |
7740 | /// in order to eliminate the icmp. | ||||
7741 | Instruction *InstCombiner::transformZExtICmp(ICmpInst *ICI, Instruction &CI, | ||||
7742 | bool DoXform) { | ||||
7743 | // If we are just checking for a icmp eq of a single bit and zext'ing it | ||||
7744 | // to an integer, then shift the bit to the appropriate place and then | ||||
7745 | // cast to integer to avoid the comparison. | ||||
7746 | if (ConstantInt *Op1C = dyn_cast<ConstantInt>(ICI->getOperand(1))) { | ||||
7747 | const APInt &Op1CV = Op1C->getValue(); | ||||
7748 | |||||
7749 | // zext (x <s 0) to i32 --> x>>u31 true if signbit set. | ||||
7750 | // zext (x >s -1) to i32 --> (x>>u31)^1 true if signbit clear. | ||||
7751 | if ((ICI->getPredicate() == ICmpInst::ICMP_SLT && Op1CV == 0) || | ||||
7752 | (ICI->getPredicate() == ICmpInst::ICMP_SGT &&Op1CV.isAllOnesValue())) { | ||||
7753 | if (!DoXform) return ICI; | ||||
7754 | |||||
7755 | Value *In = ICI->getOperand(0); | ||||
7756 | Value *Sh = ConstantInt::get(In->getType(), | ||||
7757 | In->getType()->getPrimitiveSizeInBits()-1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7758 | In = InsertNewInstBefore(BinaryOperator::CreateLShr(In, Sh, |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 7759 | In->getName()+".lobit"), |
7760 | CI); | ||||
7761 | if (In->getType() != CI.getType()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7762 | In = CastInst::CreateIntegerCast(In, CI.getType(), |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 7763 | false/*ZExt*/, "tmp", &CI); |
7764 | |||||
7765 | if (ICI->getPredicate() == ICmpInst::ICMP_SGT) { | ||||
7766 | Constant *One = ConstantInt::get(In->getType(), 1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7767 | In = InsertNewInstBefore(BinaryOperator::CreateXor(In, One, |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 7768 | In->getName()+".not"), |
7769 | CI); | ||||
7770 | } | ||||
7771 | |||||
7772 | return ReplaceInstUsesWith(CI, In); | ||||
7773 | } | ||||
7774 | |||||
7775 | |||||
7776 | |||||
7777 | // zext (X == 0) to i32 --> X^1 iff X has only the low bit set. | ||||
7778 | // zext (X == 0) to i32 --> (X>>1)^1 iff X has only the 2nd bit set. | ||||
7779 | // zext (X == 1) to i32 --> X iff X has only the low bit set. | ||||
7780 | // zext (X == 2) to i32 --> X>>1 iff X has only the 2nd bit set. | ||||
7781 | // zext (X != 0) to i32 --> X iff X has only the low bit set. | ||||
7782 | // zext (X != 0) to i32 --> X>>1 iff X has only the 2nd bit set. | ||||
7783 | // zext (X != 1) to i32 --> X^1 iff X has only the low bit set. | ||||
7784 | // zext (X != 2) to i32 --> (X>>1)^1 iff X has only the 2nd bit set. | ||||
7785 | if ((Op1CV == 0 || Op1CV.isPowerOf2()) && | ||||
7786 | // This only works for EQ and NE | ||||
7787 | ICI->isEquality()) { | ||||
7788 | // If Op1C some other power of two, convert: | ||||
7789 | uint32_t BitWidth = Op1C->getType()->getBitWidth(); | ||||
7790 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
7791 | APInt TypeMask(APInt::getAllOnesValue(BitWidth)); | ||||
7792 | ComputeMaskedBits(ICI->getOperand(0), TypeMask, KnownZero, KnownOne); | ||||
7793 | |||||
7794 | APInt KnownZeroMask(~KnownZero); | ||||
7795 | if (KnownZeroMask.isPowerOf2()) { // Exactly 1 possible 1? | ||||
7796 | if (!DoXform) return ICI; | ||||
7797 | |||||
7798 | bool isNE = ICI->getPredicate() == ICmpInst::ICMP_NE; | ||||
7799 | if (Op1CV != 0 && (Op1CV != KnownZeroMask)) { | ||||
7800 | // (X&4) == 2 --> false | ||||
7801 | // (X&4) != 2 --> true | ||||
7802 | Constant *Res = ConstantInt::get(Type::Int1Ty, isNE); | ||||
7803 | Res = ConstantExpr::getZExt(Res, CI.getType()); | ||||
7804 | return ReplaceInstUsesWith(CI, Res); | ||||
7805 | } | ||||
7806 | |||||
7807 | uint32_t ShiftAmt = KnownZeroMask.logBase2(); | ||||
7808 | Value *In = ICI->getOperand(0); | ||||
7809 | if (ShiftAmt) { | ||||
7810 | // Perform a logical shr by shiftamt. | ||||
7811 | // Insert the shift to put the result in the low bit. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7812 | In = InsertNewInstBefore(BinaryOperator::CreateLShr(In, |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 7813 | ConstantInt::get(In->getType(), ShiftAmt), |
7814 | In->getName()+".lobit"), CI); | ||||
7815 | } | ||||
7816 | |||||
7817 | if ((Op1CV != 0) == isNE) { // Toggle the low bit. | ||||
7818 | Constant *One = ConstantInt::get(In->getType(), 1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7819 | In = BinaryOperator::CreateXor(In, One, "tmp"); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 7820 | InsertNewInstBefore(cast<Instruction>(In), CI); |
7821 | } | ||||
7822 | |||||
7823 | if (CI.getType() == In->getType()) | ||||
7824 | return ReplaceInstUsesWith(CI, In); | ||||
7825 | else | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7826 | return CastInst::CreateIntegerCast(In, CI.getType(), false/*ZExt*/); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 7827 | } |
7828 | } | ||||
7829 | } | ||||
7830 | |||||
7831 | return 0; | ||||
7832 | } | ||||
7833 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7834 | Instruction *InstCombiner::visitZExt(ZExtInst &CI) { |
7835 | // If one of the common conversion will work .. | ||||
7836 | if (Instruction *Result = commonIntCastTransforms(CI)) | ||||
7837 | return Result; | ||||
7838 | |||||
7839 | Value *Src = CI.getOperand(0); | ||||
7840 | |||||
7841 | // If this is a cast of a cast | ||||
7842 | if (CastInst *CSrc = dyn_cast<CastInst>(Src)) { // A->B->C cast | ||||
7843 | // If this is a TRUNC followed by a ZEXT then we are dealing with integral | ||||
7844 | // types and if the sizes are just right we can convert this into a logical | ||||
7845 | // 'and' which will be much cheaper than the pair of casts. | ||||
7846 | if (isa<TruncInst>(CSrc)) { | ||||
7847 | // Get the sizes of the types involved | ||||
7848 | Value *A = CSrc->getOperand(0); | ||||
7849 | uint32_t SrcSize = A->getType()->getPrimitiveSizeInBits(); | ||||
7850 | uint32_t MidSize = CSrc->getType()->getPrimitiveSizeInBits(); | ||||
7851 | uint32_t DstSize = CI.getType()->getPrimitiveSizeInBits(); | ||||
7852 | // If we're actually extending zero bits and the trunc is a no-op | ||||
7853 | if (MidSize < DstSize && SrcSize == DstSize) { | ||||
7854 | // Replace both of the casts with an And of the type mask. | ||||
7855 | APInt AndValue(APInt::getLowBitsSet(SrcSize, MidSize)); | ||||
7856 | Constant *AndConst = ConstantInt::get(AndValue); | ||||
7857 | Instruction *And = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7858 | BinaryOperator::CreateAnd(CSrc->getOperand(0), AndConst); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7859 | // Unfortunately, if the type changed, we need to cast it back. |
7860 | if (And->getType() != CI.getType()) { | ||||
7861 | And->setName(CSrc->getName()+".mask"); | ||||
7862 | InsertNewInstBefore(And, CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7863 | And = CastInst::CreateIntegerCast(And, CI.getType(), false/*ZExt*/); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7864 | } |
7865 | return And; | ||||
7866 | } | ||||
7867 | } | ||||
7868 | } | ||||
7869 | |||||
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 7870 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(Src)) |
7871 | return transformZExtICmp(ICI, CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7872 | |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 7873 | BinaryOperator *SrcI = dyn_cast<BinaryOperator>(Src); |
7874 | if (SrcI && SrcI->getOpcode() == Instruction::Or) { | ||||
7875 | // zext (or icmp, icmp) --> or (zext icmp), (zext icmp) if at least one | ||||
7876 | // of the (zext icmp) will be transformed. | ||||
7877 | ICmpInst *LHS = dyn_cast<ICmpInst>(SrcI->getOperand(0)); | ||||
7878 | ICmpInst *RHS = dyn_cast<ICmpInst>(SrcI->getOperand(1)); | ||||
7879 | if (LHS && RHS && LHS->hasOneUse() && RHS->hasOneUse() && | ||||
7880 | (transformZExtICmp(LHS, CI, false) || | ||||
7881 | transformZExtICmp(RHS, CI, false))) { | ||||
7882 | Value *LCast = InsertCastBefore(Instruction::ZExt, LHS, CI.getType(), CI); | ||||
7883 | Value *RCast = InsertCastBefore(Instruction::ZExt, RHS, CI.getType(), CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7884 | return BinaryOperator::Create(Instruction::Or, LCast, RCast); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7885 | } |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 7886 | } |
7887 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7888 | return 0; |
7889 | } | ||||
7890 | |||||
7891 | Instruction *InstCombiner::visitSExt(SExtInst &CI) { | ||||
7892 | if (Instruction *I = commonIntCastTransforms(CI)) | ||||
7893 | return I; | ||||
7894 | |||||
7895 | Value *Src = CI.getOperand(0); | ||||
7896 | |||||
7897 | // sext (x <s 0) -> ashr x, 31 -> all ones if signed | ||||
7898 | // sext (x >s -1) -> ashr x, 31 -> all ones if not signed | ||||
7899 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(Src)) { | ||||
7900 | // If we are just checking for a icmp eq of a single bit and zext'ing it | ||||
7901 | // to an integer, then shift the bit to the appropriate place and then | ||||
7902 | // cast to integer to avoid the comparison. | ||||
7903 | if (ConstantInt *Op1C = dyn_cast<ConstantInt>(ICI->getOperand(1))) { | ||||
7904 | const APInt &Op1CV = Op1C->getValue(); | ||||
7905 | |||||
7906 | // sext (x <s 0) to i32 --> x>>s31 true if signbit set. | ||||
7907 | // sext (x >s -1) to i32 --> (x>>s31)^-1 true if signbit clear. | ||||
7908 | if ((ICI->getPredicate() == ICmpInst::ICMP_SLT && Op1CV == 0) || | ||||
7909 | (ICI->getPredicate() == ICmpInst::ICMP_SGT &&Op1CV.isAllOnesValue())){ | ||||
7910 | Value *In = ICI->getOperand(0); | ||||
7911 | Value *Sh = ConstantInt::get(In->getType(), | ||||
7912 | In->getType()->getPrimitiveSizeInBits()-1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7913 | In = InsertNewInstBefore(BinaryOperator::CreateAShr(In, Sh, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7914 | In->getName()+".lobit"), |
7915 | CI); | ||||
7916 | if (In->getType() != CI.getType()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7917 | In = CastInst::CreateIntegerCast(In, CI.getType(), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7918 | true/*SExt*/, "tmp", &CI); |
7919 | |||||
7920 | if (ICI->getPredicate() == ICmpInst::ICMP_SGT) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7921 | In = InsertNewInstBefore(BinaryOperator::CreateNot(In, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7922 | In->getName()+".not"), CI); |
7923 | |||||
7924 | return ReplaceInstUsesWith(CI, In); | ||||
7925 | } | ||||
7926 | } | ||||
7927 | } | ||||
Dan Gohman | f0f1202 | 2008-05-20 21:01:12 +0000 | [diff] [blame] | 7928 | |
7929 | // See if the value being truncated is already sign extended. If so, just | ||||
7930 | // eliminate the trunc/sext pair. | ||||
7931 | if (getOpcode(Src) == Instruction::Trunc) { | ||||
7932 | Value *Op = cast<User>(Src)->getOperand(0); | ||||
7933 | unsigned OpBits = cast<IntegerType>(Op->getType())->getBitWidth(); | ||||
7934 | unsigned MidBits = cast<IntegerType>(Src->getType())->getBitWidth(); | ||||
7935 | unsigned DestBits = cast<IntegerType>(CI.getType())->getBitWidth(); | ||||
7936 | unsigned NumSignBits = ComputeNumSignBits(Op); | ||||
7937 | |||||
7938 | if (OpBits == DestBits) { | ||||
7939 | // Op is i32, Mid is i8, and Dest is i32. If Op has more than 24 sign | ||||
7940 | // bits, it is already ready. | ||||
7941 | if (NumSignBits > DestBits-MidBits) | ||||
7942 | return ReplaceInstUsesWith(CI, Op); | ||||
7943 | } else if (OpBits < DestBits) { | ||||
7944 | // Op is i32, Mid is i8, and Dest is i64. If Op has more than 24 sign | ||||
7945 | // bits, just sext from i32. | ||||
7946 | if (NumSignBits > OpBits-MidBits) | ||||
7947 | return new SExtInst(Op, CI.getType(), "tmp"); | ||||
7948 | } else { | ||||
7949 | // Op is i64, Mid is i8, and Dest is i32. If Op has more than 56 sign | ||||
7950 | // bits, just truncate to i32. | ||||
7951 | if (NumSignBits > OpBits-MidBits) | ||||
7952 | return new TruncInst(Op, CI.getType(), "tmp"); | ||||
7953 | } | ||||
7954 | } | ||||
Chris Lattner | 8a2d059 | 2008-08-06 07:35:52 +0000 | [diff] [blame] | 7955 | |
7956 | // If the input is a shl/ashr pair of a same constant, then this is a sign | ||||
7957 | // extension from a smaller value. If we could trust arbitrary bitwidth | ||||
7958 | // integers, we could turn this into a truncate to the smaller bit and then | ||||
7959 | // use a sext for the whole extension. Since we don't, look deeper and check | ||||
7960 | // for a truncate. If the source and dest are the same type, eliminate the | ||||
7961 | // trunc and extend and just do shifts. For example, turn: | ||||
7962 | // %a = trunc i32 %i to i8 | ||||
7963 | // %b = shl i8 %a, 6 | ||||
7964 | // %c = ashr i8 %b, 6 | ||||
7965 | // %d = sext i8 %c to i32 | ||||
7966 | // into: | ||||
7967 | // %a = shl i32 %i, 30 | ||||
7968 | // %d = ashr i32 %a, 30 | ||||
7969 | Value *A = 0; | ||||
7970 | ConstantInt *BA = 0, *CA = 0; | ||||
7971 | if (match(Src, m_AShr(m_Shl(m_Value(A), m_ConstantInt(BA)), | ||||
7972 | m_ConstantInt(CA))) && | ||||
7973 | BA == CA && isa<TruncInst>(A)) { | ||||
7974 | Value *I = cast<TruncInst>(A)->getOperand(0); | ||||
7975 | if (I->getType() == CI.getType()) { | ||||
7976 | unsigned MidSize = Src->getType()->getPrimitiveSizeInBits(); | ||||
7977 | unsigned SrcDstSize = CI.getType()->getPrimitiveSizeInBits(); | ||||
7978 | unsigned ShAmt = CA->getZExtValue()+SrcDstSize-MidSize; | ||||
7979 | Constant *ShAmtV = ConstantInt::get(CI.getType(), ShAmt); | ||||
7980 | I = InsertNewInstBefore(BinaryOperator::CreateShl(I, ShAmtV, | ||||
7981 | CI.getName()), CI); | ||||
7982 | return BinaryOperator::CreateAShr(I, ShAmtV); | ||||
7983 | } | ||||
7984 | } | ||||
7985 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7986 | return 0; |
7987 | } | ||||
7988 | |||||
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 7989 | /// FitsInFPType - Return a Constant* for the specified FP constant if it fits |
7990 | /// in the specified FP type without changing its value. | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 7991 | static Constant *FitsInFPType(ConstantFP *CFP, const fltSemantics &Sem) { |
Dale Johannesen | 6e547b4 | 2008-10-09 23:00:39 +0000 | [diff] [blame] | 7992 | bool losesInfo; |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 7993 | APFloat F = CFP->getValueAPF(); |
Dale Johannesen | 6e547b4 | 2008-10-09 23:00:39 +0000 | [diff] [blame] | 7994 | (void)F.convert(Sem, APFloat::rmNearestTiesToEven, &losesInfo); |
7995 | if (!losesInfo) | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 7996 | return ConstantFP::get(F); |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 7997 | return 0; |
7998 | } | ||||
7999 | |||||
8000 | /// LookThroughFPExtensions - If this is an fp extension instruction, look | ||||
8001 | /// through it until we get the source value. | ||||
8002 | static Value *LookThroughFPExtensions(Value *V) { | ||||
8003 | if (Instruction *I = dyn_cast<Instruction>(V)) | ||||
8004 | if (I->getOpcode() == Instruction::FPExt) | ||||
8005 | return LookThroughFPExtensions(I->getOperand(0)); | ||||
8006 | |||||
8007 | // If this value is a constant, return the constant in the smallest FP type | ||||
8008 | // that can accurately represent it. This allows us to turn | ||||
8009 | // (float)((double)X+2.0) into x+2.0f. | ||||
8010 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(V)) { | ||||
8011 | if (CFP->getType() == Type::PPC_FP128Ty) | ||||
8012 | return V; // No constant folding of this. | ||||
8013 | // See if the value can be truncated to float and then reextended. | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 8014 | if (Value *V = FitsInFPType(CFP, APFloat::IEEEsingle)) |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8015 | return V; |
8016 | if (CFP->getType() == Type::DoubleTy) | ||||
8017 | return V; // Won't shrink. | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 8018 | if (Value *V = FitsInFPType(CFP, APFloat::IEEEdouble)) |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8019 | return V; |
8020 | // Don't try to shrink to various long double types. | ||||
8021 | } | ||||
8022 | |||||
8023 | return V; | ||||
8024 | } | ||||
8025 | |||||
8026 | Instruction *InstCombiner::visitFPTrunc(FPTruncInst &CI) { | ||||
8027 | if (Instruction *I = commonCastTransforms(CI)) | ||||
8028 | return I; | ||||
8029 | |||||
8030 | // If we have fptrunc(add (fpextend x), (fpextend y)), where x and y are | ||||
8031 | // smaller than the destination type, we can eliminate the truncate by doing | ||||
8032 | // the add as the smaller type. This applies to add/sub/mul/div as well as | ||||
8033 | // many builtins (sqrt, etc). | ||||
8034 | BinaryOperator *OpI = dyn_cast<BinaryOperator>(CI.getOperand(0)); | ||||
8035 | if (OpI && OpI->hasOneUse()) { | ||||
8036 | switch (OpI->getOpcode()) { | ||||
8037 | default: break; | ||||
8038 | case Instruction::Add: | ||||
8039 | case Instruction::Sub: | ||||
8040 | case Instruction::Mul: | ||||
8041 | case Instruction::FDiv: | ||||
8042 | case Instruction::FRem: | ||||
8043 | const Type *SrcTy = OpI->getType(); | ||||
8044 | Value *LHSTrunc = LookThroughFPExtensions(OpI->getOperand(0)); | ||||
8045 | Value *RHSTrunc = LookThroughFPExtensions(OpI->getOperand(1)); | ||||
8046 | if (LHSTrunc->getType() != SrcTy && | ||||
8047 | RHSTrunc->getType() != SrcTy) { | ||||
8048 | unsigned DstSize = CI.getType()->getPrimitiveSizeInBits(); | ||||
8049 | // If the source types were both smaller than the destination type of | ||||
8050 | // the cast, do this xform. | ||||
8051 | if (LHSTrunc->getType()->getPrimitiveSizeInBits() <= DstSize && | ||||
8052 | RHSTrunc->getType()->getPrimitiveSizeInBits() <= DstSize) { | ||||
8053 | LHSTrunc = InsertCastBefore(Instruction::FPExt, LHSTrunc, | ||||
8054 | CI.getType(), CI); | ||||
8055 | RHSTrunc = InsertCastBefore(Instruction::FPExt, RHSTrunc, | ||||
8056 | CI.getType(), CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8057 | return BinaryOperator::Create(OpI->getOpcode(), LHSTrunc, RHSTrunc); |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8058 | } |
8059 | } | ||||
8060 | break; | ||||
8061 | } | ||||
8062 | } | ||||
8063 | return 0; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8064 | } |
8065 | |||||
8066 | Instruction *InstCombiner::visitFPExt(CastInst &CI) { | ||||
8067 | return commonCastTransforms(CI); | ||||
8068 | } | ||||
8069 | |||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8070 | Instruction *InstCombiner::visitFPToUI(FPToUIInst &FI) { |
Chris Lattner | 5f4d691 | 2008-08-06 05:13:06 +0000 | [diff] [blame] | 8071 | Instruction *OpI = dyn_cast<Instruction>(FI.getOperand(0)); |
8072 | if (OpI == 0) | ||||
8073 | return commonCastTransforms(FI); | ||||
8074 | |||||
8075 | // fptoui(uitofp(X)) --> X | ||||
8076 | // fptoui(sitofp(X)) --> X | ||||
8077 | // This is safe if the intermediate type has enough bits in its mantissa to | ||||
8078 | // accurately represent all values of X. For example, do not do this with | ||||
8079 | // i64->float->i64. This is also safe for sitofp case, because any negative | ||||
8080 | // 'X' value would cause an undefined result for the fptoui. | ||||
8081 | if ((isa<UIToFPInst>(OpI) || isa<SIToFPInst>(OpI)) && | ||||
8082 | OpI->getOperand(0)->getType() == FI.getType() && | ||||
8083 | (int)FI.getType()->getPrimitiveSizeInBits() < /*extra bit for sign */ | ||||
8084 | OpI->getType()->getFPMantissaWidth()) | ||||
8085 | return ReplaceInstUsesWith(FI, OpI->getOperand(0)); | ||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8086 | |
8087 | return commonCastTransforms(FI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8088 | } |
8089 | |||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8090 | Instruction *InstCombiner::visitFPToSI(FPToSIInst &FI) { |
Chris Lattner | 5f4d691 | 2008-08-06 05:13:06 +0000 | [diff] [blame] | 8091 | Instruction *OpI = dyn_cast<Instruction>(FI.getOperand(0)); |
8092 | if (OpI == 0) | ||||
8093 | return commonCastTransforms(FI); | ||||
8094 | |||||
8095 | // fptosi(sitofp(X)) --> X | ||||
8096 | // fptosi(uitofp(X)) --> X | ||||
8097 | // This is safe if the intermediate type has enough bits in its mantissa to | ||||
8098 | // accurately represent all values of X. For example, do not do this with | ||||
8099 | // i64->float->i64. This is also safe for sitofp case, because any negative | ||||
8100 | // 'X' value would cause an undefined result for the fptoui. | ||||
8101 | if ((isa<UIToFPInst>(OpI) || isa<SIToFPInst>(OpI)) && | ||||
8102 | OpI->getOperand(0)->getType() == FI.getType() && | ||||
8103 | (int)FI.getType()->getPrimitiveSizeInBits() <= | ||||
8104 | OpI->getType()->getFPMantissaWidth()) | ||||
8105 | return ReplaceInstUsesWith(FI, OpI->getOperand(0)); | ||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8106 | |
8107 | return commonCastTransforms(FI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8108 | } |
8109 | |||||
8110 | Instruction *InstCombiner::visitUIToFP(CastInst &CI) { | ||||
8111 | return commonCastTransforms(CI); | ||||
8112 | } | ||||
8113 | |||||
8114 | Instruction *InstCombiner::visitSIToFP(CastInst &CI) { | ||||
8115 | return commonCastTransforms(CI); | ||||
8116 | } | ||||
8117 | |||||
8118 | Instruction *InstCombiner::visitPtrToInt(CastInst &CI) { | ||||
8119 | return commonPointerCastTransforms(CI); | ||||
8120 | } | ||||
8121 | |||||
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8122 | Instruction *InstCombiner::visitIntToPtr(IntToPtrInst &CI) { |
8123 | if (Instruction *I = commonCastTransforms(CI)) | ||||
8124 | return I; | ||||
8125 | |||||
8126 | const Type *DestPointee = cast<PointerType>(CI.getType())->getElementType(); | ||||
8127 | if (!DestPointee->isSized()) return 0; | ||||
8128 | |||||
8129 | // If this is inttoptr(add (ptrtoint x), cst), try to turn this into a GEP. | ||||
8130 | ConstantInt *Cst; | ||||
8131 | Value *X; | ||||
8132 | if (match(CI.getOperand(0), m_Add(m_Cast<PtrToIntInst>(m_Value(X)), | ||||
8133 | m_ConstantInt(Cst)))) { | ||||
8134 | // If the source and destination operands have the same type, see if this | ||||
8135 | // is a single-index GEP. | ||||
8136 | if (X->getType() == CI.getType()) { | ||||
8137 | // Get the size of the pointee type. | ||||
Bill Wendling | 9594af0 | 2008-03-14 05:12:19 +0000 | [diff] [blame] | 8138 | uint64_t Size = TD->getABITypeSize(DestPointee); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8139 | |
8140 | // Convert the constant to intptr type. | ||||
8141 | APInt Offset = Cst->getValue(); | ||||
8142 | Offset.sextOrTrunc(TD->getPointerSizeInBits()); | ||||
8143 | |||||
8144 | // If Offset is evenly divisible by Size, we can do this xform. | ||||
8145 | if (Size && !APIntOps::srem(Offset, APInt(Offset.getBitWidth(), Size))){ | ||||
8146 | Offset = APIntOps::sdiv(Offset, APInt(Offset.getBitWidth(), Size)); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8147 | return GetElementPtrInst::Create(X, ConstantInt::get(Offset)); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8148 | } |
8149 | } | ||||
8150 | // TODO: Could handle other cases, e.g. where add is indexing into field of | ||||
8151 | // struct etc. | ||||
8152 | } else if (CI.getOperand(0)->hasOneUse() && | ||||
8153 | match(CI.getOperand(0), m_Add(m_Value(X), m_ConstantInt(Cst)))) { | ||||
8154 | // Otherwise, if this is inttoptr(add x, cst), try to turn this into an | ||||
8155 | // "inttoptr+GEP" instead of "add+intptr". | ||||
8156 | |||||
8157 | // Get the size of the pointee type. | ||||
8158 | uint64_t Size = TD->getABITypeSize(DestPointee); | ||||
8159 | |||||
8160 | // Convert the constant to intptr type. | ||||
8161 | APInt Offset = Cst->getValue(); | ||||
8162 | Offset.sextOrTrunc(TD->getPointerSizeInBits()); | ||||
8163 | |||||
8164 | // If Offset is evenly divisible by Size, we can do this xform. | ||||
8165 | if (Size && !APIntOps::srem(Offset, APInt(Offset.getBitWidth(), Size))){ | ||||
8166 | Offset = APIntOps::sdiv(Offset, APInt(Offset.getBitWidth(), Size)); | ||||
8167 | |||||
8168 | Instruction *P = InsertNewInstBefore(new IntToPtrInst(X, CI.getType(), | ||||
8169 | "tmp"), CI); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8170 | return GetElementPtrInst::Create(P, ConstantInt::get(Offset), "tmp"); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8171 | } |
8172 | } | ||||
8173 | return 0; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8174 | } |
8175 | |||||
8176 | Instruction *InstCombiner::visitBitCast(BitCastInst &CI) { | ||||
8177 | // If the operands are integer typed then apply the integer transforms, | ||||
8178 | // otherwise just apply the common ones. | ||||
8179 | Value *Src = CI.getOperand(0); | ||||
8180 | const Type *SrcTy = Src->getType(); | ||||
8181 | const Type *DestTy = CI.getType(); | ||||
8182 | |||||
8183 | if (SrcTy->isInteger() && DestTy->isInteger()) { | ||||
8184 | if (Instruction *Result = commonIntCastTransforms(CI)) | ||||
8185 | return Result; | ||||
8186 | } else if (isa<PointerType>(SrcTy)) { | ||||
8187 | if (Instruction *I = commonPointerCastTransforms(CI)) | ||||
8188 | return I; | ||||
8189 | } else { | ||||
8190 | if (Instruction *Result = commonCastTransforms(CI)) | ||||
8191 | return Result; | ||||
8192 | } | ||||
8193 | |||||
8194 | |||||
8195 | // Get rid of casts from one type to the same type. These are useless and can | ||||
8196 | // be replaced by the operand. | ||||
8197 | if (DestTy == Src->getType()) | ||||
8198 | return ReplaceInstUsesWith(CI, Src); | ||||
8199 | |||||
8200 | if (const PointerType *DstPTy = dyn_cast<PointerType>(DestTy)) { | ||||
8201 | const PointerType *SrcPTy = cast<PointerType>(SrcTy); | ||||
8202 | const Type *DstElTy = DstPTy->getElementType(); | ||||
8203 | const Type *SrcElTy = SrcPTy->getElementType(); | ||||
8204 | |||||
Nate Begeman | df5b361 | 2008-03-31 00:22:16 +0000 | [diff] [blame] | 8205 | // If the address spaces don't match, don't eliminate the bitcast, which is |
8206 | // required for changing types. | ||||
8207 | if (SrcPTy->getAddressSpace() != DstPTy->getAddressSpace()) | ||||
8208 | return 0; | ||||
8209 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8210 | // If we are casting a malloc or alloca to a pointer to a type of the same |
8211 | // size, rewrite the allocation instruction to allocate the "right" type. | ||||
8212 | if (AllocationInst *AI = dyn_cast<AllocationInst>(Src)) | ||||
8213 | if (Instruction *V = PromoteCastOfAllocation(CI, *AI)) | ||||
8214 | return V; | ||||
8215 | |||||
8216 | // If the source and destination are pointers, and this cast is equivalent | ||||
8217 | // to a getelementptr X, 0, 0, 0... turn it into the appropriate gep. | ||||
8218 | // This can enhance SROA and other transforms that want type-safe pointers. | ||||
8219 | Constant *ZeroUInt = Constant::getNullValue(Type::Int32Ty); | ||||
8220 | unsigned NumZeros = 0; | ||||
8221 | while (SrcElTy != DstElTy && | ||||
8222 | isa<CompositeType>(SrcElTy) && !isa<PointerType>(SrcElTy) && | ||||
8223 | SrcElTy->getNumContainedTypes() /* not "{}" */) { | ||||
8224 | SrcElTy = cast<CompositeType>(SrcElTy)->getTypeAtIndex(ZeroUInt); | ||||
8225 | ++NumZeros; | ||||
8226 | } | ||||
8227 | |||||
8228 | // If we found a path from the src to dest, create the getelementptr now. | ||||
8229 | if (SrcElTy == DstElTy) { | ||||
8230 | SmallVector<Value*, 8> Idxs(NumZeros+1, ZeroUInt); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8231 | return GetElementPtrInst::Create(Src, Idxs.begin(), Idxs.end(), "", |
8232 | ((Instruction*) NULL)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8233 | } |
8234 | } | ||||
8235 | |||||
8236 | if (ShuffleVectorInst *SVI = dyn_cast<ShuffleVectorInst>(Src)) { | ||||
8237 | if (SVI->hasOneUse()) { | ||||
8238 | // Okay, we have (bitconvert (shuffle ..)). Check to see if this is | ||||
8239 | // a bitconvert to a vector with the same # elts. | ||||
8240 | if (isa<VectorType>(DestTy) && | ||||
8241 | cast<VectorType>(DestTy)->getNumElements() == | ||||
8242 | SVI->getType()->getNumElements()) { | ||||
8243 | CastInst *Tmp; | ||||
8244 | // If either of the operands is a cast from CI.getType(), then | ||||
8245 | // evaluating the shuffle in the casted destination's type will allow | ||||
8246 | // us to eliminate at least one cast. | ||||
8247 | if (((Tmp = dyn_cast<CastInst>(SVI->getOperand(0))) && | ||||
8248 | Tmp->getOperand(0)->getType() == DestTy) || | ||||
8249 | ((Tmp = dyn_cast<CastInst>(SVI->getOperand(1))) && | ||||
8250 | Tmp->getOperand(0)->getType() == DestTy)) { | ||||
8251 | Value *LHS = InsertOperandCastBefore(Instruction::BitCast, | ||||
8252 | SVI->getOperand(0), DestTy, &CI); | ||||
8253 | Value *RHS = InsertOperandCastBefore(Instruction::BitCast, | ||||
8254 | SVI->getOperand(1), DestTy, &CI); | ||||
8255 | // Return a new shuffle vector. Use the same element ID's, as we | ||||
8256 | // know the vector types match #elts. | ||||
8257 | return new ShuffleVectorInst(LHS, RHS, SVI->getOperand(2)); | ||||
8258 | } | ||||
8259 | } | ||||
8260 | } | ||||
8261 | } | ||||
8262 | return 0; | ||||
8263 | } | ||||
8264 | |||||
8265 | /// GetSelectFoldableOperands - We want to turn code that looks like this: | ||||
8266 | /// %C = or %A, %B | ||||
8267 | /// %D = select %cond, %C, %A | ||||
8268 | /// into: | ||||
8269 | /// %C = select %cond, %B, 0 | ||||
8270 | /// %D = or %A, %C | ||||
8271 | /// | ||||
8272 | /// Assuming that the specified instruction is an operand to the select, return | ||||
8273 | /// a bitmask indicating which operands of this instruction are foldable if they | ||||
8274 | /// equal the other incoming value of the select. | ||||
8275 | /// | ||||
8276 | static unsigned GetSelectFoldableOperands(Instruction *I) { | ||||
8277 | switch (I->getOpcode()) { | ||||
8278 | case Instruction::Add: | ||||
8279 | case Instruction::Mul: | ||||
8280 | case Instruction::And: | ||||
8281 | case Instruction::Or: | ||||
8282 | case Instruction::Xor: | ||||
8283 | return 3; // Can fold through either operand. | ||||
8284 | case Instruction::Sub: // Can only fold on the amount subtracted. | ||||
8285 | case Instruction::Shl: // Can only fold on the shift amount. | ||||
8286 | case Instruction::LShr: | ||||
8287 | case Instruction::AShr: | ||||
8288 | return 1; | ||||
8289 | default: | ||||
8290 | return 0; // Cannot fold | ||||
8291 | } | ||||
8292 | } | ||||
8293 | |||||
8294 | /// GetSelectFoldableConstant - For the same transformation as the previous | ||||
8295 | /// function, return the identity constant that goes into the select. | ||||
8296 | static Constant *GetSelectFoldableConstant(Instruction *I) { | ||||
8297 | switch (I->getOpcode()) { | ||||
8298 | default: assert(0 && "This cannot happen!"); abort(); | ||||
8299 | case Instruction::Add: | ||||
8300 | case Instruction::Sub: | ||||
8301 | case Instruction::Or: | ||||
8302 | case Instruction::Xor: | ||||
8303 | case Instruction::Shl: | ||||
8304 | case Instruction::LShr: | ||||
8305 | case Instruction::AShr: | ||||
8306 | return Constant::getNullValue(I->getType()); | ||||
8307 | case Instruction::And: | ||||
8308 | return Constant::getAllOnesValue(I->getType()); | ||||
8309 | case Instruction::Mul: | ||||
8310 | return ConstantInt::get(I->getType(), 1); | ||||
8311 | } | ||||
8312 | } | ||||
8313 | |||||
8314 | /// FoldSelectOpOp - Here we have (select c, TI, FI), and we know that TI and FI | ||||
8315 | /// have the same opcode and only one use each. Try to simplify this. | ||||
8316 | Instruction *InstCombiner::FoldSelectOpOp(SelectInst &SI, Instruction *TI, | ||||
8317 | Instruction *FI) { | ||||
8318 | if (TI->getNumOperands() == 1) { | ||||
8319 | // If this is a non-volatile load or a cast from the same type, | ||||
8320 | // merge. | ||||
8321 | if (TI->isCast()) { | ||||
8322 | if (TI->getOperand(0)->getType() != FI->getOperand(0)->getType()) | ||||
8323 | return 0; | ||||
8324 | } else { | ||||
8325 | return 0; // unknown unary op. | ||||
8326 | } | ||||
8327 | |||||
8328 | // Fold this by inserting a select from the input values. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8329 | SelectInst *NewSI = SelectInst::Create(SI.getCondition(), TI->getOperand(0), |
8330 | FI->getOperand(0), SI.getName()+".v"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8331 | InsertNewInstBefore(NewSI, SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8332 | return CastInst::Create(Instruction::CastOps(TI->getOpcode()), NewSI, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8333 | TI->getType()); |
8334 | } | ||||
8335 | |||||
8336 | // Only handle binary operators here. | ||||
8337 | if (!isa<BinaryOperator>(TI)) | ||||
8338 | return 0; | ||||
8339 | |||||
8340 | // Figure out if the operations have any operands in common. | ||||
8341 | Value *MatchOp, *OtherOpT, *OtherOpF; | ||||
8342 | bool MatchIsOpZero; | ||||
8343 | if (TI->getOperand(0) == FI->getOperand(0)) { | ||||
8344 | MatchOp = TI->getOperand(0); | ||||
8345 | OtherOpT = TI->getOperand(1); | ||||
8346 | OtherOpF = FI->getOperand(1); | ||||
8347 | MatchIsOpZero = true; | ||||
8348 | } else if (TI->getOperand(1) == FI->getOperand(1)) { | ||||
8349 | MatchOp = TI->getOperand(1); | ||||
8350 | OtherOpT = TI->getOperand(0); | ||||
8351 | OtherOpF = FI->getOperand(0); | ||||
8352 | MatchIsOpZero = false; | ||||
8353 | } else if (!TI->isCommutative()) { | ||||
8354 | return 0; | ||||
8355 | } else if (TI->getOperand(0) == FI->getOperand(1)) { | ||||
8356 | MatchOp = TI->getOperand(0); | ||||
8357 | OtherOpT = TI->getOperand(1); | ||||
8358 | OtherOpF = FI->getOperand(0); | ||||
8359 | MatchIsOpZero = true; | ||||
8360 | } else if (TI->getOperand(1) == FI->getOperand(0)) { | ||||
8361 | MatchOp = TI->getOperand(1); | ||||
8362 | OtherOpT = TI->getOperand(0); | ||||
8363 | OtherOpF = FI->getOperand(1); | ||||
8364 | MatchIsOpZero = true; | ||||
8365 | } else { | ||||
8366 | return 0; | ||||
8367 | } | ||||
8368 | |||||
8369 | // If we reach here, they do have operations in common. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8370 | SelectInst *NewSI = SelectInst::Create(SI.getCondition(), OtherOpT, |
8371 | OtherOpF, SI.getName()+".v"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8372 | InsertNewInstBefore(NewSI, SI); |
8373 | |||||
8374 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(TI)) { | ||||
8375 | if (MatchIsOpZero) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8376 | return BinaryOperator::Create(BO->getOpcode(), MatchOp, NewSI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8377 | else |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8378 | return BinaryOperator::Create(BO->getOpcode(), NewSI, MatchOp); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8379 | } |
8380 | assert(0 && "Shouldn't get here"); | ||||
8381 | return 0; | ||||
8382 | } | ||||
8383 | |||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 8384 | /// visitSelectInstWithICmp - Visit a SelectInst that has an |
8385 | /// ICmpInst as its first operand. | ||||
8386 | /// | ||||
8387 | Instruction *InstCombiner::visitSelectInstWithICmp(SelectInst &SI, | ||||
8388 | ICmpInst *ICI) { | ||||
8389 | bool Changed = false; | ||||
8390 | ICmpInst::Predicate Pred = ICI->getPredicate(); | ||||
8391 | Value *CmpLHS = ICI->getOperand(0); | ||||
8392 | Value *CmpRHS = ICI->getOperand(1); | ||||
8393 | Value *TrueVal = SI.getTrueValue(); | ||||
8394 | Value *FalseVal = SI.getFalseValue(); | ||||
8395 | |||||
8396 | // Check cases where the comparison is with a constant that | ||||
8397 | // can be adjusted to fit the min/max idiom. We may edit ICI in | ||||
8398 | // place here, so make sure the select is the only user. | ||||
8399 | if (ICI->hasOneUse()) | ||||
8400 | if (ConstantInt *CI = dyn_cast<ConstantInt>(CmpRHS)) | ||||
8401 | switch (Pred) { | ||||
8402 | default: break; | ||||
8403 | case ICmpInst::ICMP_ULT: | ||||
8404 | case ICmpInst::ICMP_SLT: { | ||||
8405 | // X < MIN ? T : F --> F | ||||
8406 | if (CI->isMinValue(Pred == ICmpInst::ICMP_SLT)) | ||||
8407 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8408 | // X < C ? X : C-1 --> X > C-1 ? C-1 : X | ||||
8409 | Constant *AdjustedRHS = SubOne(CI); | ||||
8410 | if ((CmpLHS == TrueVal && AdjustedRHS == FalseVal) || | ||||
8411 | (CmpLHS == FalseVal && AdjustedRHS == TrueVal)) { | ||||
8412 | Pred = ICmpInst::getSwappedPredicate(Pred); | ||||
8413 | CmpRHS = AdjustedRHS; | ||||
8414 | std::swap(FalseVal, TrueVal); | ||||
8415 | ICI->setPredicate(Pred); | ||||
8416 | ICI->setOperand(1, CmpRHS); | ||||
8417 | SI.setOperand(1, TrueVal); | ||||
8418 | SI.setOperand(2, FalseVal); | ||||
8419 | Changed = true; | ||||
8420 | } | ||||
8421 | break; | ||||
8422 | } | ||||
8423 | case ICmpInst::ICMP_UGT: | ||||
8424 | case ICmpInst::ICMP_SGT: { | ||||
8425 | // X > MAX ? T : F --> F | ||||
8426 | if (CI->isMaxValue(Pred == ICmpInst::ICMP_SGT)) | ||||
8427 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8428 | // X > C ? X : C+1 --> X < C+1 ? C+1 : X | ||||
8429 | Constant *AdjustedRHS = AddOne(CI); | ||||
8430 | if ((CmpLHS == TrueVal && AdjustedRHS == FalseVal) || | ||||
8431 | (CmpLHS == FalseVal && AdjustedRHS == TrueVal)) { | ||||
8432 | Pred = ICmpInst::getSwappedPredicate(Pred); | ||||
8433 | CmpRHS = AdjustedRHS; | ||||
8434 | std::swap(FalseVal, TrueVal); | ||||
8435 | ICI->setPredicate(Pred); | ||||
8436 | ICI->setOperand(1, CmpRHS); | ||||
8437 | SI.setOperand(1, TrueVal); | ||||
8438 | SI.setOperand(2, FalseVal); | ||||
8439 | Changed = true; | ||||
8440 | } | ||||
8441 | break; | ||||
8442 | } | ||||
8443 | } | ||||
8444 | |||||
8445 | if (CmpLHS == TrueVal && CmpRHS == FalseVal) { | ||||
8446 | // Transform (X == Y) ? X : Y -> Y | ||||
8447 | if (Pred == ICmpInst::ICMP_EQ) | ||||
8448 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8449 | // Transform (X != Y) ? X : Y -> X | ||||
8450 | if (Pred == ICmpInst::ICMP_NE) | ||||
8451 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8452 | /// NOTE: if we wanted to, this is where to detect integer MIN/MAX | ||||
8453 | |||||
8454 | } else if (CmpLHS == FalseVal && CmpRHS == TrueVal) { | ||||
8455 | // Transform (X == Y) ? Y : X -> X | ||||
8456 | if (Pred == ICmpInst::ICMP_EQ) | ||||
8457 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8458 | // Transform (X != Y) ? Y : X -> Y | ||||
8459 | if (Pred == ICmpInst::ICMP_NE) | ||||
8460 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8461 | /// NOTE: if we wanted to, this is where to detect integer MIN/MAX | ||||
8462 | } | ||||
8463 | |||||
8464 | /// NOTE: if we wanted to, this is where to detect integer ABS | ||||
8465 | |||||
8466 | return Changed ? &SI : 0; | ||||
8467 | } | ||||
8468 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8469 | Instruction *InstCombiner::visitSelectInst(SelectInst &SI) { |
8470 | Value *CondVal = SI.getCondition(); | ||||
8471 | Value *TrueVal = SI.getTrueValue(); | ||||
8472 | Value *FalseVal = SI.getFalseValue(); | ||||
8473 | |||||
8474 | // select true, X, Y -> X | ||||
8475 | // select false, X, Y -> Y | ||||
8476 | if (ConstantInt *C = dyn_cast<ConstantInt>(CondVal)) | ||||
8477 | return ReplaceInstUsesWith(SI, C->getZExtValue() ? TrueVal : FalseVal); | ||||
8478 | |||||
8479 | // select C, X, X -> X | ||||
8480 | if (TrueVal == FalseVal) | ||||
8481 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8482 | |||||
8483 | if (isa<UndefValue>(TrueVal)) // select C, undef, X -> X | ||||
8484 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8485 | if (isa<UndefValue>(FalseVal)) // select C, X, undef -> X | ||||
8486 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8487 | if (isa<UndefValue>(CondVal)) { // select undef, X, Y -> X or Y | ||||
8488 | if (isa<Constant>(TrueVal)) | ||||
8489 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8490 | else | ||||
8491 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8492 | } | ||||
8493 | |||||
8494 | if (SI.getType() == Type::Int1Ty) { | ||||
8495 | if (ConstantInt *C = dyn_cast<ConstantInt>(TrueVal)) { | ||||
8496 | if (C->getZExtValue()) { | ||||
8497 | // Change: A = select B, true, C --> A = or B, C | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8498 | return BinaryOperator::CreateOr(CondVal, FalseVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8499 | } else { |
8500 | // Change: A = select B, false, C --> A = and !B, C | ||||
8501 | Value *NotCond = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8502 | InsertNewInstBefore(BinaryOperator::CreateNot(CondVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8503 | "not."+CondVal->getName()), SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8504 | return BinaryOperator::CreateAnd(NotCond, FalseVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8505 | } |
8506 | } else if (ConstantInt *C = dyn_cast<ConstantInt>(FalseVal)) { | ||||
8507 | if (C->getZExtValue() == false) { | ||||
8508 | // Change: A = select B, C, false --> A = and B, C | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8509 | return BinaryOperator::CreateAnd(CondVal, TrueVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8510 | } else { |
8511 | // Change: A = select B, C, true --> A = or !B, C | ||||
8512 | Value *NotCond = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8513 | InsertNewInstBefore(BinaryOperator::CreateNot(CondVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8514 | "not."+CondVal->getName()), SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8515 | return BinaryOperator::CreateOr(NotCond, TrueVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8516 | } |
8517 | } | ||||
Chris Lattner | 53f85a7 | 2007-11-25 21:27:53 +0000 | [diff] [blame] | 8518 | |
8519 | // select a, b, a -> a&b | ||||
8520 | // select a, a, b -> a|b | ||||
8521 | if (CondVal == TrueVal) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8522 | return BinaryOperator::CreateOr(CondVal, FalseVal); |
Chris Lattner | 53f85a7 | 2007-11-25 21:27:53 +0000 | [diff] [blame] | 8523 | else if (CondVal == FalseVal) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8524 | return BinaryOperator::CreateAnd(CondVal, TrueVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8525 | } |
8526 | |||||
8527 | // Selecting between two integer constants? | ||||
8528 | if (ConstantInt *TrueValC = dyn_cast<ConstantInt>(TrueVal)) | ||||
8529 | if (ConstantInt *FalseValC = dyn_cast<ConstantInt>(FalseVal)) { | ||||
8530 | // select C, 1, 0 -> zext C to int | ||||
8531 | if (FalseValC->isZero() && TrueValC->getValue() == 1) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8532 | return CastInst::Create(Instruction::ZExt, CondVal, SI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8533 | } else if (TrueValC->isZero() && FalseValC->getValue() == 1) { |
8534 | // select C, 0, 1 -> zext !C to int | ||||
8535 | Value *NotCond = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8536 | InsertNewInstBefore(BinaryOperator::CreateNot(CondVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8537 | "not."+CondVal->getName()), SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8538 | return CastInst::Create(Instruction::ZExt, NotCond, SI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8539 | } |
8540 | |||||
8541 | // FIXME: Turn select 0/-1 and -1/0 into sext from condition! | ||||
8542 | |||||
8543 | if (ICmpInst *IC = dyn_cast<ICmpInst>(SI.getCondition())) { | ||||
8544 | |||||
8545 | // (x <s 0) ? -1 : 0 -> ashr x, 31 | ||||
8546 | if (TrueValC->isAllOnesValue() && FalseValC->isZero()) | ||||
8547 | if (ConstantInt *CmpCst = dyn_cast<ConstantInt>(IC->getOperand(1))) { | ||||
8548 | if (IC->getPredicate() == ICmpInst::ICMP_SLT && CmpCst->isZero()) { | ||||
8549 | // The comparison constant and the result are not neccessarily the | ||||
8550 | // same width. Make an all-ones value by inserting a AShr. | ||||
8551 | Value *X = IC->getOperand(0); | ||||
8552 | uint32_t Bits = X->getType()->getPrimitiveSizeInBits(); | ||||
8553 | Constant *ShAmt = ConstantInt::get(X->getType(), Bits-1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8554 | Instruction *SRA = BinaryOperator::Create(Instruction::AShr, X, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8555 | ShAmt, "ones"); |
8556 | InsertNewInstBefore(SRA, SI); | ||||
8557 | |||||
8558 | // Finally, convert to the type of the select RHS. We figure out | ||||
8559 | // if this requires a SExt, Trunc or BitCast based on the sizes. | ||||
8560 | Instruction::CastOps opc = Instruction::BitCast; | ||||
8561 | uint32_t SRASize = SRA->getType()->getPrimitiveSizeInBits(); | ||||
8562 | uint32_t SISize = SI.getType()->getPrimitiveSizeInBits(); | ||||
8563 | if (SRASize < SISize) | ||||
8564 | opc = Instruction::SExt; | ||||
8565 | else if (SRASize > SISize) | ||||
8566 | opc = Instruction::Trunc; | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8567 | return CastInst::Create(opc, SRA, SI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8568 | } |
8569 | } | ||||
8570 | |||||
8571 | |||||
8572 | // If one of the constants is zero (we know they can't both be) and we | ||||
8573 | // have an icmp instruction with zero, and we have an 'and' with the | ||||
8574 | // non-constant value, eliminate this whole mess. This corresponds to | ||||
8575 | // cases like this: ((X & 27) ? 27 : 0) | ||||
8576 | if (TrueValC->isZero() || FalseValC->isZero()) | ||||
8577 | if (IC->isEquality() && isa<ConstantInt>(IC->getOperand(1)) && | ||||
8578 | cast<Constant>(IC->getOperand(1))->isNullValue()) | ||||
8579 | if (Instruction *ICA = dyn_cast<Instruction>(IC->getOperand(0))) | ||||
8580 | if (ICA->getOpcode() == Instruction::And && | ||||
8581 | isa<ConstantInt>(ICA->getOperand(1)) && | ||||
8582 | (ICA->getOperand(1) == TrueValC || | ||||
8583 | ICA->getOperand(1) == FalseValC) && | ||||
8584 | isOneBitSet(cast<ConstantInt>(ICA->getOperand(1)))) { | ||||
8585 | // Okay, now we know that everything is set up, we just don't | ||||
8586 | // know whether we have a icmp_ne or icmp_eq and whether the | ||||
8587 | // true or false val is the zero. | ||||
8588 | bool ShouldNotVal = !TrueValC->isZero(); | ||||
8589 | ShouldNotVal ^= IC->getPredicate() == ICmpInst::ICMP_NE; | ||||
8590 | Value *V = ICA; | ||||
8591 | if (ShouldNotVal) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8592 | V = InsertNewInstBefore(BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8593 | Instruction::Xor, V, ICA->getOperand(1)), SI); |
8594 | return ReplaceInstUsesWith(SI, V); | ||||
8595 | } | ||||
8596 | } | ||||
8597 | } | ||||
8598 | |||||
8599 | // See if we are selecting two values based on a comparison of the two values. | ||||
8600 | if (FCmpInst *FCI = dyn_cast<FCmpInst>(CondVal)) { | ||||
8601 | if (FCI->getOperand(0) == TrueVal && FCI->getOperand(1) == FalseVal) { | ||||
8602 | // Transform (X == Y) ? X : Y -> Y | ||||
Dale Johannesen | 2e1b769 | 2007-10-03 17:45:27 +0000 | [diff] [blame] | 8603 | if (FCI->getPredicate() == FCmpInst::FCMP_OEQ) { |
8604 | // This is not safe in general for floating point: | ||||
8605 | // consider X== -0, Y== +0. | ||||
8606 | // It becomes safe if either operand is a nonzero constant. | ||||
8607 | ConstantFP *CFPt, *CFPf; | ||||
8608 | if (((CFPt = dyn_cast<ConstantFP>(TrueVal)) && | ||||
8609 | !CFPt->getValueAPF().isZero()) || | ||||
8610 | ((CFPf = dyn_cast<ConstantFP>(FalseVal)) && | ||||
8611 | !CFPf->getValueAPF().isZero())) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8612 | return ReplaceInstUsesWith(SI, FalseVal); |
Dale Johannesen | 2e1b769 | 2007-10-03 17:45:27 +0000 | [diff] [blame] | 8613 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8614 | // Transform (X != Y) ? X : Y -> X |
8615 | if (FCI->getPredicate() == FCmpInst::FCMP_ONE) | ||||
8616 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 8617 | // NOTE: if we wanted to, this is where to detect MIN/MAX |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8618 | |
8619 | } else if (FCI->getOperand(0) == FalseVal && FCI->getOperand(1) == TrueVal){ | ||||
8620 | // Transform (X == Y) ? Y : X -> X | ||||
Dale Johannesen | 2e1b769 | 2007-10-03 17:45:27 +0000 | [diff] [blame] | 8621 | if (FCI->getPredicate() == FCmpInst::FCMP_OEQ) { |
8622 | // This is not safe in general for floating point: | ||||
8623 | // consider X== -0, Y== +0. | ||||
8624 | // It becomes safe if either operand is a nonzero constant. | ||||
8625 | ConstantFP *CFPt, *CFPf; | ||||
8626 | if (((CFPt = dyn_cast<ConstantFP>(TrueVal)) && | ||||
8627 | !CFPt->getValueAPF().isZero()) || | ||||
8628 | ((CFPf = dyn_cast<ConstantFP>(FalseVal)) && | ||||
8629 | !CFPf->getValueAPF().isZero())) | ||||
8630 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8631 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8632 | // Transform (X != Y) ? Y : X -> Y |
8633 | if (FCI->getPredicate() == FCmpInst::FCMP_ONE) | ||||
8634 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 8635 | // NOTE: if we wanted to, this is where to detect MIN/MAX |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8636 | } |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 8637 | // NOTE: if we wanted to, this is where to detect ABS |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8638 | } |
8639 | |||||
8640 | // 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] | 8641 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(CondVal)) |
8642 | if (Instruction *Result = visitSelectInstWithICmp(SI, ICI)) | ||||
8643 | return Result; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8644 | |
8645 | if (Instruction *TI = dyn_cast<Instruction>(TrueVal)) | ||||
8646 | if (Instruction *FI = dyn_cast<Instruction>(FalseVal)) | ||||
8647 | if (TI->hasOneUse() && FI->hasOneUse()) { | ||||
8648 | Instruction *AddOp = 0, *SubOp = 0; | ||||
8649 | |||||
8650 | // Turn (select C, (op X, Y), (op X, Z)) -> (op X, (select C, Y, Z)) | ||||
8651 | if (TI->getOpcode() == FI->getOpcode()) | ||||
8652 | if (Instruction *IV = FoldSelectOpOp(SI, TI, FI)) | ||||
8653 | return IV; | ||||
8654 | |||||
8655 | // Turn select C, (X+Y), (X-Y) --> (X+(select C, Y, (-Y))). This is | ||||
8656 | // even legal for FP. | ||||
8657 | if (TI->getOpcode() == Instruction::Sub && | ||||
8658 | FI->getOpcode() == Instruction::Add) { | ||||
8659 | AddOp = FI; SubOp = TI; | ||||
8660 | } else if (FI->getOpcode() == Instruction::Sub && | ||||
8661 | TI->getOpcode() == Instruction::Add) { | ||||
8662 | AddOp = TI; SubOp = FI; | ||||
8663 | } | ||||
8664 | |||||
8665 | if (AddOp) { | ||||
8666 | Value *OtherAddOp = 0; | ||||
8667 | if (SubOp->getOperand(0) == AddOp->getOperand(0)) { | ||||
8668 | OtherAddOp = AddOp->getOperand(1); | ||||
8669 | } else if (SubOp->getOperand(0) == AddOp->getOperand(1)) { | ||||
8670 | OtherAddOp = AddOp->getOperand(0); | ||||
8671 | } | ||||
8672 | |||||
8673 | if (OtherAddOp) { | ||||
8674 | // So at this point we know we have (Y -> OtherAddOp): | ||||
8675 | // select C, (add X, Y), (sub X, Z) | ||||
8676 | Value *NegVal; // Compute -Z | ||||
8677 | if (Constant *C = dyn_cast<Constant>(SubOp->getOperand(1))) { | ||||
8678 | NegVal = ConstantExpr::getNeg(C); | ||||
8679 | } else { | ||||
8680 | NegVal = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8681 | BinaryOperator::CreateNeg(SubOp->getOperand(1), "tmp"), SI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8682 | } |
8683 | |||||
8684 | Value *NewTrueOp = OtherAddOp; | ||||
8685 | Value *NewFalseOp = NegVal; | ||||
8686 | if (AddOp != TI) | ||||
8687 | std::swap(NewTrueOp, NewFalseOp); | ||||
8688 | Instruction *NewSel = | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 8689 | SelectInst::Create(CondVal, NewTrueOp, |
8690 | NewFalseOp, SI.getName() + ".p"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8691 | |
8692 | NewSel = InsertNewInstBefore(NewSel, SI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8693 | return BinaryOperator::CreateAdd(SubOp->getOperand(0), NewSel); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8694 | } |
8695 | } | ||||
8696 | } | ||||
8697 | |||||
8698 | // See if we can fold the select into one of our operands. | ||||
8699 | if (SI.getType()->isInteger()) { | ||||
8700 | // See the comment above GetSelectFoldableOperands for a description of the | ||||
8701 | // transformation we are doing here. | ||||
8702 | if (Instruction *TVI = dyn_cast<Instruction>(TrueVal)) | ||||
8703 | if (TVI->hasOneUse() && TVI->getNumOperands() == 2 && | ||||
8704 | !isa<Constant>(FalseVal)) | ||||
8705 | if (unsigned SFO = GetSelectFoldableOperands(TVI)) { | ||||
8706 | unsigned OpToFold = 0; | ||||
8707 | if ((SFO & 1) && FalseVal == TVI->getOperand(0)) { | ||||
8708 | OpToFold = 1; | ||||
8709 | } else if ((SFO & 2) && FalseVal == TVI->getOperand(1)) { | ||||
8710 | OpToFold = 2; | ||||
8711 | } | ||||
8712 | |||||
8713 | if (OpToFold) { | ||||
8714 | Constant *C = GetSelectFoldableConstant(TVI); | ||||
8715 | Instruction *NewSel = | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 8716 | SelectInst::Create(SI.getCondition(), |
8717 | TVI->getOperand(2-OpToFold), C); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8718 | InsertNewInstBefore(NewSel, SI); |
8719 | NewSel->takeName(TVI); | ||||
8720 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(TVI)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8721 | return BinaryOperator::Create(BO->getOpcode(), FalseVal, NewSel); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8722 | else { |
8723 | assert(0 && "Unknown instruction!!"); | ||||
8724 | } | ||||
8725 | } | ||||
8726 | } | ||||
8727 | |||||
8728 | if (Instruction *FVI = dyn_cast<Instruction>(FalseVal)) | ||||
8729 | if (FVI->hasOneUse() && FVI->getNumOperands() == 2 && | ||||
8730 | !isa<Constant>(TrueVal)) | ||||
8731 | if (unsigned SFO = GetSelectFoldableOperands(FVI)) { | ||||
8732 | unsigned OpToFold = 0; | ||||
8733 | if ((SFO & 1) && TrueVal == FVI->getOperand(0)) { | ||||
8734 | OpToFold = 1; | ||||
8735 | } else if ((SFO & 2) && TrueVal == FVI->getOperand(1)) { | ||||
8736 | OpToFold = 2; | ||||
8737 | } | ||||
8738 | |||||
8739 | if (OpToFold) { | ||||
8740 | Constant *C = GetSelectFoldableConstant(FVI); | ||||
8741 | Instruction *NewSel = | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 8742 | SelectInst::Create(SI.getCondition(), C, |
8743 | FVI->getOperand(2-OpToFold)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8744 | InsertNewInstBefore(NewSel, SI); |
8745 | NewSel->takeName(FVI); | ||||
8746 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(FVI)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8747 | return BinaryOperator::Create(BO->getOpcode(), TrueVal, NewSel); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8748 | else |
8749 | assert(0 && "Unknown instruction!!"); | ||||
8750 | } | ||||
8751 | } | ||||
8752 | } | ||||
8753 | |||||
8754 | if (BinaryOperator::isNot(CondVal)) { | ||||
8755 | SI.setOperand(0, BinaryOperator::getNotArgument(CondVal)); | ||||
8756 | SI.setOperand(1, FalseVal); | ||||
8757 | SI.setOperand(2, TrueVal); | ||||
8758 | return &SI; | ||||
8759 | } | ||||
8760 | |||||
8761 | return 0; | ||||
8762 | } | ||||
8763 | |||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 8764 | /// EnforceKnownAlignment - If the specified pointer points to an object that |
8765 | /// we control, modify the object's alignment to PrefAlign. This isn't | ||||
8766 | /// often possible though. If alignment is important, a more reliable approach | ||||
8767 | /// is to simply align all global variables and allocation instructions to | ||||
8768 | /// their preferred alignment from the beginning. | ||||
8769 | /// | ||||
8770 | static unsigned EnforceKnownAlignment(Value *V, | ||||
8771 | unsigned Align, unsigned PrefAlign) { | ||||
Chris Lattner | 47cf345 | 2007-08-09 19:05:49 +0000 | [diff] [blame] | 8772 | |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 8773 | User *U = dyn_cast<User>(V); |
8774 | if (!U) return Align; | ||||
8775 | |||||
8776 | switch (getOpcode(U)) { | ||||
8777 | default: break; | ||||
8778 | case Instruction::BitCast: | ||||
8779 | return EnforceKnownAlignment(U->getOperand(0), Align, PrefAlign); | ||||
8780 | case Instruction::GetElementPtr: { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8781 | // If all indexes are zero, it is just the alignment of the base pointer. |
8782 | bool AllZeroOperands = true; | ||||
Gabor Greif | e92fbe2 | 2008-06-12 21:51:29 +0000 | [diff] [blame] | 8783 | 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] | 8784 | if (!isa<Constant>(*i) || |
8785 | !cast<Constant>(*i)->isNullValue()) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8786 | AllZeroOperands = false; |
8787 | break; | ||||
8788 | } | ||||
Chris Lattner | 47cf345 | 2007-08-09 19:05:49 +0000 | [diff] [blame] | 8789 | |
8790 | if (AllZeroOperands) { | ||||
8791 | // Treat this like a bitcast. | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 8792 | return EnforceKnownAlignment(U->getOperand(0), Align, PrefAlign); |
Chris Lattner | 47cf345 | 2007-08-09 19:05:49 +0000 | [diff] [blame] | 8793 | } |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 8794 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8795 | } |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 8796 | } |
8797 | |||||
8798 | if (GlobalValue *GV = dyn_cast<GlobalValue>(V)) { | ||||
8799 | // If there is a large requested alignment and we can, bump up the alignment | ||||
8800 | // of the global. | ||||
8801 | if (!GV->isDeclaration()) { | ||||
8802 | GV->setAlignment(PrefAlign); | ||||
8803 | Align = PrefAlign; | ||||
8804 | } | ||||
8805 | } else if (AllocationInst *AI = dyn_cast<AllocationInst>(V)) { | ||||
8806 | // If there is a requested alignment and if this is an alloca, round up. We | ||||
8807 | // don't do this for malloc, because some systems can't respect the request. | ||||
8808 | if (isa<AllocaInst>(AI)) { | ||||
8809 | AI->setAlignment(PrefAlign); | ||||
8810 | Align = PrefAlign; | ||||
8811 | } | ||||
8812 | } | ||||
8813 | |||||
8814 | return Align; | ||||
8815 | } | ||||
8816 | |||||
8817 | /// GetOrEnforceKnownAlignment - If the specified pointer has an alignment that | ||||
8818 | /// we can determine, return it, otherwise return 0. If PrefAlign is specified, | ||||
8819 | /// and it is more than the alignment of the ultimate object, see if we can | ||||
8820 | /// increase the alignment of the ultimate object, making this check succeed. | ||||
8821 | unsigned InstCombiner::GetOrEnforceKnownAlignment(Value *V, | ||||
8822 | unsigned PrefAlign) { | ||||
8823 | unsigned BitWidth = TD ? TD->getTypeSizeInBits(V->getType()) : | ||||
8824 | sizeof(PrefAlign) * CHAR_BIT; | ||||
8825 | APInt Mask = APInt::getAllOnesValue(BitWidth); | ||||
8826 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
8827 | ComputeMaskedBits(V, Mask, KnownZero, KnownOne); | ||||
8828 | unsigned TrailZ = KnownZero.countTrailingOnes(); | ||||
8829 | unsigned Align = 1u << std::min(BitWidth - 1, TrailZ); | ||||
8830 | |||||
8831 | if (PrefAlign > Align) | ||||
8832 | Align = EnforceKnownAlignment(V, Align, PrefAlign); | ||||
8833 | |||||
8834 | // We don't need to make any adjustment. | ||||
8835 | return Align; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8836 | } |
8837 | |||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 8838 | Instruction *InstCombiner::SimplifyMemTransfer(MemIntrinsic *MI) { |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 8839 | unsigned DstAlign = GetOrEnforceKnownAlignment(MI->getOperand(1)); |
8840 | unsigned SrcAlign = GetOrEnforceKnownAlignment(MI->getOperand(2)); | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 8841 | unsigned MinAlign = std::min(DstAlign, SrcAlign); |
8842 | unsigned CopyAlign = MI->getAlignment()->getZExtValue(); | ||||
8843 | |||||
8844 | if (CopyAlign < MinAlign) { | ||||
8845 | MI->setAlignment(ConstantInt::get(Type::Int32Ty, MinAlign)); | ||||
8846 | return MI; | ||||
8847 | } | ||||
8848 | |||||
8849 | // If MemCpyInst length is 1/2/4/8 bytes then replace memcpy with | ||||
8850 | // load/store. | ||||
8851 | ConstantInt *MemOpLength = dyn_cast<ConstantInt>(MI->getOperand(3)); | ||||
8852 | if (MemOpLength == 0) return 0; | ||||
8853 | |||||
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 8854 | // Source and destination pointer types are always "i8*" for intrinsic. See |
8855 | // if the size is something we can handle with a single primitive load/store. | ||||
8856 | // A single load+store correctly handles overlapping memory in the memmove | ||||
8857 | // case. | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 8858 | unsigned Size = MemOpLength->getZExtValue(); |
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 8859 | if (Size == 0) return MI; // Delete this mem transfer. |
8860 | |||||
8861 | if (Size > 8 || (Size&(Size-1))) | ||||
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 8862 | return 0; // If not 1/2/4/8 bytes, exit. |
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 8863 | |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 8864 | // Use an integer load+store unless we can find something better. |
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 8865 | Type *NewPtrTy = PointerType::getUnqual(IntegerType::get(Size<<3)); |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 8866 | |
8867 | // Memcpy forces the use of i8* for the source and destination. That means | ||||
8868 | // that if you're using memcpy to move one double around, you'll get a cast | ||||
8869 | // from double* to i8*. We'd much rather use a double load+store rather than | ||||
8870 | // an i64 load+store, here because this improves the odds that the source or | ||||
8871 | // dest address will be promotable. See if we can find a better type than the | ||||
8872 | // integer datatype. | ||||
8873 | if (Value *Op = getBitCastOperand(MI->getOperand(1))) { | ||||
8874 | const Type *SrcETy = cast<PointerType>(Op->getType())->getElementType(); | ||||
8875 | if (SrcETy->isSized() && TD->getTypeStoreSize(SrcETy) == Size) { | ||||
8876 | // The SrcETy might be something like {{{double}}} or [1 x double]. Rip | ||||
8877 | // down through these levels if so. | ||||
Dan Gohman | b8e94f6 | 2008-05-23 01:52:21 +0000 | [diff] [blame] | 8878 | while (!SrcETy->isSingleValueType()) { |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 8879 | if (const StructType *STy = dyn_cast<StructType>(SrcETy)) { |
8880 | if (STy->getNumElements() == 1) | ||||
8881 | SrcETy = STy->getElementType(0); | ||||
8882 | else | ||||
8883 | break; | ||||
8884 | } else if (const ArrayType *ATy = dyn_cast<ArrayType>(SrcETy)) { | ||||
8885 | if (ATy->getNumElements() == 1) | ||||
8886 | SrcETy = ATy->getElementType(); | ||||
8887 | else | ||||
8888 | break; | ||||
8889 | } else | ||||
8890 | break; | ||||
8891 | } | ||||
8892 | |||||
Dan Gohman | b8e94f6 | 2008-05-23 01:52:21 +0000 | [diff] [blame] | 8893 | if (SrcETy->isSingleValueType()) |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 8894 | NewPtrTy = PointerType::getUnqual(SrcETy); |
8895 | } | ||||
8896 | } | ||||
8897 | |||||
8898 | |||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 8899 | // If the memcpy/memmove provides better alignment info than we can |
8900 | // infer, use it. | ||||
8901 | SrcAlign = std::max(SrcAlign, CopyAlign); | ||||
8902 | DstAlign = std::max(DstAlign, CopyAlign); | ||||
8903 | |||||
8904 | Value *Src = InsertBitCastBefore(MI->getOperand(2), NewPtrTy, *MI); | ||||
8905 | Value *Dest = InsertBitCastBefore(MI->getOperand(1), NewPtrTy, *MI); | ||||
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 8906 | Instruction *L = new LoadInst(Src, "tmp", false, SrcAlign); |
8907 | InsertNewInstBefore(L, *MI); | ||||
8908 | InsertNewInstBefore(new StoreInst(L, Dest, false, DstAlign), *MI); | ||||
8909 | |||||
8910 | // Set the size of the copy to 0, it will be deleted on the next iteration. | ||||
8911 | MI->setOperand(3, Constant::getNullValue(MemOpLength->getType())); | ||||
8912 | return MI; | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 8913 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8914 | |
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 8915 | Instruction *InstCombiner::SimplifyMemSet(MemSetInst *MI) { |
8916 | unsigned Alignment = GetOrEnforceKnownAlignment(MI->getDest()); | ||||
8917 | if (MI->getAlignment()->getZExtValue() < Alignment) { | ||||
8918 | MI->setAlignment(ConstantInt::get(Type::Int32Ty, Alignment)); | ||||
8919 | return MI; | ||||
8920 | } | ||||
8921 | |||||
8922 | // Extract the length and alignment and fill if they are constant. | ||||
8923 | ConstantInt *LenC = dyn_cast<ConstantInt>(MI->getLength()); | ||||
8924 | ConstantInt *FillC = dyn_cast<ConstantInt>(MI->getValue()); | ||||
8925 | if (!LenC || !FillC || FillC->getType() != Type::Int8Ty) | ||||
8926 | return 0; | ||||
8927 | uint64_t Len = LenC->getZExtValue(); | ||||
8928 | Alignment = MI->getAlignment()->getZExtValue(); | ||||
8929 | |||||
8930 | // If the length is zero, this is a no-op | ||||
8931 | if (Len == 0) return MI; // memset(d,c,0,a) -> noop | ||||
8932 | |||||
8933 | // memset(s,c,n) -> store s, c (for n=1,2,4,8) | ||||
8934 | if (Len <= 8 && isPowerOf2_32((uint32_t)Len)) { | ||||
8935 | const Type *ITy = IntegerType::get(Len*8); // n=1 -> i8. | ||||
8936 | |||||
8937 | Value *Dest = MI->getDest(); | ||||
8938 | Dest = InsertBitCastBefore(Dest, PointerType::getUnqual(ITy), *MI); | ||||
8939 | |||||
8940 | // Alignment 0 is identity for alignment 1 for memset, but not store. | ||||
8941 | if (Alignment == 0) Alignment = 1; | ||||
8942 | |||||
8943 | // Extract the fill value and store. | ||||
8944 | uint64_t Fill = FillC->getZExtValue()*0x0101010101010101ULL; | ||||
8945 | InsertNewInstBefore(new StoreInst(ConstantInt::get(ITy, Fill), Dest, false, | ||||
8946 | Alignment), *MI); | ||||
8947 | |||||
8948 | // Set the size of the copy to 0, it will be deleted on the next iteration. | ||||
8949 | MI->setLength(Constant::getNullValue(LenC->getType())); | ||||
8950 | return MI; | ||||
8951 | } | ||||
8952 | |||||
8953 | return 0; | ||||
8954 | } | ||||
8955 | |||||
8956 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8957 | /// visitCallInst - CallInst simplification. This mostly only handles folding |
8958 | /// of intrinsic instructions. For normal calls, it allows visitCallSite to do | ||||
8959 | /// the heavy lifting. | ||||
8960 | /// | ||||
8961 | Instruction *InstCombiner::visitCallInst(CallInst &CI) { | ||||
8962 | IntrinsicInst *II = dyn_cast<IntrinsicInst>(&CI); | ||||
8963 | if (!II) return visitCallSite(&CI); | ||||
8964 | |||||
8965 | // Intrinsics cannot occur in an invoke, so handle them here instead of in | ||||
8966 | // visitCallSite. | ||||
8967 | if (MemIntrinsic *MI = dyn_cast<MemIntrinsic>(II)) { | ||||
8968 | bool Changed = false; | ||||
8969 | |||||
8970 | // memmove/cpy/set of zero bytes is a noop. | ||||
8971 | if (Constant *NumBytes = dyn_cast<Constant>(MI->getLength())) { | ||||
8972 | if (NumBytes->isNullValue()) return EraseInstFromFunction(CI); | ||||
8973 | |||||
8974 | if (ConstantInt *CI = dyn_cast<ConstantInt>(NumBytes)) | ||||
8975 | if (CI->getZExtValue() == 1) { | ||||
8976 | // Replace the instruction with just byte operations. We would | ||||
8977 | // transform other cases to loads/stores, but we don't know if | ||||
8978 | // alignment is sufficient. | ||||
8979 | } | ||||
8980 | } | ||||
8981 | |||||
8982 | // If we have a memmove and the source operation is a constant global, | ||||
8983 | // then the source and dest pointers can't alias, so we can change this | ||||
8984 | // into a call to memcpy. | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 8985 | if (MemMoveInst *MMI = dyn_cast<MemMoveInst>(MI)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8986 | if (GlobalVariable *GVSrc = dyn_cast<GlobalVariable>(MMI->getSource())) |
8987 | if (GVSrc->isConstant()) { | ||||
8988 | Module *M = CI.getParent()->getParent()->getParent(); | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 8989 | Intrinsic::ID MemCpyID; |
8990 | if (CI.getOperand(3)->getType() == Type::Int32Ty) | ||||
8991 | MemCpyID = Intrinsic::memcpy_i32; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8992 | else |
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 8993 | MemCpyID = Intrinsic::memcpy_i64; |
8994 | CI.setOperand(0, Intrinsic::getDeclaration(M, MemCpyID)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8995 | Changed = true; |
8996 | } | ||||
Chris Lattner | 59b27d9 | 2008-05-28 05:30:41 +0000 | [diff] [blame] | 8997 | |
8998 | // memmove(x,x,size) -> noop. | ||||
8999 | if (MMI->getSource() == MMI->getDest()) | ||||
9000 | return EraseInstFromFunction(CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9001 | } |
9002 | |||||
9003 | // If we can determine a pointer alignment that is bigger than currently | ||||
9004 | // set, update the alignment. | ||||
9005 | if (isa<MemCpyInst>(MI) || isa<MemMoveInst>(MI)) { | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9006 | if (Instruction *I = SimplifyMemTransfer(MI)) |
9007 | return I; | ||||
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 9008 | } else if (MemSetInst *MSI = dyn_cast<MemSetInst>(MI)) { |
9009 | if (Instruction *I = SimplifyMemSet(MSI)) | ||||
9010 | return I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9011 | } |
9012 | |||||
9013 | if (Changed) return II; | ||||
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9014 | } |
9015 | |||||
9016 | switch (II->getIntrinsicID()) { | ||||
9017 | default: break; | ||||
9018 | case Intrinsic::bswap: | ||||
9019 | // bswap(bswap(x)) -> x | ||||
9020 | if (IntrinsicInst *Operand = dyn_cast<IntrinsicInst>(II->getOperand(1))) | ||||
9021 | if (Operand->getIntrinsicID() == Intrinsic::bswap) | ||||
9022 | return ReplaceInstUsesWith(CI, Operand->getOperand(1)); | ||||
9023 | break; | ||||
9024 | case Intrinsic::ppc_altivec_lvx: | ||||
9025 | case Intrinsic::ppc_altivec_lvxl: | ||||
9026 | case Intrinsic::x86_sse_loadu_ps: | ||||
9027 | case Intrinsic::x86_sse2_loadu_pd: | ||||
9028 | case Intrinsic::x86_sse2_loadu_dq: | ||||
9029 | // Turn PPC lvx -> load if the pointer is known aligned. | ||||
9030 | // Turn X86 loadups -> load if the pointer is known aligned. | ||||
9031 | if (GetOrEnforceKnownAlignment(II->getOperand(1), 16) >= 16) { | ||||
9032 | Value *Ptr = InsertBitCastBefore(II->getOperand(1), | ||||
9033 | PointerType::getUnqual(II->getType()), | ||||
9034 | CI); | ||||
9035 | return new LoadInst(Ptr); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9036 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9037 | break; |
9038 | case Intrinsic::ppc_altivec_stvx: | ||||
9039 | case Intrinsic::ppc_altivec_stvxl: | ||||
9040 | // Turn stvx -> store if the pointer is known aligned. | ||||
9041 | if (GetOrEnforceKnownAlignment(II->getOperand(2), 16) >= 16) { | ||||
9042 | const Type *OpPtrTy = | ||||
9043 | PointerType::getUnqual(II->getOperand(1)->getType()); | ||||
9044 | Value *Ptr = InsertBitCastBefore(II->getOperand(2), OpPtrTy, CI); | ||||
9045 | return new StoreInst(II->getOperand(1), Ptr); | ||||
9046 | } | ||||
9047 | break; | ||||
9048 | case Intrinsic::x86_sse_storeu_ps: | ||||
9049 | case Intrinsic::x86_sse2_storeu_pd: | ||||
9050 | case Intrinsic::x86_sse2_storeu_dq: | ||||
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9051 | // Turn X86 storeu -> store if the pointer is known aligned. |
9052 | if (GetOrEnforceKnownAlignment(II->getOperand(1), 16) >= 16) { | ||||
9053 | const Type *OpPtrTy = | ||||
9054 | PointerType::getUnqual(II->getOperand(2)->getType()); | ||||
9055 | Value *Ptr = InsertBitCastBefore(II->getOperand(1), OpPtrTy, CI); | ||||
9056 | return new StoreInst(II->getOperand(2), Ptr); | ||||
9057 | } | ||||
9058 | break; | ||||
9059 | |||||
9060 | case Intrinsic::x86_sse_cvttss2si: { | ||||
9061 | // These intrinsics only demands the 0th element of its input vector. If | ||||
9062 | // we can simplify the input based on that, do so now. | ||||
9063 | uint64_t UndefElts; | ||||
9064 | if (Value *V = SimplifyDemandedVectorElts(II->getOperand(1), 1, | ||||
9065 | UndefElts)) { | ||||
9066 | II->setOperand(1, V); | ||||
9067 | return II; | ||||
9068 | } | ||||
9069 | break; | ||||
9070 | } | ||||
9071 | |||||
9072 | case Intrinsic::ppc_altivec_vperm: | ||||
9073 | // Turn vperm(V1,V2,mask) -> shuffle(V1,V2,mask) if mask is a constant. | ||||
9074 | if (ConstantVector *Mask = dyn_cast<ConstantVector>(II->getOperand(3))) { | ||||
9075 | assert(Mask->getNumOperands() == 16 && "Bad type for intrinsic!"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9076 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9077 | // Check that all of the elements are integer constants or undefs. |
9078 | bool AllEltsOk = true; | ||||
9079 | for (unsigned i = 0; i != 16; ++i) { | ||||
9080 | if (!isa<ConstantInt>(Mask->getOperand(i)) && | ||||
9081 | !isa<UndefValue>(Mask->getOperand(i))) { | ||||
9082 | AllEltsOk = false; | ||||
9083 | break; | ||||
9084 | } | ||||
9085 | } | ||||
9086 | |||||
9087 | if (AllEltsOk) { | ||||
9088 | // Cast the input vectors to byte vectors. | ||||
9089 | Value *Op0 =InsertBitCastBefore(II->getOperand(1),Mask->getType(),CI); | ||||
9090 | Value *Op1 =InsertBitCastBefore(II->getOperand(2),Mask->getType(),CI); | ||||
9091 | Value *Result = UndefValue::get(Op0->getType()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9092 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9093 | // Only extract each element once. |
9094 | Value *ExtractedElts[32]; | ||||
9095 | memset(ExtractedElts, 0, sizeof(ExtractedElts)); | ||||
9096 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9097 | for (unsigned i = 0; i != 16; ++i) { |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9098 | if (isa<UndefValue>(Mask->getOperand(i))) |
9099 | continue; | ||||
9100 | unsigned Idx=cast<ConstantInt>(Mask->getOperand(i))->getZExtValue(); | ||||
9101 | Idx &= 31; // Match the hardware behavior. | ||||
9102 | |||||
9103 | if (ExtractedElts[Idx] == 0) { | ||||
9104 | Instruction *Elt = | ||||
9105 | new ExtractElementInst(Idx < 16 ? Op0 : Op1, Idx&15, "tmp"); | ||||
9106 | InsertNewInstBefore(Elt, CI); | ||||
9107 | ExtractedElts[Idx] = Elt; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9108 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9109 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9110 | // Insert this value into the result vector. |
9111 | Result = InsertElementInst::Create(Result, ExtractedElts[Idx], | ||||
9112 | i, "tmp"); | ||||
9113 | InsertNewInstBefore(cast<Instruction>(Result), CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9114 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9115 | return CastInst::Create(Instruction::BitCast, Result, CI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9116 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9117 | } |
9118 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9119 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9120 | case Intrinsic::stackrestore: { |
9121 | // If the save is right next to the restore, remove the restore. This can | ||||
9122 | // happen when variable allocas are DCE'd. | ||||
9123 | if (IntrinsicInst *SS = dyn_cast<IntrinsicInst>(II->getOperand(1))) { | ||||
9124 | if (SS->getIntrinsicID() == Intrinsic::stacksave) { | ||||
9125 | BasicBlock::iterator BI = SS; | ||||
9126 | if (&*++BI == II) | ||||
9127 | return EraseInstFromFunction(CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9128 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9129 | } |
9130 | |||||
9131 | // Scan down this block to see if there is another stack restore in the | ||||
9132 | // same block without an intervening call/alloca. | ||||
9133 | BasicBlock::iterator BI = II; | ||||
9134 | TerminatorInst *TI = II->getParent()->getTerminator(); | ||||
9135 | bool CannotRemove = false; | ||||
9136 | for (++BI; &*BI != TI; ++BI) { | ||||
9137 | if (isa<AllocaInst>(BI)) { | ||||
9138 | CannotRemove = true; | ||||
9139 | break; | ||||
9140 | } | ||||
Chris Lattner | a6b477c | 2008-06-25 05:59:28 +0000 | [diff] [blame] | 9141 | if (CallInst *BCI = dyn_cast<CallInst>(BI)) { |
9142 | if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(BCI)) { | ||||
9143 | // If there is a stackrestore below this one, remove this one. | ||||
9144 | if (II->getIntrinsicID() == Intrinsic::stackrestore) | ||||
9145 | return EraseInstFromFunction(CI); | ||||
9146 | // Otherwise, ignore the intrinsic. | ||||
9147 | } else { | ||||
9148 | // If we found a non-intrinsic call, we can't remove the stack | ||||
9149 | // restore. | ||||
Chris Lattner | 416d91c | 2008-02-18 06:12:38 +0000 | [diff] [blame] | 9150 | CannotRemove = true; |
9151 | break; | ||||
9152 | } | ||||
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9153 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9154 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9155 | |
9156 | // If the stack restore is in a return/unwind block and if there are no | ||||
9157 | // allocas or calls between the restore and the return, nuke the restore. | ||||
9158 | if (!CannotRemove && (isa<ReturnInst>(TI) || isa<UnwindInst>(TI))) | ||||
9159 | return EraseInstFromFunction(CI); | ||||
9160 | break; | ||||
9161 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9162 | } |
9163 | |||||
9164 | return visitCallSite(II); | ||||
9165 | } | ||||
9166 | |||||
9167 | // InvokeInst simplification | ||||
9168 | // | ||||
9169 | Instruction *InstCombiner::visitInvokeInst(InvokeInst &II) { | ||||
9170 | return visitCallSite(&II); | ||||
9171 | } | ||||
9172 | |||||
Dale Johannesen | 9602183 | 2008-04-25 21:16:07 +0000 | [diff] [blame] | 9173 | /// isSafeToEliminateVarargsCast - If this cast does not affect the value |
9174 | /// 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] | 9175 | static bool isSafeToEliminateVarargsCast(const CallSite CS, |
9176 | const CastInst * const CI, | ||||
9177 | const TargetData * const TD, | ||||
9178 | const int ix) { | ||||
9179 | if (!CI->isLosslessCast()) | ||||
9180 | return false; | ||||
9181 | |||||
9182 | // The size of ByVal arguments is derived from the type, so we | ||||
9183 | // can't change to a type with a different size. If the size were | ||||
9184 | // passed explicitly we could avoid this check. | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9185 | if (!CS.paramHasAttr(ix, Attribute::ByVal)) |
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9186 | return true; |
9187 | |||||
9188 | const Type* SrcTy = | ||||
9189 | cast<PointerType>(CI->getOperand(0)->getType())->getElementType(); | ||||
9190 | const Type* DstTy = cast<PointerType>(CI->getType())->getElementType(); | ||||
9191 | if (!SrcTy->isSized() || !DstTy->isSized()) | ||||
9192 | return false; | ||||
9193 | if (TD->getABITypeSize(SrcTy) != TD->getABITypeSize(DstTy)) | ||||
9194 | return false; | ||||
9195 | return true; | ||||
9196 | } | ||||
9197 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9198 | // visitCallSite - Improvements for call and invoke instructions. |
9199 | // | ||||
9200 | Instruction *InstCombiner::visitCallSite(CallSite CS) { | ||||
9201 | bool Changed = false; | ||||
9202 | |||||
9203 | // If the callee is a constexpr cast of a function, attempt to move the cast | ||||
9204 | // to the arguments of the call/invoke. | ||||
9205 | if (transformConstExprCastCall(CS)) return 0; | ||||
9206 | |||||
9207 | Value *Callee = CS.getCalledValue(); | ||||
9208 | |||||
9209 | if (Function *CalleeF = dyn_cast<Function>(Callee)) | ||||
9210 | if (CalleeF->getCallingConv() != CS.getCallingConv()) { | ||||
9211 | Instruction *OldCall = CS.getInstruction(); | ||||
9212 | // If the call and callee calling conventions don't match, this call must | ||||
9213 | // be unreachable, as the call is undefined. | ||||
9214 | new StoreInst(ConstantInt::getTrue(), | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 9215 | UndefValue::get(PointerType::getUnqual(Type::Int1Ty)), |
9216 | OldCall); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9217 | if (!OldCall->use_empty()) |
9218 | OldCall->replaceAllUsesWith(UndefValue::get(OldCall->getType())); | ||||
9219 | if (isa<CallInst>(OldCall)) // Not worth removing an invoke here. | ||||
9220 | return EraseInstFromFunction(*OldCall); | ||||
9221 | return 0; | ||||
9222 | } | ||||
9223 | |||||
9224 | if (isa<ConstantPointerNull>(Callee) || isa<UndefValue>(Callee)) { | ||||
9225 | // This instruction is not reachable, just remove it. We insert a store to | ||||
9226 | // undef so that we know that this code is not reachable, despite the fact | ||||
9227 | // that we can't modify the CFG here. | ||||
9228 | new StoreInst(ConstantInt::getTrue(), | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 9229 | UndefValue::get(PointerType::getUnqual(Type::Int1Ty)), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9230 | CS.getInstruction()); |
9231 | |||||
9232 | if (!CS.getInstruction()->use_empty()) | ||||
9233 | CS.getInstruction()-> | ||||
9234 | replaceAllUsesWith(UndefValue::get(CS.getInstruction()->getType())); | ||||
9235 | |||||
9236 | if (InvokeInst *II = dyn_cast<InvokeInst>(CS.getInstruction())) { | ||||
9237 | // Don't break the CFG, insert a dummy cond branch. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9238 | BranchInst::Create(II->getNormalDest(), II->getUnwindDest(), |
9239 | ConstantInt::getTrue(), II); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9240 | } |
9241 | return EraseInstFromFunction(*CS.getInstruction()); | ||||
9242 | } | ||||
9243 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9244 | if (BitCastInst *BC = dyn_cast<BitCastInst>(Callee)) |
9245 | if (IntrinsicInst *In = dyn_cast<IntrinsicInst>(BC->getOperand(0))) | ||||
9246 | if (In->getIntrinsicID() == Intrinsic::init_trampoline) | ||||
9247 | return transformCallThroughTrampoline(CS); | ||||
9248 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9249 | const PointerType *PTy = cast<PointerType>(Callee->getType()); |
9250 | const FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); | ||||
9251 | if (FTy->isVarArg()) { | ||||
Dale Johannesen | 502336c | 2008-04-23 01:03:05 +0000 | [diff] [blame] | 9252 | int ix = FTy->getNumParams() + (isa<InvokeInst>(Callee) ? 3 : 1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9253 | // See if we can optimize any arguments passed through the varargs area of |
9254 | // the call. | ||||
9255 | for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(), | ||||
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9256 | E = CS.arg_end(); I != E; ++I, ++ix) { |
9257 | CastInst *CI = dyn_cast<CastInst>(*I); | ||||
9258 | if (CI && isSafeToEliminateVarargsCast(CS, CI, TD, ix)) { | ||||
9259 | *I = CI->getOperand(0); | ||||
9260 | Changed = true; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9261 | } |
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9262 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9263 | } |
9264 | |||||
Duncan Sands | 2937e35 | 2007-12-19 21:13:37 +0000 | [diff] [blame] | 9265 | if (isa<InlineAsm>(Callee) && !CS.doesNotThrow()) { |
Duncan Sands | 7868f3c | 2007-12-16 15:51:49 +0000 | [diff] [blame] | 9266 | // Inline asm calls cannot throw - mark them 'nounwind'. |
Duncan Sands | 2937e35 | 2007-12-19 21:13:37 +0000 | [diff] [blame] | 9267 | CS.setDoesNotThrow(); |
Duncan Sands | 7868f3c | 2007-12-16 15:51:49 +0000 | [diff] [blame] | 9268 | Changed = true; |
9269 | } | ||||
9270 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9271 | return Changed ? CS.getInstruction() : 0; |
9272 | } | ||||
9273 | |||||
9274 | // transformConstExprCastCall - If the callee is a constexpr cast of a function, | ||||
9275 | // attempt to move the cast to the arguments of the call/invoke. | ||||
9276 | // | ||||
9277 | bool InstCombiner::transformConstExprCastCall(CallSite CS) { | ||||
9278 | if (!isa<ConstantExpr>(CS.getCalledValue())) return false; | ||||
9279 | ConstantExpr *CE = cast<ConstantExpr>(CS.getCalledValue()); | ||||
9280 | if (CE->getOpcode() != Instruction::BitCast || | ||||
9281 | !isa<Function>(CE->getOperand(0))) | ||||
9282 | return false; | ||||
9283 | Function *Callee = cast<Function>(CE->getOperand(0)); | ||||
9284 | Instruction *Caller = CS.getInstruction(); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9285 | const AttrListPtr &CallerPAL = CS.getAttributes(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9286 | |
9287 | // Okay, this is a cast from a function to a different type. Unless doing so | ||||
9288 | // would cause a type conversion of one of our arguments, change this call to | ||||
9289 | // be a direct call with arguments casted to the appropriate types. | ||||
9290 | // | ||||
9291 | const FunctionType *FT = Callee->getFunctionType(); | ||||
9292 | const Type *OldRetTy = Caller->getType(); | ||||
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9293 | const Type *NewRetTy = FT->getReturnType(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9294 | |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9295 | if (isa<StructType>(NewRetTy)) |
Devang Patel | d091d32 | 2008-03-11 18:04:06 +0000 | [diff] [blame] | 9296 | return false; // TODO: Handle multiple return values. |
9297 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9298 | // Check to see if we are changing the return type... |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9299 | if (OldRetTy != NewRetTy) { |
Bill Wendling | d9644a4 | 2008-05-14 22:45:20 +0000 | [diff] [blame] | 9300 | if (Callee->isDeclaration() && |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9301 | // Conversion is ok if changing from one pointer type to another or from |
9302 | // a pointer to an integer of the same size. | ||||
9303 | !((isa<PointerType>(OldRetTy) || OldRetTy == TD->getIntPtrType()) && | ||||
Duncan Sands | 886cadb | 2008-06-17 15:55:30 +0000 | [diff] [blame] | 9304 | (isa<PointerType>(NewRetTy) || NewRetTy == TD->getIntPtrType()))) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9305 | return false; // Cannot transform this return value. |
9306 | |||||
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9307 | if (!Caller->use_empty() && |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9308 | // void -> non-void is handled specially |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9309 | NewRetTy != Type::VoidTy && !CastInst::isCastable(NewRetTy, OldRetTy)) |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9310 | return false; // Cannot transform this return value. |
9311 | |||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9312 | if (!CallerPAL.isEmpty() && !Caller->use_empty()) { |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9313 | Attributes RAttrs = CallerPAL.getRetAttributes(); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9314 | if (RAttrs & Attribute::typeIncompatible(NewRetTy)) |
Duncan Sands | dbe97dc | 2008-01-07 17:16:06 +0000 | [diff] [blame] | 9315 | return false; // Attribute not compatible with transformed value. |
9316 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9317 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9318 | // If the callsite is an invoke instruction, and the return value is used by |
9319 | // a PHI node in a successor, we cannot change the return type of the call | ||||
9320 | // because there is no place to put the cast instruction (without breaking | ||||
9321 | // the critical edge). Bail out in this case. | ||||
9322 | if (!Caller->use_empty()) | ||||
9323 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) | ||||
9324 | for (Value::use_iterator UI = II->use_begin(), E = II->use_end(); | ||||
9325 | UI != E; ++UI) | ||||
9326 | if (PHINode *PN = dyn_cast<PHINode>(*UI)) | ||||
9327 | if (PN->getParent() == II->getNormalDest() || | ||||
9328 | PN->getParent() == II->getUnwindDest()) | ||||
9329 | return false; | ||||
9330 | } | ||||
9331 | |||||
9332 | unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin()); | ||||
9333 | unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs); | ||||
9334 | |||||
9335 | CallSite::arg_iterator AI = CS.arg_begin(); | ||||
9336 | for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) { | ||||
9337 | const Type *ParamTy = FT->getParamType(i); | ||||
9338 | const Type *ActTy = (*AI)->getType(); | ||||
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9339 | |
9340 | if (!CastInst::isCastable(ActTy, ParamTy)) | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9341 | return false; // Cannot transform this parameter value. |
9342 | |||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9343 | if (CallerPAL.getParamAttributes(i + 1) |
9344 | & Attribute::typeIncompatible(ParamTy)) | ||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9345 | return false; // Attribute not compatible with transformed value. |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9346 | |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9347 | // Converting from one pointer type to another or between a pointer and an |
9348 | // 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] | 9349 | bool isConvertible = ActTy == ParamTy || |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9350 | ((isa<PointerType>(ParamTy) || ParamTy == TD->getIntPtrType()) && |
9351 | (isa<PointerType>(ActTy) || ActTy == TD->getIntPtrType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9352 | if (Callee->isDeclaration() && !isConvertible) return false; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9353 | } |
9354 | |||||
9355 | if (FT->getNumParams() < NumActualArgs && !FT->isVarArg() && | ||||
9356 | Callee->isDeclaration()) | ||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9357 | return false; // Do not delete arguments unless we have a function body. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9358 | |
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9359 | if (FT->getNumParams() < NumActualArgs && FT->isVarArg() && |
9360 | !CallerPAL.isEmpty()) | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9361 | // 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] | 9362 | // won't be dropping them. Check that these extra arguments have attributes |
9363 | // that are compatible with being a vararg call argument. | ||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9364 | for (unsigned i = CallerPAL.getNumSlots(); i; --i) { |
9365 | if (CallerPAL.getSlot(i - 1).Index <= FT->getNumParams()) | ||||
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9366 | break; |
Devang Patel | e480dfa | 2008-09-23 23:03:40 +0000 | [diff] [blame] | 9367 | Attributes PAttrs = CallerPAL.getSlot(i - 1).Attrs; |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9368 | if (PAttrs & Attribute::VarArgsIncompatible) |
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9369 | return false; |
9370 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9371 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9372 | // Okay, we decided that this is a safe thing to do: go ahead and start |
9373 | // inserting cast instructions as necessary... | ||||
9374 | std::vector<Value*> Args; | ||||
9375 | Args.reserve(NumActualArgs); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9376 | SmallVector<AttributeWithIndex, 8> attrVec; |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9377 | attrVec.reserve(NumCommonArgs); |
9378 | |||||
9379 | // Get any return attributes. | ||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9380 | Attributes RAttrs = CallerPAL.getRetAttributes(); |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9381 | |
9382 | // If the return value is not being used, the type may not be compatible | ||||
9383 | // with the existing attributes. Wipe out any problematic attributes. | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9384 | RAttrs &= ~Attribute::typeIncompatible(NewRetTy); |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9385 | |
9386 | // Add the new return attributes. | ||||
9387 | if (RAttrs) | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9388 | attrVec.push_back(AttributeWithIndex::get(0, RAttrs)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9389 | |
9390 | AI = CS.arg_begin(); | ||||
9391 | for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) { | ||||
9392 | const Type *ParamTy = FT->getParamType(i); | ||||
9393 | if ((*AI)->getType() == ParamTy) { | ||||
9394 | Args.push_back(*AI); | ||||
9395 | } else { | ||||
9396 | Instruction::CastOps opcode = CastInst::getCastOpcode(*AI, | ||||
9397 | false, ParamTy, false); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9398 | CastInst *NewCast = CastInst::Create(opcode, *AI, ParamTy, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9399 | Args.push_back(InsertNewInstBefore(NewCast, *Caller)); |
9400 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9401 | |
9402 | // Add any parameter attributes. | ||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9403 | if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9404 | attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9405 | } |
9406 | |||||
9407 | // If the function takes more arguments than the call was taking, add them | ||||
9408 | // now... | ||||
9409 | for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i) | ||||
9410 | Args.push_back(Constant::getNullValue(FT->getParamType(i))); | ||||
9411 | |||||
9412 | // If we are removing arguments to the function, emit an obnoxious warning... | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 9413 | if (FT->getNumParams() < NumActualArgs) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9414 | if (!FT->isVarArg()) { |
9415 | cerr << "WARNING: While resolving call to function '" | ||||
9416 | << Callee->getName() << "' arguments were dropped!\n"; | ||||
9417 | } else { | ||||
9418 | // Add all of the arguments in their promoted form to the arg list... | ||||
9419 | for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) { | ||||
9420 | const Type *PTy = getPromotedType((*AI)->getType()); | ||||
9421 | if (PTy != (*AI)->getType()) { | ||||
9422 | // Must promote to pass through va_arg area! | ||||
9423 | Instruction::CastOps opcode = CastInst::getCastOpcode(*AI, false, | ||||
9424 | PTy, false); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9425 | Instruction *Cast = CastInst::Create(opcode, *AI, PTy, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9426 | InsertNewInstBefore(Cast, *Caller); |
9427 | Args.push_back(Cast); | ||||
9428 | } else { | ||||
9429 | Args.push_back(*AI); | ||||
9430 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9431 | |
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9432 | // Add any parameter attributes. |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9433 | if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9434 | attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); |
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9435 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9436 | } |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 9437 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9438 | |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9439 | if (Attributes FnAttrs = CallerPAL.getFnAttributes()) |
9440 | attrVec.push_back(AttributeWithIndex::get(~0, FnAttrs)); | ||||
9441 | |||||
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9442 | if (NewRetTy == Type::VoidTy) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9443 | Caller->setName(""); // Void type should not have a name. |
9444 | |||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9445 | const AttrListPtr &NewCallerPAL = AttrListPtr::get(attrVec.begin(),attrVec.end()); |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9446 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9447 | Instruction *NC; |
9448 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9449 | NC = InvokeInst::Create(Callee, II->getNormalDest(), II->getUnwindDest(), |
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9450 | Args.begin(), Args.end(), |
9451 | Caller->getName(), Caller); | ||||
Reid Spencer | 6b0b09a | 2007-07-30 19:53:57 +0000 | [diff] [blame] | 9452 | cast<InvokeInst>(NC)->setCallingConv(II->getCallingConv()); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9453 | cast<InvokeInst>(NC)->setAttributes(NewCallerPAL); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9454 | } else { |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9455 | NC = CallInst::Create(Callee, Args.begin(), Args.end(), |
9456 | Caller->getName(), Caller); | ||||
Duncan Sands | f5588dc | 2007-11-27 13:23:08 +0000 | [diff] [blame] | 9457 | CallInst *CI = cast<CallInst>(Caller); |
9458 | if (CI->isTailCall()) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9459 | cast<CallInst>(NC)->setTailCall(); |
Duncan Sands | f5588dc | 2007-11-27 13:23:08 +0000 | [diff] [blame] | 9460 | cast<CallInst>(NC)->setCallingConv(CI->getCallingConv()); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9461 | cast<CallInst>(NC)->setAttributes(NewCallerPAL); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9462 | } |
9463 | |||||
9464 | // Insert a cast of the return type as necessary. | ||||
9465 | Value *NV = NC; | ||||
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9466 | if (OldRetTy != NV->getType() && !Caller->use_empty()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9467 | if (NV->getType() != Type::VoidTy) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9468 | Instruction::CastOps opcode = CastInst::getCastOpcode(NC, false, |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9469 | OldRetTy, false); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9470 | NV = NC = CastInst::Create(opcode, NC, OldRetTy, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9471 | |
9472 | // If this is an invoke instruction, we should insert it after the first | ||||
9473 | // non-phi, instruction in the normal successor block. | ||||
9474 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { | ||||
Dan Gohman | 514277c | 2008-05-23 21:05:58 +0000 | [diff] [blame] | 9475 | BasicBlock::iterator I = II->getNormalDest()->getFirstNonPHI(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9476 | InsertNewInstBefore(NC, *I); |
9477 | } else { | ||||
9478 | // Otherwise, it's a call, just insert cast right after the call instr | ||||
9479 | InsertNewInstBefore(NC, *Caller); | ||||
9480 | } | ||||
9481 | AddUsersToWorkList(*Caller); | ||||
9482 | } else { | ||||
9483 | NV = UndefValue::get(Caller->getType()); | ||||
9484 | } | ||||
9485 | } | ||||
9486 | |||||
9487 | if (Caller->getType() != Type::VoidTy && !Caller->use_empty()) | ||||
9488 | Caller->replaceAllUsesWith(NV); | ||||
9489 | Caller->eraseFromParent(); | ||||
9490 | RemoveFromWorkList(Caller); | ||||
9491 | return true; | ||||
9492 | } | ||||
9493 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9494 | // transformCallThroughTrampoline - Turn a call to a function created by the |
9495 | // init_trampoline intrinsic into a direct call to the underlying function. | ||||
9496 | // | ||||
9497 | Instruction *InstCombiner::transformCallThroughTrampoline(CallSite CS) { | ||||
9498 | Value *Callee = CS.getCalledValue(); | ||||
9499 | const PointerType *PTy = cast<PointerType>(Callee->getType()); | ||||
9500 | const FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9501 | const AttrListPtr &Attrs = CS.getAttributes(); |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9502 | |
9503 | // If the call already has the 'nest' attribute somewhere then give up - | ||||
9504 | // otherwise 'nest' would occur twice after splicing in the chain. | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9505 | if (Attrs.hasAttrSomewhere(Attribute::Nest)) |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9506 | return 0; |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9507 | |
9508 | IntrinsicInst *Tramp = | ||||
9509 | cast<IntrinsicInst>(cast<BitCastInst>(Callee)->getOperand(0)); | ||||
9510 | |||||
Anton Korobeynikov | 48fc88f | 2008-05-07 22:54:15 +0000 | [diff] [blame] | 9511 | Function *NestF = cast<Function>(Tramp->getOperand(2)->stripPointerCasts()); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9512 | const PointerType *NestFPTy = cast<PointerType>(NestF->getType()); |
9513 | const FunctionType *NestFTy = cast<FunctionType>(NestFPTy->getElementType()); | ||||
9514 | |||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9515 | const AttrListPtr &NestAttrs = NestF->getAttributes(); |
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9516 | if (!NestAttrs.isEmpty()) { |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9517 | unsigned NestIdx = 1; |
9518 | const Type *NestTy = 0; | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9519 | Attributes NestAttr = Attribute::None; |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9520 | |
9521 | // Look for a parameter marked with the 'nest' attribute. | ||||
9522 | for (FunctionType::param_iterator I = NestFTy->param_begin(), | ||||
9523 | E = NestFTy->param_end(); I != E; ++NestIdx, ++I) | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9524 | if (NestAttrs.paramHasAttr(NestIdx, Attribute::Nest)) { |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9525 | // Record the parameter type and any other attributes. |
9526 | NestTy = *I; | ||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9527 | NestAttr = NestAttrs.getParamAttributes(NestIdx); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9528 | break; |
9529 | } | ||||
9530 | |||||
9531 | if (NestTy) { | ||||
9532 | Instruction *Caller = CS.getInstruction(); | ||||
9533 | std::vector<Value*> NewArgs; | ||||
9534 | NewArgs.reserve(unsigned(CS.arg_end()-CS.arg_begin())+1); | ||||
9535 | |||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9536 | SmallVector<AttributeWithIndex, 8> NewAttrs; |
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9537 | NewAttrs.reserve(Attrs.getNumSlots() + 1); |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9538 | |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9539 | // Insert the nest argument into the call argument list, which may |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9540 | // mean appending it. Likewise for attributes. |
9541 | |||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9542 | // Add any result attributes. |
9543 | if (Attributes Attr = Attrs.getRetAttributes()) | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9544 | NewAttrs.push_back(AttributeWithIndex::get(0, Attr)); |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9545 | |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9546 | { |
9547 | unsigned Idx = 1; | ||||
9548 | CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); | ||||
9549 | do { | ||||
9550 | if (Idx == NestIdx) { | ||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9551 | // Add the chain argument and attributes. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9552 | Value *NestVal = Tramp->getOperand(3); |
9553 | if (NestVal->getType() != NestTy) | ||||
9554 | NestVal = new BitCastInst(NestVal, NestTy, "nest", Caller); | ||||
9555 | NewArgs.push_back(NestVal); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9556 | NewAttrs.push_back(AttributeWithIndex::get(NestIdx, NestAttr)); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9557 | } |
9558 | |||||
9559 | if (I == E) | ||||
9560 | break; | ||||
9561 | |||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9562 | // Add the original argument and attributes. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9563 | NewArgs.push_back(*I); |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9564 | if (Attributes Attr = Attrs.getParamAttributes(Idx)) |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9565 | NewAttrs.push_back |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9566 | (AttributeWithIndex::get(Idx + (Idx >= NestIdx), Attr)); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9567 | |
9568 | ++Idx, ++I; | ||||
9569 | } while (1); | ||||
9570 | } | ||||
9571 | |||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9572 | // Add any function attributes. |
9573 | if (Attributes Attr = Attrs.getFnAttributes()) | ||||
9574 | NewAttrs.push_back(AttributeWithIndex::get(~0, Attr)); | ||||
9575 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9576 | // The trampoline may have been bitcast to a bogus type (FTy). |
9577 | // Handle this by synthesizing a new function type, equal to FTy | ||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9578 | // with the chain parameter inserted. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9579 | |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9580 | std::vector<const Type*> NewTypes; |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9581 | NewTypes.reserve(FTy->getNumParams()+1); |
9582 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9583 | // 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] | 9584 | // mean appending it. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9585 | { |
9586 | unsigned Idx = 1; | ||||
9587 | FunctionType::param_iterator I = FTy->param_begin(), | ||||
9588 | E = FTy->param_end(); | ||||
9589 | |||||
9590 | do { | ||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9591 | if (Idx == NestIdx) |
9592 | // Add the chain's type. | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9593 | NewTypes.push_back(NestTy); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9594 | |
9595 | if (I == E) | ||||
9596 | break; | ||||
9597 | |||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9598 | // Add the original type. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9599 | NewTypes.push_back(*I); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9600 | |
9601 | ++Idx, ++I; | ||||
9602 | } while (1); | ||||
9603 | } | ||||
9604 | |||||
9605 | // Replace the trampoline call with a direct call. Let the generic | ||||
9606 | // code sort out any function type mismatches. | ||||
9607 | FunctionType *NewFTy = | ||||
Duncan Sands | f5588dc | 2007-11-27 13:23:08 +0000 | [diff] [blame] | 9608 | FunctionType::get(FTy->getReturnType(), NewTypes, FTy->isVarArg()); |
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 9609 | Constant *NewCallee = NestF->getType() == PointerType::getUnqual(NewFTy) ? |
9610 | NestF : ConstantExpr::getBitCast(NestF, PointerType::getUnqual(NewFTy)); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9611 | const AttrListPtr &NewPAL = AttrListPtr::get(NewAttrs.begin(),NewAttrs.end()); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9612 | |
9613 | Instruction *NewCaller; | ||||
9614 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9615 | NewCaller = InvokeInst::Create(NewCallee, |
9616 | II->getNormalDest(), II->getUnwindDest(), | ||||
9617 | NewArgs.begin(), NewArgs.end(), | ||||
9618 | Caller->getName(), Caller); | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9619 | cast<InvokeInst>(NewCaller)->setCallingConv(II->getCallingConv()); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9620 | cast<InvokeInst>(NewCaller)->setAttributes(NewPAL); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9621 | } else { |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9622 | NewCaller = CallInst::Create(NewCallee, NewArgs.begin(), NewArgs.end(), |
9623 | Caller->getName(), Caller); | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9624 | if (cast<CallInst>(Caller)->isTailCall()) |
9625 | cast<CallInst>(NewCaller)->setTailCall(); | ||||
9626 | cast<CallInst>(NewCaller)-> | ||||
9627 | setCallingConv(cast<CallInst>(Caller)->getCallingConv()); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9628 | cast<CallInst>(NewCaller)->setAttributes(NewPAL); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9629 | } |
9630 | if (Caller->getType() != Type::VoidTy && !Caller->use_empty()) | ||||
9631 | Caller->replaceAllUsesWith(NewCaller); | ||||
9632 | Caller->eraseFromParent(); | ||||
9633 | RemoveFromWorkList(Caller); | ||||
9634 | return 0; | ||||
9635 | } | ||||
9636 | } | ||||
9637 | |||||
9638 | // Replace the trampoline call with a direct call. Since there is no 'nest' | ||||
9639 | // parameter, there is no need to adjust the argument list. Let the generic | ||||
9640 | // code sort out any function type mismatches. | ||||
9641 | Constant *NewCallee = | ||||
9642 | NestF->getType() == PTy ? NestF : ConstantExpr::getBitCast(NestF, PTy); | ||||
9643 | CS.setCalledFunction(NewCallee); | ||||
9644 | return CS.getInstruction(); | ||||
9645 | } | ||||
9646 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9647 | /// FoldPHIArgBinOpIntoPHI - If we have something like phi [add (a,b), add(c,d)] |
9648 | /// and if a/b/c/d and the add's all have a single use, turn this into two phi's | ||||
9649 | /// and a single binop. | ||||
9650 | Instruction *InstCombiner::FoldPHIArgBinOpIntoPHI(PHINode &PN) { | ||||
9651 | Instruction *FirstInst = cast<Instruction>(PN.getIncomingValue(0)); | ||||
9652 | assert(isa<BinaryOperator>(FirstInst) || isa<GetElementPtrInst>(FirstInst) || | ||||
9653 | isa<CmpInst>(FirstInst)); | ||||
9654 | unsigned Opc = FirstInst->getOpcode(); | ||||
9655 | Value *LHSVal = FirstInst->getOperand(0); | ||||
9656 | Value *RHSVal = FirstInst->getOperand(1); | ||||
9657 | |||||
9658 | const Type *LHSType = LHSVal->getType(); | ||||
9659 | const Type *RHSType = RHSVal->getType(); | ||||
9660 | |||||
9661 | // Scan to see if all operands are the same opcode, all have one use, and all | ||||
9662 | // kill their operands (i.e. the operands have one use). | ||||
9663 | for (unsigned i = 0; i != PN.getNumIncomingValues(); ++i) { | ||||
9664 | Instruction *I = dyn_cast<Instruction>(PN.getIncomingValue(i)); | ||||
9665 | if (!I || I->getOpcode() != Opc || !I->hasOneUse() || | ||||
9666 | // Verify type of the LHS matches so we don't fold cmp's of different | ||||
9667 | // types or GEP's with different index types. | ||||
9668 | I->getOperand(0)->getType() != LHSType || | ||||
9669 | I->getOperand(1)->getType() != RHSType) | ||||
9670 | return 0; | ||||
9671 | |||||
9672 | // If they are CmpInst instructions, check their predicates | ||||
9673 | if (Opc == Instruction::ICmp || Opc == Instruction::FCmp) | ||||
9674 | if (cast<CmpInst>(I)->getPredicate() != | ||||
9675 | cast<CmpInst>(FirstInst)->getPredicate()) | ||||
9676 | return 0; | ||||
9677 | |||||
9678 | // Keep track of which operand needs a phi node. | ||||
9679 | if (I->getOperand(0) != LHSVal) LHSVal = 0; | ||||
9680 | if (I->getOperand(1) != RHSVal) RHSVal = 0; | ||||
9681 | } | ||||
9682 | |||||
9683 | // Otherwise, this is safe to transform, determine if it is profitable. | ||||
9684 | |||||
9685 | // If this is a GEP, and if the index (not the pointer) needs a PHI, bail out. | ||||
9686 | // Indexes are often folded into load/store instructions, so we don't want to | ||||
9687 | // hide them behind a phi. | ||||
9688 | if (isa<GetElementPtrInst>(FirstInst) && RHSVal == 0) | ||||
9689 | return 0; | ||||
9690 | |||||
9691 | Value *InLHS = FirstInst->getOperand(0); | ||||
9692 | Value *InRHS = FirstInst->getOperand(1); | ||||
9693 | PHINode *NewLHS = 0, *NewRHS = 0; | ||||
9694 | if (LHSVal == 0) { | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9695 | NewLHS = PHINode::Create(LHSType, |
9696 | FirstInst->getOperand(0)->getName() + ".pn"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9697 | NewLHS->reserveOperandSpace(PN.getNumOperands()/2); |
9698 | NewLHS->addIncoming(InLHS, PN.getIncomingBlock(0)); | ||||
9699 | InsertNewInstBefore(NewLHS, PN); | ||||
9700 | LHSVal = NewLHS; | ||||
9701 | } | ||||
9702 | |||||
9703 | if (RHSVal == 0) { | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9704 | NewRHS = PHINode::Create(RHSType, |
9705 | FirstInst->getOperand(1)->getName() + ".pn"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9706 | NewRHS->reserveOperandSpace(PN.getNumOperands()/2); |
9707 | NewRHS->addIncoming(InRHS, PN.getIncomingBlock(0)); | ||||
9708 | InsertNewInstBefore(NewRHS, PN); | ||||
9709 | RHSVal = NewRHS; | ||||
9710 | } | ||||
9711 | |||||
9712 | // Add all operands to the new PHIs. | ||||
9713 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
9714 | if (NewLHS) { | ||||
9715 | Value *NewInLHS =cast<Instruction>(PN.getIncomingValue(i))->getOperand(0); | ||||
9716 | NewLHS->addIncoming(NewInLHS, PN.getIncomingBlock(i)); | ||||
9717 | } | ||||
9718 | if (NewRHS) { | ||||
9719 | Value *NewInRHS =cast<Instruction>(PN.getIncomingValue(i))->getOperand(1); | ||||
9720 | NewRHS->addIncoming(NewInRHS, PN.getIncomingBlock(i)); | ||||
9721 | } | ||||
9722 | } | ||||
9723 | |||||
9724 | if (BinaryOperator *BinOp = dyn_cast<BinaryOperator>(FirstInst)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9725 | return BinaryOperator::Create(BinOp->getOpcode(), LHSVal, RHSVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9726 | else if (CmpInst *CIOp = dyn_cast<CmpInst>(FirstInst)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9727 | return CmpInst::Create(CIOp->getOpcode(), CIOp->getPredicate(), LHSVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9728 | RHSVal); |
9729 | else { | ||||
9730 | assert(isa<GetElementPtrInst>(FirstInst)); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9731 | return GetElementPtrInst::Create(LHSVal, RHSVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9732 | } |
9733 | } | ||||
9734 | |||||
9735 | /// isSafeToSinkLoad - Return true if we know that it is safe sink the load out | ||||
9736 | /// of the block that defines it. This means that it must be obvious the value | ||||
9737 | /// of the load is not changed from the point of the load to the end of the | ||||
9738 | /// block it is in. | ||||
9739 | /// | ||||
9740 | /// Finally, it is safe, but not profitable, to sink a load targetting a | ||||
9741 | /// non-address-taken alloca. Doing so will cause us to not promote the alloca | ||||
9742 | /// to a register. | ||||
9743 | static bool isSafeToSinkLoad(LoadInst *L) { | ||||
9744 | BasicBlock::iterator BBI = L, E = L->getParent()->end(); | ||||
9745 | |||||
9746 | for (++BBI; BBI != E; ++BBI) | ||||
9747 | if (BBI->mayWriteToMemory()) | ||||
9748 | return false; | ||||
9749 | |||||
9750 | // Check for non-address taken alloca. If not address-taken already, it isn't | ||||
9751 | // profitable to do this xform. | ||||
9752 | if (AllocaInst *AI = dyn_cast<AllocaInst>(L->getOperand(0))) { | ||||
9753 | bool isAddressTaken = false; | ||||
9754 | for (Value::use_iterator UI = AI->use_begin(), E = AI->use_end(); | ||||
9755 | UI != E; ++UI) { | ||||
9756 | if (isa<LoadInst>(UI)) continue; | ||||
9757 | if (StoreInst *SI = dyn_cast<StoreInst>(*UI)) { | ||||
9758 | // If storing TO the alloca, then the address isn't taken. | ||||
9759 | if (SI->getOperand(1) == AI) continue; | ||||
9760 | } | ||||
9761 | isAddressTaken = true; | ||||
9762 | break; | ||||
9763 | } | ||||
9764 | |||||
9765 | if (!isAddressTaken) | ||||
9766 | return false; | ||||
9767 | } | ||||
9768 | |||||
9769 | return true; | ||||
9770 | } | ||||
9771 | |||||
9772 | |||||
9773 | // FoldPHIArgOpIntoPHI - If all operands to a PHI node are the same "unary" | ||||
9774 | // operator and they all are only used by the PHI, PHI together their | ||||
9775 | // inputs, and do the operation once, to the result of the PHI. | ||||
9776 | Instruction *InstCombiner::FoldPHIArgOpIntoPHI(PHINode &PN) { | ||||
9777 | Instruction *FirstInst = cast<Instruction>(PN.getIncomingValue(0)); | ||||
9778 | |||||
9779 | // Scan the instruction, looking for input operations that can be folded away. | ||||
9780 | // If all input operands to the phi are the same instruction (e.g. a cast from | ||||
9781 | // the same type or "+42") we can pull the operation through the PHI, reducing | ||||
9782 | // code size and simplifying code. | ||||
9783 | Constant *ConstantOp = 0; | ||||
9784 | const Type *CastSrcTy = 0; | ||||
9785 | bool isVolatile = false; | ||||
9786 | if (isa<CastInst>(FirstInst)) { | ||||
9787 | CastSrcTy = FirstInst->getOperand(0)->getType(); | ||||
9788 | } else if (isa<BinaryOperator>(FirstInst) || isa<CmpInst>(FirstInst)) { | ||||
9789 | // Can fold binop, compare or shift here if the RHS is a constant, | ||||
9790 | // otherwise call FoldPHIArgBinOpIntoPHI. | ||||
9791 | ConstantOp = dyn_cast<Constant>(FirstInst->getOperand(1)); | ||||
9792 | if (ConstantOp == 0) | ||||
9793 | return FoldPHIArgBinOpIntoPHI(PN); | ||||
9794 | } else if (LoadInst *LI = dyn_cast<LoadInst>(FirstInst)) { | ||||
9795 | isVolatile = LI->isVolatile(); | ||||
9796 | // We can't sink the load if the loaded value could be modified between the | ||||
9797 | // load and the PHI. | ||||
9798 | if (LI->getParent() != PN.getIncomingBlock(0) || | ||||
9799 | !isSafeToSinkLoad(LI)) | ||||
9800 | return 0; | ||||
Chris Lattner | 2d9fdd8 | 2008-07-08 17:18:32 +0000 | [diff] [blame] | 9801 | |
9802 | // If the PHI is of volatile loads and the load block has multiple | ||||
9803 | // successors, sinking it would remove a load of the volatile value from | ||||
9804 | // the path through the other successor. | ||||
9805 | if (isVolatile && | ||||
9806 | LI->getParent()->getTerminator()->getNumSuccessors() != 1) | ||||
9807 | return 0; | ||||
9808 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9809 | } else if (isa<GetElementPtrInst>(FirstInst)) { |
9810 | if (FirstInst->getNumOperands() == 2) | ||||
9811 | return FoldPHIArgBinOpIntoPHI(PN); | ||||
9812 | // Can't handle general GEPs yet. | ||||
9813 | return 0; | ||||
9814 | } else { | ||||
9815 | return 0; // Cannot fold this operation. | ||||
9816 | } | ||||
9817 | |||||
9818 | // Check to see if all arguments are the same operation. | ||||
9819 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
9820 | if (!isa<Instruction>(PN.getIncomingValue(i))) return 0; | ||||
9821 | Instruction *I = cast<Instruction>(PN.getIncomingValue(i)); | ||||
9822 | if (!I->hasOneUse() || !I->isSameOperationAs(FirstInst)) | ||||
9823 | return 0; | ||||
9824 | if (CastSrcTy) { | ||||
9825 | if (I->getOperand(0)->getType() != CastSrcTy) | ||||
9826 | return 0; // Cast operation must match. | ||||
9827 | } else if (LoadInst *LI = dyn_cast<LoadInst>(I)) { | ||||
9828 | // We can't sink the load if the loaded value could be modified between | ||||
9829 | // the load and the PHI. | ||||
9830 | if (LI->isVolatile() != isVolatile || | ||||
9831 | LI->getParent() != PN.getIncomingBlock(i) || | ||||
9832 | !isSafeToSinkLoad(LI)) | ||||
9833 | return 0; | ||||
Chris Lattner | f786701 | 2008-04-29 17:28:22 +0000 | [diff] [blame] | 9834 | |
Chris Lattner | 2d9fdd8 | 2008-07-08 17:18:32 +0000 | [diff] [blame] | 9835 | // If the PHI is of volatile loads and the load block has multiple |
9836 | // successors, sinking it would remove a load of the volatile value from | ||||
9837 | // the path through the other successor. | ||||
Chris Lattner | f786701 | 2008-04-29 17:28:22 +0000 | [diff] [blame] | 9838 | if (isVolatile && |
9839 | LI->getParent()->getTerminator()->getNumSuccessors() != 1) | ||||
9840 | return 0; | ||||
9841 | |||||
9842 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9843 | } else if (I->getOperand(1) != ConstantOp) { |
9844 | return 0; | ||||
9845 | } | ||||
9846 | } | ||||
9847 | |||||
9848 | // Okay, they are all the same operation. Create a new PHI node of the | ||||
9849 | // 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] | 9850 | PHINode *NewPN = PHINode::Create(FirstInst->getOperand(0)->getType(), |
9851 | PN.getName()+".in"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9852 | NewPN->reserveOperandSpace(PN.getNumOperands()/2); |
9853 | |||||
9854 | Value *InVal = FirstInst->getOperand(0); | ||||
9855 | NewPN->addIncoming(InVal, PN.getIncomingBlock(0)); | ||||
9856 | |||||
9857 | // Add all operands to the new PHI. | ||||
9858 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
9859 | Value *NewInVal = cast<Instruction>(PN.getIncomingValue(i))->getOperand(0); | ||||
9860 | if (NewInVal != InVal) | ||||
9861 | InVal = 0; | ||||
9862 | NewPN->addIncoming(NewInVal, PN.getIncomingBlock(i)); | ||||
9863 | } | ||||
9864 | |||||
9865 | Value *PhiVal; | ||||
9866 | if (InVal) { | ||||
9867 | // The new PHI unions all of the same values together. This is really | ||||
9868 | // common, so we handle it intelligently here for compile-time speed. | ||||
9869 | PhiVal = InVal; | ||||
9870 | delete NewPN; | ||||
9871 | } else { | ||||
9872 | InsertNewInstBefore(NewPN, PN); | ||||
9873 | PhiVal = NewPN; | ||||
9874 | } | ||||
9875 | |||||
9876 | // Insert and return the new operation. | ||||
9877 | if (CastInst* FirstCI = dyn_cast<CastInst>(FirstInst)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9878 | return CastInst::Create(FirstCI->getOpcode(), PhiVal, PN.getType()); |
Chris Lattner | fc984e9 | 2008-04-29 17:13:43 +0000 | [diff] [blame] | 9879 | if (BinaryOperator *BinOp = dyn_cast<BinaryOperator>(FirstInst)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9880 | return BinaryOperator::Create(BinOp->getOpcode(), PhiVal, ConstantOp); |
Chris Lattner | fc984e9 | 2008-04-29 17:13:43 +0000 | [diff] [blame] | 9881 | if (CmpInst *CIOp = dyn_cast<CmpInst>(FirstInst)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9882 | return CmpInst::Create(CIOp->getOpcode(), CIOp->getPredicate(), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9883 | PhiVal, ConstantOp); |
Chris Lattner | fc984e9 | 2008-04-29 17:13:43 +0000 | [diff] [blame] | 9884 | assert(isa<LoadInst>(FirstInst) && "Unknown operation"); |
9885 | |||||
9886 | // If this was a volatile load that we are merging, make sure to loop through | ||||
9887 | // and mark all the input loads as non-volatile. If we don't do this, we will | ||||
9888 | // insert a new volatile load and the old ones will not be deletable. | ||||
9889 | if (isVolatile) | ||||
9890 | for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) | ||||
9891 | cast<LoadInst>(PN.getIncomingValue(i))->setVolatile(false); | ||||
9892 | |||||
9893 | return new LoadInst(PhiVal, "", isVolatile); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9894 | } |
9895 | |||||
9896 | /// DeadPHICycle - Return true if this PHI node is only used by a PHI node cycle | ||||
9897 | /// that is dead. | ||||
9898 | static bool DeadPHICycle(PHINode *PN, | ||||
9899 | SmallPtrSet<PHINode*, 16> &PotentiallyDeadPHIs) { | ||||
9900 | if (PN->use_empty()) return true; | ||||
9901 | if (!PN->hasOneUse()) return false; | ||||
9902 | |||||
9903 | // Remember this node, and if we find the cycle, return. | ||||
9904 | if (!PotentiallyDeadPHIs.insert(PN)) | ||||
9905 | return true; | ||||
Chris Lattner | adf2e34 | 2007-08-28 04:23:55 +0000 | [diff] [blame] | 9906 | |
9907 | // Don't scan crazily complex things. | ||||
9908 | if (PotentiallyDeadPHIs.size() == 16) | ||||
9909 | return false; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9910 | |
9911 | if (PHINode *PU = dyn_cast<PHINode>(PN->use_back())) | ||||
9912 | return DeadPHICycle(PU, PotentiallyDeadPHIs); | ||||
9913 | |||||
9914 | return false; | ||||
9915 | } | ||||
9916 | |||||
Chris Lattner | 27b695d | 2007-11-06 21:52:06 +0000 | [diff] [blame] | 9917 | /// PHIsEqualValue - Return true if this phi node is always equal to |
9918 | /// NonPhiInVal. This happens with mutually cyclic phi nodes like: | ||||
9919 | /// z = some value; x = phi (y, z); y = phi (x, z) | ||||
9920 | static bool PHIsEqualValue(PHINode *PN, Value *NonPhiInVal, | ||||
9921 | SmallPtrSet<PHINode*, 16> &ValueEqualPHIs) { | ||||
9922 | // See if we already saw this PHI node. | ||||
9923 | if (!ValueEqualPHIs.insert(PN)) | ||||
9924 | return true; | ||||
9925 | |||||
9926 | // Don't scan crazily complex things. | ||||
9927 | if (ValueEqualPHIs.size() == 16) | ||||
9928 | return false; | ||||
9929 | |||||
9930 | // Scan the operands to see if they are either phi nodes or are equal to | ||||
9931 | // the value. | ||||
9932 | for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) { | ||||
9933 | Value *Op = PN->getIncomingValue(i); | ||||
9934 | if (PHINode *OpPN = dyn_cast<PHINode>(Op)) { | ||||
9935 | if (!PHIsEqualValue(OpPN, NonPhiInVal, ValueEqualPHIs)) | ||||
9936 | return false; | ||||
9937 | } else if (Op != NonPhiInVal) | ||||
9938 | return false; | ||||
9939 | } | ||||
9940 | |||||
9941 | return true; | ||||
9942 | } | ||||
9943 | |||||
9944 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9945 | // PHINode simplification |
9946 | // | ||||
9947 | Instruction *InstCombiner::visitPHINode(PHINode &PN) { | ||||
9948 | // If LCSSA is around, don't mess with Phi nodes | ||||
9949 | if (MustPreserveLCSSA) return 0; | ||||
9950 | |||||
9951 | if (Value *V = PN.hasConstantValue()) | ||||
9952 | return ReplaceInstUsesWith(PN, V); | ||||
9953 | |||||
9954 | // If all PHI operands are the same operation, pull them through the PHI, | ||||
9955 | // reducing code size. | ||||
9956 | if (isa<Instruction>(PN.getIncomingValue(0)) && | ||||
9957 | PN.getIncomingValue(0)->hasOneUse()) | ||||
9958 | if (Instruction *Result = FoldPHIArgOpIntoPHI(PN)) | ||||
9959 | return Result; | ||||
9960 | |||||
9961 | // If this is a trivial cycle in the PHI node graph, remove it. Basically, if | ||||
9962 | // this PHI only has a single use (a PHI), and if that PHI only has one use (a | ||||
9963 | // PHI)... break the cycle. | ||||
9964 | if (PN.hasOneUse()) { | ||||
9965 | Instruction *PHIUser = cast<Instruction>(PN.use_back()); | ||||
9966 | if (PHINode *PU = dyn_cast<PHINode>(PHIUser)) { | ||||
9967 | SmallPtrSet<PHINode*, 16> PotentiallyDeadPHIs; | ||||
9968 | PotentiallyDeadPHIs.insert(&PN); | ||||
9969 | if (DeadPHICycle(PU, PotentiallyDeadPHIs)) | ||||
9970 | return ReplaceInstUsesWith(PN, UndefValue::get(PN.getType())); | ||||
9971 | } | ||||
9972 | |||||
9973 | // If this phi has a single use, and if that use just computes a value for | ||||
9974 | // the next iteration of a loop, delete the phi. This occurs with unused | ||||
9975 | // induction variables, e.g. "for (int j = 0; ; ++j);". Detecting this | ||||
9976 | // common case here is good because the only other things that catch this | ||||
9977 | // are induction variable analysis (sometimes) and ADCE, which is only run | ||||
9978 | // late. | ||||
9979 | if (PHIUser->hasOneUse() && | ||||
9980 | (isa<BinaryOperator>(PHIUser) || isa<GetElementPtrInst>(PHIUser)) && | ||||
9981 | PHIUser->use_back() == &PN) { | ||||
9982 | return ReplaceInstUsesWith(PN, UndefValue::get(PN.getType())); | ||||
9983 | } | ||||
9984 | } | ||||
9985 | |||||
Chris Lattner | 27b695d | 2007-11-06 21:52:06 +0000 | [diff] [blame] | 9986 | // We sometimes end up with phi cycles that non-obviously end up being the |
9987 | // same value, for example: | ||||
9988 | // z = some value; x = phi (y, z); y = phi (x, z) | ||||
9989 | // where the phi nodes don't necessarily need to be in the same block. Do a | ||||
9990 | // quick check to see if the PHI node only contains a single non-phi value, if | ||||
9991 | // so, scan to see if the phi cycle is actually equal to that value. | ||||
9992 | { | ||||
9993 | unsigned InValNo = 0, NumOperandVals = PN.getNumIncomingValues(); | ||||
9994 | // Scan for the first non-phi operand. | ||||
9995 | while (InValNo != NumOperandVals && | ||||
9996 | isa<PHINode>(PN.getIncomingValue(InValNo))) | ||||
9997 | ++InValNo; | ||||
9998 | |||||
9999 | if (InValNo != NumOperandVals) { | ||||
10000 | Value *NonPhiInVal = PN.getOperand(InValNo); | ||||
10001 | |||||
10002 | // Scan the rest of the operands to see if there are any conflicts, if so | ||||
10003 | // there is no need to recursively scan other phis. | ||||
10004 | for (++InValNo; InValNo != NumOperandVals; ++InValNo) { | ||||
10005 | Value *OpVal = PN.getIncomingValue(InValNo); | ||||
10006 | if (OpVal != NonPhiInVal && !isa<PHINode>(OpVal)) | ||||
10007 | break; | ||||
10008 | } | ||||
10009 | |||||
10010 | // If we scanned over all operands, then we have one unique value plus | ||||
10011 | // phi values. Scan PHI nodes to see if they all merge in each other or | ||||
10012 | // the value. | ||||
10013 | if (InValNo == NumOperandVals) { | ||||
10014 | SmallPtrSet<PHINode*, 16> ValueEqualPHIs; | ||||
10015 | if (PHIsEqualValue(&PN, NonPhiInVal, ValueEqualPHIs)) | ||||
10016 | return ReplaceInstUsesWith(PN, NonPhiInVal); | ||||
10017 | } | ||||
10018 | } | ||||
10019 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10020 | return 0; |
10021 | } | ||||
10022 | |||||
10023 | static Value *InsertCastToIntPtrTy(Value *V, const Type *DTy, | ||||
10024 | Instruction *InsertPoint, | ||||
10025 | InstCombiner *IC) { | ||||
10026 | unsigned PtrSize = DTy->getPrimitiveSizeInBits(); | ||||
10027 | unsigned VTySize = V->getType()->getPrimitiveSizeInBits(); | ||||
10028 | // We must cast correctly to the pointer type. Ensure that we | ||||
10029 | // sign extend the integer value if it is smaller as this is | ||||
10030 | // used for address computation. | ||||
10031 | Instruction::CastOps opcode = | ||||
10032 | (VTySize < PtrSize ? Instruction::SExt : | ||||
10033 | (VTySize == PtrSize ? Instruction::BitCast : Instruction::Trunc)); | ||||
10034 | return IC->InsertCastBefore(opcode, V, DTy, *InsertPoint); | ||||
10035 | } | ||||
10036 | |||||
10037 | |||||
10038 | Instruction *InstCombiner::visitGetElementPtrInst(GetElementPtrInst &GEP) { | ||||
10039 | Value *PtrOp = GEP.getOperand(0); | ||||
10040 | // Is it 'getelementptr %P, i32 0' or 'getelementptr %P' | ||||
10041 | // If so, eliminate the noop. | ||||
10042 | if (GEP.getNumOperands() == 1) | ||||
10043 | return ReplaceInstUsesWith(GEP, PtrOp); | ||||
10044 | |||||
10045 | if (isa<UndefValue>(GEP.getOperand(0))) | ||||
10046 | return ReplaceInstUsesWith(GEP, UndefValue::get(GEP.getType())); | ||||
10047 | |||||
10048 | bool HasZeroPointerIndex = false; | ||||
10049 | if (Constant *C = dyn_cast<Constant>(GEP.getOperand(1))) | ||||
10050 | HasZeroPointerIndex = C->isNullValue(); | ||||
10051 | |||||
10052 | if (GEP.getNumOperands() == 2 && HasZeroPointerIndex) | ||||
10053 | return ReplaceInstUsesWith(GEP, PtrOp); | ||||
10054 | |||||
10055 | // Eliminate unneeded casts for indices. | ||||
10056 | bool MadeChange = false; | ||||
10057 | |||||
10058 | gep_type_iterator GTI = gep_type_begin(GEP); | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10059 | for (User::op_iterator i = GEP.op_begin() + 1, e = GEP.op_end(); |
10060 | i != e; ++i, ++GTI) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10061 | if (isa<SequentialType>(*GTI)) { |
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10062 | if (CastInst *CI = dyn_cast<CastInst>(*i)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10063 | if (CI->getOpcode() == Instruction::ZExt || |
10064 | CI->getOpcode() == Instruction::SExt) { | ||||
10065 | const Type *SrcTy = CI->getOperand(0)->getType(); | ||||
10066 | // We can eliminate a cast from i32 to i64 iff the target | ||||
10067 | // is a 32-bit pointer target. | ||||
10068 | if (SrcTy->getPrimitiveSizeInBits() >= TD->getPointerSizeInBits()) { | ||||
10069 | MadeChange = true; | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10070 | *i = CI->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10071 | } |
10072 | } | ||||
10073 | } | ||||
10074 | // 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] | 10075 | // to what we need. If narrower, sign-extend it to what we need. |
10076 | // If the incoming value needs a cast instruction, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10077 | // insert it. This explicit cast can make subsequent optimizations more |
10078 | // obvious. | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10079 | Value *Op = *i; |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10080 | if (TD->getTypeSizeInBits(Op->getType()) > TD->getPointerSizeInBits()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10081 | if (Constant *C = dyn_cast<Constant>(Op)) { |
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10082 | *i = ConstantExpr::getTrunc(C, TD->getIntPtrType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10083 | MadeChange = true; |
10084 | } else { | ||||
10085 | Op = InsertCastBefore(Instruction::Trunc, Op, TD->getIntPtrType(), | ||||
10086 | GEP); | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10087 | *i = Op; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10088 | MadeChange = true; |
10089 | } | ||||
Dan Gohman | 5d639ed | 2008-09-11 23:06:38 +0000 | [diff] [blame] | 10090 | } else if (TD->getTypeSizeInBits(Op->getType()) < TD->getPointerSizeInBits()) { |
10091 | if (Constant *C = dyn_cast<Constant>(Op)) { | ||||
10092 | *i = ConstantExpr::getSExt(C, TD->getIntPtrType()); | ||||
10093 | MadeChange = true; | ||||
10094 | } else { | ||||
10095 | Op = InsertCastBefore(Instruction::SExt, Op, TD->getIntPtrType(), | ||||
10096 | GEP); | ||||
10097 | *i = Op; | ||||
10098 | MadeChange = true; | ||||
10099 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10100 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10101 | } |
10102 | } | ||||
10103 | if (MadeChange) return &GEP; | ||||
10104 | |||||
10105 | // If this GEP instruction doesn't move the pointer, and if the input operand | ||||
10106 | // is a bitcast of another pointer, just replace the GEP with a bitcast of the | ||||
10107 | // real input to the dest type. | ||||
Chris Lattner | c59171a | 2007-10-12 05:30:59 +0000 | [diff] [blame] | 10108 | if (GEP.hasAllZeroIndices()) { |
10109 | if (BitCastInst *BCI = dyn_cast<BitCastInst>(GEP.getOperand(0))) { | ||||
10110 | // If the bitcast is of an allocation, and the allocation will be | ||||
10111 | // converted to match the type of the cast, don't touch this. | ||||
10112 | if (isa<AllocationInst>(BCI->getOperand(0))) { | ||||
10113 | // See if the bitcast simplifies, if so, don't nuke this GEP yet. | ||||
Chris Lattner | 551a587 | 2007-10-12 18:05:47 +0000 | [diff] [blame] | 10114 | if (Instruction *I = visitBitCast(*BCI)) { |
10115 | if (I != BCI) { | ||||
10116 | I->takeName(BCI); | ||||
10117 | BCI->getParent()->getInstList().insert(BCI, I); | ||||
10118 | ReplaceInstUsesWith(*BCI, I); | ||||
10119 | } | ||||
Chris Lattner | c59171a | 2007-10-12 05:30:59 +0000 | [diff] [blame] | 10120 | return &GEP; |
Chris Lattner | 551a587 | 2007-10-12 18:05:47 +0000 | [diff] [blame] | 10121 | } |
Chris Lattner | c59171a | 2007-10-12 05:30:59 +0000 | [diff] [blame] | 10122 | } |
10123 | return new BitCastInst(BCI->getOperand(0), GEP.getType()); | ||||
10124 | } | ||||
10125 | } | ||||
10126 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10127 | // Combine Indices - If the source pointer to this getelementptr instruction |
10128 | // is a getelementptr instruction, combine the indices of the two | ||||
10129 | // getelementptr instructions into a single instruction. | ||||
10130 | // | ||||
10131 | SmallVector<Value*, 8> SrcGEPOperands; | ||||
10132 | if (User *Src = dyn_castGetElementPtr(PtrOp)) | ||||
10133 | SrcGEPOperands.append(Src->op_begin(), Src->op_end()); | ||||
10134 | |||||
10135 | if (!SrcGEPOperands.empty()) { | ||||
10136 | // Note that if our source is a gep chain itself that we wait for that | ||||
10137 | // chain to be resolved before we perform this transformation. This | ||||
10138 | // avoids us creating a TON of code in some cases. | ||||
10139 | // | ||||
10140 | if (isa<GetElementPtrInst>(SrcGEPOperands[0]) && | ||||
10141 | cast<Instruction>(SrcGEPOperands[0])->getNumOperands() == 2) | ||||
10142 | return 0; // Wait until our source is folded to completion. | ||||
10143 | |||||
10144 | SmallVector<Value*, 8> Indices; | ||||
10145 | |||||
10146 | // Find out whether the last index in the source GEP is a sequential idx. | ||||
10147 | bool EndsWithSequential = false; | ||||
10148 | for (gep_type_iterator I = gep_type_begin(*cast<User>(PtrOp)), | ||||
10149 | E = gep_type_end(*cast<User>(PtrOp)); I != E; ++I) | ||||
10150 | EndsWithSequential = !isa<StructType>(*I); | ||||
10151 | |||||
10152 | // Can we combine the two pointer arithmetics offsets? | ||||
10153 | if (EndsWithSequential) { | ||||
10154 | // Replace: gep (gep %P, long B), long A, ... | ||||
10155 | // With: T = long A+B; gep %P, T, ... | ||||
10156 | // | ||||
10157 | Value *Sum, *SO1 = SrcGEPOperands.back(), *GO1 = GEP.getOperand(1); | ||||
10158 | if (SO1 == Constant::getNullValue(SO1->getType())) { | ||||
10159 | Sum = GO1; | ||||
10160 | } else if (GO1 == Constant::getNullValue(GO1->getType())) { | ||||
10161 | Sum = SO1; | ||||
10162 | } else { | ||||
10163 | // If they aren't the same type, convert both to an integer of the | ||||
10164 | // target's pointer size. | ||||
10165 | if (SO1->getType() != GO1->getType()) { | ||||
10166 | if (Constant *SO1C = dyn_cast<Constant>(SO1)) { | ||||
10167 | SO1 = ConstantExpr::getIntegerCast(SO1C, GO1->getType(), true); | ||||
10168 | } else if (Constant *GO1C = dyn_cast<Constant>(GO1)) { | ||||
10169 | GO1 = ConstantExpr::getIntegerCast(GO1C, SO1->getType(), true); | ||||
10170 | } else { | ||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 10171 | unsigned PS = TD->getPointerSizeInBits(); |
10172 | if (TD->getTypeSizeInBits(SO1->getType()) == PS) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10173 | // Convert GO1 to SO1's type. |
10174 | GO1 = InsertCastToIntPtrTy(GO1, SO1->getType(), &GEP, this); | ||||
10175 | |||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 10176 | } else if (TD->getTypeSizeInBits(GO1->getType()) == PS) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10177 | // Convert SO1 to GO1's type. |
10178 | SO1 = InsertCastToIntPtrTy(SO1, GO1->getType(), &GEP, this); | ||||
10179 | } else { | ||||
10180 | const Type *PT = TD->getIntPtrType(); | ||||
10181 | SO1 = InsertCastToIntPtrTy(SO1, PT, &GEP, this); | ||||
10182 | GO1 = InsertCastToIntPtrTy(GO1, PT, &GEP, this); | ||||
10183 | } | ||||
10184 | } | ||||
10185 | } | ||||
10186 | if (isa<Constant>(SO1) && isa<Constant>(GO1)) | ||||
10187 | Sum = ConstantExpr::getAdd(cast<Constant>(SO1), cast<Constant>(GO1)); | ||||
10188 | else { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10189 | Sum = BinaryOperator::CreateAdd(SO1, GO1, PtrOp->getName()+".sum"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10190 | InsertNewInstBefore(cast<Instruction>(Sum), GEP); |
10191 | } | ||||
10192 | } | ||||
10193 | |||||
10194 | // Recycle the GEP we already have if possible. | ||||
10195 | if (SrcGEPOperands.size() == 2) { | ||||
10196 | GEP.setOperand(0, SrcGEPOperands[0]); | ||||
10197 | GEP.setOperand(1, Sum); | ||||
10198 | return &GEP; | ||||
10199 | } else { | ||||
10200 | Indices.insert(Indices.end(), SrcGEPOperands.begin()+1, | ||||
10201 | SrcGEPOperands.end()-1); | ||||
10202 | Indices.push_back(Sum); | ||||
10203 | Indices.insert(Indices.end(), GEP.op_begin()+2, GEP.op_end()); | ||||
10204 | } | ||||
10205 | } else if (isa<Constant>(*GEP.idx_begin()) && | ||||
10206 | cast<Constant>(*GEP.idx_begin())->isNullValue() && | ||||
10207 | SrcGEPOperands.size() != 1) { | ||||
10208 | // Otherwise we can do the fold if the first index of the GEP is a zero | ||||
10209 | Indices.insert(Indices.end(), SrcGEPOperands.begin()+1, | ||||
10210 | SrcGEPOperands.end()); | ||||
10211 | Indices.insert(Indices.end(), GEP.idx_begin()+1, GEP.idx_end()); | ||||
10212 | } | ||||
10213 | |||||
10214 | if (!Indices.empty()) | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10215 | return GetElementPtrInst::Create(SrcGEPOperands[0], Indices.begin(), |
10216 | Indices.end(), GEP.getName()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10217 | |
10218 | } else if (GlobalValue *GV = dyn_cast<GlobalValue>(PtrOp)) { | ||||
10219 | // GEP of global variable. If all of the indices for this GEP are | ||||
10220 | // constants, we can promote this to a constexpr instead of an instruction. | ||||
10221 | |||||
10222 | // Scan for nonconstants... | ||||
10223 | SmallVector<Constant*, 8> Indices; | ||||
10224 | User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end(); | ||||
10225 | for (; I != E && isa<Constant>(*I); ++I) | ||||
10226 | Indices.push_back(cast<Constant>(*I)); | ||||
10227 | |||||
10228 | if (I == E) { // If they are all constants... | ||||
10229 | Constant *CE = ConstantExpr::getGetElementPtr(GV, | ||||
10230 | &Indices[0],Indices.size()); | ||||
10231 | |||||
10232 | // Replace all uses of the GEP with the new constexpr... | ||||
10233 | return ReplaceInstUsesWith(GEP, CE); | ||||
10234 | } | ||||
10235 | } else if (Value *X = getBitCastOperand(PtrOp)) { // Is the operand a cast? | ||||
10236 | if (!isa<PointerType>(X->getType())) { | ||||
10237 | // Not interesting. Source pointer must be a cast from pointer. | ||||
10238 | } else if (HasZeroPointerIndex) { | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10239 | // transform: GEP (bitcast [10 x i8]* X to [0 x i8]*), i32 0, ... |
10240 | // into : GEP [10 x i8]* X, i32 0, ... | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10241 | // |
10242 | // This occurs when the program declares an array extern like "int X[];" | ||||
10243 | // | ||||
10244 | const PointerType *CPTy = cast<PointerType>(PtrOp->getType()); | ||||
10245 | const PointerType *XTy = cast<PointerType>(X->getType()); | ||||
10246 | if (const ArrayType *XATy = | ||||
10247 | dyn_cast<ArrayType>(XTy->getElementType())) | ||||
10248 | if (const ArrayType *CATy = | ||||
10249 | dyn_cast<ArrayType>(CPTy->getElementType())) | ||||
10250 | if (CATy->getElementType() == XATy->getElementType()) { | ||||
10251 | // At this point, we know that the cast source type is a pointer | ||||
10252 | // to an array of the same type as the destination pointer | ||||
10253 | // array. Because the array type is never stepped over (there | ||||
10254 | // is a leading zero) we can fold the cast into this GEP. | ||||
10255 | GEP.setOperand(0, X); | ||||
10256 | return &GEP; | ||||
10257 | } | ||||
10258 | } else if (GEP.getNumOperands() == 2) { | ||||
10259 | // Transform things like: | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10260 | // %t = getelementptr i32* bitcast ([2 x i32]* %str to i32*), i32 %V |
10261 | // into: %t1 = getelementptr [2 x i32]* %str, i32 0, i32 %V; bitcast | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10262 | const Type *SrcElTy = cast<PointerType>(X->getType())->getElementType(); |
10263 | const Type *ResElTy=cast<PointerType>(PtrOp->getType())->getElementType(); | ||||
10264 | if (isa<ArrayType>(SrcElTy) && | ||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 10265 | TD->getABITypeSize(cast<ArrayType>(SrcElTy)->getElementType()) == |
10266 | TD->getABITypeSize(ResElTy)) { | ||||
David Greene | 393be88 | 2007-09-04 15:46:09 +0000 | [diff] [blame] | 10267 | Value *Idx[2]; |
10268 | Idx[0] = Constant::getNullValue(Type::Int32Ty); | ||||
10269 | Idx[1] = GEP.getOperand(1); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10270 | Value *V = InsertNewInstBefore( |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10271 | GetElementPtrInst::Create(X, Idx, Idx + 2, GEP.getName()), GEP); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10272 | // V and GEP are both pointer types --> BitCast |
10273 | return new BitCastInst(V, GEP.getType()); | ||||
10274 | } | ||||
10275 | |||||
10276 | // Transform things like: | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10277 | // getelementptr i8* bitcast ([100 x double]* X to i8*), i32 %tmp |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10278 | // (where tmp = 8*tmp2) into: |
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10279 | // getelementptr [100 x double]* %arr, i32 0, i32 %tmp2; bitcast |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10280 | |
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10281 | if (isa<ArrayType>(SrcElTy) && ResElTy == Type::Int8Ty) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10282 | uint64_t ArrayEltSize = |
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 10283 | TD->getABITypeSize(cast<ArrayType>(SrcElTy)->getElementType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10284 | |
10285 | // Check to see if "tmp" is a scale by a multiple of ArrayEltSize. We | ||||
10286 | // allow either a mul, shift, or constant here. | ||||
10287 | Value *NewIdx = 0; | ||||
10288 | ConstantInt *Scale = 0; | ||||
10289 | if (ArrayEltSize == 1) { | ||||
10290 | NewIdx = GEP.getOperand(1); | ||||
10291 | Scale = ConstantInt::get(NewIdx->getType(), 1); | ||||
10292 | } else if (ConstantInt *CI = dyn_cast<ConstantInt>(GEP.getOperand(1))) { | ||||
10293 | NewIdx = ConstantInt::get(CI->getType(), 1); | ||||
10294 | Scale = CI; | ||||
10295 | } else if (Instruction *Inst =dyn_cast<Instruction>(GEP.getOperand(1))){ | ||||
10296 | if (Inst->getOpcode() == Instruction::Shl && | ||||
10297 | isa<ConstantInt>(Inst->getOperand(1))) { | ||||
10298 | ConstantInt *ShAmt = cast<ConstantInt>(Inst->getOperand(1)); | ||||
10299 | uint32_t ShAmtVal = ShAmt->getLimitedValue(64); | ||||
10300 | Scale = ConstantInt::get(Inst->getType(), 1ULL << ShAmtVal); | ||||
10301 | NewIdx = Inst->getOperand(0); | ||||
10302 | } else if (Inst->getOpcode() == Instruction::Mul && | ||||
10303 | isa<ConstantInt>(Inst->getOperand(1))) { | ||||
10304 | Scale = cast<ConstantInt>(Inst->getOperand(1)); | ||||
10305 | NewIdx = Inst->getOperand(0); | ||||
10306 | } | ||||
10307 | } | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10308 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10309 | // 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] | 10310 | // out, perform the transformation. Note, we don't know whether Scale is |
10311 | // signed or not. We'll use unsigned version of division/modulo | ||||
10312 | // operation after making sure Scale doesn't have the sign bit set. | ||||
10313 | if (Scale && Scale->getSExtValue() >= 0LL && | ||||
10314 | Scale->getZExtValue() % ArrayEltSize == 0) { | ||||
10315 | Scale = ConstantInt::get(Scale->getType(), | ||||
10316 | Scale->getZExtValue() / ArrayEltSize); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10317 | if (Scale->getZExtValue() != 1) { |
10318 | Constant *C = ConstantExpr::getIntegerCast(Scale, NewIdx->getType(), | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10319 | false /*ZExt*/); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10320 | Instruction *Sc = BinaryOperator::CreateMul(NewIdx, C, "idxscale"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10321 | NewIdx = InsertNewInstBefore(Sc, GEP); |
10322 | } | ||||
10323 | |||||
10324 | // Insert the new GEP instruction. | ||||
David Greene | 393be88 | 2007-09-04 15:46:09 +0000 | [diff] [blame] | 10325 | Value *Idx[2]; |
10326 | Idx[0] = Constant::getNullValue(Type::Int32Ty); | ||||
10327 | Idx[1] = NewIdx; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10328 | Instruction *NewGEP = |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10329 | GetElementPtrInst::Create(X, Idx, Idx + 2, GEP.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10330 | NewGEP = InsertNewInstBefore(NewGEP, GEP); |
10331 | // The NewGEP must be pointer typed, so must the old one -> BitCast | ||||
10332 | return new BitCastInst(NewGEP, GEP.getType()); | ||||
10333 | } | ||||
10334 | } | ||||
10335 | } | ||||
10336 | } | ||||
10337 | |||||
10338 | return 0; | ||||
10339 | } | ||||
10340 | |||||
10341 | Instruction *InstCombiner::visitAllocationInst(AllocationInst &AI) { | ||||
10342 | // 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] | 10343 | if (AI.isArrayAllocation()) { // Check C != 1 |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10344 | if (const ConstantInt *C = dyn_cast<ConstantInt>(AI.getArraySize())) { |
10345 | const Type *NewTy = | ||||
10346 | ArrayType::get(AI.getAllocatedType(), C->getZExtValue()); | ||||
10347 | AllocationInst *New = 0; | ||||
10348 | |||||
10349 | // Create and insert the replacement instruction... | ||||
10350 | if (isa<MallocInst>(AI)) | ||||
10351 | New = new MallocInst(NewTy, 0, AI.getAlignment(), AI.getName()); | ||||
10352 | else { | ||||
10353 | assert(isa<AllocaInst>(AI) && "Unknown type of allocation inst!"); | ||||
10354 | New = new AllocaInst(NewTy, 0, AI.getAlignment(), AI.getName()); | ||||
10355 | } | ||||
10356 | |||||
10357 | InsertNewInstBefore(New, AI); | ||||
10358 | |||||
10359 | // Scan to the end of the allocation instructions, to skip over a block of | ||||
10360 | // allocas if possible... | ||||
10361 | // | ||||
10362 | BasicBlock::iterator It = New; | ||||
10363 | while (isa<AllocationInst>(*It)) ++It; | ||||
10364 | |||||
10365 | // Now that I is pointing to the first non-allocation-inst in the block, | ||||
10366 | // insert our getelementptr instruction... | ||||
10367 | // | ||||
10368 | Value *NullIdx = Constant::getNullValue(Type::Int32Ty); | ||||
David Greene | 393be88 | 2007-09-04 15:46:09 +0000 | [diff] [blame] | 10369 | Value *Idx[2]; |
10370 | Idx[0] = NullIdx; | ||||
10371 | Idx[1] = NullIdx; | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10372 | Value *V = GetElementPtrInst::Create(New, Idx, Idx + 2, |
10373 | New->getName()+".sub", It); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10374 | |
10375 | // Now make everything use the getelementptr instead of the original | ||||
10376 | // allocation. | ||||
10377 | return ReplaceInstUsesWith(AI, V); | ||||
10378 | } else if (isa<UndefValue>(AI.getArraySize())) { | ||||
10379 | return ReplaceInstUsesWith(AI, Constant::getNullValue(AI.getType())); | ||||
10380 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10381 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10382 | |
10383 | // If alloca'ing a zero byte object, replace the alloca with a null pointer. | ||||
10384 | // Note that we only do this for alloca's, because malloc should allocate and | ||||
10385 | // return a unique pointer, even for a zero byte allocation. | ||||
10386 | if (isa<AllocaInst>(AI) && AI.getAllocatedType()->isSized() && | ||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 10387 | TD->getABITypeSize(AI.getAllocatedType()) == 0) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10388 | return ReplaceInstUsesWith(AI, Constant::getNullValue(AI.getType())); |
10389 | |||||
10390 | return 0; | ||||
10391 | } | ||||
10392 | |||||
10393 | Instruction *InstCombiner::visitFreeInst(FreeInst &FI) { | ||||
10394 | Value *Op = FI.getOperand(0); | ||||
10395 | |||||
10396 | // free undef -> unreachable. | ||||
10397 | if (isa<UndefValue>(Op)) { | ||||
10398 | // Insert a new store to null because we cannot modify the CFG here. | ||||
10399 | new StoreInst(ConstantInt::getTrue(), | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 10400 | UndefValue::get(PointerType::getUnqual(Type::Int1Ty)), &FI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10401 | return EraseInstFromFunction(FI); |
10402 | } | ||||
10403 | |||||
10404 | // If we have 'free null' delete the instruction. This can happen in stl code | ||||
10405 | // when lots of inlining happens. | ||||
10406 | if (isa<ConstantPointerNull>(Op)) | ||||
10407 | return EraseInstFromFunction(FI); | ||||
10408 | |||||
10409 | // Change free <ty>* (cast <ty2>* X to <ty>*) into free <ty2>* X | ||||
10410 | if (BitCastInst *CI = dyn_cast<BitCastInst>(Op)) { | ||||
10411 | FI.setOperand(0, CI->getOperand(0)); | ||||
10412 | return &FI; | ||||
10413 | } | ||||
10414 | |||||
10415 | // Change free (gep X, 0,0,0,0) into free(X) | ||||
10416 | if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(Op)) { | ||||
10417 | if (GEPI->hasAllZeroIndices()) { | ||||
10418 | AddToWorkList(GEPI); | ||||
10419 | FI.setOperand(0, GEPI->getOperand(0)); | ||||
10420 | return &FI; | ||||
10421 | } | ||||
10422 | } | ||||
10423 | |||||
10424 | // Change free(malloc) into nothing, if the malloc has a single use. | ||||
10425 | if (MallocInst *MI = dyn_cast<MallocInst>(Op)) | ||||
10426 | if (MI->hasOneUse()) { | ||||
10427 | EraseInstFromFunction(FI); | ||||
10428 | return EraseInstFromFunction(*MI); | ||||
10429 | } | ||||
10430 | |||||
10431 | return 0; | ||||
10432 | } | ||||
10433 | |||||
10434 | |||||
10435 | /// InstCombineLoadCast - Fold 'load (cast P)' -> cast (load P)' when possible. | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 10436 | static Instruction *InstCombineLoadCast(InstCombiner &IC, LoadInst &LI, |
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 10437 | const TargetData *TD) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10438 | User *CI = cast<User>(LI.getOperand(0)); |
10439 | Value *CastOp = CI->getOperand(0); | ||||
10440 | |||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 10441 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(CI)) { |
10442 | // Instead of loading constant c string, use corresponding integer value | ||||
10443 | // directly if string length is small enough. | ||||
Evan Cheng | 833501d | 2008-06-30 07:31:25 +0000 | [diff] [blame] | 10444 | std::string Str; |
10445 | if (GetConstantStringInfo(CE->getOperand(0), Str) && !Str.empty()) { | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 10446 | unsigned len = Str.length(); |
10447 | const Type *Ty = cast<PointerType>(CE->getType())->getElementType(); | ||||
10448 | unsigned numBits = Ty->getPrimitiveSizeInBits(); | ||||
10449 | // Replace LI with immediate integer store. | ||||
10450 | if ((numBits >> 3) == len + 1) { | ||||
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 10451 | APInt StrVal(numBits, 0); |
10452 | APInt SingleChar(numBits, 0); | ||||
10453 | if (TD->isLittleEndian()) { | ||||
10454 | for (signed i = len-1; i >= 0; i--) { | ||||
10455 | SingleChar = (uint64_t) Str[i]; | ||||
10456 | StrVal = (StrVal << 8) | SingleChar; | ||||
10457 | } | ||||
10458 | } else { | ||||
10459 | for (unsigned i = 0; i < len; i++) { | ||||
10460 | SingleChar = (uint64_t) Str[i]; | ||||
10461 | StrVal = (StrVal << 8) | SingleChar; | ||||
10462 | } | ||||
10463 | // Append NULL at the end. | ||||
10464 | SingleChar = 0; | ||||
10465 | StrVal = (StrVal << 8) | SingleChar; | ||||
10466 | } | ||||
10467 | Value *NL = ConstantInt::get(StrVal); | ||||
10468 | return IC.ReplaceInstUsesWith(LI, NL); | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 10469 | } |
10470 | } | ||||
10471 | } | ||||
10472 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10473 | const Type *DestPTy = cast<PointerType>(CI->getType())->getElementType(); |
10474 | if (const PointerType *SrcTy = dyn_cast<PointerType>(CastOp->getType())) { | ||||
10475 | const Type *SrcPTy = SrcTy->getElementType(); | ||||
10476 | |||||
10477 | if (DestPTy->isInteger() || isa<PointerType>(DestPTy) || | ||||
10478 | isa<VectorType>(DestPTy)) { | ||||
10479 | // If the source is an array, the code below will not succeed. Check to | ||||
10480 | // see if a trivial 'gep P, 0, 0' will help matters. Only do this for | ||||
10481 | // constants. | ||||
10482 | if (const ArrayType *ASrcTy = dyn_cast<ArrayType>(SrcPTy)) | ||||
10483 | if (Constant *CSrc = dyn_cast<Constant>(CastOp)) | ||||
10484 | if (ASrcTy->getNumElements() != 0) { | ||||
10485 | Value *Idxs[2]; | ||||
10486 | Idxs[0] = Idxs[1] = Constant::getNullValue(Type::Int32Ty); | ||||
10487 | CastOp = ConstantExpr::getGetElementPtr(CSrc, Idxs, 2); | ||||
10488 | SrcTy = cast<PointerType>(CastOp->getType()); | ||||
10489 | SrcPTy = SrcTy->getElementType(); | ||||
10490 | } | ||||
10491 | |||||
10492 | if ((SrcPTy->isInteger() || isa<PointerType>(SrcPTy) || | ||||
10493 | isa<VectorType>(SrcPTy)) && | ||||
10494 | // Do not allow turning this into a load of an integer, which is then | ||||
10495 | // casted to a pointer, this pessimizes pointer analysis a lot. | ||||
10496 | (isa<PointerType>(SrcPTy) == isa<PointerType>(LI.getType())) && | ||||
10497 | IC.getTargetData().getTypeSizeInBits(SrcPTy) == | ||||
10498 | IC.getTargetData().getTypeSizeInBits(DestPTy)) { | ||||
10499 | |||||
10500 | // Okay, we are casting from one integer or pointer type to another of | ||||
10501 | // the same size. Instead of casting the pointer before the load, cast | ||||
10502 | // the result of the loaded value. | ||||
10503 | Value *NewLoad = IC.InsertNewInstBefore(new LoadInst(CastOp, | ||||
10504 | CI->getName(), | ||||
10505 | LI.isVolatile()),LI); | ||||
10506 | // Now cast the result of the load. | ||||
10507 | return new BitCastInst(NewLoad, LI.getType()); | ||||
10508 | } | ||||
10509 | } | ||||
10510 | } | ||||
10511 | return 0; | ||||
10512 | } | ||||
10513 | |||||
10514 | /// isSafeToLoadUnconditionally - Return true if we know that executing a load | ||||
10515 | /// from this value cannot trap. If it is not obviously safe to load from the | ||||
10516 | /// specified pointer, we do a quick local scan of the basic block containing | ||||
10517 | /// ScanFrom, to determine if the address is already accessed. | ||||
10518 | static bool isSafeToLoadUnconditionally(Value *V, Instruction *ScanFrom) { | ||||
Duncan Sands | 9b27dbe | 2007-09-19 10:10:31 +0000 | [diff] [blame] | 10519 | // If it is an alloca it is always safe to load from. |
10520 | if (isa<AllocaInst>(V)) return true; | ||||
10521 | |||||
Duncan Sands | e40a94a | 2007-09-19 10:25:38 +0000 | [diff] [blame] | 10522 | // 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] | 10523 | if (const GlobalValue *GV = dyn_cast<GlobalVariable>(V)) |
Duncan Sands | e40a94a | 2007-09-19 10:25:38 +0000 | [diff] [blame] | 10524 | // Don't try to evaluate aliases. External weak GV can be null. |
Duncan Sands | 9b27dbe | 2007-09-19 10:10:31 +0000 | [diff] [blame] | 10525 | return !isa<GlobalAlias>(GV) && !GV->hasExternalWeakLinkage(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10526 | |
10527 | // Otherwise, be a little bit agressive by scanning the local block where we | ||||
10528 | // want to check to see if the pointer is already being loaded or stored | ||||
10529 | // from/to. If so, the previous load or store would have already trapped, | ||||
10530 | // so there is no harm doing an extra load (also, CSE will later eliminate | ||||
10531 | // the load entirely). | ||||
10532 | BasicBlock::iterator BBI = ScanFrom, E = ScanFrom->getParent()->begin(); | ||||
10533 | |||||
10534 | while (BBI != E) { | ||||
10535 | --BBI; | ||||
10536 | |||||
Chris Lattner | 476983a | 2008-06-20 05:12:56 +0000 | [diff] [blame] | 10537 | // If we see a free or a call (which might do a free) the pointer could be |
10538 | // marked invalid. | ||||
10539 | if (isa<FreeInst>(BBI) || isa<CallInst>(BBI)) | ||||
10540 | return false; | ||||
10541 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10542 | if (LoadInst *LI = dyn_cast<LoadInst>(BBI)) { |
10543 | if (LI->getOperand(0) == V) return true; | ||||
Chris Lattner | 476983a | 2008-06-20 05:12:56 +0000 | [diff] [blame] | 10544 | } else if (StoreInst *SI = dyn_cast<StoreInst>(BBI)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10545 | if (SI->getOperand(1) == V) return true; |
Chris Lattner | 476983a | 2008-06-20 05:12:56 +0000 | [diff] [blame] | 10546 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10547 | |
10548 | } | ||||
10549 | return false; | ||||
10550 | } | ||||
10551 | |||||
10552 | Instruction *InstCombiner::visitLoadInst(LoadInst &LI) { | ||||
10553 | Value *Op = LI.getOperand(0); | ||||
10554 | |||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 10555 | // Attempt to improve the alignment. |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 10556 | unsigned KnownAlign = GetOrEnforceKnownAlignment(Op); |
10557 | if (KnownAlign > | ||||
10558 | (LI.getAlignment() == 0 ? TD->getABITypeAlignment(LI.getType()) : | ||||
10559 | LI.getAlignment())) | ||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 10560 | LI.setAlignment(KnownAlign); |
10561 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10562 | // load (cast X) --> cast (load X) iff safe |
10563 | if (isa<CastInst>(Op)) | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 10564 | if (Instruction *Res = InstCombineLoadCast(*this, LI, TD)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10565 | return Res; |
10566 | |||||
10567 | // None of the following transforms are legal for volatile loads. | ||||
10568 | if (LI.isVolatile()) return 0; | ||||
10569 | |||||
10570 | if (&LI.getParent()->front() != &LI) { | ||||
10571 | BasicBlock::iterator BBI = &LI; --BBI; | ||||
10572 | // If the instruction immediately before this is a store to the same | ||||
10573 | // address, do a simple form of store->load forwarding. | ||||
10574 | if (StoreInst *SI = dyn_cast<StoreInst>(BBI)) | ||||
10575 | if (SI->getOperand(1) == LI.getOperand(0)) | ||||
10576 | return ReplaceInstUsesWith(LI, SI->getOperand(0)); | ||||
10577 | if (LoadInst *LIB = dyn_cast<LoadInst>(BBI)) | ||||
10578 | if (LIB->getOperand(0) == LI.getOperand(0)) | ||||
10579 | return ReplaceInstUsesWith(LI, LIB); | ||||
10580 | } | ||||
10581 | |||||
Christopher Lamb | 2c17539 | 2007-12-29 07:56:53 +0000 | [diff] [blame] | 10582 | if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(Op)) { |
10583 | const Value *GEPI0 = GEPI->getOperand(0); | ||||
10584 | // TODO: Consider a target hook for valid address spaces for this xform. | ||||
10585 | if (isa<ConstantPointerNull>(GEPI0) && | ||||
10586 | cast<PointerType>(GEPI0->getType())->getAddressSpace() == 0) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10587 | // Insert a new store to null instruction before the load to indicate |
10588 | // that this code is not reachable. We do this instead of inserting | ||||
10589 | // an unreachable instruction directly because we cannot modify the | ||||
10590 | // CFG. | ||||
10591 | new StoreInst(UndefValue::get(LI.getType()), | ||||
10592 | Constant::getNullValue(Op->getType()), &LI); | ||||
10593 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
10594 | } | ||||
Christopher Lamb | 2c17539 | 2007-12-29 07:56:53 +0000 | [diff] [blame] | 10595 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10596 | |
10597 | if (Constant *C = dyn_cast<Constant>(Op)) { | ||||
10598 | // load null/undef -> undef | ||||
Christopher Lamb | 2c17539 | 2007-12-29 07:56:53 +0000 | [diff] [blame] | 10599 | // TODO: Consider a target hook for valid address spaces for this xform. |
10600 | if (isa<UndefValue>(C) || (C->isNullValue() && | ||||
10601 | cast<PointerType>(Op->getType())->getAddressSpace() == 0)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10602 | // Insert a new store to null instruction before the load to indicate that |
10603 | // this code is not reachable. We do this instead of inserting an | ||||
10604 | // unreachable instruction directly because we cannot modify the CFG. | ||||
10605 | new StoreInst(UndefValue::get(LI.getType()), | ||||
10606 | Constant::getNullValue(Op->getType()), &LI); | ||||
10607 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
10608 | } | ||||
10609 | |||||
10610 | // Instcombine load (constant global) into the value loaded. | ||||
10611 | if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op)) | ||||
10612 | if (GV->isConstant() && !GV->isDeclaration()) | ||||
10613 | return ReplaceInstUsesWith(LI, GV->getInitializer()); | ||||
10614 | |||||
10615 | // Instcombine load (constantexpr_GEP global, 0, ...) into the value loaded. | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10616 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Op)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10617 | if (CE->getOpcode() == Instruction::GetElementPtr) { |
10618 | if (GlobalVariable *GV = dyn_cast<GlobalVariable>(CE->getOperand(0))) | ||||
10619 | if (GV->isConstant() && !GV->isDeclaration()) | ||||
10620 | if (Constant *V = | ||||
10621 | ConstantFoldLoadThroughGEPConstantExpr(GV->getInitializer(), CE)) | ||||
10622 | return ReplaceInstUsesWith(LI, V); | ||||
10623 | if (CE->getOperand(0)->isNullValue()) { | ||||
10624 | // Insert a new store to null instruction before the load to indicate | ||||
10625 | // that this code is not reachable. We do this instead of inserting | ||||
10626 | // an unreachable instruction directly because we cannot modify the | ||||
10627 | // CFG. | ||||
10628 | new StoreInst(UndefValue::get(LI.getType()), | ||||
10629 | Constant::getNullValue(Op->getType()), &LI); | ||||
10630 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
10631 | } | ||||
10632 | |||||
10633 | } else if (CE->isCast()) { | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 10634 | if (Instruction *Res = InstCombineLoadCast(*this, LI, TD)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10635 | return Res; |
10636 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10637 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10638 | } |
Chris Lattner | 0270a11 | 2007-08-11 18:48:48 +0000 | [diff] [blame] | 10639 | |
10640 | // If this load comes from anywhere in a constant global, and if the global | ||||
10641 | // is all undef or zero, we know what it loads. | ||||
Duncan Sands | 52fb873 | 2008-10-01 15:25:41 +0000 | [diff] [blame] | 10642 | if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op->getUnderlyingObject())){ |
Chris Lattner | 0270a11 | 2007-08-11 18:48:48 +0000 | [diff] [blame] | 10643 | if (GV->isConstant() && GV->hasInitializer()) { |
10644 | if (GV->getInitializer()->isNullValue()) | ||||
10645 | return ReplaceInstUsesWith(LI, Constant::getNullValue(LI.getType())); | ||||
10646 | else if (isa<UndefValue>(GV->getInitializer())) | ||||
10647 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
10648 | } | ||||
10649 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10650 | |
10651 | if (Op->hasOneUse()) { | ||||
10652 | // Change select and PHI nodes to select values instead of addresses: this | ||||
10653 | // helps alias analysis out a lot, allows many others simplifications, and | ||||
10654 | // exposes redundancy in the code. | ||||
10655 | // | ||||
10656 | // Note that we cannot do the transformation unless we know that the | ||||
10657 | // introduced loads cannot trap! Something like this is valid as long as | ||||
10658 | // the condition is always false: load (select bool %C, int* null, int* %G), | ||||
10659 | // but it would not be valid if we transformed it to load from null | ||||
10660 | // unconditionally. | ||||
10661 | // | ||||
10662 | if (SelectInst *SI = dyn_cast<SelectInst>(Op)) { | ||||
10663 | // load (select (Cond, &V1, &V2)) --> select(Cond, load &V1, load &V2). | ||||
10664 | if (isSafeToLoadUnconditionally(SI->getOperand(1), SI) && | ||||
10665 | isSafeToLoadUnconditionally(SI->getOperand(2), SI)) { | ||||
10666 | Value *V1 = InsertNewInstBefore(new LoadInst(SI->getOperand(1), | ||||
10667 | SI->getOperand(1)->getName()+".val"), LI); | ||||
10668 | Value *V2 = InsertNewInstBefore(new LoadInst(SI->getOperand(2), | ||||
10669 | SI->getOperand(2)->getName()+".val"), LI); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10670 | return SelectInst::Create(SI->getCondition(), V1, V2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10671 | } |
10672 | |||||
10673 | // load (select (cond, null, P)) -> load P | ||||
10674 | if (Constant *C = dyn_cast<Constant>(SI->getOperand(1))) | ||||
10675 | if (C->isNullValue()) { | ||||
10676 | LI.setOperand(0, SI->getOperand(2)); | ||||
10677 | return &LI; | ||||
10678 | } | ||||
10679 | |||||
10680 | // load (select (cond, P, null)) -> load P | ||||
10681 | if (Constant *C = dyn_cast<Constant>(SI->getOperand(2))) | ||||
10682 | if (C->isNullValue()) { | ||||
10683 | LI.setOperand(0, SI->getOperand(1)); | ||||
10684 | return &LI; | ||||
10685 | } | ||||
10686 | } | ||||
10687 | } | ||||
10688 | return 0; | ||||
10689 | } | ||||
10690 | |||||
10691 | /// InstCombineStoreToCast - Fold store V, (cast P) -> store (cast V), P | ||||
10692 | /// when possible. | ||||
10693 | static Instruction *InstCombineStoreToCast(InstCombiner &IC, StoreInst &SI) { | ||||
10694 | User *CI = cast<User>(SI.getOperand(1)); | ||||
10695 | Value *CastOp = CI->getOperand(0); | ||||
10696 | |||||
10697 | const Type *DestPTy = cast<PointerType>(CI->getType())->getElementType(); | ||||
10698 | if (const PointerType *SrcTy = dyn_cast<PointerType>(CastOp->getType())) { | ||||
10699 | const Type *SrcPTy = SrcTy->getElementType(); | ||||
10700 | |||||
10701 | if (DestPTy->isInteger() || isa<PointerType>(DestPTy)) { | ||||
10702 | // If the source is an array, the code below will not succeed. Check to | ||||
10703 | // see if a trivial 'gep P, 0, 0' will help matters. Only do this for | ||||
10704 | // constants. | ||||
10705 | if (const ArrayType *ASrcTy = dyn_cast<ArrayType>(SrcPTy)) | ||||
10706 | if (Constant *CSrc = dyn_cast<Constant>(CastOp)) | ||||
10707 | if (ASrcTy->getNumElements() != 0) { | ||||
10708 | Value* Idxs[2]; | ||||
10709 | Idxs[0] = Idxs[1] = Constant::getNullValue(Type::Int32Ty); | ||||
10710 | CastOp = ConstantExpr::getGetElementPtr(CSrc, Idxs, 2); | ||||
10711 | SrcTy = cast<PointerType>(CastOp->getType()); | ||||
10712 | SrcPTy = SrcTy->getElementType(); | ||||
10713 | } | ||||
10714 | |||||
10715 | if ((SrcPTy->isInteger() || isa<PointerType>(SrcPTy)) && | ||||
10716 | IC.getTargetData().getTypeSizeInBits(SrcPTy) == | ||||
10717 | IC.getTargetData().getTypeSizeInBits(DestPTy)) { | ||||
10718 | |||||
10719 | // Okay, we are casting from one integer or pointer type to another of | ||||
10720 | // the same size. Instead of casting the pointer before | ||||
10721 | // the store, cast the value to be stored. | ||||
10722 | Value *NewCast; | ||||
10723 | Value *SIOp0 = SI.getOperand(0); | ||||
10724 | Instruction::CastOps opcode = Instruction::BitCast; | ||||
10725 | const Type* CastSrcTy = SIOp0->getType(); | ||||
10726 | const Type* CastDstTy = SrcPTy; | ||||
10727 | if (isa<PointerType>(CastDstTy)) { | ||||
10728 | if (CastSrcTy->isInteger()) | ||||
10729 | opcode = Instruction::IntToPtr; | ||||
10730 | } else if (isa<IntegerType>(CastDstTy)) { | ||||
10731 | if (isa<PointerType>(SIOp0->getType())) | ||||
10732 | opcode = Instruction::PtrToInt; | ||||
10733 | } | ||||
10734 | if (Constant *C = dyn_cast<Constant>(SIOp0)) | ||||
10735 | NewCast = ConstantExpr::getCast(opcode, C, CastDstTy); | ||||
10736 | else | ||||
10737 | NewCast = IC.InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10738 | CastInst::Create(opcode, SIOp0, CastDstTy, SIOp0->getName()+".c"), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10739 | SI); |
10740 | return new StoreInst(NewCast, CastOp); | ||||
10741 | } | ||||
10742 | } | ||||
10743 | } | ||||
10744 | return 0; | ||||
10745 | } | ||||
10746 | |||||
10747 | Instruction *InstCombiner::visitStoreInst(StoreInst &SI) { | ||||
10748 | Value *Val = SI.getOperand(0); | ||||
10749 | Value *Ptr = SI.getOperand(1); | ||||
10750 | |||||
10751 | if (isa<UndefValue>(Ptr)) { // store X, undef -> noop (even if volatile) | ||||
10752 | EraseInstFromFunction(SI); | ||||
10753 | ++NumCombined; | ||||
10754 | return 0; | ||||
10755 | } | ||||
10756 | |||||
10757 | // If the RHS is an alloca with a single use, zapify the store, making the | ||||
10758 | // alloca dead. | ||||
Chris Lattner | a02bacc | 2008-04-29 04:58:38 +0000 | [diff] [blame] | 10759 | if (Ptr->hasOneUse() && !SI.isVolatile()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10760 | if (isa<AllocaInst>(Ptr)) { |
10761 | EraseInstFromFunction(SI); | ||||
10762 | ++NumCombined; | ||||
10763 | return 0; | ||||
10764 | } | ||||
10765 | |||||
10766 | if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Ptr)) | ||||
10767 | if (isa<AllocaInst>(GEP->getOperand(0)) && | ||||
10768 | GEP->getOperand(0)->hasOneUse()) { | ||||
10769 | EraseInstFromFunction(SI); | ||||
10770 | ++NumCombined; | ||||
10771 | return 0; | ||||
10772 | } | ||||
10773 | } | ||||
10774 | |||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 10775 | // Attempt to improve the alignment. |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 10776 | unsigned KnownAlign = GetOrEnforceKnownAlignment(Ptr); |
10777 | if (KnownAlign > | ||||
10778 | (SI.getAlignment() == 0 ? TD->getABITypeAlignment(Val->getType()) : | ||||
10779 | SI.getAlignment())) | ||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 10780 | SI.setAlignment(KnownAlign); |
10781 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10782 | // Do really simple DSE, to catch cases where there are several consequtive |
10783 | // stores to the same location, separated by a few arithmetic operations. This | ||||
10784 | // situation often occurs with bitfield accesses. | ||||
10785 | BasicBlock::iterator BBI = &SI; | ||||
10786 | for (unsigned ScanInsts = 6; BBI != SI.getParent()->begin() && ScanInsts; | ||||
10787 | --ScanInsts) { | ||||
10788 | --BBI; | ||||
10789 | |||||
10790 | if (StoreInst *PrevSI = dyn_cast<StoreInst>(BBI)) { | ||||
10791 | // Prev store isn't volatile, and stores to the same location? | ||||
10792 | if (!PrevSI->isVolatile() && PrevSI->getOperand(1) == SI.getOperand(1)) { | ||||
10793 | ++NumDeadStore; | ||||
10794 | ++BBI; | ||||
10795 | EraseInstFromFunction(*PrevSI); | ||||
10796 | continue; | ||||
10797 | } | ||||
10798 | break; | ||||
10799 | } | ||||
10800 | |||||
10801 | // If this is a load, we have to stop. However, if the loaded value is from | ||||
10802 | // the pointer we're loading and is producing the pointer we're storing, | ||||
10803 | // then *this* store is dead (X = load P; store X -> P). | ||||
10804 | if (LoadInst *LI = dyn_cast<LoadInst>(BBI)) { | ||||
Chris Lattner | 24905f7 | 2007-09-07 05:33:03 +0000 | [diff] [blame] | 10805 | if (LI == Val && LI->getOperand(0) == Ptr && !SI.isVolatile()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10806 | EraseInstFromFunction(SI); |
10807 | ++NumCombined; | ||||
10808 | return 0; | ||||
10809 | } | ||||
10810 | // Otherwise, this is a load from some other location. Stores before it | ||||
10811 | // may not be dead. | ||||
10812 | break; | ||||
10813 | } | ||||
10814 | |||||
10815 | // Don't skip over loads or things that can modify memory. | ||||
Chris Lattner | 8450428 | 2008-05-08 17:20:30 +0000 | [diff] [blame] | 10816 | if (BBI->mayWriteToMemory() || BBI->mayReadFromMemory()) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10817 | break; |
10818 | } | ||||
10819 | |||||
10820 | |||||
10821 | if (SI.isVolatile()) return 0; // Don't hack volatile stores. | ||||
10822 | |||||
10823 | // store X, null -> turns into 'unreachable' in SimplifyCFG | ||||
10824 | if (isa<ConstantPointerNull>(Ptr)) { | ||||
10825 | if (!isa<UndefValue>(Val)) { | ||||
10826 | SI.setOperand(0, UndefValue::get(Val->getType())); | ||||
10827 | if (Instruction *U = dyn_cast<Instruction>(Val)) | ||||
10828 | AddToWorkList(U); // Dropped a use. | ||||
10829 | ++NumCombined; | ||||
10830 | } | ||||
10831 | return 0; // Do not modify these! | ||||
10832 | } | ||||
10833 | |||||
10834 | // store undef, Ptr -> noop | ||||
10835 | if (isa<UndefValue>(Val)) { | ||||
10836 | EraseInstFromFunction(SI); | ||||
10837 | ++NumCombined; | ||||
10838 | return 0; | ||||
10839 | } | ||||
10840 | |||||
10841 | // If the pointer destination is a cast, see if we can fold the cast into the | ||||
10842 | // source instead. | ||||
10843 | if (isa<CastInst>(Ptr)) | ||||
10844 | if (Instruction *Res = InstCombineStoreToCast(*this, SI)) | ||||
10845 | return Res; | ||||
10846 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr)) | ||||
10847 | if (CE->isCast()) | ||||
10848 | if (Instruction *Res = InstCombineStoreToCast(*this, SI)) | ||||
10849 | return Res; | ||||
10850 | |||||
10851 | |||||
10852 | // If this store is the last instruction in the basic block, and if the block | ||||
10853 | // ends with an unconditional branch, try to move it to the successor block. | ||||
10854 | BBI = &SI; ++BBI; | ||||
10855 | if (BranchInst *BI = dyn_cast<BranchInst>(BBI)) | ||||
10856 | if (BI->isUnconditional()) | ||||
10857 | if (SimplifyStoreAtEndOfBlock(SI)) | ||||
10858 | return 0; // xform done! | ||||
10859 | |||||
10860 | return 0; | ||||
10861 | } | ||||
10862 | |||||
10863 | /// SimplifyStoreAtEndOfBlock - Turn things like: | ||||
10864 | /// if () { *P = v1; } else { *P = v2 } | ||||
10865 | /// into a phi node with a store in the successor. | ||||
10866 | /// | ||||
10867 | /// Simplify things like: | ||||
10868 | /// *P = v1; if () { *P = v2; } | ||||
10869 | /// into a phi node with a store in the successor. | ||||
10870 | /// | ||||
10871 | bool InstCombiner::SimplifyStoreAtEndOfBlock(StoreInst &SI) { | ||||
10872 | BasicBlock *StoreBB = SI.getParent(); | ||||
10873 | |||||
10874 | // Check to see if the successor block has exactly two incoming edges. If | ||||
10875 | // so, see if the other predecessor contains a store to the same location. | ||||
10876 | // if so, insert a PHI node (if needed) and move the stores down. | ||||
10877 | BasicBlock *DestBB = StoreBB->getTerminator()->getSuccessor(0); | ||||
10878 | |||||
10879 | // Determine whether Dest has exactly two predecessors and, if so, compute | ||||
10880 | // the other predecessor. | ||||
10881 | pred_iterator PI = pred_begin(DestBB); | ||||
10882 | BasicBlock *OtherBB = 0; | ||||
10883 | if (*PI != StoreBB) | ||||
10884 | OtherBB = *PI; | ||||
10885 | ++PI; | ||||
10886 | if (PI == pred_end(DestBB)) | ||||
10887 | return false; | ||||
10888 | |||||
10889 | if (*PI != StoreBB) { | ||||
10890 | if (OtherBB) | ||||
10891 | return false; | ||||
10892 | OtherBB = *PI; | ||||
10893 | } | ||||
10894 | if (++PI != pred_end(DestBB)) | ||||
10895 | return false; | ||||
Eli Friedman | ab39f9a | 2008-06-13 21:17:49 +0000 | [diff] [blame] | 10896 | |
10897 | // Bail out if all the relevant blocks aren't distinct (this can happen, | ||||
10898 | // for example, if SI is in an infinite loop) | ||||
10899 | if (StoreBB == DestBB || OtherBB == DestBB) | ||||
10900 | return false; | ||||
10901 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10902 | // Verify that the other block ends in a branch and is not otherwise empty. |
10903 | BasicBlock::iterator BBI = OtherBB->getTerminator(); | ||||
10904 | BranchInst *OtherBr = dyn_cast<BranchInst>(BBI); | ||||
10905 | if (!OtherBr || BBI == OtherBB->begin()) | ||||
10906 | return false; | ||||
10907 | |||||
10908 | // If the other block ends in an unconditional branch, check for the 'if then | ||||
10909 | // else' case. there is an instruction before the branch. | ||||
10910 | StoreInst *OtherStore = 0; | ||||
10911 | if (OtherBr->isUnconditional()) { | ||||
10912 | // If this isn't a store, or isn't a store to the same location, bail out. | ||||
10913 | --BBI; | ||||
10914 | OtherStore = dyn_cast<StoreInst>(BBI); | ||||
10915 | if (!OtherStore || OtherStore->getOperand(1) != SI.getOperand(1)) | ||||
10916 | return false; | ||||
10917 | } else { | ||||
10918 | // Otherwise, the other block ended with a conditional branch. If one of the | ||||
10919 | // destinations is StoreBB, then we have the if/then case. | ||||
10920 | if (OtherBr->getSuccessor(0) != StoreBB && | ||||
10921 | OtherBr->getSuccessor(1) != StoreBB) | ||||
10922 | return false; | ||||
10923 | |||||
10924 | // Okay, we know that OtherBr now goes to Dest and StoreBB, so this is an | ||||
10925 | // if/then triangle. See if there is a store to the same ptr as SI that | ||||
10926 | // lives in OtherBB. | ||||
10927 | for (;; --BBI) { | ||||
10928 | // Check to see if we find the matching store. | ||||
10929 | if ((OtherStore = dyn_cast<StoreInst>(BBI))) { | ||||
10930 | if (OtherStore->getOperand(1) != SI.getOperand(1)) | ||||
10931 | return false; | ||||
10932 | break; | ||||
10933 | } | ||||
Eli Friedman | 3a311d5 | 2008-06-13 22:02:12 +0000 | [diff] [blame] | 10934 | // If we find something that may be using or overwriting the stored |
10935 | // value, or if we run out of instructions, we can't do the xform. | ||||
10936 | if (BBI->mayReadFromMemory() || BBI->mayWriteToMemory() || | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10937 | BBI == OtherBB->begin()) |
10938 | return false; | ||||
10939 | } | ||||
10940 | |||||
10941 | // In order to eliminate the store in OtherBr, we have to | ||||
Eli Friedman | 3a311d5 | 2008-06-13 22:02:12 +0000 | [diff] [blame] | 10942 | // make sure nothing reads or overwrites the stored value in |
10943 | // StoreBB. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10944 | for (BasicBlock::iterator I = StoreBB->begin(); &*I != &SI; ++I) { |
10945 | // FIXME: This should really be AA driven. | ||||
Eli Friedman | 3a311d5 | 2008-06-13 22:02:12 +0000 | [diff] [blame] | 10946 | if (I->mayReadFromMemory() || I->mayWriteToMemory()) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10947 | return false; |
10948 | } | ||||
10949 | } | ||||
10950 | |||||
10951 | // Insert a PHI node now if we need it. | ||||
10952 | Value *MergedVal = OtherStore->getOperand(0); | ||||
10953 | if (MergedVal != SI.getOperand(0)) { | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10954 | PHINode *PN = PHINode::Create(MergedVal->getType(), "storemerge"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10955 | PN->reserveOperandSpace(2); |
10956 | PN->addIncoming(SI.getOperand(0), SI.getParent()); | ||||
10957 | PN->addIncoming(OtherStore->getOperand(0), OtherBB); | ||||
10958 | MergedVal = InsertNewInstBefore(PN, DestBB->front()); | ||||
10959 | } | ||||
10960 | |||||
10961 | // Advance to a place where it is safe to insert the new store and | ||||
10962 | // insert it. | ||||
Dan Gohman | 514277c | 2008-05-23 21:05:58 +0000 | [diff] [blame] | 10963 | BBI = DestBB->getFirstNonPHI(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10964 | InsertNewInstBefore(new StoreInst(MergedVal, SI.getOperand(1), |
10965 | OtherStore->isVolatile()), *BBI); | ||||
10966 | |||||
10967 | // Nuke the old stores. | ||||
10968 | EraseInstFromFunction(SI); | ||||
10969 | EraseInstFromFunction(*OtherStore); | ||||
10970 | ++NumCombined; | ||||
10971 | return true; | ||||
10972 | } | ||||
10973 | |||||
10974 | |||||
10975 | Instruction *InstCombiner::visitBranchInst(BranchInst &BI) { | ||||
10976 | // Change br (not X), label True, label False to: br X, label False, True | ||||
10977 | Value *X = 0; | ||||
10978 | BasicBlock *TrueDest; | ||||
10979 | BasicBlock *FalseDest; | ||||
10980 | if (match(&BI, m_Br(m_Not(m_Value(X)), TrueDest, FalseDest)) && | ||||
10981 | !isa<Constant>(X)) { | ||||
10982 | // Swap Destinations and condition... | ||||
10983 | BI.setCondition(X); | ||||
10984 | BI.setSuccessor(0, FalseDest); | ||||
10985 | BI.setSuccessor(1, TrueDest); | ||||
10986 | return &BI; | ||||
10987 | } | ||||
10988 | |||||
10989 | // Cannonicalize fcmp_one -> fcmp_oeq | ||||
10990 | FCmpInst::Predicate FPred; Value *Y; | ||||
10991 | if (match(&BI, m_Br(m_FCmp(FPred, m_Value(X), m_Value(Y)), | ||||
10992 | TrueDest, FalseDest))) | ||||
10993 | if ((FPred == FCmpInst::FCMP_ONE || FPred == FCmpInst::FCMP_OLE || | ||||
10994 | FPred == FCmpInst::FCMP_OGE) && BI.getCondition()->hasOneUse()) { | ||||
10995 | FCmpInst *I = cast<FCmpInst>(BI.getCondition()); | ||||
10996 | FCmpInst::Predicate NewPred = FCmpInst::getInversePredicate(FPred); | ||||
10997 | Instruction *NewSCC = new FCmpInst(NewPred, X, Y, "", I); | ||||
10998 | NewSCC->takeName(I); | ||||
10999 | // Swap Destinations and condition... | ||||
11000 | BI.setCondition(NewSCC); | ||||
11001 | BI.setSuccessor(0, FalseDest); | ||||
11002 | BI.setSuccessor(1, TrueDest); | ||||
11003 | RemoveFromWorkList(I); | ||||
11004 | I->eraseFromParent(); | ||||
11005 | AddToWorkList(NewSCC); | ||||
11006 | return &BI; | ||||
11007 | } | ||||
11008 | |||||
11009 | // Cannonicalize icmp_ne -> icmp_eq | ||||
11010 | ICmpInst::Predicate IPred; | ||||
11011 | if (match(&BI, m_Br(m_ICmp(IPred, m_Value(X), m_Value(Y)), | ||||
11012 | TrueDest, FalseDest))) | ||||
11013 | if ((IPred == ICmpInst::ICMP_NE || IPred == ICmpInst::ICMP_ULE || | ||||
11014 | IPred == ICmpInst::ICMP_SLE || IPred == ICmpInst::ICMP_UGE || | ||||
11015 | IPred == ICmpInst::ICMP_SGE) && BI.getCondition()->hasOneUse()) { | ||||
11016 | ICmpInst *I = cast<ICmpInst>(BI.getCondition()); | ||||
11017 | ICmpInst::Predicate NewPred = ICmpInst::getInversePredicate(IPred); | ||||
11018 | Instruction *NewSCC = new ICmpInst(NewPred, X, Y, "", I); | ||||
11019 | NewSCC->takeName(I); | ||||
11020 | // Swap Destinations and condition... | ||||
11021 | BI.setCondition(NewSCC); | ||||
11022 | BI.setSuccessor(0, FalseDest); | ||||
11023 | BI.setSuccessor(1, TrueDest); | ||||
11024 | RemoveFromWorkList(I); | ||||
11025 | I->eraseFromParent();; | ||||
11026 | AddToWorkList(NewSCC); | ||||
11027 | return &BI; | ||||
11028 | } | ||||
11029 | |||||
11030 | return 0; | ||||
11031 | } | ||||
11032 | |||||
11033 | Instruction *InstCombiner::visitSwitchInst(SwitchInst &SI) { | ||||
11034 | Value *Cond = SI.getCondition(); | ||||
11035 | if (Instruction *I = dyn_cast<Instruction>(Cond)) { | ||||
11036 | if (I->getOpcode() == Instruction::Add) | ||||
11037 | if (ConstantInt *AddRHS = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
11038 | // change 'switch (X+4) case 1:' into 'switch (X) case -3' | ||||
11039 | for (unsigned i = 2, e = SI.getNumOperands(); i != e; i += 2) | ||||
11040 | SI.setOperand(i,ConstantExpr::getSub(cast<Constant>(SI.getOperand(i)), | ||||
11041 | AddRHS)); | ||||
11042 | SI.setOperand(0, I->getOperand(0)); | ||||
11043 | AddToWorkList(I); | ||||
11044 | return &SI; | ||||
11045 | } | ||||
11046 | } | ||||
11047 | return 0; | ||||
11048 | } | ||||
11049 | |||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 11050 | Instruction *InstCombiner::visitExtractValueInst(ExtractValueInst &EV) { |
Matthijs Kooijman | 45e8eb4 | 2008-07-16 12:55:45 +0000 | [diff] [blame] | 11051 | Value *Agg = EV.getAggregateOperand(); |
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 11052 | |
Matthijs Kooijman | 45e8eb4 | 2008-07-16 12:55:45 +0000 | [diff] [blame] | 11053 | if (!EV.hasIndices()) |
11054 | return ReplaceInstUsesWith(EV, Agg); | ||||
11055 | |||||
11056 | if (Constant *C = dyn_cast<Constant>(Agg)) { | ||||
11057 | if (isa<UndefValue>(C)) | ||||
11058 | return ReplaceInstUsesWith(EV, UndefValue::get(EV.getType())); | ||||
11059 | |||||
11060 | if (isa<ConstantAggregateZero>(C)) | ||||
11061 | return ReplaceInstUsesWith(EV, Constant::getNullValue(EV.getType())); | ||||
11062 | |||||
11063 | if (isa<ConstantArray>(C) || isa<ConstantStruct>(C)) { | ||||
11064 | // Extract the element indexed by the first index out of the constant | ||||
11065 | Value *V = C->getOperand(*EV.idx_begin()); | ||||
11066 | if (EV.getNumIndices() > 1) | ||||
11067 | // Extract the remaining indices out of the constant indexed by the | ||||
11068 | // first index | ||||
11069 | return ExtractValueInst::Create(V, EV.idx_begin() + 1, EV.idx_end()); | ||||
11070 | else | ||||
11071 | return ReplaceInstUsesWith(EV, V); | ||||
11072 | } | ||||
11073 | return 0; // Can't handle other constants | ||||
11074 | } | ||||
11075 | if (InsertValueInst *IV = dyn_cast<InsertValueInst>(Agg)) { | ||||
11076 | // We're extracting from an insertvalue instruction, compare the indices | ||||
11077 | const unsigned *exti, *exte, *insi, *inse; | ||||
11078 | for (exti = EV.idx_begin(), insi = IV->idx_begin(), | ||||
11079 | exte = EV.idx_end(), inse = IV->idx_end(); | ||||
11080 | exti != exte && insi != inse; | ||||
11081 | ++exti, ++insi) { | ||||
11082 | if (*insi != *exti) | ||||
11083 | // The insert and extract both reference distinctly different elements. | ||||
11084 | // This means the extract is not influenced by the insert, and we can | ||||
11085 | // replace the aggregate operand of the extract with the aggregate | ||||
11086 | // operand of the insert. i.e., replace | ||||
11087 | // %I = insertvalue { i32, { i32 } } %A, { i32 } { i32 42 }, 1 | ||||
11088 | // %E = extractvalue { i32, { i32 } } %I, 0 | ||||
11089 | // with | ||||
11090 | // %E = extractvalue { i32, { i32 } } %A, 0 | ||||
11091 | return ExtractValueInst::Create(IV->getAggregateOperand(), | ||||
11092 | EV.idx_begin(), EV.idx_end()); | ||||
11093 | } | ||||
11094 | if (exti == exte && insi == inse) | ||||
11095 | // Both iterators are at the end: Index lists are identical. Replace | ||||
11096 | // %B = insertvalue { i32, { i32 } } %A, i32 42, 1, 0 | ||||
11097 | // %C = extractvalue { i32, { i32 } } %B, 1, 0 | ||||
11098 | // with "i32 42" | ||||
11099 | return ReplaceInstUsesWith(EV, IV->getInsertedValueOperand()); | ||||
11100 | if (exti == exte) { | ||||
11101 | // The extract list is a prefix of the insert list. i.e. replace | ||||
11102 | // %I = insertvalue { i32, { i32 } } %A, i32 42, 1, 0 | ||||
11103 | // %E = extractvalue { i32, { i32 } } %I, 1 | ||||
11104 | // with | ||||
11105 | // %X = extractvalue { i32, { i32 } } %A, 1 | ||||
11106 | // %E = insertvalue { i32 } %X, i32 42, 0 | ||||
11107 | // by switching the order of the insert and extract (though the | ||||
11108 | // insertvalue should be left in, since it may have other uses). | ||||
11109 | Value *NewEV = InsertNewInstBefore( | ||||
11110 | ExtractValueInst::Create(IV->getAggregateOperand(), | ||||
11111 | EV.idx_begin(), EV.idx_end()), | ||||
11112 | EV); | ||||
11113 | return InsertValueInst::Create(NewEV, IV->getInsertedValueOperand(), | ||||
11114 | insi, inse); | ||||
11115 | } | ||||
11116 | if (insi == inse) | ||||
11117 | // The insert list is a prefix of the extract list | ||||
11118 | // We can simply remove the common indices from the extract and make it | ||||
11119 | // operate on the inserted value instead of the insertvalue result. | ||||
11120 | // i.e., replace | ||||
11121 | // %I = insertvalue { i32, { i32 } } %A, { i32 } { i32 42 }, 1 | ||||
11122 | // %E = extractvalue { i32, { i32 } } %I, 1, 0 | ||||
11123 | // with | ||||
11124 | // %E extractvalue { i32 } { i32 42 }, 0 | ||||
11125 | return ExtractValueInst::Create(IV->getInsertedValueOperand(), | ||||
11126 | exti, exte); | ||||
11127 | } | ||||
11128 | // Can't simplify extracts from other values. Note that nested extracts are | ||||
11129 | // already simplified implicitely by the above (extract ( extract (insert) ) | ||||
11130 | // will be translated into extract ( insert ( extract ) ) first and then just | ||||
11131 | // the value inserted, if appropriate). | ||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 11132 | return 0; |
11133 | } | ||||
11134 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11135 | /// CheapToScalarize - Return true if the value is cheaper to scalarize than it |
11136 | /// is to leave as a vector operation. | ||||
11137 | static bool CheapToScalarize(Value *V, bool isConstant) { | ||||
11138 | if (isa<ConstantAggregateZero>(V)) | ||||
11139 | return true; | ||||
11140 | if (ConstantVector *C = dyn_cast<ConstantVector>(V)) { | ||||
11141 | if (isConstant) return true; | ||||
11142 | // If all elts are the same, we can extract. | ||||
11143 | Constant *Op0 = C->getOperand(0); | ||||
11144 | for (unsigned i = 1; i < C->getNumOperands(); ++i) | ||||
11145 | if (C->getOperand(i) != Op0) | ||||
11146 | return false; | ||||
11147 | return true; | ||||
11148 | } | ||||
11149 | Instruction *I = dyn_cast<Instruction>(V); | ||||
11150 | if (!I) return false; | ||||
11151 | |||||
11152 | // Insert element gets simplified to the inserted element or is deleted if | ||||
11153 | // this is constant idx extract element and its a constant idx insertelt. | ||||
11154 | if (I->getOpcode() == Instruction::InsertElement && isConstant && | ||||
11155 | isa<ConstantInt>(I->getOperand(2))) | ||||
11156 | return true; | ||||
11157 | if (I->getOpcode() == Instruction::Load && I->hasOneUse()) | ||||
11158 | return true; | ||||
11159 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(I)) | ||||
11160 | if (BO->hasOneUse() && | ||||
11161 | (CheapToScalarize(BO->getOperand(0), isConstant) || | ||||
11162 | CheapToScalarize(BO->getOperand(1), isConstant))) | ||||
11163 | return true; | ||||
11164 | if (CmpInst *CI = dyn_cast<CmpInst>(I)) | ||||
11165 | if (CI->hasOneUse() && | ||||
11166 | (CheapToScalarize(CI->getOperand(0), isConstant) || | ||||
11167 | CheapToScalarize(CI->getOperand(1), isConstant))) | ||||
11168 | return true; | ||||
11169 | |||||
11170 | return false; | ||||
11171 | } | ||||
11172 | |||||
11173 | /// Read and decode a shufflevector mask. | ||||
11174 | /// | ||||
11175 | /// It turns undef elements into values that are larger than the number of | ||||
11176 | /// elements in the input. | ||||
11177 | static std::vector<unsigned> getShuffleMask(const ShuffleVectorInst *SVI) { | ||||
11178 | unsigned NElts = SVI->getType()->getNumElements(); | ||||
11179 | if (isa<ConstantAggregateZero>(SVI->getOperand(2))) | ||||
11180 | return std::vector<unsigned>(NElts, 0); | ||||
11181 | if (isa<UndefValue>(SVI->getOperand(2))) | ||||
11182 | return std::vector<unsigned>(NElts, 2*NElts); | ||||
11183 | |||||
11184 | std::vector<unsigned> Result; | ||||
11185 | const ConstantVector *CP = cast<ConstantVector>(SVI->getOperand(2)); | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 11186 | for (User::const_op_iterator i = CP->op_begin(), e = CP->op_end(); i!=e; ++i) |
11187 | if (isa<UndefValue>(*i)) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11188 | Result.push_back(NElts*2); // undef -> 8 |
11189 | else | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 11190 | Result.push_back(cast<ConstantInt>(*i)->getZExtValue()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11191 | return Result; |
11192 | } | ||||
11193 | |||||
11194 | /// FindScalarElement - Given a vector and an element number, see if the scalar | ||||
11195 | /// value is already around as a register, for example if it were inserted then | ||||
11196 | /// extracted from the vector. | ||||
11197 | static Value *FindScalarElement(Value *V, unsigned EltNo) { | ||||
11198 | assert(isa<VectorType>(V->getType()) && "Not looking at a vector?"); | ||||
11199 | const VectorType *PTy = cast<VectorType>(V->getType()); | ||||
11200 | unsigned Width = PTy->getNumElements(); | ||||
11201 | if (EltNo >= Width) // Out of range access. | ||||
11202 | return UndefValue::get(PTy->getElementType()); | ||||
11203 | |||||
11204 | if (isa<UndefValue>(V)) | ||||
11205 | return UndefValue::get(PTy->getElementType()); | ||||
11206 | else if (isa<ConstantAggregateZero>(V)) | ||||
11207 | return Constant::getNullValue(PTy->getElementType()); | ||||
11208 | else if (ConstantVector *CP = dyn_cast<ConstantVector>(V)) | ||||
11209 | return CP->getOperand(EltNo); | ||||
11210 | else if (InsertElementInst *III = dyn_cast<InsertElementInst>(V)) { | ||||
11211 | // If this is an insert to a variable element, we don't know what it is. | ||||
11212 | if (!isa<ConstantInt>(III->getOperand(2))) | ||||
11213 | return 0; | ||||
11214 | unsigned IIElt = cast<ConstantInt>(III->getOperand(2))->getZExtValue(); | ||||
11215 | |||||
11216 | // If this is an insert to the element we are looking for, return the | ||||
11217 | // inserted value. | ||||
11218 | if (EltNo == IIElt) | ||||
11219 | return III->getOperand(1); | ||||
11220 | |||||
11221 | // Otherwise, the insertelement doesn't modify the value, recurse on its | ||||
11222 | // vector input. | ||||
11223 | return FindScalarElement(III->getOperand(0), EltNo); | ||||
11224 | } else if (ShuffleVectorInst *SVI = dyn_cast<ShuffleVectorInst>(V)) { | ||||
11225 | unsigned InEl = getShuffleMask(SVI)[EltNo]; | ||||
11226 | if (InEl < Width) | ||||
11227 | return FindScalarElement(SVI->getOperand(0), InEl); | ||||
11228 | else if (InEl < Width*2) | ||||
11229 | return FindScalarElement(SVI->getOperand(1), InEl - Width); | ||||
11230 | else | ||||
11231 | return UndefValue::get(PTy->getElementType()); | ||||
11232 | } | ||||
11233 | |||||
11234 | // Otherwise, we don't know. | ||||
11235 | return 0; | ||||
11236 | } | ||||
11237 | |||||
11238 | Instruction *InstCombiner::visitExtractElementInst(ExtractElementInst &EI) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11239 | // If vector val is undef, replace extract with scalar undef. |
11240 | if (isa<UndefValue>(EI.getOperand(0))) | ||||
11241 | return ReplaceInstUsesWith(EI, UndefValue::get(EI.getType())); | ||||
11242 | |||||
11243 | // If vector val is constant 0, replace extract with scalar 0. | ||||
11244 | if (isa<ConstantAggregateZero>(EI.getOperand(0))) | ||||
11245 | return ReplaceInstUsesWith(EI, Constant::getNullValue(EI.getType())); | ||||
11246 | |||||
11247 | if (ConstantVector *C = dyn_cast<ConstantVector>(EI.getOperand(0))) { | ||||
Matthijs Kooijman | dd3425f | 2008-06-11 09:00:12 +0000 | [diff] [blame] | 11248 | // If vector val is constant with all elements the same, replace EI with |
11249 | // that element. When the elements are not identical, we cannot replace yet | ||||
11250 | // (we do that below, but only when the index is constant). | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11251 | Constant *op0 = C->getOperand(0); |
11252 | for (unsigned i = 1; i < C->getNumOperands(); ++i) | ||||
11253 | if (C->getOperand(i) != op0) { | ||||
11254 | op0 = 0; | ||||
11255 | break; | ||||
11256 | } | ||||
11257 | if (op0) | ||||
11258 | return ReplaceInstUsesWith(EI, op0); | ||||
11259 | } | ||||
11260 | |||||
11261 | // If extracting a specified index from the vector, see if we can recursively | ||||
11262 | // find a previously computed scalar that was inserted into the vector. | ||||
11263 | if (ConstantInt *IdxC = dyn_cast<ConstantInt>(EI.getOperand(1))) { | ||||
11264 | unsigned IndexVal = IdxC->getZExtValue(); | ||||
11265 | unsigned VectorWidth = | ||||
11266 | cast<VectorType>(EI.getOperand(0)->getType())->getNumElements(); | ||||
11267 | |||||
11268 | // If this is extracting an invalid index, turn this into undef, to avoid | ||||
11269 | // crashing the code below. | ||||
11270 | if (IndexVal >= VectorWidth) | ||||
11271 | return ReplaceInstUsesWith(EI, UndefValue::get(EI.getType())); | ||||
11272 | |||||
11273 | // This instruction only demands the single element from the input vector. | ||||
11274 | // If the input vector has a single use, simplify it based on this use | ||||
11275 | // property. | ||||
11276 | if (EI.getOperand(0)->hasOneUse() && VectorWidth != 1) { | ||||
11277 | uint64_t UndefElts; | ||||
11278 | if (Value *V = SimplifyDemandedVectorElts(EI.getOperand(0), | ||||
11279 | 1 << IndexVal, | ||||
11280 | UndefElts)) { | ||||
11281 | EI.setOperand(0, V); | ||||
11282 | return &EI; | ||||
11283 | } | ||||
11284 | } | ||||
11285 | |||||
11286 | if (Value *Elt = FindScalarElement(EI.getOperand(0), IndexVal)) | ||||
11287 | return ReplaceInstUsesWith(EI, Elt); | ||||
11288 | |||||
11289 | // If the this extractelement is directly using a bitcast from a vector of | ||||
11290 | // the same number of elements, see if we can find the source element from | ||||
11291 | // it. In this case, we will end up needing to bitcast the scalars. | ||||
11292 | if (BitCastInst *BCI = dyn_cast<BitCastInst>(EI.getOperand(0))) { | ||||
11293 | if (const VectorType *VT = | ||||
11294 | dyn_cast<VectorType>(BCI->getOperand(0)->getType())) | ||||
11295 | if (VT->getNumElements() == VectorWidth) | ||||
11296 | if (Value *Elt = FindScalarElement(BCI->getOperand(0), IndexVal)) | ||||
11297 | return new BitCastInst(Elt, EI.getType()); | ||||
11298 | } | ||||
11299 | } | ||||
11300 | |||||
11301 | if (Instruction *I = dyn_cast<Instruction>(EI.getOperand(0))) { | ||||
11302 | if (I->hasOneUse()) { | ||||
11303 | // Push extractelement into predecessor operation if legal and | ||||
11304 | // profitable to do so | ||||
11305 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(I)) { | ||||
11306 | bool isConstantElt = isa<ConstantInt>(EI.getOperand(1)); | ||||
11307 | if (CheapToScalarize(BO, isConstantElt)) { | ||||
11308 | ExtractElementInst *newEI0 = | ||||
11309 | new ExtractElementInst(BO->getOperand(0), EI.getOperand(1), | ||||
11310 | EI.getName()+".lhs"); | ||||
11311 | ExtractElementInst *newEI1 = | ||||
11312 | new ExtractElementInst(BO->getOperand(1), EI.getOperand(1), | ||||
11313 | EI.getName()+".rhs"); | ||||
11314 | InsertNewInstBefore(newEI0, EI); | ||||
11315 | InsertNewInstBefore(newEI1, EI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 11316 | return BinaryOperator::Create(BO->getOpcode(), newEI0, newEI1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11317 | } |
11318 | } else if (isa<LoadInst>(I)) { | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 11319 | unsigned AS = |
11320 | cast<PointerType>(I->getOperand(0)->getType())->getAddressSpace(); | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 11321 | Value *Ptr = InsertBitCastBefore(I->getOperand(0), |
11322 | PointerType::get(EI.getType(), AS),EI); | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 11323 | GetElementPtrInst *GEP = |
11324 | GetElementPtrInst::Create(Ptr, EI.getOperand(1), I->getName()+".gep"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11325 | InsertNewInstBefore(GEP, EI); |
11326 | return new LoadInst(GEP); | ||||
11327 | } | ||||
11328 | } | ||||
11329 | if (InsertElementInst *IE = dyn_cast<InsertElementInst>(I)) { | ||||
11330 | // Extracting the inserted element? | ||||
11331 | if (IE->getOperand(2) == EI.getOperand(1)) | ||||
11332 | return ReplaceInstUsesWith(EI, IE->getOperand(1)); | ||||
11333 | // If the inserted and extracted elements are constants, they must not | ||||
11334 | // be the same value, extract from the pre-inserted value instead. | ||||
11335 | if (isa<Constant>(IE->getOperand(2)) && | ||||
11336 | isa<Constant>(EI.getOperand(1))) { | ||||
11337 | AddUsesToWorkList(EI); | ||||
11338 | EI.setOperand(0, IE->getOperand(0)); | ||||
11339 | return &EI; | ||||
11340 | } | ||||
11341 | } else if (ShuffleVectorInst *SVI = dyn_cast<ShuffleVectorInst>(I)) { | ||||
11342 | // If this is extracting an element from a shufflevector, figure out where | ||||
11343 | // it came from and extract from the appropriate input element instead. | ||||
11344 | if (ConstantInt *Elt = dyn_cast<ConstantInt>(EI.getOperand(1))) { | ||||
11345 | unsigned SrcIdx = getShuffleMask(SVI)[Elt->getZExtValue()]; | ||||
11346 | Value *Src; | ||||
11347 | if (SrcIdx < SVI->getType()->getNumElements()) | ||||
11348 | Src = SVI->getOperand(0); | ||||
11349 | else if (SrcIdx < SVI->getType()->getNumElements()*2) { | ||||
11350 | SrcIdx -= SVI->getType()->getNumElements(); | ||||
11351 | Src = SVI->getOperand(1); | ||||
11352 | } else { | ||||
11353 | return ReplaceInstUsesWith(EI, UndefValue::get(EI.getType())); | ||||
11354 | } | ||||
11355 | return new ExtractElementInst(Src, SrcIdx); | ||||
11356 | } | ||||
11357 | } | ||||
11358 | } | ||||
11359 | return 0; | ||||
11360 | } | ||||
11361 | |||||
11362 | /// CollectSingleShuffleElements - If V is a shuffle of values that ONLY returns | ||||
11363 | /// elements from either LHS or RHS, return the shuffle mask and true. | ||||
11364 | /// Otherwise, return false. | ||||
11365 | static bool CollectSingleShuffleElements(Value *V, Value *LHS, Value *RHS, | ||||
11366 | std::vector<Constant*> &Mask) { | ||||
11367 | assert(V->getType() == LHS->getType() && V->getType() == RHS->getType() && | ||||
11368 | "Invalid CollectSingleShuffleElements"); | ||||
11369 | unsigned NumElts = cast<VectorType>(V->getType())->getNumElements(); | ||||
11370 | |||||
11371 | if (isa<UndefValue>(V)) { | ||||
11372 | Mask.assign(NumElts, UndefValue::get(Type::Int32Ty)); | ||||
11373 | return true; | ||||
11374 | } else if (V == LHS) { | ||||
11375 | for (unsigned i = 0; i != NumElts; ++i) | ||||
11376 | Mask.push_back(ConstantInt::get(Type::Int32Ty, i)); | ||||
11377 | return true; | ||||
11378 | } else if (V == RHS) { | ||||
11379 | for (unsigned i = 0; i != NumElts; ++i) | ||||
11380 | Mask.push_back(ConstantInt::get(Type::Int32Ty, i+NumElts)); | ||||
11381 | return true; | ||||
11382 | } else if (InsertElementInst *IEI = dyn_cast<InsertElementInst>(V)) { | ||||
11383 | // If this is an insert of an extract from some other vector, include it. | ||||
11384 | Value *VecOp = IEI->getOperand(0); | ||||
11385 | Value *ScalarOp = IEI->getOperand(1); | ||||
11386 | Value *IdxOp = IEI->getOperand(2); | ||||
11387 | |||||
11388 | if (!isa<ConstantInt>(IdxOp)) | ||||
11389 | return false; | ||||
11390 | unsigned InsertedIdx = cast<ConstantInt>(IdxOp)->getZExtValue(); | ||||
11391 | |||||
11392 | if (isa<UndefValue>(ScalarOp)) { // inserting undef into vector. | ||||
11393 | // Okay, we can handle this if the vector we are insertinting into is | ||||
11394 | // transitively ok. | ||||
11395 | if (CollectSingleShuffleElements(VecOp, LHS, RHS, Mask)) { | ||||
11396 | // If so, update the mask to reflect the inserted undef. | ||||
11397 | Mask[InsertedIdx] = UndefValue::get(Type::Int32Ty); | ||||
11398 | return true; | ||||
11399 | } | ||||
11400 | } else if (ExtractElementInst *EI = dyn_cast<ExtractElementInst>(ScalarOp)){ | ||||
11401 | if (isa<ConstantInt>(EI->getOperand(1)) && | ||||
11402 | EI->getOperand(0)->getType() == V->getType()) { | ||||
11403 | unsigned ExtractedIdx = | ||||
11404 | cast<ConstantInt>(EI->getOperand(1))->getZExtValue(); | ||||
11405 | |||||
11406 | // This must be extracting from either LHS or RHS. | ||||
11407 | if (EI->getOperand(0) == LHS || EI->getOperand(0) == RHS) { | ||||
11408 | // Okay, we can handle this if the vector we are insertinting into is | ||||
11409 | // transitively ok. | ||||
11410 | if (CollectSingleShuffleElements(VecOp, LHS, RHS, Mask)) { | ||||
11411 | // If so, update the mask to reflect the inserted value. | ||||
11412 | if (EI->getOperand(0) == LHS) { | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 11413 | Mask[InsertedIdx % NumElts] = |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11414 | ConstantInt::get(Type::Int32Ty, ExtractedIdx); |
11415 | } else { | ||||
11416 | assert(EI->getOperand(0) == RHS); | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 11417 | Mask[InsertedIdx % NumElts] = |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11418 | ConstantInt::get(Type::Int32Ty, ExtractedIdx+NumElts); |
11419 | |||||
11420 | } | ||||
11421 | return true; | ||||
11422 | } | ||||
11423 | } | ||||
11424 | } | ||||
11425 | } | ||||
11426 | } | ||||
11427 | // TODO: Handle shufflevector here! | ||||
11428 | |||||
11429 | return false; | ||||
11430 | } | ||||
11431 | |||||
11432 | /// CollectShuffleElements - We are building a shuffle of V, using RHS as the | ||||
11433 | /// RHS of the shuffle instruction, if it is not null. Return a shuffle mask | ||||
11434 | /// that computes V and the LHS value of the shuffle. | ||||
11435 | static Value *CollectShuffleElements(Value *V, std::vector<Constant*> &Mask, | ||||
11436 | Value *&RHS) { | ||||
11437 | assert(isa<VectorType>(V->getType()) && | ||||
11438 | (RHS == 0 || V->getType() == RHS->getType()) && | ||||
11439 | "Invalid shuffle!"); | ||||
11440 | unsigned NumElts = cast<VectorType>(V->getType())->getNumElements(); | ||||
11441 | |||||
11442 | if (isa<UndefValue>(V)) { | ||||
11443 | Mask.assign(NumElts, UndefValue::get(Type::Int32Ty)); | ||||
11444 | return V; | ||||
11445 | } else if (isa<ConstantAggregateZero>(V)) { | ||||
11446 | Mask.assign(NumElts, ConstantInt::get(Type::Int32Ty, 0)); | ||||
11447 | return V; | ||||
11448 | } else if (InsertElementInst *IEI = dyn_cast<InsertElementInst>(V)) { | ||||
11449 | // If this is an insert of an extract from some other vector, include it. | ||||
11450 | Value *VecOp = IEI->getOperand(0); | ||||
11451 | Value *ScalarOp = IEI->getOperand(1); | ||||
11452 | Value *IdxOp = IEI->getOperand(2); | ||||
11453 | |||||
11454 | if (ExtractElementInst *EI = dyn_cast<ExtractElementInst>(ScalarOp)) { | ||||
11455 | if (isa<ConstantInt>(EI->getOperand(1)) && isa<ConstantInt>(IdxOp) && | ||||
11456 | EI->getOperand(0)->getType() == V->getType()) { | ||||
11457 | unsigned ExtractedIdx = | ||||
11458 | cast<ConstantInt>(EI->getOperand(1))->getZExtValue(); | ||||
11459 | unsigned InsertedIdx = cast<ConstantInt>(IdxOp)->getZExtValue(); | ||||
11460 | |||||
11461 | // Either the extracted from or inserted into vector must be RHSVec, | ||||
11462 | // otherwise we'd end up with a shuffle of three inputs. | ||||
11463 | if (EI->getOperand(0) == RHS || RHS == 0) { | ||||
11464 | RHS = EI->getOperand(0); | ||||
11465 | Value *V = CollectShuffleElements(VecOp, Mask, RHS); | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 11466 | Mask[InsertedIdx % NumElts] = |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11467 | ConstantInt::get(Type::Int32Ty, NumElts+ExtractedIdx); |
11468 | return V; | ||||
11469 | } | ||||
11470 | |||||
11471 | if (VecOp == RHS) { | ||||
11472 | Value *V = CollectShuffleElements(EI->getOperand(0), Mask, RHS); | ||||
11473 | // Everything but the extracted element is replaced with the RHS. | ||||
11474 | for (unsigned i = 0; i != NumElts; ++i) { | ||||
11475 | if (i != InsertedIdx) | ||||
11476 | Mask[i] = ConstantInt::get(Type::Int32Ty, NumElts+i); | ||||
11477 | } | ||||
11478 | return V; | ||||
11479 | } | ||||
11480 | |||||
11481 | // If this insertelement is a chain that comes from exactly these two | ||||
11482 | // vectors, return the vector and the effective shuffle. | ||||
11483 | if (CollectSingleShuffleElements(IEI, EI->getOperand(0), RHS, Mask)) | ||||
11484 | return EI->getOperand(0); | ||||
11485 | |||||
11486 | } | ||||
11487 | } | ||||
11488 | } | ||||
11489 | // TODO: Handle shufflevector here! | ||||
11490 | |||||
11491 | // Otherwise, can't do anything fancy. Return an identity vector. | ||||
11492 | for (unsigned i = 0; i != NumElts; ++i) | ||||
11493 | Mask.push_back(ConstantInt::get(Type::Int32Ty, i)); | ||||
11494 | return V; | ||||
11495 | } | ||||
11496 | |||||
11497 | Instruction *InstCombiner::visitInsertElementInst(InsertElementInst &IE) { | ||||
11498 | Value *VecOp = IE.getOperand(0); | ||||
11499 | Value *ScalarOp = IE.getOperand(1); | ||||
11500 | Value *IdxOp = IE.getOperand(2); | ||||
11501 | |||||
11502 | // Inserting an undef or into an undefined place, remove this. | ||||
11503 | if (isa<UndefValue>(ScalarOp) || isa<UndefValue>(IdxOp)) | ||||
11504 | ReplaceInstUsesWith(IE, VecOp); | ||||
11505 | |||||
11506 | // If the inserted element was extracted from some other vector, and if the | ||||
11507 | // indexes are constant, try to turn this into a shufflevector operation. | ||||
11508 | if (ExtractElementInst *EI = dyn_cast<ExtractElementInst>(ScalarOp)) { | ||||
11509 | if (isa<ConstantInt>(EI->getOperand(1)) && isa<ConstantInt>(IdxOp) && | ||||
11510 | EI->getOperand(0)->getType() == IE.getType()) { | ||||
11511 | unsigned NumVectorElts = IE.getType()->getNumElements(); | ||||
11512 | unsigned ExtractedIdx = | ||||
11513 | cast<ConstantInt>(EI->getOperand(1))->getZExtValue(); | ||||
11514 | unsigned InsertedIdx = cast<ConstantInt>(IdxOp)->getZExtValue(); | ||||
11515 | |||||
11516 | if (ExtractedIdx >= NumVectorElts) // Out of range extract. | ||||
11517 | return ReplaceInstUsesWith(IE, VecOp); | ||||
11518 | |||||
11519 | if (InsertedIdx >= NumVectorElts) // Out of range insert. | ||||
11520 | return ReplaceInstUsesWith(IE, UndefValue::get(IE.getType())); | ||||
11521 | |||||
11522 | // If we are extracting a value from a vector, then inserting it right | ||||
11523 | // back into the same place, just use the input vector. | ||||
11524 | if (EI->getOperand(0) == VecOp && ExtractedIdx == InsertedIdx) | ||||
11525 | return ReplaceInstUsesWith(IE, VecOp); | ||||
11526 | |||||
11527 | // We could theoretically do this for ANY input. However, doing so could | ||||
11528 | // turn chains of insertelement instructions into a chain of shufflevector | ||||
11529 | // instructions, and right now we do not merge shufflevectors. As such, | ||||
11530 | // only do this in a situation where it is clear that there is benefit. | ||||
11531 | if (isa<UndefValue>(VecOp) || isa<ConstantAggregateZero>(VecOp)) { | ||||
11532 | // Turn this into shuffle(EIOp0, VecOp, Mask). The result has all of | ||||
11533 | // the values of VecOp, except then one read from EIOp0. | ||||
11534 | // Build a new shuffle mask. | ||||
11535 | std::vector<Constant*> Mask; | ||||
11536 | if (isa<UndefValue>(VecOp)) | ||||
11537 | Mask.assign(NumVectorElts, UndefValue::get(Type::Int32Ty)); | ||||
11538 | else { | ||||
11539 | assert(isa<ConstantAggregateZero>(VecOp) && "Unknown thing"); | ||||
11540 | Mask.assign(NumVectorElts, ConstantInt::get(Type::Int32Ty, | ||||
11541 | NumVectorElts)); | ||||
11542 | } | ||||
11543 | Mask[InsertedIdx] = ConstantInt::get(Type::Int32Ty, ExtractedIdx); | ||||
11544 | return new ShuffleVectorInst(EI->getOperand(0), VecOp, | ||||
11545 | ConstantVector::get(Mask)); | ||||
11546 | } | ||||
11547 | |||||
11548 | // If this insertelement isn't used by some other insertelement, turn it | ||||
11549 | // (and any insertelements it points to), into one big shuffle. | ||||
11550 | if (!IE.hasOneUse() || !isa<InsertElementInst>(IE.use_back())) { | ||||
11551 | std::vector<Constant*> Mask; | ||||
11552 | Value *RHS = 0; | ||||
11553 | Value *LHS = CollectShuffleElements(&IE, Mask, RHS); | ||||
11554 | if (RHS == 0) RHS = UndefValue::get(LHS->getType()); | ||||
11555 | // We now have a shuffle of LHS, RHS, Mask. | ||||
11556 | return new ShuffleVectorInst(LHS, RHS, ConstantVector::get(Mask)); | ||||
11557 | } | ||||
11558 | } | ||||
11559 | } | ||||
11560 | |||||
11561 | return 0; | ||||
11562 | } | ||||
11563 | |||||
11564 | |||||
11565 | Instruction *InstCombiner::visitShuffleVectorInst(ShuffleVectorInst &SVI) { | ||||
11566 | Value *LHS = SVI.getOperand(0); | ||||
11567 | Value *RHS = SVI.getOperand(1); | ||||
11568 | std::vector<unsigned> Mask = getShuffleMask(&SVI); | ||||
11569 | |||||
11570 | bool MadeChange = false; | ||||
11571 | |||||
11572 | // Undefined shuffle mask -> undefined value. | ||||
11573 | if (isa<UndefValue>(SVI.getOperand(2))) | ||||
11574 | return ReplaceInstUsesWith(SVI, UndefValue::get(SVI.getType())); | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 11575 | |
11576 | uint64_t UndefElts; | ||||
11577 | unsigned VWidth = cast<VectorType>(SVI.getType())->getNumElements(); | ||||
11578 | uint64_t AllOnesEltMask = ~0ULL >> (64-VWidth); | ||||
11579 | if (VWidth <= 64 && | ||||
Dan Gohman | 83b702d | 2008-09-11 22:47:57 +0000 | [diff] [blame] | 11580 | SimplifyDemandedVectorElts(&SVI, AllOnesEltMask, UndefElts)) { |
11581 | LHS = SVI.getOperand(0); | ||||
11582 | RHS = SVI.getOperand(1); | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 11583 | MadeChange = true; |
Dan Gohman | 83b702d | 2008-09-11 22:47:57 +0000 | [diff] [blame] | 11584 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11585 | |
11586 | // Canonicalize shuffle(x ,x,mask) -> shuffle(x, undef,mask') | ||||
11587 | // Canonicalize shuffle(undef,x,mask) -> shuffle(x, undef,mask'). | ||||
11588 | if (LHS == RHS || isa<UndefValue>(LHS)) { | ||||
11589 | if (isa<UndefValue>(LHS) && LHS == RHS) { | ||||
11590 | // shuffle(undef,undef,mask) -> undef. | ||||
11591 | return ReplaceInstUsesWith(SVI, LHS); | ||||
11592 | } | ||||
11593 | |||||
11594 | // Remap any references to RHS to use LHS. | ||||
11595 | std::vector<Constant*> Elts; | ||||
11596 | for (unsigned i = 0, e = Mask.size(); i != e; ++i) { | ||||
11597 | if (Mask[i] >= 2*e) | ||||
11598 | Elts.push_back(UndefValue::get(Type::Int32Ty)); | ||||
11599 | else { | ||||
11600 | if ((Mask[i] >= e && isa<UndefValue>(RHS)) || | ||||
Dan Gohman | bba96b9 | 2008-08-06 18:17:32 +0000 | [diff] [blame] | 11601 | (Mask[i] < e && isa<UndefValue>(LHS))) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11602 | Mask[i] = 2*e; // Turn into undef. |
Dan Gohman | bba96b9 | 2008-08-06 18:17:32 +0000 | [diff] [blame] | 11603 | Elts.push_back(UndefValue::get(Type::Int32Ty)); |
11604 | } else { | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 11605 | Mask[i] = Mask[i] % e; // Force to LHS. |
Dan Gohman | bba96b9 | 2008-08-06 18:17:32 +0000 | [diff] [blame] | 11606 | Elts.push_back(ConstantInt::get(Type::Int32Ty, Mask[i])); |
11607 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11608 | } |
11609 | } | ||||
11610 | SVI.setOperand(0, SVI.getOperand(1)); | ||||
11611 | SVI.setOperand(1, UndefValue::get(RHS->getType())); | ||||
11612 | SVI.setOperand(2, ConstantVector::get(Elts)); | ||||
11613 | LHS = SVI.getOperand(0); | ||||
11614 | RHS = SVI.getOperand(1); | ||||
11615 | MadeChange = true; | ||||
11616 | } | ||||
11617 | |||||
11618 | // Analyze the shuffle, are the LHS or RHS and identity shuffles? | ||||
11619 | bool isLHSID = true, isRHSID = true; | ||||
11620 | |||||
11621 | for (unsigned i = 0, e = Mask.size(); i != e; ++i) { | ||||
11622 | if (Mask[i] >= e*2) continue; // Ignore undef values. | ||||
11623 | // Is this an identity shuffle of the LHS value? | ||||
11624 | isLHSID &= (Mask[i] == i); | ||||
11625 | |||||
11626 | // Is this an identity shuffle of the RHS value? | ||||
11627 | isRHSID &= (Mask[i]-e == i); | ||||
11628 | } | ||||
11629 | |||||
11630 | // Eliminate identity shuffles. | ||||
11631 | if (isLHSID) return ReplaceInstUsesWith(SVI, LHS); | ||||
11632 | if (isRHSID) return ReplaceInstUsesWith(SVI, RHS); | ||||
11633 | |||||
11634 | // If the LHS is a shufflevector itself, see if we can combine it with this | ||||
11635 | // one without producing an unusual shuffle. Here we are really conservative: | ||||
11636 | // we are absolutely afraid of producing a shuffle mask not in the input | ||||
11637 | // program, because the code gen may not be smart enough to turn a merged | ||||
11638 | // shuffle into two specific shuffles: it may produce worse code. As such, | ||||
11639 | // we only merge two shuffles if the result is one of the two input shuffle | ||||
11640 | // masks. In this case, merging the shuffles just removes one instruction, | ||||
11641 | // which we know is safe. This is good for things like turning: | ||||
11642 | // (splat(splat)) -> splat. | ||||
11643 | if (ShuffleVectorInst *LHSSVI = dyn_cast<ShuffleVectorInst>(LHS)) { | ||||
11644 | if (isa<UndefValue>(RHS)) { | ||||
11645 | std::vector<unsigned> LHSMask = getShuffleMask(LHSSVI); | ||||
11646 | |||||
11647 | std::vector<unsigned> NewMask; | ||||
11648 | for (unsigned i = 0, e = Mask.size(); i != e; ++i) | ||||
11649 | if (Mask[i] >= 2*e) | ||||
11650 | NewMask.push_back(2*e); | ||||
11651 | else | ||||
11652 | NewMask.push_back(LHSMask[Mask[i]]); | ||||
11653 | |||||
11654 | // If the result mask is equal to the src shuffle or this shuffle mask, do | ||||
11655 | // the replacement. | ||||
11656 | if (NewMask == LHSMask || NewMask == Mask) { | ||||
11657 | std::vector<Constant*> Elts; | ||||
11658 | for (unsigned i = 0, e = NewMask.size(); i != e; ++i) { | ||||
11659 | if (NewMask[i] >= e*2) { | ||||
11660 | Elts.push_back(UndefValue::get(Type::Int32Ty)); | ||||
11661 | } else { | ||||
11662 | Elts.push_back(ConstantInt::get(Type::Int32Ty, NewMask[i])); | ||||
11663 | } | ||||
11664 | } | ||||
11665 | return new ShuffleVectorInst(LHSSVI->getOperand(0), | ||||
11666 | LHSSVI->getOperand(1), | ||||
11667 | ConstantVector::get(Elts)); | ||||
11668 | } | ||||
11669 | } | ||||
11670 | } | ||||
11671 | |||||
11672 | return MadeChange ? &SVI : 0; | ||||
11673 | } | ||||
11674 | |||||
11675 | |||||
11676 | |||||
11677 | |||||
11678 | /// TryToSinkInstruction - Try to move the specified instruction from its | ||||
11679 | /// current block into the beginning of DestBlock, which can only happen if it's | ||||
11680 | /// safe to move the instruction past all of the instructions between it and the | ||||
11681 | /// end of its block. | ||||
11682 | static bool TryToSinkInstruction(Instruction *I, BasicBlock *DestBlock) { | ||||
11683 | assert(I->hasOneUse() && "Invariants didn't hold!"); | ||||
11684 | |||||
11685 | // Cannot move control-flow-involving, volatile loads, vaarg, etc. | ||||
Chris Lattner | cb19a1c | 2008-05-09 15:07:33 +0000 | [diff] [blame] | 11686 | if (isa<PHINode>(I) || I->mayWriteToMemory() || isa<TerminatorInst>(I)) |
11687 | return false; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11688 | |
11689 | // Do not sink alloca instructions out of the entry block. | ||||
11690 | if (isa<AllocaInst>(I) && I->getParent() == | ||||
11691 | &DestBlock->getParent()->getEntryBlock()) | ||||
11692 | return false; | ||||
11693 | |||||
11694 | // We can only sink load instructions if there is nothing between the load and | ||||
11695 | // the end of block that could change the value. | ||||
Chris Lattner | 0db40a6 | 2008-05-08 17:37:37 +0000 | [diff] [blame] | 11696 | if (I->mayReadFromMemory()) { |
11697 | for (BasicBlock::iterator Scan = I, E = I->getParent()->end(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11698 | Scan != E; ++Scan) |
11699 | if (Scan->mayWriteToMemory()) | ||||
11700 | return false; | ||||
11701 | } | ||||
11702 | |||||
Dan Gohman | 514277c | 2008-05-23 21:05:58 +0000 | [diff] [blame] | 11703 | BasicBlock::iterator InsertPos = DestBlock->getFirstNonPHI(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11704 | |
11705 | I->moveBefore(InsertPos); | ||||
11706 | ++NumSunkInst; | ||||
11707 | return true; | ||||
11708 | } | ||||
11709 | |||||
11710 | |||||
11711 | /// AddReachableCodeToWorklist - Walk the function in depth-first order, adding | ||||
11712 | /// all reachable code to the worklist. | ||||
11713 | /// | ||||
11714 | /// This has a couple of tricks to make the code faster and more powerful. In | ||||
11715 | /// particular, we constant fold and DCE instructions as we go, to avoid adding | ||||
11716 | /// them to the worklist (this significantly speeds up instcombine on code where | ||||
11717 | /// many instructions are dead or constant). Additionally, if we find a branch | ||||
11718 | /// whose condition is a known constant, we only visit the reachable successors. | ||||
11719 | /// | ||||
11720 | static void AddReachableCodeToWorklist(BasicBlock *BB, | ||||
11721 | SmallPtrSet<BasicBlock*, 64> &Visited, | ||||
11722 | InstCombiner &IC, | ||||
11723 | const TargetData *TD) { | ||||
Chris Lattner | a06291a | 2008-08-15 04:03:01 +0000 | [diff] [blame] | 11724 | SmallVector<BasicBlock*, 256> Worklist; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11725 | Worklist.push_back(BB); |
11726 | |||||
11727 | while (!Worklist.empty()) { | ||||
11728 | BB = Worklist.back(); | ||||
11729 | Worklist.pop_back(); | ||||
11730 | |||||
11731 | // We have now visited this block! If we've already been here, ignore it. | ||||
11732 | if (!Visited.insert(BB)) continue; | ||||
11733 | |||||
11734 | for (BasicBlock::iterator BBI = BB->begin(), E = BB->end(); BBI != E; ) { | ||||
11735 | Instruction *Inst = BBI++; | ||||
11736 | |||||
11737 | // DCE instruction if trivially dead. | ||||
11738 | if (isInstructionTriviallyDead(Inst)) { | ||||
11739 | ++NumDeadInst; | ||||
11740 | DOUT << "IC: DCE: " << *Inst; | ||||
11741 | Inst->eraseFromParent(); | ||||
11742 | continue; | ||||
11743 | } | ||||
11744 | |||||
11745 | // ConstantProp instruction if trivially constant. | ||||
11746 | if (Constant *C = ConstantFoldInstruction(Inst, TD)) { | ||||
11747 | DOUT << "IC: ConstFold to: " << *C << " from: " << *Inst; | ||||
11748 | Inst->replaceAllUsesWith(C); | ||||
11749 | ++NumConstProp; | ||||
11750 | Inst->eraseFromParent(); | ||||
11751 | continue; | ||||
11752 | } | ||||
Chris Lattner | e0f462d | 2007-07-20 22:06:41 +0000 | [diff] [blame] | 11753 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11754 | IC.AddToWorkList(Inst); |
11755 | } | ||||
11756 | |||||
11757 | // Recursively visit successors. If this is a branch or switch on a | ||||
11758 | // constant, only visit the reachable successor. | ||||
11759 | TerminatorInst *TI = BB->getTerminator(); | ||||
11760 | if (BranchInst *BI = dyn_cast<BranchInst>(TI)) { | ||||
11761 | if (BI->isConditional() && isa<ConstantInt>(BI->getCondition())) { | ||||
11762 | bool CondVal = cast<ConstantInt>(BI->getCondition())->getZExtValue(); | ||||
Nick Lewycky | d551cf1 | 2008-03-09 08:50:23 +0000 | [diff] [blame] | 11763 | BasicBlock *ReachableBB = BI->getSuccessor(!CondVal); |
Nick Lewycky | d8aa33a | 2008-04-25 16:53:59 +0000 | [diff] [blame] | 11764 | Worklist.push_back(ReachableBB); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11765 | continue; |
11766 | } | ||||
11767 | } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) { | ||||
11768 | if (ConstantInt *Cond = dyn_cast<ConstantInt>(SI->getCondition())) { | ||||
11769 | // See if this is an explicit destination. | ||||
11770 | for (unsigned i = 1, e = SI->getNumSuccessors(); i != e; ++i) | ||||
11771 | if (SI->getCaseValue(i) == Cond) { | ||||
Nick Lewycky | d551cf1 | 2008-03-09 08:50:23 +0000 | [diff] [blame] | 11772 | BasicBlock *ReachableBB = SI->getSuccessor(i); |
Nick Lewycky | d8aa33a | 2008-04-25 16:53:59 +0000 | [diff] [blame] | 11773 | Worklist.push_back(ReachableBB); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11774 | continue; |
11775 | } | ||||
11776 | |||||
11777 | // Otherwise it is the default destination. | ||||
11778 | Worklist.push_back(SI->getSuccessor(0)); | ||||
11779 | continue; | ||||
11780 | } | ||||
11781 | } | ||||
11782 | |||||
11783 | for (unsigned i = 0, e = TI->getNumSuccessors(); i != e; ++i) | ||||
11784 | Worklist.push_back(TI->getSuccessor(i)); | ||||
11785 | } | ||||
11786 | } | ||||
11787 | |||||
11788 | bool InstCombiner::DoOneIteration(Function &F, unsigned Iteration) { | ||||
11789 | bool Changed = false; | ||||
11790 | TD = &getAnalysis<TargetData>(); | ||||
11791 | |||||
11792 | DEBUG(DOUT << "\n\nINSTCOMBINE ITERATION #" << Iteration << " on " | ||||
11793 | << F.getNameStr() << "\n"); | ||||
11794 | |||||
11795 | { | ||||
11796 | // Do a depth-first traversal of the function, populate the worklist with | ||||
11797 | // the reachable instructions. Ignore blocks that are not reachable. Keep | ||||
11798 | // track of which blocks we visit. | ||||
11799 | SmallPtrSet<BasicBlock*, 64> Visited; | ||||
11800 | AddReachableCodeToWorklist(F.begin(), Visited, *this, TD); | ||||
11801 | |||||
11802 | // Do a quick scan over the function. If we find any blocks that are | ||||
11803 | // unreachable, remove any instructions inside of them. This prevents | ||||
11804 | // the instcombine code from having to deal with some bad special cases. | ||||
11805 | for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) | ||||
11806 | if (!Visited.count(BB)) { | ||||
11807 | Instruction *Term = BB->getTerminator(); | ||||
11808 | while (Term != BB->begin()) { // Remove instrs bottom-up | ||||
11809 | BasicBlock::iterator I = Term; --I; | ||||
11810 | |||||
11811 | DOUT << "IC: DCE: " << *I; | ||||
11812 | ++NumDeadInst; | ||||
11813 | |||||
11814 | if (!I->use_empty()) | ||||
11815 | I->replaceAllUsesWith(UndefValue::get(I->getType())); | ||||
11816 | I->eraseFromParent(); | ||||
11817 | } | ||||
11818 | } | ||||
11819 | } | ||||
11820 | |||||
11821 | while (!Worklist.empty()) { | ||||
11822 | Instruction *I = RemoveOneFromWorkList(); | ||||
11823 | if (I == 0) continue; // skip null values. | ||||
11824 | |||||
11825 | // Check to see if we can DCE the instruction. | ||||
11826 | if (isInstructionTriviallyDead(I)) { | ||||
11827 | // Add operands to the worklist. | ||||
11828 | if (I->getNumOperands() < 4) | ||||
11829 | AddUsesToWorkList(*I); | ||||
11830 | ++NumDeadInst; | ||||
11831 | |||||
11832 | DOUT << "IC: DCE: " << *I; | ||||
11833 | |||||
11834 | I->eraseFromParent(); | ||||
11835 | RemoveFromWorkList(I); | ||||
11836 | continue; | ||||
11837 | } | ||||
11838 | |||||
11839 | // Instruction isn't dead, see if we can constant propagate it. | ||||
11840 | if (Constant *C = ConstantFoldInstruction(I, TD)) { | ||||
11841 | DOUT << "IC: ConstFold to: " << *C << " from: " << *I; | ||||
11842 | |||||
11843 | // Add operands to the worklist. | ||||
11844 | AddUsesToWorkList(*I); | ||||
11845 | ReplaceInstUsesWith(*I, C); | ||||
11846 | |||||
11847 | ++NumConstProp; | ||||
11848 | I->eraseFromParent(); | ||||
11849 | RemoveFromWorkList(I); | ||||
11850 | continue; | ||||
11851 | } | ||||
11852 | |||||
Nick Lewycky | adb6792 | 2008-05-25 20:56:15 +0000 | [diff] [blame] | 11853 | if (TD && I->getType()->getTypeID() == Type::VoidTyID) { |
11854 | // See if we can constant fold its operands. | ||||
11855 | for (User::op_iterator i = I->op_begin(), e = I->op_end(); i != e; ++i) { | ||||
11856 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(i)) { | ||||
11857 | if (Constant *NewC = ConstantFoldConstantExpression(CE, TD)) | ||||
11858 | i->set(NewC); | ||||
11859 | } | ||||
11860 | } | ||||
11861 | } | ||||
11862 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11863 | // 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] | 11864 | if (I->hasOneUse()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11865 | BasicBlock *BB = I->getParent(); |
11866 | BasicBlock *UserParent = cast<Instruction>(I->use_back())->getParent(); | ||||
11867 | if (UserParent != BB) { | ||||
11868 | bool UserIsSuccessor = false; | ||||
11869 | // See if the user is one of our successors. | ||||
11870 | for (succ_iterator SI = succ_begin(BB), E = succ_end(BB); SI != E; ++SI) | ||||
11871 | if (*SI == UserParent) { | ||||
11872 | UserIsSuccessor = true; | ||||
11873 | break; | ||||
11874 | } | ||||
11875 | |||||
11876 | // If the user is one of our immediate successors, and if that successor | ||||
11877 | // only has us as a predecessors (we'd have to split the critical edge | ||||
11878 | // otherwise), we can keep going. | ||||
11879 | if (UserIsSuccessor && !isa<PHINode>(I->use_back()) && | ||||
11880 | next(pred_begin(UserParent)) == pred_end(UserParent)) | ||||
11881 | // Okay, the CFG is simple enough, try to sink this instruction. | ||||
11882 | Changed |= TryToSinkInstruction(I, UserParent); | ||||
11883 | } | ||||
11884 | } | ||||
11885 | |||||
11886 | // Now that we have an instruction, try combining it to simplify it... | ||||
11887 | #ifndef NDEBUG | ||||
11888 | std::string OrigI; | ||||
11889 | #endif | ||||
11890 | DEBUG(std::ostringstream SS; I->print(SS); OrigI = SS.str();); | ||||
11891 | if (Instruction *Result = visit(*I)) { | ||||
11892 | ++NumCombined; | ||||
11893 | // Should we replace the old instruction with a new one? | ||||
11894 | if (Result != I) { | ||||
11895 | DOUT << "IC: Old = " << *I | ||||
11896 | << " New = " << *Result; | ||||
11897 | |||||
11898 | // Everything uses the new instruction now. | ||||
11899 | I->replaceAllUsesWith(Result); | ||||
11900 | |||||
11901 | // Push the new instruction and any users onto the worklist. | ||||
11902 | AddToWorkList(Result); | ||||
11903 | AddUsersToWorkList(*Result); | ||||
11904 | |||||
11905 | // Move the name to the new instruction first. | ||||
11906 | Result->takeName(I); | ||||
11907 | |||||
11908 | // Insert the new instruction into the basic block... | ||||
11909 | BasicBlock *InstParent = I->getParent(); | ||||
11910 | BasicBlock::iterator InsertPos = I; | ||||
11911 | |||||
11912 | if (!isa<PHINode>(Result)) // If combining a PHI, don't insert | ||||
11913 | while (isa<PHINode>(InsertPos)) // middle of a block of PHIs. | ||||
11914 | ++InsertPos; | ||||
11915 | |||||
11916 | InstParent->getInstList().insert(InsertPos, Result); | ||||
11917 | |||||
11918 | // Make sure that we reprocess all operands now that we reduced their | ||||
11919 | // use counts. | ||||
11920 | AddUsesToWorkList(*I); | ||||
11921 | |||||
11922 | // Instructions can end up on the worklist more than once. Make sure | ||||
11923 | // we do not process an instruction that has been deleted. | ||||
11924 | RemoveFromWorkList(I); | ||||
11925 | |||||
11926 | // Erase the old instruction. | ||||
11927 | InstParent->getInstList().erase(I); | ||||
11928 | } else { | ||||
11929 | #ifndef NDEBUG | ||||
11930 | DOUT << "IC: Mod = " << OrigI | ||||
11931 | << " New = " << *I; | ||||
11932 | #endif | ||||
11933 | |||||
11934 | // If the instruction was modified, it's possible that it is now dead. | ||||
11935 | // if so, remove it. | ||||
11936 | if (isInstructionTriviallyDead(I)) { | ||||
11937 | // Make sure we process all operands now that we are reducing their | ||||
11938 | // use counts. | ||||
11939 | AddUsesToWorkList(*I); | ||||
11940 | |||||
11941 | // Instructions may end up in the worklist more than once. Erase all | ||||
11942 | // occurrences of this instruction. | ||||
11943 | RemoveFromWorkList(I); | ||||
11944 | I->eraseFromParent(); | ||||
11945 | } else { | ||||
11946 | AddToWorkList(I); | ||||
11947 | AddUsersToWorkList(*I); | ||||
11948 | } | ||||
11949 | } | ||||
11950 | Changed = true; | ||||
11951 | } | ||||
11952 | } | ||||
11953 | |||||
11954 | assert(WorklistMap.empty() && "Worklist empty, but map not?"); | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 11955 | |
11956 | // Do an explicit clear, this shrinks the map if needed. | ||||
11957 | WorklistMap.clear(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11958 | return Changed; |
11959 | } | ||||
11960 | |||||
11961 | |||||
11962 | bool InstCombiner::runOnFunction(Function &F) { | ||||
11963 | MustPreserveLCSSA = mustPreserveAnalysisID(LCSSAID); | ||||
11964 | |||||
11965 | bool EverMadeChange = false; | ||||
11966 | |||||
11967 | // Iterate while there is work to do. | ||||
11968 | unsigned Iteration = 0; | ||||
Bill Wendling | d9644a4 | 2008-05-14 22:45:20 +0000 | [diff] [blame] | 11969 | while (DoOneIteration(F, Iteration++)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11970 | EverMadeChange = true; |
11971 | return EverMadeChange; | ||||
11972 | } | ||||
11973 | |||||
11974 | FunctionPass *llvm::createInstructionCombiningPass() { | ||||
11975 | return new InstCombiner(); | ||||
11976 | } | ||||
11977 | |||||
Chris Lattner | 6297fc7 | 2008-08-11 22:06:05 +0000 | [diff] [blame] | 11978 |