Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1 | //===- InstructionCombining.cpp - Combine multiple instructions -----------===// |
2 | // | ||||
3 | // The LLVM Compiler Infrastructure | ||||
4 | // | ||||
Chris Lattner | 081ce94 | 2007-12-29 20:36:04 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
6 | // License. See LICENSE.TXT for details. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7 | // |
8 | //===----------------------------------------------------------------------===// | ||||
9 | // | ||||
10 | // InstructionCombining - Combine instructions to form fewer, simple | ||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 11 | // instructions. This pass does not modify the CFG. This pass is where |
12 | // algebraic simplification happens. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 13 | // |
14 | // This pass combines things like: | ||||
15 | // %Y = add i32 %X, 1 | ||||
16 | // %Z = add i32 %Y, 1 | ||||
17 | // into: | ||||
18 | // %Z = add i32 %X, 2 | ||||
19 | // | ||||
20 | // This is a simple worklist driven algorithm. | ||||
21 | // | ||||
22 | // This pass guarantees that the following canonicalizations are performed on | ||||
23 | // the program: | ||||
24 | // 1. If a binary operator has a constant operand, it is moved to the RHS | ||||
25 | // 2. Bitwise operators with constant operands are always grouped so that | ||||
26 | // shifts are performed first, then or's, then and's, then xor's. | ||||
27 | // 3. Compare instructions are converted from <,>,<=,>= to ==,!= if possible | ||||
28 | // 4. All cmp instructions on boolean values are replaced with logical ops | ||||
29 | // 5. add X, X is represented as (X*2) => (X << 1) | ||||
30 | // 6. Multiplies with a power-of-two constant argument are transformed into | ||||
31 | // shifts. | ||||
32 | // ... etc. | ||||
33 | // | ||||
34 | //===----------------------------------------------------------------------===// | ||||
35 | |||||
36 | #define DEBUG_TYPE "instcombine" | ||||
37 | #include "llvm/Transforms/Scalar.h" | ||||
38 | #include "llvm/IntrinsicInst.h" | ||||
39 | #include "llvm/Pass.h" | ||||
40 | #include "llvm/DerivedTypes.h" | ||||
41 | #include "llvm/GlobalVariable.h" | ||||
42 | #include "llvm/Analysis/ConstantFolding.h" | ||||
Chris Lattner | a432bc7 | 2008-06-02 01:18:21 +0000 | [diff] [blame] | 43 | #include "llvm/Analysis/ValueTracking.h" |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 44 | #include "llvm/Target/TargetData.h" |
45 | #include "llvm/Transforms/Utils/BasicBlockUtils.h" | ||||
46 | #include "llvm/Transforms/Utils/Local.h" | ||||
47 | #include "llvm/Support/CallSite.h" | ||||
Nick Lewycky | 0185bbf | 2008-02-03 16:33:09 +0000 | [diff] [blame] | 48 | #include "llvm/Support/ConstantRange.h" |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 49 | #include "llvm/Support/Debug.h" |
50 | #include "llvm/Support/GetElementPtrTypeIterator.h" | ||||
51 | #include "llvm/Support/InstVisitor.h" | ||||
52 | #include "llvm/Support/MathExtras.h" | ||||
53 | #include "llvm/Support/PatternMatch.h" | ||||
54 | #include "llvm/Support/Compiler.h" | ||||
55 | #include "llvm/ADT/DenseMap.h" | ||||
56 | #include "llvm/ADT/SmallVector.h" | ||||
57 | #include "llvm/ADT/SmallPtrSet.h" | ||||
58 | #include "llvm/ADT/Statistic.h" | ||||
59 | #include "llvm/ADT/STLExtras.h" | ||||
60 | #include <algorithm> | ||||
Edwin Török | a0e6fce | 2008-04-20 08:33:11 +0000 | [diff] [blame] | 61 | #include <climits> |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 62 | #include <sstream> |
63 | using namespace llvm; | ||||
64 | using namespace llvm::PatternMatch; | ||||
65 | |||||
66 | STATISTIC(NumCombined , "Number of insts combined"); | ||||
67 | STATISTIC(NumConstProp, "Number of constant folds"); | ||||
68 | STATISTIC(NumDeadInst , "Number of dead inst eliminated"); | ||||
69 | STATISTIC(NumDeadStore, "Number of dead stores eliminated"); | ||||
70 | STATISTIC(NumSunkInst , "Number of instructions sunk"); | ||||
71 | |||||
72 | namespace { | ||||
73 | class VISIBILITY_HIDDEN InstCombiner | ||||
74 | : public FunctionPass, | ||||
75 | public InstVisitor<InstCombiner, Instruction*> { | ||||
76 | // Worklist of all of the instructions that need to be simplified. | ||||
Chris Lattner | a06291a | 2008-08-15 04:03:01 +0000 | [diff] [blame] | 77 | SmallVector<Instruction*, 256> Worklist; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 78 | DenseMap<Instruction*, unsigned> WorklistMap; |
79 | TargetData *TD; | ||||
80 | bool MustPreserveLCSSA; | ||||
81 | public: | ||||
82 | static char ID; // Pass identification, replacement for typeid | ||||
Dan Gohman | 26f8c27 | 2008-09-04 17:05:41 +0000 | [diff] [blame] | 83 | InstCombiner() : FunctionPass(&ID) {} |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 84 | |
85 | /// AddToWorkList - Add the specified instruction to the worklist if it | ||||
86 | /// isn't already in it. | ||||
87 | void AddToWorkList(Instruction *I) { | ||||
Dan Gohman | 55d1966 | 2008-07-07 17:46:23 +0000 | [diff] [blame] | 88 | if (WorklistMap.insert(std::make_pair(I, Worklist.size())).second) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 89 | Worklist.push_back(I); |
90 | } | ||||
91 | |||||
92 | // RemoveFromWorkList - remove I from the worklist if it exists. | ||||
93 | void RemoveFromWorkList(Instruction *I) { | ||||
94 | DenseMap<Instruction*, unsigned>::iterator It = WorklistMap.find(I); | ||||
95 | if (It == WorklistMap.end()) return; // Not in worklist. | ||||
96 | |||||
97 | // Don't bother moving everything down, just null out the slot. | ||||
98 | Worklist[It->second] = 0; | ||||
99 | |||||
100 | WorklistMap.erase(It); | ||||
101 | } | ||||
102 | |||||
103 | Instruction *RemoveOneFromWorkList() { | ||||
104 | Instruction *I = Worklist.back(); | ||||
105 | Worklist.pop_back(); | ||||
106 | WorklistMap.erase(I); | ||||
107 | return I; | ||||
108 | } | ||||
109 | |||||
110 | |||||
111 | /// AddUsersToWorkList - When an instruction is simplified, add all users of | ||||
112 | /// the instruction to the work lists because they might get more simplified | ||||
113 | /// now. | ||||
114 | /// | ||||
115 | void AddUsersToWorkList(Value &I) { | ||||
116 | for (Value::use_iterator UI = I.use_begin(), UE = I.use_end(); | ||||
117 | UI != UE; ++UI) | ||||
118 | AddToWorkList(cast<Instruction>(*UI)); | ||||
119 | } | ||||
120 | |||||
121 | /// AddUsesToWorkList - When an instruction is simplified, add operands to | ||||
122 | /// the work lists because they might get more simplified now. | ||||
123 | /// | ||||
124 | void AddUsesToWorkList(Instruction &I) { | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 125 | for (User::op_iterator i = I.op_begin(), e = I.op_end(); i != e; ++i) |
126 | if (Instruction *Op = dyn_cast<Instruction>(*i)) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 127 | AddToWorkList(Op); |
128 | } | ||||
129 | |||||
130 | /// AddSoonDeadInstToWorklist - The specified instruction is about to become | ||||
131 | /// dead. Add all of its operands to the worklist, turning them into | ||||
132 | /// undef's to reduce the number of uses of those instructions. | ||||
133 | /// | ||||
134 | /// Return the specified operand before it is turned into an undef. | ||||
135 | /// | ||||
136 | Value *AddSoonDeadInstToWorklist(Instruction &I, unsigned op) { | ||||
137 | Value *R = I.getOperand(op); | ||||
138 | |||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 139 | for (User::op_iterator i = I.op_begin(), e = I.op_end(); i != e; ++i) |
140 | if (Instruction *Op = dyn_cast<Instruction>(*i)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 141 | AddToWorkList(Op); |
142 | // Set the operand to undef to drop the use. | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 143 | *i = UndefValue::get(Op->getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 144 | } |
145 | |||||
146 | return R; | ||||
147 | } | ||||
148 | |||||
149 | public: | ||||
150 | virtual bool runOnFunction(Function &F); | ||||
151 | |||||
152 | bool DoOneIteration(Function &F, unsigned ItNum); | ||||
153 | |||||
154 | virtual void getAnalysisUsage(AnalysisUsage &AU) const { | ||||
155 | AU.addRequired<TargetData>(); | ||||
156 | AU.addPreservedID(LCSSAID); | ||||
157 | AU.setPreservesCFG(); | ||||
158 | } | ||||
159 | |||||
160 | TargetData &getTargetData() const { return *TD; } | ||||
161 | |||||
162 | // Visitation implementation - Implement instruction combining for different | ||||
163 | // instruction types. The semantics are as follows: | ||||
164 | // Return Value: | ||||
165 | // null - No change was made | ||||
166 | // I - Change was made, I is still valid, I may be dead though | ||||
167 | // otherwise - Change was made, replace I with returned instruction | ||||
168 | // | ||||
169 | Instruction *visitAdd(BinaryOperator &I); | ||||
170 | Instruction *visitSub(BinaryOperator &I); | ||||
171 | Instruction *visitMul(BinaryOperator &I); | ||||
172 | Instruction *visitURem(BinaryOperator &I); | ||||
173 | Instruction *visitSRem(BinaryOperator &I); | ||||
174 | Instruction *visitFRem(BinaryOperator &I); | ||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 175 | bool SimplifyDivRemOfSelect(BinaryOperator &I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 176 | Instruction *commonRemTransforms(BinaryOperator &I); |
177 | Instruction *commonIRemTransforms(BinaryOperator &I); | ||||
178 | Instruction *commonDivTransforms(BinaryOperator &I); | ||||
179 | Instruction *commonIDivTransforms(BinaryOperator &I); | ||||
180 | Instruction *visitUDiv(BinaryOperator &I); | ||||
181 | Instruction *visitSDiv(BinaryOperator &I); | ||||
182 | Instruction *visitFDiv(BinaryOperator &I); | ||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 183 | Instruction *FoldAndOfICmps(Instruction &I, ICmpInst *LHS, ICmpInst *RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 184 | Instruction *visitAnd(BinaryOperator &I); |
Chris Lattner | 0c678e5 | 2008-11-16 05:20:07 +0000 | [diff] [blame] | 185 | Instruction *FoldOrOfICmps(Instruction &I, ICmpInst *LHS, ICmpInst *RHS); |
Bill Wendling | 9912f71 | 2008-12-01 08:32:40 +0000 | [diff] [blame] | 186 | Instruction *FoldOrWithConstants(BinaryOperator &I, Value *Op, |
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 187 | Value *A, Value *B, Value *C); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 188 | Instruction *visitOr (BinaryOperator &I); |
189 | Instruction *visitXor(BinaryOperator &I); | ||||
190 | Instruction *visitShl(BinaryOperator &I); | ||||
191 | Instruction *visitAShr(BinaryOperator &I); | ||||
192 | Instruction *visitLShr(BinaryOperator &I); | ||||
193 | Instruction *commonShiftTransforms(BinaryOperator &I); | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 194 | Instruction *FoldFCmp_IntToFP_Cst(FCmpInst &I, Instruction *LHSI, |
195 | Constant *RHSC); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 196 | Instruction *visitFCmpInst(FCmpInst &I); |
197 | Instruction *visitICmpInst(ICmpInst &I); | ||||
198 | Instruction *visitICmpInstWithCastAndCast(ICmpInst &ICI); | ||||
199 | Instruction *visitICmpInstWithInstAndIntCst(ICmpInst &ICI, | ||||
200 | Instruction *LHS, | ||||
201 | ConstantInt *RHS); | ||||
202 | Instruction *FoldICmpDivCst(ICmpInst &ICI, BinaryOperator *DivI, | ||||
203 | ConstantInt *DivRHS); | ||||
204 | |||||
205 | Instruction *FoldGEPICmp(User *GEPLHS, Value *RHS, | ||||
206 | ICmpInst::Predicate Cond, Instruction &I); | ||||
207 | Instruction *FoldShiftByConstant(Value *Op0, ConstantInt *Op1, | ||||
208 | BinaryOperator &I); | ||||
209 | Instruction *commonCastTransforms(CastInst &CI); | ||||
210 | Instruction *commonIntCastTransforms(CastInst &CI); | ||||
211 | Instruction *commonPointerCastTransforms(CastInst &CI); | ||||
212 | Instruction *visitTrunc(TruncInst &CI); | ||||
213 | Instruction *visitZExt(ZExtInst &CI); | ||||
214 | Instruction *visitSExt(SExtInst &CI); | ||||
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 215 | Instruction *visitFPTrunc(FPTruncInst &CI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 216 | Instruction *visitFPExt(CastInst &CI); |
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 217 | Instruction *visitFPToUI(FPToUIInst &FI); |
218 | Instruction *visitFPToSI(FPToSIInst &FI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 219 | Instruction *visitUIToFP(CastInst &CI); |
220 | Instruction *visitSIToFP(CastInst &CI); | ||||
221 | Instruction *visitPtrToInt(CastInst &CI); | ||||
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 222 | Instruction *visitIntToPtr(IntToPtrInst &CI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 223 | Instruction *visitBitCast(BitCastInst &CI); |
224 | Instruction *FoldSelectOpOp(SelectInst &SI, Instruction *TI, | ||||
225 | Instruction *FI); | ||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 226 | Instruction *visitSelectInst(SelectInst &SI); |
227 | Instruction *visitSelectInstWithICmp(SelectInst &SI, ICmpInst *ICI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 228 | Instruction *visitCallInst(CallInst &CI); |
229 | Instruction *visitInvokeInst(InvokeInst &II); | ||||
230 | Instruction *visitPHINode(PHINode &PN); | ||||
231 | Instruction *visitGetElementPtrInst(GetElementPtrInst &GEP); | ||||
232 | Instruction *visitAllocationInst(AllocationInst &AI); | ||||
233 | Instruction *visitFreeInst(FreeInst &FI); | ||||
234 | Instruction *visitLoadInst(LoadInst &LI); | ||||
235 | Instruction *visitStoreInst(StoreInst &SI); | ||||
236 | Instruction *visitBranchInst(BranchInst &BI); | ||||
237 | Instruction *visitSwitchInst(SwitchInst &SI); | ||||
238 | Instruction *visitInsertElementInst(InsertElementInst &IE); | ||||
239 | Instruction *visitExtractElementInst(ExtractElementInst &EI); | ||||
240 | Instruction *visitShuffleVectorInst(ShuffleVectorInst &SVI); | ||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 241 | Instruction *visitExtractValueInst(ExtractValueInst &EV); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 242 | |
243 | // visitInstruction - Specify what to return for unhandled instructions... | ||||
244 | Instruction *visitInstruction(Instruction &I) { return 0; } | ||||
245 | |||||
246 | private: | ||||
247 | Instruction *visitCallSite(CallSite CS); | ||||
248 | bool transformConstExprCastCall(CallSite CS); | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 249 | Instruction *transformCallThroughTrampoline(CallSite CS); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 250 | Instruction *transformZExtICmp(ICmpInst *ICI, Instruction &CI, |
251 | bool DoXform = true); | ||||
Chris Lattner | 3554f97 | 2008-05-20 05:46:13 +0000 | [diff] [blame] | 252 | bool WillNotOverflowSignedAdd(Value *LHS, Value *RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 253 | |
254 | public: | ||||
255 | // InsertNewInstBefore - insert an instruction New before instruction Old | ||||
256 | // in the program. Add the new instruction to the worklist. | ||||
257 | // | ||||
258 | Instruction *InsertNewInstBefore(Instruction *New, Instruction &Old) { | ||||
259 | assert(New && New->getParent() == 0 && | ||||
260 | "New instruction already inserted into a basic block!"); | ||||
261 | BasicBlock *BB = Old.getParent(); | ||||
262 | BB->getInstList().insert(&Old, New); // Insert inst | ||||
263 | AddToWorkList(New); | ||||
264 | return New; | ||||
265 | } | ||||
266 | |||||
267 | /// InsertCastBefore - Insert a cast of V to TY before the instruction POS. | ||||
268 | /// This also adds the cast to the worklist. Finally, this returns the | ||||
269 | /// cast. | ||||
270 | Value *InsertCastBefore(Instruction::CastOps opc, Value *V, const Type *Ty, | ||||
271 | Instruction &Pos) { | ||||
272 | if (V->getType() == Ty) return V; | ||||
273 | |||||
274 | if (Constant *CV = dyn_cast<Constant>(V)) | ||||
275 | return ConstantExpr::getCast(opc, CV, Ty); | ||||
276 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 277 | Instruction *C = CastInst::Create(opc, V, Ty, V->getName(), &Pos); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 278 | AddToWorkList(C); |
279 | return C; | ||||
280 | } | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 281 | |
282 | Value *InsertBitCastBefore(Value *V, const Type *Ty, Instruction &Pos) { | ||||
283 | return InsertCastBefore(Instruction::BitCast, V, Ty, Pos); | ||||
284 | } | ||||
285 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 286 | |
287 | // ReplaceInstUsesWith - This method is to be used when an instruction is | ||||
288 | // found to be dead, replacable with another preexisting expression. Here | ||||
289 | // we add all uses of I to the worklist, replace all uses of I with the new | ||||
290 | // value, then return I, so that the inst combiner will know that I was | ||||
291 | // modified. | ||||
292 | // | ||||
293 | Instruction *ReplaceInstUsesWith(Instruction &I, Value *V) { | ||||
294 | AddUsersToWorkList(I); // Add all modified instrs to worklist | ||||
295 | if (&I != V) { | ||||
296 | I.replaceAllUsesWith(V); | ||||
297 | return &I; | ||||
298 | } else { | ||||
299 | // If we are replacing the instruction with itself, this must be in a | ||||
300 | // segment of unreachable code, so just clobber the instruction. | ||||
301 | I.replaceAllUsesWith(UndefValue::get(I.getType())); | ||||
302 | return &I; | ||||
303 | } | ||||
304 | } | ||||
305 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 306 | // EraseInstFromFunction - When dealing with an instruction that has side |
307 | // effects or produces a void value, we can't rely on DCE to delete the | ||||
308 | // instruction. Instead, visit methods should return the value returned by | ||||
309 | // this function. | ||||
310 | Instruction *EraseInstFromFunction(Instruction &I) { | ||||
311 | assert(I.use_empty() && "Cannot erase instruction that is used!"); | ||||
312 | AddUsesToWorkList(I); | ||||
313 | RemoveFromWorkList(&I); | ||||
314 | I.eraseFromParent(); | ||||
315 | return 0; // Don't do anything with FI | ||||
316 | } | ||||
Chris Lattner | a432bc7 | 2008-06-02 01:18:21 +0000 | [diff] [blame] | 317 | |
318 | void ComputeMaskedBits(Value *V, const APInt &Mask, APInt &KnownZero, | ||||
319 | APInt &KnownOne, unsigned Depth = 0) const { | ||||
320 | return llvm::ComputeMaskedBits(V, Mask, KnownZero, KnownOne, TD, Depth); | ||||
321 | } | ||||
322 | |||||
323 | bool MaskedValueIsZero(Value *V, const APInt &Mask, | ||||
324 | unsigned Depth = 0) const { | ||||
325 | return llvm::MaskedValueIsZero(V, Mask, TD, Depth); | ||||
326 | } | ||||
327 | unsigned ComputeNumSignBits(Value *Op, unsigned Depth = 0) const { | ||||
328 | return llvm::ComputeNumSignBits(Op, TD, Depth); | ||||
329 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 330 | |
331 | private: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 332 | |
333 | /// SimplifyCommutative - This performs a few simplifications for | ||||
334 | /// commutative operators. | ||||
335 | bool SimplifyCommutative(BinaryOperator &I); | ||||
336 | |||||
337 | /// SimplifyCompare - This reorders the operands of a CmpInst to get them in | ||||
338 | /// most-complex to least-complex order. | ||||
339 | bool SimplifyCompare(CmpInst &I); | ||||
340 | |||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 341 | /// SimplifyDemandedUseBits - Attempts to replace V with a simpler value |
342 | /// based on the demanded bits. | ||||
343 | Value *SimplifyDemandedUseBits(Value *V, APInt DemandedMask, | ||||
344 | APInt& KnownZero, APInt& KnownOne, | ||||
345 | unsigned Depth); | ||||
346 | bool SimplifyDemandedBits(Use &U, APInt DemandedMask, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 347 | APInt& KnownZero, APInt& KnownOne, |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 348 | unsigned Depth=0); |
349 | |||||
350 | /// SimplifyDemandedInstructionBits - Inst is an integer instruction that | ||||
351 | /// SimplifyDemandedBits knows about. See if the instruction has any | ||||
352 | /// properties that allow us to simplify its operands. | ||||
353 | bool SimplifyDemandedInstructionBits(Instruction &Inst); | ||||
354 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 355 | Value *SimplifyDemandedVectorElts(Value *V, uint64_t DemandedElts, |
356 | uint64_t &UndefElts, unsigned Depth = 0); | ||||
357 | |||||
358 | // FoldOpIntoPhi - Given a binary operator or cast instruction which has a | ||||
359 | // PHI node as operand #0, see if we can fold the instruction into the PHI | ||||
360 | // (which is only possible if all operands to the PHI are constants). | ||||
361 | Instruction *FoldOpIntoPhi(Instruction &I); | ||||
362 | |||||
363 | // FoldPHIArgOpIntoPHI - If all operands to a PHI node are the same "unary" | ||||
364 | // operator and they all are only used by the PHI, PHI together their | ||||
365 | // inputs, and do the operation once, to the result of the PHI. | ||||
366 | Instruction *FoldPHIArgOpIntoPHI(PHINode &PN); | ||||
367 | Instruction *FoldPHIArgBinOpIntoPHI(PHINode &PN); | ||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 368 | Instruction *FoldPHIArgGEPIntoPHI(PHINode &PN); |
369 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 370 | |
371 | Instruction *OptAndOp(Instruction *Op, ConstantInt *OpRHS, | ||||
372 | ConstantInt *AndRHS, BinaryOperator &TheAnd); | ||||
373 | |||||
374 | Value *FoldLogicalPlusAnd(Value *LHS, Value *RHS, ConstantInt *Mask, | ||||
375 | bool isSub, Instruction &I); | ||||
376 | Instruction *InsertRangeTest(Value *V, Constant *Lo, Constant *Hi, | ||||
377 | bool isSigned, bool Inside, Instruction &IB); | ||||
378 | Instruction *PromoteCastOfAllocation(BitCastInst &CI, AllocationInst &AI); | ||||
379 | Instruction *MatchBSwap(BinaryOperator &I); | ||||
380 | bool SimplifyStoreAtEndOfBlock(StoreInst &SI); | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 381 | Instruction *SimplifyMemTransfer(MemIntrinsic *MI); |
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 382 | Instruction *SimplifyMemSet(MemSetInst *MI); |
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 383 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 384 | |
385 | Value *EvaluateInDifferentType(Value *V, const Type *Ty, bool isSigned); | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 386 | |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 387 | bool CanEvaluateInDifferentType(Value *V, const IntegerType *Ty, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 388 | unsigned CastOpc, int &NumCastsRemoved); |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 389 | unsigned GetOrEnforceKnownAlignment(Value *V, |
390 | unsigned PrefAlign = 0); | ||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 391 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 392 | }; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 393 | } |
394 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 395 | char InstCombiner::ID = 0; |
396 | static RegisterPass<InstCombiner> | ||||
397 | X("instcombine", "Combine redundant instructions"); | ||||
398 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 399 | // getComplexity: Assign a complexity or rank value to LLVM Values... |
400 | // 0 -> undef, 1 -> Const, 2 -> Other, 3 -> Arg, 3 -> Unary, 4 -> OtherInst | ||||
401 | static unsigned getComplexity(Value *V) { | ||||
402 | if (isa<Instruction>(V)) { | ||||
403 | if (BinaryOperator::isNeg(V) || BinaryOperator::isNot(V)) | ||||
404 | return 3; | ||||
405 | return 4; | ||||
406 | } | ||||
407 | if (isa<Argument>(V)) return 3; | ||||
408 | return isa<Constant>(V) ? (isa<UndefValue>(V) ? 0 : 1) : 2; | ||||
409 | } | ||||
410 | |||||
411 | // isOnlyUse - Return true if this instruction will be deleted if we stop using | ||||
412 | // it. | ||||
413 | static bool isOnlyUse(Value *V) { | ||||
414 | return V->hasOneUse() || isa<Constant>(V); | ||||
415 | } | ||||
416 | |||||
417 | // getPromotedType - Return the specified type promoted as it would be to pass | ||||
418 | // though a va_arg area... | ||||
419 | static const Type *getPromotedType(const Type *Ty) { | ||||
420 | if (const IntegerType* ITy = dyn_cast<IntegerType>(Ty)) { | ||||
421 | if (ITy->getBitWidth() < 32) | ||||
422 | return Type::Int32Ty; | ||||
423 | } | ||||
424 | return Ty; | ||||
425 | } | ||||
426 | |||||
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 427 | /// getBitCastOperand - If the specified operand is a CastInst, a constant |
428 | /// expression bitcast, or a GetElementPtrInst with all zero indices, return the | ||||
429 | /// operand value, otherwise return null. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 430 | static Value *getBitCastOperand(Value *V) { |
431 | if (BitCastInst *I = dyn_cast<BitCastInst>(V)) | ||||
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 432 | // BitCastInst? |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 433 | return I->getOperand(0); |
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 434 | else if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(V)) { |
435 | // GetElementPtrInst? | ||||
436 | if (GEP->hasAllZeroIndices()) | ||||
437 | return GEP->getOperand(0); | ||||
438 | } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(V)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 439 | if (CE->getOpcode() == Instruction::BitCast) |
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 440 | // BitCast ConstantExp? |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 441 | return CE->getOperand(0); |
Matthijs Kooijman | 5e2a318 | 2008-10-13 15:17:01 +0000 | [diff] [blame] | 442 | else if (CE->getOpcode() == Instruction::GetElementPtr) { |
443 | // GetElementPtr ConstantExp? | ||||
444 | for (User::op_iterator I = CE->op_begin() + 1, E = CE->op_end(); | ||||
445 | I != E; ++I) { | ||||
446 | ConstantInt *CI = dyn_cast<ConstantInt>(I); | ||||
447 | if (!CI || !CI->isZero()) | ||||
448 | // Any non-zero indices? Not cast-like. | ||||
449 | return 0; | ||||
450 | } | ||||
451 | // All-zero indices? This is just like casting. | ||||
452 | return CE->getOperand(0); | ||||
453 | } | ||||
454 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 455 | return 0; |
456 | } | ||||
457 | |||||
458 | /// This function is a wrapper around CastInst::isEliminableCastPair. It | ||||
459 | /// simply extracts arguments and returns what that function returns. | ||||
460 | static Instruction::CastOps | ||||
461 | isEliminableCastPair( | ||||
462 | const CastInst *CI, ///< The first cast instruction | ||||
463 | unsigned opcode, ///< The opcode of the second cast instruction | ||||
464 | const Type *DstTy, ///< The target type for the second cast instruction | ||||
465 | TargetData *TD ///< The target data for pointer size | ||||
466 | ) { | ||||
467 | |||||
468 | const Type *SrcTy = CI->getOperand(0)->getType(); // A from above | ||||
469 | const Type *MidTy = CI->getType(); // B from above | ||||
470 | |||||
471 | // Get the opcodes of the two Cast instructions | ||||
472 | Instruction::CastOps firstOp = Instruction::CastOps(CI->getOpcode()); | ||||
473 | Instruction::CastOps secondOp = Instruction::CastOps(opcode); | ||||
474 | |||||
475 | return Instruction::CastOps( | ||||
476 | CastInst::isEliminableCastPair(firstOp, secondOp, SrcTy, MidTy, | ||||
477 | DstTy, TD->getIntPtrType())); | ||||
478 | } | ||||
479 | |||||
480 | /// ValueRequiresCast - Return true if the cast from "V to Ty" actually results | ||||
481 | /// in any code being generated. It does not require codegen if V is simple | ||||
482 | /// enough or if the cast can be folded into other casts. | ||||
483 | static bool ValueRequiresCast(Instruction::CastOps opcode, const Value *V, | ||||
484 | const Type *Ty, TargetData *TD) { | ||||
485 | if (V->getType() == Ty || isa<Constant>(V)) return false; | ||||
486 | |||||
487 | // If this is another cast that can be eliminated, it isn't codegen either. | ||||
488 | if (const CastInst *CI = dyn_cast<CastInst>(V)) | ||||
489 | if (isEliminableCastPair(CI, opcode, Ty, TD)) | ||||
490 | return false; | ||||
491 | return true; | ||||
492 | } | ||||
493 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 494 | // SimplifyCommutative - This performs a few simplifications for commutative |
495 | // operators: | ||||
496 | // | ||||
497 | // 1. Order operands such that they are listed from right (least complex) to | ||||
498 | // left (most complex). This puts constants before unary operators before | ||||
499 | // binary operators. | ||||
500 | // | ||||
501 | // 2. Transform: (op (op V, C1), C2) ==> (op V, (op C1, C2)) | ||||
502 | // 3. Transform: (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2)) | ||||
503 | // | ||||
504 | bool InstCombiner::SimplifyCommutative(BinaryOperator &I) { | ||||
505 | bool Changed = false; | ||||
506 | if (getComplexity(I.getOperand(0)) < getComplexity(I.getOperand(1))) | ||||
507 | Changed = !I.swapOperands(); | ||||
508 | |||||
509 | if (!I.isAssociative()) return Changed; | ||||
510 | Instruction::BinaryOps Opcode = I.getOpcode(); | ||||
511 | if (BinaryOperator *Op = dyn_cast<BinaryOperator>(I.getOperand(0))) | ||||
512 | if (Op->getOpcode() == Opcode && isa<Constant>(Op->getOperand(1))) { | ||||
513 | if (isa<Constant>(I.getOperand(1))) { | ||||
514 | Constant *Folded = ConstantExpr::get(I.getOpcode(), | ||||
515 | cast<Constant>(I.getOperand(1)), | ||||
516 | cast<Constant>(Op->getOperand(1))); | ||||
517 | I.setOperand(0, Op->getOperand(0)); | ||||
518 | I.setOperand(1, Folded); | ||||
519 | return true; | ||||
520 | } else if (BinaryOperator *Op1=dyn_cast<BinaryOperator>(I.getOperand(1))) | ||||
521 | if (Op1->getOpcode() == Opcode && isa<Constant>(Op1->getOperand(1)) && | ||||
522 | isOnlyUse(Op) && isOnlyUse(Op1)) { | ||||
523 | Constant *C1 = cast<Constant>(Op->getOperand(1)); | ||||
524 | Constant *C2 = cast<Constant>(Op1->getOperand(1)); | ||||
525 | |||||
526 | // Fold (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2)) | ||||
527 | Constant *Folded = ConstantExpr::get(I.getOpcode(), C1, C2); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 528 | Instruction *New = BinaryOperator::Create(Opcode, Op->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 529 | Op1->getOperand(0), |
530 | Op1->getName(), &I); | ||||
531 | AddToWorkList(New); | ||||
532 | I.setOperand(0, New); | ||||
533 | I.setOperand(1, Folded); | ||||
534 | return true; | ||||
535 | } | ||||
536 | } | ||||
537 | return Changed; | ||||
538 | } | ||||
539 | |||||
540 | /// SimplifyCompare - For a CmpInst this function just orders the operands | ||||
541 | /// so that theyare listed from right (least complex) to left (most complex). | ||||
542 | /// This puts constants before unary operators before binary operators. | ||||
543 | bool InstCombiner::SimplifyCompare(CmpInst &I) { | ||||
544 | if (getComplexity(I.getOperand(0)) >= getComplexity(I.getOperand(1))) | ||||
545 | return false; | ||||
546 | I.swapOperands(); | ||||
547 | // Compare instructions are not associative so there's nothing else we can do. | ||||
548 | return true; | ||||
549 | } | ||||
550 | |||||
551 | // dyn_castNegVal - Given a 'sub' instruction, return the RHS of the instruction | ||||
552 | // if the LHS is a constant zero (which is the 'negate' form). | ||||
553 | // | ||||
554 | static inline Value *dyn_castNegVal(Value *V) { | ||||
555 | if (BinaryOperator::isNeg(V)) | ||||
556 | return BinaryOperator::getNegArgument(V); | ||||
557 | |||||
558 | // Constants can be considered to be negated values if they can be folded. | ||||
559 | if (ConstantInt *C = dyn_cast<ConstantInt>(V)) | ||||
560 | return ConstantExpr::getNeg(C); | ||||
Nick Lewycky | 58867bc | 2008-05-23 04:54:45 +0000 | [diff] [blame] | 561 | |
562 | if (ConstantVector *C = dyn_cast<ConstantVector>(V)) | ||||
563 | if (C->getType()->getElementType()->isInteger()) | ||||
564 | return ConstantExpr::getNeg(C); | ||||
565 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 566 | return 0; |
567 | } | ||||
568 | |||||
569 | static inline Value *dyn_castNotVal(Value *V) { | ||||
570 | if (BinaryOperator::isNot(V)) | ||||
571 | return BinaryOperator::getNotArgument(V); | ||||
572 | |||||
573 | // Constants can be considered to be not'ed values... | ||||
574 | if (ConstantInt *C = dyn_cast<ConstantInt>(V)) | ||||
575 | return ConstantInt::get(~C->getValue()); | ||||
576 | return 0; | ||||
577 | } | ||||
578 | |||||
579 | // dyn_castFoldableMul - If this value is a multiply that can be folded into | ||||
580 | // other computations (because it has a constant operand), return the | ||||
581 | // non-constant operand of the multiply, and set CST to point to the multiplier. | ||||
582 | // Otherwise, return null. | ||||
583 | // | ||||
584 | static inline Value *dyn_castFoldableMul(Value *V, ConstantInt *&CST) { | ||||
585 | if (V->hasOneUse() && V->getType()->isInteger()) | ||||
586 | if (Instruction *I = dyn_cast<Instruction>(V)) { | ||||
587 | if (I->getOpcode() == Instruction::Mul) | ||||
588 | if ((CST = dyn_cast<ConstantInt>(I->getOperand(1)))) | ||||
589 | return I->getOperand(0); | ||||
590 | if (I->getOpcode() == Instruction::Shl) | ||||
591 | if ((CST = dyn_cast<ConstantInt>(I->getOperand(1)))) { | ||||
592 | // The multiplier is really 1 << CST. | ||||
593 | uint32_t BitWidth = cast<IntegerType>(V->getType())->getBitWidth(); | ||||
594 | uint32_t CSTVal = CST->getLimitedValue(BitWidth); | ||||
595 | CST = ConstantInt::get(APInt(BitWidth, 1).shl(CSTVal)); | ||||
596 | return I->getOperand(0); | ||||
597 | } | ||||
598 | } | ||||
599 | return 0; | ||||
600 | } | ||||
601 | |||||
602 | /// dyn_castGetElementPtr - If this is a getelementptr instruction or constant | ||||
603 | /// expression, return it. | ||||
604 | static User *dyn_castGetElementPtr(Value *V) { | ||||
605 | if (isa<GetElementPtrInst>(V)) return cast<User>(V); | ||||
606 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(V)) | ||||
607 | if (CE->getOpcode() == Instruction::GetElementPtr) | ||||
608 | return cast<User>(V); | ||||
609 | return false; | ||||
610 | } | ||||
611 | |||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 612 | /// getOpcode - If this is an Instruction or a ConstantExpr, return the |
613 | /// opcode value. Otherwise return UserOp1. | ||||
Dan Gohman | 8c39786 | 2008-05-29 19:53:46 +0000 | [diff] [blame] | 614 | static unsigned getOpcode(const Value *V) { |
615 | if (const Instruction *I = dyn_cast<Instruction>(V)) | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 616 | return I->getOpcode(); |
Dan Gohman | 8c39786 | 2008-05-29 19:53:46 +0000 | [diff] [blame] | 617 | if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(V)) |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 618 | return CE->getOpcode(); |
619 | // Use UserOp1 to mean there's no opcode. | ||||
620 | return Instruction::UserOp1; | ||||
621 | } | ||||
622 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 623 | /// AddOne - Add one to a ConstantInt |
624 | static ConstantInt *AddOne(ConstantInt *C) { | ||||
625 | APInt Val(C->getValue()); | ||||
626 | return ConstantInt::get(++Val); | ||||
627 | } | ||||
628 | /// SubOne - Subtract one from a ConstantInt | ||||
629 | static ConstantInt *SubOne(ConstantInt *C) { | ||||
630 | APInt Val(C->getValue()); | ||||
631 | return ConstantInt::get(--Val); | ||||
632 | } | ||||
633 | /// Add - Add two ConstantInts together | ||||
634 | static ConstantInt *Add(ConstantInt *C1, ConstantInt *C2) { | ||||
635 | return ConstantInt::get(C1->getValue() + C2->getValue()); | ||||
636 | } | ||||
637 | /// And - Bitwise AND two ConstantInts together | ||||
638 | static ConstantInt *And(ConstantInt *C1, ConstantInt *C2) { | ||||
639 | return ConstantInt::get(C1->getValue() & C2->getValue()); | ||||
640 | } | ||||
641 | /// Subtract - Subtract one ConstantInt from another | ||||
642 | static ConstantInt *Subtract(ConstantInt *C1, ConstantInt *C2) { | ||||
643 | return ConstantInt::get(C1->getValue() - C2->getValue()); | ||||
644 | } | ||||
645 | /// Multiply - Multiply two ConstantInts together | ||||
646 | static ConstantInt *Multiply(ConstantInt *C1, ConstantInt *C2) { | ||||
647 | return ConstantInt::get(C1->getValue() * C2->getValue()); | ||||
648 | } | ||||
Nick Lewycky | 9d798f9 | 2008-02-18 22:48:05 +0000 | [diff] [blame] | 649 | /// MultiplyOverflows - True if the multiply can not be expressed in an int |
650 | /// this size. | ||||
651 | static bool MultiplyOverflows(ConstantInt *C1, ConstantInt *C2, bool sign) { | ||||
652 | uint32_t W = C1->getBitWidth(); | ||||
653 | APInt LHSExt = C1->getValue(), RHSExt = C2->getValue(); | ||||
654 | if (sign) { | ||||
655 | LHSExt.sext(W * 2); | ||||
656 | RHSExt.sext(W * 2); | ||||
657 | } else { | ||||
658 | LHSExt.zext(W * 2); | ||||
659 | RHSExt.zext(W * 2); | ||||
660 | } | ||||
661 | |||||
662 | APInt MulExt = LHSExt * RHSExt; | ||||
663 | |||||
664 | if (sign) { | ||||
665 | APInt Min = APInt::getSignedMinValue(W).sext(W * 2); | ||||
666 | APInt Max = APInt::getSignedMaxValue(W).sext(W * 2); | ||||
667 | return MulExt.slt(Min) || MulExt.sgt(Max); | ||||
668 | } else | ||||
669 | return MulExt.ugt(APInt::getLowBitsSet(W * 2, W)); | ||||
670 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 671 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 672 | |
673 | /// ShrinkDemandedConstant - Check to see if the specified operand of the | ||||
674 | /// specified instruction is a constant integer. If so, check to see if there | ||||
675 | /// are any bits set in the constant that are not demanded. If so, shrink the | ||||
676 | /// constant and return true. | ||||
677 | static bool ShrinkDemandedConstant(Instruction *I, unsigned OpNo, | ||||
678 | APInt Demanded) { | ||||
679 | assert(I && "No instruction?"); | ||||
680 | assert(OpNo < I->getNumOperands() && "Operand index too large"); | ||||
681 | |||||
682 | // If the operand is not a constant integer, nothing to do. | ||||
683 | ConstantInt *OpC = dyn_cast<ConstantInt>(I->getOperand(OpNo)); | ||||
684 | if (!OpC) return false; | ||||
685 | |||||
686 | // If there are no bits set that aren't demanded, nothing to do. | ||||
687 | Demanded.zextOrTrunc(OpC->getValue().getBitWidth()); | ||||
688 | if ((~Demanded & OpC->getValue()) == 0) | ||||
689 | return false; | ||||
690 | |||||
691 | // This instruction is producing bits that are not demanded. Shrink the RHS. | ||||
692 | Demanded &= OpC->getValue(); | ||||
693 | I->setOperand(OpNo, ConstantInt::get(Demanded)); | ||||
694 | return true; | ||||
695 | } | ||||
696 | |||||
697 | // ComputeSignedMinMaxValuesFromKnownBits - Given a signed integer type and a | ||||
698 | // set of known zero and one bits, compute the maximum and minimum values that | ||||
699 | // could have the specified known zero and known one bits, returning them in | ||||
700 | // min/max. | ||||
701 | static void ComputeSignedMinMaxValuesFromKnownBits(const Type *Ty, | ||||
702 | const APInt& KnownZero, | ||||
703 | const APInt& KnownOne, | ||||
704 | APInt& Min, APInt& Max) { | ||||
705 | uint32_t BitWidth = cast<IntegerType>(Ty)->getBitWidth(); | ||||
706 | assert(KnownZero.getBitWidth() == BitWidth && | ||||
707 | KnownOne.getBitWidth() == BitWidth && | ||||
708 | Min.getBitWidth() == BitWidth && Max.getBitWidth() == BitWidth && | ||||
709 | "Ty, KnownZero, KnownOne and Min, Max must have equal bitwidth."); | ||||
710 | APInt UnknownBits = ~(KnownZero|KnownOne); | ||||
711 | |||||
712 | // The minimum value is when all unknown bits are zeros, EXCEPT for the sign | ||||
713 | // bit if it is unknown. | ||||
714 | Min = KnownOne; | ||||
715 | Max = KnownOne|UnknownBits; | ||||
716 | |||||
717 | if (UnknownBits[BitWidth-1]) { // Sign bit is unknown | ||||
718 | Min.set(BitWidth-1); | ||||
719 | Max.clear(BitWidth-1); | ||||
720 | } | ||||
721 | } | ||||
722 | |||||
723 | // ComputeUnsignedMinMaxValuesFromKnownBits - Given an unsigned integer type and | ||||
724 | // a 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 ComputeUnsignedMinMaxValuesFromKnownBits(const Type *Ty, | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 728 | const APInt &KnownZero, |
729 | const APInt &KnownOne, | ||||
730 | APInt &Min, APInt &Max) { | ||||
731 | uint32_t BitWidth = cast<IntegerType>(Ty)->getBitWidth(); BitWidth = BitWidth; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 732 | assert(KnownZero.getBitWidth() == BitWidth && |
733 | KnownOne.getBitWidth() == BitWidth && | ||||
734 | Min.getBitWidth() == BitWidth && Max.getBitWidth() && | ||||
735 | "Ty, KnownZero, KnownOne and Min, Max must have equal bitwidth."); | ||||
736 | APInt UnknownBits = ~(KnownZero|KnownOne); | ||||
737 | |||||
738 | // The minimum value is when the unknown bits are all zeros. | ||||
739 | Min = KnownOne; | ||||
740 | // The maximum value is when the unknown bits are all ones. | ||||
741 | Max = KnownOne|UnknownBits; | ||||
742 | } | ||||
743 | |||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 744 | /// SimplifyDemandedInstructionBits - Inst is an integer instruction that |
745 | /// SimplifyDemandedBits knows about. See if the instruction has any | ||||
746 | /// properties that allow us to simplify its operands. | ||||
747 | bool InstCombiner::SimplifyDemandedInstructionBits(Instruction &Inst) { | ||||
748 | unsigned BitWidth = cast<IntegerType>(Inst.getType())->getBitWidth(); | ||||
749 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
750 | APInt DemandedMask(APInt::getAllOnesValue(BitWidth)); | ||||
751 | |||||
752 | Value *V = SimplifyDemandedUseBits(&Inst, DemandedMask, | ||||
753 | KnownZero, KnownOne, 0); | ||||
754 | if (V == 0) return false; | ||||
755 | if (V == &Inst) return true; | ||||
756 | ReplaceInstUsesWith(Inst, V); | ||||
757 | return true; | ||||
758 | } | ||||
759 | |||||
760 | /// SimplifyDemandedBits - This form of SimplifyDemandedBits simplifies the | ||||
761 | /// specified instruction operand if possible, updating it in place. It returns | ||||
762 | /// true if it made any change and false otherwise. | ||||
763 | bool InstCombiner::SimplifyDemandedBits(Use &U, APInt DemandedMask, | ||||
764 | APInt &KnownZero, APInt &KnownOne, | ||||
765 | unsigned Depth) { | ||||
766 | Value *NewVal = SimplifyDemandedUseBits(U.get(), DemandedMask, | ||||
767 | KnownZero, KnownOne, Depth); | ||||
768 | if (NewVal == 0) return false; | ||||
769 | U.set(NewVal); | ||||
770 | return true; | ||||
771 | } | ||||
772 | |||||
773 | |||||
774 | /// SimplifyDemandedUseBits - This function attempts to replace V with a simpler | ||||
775 | /// value based on the demanded bits. When this function is called, it is known | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 776 | /// that only the bits set in DemandedMask of the result of V are ever used |
777 | /// downstream. Consequently, depending on the mask and V, it may be possible | ||||
778 | /// to replace V with a constant or one of its operands. In such cases, this | ||||
779 | /// function does the replacement and returns true. In all other cases, it | ||||
780 | /// returns false after analyzing the expression and setting KnownOne and known | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 781 | /// to be one in the expression. KnownZero contains all the bits that are known |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 782 | /// to be zero in the expression. These are provided to potentially allow the |
783 | /// caller (which might recursively be SimplifyDemandedBits itself) to simplify | ||||
784 | /// the expression. KnownOne and KnownZero always follow the invariant that | ||||
785 | /// KnownOne & KnownZero == 0. That is, a bit can't be both 1 and 0. Note that | ||||
786 | /// the bits in KnownOne and KnownZero may only be accurate for those bits set | ||||
787 | /// in DemandedMask. Note also that the bitwidth of V, DemandedMask, KnownZero | ||||
788 | /// and KnownOne must all be the same. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 789 | /// |
790 | /// This returns null if it did not change anything and it permits no | ||||
791 | /// simplification. This returns V itself if it did some simplification of V's | ||||
792 | /// operands based on the information about what bits are demanded. This returns | ||||
793 | /// some other non-null value if it found out that V is equal to another value | ||||
794 | /// in the context where the specified bits are demanded, but not for all users. | ||||
795 | Value *InstCombiner::SimplifyDemandedUseBits(Value *V, APInt DemandedMask, | ||||
796 | APInt &KnownZero, APInt &KnownOne, | ||||
797 | unsigned Depth) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 798 | assert(V != 0 && "Null pointer of Value???"); |
799 | assert(Depth <= 6 && "Limit Search Depth"); | ||||
800 | uint32_t BitWidth = DemandedMask.getBitWidth(); | ||||
801 | const IntegerType *VTy = cast<IntegerType>(V->getType()); | ||||
802 | assert(VTy->getBitWidth() == BitWidth && | ||||
803 | KnownZero.getBitWidth() == BitWidth && | ||||
804 | KnownOne.getBitWidth() == BitWidth && | ||||
805 | "Value *V, DemandedMask, KnownZero and KnownOne \ | ||||
806 | must have same BitWidth"); | ||||
807 | if (ConstantInt *CI = dyn_cast<ConstantInt>(V)) { | ||||
808 | // We know all of the bits for a constant! | ||||
809 | KnownOne = CI->getValue() & DemandedMask; | ||||
810 | KnownZero = ~KnownOne & DemandedMask; | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 811 | return 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 812 | } |
813 | |||||
Chris Lattner | c5d7e4e | 2009-01-31 07:26:06 +0000 | [diff] [blame] | 814 | KnownZero.clear(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 815 | KnownOne.clear(); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 816 | if (DemandedMask == 0) { // Not demanding any bits from V. |
817 | if (isa<UndefValue>(V)) | ||||
818 | return 0; | ||||
819 | return UndefValue::get(VTy); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 820 | } |
821 | |||||
Chris Lattner | 0881733 | 2009-01-31 08:24:16 +0000 | [diff] [blame] | 822 | if (Depth == 6) // Limit search depth. |
823 | return 0; | ||||
824 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 825 | Instruction *I = dyn_cast<Instruction>(V); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 826 | if (!I) return 0; // Only analyze instructions. |
Chris Lattner | 0881733 | 2009-01-31 08:24:16 +0000 | [diff] [blame] | 827 | |
Chris Lattner | cd8d44c | 2009-01-31 08:40:03 +0000 | [diff] [blame] | 828 | APInt LHSKnownZero(BitWidth, 0), LHSKnownOne(BitWidth, 0); |
829 | APInt &RHSKnownZero = KnownZero, &RHSKnownOne = KnownOne; | ||||
830 | |||||
Chris Lattner | 0881733 | 2009-01-31 08:24:16 +0000 | [diff] [blame] | 831 | // If there are multiple uses of this value and we aren't at the root, then |
832 | // we can't do any simplifications of the operands, because DemandedMask | ||||
833 | // only reflects the bits demanded by *one* of the users. | ||||
834 | if (Depth != 0 && !I->hasOneUse()) { | ||||
Chris Lattner | cd8d44c | 2009-01-31 08:40:03 +0000 | [diff] [blame] | 835 | // Despite the fact that we can't simplify this instruction in all User's |
836 | // context, we can at least compute the knownzero/knownone bits, and we can | ||||
837 | // do simplifications that apply to *just* the one user if we know that | ||||
838 | // this instruction has a simpler value in that context. | ||||
839 | if (I->getOpcode() == Instruction::And) { | ||||
840 | // If either the LHS or the RHS are Zero, the result is zero. | ||||
841 | ComputeMaskedBits(I->getOperand(1), DemandedMask, | ||||
842 | RHSKnownZero, RHSKnownOne, Depth+1); | ||||
843 | ComputeMaskedBits(I->getOperand(0), DemandedMask & ~RHSKnownZero, | ||||
844 | LHSKnownZero, LHSKnownOne, Depth+1); | ||||
845 | |||||
846 | // If all of the demanded bits are known 1 on one side, return the other. | ||||
847 | // These bits cannot contribute to the result of the 'and' in this | ||||
848 | // context. | ||||
849 | if ((DemandedMask & ~LHSKnownZero & RHSKnownOne) == | ||||
850 | (DemandedMask & ~LHSKnownZero)) | ||||
851 | return I->getOperand(0); | ||||
852 | if ((DemandedMask & ~RHSKnownZero & LHSKnownOne) == | ||||
853 | (DemandedMask & ~RHSKnownZero)) | ||||
854 | return I->getOperand(1); | ||||
855 | |||||
856 | // If all of the demanded bits in the inputs are known zeros, return zero. | ||||
857 | if ((DemandedMask & (RHSKnownZero|LHSKnownZero)) == DemandedMask) | ||||
858 | return Constant::getNullValue(VTy); | ||||
859 | |||||
860 | } else if (I->getOpcode() == Instruction::Or) { | ||||
861 | // We can simplify (X|Y) -> X or Y in the user's context if we know that | ||||
862 | // only bits from X or Y are demanded. | ||||
863 | |||||
864 | // If either the LHS or the RHS are One, the result is One. | ||||
865 | ComputeMaskedBits(I->getOperand(1), DemandedMask, | ||||
866 | RHSKnownZero, RHSKnownOne, Depth+1); | ||||
867 | ComputeMaskedBits(I->getOperand(0), DemandedMask & ~RHSKnownOne, | ||||
868 | LHSKnownZero, LHSKnownOne, Depth+1); | ||||
869 | |||||
870 | // If all of the demanded bits are known zero on one side, return the | ||||
871 | // other. These bits cannot contribute to the result of the 'or' in this | ||||
872 | // context. | ||||
873 | if ((DemandedMask & ~LHSKnownOne & RHSKnownZero) == | ||||
874 | (DemandedMask & ~LHSKnownOne)) | ||||
875 | return I->getOperand(0); | ||||
876 | if ((DemandedMask & ~RHSKnownOne & LHSKnownZero) == | ||||
877 | (DemandedMask & ~RHSKnownOne)) | ||||
878 | return I->getOperand(1); | ||||
879 | |||||
880 | // If all of the potentially set bits on one side are known to be set on | ||||
881 | // the other side, just use the 'other' side. | ||||
882 | if ((DemandedMask & (~RHSKnownZero) & LHSKnownOne) == | ||||
883 | (DemandedMask & (~RHSKnownZero))) | ||||
884 | return I->getOperand(0); | ||||
885 | if ((DemandedMask & (~LHSKnownZero) & RHSKnownOne) == | ||||
886 | (DemandedMask & (~LHSKnownZero))) | ||||
887 | return I->getOperand(1); | ||||
888 | } | ||||
889 | |||||
Chris Lattner | 0881733 | 2009-01-31 08:24:16 +0000 | [diff] [blame] | 890 | // Compute the KnownZero/KnownOne bits to simplify things downstream. |
891 | ComputeMaskedBits(I, DemandedMask, KnownZero, KnownOne, Depth); | ||||
892 | return 0; | ||||
893 | } | ||||
894 | |||||
895 | // If this is the root being simplified, allow it to have multiple uses, | ||||
896 | // just set the DemandedMask to all bits so that we can try to simplify the | ||||
897 | // operands. This allows visitTruncInst (for example) to simplify the | ||||
898 | // operand of a trunc without duplicating all the logic below. | ||||
899 | if (Depth == 0 && !V->hasOneUse()) | ||||
900 | DemandedMask = APInt::getAllOnesValue(BitWidth); | ||||
901 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 902 | switch (I->getOpcode()) { |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 903 | default: |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 904 | ComputeMaskedBits(I, DemandedMask, RHSKnownZero, RHSKnownOne, Depth); |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 905 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 906 | case Instruction::And: |
907 | // If either the LHS or the RHS are Zero, the result is zero. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 908 | if (SimplifyDemandedBits(I->getOperandUse(1), DemandedMask, |
909 | RHSKnownZero, RHSKnownOne, Depth+1) || | ||||
910 | SimplifyDemandedBits(I->getOperandUse(0), DemandedMask & ~RHSKnownZero, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 911 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 912 | return I; |
913 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
914 | assert(!(LHSKnownZero & LHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 915 | |
916 | // If all of the demanded bits are known 1 on one side, return the other. | ||||
917 | // These bits cannot contribute to the result of the 'and'. | ||||
918 | if ((DemandedMask & ~LHSKnownZero & RHSKnownOne) == | ||||
919 | (DemandedMask & ~LHSKnownZero)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 920 | return I->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 921 | if ((DemandedMask & ~RHSKnownZero & LHSKnownOne) == |
922 | (DemandedMask & ~RHSKnownZero)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 923 | return I->getOperand(1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 924 | |
925 | // If all of the demanded bits in the inputs are known zeros, return zero. | ||||
926 | if ((DemandedMask & (RHSKnownZero|LHSKnownZero)) == DemandedMask) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 927 | return Constant::getNullValue(VTy); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 928 | |
929 | // If the RHS is a constant, see if we can simplify it. | ||||
930 | if (ShrinkDemandedConstant(I, 1, DemandedMask & ~LHSKnownZero)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 931 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 932 | |
933 | // Output known-1 bits are only known if set in both the LHS & RHS. | ||||
934 | RHSKnownOne &= LHSKnownOne; | ||||
935 | // Output known-0 are known to be clear if zero in either the LHS | RHS. | ||||
936 | RHSKnownZero |= LHSKnownZero; | ||||
937 | break; | ||||
938 | case Instruction::Or: | ||||
939 | // If either the LHS or the RHS are One, the result is One. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 940 | if (SimplifyDemandedBits(I->getOperandUse(1), DemandedMask, |
941 | RHSKnownZero, RHSKnownOne, Depth+1) || | ||||
942 | SimplifyDemandedBits(I->getOperandUse(0), DemandedMask & ~RHSKnownOne, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 943 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 944 | return I; |
945 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
946 | assert(!(LHSKnownZero & LHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 947 | |
948 | // If all of the demanded bits are known zero on one side, return the other. | ||||
949 | // These bits cannot contribute to the result of the 'or'. | ||||
950 | if ((DemandedMask & ~LHSKnownOne & RHSKnownZero) == | ||||
951 | (DemandedMask & ~LHSKnownOne)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 952 | return I->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 953 | if ((DemandedMask & ~RHSKnownOne & LHSKnownZero) == |
954 | (DemandedMask & ~RHSKnownOne)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 955 | return I->getOperand(1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 956 | |
957 | // If all of the potentially set bits on one side are known to be set on | ||||
958 | // the other side, just use the 'other' side. | ||||
959 | if ((DemandedMask & (~RHSKnownZero) & LHSKnownOne) == | ||||
960 | (DemandedMask & (~RHSKnownZero))) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 961 | return I->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 962 | if ((DemandedMask & (~LHSKnownZero) & RHSKnownOne) == |
963 | (DemandedMask & (~LHSKnownZero))) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 964 | return I->getOperand(1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 965 | |
966 | // If the RHS is a constant, see if we can simplify it. | ||||
967 | if (ShrinkDemandedConstant(I, 1, DemandedMask)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 968 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 969 | |
970 | // Output known-0 bits are only known if clear in both the LHS & RHS. | ||||
971 | RHSKnownZero &= LHSKnownZero; | ||||
972 | // Output known-1 are known to be set if set in either the LHS | RHS. | ||||
973 | RHSKnownOne |= LHSKnownOne; | ||||
974 | break; | ||||
975 | case Instruction::Xor: { | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 976 | if (SimplifyDemandedBits(I->getOperandUse(1), DemandedMask, |
977 | RHSKnownZero, RHSKnownOne, Depth+1) || | ||||
978 | SimplifyDemandedBits(I->getOperandUse(0), DemandedMask, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 979 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 980 | return I; |
981 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
982 | assert(!(LHSKnownZero & LHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 983 | |
984 | // If all of the demanded bits are known zero on one side, return the other. | ||||
985 | // These bits cannot contribute to the result of the 'xor'. | ||||
986 | if ((DemandedMask & RHSKnownZero) == DemandedMask) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 987 | return I->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 988 | if ((DemandedMask & LHSKnownZero) == DemandedMask) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 989 | return I->getOperand(1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 990 | |
991 | // Output known-0 bits are known if clear or set in both the LHS & RHS. | ||||
992 | APInt KnownZeroOut = (RHSKnownZero & LHSKnownZero) | | ||||
993 | (RHSKnownOne & LHSKnownOne); | ||||
994 | // Output known-1 are known to be set if set in only one of the LHS, RHS. | ||||
995 | APInt KnownOneOut = (RHSKnownZero & LHSKnownOne) | | ||||
996 | (RHSKnownOne & LHSKnownZero); | ||||
997 | |||||
998 | // If all of the demanded bits are known to be zero on one side or the | ||||
999 | // other, turn this into an *inclusive* or. | ||||
1000 | // e.g. (A & C1)^(B & C2) -> (A & C1)|(B & C2) iff C1&C2 == 0 | ||||
1001 | if ((DemandedMask & ~RHSKnownZero & ~LHSKnownZero) == 0) { | ||||
1002 | Instruction *Or = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1003 | BinaryOperator::CreateOr(I->getOperand(0), I->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1004 | I->getName()); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1005 | return InsertNewInstBefore(Or, *I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1006 | } |
1007 | |||||
1008 | // If all of the demanded bits on one side are known, and all of the set | ||||
1009 | // bits on that side are also known to be set on the other side, turn this | ||||
1010 | // into an AND, as we know the bits will be cleared. | ||||
1011 | // e.g. (X | C1) ^ C2 --> (X | C1) & ~C2 iff (C1&C2) == C2 | ||||
1012 | if ((DemandedMask & (RHSKnownZero|RHSKnownOne)) == DemandedMask) { | ||||
1013 | // all known | ||||
1014 | if ((RHSKnownOne & LHSKnownOne) == RHSKnownOne) { | ||||
1015 | Constant *AndC = ConstantInt::get(~RHSKnownOne & DemandedMask); | ||||
1016 | Instruction *And = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1017 | BinaryOperator::CreateAnd(I->getOperand(0), AndC, "tmp"); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1018 | return InsertNewInstBefore(And, *I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1019 | } |
1020 | } | ||||
1021 | |||||
1022 | // If the RHS is a constant, see if we can simplify it. | ||||
1023 | // FIXME: for XOR, we prefer to force bits to 1 if they will make a -1. | ||||
1024 | if (ShrinkDemandedConstant(I, 1, DemandedMask)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1025 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1026 | |
1027 | RHSKnownZero = KnownZeroOut; | ||||
1028 | RHSKnownOne = KnownOneOut; | ||||
1029 | break; | ||||
1030 | } | ||||
1031 | case Instruction::Select: | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1032 | if (SimplifyDemandedBits(I->getOperandUse(2), DemandedMask, |
1033 | RHSKnownZero, RHSKnownOne, Depth+1) || | ||||
1034 | SimplifyDemandedBits(I->getOperandUse(1), DemandedMask, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1035 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1036 | return I; |
1037 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
1038 | assert(!(LHSKnownZero & LHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1039 | |
1040 | // If the operands are constants, see if we can simplify them. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1041 | if (ShrinkDemandedConstant(I, 1, DemandedMask) || |
1042 | ShrinkDemandedConstant(I, 2, DemandedMask)) | ||||
1043 | return I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1044 | |
1045 | // Only known if known in both the LHS and RHS. | ||||
1046 | RHSKnownOne &= LHSKnownOne; | ||||
1047 | RHSKnownZero &= LHSKnownZero; | ||||
1048 | break; | ||||
1049 | case Instruction::Trunc: { | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1050 | unsigned truncBf = I->getOperand(0)->getType()->getPrimitiveSizeInBits(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1051 | DemandedMask.zext(truncBf); |
1052 | RHSKnownZero.zext(truncBf); | ||||
1053 | RHSKnownOne.zext(truncBf); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1054 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMask, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1055 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1056 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1057 | DemandedMask.trunc(BitWidth); |
1058 | RHSKnownZero.trunc(BitWidth); | ||||
1059 | RHSKnownOne.trunc(BitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1060 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1061 | break; |
1062 | } | ||||
1063 | case Instruction::BitCast: | ||||
1064 | if (!I->getOperand(0)->getType()->isInteger()) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1065 | return false; // vector->int or fp->int? |
1066 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMask, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1067 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1068 | return I; |
1069 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1070 | break; |
1071 | case Instruction::ZExt: { | ||||
1072 | // Compute the bits in the result that are not present in the input. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1073 | unsigned SrcBitWidth =I->getOperand(0)->getType()->getPrimitiveSizeInBits(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1074 | |
1075 | DemandedMask.trunc(SrcBitWidth); | ||||
1076 | RHSKnownZero.trunc(SrcBitWidth); | ||||
1077 | RHSKnownOne.trunc(SrcBitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1078 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMask, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1079 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1080 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1081 | DemandedMask.zext(BitWidth); |
1082 | RHSKnownZero.zext(BitWidth); | ||||
1083 | RHSKnownOne.zext(BitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1084 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1085 | // The top bits are known to be zero. |
1086 | RHSKnownZero |= APInt::getHighBitsSet(BitWidth, BitWidth - SrcBitWidth); | ||||
1087 | break; | ||||
1088 | } | ||||
1089 | case Instruction::SExt: { | ||||
1090 | // Compute the bits in the result that are not present in the input. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1091 | unsigned SrcBitWidth =I->getOperand(0)->getType()->getPrimitiveSizeInBits(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1092 | |
1093 | APInt InputDemandedBits = DemandedMask & | ||||
1094 | APInt::getLowBitsSet(BitWidth, SrcBitWidth); | ||||
1095 | |||||
1096 | APInt NewBits(APInt::getHighBitsSet(BitWidth, BitWidth - SrcBitWidth)); | ||||
1097 | // If any of the sign extended bits are demanded, we know that the sign | ||||
1098 | // bit is demanded. | ||||
1099 | if ((NewBits & DemandedMask) != 0) | ||||
1100 | InputDemandedBits.set(SrcBitWidth-1); | ||||
1101 | |||||
1102 | InputDemandedBits.trunc(SrcBitWidth); | ||||
1103 | RHSKnownZero.trunc(SrcBitWidth); | ||||
1104 | RHSKnownOne.trunc(SrcBitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1105 | if (SimplifyDemandedBits(I->getOperandUse(0), InputDemandedBits, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1106 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1107 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1108 | InputDemandedBits.zext(BitWidth); |
1109 | RHSKnownZero.zext(BitWidth); | ||||
1110 | RHSKnownOne.zext(BitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1111 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1112 | |
1113 | // If the sign bit of the input is known set or clear, then we know the | ||||
1114 | // top bits of the result. | ||||
1115 | |||||
1116 | // If the input sign bit is known zero, or if the NewBits are not demanded | ||||
1117 | // convert this into a zero extension. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1118 | if (RHSKnownZero[SrcBitWidth-1] || (NewBits & ~DemandedMask) == NewBits) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1119 | // Convert to ZExt cast |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1120 | CastInst *NewCast = new ZExtInst(I->getOperand(0), VTy, I->getName()); |
1121 | return InsertNewInstBefore(NewCast, *I); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1122 | } else if (RHSKnownOne[SrcBitWidth-1]) { // Input sign bit known set |
1123 | RHSKnownOne |= NewBits; | ||||
1124 | } | ||||
1125 | break; | ||||
1126 | } | ||||
1127 | case Instruction::Add: { | ||||
1128 | // Figure out what the input bits are. If the top bits of the and result | ||||
1129 | // are not demanded, then the add doesn't demand them from its input | ||||
1130 | // either. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1131 | unsigned NLZ = DemandedMask.countLeadingZeros(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1132 | |
1133 | // If there is a constant on the RHS, there are a variety of xformations | ||||
1134 | // we can do. | ||||
1135 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1136 | // If null, this should be simplified elsewhere. Some of the xforms here | ||||
1137 | // won't work if the RHS is zero. | ||||
1138 | if (RHS->isZero()) | ||||
1139 | break; | ||||
1140 | |||||
1141 | // If the top bit of the output is demanded, demand everything from the | ||||
1142 | // input. Otherwise, we demand all the input bits except NLZ top bits. | ||||
1143 | APInt InDemandedBits(APInt::getLowBitsSet(BitWidth, BitWidth - NLZ)); | ||||
1144 | |||||
1145 | // Find information about known zero/one bits in the input. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1146 | if (SimplifyDemandedBits(I->getOperandUse(0), InDemandedBits, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1147 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1148 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1149 | |
1150 | // If the RHS of the add has bits set that can't affect the input, reduce | ||||
1151 | // the constant. | ||||
1152 | if (ShrinkDemandedConstant(I, 1, InDemandedBits)) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1153 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1154 | |
1155 | // Avoid excess work. | ||||
1156 | if (LHSKnownZero == 0 && LHSKnownOne == 0) | ||||
1157 | break; | ||||
1158 | |||||
1159 | // Turn it into OR if input bits are zero. | ||||
1160 | if ((LHSKnownZero & RHS->getValue()) == RHS->getValue()) { | ||||
1161 | Instruction *Or = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1162 | BinaryOperator::CreateOr(I->getOperand(0), I->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1163 | I->getName()); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1164 | return InsertNewInstBefore(Or, *I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1165 | } |
1166 | |||||
1167 | // We can say something about the output known-zero and known-one bits, | ||||
1168 | // depending on potential carries from the input constant and the | ||||
1169 | // unknowns. For example if the LHS is known to have at most the 0x0F0F0 | ||||
1170 | // bits set and the RHS constant is 0x01001, then we know we have a known | ||||
1171 | // one mask of 0x00001 and a known zero mask of 0xE0F0E. | ||||
1172 | |||||
1173 | // To compute this, we first compute the potential carry bits. These are | ||||
1174 | // the bits which may be modified. I'm not aware of a better way to do | ||||
1175 | // this scan. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1176 | const APInt &RHSVal = RHS->getValue(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1177 | APInt CarryBits((~LHSKnownZero + RHSVal) ^ (~LHSKnownZero ^ RHSVal)); |
1178 | |||||
1179 | // Now that we know which bits have carries, compute the known-1/0 sets. | ||||
1180 | |||||
1181 | // Bits are known one if they are known zero in one operand and one in the | ||||
1182 | // other, and there is no input carry. | ||||
1183 | RHSKnownOne = ((LHSKnownZero & RHSVal) | | ||||
1184 | (LHSKnownOne & ~RHSVal)) & ~CarryBits; | ||||
1185 | |||||
1186 | // Bits are known zero if they are known zero in both operands and there | ||||
1187 | // is no input carry. | ||||
1188 | RHSKnownZero = LHSKnownZero & ~RHSVal & ~CarryBits; | ||||
1189 | } else { | ||||
1190 | // If the high-bits of this ADD are not demanded, then it does not demand | ||||
1191 | // the high bits of its LHS or RHS. | ||||
1192 | if (DemandedMask[BitWidth-1] == 0) { | ||||
1193 | // Right fill the mask of bits for this ADD to demand the most | ||||
1194 | // significant bit and all those below it. | ||||
1195 | APInt DemandedFromOps(APInt::getLowBitsSet(BitWidth, BitWidth-NLZ)); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1196 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedFromOps, |
1197 | LHSKnownZero, LHSKnownOne, Depth+1) || | ||||
1198 | SimplifyDemandedBits(I->getOperandUse(1), DemandedFromOps, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1199 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1200 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1201 | } |
1202 | } | ||||
1203 | break; | ||||
1204 | } | ||||
1205 | case Instruction::Sub: | ||||
1206 | // If the high-bits of this SUB are not demanded, then it does not demand | ||||
1207 | // the high bits of its LHS or RHS. | ||||
1208 | if (DemandedMask[BitWidth-1] == 0) { | ||||
1209 | // Right fill the mask of bits for this SUB to demand the most | ||||
1210 | // significant bit and all those below it. | ||||
1211 | uint32_t NLZ = DemandedMask.countLeadingZeros(); | ||||
1212 | APInt DemandedFromOps(APInt::getLowBitsSet(BitWidth, BitWidth-NLZ)); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1213 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedFromOps, |
1214 | LHSKnownZero, LHSKnownOne, Depth+1) || | ||||
1215 | SimplifyDemandedBits(I->getOperandUse(1), DemandedFromOps, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1216 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1217 | return I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1218 | } |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1219 | // Otherwise just hand the sub off to ComputeMaskedBits to fill in |
1220 | // the known zeros and ones. | ||||
1221 | ComputeMaskedBits(V, DemandedMask, RHSKnownZero, RHSKnownOne, Depth); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1222 | break; |
1223 | case Instruction::Shl: | ||||
1224 | if (ConstantInt *SA = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1225 | uint64_t ShiftAmt = SA->getLimitedValue(BitWidth); | ||||
1226 | APInt DemandedMaskIn(DemandedMask.lshr(ShiftAmt)); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1227 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMaskIn, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1228 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1229 | return I; |
1230 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1231 | RHSKnownZero <<= ShiftAmt; |
1232 | RHSKnownOne <<= ShiftAmt; | ||||
1233 | // low bits known zero. | ||||
1234 | if (ShiftAmt) | ||||
1235 | RHSKnownZero |= APInt::getLowBitsSet(BitWidth, ShiftAmt); | ||||
1236 | } | ||||
1237 | break; | ||||
1238 | case Instruction::LShr: | ||||
1239 | // For a logical shift right | ||||
1240 | if (ConstantInt *SA = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1241 | uint64_t ShiftAmt = SA->getLimitedValue(BitWidth); | ||||
1242 | |||||
1243 | // Unsigned shift right. | ||||
1244 | APInt DemandedMaskIn(DemandedMask.shl(ShiftAmt)); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1245 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMaskIn, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1246 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1247 | return I; |
1248 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1249 | RHSKnownZero = APIntOps::lshr(RHSKnownZero, ShiftAmt); |
1250 | RHSKnownOne = APIntOps::lshr(RHSKnownOne, ShiftAmt); | ||||
1251 | if (ShiftAmt) { | ||||
1252 | // Compute the new bits that are at the top now. | ||||
1253 | APInt HighBits(APInt::getHighBitsSet(BitWidth, ShiftAmt)); | ||||
1254 | RHSKnownZero |= HighBits; // high bits known zero. | ||||
1255 | } | ||||
1256 | } | ||||
1257 | break; | ||||
1258 | case Instruction::AShr: | ||||
1259 | // If this is an arithmetic shift right and only the low-bit is set, we can | ||||
1260 | // always convert this into a logical shr, even if the shift amount is | ||||
1261 | // variable. The low bit of the shift cannot be an input sign bit unless | ||||
1262 | // the shift amount is >= the size of the datatype, which is undefined. | ||||
1263 | if (DemandedMask == 1) { | ||||
1264 | // Perform the logical shift right. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1265 | Instruction *NewVal = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1266 | I->getOperand(0), I->getOperand(1), I->getName()); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1267 | return InsertNewInstBefore(NewVal, *I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1268 | } |
1269 | |||||
1270 | // If the sign bit is the only bit demanded by this ashr, then there is no | ||||
1271 | // need to do it, the shift doesn't change the high bit. | ||||
1272 | if (DemandedMask.isSignBit()) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1273 | return I->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1274 | |
1275 | if (ConstantInt *SA = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
1276 | uint32_t ShiftAmt = SA->getLimitedValue(BitWidth); | ||||
1277 | |||||
1278 | // Signed shift right. | ||||
1279 | APInt DemandedMaskIn(DemandedMask.shl(ShiftAmt)); | ||||
1280 | // If any of the "high bits" are demanded, we should set the sign bit as | ||||
1281 | // demanded. | ||||
1282 | if (DemandedMask.countLeadingZeros() <= ShiftAmt) | ||||
1283 | DemandedMaskIn.set(BitWidth-1); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1284 | if (SimplifyDemandedBits(I->getOperandUse(0), DemandedMaskIn, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1285 | RHSKnownZero, RHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1286 | return I; |
1287 | assert(!(RHSKnownZero & RHSKnownOne) && "Bits known to be one AND zero?"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1288 | // Compute the new bits that are at the top now. |
1289 | APInt HighBits(APInt::getHighBitsSet(BitWidth, ShiftAmt)); | ||||
1290 | RHSKnownZero = APIntOps::lshr(RHSKnownZero, ShiftAmt); | ||||
1291 | RHSKnownOne = APIntOps::lshr(RHSKnownOne, ShiftAmt); | ||||
1292 | |||||
1293 | // Handle the sign bits. | ||||
1294 | APInt SignBit(APInt::getSignBit(BitWidth)); | ||||
1295 | // Adjust to where it is now in the mask. | ||||
1296 | SignBit = APIntOps::lshr(SignBit, ShiftAmt); | ||||
1297 | |||||
1298 | // If the input sign bit is known to be zero, or if none of the top bits | ||||
1299 | // are demanded, turn this into an unsigned shift right. | ||||
Zhou Sheng | 533604e | 2008-06-06 08:32:05 +0000 | [diff] [blame] | 1300 | if (BitWidth <= ShiftAmt || RHSKnownZero[BitWidth-ShiftAmt-1] || |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1301 | (HighBits & ~DemandedMask) == HighBits) { |
1302 | // Perform the logical shift right. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1303 | Instruction *NewVal = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1304 | I->getOperand(0), SA, I->getName()); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1305 | return InsertNewInstBefore(NewVal, *I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1306 | } else if ((RHSKnownOne & SignBit) != 0) { // New bits are known one. |
1307 | RHSKnownOne |= HighBits; | ||||
1308 | } | ||||
1309 | } | ||||
1310 | break; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1311 | case Instruction::SRem: |
1312 | if (ConstantInt *Rem = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
Nick Lewycky | cfaaece | 2008-11-02 02:41:50 +0000 | [diff] [blame] | 1313 | APInt RA = Rem->getValue().abs(); |
1314 | if (RA.isPowerOf2()) { | ||||
Nick Lewycky | 245de42 | 2008-07-12 05:04:38 +0000 | [diff] [blame] | 1315 | if (DemandedMask.ule(RA)) // srem won't affect demanded bits |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1316 | return I->getOperand(0); |
Nick Lewycky | 245de42 | 2008-07-12 05:04:38 +0000 | [diff] [blame] | 1317 | |
Nick Lewycky | cfaaece | 2008-11-02 02:41:50 +0000 | [diff] [blame] | 1318 | APInt LowBits = RA - 1; |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1319 | APInt Mask2 = LowBits | APInt::getSignBit(BitWidth); |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1320 | if (SimplifyDemandedBits(I->getOperandUse(0), Mask2, |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1321 | LHSKnownZero, LHSKnownOne, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1322 | return I; |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1323 | |
1324 | if (LHSKnownZero[BitWidth-1] || ((LHSKnownZero & LowBits) == LowBits)) | ||||
1325 | LHSKnownZero |= ~LowBits; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1326 | |
1327 | KnownZero |= LHSKnownZero & DemandedMask; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1328 | |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1329 | assert(!(KnownZero & KnownOne) && "Bits known to be one AND zero?"); |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1330 | } |
1331 | } | ||||
1332 | break; | ||||
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1333 | case Instruction::URem: { |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1334 | APInt KnownZero2(BitWidth, 0), KnownOne2(BitWidth, 0); |
1335 | APInt AllOnes = APInt::getAllOnesValue(BitWidth); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1336 | if (SimplifyDemandedBits(I->getOperandUse(0), AllOnes, |
1337 | KnownZero2, KnownOne2, Depth+1) || | ||||
1338 | SimplifyDemandedBits(I->getOperandUse(1), AllOnes, | ||||
Dan Gohman | 23ea06d | 2008-05-01 19:13:24 +0000 | [diff] [blame] | 1339 | KnownZero2, KnownOne2, Depth+1)) |
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1340 | return I; |
Dan Gohman | 23ea06d | 2008-05-01 19:13:24 +0000 | [diff] [blame] | 1341 | |
Chris Lattner | ee5417c | 2009-01-21 18:09:24 +0000 | [diff] [blame] | 1342 | unsigned Leaders = KnownZero2.countLeadingOnes(); |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1343 | Leaders = std::max(Leaders, |
1344 | KnownZero2.countLeadingOnes()); | ||||
1345 | KnownZero = APInt::getHighBitsSet(BitWidth, Leaders) & DemandedMask; | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 1346 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1347 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 1348 | case Instruction::Call: |
1349 | if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(I)) { | ||||
1350 | switch (II->getIntrinsicID()) { | ||||
1351 | default: break; | ||||
1352 | case Intrinsic::bswap: { | ||||
1353 | // If the only bits demanded come from one byte of the bswap result, | ||||
1354 | // just shift the input byte into position to eliminate the bswap. | ||||
1355 | unsigned NLZ = DemandedMask.countLeadingZeros(); | ||||
1356 | unsigned NTZ = DemandedMask.countTrailingZeros(); | ||||
1357 | |||||
1358 | // Round NTZ down to the next byte. If we have 11 trailing zeros, then | ||||
1359 | // we need all the bits down to bit 8. Likewise, round NLZ. If we | ||||
1360 | // have 14 leading zeros, round to 8. | ||||
1361 | NLZ &= ~7; | ||||
1362 | NTZ &= ~7; | ||||
1363 | // If we need exactly one byte, we can do this transformation. | ||||
1364 | if (BitWidth-NLZ-NTZ == 8) { | ||||
1365 | unsigned ResultBit = NTZ; | ||||
1366 | unsigned InputBit = BitWidth-NTZ-8; | ||||
1367 | |||||
1368 | // Replace this with either a left or right shift to get the byte into | ||||
1369 | // the right place. | ||||
1370 | Instruction *NewVal; | ||||
1371 | if (InputBit > ResultBit) | ||||
1372 | NewVal = BinaryOperator::CreateLShr(I->getOperand(1), | ||||
1373 | ConstantInt::get(I->getType(), InputBit-ResultBit)); | ||||
1374 | else | ||||
1375 | NewVal = BinaryOperator::CreateShl(I->getOperand(1), | ||||
1376 | ConstantInt::get(I->getType(), ResultBit-InputBit)); | ||||
1377 | NewVal->takeName(I); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1378 | return InsertNewInstBefore(NewVal, *I); |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 1379 | } |
1380 | |||||
1381 | // TODO: Could compute known zero/one bits based on the input. | ||||
1382 | break; | ||||
1383 | } | ||||
1384 | } | ||||
1385 | } | ||||
Chris Lattner | 4946e22 | 2008-06-18 18:11:55 +0000 | [diff] [blame] | 1386 | ComputeMaskedBits(V, DemandedMask, RHSKnownZero, RHSKnownOne, Depth); |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 1387 | break; |
Dan Gohman | bec1605 | 2008-04-28 17:02:21 +0000 | [diff] [blame] | 1388 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1389 | |
1390 | // If the client is only demanding bits that we know, return the known | ||||
1391 | // constant. | ||||
1392 | if ((DemandedMask & (RHSKnownZero|RHSKnownOne)) == DemandedMask) | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 1393 | return ConstantInt::get(RHSKnownOne); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1394 | return false; |
1395 | } | ||||
1396 | |||||
1397 | |||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1398 | /// SimplifyDemandedVectorElts - The specified value produces a vector with |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1399 | /// 64 or fewer elements. DemandedElts contains the set of elements that are |
1400 | /// actually used by the caller. This method analyzes which elements of the | ||||
1401 | /// operand are undef and returns that information in UndefElts. | ||||
1402 | /// | ||||
1403 | /// If the information about demanded elements can be used to simplify the | ||||
1404 | /// operation, the operation is simplified, then the resultant value is | ||||
1405 | /// returned. This returns null if no change was made. | ||||
1406 | Value *InstCombiner::SimplifyDemandedVectorElts(Value *V, uint64_t DemandedElts, | ||||
1407 | uint64_t &UndefElts, | ||||
1408 | unsigned Depth) { | ||||
1409 | unsigned VWidth = cast<VectorType>(V->getType())->getNumElements(); | ||||
1410 | assert(VWidth <= 64 && "Vector too wide to analyze!"); | ||||
1411 | uint64_t EltMask = ~0ULL >> (64-VWidth); | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1412 | assert((DemandedElts & ~EltMask) == 0 && "Invalid DemandedElts!"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1413 | |
1414 | if (isa<UndefValue>(V)) { | ||||
1415 | // If the entire vector is undefined, just return this info. | ||||
1416 | UndefElts = EltMask; | ||||
1417 | return 0; | ||||
1418 | } else if (DemandedElts == 0) { // If nothing is demanded, provide undef. | ||||
1419 | UndefElts = EltMask; | ||||
1420 | return UndefValue::get(V->getType()); | ||||
1421 | } | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1422 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1423 | UndefElts = 0; |
1424 | if (ConstantVector *CP = dyn_cast<ConstantVector>(V)) { | ||||
1425 | const Type *EltTy = cast<VectorType>(V->getType())->getElementType(); | ||||
1426 | Constant *Undef = UndefValue::get(EltTy); | ||||
1427 | |||||
1428 | std::vector<Constant*> Elts; | ||||
1429 | for (unsigned i = 0; i != VWidth; ++i) | ||||
1430 | if (!(DemandedElts & (1ULL << i))) { // If not demanded, set to undef. | ||||
1431 | Elts.push_back(Undef); | ||||
1432 | UndefElts |= (1ULL << i); | ||||
1433 | } else if (isa<UndefValue>(CP->getOperand(i))) { // Already undef. | ||||
1434 | Elts.push_back(Undef); | ||||
1435 | UndefElts |= (1ULL << i); | ||||
1436 | } else { // Otherwise, defined. | ||||
1437 | Elts.push_back(CP->getOperand(i)); | ||||
1438 | } | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1439 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1440 | // If we changed the constant, return it. |
1441 | Constant *NewCP = ConstantVector::get(Elts); | ||||
1442 | return NewCP != CP ? NewCP : 0; | ||||
1443 | } else if (isa<ConstantAggregateZero>(V)) { | ||||
1444 | // Simplify the CAZ to a ConstantVector where the non-demanded elements are | ||||
1445 | // set to undef. | ||||
Mon P Wang | 927daf5 | 2008-11-06 22:52:21 +0000 | [diff] [blame] | 1446 | |
1447 | // Check if this is identity. If so, return 0 since we are not simplifying | ||||
1448 | // anything. | ||||
1449 | if (DemandedElts == ((1ULL << VWidth) -1)) | ||||
1450 | return 0; | ||||
1451 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1452 | const Type *EltTy = cast<VectorType>(V->getType())->getElementType(); |
1453 | Constant *Zero = Constant::getNullValue(EltTy); | ||||
1454 | Constant *Undef = UndefValue::get(EltTy); | ||||
1455 | std::vector<Constant*> Elts; | ||||
1456 | for (unsigned i = 0; i != VWidth; ++i) | ||||
1457 | Elts.push_back((DemandedElts & (1ULL << i)) ? Zero : Undef); | ||||
1458 | UndefElts = DemandedElts ^ EltMask; | ||||
1459 | return ConstantVector::get(Elts); | ||||
1460 | } | ||||
1461 | |||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1462 | // Limit search depth. |
1463 | if (Depth == 10) | ||||
1464 | return false; | ||||
1465 | |||||
1466 | // If multiple users are using the root value, procede with | ||||
1467 | // simplification conservatively assuming that all elements | ||||
1468 | // are needed. | ||||
1469 | if (!V->hasOneUse()) { | ||||
1470 | // Quit if we find multiple users of a non-root value though. | ||||
1471 | // They'll be handled when it's their turn to be visited by | ||||
1472 | // the main instcombine process. | ||||
1473 | if (Depth != 0) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1474 | // TODO: Just compute the UndefElts information recursively. |
1475 | return false; | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1476 | |
1477 | // Conservatively assume that all elements are needed. | ||||
1478 | DemandedElts = EltMask; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1479 | } |
1480 | |||||
1481 | Instruction *I = dyn_cast<Instruction>(V); | ||||
1482 | if (!I) return false; // Only analyze instructions. | ||||
1483 | |||||
1484 | bool MadeChange = false; | ||||
1485 | uint64_t UndefElts2; | ||||
1486 | Value *TmpV; | ||||
1487 | switch (I->getOpcode()) { | ||||
1488 | default: break; | ||||
1489 | |||||
1490 | case Instruction::InsertElement: { | ||||
1491 | // If this is a variable index, we don't know which element it overwrites. | ||||
1492 | // demand exactly the same input as we produce. | ||||
1493 | ConstantInt *Idx = dyn_cast<ConstantInt>(I->getOperand(2)); | ||||
1494 | if (Idx == 0) { | ||||
1495 | // Note that we can't propagate undef elt info, because we don't know | ||||
1496 | // which elt is getting updated. | ||||
1497 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts, | ||||
1498 | UndefElts2, Depth+1); | ||||
1499 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1500 | break; | ||||
1501 | } | ||||
1502 | |||||
1503 | // If this is inserting an element that isn't demanded, remove this | ||||
1504 | // insertelement. | ||||
1505 | unsigned IdxNo = Idx->getZExtValue(); | ||||
1506 | if (IdxNo >= VWidth || (DemandedElts & (1ULL << IdxNo)) == 0) | ||||
1507 | return AddSoonDeadInstToWorklist(*I, 0); | ||||
1508 | |||||
1509 | // Otherwise, the element inserted overwrites whatever was there, so the | ||||
1510 | // input demanded set is simpler than the output set. | ||||
1511 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), | ||||
1512 | DemandedElts & ~(1ULL << IdxNo), | ||||
1513 | UndefElts, Depth+1); | ||||
1514 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1515 | |||||
1516 | // The inserted element is defined. | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1517 | UndefElts &= ~(1ULL << IdxNo); |
1518 | break; | ||||
1519 | } | ||||
1520 | case Instruction::ShuffleVector: { | ||||
1521 | ShuffleVectorInst *Shuffle = cast<ShuffleVectorInst>(I); | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1522 | uint64_t LHSVWidth = |
1523 | cast<VectorType>(Shuffle->getOperand(0)->getType())->getNumElements(); | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1524 | uint64_t LeftDemanded = 0, RightDemanded = 0; |
1525 | for (unsigned i = 0; i < VWidth; i++) { | ||||
1526 | if (DemandedElts & (1ULL << i)) { | ||||
1527 | unsigned MaskVal = Shuffle->getMaskValue(i); | ||||
1528 | if (MaskVal != -1u) { | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1529 | assert(MaskVal < LHSVWidth * 2 && |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1530 | "shufflevector mask index out of range!"); |
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1531 | if (MaskVal < LHSVWidth) |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1532 | LeftDemanded |= 1ULL << MaskVal; |
1533 | else | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1534 | RightDemanded |= 1ULL << (MaskVal - LHSVWidth); |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1535 | } |
1536 | } | ||||
1537 | } | ||||
1538 | |||||
1539 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), LeftDemanded, | ||||
1540 | UndefElts2, Depth+1); | ||||
1541 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1542 | |||||
1543 | uint64_t UndefElts3; | ||||
1544 | TmpV = SimplifyDemandedVectorElts(I->getOperand(1), RightDemanded, | ||||
1545 | UndefElts3, Depth+1); | ||||
1546 | if (TmpV) { I->setOperand(1, TmpV); MadeChange = true; } | ||||
1547 | |||||
1548 | bool NewUndefElts = false; | ||||
1549 | for (unsigned i = 0; i < VWidth; i++) { | ||||
1550 | unsigned MaskVal = Shuffle->getMaskValue(i); | ||||
Dan Gohman | 24f6ee2 | 2008-09-10 01:09:32 +0000 | [diff] [blame] | 1551 | if (MaskVal == -1u) { |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1552 | uint64_t NewBit = 1ULL << i; |
1553 | UndefElts |= NewBit; | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1554 | } else if (MaskVal < LHSVWidth) { |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1555 | uint64_t NewBit = ((UndefElts2 >> MaskVal) & 1) << i; |
1556 | NewUndefElts |= NewBit; | ||||
1557 | UndefElts |= NewBit; | ||||
1558 | } else { | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 1559 | uint64_t NewBit = ((UndefElts3 >> (MaskVal - LHSVWidth)) & 1) << i; |
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 1560 | NewUndefElts |= NewBit; |
1561 | UndefElts |= NewBit; | ||||
1562 | } | ||||
1563 | } | ||||
1564 | |||||
1565 | if (NewUndefElts) { | ||||
1566 | // Add additional discovered undefs. | ||||
1567 | std::vector<Constant*> Elts; | ||||
1568 | for (unsigned i = 0; i < VWidth; ++i) { | ||||
1569 | if (UndefElts & (1ULL << i)) | ||||
1570 | Elts.push_back(UndefValue::get(Type::Int32Ty)); | ||||
1571 | else | ||||
1572 | Elts.push_back(ConstantInt::get(Type::Int32Ty, | ||||
1573 | Shuffle->getMaskValue(i))); | ||||
1574 | } | ||||
1575 | I->setOperand(2, ConstantVector::get(Elts)); | ||||
1576 | MadeChange = true; | ||||
1577 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1578 | break; |
1579 | } | ||||
1580 | case Instruction::BitCast: { | ||||
1581 | // Vector->vector casts only. | ||||
1582 | const VectorType *VTy = dyn_cast<VectorType>(I->getOperand(0)->getType()); | ||||
1583 | if (!VTy) break; | ||||
1584 | unsigned InVWidth = VTy->getNumElements(); | ||||
1585 | uint64_t InputDemandedElts = 0; | ||||
1586 | unsigned Ratio; | ||||
1587 | |||||
1588 | if (VWidth == InVWidth) { | ||||
1589 | // If we are converting from <4 x i32> -> <4 x f32>, we demand the same | ||||
1590 | // elements as are demanded of us. | ||||
1591 | Ratio = 1; | ||||
1592 | InputDemandedElts = DemandedElts; | ||||
1593 | } else if (VWidth > InVWidth) { | ||||
1594 | // Untested so far. | ||||
1595 | break; | ||||
1596 | |||||
1597 | // If there are more elements in the result than there are in the source, | ||||
1598 | // then an input element is live if any of the corresponding output | ||||
1599 | // elements are live. | ||||
1600 | Ratio = VWidth/InVWidth; | ||||
1601 | for (unsigned OutIdx = 0; OutIdx != VWidth; ++OutIdx) { | ||||
1602 | if (DemandedElts & (1ULL << OutIdx)) | ||||
1603 | InputDemandedElts |= 1ULL << (OutIdx/Ratio); | ||||
1604 | } | ||||
1605 | } else { | ||||
1606 | // Untested so far. | ||||
1607 | break; | ||||
1608 | |||||
1609 | // If there are more elements in the source than there are in the result, | ||||
1610 | // then an input element is live if the corresponding output element is | ||||
1611 | // live. | ||||
1612 | Ratio = InVWidth/VWidth; | ||||
1613 | for (unsigned InIdx = 0; InIdx != InVWidth; ++InIdx) | ||||
1614 | if (DemandedElts & (1ULL << InIdx/Ratio)) | ||||
1615 | InputDemandedElts |= 1ULL << InIdx; | ||||
1616 | } | ||||
1617 | |||||
1618 | // div/rem demand all inputs, because they don't want divide by zero. | ||||
1619 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), InputDemandedElts, | ||||
1620 | UndefElts2, Depth+1); | ||||
1621 | if (TmpV) { | ||||
1622 | I->setOperand(0, TmpV); | ||||
1623 | MadeChange = true; | ||||
1624 | } | ||||
1625 | |||||
1626 | UndefElts = UndefElts2; | ||||
1627 | if (VWidth > InVWidth) { | ||||
1628 | assert(0 && "Unimp"); | ||||
1629 | // If there are more elements in the result than there are in the source, | ||||
1630 | // then an output element is undef if the corresponding input element is | ||||
1631 | // undef. | ||||
1632 | for (unsigned OutIdx = 0; OutIdx != VWidth; ++OutIdx) | ||||
1633 | if (UndefElts2 & (1ULL << (OutIdx/Ratio))) | ||||
1634 | UndefElts |= 1ULL << OutIdx; | ||||
1635 | } else if (VWidth < InVWidth) { | ||||
1636 | assert(0 && "Unimp"); | ||||
1637 | // If there are more elements in the source than there are in the result, | ||||
1638 | // then a result element is undef if all of the corresponding input | ||||
1639 | // elements are undef. | ||||
1640 | UndefElts = ~0ULL >> (64-VWidth); // Start out all undef. | ||||
1641 | for (unsigned InIdx = 0; InIdx != InVWidth; ++InIdx) | ||||
1642 | if ((UndefElts2 & (1ULL << InIdx)) == 0) // Not undef? | ||||
1643 | UndefElts &= ~(1ULL << (InIdx/Ratio)); // Clear undef bit. | ||||
1644 | } | ||||
1645 | break; | ||||
1646 | } | ||||
1647 | case Instruction::And: | ||||
1648 | case Instruction::Or: | ||||
1649 | case Instruction::Xor: | ||||
1650 | case Instruction::Add: | ||||
1651 | case Instruction::Sub: | ||||
1652 | case Instruction::Mul: | ||||
1653 | // div/rem demand all inputs, because they don't want divide by zero. | ||||
1654 | TmpV = SimplifyDemandedVectorElts(I->getOperand(0), DemandedElts, | ||||
1655 | UndefElts, Depth+1); | ||||
1656 | if (TmpV) { I->setOperand(0, TmpV); MadeChange = true; } | ||||
1657 | TmpV = SimplifyDemandedVectorElts(I->getOperand(1), DemandedElts, | ||||
1658 | UndefElts2, Depth+1); | ||||
1659 | if (TmpV) { I->setOperand(1, TmpV); MadeChange = true; } | ||||
1660 | |||||
1661 | // Output elements are undefined if both are undefined. Consider things | ||||
1662 | // like undef&0. The result is known zero, not undef. | ||||
1663 | UndefElts &= UndefElts2; | ||||
1664 | break; | ||||
1665 | |||||
1666 | case Instruction::Call: { | ||||
1667 | IntrinsicInst *II = dyn_cast<IntrinsicInst>(I); | ||||
1668 | if (!II) break; | ||||
1669 | switch (II->getIntrinsicID()) { | ||||
1670 | default: break; | ||||
1671 | |||||
1672 | // Binary vector operations that work column-wise. A dest element is a | ||||
1673 | // function of the corresponding input elements from the two inputs. | ||||
1674 | case Intrinsic::x86_sse_sub_ss: | ||||
1675 | case Intrinsic::x86_sse_mul_ss: | ||||
1676 | case Intrinsic::x86_sse_min_ss: | ||||
1677 | case Intrinsic::x86_sse_max_ss: | ||||
1678 | case Intrinsic::x86_sse2_sub_sd: | ||||
1679 | case Intrinsic::x86_sse2_mul_sd: | ||||
1680 | case Intrinsic::x86_sse2_min_sd: | ||||
1681 | case Intrinsic::x86_sse2_max_sd: | ||||
1682 | TmpV = SimplifyDemandedVectorElts(II->getOperand(1), DemandedElts, | ||||
1683 | UndefElts, Depth+1); | ||||
1684 | if (TmpV) { II->setOperand(1, TmpV); MadeChange = true; } | ||||
1685 | TmpV = SimplifyDemandedVectorElts(II->getOperand(2), DemandedElts, | ||||
1686 | UndefElts2, Depth+1); | ||||
1687 | if (TmpV) { II->setOperand(2, TmpV); MadeChange = true; } | ||||
1688 | |||||
1689 | // If only the low elt is demanded and this is a scalarizable intrinsic, | ||||
1690 | // scalarize it now. | ||||
1691 | if (DemandedElts == 1) { | ||||
1692 | switch (II->getIntrinsicID()) { | ||||
1693 | default: break; | ||||
1694 | case Intrinsic::x86_sse_sub_ss: | ||||
1695 | case Intrinsic::x86_sse_mul_ss: | ||||
1696 | case Intrinsic::x86_sse2_sub_sd: | ||||
1697 | case Intrinsic::x86_sse2_mul_sd: | ||||
1698 | // TODO: Lower MIN/MAX/ABS/etc | ||||
1699 | Value *LHS = II->getOperand(1); | ||||
1700 | Value *RHS = II->getOperand(2); | ||||
1701 | // Extract the element as scalars. | ||||
1702 | LHS = InsertNewInstBefore(new ExtractElementInst(LHS, 0U,"tmp"), *II); | ||||
1703 | RHS = InsertNewInstBefore(new ExtractElementInst(RHS, 0U,"tmp"), *II); | ||||
1704 | |||||
1705 | switch (II->getIntrinsicID()) { | ||||
1706 | default: assert(0 && "Case stmts out of sync!"); | ||||
1707 | case Intrinsic::x86_sse_sub_ss: | ||||
1708 | case Intrinsic::x86_sse2_sub_sd: | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1709 | TmpV = InsertNewInstBefore(BinaryOperator::CreateSub(LHS, RHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1710 | II->getName()), *II); |
1711 | break; | ||||
1712 | case Intrinsic::x86_sse_mul_ss: | ||||
1713 | case Intrinsic::x86_sse2_mul_sd: | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1714 | TmpV = InsertNewInstBefore(BinaryOperator::CreateMul(LHS, RHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1715 | II->getName()), *II); |
1716 | break; | ||||
1717 | } | ||||
1718 | |||||
1719 | Instruction *New = | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 1720 | InsertElementInst::Create(UndefValue::get(II->getType()), TmpV, 0U, |
1721 | II->getName()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1722 | InsertNewInstBefore(New, *II); |
1723 | AddSoonDeadInstToWorklist(*II, 0); | ||||
1724 | return New; | ||||
1725 | } | ||||
1726 | } | ||||
1727 | |||||
1728 | // Output elements are undefined if both are undefined. Consider things | ||||
1729 | // like undef&0. The result is known zero, not undef. | ||||
1730 | UndefElts &= UndefElts2; | ||||
1731 | break; | ||||
1732 | } | ||||
1733 | break; | ||||
1734 | } | ||||
1735 | } | ||||
1736 | return MadeChange ? I : 0; | ||||
1737 | } | ||||
1738 | |||||
Dan Gohman | 5d56fd4 | 2008-05-19 22:14:15 +0000 | [diff] [blame] | 1739 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1740 | /// AssociativeOpt - Perform an optimization on an associative operator. This |
1741 | /// function is designed to check a chain of associative operators for a | ||||
1742 | /// potential to apply a certain optimization. Since the optimization may be | ||||
1743 | /// applicable if the expression was reassociated, this checks the chain, then | ||||
1744 | /// reassociates the expression as necessary to expose the optimization | ||||
1745 | /// opportunity. This makes use of a special Functor, which must define | ||||
1746 | /// 'shouldApply' and 'apply' methods. | ||||
1747 | /// | ||||
1748 | template<typename Functor> | ||||
Dan Gohman | d8bcf5b | 2008-05-20 01:14:05 +0000 | [diff] [blame] | 1749 | static Instruction *AssociativeOpt(BinaryOperator &Root, const Functor &F) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1750 | unsigned Opcode = Root.getOpcode(); |
1751 | Value *LHS = Root.getOperand(0); | ||||
1752 | |||||
1753 | // Quick check, see if the immediate LHS matches... | ||||
1754 | if (F.shouldApply(LHS)) | ||||
1755 | return F.apply(Root); | ||||
1756 | |||||
1757 | // Otherwise, if the LHS is not of the same opcode as the root, return. | ||||
1758 | Instruction *LHSI = dyn_cast<Instruction>(LHS); | ||||
1759 | while (LHSI && LHSI->getOpcode() == Opcode && LHSI->hasOneUse()) { | ||||
1760 | // Should we apply this transform to the RHS? | ||||
1761 | bool ShouldApply = F.shouldApply(LHSI->getOperand(1)); | ||||
1762 | |||||
1763 | // If not to the RHS, check to see if we should apply to the LHS... | ||||
1764 | if (!ShouldApply && F.shouldApply(LHSI->getOperand(0))) { | ||||
1765 | cast<BinaryOperator>(LHSI)->swapOperands(); // Make the LHS the RHS | ||||
1766 | ShouldApply = true; | ||||
1767 | } | ||||
1768 | |||||
1769 | // If the functor wants to apply the optimization to the RHS of LHSI, | ||||
1770 | // reassociate the expression from ((? op A) op B) to (? op (A op B)) | ||||
1771 | if (ShouldApply) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1772 | // Now all of the instructions are in the current basic block, go ahead |
1773 | // and perform the reassociation. | ||||
1774 | Instruction *TmpLHSI = cast<Instruction>(Root.getOperand(0)); | ||||
1775 | |||||
1776 | // First move the selected RHS to the LHS of the root... | ||||
1777 | Root.setOperand(0, LHSI->getOperand(1)); | ||||
1778 | |||||
1779 | // Make what used to be the LHS of the root be the user of the root... | ||||
1780 | Value *ExtraOperand = TmpLHSI->getOperand(1); | ||||
1781 | if (&Root == TmpLHSI) { | ||||
1782 | Root.replaceAllUsesWith(Constant::getNullValue(TmpLHSI->getType())); | ||||
1783 | return 0; | ||||
1784 | } | ||||
1785 | Root.replaceAllUsesWith(TmpLHSI); // Users now use TmpLHSI | ||||
1786 | TmpLHSI->setOperand(1, &Root); // TmpLHSI now uses the root | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1787 | BasicBlock::iterator ARI = &Root; ++ARI; |
Dan Gohman | 0bb9a3d | 2008-06-19 17:47:47 +0000 | [diff] [blame] | 1788 | TmpLHSI->moveBefore(ARI); // Move TmpLHSI to after Root |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1789 | ARI = Root; |
1790 | |||||
1791 | // Now propagate the ExtraOperand down the chain of instructions until we | ||||
1792 | // get to LHSI. | ||||
1793 | while (TmpLHSI != LHSI) { | ||||
1794 | Instruction *NextLHSI = cast<Instruction>(TmpLHSI->getOperand(0)); | ||||
1795 | // Move the instruction to immediately before the chain we are | ||||
1796 | // constructing to avoid breaking dominance properties. | ||||
Dan Gohman | 0bb9a3d | 2008-06-19 17:47:47 +0000 | [diff] [blame] | 1797 | NextLHSI->moveBefore(ARI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1798 | ARI = NextLHSI; |
1799 | |||||
1800 | Value *NextOp = NextLHSI->getOperand(1); | ||||
1801 | NextLHSI->setOperand(1, ExtraOperand); | ||||
1802 | TmpLHSI = NextLHSI; | ||||
1803 | ExtraOperand = NextOp; | ||||
1804 | } | ||||
1805 | |||||
1806 | // Now that the instructions are reassociated, have the functor perform | ||||
1807 | // the transformation... | ||||
1808 | return F.apply(Root); | ||||
1809 | } | ||||
1810 | |||||
1811 | LHSI = dyn_cast<Instruction>(LHSI->getOperand(0)); | ||||
1812 | } | ||||
1813 | return 0; | ||||
1814 | } | ||||
1815 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 1816 | namespace { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1817 | |
Nick Lewycky | 27f6c13 | 2008-05-23 04:34:58 +0000 | [diff] [blame] | 1818 | // AddRHS - Implements: X + X --> X << 1 |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1819 | struct AddRHS { |
1820 | Value *RHS; | ||||
1821 | AddRHS(Value *rhs) : RHS(rhs) {} | ||||
1822 | bool shouldApply(Value *LHS) const { return LHS == RHS; } | ||||
1823 | Instruction *apply(BinaryOperator &Add) const { | ||||
Nick Lewycky | 27f6c13 | 2008-05-23 04:34:58 +0000 | [diff] [blame] | 1824 | return BinaryOperator::CreateShl(Add.getOperand(0), |
1825 | ConstantInt::get(Add.getType(), 1)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1826 | } |
1827 | }; | ||||
1828 | |||||
1829 | // AddMaskingAnd - Implements (A & C1)+(B & C2) --> (A & C1)|(B & C2) | ||||
1830 | // iff C1&C2 == 0 | ||||
1831 | struct AddMaskingAnd { | ||||
1832 | Constant *C2; | ||||
1833 | AddMaskingAnd(Constant *c) : C2(c) {} | ||||
1834 | bool shouldApply(Value *LHS) const { | ||||
1835 | ConstantInt *C1; | ||||
1836 | return match(LHS, m_And(m_Value(), m_ConstantInt(C1))) && | ||||
1837 | ConstantExpr::getAnd(C1, C2)->isNullValue(); | ||||
1838 | } | ||||
1839 | Instruction *apply(BinaryOperator &Add) const { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1840 | return BinaryOperator::CreateOr(Add.getOperand(0), Add.getOperand(1)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1841 | } |
1842 | }; | ||||
1843 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 1844 | } |
1845 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1846 | static Value *FoldOperationIntoSelectOperand(Instruction &I, Value *SO, |
1847 | InstCombiner *IC) { | ||||
1848 | if (CastInst *CI = dyn_cast<CastInst>(&I)) { | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 1849 | return IC->InsertCastBefore(CI->getOpcode(), SO, I.getType(), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1850 | } |
1851 | |||||
1852 | // Figure out if the constant is the left or the right argument. | ||||
1853 | bool ConstIsRHS = isa<Constant>(I.getOperand(1)); | ||||
1854 | Constant *ConstOperand = cast<Constant>(I.getOperand(ConstIsRHS)); | ||||
1855 | |||||
1856 | if (Constant *SOC = dyn_cast<Constant>(SO)) { | ||||
1857 | if (ConstIsRHS) | ||||
1858 | return ConstantExpr::get(I.getOpcode(), SOC, ConstOperand); | ||||
1859 | return ConstantExpr::get(I.getOpcode(), ConstOperand, SOC); | ||||
1860 | } | ||||
1861 | |||||
1862 | Value *Op0 = SO, *Op1 = ConstOperand; | ||||
1863 | if (!ConstIsRHS) | ||||
1864 | std::swap(Op0, Op1); | ||||
1865 | Instruction *New; | ||||
1866 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(&I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1867 | New = BinaryOperator::Create(BO->getOpcode(), Op0, Op1,SO->getName()+".op"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1868 | else if (CmpInst *CI = dyn_cast<CmpInst>(&I)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1869 | New = CmpInst::Create(CI->getOpcode(), CI->getPredicate(), Op0, Op1, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1870 | SO->getName()+".cmp"); |
1871 | else { | ||||
1872 | assert(0 && "Unknown binary instruction type!"); | ||||
1873 | abort(); | ||||
1874 | } | ||||
1875 | return IC->InsertNewInstBefore(New, I); | ||||
1876 | } | ||||
1877 | |||||
1878 | // FoldOpIntoSelect - Given an instruction with a select as one operand and a | ||||
1879 | // constant as the other operand, try to fold the binary operator into the | ||||
1880 | // select arguments. This also works for Cast instructions, which obviously do | ||||
1881 | // not have a second operand. | ||||
1882 | static Instruction *FoldOpIntoSelect(Instruction &Op, SelectInst *SI, | ||||
1883 | InstCombiner *IC) { | ||||
1884 | // Don't modify shared select instructions | ||||
1885 | if (!SI->hasOneUse()) return 0; | ||||
1886 | Value *TV = SI->getOperand(1); | ||||
1887 | Value *FV = SI->getOperand(2); | ||||
1888 | |||||
1889 | if (isa<Constant>(TV) || isa<Constant>(FV)) { | ||||
1890 | // Bool selects with constant operands can be folded to logical ops. | ||||
1891 | if (SI->getType() == Type::Int1Ty) return 0; | ||||
1892 | |||||
1893 | Value *SelectTrueVal = FoldOperationIntoSelectOperand(Op, TV, IC); | ||||
1894 | Value *SelectFalseVal = FoldOperationIntoSelectOperand(Op, FV, IC); | ||||
1895 | |||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 1896 | return SelectInst::Create(SI->getCondition(), SelectTrueVal, |
1897 | SelectFalseVal); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1898 | } |
1899 | return 0; | ||||
1900 | } | ||||
1901 | |||||
1902 | |||||
1903 | /// FoldOpIntoPhi - Given a binary operator or cast instruction which has a PHI | ||||
1904 | /// node as operand #0, see if we can fold the instruction into the PHI (which | ||||
1905 | /// is only possible if all operands to the PHI are constants). | ||||
1906 | Instruction *InstCombiner::FoldOpIntoPhi(Instruction &I) { | ||||
1907 | PHINode *PN = cast<PHINode>(I.getOperand(0)); | ||||
1908 | unsigned NumPHIValues = PN->getNumIncomingValues(); | ||||
1909 | if (!PN->hasOneUse() || NumPHIValues == 0) return 0; | ||||
1910 | |||||
1911 | // Check to see if all of the operands of the PHI are constants. If there is | ||||
1912 | // one non-constant value, remember the BB it is. If there is more than one | ||||
1913 | // or if *it* is a PHI, bail out. | ||||
1914 | BasicBlock *NonConstBB = 0; | ||||
1915 | for (unsigned i = 0; i != NumPHIValues; ++i) | ||||
1916 | if (!isa<Constant>(PN->getIncomingValue(i))) { | ||||
1917 | if (NonConstBB) return 0; // More than one non-const value. | ||||
1918 | if (isa<PHINode>(PN->getIncomingValue(i))) return 0; // Itself a phi. | ||||
1919 | NonConstBB = PN->getIncomingBlock(i); | ||||
1920 | |||||
1921 | // If the incoming non-constant value is in I's block, we have an infinite | ||||
1922 | // loop. | ||||
1923 | if (NonConstBB == I.getParent()) | ||||
1924 | return 0; | ||||
1925 | } | ||||
1926 | |||||
1927 | // If there is exactly one non-constant value, we can insert a copy of the | ||||
1928 | // operation in that block. However, if this is a critical edge, we would be | ||||
1929 | // inserting the computation one some other paths (e.g. inside a loop). Only | ||||
1930 | // do this if the pred block is unconditionally branching into the phi block. | ||||
1931 | if (NonConstBB) { | ||||
1932 | BranchInst *BI = dyn_cast<BranchInst>(NonConstBB->getTerminator()); | ||||
1933 | if (!BI || !BI->isUnconditional()) return 0; | ||||
1934 | } | ||||
1935 | |||||
1936 | // Okay, we can do the transformation: create the new PHI node. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 1937 | PHINode *NewPN = PHINode::Create(I.getType(), ""); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1938 | NewPN->reserveOperandSpace(PN->getNumOperands()/2); |
1939 | InsertNewInstBefore(NewPN, *PN); | ||||
1940 | NewPN->takeName(PN); | ||||
1941 | |||||
1942 | // Next, add all of the operands to the PHI. | ||||
1943 | if (I.getNumOperands() == 2) { | ||||
1944 | Constant *C = cast<Constant>(I.getOperand(1)); | ||||
1945 | for (unsigned i = 0; i != NumPHIValues; ++i) { | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 1946 | Value *InV = 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1947 | if (Constant *InC = dyn_cast<Constant>(PN->getIncomingValue(i))) { |
1948 | if (CmpInst *CI = dyn_cast<CmpInst>(&I)) | ||||
1949 | InV = ConstantExpr::getCompare(CI->getPredicate(), InC, C); | ||||
1950 | else | ||||
1951 | InV = ConstantExpr::get(I.getOpcode(), InC, C); | ||||
1952 | } else { | ||||
1953 | assert(PN->getIncomingBlock(i) == NonConstBB); | ||||
1954 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(&I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1955 | InV = BinaryOperator::Create(BO->getOpcode(), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1956 | PN->getIncomingValue(i), C, "phitmp", |
1957 | NonConstBB->getTerminator()); | ||||
1958 | else if (CmpInst *CI = dyn_cast<CmpInst>(&I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1959 | InV = CmpInst::Create(CI->getOpcode(), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1960 | CI->getPredicate(), |
1961 | PN->getIncomingValue(i), C, "phitmp", | ||||
1962 | NonConstBB->getTerminator()); | ||||
1963 | else | ||||
1964 | assert(0 && "Unknown binop!"); | ||||
1965 | |||||
1966 | AddToWorkList(cast<Instruction>(InV)); | ||||
1967 | } | ||||
1968 | NewPN->addIncoming(InV, PN->getIncomingBlock(i)); | ||||
1969 | } | ||||
1970 | } else { | ||||
1971 | CastInst *CI = cast<CastInst>(&I); | ||||
1972 | const Type *RetTy = CI->getType(); | ||||
1973 | for (unsigned i = 0; i != NumPHIValues; ++i) { | ||||
1974 | Value *InV; | ||||
1975 | if (Constant *InC = dyn_cast<Constant>(PN->getIncomingValue(i))) { | ||||
1976 | InV = ConstantExpr::getCast(CI->getOpcode(), InC, RetTy); | ||||
1977 | } else { | ||||
1978 | assert(PN->getIncomingBlock(i) == NonConstBB); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 1979 | InV = CastInst::Create(CI->getOpcode(), PN->getIncomingValue(i), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 1980 | I.getType(), "phitmp", |
1981 | NonConstBB->getTerminator()); | ||||
1982 | AddToWorkList(cast<Instruction>(InV)); | ||||
1983 | } | ||||
1984 | NewPN->addIncoming(InV, PN->getIncomingBlock(i)); | ||||
1985 | } | ||||
1986 | } | ||||
1987 | return ReplaceInstUsesWith(I, NewPN); | ||||
1988 | } | ||||
1989 | |||||
Chris Lattner | 5547616 | 2008-01-29 06:52:45 +0000 | [diff] [blame] | 1990 | |
Chris Lattner | 3554f97 | 2008-05-20 05:46:13 +0000 | [diff] [blame] | 1991 | /// WillNotOverflowSignedAdd - Return true if we can prove that: |
1992 | /// (sext (add LHS, RHS)) === (add (sext LHS), (sext RHS)) | ||||
1993 | /// This basically requires proving that the add in the original type would not | ||||
1994 | /// overflow to change the sign bit or have a carry out. | ||||
1995 | bool InstCombiner::WillNotOverflowSignedAdd(Value *LHS, Value *RHS) { | ||||
1996 | // There are different heuristics we can use for this. Here are some simple | ||||
1997 | // ones. | ||||
1998 | |||||
1999 | // Add has the property that adding any two 2's complement numbers can only | ||||
2000 | // have one carry bit which can change a sign. As such, if LHS and RHS each | ||||
2001 | // have at least two sign bits, we know that the addition of the two values will | ||||
2002 | // sign extend fine. | ||||
2003 | if (ComputeNumSignBits(LHS) > 1 && ComputeNumSignBits(RHS) > 1) | ||||
2004 | return true; | ||||
2005 | |||||
2006 | |||||
2007 | // If one of the operands only has one non-zero bit, and if the other operand | ||||
2008 | // has a known-zero bit in a more significant place than it (not including the | ||||
2009 | // sign bit) the ripple may go up to and fill the zero, but won't change the | ||||
2010 | // sign. For example, (X & ~4) + 1. | ||||
2011 | |||||
2012 | // TODO: Implement. | ||||
2013 | |||||
2014 | return false; | ||||
2015 | } | ||||
2016 | |||||
Chris Lattner | 5547616 | 2008-01-29 06:52:45 +0000 | [diff] [blame] | 2017 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2018 | Instruction *InstCombiner::visitAdd(BinaryOperator &I) { |
2019 | bool Changed = SimplifyCommutative(I); | ||||
2020 | Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); | ||||
2021 | |||||
2022 | if (Constant *RHSC = dyn_cast<Constant>(RHS)) { | ||||
2023 | // X + undef -> undef | ||||
2024 | if (isa<UndefValue>(RHS)) | ||||
2025 | return ReplaceInstUsesWith(I, RHS); | ||||
2026 | |||||
2027 | // X + 0 --> X | ||||
2028 | if (!I.getType()->isFPOrFPVector()) { // NOTE: -0 + +0 = +0. | ||||
2029 | if (RHSC->isNullValue()) | ||||
2030 | return ReplaceInstUsesWith(I, LHS); | ||||
2031 | } else if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHSC)) { | ||||
Dale Johannesen | 2fc2078 | 2007-09-14 22:26:36 +0000 | [diff] [blame] | 2032 | if (CFP->isExactlyValue(ConstantFP::getNegativeZero |
2033 | (I.getType())->getValueAPF())) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2034 | return ReplaceInstUsesWith(I, LHS); |
2035 | } | ||||
2036 | |||||
2037 | if (ConstantInt *CI = dyn_cast<ConstantInt>(RHSC)) { | ||||
2038 | // X + (signbit) --> X ^ signbit | ||||
2039 | const APInt& Val = CI->getValue(); | ||||
2040 | uint32_t BitWidth = Val.getBitWidth(); | ||||
2041 | if (Val == APInt::getSignBit(BitWidth)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2042 | return BinaryOperator::CreateXor(LHS, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2043 | |
2044 | // See if SimplifyDemandedBits can simplify this. This handles stuff like | ||||
2045 | // (X & 254)+1 -> (X&254)|1 | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 2046 | if (!isa<VectorType>(I.getType()) && SimplifyDemandedInstructionBits(I)) |
2047 | return &I; | ||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 2048 | |
2049 | // zext(i1) - 1 -> select i1, 0, -1 | ||||
2050 | if (ZExtInst *ZI = dyn_cast<ZExtInst>(LHS)) | ||||
2051 | if (CI->isAllOnesValue() && | ||||
2052 | ZI->getOperand(0)->getType() == Type::Int1Ty) | ||||
2053 | return SelectInst::Create(ZI->getOperand(0), | ||||
2054 | Constant::getNullValue(I.getType()), | ||||
2055 | ConstantInt::getAllOnesValue(I.getType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2056 | } |
2057 | |||||
2058 | if (isa<PHINode>(LHS)) | ||||
2059 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2060 | return NV; | ||||
2061 | |||||
2062 | ConstantInt *XorRHS = 0; | ||||
2063 | Value *XorLHS = 0; | ||||
2064 | if (isa<ConstantInt>(RHSC) && | ||||
2065 | match(LHS, m_Xor(m_Value(XorLHS), m_ConstantInt(XorRHS)))) { | ||||
2066 | uint32_t TySizeBits = I.getType()->getPrimitiveSizeInBits(); | ||||
2067 | const APInt& RHSVal = cast<ConstantInt>(RHSC)->getValue(); | ||||
2068 | |||||
2069 | uint32_t Size = TySizeBits / 2; | ||||
2070 | APInt C0080Val(APInt(TySizeBits, 1ULL).shl(Size - 1)); | ||||
2071 | APInt CFF80Val(-C0080Val); | ||||
2072 | do { | ||||
2073 | if (TySizeBits > Size) { | ||||
2074 | // If we have ADD(XOR(AND(X, 0xFF), 0x80), 0xF..F80), it's a sext. | ||||
2075 | // If we have ADD(XOR(AND(X, 0xFF), 0xF..F80), 0x80), it's a sext. | ||||
2076 | if ((RHSVal == CFF80Val && XorRHS->getValue() == C0080Val) || | ||||
2077 | (RHSVal == C0080Val && XorRHS->getValue() == CFF80Val)) { | ||||
2078 | // This is a sign extend if the top bits are known zero. | ||||
2079 | if (!MaskedValueIsZero(XorLHS, | ||||
2080 | APInt::getHighBitsSet(TySizeBits, TySizeBits - Size))) | ||||
2081 | Size = 0; // Not a sign ext, but can't be any others either. | ||||
2082 | break; | ||||
2083 | } | ||||
2084 | } | ||||
2085 | Size >>= 1; | ||||
2086 | C0080Val = APIntOps::lshr(C0080Val, Size); | ||||
2087 | CFF80Val = APIntOps::ashr(CFF80Val, Size); | ||||
2088 | } while (Size >= 1); | ||||
2089 | |||||
2090 | // FIXME: This shouldn't be necessary. When the backends can handle types | ||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 2091 | // with funny bit widths then this switch statement should be removed. It |
2092 | // is just here to get the size of the "middle" type back up to something | ||||
2093 | // that the back ends can handle. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2094 | const Type *MiddleType = 0; |
2095 | switch (Size) { | ||||
2096 | default: break; | ||||
2097 | case 32: MiddleType = Type::Int32Ty; break; | ||||
2098 | case 16: MiddleType = Type::Int16Ty; break; | ||||
2099 | case 8: MiddleType = Type::Int8Ty; break; | ||||
2100 | } | ||||
2101 | if (MiddleType) { | ||||
2102 | Instruction *NewTrunc = new TruncInst(XorLHS, MiddleType, "sext"); | ||||
2103 | InsertNewInstBefore(NewTrunc, I); | ||||
2104 | return new SExtInst(NewTrunc, I.getType(), I.getName()); | ||||
2105 | } | ||||
2106 | } | ||||
2107 | } | ||||
2108 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2109 | if (I.getType() == Type::Int1Ty) |
2110 | return BinaryOperator::CreateXor(LHS, RHS); | ||||
2111 | |||||
Nick Lewycky | 4d474cd | 2008-05-23 04:39:38 +0000 | [diff] [blame] | 2112 | // X + X --> X << 1 |
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2113 | if (I.getType()->isInteger()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2114 | if (Instruction *Result = AssociativeOpt(I, AddRHS(RHS))) return Result; |
2115 | |||||
2116 | if (Instruction *RHSI = dyn_cast<Instruction>(RHS)) { | ||||
2117 | if (RHSI->getOpcode() == Instruction::Sub) | ||||
2118 | if (LHS == RHSI->getOperand(1)) // A + (B - A) --> B | ||||
2119 | return ReplaceInstUsesWith(I, RHSI->getOperand(0)); | ||||
2120 | } | ||||
2121 | if (Instruction *LHSI = dyn_cast<Instruction>(LHS)) { | ||||
2122 | if (LHSI->getOpcode() == Instruction::Sub) | ||||
2123 | if (RHS == LHSI->getOperand(1)) // (B - A) + A --> B | ||||
2124 | return ReplaceInstUsesWith(I, LHSI->getOperand(0)); | ||||
2125 | } | ||||
2126 | } | ||||
2127 | |||||
2128 | // -A + B --> B - A | ||||
Chris Lattner | 53c9fbf | 2008-02-17 21:03:36 +0000 | [diff] [blame] | 2129 | // -A + -B --> -(A + B) |
2130 | if (Value *LHSV = dyn_castNegVal(LHS)) { | ||||
Chris Lattner | 322a919 | 2008-02-18 17:50:16 +0000 | [diff] [blame] | 2131 | if (LHS->getType()->isIntOrIntVector()) { |
2132 | if (Value *RHSV = dyn_castNegVal(RHS)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2133 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSV, RHSV, "sum"); |
Chris Lattner | 322a919 | 2008-02-18 17:50:16 +0000 | [diff] [blame] | 2134 | InsertNewInstBefore(NewAdd, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2135 | return BinaryOperator::CreateNeg(NewAdd); |
Chris Lattner | 322a919 | 2008-02-18 17:50:16 +0000 | [diff] [blame] | 2136 | } |
Chris Lattner | 53c9fbf | 2008-02-17 21:03:36 +0000 | [diff] [blame] | 2137 | } |
2138 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2139 | return BinaryOperator::CreateSub(RHS, LHSV); |
Chris Lattner | 53c9fbf | 2008-02-17 21:03:36 +0000 | [diff] [blame] | 2140 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2141 | |
2142 | // A + -B --> A - B | ||||
2143 | if (!isa<Constant>(RHS)) | ||||
2144 | if (Value *V = dyn_castNegVal(RHS)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2145 | return BinaryOperator::CreateSub(LHS, V); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2146 | |
2147 | |||||
2148 | ConstantInt *C2; | ||||
2149 | if (Value *X = dyn_castFoldableMul(LHS, C2)) { | ||||
2150 | if (X == RHS) // X*C + X --> X * (C+1) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2151 | return BinaryOperator::CreateMul(RHS, AddOne(C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2152 | |
2153 | // X*C1 + X*C2 --> X * (C1+C2) | ||||
2154 | ConstantInt *C1; | ||||
2155 | if (X == dyn_castFoldableMul(RHS, C1)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2156 | return BinaryOperator::CreateMul(X, Add(C1, C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2157 | } |
2158 | |||||
2159 | // X + X*C --> X * (C+1) | ||||
2160 | if (dyn_castFoldableMul(RHS, C2) == LHS) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2161 | return BinaryOperator::CreateMul(LHS, AddOne(C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2162 | |
2163 | // X + ~X --> -1 since ~X = -X-1 | ||||
2164 | if (dyn_castNotVal(LHS) == RHS || dyn_castNotVal(RHS) == LHS) | ||||
2165 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
2166 | |||||
2167 | |||||
2168 | // (A & C1)+(B & C2) --> (A & C1)|(B & C2) iff C1&C2 == 0 | ||||
2169 | if (match(RHS, m_And(m_Value(), m_ConstantInt(C2)))) | ||||
2170 | if (Instruction *R = AssociativeOpt(I, AddMaskingAnd(C2))) | ||||
2171 | return R; | ||||
Chris Lattner | c1575ce | 2008-05-19 20:01:56 +0000 | [diff] [blame] | 2172 | |
2173 | // A+B --> A|B iff A and B have no bits set in common. | ||||
2174 | if (const IntegerType *IT = dyn_cast<IntegerType>(I.getType())) { | ||||
2175 | APInt Mask = APInt::getAllOnesValue(IT->getBitWidth()); | ||||
2176 | APInt LHSKnownOne(IT->getBitWidth(), 0); | ||||
2177 | APInt LHSKnownZero(IT->getBitWidth(), 0); | ||||
2178 | ComputeMaskedBits(LHS, Mask, LHSKnownZero, LHSKnownOne); | ||||
2179 | if (LHSKnownZero != 0) { | ||||
2180 | APInt RHSKnownOne(IT->getBitWidth(), 0); | ||||
2181 | APInt RHSKnownZero(IT->getBitWidth(), 0); | ||||
2182 | ComputeMaskedBits(RHS, Mask, RHSKnownZero, RHSKnownOne); | ||||
2183 | |||||
2184 | // No bits in common -> bitwise or. | ||||
Chris Lattner | 130443c | 2008-05-19 20:03:53 +0000 | [diff] [blame] | 2185 | if ((LHSKnownZero|RHSKnownZero).isAllOnesValue()) |
Chris Lattner | c1575ce | 2008-05-19 20:01:56 +0000 | [diff] [blame] | 2186 | return BinaryOperator::CreateOr(LHS, RHS); |
Chris Lattner | c1575ce | 2008-05-19 20:01:56 +0000 | [diff] [blame] | 2187 | } |
2188 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2189 | |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2190 | // W*X + Y*Z --> W * (X+Z) iff W == Y |
Nick Lewycky | 5d03b51 | 2008-02-03 08:19:11 +0000 | [diff] [blame] | 2191 | if (I.getType()->isIntOrIntVector()) { |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2192 | Value *W, *X, *Y, *Z; |
2193 | if (match(LHS, m_Mul(m_Value(W), m_Value(X))) && | ||||
2194 | match(RHS, m_Mul(m_Value(Y), m_Value(Z)))) { | ||||
2195 | if (W != Y) { | ||||
2196 | if (W == Z) { | ||||
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 2197 | std::swap(Y, Z); |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2198 | } else if (Y == X) { |
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 2199 | std::swap(W, X); |
2200 | } else if (X == Z) { | ||||
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2201 | std::swap(Y, Z); |
2202 | std::swap(W, X); | ||||
2203 | } | ||||
2204 | } | ||||
2205 | |||||
2206 | if (W == Y) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2207 | Value *NewAdd = InsertNewInstBefore(BinaryOperator::CreateAdd(X, Z, |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2208 | LHS->getName()), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2209 | return BinaryOperator::CreateMul(W, NewAdd); |
Nick Lewycky | 83598a7 | 2008-02-03 07:42:09 +0000 | [diff] [blame] | 2210 | } |
2211 | } | ||||
2212 | } | ||||
2213 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2214 | if (ConstantInt *CRHS = dyn_cast<ConstantInt>(RHS)) { |
2215 | Value *X = 0; | ||||
2216 | 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] | 2217 | return BinaryOperator::CreateSub(SubOne(CRHS), X); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2218 | |
2219 | // (X & FF00) + xx00 -> (X+xx00) & FF00 | ||||
2220 | if (LHS->hasOneUse() && match(LHS, m_And(m_Value(X), m_ConstantInt(C2)))) { | ||||
2221 | Constant *Anded = And(CRHS, C2); | ||||
2222 | if (Anded == CRHS) { | ||||
2223 | // See if all bits from the first bit set in the Add RHS up are included | ||||
2224 | // in the mask. First, get the rightmost bit. | ||||
2225 | const APInt& AddRHSV = CRHS->getValue(); | ||||
2226 | |||||
2227 | // Form a mask of all bits from the lowest bit added through the top. | ||||
2228 | APInt AddRHSHighBits(~((AddRHSV & -AddRHSV)-1)); | ||||
2229 | |||||
2230 | // See if the and mask includes all of these bits. | ||||
2231 | APInt AddRHSHighBitsAnd(AddRHSHighBits & C2->getValue()); | ||||
2232 | |||||
2233 | if (AddRHSHighBits == AddRHSHighBitsAnd) { | ||||
2234 | // Okay, the xform is safe. Insert the new add pronto. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2235 | Value *NewAdd = InsertNewInstBefore(BinaryOperator::CreateAdd(X, CRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2236 | LHS->getName()), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2237 | return BinaryOperator::CreateAnd(NewAdd, C2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2238 | } |
2239 | } | ||||
2240 | } | ||||
2241 | |||||
2242 | // Try to fold constant add into select arguments. | ||||
2243 | if (SelectInst *SI = dyn_cast<SelectInst>(LHS)) | ||||
2244 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2245 | return R; | ||||
2246 | } | ||||
2247 | |||||
2248 | // add (cast *A to intptrtype) B -> | ||||
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2249 | // cast (GEP (cast *A to sbyte*) B) --> intptrtype |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2250 | { |
2251 | CastInst *CI = dyn_cast<CastInst>(LHS); | ||||
2252 | Value *Other = RHS; | ||||
2253 | if (!CI) { | ||||
2254 | CI = dyn_cast<CastInst>(RHS); | ||||
2255 | Other = LHS; | ||||
2256 | } | ||||
2257 | if (CI && CI->getType()->isSized() && | ||||
2258 | (CI->getType()->getPrimitiveSizeInBits() == | ||||
2259 | TD->getIntPtrType()->getPrimitiveSizeInBits()) | ||||
2260 | && isa<PointerType>(CI->getOperand(0)->getType())) { | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 2261 | unsigned AS = |
2262 | cast<PointerType>(CI->getOperand(0)->getType())->getAddressSpace(); | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 2263 | Value *I2 = InsertBitCastBefore(CI->getOperand(0), |
2264 | PointerType::get(Type::Int8Ty, AS), I); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2265 | I2 = InsertNewInstBefore(GetElementPtrInst::Create(I2, Other, "ctg2"), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2266 | return new PtrToIntInst(I2, CI->getType()); |
2267 | } | ||||
2268 | } | ||||
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2269 | |
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2270 | // add (select X 0 (sub n A)) A --> select X A n |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2271 | { |
2272 | SelectInst *SI = dyn_cast<SelectInst>(LHS); | ||||
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 2273 | Value *A = RHS; |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2274 | if (!SI) { |
2275 | SI = dyn_cast<SelectInst>(RHS); | ||||
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 2276 | A = LHS; |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2277 | } |
Chris Lattner | bf0c5f3 | 2007-12-20 01:56:58 +0000 | [diff] [blame] | 2278 | if (SI && SI->hasOneUse()) { |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2279 | Value *TV = SI->getTrueValue(); |
2280 | Value *FV = SI->getFalseValue(); | ||||
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 2281 | Value *N; |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2282 | |
2283 | // Can we fold the add into the argument of the select? | ||||
2284 | // We check both true and false select arguments for a matching subtract. | ||||
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 2285 | if (match(FV, m_Zero()) && match(TV, m_Sub(m_Value(N), m_Specific(A)))) |
2286 | // Fold the add into the true select value. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2287 | return SelectInst::Create(SI->getCondition(), N, A); |
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 2288 | if (match(TV, m_Zero()) && match(FV, m_Sub(m_Value(N), m_Specific(A)))) |
2289 | // Fold the add into the false select value. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2290 | return SelectInst::Create(SI->getCondition(), A, N); |
Christopher Lamb | 244ec28 | 2007-12-18 09:34:41 +0000 | [diff] [blame] | 2291 | } |
2292 | } | ||||
Chris Lattner | 5547616 | 2008-01-29 06:52:45 +0000 | [diff] [blame] | 2293 | |
2294 | // Check for X+0.0. Simplify it to X if we know X is not -0.0. | ||||
2295 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHS)) | ||||
2296 | if (CFP->getValueAPF().isPosZero() && CannotBeNegativeZero(LHS)) | ||||
2297 | return ReplaceInstUsesWith(I, LHS); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2298 | |
Chris Lattner | 3554f97 | 2008-05-20 05:46:13 +0000 | [diff] [blame] | 2299 | // Check for (add (sext x), y), see if we can merge this into an |
2300 | // integer add followed by a sext. | ||||
2301 | if (SExtInst *LHSConv = dyn_cast<SExtInst>(LHS)) { | ||||
2302 | // (add (sext x), cst) --> (sext (add x, cst')) | ||||
2303 | if (ConstantInt *RHSC = dyn_cast<ConstantInt>(RHS)) { | ||||
2304 | Constant *CI = | ||||
2305 | ConstantExpr::getTrunc(RHSC, LHSConv->getOperand(0)->getType()); | ||||
2306 | if (LHSConv->hasOneUse() && | ||||
2307 | ConstantExpr::getSExt(CI, I.getType()) == RHSC && | ||||
2308 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), CI)) { | ||||
2309 | // Insert the new, smaller add. | ||||
2310 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2311 | CI, "addconv"); | ||||
2312 | InsertNewInstBefore(NewAdd, I); | ||||
2313 | return new SExtInst(NewAdd, I.getType()); | ||||
2314 | } | ||||
2315 | } | ||||
2316 | |||||
2317 | // (add (sext x), (sext y)) --> (sext (add int x, y)) | ||||
2318 | if (SExtInst *RHSConv = dyn_cast<SExtInst>(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 sexts), and if the | ||||
2321 | // 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 SExtInst(NewAdd, I.getType()); | ||||
2332 | } | ||||
2333 | } | ||||
2334 | } | ||||
2335 | |||||
2336 | // Check for (add double (sitofp x), y), see if we can merge this into an | ||||
2337 | // integer add followed by a promotion. | ||||
2338 | if (SIToFPInst *LHSConv = dyn_cast<SIToFPInst>(LHS)) { | ||||
2339 | // (add double (sitofp x), fpcst) --> (sitofp (add int x, intcst)) | ||||
2340 | // ... if the constant fits in the integer value. This is useful for things | ||||
2341 | // like (double)(x & 1234) + 4.0 -> (double)((X & 1234)+4) which no longer | ||||
2342 | // requires a constant pool load, and generally allows the add to be better | ||||
2343 | // instcombined. | ||||
2344 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHS)) { | ||||
2345 | Constant *CI = | ||||
2346 | ConstantExpr::getFPToSI(CFP, LHSConv->getOperand(0)->getType()); | ||||
2347 | if (LHSConv->hasOneUse() && | ||||
2348 | ConstantExpr::getSIToFP(CI, I.getType()) == CFP && | ||||
2349 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), CI)) { | ||||
2350 | // Insert the new integer add. | ||||
2351 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2352 | CI, "addconv"); | ||||
2353 | InsertNewInstBefore(NewAdd, I); | ||||
2354 | return new SIToFPInst(NewAdd, I.getType()); | ||||
2355 | } | ||||
2356 | } | ||||
2357 | |||||
2358 | // (add double (sitofp x), (sitofp y)) --> (sitofp (add int x, y)) | ||||
2359 | if (SIToFPInst *RHSConv = dyn_cast<SIToFPInst>(RHS)) { | ||||
2360 | // Only do this if x/y have the same type, if at last one of them has a | ||||
2361 | // single use (so we don't increase the number of int->fp conversions), | ||||
2362 | // and if the integer add will not overflow. | ||||
2363 | if (LHSConv->getOperand(0)->getType()==RHSConv->getOperand(0)->getType()&& | ||||
2364 | (LHSConv->hasOneUse() || RHSConv->hasOneUse()) && | ||||
2365 | WillNotOverflowSignedAdd(LHSConv->getOperand(0), | ||||
2366 | RHSConv->getOperand(0))) { | ||||
2367 | // Insert the new integer add. | ||||
2368 | Instruction *NewAdd = BinaryOperator::CreateAdd(LHSConv->getOperand(0), | ||||
2369 | RHSConv->getOperand(0), | ||||
2370 | "addconv"); | ||||
2371 | InsertNewInstBefore(NewAdd, I); | ||||
2372 | return new SIToFPInst(NewAdd, I.getType()); | ||||
2373 | } | ||||
2374 | } | ||||
2375 | } | ||||
2376 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2377 | return Changed ? &I : 0; |
2378 | } | ||||
2379 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2380 | Instruction *InstCombiner::visitSub(BinaryOperator &I) { |
2381 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2382 | |||||
Chris Lattner | 27fbef4 | 2008-07-17 06:07:20 +0000 | [diff] [blame] | 2383 | if (Op0 == Op1 && // sub X, X -> 0 |
2384 | !I.getType()->isFPOrFPVector()) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2385 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
2386 | |||||
2387 | // If this is a 'B = x-(-A)', change to B = x+A... | ||||
2388 | if (Value *V = dyn_castNegVal(Op1)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2389 | return BinaryOperator::CreateAdd(Op0, V); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2390 | |
2391 | if (isa<UndefValue>(Op0)) | ||||
2392 | return ReplaceInstUsesWith(I, Op0); // undef - X -> undef | ||||
2393 | if (isa<UndefValue>(Op1)) | ||||
2394 | return ReplaceInstUsesWith(I, Op1); // X - undef -> undef | ||||
2395 | |||||
2396 | if (ConstantInt *C = dyn_cast<ConstantInt>(Op0)) { | ||||
2397 | // Replace (-1 - A) with (~A)... | ||||
2398 | if (C->isAllOnesValue()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2399 | return BinaryOperator::CreateNot(Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2400 | |
2401 | // C - ~X == X + (1+C) | ||||
2402 | Value *X = 0; | ||||
2403 | if (match(Op1, m_Not(m_Value(X)))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2404 | return BinaryOperator::CreateAdd(X, AddOne(C)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2405 | |
2406 | // -(X >>u 31) -> (X >>s 31) | ||||
2407 | // -(X >>s 31) -> (X >>u 31) | ||||
2408 | if (C->isZero()) { | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2409 | if (BinaryOperator *SI = dyn_cast<BinaryOperator>(Op1)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2410 | if (SI->getOpcode() == Instruction::LShr) { |
2411 | if (ConstantInt *CU = dyn_cast<ConstantInt>(SI->getOperand(1))) { | ||||
2412 | // Check to see if we are shifting out everything but the sign bit. | ||||
2413 | if (CU->getLimitedValue(SI->getType()->getPrimitiveSizeInBits()) == | ||||
2414 | SI->getType()->getPrimitiveSizeInBits()-1) { | ||||
2415 | // Ok, the transformation is safe. Insert AShr. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2416 | return BinaryOperator::Create(Instruction::AShr, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2417 | SI->getOperand(0), CU, SI->getName()); |
2418 | } | ||||
2419 | } | ||||
2420 | } | ||||
2421 | else if (SI->getOpcode() == Instruction::AShr) { | ||||
2422 | if (ConstantInt *CU = dyn_cast<ConstantInt>(SI->getOperand(1))) { | ||||
2423 | // Check to see if we are shifting out everything but the sign bit. | ||||
2424 | if (CU->getLimitedValue(SI->getType()->getPrimitiveSizeInBits()) == | ||||
2425 | SI->getType()->getPrimitiveSizeInBits()-1) { | ||||
2426 | // Ok, the transformation is safe. Insert LShr. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2427 | return BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2428 | SI->getOperand(0), CU, SI->getName()); |
2429 | } | ||||
2430 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2431 | } |
2432 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2433 | } |
2434 | |||||
2435 | // Try to fold constant sub into select arguments. | ||||
2436 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) | ||||
2437 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2438 | return R; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2439 | } |
2440 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2441 | if (I.getType() == Type::Int1Ty) |
2442 | return BinaryOperator::CreateXor(Op0, Op1); | ||||
2443 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2444 | if (BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1)) { |
2445 | if (Op1I->getOpcode() == Instruction::Add && | ||||
2446 | !Op0->getType()->isFPOrFPVector()) { | ||||
2447 | if (Op1I->getOperand(0) == Op0) // X-(X+Y) == -Y | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2448 | return BinaryOperator::CreateNeg(Op1I->getOperand(1), I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2449 | else if (Op1I->getOperand(1) == Op0) // X-(Y+X) == -Y |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2450 | return BinaryOperator::CreateNeg(Op1I->getOperand(0), I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2451 | else if (ConstantInt *CI1 = dyn_cast<ConstantInt>(I.getOperand(0))) { |
2452 | if (ConstantInt *CI2 = dyn_cast<ConstantInt>(Op1I->getOperand(1))) | ||||
2453 | // C1-(X+C2) --> (C1-C2)-X | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2454 | return BinaryOperator::CreateSub(Subtract(CI1, CI2), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2455 | Op1I->getOperand(0)); |
2456 | } | ||||
2457 | } | ||||
2458 | |||||
2459 | if (Op1I->hasOneUse()) { | ||||
2460 | // Replace (x - (y - z)) with (x + (z - y)) if the (y - z) subexpression | ||||
2461 | // is not used by anyone else... | ||||
2462 | // | ||||
2463 | if (Op1I->getOpcode() == Instruction::Sub && | ||||
2464 | !Op1I->getType()->isFPOrFPVector()) { | ||||
2465 | // Swap the two operands of the subexpr... | ||||
2466 | Value *IIOp0 = Op1I->getOperand(0), *IIOp1 = Op1I->getOperand(1); | ||||
2467 | Op1I->setOperand(0, IIOp1); | ||||
2468 | Op1I->setOperand(1, IIOp0); | ||||
2469 | |||||
2470 | // Create the new top level add instruction... | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2471 | return BinaryOperator::CreateAdd(Op0, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2472 | } |
2473 | |||||
2474 | // Replace (A - (A & B)) with (A & ~B) if this is the only use of (A&B)... | ||||
2475 | // | ||||
2476 | if (Op1I->getOpcode() == Instruction::And && | ||||
2477 | (Op1I->getOperand(0) == Op0 || Op1I->getOperand(1) == Op0)) { | ||||
2478 | Value *OtherOp = Op1I->getOperand(Op1I->getOperand(0) == Op0); | ||||
2479 | |||||
2480 | Value *NewNot = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2481 | InsertNewInstBefore(BinaryOperator::CreateNot(OtherOp, "B.not"), I); |
2482 | return BinaryOperator::CreateAnd(Op0, NewNot); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2483 | } |
2484 | |||||
2485 | // 0 - (X sdiv C) -> (X sdiv -C) | ||||
2486 | if (Op1I->getOpcode() == Instruction::SDiv) | ||||
2487 | if (ConstantInt *CSI = dyn_cast<ConstantInt>(Op0)) | ||||
2488 | if (CSI->isZero()) | ||||
2489 | if (Constant *DivRHS = dyn_cast<Constant>(Op1I->getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2490 | return BinaryOperator::CreateSDiv(Op1I->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2491 | ConstantExpr::getNeg(DivRHS)); |
2492 | |||||
2493 | // X - X*C --> X * (1-C) | ||||
2494 | ConstantInt *C2 = 0; | ||||
2495 | if (dyn_castFoldableMul(Op1I, C2) == Op0) { | ||||
2496 | Constant *CP1 = Subtract(ConstantInt::get(I.getType(), 1), C2); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2497 | return BinaryOperator::CreateMul(Op0, CP1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2498 | } |
2499 | } | ||||
2500 | } | ||||
2501 | |||||
2502 | if (!Op0->getType()->isFPOrFPVector()) | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2503 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2504 | if (Op0I->getOpcode() == Instruction::Add) { |
2505 | if (Op0I->getOperand(0) == Op1) // (Y+X)-Y == X | ||||
2506 | return ReplaceInstUsesWith(I, Op0I->getOperand(1)); | ||||
2507 | else if (Op0I->getOperand(1) == Op1) // (X+Y)-Y == X | ||||
2508 | return ReplaceInstUsesWith(I, Op0I->getOperand(0)); | ||||
2509 | } else if (Op0I->getOpcode() == Instruction::Sub) { | ||||
2510 | if (Op0I->getOperand(0) == Op1) // (X-Y)-X == -Y | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2511 | return BinaryOperator::CreateNeg(Op0I->getOperand(1), I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2512 | } |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 2513 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2514 | |
2515 | ConstantInt *C1; | ||||
2516 | if (Value *X = dyn_castFoldableMul(Op0, C1)) { | ||||
2517 | if (X == Op1) // X*C - X --> X * (C-1) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2518 | return BinaryOperator::CreateMul(Op1, SubOne(C1)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2519 | |
2520 | ConstantInt *C2; // X*C1 - X*C2 -> X * (C1-C2) | ||||
2521 | if (X == dyn_castFoldableMul(Op1, C2)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2522 | return BinaryOperator::CreateMul(X, Subtract(C1, C2)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2523 | } |
2524 | return 0; | ||||
2525 | } | ||||
2526 | |||||
2527 | /// isSignBitCheck - Given an exploded icmp instruction, return true if the | ||||
2528 | /// comparison only checks the sign bit. If it only checks the sign bit, set | ||||
2529 | /// TrueIfSigned if the result of the comparison is true when the input value is | ||||
2530 | /// signed. | ||||
2531 | static bool isSignBitCheck(ICmpInst::Predicate pred, ConstantInt *RHS, | ||||
2532 | bool &TrueIfSigned) { | ||||
2533 | switch (pred) { | ||||
2534 | case ICmpInst::ICMP_SLT: // True if LHS s< 0 | ||||
2535 | TrueIfSigned = true; | ||||
2536 | return RHS->isZero(); | ||||
2537 | case ICmpInst::ICMP_SLE: // True if LHS s<= RHS and RHS == -1 | ||||
2538 | TrueIfSigned = true; | ||||
2539 | return RHS->isAllOnesValue(); | ||||
2540 | case ICmpInst::ICMP_SGT: // True if LHS s> -1 | ||||
2541 | TrueIfSigned = false; | ||||
2542 | return RHS->isAllOnesValue(); | ||||
2543 | case ICmpInst::ICMP_UGT: | ||||
2544 | // True if LHS u> RHS and RHS == high-bit-mask - 1 | ||||
2545 | TrueIfSigned = true; | ||||
2546 | return RHS->getValue() == | ||||
2547 | APInt::getSignedMaxValue(RHS->getType()->getPrimitiveSizeInBits()); | ||||
2548 | case ICmpInst::ICMP_UGE: | ||||
2549 | // True if LHS u>= RHS and RHS == high-bit-mask (2^7, 2^15, 2^31, etc) | ||||
2550 | TrueIfSigned = true; | ||||
Chris Lattner | 60813c2 | 2008-06-02 01:29:46 +0000 | [diff] [blame] | 2551 | return RHS->getValue().isSignBit(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2552 | default: |
2553 | return false; | ||||
2554 | } | ||||
2555 | } | ||||
2556 | |||||
2557 | Instruction *InstCombiner::visitMul(BinaryOperator &I) { | ||||
2558 | bool Changed = SimplifyCommutative(I); | ||||
2559 | Value *Op0 = I.getOperand(0); | ||||
2560 | |||||
2561 | if (isa<UndefValue>(I.getOperand(1))) // undef * X -> 0 | ||||
2562 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2563 | |||||
2564 | // Simplify mul instructions with a constant RHS... | ||||
2565 | if (Constant *Op1 = dyn_cast<Constant>(I.getOperand(1))) { | ||||
2566 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) { | ||||
2567 | |||||
2568 | // ((X << C1)*C2) == (X * (C2 << C1)) | ||||
2569 | if (BinaryOperator *SI = dyn_cast<BinaryOperator>(Op0)) | ||||
2570 | if (SI->getOpcode() == Instruction::Shl) | ||||
2571 | if (Constant *ShOp = dyn_cast<Constant>(SI->getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2572 | return BinaryOperator::CreateMul(SI->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2573 | ConstantExpr::getShl(CI, ShOp)); |
2574 | |||||
2575 | if (CI->isZero()) | ||||
2576 | return ReplaceInstUsesWith(I, Op1); // X * 0 == 0 | ||||
2577 | if (CI->equalsInt(1)) // X * 1 == X | ||||
2578 | return ReplaceInstUsesWith(I, Op0); | ||||
2579 | if (CI->isAllOnesValue()) // X * -1 == 0 - X | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2580 | return BinaryOperator::CreateNeg(Op0, I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2581 | |
2582 | const APInt& Val = cast<ConstantInt>(CI)->getValue(); | ||||
2583 | if (Val.isPowerOf2()) { // Replace X*(2^C) with X << C | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2584 | return BinaryOperator::CreateShl(Op0, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2585 | ConstantInt::get(Op0->getType(), Val.logBase2())); |
2586 | } | ||||
2587 | } else if (ConstantFP *Op1F = dyn_cast<ConstantFP>(Op1)) { | ||||
2588 | if (Op1F->isNullValue()) | ||||
2589 | return ReplaceInstUsesWith(I, Op1); | ||||
2590 | |||||
2591 | // "In IEEE floating point, x*1 is not equivalent to x for nans. However, | ||||
2592 | // 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] | 2593 | if (Op1F->isExactlyValue(1.0)) |
2594 | return ReplaceInstUsesWith(I, Op0); // Eliminate 'mul double %X, 1.0' | ||||
2595 | } else if (isa<VectorType>(Op1->getType())) { | ||||
2596 | if (isa<ConstantAggregateZero>(Op1)) | ||||
2597 | return ReplaceInstUsesWith(I, Op1); | ||||
Nick Lewycky | 9441873 | 2008-11-27 20:21:08 +0000 | [diff] [blame] | 2598 | |
2599 | if (ConstantVector *Op1V = dyn_cast<ConstantVector>(Op1)) { | ||||
2600 | if (Op1V->isAllOnesValue()) // X * -1 == 0 - X | ||||
2601 | return BinaryOperator::CreateNeg(Op0, I.getName()); | ||||
2602 | |||||
2603 | // As above, vector X*splat(1.0) -> X in all defined cases. | ||||
2604 | if (Constant *Splat = Op1V->getSplatValue()) { | ||||
2605 | if (ConstantFP *F = dyn_cast<ConstantFP>(Splat)) | ||||
2606 | if (F->isExactlyValue(1.0)) | ||||
2607 | return ReplaceInstUsesWith(I, Op0); | ||||
2608 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Splat)) | ||||
2609 | if (CI->equalsInt(1)) | ||||
2610 | return ReplaceInstUsesWith(I, Op0); | ||||
2611 | } | ||||
2612 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2613 | } |
2614 | |||||
2615 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) | ||||
2616 | if (Op0I->getOpcode() == Instruction::Add && Op0I->hasOneUse() && | ||||
Chris Lattner | 5819408 | 2008-05-18 04:11:26 +0000 | [diff] [blame] | 2617 | isa<ConstantInt>(Op0I->getOperand(1)) && isa<ConstantInt>(Op1)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2618 | // Canonicalize (X+C1)*C2 -> X*C2+C1*C2. |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2619 | Instruction *Add = BinaryOperator::CreateMul(Op0I->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2620 | Op1, "tmp"); |
2621 | InsertNewInstBefore(Add, I); | ||||
2622 | Value *C1C2 = ConstantExpr::getMul(Op1, | ||||
2623 | cast<Constant>(Op0I->getOperand(1))); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2624 | return BinaryOperator::CreateAdd(Add, C1C2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2625 | |
2626 | } | ||||
2627 | |||||
2628 | // Try to fold constant mul into select arguments. | ||||
2629 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
2630 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2631 | return R; | ||||
2632 | |||||
2633 | if (isa<PHINode>(Op0)) | ||||
2634 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2635 | return NV; | ||||
2636 | } | ||||
2637 | |||||
2638 | if (Value *Op0v = dyn_castNegVal(Op0)) // -X * -Y = X*Y | ||||
2639 | if (Value *Op1v = dyn_castNegVal(I.getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2640 | return BinaryOperator::CreateMul(Op0v, Op1v); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2641 | |
Nick Lewycky | 1c24640 | 2008-11-21 07:33:58 +0000 | [diff] [blame] | 2642 | // (X / Y) * Y = X - (X % Y) |
2643 | // (X / Y) * -Y = (X % Y) - X | ||||
2644 | { | ||||
2645 | Value *Op1 = I.getOperand(1); | ||||
2646 | BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0); | ||||
2647 | if (!BO || | ||||
2648 | (BO->getOpcode() != Instruction::UDiv && | ||||
2649 | BO->getOpcode() != Instruction::SDiv)) { | ||||
2650 | Op1 = Op0; | ||||
2651 | BO = dyn_cast<BinaryOperator>(I.getOperand(1)); | ||||
2652 | } | ||||
2653 | Value *Neg = dyn_castNegVal(Op1); | ||||
2654 | if (BO && BO->hasOneUse() && | ||||
2655 | (BO->getOperand(1) == Op1 || BO->getOperand(1) == Neg) && | ||||
2656 | (BO->getOpcode() == Instruction::UDiv || | ||||
2657 | BO->getOpcode() == Instruction::SDiv)) { | ||||
2658 | Value *Op0BO = BO->getOperand(0), *Op1BO = BO->getOperand(1); | ||||
2659 | |||||
2660 | Instruction *Rem; | ||||
2661 | if (BO->getOpcode() == Instruction::UDiv) | ||||
2662 | Rem = BinaryOperator::CreateURem(Op0BO, Op1BO); | ||||
2663 | else | ||||
2664 | Rem = BinaryOperator::CreateSRem(Op0BO, Op1BO); | ||||
2665 | |||||
2666 | InsertNewInstBefore(Rem, I); | ||||
2667 | Rem->takeName(BO); | ||||
2668 | |||||
2669 | if (Op1BO == Op1) | ||||
2670 | return BinaryOperator::CreateSub(Op0BO, Rem); | ||||
2671 | else | ||||
2672 | return BinaryOperator::CreateSub(Rem, Op0BO); | ||||
2673 | } | ||||
2674 | } | ||||
2675 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2676 | if (I.getType() == Type::Int1Ty) |
2677 | return BinaryOperator::CreateAnd(Op0, I.getOperand(1)); | ||||
2678 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2679 | // If one of the operands of the multiply is a cast from a boolean value, then |
2680 | // we know the bool is either zero or one, so this is a 'masking' multiply. | ||||
2681 | // See if we can simplify things based on how the boolean was originally | ||||
2682 | // formed. | ||||
2683 | CastInst *BoolCast = 0; | ||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2684 | if (ZExtInst *CI = dyn_cast<ZExtInst>(Op0)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2685 | if (CI->getOperand(0)->getType() == Type::Int1Ty) |
2686 | BoolCast = CI; | ||||
2687 | if (!BoolCast) | ||||
2688 | if (ZExtInst *CI = dyn_cast<ZExtInst>(I.getOperand(1))) | ||||
2689 | if (CI->getOperand(0)->getType() == Type::Int1Ty) | ||||
2690 | BoolCast = CI; | ||||
2691 | if (BoolCast) { | ||||
2692 | if (ICmpInst *SCI = dyn_cast<ICmpInst>(BoolCast->getOperand(0))) { | ||||
2693 | Value *SCIOp0 = SCI->getOperand(0), *SCIOp1 = SCI->getOperand(1); | ||||
2694 | const Type *SCOpTy = SCIOp0->getType(); | ||||
2695 | bool TIS = false; | ||||
2696 | |||||
2697 | // If the icmp is true iff the sign bit of X is set, then convert this | ||||
2698 | // multiply into a shift/and combination. | ||||
2699 | if (isa<ConstantInt>(SCIOp1) && | ||||
2700 | isSignBitCheck(SCI->getPredicate(), cast<ConstantInt>(SCIOp1), TIS) && | ||||
2701 | TIS) { | ||||
2702 | // Shift the X value right to turn it into "all signbits". | ||||
2703 | Constant *Amt = ConstantInt::get(SCIOp0->getType(), | ||||
2704 | SCOpTy->getPrimitiveSizeInBits()-1); | ||||
2705 | Value *V = | ||||
2706 | InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2707 | BinaryOperator::Create(Instruction::AShr, SCIOp0, Amt, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2708 | BoolCast->getOperand(0)->getName()+ |
2709 | ".mask"), I); | ||||
2710 | |||||
2711 | // If the multiply type is not the same as the source type, sign extend | ||||
2712 | // or truncate to the multiply type. | ||||
2713 | if (I.getType() != V->getType()) { | ||||
2714 | uint32_t SrcBits = V->getType()->getPrimitiveSizeInBits(); | ||||
2715 | uint32_t DstBits = I.getType()->getPrimitiveSizeInBits(); | ||||
2716 | Instruction::CastOps opcode = | ||||
2717 | (SrcBits == DstBits ? Instruction::BitCast : | ||||
2718 | (SrcBits < DstBits ? Instruction::SExt : Instruction::Trunc)); | ||||
2719 | V = InsertCastBefore(opcode, V, I.getType(), I); | ||||
2720 | } | ||||
2721 | |||||
2722 | Value *OtherOp = Op0 == BoolCast ? I.getOperand(1) : Op0; | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2723 | return BinaryOperator::CreateAnd(V, OtherOp); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2724 | } |
2725 | } | ||||
2726 | } | ||||
2727 | |||||
2728 | return Changed ? &I : 0; | ||||
2729 | } | ||||
2730 | |||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 2731 | /// SimplifyDivRemOfSelect - Try to fold a divide or remainder of a select |
2732 | /// instruction. | ||||
2733 | bool InstCombiner::SimplifyDivRemOfSelect(BinaryOperator &I) { | ||||
2734 | SelectInst *SI = cast<SelectInst>(I.getOperand(1)); | ||||
2735 | |||||
2736 | // div/rem X, (Cond ? 0 : Y) -> div/rem X, Y | ||||
2737 | int NonNullOperand = -1; | ||||
2738 | if (Constant *ST = dyn_cast<Constant>(SI->getOperand(1))) | ||||
2739 | if (ST->isNullValue()) | ||||
2740 | NonNullOperand = 2; | ||||
2741 | // div/rem X, (Cond ? Y : 0) -> div/rem X, Y | ||||
2742 | if (Constant *ST = dyn_cast<Constant>(SI->getOperand(2))) | ||||
2743 | if (ST->isNullValue()) | ||||
2744 | NonNullOperand = 1; | ||||
2745 | |||||
2746 | if (NonNullOperand == -1) | ||||
2747 | return false; | ||||
2748 | |||||
2749 | Value *SelectCond = SI->getOperand(0); | ||||
2750 | |||||
2751 | // Change the div/rem to use 'Y' instead of the select. | ||||
2752 | I.setOperand(1, SI->getOperand(NonNullOperand)); | ||||
2753 | |||||
2754 | // Okay, we know we replace the operand of the div/rem with 'Y' with no | ||||
2755 | // problem. However, the select, or the condition of the select may have | ||||
2756 | // multiple uses. Based on our knowledge that the operand must be non-zero, | ||||
2757 | // propagate the known value for the select into other uses of it, and | ||||
2758 | // propagate a known value of the condition into its other users. | ||||
2759 | |||||
2760 | // If the select and condition only have a single use, don't bother with this, | ||||
2761 | // early exit. | ||||
2762 | if (SI->use_empty() && SelectCond->hasOneUse()) | ||||
2763 | return true; | ||||
2764 | |||||
2765 | // Scan the current block backward, looking for other uses of SI. | ||||
2766 | BasicBlock::iterator BBI = &I, BBFront = I.getParent()->begin(); | ||||
2767 | |||||
2768 | while (BBI != BBFront) { | ||||
2769 | --BBI; | ||||
2770 | // If we found a call to a function, we can't assume it will return, so | ||||
2771 | // information from below it cannot be propagated above it. | ||||
2772 | if (isa<CallInst>(BBI) && !isa<IntrinsicInst>(BBI)) | ||||
2773 | break; | ||||
2774 | |||||
2775 | // Replace uses of the select or its condition with the known values. | ||||
2776 | for (Instruction::op_iterator I = BBI->op_begin(), E = BBI->op_end(); | ||||
2777 | I != E; ++I) { | ||||
2778 | if (*I == SI) { | ||||
2779 | *I = SI->getOperand(NonNullOperand); | ||||
2780 | AddToWorkList(BBI); | ||||
2781 | } else if (*I == SelectCond) { | ||||
2782 | *I = NonNullOperand == 1 ? ConstantInt::getTrue() : | ||||
2783 | ConstantInt::getFalse(); | ||||
2784 | AddToWorkList(BBI); | ||||
2785 | } | ||||
2786 | } | ||||
2787 | |||||
2788 | // If we past the instruction, quit looking for it. | ||||
2789 | if (&*BBI == SI) | ||||
2790 | SI = 0; | ||||
2791 | if (&*BBI == SelectCond) | ||||
2792 | SelectCond = 0; | ||||
2793 | |||||
2794 | // If we ran out of things to eliminate, break out of the loop. | ||||
2795 | if (SelectCond == 0 && SI == 0) | ||||
2796 | break; | ||||
2797 | |||||
2798 | } | ||||
2799 | return true; | ||||
2800 | } | ||||
2801 | |||||
2802 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2803 | /// This function implements the transforms on div instructions that work |
2804 | /// regardless of the kind of div instruction it is (udiv, sdiv, or fdiv). It is | ||||
2805 | /// used by the visitors to those instructions. | ||||
2806 | /// @brief Transforms common to all three div instructions | ||||
2807 | Instruction *InstCombiner::commonDivTransforms(BinaryOperator &I) { | ||||
2808 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2809 | |||||
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 2810 | // undef / X -> 0 for integer. |
2811 | // undef / X -> undef for FP (the undef could be a snan). | ||||
2812 | if (isa<UndefValue>(Op0)) { | ||||
2813 | if (Op0->getType()->isFPOrFPVector()) | ||||
2814 | return ReplaceInstUsesWith(I, Op0); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2815 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 2816 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2817 | |
2818 | // X / undef -> undef | ||||
2819 | if (isa<UndefValue>(Op1)) | ||||
2820 | return ReplaceInstUsesWith(I, Op1); | ||||
2821 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2822 | return 0; |
2823 | } | ||||
2824 | |||||
2825 | /// This function implements the transforms common to both integer division | ||||
2826 | /// instructions (udiv and sdiv). It is called by the visitors to those integer | ||||
2827 | /// division instructions. | ||||
2828 | /// @brief Common integer divide transforms | ||||
2829 | Instruction *InstCombiner::commonIDivTransforms(BinaryOperator &I) { | ||||
2830 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2831 | |||||
Chris Lattner | cefb36c | 2008-05-16 02:59:42 +0000 | [diff] [blame] | 2832 | // (sdiv X, X) --> 1 (udiv X, X) --> 1 |
Nick Lewycky | 386c013 | 2008-05-23 03:26:47 +0000 | [diff] [blame] | 2833 | if (Op0 == Op1) { |
2834 | if (const VectorType *Ty = dyn_cast<VectorType>(I.getType())) { | ||||
2835 | ConstantInt *CI = ConstantInt::get(Ty->getElementType(), 1); | ||||
2836 | std::vector<Constant*> Elts(Ty->getNumElements(), CI); | ||||
2837 | return ReplaceInstUsesWith(I, ConstantVector::get(Elts)); | ||||
2838 | } | ||||
2839 | |||||
2840 | ConstantInt *CI = ConstantInt::get(I.getType(), 1); | ||||
2841 | return ReplaceInstUsesWith(I, CI); | ||||
2842 | } | ||||
Chris Lattner | cefb36c | 2008-05-16 02:59:42 +0000 | [diff] [blame] | 2843 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2844 | if (Instruction *Common = commonDivTransforms(I)) |
2845 | return Common; | ||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 2846 | |
2847 | // Handle cases involving: [su]div X, (select Cond, Y, Z) | ||||
2848 | // This does not apply for fdiv. | ||||
2849 | if (isa<SelectInst>(Op1) && SimplifyDivRemOfSelect(I)) | ||||
2850 | return &I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2851 | |
2852 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
2853 | // div X, 1 == X | ||||
2854 | if (RHS->equalsInt(1)) | ||||
2855 | return ReplaceInstUsesWith(I, Op0); | ||||
2856 | |||||
2857 | // (X / C1) / C2 -> X / (C1*C2) | ||||
2858 | if (Instruction *LHS = dyn_cast<Instruction>(Op0)) | ||||
2859 | if (Instruction::BinaryOps(LHS->getOpcode()) == I.getOpcode()) | ||||
2860 | if (ConstantInt *LHSRHS = dyn_cast<ConstantInt>(LHS->getOperand(1))) { | ||||
Nick Lewycky | 9d798f9 | 2008-02-18 22:48:05 +0000 | [diff] [blame] | 2861 | if (MultiplyOverflows(RHS, LHSRHS, I.getOpcode()==Instruction::SDiv)) |
2862 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2863 | else | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2864 | return BinaryOperator::Create(I.getOpcode(), LHS->getOperand(0), |
Nick Lewycky | 9d798f9 | 2008-02-18 22:48:05 +0000 | [diff] [blame] | 2865 | Multiply(RHS, LHSRHS)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2866 | } |
2867 | |||||
2868 | if (!RHS->isZero()) { // avoid X udiv 0 | ||||
2869 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
2870 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
2871 | return R; | ||||
2872 | if (isa<PHINode>(Op0)) | ||||
2873 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
2874 | return NV; | ||||
2875 | } | ||||
2876 | } | ||||
2877 | |||||
2878 | // 0 / X == 0, we don't need to preserve faults! | ||||
2879 | if (ConstantInt *LHS = dyn_cast<ConstantInt>(Op0)) | ||||
2880 | if (LHS->equalsInt(0)) | ||||
2881 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
2882 | |||||
Nick Lewycky | d4b6367 | 2008-05-31 17:59:52 +0000 | [diff] [blame] | 2883 | // It can't be division by zero, hence it must be division by one. |
2884 | if (I.getType() == Type::Int1Ty) | ||||
2885 | return ReplaceInstUsesWith(I, Op0); | ||||
2886 | |||||
Nick Lewycky | 9441873 | 2008-11-27 20:21:08 +0000 | [diff] [blame] | 2887 | if (ConstantVector *Op1V = dyn_cast<ConstantVector>(Op1)) { |
2888 | if (ConstantInt *X = cast_or_null<ConstantInt>(Op1V->getSplatValue())) | ||||
2889 | // div X, 1 == X | ||||
2890 | if (X->isOne()) | ||||
2891 | return ReplaceInstUsesWith(I, Op0); | ||||
2892 | } | ||||
2893 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2894 | return 0; |
2895 | } | ||||
2896 | |||||
2897 | Instruction *InstCombiner::visitUDiv(BinaryOperator &I) { | ||||
2898 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2899 | |||||
2900 | // Handle the integer div common cases | ||||
2901 | if (Instruction *Common = commonIDivTransforms(I)) | ||||
2902 | return Common; | ||||
2903 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2904 | if (ConstantInt *C = dyn_cast<ConstantInt>(Op1)) { |
Nick Lewycky | 240182a | 2008-11-27 22:41:10 +0000 | [diff] [blame] | 2905 | // X udiv C^2 -> X >> C |
2906 | // Check to see if this is an unsigned division with an exact power of 2, | ||||
2907 | // if so, convert to a right shift. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2908 | if (C->getValue().isPowerOf2()) // 0 not included in isPowerOf2 |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2909 | return BinaryOperator::CreateLShr(Op0, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2910 | ConstantInt::get(Op0->getType(), C->getValue().logBase2())); |
Nick Lewycky | 240182a | 2008-11-27 22:41:10 +0000 | [diff] [blame] | 2911 | |
2912 | // X udiv C, where C >= signbit | ||||
2913 | if (C->getValue().isNegative()) { | ||||
2914 | Value *IC = InsertNewInstBefore(new ICmpInst(ICmpInst::ICMP_ULT, Op0, C), | ||||
2915 | I); | ||||
2916 | return SelectInst::Create(IC, Constant::getNullValue(I.getType()), | ||||
2917 | ConstantInt::get(I.getType(), 1)); | ||||
2918 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2919 | } |
2920 | |||||
2921 | // X udiv (C1 << N), where C1 is "1<<C2" --> X >> (N+C2) | ||||
2922 | if (BinaryOperator *RHSI = dyn_cast<BinaryOperator>(I.getOperand(1))) { | ||||
2923 | if (RHSI->getOpcode() == Instruction::Shl && | ||||
2924 | isa<ConstantInt>(RHSI->getOperand(0))) { | ||||
2925 | const APInt& C1 = cast<ConstantInt>(RHSI->getOperand(0))->getValue(); | ||||
2926 | if (C1.isPowerOf2()) { | ||||
2927 | Value *N = RHSI->getOperand(1); | ||||
2928 | const Type *NTy = N->getType(); | ||||
2929 | if (uint32_t C2 = C1.logBase2()) { | ||||
2930 | Constant *C2V = ConstantInt::get(NTy, C2); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2931 | N = InsertNewInstBefore(BinaryOperator::CreateAdd(N, C2V, "tmp"), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2932 | } |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2933 | return BinaryOperator::CreateLShr(Op0, N); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2934 | } |
2935 | } | ||||
2936 | } | ||||
2937 | |||||
2938 | // udiv X, (Select Cond, C1, C2) --> Select Cond, (shr X, C1), (shr X, C2) | ||||
2939 | // where C1&C2 are powers of two. | ||||
2940 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) | ||||
2941 | if (ConstantInt *STO = dyn_cast<ConstantInt>(SI->getOperand(1))) | ||||
2942 | if (ConstantInt *SFO = dyn_cast<ConstantInt>(SI->getOperand(2))) { | ||||
2943 | const APInt &TVA = STO->getValue(), &FVA = SFO->getValue(); | ||||
2944 | if (TVA.isPowerOf2() && FVA.isPowerOf2()) { | ||||
2945 | // Compute the shift amounts | ||||
2946 | uint32_t TSA = TVA.logBase2(), FSA = FVA.logBase2(); | ||||
2947 | // Construct the "on true" case of the select | ||||
2948 | Constant *TC = ConstantInt::get(Op0->getType(), TSA); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2949 | Instruction *TSI = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2950 | Op0, TC, SI->getName()+".t"); |
2951 | TSI = InsertNewInstBefore(TSI, I); | ||||
2952 | |||||
2953 | // Construct the "on false" case of the select | ||||
2954 | Constant *FC = ConstantInt::get(Op0->getType(), FSA); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2955 | Instruction *FSI = BinaryOperator::CreateLShr( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2956 | Op0, FC, SI->getName()+".f"); |
2957 | FSI = InsertNewInstBefore(FSI, I); | ||||
2958 | |||||
2959 | // construct the select instruction and return it. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 2960 | return SelectInst::Create(SI->getOperand(0), TSI, FSI, SI->getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2961 | } |
2962 | } | ||||
2963 | return 0; | ||||
2964 | } | ||||
2965 | |||||
2966 | Instruction *InstCombiner::visitSDiv(BinaryOperator &I) { | ||||
2967 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
2968 | |||||
2969 | // Handle the integer div common cases | ||||
2970 | if (Instruction *Common = commonIDivTransforms(I)) | ||||
2971 | return Common; | ||||
2972 | |||||
2973 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
2974 | // sdiv X, -1 == -X | ||||
2975 | if (RHS->isAllOnesValue()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 2976 | return BinaryOperator::CreateNeg(Op0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2977 | } |
2978 | |||||
2979 | // If the sign bits of both operands are zero (i.e. we can prove they are | ||||
2980 | // unsigned inputs), turn this into a udiv. | ||||
2981 | if (I.getType()->isInteger()) { | ||||
2982 | APInt Mask(APInt::getSignBit(I.getType()->getPrimitiveSizeInBits())); | ||||
2983 | if (MaskedValueIsZero(Op1, Mask) && MaskedValueIsZero(Op0, Mask)) { | ||||
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 2984 | // 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] | 2985 | return BinaryOperator::CreateUDiv(Op0, Op1, I.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2986 | } |
2987 | } | ||||
2988 | |||||
2989 | return 0; | ||||
2990 | } | ||||
2991 | |||||
2992 | Instruction *InstCombiner::visitFDiv(BinaryOperator &I) { | ||||
2993 | return commonDivTransforms(I); | ||||
2994 | } | ||||
2995 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 2996 | /// This function implements the transforms on rem instructions that work |
2997 | /// regardless of the kind of rem instruction it is (urem, srem, or frem). It | ||||
2998 | /// is used by the visitors to those instructions. | ||||
2999 | /// @brief Transforms common to all three rem instructions | ||||
3000 | Instruction *InstCombiner::commonRemTransforms(BinaryOperator &I) { | ||||
3001 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3002 | |||||
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 3003 | if (isa<UndefValue>(Op0)) { // undef % X -> 0 |
3004 | if (I.getType()->isFPOrFPVector()) | ||||
3005 | return ReplaceInstUsesWith(I, Op0); // X % undef -> undef (could be SNaN) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3006 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
Chris Lattner | 653ef3c | 2008-02-19 06:12:18 +0000 | [diff] [blame] | 3007 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3008 | if (isa<UndefValue>(Op1)) |
3009 | return ReplaceInstUsesWith(I, Op1); // X % undef -> undef | ||||
3010 | |||||
3011 | // Handle cases involving: rem X, (select Cond, Y, Z) | ||||
Chris Lattner | 76972db | 2008-07-14 00:15:52 +0000 | [diff] [blame] | 3012 | if (isa<SelectInst>(Op1) && SimplifyDivRemOfSelect(I)) |
3013 | return &I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3014 | |
3015 | return 0; | ||||
3016 | } | ||||
3017 | |||||
3018 | /// This function implements the transforms common to both integer remainder | ||||
3019 | /// instructions (urem and srem). It is called by the visitors to those integer | ||||
3020 | /// remainder instructions. | ||||
3021 | /// @brief Common integer remainder transforms | ||||
3022 | Instruction *InstCombiner::commonIRemTransforms(BinaryOperator &I) { | ||||
3023 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3024 | |||||
3025 | if (Instruction *common = commonRemTransforms(I)) | ||||
3026 | return common; | ||||
3027 | |||||
Dale Johannesen | a51f737 | 2009-01-21 00:35:19 +0000 | [diff] [blame] | 3028 | // 0 % X == 0 for integer, we don't need to preserve faults! |
3029 | if (Constant *LHS = dyn_cast<Constant>(Op0)) | ||||
3030 | if (LHS->isNullValue()) | ||||
3031 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
3032 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3033 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { |
3034 | // X % 0 == undef, we don't need to preserve faults! | ||||
3035 | if (RHS->equalsInt(0)) | ||||
3036 | return ReplaceInstUsesWith(I, UndefValue::get(I.getType())); | ||||
3037 | |||||
3038 | if (RHS->equalsInt(1)) // X % 1 == 0 | ||||
3039 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
3040 | |||||
3041 | if (Instruction *Op0I = dyn_cast<Instruction>(Op0)) { | ||||
3042 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0I)) { | ||||
3043 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
3044 | return R; | ||||
3045 | } else if (isa<PHINode>(Op0I)) { | ||||
3046 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
3047 | return NV; | ||||
3048 | } | ||||
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 3049 | |
3050 | // See if we can fold away this rem instruction. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 3051 | if (SimplifyDemandedInstructionBits(I)) |
Nick Lewycky | c1372c8 | 2008-03-06 06:48:30 +0000 | [diff] [blame] | 3052 | return &I; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3053 | } |
3054 | } | ||||
3055 | |||||
3056 | return 0; | ||||
3057 | } | ||||
3058 | |||||
3059 | Instruction *InstCombiner::visitURem(BinaryOperator &I) { | ||||
3060 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3061 | |||||
3062 | if (Instruction *common = commonIRemTransforms(I)) | ||||
3063 | return common; | ||||
3064 | |||||
3065 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
3066 | // X urem C^2 -> X and C | ||||
3067 | // Check to see if this is an unsigned remainder with an exact power of 2, | ||||
3068 | // if so, convert to a bitwise and. | ||||
3069 | if (ConstantInt *C = dyn_cast<ConstantInt>(RHS)) | ||||
3070 | if (C->getValue().isPowerOf2()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3071 | return BinaryOperator::CreateAnd(Op0, SubOne(C)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3072 | } |
3073 | |||||
3074 | if (Instruction *RHSI = dyn_cast<Instruction>(I.getOperand(1))) { | ||||
3075 | // Turn A % (C << N), where C is 2^k, into A & ((C << N)-1) | ||||
3076 | if (RHSI->getOpcode() == Instruction::Shl && | ||||
3077 | isa<ConstantInt>(RHSI->getOperand(0))) { | ||||
3078 | if (cast<ConstantInt>(RHSI->getOperand(0))->getValue().isPowerOf2()) { | ||||
3079 | Constant *N1 = ConstantInt::getAllOnesValue(I.getType()); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3080 | Value *Add = InsertNewInstBefore(BinaryOperator::CreateAdd(RHSI, N1, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3081 | "tmp"), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3082 | return BinaryOperator::CreateAnd(Op0, Add); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3083 | } |
3084 | } | ||||
3085 | } | ||||
3086 | |||||
3087 | // urem X, (select Cond, 2^C1, 2^C2) --> select Cond, (and X, C1), (and X, C2) | ||||
3088 | // where C1&C2 are powers of two. | ||||
3089 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) { | ||||
3090 | if (ConstantInt *STO = dyn_cast<ConstantInt>(SI->getOperand(1))) | ||||
3091 | if (ConstantInt *SFO = dyn_cast<ConstantInt>(SI->getOperand(2))) { | ||||
3092 | // STO == 0 and SFO == 0 handled above. | ||||
3093 | if ((STO->getValue().isPowerOf2()) && | ||||
3094 | (SFO->getValue().isPowerOf2())) { | ||||
3095 | Value *TrueAnd = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3096 | BinaryOperator::CreateAnd(Op0, SubOne(STO), SI->getName()+".t"), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3097 | Value *FalseAnd = InsertNewInstBefore( |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3098 | BinaryOperator::CreateAnd(Op0, SubOne(SFO), SI->getName()+".f"), I); |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 3099 | return SelectInst::Create(SI->getOperand(0), TrueAnd, FalseAnd); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3100 | } |
3101 | } | ||||
3102 | } | ||||
3103 | |||||
3104 | return 0; | ||||
3105 | } | ||||
3106 | |||||
3107 | Instruction *InstCombiner::visitSRem(BinaryOperator &I) { | ||||
3108 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3109 | |||||
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3110 | // Handle the integer rem common cases |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3111 | if (Instruction *common = commonIRemTransforms(I)) |
3112 | return common; | ||||
3113 | |||||
3114 | if (Value *RHSNeg = dyn_castNegVal(Op1)) | ||||
Nick Lewycky | cfadfbd | 2008-09-03 06:24:21 +0000 | [diff] [blame] | 3115 | if (!isa<Constant>(RHSNeg) || |
3116 | (isa<ConstantInt>(RHSNeg) && | ||||
3117 | cast<ConstantInt>(RHSNeg)->getValue().isStrictlyPositive())) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3118 | // X % -Y -> X % Y |
3119 | AddUsesToWorkList(I); | ||||
3120 | I.setOperand(1, RHSNeg); | ||||
3121 | return &I; | ||||
3122 | } | ||||
Nick Lewycky | 5515c7a | 2008-09-30 06:08:34 +0000 | [diff] [blame] | 3123 | |
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3124 | // 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] | 3125 | // unsigned inputs), turn this into a urem. |
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3126 | if (I.getType()->isInteger()) { |
3127 | APInt Mask(APInt::getSignBit(I.getType()->getPrimitiveSizeInBits())); | ||||
3128 | if (MaskedValueIsZero(Op1, Mask) && MaskedValueIsZero(Op0, Mask)) { | ||||
3129 | // 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] | 3130 | return BinaryOperator::CreateURem(Op0, Op1, I.getName()); |
Dan Gohman | db3dd96 | 2007-11-05 23:16:33 +0000 | [diff] [blame] | 3131 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3132 | } |
3133 | |||||
Nick Lewycky | da9fa43 | 2008-12-18 06:31:11 +0000 | [diff] [blame] | 3134 | // If it's a constant vector, flip any negative values positive. |
Nick Lewycky | fd74683 | 2008-12-20 16:48:00 +0000 | [diff] [blame] | 3135 | if (ConstantVector *RHSV = dyn_cast<ConstantVector>(Op1)) { |
3136 | unsigned VWidth = RHSV->getNumOperands(); | ||||
Nick Lewycky | da9fa43 | 2008-12-18 06:31:11 +0000 | [diff] [blame] | 3137 | |
Nick Lewycky | fd74683 | 2008-12-20 16:48:00 +0000 | [diff] [blame] | 3138 | bool hasNegative = false; |
3139 | for (unsigned i = 0; !hasNegative && i != VWidth; ++i) | ||||
3140 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(RHSV->getOperand(i))) | ||||
3141 | if (RHS->getValue().isNegative()) | ||||
3142 | hasNegative = true; | ||||
3143 | |||||
3144 | if (hasNegative) { | ||||
3145 | std::vector<Constant *> Elts(VWidth); | ||||
Nick Lewycky | da9fa43 | 2008-12-18 06:31:11 +0000 | [diff] [blame] | 3146 | for (unsigned i = 0; i != VWidth; ++i) { |
3147 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(RHSV->getOperand(i))) { | ||||
3148 | if (RHS->getValue().isNegative()) | ||||
3149 | Elts[i] = cast<ConstantInt>(ConstantExpr::getNeg(RHS)); | ||||
3150 | else | ||||
3151 | Elts[i] = RHS; | ||||
3152 | } | ||||
3153 | } | ||||
3154 | |||||
3155 | Constant *NewRHSV = ConstantVector::get(Elts); | ||||
3156 | if (NewRHSV != RHSV) { | ||||
Nick Lewycky | 338ecd5 | 2008-12-18 06:42:28 +0000 | [diff] [blame] | 3157 | AddUsesToWorkList(I); |
Nick Lewycky | da9fa43 | 2008-12-18 06:31:11 +0000 | [diff] [blame] | 3158 | I.setOperand(1, NewRHSV); |
3159 | return &I; | ||||
3160 | } | ||||
3161 | } | ||||
3162 | } | ||||
3163 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3164 | return 0; |
3165 | } | ||||
3166 | |||||
3167 | Instruction *InstCombiner::visitFRem(BinaryOperator &I) { | ||||
3168 | return commonRemTransforms(I); | ||||
3169 | } | ||||
3170 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3171 | // isOneBitSet - Return true if there is exactly one bit set in the specified |
3172 | // constant. | ||||
3173 | static bool isOneBitSet(const ConstantInt *CI) { | ||||
3174 | return CI->getValue().isPowerOf2(); | ||||
3175 | } | ||||
3176 | |||||
3177 | // isHighOnes - Return true if the constant is of the form 1+0+. | ||||
3178 | // This is the same as lowones(~X). | ||||
3179 | static bool isHighOnes(const ConstantInt *CI) { | ||||
3180 | return (~CI->getValue() + 1).isPowerOf2(); | ||||
3181 | } | ||||
3182 | |||||
3183 | /// getICmpCode - Encode a icmp predicate into a three bit mask. These bits | ||||
3184 | /// are carefully arranged to allow folding of expressions such as: | ||||
3185 | /// | ||||
3186 | /// (A < B) | (A > B) --> (A != B) | ||||
3187 | /// | ||||
3188 | /// Note that this is only valid if the first and second predicates have the | ||||
3189 | /// same sign. Is illegal to do: (A u< B) | (A s> B) | ||||
3190 | /// | ||||
3191 | /// Three bits are used to represent the condition, as follows: | ||||
3192 | /// 0 A > B | ||||
3193 | /// 1 A == B | ||||
3194 | /// 2 A < B | ||||
3195 | /// | ||||
3196 | /// <=> Value Definition | ||||
3197 | /// 000 0 Always false | ||||
3198 | /// 001 1 A > B | ||||
3199 | /// 010 2 A == B | ||||
3200 | /// 011 3 A >= B | ||||
3201 | /// 100 4 A < B | ||||
3202 | /// 101 5 A != B | ||||
3203 | /// 110 6 A <= B | ||||
3204 | /// 111 7 Always true | ||||
3205 | /// | ||||
3206 | static unsigned getICmpCode(const ICmpInst *ICI) { | ||||
3207 | switch (ICI->getPredicate()) { | ||||
3208 | // False -> 0 | ||||
3209 | case ICmpInst::ICMP_UGT: return 1; // 001 | ||||
3210 | case ICmpInst::ICMP_SGT: return 1; // 001 | ||||
3211 | case ICmpInst::ICMP_EQ: return 2; // 010 | ||||
3212 | case ICmpInst::ICMP_UGE: return 3; // 011 | ||||
3213 | case ICmpInst::ICMP_SGE: return 3; // 011 | ||||
3214 | case ICmpInst::ICMP_ULT: return 4; // 100 | ||||
3215 | case ICmpInst::ICMP_SLT: return 4; // 100 | ||||
3216 | case ICmpInst::ICMP_NE: return 5; // 101 | ||||
3217 | case ICmpInst::ICMP_ULE: return 6; // 110 | ||||
3218 | case ICmpInst::ICMP_SLE: return 6; // 110 | ||||
3219 | // True -> 7 | ||||
3220 | default: | ||||
3221 | assert(0 && "Invalid ICmp predicate!"); | ||||
3222 | return 0; | ||||
3223 | } | ||||
3224 | } | ||||
3225 | |||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3226 | /// getFCmpCode - Similar to getICmpCode but for FCmpInst. This encodes a fcmp |
3227 | /// predicate into a three bit mask. It also returns whether it is an ordered | ||||
3228 | /// predicate by reference. | ||||
3229 | static unsigned getFCmpCode(FCmpInst::Predicate CC, bool &isOrdered) { | ||||
3230 | isOrdered = false; | ||||
3231 | switch (CC) { | ||||
3232 | case FCmpInst::FCMP_ORD: isOrdered = true; return 0; // 000 | ||||
3233 | case FCmpInst::FCMP_UNO: return 0; // 000 | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 3234 | case FCmpInst::FCMP_OGT: isOrdered = true; return 1; // 001 |
3235 | case FCmpInst::FCMP_UGT: return 1; // 001 | ||||
3236 | case FCmpInst::FCMP_OEQ: isOrdered = true; return 2; // 010 | ||||
3237 | case FCmpInst::FCMP_UEQ: return 2; // 010 | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3238 | case FCmpInst::FCMP_OGE: isOrdered = true; return 3; // 011 |
3239 | case FCmpInst::FCMP_UGE: return 3; // 011 | ||||
3240 | case FCmpInst::FCMP_OLT: isOrdered = true; return 4; // 100 | ||||
3241 | case FCmpInst::FCMP_ULT: return 4; // 100 | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 3242 | case FCmpInst::FCMP_ONE: isOrdered = true; return 5; // 101 |
3243 | case FCmpInst::FCMP_UNE: return 5; // 101 | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3244 | case FCmpInst::FCMP_OLE: isOrdered = true; return 6; // 110 |
3245 | case FCmpInst::FCMP_ULE: return 6; // 110 | ||||
Evan Cheng | 7298805 | 2008-10-14 18:44:08 +0000 | [diff] [blame] | 3246 | // True -> 7 |
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3247 | default: |
3248 | // Not expecting FCMP_FALSE and FCMP_TRUE; | ||||
3249 | assert(0 && "Unexpected FCmp predicate!"); | ||||
3250 | return 0; | ||||
3251 | } | ||||
3252 | } | ||||
3253 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3254 | /// getICmpValue - This is the complement of getICmpCode, which turns an |
3255 | /// 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] | 3256 | /// new ICmp instruction. The sign is passed in to determine which kind |
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3257 | /// of predicate to use in the new icmp instruction. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3258 | static Value *getICmpValue(bool sign, unsigned code, Value *LHS, Value *RHS) { |
3259 | switch (code) { | ||||
3260 | default: assert(0 && "Illegal ICmp code!"); | ||||
3261 | case 0: return ConstantInt::getFalse(); | ||||
3262 | case 1: | ||||
3263 | if (sign) | ||||
3264 | return new ICmpInst(ICmpInst::ICMP_SGT, LHS, RHS); | ||||
3265 | else | ||||
3266 | return new ICmpInst(ICmpInst::ICMP_UGT, LHS, RHS); | ||||
3267 | case 2: return new ICmpInst(ICmpInst::ICMP_EQ, LHS, RHS); | ||||
3268 | case 3: | ||||
3269 | if (sign) | ||||
3270 | return new ICmpInst(ICmpInst::ICMP_SGE, LHS, RHS); | ||||
3271 | else | ||||
3272 | return new ICmpInst(ICmpInst::ICMP_UGE, LHS, RHS); | ||||
3273 | case 4: | ||||
3274 | if (sign) | ||||
3275 | return new ICmpInst(ICmpInst::ICMP_SLT, LHS, RHS); | ||||
3276 | else | ||||
3277 | return new ICmpInst(ICmpInst::ICMP_ULT, LHS, RHS); | ||||
3278 | case 5: return new ICmpInst(ICmpInst::ICMP_NE, LHS, RHS); | ||||
3279 | case 6: | ||||
3280 | if (sign) | ||||
3281 | return new ICmpInst(ICmpInst::ICMP_SLE, LHS, RHS); | ||||
3282 | else | ||||
3283 | return new ICmpInst(ICmpInst::ICMP_ULE, LHS, RHS); | ||||
3284 | case 7: return ConstantInt::getTrue(); | ||||
3285 | } | ||||
3286 | } | ||||
3287 | |||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3288 | /// getFCmpValue - This is the complement of getFCmpCode, which turns an |
3289 | /// opcode and two operands into either a FCmp instruction. isordered is passed | ||||
3290 | /// in to determine which kind of predicate to use in the new fcmp instruction. | ||||
3291 | static Value *getFCmpValue(bool isordered, unsigned code, | ||||
3292 | Value *LHS, Value *RHS) { | ||||
3293 | switch (code) { | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 3294 | default: assert(0 && "Illegal FCmp code!"); |
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3295 | case 0: |
3296 | if (isordered) | ||||
3297 | return new FCmpInst(FCmpInst::FCMP_ORD, LHS, RHS); | ||||
3298 | else | ||||
3299 | return new FCmpInst(FCmpInst::FCMP_UNO, LHS, RHS); | ||||
3300 | case 1: | ||||
3301 | if (isordered) | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3302 | return new FCmpInst(FCmpInst::FCMP_OGT, LHS, RHS); |
3303 | else | ||||
3304 | return new FCmpInst(FCmpInst::FCMP_UGT, LHS, RHS); | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 3305 | case 2: |
3306 | if (isordered) | ||||
3307 | return new FCmpInst(FCmpInst::FCMP_OEQ, LHS, RHS); | ||||
3308 | else | ||||
3309 | return new FCmpInst(FCmpInst::FCMP_UEQ, LHS, RHS); | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3310 | case 3: |
3311 | if (isordered) | ||||
3312 | return new FCmpInst(FCmpInst::FCMP_OGE, LHS, RHS); | ||||
3313 | else | ||||
3314 | return new FCmpInst(FCmpInst::FCMP_UGE, LHS, RHS); | ||||
3315 | case 4: | ||||
3316 | if (isordered) | ||||
3317 | return new FCmpInst(FCmpInst::FCMP_OLT, LHS, RHS); | ||||
3318 | else | ||||
3319 | return new FCmpInst(FCmpInst::FCMP_ULT, LHS, RHS); | ||||
3320 | case 5: | ||||
3321 | if (isordered) | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 3322 | return new FCmpInst(FCmpInst::FCMP_ONE, LHS, RHS); |
3323 | else | ||||
3324 | return new FCmpInst(FCmpInst::FCMP_UNE, LHS, RHS); | ||||
3325 | case 6: | ||||
3326 | if (isordered) | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3327 | return new FCmpInst(FCmpInst::FCMP_OLE, LHS, RHS); |
3328 | else | ||||
3329 | return new FCmpInst(FCmpInst::FCMP_ULE, LHS, RHS); | ||||
Evan Cheng | 7298805 | 2008-10-14 18:44:08 +0000 | [diff] [blame] | 3330 | case 7: return ConstantInt::getTrue(); |
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 3331 | } |
3332 | } | ||||
3333 | |||||
Chris Lattner | 2972b82 | 2008-11-16 04:55:20 +0000 | [diff] [blame] | 3334 | /// PredicatesFoldable - Return true if both predicates match sign or if at |
3335 | /// least one of them is an equality comparison (which is signless). | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3336 | static bool PredicatesFoldable(ICmpInst::Predicate p1, ICmpInst::Predicate p2) { |
3337 | return (ICmpInst::isSignedPredicate(p1) == ICmpInst::isSignedPredicate(p2)) || | ||||
Chris Lattner | 2972b82 | 2008-11-16 04:55:20 +0000 | [diff] [blame] | 3338 | (ICmpInst::isSignedPredicate(p1) && ICmpInst::isEquality(p2)) || |
3339 | (ICmpInst::isSignedPredicate(p2) && ICmpInst::isEquality(p1)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3340 | } |
3341 | |||||
3342 | namespace { | ||||
3343 | // FoldICmpLogical - Implements (icmp1 A, B) & (icmp2 A, B) --> (icmp3 A, B) | ||||
3344 | struct FoldICmpLogical { | ||||
3345 | InstCombiner &IC; | ||||
3346 | Value *LHS, *RHS; | ||||
3347 | ICmpInst::Predicate pred; | ||||
3348 | FoldICmpLogical(InstCombiner &ic, ICmpInst *ICI) | ||||
3349 | : IC(ic), LHS(ICI->getOperand(0)), RHS(ICI->getOperand(1)), | ||||
3350 | pred(ICI->getPredicate()) {} | ||||
3351 | bool shouldApply(Value *V) const { | ||||
3352 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(V)) | ||||
3353 | if (PredicatesFoldable(pred, ICI->getPredicate())) | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 3354 | return ((ICI->getOperand(0) == LHS && ICI->getOperand(1) == RHS) || |
3355 | (ICI->getOperand(0) == RHS && ICI->getOperand(1) == LHS)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3356 | return false; |
3357 | } | ||||
3358 | Instruction *apply(Instruction &Log) const { | ||||
3359 | ICmpInst *ICI = cast<ICmpInst>(Log.getOperand(0)); | ||||
3360 | if (ICI->getOperand(0) != LHS) { | ||||
3361 | assert(ICI->getOperand(1) == LHS); | ||||
3362 | ICI->swapOperands(); // Swap the LHS and RHS of the ICmp | ||||
3363 | } | ||||
3364 | |||||
3365 | ICmpInst *RHSICI = cast<ICmpInst>(Log.getOperand(1)); | ||||
3366 | unsigned LHSCode = getICmpCode(ICI); | ||||
3367 | unsigned RHSCode = getICmpCode(RHSICI); | ||||
3368 | unsigned Code; | ||||
3369 | switch (Log.getOpcode()) { | ||||
3370 | case Instruction::And: Code = LHSCode & RHSCode; break; | ||||
3371 | case Instruction::Or: Code = LHSCode | RHSCode; break; | ||||
3372 | case Instruction::Xor: Code = LHSCode ^ RHSCode; break; | ||||
3373 | default: assert(0 && "Illegal logical opcode!"); return 0; | ||||
3374 | } | ||||
3375 | |||||
3376 | bool isSigned = ICmpInst::isSignedPredicate(RHSICI->getPredicate()) || | ||||
3377 | ICmpInst::isSignedPredicate(ICI->getPredicate()); | ||||
3378 | |||||
3379 | Value *RV = getICmpValue(isSigned, Code, LHS, RHS); | ||||
3380 | if (Instruction *I = dyn_cast<Instruction>(RV)) | ||||
3381 | return I; | ||||
3382 | // Otherwise, it's a constant boolean value... | ||||
3383 | return IC.ReplaceInstUsesWith(Log, RV); | ||||
3384 | } | ||||
3385 | }; | ||||
3386 | } // end anonymous namespace | ||||
3387 | |||||
3388 | // OptAndOp - This handles expressions of the form ((val OP C1) & C2). Where | ||||
3389 | // the Op parameter is 'OP', OpRHS is 'C1', and AndRHS is 'C2'. Op is | ||||
3390 | // guaranteed to be a binary operator. | ||||
3391 | Instruction *InstCombiner::OptAndOp(Instruction *Op, | ||||
3392 | ConstantInt *OpRHS, | ||||
3393 | ConstantInt *AndRHS, | ||||
3394 | BinaryOperator &TheAnd) { | ||||
3395 | Value *X = Op->getOperand(0); | ||||
3396 | Constant *Together = 0; | ||||
3397 | if (!Op->isShift()) | ||||
3398 | Together = And(AndRHS, OpRHS); | ||||
3399 | |||||
3400 | switch (Op->getOpcode()) { | ||||
3401 | case Instruction::Xor: | ||||
3402 | if (Op->hasOneUse()) { | ||||
3403 | // (X ^ C1) & C2 --> (X & C2) ^ (C1&C2) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3404 | Instruction *And = BinaryOperator::CreateAnd(X, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3405 | InsertNewInstBefore(And, TheAnd); |
3406 | And->takeName(Op); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3407 | return BinaryOperator::CreateXor(And, Together); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3408 | } |
3409 | break; | ||||
3410 | case Instruction::Or: | ||||
3411 | if (Together == AndRHS) // (X | C) & C --> C | ||||
3412 | return ReplaceInstUsesWith(TheAnd, AndRHS); | ||||
3413 | |||||
3414 | if (Op->hasOneUse() && Together != OpRHS) { | ||||
3415 | // (X | C1) & C2 --> (X | (C1&C2)) & C2 | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3416 | Instruction *Or = BinaryOperator::CreateOr(X, Together); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3417 | InsertNewInstBefore(Or, TheAnd); |
3418 | Or->takeName(Op); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3419 | return BinaryOperator::CreateAnd(Or, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3420 | } |
3421 | break; | ||||
3422 | case Instruction::Add: | ||||
3423 | if (Op->hasOneUse()) { | ||||
3424 | // Adding a one to a single bit bit-field should be turned into an XOR | ||||
3425 | // of the bit. First thing to check is to see if this AND is with a | ||||
3426 | // single bit constant. | ||||
3427 | const APInt& AndRHSV = cast<ConstantInt>(AndRHS)->getValue(); | ||||
3428 | |||||
3429 | // If there is only one bit set... | ||||
3430 | if (isOneBitSet(cast<ConstantInt>(AndRHS))) { | ||||
3431 | // Ok, at this point, we know that we are masking the result of the | ||||
3432 | // ADD down to exactly one bit. If the constant we are adding has | ||||
3433 | // no bits set below this bit, then we can eliminate the ADD. | ||||
3434 | const APInt& AddRHS = cast<ConstantInt>(OpRHS)->getValue(); | ||||
3435 | |||||
3436 | // Check to see if any bits below the one bit set in AndRHSV are set. | ||||
3437 | if ((AddRHS & (AndRHSV-1)) == 0) { | ||||
3438 | // If not, the only thing that can effect the output of the AND is | ||||
3439 | // the bit specified by AndRHSV. If that bit is set, the effect of | ||||
3440 | // the XOR is to toggle the bit. If it is clear, then the ADD has | ||||
3441 | // no effect. | ||||
3442 | if ((AddRHS & AndRHSV) == 0) { // Bit is not set, noop | ||||
3443 | TheAnd.setOperand(0, X); | ||||
3444 | return &TheAnd; | ||||
3445 | } else { | ||||
3446 | // Pull the XOR out of the AND. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3447 | Instruction *NewAnd = BinaryOperator::CreateAnd(X, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3448 | InsertNewInstBefore(NewAnd, TheAnd); |
3449 | NewAnd->takeName(Op); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3450 | return BinaryOperator::CreateXor(NewAnd, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3451 | } |
3452 | } | ||||
3453 | } | ||||
3454 | } | ||||
3455 | break; | ||||
3456 | |||||
3457 | case Instruction::Shl: { | ||||
3458 | // We know that the AND will not produce any of the bits shifted in, so if | ||||
3459 | // the anded constant includes them, clear them now! | ||||
3460 | // | ||||
3461 | uint32_t BitWidth = AndRHS->getType()->getBitWidth(); | ||||
3462 | uint32_t OpRHSVal = OpRHS->getLimitedValue(BitWidth); | ||||
3463 | APInt ShlMask(APInt::getHighBitsSet(BitWidth, BitWidth-OpRHSVal)); | ||||
3464 | ConstantInt *CI = ConstantInt::get(AndRHS->getValue() & ShlMask); | ||||
3465 | |||||
3466 | if (CI->getValue() == ShlMask) { | ||||
3467 | // Masking out bits that the shift already masks | ||||
3468 | return ReplaceInstUsesWith(TheAnd, Op); // No need for the and. | ||||
3469 | } else if (CI != AndRHS) { // Reducing bits set in and. | ||||
3470 | TheAnd.setOperand(1, CI); | ||||
3471 | return &TheAnd; | ||||
3472 | } | ||||
3473 | break; | ||||
3474 | } | ||||
3475 | case Instruction::LShr: | ||||
3476 | { | ||||
3477 | // We know that the AND will not produce any of the bits shifted in, so if | ||||
3478 | // the anded constant includes them, clear them now! This only applies to | ||||
3479 | // unsigned shifts, because a signed shr may bring in set bits! | ||||
3480 | // | ||||
3481 | uint32_t BitWidth = AndRHS->getType()->getBitWidth(); | ||||
3482 | uint32_t OpRHSVal = OpRHS->getLimitedValue(BitWidth); | ||||
3483 | APInt ShrMask(APInt::getLowBitsSet(BitWidth, BitWidth - OpRHSVal)); | ||||
3484 | ConstantInt *CI = ConstantInt::get(AndRHS->getValue() & ShrMask); | ||||
3485 | |||||
3486 | if (CI->getValue() == ShrMask) { | ||||
3487 | // Masking out bits that the shift already masks. | ||||
3488 | return ReplaceInstUsesWith(TheAnd, Op); | ||||
3489 | } else if (CI != AndRHS) { | ||||
3490 | TheAnd.setOperand(1, CI); // Reduce bits set in and cst. | ||||
3491 | return &TheAnd; | ||||
3492 | } | ||||
3493 | break; | ||||
3494 | } | ||||
3495 | case Instruction::AShr: | ||||
3496 | // Signed shr. | ||||
3497 | // See if this is shifting in some sign extension, then masking it out | ||||
3498 | // with an and. | ||||
3499 | if (Op->hasOneUse()) { | ||||
3500 | uint32_t BitWidth = AndRHS->getType()->getBitWidth(); | ||||
3501 | uint32_t OpRHSVal = OpRHS->getLimitedValue(BitWidth); | ||||
3502 | APInt ShrMask(APInt::getLowBitsSet(BitWidth, BitWidth - OpRHSVal)); | ||||
3503 | Constant *C = ConstantInt::get(AndRHS->getValue() & ShrMask); | ||||
3504 | if (C == AndRHS) { // Masking out bits shifted in. | ||||
3505 | // (Val ashr C1) & C2 -> (Val lshr C1) & C2 | ||||
3506 | // Make the argument unsigned. | ||||
3507 | Value *ShVal = Op->getOperand(0); | ||||
3508 | ShVal = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3509 | BinaryOperator::CreateLShr(ShVal, OpRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3510 | Op->getName()), TheAnd); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3511 | return BinaryOperator::CreateAnd(ShVal, AndRHS, TheAnd.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3512 | } |
3513 | } | ||||
3514 | break; | ||||
3515 | } | ||||
3516 | return 0; | ||||
3517 | } | ||||
3518 | |||||
3519 | |||||
3520 | /// InsertRangeTest - Emit a computation of: (V >= Lo && V < Hi) if Inside is | ||||
3521 | /// true, otherwise (V < Lo || V >= Hi). In pratice, we emit the more efficient | ||||
3522 | /// (V-Lo) <u Hi-Lo. This method expects that Lo <= Hi. isSigned indicates | ||||
3523 | /// whether to treat the V, Lo and HI as signed or not. IB is the location to | ||||
3524 | /// insert new instructions. | ||||
3525 | Instruction *InstCombiner::InsertRangeTest(Value *V, Constant *Lo, Constant *Hi, | ||||
3526 | bool isSigned, bool Inside, | ||||
3527 | Instruction &IB) { | ||||
3528 | assert(cast<ConstantInt>(ConstantExpr::getICmp((isSigned ? | ||||
3529 | ICmpInst::ICMP_SLE:ICmpInst::ICMP_ULE), Lo, Hi))->getZExtValue() && | ||||
3530 | "Lo is not <= Hi in range emission code!"); | ||||
3531 | |||||
3532 | if (Inside) { | ||||
3533 | if (Lo == Hi) // Trivially false. | ||||
3534 | return new ICmpInst(ICmpInst::ICMP_NE, V, V); | ||||
3535 | |||||
3536 | // V >= Min && V < Hi --> V < Hi | ||||
3537 | if (cast<ConstantInt>(Lo)->isMinValue(isSigned)) { | ||||
3538 | ICmpInst::Predicate pred = (isSigned ? | ||||
3539 | ICmpInst::ICMP_SLT : ICmpInst::ICMP_ULT); | ||||
3540 | return new ICmpInst(pred, V, Hi); | ||||
3541 | } | ||||
3542 | |||||
3543 | // Emit V-Lo <u Hi-Lo | ||||
3544 | Constant *NegLo = ConstantExpr::getNeg(Lo); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3545 | Instruction *Add = BinaryOperator::CreateAdd(V, NegLo, V->getName()+".off"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3546 | InsertNewInstBefore(Add, IB); |
3547 | Constant *UpperBound = ConstantExpr::getAdd(NegLo, Hi); | ||||
3548 | return new ICmpInst(ICmpInst::ICMP_ULT, Add, UpperBound); | ||||
3549 | } | ||||
3550 | |||||
3551 | if (Lo == Hi) // Trivially true. | ||||
3552 | return new ICmpInst(ICmpInst::ICMP_EQ, V, V); | ||||
3553 | |||||
3554 | // V < Min || V >= Hi -> V > Hi-1 | ||||
3555 | Hi = SubOne(cast<ConstantInt>(Hi)); | ||||
3556 | if (cast<ConstantInt>(Lo)->isMinValue(isSigned)) { | ||||
3557 | ICmpInst::Predicate pred = (isSigned ? | ||||
3558 | ICmpInst::ICMP_SGT : ICmpInst::ICMP_UGT); | ||||
3559 | return new ICmpInst(pred, V, Hi); | ||||
3560 | } | ||||
3561 | |||||
3562 | // Emit V-Lo >u Hi-1-Lo | ||||
3563 | // Note that Hi has already had one subtracted from it, above. | ||||
3564 | ConstantInt *NegLo = cast<ConstantInt>(ConstantExpr::getNeg(Lo)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3565 | Instruction *Add = BinaryOperator::CreateAdd(V, NegLo, V->getName()+".off"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3566 | InsertNewInstBefore(Add, IB); |
3567 | Constant *LowerBound = ConstantExpr::getAdd(NegLo, Hi); | ||||
3568 | return new ICmpInst(ICmpInst::ICMP_UGT, Add, LowerBound); | ||||
3569 | } | ||||
3570 | |||||
3571 | // isRunOfOnes - Returns true iff Val consists of one contiguous run of 1s with | ||||
3572 | // any number of 0s on either side. The 1s are allowed to wrap from LSB to | ||||
3573 | // MSB, so 0x000FFF0, 0x0000FFFF, and 0xFF0000FF are all runs. 0x0F0F0000 is | ||||
3574 | // not, since all 1s are not contiguous. | ||||
3575 | static bool isRunOfOnes(ConstantInt *Val, uint32_t &MB, uint32_t &ME) { | ||||
3576 | const APInt& V = Val->getValue(); | ||||
3577 | uint32_t BitWidth = Val->getType()->getBitWidth(); | ||||
3578 | if (!APIntOps::isShiftedMask(BitWidth, V)) return false; | ||||
3579 | |||||
3580 | // look for the first zero bit after the run of ones | ||||
3581 | MB = BitWidth - ((V - 1) ^ V).countLeadingZeros(); | ||||
3582 | // look for the first non-zero bit | ||||
3583 | ME = V.getActiveBits(); | ||||
3584 | return true; | ||||
3585 | } | ||||
3586 | |||||
3587 | /// FoldLogicalPlusAnd - This is part of an expression (LHS +/- RHS) & Mask, | ||||
3588 | /// where isSub determines whether the operator is a sub. If we can fold one of | ||||
3589 | /// the following xforms: | ||||
3590 | /// | ||||
3591 | /// ((A & N) +/- B) & Mask -> (A +/- B) & Mask iff N&Mask == Mask | ||||
3592 | /// ((A | N) +/- B) & Mask -> (A +/- B) & Mask iff N&Mask == 0 | ||||
3593 | /// ((A ^ N) +/- B) & Mask -> (A +/- B) & Mask iff N&Mask == 0 | ||||
3594 | /// | ||||
3595 | /// return (A +/- B). | ||||
3596 | /// | ||||
3597 | Value *InstCombiner::FoldLogicalPlusAnd(Value *LHS, Value *RHS, | ||||
3598 | ConstantInt *Mask, bool isSub, | ||||
3599 | Instruction &I) { | ||||
3600 | Instruction *LHSI = dyn_cast<Instruction>(LHS); | ||||
3601 | if (!LHSI || LHSI->getNumOperands() != 2 || | ||||
3602 | !isa<ConstantInt>(LHSI->getOperand(1))) return 0; | ||||
3603 | |||||
3604 | ConstantInt *N = cast<ConstantInt>(LHSI->getOperand(1)); | ||||
3605 | |||||
3606 | switch (LHSI->getOpcode()) { | ||||
3607 | default: return 0; | ||||
3608 | case Instruction::And: | ||||
3609 | if (And(N, Mask) == Mask) { | ||||
3610 | // If the AndRHS is a power of two minus one (0+1+), this is simple. | ||||
3611 | if ((Mask->getValue().countLeadingZeros() + | ||||
3612 | Mask->getValue().countPopulation()) == | ||||
3613 | Mask->getValue().getBitWidth()) | ||||
3614 | break; | ||||
3615 | |||||
3616 | // Otherwise, if Mask is 0+1+0+, and if B is known to have the low 0+ | ||||
3617 | // part, we don't need any explicit masks to take them out of A. If that | ||||
3618 | // is all N is, ignore it. | ||||
3619 | uint32_t MB = 0, ME = 0; | ||||
3620 | if (isRunOfOnes(Mask, MB, ME)) { // begin/end bit of run, inclusive | ||||
3621 | uint32_t BitWidth = cast<IntegerType>(RHS->getType())->getBitWidth(); | ||||
3622 | APInt Mask(APInt::getLowBitsSet(BitWidth, MB-1)); | ||||
3623 | if (MaskedValueIsZero(RHS, Mask)) | ||||
3624 | break; | ||||
3625 | } | ||||
3626 | } | ||||
3627 | return 0; | ||||
3628 | case Instruction::Or: | ||||
3629 | case Instruction::Xor: | ||||
3630 | // If the AndRHS is a power of two minus one (0+1+), and N&Mask == 0 | ||||
3631 | if ((Mask->getValue().countLeadingZeros() + | ||||
3632 | Mask->getValue().countPopulation()) == Mask->getValue().getBitWidth() | ||||
3633 | && And(N, Mask)->isZero()) | ||||
3634 | break; | ||||
3635 | return 0; | ||||
3636 | } | ||||
3637 | |||||
3638 | Instruction *New; | ||||
3639 | if (isSub) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3640 | New = BinaryOperator::CreateSub(LHSI->getOperand(0), RHS, "fold"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3641 | else |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3642 | New = BinaryOperator::CreateAdd(LHSI->getOperand(0), RHS, "fold"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3643 | return InsertNewInstBefore(New, I); |
3644 | } | ||||
3645 | |||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3646 | /// FoldAndOfICmps - Fold (icmp)&(icmp) if possible. |
3647 | Instruction *InstCombiner::FoldAndOfICmps(Instruction &I, | ||||
3648 | ICmpInst *LHS, ICmpInst *RHS) { | ||||
Chris Lattner | f380348 | 2008-11-16 05:10:52 +0000 | [diff] [blame] | 3649 | Value *Val, *Val2; |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3650 | ConstantInt *LHSCst, *RHSCst; |
3651 | ICmpInst::Predicate LHSCC, RHSCC; | ||||
3652 | |||||
Chris Lattner | f380348 | 2008-11-16 05:10:52 +0000 | [diff] [blame] | 3653 | // This only handles icmp of constants: (icmp1 A, C1) & (icmp2 B, C2). |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3654 | if (!match(LHS, m_ICmp(LHSCC, m_Value(Val), m_ConstantInt(LHSCst))) || |
Chris Lattner | f380348 | 2008-11-16 05:10:52 +0000 | [diff] [blame] | 3655 | !match(RHS, m_ICmp(RHSCC, m_Value(Val2), m_ConstantInt(RHSCst)))) |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3656 | return 0; |
Chris Lattner | f380348 | 2008-11-16 05:10:52 +0000 | [diff] [blame] | 3657 | |
3658 | // (icmp ult A, C) & (icmp ult B, C) --> (icmp ult (A|B), C) | ||||
3659 | // where C is a power of 2 | ||||
3660 | if (LHSCst == RHSCst && LHSCC == RHSCC && LHSCC == ICmpInst::ICMP_ULT && | ||||
3661 | LHSCst->getValue().isPowerOf2()) { | ||||
3662 | Instruction *NewOr = BinaryOperator::CreateOr(Val, Val2); | ||||
3663 | InsertNewInstBefore(NewOr, I); | ||||
3664 | return new ICmpInst(LHSCC, NewOr, LHSCst); | ||||
3665 | } | ||||
3666 | |||||
3667 | // From here on, we only handle: | ||||
3668 | // (icmp1 A, C1) & (icmp2 A, C2) --> something simpler. | ||||
3669 | if (Val != Val2) return 0; | ||||
3670 | |||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3671 | // ICMP_[US][GL]E X, CST is folded to ICMP_[US][GL]T elsewhere. |
3672 | if (LHSCC == ICmpInst::ICMP_UGE || LHSCC == ICmpInst::ICMP_ULE || | ||||
3673 | RHSCC == ICmpInst::ICMP_UGE || RHSCC == ICmpInst::ICMP_ULE || | ||||
3674 | LHSCC == ICmpInst::ICMP_SGE || LHSCC == ICmpInst::ICMP_SLE || | ||||
3675 | RHSCC == ICmpInst::ICMP_SGE || RHSCC == ICmpInst::ICMP_SLE) | ||||
3676 | return 0; | ||||
3677 | |||||
3678 | // We can't fold (ugt x, C) & (sgt x, C2). | ||||
3679 | if (!PredicatesFoldable(LHSCC, RHSCC)) | ||||
3680 | return 0; | ||||
3681 | |||||
3682 | // Ensure that the larger constant is on the RHS. | ||||
Chris Lattner | 665298f | 2008-11-16 05:14:43 +0000 | [diff] [blame] | 3683 | bool ShouldSwap; |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3684 | if (ICmpInst::isSignedPredicate(LHSCC) || |
3685 | (ICmpInst::isEquality(LHSCC) && | ||||
3686 | ICmpInst::isSignedPredicate(RHSCC))) | ||||
Chris Lattner | 665298f | 2008-11-16 05:14:43 +0000 | [diff] [blame] | 3687 | ShouldSwap = LHSCst->getValue().sgt(RHSCst->getValue()); |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3688 | else |
Chris Lattner | 665298f | 2008-11-16 05:14:43 +0000 | [diff] [blame] | 3689 | ShouldSwap = LHSCst->getValue().ugt(RHSCst->getValue()); |
3690 | |||||
3691 | if (ShouldSwap) { | ||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3692 | std::swap(LHS, RHS); |
3693 | std::swap(LHSCst, RHSCst); | ||||
3694 | std::swap(LHSCC, RHSCC); | ||||
3695 | } | ||||
3696 | |||||
3697 | // At this point, we know we have have two icmp instructions | ||||
3698 | // comparing a value against two constants and and'ing the result | ||||
3699 | // together. Because of the above check, we know that we only have | ||||
3700 | // icmp eq, icmp ne, icmp [su]lt, and icmp [SU]gt here. We also know | ||||
3701 | // (from the FoldICmpLogical check above), that the two constants | ||||
3702 | // are not equal and that the larger constant is on the RHS | ||||
3703 | assert(LHSCst != RHSCst && "Compares not folded above?"); | ||||
3704 | |||||
3705 | switch (LHSCC) { | ||||
3706 | default: assert(0 && "Unknown integer condition code!"); | ||||
3707 | case ICmpInst::ICMP_EQ: | ||||
3708 | switch (RHSCC) { | ||||
3709 | default: assert(0 && "Unknown integer condition code!"); | ||||
3710 | case ICmpInst::ICMP_EQ: // (X == 13 & X == 15) -> false | ||||
3711 | case ICmpInst::ICMP_UGT: // (X == 13 & X > 15) -> false | ||||
3712 | case ICmpInst::ICMP_SGT: // (X == 13 & X > 15) -> false | ||||
3713 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
3714 | case ICmpInst::ICMP_NE: // (X == 13 & X != 15) -> X == 13 | ||||
3715 | case ICmpInst::ICMP_ULT: // (X == 13 & X < 15) -> X == 13 | ||||
3716 | case ICmpInst::ICMP_SLT: // (X == 13 & X < 15) -> X == 13 | ||||
3717 | return ReplaceInstUsesWith(I, LHS); | ||||
3718 | } | ||||
3719 | case ICmpInst::ICMP_NE: | ||||
3720 | switch (RHSCC) { | ||||
3721 | default: assert(0 && "Unknown integer condition code!"); | ||||
3722 | case ICmpInst::ICMP_ULT: | ||||
3723 | if (LHSCst == SubOne(RHSCst)) // (X != 13 & X u< 14) -> X < 13 | ||||
3724 | return new ICmpInst(ICmpInst::ICMP_ULT, Val, LHSCst); | ||||
3725 | break; // (X != 13 & X u< 15) -> no change | ||||
3726 | case ICmpInst::ICMP_SLT: | ||||
3727 | if (LHSCst == SubOne(RHSCst)) // (X != 13 & X s< 14) -> X < 13 | ||||
3728 | return new ICmpInst(ICmpInst::ICMP_SLT, Val, LHSCst); | ||||
3729 | break; // (X != 13 & X s< 15) -> no change | ||||
3730 | case ICmpInst::ICMP_EQ: // (X != 13 & X == 15) -> X == 15 | ||||
3731 | case ICmpInst::ICMP_UGT: // (X != 13 & X u> 15) -> X u> 15 | ||||
3732 | case ICmpInst::ICMP_SGT: // (X != 13 & X s> 15) -> X s> 15 | ||||
3733 | return ReplaceInstUsesWith(I, RHS); | ||||
3734 | case ICmpInst::ICMP_NE: | ||||
3735 | if (LHSCst == SubOne(RHSCst)){// (X != 13 & X != 14) -> X-13 >u 1 | ||||
3736 | Constant *AddCST = ConstantExpr::getNeg(LHSCst); | ||||
3737 | Instruction *Add = BinaryOperator::CreateAdd(Val, AddCST, | ||||
3738 | Val->getName()+".off"); | ||||
3739 | InsertNewInstBefore(Add, I); | ||||
3740 | return new ICmpInst(ICmpInst::ICMP_UGT, Add, | ||||
3741 | ConstantInt::get(Add->getType(), 1)); | ||||
3742 | } | ||||
3743 | break; // (X != 13 & X != 15) -> no change | ||||
3744 | } | ||||
3745 | break; | ||||
3746 | case ICmpInst::ICMP_ULT: | ||||
3747 | switch (RHSCC) { | ||||
3748 | default: assert(0 && "Unknown integer condition code!"); | ||||
3749 | case ICmpInst::ICMP_EQ: // (X u< 13 & X == 15) -> false | ||||
3750 | case ICmpInst::ICMP_UGT: // (X u< 13 & X u> 15) -> false | ||||
3751 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
3752 | case ICmpInst::ICMP_SGT: // (X u< 13 & X s> 15) -> no change | ||||
3753 | break; | ||||
3754 | case ICmpInst::ICMP_NE: // (X u< 13 & X != 15) -> X u< 13 | ||||
3755 | case ICmpInst::ICMP_ULT: // (X u< 13 & X u< 15) -> X u< 13 | ||||
3756 | return ReplaceInstUsesWith(I, LHS); | ||||
3757 | case ICmpInst::ICMP_SLT: // (X u< 13 & X s< 15) -> no change | ||||
3758 | break; | ||||
3759 | } | ||||
3760 | break; | ||||
3761 | case ICmpInst::ICMP_SLT: | ||||
3762 | switch (RHSCC) { | ||||
3763 | default: assert(0 && "Unknown integer condition code!"); | ||||
3764 | case ICmpInst::ICMP_EQ: // (X s< 13 & X == 15) -> false | ||||
3765 | case ICmpInst::ICMP_SGT: // (X s< 13 & X s> 15) -> false | ||||
3766 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
3767 | case ICmpInst::ICMP_UGT: // (X s< 13 & X u> 15) -> no change | ||||
3768 | break; | ||||
3769 | case ICmpInst::ICMP_NE: // (X s< 13 & X != 15) -> X < 13 | ||||
3770 | case ICmpInst::ICMP_SLT: // (X s< 13 & X s< 15) -> X < 13 | ||||
3771 | return ReplaceInstUsesWith(I, LHS); | ||||
3772 | case ICmpInst::ICMP_ULT: // (X s< 13 & X u< 15) -> no change | ||||
3773 | break; | ||||
3774 | } | ||||
3775 | break; | ||||
3776 | case ICmpInst::ICMP_UGT: | ||||
3777 | switch (RHSCC) { | ||||
3778 | default: assert(0 && "Unknown integer condition code!"); | ||||
3779 | case ICmpInst::ICMP_EQ: // (X u> 13 & X == 15) -> X == 15 | ||||
3780 | case ICmpInst::ICMP_UGT: // (X u> 13 & X u> 15) -> X u> 15 | ||||
3781 | return ReplaceInstUsesWith(I, RHS); | ||||
3782 | case ICmpInst::ICMP_SGT: // (X u> 13 & X s> 15) -> no change | ||||
3783 | break; | ||||
3784 | case ICmpInst::ICMP_NE: | ||||
3785 | if (RHSCst == AddOne(LHSCst)) // (X u> 13 & X != 14) -> X u> 14 | ||||
3786 | return new ICmpInst(LHSCC, Val, RHSCst); | ||||
3787 | break; // (X u> 13 & X != 15) -> no change | ||||
Chris Lattner | 0c678e5 | 2008-11-16 05:20:07 +0000 | [diff] [blame] | 3788 | case ICmpInst::ICMP_ULT: // (X u> 13 & X u< 15) -> (X-14) <u 1 |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3789 | return InsertRangeTest(Val, AddOne(LHSCst), RHSCst, false, true, I); |
3790 | case ICmpInst::ICMP_SLT: // (X u> 13 & X s< 15) -> no change | ||||
3791 | break; | ||||
3792 | } | ||||
3793 | break; | ||||
3794 | case ICmpInst::ICMP_SGT: | ||||
3795 | switch (RHSCC) { | ||||
3796 | default: assert(0 && "Unknown integer condition code!"); | ||||
3797 | case ICmpInst::ICMP_EQ: // (X s> 13 & X == 15) -> X == 15 | ||||
3798 | case ICmpInst::ICMP_SGT: // (X s> 13 & X s> 15) -> X s> 15 | ||||
3799 | return ReplaceInstUsesWith(I, RHS); | ||||
3800 | case ICmpInst::ICMP_UGT: // (X s> 13 & X u> 15) -> no change | ||||
3801 | break; | ||||
3802 | case ICmpInst::ICMP_NE: | ||||
3803 | if (RHSCst == AddOne(LHSCst)) // (X s> 13 & X != 14) -> X s> 14 | ||||
3804 | return new ICmpInst(LHSCC, Val, RHSCst); | ||||
3805 | break; // (X s> 13 & X != 15) -> no change | ||||
Chris Lattner | 0c678e5 | 2008-11-16 05:20:07 +0000 | [diff] [blame] | 3806 | case ICmpInst::ICMP_SLT: // (X s> 13 & X s< 15) -> (X-14) s< 1 |
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3807 | return InsertRangeTest(Val, AddOne(LHSCst), RHSCst, true, true, I); |
3808 | case ICmpInst::ICMP_ULT: // (X s> 13 & X u< 15) -> no change | ||||
3809 | break; | ||||
3810 | } | ||||
3811 | break; | ||||
3812 | } | ||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 3813 | |
3814 | return 0; | ||||
3815 | } | ||||
3816 | |||||
3817 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3818 | Instruction *InstCombiner::visitAnd(BinaryOperator &I) { |
3819 | bool Changed = SimplifyCommutative(I); | ||||
3820 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
3821 | |||||
3822 | if (isa<UndefValue>(Op1)) // X & undef -> 0 | ||||
3823 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
3824 | |||||
3825 | // and X, X = X | ||||
3826 | if (Op0 == Op1) | ||||
3827 | return ReplaceInstUsesWith(I, Op1); | ||||
3828 | |||||
3829 | // See if we can simplify any instructions used by the instruction whose sole | ||||
3830 | // purpose is to compute bits we don't care about. | ||||
3831 | if (!isa<VectorType>(I.getType())) { | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 3832 | if (SimplifyDemandedInstructionBits(I)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3833 | return &I; |
3834 | } else { | ||||
3835 | if (ConstantVector *CP = dyn_cast<ConstantVector>(Op1)) { | ||||
3836 | if (CP->isAllOnesValue()) // X & <-1,-1> -> X | ||||
3837 | return ReplaceInstUsesWith(I, I.getOperand(0)); | ||||
3838 | } else if (isa<ConstantAggregateZero>(Op1)) { | ||||
3839 | return ReplaceInstUsesWith(I, Op1); // X & <0,0> -> <0,0> | ||||
3840 | } | ||||
3841 | } | ||||
3842 | |||||
3843 | if (ConstantInt *AndRHS = dyn_cast<ConstantInt>(Op1)) { | ||||
3844 | const APInt& AndRHSMask = AndRHS->getValue(); | ||||
3845 | APInt NotAndRHS(~AndRHSMask); | ||||
3846 | |||||
3847 | // Optimize a variety of ((val OP C1) & C2) combinations... | ||||
3848 | if (isa<BinaryOperator>(Op0)) { | ||||
3849 | Instruction *Op0I = cast<Instruction>(Op0); | ||||
3850 | Value *Op0LHS = Op0I->getOperand(0); | ||||
3851 | Value *Op0RHS = Op0I->getOperand(1); | ||||
3852 | switch (Op0I->getOpcode()) { | ||||
3853 | case Instruction::Xor: | ||||
3854 | case Instruction::Or: | ||||
3855 | // If the mask is only needed on one incoming arm, push it up. | ||||
3856 | if (Op0I->hasOneUse()) { | ||||
3857 | if (MaskedValueIsZero(Op0LHS, NotAndRHS)) { | ||||
3858 | // Not masking anything out for the LHS, move to RHS. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3859 | Instruction *NewRHS = BinaryOperator::CreateAnd(Op0RHS, AndRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3860 | Op0RHS->getName()+".masked"); |
3861 | InsertNewInstBefore(NewRHS, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3862 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3863 | cast<BinaryOperator>(Op0I)->getOpcode(), Op0LHS, NewRHS); |
3864 | } | ||||
3865 | if (!isa<Constant>(Op0RHS) && | ||||
3866 | MaskedValueIsZero(Op0RHS, NotAndRHS)) { | ||||
3867 | // Not masking anything out for the RHS, move to LHS. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3868 | Instruction *NewLHS = BinaryOperator::CreateAnd(Op0LHS, AndRHS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3869 | Op0LHS->getName()+".masked"); |
3870 | InsertNewInstBefore(NewLHS, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3871 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3872 | cast<BinaryOperator>(Op0I)->getOpcode(), NewLHS, Op0RHS); |
3873 | } | ||||
3874 | } | ||||
3875 | |||||
3876 | break; | ||||
3877 | case Instruction::Add: | ||||
3878 | // ((A & N) + B) & AndRHS -> (A + B) & AndRHS iff N&AndRHS == AndRHS. | ||||
3879 | // ((A | N) + B) & AndRHS -> (A + B) & AndRHS iff N&AndRHS == 0 | ||||
3880 | // ((A ^ N) + B) & AndRHS -> (A + B) & AndRHS iff N&AndRHS == 0 | ||||
3881 | if (Value *V = FoldLogicalPlusAnd(Op0LHS, Op0RHS, AndRHS, false, I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3882 | return BinaryOperator::CreateAnd(V, AndRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3883 | if (Value *V = FoldLogicalPlusAnd(Op0RHS, Op0LHS, AndRHS, false, I)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3884 | return BinaryOperator::CreateAnd(V, AndRHS); // Add commutes |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3885 | break; |
3886 | |||||
3887 | case Instruction::Sub: | ||||
3888 | // ((A & N) - B) & AndRHS -> (A - B) & AndRHS iff N&AndRHS == AndRHS. | ||||
3889 | // ((A | N) - B) & AndRHS -> (A - B) & AndRHS iff N&AndRHS == 0 | ||||
3890 | // ((A ^ N) - B) & AndRHS -> (A - B) & AndRHS iff N&AndRHS == 0 | ||||
3891 | if (Value *V = FoldLogicalPlusAnd(Op0LHS, Op0RHS, AndRHS, true, I)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3892 | return BinaryOperator::CreateAnd(V, AndRHS); |
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 3893 | |
Nick Lewycky | a349ba4 | 2008-07-10 05:51:40 +0000 | [diff] [blame] | 3894 | // (A - N) & AndRHS -> -N & AndRHS iff A&AndRHS==0 and AndRHS |
3895 | // has 1's for all bits that the subtraction with A might affect. | ||||
3896 | if (Op0I->hasOneUse()) { | ||||
3897 | uint32_t BitWidth = AndRHSMask.getBitWidth(); | ||||
3898 | uint32_t Zeros = AndRHSMask.countLeadingZeros(); | ||||
3899 | APInt Mask = APInt::getLowBitsSet(BitWidth, BitWidth - Zeros); | ||||
3900 | |||||
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 3901 | ConstantInt *A = dyn_cast<ConstantInt>(Op0LHS); |
Nick Lewycky | a349ba4 | 2008-07-10 05:51:40 +0000 | [diff] [blame] | 3902 | if (!(A && A->isZero()) && // avoid infinite recursion. |
3903 | MaskedValueIsZero(Op0LHS, Mask)) { | ||||
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 3904 | Instruction *NewNeg = BinaryOperator::CreateNeg(Op0RHS); |
3905 | InsertNewInstBefore(NewNeg, I); | ||||
3906 | return BinaryOperator::CreateAnd(NewNeg, AndRHS); | ||||
3907 | } | ||||
3908 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3909 | break; |
Nick Lewycky | 659ed4d | 2008-07-09 05:20:13 +0000 | [diff] [blame] | 3910 | |
3911 | case Instruction::Shl: | ||||
3912 | case Instruction::LShr: | ||||
3913 | // (1 << x) & 1 --> zext(x == 0) | ||||
3914 | // (1 >> x) & 1 --> zext(x == 0) | ||||
Nick Lewycky | f1b1222 | 2008-07-09 07:35:26 +0000 | [diff] [blame] | 3915 | if (AndRHSMask == 1 && Op0LHS == AndRHS) { |
Nick Lewycky | 659ed4d | 2008-07-09 05:20:13 +0000 | [diff] [blame] | 3916 | Instruction *NewICmp = new ICmpInst(ICmpInst::ICMP_EQ, Op0RHS, |
3917 | Constant::getNullValue(I.getType())); | ||||
3918 | InsertNewInstBefore(NewICmp, I); | ||||
3919 | return new ZExtInst(NewICmp, I.getType()); | ||||
3920 | } | ||||
3921 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3922 | } |
3923 | |||||
3924 | if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) | ||||
3925 | if (Instruction *Res = OptAndOp(Op0I, Op0CI, AndRHS, I)) | ||||
3926 | return Res; | ||||
3927 | } else if (CastInst *CI = dyn_cast<CastInst>(Op0)) { | ||||
3928 | // If this is an integer truncation or change from signed-to-unsigned, and | ||||
3929 | // if the source is an and/or with immediate, transform it. This | ||||
3930 | // frequently occurs for bitfield accesses. | ||||
3931 | if (Instruction *CastOp = dyn_cast<Instruction>(CI->getOperand(0))) { | ||||
3932 | if ((isa<TruncInst>(CI) || isa<BitCastInst>(CI)) && | ||||
3933 | CastOp->getNumOperands() == 2) | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 3934 | if (ConstantInt *AndCI = dyn_cast<ConstantInt>(CastOp->getOperand(1))) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3935 | if (CastOp->getOpcode() == Instruction::And) { |
3936 | // Change: and (cast (and X, C1) to T), C2 | ||||
3937 | // into : and (cast X to T), trunc_or_bitcast(C1)&C2 | ||||
3938 | // This will fold the two constants together, which may allow | ||||
3939 | // other simplifications. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3940 | Instruction *NewCast = CastInst::CreateTruncOrBitCast( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3941 | CastOp->getOperand(0), I.getType(), |
3942 | CastOp->getName()+".shrunk"); | ||||
3943 | NewCast = InsertNewInstBefore(NewCast, I); | ||||
3944 | // trunc_or_bitcast(C1)&C2 | ||||
3945 | Constant *C3 = ConstantExpr::getTruncOrBitCast(AndCI,I.getType()); | ||||
3946 | C3 = ConstantExpr::getAnd(C3, AndRHS); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3947 | return BinaryOperator::CreateAnd(NewCast, C3); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3948 | } else if (CastOp->getOpcode() == Instruction::Or) { |
3949 | // Change: and (cast (or X, C1) to T), C2 | ||||
3950 | // into : trunc(C1)&C2 iff trunc(C1)&C2 == C2 | ||||
3951 | Constant *C3 = ConstantExpr::getTruncOrBitCast(AndCI,I.getType()); | ||||
3952 | if (ConstantExpr::getAnd(C3, AndRHS) == AndRHS) // trunc(C1)&C2 | ||||
3953 | return ReplaceInstUsesWith(I, AndRHS); | ||||
3954 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 3955 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3956 | } |
3957 | } | ||||
3958 | |||||
3959 | // Try to fold constant and into select arguments. | ||||
3960 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
3961 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
3962 | return R; | ||||
3963 | if (isa<PHINode>(Op0)) | ||||
3964 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
3965 | return NV; | ||||
3966 | } | ||||
3967 | |||||
3968 | Value *Op0NotVal = dyn_castNotVal(Op0); | ||||
3969 | Value *Op1NotVal = dyn_castNotVal(Op1); | ||||
3970 | |||||
3971 | if (Op0NotVal == Op1 || Op1NotVal == Op0) // A & ~A == ~A & A == 0 | ||||
3972 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
3973 | |||||
3974 | // (~A & ~B) == (~(A | B)) - De Morgan's Law | ||||
3975 | if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3976 | Instruction *Or = BinaryOperator::CreateOr(Op0NotVal, Op1NotVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3977 | I.getName()+".demorgan"); |
3978 | InsertNewInstBefore(Or, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3979 | return BinaryOperator::CreateNot(Or); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3980 | } |
3981 | |||||
3982 | { | ||||
3983 | Value *A = 0, *B = 0, *C = 0, *D = 0; | ||||
3984 | if (match(Op0, m_Or(m_Value(A), m_Value(B)))) { | ||||
3985 | if (A == Op1 || B == Op1) // (A | ?) & A --> A | ||||
3986 | return ReplaceInstUsesWith(I, Op1); | ||||
3987 | |||||
3988 | // (A|B) & ~(A&B) -> A^B | ||||
3989 | if (match(Op1, m_Not(m_And(m_Value(C), m_Value(D))))) { | ||||
3990 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 3991 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 3992 | } |
3993 | } | ||||
3994 | |||||
3995 | if (match(Op1, m_Or(m_Value(A), m_Value(B)))) { | ||||
3996 | if (A == Op0 || B == Op0) // A & (A | ?) --> A | ||||
3997 | return ReplaceInstUsesWith(I, Op0); | ||||
3998 | |||||
3999 | // ~(A&B) & (A|B) -> A^B | ||||
4000 | if (match(Op0, m_Not(m_And(m_Value(C), m_Value(D))))) { | ||||
4001 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4002 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4003 | } |
4004 | } | ||||
4005 | |||||
4006 | if (Op0->hasOneUse() && | ||||
4007 | match(Op0, m_Xor(m_Value(A), m_Value(B)))) { | ||||
4008 | if (A == Op1) { // (A^B)&A -> A&(A^B) | ||||
4009 | I.swapOperands(); // Simplify below | ||||
4010 | std::swap(Op0, Op1); | ||||
4011 | } else if (B == Op1) { // (A^B)&B -> B&(B^A) | ||||
4012 | cast<BinaryOperator>(Op0)->swapOperands(); | ||||
4013 | I.swapOperands(); // Simplify below | ||||
4014 | std::swap(Op0, Op1); | ||||
4015 | } | ||||
4016 | } | ||||
Bill Wendling | ce5e0af | 2008-11-30 13:08:13 +0000 | [diff] [blame] | 4017 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4018 | if (Op1->hasOneUse() && |
4019 | match(Op1, m_Xor(m_Value(A), m_Value(B)))) { | ||||
4020 | if (B == Op0) { // B&(A^B) -> B&(B^A) | ||||
4021 | cast<BinaryOperator>(Op1)->swapOperands(); | ||||
4022 | std::swap(A, B); | ||||
4023 | } | ||||
4024 | if (A == Op0) { // A&(A^B) -> A & ~B | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4025 | Instruction *NotB = BinaryOperator::CreateNot(B, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4026 | InsertNewInstBefore(NotB, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4027 | return BinaryOperator::CreateAnd(A, NotB); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4028 | } |
4029 | } | ||||
Bill Wendling | ce5e0af | 2008-11-30 13:08:13 +0000 | [diff] [blame] | 4030 | |
4031 | // (A&((~A)|B)) -> A&B | ||||
Chris Lattner | 9db479f | 2008-12-01 05:16:26 +0000 | [diff] [blame] | 4032 | if (match(Op0, m_Or(m_Not(m_Specific(Op1)), m_Value(A))) || |
4033 | match(Op0, m_Or(m_Value(A), m_Not(m_Specific(Op1))))) | ||||
4034 | return BinaryOperator::CreateAnd(A, Op1); | ||||
4035 | if (match(Op1, m_Or(m_Not(m_Specific(Op0)), m_Value(A))) || | ||||
4036 | match(Op1, m_Or(m_Value(A), m_Not(m_Specific(Op0))))) | ||||
4037 | return BinaryOperator::CreateAnd(A, Op0); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4038 | } |
4039 | |||||
4040 | if (ICmpInst *RHS = dyn_cast<ICmpInst>(Op1)) { | ||||
4041 | // (icmp1 A, B) & (icmp2 A, B) --> (icmp3 A, B) | ||||
4042 | if (Instruction *R = AssociativeOpt(I, FoldICmpLogical(*this, RHS))) | ||||
4043 | return R; | ||||
4044 | |||||
Chris Lattner | 0631ea7 | 2008-11-16 05:06:21 +0000 | [diff] [blame] | 4045 | if (ICmpInst *LHS = dyn_cast<ICmpInst>(Op0)) |
4046 | if (Instruction *Res = FoldAndOfICmps(I, LHS, RHS)) | ||||
4047 | return Res; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4048 | } |
4049 | |||||
4050 | // fold (and (cast A), (cast B)) -> (cast (and A, B)) | ||||
4051 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) | ||||
4052 | if (CastInst *Op1C = dyn_cast<CastInst>(Op1)) | ||||
4053 | if (Op0C->getOpcode() == Op1C->getOpcode()) { // same cast kind ? | ||||
4054 | const Type *SrcTy = Op0C->getOperand(0)->getType(); | ||||
4055 | if (SrcTy == Op1C->getOperand(0)->getType() && SrcTy->isInteger() && | ||||
4056 | // Only do this if the casts both really cause code to be generated. | ||||
4057 | ValueRequiresCast(Op0C->getOpcode(), Op0C->getOperand(0), | ||||
4058 | I.getType(), TD) && | ||||
4059 | ValueRequiresCast(Op1C->getOpcode(), Op1C->getOperand(0), | ||||
4060 | I.getType(), TD)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4061 | Instruction *NewOp = BinaryOperator::CreateAnd(Op0C->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4062 | Op1C->getOperand(0), |
4063 | I.getName()); | ||||
4064 | InsertNewInstBefore(NewOp, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4065 | return CastInst::Create(Op0C->getOpcode(), NewOp, I.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4066 | } |
4067 | } | ||||
4068 | |||||
4069 | // (X >> Z) & (Y >> Z) -> (X&Y) >> Z for all shifts. | ||||
4070 | if (BinaryOperator *SI1 = dyn_cast<BinaryOperator>(Op1)) { | ||||
4071 | if (BinaryOperator *SI0 = dyn_cast<BinaryOperator>(Op0)) | ||||
4072 | if (SI0->isShift() && SI0->getOpcode() == SI1->getOpcode() && | ||||
4073 | SI0->getOperand(1) == SI1->getOperand(1) && | ||||
4074 | (SI0->hasOneUse() || SI1->hasOneUse())) { | ||||
4075 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4076 | InsertNewInstBefore(BinaryOperator::CreateAnd(SI0->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4077 | SI1->getOperand(0), |
4078 | SI0->getName()), I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4079 | return BinaryOperator::Create(SI1->getOpcode(), NewOp, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4080 | SI1->getOperand(1)); |
4081 | } | ||||
4082 | } | ||||
4083 | |||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 4084 | // If and'ing two fcmp, try combine them into one. |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4085 | if (FCmpInst *LHS = dyn_cast<FCmpInst>(I.getOperand(0))) { |
4086 | if (FCmpInst *RHS = dyn_cast<FCmpInst>(I.getOperand(1))) { | ||||
4087 | if (LHS->getPredicate() == FCmpInst::FCMP_ORD && | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 4088 | RHS->getPredicate() == FCmpInst::FCMP_ORD) { |
4089 | // (fcmp ord x, c) & (fcmp ord y, c) -> (fcmp ord x, y) | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4090 | if (ConstantFP *LHSC = dyn_cast<ConstantFP>(LHS->getOperand(1))) |
4091 | if (ConstantFP *RHSC = dyn_cast<ConstantFP>(RHS->getOperand(1))) { | ||||
4092 | // If either of the constants are nans, then the whole thing returns | ||||
4093 | // false. | ||||
Chris Lattner | a6c7dce | 2007-10-24 18:54:45 +0000 | [diff] [blame] | 4094 | if (LHSC->getValueAPF().isNaN() || RHSC->getValueAPF().isNaN()) |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4095 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
4096 | return new FCmpInst(FCmpInst::FCMP_ORD, LHS->getOperand(0), | ||||
4097 | RHS->getOperand(0)); | ||||
4098 | } | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 4099 | } else { |
4100 | Value *Op0LHS, *Op0RHS, *Op1LHS, *Op1RHS; | ||||
4101 | FCmpInst::Predicate Op0CC, Op1CC; | ||||
4102 | if (match(Op0, m_FCmp(Op0CC, m_Value(Op0LHS), m_Value(Op0RHS))) && | ||||
4103 | match(Op1, m_FCmp(Op1CC, m_Value(Op1LHS), m_Value(Op1RHS)))) { | ||||
Evan Cheng | f1f2cea | 2008-10-14 18:13:38 +0000 | [diff] [blame] | 4104 | if (Op0LHS == Op1RHS && Op0RHS == Op1LHS) { |
4105 | // Swap RHS operands to match LHS. | ||||
4106 | Op1CC = FCmpInst::getSwappedPredicate(Op1CC); | ||||
4107 | std::swap(Op1LHS, Op1RHS); | ||||
4108 | } | ||||
Evan Cheng | 0ac3a4d | 2008-10-14 17:15:11 +0000 | [diff] [blame] | 4109 | if (Op0LHS == Op1LHS && Op0RHS == Op1RHS) { |
4110 | // Simplify (fcmp cc0 x, y) & (fcmp cc1 x, y). | ||||
4111 | if (Op0CC == Op1CC) | ||||
4112 | return new FCmpInst((FCmpInst::Predicate)Op0CC, Op0LHS, Op0RHS); | ||||
4113 | else if (Op0CC == FCmpInst::FCMP_FALSE || | ||||
4114 | Op1CC == FCmpInst::FCMP_FALSE) | ||||
4115 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
4116 | else if (Op0CC == FCmpInst::FCMP_TRUE) | ||||
4117 | return ReplaceInstUsesWith(I, Op1); | ||||
4118 | else if (Op1CC == FCmpInst::FCMP_TRUE) | ||||
4119 | return ReplaceInstUsesWith(I, Op0); | ||||
4120 | bool Op0Ordered; | ||||
4121 | bool Op1Ordered; | ||||
4122 | unsigned Op0Pred = getFCmpCode(Op0CC, Op0Ordered); | ||||
4123 | unsigned Op1Pred = getFCmpCode(Op1CC, Op1Ordered); | ||||
4124 | if (Op1Pred == 0) { | ||||
4125 | std::swap(Op0, Op1); | ||||
4126 | std::swap(Op0Pred, Op1Pred); | ||||
4127 | std::swap(Op0Ordered, Op1Ordered); | ||||
4128 | } | ||||
4129 | if (Op0Pred == 0) { | ||||
4130 | // uno && ueq -> uno && (uno || eq) -> ueq | ||||
4131 | // ord && olt -> ord && (ord && lt) -> olt | ||||
4132 | if (Op0Ordered == Op1Ordered) | ||||
4133 | return ReplaceInstUsesWith(I, Op1); | ||||
4134 | // uno && oeq -> uno && (ord && eq) -> false | ||||
4135 | // uno && ord -> false | ||||
4136 | if (!Op0Ordered) | ||||
4137 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
4138 | // ord && ueq -> ord && (uno || eq) -> oeq | ||||
4139 | return cast<Instruction>(getFCmpValue(true, Op1Pred, | ||||
4140 | Op0LHS, Op0RHS)); | ||||
4141 | } | ||||
4142 | } | ||||
4143 | } | ||||
4144 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4145 | } |
4146 | } | ||||
Nick Lewycky | ffed71b | 2008-07-09 04:32:37 +0000 | [diff] [blame] | 4147 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4148 | return Changed ? &I : 0; |
4149 | } | ||||
4150 | |||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4151 | /// CollectBSwapParts - Analyze the specified subexpression and see if it is |
4152 | /// capable of providing pieces of a bswap. The subexpression provides pieces | ||||
4153 | /// of a bswap if it is proven that each of the non-zero bytes in the output of | ||||
4154 | /// the expression came from the corresponding "byte swapped" byte in some other | ||||
4155 | /// value. For example, if the current subexpression is "(shl i32 %X, 24)" then | ||||
4156 | /// we know that the expression deposits the low byte of %X into the high byte | ||||
4157 | /// of the bswap result and that all other bytes are zero. This expression is | ||||
4158 | /// accepted, the high byte of ByteValues is set to X to indicate a correct | ||||
4159 | /// match. | ||||
4160 | /// | ||||
4161 | /// This function returns true if the match was unsuccessful and false if so. | ||||
4162 | /// On entry to the function the "OverallLeftShift" is a signed integer value | ||||
4163 | /// indicating the number of bytes that the subexpression is later shifted. For | ||||
4164 | /// example, if the expression is later right shifted by 16 bits, the | ||||
4165 | /// OverallLeftShift value would be -2 on entry. This is used to specify which | ||||
4166 | /// byte of ByteValues is actually being set. | ||||
4167 | /// | ||||
4168 | /// Similarly, ByteMask is a bitmask where a bit is clear if its corresponding | ||||
4169 | /// byte is masked to zero by a user. For example, in (X & 255), X will be | ||||
4170 | /// processed with a bytemask of 1. Because bytemask is 32-bits, this limits | ||||
4171 | /// this function to working on up to 32-byte (256 bit) values. ByteMask is | ||||
4172 | /// always in the local (OverallLeftShift) coordinate space. | ||||
4173 | /// | ||||
4174 | static bool CollectBSwapParts(Value *V, int OverallLeftShift, uint32_t ByteMask, | ||||
4175 | SmallVector<Value*, 8> &ByteValues) { | ||||
4176 | if (Instruction *I = dyn_cast<Instruction>(V)) { | ||||
4177 | // If this is an or instruction, it may be an inner node of the bswap. | ||||
4178 | if (I->getOpcode() == Instruction::Or) { | ||||
4179 | return CollectBSwapParts(I->getOperand(0), OverallLeftShift, ByteMask, | ||||
4180 | ByteValues) || | ||||
4181 | CollectBSwapParts(I->getOperand(1), OverallLeftShift, ByteMask, | ||||
4182 | ByteValues); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4183 | } |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4184 | |
4185 | // If this is a logical shift by a constant multiple of 8, recurse with | ||||
4186 | // OverallLeftShift and ByteMask adjusted. | ||||
4187 | if (I->isLogicalShift() && isa<ConstantInt>(I->getOperand(1))) { | ||||
4188 | unsigned ShAmt = | ||||
4189 | cast<ConstantInt>(I->getOperand(1))->getLimitedValue(~0U); | ||||
4190 | // Ensure the shift amount is defined and of a byte value. | ||||
4191 | if ((ShAmt & 7) || (ShAmt > 8*ByteValues.size())) | ||||
4192 | return true; | ||||
4193 | |||||
4194 | unsigned ByteShift = ShAmt >> 3; | ||||
4195 | if (I->getOpcode() == Instruction::Shl) { | ||||
4196 | // X << 2 -> collect(X, +2) | ||||
4197 | OverallLeftShift += ByteShift; | ||||
4198 | ByteMask >>= ByteShift; | ||||
4199 | } else { | ||||
4200 | // X >>u 2 -> collect(X, -2) | ||||
4201 | OverallLeftShift -= ByteShift; | ||||
4202 | ByteMask <<= ByteShift; | ||||
Chris Lattner | 4444859 | 2008-10-08 06:42:28 +0000 | [diff] [blame] | 4203 | ByteMask &= (~0U >> (32-ByteValues.size())); |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4204 | } |
4205 | |||||
4206 | if (OverallLeftShift >= (int)ByteValues.size()) return true; | ||||
4207 | if (OverallLeftShift <= -(int)ByteValues.size()) return true; | ||||
4208 | |||||
4209 | return CollectBSwapParts(I->getOperand(0), OverallLeftShift, ByteMask, | ||||
4210 | ByteValues); | ||||
4211 | } | ||||
4212 | |||||
4213 | // If this is a logical 'and' with a mask that clears bytes, clear the | ||||
4214 | // corresponding bytes in ByteMask. | ||||
4215 | if (I->getOpcode() == Instruction::And && | ||||
4216 | isa<ConstantInt>(I->getOperand(1))) { | ||||
4217 | // Scan every byte of the and mask, seeing if the byte is either 0 or 255. | ||||
4218 | unsigned NumBytes = ByteValues.size(); | ||||
4219 | APInt Byte(I->getType()->getPrimitiveSizeInBits(), 255); | ||||
4220 | const APInt &AndMask = cast<ConstantInt>(I->getOperand(1))->getValue(); | ||||
4221 | |||||
4222 | for (unsigned i = 0; i != NumBytes; ++i, Byte <<= 8) { | ||||
4223 | // If this byte is masked out by a later operation, we don't care what | ||||
4224 | // the and mask is. | ||||
4225 | if ((ByteMask & (1 << i)) == 0) | ||||
4226 | continue; | ||||
4227 | |||||
4228 | // If the AndMask is all zeros for this byte, clear the bit. | ||||
4229 | APInt MaskB = AndMask & Byte; | ||||
4230 | if (MaskB == 0) { | ||||
4231 | ByteMask &= ~(1U << i); | ||||
4232 | continue; | ||||
4233 | } | ||||
4234 | |||||
4235 | // If the AndMask is not all ones for this byte, it's not a bytezap. | ||||
4236 | if (MaskB != Byte) | ||||
4237 | return true; | ||||
4238 | |||||
4239 | // Otherwise, this byte is kept. | ||||
4240 | } | ||||
4241 | |||||
4242 | return CollectBSwapParts(I->getOperand(0), OverallLeftShift, ByteMask, | ||||
4243 | ByteValues); | ||||
4244 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4245 | } |
4246 | |||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4247 | // Okay, we got to something that isn't a shift, 'or' or 'and'. This must be |
4248 | // the input value to the bswap. Some observations: 1) if more than one byte | ||||
4249 | // is demanded from this input, then it could not be successfully assembled | ||||
4250 | // into a byteswap. At least one of the two bytes would not be aligned with | ||||
4251 | // their ultimate destination. | ||||
4252 | if (!isPowerOf2_32(ByteMask)) return true; | ||||
4253 | unsigned InputByteNo = CountTrailingZeros_32(ByteMask); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4254 | |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4255 | // 2) The input and ultimate destinations must line up: if byte 3 of an i32 |
4256 | // is demanded, it needs to go into byte 0 of the result. This means that the | ||||
4257 | // byte needs to be shifted until it lands in the right byte bucket. The | ||||
4258 | // shift amount depends on the position: if the byte is coming from the high | ||||
4259 | // part of the value (e.g. byte 3) then it must be shifted right. If from the | ||||
4260 | // low part, it must be shifted left. | ||||
4261 | unsigned DestByteNo = InputByteNo + OverallLeftShift; | ||||
4262 | if (InputByteNo < ByteValues.size()/2) { | ||||
4263 | if (ByteValues.size()-1-DestByteNo != InputByteNo) | ||||
4264 | return true; | ||||
4265 | } else { | ||||
4266 | if (ByteValues.size()-1-DestByteNo != InputByteNo) | ||||
4267 | return true; | ||||
4268 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4269 | |
4270 | // If the destination byte value is already defined, the values are or'd | ||||
4271 | // 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] | 4272 | if (ByteValues[DestByteNo] && ByteValues[DestByteNo] != V) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4273 | return true; |
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4274 | ByteValues[DestByteNo] = V; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4275 | return false; |
4276 | } | ||||
4277 | |||||
4278 | /// MatchBSwap - Given an OR instruction, check to see if this is a bswap idiom. | ||||
4279 | /// If so, insert the new bswap intrinsic and return it. | ||||
4280 | Instruction *InstCombiner::MatchBSwap(BinaryOperator &I) { | ||||
4281 | const IntegerType *ITy = dyn_cast<IntegerType>(I.getType()); | ||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4282 | if (!ITy || ITy->getBitWidth() % 16 || |
4283 | // ByteMask only allows up to 32-byte values. | ||||
4284 | ITy->getBitWidth() > 32*8) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4285 | return 0; // Can only bswap pairs of bytes. Can't do vectors. |
4286 | |||||
4287 | /// ByteValues - For each byte of the result, we keep track of which value | ||||
4288 | /// defines each byte. | ||||
4289 | SmallVector<Value*, 8> ByteValues; | ||||
4290 | ByteValues.resize(ITy->getBitWidth()/8); | ||||
4291 | |||||
4292 | // Try to find all the pieces corresponding to the bswap. | ||||
Chris Lattner | 567f511 | 2008-10-05 02:13:19 +0000 | [diff] [blame] | 4293 | uint32_t ByteMask = ~0U >> (32-ByteValues.size()); |
4294 | if (CollectBSwapParts(&I, 0, ByteMask, ByteValues)) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4295 | return 0; |
4296 | |||||
4297 | // Check to see if all of the bytes come from the same value. | ||||
4298 | Value *V = ByteValues[0]; | ||||
4299 | if (V == 0) return 0; // Didn't find a byte? Must be zero. | ||||
4300 | |||||
4301 | // Check to make sure that all of the bytes come from the same value. | ||||
4302 | for (unsigned i = 1, e = ByteValues.size(); i != e; ++i) | ||||
4303 | if (ByteValues[i] != V) | ||||
4304 | return 0; | ||||
Chandler Carruth | a228e39 | 2007-08-04 01:51:18 +0000 | [diff] [blame] | 4305 | const Type *Tys[] = { ITy }; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4306 | Module *M = I.getParent()->getParent()->getParent(); |
Chandler Carruth | a228e39 | 2007-08-04 01:51:18 +0000 | [diff] [blame] | 4307 | Function *F = Intrinsic::getDeclaration(M, Intrinsic::bswap, Tys, 1); |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 4308 | return CallInst::Create(F, V); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4309 | } |
4310 | |||||
Chris Lattner | dd7772b | 2008-11-16 04:24:12 +0000 | [diff] [blame] | 4311 | /// MatchSelectFromAndOr - We have an expression of the form (A&C)|(B&D). Check |
4312 | /// If A is (cond?-1:0) and either B or D is ~(cond?-1,0) or (cond?0,-1), then | ||||
4313 | /// we can simplify this expression to "cond ? C : D or B". | ||||
4314 | static Instruction *MatchSelectFromAndOr(Value *A, Value *B, | ||||
4315 | Value *C, Value *D) { | ||||
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4316 | // If A is not a select of -1/0, this cannot match. |
Chris Lattner | 641ea46 | 2008-11-16 04:46:19 +0000 | [diff] [blame] | 4317 | Value *Cond = 0; |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 4318 | if (!match(A, m_SelectCst<-1, 0>(m_Value(Cond)))) |
Chris Lattner | dd7772b | 2008-11-16 04:24:12 +0000 | [diff] [blame] | 4319 | return 0; |
4320 | |||||
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4321 | // ((cond?-1:0)&C) | (B&(cond?0:-1)) -> cond ? C : B. |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 4322 | if (match(D, m_SelectCst<0, -1>(m_Specific(Cond)))) |
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4323 | return SelectInst::Create(Cond, C, B); |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 4324 | if (match(D, m_Not(m_SelectCst<-1, 0>(m_Specific(Cond))))) |
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4325 | return SelectInst::Create(Cond, C, B); |
4326 | // ((cond?-1:0)&C) | ((cond?0:-1)&D) -> cond ? C : D. | ||||
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 4327 | if (match(B, m_SelectCst<0, -1>(m_Specific(Cond)))) |
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4328 | return SelectInst::Create(Cond, C, D); |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 4329 | if (match(B, m_Not(m_SelectCst<-1, 0>(m_Specific(Cond))))) |
Chris Lattner | d09b5ba | 2008-11-16 04:26:55 +0000 | [diff] [blame] | 4330 | return SelectInst::Create(Cond, C, D); |
Chris Lattner | dd7772b | 2008-11-16 04:24:12 +0000 | [diff] [blame] | 4331 | return 0; |
4332 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4333 | |
Chris Lattner | 0c678e5 | 2008-11-16 05:20:07 +0000 | [diff] [blame] | 4334 | /// FoldOrOfICmps - Fold (icmp)|(icmp) if possible. |
4335 | Instruction *InstCombiner::FoldOrOfICmps(Instruction &I, | ||||
4336 | ICmpInst *LHS, ICmpInst *RHS) { | ||||
4337 | Value *Val, *Val2; | ||||
4338 | ConstantInt *LHSCst, *RHSCst; | ||||
4339 | ICmpInst::Predicate LHSCC, RHSCC; | ||||
4340 | |||||
4341 | // This only handles icmp of constants: (icmp1 A, C1) | (icmp2 B, C2). | ||||
4342 | if (!match(LHS, m_ICmp(LHSCC, m_Value(Val), m_ConstantInt(LHSCst))) || | ||||
4343 | !match(RHS, m_ICmp(RHSCC, m_Value(Val2), m_ConstantInt(RHSCst)))) | ||||
4344 | return 0; | ||||
4345 | |||||
4346 | // From here on, we only handle: | ||||
4347 | // (icmp1 A, C1) | (icmp2 A, C2) --> something simpler. | ||||
4348 | if (Val != Val2) return 0; | ||||
4349 | |||||
4350 | // ICMP_[US][GL]E X, CST is folded to ICMP_[US][GL]T elsewhere. | ||||
4351 | if (LHSCC == ICmpInst::ICMP_UGE || LHSCC == ICmpInst::ICMP_ULE || | ||||
4352 | RHSCC == ICmpInst::ICMP_UGE || RHSCC == ICmpInst::ICMP_ULE || | ||||
4353 | LHSCC == ICmpInst::ICMP_SGE || LHSCC == ICmpInst::ICMP_SLE || | ||||
4354 | RHSCC == ICmpInst::ICMP_SGE || RHSCC == ICmpInst::ICMP_SLE) | ||||
4355 | return 0; | ||||
4356 | |||||
4357 | // We can't fold (ugt x, C) | (sgt x, C2). | ||||
4358 | if (!PredicatesFoldable(LHSCC, RHSCC)) | ||||
4359 | return 0; | ||||
4360 | |||||
4361 | // Ensure that the larger constant is on the RHS. | ||||
4362 | bool ShouldSwap; | ||||
4363 | if (ICmpInst::isSignedPredicate(LHSCC) || | ||||
4364 | (ICmpInst::isEquality(LHSCC) && | ||||
4365 | ICmpInst::isSignedPredicate(RHSCC))) | ||||
4366 | ShouldSwap = LHSCst->getValue().sgt(RHSCst->getValue()); | ||||
4367 | else | ||||
4368 | ShouldSwap = LHSCst->getValue().ugt(RHSCst->getValue()); | ||||
4369 | |||||
4370 | if (ShouldSwap) { | ||||
4371 | std::swap(LHS, RHS); | ||||
4372 | std::swap(LHSCst, RHSCst); | ||||
4373 | std::swap(LHSCC, RHSCC); | ||||
4374 | } | ||||
4375 | |||||
4376 | // At this point, we know we have have two icmp instructions | ||||
4377 | // comparing a value against two constants and or'ing the result | ||||
4378 | // together. Because of the above check, we know that we only have | ||||
4379 | // ICMP_EQ, ICMP_NE, ICMP_LT, and ICMP_GT here. We also know (from the | ||||
4380 | // FoldICmpLogical check above), that the two constants are not | ||||
4381 | // equal. | ||||
4382 | assert(LHSCst != RHSCst && "Compares not folded above?"); | ||||
4383 | |||||
4384 | switch (LHSCC) { | ||||
4385 | default: assert(0 && "Unknown integer condition code!"); | ||||
4386 | case ICmpInst::ICMP_EQ: | ||||
4387 | switch (RHSCC) { | ||||
4388 | default: assert(0 && "Unknown integer condition code!"); | ||||
4389 | case ICmpInst::ICMP_EQ: | ||||
4390 | if (LHSCst == SubOne(RHSCst)) { // (X == 13 | X == 14) -> X-13 <u 2 | ||||
4391 | Constant *AddCST = ConstantExpr::getNeg(LHSCst); | ||||
4392 | Instruction *Add = BinaryOperator::CreateAdd(Val, AddCST, | ||||
4393 | Val->getName()+".off"); | ||||
4394 | InsertNewInstBefore(Add, I); | ||||
4395 | AddCST = Subtract(AddOne(RHSCst), LHSCst); | ||||
4396 | return new ICmpInst(ICmpInst::ICMP_ULT, Add, AddCST); | ||||
4397 | } | ||||
4398 | break; // (X == 13 | X == 15) -> no change | ||||
4399 | case ICmpInst::ICMP_UGT: // (X == 13 | X u> 14) -> no change | ||||
4400 | case ICmpInst::ICMP_SGT: // (X == 13 | X s> 14) -> no change | ||||
4401 | break; | ||||
4402 | case ICmpInst::ICMP_NE: // (X == 13 | X != 15) -> X != 15 | ||||
4403 | case ICmpInst::ICMP_ULT: // (X == 13 | X u< 15) -> X u< 15 | ||||
4404 | case ICmpInst::ICMP_SLT: // (X == 13 | X s< 15) -> X s< 15 | ||||
4405 | return ReplaceInstUsesWith(I, RHS); | ||||
4406 | } | ||||
4407 | break; | ||||
4408 | case ICmpInst::ICMP_NE: | ||||
4409 | switch (RHSCC) { | ||||
4410 | default: assert(0 && "Unknown integer condition code!"); | ||||
4411 | case ICmpInst::ICMP_EQ: // (X != 13 | X == 15) -> X != 13 | ||||
4412 | case ICmpInst::ICMP_UGT: // (X != 13 | X u> 15) -> X != 13 | ||||
4413 | case ICmpInst::ICMP_SGT: // (X != 13 | X s> 15) -> X != 13 | ||||
4414 | return ReplaceInstUsesWith(I, LHS); | ||||
4415 | case ICmpInst::ICMP_NE: // (X != 13 | X != 15) -> true | ||||
4416 | case ICmpInst::ICMP_ULT: // (X != 13 | X u< 15) -> true | ||||
4417 | case ICmpInst::ICMP_SLT: // (X != 13 | X s< 15) -> true | ||||
4418 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4419 | } | ||||
4420 | break; | ||||
4421 | case ICmpInst::ICMP_ULT: | ||||
4422 | switch (RHSCC) { | ||||
4423 | default: assert(0 && "Unknown integer condition code!"); | ||||
4424 | case ICmpInst::ICMP_EQ: // (X u< 13 | X == 14) -> no change | ||||
4425 | break; | ||||
4426 | case ICmpInst::ICMP_UGT: // (X u< 13 | X u> 15) -> (X-13) u> 2 | ||||
4427 | // If RHSCst is [us]MAXINT, it is always false. Not handling | ||||
4428 | // this can cause overflow. | ||||
4429 | if (RHSCst->isMaxValue(false)) | ||||
4430 | return ReplaceInstUsesWith(I, LHS); | ||||
4431 | return InsertRangeTest(Val, LHSCst, AddOne(RHSCst), false, false, I); | ||||
4432 | case ICmpInst::ICMP_SGT: // (X u< 13 | X s> 15) -> no change | ||||
4433 | break; | ||||
4434 | case ICmpInst::ICMP_NE: // (X u< 13 | X != 15) -> X != 15 | ||||
4435 | case ICmpInst::ICMP_ULT: // (X u< 13 | X u< 15) -> X u< 15 | ||||
4436 | return ReplaceInstUsesWith(I, RHS); | ||||
4437 | case ICmpInst::ICMP_SLT: // (X u< 13 | X s< 15) -> no change | ||||
4438 | break; | ||||
4439 | } | ||||
4440 | break; | ||||
4441 | case ICmpInst::ICMP_SLT: | ||||
4442 | switch (RHSCC) { | ||||
4443 | default: assert(0 && "Unknown integer condition code!"); | ||||
4444 | case ICmpInst::ICMP_EQ: // (X s< 13 | X == 14) -> no change | ||||
4445 | break; | ||||
4446 | case ICmpInst::ICMP_SGT: // (X s< 13 | X s> 15) -> (X-13) s> 2 | ||||
4447 | // If RHSCst is [us]MAXINT, it is always false. Not handling | ||||
4448 | // this can cause overflow. | ||||
4449 | if (RHSCst->isMaxValue(true)) | ||||
4450 | return ReplaceInstUsesWith(I, LHS); | ||||
4451 | return InsertRangeTest(Val, LHSCst, AddOne(RHSCst), true, false, I); | ||||
4452 | case ICmpInst::ICMP_UGT: // (X s< 13 | X u> 15) -> no change | ||||
4453 | break; | ||||
4454 | case ICmpInst::ICMP_NE: // (X s< 13 | X != 15) -> X != 15 | ||||
4455 | case ICmpInst::ICMP_SLT: // (X s< 13 | X s< 15) -> X s< 15 | ||||
4456 | return ReplaceInstUsesWith(I, RHS); | ||||
4457 | case ICmpInst::ICMP_ULT: // (X s< 13 | X u< 15) -> no change | ||||
4458 | break; | ||||
4459 | } | ||||
4460 | break; | ||||
4461 | case ICmpInst::ICMP_UGT: | ||||
4462 | switch (RHSCC) { | ||||
4463 | default: assert(0 && "Unknown integer condition code!"); | ||||
4464 | case ICmpInst::ICMP_EQ: // (X u> 13 | X == 15) -> X u> 13 | ||||
4465 | case ICmpInst::ICMP_UGT: // (X u> 13 | X u> 15) -> X u> 13 | ||||
4466 | return ReplaceInstUsesWith(I, LHS); | ||||
4467 | case ICmpInst::ICMP_SGT: // (X u> 13 | X s> 15) -> no change | ||||
4468 | break; | ||||
4469 | case ICmpInst::ICMP_NE: // (X u> 13 | X != 15) -> true | ||||
4470 | case ICmpInst::ICMP_ULT: // (X u> 13 | X u< 15) -> true | ||||
4471 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4472 | case ICmpInst::ICMP_SLT: // (X u> 13 | X s< 15) -> no change | ||||
4473 | break; | ||||
4474 | } | ||||
4475 | break; | ||||
4476 | case ICmpInst::ICMP_SGT: | ||||
4477 | switch (RHSCC) { | ||||
4478 | default: assert(0 && "Unknown integer condition code!"); | ||||
4479 | case ICmpInst::ICMP_EQ: // (X s> 13 | X == 15) -> X > 13 | ||||
4480 | case ICmpInst::ICMP_SGT: // (X s> 13 | X s> 15) -> X > 13 | ||||
4481 | return ReplaceInstUsesWith(I, LHS); | ||||
4482 | case ICmpInst::ICMP_UGT: // (X s> 13 | X u> 15) -> no change | ||||
4483 | break; | ||||
4484 | case ICmpInst::ICMP_NE: // (X s> 13 | X != 15) -> true | ||||
4485 | case ICmpInst::ICMP_SLT: // (X s> 13 | X s< 15) -> true | ||||
4486 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4487 | case ICmpInst::ICMP_ULT: // (X s> 13 | X u< 15) -> no change | ||||
4488 | break; | ||||
4489 | } | ||||
4490 | break; | ||||
4491 | } | ||||
4492 | return 0; | ||||
4493 | } | ||||
4494 | |||||
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4495 | /// FoldOrWithConstants - This helper function folds: |
4496 | /// | ||||
Bill Wendling | 236a119 | 2008-12-02 05:09:00 +0000 | [diff] [blame] | 4497 | /// ((A | B) & C1) | (B & C2) |
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4498 | /// |
4499 | /// into: | ||||
4500 | /// | ||||
Bill Wendling | 236a119 | 2008-12-02 05:09:00 +0000 | [diff] [blame] | 4501 | /// (A & C1) | B |
Bill Wendling | 9912f71 | 2008-12-01 08:32:40 +0000 | [diff] [blame] | 4502 | /// |
Bill Wendling | 236a119 | 2008-12-02 05:09:00 +0000 | [diff] [blame] | 4503 | /// when the XOR of the two constants is "all ones" (-1). |
Bill Wendling | 9912f71 | 2008-12-01 08:32:40 +0000 | [diff] [blame] | 4504 | Instruction *InstCombiner::FoldOrWithConstants(BinaryOperator &I, Value *Op, |
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4505 | Value *A, Value *B, Value *C) { |
Bill Wendling | fc5b8e6 | 2008-12-02 05:06:43 +0000 | [diff] [blame] | 4506 | ConstantInt *CI1 = dyn_cast<ConstantInt>(C); |
4507 | if (!CI1) return 0; | ||||
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4508 | |
Bill Wendling | 0a0dcaf | 2008-12-02 06:24:20 +0000 | [diff] [blame] | 4509 | Value *V1 = 0; |
4510 | ConstantInt *CI2 = 0; | ||||
4511 | if (!match(Op, m_And(m_Value(V1), m_ConstantInt(CI2)))) return 0; | ||||
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4512 | |
Bill Wendling | 86ee316 | 2008-12-02 06:18:11 +0000 | [diff] [blame] | 4513 | APInt Xor = CI1->getValue() ^ CI2->getValue(); |
4514 | if (!Xor.isAllOnesValue()) return 0; | ||||
4515 | |||||
Bill Wendling | 0a0dcaf | 2008-12-02 06:24:20 +0000 | [diff] [blame] | 4516 | if (V1 == A || V1 == B) { |
Bill Wendling | 86ee316 | 2008-12-02 06:18:11 +0000 | [diff] [blame] | 4517 | Instruction *NewOp = |
Bill Wendling | 6c8ecbb | 2008-12-02 06:22:04 +0000 | [diff] [blame] | 4518 | InsertNewInstBefore(BinaryOperator::CreateAnd((V1 == A) ? B : A, CI1), I); |
4519 | return BinaryOperator::CreateOr(NewOp, V1); | ||||
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4520 | } |
4521 | |||||
4522 | return 0; | ||||
4523 | } | ||||
4524 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4525 | Instruction *InstCombiner::visitOr(BinaryOperator &I) { |
4526 | bool Changed = SimplifyCommutative(I); | ||||
4527 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
4528 | |||||
4529 | if (isa<UndefValue>(Op1)) // X | undef -> -1 | ||||
4530 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4531 | |||||
4532 | // or X, X = X | ||||
4533 | if (Op0 == Op1) | ||||
4534 | return ReplaceInstUsesWith(I, Op0); | ||||
4535 | |||||
4536 | // See if we can simplify any instructions used by the instruction whose sole | ||||
4537 | // purpose is to compute bits we don't care about. | ||||
4538 | if (!isa<VectorType>(I.getType())) { | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 4539 | if (SimplifyDemandedInstructionBits(I)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4540 | return &I; |
4541 | } else if (isa<ConstantAggregateZero>(Op1)) { | ||||
4542 | return ReplaceInstUsesWith(I, Op0); // X | <0,0> -> X | ||||
4543 | } else if (ConstantVector *CP = dyn_cast<ConstantVector>(Op1)) { | ||||
4544 | if (CP->isAllOnesValue()) // X | <-1,-1> -> <-1,-1> | ||||
4545 | return ReplaceInstUsesWith(I, I.getOperand(1)); | ||||
4546 | } | ||||
4547 | |||||
4548 | |||||
4549 | |||||
4550 | // or X, -1 == -1 | ||||
4551 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
4552 | ConstantInt *C1 = 0; Value *X = 0; | ||||
4553 | // (X & C1) | C2 --> (X | C2) & (C1|C2) | ||||
4554 | 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] | 4555 | Instruction *Or = BinaryOperator::CreateOr(X, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4556 | InsertNewInstBefore(Or, I); |
4557 | Or->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4558 | return BinaryOperator::CreateAnd(Or, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4559 | ConstantInt::get(RHS->getValue() | C1->getValue())); |
4560 | } | ||||
4561 | |||||
4562 | // (X ^ C1) | C2 --> (X | C2) ^ (C1&~C2) | ||||
4563 | 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] | 4564 | Instruction *Or = BinaryOperator::CreateOr(X, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4565 | InsertNewInstBefore(Or, I); |
4566 | Or->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4567 | return BinaryOperator::CreateXor(Or, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4568 | ConstantInt::get(C1->getValue() & ~RHS->getValue())); |
4569 | } | ||||
4570 | |||||
4571 | // Try to fold constant and into select arguments. | ||||
4572 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
4573 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
4574 | return R; | ||||
4575 | if (isa<PHINode>(Op0)) | ||||
4576 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
4577 | return NV; | ||||
4578 | } | ||||
4579 | |||||
4580 | Value *A = 0, *B = 0; | ||||
4581 | ConstantInt *C1 = 0, *C2 = 0; | ||||
4582 | |||||
4583 | if (match(Op0, m_And(m_Value(A), m_Value(B)))) | ||||
4584 | if (A == Op1 || B == Op1) // (A & ?) | A --> A | ||||
4585 | return ReplaceInstUsesWith(I, Op1); | ||||
4586 | if (match(Op1, m_And(m_Value(A), m_Value(B)))) | ||||
4587 | if (A == Op0 || B == Op0) // A | (A & ?) --> A | ||||
4588 | return ReplaceInstUsesWith(I, Op0); | ||||
4589 | |||||
4590 | // (A | B) | C and A | (B | C) -> bswap if possible. | ||||
4591 | // (A >> B) | (C << D) and (A << B) | (B >> C) -> bswap if possible. | ||||
4592 | if (match(Op0, m_Or(m_Value(), m_Value())) || | ||||
4593 | match(Op1, m_Or(m_Value(), m_Value())) || | ||||
4594 | (match(Op0, m_Shift(m_Value(), m_Value())) && | ||||
4595 | match(Op1, m_Shift(m_Value(), m_Value())))) { | ||||
4596 | if (Instruction *BSwap = MatchBSwap(I)) | ||||
4597 | return BSwap; | ||||
4598 | } | ||||
4599 | |||||
4600 | // (X^C)|Y -> (X|Y)^C iff Y&C == 0 | ||||
4601 | if (Op0->hasOneUse() && match(Op0, m_Xor(m_Value(A), m_ConstantInt(C1))) && | ||||
4602 | MaskedValueIsZero(Op1, C1->getValue())) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4603 | Instruction *NOr = BinaryOperator::CreateOr(A, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4604 | InsertNewInstBefore(NOr, I); |
4605 | NOr->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4606 | return BinaryOperator::CreateXor(NOr, C1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4607 | } |
4608 | |||||
4609 | // Y|(X^C) -> (X|Y)^C iff Y&C == 0 | ||||
4610 | if (Op1->hasOneUse() && match(Op1, m_Xor(m_Value(A), m_ConstantInt(C1))) && | ||||
4611 | MaskedValueIsZero(Op0, C1->getValue())) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4612 | Instruction *NOr = BinaryOperator::CreateOr(A, Op0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4613 | InsertNewInstBefore(NOr, I); |
4614 | NOr->takeName(Op0); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4615 | return BinaryOperator::CreateXor(NOr, C1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4616 | } |
4617 | |||||
4618 | // (A & C)|(B & D) | ||||
4619 | Value *C = 0, *D = 0; | ||||
4620 | if (match(Op0, m_And(m_Value(A), m_Value(C))) && | ||||
4621 | match(Op1, m_And(m_Value(B), m_Value(D)))) { | ||||
4622 | Value *V1 = 0, *V2 = 0, *V3 = 0; | ||||
4623 | C1 = dyn_cast<ConstantInt>(C); | ||||
4624 | C2 = dyn_cast<ConstantInt>(D); | ||||
4625 | if (C1 && C2) { // (A & C1)|(B & C2) | ||||
4626 | // If we have: ((V + N) & C1) | (V & C2) | ||||
4627 | // .. and C2 = ~C1 and C2 is 0+1+ and (N & C2) == 0 | ||||
4628 | // replace with V+N. | ||||
4629 | if (C1->getValue() == ~C2->getValue()) { | ||||
4630 | if ((C2->getValue() & (C2->getValue()+1)) == 0 && // C2 == 0+1+ | ||||
4631 | match(A, m_Add(m_Value(V1), m_Value(V2)))) { | ||||
4632 | // Add commutes, try both ways. | ||||
4633 | if (V1 == B && MaskedValueIsZero(V2, C2->getValue())) | ||||
4634 | return ReplaceInstUsesWith(I, A); | ||||
4635 | if (V2 == B && MaskedValueIsZero(V1, C2->getValue())) | ||||
4636 | return ReplaceInstUsesWith(I, A); | ||||
4637 | } | ||||
4638 | // Or commutes, try both ways. | ||||
4639 | if ((C1->getValue() & (C1->getValue()+1)) == 0 && | ||||
4640 | match(B, m_Add(m_Value(V1), m_Value(V2)))) { | ||||
4641 | // Add commutes, try both ways. | ||||
4642 | if (V1 == A && MaskedValueIsZero(V2, C1->getValue())) | ||||
4643 | return ReplaceInstUsesWith(I, B); | ||||
4644 | if (V2 == A && MaskedValueIsZero(V1, C1->getValue())) | ||||
4645 | return ReplaceInstUsesWith(I, B); | ||||
4646 | } | ||||
4647 | } | ||||
4648 | V1 = 0; V2 = 0; V3 = 0; | ||||
4649 | } | ||||
4650 | |||||
4651 | // Check to see if we have any common things being and'ed. If so, find the | ||||
4652 | // terms for V1 & (V2|V3). | ||||
4653 | if (isOnlyUse(Op0) || isOnlyUse(Op1)) { | ||||
4654 | if (A == B) // (A & C)|(A & D) == A & (C|D) | ||||
4655 | V1 = A, V2 = C, V3 = D; | ||||
4656 | else if (A == D) // (A & C)|(B & A) == A & (B|C) | ||||
4657 | V1 = A, V2 = B, V3 = C; | ||||
4658 | else if (C == B) // (A & C)|(C & D) == C & (A|D) | ||||
4659 | V1 = C, V2 = A, V3 = D; | ||||
4660 | else if (C == D) // (A & C)|(B & C) == C & (A|B) | ||||
4661 | V1 = C, V2 = A, V3 = B; | ||||
4662 | |||||
4663 | if (V1) { | ||||
4664 | Value *Or = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4665 | InsertNewInstBefore(BinaryOperator::CreateOr(V2, V3, "tmp"), I); |
4666 | return BinaryOperator::CreateAnd(V1, Or); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4667 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4668 | } |
Dan Gohman | 279952c | 2008-10-28 22:38:57 +0000 | [diff] [blame] | 4669 | |
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 4670 | // (A & (C0?-1:0)) | (B & ~(C0?-1:0)) -> C0 ? A : B, and commuted variants |
Chris Lattner | dd7772b | 2008-11-16 04:24:12 +0000 | [diff] [blame] | 4671 | if (Instruction *Match = MatchSelectFromAndOr(A, B, C, D)) |
4672 | return Match; | ||||
4673 | if (Instruction *Match = MatchSelectFromAndOr(B, A, D, C)) | ||||
4674 | return Match; | ||||
4675 | if (Instruction *Match = MatchSelectFromAndOr(C, B, A, D)) | ||||
4676 | return Match; | ||||
4677 | if (Instruction *Match = MatchSelectFromAndOr(D, A, B, C)) | ||||
4678 | return Match; | ||||
Bill Wendling | 22ca835 | 2008-11-30 13:52:49 +0000 | [diff] [blame] | 4679 | |
Bill Wendling | 22ca835 | 2008-11-30 13:52:49 +0000 | [diff] [blame] | 4680 | // ((A&~B)|(~A&B)) -> A^B |
Bill Wendling | c1f3113 | 2008-12-01 08:09:47 +0000 | [diff] [blame] | 4681 | if ((match(C, m_Not(m_Specific(D))) && |
4682 | match(B, m_Not(m_Specific(A))))) | ||||
4683 | return BinaryOperator::CreateXor(A, D); | ||||
Bill Wendling | 22ca835 | 2008-11-30 13:52:49 +0000 | [diff] [blame] | 4684 | // ((~B&A)|(~A&B)) -> A^B |
Bill Wendling | c1f3113 | 2008-12-01 08:09:47 +0000 | [diff] [blame] | 4685 | if ((match(A, m_Not(m_Specific(D))) && |
4686 | match(B, m_Not(m_Specific(C))))) | ||||
4687 | return BinaryOperator::CreateXor(C, D); | ||||
Bill Wendling | 22ca835 | 2008-11-30 13:52:49 +0000 | [diff] [blame] | 4688 | // ((A&~B)|(B&~A)) -> A^B |
Bill Wendling | c1f3113 | 2008-12-01 08:09:47 +0000 | [diff] [blame] | 4689 | if ((match(C, m_Not(m_Specific(B))) && |
4690 | match(D, m_Not(m_Specific(A))))) | ||||
4691 | return BinaryOperator::CreateXor(A, B); | ||||
Bill Wendling | 22ca835 | 2008-11-30 13:52:49 +0000 | [diff] [blame] | 4692 | // ((~B&A)|(B&~A)) -> A^B |
Bill Wendling | c1f3113 | 2008-12-01 08:09:47 +0000 | [diff] [blame] | 4693 | if ((match(A, m_Not(m_Specific(B))) && |
4694 | match(D, m_Not(m_Specific(C))))) | ||||
4695 | return BinaryOperator::CreateXor(C, B); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4696 | } |
4697 | |||||
4698 | // (X >> Z) | (Y >> Z) -> (X|Y) >> Z for all shifts. | ||||
4699 | if (BinaryOperator *SI1 = dyn_cast<BinaryOperator>(Op1)) { | ||||
4700 | if (BinaryOperator *SI0 = dyn_cast<BinaryOperator>(Op0)) | ||||
4701 | if (SI0->isShift() && SI0->getOpcode() == SI1->getOpcode() && | ||||
4702 | SI0->getOperand(1) == SI1->getOperand(1) && | ||||
4703 | (SI0->hasOneUse() || SI1->hasOneUse())) { | ||||
4704 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4705 | InsertNewInstBefore(BinaryOperator::CreateOr(SI0->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4706 | SI1->getOperand(0), |
4707 | SI0->getName()), I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4708 | return BinaryOperator::Create(SI1->getOpcode(), NewOp, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4709 | SI1->getOperand(1)); |
4710 | } | ||||
4711 | } | ||||
4712 | |||||
Bill Wendling | d8ce237 | 2008-12-01 01:07:11 +0000 | [diff] [blame] | 4713 | // ((A|B)&1)|(B&-2) -> (A&1) | B |
4714 | if (match(Op0, m_And(m_Or(m_Value(A), m_Value(B)), m_Value(C))) || | ||||
4715 | match(Op0, m_And(m_Value(C), m_Or(m_Value(A), m_Value(B))))) { | ||||
Bill Wendling | 9912f71 | 2008-12-01 08:32:40 +0000 | [diff] [blame] | 4716 | Instruction *Ret = FoldOrWithConstants(I, Op1, A, B, C); |
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4717 | if (Ret) return Ret; |
Bill Wendling | d8ce237 | 2008-12-01 01:07:11 +0000 | [diff] [blame] | 4718 | } |
4719 | // (B&-2)|((A|B)&1) -> (A&1) | B | ||||
4720 | if (match(Op1, m_And(m_Or(m_Value(A), m_Value(B)), m_Value(C))) || | ||||
4721 | match(Op1, m_And(m_Value(C), m_Or(m_Value(A), m_Value(B))))) { | ||||
Bill Wendling | 9912f71 | 2008-12-01 08:32:40 +0000 | [diff] [blame] | 4722 | Instruction *Ret = FoldOrWithConstants(I, Op0, A, B, C); |
Bill Wendling | dae376a | 2008-12-01 08:23:25 +0000 | [diff] [blame] | 4723 | if (Ret) return Ret; |
Bill Wendling | d8ce237 | 2008-12-01 01:07:11 +0000 | [diff] [blame] | 4724 | } |
4725 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4726 | if (match(Op0, m_Not(m_Value(A)))) { // ~A | Op1 |
4727 | if (A == Op1) // ~A | A == -1 | ||||
4728 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4729 | } else { | ||||
4730 | A = 0; | ||||
4731 | } | ||||
4732 | // Note, A is still live here! | ||||
4733 | if (match(Op1, m_Not(m_Value(B)))) { // Op0 | ~B | ||||
4734 | if (Op0 == B) | ||||
4735 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4736 | |||||
4737 | // (~A | ~B) == (~(A & B)) - De Morgan's Law | ||||
4738 | if (A && isOnlyUse(Op0) && isOnlyUse(Op1)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4739 | Value *And = InsertNewInstBefore(BinaryOperator::CreateAnd(A, B, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4740 | I.getName()+".demorgan"), I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4741 | return BinaryOperator::CreateNot(And); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4742 | } |
4743 | } | ||||
4744 | |||||
4745 | // (icmp1 A, B) | (icmp2 A, B) --> (icmp3 A, B) | ||||
4746 | if (ICmpInst *RHS = dyn_cast<ICmpInst>(I.getOperand(1))) { | ||||
4747 | if (Instruction *R = AssociativeOpt(I, FoldICmpLogical(*this, RHS))) | ||||
4748 | return R; | ||||
4749 | |||||
Chris Lattner | 0c678e5 | 2008-11-16 05:20:07 +0000 | [diff] [blame] | 4750 | if (ICmpInst *LHS = dyn_cast<ICmpInst>(I.getOperand(0))) |
4751 | if (Instruction *Res = FoldOrOfICmps(I, LHS, RHS)) | ||||
4752 | return Res; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4753 | } |
4754 | |||||
4755 | // fold (or (cast A), (cast B)) -> (cast (or A, B)) | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4756 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4757 | if (CastInst *Op1C = dyn_cast<CastInst>(Op1)) |
4758 | if (Op0C->getOpcode() == Op1C->getOpcode()) {// same cast kind ? | ||||
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 4759 | if (!isa<ICmpInst>(Op0C->getOperand(0)) || |
4760 | !isa<ICmpInst>(Op1C->getOperand(0))) { | ||||
4761 | const Type *SrcTy = Op0C->getOperand(0)->getType(); | ||||
4762 | if (SrcTy == Op1C->getOperand(0)->getType() && SrcTy->isInteger() && | ||||
4763 | // Only do this if the casts both really cause code to be | ||||
4764 | // generated. | ||||
4765 | ValueRequiresCast(Op0C->getOpcode(), Op0C->getOperand(0), | ||||
4766 | I.getType(), TD) && | ||||
4767 | ValueRequiresCast(Op1C->getOpcode(), Op1C->getOperand(0), | ||||
4768 | I.getType(), TD)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4769 | Instruction *NewOp = BinaryOperator::CreateOr(Op0C->getOperand(0), |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 4770 | Op1C->getOperand(0), |
4771 | I.getName()); | ||||
4772 | InsertNewInstBefore(NewOp, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4773 | return CastInst::Create(Op0C->getOpcode(), NewOp, I.getType()); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 4774 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4775 | } |
4776 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4777 | } |
4778 | |||||
4779 | |||||
4780 | // (fcmp uno x, c) | (fcmp uno y, c) -> (fcmp uno x, y) | ||||
4781 | if (FCmpInst *LHS = dyn_cast<FCmpInst>(I.getOperand(0))) { | ||||
4782 | if (FCmpInst *RHS = dyn_cast<FCmpInst>(I.getOperand(1))) { | ||||
4783 | if (LHS->getPredicate() == FCmpInst::FCMP_UNO && | ||||
Chris Lattner | be9e63e | 2008-02-29 06:09:11 +0000 | [diff] [blame] | 4784 | RHS->getPredicate() == FCmpInst::FCMP_UNO && |
Evan Cheng | 7298805 | 2008-10-14 18:44:08 +0000 | [diff] [blame] | 4785 | LHS->getOperand(0)->getType() == RHS->getOperand(0)->getType()) { |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4786 | if (ConstantFP *LHSC = dyn_cast<ConstantFP>(LHS->getOperand(1))) |
4787 | if (ConstantFP *RHSC = dyn_cast<ConstantFP>(RHS->getOperand(1))) { | ||||
4788 | // If either of the constants are nans, then the whole thing returns | ||||
4789 | // true. | ||||
Chris Lattner | a6c7dce | 2007-10-24 18:54:45 +0000 | [diff] [blame] | 4790 | if (LHSC->getValueAPF().isNaN() || RHSC->getValueAPF().isNaN()) |
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4791 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
4792 | |||||
4793 | // Otherwise, no need to compare the two constants, compare the | ||||
4794 | // rest. | ||||
4795 | return new FCmpInst(FCmpInst::FCMP_UNO, LHS->getOperand(0), | ||||
4796 | RHS->getOperand(0)); | ||||
4797 | } | ||||
Evan Cheng | 7298805 | 2008-10-14 18:44:08 +0000 | [diff] [blame] | 4798 | } else { |
4799 | Value *Op0LHS, *Op0RHS, *Op1LHS, *Op1RHS; | ||||
4800 | FCmpInst::Predicate Op0CC, Op1CC; | ||||
4801 | if (match(Op0, m_FCmp(Op0CC, m_Value(Op0LHS), m_Value(Op0RHS))) && | ||||
4802 | match(Op1, m_FCmp(Op1CC, m_Value(Op1LHS), m_Value(Op1RHS)))) { | ||||
4803 | if (Op0LHS == Op1RHS && Op0RHS == Op1LHS) { | ||||
4804 | // Swap RHS operands to match LHS. | ||||
4805 | Op1CC = FCmpInst::getSwappedPredicate(Op1CC); | ||||
4806 | std::swap(Op1LHS, Op1RHS); | ||||
4807 | } | ||||
4808 | if (Op0LHS == Op1LHS && Op0RHS == Op1RHS) { | ||||
4809 | // Simplify (fcmp cc0 x, y) | (fcmp cc1 x, y). | ||||
4810 | if (Op0CC == Op1CC) | ||||
4811 | return new FCmpInst((FCmpInst::Predicate)Op0CC, Op0LHS, Op0RHS); | ||||
4812 | else if (Op0CC == FCmpInst::FCMP_TRUE || | ||||
4813 | Op1CC == FCmpInst::FCMP_TRUE) | ||||
4814 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
4815 | else if (Op0CC == FCmpInst::FCMP_FALSE) | ||||
4816 | return ReplaceInstUsesWith(I, Op1); | ||||
4817 | else if (Op1CC == FCmpInst::FCMP_FALSE) | ||||
4818 | return ReplaceInstUsesWith(I, Op0); | ||||
4819 | bool Op0Ordered; | ||||
4820 | bool Op1Ordered; | ||||
4821 | unsigned Op0Pred = getFCmpCode(Op0CC, Op0Ordered); | ||||
4822 | unsigned Op1Pred = getFCmpCode(Op1CC, Op1Ordered); | ||||
4823 | if (Op0Ordered == Op1Ordered) { | ||||
4824 | // If both are ordered or unordered, return a new fcmp with | ||||
4825 | // or'ed predicates. | ||||
4826 | Value *RV = getFCmpValue(Op0Ordered, Op0Pred|Op1Pred, | ||||
4827 | Op0LHS, Op0RHS); | ||||
4828 | if (Instruction *I = dyn_cast<Instruction>(RV)) | ||||
4829 | return I; | ||||
4830 | // Otherwise, it's a constant boolean value... | ||||
4831 | return ReplaceInstUsesWith(I, RV); | ||||
4832 | } | ||||
4833 | } | ||||
4834 | } | ||||
4835 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 4836 | } |
4837 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4838 | |
4839 | return Changed ? &I : 0; | ||||
4840 | } | ||||
4841 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 4842 | namespace { |
4843 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4844 | // XorSelf - Implements: X ^ X --> 0 |
4845 | struct XorSelf { | ||||
4846 | Value *RHS; | ||||
4847 | XorSelf(Value *rhs) : RHS(rhs) {} | ||||
4848 | bool shouldApply(Value *LHS) const { return LHS == RHS; } | ||||
4849 | Instruction *apply(BinaryOperator &Xor) const { | ||||
4850 | return &Xor; | ||||
4851 | } | ||||
4852 | }; | ||||
4853 | |||||
Dan Gohman | 089efff | 2008-05-13 00:00:25 +0000 | [diff] [blame] | 4854 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4855 | |
4856 | Instruction *InstCombiner::visitXor(BinaryOperator &I) { | ||||
4857 | bool Changed = SimplifyCommutative(I); | ||||
4858 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
4859 | |||||
Evan Cheng | e5cd803 | 2008-03-25 20:07:13 +0000 | [diff] [blame] | 4860 | if (isa<UndefValue>(Op1)) { |
4861 | if (isa<UndefValue>(Op0)) | ||||
4862 | // Handle undef ^ undef -> 0 special case. This is a common | ||||
4863 | // idiom (misuse). | ||||
4864 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4865 | return ReplaceInstUsesWith(I, Op1); // X ^ undef -> undef |
Evan Cheng | e5cd803 | 2008-03-25 20:07:13 +0000 | [diff] [blame] | 4866 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4867 | |
4868 | // xor X, X = 0, even if X is nested in a sequence of Xor's. | ||||
4869 | if (Instruction *Result = AssociativeOpt(I, XorSelf(Op1))) { | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 4870 | assert(Result == &I && "AssociativeOpt didn't work?"); Result=Result; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4871 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); |
4872 | } | ||||
4873 | |||||
4874 | // See if we can simplify any instructions used by the instruction whose sole | ||||
4875 | // purpose is to compute bits we don't care about. | ||||
4876 | if (!isa<VectorType>(I.getType())) { | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 4877 | if (SimplifyDemandedInstructionBits(I)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4878 | return &I; |
4879 | } else if (isa<ConstantAggregateZero>(Op1)) { | ||||
4880 | return ReplaceInstUsesWith(I, Op0); // X ^ <0,0> -> X | ||||
4881 | } | ||||
4882 | |||||
4883 | // Is this a ~ operation? | ||||
4884 | if (Value *NotOp = dyn_castNotVal(&I)) { | ||||
4885 | // ~(~X & Y) --> (X | ~Y) - De Morgan's Law | ||||
4886 | // ~(~X | Y) === (X & ~Y) - De Morgan's Law | ||||
4887 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(NotOp)) { | ||||
4888 | if (Op0I->getOpcode() == Instruction::And || | ||||
4889 | Op0I->getOpcode() == Instruction::Or) { | ||||
4890 | if (dyn_castNotVal(Op0I->getOperand(1))) Op0I->swapOperands(); | ||||
4891 | if (Value *Op0NotVal = dyn_castNotVal(Op0I->getOperand(0))) { | ||||
4892 | Instruction *NotY = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4893 | BinaryOperator::CreateNot(Op0I->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4894 | Op0I->getOperand(1)->getName()+".not"); |
4895 | InsertNewInstBefore(NotY, I); | ||||
4896 | if (Op0I->getOpcode() == Instruction::And) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4897 | return BinaryOperator::CreateOr(Op0NotVal, NotY); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4898 | else |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4899 | return BinaryOperator::CreateAnd(Op0NotVal, NotY); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4900 | } |
4901 | } | ||||
4902 | } | ||||
4903 | } | ||||
4904 | |||||
4905 | |||||
4906 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(Op1)) { | ||||
Nick Lewycky | 1405e92 | 2007-08-06 20:04:16 +0000 | [diff] [blame] | 4907 | if (RHS == ConstantInt::getTrue() && Op0->hasOneUse()) { |
Bill Wendling | 6174195 | 2009-01-01 01:18:23 +0000 | [diff] [blame] | 4908 | // xor (cmp A, B), true = not (cmp A, B) = !cmp A, B |
Nick Lewycky | 1405e92 | 2007-08-06 20:04:16 +0000 | [diff] [blame] | 4909 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(Op0)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4910 | return new ICmpInst(ICI->getInversePredicate(), |
4911 | ICI->getOperand(0), ICI->getOperand(1)); | ||||
4912 | |||||
Nick Lewycky | 1405e92 | 2007-08-06 20:04:16 +0000 | [diff] [blame] | 4913 | if (FCmpInst *FCI = dyn_cast<FCmpInst>(Op0)) |
4914 | return new FCmpInst(FCI->getInversePredicate(), | ||||
4915 | FCI->getOperand(0), FCI->getOperand(1)); | ||||
4916 | } | ||||
4917 | |||||
Nick Lewycky | 0aa63aa | 2008-05-31 19:01:33 +0000 | [diff] [blame] | 4918 | // fold (xor(zext(cmp)), 1) and (xor(sext(cmp)), -1) to ext(!cmp). |
4919 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) { | ||||
4920 | if (CmpInst *CI = dyn_cast<CmpInst>(Op0C->getOperand(0))) { | ||||
4921 | if (CI->hasOneUse() && Op0C->hasOneUse()) { | ||||
4922 | Instruction::CastOps Opcode = Op0C->getOpcode(); | ||||
4923 | if (Opcode == Instruction::ZExt || Opcode == Instruction::SExt) { | ||||
4924 | if (RHS == ConstantExpr::getCast(Opcode, ConstantInt::getTrue(), | ||||
4925 | Op0C->getDestTy())) { | ||||
4926 | Instruction *NewCI = InsertNewInstBefore(CmpInst::Create( | ||||
4927 | CI->getOpcode(), CI->getInversePredicate(), | ||||
4928 | CI->getOperand(0), CI->getOperand(1)), I); | ||||
4929 | NewCI->takeName(CI); | ||||
4930 | return CastInst::Create(Opcode, NewCI, Op0C->getType()); | ||||
4931 | } | ||||
4932 | } | ||||
4933 | } | ||||
4934 | } | ||||
4935 | } | ||||
4936 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4937 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) { |
4938 | // ~(c-X) == X-c-1 == X+(-c-1) | ||||
4939 | if (Op0I->getOpcode() == Instruction::Sub && RHS->isAllOnesValue()) | ||||
4940 | if (Constant *Op0I0C = dyn_cast<Constant>(Op0I->getOperand(0))) { | ||||
4941 | Constant *NegOp0I0C = ConstantExpr::getNeg(Op0I0C); | ||||
4942 | Constant *ConstantRHS = ConstantExpr::getSub(NegOp0I0C, | ||||
4943 | ConstantInt::get(I.getType(), 1)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4944 | return BinaryOperator::CreateAdd(Op0I->getOperand(1), ConstantRHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4945 | } |
4946 | |||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 4947 | if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4948 | if (Op0I->getOpcode() == Instruction::Add) { |
4949 | // ~(X-c) --> (-c-1)-X | ||||
4950 | if (RHS->isAllOnesValue()) { | ||||
4951 | Constant *NegOp0CI = ConstantExpr::getNeg(Op0CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4952 | return BinaryOperator::CreateSub( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4953 | ConstantExpr::getSub(NegOp0CI, |
4954 | ConstantInt::get(I.getType(), 1)), | ||||
4955 | Op0I->getOperand(0)); | ||||
4956 | } else if (RHS->getValue().isSignBit()) { | ||||
4957 | // (X + C) ^ signbit -> (X + C + signbit) | ||||
4958 | Constant *C = ConstantInt::get(RHS->getValue() + Op0CI->getValue()); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 4959 | return BinaryOperator::CreateAdd(Op0I->getOperand(0), C); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4960 | |
4961 | } | ||||
4962 | } else if (Op0I->getOpcode() == Instruction::Or) { | ||||
4963 | // (X|C1)^C2 -> X^(C1|C2) iff X&~C1 == 0 | ||||
4964 | if (MaskedValueIsZero(Op0I->getOperand(0), Op0CI->getValue())) { | ||||
4965 | Constant *NewRHS = ConstantExpr::getOr(Op0CI, RHS); | ||||
4966 | // Anything in both C1 and C2 is known to be zero, remove it from | ||||
4967 | // NewRHS. | ||||
4968 | Constant *CommonBits = And(Op0CI, RHS); | ||||
4969 | NewRHS = ConstantExpr::getAnd(NewRHS, | ||||
4970 | ConstantExpr::getNot(CommonBits)); | ||||
4971 | AddToWorkList(Op0I); | ||||
4972 | I.setOperand(0, Op0I->getOperand(0)); | ||||
4973 | I.setOperand(1, NewRHS); | ||||
4974 | return &I; | ||||
4975 | } | ||||
4976 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 4977 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 4978 | } |
4979 | |||||
4980 | // Try to fold constant and into select arguments. | ||||
4981 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
4982 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
4983 | return R; | ||||
4984 | if (isa<PHINode>(Op0)) | ||||
4985 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
4986 | return NV; | ||||
4987 | } | ||||
4988 | |||||
4989 | if (Value *X = dyn_castNotVal(Op0)) // ~A ^ A == -1 | ||||
4990 | if (X == Op1) | ||||
4991 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4992 | |||||
4993 | if (Value *X = dyn_castNotVal(Op1)) // A ^ ~A == -1 | ||||
4994 | if (X == Op0) | ||||
4995 | return ReplaceInstUsesWith(I, Constant::getAllOnesValue(I.getType())); | ||||
4996 | |||||
4997 | |||||
4998 | BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1); | ||||
4999 | if (Op1I) { | ||||
5000 | Value *A, *B; | ||||
5001 | if (match(Op1I, m_Or(m_Value(A), m_Value(B)))) { | ||||
5002 | if (A == Op0) { // B^(B|A) == (A|B)^B | ||||
5003 | Op1I->swapOperands(); | ||||
5004 | I.swapOperands(); | ||||
5005 | std::swap(Op0, Op1); | ||||
5006 | } else if (B == Op0) { // B^(A|B) == (A|B)^B | ||||
5007 | I.swapOperands(); // Simplified below. | ||||
5008 | std::swap(Op0, Op1); | ||||
5009 | } | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 5010 | } else if (match(Op1I, m_Xor(m_Specific(Op0), m_Value(B)))) { |
5011 | return ReplaceInstUsesWith(I, B); // A^(A^B) == B | ||||
5012 | } else if (match(Op1I, m_Xor(m_Value(A), m_Specific(Op0)))) { | ||||
5013 | return ReplaceInstUsesWith(I, A); // A^(B^A) == B | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5014 | } else if (match(Op1I, m_And(m_Value(A), m_Value(B))) && Op1I->hasOneUse()){ |
5015 | if (A == Op0) { // A^(A&B) -> A^(B&A) | ||||
5016 | Op1I->swapOperands(); | ||||
5017 | std::swap(A, B); | ||||
5018 | } | ||||
5019 | if (B == Op0) { // A^(B&A) -> (B&A)^A | ||||
5020 | I.swapOperands(); // Simplified below. | ||||
5021 | std::swap(Op0, Op1); | ||||
5022 | } | ||||
5023 | } | ||||
5024 | } | ||||
5025 | |||||
5026 | BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0); | ||||
5027 | if (Op0I) { | ||||
5028 | Value *A, *B; | ||||
5029 | if (match(Op0I, m_Or(m_Value(A), m_Value(B))) && Op0I->hasOneUse()) { | ||||
5030 | if (A == Op1) // (B|A)^B == (A|B)^B | ||||
5031 | std::swap(A, B); | ||||
5032 | if (B == Op1) { // (A|B)^B == A & ~B | ||||
5033 | Instruction *NotB = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5034 | InsertNewInstBefore(BinaryOperator::CreateNot(Op1, "tmp"), I); |
5035 | return BinaryOperator::CreateAnd(A, NotB); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5036 | } |
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 5037 | } else if (match(Op0I, m_Xor(m_Specific(Op1), m_Value(B)))) { |
5038 | return ReplaceInstUsesWith(I, B); // (A^B)^A == B | ||||
5039 | } else if (match(Op0I, m_Xor(m_Value(A), m_Specific(Op1)))) { | ||||
5040 | return ReplaceInstUsesWith(I, A); // (B^A)^A == B | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5041 | } else if (match(Op0I, m_And(m_Value(A), m_Value(B))) && Op0I->hasOneUse()){ |
5042 | if (A == Op1) // (A&B)^A -> (B&A)^A | ||||
5043 | std::swap(A, B); | ||||
5044 | if (B == Op1 && // (B&A)^A == ~B & A | ||||
5045 | !isa<ConstantInt>(Op1)) { // Canonical form is (B&C)^C | ||||
5046 | Instruction *N = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5047 | InsertNewInstBefore(BinaryOperator::CreateNot(A, "tmp"), I); |
5048 | return BinaryOperator::CreateAnd(N, Op1); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5049 | } |
5050 | } | ||||
5051 | } | ||||
5052 | |||||
5053 | // (X >> Z) ^ (Y >> Z) -> (X^Y) >> Z for all shifts. | ||||
5054 | if (Op0I && Op1I && Op0I->isShift() && | ||||
5055 | Op0I->getOpcode() == Op1I->getOpcode() && | ||||
5056 | Op0I->getOperand(1) == Op1I->getOperand(1) && | ||||
5057 | (Op1I->hasOneUse() || Op1I->hasOneUse())) { | ||||
5058 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5059 | InsertNewInstBefore(BinaryOperator::CreateXor(Op0I->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5060 | Op1I->getOperand(0), |
5061 | Op0I->getName()), I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5062 | return BinaryOperator::Create(Op1I->getOpcode(), NewOp, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5063 | Op1I->getOperand(1)); |
5064 | } | ||||
5065 | |||||
5066 | if (Op0I && Op1I) { | ||||
5067 | Value *A, *B, *C, *D; | ||||
5068 | // (A & B)^(A | B) -> A ^ B | ||||
5069 | if (match(Op0I, m_And(m_Value(A), m_Value(B))) && | ||||
5070 | match(Op1I, m_Or(m_Value(C), m_Value(D)))) { | ||||
5071 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5072 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5073 | } |
5074 | // (A | B)^(A & B) -> A ^ B | ||||
5075 | if (match(Op0I, m_Or(m_Value(A), m_Value(B))) && | ||||
5076 | match(Op1I, m_And(m_Value(C), m_Value(D)))) { | ||||
5077 | if ((A == C && B == D) || (A == D && B == C)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5078 | return BinaryOperator::CreateXor(A, B); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5079 | } |
5080 | |||||
5081 | // (A & B)^(C & D) | ||||
5082 | if ((Op0I->hasOneUse() || Op1I->hasOneUse()) && | ||||
5083 | match(Op0I, m_And(m_Value(A), m_Value(B))) && | ||||
5084 | match(Op1I, m_And(m_Value(C), m_Value(D)))) { | ||||
5085 | // (X & Y)^(X & Y) -> (Y^Z) & X | ||||
5086 | Value *X = 0, *Y = 0, *Z = 0; | ||||
5087 | if (A == C) | ||||
5088 | X = A, Y = B, Z = D; | ||||
5089 | else if (A == D) | ||||
5090 | X = A, Y = B, Z = C; | ||||
5091 | else if (B == C) | ||||
5092 | X = B, Y = A, Z = D; | ||||
5093 | else if (B == D) | ||||
5094 | X = B, Y = A, Z = C; | ||||
5095 | |||||
5096 | if (X) { | ||||
5097 | Instruction *NewOp = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5098 | InsertNewInstBefore(BinaryOperator::CreateXor(Y, Z, Op0->getName()), I); |
5099 | return BinaryOperator::CreateAnd(NewOp, X); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5100 | } |
5101 | } | ||||
5102 | } | ||||
5103 | |||||
5104 | // (icmp1 A, B) ^ (icmp2 A, B) --> (icmp3 A, B) | ||||
5105 | if (ICmpInst *RHS = dyn_cast<ICmpInst>(I.getOperand(1))) | ||||
5106 | if (Instruction *R = AssociativeOpt(I, FoldICmpLogical(*this, RHS))) | ||||
5107 | return R; | ||||
5108 | |||||
5109 | // fold (xor (cast A), (cast B)) -> (cast (xor A, B)) | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 5110 | if (CastInst *Op0C = dyn_cast<CastInst>(Op0)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5111 | if (CastInst *Op1C = dyn_cast<CastInst>(Op1)) |
5112 | if (Op0C->getOpcode() == Op1C->getOpcode()) { // same cast kind? | ||||
5113 | const Type *SrcTy = Op0C->getOperand(0)->getType(); | ||||
5114 | if (SrcTy == Op1C->getOperand(0)->getType() && SrcTy->isInteger() && | ||||
5115 | // Only do this if the casts both really cause code to be generated. | ||||
5116 | ValueRequiresCast(Op0C->getOpcode(), Op0C->getOperand(0), | ||||
5117 | I.getType(), TD) && | ||||
5118 | ValueRequiresCast(Op1C->getOpcode(), Op1C->getOperand(0), | ||||
5119 | I.getType(), TD)) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5120 | Instruction *NewOp = BinaryOperator::CreateXor(Op0C->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5121 | Op1C->getOperand(0), |
5122 | I.getName()); | ||||
5123 | InsertNewInstBefore(NewOp, I); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5124 | return CastInst::Create(Op0C->getOpcode(), NewOp, I.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5125 | } |
5126 | } | ||||
Chris Lattner | 9188243 | 2007-10-24 05:38:08 +0000 | [diff] [blame] | 5127 | } |
Nick Lewycky | 0aa63aa | 2008-05-31 19:01:33 +0000 | [diff] [blame] | 5128 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5129 | return Changed ? &I : 0; |
5130 | } | ||||
5131 | |||||
5132 | /// AddWithOverflow - Compute Result = In1+In2, returning true if the result | ||||
5133 | /// overflowed for this type. | ||||
5134 | static bool AddWithOverflow(ConstantInt *&Result, ConstantInt *In1, | ||||
5135 | ConstantInt *In2, bool IsSigned = false) { | ||||
5136 | Result = cast<ConstantInt>(Add(In1, In2)); | ||||
5137 | |||||
5138 | if (IsSigned) | ||||
5139 | if (In2->getValue().isNegative()) | ||||
5140 | return Result->getValue().sgt(In1->getValue()); | ||||
5141 | else | ||||
5142 | return Result->getValue().slt(In1->getValue()); | ||||
5143 | else | ||||
5144 | return Result->getValue().ult(In1->getValue()); | ||||
5145 | } | ||||
5146 | |||||
Dan Gohman | b80d561 | 2008-09-10 23:30:57 +0000 | [diff] [blame] | 5147 | /// SubWithOverflow - Compute Result = In1-In2, returning true if the result |
5148 | /// overflowed for this type. | ||||
5149 | static bool SubWithOverflow(ConstantInt *&Result, ConstantInt *In1, | ||||
5150 | ConstantInt *In2, bool IsSigned = false) { | ||||
Dan Gohman | 2c3b489 | 2008-09-11 18:53:02 +0000 | [diff] [blame] | 5151 | Result = cast<ConstantInt>(Subtract(In1, In2)); |
Dan Gohman | b80d561 | 2008-09-10 23:30:57 +0000 | [diff] [blame] | 5152 | |
5153 | if (IsSigned) | ||||
5154 | if (In2->getValue().isNegative()) | ||||
5155 | return Result->getValue().slt(In1->getValue()); | ||||
5156 | else | ||||
5157 | return Result->getValue().sgt(In1->getValue()); | ||||
5158 | else | ||||
5159 | return Result->getValue().ugt(In1->getValue()); | ||||
5160 | } | ||||
5161 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5162 | /// EmitGEPOffset - Given a getelementptr instruction/constantexpr, emit the |
5163 | /// code necessary to compute the offset from the base pointer (without adding | ||||
5164 | /// in the base pointer). Return the result as a signed integer of intptr size. | ||||
5165 | static Value *EmitGEPOffset(User *GEP, Instruction &I, InstCombiner &IC) { | ||||
5166 | TargetData &TD = IC.getTargetData(); | ||||
5167 | gep_type_iterator GTI = gep_type_begin(GEP); | ||||
5168 | const Type *IntPtrTy = TD.getIntPtrType(); | ||||
5169 | Value *Result = Constant::getNullValue(IntPtrTy); | ||||
5170 | |||||
5171 | // Build a mask for high order bits. | ||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5172 | unsigned IntPtrWidth = TD.getPointerSizeInBits(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5173 | uint64_t PtrSizeMask = ~0ULL >> (64-IntPtrWidth); |
5174 | |||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 5175 | for (User::op_iterator i = GEP->op_begin() + 1, e = GEP->op_end(); i != e; |
5176 | ++i, ++GTI) { | ||||
5177 | Value *Op = *i; | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 5178 | uint64_t Size = TD.getTypePaddedSize(GTI.getIndexedType()) & PtrSizeMask; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5179 | if (ConstantInt *OpC = dyn_cast<ConstantInt>(Op)) { |
5180 | if (OpC->isZero()) continue; | ||||
5181 | |||||
5182 | // Handle a struct index, which adds its field offset to the pointer. | ||||
5183 | if (const StructType *STy = dyn_cast<StructType>(*GTI)) { | ||||
5184 | Size = TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue()); | ||||
5185 | |||||
5186 | if (ConstantInt *RC = dyn_cast<ConstantInt>(Result)) | ||||
5187 | Result = ConstantInt::get(RC->getValue() + APInt(IntPtrWidth, Size)); | ||||
5188 | else | ||||
5189 | Result = IC.InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5190 | BinaryOperator::CreateAdd(Result, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5191 | ConstantInt::get(IntPtrTy, Size), |
5192 | GEP->getName()+".offs"), I); | ||||
5193 | continue; | ||||
5194 | } | ||||
5195 | |||||
5196 | Constant *Scale = ConstantInt::get(IntPtrTy, Size); | ||||
5197 | Constant *OC = ConstantExpr::getIntegerCast(OpC, IntPtrTy, true /*SExt*/); | ||||
5198 | Scale = ConstantExpr::getMul(OC, Scale); | ||||
5199 | if (Constant *RC = dyn_cast<Constant>(Result)) | ||||
5200 | Result = ConstantExpr::getAdd(RC, Scale); | ||||
5201 | else { | ||||
5202 | // Emit an add instruction. | ||||
5203 | Result = IC.InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5204 | BinaryOperator::CreateAdd(Result, Scale, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5205 | GEP->getName()+".offs"), I); |
5206 | } | ||||
5207 | continue; | ||||
5208 | } | ||||
5209 | // Convert to correct type. | ||||
5210 | if (Op->getType() != IntPtrTy) { | ||||
5211 | if (Constant *OpC = dyn_cast<Constant>(Op)) | ||||
5212 | Op = ConstantExpr::getSExt(OpC, IntPtrTy); | ||||
5213 | else | ||||
5214 | Op = IC.InsertNewInstBefore(new SExtInst(Op, IntPtrTy, | ||||
5215 | Op->getName()+".c"), I); | ||||
5216 | } | ||||
5217 | if (Size != 1) { | ||||
5218 | Constant *Scale = ConstantInt::get(IntPtrTy, Size); | ||||
5219 | if (Constant *OpC = dyn_cast<Constant>(Op)) | ||||
5220 | Op = ConstantExpr::getMul(OpC, Scale); | ||||
5221 | 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] | 5222 | Op = IC.InsertNewInstBefore(BinaryOperator::CreateMul(Op, Scale, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5223 | GEP->getName()+".idx"), I); |
5224 | } | ||||
5225 | |||||
5226 | // Emit an add instruction. | ||||
5227 | if (isa<Constant>(Op) && isa<Constant>(Result)) | ||||
5228 | Result = ConstantExpr::getAdd(cast<Constant>(Op), | ||||
5229 | cast<Constant>(Result)); | ||||
5230 | else | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5231 | Result = IC.InsertNewInstBefore(BinaryOperator::CreateAdd(Op, Result, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5232 | GEP->getName()+".offs"), I); |
5233 | } | ||||
5234 | return Result; | ||||
5235 | } | ||||
5236 | |||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5237 | |
5238 | /// EvaluateGEPOffsetExpression - Return an value that can be used to compare of | ||||
5239 | /// the *offset* implied by GEP to zero. For example, if we have &A[i], we want | ||||
5240 | /// to return 'i' for "icmp ne i, 0". Note that, in general, indices can be | ||||
5241 | /// complex, and scales are involved. The above expression would also be legal | ||||
5242 | /// to codegen as "icmp ne (i*4), 0" (assuming A is a pointer to i32). This | ||||
5243 | /// later form is less amenable to optimization though, and we are allowed to | ||||
5244 | /// generate the first by knowing that pointer arithmetic doesn't overflow. | ||||
5245 | /// | ||||
5246 | /// If we can't emit an optimized form for this expression, this returns null. | ||||
5247 | /// | ||||
5248 | static Value *EvaluateGEPOffsetExpression(User *GEP, Instruction &I, | ||||
5249 | InstCombiner &IC) { | ||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5250 | TargetData &TD = IC.getTargetData(); |
5251 | gep_type_iterator GTI = gep_type_begin(GEP); | ||||
5252 | |||||
5253 | // Check to see if this gep only has a single variable index. If so, and if | ||||
5254 | // any constant indices are a multiple of its scale, then we can compute this | ||||
5255 | // in terms of the scale of the variable index. For example, if the GEP | ||||
5256 | // implies an offset of "12 + i*4", then we can codegen this as "3 + i", | ||||
5257 | // because the expression will cross zero at the same point. | ||||
5258 | unsigned i, e = GEP->getNumOperands(); | ||||
5259 | int64_t Offset = 0; | ||||
5260 | for (i = 1; i != e; ++i, ++GTI) { | ||||
5261 | if (ConstantInt *CI = dyn_cast<ConstantInt>(GEP->getOperand(i))) { | ||||
5262 | // Compute the aggregate offset of constant indices. | ||||
5263 | if (CI->isZero()) continue; | ||||
5264 | |||||
5265 | // Handle a struct index, which adds its field offset to the pointer. | ||||
5266 | if (const StructType *STy = dyn_cast<StructType>(*GTI)) { | ||||
5267 | Offset += TD.getStructLayout(STy)->getElementOffset(CI->getZExtValue()); | ||||
5268 | } else { | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 5269 | uint64_t Size = TD.getTypePaddedSize(GTI.getIndexedType()); |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5270 | Offset += Size*CI->getSExtValue(); |
5271 | } | ||||
5272 | } else { | ||||
5273 | // Found our variable index. | ||||
5274 | break; | ||||
5275 | } | ||||
5276 | } | ||||
5277 | |||||
5278 | // If there are no variable indices, we must have a constant offset, just | ||||
5279 | // evaluate it the general way. | ||||
5280 | if (i == e) return 0; | ||||
5281 | |||||
5282 | Value *VariableIdx = GEP->getOperand(i); | ||||
5283 | // Determine the scale factor of the variable element. For example, this is | ||||
5284 | // 4 if the variable index is into an array of i32. | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 5285 | uint64_t VariableScale = TD.getTypePaddedSize(GTI.getIndexedType()); |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5286 | |
5287 | // Verify that there are no other variable indices. If so, emit the hard way. | ||||
5288 | for (++i, ++GTI; i != e; ++i, ++GTI) { | ||||
5289 | ConstantInt *CI = dyn_cast<ConstantInt>(GEP->getOperand(i)); | ||||
5290 | if (!CI) return 0; | ||||
5291 | |||||
5292 | // Compute the aggregate offset of constant indices. | ||||
5293 | if (CI->isZero()) continue; | ||||
5294 | |||||
5295 | // Handle a struct index, which adds its field offset to the pointer. | ||||
5296 | if (const StructType *STy = dyn_cast<StructType>(*GTI)) { | ||||
5297 | Offset += TD.getStructLayout(STy)->getElementOffset(CI->getZExtValue()); | ||||
5298 | } else { | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 5299 | uint64_t Size = TD.getTypePaddedSize(GTI.getIndexedType()); |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5300 | Offset += Size*CI->getSExtValue(); |
5301 | } | ||||
5302 | } | ||||
5303 | |||||
5304 | // Okay, we know we have a single variable index, which must be a | ||||
5305 | // pointer/array/vector index. If there is no offset, life is simple, return | ||||
5306 | // the index. | ||||
5307 | unsigned IntPtrWidth = TD.getPointerSizeInBits(); | ||||
5308 | if (Offset == 0) { | ||||
5309 | // Cast to intptrty in case a truncation occurs. If an extension is needed, | ||||
5310 | // we don't need to bother extending: the extension won't affect where the | ||||
5311 | // computation crosses zero. | ||||
5312 | if (VariableIdx->getType()->getPrimitiveSizeInBits() > IntPtrWidth) | ||||
5313 | VariableIdx = new TruncInst(VariableIdx, TD.getIntPtrType(), | ||||
5314 | VariableIdx->getNameStart(), &I); | ||||
5315 | return VariableIdx; | ||||
5316 | } | ||||
5317 | |||||
5318 | // Otherwise, there is an index. The computation we will do will be modulo | ||||
5319 | // the pointer size, so get it. | ||||
5320 | uint64_t PtrSizeMask = ~0ULL >> (64-IntPtrWidth); | ||||
5321 | |||||
5322 | Offset &= PtrSizeMask; | ||||
5323 | VariableScale &= PtrSizeMask; | ||||
5324 | |||||
5325 | // To do this transformation, any constant index must be a multiple of the | ||||
5326 | // variable scale factor. For example, we can evaluate "12 + 4*i" as "3 + i", | ||||
5327 | // but we can't evaluate "10 + 3*i" in terms of i. Check that the offset is a | ||||
5328 | // multiple of the variable scale. | ||||
5329 | int64_t NewOffs = Offset / (int64_t)VariableScale; | ||||
5330 | if (Offset != NewOffs*(int64_t)VariableScale) | ||||
5331 | return 0; | ||||
5332 | |||||
5333 | // Okay, we can do this evaluation. Start by converting the index to intptr. | ||||
5334 | const Type *IntPtrTy = TD.getIntPtrType(); | ||||
5335 | if (VariableIdx->getType() != IntPtrTy) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5336 | VariableIdx = CastInst::CreateIntegerCast(VariableIdx, IntPtrTy, |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5337 | true /*SExt*/, |
5338 | VariableIdx->getNameStart(), &I); | ||||
5339 | Constant *OffsetVal = ConstantInt::get(IntPtrTy, NewOffs); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5340 | return BinaryOperator::CreateAdd(VariableIdx, OffsetVal, "offset", &I); |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5341 | } |
5342 | |||||
5343 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5344 | /// FoldGEPICmp - Fold comparisons between a GEP instruction and something |
5345 | /// else. At this point we know that the GEP is on the LHS of the comparison. | ||||
5346 | Instruction *InstCombiner::FoldGEPICmp(User *GEPLHS, Value *RHS, | ||||
5347 | ICmpInst::Predicate Cond, | ||||
5348 | Instruction &I) { | ||||
5349 | assert(dyn_castGetElementPtr(GEPLHS) && "LHS is not a getelementptr!"); | ||||
5350 | |||||
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5351 | // Look through bitcasts. |
5352 | if (BitCastInst *BCI = dyn_cast<BitCastInst>(RHS)) | ||||
5353 | RHS = BCI->getOperand(0); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5354 | |
5355 | Value *PtrBase = GEPLHS->getOperand(0); | ||||
5356 | if (PtrBase == RHS) { | ||||
Chris Lattner | af97d02 | 2008-02-05 04:45:32 +0000 | [diff] [blame] | 5357 | // ((gep Ptr, OFFSET) cmp Ptr) ---> (OFFSET cmp 0). |
Chris Lattner | eba7586 | 2008-04-22 02:53:33 +0000 | [diff] [blame] | 5358 | // This transformation (ignoring the base and scales) is valid because we |
5359 | // know pointers can't overflow. See if we can output an optimized form. | ||||
5360 | Value *Offset = EvaluateGEPOffsetExpression(GEPLHS, I, *this); | ||||
5361 | |||||
5362 | // If not, synthesize the offset the hard way. | ||||
5363 | if (Offset == 0) | ||||
5364 | Offset = EmitGEPOffset(GEPLHS, I, *this); | ||||
Chris Lattner | af97d02 | 2008-02-05 04:45:32 +0000 | [diff] [blame] | 5365 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), Offset, |
5366 | Constant::getNullValue(Offset->getType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5367 | } else if (User *GEPRHS = dyn_castGetElementPtr(RHS)) { |
5368 | // If the base pointers are different, but the indices are the same, just | ||||
5369 | // compare the base pointer. | ||||
5370 | if (PtrBase != GEPRHS->getOperand(0)) { | ||||
5371 | bool IndicesTheSame = GEPLHS->getNumOperands()==GEPRHS->getNumOperands(); | ||||
5372 | IndicesTheSame &= GEPLHS->getOperand(0)->getType() == | ||||
5373 | GEPRHS->getOperand(0)->getType(); | ||||
5374 | if (IndicesTheSame) | ||||
5375 | for (unsigned i = 1, e = GEPLHS->getNumOperands(); i != e; ++i) | ||||
5376 | if (GEPLHS->getOperand(i) != GEPRHS->getOperand(i)) { | ||||
5377 | IndicesTheSame = false; | ||||
5378 | break; | ||||
5379 | } | ||||
5380 | |||||
5381 | // If all indices are the same, just compare the base pointers. | ||||
5382 | if (IndicesTheSame) | ||||
5383 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), | ||||
5384 | GEPLHS->getOperand(0), GEPRHS->getOperand(0)); | ||||
5385 | |||||
5386 | // Otherwise, the base pointers are different and the indices are | ||||
5387 | // different, bail out. | ||||
5388 | return 0; | ||||
5389 | } | ||||
5390 | |||||
5391 | // If one of the GEPs has all zero indices, recurse. | ||||
5392 | bool AllZeros = true; | ||||
5393 | for (unsigned i = 1, e = GEPLHS->getNumOperands(); i != e; ++i) | ||||
5394 | if (!isa<Constant>(GEPLHS->getOperand(i)) || | ||||
5395 | !cast<Constant>(GEPLHS->getOperand(i))->isNullValue()) { | ||||
5396 | AllZeros = false; | ||||
5397 | break; | ||||
5398 | } | ||||
5399 | if (AllZeros) | ||||
5400 | return FoldGEPICmp(GEPRHS, GEPLHS->getOperand(0), | ||||
5401 | ICmpInst::getSwappedPredicate(Cond), I); | ||||
5402 | |||||
5403 | // If the other GEP has all zero indices, recurse. | ||||
5404 | AllZeros = true; | ||||
5405 | for (unsigned i = 1, e = GEPRHS->getNumOperands(); i != e; ++i) | ||||
5406 | if (!isa<Constant>(GEPRHS->getOperand(i)) || | ||||
5407 | !cast<Constant>(GEPRHS->getOperand(i))->isNullValue()) { | ||||
5408 | AllZeros = false; | ||||
5409 | break; | ||||
5410 | } | ||||
5411 | if (AllZeros) | ||||
5412 | return FoldGEPICmp(GEPLHS, GEPRHS->getOperand(0), Cond, I); | ||||
5413 | |||||
5414 | if (GEPLHS->getNumOperands() == GEPRHS->getNumOperands()) { | ||||
5415 | // If the GEPs only differ by one index, compare it. | ||||
5416 | unsigned NumDifferences = 0; // Keep track of # differences. | ||||
5417 | unsigned DiffOperand = 0; // The operand that differs. | ||||
5418 | for (unsigned i = 1, e = GEPRHS->getNumOperands(); i != e; ++i) | ||||
5419 | if (GEPLHS->getOperand(i) != GEPRHS->getOperand(i)) { | ||||
5420 | if (GEPLHS->getOperand(i)->getType()->getPrimitiveSizeInBits() != | ||||
5421 | GEPRHS->getOperand(i)->getType()->getPrimitiveSizeInBits()) { | ||||
5422 | // Irreconcilable differences. | ||||
5423 | NumDifferences = 2; | ||||
5424 | break; | ||||
5425 | } else { | ||||
5426 | if (NumDifferences++) break; | ||||
5427 | DiffOperand = i; | ||||
5428 | } | ||||
5429 | } | ||||
5430 | |||||
5431 | if (NumDifferences == 0) // SAME GEP? | ||||
5432 | return ReplaceInstUsesWith(I, // No comparison is needed here. | ||||
Nick Lewycky | 2de09a9 | 2007-09-06 02:40:25 +0000 | [diff] [blame] | 5433 | ConstantInt::get(Type::Int1Ty, |
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 5434 | ICmpInst::isTrueWhenEqual(Cond))); |
Nick Lewycky | 2de09a9 | 2007-09-06 02:40:25 +0000 | [diff] [blame] | 5435 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5436 | else if (NumDifferences == 1) { |
5437 | Value *LHSV = GEPLHS->getOperand(DiffOperand); | ||||
5438 | Value *RHSV = GEPRHS->getOperand(DiffOperand); | ||||
5439 | // Make sure we do a signed comparison here. | ||||
5440 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), LHSV, RHSV); | ||||
5441 | } | ||||
5442 | } | ||||
5443 | |||||
5444 | // Only lower this if the icmp is the only user of the GEP or if we expect | ||||
5445 | // the result to fold to a constant! | ||||
5446 | if ((isa<ConstantExpr>(GEPLHS) || GEPLHS->hasOneUse()) && | ||||
5447 | (isa<ConstantExpr>(GEPRHS) || GEPRHS->hasOneUse())) { | ||||
5448 | // ((gep Ptr, OFFSET1) cmp (gep Ptr, OFFSET2) ---> (OFFSET1 cmp OFFSET2) | ||||
5449 | Value *L = EmitGEPOffset(GEPLHS, I, *this); | ||||
5450 | Value *R = EmitGEPOffset(GEPRHS, I, *this); | ||||
5451 | return new ICmpInst(ICmpInst::getSignedPredicate(Cond), L, R); | ||||
5452 | } | ||||
5453 | } | ||||
5454 | return 0; | ||||
5455 | } | ||||
5456 | |||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5457 | /// FoldFCmp_IntToFP_Cst - Fold fcmp ([us]itofp x, cst) if possible. |
5458 | /// | ||||
5459 | Instruction *InstCombiner::FoldFCmp_IntToFP_Cst(FCmpInst &I, | ||||
5460 | Instruction *LHSI, | ||||
5461 | Constant *RHSC) { | ||||
5462 | if (!isa<ConstantFP>(RHSC)) return 0; | ||||
5463 | const APFloat &RHS = cast<ConstantFP>(RHSC)->getValueAPF(); | ||||
5464 | |||||
5465 | // Get the width of the mantissa. We don't want to hack on conversions that | ||||
5466 | // might lose information from the integer, e.g. "i64 -> float" | ||||
Chris Lattner | 9ce836b | 2008-05-19 21:17:23 +0000 | [diff] [blame] | 5467 | int MantissaWidth = LHSI->getType()->getFPMantissaWidth(); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5468 | if (MantissaWidth == -1) return 0; // Unknown. |
5469 | |||||
5470 | // Check to see that the input is converted from an integer type that is small | ||||
5471 | // enough that preserves all bits. TODO: check here for "known" sign bits. | ||||
5472 | // This would allow us to handle (fptosi (x >>s 62) to float) if x is i64 f.e. | ||||
5473 | unsigned InputSize = LHSI->getOperand(0)->getType()->getPrimitiveSizeInBits(); | ||||
5474 | |||||
5475 | // If this is a uitofp instruction, we need an extra bit to hold the sign. | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5476 | bool LHSUnsigned = isa<UIToFPInst>(LHSI); |
5477 | if (LHSUnsigned) | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5478 | ++InputSize; |
5479 | |||||
5480 | // If the conversion would lose info, don't hack on this. | ||||
5481 | if ((int)InputSize > MantissaWidth) | ||||
5482 | return 0; | ||||
5483 | |||||
5484 | // Otherwise, we can potentially simplify the comparison. We know that it | ||||
5485 | // will always come through as an integer value and we know the constant is | ||||
5486 | // not a NAN (it would have been previously simplified). | ||||
5487 | assert(!RHS.isNaN() && "NaN comparison not already folded!"); | ||||
5488 | |||||
5489 | ICmpInst::Predicate Pred; | ||||
5490 | switch (I.getPredicate()) { | ||||
5491 | default: assert(0 && "Unexpected predicate!"); | ||||
5492 | case FCmpInst::FCMP_UEQ: | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5493 | case FCmpInst::FCMP_OEQ: |
5494 | Pred = ICmpInst::ICMP_EQ; | ||||
5495 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5496 | case FCmpInst::FCMP_UGT: |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5497 | case FCmpInst::FCMP_OGT: |
5498 | Pred = LHSUnsigned ? ICmpInst::ICMP_UGT : ICmpInst::ICMP_SGT; | ||||
5499 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5500 | case FCmpInst::FCMP_UGE: |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5501 | case FCmpInst::FCMP_OGE: |
5502 | Pred = LHSUnsigned ? ICmpInst::ICMP_UGE : ICmpInst::ICMP_SGE; | ||||
5503 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5504 | case FCmpInst::FCMP_ULT: |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5505 | case FCmpInst::FCMP_OLT: |
5506 | Pred = LHSUnsigned ? ICmpInst::ICMP_ULT : ICmpInst::ICMP_SLT; | ||||
5507 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5508 | case FCmpInst::FCMP_ULE: |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5509 | case FCmpInst::FCMP_OLE: |
5510 | Pred = LHSUnsigned ? ICmpInst::ICMP_ULE : ICmpInst::ICMP_SLE; | ||||
5511 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5512 | case FCmpInst::FCMP_UNE: |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5513 | case FCmpInst::FCMP_ONE: |
5514 | Pred = ICmpInst::ICMP_NE; | ||||
5515 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5516 | case FCmpInst::FCMP_ORD: |
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5517 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5518 | case FCmpInst::FCMP_UNO: |
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5519 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5520 | } |
5521 | |||||
5522 | const IntegerType *IntTy = cast<IntegerType>(LHSI->getOperand(0)->getType()); | ||||
5523 | |||||
5524 | // Now we know that the APFloat is a normal number, zero or inf. | ||||
5525 | |||||
Chris Lattner | f13ff49 | 2008-05-20 03:50:52 +0000 | [diff] [blame] | 5526 | // 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] | 5527 | // comparing an i8 to 300.0. |
5528 | unsigned IntWidth = IntTy->getPrimitiveSizeInBits(); | ||||
5529 | |||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5530 | if (!LHSUnsigned) { |
5531 | // If the RHS value is > SignedMax, fold the comparison. This handles +INF | ||||
5532 | // and large values. | ||||
5533 | APFloat SMax(RHS.getSemantics(), APFloat::fcZero, false); | ||||
5534 | SMax.convertFromAPInt(APInt::getSignedMaxValue(IntWidth), true, | ||||
5535 | APFloat::rmNearestTiesToEven); | ||||
5536 | if (SMax.compare(RHS) == APFloat::cmpLessThan) { // smax < 13123.0 | ||||
5537 | if (Pred == ICmpInst::ICMP_NE || Pred == ICmpInst::ICMP_SLT || | ||||
5538 | Pred == ICmpInst::ICMP_SLE) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5539 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
5540 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5541 | } |
5542 | } else { | ||||
5543 | // If the RHS value is > UnsignedMax, fold the comparison. This handles | ||||
5544 | // +INF and large values. | ||||
5545 | APFloat UMax(RHS.getSemantics(), APFloat::fcZero, false); | ||||
5546 | UMax.convertFromAPInt(APInt::getMaxValue(IntWidth), false, | ||||
5547 | APFloat::rmNearestTiesToEven); | ||||
5548 | if (UMax.compare(RHS) == APFloat::cmpLessThan) { // umax < 13123.0 | ||||
5549 | if (Pred == ICmpInst::ICMP_NE || Pred == ICmpInst::ICMP_ULT || | ||||
5550 | Pred == ICmpInst::ICMP_ULE) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5551 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
5552 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5553 | } |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5554 | } |
5555 | |||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5556 | if (!LHSUnsigned) { |
5557 | // See if the RHS value is < SignedMin. | ||||
5558 | APFloat SMin(RHS.getSemantics(), APFloat::fcZero, false); | ||||
5559 | SMin.convertFromAPInt(APInt::getSignedMinValue(IntWidth), true, | ||||
5560 | APFloat::rmNearestTiesToEven); | ||||
5561 | if (SMin.compare(RHS) == APFloat::cmpGreaterThan) { // smin > 12312.0 | ||||
5562 | if (Pred == ICmpInst::ICMP_NE || Pred == ICmpInst::ICMP_SGT || | ||||
5563 | Pred == ICmpInst::ICMP_SGE) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5564 | return ReplaceInstUsesWith(I,ConstantInt::getTrue()); |
5565 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5566 | } |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5567 | } |
5568 | |||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5569 | // Okay, now we know that the FP constant fits in the range [SMIN, SMAX] or |
5570 | // [0, UMAX], but it may still be fractional. See if it is fractional by | ||||
5571 | // casting the FP value to the integer value and back, checking for equality. | ||||
5572 | // Don't do this for zero, because -0.0 is not fractional. | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5573 | Constant *RHSInt = ConstantExpr::getFPToSI(RHSC, IntTy); |
5574 | if (!RHS.isZero() && | ||||
5575 | ConstantExpr::getSIToFP(RHSInt, RHSC->getType()) != RHSC) { | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5576 | // If we had a comparison against a fractional value, we have to adjust the |
5577 | // compare predicate and sometimes the value. RHSC is rounded towards zero | ||||
5578 | // at this point. | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5579 | switch (Pred) { |
5580 | default: assert(0 && "Unexpected integer comparison!"); | ||||
5581 | case ICmpInst::ICMP_NE: // (float)int != 4.4 --> true | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5582 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5583 | case ICmpInst::ICMP_EQ: // (float)int == 4.4 --> false |
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5584 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5585 | case ICmpInst::ICMP_ULE: |
5586 | // (float)int <= 4.4 --> int <= 4 | ||||
5587 | // (float)int <= -4.4 --> false | ||||
5588 | if (RHS.isNegative()) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5589 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5590 | break; |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5591 | case ICmpInst::ICMP_SLE: |
5592 | // (float)int <= 4.4 --> int <= 4 | ||||
5593 | // (float)int <= -4.4 --> int < -4 | ||||
5594 | if (RHS.isNegative()) | ||||
5595 | Pred = ICmpInst::ICMP_SLT; | ||||
5596 | break; | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5597 | case ICmpInst::ICMP_ULT: |
5598 | // (float)int < -4.4 --> false | ||||
5599 | // (float)int < 4.4 --> int <= 4 | ||||
5600 | if (RHS.isNegative()) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5601 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5602 | Pred = ICmpInst::ICMP_ULE; |
5603 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5604 | case ICmpInst::ICMP_SLT: |
5605 | // (float)int < -4.4 --> int < -4 | ||||
5606 | // (float)int < 4.4 --> int <= 4 | ||||
5607 | if (!RHS.isNegative()) | ||||
5608 | Pred = ICmpInst::ICMP_SLE; | ||||
5609 | break; | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5610 | case ICmpInst::ICMP_UGT: |
5611 | // (float)int > 4.4 --> int > 4 | ||||
5612 | // (float)int > -4.4 --> true | ||||
5613 | if (RHS.isNegative()) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5614 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5615 | break; |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5616 | case ICmpInst::ICMP_SGT: |
5617 | // (float)int > 4.4 --> int > 4 | ||||
5618 | // (float)int > -4.4 --> int >= -4 | ||||
5619 | if (RHS.isNegative()) | ||||
5620 | Pred = ICmpInst::ICMP_SGE; | ||||
5621 | break; | ||||
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5622 | case ICmpInst::ICMP_UGE: |
5623 | // (float)int >= -4.4 --> true | ||||
5624 | // (float)int >= 4.4 --> int > 4 | ||||
5625 | if (!RHS.isNegative()) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5626 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Bill Wendling | 20636df | 2008-11-09 04:26:50 +0000 | [diff] [blame] | 5627 | Pred = ICmpInst::ICMP_UGT; |
5628 | break; | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5629 | case ICmpInst::ICMP_SGE: |
5630 | // (float)int >= -4.4 --> int >= -4 | ||||
5631 | // (float)int >= 4.4 --> int > 4 | ||||
5632 | if (!RHS.isNegative()) | ||||
5633 | Pred = ICmpInst::ICMP_SGT; | ||||
5634 | break; | ||||
5635 | } | ||||
5636 | } | ||||
5637 | |||||
5638 | // Lower this FP comparison into an appropriate integer version of the | ||||
5639 | // comparison. | ||||
5640 | return new ICmpInst(Pred, LHSI->getOperand(0), RHSInt); | ||||
5641 | } | ||||
5642 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5643 | Instruction *InstCombiner::visitFCmpInst(FCmpInst &I) { |
5644 | bool Changed = SimplifyCompare(I); | ||||
5645 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
5646 | |||||
5647 | // Fold trivial predicates. | ||||
5648 | if (I.getPredicate() == FCmpInst::FCMP_FALSE) | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5649 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5650 | if (I.getPredicate() == FCmpInst::FCMP_TRUE) |
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5651 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5652 | |
5653 | // Simplify 'fcmp pred X, X' | ||||
5654 | if (Op0 == Op1) { | ||||
5655 | switch (I.getPredicate()) { | ||||
5656 | default: assert(0 && "Unknown predicate!"); | ||||
5657 | case FCmpInst::FCMP_UEQ: // True if unordered or equal | ||||
5658 | case FCmpInst::FCMP_UGE: // True if unordered, greater than, or equal | ||||
5659 | case FCmpInst::FCMP_ULE: // True if unordered, less than, or equal | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5660 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5661 | case FCmpInst::FCMP_OGT: // True if ordered and greater than |
5662 | case FCmpInst::FCMP_OLT: // True if ordered and less than | ||||
5663 | case FCmpInst::FCMP_ONE: // True if ordered and operands are unequal | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5664 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5665 | |
5666 | case FCmpInst::FCMP_UNO: // True if unordered: isnan(X) | isnan(Y) | ||||
5667 | case FCmpInst::FCMP_ULT: // True if unordered or less than | ||||
5668 | case FCmpInst::FCMP_UGT: // True if unordered or greater than | ||||
5669 | case FCmpInst::FCMP_UNE: // True if unordered or not equal | ||||
5670 | // Canonicalize these to be 'fcmp uno %X, 0.0'. | ||||
5671 | I.setPredicate(FCmpInst::FCMP_UNO); | ||||
5672 | I.setOperand(1, Constant::getNullValue(Op0->getType())); | ||||
5673 | return &I; | ||||
5674 | |||||
5675 | case FCmpInst::FCMP_ORD: // True if ordered (no nans) | ||||
5676 | case FCmpInst::FCMP_OEQ: // True if ordered and equal | ||||
5677 | case FCmpInst::FCMP_OGE: // True if ordered and greater than or equal | ||||
5678 | case FCmpInst::FCMP_OLE: // True if ordered and less than or equal | ||||
5679 | // Canonicalize these to be 'fcmp ord %X, 0.0'. | ||||
5680 | I.setPredicate(FCmpInst::FCMP_ORD); | ||||
5681 | I.setOperand(1, Constant::getNullValue(Op0->getType())); | ||||
5682 | return &I; | ||||
5683 | } | ||||
5684 | } | ||||
5685 | |||||
5686 | if (isa<UndefValue>(Op1)) // fcmp pred X, undef -> undef | ||||
5687 | return ReplaceInstUsesWith(I, UndefValue::get(Type::Int1Ty)); | ||||
5688 | |||||
5689 | // Handle fcmp with constant RHS | ||||
5690 | if (Constant *RHSC = dyn_cast<Constant>(Op1)) { | ||||
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5691 | // If the constant is a nan, see if we can fold the comparison based on it. |
5692 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHSC)) { | ||||
5693 | if (CFP->getValueAPF().isNaN()) { | ||||
5694 | if (FCmpInst::isOrdered(I.getPredicate())) // True if ordered and... | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5695 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | f13ff49 | 2008-05-20 03:50:52 +0000 | [diff] [blame] | 5696 | assert(FCmpInst::isUnordered(I.getPredicate()) && |
5697 | "Comparison must be either ordered or unordered!"); | ||||
5698 | // True if unordered. | ||||
Eli Friedman | c9c9624 | 2008-11-30 22:48:49 +0000 | [diff] [blame] | 5699 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5700 | } |
5701 | } | ||||
5702 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5703 | if (Instruction *LHSI = dyn_cast<Instruction>(Op0)) |
5704 | switch (LHSI->getOpcode()) { | ||||
5705 | case Instruction::PHI: | ||||
Chris Lattner | a2417ba | 2008-06-08 20:52:11 +0000 | [diff] [blame] | 5706 | // Only fold fcmp into the PHI if the phi and fcmp are in the same |
5707 | // block. If in the same block, we're encouraging jump threading. If | ||||
5708 | // not, we are just pessimizing the code by making an i1 phi. | ||||
5709 | if (LHSI->getParent() == I.getParent()) | ||||
5710 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
5711 | return NV; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5712 | break; |
Chris Lattner | e6b62d9 | 2008-05-19 20:18:56 +0000 | [diff] [blame] | 5713 | case Instruction::SIToFP: |
5714 | case Instruction::UIToFP: | ||||
5715 | if (Instruction *NV = FoldFCmp_IntToFP_Cst(I, LHSI, RHSC)) | ||||
5716 | return NV; | ||||
5717 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5718 | case Instruction::Select: |
5719 | // If either operand of the select is a constant, we can fold the | ||||
5720 | // comparison into the select arms, which will cause one to be | ||||
5721 | // constant folded and the select turned into a bitwise or. | ||||
5722 | Value *Op1 = 0, *Op2 = 0; | ||||
5723 | if (LHSI->hasOneUse()) { | ||||
5724 | if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(1))) { | ||||
5725 | // Fold the known value into the constant operand. | ||||
5726 | Op1 = ConstantExpr::getCompare(I.getPredicate(), C, RHSC); | ||||
5727 | // Insert a new FCmp of the other select operand. | ||||
5728 | Op2 = InsertNewInstBefore(new FCmpInst(I.getPredicate(), | ||||
5729 | LHSI->getOperand(2), RHSC, | ||||
5730 | I.getName()), I); | ||||
5731 | } else if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(2))) { | ||||
5732 | // Fold the known value into the constant operand. | ||||
5733 | Op2 = ConstantExpr::getCompare(I.getPredicate(), C, RHSC); | ||||
5734 | // Insert a new FCmp of the other select operand. | ||||
5735 | Op1 = InsertNewInstBefore(new FCmpInst(I.getPredicate(), | ||||
5736 | LHSI->getOperand(1), RHSC, | ||||
5737 | I.getName()), I); | ||||
5738 | } | ||||
5739 | } | ||||
5740 | |||||
5741 | if (Op1) | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 5742 | return SelectInst::Create(LHSI->getOperand(0), Op1, Op2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5743 | break; |
5744 | } | ||||
5745 | } | ||||
5746 | |||||
5747 | return Changed ? &I : 0; | ||||
5748 | } | ||||
5749 | |||||
5750 | Instruction *InstCombiner::visitICmpInst(ICmpInst &I) { | ||||
5751 | bool Changed = SimplifyCompare(I); | ||||
5752 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
5753 | const Type *Ty = Op0->getType(); | ||||
5754 | |||||
5755 | // icmp X, X | ||||
5756 | if (Op0 == Op1) | ||||
5757 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, | ||||
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 5758 | I.isTrueWhenEqual())); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5759 | |
5760 | if (isa<UndefValue>(Op1)) // X icmp undef -> undef | ||||
5761 | return ReplaceInstUsesWith(I, UndefValue::get(Type::Int1Ty)); | ||||
Christopher Lamb | f78cd32 | 2007-12-18 21:32:20 +0000 | [diff] [blame] | 5762 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5763 | // icmp <global/alloca*/null>, <global/alloca*/null> - Global/Stack value |
5764 | // addresses never equal each other! We already know that Op0 != Op1. | ||||
5765 | if ((isa<GlobalValue>(Op0) || isa<AllocaInst>(Op0) || | ||||
5766 | isa<ConstantPointerNull>(Op0)) && | ||||
5767 | (isa<GlobalValue>(Op1) || isa<AllocaInst>(Op1) || | ||||
5768 | isa<ConstantPointerNull>(Op1))) | ||||
5769 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, | ||||
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 5770 | !I.isTrueWhenEqual())); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5771 | |
5772 | // icmp's with boolean values can always be turned into bitwise operations | ||||
5773 | if (Ty == Type::Int1Ty) { | ||||
5774 | switch (I.getPredicate()) { | ||||
5775 | default: assert(0 && "Invalid icmp instruction!"); | ||||
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5776 | case ICmpInst::ICMP_EQ: { // icmp eq i1 A, B -> ~(A^B) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5777 | Instruction *Xor = BinaryOperator::CreateXor(Op0, Op1, I.getName()+"tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5778 | InsertNewInstBefore(Xor, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5779 | return BinaryOperator::CreateNot(Xor); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5780 | } |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5781 | case ICmpInst::ICMP_NE: // icmp eq i1 A, B -> A^B |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5782 | return BinaryOperator::CreateXor(Op0, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5783 | |
5784 | case ICmpInst::ICMP_UGT: | ||||
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5785 | std::swap(Op0, Op1); // Change icmp ugt -> icmp ult |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5786 | // FALL THROUGH |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5787 | case ICmpInst::ICMP_ULT:{ // icmp ult i1 A, B -> ~A & B |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5788 | Instruction *Not = BinaryOperator::CreateNot(Op0, I.getName()+"tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5789 | InsertNewInstBefore(Not, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5790 | return BinaryOperator::CreateAnd(Not, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5791 | } |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5792 | case ICmpInst::ICMP_SGT: |
5793 | std::swap(Op0, Op1); // Change icmp sgt -> icmp slt | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5794 | // FALL THROUGH |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5795 | case ICmpInst::ICMP_SLT: { // icmp slt i1 A, B -> A & ~B |
5796 | Instruction *Not = BinaryOperator::CreateNot(Op1, I.getName()+"tmp"); | ||||
5797 | InsertNewInstBefore(Not, I); | ||||
5798 | return BinaryOperator::CreateAnd(Not, Op0); | ||||
5799 | } | ||||
5800 | case ICmpInst::ICMP_UGE: | ||||
5801 | std::swap(Op0, Op1); // Change icmp uge -> icmp ule | ||||
5802 | // FALL THROUGH | ||||
5803 | case ICmpInst::ICMP_ULE: { // icmp ule i1 A, B -> ~A | B | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5804 | Instruction *Not = BinaryOperator::CreateNot(Op0, I.getName()+"tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5805 | InsertNewInstBefore(Not, I); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 5806 | return BinaryOperator::CreateOr(Not, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5807 | } |
Chris Lattner | a02893d | 2008-07-11 04:20:58 +0000 | [diff] [blame] | 5808 | case ICmpInst::ICMP_SGE: |
5809 | std::swap(Op0, Op1); // Change icmp sge -> icmp sle | ||||
5810 | // FALL THROUGH | ||||
5811 | case ICmpInst::ICMP_SLE: { // icmp sle i1 A, B -> A | ~B | ||||
5812 | Instruction *Not = BinaryOperator::CreateNot(Op1, I.getName()+"tmp"); | ||||
5813 | InsertNewInstBefore(Not, I); | ||||
5814 | return BinaryOperator::CreateOr(Not, Op0); | ||||
5815 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5816 | } |
5817 | } | ||||
5818 | |||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 5819 | // See if we are doing a comparison with a constant. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5820 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) { |
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 5821 | Value *A, *B; |
Christopher Lamb | fa6b310 | 2007-12-20 07:21:11 +0000 | [diff] [blame] | 5822 | |
Chris Lattner | be6c54a | 2008-01-05 01:18:20 +0000 | [diff] [blame] | 5823 | // (icmp ne/eq (sub A B) 0) -> (icmp ne/eq A, B) |
5824 | if (I.isEquality() && CI->isNullValue() && | ||||
5825 | match(Op0, m_Sub(m_Value(A), m_Value(B)))) { | ||||
5826 | // (icmp cond A B) if cond is equality | ||||
5827 | return new ICmpInst(I.getPredicate(), A, B); | ||||
Owen Anderson | 42f61ed | 2007-12-28 07:42:12 +0000 | [diff] [blame] | 5828 | } |
Christopher Lamb | fa6b310 | 2007-12-20 07:21:11 +0000 | [diff] [blame] | 5829 | |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 5830 | // If we have an icmp le or icmp ge instruction, turn it into the |
5831 | // appropriate icmp lt or icmp gt instruction. This allows us to rely on | ||||
5832 | // them being folded in the code below. | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5833 | switch (I.getPredicate()) { |
5834 | default: break; | ||||
5835 | case ICmpInst::ICMP_ULE: | ||||
5836 | if (CI->isMaxValue(false)) // A <=u MAX -> TRUE | ||||
5837 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5838 | return new ICmpInst(ICmpInst::ICMP_ULT, Op0, AddOne(CI)); | ||||
5839 | case ICmpInst::ICMP_SLE: | ||||
5840 | if (CI->isMaxValue(true)) // A <=s MAX -> TRUE | ||||
5841 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5842 | return new ICmpInst(ICmpInst::ICMP_SLT, Op0, AddOne(CI)); | ||||
5843 | case ICmpInst::ICMP_UGE: | ||||
5844 | if (CI->isMinValue(false)) // A >=u MIN -> TRUE | ||||
5845 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5846 | return new ICmpInst( ICmpInst::ICMP_UGT, Op0, SubOne(CI)); | ||||
5847 | case ICmpInst::ICMP_SGE: | ||||
5848 | if (CI->isMinValue(true)) // A >=s MIN -> TRUE | ||||
5849 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5850 | return new ICmpInst(ICmpInst::ICMP_SGT, Op0, SubOne(CI)); | ||||
5851 | } | ||||
5852 | |||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5853 | // See if we can fold the comparison based on range information we can get |
5854 | // by checking whether bits are known to be zero or one in the input. | ||||
5855 | uint32_t BitWidth = cast<IntegerType>(Ty)->getBitWidth(); | ||||
5856 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
5857 | |||||
5858 | // If this comparison is a normal comparison, it demands all | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5859 | // 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] | 5860 | bool UnusedBit; |
5861 | bool isSignBit = isSignBitCheck(I.getPredicate(), CI, UnusedBit); | ||||
5862 | |||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 5863 | if (SimplifyDemandedBits(I.getOperandUse(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5864 | isSignBit ? APInt::getSignBit(BitWidth) |
5865 | : APInt::getAllOnesValue(BitWidth), | ||||
5866 | KnownZero, KnownOne, 0)) | ||||
5867 | return &I; | ||||
5868 | |||||
5869 | // 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] | 5870 | // in. Compute the Min, Max and RHS values based on the known bits. For the |
5871 | // EQ and NE we use unsigned values. | ||||
5872 | APInt Min(BitWidth, 0), Max(BitWidth, 0); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5873 | if (ICmpInst::isSignedPredicate(I.getPredicate())) |
5874 | ComputeSignedMinMaxValuesFromKnownBits(Ty, KnownZero, KnownOne, Min, Max); | ||||
5875 | else | ||||
5876 | ComputeUnsignedMinMaxValuesFromKnownBits(Ty, KnownZero, KnownOne,Min,Max); | ||||
5877 | |||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5878 | // If Min and Max are known to be the same, then SimplifyDemandedBits |
5879 | // figured out that the LHS is a constant. Just constant fold this now so | ||||
5880 | // that code below can assume that Min != Max. | ||||
5881 | if (Min == Max) | ||||
5882 | return ReplaceInstUsesWith(I, ConstantExpr::getICmp(I.getPredicate(), | ||||
5883 | ConstantInt::get(Min), | ||||
5884 | CI)); | ||||
5885 | |||||
5886 | // Based on the range information we know about the LHS, see if we can | ||||
5887 | // simplify this comparison. For example, (x&4) < 8 is always true. | ||||
5888 | const APInt &RHSVal = CI->getValue(); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5889 | switch (I.getPredicate()) { // LE/GE have been folded already. |
5890 | default: assert(0 && "Unknown icmp opcode!"); | ||||
5891 | case ICmpInst::ICMP_EQ: | ||||
5892 | if (Max.ult(RHSVal) || Min.ugt(RHSVal)) | ||||
5893 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); | ||||
5894 | break; | ||||
5895 | case ICmpInst::ICMP_NE: | ||||
5896 | if (Max.ult(RHSVal) || Min.ugt(RHSVal)) | ||||
5897 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); | ||||
5898 | break; | ||||
5899 | case ICmpInst::ICMP_ULT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5900 | if (Max.ult(RHSVal)) // A <u C -> true iff max(A) < C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5901 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5902 | if (Min.uge(RHSVal)) // A <u C -> false iff min(A) >= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5903 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5904 | if (RHSVal == Max) // A <u MAX -> A != MAX |
5905 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
5906 | if (RHSVal == Min+1) // A <u MIN+1 -> A == MIN | ||||
5907 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, SubOne(CI)); | ||||
5908 | |||||
5909 | // (x <u 2147483648) -> (x >s -1) -> true if sign bit clear | ||||
5910 | if (CI->isMinValue(true)) | ||||
5911 | return new ICmpInst(ICmpInst::ICMP_SGT, Op0, | ||||
5912 | ConstantInt::getAllOnesValue(Op0->getType())); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5913 | break; |
5914 | case ICmpInst::ICMP_UGT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5915 | if (Min.ugt(RHSVal)) // A >u C -> true iff min(A) > C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5916 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5917 | if (Max.ule(RHSVal)) // A >u C -> false iff max(A) <= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5918 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5919 | |
5920 | if (RHSVal == Min) // A >u MIN -> A != MIN | ||||
5921 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
5922 | if (RHSVal == Max-1) // A >u MAX-1 -> A == MAX | ||||
5923 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, AddOne(CI)); | ||||
5924 | |||||
5925 | // (x >u 2147483647) -> (x <s 0) -> true if sign bit set | ||||
5926 | if (CI->isMaxValue(true)) | ||||
5927 | return new ICmpInst(ICmpInst::ICMP_SLT, Op0, | ||||
5928 | ConstantInt::getNullValue(Op0->getType())); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5929 | break; |
5930 | case ICmpInst::ICMP_SLT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5931 | if (Max.slt(RHSVal)) // A <s C -> true iff max(A) < C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5932 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | 611b43e | 2008-07-11 06:40:29 +0000 | [diff] [blame] | 5933 | if (Min.sge(RHSVal)) // A <s C -> false iff min(A) >= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5934 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5935 | if (RHSVal == Max) // A <s MAX -> A != MAX |
5936 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
Chris Lattner | 3496f3e | 2008-07-11 06:36:01 +0000 | [diff] [blame] | 5937 | if (RHSVal == Min+1) // A <s MIN+1 -> A == MIN |
Chris Lattner | 55ab315 | 2008-07-11 06:38:16 +0000 | [diff] [blame] | 5938 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, SubOne(CI)); |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5939 | break; |
5940 | case ICmpInst::ICMP_SGT: | ||||
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5941 | if (Min.sgt(RHSVal)) // A >s C -> true iff min(A) > C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5942 | return ReplaceInstUsesWith(I, ConstantInt::getTrue()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5943 | if (Max.sle(RHSVal)) // A >s C -> false iff max(A) <= C |
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5944 | return ReplaceInstUsesWith(I, ConstantInt::getFalse()); |
Chris Lattner | a130865 | 2008-07-11 05:40:05 +0000 | [diff] [blame] | 5945 | |
5946 | if (RHSVal == Min) // A >s MIN -> A != MIN | ||||
5947 | return new ICmpInst(ICmpInst::ICMP_NE, Op0, Op1); | ||||
5948 | if (RHSVal == Max-1) // A >s MAX-1 -> A == MAX | ||||
5949 | return new ICmpInst(ICmpInst::ICMP_EQ, Op0, AddOne(CI)); | ||||
Chris Lattner | 62d0f23 | 2008-07-11 05:08:55 +0000 | [diff] [blame] | 5950 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5951 | } |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 5952 | } |
5953 | |||||
5954 | // Test if the ICmpInst instruction is used exclusively by a select as | ||||
5955 | // part of a minimum or maximum operation. If so, refrain from doing | ||||
5956 | // any other folding. This helps out other analyses which understand | ||||
5957 | // non-obfuscated minimum and maximum idioms, such as ScalarEvolution | ||||
5958 | // and CodeGen. And in this case, at least one of the comparison | ||||
5959 | // operands has at least one user besides the compare (the select), | ||||
5960 | // which would often largely negate the benefit of folding anyway. | ||||
5961 | if (I.hasOneUse()) | ||||
5962 | if (SelectInst *SI = dyn_cast<SelectInst>(*I.use_begin())) | ||||
5963 | if ((SI->getOperand(1) == Op0 && SI->getOperand(2) == Op1) || | ||||
5964 | (SI->getOperand(2) == Op0 && SI->getOperand(1) == Op1)) | ||||
5965 | return 0; | ||||
5966 | |||||
5967 | // See if we are doing a comparison between a constant and an instruction that | ||||
5968 | // can be folded into the comparison. | ||||
5969 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 5970 | // Since the RHS is a ConstantInt (CI), if the left hand side is an |
5971 | // instruction, see if that instruction also has constants so that the | ||||
5972 | // instruction can be folded into the icmp | ||||
5973 | if (Instruction *LHSI = dyn_cast<Instruction>(Op0)) | ||||
5974 | if (Instruction *Res = visitICmpInstWithInstAndIntCst(I, LHSI, CI)) | ||||
5975 | return Res; | ||||
5976 | } | ||||
5977 | |||||
5978 | // Handle icmp with constant (but not simple integer constant) RHS | ||||
5979 | if (Constant *RHSC = dyn_cast<Constant>(Op1)) { | ||||
5980 | if (Instruction *LHSI = dyn_cast<Instruction>(Op0)) | ||||
5981 | switch (LHSI->getOpcode()) { | ||||
5982 | case Instruction::GetElementPtr: | ||||
5983 | if (RHSC->isNullValue()) { | ||||
5984 | // icmp pred GEP (P, int 0, int 0, int 0), null -> icmp pred P, null | ||||
5985 | bool isAllZeros = true; | ||||
5986 | for (unsigned i = 1, e = LHSI->getNumOperands(); i != e; ++i) | ||||
5987 | if (!isa<Constant>(LHSI->getOperand(i)) || | ||||
5988 | !cast<Constant>(LHSI->getOperand(i))->isNullValue()) { | ||||
5989 | isAllZeros = false; | ||||
5990 | break; | ||||
5991 | } | ||||
5992 | if (isAllZeros) | ||||
5993 | return new ICmpInst(I.getPredicate(), LHSI->getOperand(0), | ||||
5994 | Constant::getNullValue(LHSI->getOperand(0)->getType())); | ||||
5995 | } | ||||
5996 | break; | ||||
5997 | |||||
5998 | case Instruction::PHI: | ||||
Chris Lattner | a2417ba | 2008-06-08 20:52:11 +0000 | [diff] [blame] | 5999 | // Only fold icmp into the PHI if the phi and fcmp are in the same |
6000 | // block. If in the same block, we're encouraging jump threading. If | ||||
6001 | // not, we are just pessimizing the code by making an i1 phi. | ||||
6002 | if (LHSI->getParent() == I.getParent()) | ||||
6003 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
6004 | return NV; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6005 | break; |
6006 | case Instruction::Select: { | ||||
6007 | // If either operand of the select is a constant, we can fold the | ||||
6008 | // comparison into the select arms, which will cause one to be | ||||
6009 | // constant folded and the select turned into a bitwise or. | ||||
6010 | Value *Op1 = 0, *Op2 = 0; | ||||
6011 | if (LHSI->hasOneUse()) { | ||||
6012 | if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(1))) { | ||||
6013 | // Fold the known value into the constant operand. | ||||
6014 | Op1 = ConstantExpr::getICmp(I.getPredicate(), C, RHSC); | ||||
6015 | // Insert a new ICmp of the other select operand. | ||||
6016 | Op2 = InsertNewInstBefore(new ICmpInst(I.getPredicate(), | ||||
6017 | LHSI->getOperand(2), RHSC, | ||||
6018 | I.getName()), I); | ||||
6019 | } else if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(2))) { | ||||
6020 | // Fold the known value into the constant operand. | ||||
6021 | Op2 = ConstantExpr::getICmp(I.getPredicate(), C, RHSC); | ||||
6022 | // Insert a new ICmp of the other select operand. | ||||
6023 | Op1 = InsertNewInstBefore(new ICmpInst(I.getPredicate(), | ||||
6024 | LHSI->getOperand(1), RHSC, | ||||
6025 | I.getName()), I); | ||||
6026 | } | ||||
6027 | } | ||||
6028 | |||||
6029 | if (Op1) | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 6030 | return SelectInst::Create(LHSI->getOperand(0), Op1, Op2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6031 | break; |
6032 | } | ||||
6033 | case Instruction::Malloc: | ||||
6034 | // If we have (malloc != null), and if the malloc has a single use, we | ||||
6035 | // can assume it is successful and remove the malloc. | ||||
6036 | if (LHSI->hasOneUse() && isa<ConstantPointerNull>(RHSC)) { | ||||
6037 | AddToWorkList(LHSI); | ||||
6038 | return ReplaceInstUsesWith(I, ConstantInt::get(Type::Int1Ty, | ||||
Nick Lewycky | 09284cf | 2008-05-17 07:33:39 +0000 | [diff] [blame] | 6039 | !I.isTrueWhenEqual())); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6040 | } |
6041 | break; | ||||
6042 | } | ||||
6043 | } | ||||
6044 | |||||
6045 | // If we can optimize a 'icmp GEP, P' or 'icmp P, GEP', do so now. | ||||
6046 | if (User *GEP = dyn_castGetElementPtr(Op0)) | ||||
6047 | if (Instruction *NI = FoldGEPICmp(GEP, Op1, I.getPredicate(), I)) | ||||
6048 | return NI; | ||||
6049 | if (User *GEP = dyn_castGetElementPtr(Op1)) | ||||
6050 | if (Instruction *NI = FoldGEPICmp(GEP, Op0, | ||||
6051 | ICmpInst::getSwappedPredicate(I.getPredicate()), I)) | ||||
6052 | return NI; | ||||
6053 | |||||
6054 | // Test to see if the operands of the icmp are casted versions of other | ||||
6055 | // values. If the ptr->ptr cast can be stripped off both arguments, we do so | ||||
6056 | // now. | ||||
6057 | if (BitCastInst *CI = dyn_cast<BitCastInst>(Op0)) { | ||||
6058 | if (isa<PointerType>(Op0->getType()) && | ||||
6059 | (isa<Constant>(Op1) || isa<BitCastInst>(Op1))) { | ||||
6060 | // We keep moving the cast from the left operand over to the right | ||||
6061 | // operand, where it can often be eliminated completely. | ||||
6062 | Op0 = CI->getOperand(0); | ||||
6063 | |||||
6064 | // If operand #1 is a bitcast instruction, it must also be a ptr->ptr cast | ||||
6065 | // so eliminate it as well. | ||||
6066 | if (BitCastInst *CI2 = dyn_cast<BitCastInst>(Op1)) | ||||
6067 | Op1 = CI2->getOperand(0); | ||||
6068 | |||||
6069 | // 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] | 6070 | if (Op0->getType() != Op1->getType()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6071 | if (Constant *Op1C = dyn_cast<Constant>(Op1)) { |
6072 | Op1 = ConstantExpr::getBitCast(Op1C, Op0->getType()); | ||||
6073 | } else { | ||||
6074 | // Otherwise, cast the RHS right before the icmp | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 6075 | Op1 = InsertBitCastBefore(Op1, Op0->getType(), I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6076 | } |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 6077 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6078 | return new ICmpInst(I.getPredicate(), Op0, Op1); |
6079 | } | ||||
6080 | } | ||||
6081 | |||||
6082 | if (isa<CastInst>(Op0)) { | ||||
6083 | // Handle the special case of: icmp (cast bool to X), <cst> | ||||
6084 | // This comes up when you have code like | ||||
6085 | // int X = A < B; | ||||
6086 | // if (X) ... | ||||
6087 | // For generality, we handle any zero-extension of any operand comparison | ||||
6088 | // with a constant or another cast from the same type. | ||||
6089 | if (isa<ConstantInt>(Op1) || isa<CastInst>(Op1)) | ||||
6090 | if (Instruction *R = visitICmpInstWithCastAndCast(I)) | ||||
6091 | return R; | ||||
6092 | } | ||||
6093 | |||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 6094 | // See if it's the same type of instruction on the left and right. |
6095 | if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) { | ||||
6096 | if (BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1)) { | ||||
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 6097 | if (Op0I->getOpcode() == Op1I->getOpcode() && Op0I->hasOneUse() && |
6098 | Op1I->hasOneUse() && Op0I->getOperand(1) == Op1I->getOperand(1) && | ||||
6099 | I.isEquality()) { | ||||
Nick Lewycky | cfadfbd | 2008-09-03 06:24:21 +0000 | [diff] [blame] | 6100 | switch (Op0I->getOpcode()) { |
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 6101 | default: break; |
6102 | case Instruction::Add: | ||||
6103 | case Instruction::Sub: | ||||
6104 | case Instruction::Xor: | ||||
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 6105 | // a+x icmp eq/ne b+x --> a icmp b |
6106 | return new ICmpInst(I.getPredicate(), Op0I->getOperand(0), | ||||
6107 | Op1I->getOperand(0)); | ||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 6108 | break; |
6109 | case Instruction::Mul: | ||||
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 6110 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) { |
6111 | // a * Cst icmp eq/ne b * Cst --> a & Mask icmp b & Mask | ||||
6112 | // Mask = -1 >> count-trailing-zeros(Cst). | ||||
6113 | if (!CI->isZero() && !CI->isOne()) { | ||||
6114 | const APInt &AP = CI->getValue(); | ||||
6115 | ConstantInt *Mask = ConstantInt::get( | ||||
6116 | APInt::getLowBitsSet(AP.getBitWidth(), | ||||
6117 | AP.getBitWidth() - | ||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 6118 | AP.countTrailingZeros())); |
Nick Lewycky | 58ecfb2 | 2008-08-21 05:56:10 +0000 | [diff] [blame] | 6119 | Instruction *And1 = BinaryOperator::CreateAnd(Op0I->getOperand(0), |
6120 | Mask); | ||||
6121 | Instruction *And2 = BinaryOperator::CreateAnd(Op1I->getOperand(0), | ||||
6122 | Mask); | ||||
6123 | InsertNewInstBefore(And1, I); | ||||
6124 | InsertNewInstBefore(And2, I); | ||||
6125 | return new ICmpInst(I.getPredicate(), And1, And2); | ||||
Nick Lewycky | d4c5ea0 | 2008-07-11 07:20:53 +0000 | [diff] [blame] | 6126 | } |
6127 | } | ||||
6128 | break; | ||||
6129 | } | ||||
6130 | } | ||||
6131 | } | ||||
6132 | } | ||||
6133 | |||||
Chris Lattner | a4e1eef | 2008-05-09 05:19:28 +0000 | [diff] [blame] | 6134 | // ~x < ~y --> y < x |
6135 | { Value *A, *B; | ||||
6136 | if (match(Op0, m_Not(m_Value(A))) && | ||||
6137 | match(Op1, m_Not(m_Value(B)))) | ||||
6138 | return new ICmpInst(I.getPredicate(), B, A); | ||||
6139 | } | ||||
6140 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6141 | if (I.isEquality()) { |
6142 | Value *A, *B, *C, *D; | ||||
Chris Lattner | a4e1eef | 2008-05-09 05:19:28 +0000 | [diff] [blame] | 6143 | |
6144 | // -x == -y --> x == y | ||||
6145 | if (match(Op0, m_Neg(m_Value(A))) && | ||||
6146 | match(Op1, m_Neg(m_Value(B)))) | ||||
6147 | return new ICmpInst(I.getPredicate(), A, B); | ||||
6148 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6149 | if (match(Op0, m_Xor(m_Value(A), m_Value(B)))) { |
6150 | if (A == Op1 || B == Op1) { // (A^B) == A -> B == 0 | ||||
6151 | Value *OtherVal = A == Op1 ? B : A; | ||||
6152 | return new ICmpInst(I.getPredicate(), OtherVal, | ||||
6153 | Constant::getNullValue(A->getType())); | ||||
6154 | } | ||||
6155 | |||||
6156 | if (match(Op1, m_Xor(m_Value(C), m_Value(D)))) { | ||||
6157 | // A^c1 == C^c2 --> A == C^(c1^c2) | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 6158 | ConstantInt *C1, *C2; |
6159 | if (match(B, m_ConstantInt(C1)) && | ||||
6160 | match(D, m_ConstantInt(C2)) && Op1->hasOneUse()) { | ||||
6161 | Constant *NC = ConstantInt::get(C1->getValue() ^ C2->getValue()); | ||||
6162 | Instruction *Xor = BinaryOperator::CreateXor(C, NC, "tmp"); | ||||
6163 | return new ICmpInst(I.getPredicate(), A, | ||||
6164 | InsertNewInstBefore(Xor, I)); | ||||
6165 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6166 | |
6167 | // A^B == A^D -> B == D | ||||
6168 | if (A == C) return new ICmpInst(I.getPredicate(), B, D); | ||||
6169 | if (A == D) return new ICmpInst(I.getPredicate(), B, C); | ||||
6170 | if (B == C) return new ICmpInst(I.getPredicate(), A, D); | ||||
6171 | if (B == D) return new ICmpInst(I.getPredicate(), A, C); | ||||
6172 | } | ||||
6173 | } | ||||
6174 | |||||
6175 | if (match(Op1, m_Xor(m_Value(A), m_Value(B))) && | ||||
6176 | (A == Op0 || B == Op0)) { | ||||
6177 | // A == (A^B) -> B == 0 | ||||
6178 | Value *OtherVal = A == Op0 ? B : A; | ||||
6179 | return new ICmpInst(I.getPredicate(), OtherVal, | ||||
6180 | Constant::getNullValue(A->getType())); | ||||
6181 | } | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 6182 | |
6183 | // (A-B) == A -> B == 0 | ||||
6184 | if (match(Op0, m_Sub(m_Specific(Op1), m_Value(B)))) | ||||
6185 | return new ICmpInst(I.getPredicate(), B, | ||||
6186 | Constant::getNullValue(B->getType())); | ||||
6187 | |||||
6188 | // A == (A-B) -> B == 0 | ||||
6189 | if (match(Op1, m_Sub(m_Specific(Op0), m_Value(B)))) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6190 | return new ICmpInst(I.getPredicate(), B, |
6191 | Constant::getNullValue(B->getType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6192 | |
6193 | // (X&Z) == (Y&Z) -> (X^Y) & Z == 0 | ||||
6194 | if (Op0->hasOneUse() && Op1->hasOneUse() && | ||||
6195 | match(Op0, m_And(m_Value(A), m_Value(B))) && | ||||
6196 | match(Op1, m_And(m_Value(C), m_Value(D)))) { | ||||
6197 | Value *X = 0, *Y = 0, *Z = 0; | ||||
6198 | |||||
6199 | if (A == C) { | ||||
6200 | X = B; Y = D; Z = A; | ||||
6201 | } else if (A == D) { | ||||
6202 | X = B; Y = C; Z = A; | ||||
6203 | } else if (B == C) { | ||||
6204 | X = A; Y = D; Z = B; | ||||
6205 | } else if (B == D) { | ||||
6206 | X = A; Y = C; Z = B; | ||||
6207 | } | ||||
6208 | |||||
6209 | if (X) { // Build (X^Y) & Z | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6210 | Op1 = InsertNewInstBefore(BinaryOperator::CreateXor(X, Y, "tmp"), I); |
6211 | Op1 = InsertNewInstBefore(BinaryOperator::CreateAnd(Op1, Z, "tmp"), I); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6212 | I.setOperand(0, Op1); |
6213 | I.setOperand(1, Constant::getNullValue(Op1->getType())); | ||||
6214 | return &I; | ||||
6215 | } | ||||
6216 | } | ||||
6217 | } | ||||
6218 | return Changed ? &I : 0; | ||||
6219 | } | ||||
6220 | |||||
6221 | |||||
6222 | /// FoldICmpDivCst - Fold "icmp pred, ([su]div X, DivRHS), CmpRHS" where DivRHS | ||||
6223 | /// and CmpRHS are both known to be integer constants. | ||||
6224 | Instruction *InstCombiner::FoldICmpDivCst(ICmpInst &ICI, BinaryOperator *DivI, | ||||
6225 | ConstantInt *DivRHS) { | ||||
6226 | ConstantInt *CmpRHS = cast<ConstantInt>(ICI.getOperand(1)); | ||||
6227 | const APInt &CmpRHSV = CmpRHS->getValue(); | ||||
6228 | |||||
6229 | // FIXME: If the operand types don't match the type of the divide | ||||
6230 | // then don't attempt this transform. The code below doesn't have the | ||||
6231 | // logic to deal with a signed divide and an unsigned compare (and | ||||
6232 | // vice versa). This is because (x /s C1) <s C2 produces different | ||||
6233 | // results than (x /s C1) <u C2 or (x /u C1) <s C2 or even | ||||
6234 | // (x /u C1) <u C2. Simply casting the operands and result won't | ||||
6235 | // work. :( The if statement below tests that condition and bails | ||||
6236 | // if it finds it. | ||||
6237 | bool DivIsSigned = DivI->getOpcode() == Instruction::SDiv; | ||||
6238 | if (!ICI.isEquality() && DivIsSigned != ICI.isSignedPredicate()) | ||||
6239 | return 0; | ||||
6240 | if (DivRHS->isZero()) | ||||
6241 | return 0; // The ProdOV computation fails on divide by zero. | ||||
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 6242 | if (DivIsSigned && DivRHS->isAllOnesValue()) |
6243 | return 0; // The overflow computation also screws up here | ||||
6244 | if (DivRHS->isOne()) | ||||
6245 | return 0; // Not worth bothering, and eliminates some funny cases | ||||
6246 | // with INT_MIN. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6247 | |
6248 | // Compute Prod = CI * DivRHS. We are essentially solving an equation | ||||
6249 | // of form X/C1=C2. We solve for X by multiplying C1 (DivRHS) and | ||||
6250 | // C2 (CI). By solving for X we can turn this into a range check | ||||
6251 | // instead of computing a divide. | ||||
6252 | ConstantInt *Prod = Multiply(CmpRHS, DivRHS); | ||||
6253 | |||||
6254 | // Determine if the product overflows by seeing if the product is | ||||
6255 | // not equal to the divide. Make sure we do the same kind of divide | ||||
6256 | // as in the LHS instruction that we're folding. | ||||
6257 | bool ProdOV = (DivIsSigned ? ConstantExpr::getSDiv(Prod, DivRHS) : | ||||
6258 | ConstantExpr::getUDiv(Prod, DivRHS)) != CmpRHS; | ||||
6259 | |||||
6260 | // Get the ICmp opcode | ||||
6261 | ICmpInst::Predicate Pred = ICI.getPredicate(); | ||||
6262 | |||||
6263 | // Figure out the interval that is being checked. For example, a comparison | ||||
6264 | // like "X /u 5 == 0" is really checking that X is in the interval [0, 5). | ||||
6265 | // Compute this interval based on the constants involved and the signedness of | ||||
6266 | // the compare/divide. This computes a half-open interval, keeping track of | ||||
6267 | // whether either value in the interval overflows. After analysis each | ||||
6268 | // overflow variable is set to 0 if it's corresponding bound variable is valid | ||||
6269 | // -1 if overflowed off the bottom end, or +1 if overflowed off the top end. | ||||
6270 | int LoOverflow = 0, HiOverflow = 0; | ||||
6271 | ConstantInt *LoBound = 0, *HiBound = 0; | ||||
6272 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6273 | if (!DivIsSigned) { // udiv |
6274 | // e.g. X/5 op 3 --> [15, 20) | ||||
6275 | LoBound = Prod; | ||||
6276 | HiOverflow = LoOverflow = ProdOV; | ||||
6277 | if (!HiOverflow) | ||||
6278 | HiOverflow = AddWithOverflow(HiBound, LoBound, DivRHS, false); | ||||
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 6279 | } else if (DivRHS->getValue().isStrictlyPositive()) { // Divisor is > 0. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6280 | if (CmpRHSV == 0) { // (X / pos) op 0 |
6281 | // Can't overflow. e.g. X/2 op 0 --> [-1, 2) | ||||
6282 | LoBound = cast<ConstantInt>(ConstantExpr::getNeg(SubOne(DivRHS))); | ||||
6283 | HiBound = DivRHS; | ||||
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 6284 | } else if (CmpRHSV.isStrictlyPositive()) { // (X / pos) op pos |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6285 | LoBound = Prod; // e.g. X/5 op 3 --> [15, 20) |
6286 | HiOverflow = LoOverflow = ProdOV; | ||||
6287 | if (!HiOverflow) | ||||
6288 | HiOverflow = AddWithOverflow(HiBound, Prod, DivRHS, true); | ||||
6289 | } else { // (X / pos) op neg | ||||
6290 | // e.g. X/5 op -3 --> [-15-4, -15+1) --> [-19, -14) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6291 | HiBound = AddOne(Prod); |
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 6292 | LoOverflow = HiOverflow = ProdOV ? -1 : 0; |
6293 | if (!LoOverflow) { | ||||
6294 | ConstantInt* DivNeg = cast<ConstantInt>(ConstantExpr::getNeg(DivRHS)); | ||||
6295 | LoOverflow = AddWithOverflow(LoBound, HiBound, DivNeg, | ||||
6296 | true) ? -1 : 0; | ||||
6297 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6298 | } |
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 6299 | } else if (DivRHS->getValue().isNegative()) { // Divisor is < 0. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6300 | if (CmpRHSV == 0) { // (X / neg) op 0 |
6301 | // e.g. X/-5 op 0 --> [-4, 5) | ||||
6302 | LoBound = AddOne(DivRHS); | ||||
6303 | HiBound = cast<ConstantInt>(ConstantExpr::getNeg(DivRHS)); | ||||
6304 | if (HiBound == DivRHS) { // -INTMIN = INTMIN | ||||
6305 | HiOverflow = 1; // [INTMIN+1, overflow) | ||||
6306 | HiBound = 0; // e.g. X/INTMIN = 0 --> X > INTMIN | ||||
6307 | } | ||||
Dan Gohman | 5dceed1 | 2008-02-13 22:09:18 +0000 | [diff] [blame] | 6308 | } else if (CmpRHSV.isStrictlyPositive()) { // (X / neg) op pos |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6309 | // e.g. X/-5 op 3 --> [-19, -14) |
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 6310 | HiBound = AddOne(Prod); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6311 | HiOverflow = LoOverflow = ProdOV ? -1 : 0; |
6312 | if (!LoOverflow) | ||||
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 6313 | LoOverflow = AddWithOverflow(LoBound, HiBound, DivRHS, true) ? -1 : 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6314 | } else { // (X / neg) op neg |
Chris Lattner | bd85a5f | 2008-10-11 22:55:00 +0000 | [diff] [blame] | 6315 | LoBound = Prod; // e.g. X/-5 op -3 --> [15, 20) |
6316 | LoOverflow = HiOverflow = ProdOV; | ||||
Dan Gohman | 45408ea | 2008-09-11 00:25:00 +0000 | [diff] [blame] | 6317 | if (!HiOverflow) |
6318 | HiOverflow = SubWithOverflow(HiBound, Prod, DivRHS, true); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6319 | } |
6320 | |||||
6321 | // Dividing by a negative swaps the condition. LT <-> GT | ||||
6322 | Pred = ICmpInst::getSwappedPredicate(Pred); | ||||
6323 | } | ||||
6324 | |||||
6325 | Value *X = DivI->getOperand(0); | ||||
6326 | switch (Pred) { | ||||
6327 | default: assert(0 && "Unhandled icmp opcode!"); | ||||
6328 | case ICmpInst::ICMP_EQ: | ||||
6329 | if (LoOverflow && HiOverflow) | ||||
6330 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6331 | else if (HiOverflow) | ||||
6332 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SGE : | ||||
6333 | ICmpInst::ICMP_UGE, X, LoBound); | ||||
6334 | else if (LoOverflow) | ||||
6335 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SLT : | ||||
6336 | ICmpInst::ICMP_ULT, X, HiBound); | ||||
6337 | else | ||||
6338 | return InsertRangeTest(X, LoBound, HiBound, DivIsSigned, true, ICI); | ||||
6339 | case ICmpInst::ICMP_NE: | ||||
6340 | if (LoOverflow && HiOverflow) | ||||
6341 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6342 | else if (HiOverflow) | ||||
6343 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SLT : | ||||
6344 | ICmpInst::ICMP_ULT, X, LoBound); | ||||
6345 | else if (LoOverflow) | ||||
6346 | return new ICmpInst(DivIsSigned ? ICmpInst::ICMP_SGE : | ||||
6347 | ICmpInst::ICMP_UGE, X, HiBound); | ||||
6348 | else | ||||
6349 | return InsertRangeTest(X, LoBound, HiBound, DivIsSigned, false, ICI); | ||||
6350 | case ICmpInst::ICMP_ULT: | ||||
6351 | case ICmpInst::ICMP_SLT: | ||||
6352 | if (LoOverflow == +1) // Low bound is greater than input range. | ||||
6353 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6354 | if (LoOverflow == -1) // Low bound is less than input range. | ||||
6355 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6356 | return new ICmpInst(Pred, X, LoBound); | ||||
6357 | case ICmpInst::ICMP_UGT: | ||||
6358 | case ICmpInst::ICMP_SGT: | ||||
6359 | if (HiOverflow == +1) // High bound greater than input range. | ||||
6360 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6361 | else if (HiOverflow == -1) // High bound less than input range. | ||||
6362 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6363 | if (Pred == ICmpInst::ICMP_UGT) | ||||
6364 | return new ICmpInst(ICmpInst::ICMP_UGE, X, HiBound); | ||||
6365 | else | ||||
6366 | return new ICmpInst(ICmpInst::ICMP_SGE, X, HiBound); | ||||
6367 | } | ||||
6368 | } | ||||
6369 | |||||
6370 | |||||
6371 | /// visitICmpInstWithInstAndIntCst - Handle "icmp (instr, intcst)". | ||||
6372 | /// | ||||
6373 | Instruction *InstCombiner::visitICmpInstWithInstAndIntCst(ICmpInst &ICI, | ||||
6374 | Instruction *LHSI, | ||||
6375 | ConstantInt *RHS) { | ||||
6376 | const APInt &RHSV = RHS->getValue(); | ||||
6377 | |||||
6378 | switch (LHSI->getOpcode()) { | ||||
Chris Lattner | 56be123 | 2009-01-09 07:47:06 +0000 | [diff] [blame] | 6379 | case Instruction::Trunc: |
6380 | if (ICI.isEquality() && LHSI->hasOneUse()) { | ||||
6381 | // Simplify icmp eq (trunc x to i8), 42 -> icmp eq x, 42|highbits if all | ||||
6382 | // of the high bits truncated out of x are known. | ||||
6383 | unsigned DstBits = LHSI->getType()->getPrimitiveSizeInBits(), | ||||
6384 | SrcBits = LHSI->getOperand(0)->getType()->getPrimitiveSizeInBits(); | ||||
6385 | APInt Mask(APInt::getHighBitsSet(SrcBits, SrcBits-DstBits)); | ||||
6386 | APInt KnownZero(SrcBits, 0), KnownOne(SrcBits, 0); | ||||
6387 | ComputeMaskedBits(LHSI->getOperand(0), Mask, KnownZero, KnownOne); | ||||
6388 | |||||
6389 | // If all the high bits are known, we can do this xform. | ||||
6390 | if ((KnownZero|KnownOne).countLeadingOnes() >= SrcBits-DstBits) { | ||||
6391 | // Pull in the high bits from known-ones set. | ||||
6392 | APInt NewRHS(RHS->getValue()); | ||||
6393 | NewRHS.zext(SrcBits); | ||||
6394 | NewRHS |= KnownOne; | ||||
6395 | return new ICmpInst(ICI.getPredicate(), LHSI->getOperand(0), | ||||
6396 | ConstantInt::get(NewRHS)); | ||||
6397 | } | ||||
6398 | } | ||||
6399 | break; | ||||
6400 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6401 | case Instruction::Xor: // (icmp pred (xor X, XorCST), CI) |
6402 | if (ConstantInt *XorCST = dyn_cast<ConstantInt>(LHSI->getOperand(1))) { | ||||
6403 | // If this is a comparison that tests the signbit (X < 0) or (x > -1), | ||||
6404 | // fold the xor. | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 6405 | if ((ICI.getPredicate() == ICmpInst::ICMP_SLT && RHSV == 0) || |
6406 | (ICI.getPredicate() == ICmpInst::ICMP_SGT && RHSV.isAllOnesValue())) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6407 | Value *CompareVal = LHSI->getOperand(0); |
6408 | |||||
6409 | // If the sign bit of the XorCST is not set, there is no change to | ||||
6410 | // the operation, just stop using the Xor. | ||||
6411 | if (!XorCST->getValue().isNegative()) { | ||||
6412 | ICI.setOperand(0, CompareVal); | ||||
6413 | AddToWorkList(LHSI); | ||||
6414 | return &ICI; | ||||
6415 | } | ||||
6416 | |||||
6417 | // Was the old condition true if the operand is positive? | ||||
6418 | bool isTrueIfPositive = ICI.getPredicate() == ICmpInst::ICMP_SGT; | ||||
6419 | |||||
6420 | // If so, the new one isn't. | ||||
6421 | isTrueIfPositive ^= true; | ||||
6422 | |||||
6423 | if (isTrueIfPositive) | ||||
6424 | return new ICmpInst(ICmpInst::ICMP_SGT, CompareVal, SubOne(RHS)); | ||||
6425 | else | ||||
6426 | return new ICmpInst(ICmpInst::ICMP_SLT, CompareVal, AddOne(RHS)); | ||||
6427 | } | ||||
6428 | } | ||||
6429 | break; | ||||
6430 | case Instruction::And: // (icmp pred (and X, AndCST), RHS) | ||||
6431 | if (LHSI->hasOneUse() && isa<ConstantInt>(LHSI->getOperand(1)) && | ||||
6432 | LHSI->getOperand(0)->hasOneUse()) { | ||||
6433 | ConstantInt *AndCST = cast<ConstantInt>(LHSI->getOperand(1)); | ||||
6434 | |||||
6435 | // If the LHS is an AND of a truncating cast, we can widen the | ||||
6436 | // and/compare to be the input width without changing the value | ||||
6437 | // produced, eliminating a cast. | ||||
6438 | if (TruncInst *Cast = dyn_cast<TruncInst>(LHSI->getOperand(0))) { | ||||
6439 | // We can do this transformation if either the AND constant does not | ||||
6440 | // have its sign bit set or if it is an equality comparison. | ||||
6441 | // Extending a relational comparison when we're checking the sign | ||||
6442 | // bit would not work. | ||||
6443 | if (Cast->hasOneUse() && | ||||
Anton Korobeynikov | 6a4a933 | 2008-02-20 12:07:57 +0000 | [diff] [blame] | 6444 | (ICI.isEquality() || |
6445 | (AndCST->getValue().isNonNegative() && RHSV.isNonNegative()))) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6446 | uint32_t BitWidth = |
6447 | cast<IntegerType>(Cast->getOperand(0)->getType())->getBitWidth(); | ||||
6448 | APInt NewCST = AndCST->getValue(); | ||||
6449 | NewCST.zext(BitWidth); | ||||
6450 | APInt NewCI = RHSV; | ||||
6451 | NewCI.zext(BitWidth); | ||||
6452 | Instruction *NewAnd = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6453 | BinaryOperator::CreateAnd(Cast->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6454 | ConstantInt::get(NewCST),LHSI->getName()); |
6455 | InsertNewInstBefore(NewAnd, ICI); | ||||
6456 | return new ICmpInst(ICI.getPredicate(), NewAnd, | ||||
6457 | ConstantInt::get(NewCI)); | ||||
6458 | } | ||||
6459 | } | ||||
6460 | |||||
6461 | // If this is: (X >> C1) & C2 != C3 (where any shift and any compare | ||||
6462 | // could exist), turn it into (X & (C2 << C1)) != (C3 << C1). This | ||||
6463 | // happens a LOT in code produced by the C front-end, for bitfield | ||||
6464 | // access. | ||||
6465 | BinaryOperator *Shift = dyn_cast<BinaryOperator>(LHSI->getOperand(0)); | ||||
6466 | if (Shift && !Shift->isShift()) | ||||
6467 | Shift = 0; | ||||
6468 | |||||
6469 | ConstantInt *ShAmt; | ||||
6470 | ShAmt = Shift ? dyn_cast<ConstantInt>(Shift->getOperand(1)) : 0; | ||||
6471 | const Type *Ty = Shift ? Shift->getType() : 0; // Type of the shift. | ||||
6472 | const Type *AndTy = AndCST->getType(); // Type of the and. | ||||
6473 | |||||
6474 | // We can fold this as long as we can't shift unknown bits | ||||
6475 | // into the mask. This can only happen with signed shift | ||||
6476 | // rights, as they sign-extend. | ||||
6477 | if (ShAmt) { | ||||
6478 | bool CanFold = Shift->isLogicalShift(); | ||||
6479 | if (!CanFold) { | ||||
6480 | // To test for the bad case of the signed shr, see if any | ||||
6481 | // of the bits shifted in could be tested after the mask. | ||||
6482 | uint32_t TyBits = Ty->getPrimitiveSizeInBits(); | ||||
6483 | int ShAmtVal = TyBits - ShAmt->getLimitedValue(TyBits); | ||||
6484 | |||||
6485 | uint32_t BitWidth = AndTy->getPrimitiveSizeInBits(); | ||||
6486 | if ((APInt::getHighBitsSet(BitWidth, BitWidth-ShAmtVal) & | ||||
6487 | AndCST->getValue()) == 0) | ||||
6488 | CanFold = true; | ||||
6489 | } | ||||
6490 | |||||
6491 | if (CanFold) { | ||||
6492 | Constant *NewCst; | ||||
6493 | if (Shift->getOpcode() == Instruction::Shl) | ||||
6494 | NewCst = ConstantExpr::getLShr(RHS, ShAmt); | ||||
6495 | else | ||||
6496 | NewCst = ConstantExpr::getShl(RHS, ShAmt); | ||||
6497 | |||||
6498 | // Check to see if we are shifting out any of the bits being | ||||
6499 | // compared. | ||||
6500 | if (ConstantExpr::get(Shift->getOpcode(), NewCst, ShAmt) != RHS) { | ||||
6501 | // If we shifted bits out, the fold is not going to work out. | ||||
6502 | // As a special case, check to see if this means that the | ||||
6503 | // result is always true or false now. | ||||
6504 | if (ICI.getPredicate() == ICmpInst::ICMP_EQ) | ||||
6505 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6506 | if (ICI.getPredicate() == ICmpInst::ICMP_NE) | ||||
6507 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6508 | } else { | ||||
6509 | ICI.setOperand(1, NewCst); | ||||
6510 | Constant *NewAndCST; | ||||
6511 | if (Shift->getOpcode() == Instruction::Shl) | ||||
6512 | NewAndCST = ConstantExpr::getLShr(AndCST, ShAmt); | ||||
6513 | else | ||||
6514 | NewAndCST = ConstantExpr::getShl(AndCST, ShAmt); | ||||
6515 | LHSI->setOperand(1, NewAndCST); | ||||
6516 | LHSI->setOperand(0, Shift->getOperand(0)); | ||||
6517 | AddToWorkList(Shift); // Shift is dead. | ||||
6518 | AddUsesToWorkList(ICI); | ||||
6519 | return &ICI; | ||||
6520 | } | ||||
6521 | } | ||||
6522 | } | ||||
6523 | |||||
6524 | // Turn ((X >> Y) & C) == 0 into (X & (C << Y)) == 0. The later is | ||||
6525 | // preferable because it allows the C<<Y expression to be hoisted out | ||||
6526 | // of a loop if Y is invariant and X is not. | ||||
6527 | if (Shift && Shift->hasOneUse() && RHSV == 0 && | ||||
6528 | ICI.isEquality() && !Shift->isArithmeticShift() && | ||||
6529 | isa<Instruction>(Shift->getOperand(0))) { | ||||
6530 | // Compute C << Y. | ||||
6531 | Value *NS; | ||||
6532 | if (Shift->getOpcode() == Instruction::LShr) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6533 | NS = BinaryOperator::CreateShl(AndCST, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6534 | Shift->getOperand(1), "tmp"); |
6535 | } else { | ||||
6536 | // Insert a logical shift. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6537 | NS = BinaryOperator::CreateLShr(AndCST, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6538 | Shift->getOperand(1), "tmp"); |
6539 | } | ||||
6540 | InsertNewInstBefore(cast<Instruction>(NS), ICI); | ||||
6541 | |||||
6542 | // Compute X & (C << Y). | ||||
6543 | Instruction *NewAnd = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6544 | BinaryOperator::CreateAnd(Shift->getOperand(0), NS, LHSI->getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6545 | InsertNewInstBefore(NewAnd, ICI); |
6546 | |||||
6547 | ICI.setOperand(0, NewAnd); | ||||
6548 | return &ICI; | ||||
6549 | } | ||||
6550 | } | ||||
6551 | break; | ||||
6552 | |||||
6553 | case Instruction::Shl: { // (icmp pred (shl X, ShAmt), CI) | ||||
6554 | ConstantInt *ShAmt = dyn_cast<ConstantInt>(LHSI->getOperand(1)); | ||||
6555 | if (!ShAmt) break; | ||||
6556 | |||||
6557 | uint32_t TypeBits = RHSV.getBitWidth(); | ||||
6558 | |||||
6559 | // Check that the shift amount is in range. If not, don't perform | ||||
6560 | // undefined shifts. When the shift is visited it will be | ||||
6561 | // simplified. | ||||
6562 | if (ShAmt->uge(TypeBits)) | ||||
6563 | break; | ||||
6564 | |||||
6565 | if (ICI.isEquality()) { | ||||
6566 | // If we are comparing against bits always shifted out, the | ||||
6567 | // comparison cannot succeed. | ||||
6568 | Constant *Comp = | ||||
6569 | ConstantExpr::getShl(ConstantExpr::getLShr(RHS, ShAmt), ShAmt); | ||||
6570 | if (Comp != RHS) {// Comparing against a bit that we know is zero. | ||||
6571 | bool IsICMP_NE = ICI.getPredicate() == ICmpInst::ICMP_NE; | ||||
6572 | Constant *Cst = ConstantInt::get(Type::Int1Ty, IsICMP_NE); | ||||
6573 | return ReplaceInstUsesWith(ICI, Cst); | ||||
6574 | } | ||||
6575 | |||||
6576 | if (LHSI->hasOneUse()) { | ||||
6577 | // Otherwise strength reduce the shift into an and. | ||||
6578 | uint32_t ShAmtVal = (uint32_t)ShAmt->getLimitedValue(TypeBits); | ||||
6579 | Constant *Mask = | ||||
6580 | ConstantInt::get(APInt::getLowBitsSet(TypeBits, TypeBits-ShAmtVal)); | ||||
6581 | |||||
6582 | Instruction *AndI = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6583 | BinaryOperator::CreateAnd(LHSI->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6584 | Mask, LHSI->getName()+".mask"); |
6585 | Value *And = InsertNewInstBefore(AndI, ICI); | ||||
6586 | return new ICmpInst(ICI.getPredicate(), And, | ||||
6587 | ConstantInt::get(RHSV.lshr(ShAmtVal))); | ||||
6588 | } | ||||
6589 | } | ||||
6590 | |||||
6591 | // Otherwise, if this is a comparison of the sign bit, simplify to and/test. | ||||
6592 | bool TrueIfSigned = false; | ||||
6593 | if (LHSI->hasOneUse() && | ||||
6594 | isSignBitCheck(ICI.getPredicate(), RHS, TrueIfSigned)) { | ||||
6595 | // (X << 31) <s 0 --> (X&1) != 0 | ||||
6596 | Constant *Mask = ConstantInt::get(APInt(TypeBits, 1) << | ||||
6597 | (TypeBits-ShAmt->getZExtValue()-1)); | ||||
6598 | Instruction *AndI = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6599 | BinaryOperator::CreateAnd(LHSI->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6600 | Mask, LHSI->getName()+".mask"); |
6601 | Value *And = InsertNewInstBefore(AndI, ICI); | ||||
6602 | |||||
6603 | return new ICmpInst(TrueIfSigned ? ICmpInst::ICMP_NE : ICmpInst::ICMP_EQ, | ||||
6604 | And, Constant::getNullValue(And->getType())); | ||||
6605 | } | ||||
6606 | break; | ||||
6607 | } | ||||
6608 | |||||
6609 | case Instruction::LShr: // (icmp pred (shr X, ShAmt), CI) | ||||
6610 | case Instruction::AShr: { | ||||
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6611 | // Only handle equality comparisons of shift-by-constant. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6612 | ConstantInt *ShAmt = dyn_cast<ConstantInt>(LHSI->getOperand(1)); |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6613 | if (!ShAmt || !ICI.isEquality()) break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6614 | |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6615 | // Check that the shift amount is in range. If not, don't perform |
6616 | // undefined shifts. When the shift is visited it will be | ||||
6617 | // simplified. | ||||
6618 | uint32_t TypeBits = RHSV.getBitWidth(); | ||||
6619 | if (ShAmt->uge(TypeBits)) | ||||
6620 | break; | ||||
6621 | |||||
6622 | uint32_t ShAmtVal = (uint32_t)ShAmt->getLimitedValue(TypeBits); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6623 | |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6624 | // If we are comparing against bits always shifted out, the |
6625 | // comparison cannot succeed. | ||||
6626 | APInt Comp = RHSV << ShAmtVal; | ||||
6627 | if (LHSI->getOpcode() == Instruction::LShr) | ||||
6628 | Comp = Comp.lshr(ShAmtVal); | ||||
6629 | else | ||||
6630 | Comp = Comp.ashr(ShAmtVal); | ||||
6631 | |||||
6632 | if (Comp != RHSV) { // Comparing against a bit that we know is zero. | ||||
6633 | bool IsICMP_NE = ICI.getPredicate() == ICmpInst::ICMP_NE; | ||||
6634 | Constant *Cst = ConstantInt::get(Type::Int1Ty, IsICMP_NE); | ||||
6635 | return ReplaceInstUsesWith(ICI, Cst); | ||||
6636 | } | ||||
6637 | |||||
6638 | // Otherwise, check to see if the bits shifted out are known to be zero. | ||||
6639 | // If so, we can compare against the unshifted value: | ||||
6640 | // (X & 4) >> 1 == 2 --> (X & 4) == 4. | ||||
Evan Cheng | fb9292a | 2008-04-23 00:38:06 +0000 | [diff] [blame] | 6641 | if (LHSI->hasOneUse() && |
6642 | MaskedValueIsZero(LHSI->getOperand(0), | ||||
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6643 | APInt::getLowBitsSet(Comp.getBitWidth(), ShAmtVal))) { |
6644 | return new ICmpInst(ICI.getPredicate(), LHSI->getOperand(0), | ||||
6645 | ConstantExpr::getShl(RHS, ShAmt)); | ||||
6646 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6647 | |
Evan Cheng | fb9292a | 2008-04-23 00:38:06 +0000 | [diff] [blame] | 6648 | if (LHSI->hasOneUse()) { |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6649 | // Otherwise strength reduce the shift into an and. |
6650 | APInt Val(APInt::getHighBitsSet(TypeBits, TypeBits - ShAmtVal)); | ||||
6651 | Constant *Mask = ConstantInt::get(Val); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6652 | |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6653 | Instruction *AndI = |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6654 | BinaryOperator::CreateAnd(LHSI->getOperand(0), |
Chris Lattner | 5ee84f8 | 2008-03-21 05:19:58 +0000 | [diff] [blame] | 6655 | Mask, LHSI->getName()+".mask"); |
6656 | Value *And = InsertNewInstBefore(AndI, ICI); | ||||
6657 | return new ICmpInst(ICI.getPredicate(), And, | ||||
6658 | ConstantExpr::getShl(RHS, ShAmt)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6659 | } |
6660 | break; | ||||
6661 | } | ||||
6662 | |||||
6663 | case Instruction::SDiv: | ||||
6664 | case Instruction::UDiv: | ||||
6665 | // Fold: icmp pred ([us]div X, C1), C2 -> range test | ||||
6666 | // Fold this div into the comparison, producing a range check. | ||||
6667 | // Determine, based on the divide type, what the range is being | ||||
6668 | // checked. If there is an overflow on the low or high side, remember | ||||
6669 | // it, otherwise compute the range [low, hi) bounding the new value. | ||||
6670 | // See: InsertRangeTest above for the kinds of replacements possible. | ||||
6671 | if (ConstantInt *DivRHS = dyn_cast<ConstantInt>(LHSI->getOperand(1))) | ||||
6672 | if (Instruction *R = FoldICmpDivCst(ICI, cast<BinaryOperator>(LHSI), | ||||
6673 | DivRHS)) | ||||
6674 | return R; | ||||
6675 | break; | ||||
Nick Lewycky | 0185bbf | 2008-02-03 16:33:09 +0000 | [diff] [blame] | 6676 | |
6677 | case Instruction::Add: | ||||
6678 | // Fold: icmp pred (add, X, C1), C2 | ||||
6679 | |||||
6680 | if (!ICI.isEquality()) { | ||||
6681 | ConstantInt *LHSC = dyn_cast<ConstantInt>(LHSI->getOperand(1)); | ||||
6682 | if (!LHSC) break; | ||||
6683 | const APInt &LHSV = LHSC->getValue(); | ||||
6684 | |||||
6685 | ConstantRange CR = ICI.makeConstantRange(ICI.getPredicate(), RHSV) | ||||
6686 | .subtract(LHSV); | ||||
6687 | |||||
6688 | if (ICI.isSignedPredicate()) { | ||||
6689 | if (CR.getLower().isSignBit()) { | ||||
6690 | return new ICmpInst(ICmpInst::ICMP_SLT, LHSI->getOperand(0), | ||||
6691 | ConstantInt::get(CR.getUpper())); | ||||
6692 | } else if (CR.getUpper().isSignBit()) { | ||||
6693 | return new ICmpInst(ICmpInst::ICMP_SGE, LHSI->getOperand(0), | ||||
6694 | ConstantInt::get(CR.getLower())); | ||||
6695 | } | ||||
6696 | } else { | ||||
6697 | if (CR.getLower().isMinValue()) { | ||||
6698 | return new ICmpInst(ICmpInst::ICMP_ULT, LHSI->getOperand(0), | ||||
6699 | ConstantInt::get(CR.getUpper())); | ||||
6700 | } else if (CR.getUpper().isMinValue()) { | ||||
6701 | return new ICmpInst(ICmpInst::ICMP_UGE, LHSI->getOperand(0), | ||||
6702 | ConstantInt::get(CR.getLower())); | ||||
6703 | } | ||||
6704 | } | ||||
6705 | } | ||||
6706 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6707 | } |
6708 | |||||
6709 | // Simplify icmp_eq and icmp_ne instructions with integer constant RHS. | ||||
6710 | if (ICI.isEquality()) { | ||||
6711 | bool isICMP_NE = ICI.getPredicate() == ICmpInst::ICMP_NE; | ||||
6712 | |||||
6713 | // If the first operand is (add|sub|and|or|xor|rem) with a constant, and | ||||
6714 | // the second operand is a constant, simplify a bit. | ||||
6715 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(LHSI)) { | ||||
6716 | switch (BO->getOpcode()) { | ||||
6717 | case Instruction::SRem: | ||||
6718 | // If we have a signed (X % (2^c)) == 0, turn it into an unsigned one. | ||||
6719 | if (RHSV == 0 && isa<ConstantInt>(BO->getOperand(1)) &&BO->hasOneUse()){ | ||||
6720 | const APInt &V = cast<ConstantInt>(BO->getOperand(1))->getValue(); | ||||
6721 | if (V.sgt(APInt(V.getBitWidth(), 1)) && V.isPowerOf2()) { | ||||
6722 | Instruction *NewRem = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6723 | BinaryOperator::CreateURem(BO->getOperand(0), BO->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6724 | BO->getName()); |
6725 | InsertNewInstBefore(NewRem, ICI); | ||||
6726 | return new ICmpInst(ICI.getPredicate(), NewRem, | ||||
6727 | Constant::getNullValue(BO->getType())); | ||||
6728 | } | ||||
6729 | } | ||||
6730 | break; | ||||
6731 | case Instruction::Add: | ||||
6732 | // Replace ((add A, B) != C) with (A != C-B) if B & C are constants. | ||||
6733 | if (ConstantInt *BOp1C = dyn_cast<ConstantInt>(BO->getOperand(1))) { | ||||
6734 | if (BO->hasOneUse()) | ||||
6735 | return new ICmpInst(ICI.getPredicate(), BO->getOperand(0), | ||||
6736 | Subtract(RHS, BOp1C)); | ||||
6737 | } else if (RHSV == 0) { | ||||
6738 | // Replace ((add A, B) != 0) with (A != -B) if A or B is | ||||
6739 | // efficiently invertible, or if the add has just this one use. | ||||
6740 | Value *BOp0 = BO->getOperand(0), *BOp1 = BO->getOperand(1); | ||||
6741 | |||||
6742 | if (Value *NegVal = dyn_castNegVal(BOp1)) | ||||
6743 | return new ICmpInst(ICI.getPredicate(), BOp0, NegVal); | ||||
6744 | else if (Value *NegVal = dyn_castNegVal(BOp0)) | ||||
6745 | return new ICmpInst(ICI.getPredicate(), NegVal, BOp1); | ||||
6746 | else if (BO->hasOneUse()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6747 | Instruction *Neg = BinaryOperator::CreateNeg(BOp1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6748 | InsertNewInstBefore(Neg, ICI); |
6749 | Neg->takeName(BO); | ||||
6750 | return new ICmpInst(ICI.getPredicate(), BOp0, Neg); | ||||
6751 | } | ||||
6752 | } | ||||
6753 | break; | ||||
6754 | case Instruction::Xor: | ||||
6755 | // For the xor case, we can xor two constants together, eliminating | ||||
6756 | // the explicit xor. | ||||
6757 | if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1))) | ||||
6758 | return new ICmpInst(ICI.getPredicate(), BO->getOperand(0), | ||||
6759 | ConstantExpr::getXor(RHS, BOC)); | ||||
6760 | |||||
6761 | // FALLTHROUGH | ||||
6762 | case Instruction::Sub: | ||||
6763 | // Replace (([sub|xor] A, B) != 0) with (A != B) | ||||
6764 | if (RHSV == 0) | ||||
6765 | return new ICmpInst(ICI.getPredicate(), BO->getOperand(0), | ||||
6766 | BO->getOperand(1)); | ||||
6767 | break; | ||||
6768 | |||||
6769 | case Instruction::Or: | ||||
6770 | // If bits are being or'd in that are not present in the constant we | ||||
6771 | // are comparing against, then the comparison could never succeed! | ||||
6772 | if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1))) { | ||||
6773 | Constant *NotCI = ConstantExpr::getNot(RHS); | ||||
6774 | if (!ConstantExpr::getAnd(BOC, NotCI)->isNullValue()) | ||||
6775 | return ReplaceInstUsesWith(ICI, ConstantInt::get(Type::Int1Ty, | ||||
6776 | isICMP_NE)); | ||||
6777 | } | ||||
6778 | break; | ||||
6779 | |||||
6780 | case Instruction::And: | ||||
6781 | if (ConstantInt *BOC = dyn_cast<ConstantInt>(BO->getOperand(1))) { | ||||
6782 | // If bits are being compared against that are and'd out, then the | ||||
6783 | // comparison can never succeed! | ||||
6784 | if ((RHSV & ~BOC->getValue()) != 0) | ||||
6785 | return ReplaceInstUsesWith(ICI, ConstantInt::get(Type::Int1Ty, | ||||
6786 | isICMP_NE)); | ||||
6787 | |||||
6788 | // If we have ((X & C) == C), turn it into ((X & C) != 0). | ||||
6789 | if (RHS == BOC && RHSV.isPowerOf2()) | ||||
6790 | return new ICmpInst(isICMP_NE ? ICmpInst::ICMP_EQ : | ||||
6791 | ICmpInst::ICMP_NE, LHSI, | ||||
6792 | Constant::getNullValue(RHS->getType())); | ||||
6793 | |||||
6794 | // Replace (and X, (1 << size(X)-1) != 0) with x s< 0 | ||||
Chris Lattner | 60813c2 | 2008-06-02 01:29:46 +0000 | [diff] [blame] | 6795 | if (BOC->getValue().isSignBit()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6796 | Value *X = BO->getOperand(0); |
6797 | Constant *Zero = Constant::getNullValue(X->getType()); | ||||
6798 | ICmpInst::Predicate pred = isICMP_NE ? | ||||
6799 | ICmpInst::ICMP_SLT : ICmpInst::ICMP_SGE; | ||||
6800 | return new ICmpInst(pred, X, Zero); | ||||
6801 | } | ||||
6802 | |||||
6803 | // ((X & ~7) == 0) --> X < 8 | ||||
6804 | if (RHSV == 0 && isHighOnes(BOC)) { | ||||
6805 | Value *X = BO->getOperand(0); | ||||
6806 | Constant *NegX = ConstantExpr::getNeg(BOC); | ||||
6807 | ICmpInst::Predicate pred = isICMP_NE ? | ||||
6808 | ICmpInst::ICMP_UGE : ICmpInst::ICMP_ULT; | ||||
6809 | return new ICmpInst(pred, X, NegX); | ||||
6810 | } | ||||
6811 | } | ||||
6812 | default: break; | ||||
6813 | } | ||||
6814 | } else if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(LHSI)) { | ||||
6815 | // Handle icmp {eq|ne} <intrinsic>, intcst. | ||||
6816 | if (II->getIntrinsicID() == Intrinsic::bswap) { | ||||
6817 | AddToWorkList(II); | ||||
6818 | ICI.setOperand(0, II->getOperand(1)); | ||||
6819 | ICI.setOperand(1, ConstantInt::get(RHSV.byteSwap())); | ||||
6820 | return &ICI; | ||||
6821 | } | ||||
6822 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6823 | } |
6824 | return 0; | ||||
6825 | } | ||||
6826 | |||||
6827 | /// visitICmpInstWithCastAndCast - Handle icmp (cast x to y), (cast/cst). | ||||
6828 | /// We only handle extending casts so far. | ||||
6829 | /// | ||||
6830 | Instruction *InstCombiner::visitICmpInstWithCastAndCast(ICmpInst &ICI) { | ||||
6831 | const CastInst *LHSCI = cast<CastInst>(ICI.getOperand(0)); | ||||
6832 | Value *LHSCIOp = LHSCI->getOperand(0); | ||||
6833 | const Type *SrcTy = LHSCIOp->getType(); | ||||
6834 | const Type *DestTy = LHSCI->getType(); | ||||
6835 | Value *RHSCIOp; | ||||
6836 | |||||
6837 | // Turn icmp (ptrtoint x), (ptrtoint/c) into a compare of the input if the | ||||
6838 | // integer type is the same size as the pointer type. | ||||
6839 | if (LHSCI->getOpcode() == Instruction::PtrToInt && | ||||
6840 | getTargetData().getPointerSizeInBits() == | ||||
6841 | cast<IntegerType>(DestTy)->getBitWidth()) { | ||||
6842 | Value *RHSOp = 0; | ||||
6843 | if (Constant *RHSC = dyn_cast<Constant>(ICI.getOperand(1))) { | ||||
6844 | RHSOp = ConstantExpr::getIntToPtr(RHSC, SrcTy); | ||||
6845 | } else if (PtrToIntInst *RHSC = dyn_cast<PtrToIntInst>(ICI.getOperand(1))) { | ||||
6846 | RHSOp = RHSC->getOperand(0); | ||||
6847 | // If the pointer types don't match, insert a bitcast. | ||||
6848 | if (LHSCIOp->getType() != RHSOp->getType()) | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 6849 | RHSOp = InsertBitCastBefore(RHSOp, LHSCIOp->getType(), ICI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6850 | } |
6851 | |||||
6852 | if (RHSOp) | ||||
6853 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, RHSOp); | ||||
6854 | } | ||||
6855 | |||||
6856 | // The code below only handles extension cast instructions, so far. | ||||
6857 | // Enforce this. | ||||
6858 | if (LHSCI->getOpcode() != Instruction::ZExt && | ||||
6859 | LHSCI->getOpcode() != Instruction::SExt) | ||||
6860 | return 0; | ||||
6861 | |||||
6862 | bool isSignedExt = LHSCI->getOpcode() == Instruction::SExt; | ||||
6863 | bool isSignedCmp = ICI.isSignedPredicate(); | ||||
6864 | |||||
6865 | if (CastInst *CI = dyn_cast<CastInst>(ICI.getOperand(1))) { | ||||
6866 | // Not an extension from the same type? | ||||
6867 | RHSCIOp = CI->getOperand(0); | ||||
6868 | if (RHSCIOp->getType() != LHSCIOp->getType()) | ||||
6869 | return 0; | ||||
6870 | |||||
Nick Lewycky | d4264dc | 2008-01-28 03:48:02 +0000 | [diff] [blame] | 6871 | // 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] | 6872 | // and the other is a zext), then we can't handle this. |
6873 | if (CI->getOpcode() != LHSCI->getOpcode()) | ||||
6874 | return 0; | ||||
6875 | |||||
Nick Lewycky | d4264dc | 2008-01-28 03:48:02 +0000 | [diff] [blame] | 6876 | // Deal with equality cases early. |
6877 | if (ICI.isEquality()) | ||||
6878 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, RHSCIOp); | ||||
6879 | |||||
6880 | // A signed comparison of sign extended values simplifies into a | ||||
6881 | // signed comparison. | ||||
6882 | if (isSignedCmp && isSignedExt) | ||||
6883 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, RHSCIOp); | ||||
6884 | |||||
6885 | // The other three cases all fold into an unsigned comparison. | ||||
6886 | return new ICmpInst(ICI.getUnsignedPredicate(), LHSCIOp, RHSCIOp); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6887 | } |
6888 | |||||
6889 | // If we aren't dealing with a constant on the RHS, exit early | ||||
6890 | ConstantInt *CI = dyn_cast<ConstantInt>(ICI.getOperand(1)); | ||||
6891 | if (!CI) | ||||
6892 | return 0; | ||||
6893 | |||||
6894 | // Compute the constant that would happen if we truncated to SrcTy then | ||||
6895 | // reextended to DestTy. | ||||
6896 | Constant *Res1 = ConstantExpr::getTrunc(CI, SrcTy); | ||||
6897 | Constant *Res2 = ConstantExpr::getCast(LHSCI->getOpcode(), Res1, DestTy); | ||||
6898 | |||||
6899 | // If the re-extended constant didn't change... | ||||
6900 | if (Res2 == CI) { | ||||
6901 | // Make sure that sign of the Cmp and the sign of the Cast are the same. | ||||
6902 | // For example, we might have: | ||||
6903 | // %A = sext short %X to uint | ||||
6904 | // %B = icmp ugt uint %A, 1330 | ||||
6905 | // It is incorrect to transform this into | ||||
6906 | // %B = icmp ugt short %X, 1330 | ||||
6907 | // because %A may have negative value. | ||||
6908 | // | ||||
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 6909 | // However, we allow this when the compare is EQ/NE, because they are |
6910 | // signless. | ||||
6911 | if (isSignedExt == isSignedCmp || ICI.isEquality()) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6912 | return new ICmpInst(ICI.getPredicate(), LHSCIOp, Res1); |
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 6913 | return 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6914 | } |
6915 | |||||
6916 | // The re-extended constant changed so the constant cannot be represented | ||||
6917 | // in the shorter type. Consequently, we cannot emit a simple comparison. | ||||
6918 | |||||
6919 | // First, handle some easy cases. We know the result cannot be equal at this | ||||
6920 | // point so handle the ICI.isEquality() cases | ||||
6921 | if (ICI.getPredicate() == ICmpInst::ICMP_EQ) | ||||
6922 | return ReplaceInstUsesWith(ICI, ConstantInt::getFalse()); | ||||
6923 | if (ICI.getPredicate() == ICmpInst::ICMP_NE) | ||||
6924 | return ReplaceInstUsesWith(ICI, ConstantInt::getTrue()); | ||||
6925 | |||||
6926 | // Evaluate the comparison for LT (we invert for GT below). LE and GE cases | ||||
6927 | // should have been folded away previously and not enter in here. | ||||
6928 | Value *Result; | ||||
6929 | if (isSignedCmp) { | ||||
6930 | // We're performing a signed comparison. | ||||
6931 | if (cast<ConstantInt>(CI)->getValue().isNegative()) | ||||
6932 | Result = ConstantInt::getFalse(); // X < (small) --> false | ||||
6933 | else | ||||
6934 | Result = ConstantInt::getTrue(); // X < (large) --> true | ||||
6935 | } else { | ||||
6936 | // We're performing an unsigned comparison. | ||||
6937 | if (isSignedExt) { | ||||
6938 | // We're performing an unsigned comp with a sign extended value. | ||||
6939 | // This is true if the input is >= 0. [aka >s -1] | ||||
6940 | Constant *NegOne = ConstantInt::getAllOnesValue(SrcTy); | ||||
6941 | Result = InsertNewInstBefore(new ICmpInst(ICmpInst::ICMP_SGT, LHSCIOp, | ||||
6942 | NegOne, ICI.getName()), ICI); | ||||
6943 | } else { | ||||
6944 | // Unsigned extend & unsigned compare -> always true. | ||||
6945 | Result = ConstantInt::getTrue(); | ||||
6946 | } | ||||
6947 | } | ||||
6948 | |||||
6949 | // Finally, return the value computed. | ||||
6950 | if (ICI.getPredicate() == ICmpInst::ICMP_ULT || | ||||
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 6951 | ICI.getPredicate() == ICmpInst::ICMP_SLT) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6952 | return ReplaceInstUsesWith(ICI, Result); |
Chris Lattner | 3d81653 | 2008-07-11 04:09:09 +0000 | [diff] [blame] | 6953 | |
6954 | assert((ICI.getPredicate()==ICmpInst::ICMP_UGT || | ||||
6955 | ICI.getPredicate()==ICmpInst::ICMP_SGT) && | ||||
6956 | "ICmp should be folded!"); | ||||
6957 | if (Constant *CI = dyn_cast<Constant>(Result)) | ||||
6958 | return ReplaceInstUsesWith(ICI, ConstantExpr::getNot(CI)); | ||||
6959 | return BinaryOperator::CreateNot(Result); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6960 | } |
6961 | |||||
6962 | Instruction *InstCombiner::visitShl(BinaryOperator &I) { | ||||
6963 | return commonShiftTransforms(I); | ||||
6964 | } | ||||
6965 | |||||
6966 | Instruction *InstCombiner::visitLShr(BinaryOperator &I) { | ||||
6967 | return commonShiftTransforms(I); | ||||
6968 | } | ||||
6969 | |||||
6970 | Instruction *InstCombiner::visitAShr(BinaryOperator &I) { | ||||
Chris Lattner | e3c504f | 2007-12-06 01:59:46 +0000 | [diff] [blame] | 6971 | if (Instruction *R = commonShiftTransforms(I)) |
6972 | return R; | ||||
6973 | |||||
6974 | Value *Op0 = I.getOperand(0); | ||||
6975 | |||||
6976 | // ashr int -1, X = -1 (for any arithmetic shift rights of ~0) | ||||
6977 | if (ConstantInt *CSI = dyn_cast<ConstantInt>(Op0)) | ||||
6978 | if (CSI->isAllOnesValue()) | ||||
6979 | return ReplaceInstUsesWith(I, CSI); | ||||
6980 | |||||
6981 | // See if we can turn a signed shr into an unsigned shr. | ||||
Nate Begeman | bb1ce94 | 2008-07-29 15:49:41 +0000 | [diff] [blame] | 6982 | if (!isa<VectorType>(I.getType()) && |
6983 | MaskedValueIsZero(Op0, | ||||
Chris Lattner | e3c504f | 2007-12-06 01:59:46 +0000 | [diff] [blame] | 6984 | APInt::getSignBit(I.getType()->getPrimitiveSizeInBits()))) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 6985 | return BinaryOperator::CreateLShr(Op0, I.getOperand(1)); |
Chris Lattner | e3c504f | 2007-12-06 01:59:46 +0000 | [diff] [blame] | 6986 | |
6987 | return 0; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 6988 | } |
6989 | |||||
6990 | Instruction *InstCombiner::commonShiftTransforms(BinaryOperator &I) { | ||||
6991 | assert(I.getOperand(1)->getType() == I.getOperand(0)->getType()); | ||||
6992 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); | ||||
6993 | |||||
6994 | // shl X, 0 == X and shr X, 0 == X | ||||
6995 | // shl 0, X == 0 and shr 0, X == 0 | ||||
6996 | if (Op1 == Constant::getNullValue(Op1->getType()) || | ||||
6997 | Op0 == Constant::getNullValue(Op0->getType())) | ||||
6998 | return ReplaceInstUsesWith(I, Op0); | ||||
6999 | |||||
7000 | if (isa<UndefValue>(Op0)) { | ||||
7001 | if (I.getOpcode() == Instruction::AShr) // undef >>s X -> undef | ||||
7002 | return ReplaceInstUsesWith(I, Op0); | ||||
7003 | else // undef << X -> 0, undef >>u X -> 0 | ||||
7004 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
7005 | } | ||||
7006 | if (isa<UndefValue>(Op1)) { | ||||
7007 | if (I.getOpcode() == Instruction::AShr) // X >>s undef -> X | ||||
7008 | return ReplaceInstUsesWith(I, Op0); | ||||
7009 | else // X << undef, X >>u undef -> 0 | ||||
7010 | return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType())); | ||||
7011 | } | ||||
7012 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7013 | // Try to fold constant and into select arguments. |
7014 | if (isa<Constant>(Op0)) | ||||
7015 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) | ||||
7016 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
7017 | return R; | ||||
7018 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7019 | if (ConstantInt *CUI = dyn_cast<ConstantInt>(Op1)) |
7020 | if (Instruction *Res = FoldShiftByConstant(Op0, CUI, I)) | ||||
7021 | return Res; | ||||
7022 | return 0; | ||||
7023 | } | ||||
7024 | |||||
7025 | Instruction *InstCombiner::FoldShiftByConstant(Value *Op0, ConstantInt *Op1, | ||||
7026 | BinaryOperator &I) { | ||||
Chris Lattner | 0881733 | 2009-01-31 08:24:16 +0000 | [diff] [blame] | 7027 | bool isLeftShift = I.getOpcode() == Instruction::Shl; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7028 | |
7029 | // See if we can simplify any instructions used by the instruction whose sole | ||||
7030 | // purpose is to compute bits we don't care about. | ||||
7031 | uint32_t TypeBits = Op0->getType()->getPrimitiveSizeInBits(); | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 7032 | if (SimplifyDemandedInstructionBits(I)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7033 | return &I; |
7034 | |||||
7035 | // shl uint X, 32 = 0 and shr ubyte Y, 9 = 0, ... just don't eliminate shr | ||||
7036 | // of a signed value. | ||||
7037 | // | ||||
7038 | if (Op1->uge(TypeBits)) { | ||||
7039 | if (I.getOpcode() != Instruction::AShr) | ||||
7040 | return ReplaceInstUsesWith(I, Constant::getNullValue(Op0->getType())); | ||||
7041 | else { | ||||
7042 | I.setOperand(1, ConstantInt::get(I.getType(), TypeBits-1)); | ||||
7043 | return &I; | ||||
7044 | } | ||||
7045 | } | ||||
7046 | |||||
7047 | // ((X*C1) << C2) == (X * (C1 << C2)) | ||||
7048 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0)) | ||||
7049 | if (BO->getOpcode() == Instruction::Mul && isLeftShift) | ||||
7050 | if (Constant *BOOp = dyn_cast<Constant>(BO->getOperand(1))) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7051 | return BinaryOperator::CreateMul(BO->getOperand(0), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7052 | ConstantExpr::getShl(BOOp, Op1)); |
7053 | |||||
7054 | // Try to fold constant and into select arguments. | ||||
7055 | if (SelectInst *SI = dyn_cast<SelectInst>(Op0)) | ||||
7056 | if (Instruction *R = FoldOpIntoSelect(I, SI, this)) | ||||
7057 | return R; | ||||
7058 | if (isa<PHINode>(Op0)) | ||||
7059 | if (Instruction *NV = FoldOpIntoPhi(I)) | ||||
7060 | return NV; | ||||
7061 | |||||
Chris Lattner | c6d1f64 | 2007-12-22 09:07:47 +0000 | [diff] [blame] | 7062 | // Fold shift2(trunc(shift1(x,c1)), c2) -> trunc(shift2(shift1(x,c1),c2)) |
7063 | if (TruncInst *TI = dyn_cast<TruncInst>(Op0)) { | ||||
7064 | Instruction *TrOp = dyn_cast<Instruction>(TI->getOperand(0)); | ||||
7065 | // If 'shift2' is an ashr, we would have to get the sign bit into a funny | ||||
7066 | // place. Don't try to do this transformation in this case. Also, we | ||||
7067 | // require that the input operand is a shift-by-constant so that we have | ||||
7068 | // confidence that the shifts will get folded together. We could do this | ||||
7069 | // xform in more cases, but it is unlikely to be profitable. | ||||
7070 | if (TrOp && I.isLogicalShift() && TrOp->isShift() && | ||||
7071 | isa<ConstantInt>(TrOp->getOperand(1))) { | ||||
7072 | // Okay, we'll do this xform. Make the shift of shift. | ||||
7073 | Constant *ShAmt = ConstantExpr::getZExt(Op1, TrOp->getType()); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7074 | Instruction *NSh = BinaryOperator::Create(I.getOpcode(), TrOp, ShAmt, |
Chris Lattner | c6d1f64 | 2007-12-22 09:07:47 +0000 | [diff] [blame] | 7075 | I.getName()); |
7076 | InsertNewInstBefore(NSh, I); // (shift2 (shift1 & 0x00FF), c2) | ||||
7077 | |||||
7078 | // For logical shifts, the truncation has the effect of making the high | ||||
7079 | // part of the register be zeros. Emulate this by inserting an AND to | ||||
7080 | // clear the top bits as needed. This 'and' will usually be zapped by | ||||
7081 | // other xforms later if dead. | ||||
7082 | unsigned SrcSize = TrOp->getType()->getPrimitiveSizeInBits(); | ||||
7083 | unsigned DstSize = TI->getType()->getPrimitiveSizeInBits(); | ||||
7084 | APInt MaskV(APInt::getLowBitsSet(SrcSize, DstSize)); | ||||
7085 | |||||
7086 | // The mask we constructed says what the trunc would do if occurring | ||||
7087 | // between the shifts. We want to know the effect *after* the second | ||||
7088 | // shift. We know that it is a logical shift by a constant, so adjust the | ||||
7089 | // mask as appropriate. | ||||
7090 | if (I.getOpcode() == Instruction::Shl) | ||||
7091 | MaskV <<= Op1->getZExtValue(); | ||||
7092 | else { | ||||
7093 | assert(I.getOpcode() == Instruction::LShr && "Unknown logical shift"); | ||||
7094 | MaskV = MaskV.lshr(Op1->getZExtValue()); | ||||
7095 | } | ||||
7096 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7097 | Instruction *And = BinaryOperator::CreateAnd(NSh, ConstantInt::get(MaskV), |
Chris Lattner | c6d1f64 | 2007-12-22 09:07:47 +0000 | [diff] [blame] | 7098 | TI->getName()); |
7099 | InsertNewInstBefore(And, I); // shift1 & 0x00FF | ||||
7100 | |||||
7101 | // Return the value truncated to the interesting size. | ||||
7102 | return new TruncInst(And, I.getType()); | ||||
7103 | } | ||||
7104 | } | ||||
7105 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7106 | if (Op0->hasOneUse()) { |
7107 | if (BinaryOperator *Op0BO = dyn_cast<BinaryOperator>(Op0)) { | ||||
7108 | // Turn ((X >> C) + Y) << C -> (X + (Y << C)) & (~0 << C) | ||||
7109 | Value *V1, *V2; | ||||
7110 | ConstantInt *CC; | ||||
7111 | switch (Op0BO->getOpcode()) { | ||||
7112 | default: break; | ||||
7113 | case Instruction::Add: | ||||
7114 | case Instruction::And: | ||||
7115 | case Instruction::Or: | ||||
7116 | case Instruction::Xor: { | ||||
7117 | // These operators commute. | ||||
7118 | // Turn (Y + (X >> C)) << C -> (X + (Y << C)) & (~0 << C) | ||||
7119 | if (isLeftShift && Op0BO->getOperand(1)->hasOneUse() && | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 7120 | match(Op0BO->getOperand(1), m_Shr(m_Value(V1), m_Specific(Op1)))){ |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7121 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7122 | Op0BO->getOperand(0), Op1, |
7123 | Op0BO->getName()); | ||||
7124 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
7125 | Instruction *X = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7126 | BinaryOperator::Create(Op0BO->getOpcode(), YS, V1, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7127 | Op0BO->getOperand(1)->getName()); |
7128 | InsertNewInstBefore(X, I); // (X + (Y << C)) | ||||
7129 | uint32_t Op1Val = Op1->getLimitedValue(TypeBits); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7130 | return BinaryOperator::CreateAnd(X, ConstantInt::get( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7131 | APInt::getHighBitsSet(TypeBits, TypeBits-Op1Val))); |
7132 | } | ||||
7133 | |||||
7134 | // Turn (Y + ((X >> C) & CC)) << C -> ((X & (CC << C)) + (Y << C)) | ||||
7135 | Value *Op0BOOp1 = Op0BO->getOperand(1); | ||||
7136 | if (isLeftShift && Op0BOOp1->hasOneUse() && | ||||
7137 | match(Op0BOOp1, | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 7138 | m_And(m_Shr(m_Value(V1), m_Specific(Op1)), |
7139 | m_ConstantInt(CC))) && | ||||
7140 | cast<BinaryOperator>(Op0BOOp1)->getOperand(0)->hasOneUse()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7141 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7142 | Op0BO->getOperand(0), Op1, |
7143 | Op0BO->getName()); | ||||
7144 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
7145 | Instruction *XM = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7146 | BinaryOperator::CreateAnd(V1, ConstantExpr::getShl(CC, Op1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7147 | V1->getName()+".mask"); |
7148 | InsertNewInstBefore(XM, I); // X & (CC << C) | ||||
7149 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7150 | return BinaryOperator::Create(Op0BO->getOpcode(), YS, XM); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7151 | } |
7152 | } | ||||
7153 | |||||
7154 | // FALL THROUGH. | ||||
7155 | case Instruction::Sub: { | ||||
7156 | // Turn ((X >> C) + Y) << C -> (X + (Y << C)) & (~0 << C) | ||||
7157 | if (isLeftShift && Op0BO->getOperand(0)->hasOneUse() && | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 7158 | match(Op0BO->getOperand(0), m_Shr(m_Value(V1), m_Specific(Op1)))){ |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7159 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7160 | Op0BO->getOperand(1), Op1, |
7161 | Op0BO->getName()); | ||||
7162 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
7163 | Instruction *X = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7164 | BinaryOperator::Create(Op0BO->getOpcode(), V1, YS, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7165 | Op0BO->getOperand(0)->getName()); |
7166 | InsertNewInstBefore(X, I); // (X + (Y << C)) | ||||
7167 | uint32_t Op1Val = Op1->getLimitedValue(TypeBits); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7168 | return BinaryOperator::CreateAnd(X, ConstantInt::get( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7169 | APInt::getHighBitsSet(TypeBits, TypeBits-Op1Val))); |
7170 | } | ||||
7171 | |||||
7172 | // Turn (((X >> C)&CC) + Y) << C -> (X + (Y << C)) & (CC << C) | ||||
7173 | if (isLeftShift && Op0BO->getOperand(0)->hasOneUse() && | ||||
7174 | match(Op0BO->getOperand(0), | ||||
7175 | m_And(m_Shr(m_Value(V1), m_Value(V2)), | ||||
7176 | m_ConstantInt(CC))) && V2 == Op1 && | ||||
7177 | cast<BinaryOperator>(Op0BO->getOperand(0)) | ||||
7178 | ->getOperand(0)->hasOneUse()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7179 | Instruction *YS = BinaryOperator::CreateShl( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7180 | Op0BO->getOperand(1), Op1, |
7181 | Op0BO->getName()); | ||||
7182 | InsertNewInstBefore(YS, I); // (Y << C) | ||||
7183 | Instruction *XM = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7184 | BinaryOperator::CreateAnd(V1, ConstantExpr::getShl(CC, Op1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7185 | V1->getName()+".mask"); |
7186 | InsertNewInstBefore(XM, I); // X & (CC << C) | ||||
7187 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7188 | return BinaryOperator::Create(Op0BO->getOpcode(), XM, YS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7189 | } |
7190 | |||||
7191 | break; | ||||
7192 | } | ||||
7193 | } | ||||
7194 | |||||
7195 | |||||
7196 | // If the operand is an bitwise operator with a constant RHS, and the | ||||
7197 | // shift is the only use, we can pull it out of the shift. | ||||
7198 | if (ConstantInt *Op0C = dyn_cast<ConstantInt>(Op0BO->getOperand(1))) { | ||||
7199 | bool isValid = true; // Valid only for And, Or, Xor | ||||
7200 | bool highBitSet = false; // Transform if high bit of constant set? | ||||
7201 | |||||
7202 | switch (Op0BO->getOpcode()) { | ||||
7203 | default: isValid = false; break; // Do not perform transform! | ||||
7204 | case Instruction::Add: | ||||
7205 | isValid = isLeftShift; | ||||
7206 | break; | ||||
7207 | case Instruction::Or: | ||||
7208 | case Instruction::Xor: | ||||
7209 | highBitSet = false; | ||||
7210 | break; | ||||
7211 | case Instruction::And: | ||||
7212 | highBitSet = true; | ||||
7213 | break; | ||||
7214 | } | ||||
7215 | |||||
7216 | // If this is a signed shift right, and the high bit is modified | ||||
7217 | // by the logical operation, do not perform the transformation. | ||||
7218 | // The highBitSet boolean indicates the value of the high bit of | ||||
7219 | // the constant which would cause it to be modified for this | ||||
7220 | // operation. | ||||
7221 | // | ||||
Chris Lattner | 15b76e3 | 2007-12-06 06:25:04 +0000 | [diff] [blame] | 7222 | if (isValid && I.getOpcode() == Instruction::AShr) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7223 | isValid = Op0C->getValue()[TypeBits-1] == highBitSet; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7224 | |
7225 | if (isValid) { | ||||
7226 | Constant *NewRHS = ConstantExpr::get(I.getOpcode(), Op0C, Op1); | ||||
7227 | |||||
7228 | Instruction *NewShift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7229 | BinaryOperator::Create(I.getOpcode(), Op0BO->getOperand(0), Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7230 | InsertNewInstBefore(NewShift, I); |
7231 | NewShift->takeName(Op0BO); | ||||
7232 | |||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7233 | return BinaryOperator::Create(Op0BO->getOpcode(), NewShift, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7234 | NewRHS); |
7235 | } | ||||
7236 | } | ||||
7237 | } | ||||
7238 | } | ||||
7239 | |||||
7240 | // Find out if this is a shift of a shift by a constant. | ||||
7241 | BinaryOperator *ShiftOp = dyn_cast<BinaryOperator>(Op0); | ||||
7242 | if (ShiftOp && !ShiftOp->isShift()) | ||||
7243 | ShiftOp = 0; | ||||
7244 | |||||
7245 | if (ShiftOp && isa<ConstantInt>(ShiftOp->getOperand(1))) { | ||||
7246 | ConstantInt *ShiftAmt1C = cast<ConstantInt>(ShiftOp->getOperand(1)); | ||||
7247 | uint32_t ShiftAmt1 = ShiftAmt1C->getLimitedValue(TypeBits); | ||||
7248 | uint32_t ShiftAmt2 = Op1->getLimitedValue(TypeBits); | ||||
7249 | assert(ShiftAmt2 != 0 && "Should have been simplified earlier"); | ||||
7250 | if (ShiftAmt1 == 0) return 0; // Will be simplified in the future. | ||||
7251 | Value *X = ShiftOp->getOperand(0); | ||||
7252 | |||||
7253 | uint32_t AmtSum = ShiftAmt1+ShiftAmt2; // Fold into one big shift. | ||||
7254 | if (AmtSum > TypeBits) | ||||
7255 | AmtSum = TypeBits; | ||||
7256 | |||||
7257 | const IntegerType *Ty = cast<IntegerType>(I.getType()); | ||||
7258 | |||||
7259 | // Check for (X << c1) << c2 and (X >> c1) >> c2 | ||||
7260 | if (I.getOpcode() == ShiftOp->getOpcode()) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7261 | return BinaryOperator::Create(I.getOpcode(), X, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7262 | ConstantInt::get(Ty, AmtSum)); |
7263 | } else if (ShiftOp->getOpcode() == Instruction::LShr && | ||||
7264 | I.getOpcode() == Instruction::AShr) { | ||||
7265 | // ((X >>u C1) >>s C2) -> (X >>u (C1+C2)) since C1 != 0. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7266 | return BinaryOperator::CreateLShr(X, ConstantInt::get(Ty, AmtSum)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7267 | } else if (ShiftOp->getOpcode() == Instruction::AShr && |
7268 | I.getOpcode() == Instruction::LShr) { | ||||
7269 | // ((X >>s C1) >>u C2) -> ((X >>s (C1+C2)) & mask) since C1 != 0. | ||||
7270 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7271 | BinaryOperator::CreateAShr(X, ConstantInt::get(Ty, AmtSum)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7272 | InsertNewInstBefore(Shift, I); |
7273 | |||||
7274 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7275 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7276 | } |
7277 | |||||
7278 | // Okay, if we get here, one shift must be left, and the other shift must be | ||||
7279 | // right. See if the amounts are equal. | ||||
7280 | if (ShiftAmt1 == ShiftAmt2) { | ||||
7281 | // If we have ((X >>? C) << C), turn this into X & (-1 << C). | ||||
7282 | if (I.getOpcode() == Instruction::Shl) { | ||||
7283 | APInt Mask(APInt::getHighBitsSet(TypeBits, TypeBits - ShiftAmt1)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7284 | return BinaryOperator::CreateAnd(X, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7285 | } |
7286 | // If we have ((X << C) >>u C), turn this into X & (-1 >>u C). | ||||
7287 | if (I.getOpcode() == Instruction::LShr) { | ||||
7288 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt1)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7289 | return BinaryOperator::CreateAnd(X, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7290 | } |
7291 | // We can simplify ((X << C) >>s C) into a trunc + sext. | ||||
7292 | // NOTE: we could do this for any C, but that would make 'unusual' integer | ||||
7293 | // types. For now, just stick to ones well-supported by the code | ||||
7294 | // generators. | ||||
7295 | const Type *SExtType = 0; | ||||
7296 | switch (Ty->getBitWidth() - ShiftAmt1) { | ||||
7297 | case 1 : | ||||
7298 | case 8 : | ||||
7299 | case 16 : | ||||
7300 | case 32 : | ||||
7301 | case 64 : | ||||
7302 | case 128: | ||||
7303 | SExtType = IntegerType::get(Ty->getBitWidth() - ShiftAmt1); | ||||
7304 | break; | ||||
7305 | default: break; | ||||
7306 | } | ||||
7307 | if (SExtType) { | ||||
7308 | Instruction *NewTrunc = new TruncInst(X, SExtType, "sext"); | ||||
7309 | InsertNewInstBefore(NewTrunc, I); | ||||
7310 | return new SExtInst(NewTrunc, Ty); | ||||
7311 | } | ||||
7312 | // Otherwise, we can't handle it yet. | ||||
7313 | } else if (ShiftAmt1 < ShiftAmt2) { | ||||
7314 | uint32_t ShiftDiff = ShiftAmt2-ShiftAmt1; | ||||
7315 | |||||
7316 | // (X >>? C1) << C2 --> X << (C2-C1) & (-1 << C2) | ||||
7317 | if (I.getOpcode() == Instruction::Shl) { | ||||
7318 | assert(ShiftOp->getOpcode() == Instruction::LShr || | ||||
7319 | ShiftOp->getOpcode() == Instruction::AShr); | ||||
7320 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7321 | BinaryOperator::CreateShl(X, ConstantInt::get(Ty, ShiftDiff)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7322 | InsertNewInstBefore(Shift, I); |
7323 | |||||
7324 | APInt Mask(APInt::getHighBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7325 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7326 | } |
7327 | |||||
7328 | // (X << C1) >>u C2 --> X >>u (C2-C1) & (-1 >> C2) | ||||
7329 | if (I.getOpcode() == Instruction::LShr) { | ||||
7330 | assert(ShiftOp->getOpcode() == Instruction::Shl); | ||||
7331 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7332 | BinaryOperator::CreateLShr(X, ConstantInt::get(Ty, ShiftDiff)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7333 | InsertNewInstBefore(Shift, I); |
7334 | |||||
7335 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7336 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7337 | } |
7338 | |||||
7339 | // We can't handle (X << C1) >>s C2, it shifts arbitrary bits in. | ||||
7340 | } else { | ||||
7341 | assert(ShiftAmt2 < ShiftAmt1); | ||||
7342 | uint32_t ShiftDiff = ShiftAmt1-ShiftAmt2; | ||||
7343 | |||||
7344 | // (X >>? C1) << C2 --> X >>? (C1-C2) & (-1 << C2) | ||||
7345 | if (I.getOpcode() == Instruction::Shl) { | ||||
7346 | assert(ShiftOp->getOpcode() == Instruction::LShr || | ||||
7347 | ShiftOp->getOpcode() == Instruction::AShr); | ||||
7348 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7349 | BinaryOperator::Create(ShiftOp->getOpcode(), X, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7350 | ConstantInt::get(Ty, ShiftDiff)); |
7351 | InsertNewInstBefore(Shift, I); | ||||
7352 | |||||
7353 | APInt Mask(APInt::getHighBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7354 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7355 | } |
7356 | |||||
7357 | // (X << C1) >>u C2 --> X << (C1-C2) & (-1 >> C2) | ||||
7358 | if (I.getOpcode() == Instruction::LShr) { | ||||
7359 | assert(ShiftOp->getOpcode() == Instruction::Shl); | ||||
7360 | Instruction *Shift = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7361 | BinaryOperator::CreateShl(X, ConstantInt::get(Ty, ShiftDiff)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7362 | InsertNewInstBefore(Shift, I); |
7363 | |||||
7364 | APInt Mask(APInt::getLowBitsSet(TypeBits, TypeBits - ShiftAmt2)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7365 | return BinaryOperator::CreateAnd(Shift, ConstantInt::get(Mask)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7366 | } |
7367 | |||||
7368 | // We can't handle (X << C1) >>a C2, it shifts arbitrary bits in. | ||||
7369 | } | ||||
7370 | } | ||||
7371 | return 0; | ||||
7372 | } | ||||
7373 | |||||
7374 | |||||
7375 | /// DecomposeSimpleLinearExpr - Analyze 'Val', seeing if it is a simple linear | ||||
7376 | /// expression. If so, decompose it, returning some value X, such that Val is | ||||
7377 | /// X*Scale+Offset. | ||||
7378 | /// | ||||
7379 | static Value *DecomposeSimpleLinearExpr(Value *Val, unsigned &Scale, | ||||
7380 | int &Offset) { | ||||
7381 | assert(Val->getType() == Type::Int32Ty && "Unexpected allocation size type!"); | ||||
7382 | if (ConstantInt *CI = dyn_cast<ConstantInt>(Val)) { | ||||
7383 | Offset = CI->getZExtValue(); | ||||
Chris Lattner | c59171a | 2007-10-12 05:30:59 +0000 | [diff] [blame] | 7384 | Scale = 0; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7385 | return ConstantInt::get(Type::Int32Ty, 0); |
Chris Lattner | c59171a | 2007-10-12 05:30:59 +0000 | [diff] [blame] | 7386 | } else if (BinaryOperator *I = dyn_cast<BinaryOperator>(Val)) { |
7387 | if (ConstantInt *RHS = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
7388 | if (I->getOpcode() == Instruction::Shl) { | ||||
7389 | // This is a value scaled by '1 << the shift amt'. | ||||
7390 | Scale = 1U << RHS->getZExtValue(); | ||||
7391 | Offset = 0; | ||||
7392 | return I->getOperand(0); | ||||
7393 | } else if (I->getOpcode() == Instruction::Mul) { | ||||
7394 | // This value is scaled by 'RHS'. | ||||
7395 | Scale = RHS->getZExtValue(); | ||||
7396 | Offset = 0; | ||||
7397 | return I->getOperand(0); | ||||
7398 | } else if (I->getOpcode() == Instruction::Add) { | ||||
7399 | // We have X+C. Check to see if we really have (X*C2)+C1, | ||||
7400 | // where C1 is divisible by C2. | ||||
7401 | unsigned SubScale; | ||||
7402 | Value *SubVal = | ||||
7403 | DecomposeSimpleLinearExpr(I->getOperand(0), SubScale, Offset); | ||||
7404 | Offset += RHS->getZExtValue(); | ||||
7405 | Scale = SubScale; | ||||
7406 | return SubVal; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7407 | } |
7408 | } | ||||
7409 | } | ||||
7410 | |||||
7411 | // Otherwise, we can't look past this. | ||||
7412 | Scale = 1; | ||||
7413 | Offset = 0; | ||||
7414 | return Val; | ||||
7415 | } | ||||
7416 | |||||
7417 | |||||
7418 | /// PromoteCastOfAllocation - If we find a cast of an allocation instruction, | ||||
7419 | /// try to eliminate the cast by moving the type information into the alloc. | ||||
7420 | Instruction *InstCombiner::PromoteCastOfAllocation(BitCastInst &CI, | ||||
7421 | AllocationInst &AI) { | ||||
7422 | const PointerType *PTy = cast<PointerType>(CI.getType()); | ||||
7423 | |||||
7424 | // Remove any uses of AI that are dead. | ||||
7425 | assert(!CI.use_empty() && "Dead instructions should be removed earlier!"); | ||||
7426 | |||||
7427 | for (Value::use_iterator UI = AI.use_begin(), E = AI.use_end(); UI != E; ) { | ||||
7428 | Instruction *User = cast<Instruction>(*UI++); | ||||
7429 | if (isInstructionTriviallyDead(User)) { | ||||
7430 | while (UI != E && *UI == User) | ||||
7431 | ++UI; // If this instruction uses AI more than once, don't break UI. | ||||
7432 | |||||
7433 | ++NumDeadInst; | ||||
7434 | DOUT << "IC: DCE: " << *User; | ||||
7435 | EraseInstFromFunction(*User); | ||||
7436 | } | ||||
7437 | } | ||||
7438 | |||||
7439 | // Get the type really allocated and the type casted to. | ||||
7440 | const Type *AllocElTy = AI.getAllocatedType(); | ||||
7441 | const Type *CastElTy = PTy->getElementType(); | ||||
7442 | if (!AllocElTy->isSized() || !CastElTy->isSized()) return 0; | ||||
7443 | |||||
7444 | unsigned AllocElTyAlign = TD->getABITypeAlignment(AllocElTy); | ||||
7445 | unsigned CastElTyAlign = TD->getABITypeAlignment(CastElTy); | ||||
7446 | if (CastElTyAlign < AllocElTyAlign) return 0; | ||||
7447 | |||||
7448 | // If the allocation has multiple uses, only promote it if we are strictly | ||||
7449 | // increasing the alignment of the resultant allocation. If we keep it the | ||||
7450 | // same, we open the door to infinite loops of various kinds. | ||||
7451 | if (!AI.hasOneUse() && CastElTyAlign == AllocElTyAlign) return 0; | ||||
7452 | |||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 7453 | uint64_t AllocElTySize = TD->getTypePaddedSize(AllocElTy); |
7454 | uint64_t CastElTySize = TD->getTypePaddedSize(CastElTy); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7455 | if (CastElTySize == 0 || AllocElTySize == 0) return 0; |
7456 | |||||
7457 | // See if we can satisfy the modulus by pulling a scale out of the array | ||||
7458 | // size argument. | ||||
7459 | unsigned ArraySizeScale; | ||||
7460 | int ArrayOffset; | ||||
7461 | Value *NumElements = // See if the array size is a decomposable linear expr. | ||||
7462 | DecomposeSimpleLinearExpr(AI.getOperand(0), ArraySizeScale, ArrayOffset); | ||||
7463 | |||||
7464 | // If we can now satisfy the modulus, by using a non-1 scale, we really can | ||||
7465 | // do the xform. | ||||
7466 | if ((AllocElTySize*ArraySizeScale) % CastElTySize != 0 || | ||||
7467 | (AllocElTySize*ArrayOffset ) % CastElTySize != 0) return 0; | ||||
7468 | |||||
7469 | unsigned Scale = (AllocElTySize*ArraySizeScale)/CastElTySize; | ||||
7470 | Value *Amt = 0; | ||||
7471 | if (Scale == 1) { | ||||
7472 | Amt = NumElements; | ||||
7473 | } else { | ||||
7474 | // If the allocation size is constant, form a constant mul expression | ||||
7475 | Amt = ConstantInt::get(Type::Int32Ty, Scale); | ||||
7476 | if (isa<ConstantInt>(NumElements)) | ||||
7477 | Amt = Multiply(cast<ConstantInt>(NumElements), cast<ConstantInt>(Amt)); | ||||
7478 | // otherwise multiply the amount and the number of elements | ||||
7479 | else if (Scale != 1) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7480 | Instruction *Tmp = BinaryOperator::CreateMul(Amt, NumElements, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7481 | Amt = InsertNewInstBefore(Tmp, AI); |
7482 | } | ||||
7483 | } | ||||
7484 | |||||
7485 | if (int Offset = (AllocElTySize*ArrayOffset)/CastElTySize) { | ||||
7486 | Value *Off = ConstantInt::get(Type::Int32Ty, Offset, true); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7487 | Instruction *Tmp = BinaryOperator::CreateAdd(Amt, Off, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7488 | Amt = InsertNewInstBefore(Tmp, AI); |
7489 | } | ||||
7490 | |||||
7491 | AllocationInst *New; | ||||
7492 | if (isa<MallocInst>(AI)) | ||||
7493 | New = new MallocInst(CastElTy, Amt, AI.getAlignment()); | ||||
7494 | else | ||||
7495 | New = new AllocaInst(CastElTy, Amt, AI.getAlignment()); | ||||
7496 | InsertNewInstBefore(New, AI); | ||||
7497 | New->takeName(&AI); | ||||
7498 | |||||
7499 | // If the allocation has multiple uses, insert a cast and change all things | ||||
7500 | // that used it to use the new cast. This will also hack on CI, but it will | ||||
7501 | // die soon. | ||||
7502 | if (!AI.hasOneUse()) { | ||||
7503 | AddUsesToWorkList(AI); | ||||
7504 | // New is the allocation instruction, pointer typed. AI is the original | ||||
7505 | // allocation instruction, also pointer typed. Thus, cast to use is BitCast. | ||||
7506 | CastInst *NewCast = new BitCastInst(New, AI.getType(), "tmpcast"); | ||||
7507 | InsertNewInstBefore(NewCast, AI); | ||||
7508 | AI.replaceAllUsesWith(NewCast); | ||||
7509 | } | ||||
7510 | return ReplaceInstUsesWith(CI, New); | ||||
7511 | } | ||||
7512 | |||||
7513 | /// CanEvaluateInDifferentType - Return true if we can take the specified value | ||||
7514 | /// and return it as type Ty without inserting any new casts and without | ||||
7515 | /// changing the computed value. This is used by code that tries to decide | ||||
7516 | /// whether promoting or shrinking integer operations to wider or smaller types | ||||
7517 | /// will allow us to eliminate a truncate or extend. | ||||
7518 | /// | ||||
7519 | /// This is a truncation operation if Ty is smaller than V->getType(), or an | ||||
7520 | /// extension operation if Ty is larger. | ||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7521 | /// |
7522 | /// If CastOpc is a truncation, then Ty will be a type smaller than V. We | ||||
7523 | /// should return true if trunc(V) can be computed by computing V in the smaller | ||||
7524 | /// type. If V is an instruction, then trunc(inst(x,y)) can be computed as | ||||
7525 | /// inst(trunc(x),trunc(y)), which only makes sense if x and y can be | ||||
7526 | /// efficiently truncated. | ||||
7527 | /// | ||||
7528 | /// If CastOpc is a sext or zext, we are asking if the low bits of the value can | ||||
7529 | /// bit computed in a larger type, which is then and'd or sext_in_reg'd to get | ||||
7530 | /// the final result. | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7531 | bool InstCombiner::CanEvaluateInDifferentType(Value *V, const IntegerType *Ty, |
7532 | unsigned CastOpc, | ||||
7533 | int &NumCastsRemoved){ | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7534 | // We can always evaluate constants in another type. |
7535 | if (isa<ConstantInt>(V)) | ||||
7536 | return true; | ||||
7537 | |||||
7538 | Instruction *I = dyn_cast<Instruction>(V); | ||||
7539 | if (!I) return false; | ||||
7540 | |||||
7541 | const IntegerType *OrigTy = cast<IntegerType>(V->getType()); | ||||
7542 | |||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7543 | // If this is an extension or truncate, we can often eliminate it. |
7544 | if (isa<TruncInst>(I) || isa<ZExtInst>(I) || isa<SExtInst>(I)) { | ||||
7545 | // If this is a cast from the destination type, we can trivially eliminate | ||||
7546 | // it, and this will remove a cast overall. | ||||
7547 | if (I->getOperand(0)->getType() == Ty) { | ||||
7548 | // If the first operand is itself a cast, and is eliminable, do not count | ||||
7549 | // this as an eliminable cast. We would prefer to eliminate those two | ||||
7550 | // casts first. | ||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7551 | if (!isa<CastInst>(I->getOperand(0)) && I->hasOneUse()) |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7552 | ++NumCastsRemoved; |
7553 | return true; | ||||
7554 | } | ||||
7555 | } | ||||
7556 | |||||
7557 | // We can't extend or shrink something that has multiple uses: doing so would | ||||
7558 | // require duplicating the instruction in general, which isn't profitable. | ||||
7559 | if (!I->hasOneUse()) return false; | ||||
7560 | |||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7561 | unsigned Opc = I->getOpcode(); |
7562 | switch (Opc) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7563 | case Instruction::Add: |
7564 | case Instruction::Sub: | ||||
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7565 | case Instruction::Mul: |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7566 | case Instruction::And: |
7567 | case Instruction::Or: | ||||
7568 | case Instruction::Xor: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7569 | // These operators can all arbitrarily be extended or truncated. |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7570 | return CanEvaluateInDifferentType(I->getOperand(0), Ty, CastOpc, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7571 | NumCastsRemoved) && |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7572 | CanEvaluateInDifferentType(I->getOperand(1), Ty, CastOpc, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7573 | NumCastsRemoved); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7574 | |
7575 | case Instruction::Shl: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7576 | // If we are truncating the result of this SHL, and if it's a shift of a |
7577 | // constant amount, we can always perform a SHL in a smaller type. | ||||
7578 | if (ConstantInt *CI = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
7579 | uint32_t BitWidth = Ty->getBitWidth(); | ||||
7580 | if (BitWidth < OrigTy->getBitWidth() && | ||||
7581 | CI->getLimitedValue(BitWidth) < BitWidth) | ||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7582 | return CanEvaluateInDifferentType(I->getOperand(0), Ty, CastOpc, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7583 | NumCastsRemoved); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7584 | } |
7585 | break; | ||||
7586 | case Instruction::LShr: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7587 | // If this is a truncate of a logical shr, we can truncate it to a smaller |
7588 | // lshr iff we know that the bits we would otherwise be shifting in are | ||||
7589 | // already zeros. | ||||
7590 | if (ConstantInt *CI = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
7591 | uint32_t OrigBitWidth = OrigTy->getBitWidth(); | ||||
7592 | uint32_t BitWidth = Ty->getBitWidth(); | ||||
7593 | if (BitWidth < OrigBitWidth && | ||||
7594 | MaskedValueIsZero(I->getOperand(0), | ||||
7595 | APInt::getHighBitsSet(OrigBitWidth, OrigBitWidth-BitWidth)) && | ||||
7596 | CI->getLimitedValue(BitWidth) < BitWidth) { | ||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7597 | return CanEvaluateInDifferentType(I->getOperand(0), Ty, CastOpc, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7598 | NumCastsRemoved); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7599 | } |
7600 | } | ||||
7601 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7602 | case Instruction::ZExt: |
7603 | case Instruction::SExt: | ||||
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7604 | case Instruction::Trunc: |
7605 | // 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] | 7606 | // can safely replace it. Note that replacing it does not reduce the number |
7607 | // of casts in the input. | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7608 | if (Opc == CastOpc) |
7609 | return true; | ||||
7610 | |||||
7611 | // sext (zext ty1), ty2 -> zext ty2 | ||||
Evan Cheng | 7bb0d95 | 2009-01-15 17:09:07 +0000 | [diff] [blame] | 7612 | if (CastOpc == Instruction::SExt && Opc == Instruction::ZExt) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7613 | return true; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7614 | break; |
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7615 | case Instruction::Select: { |
7616 | SelectInst *SI = cast<SelectInst>(I); | ||||
7617 | return CanEvaluateInDifferentType(SI->getTrueValue(), Ty, CastOpc, | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7618 | NumCastsRemoved) && |
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7619 | CanEvaluateInDifferentType(SI->getFalseValue(), Ty, CastOpc, |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7620 | NumCastsRemoved); |
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7621 | } |
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7622 | case Instruction::PHI: { |
7623 | // We can change a phi if we can change all operands. | ||||
7624 | PHINode *PN = cast<PHINode>(I); | ||||
7625 | for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) | ||||
7626 | if (!CanEvaluateInDifferentType(PN->getIncomingValue(i), Ty, CastOpc, | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7627 | NumCastsRemoved)) |
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7628 | return false; |
7629 | return true; | ||||
7630 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7631 | default: |
7632 | // TODO: Can handle more cases here. | ||||
7633 | break; | ||||
7634 | } | ||||
7635 | |||||
7636 | return false; | ||||
7637 | } | ||||
7638 | |||||
7639 | /// EvaluateInDifferentType - Given an expression that | ||||
7640 | /// CanEvaluateInDifferentType returns true for, actually insert the code to | ||||
7641 | /// evaluate the expression. | ||||
7642 | Value *InstCombiner::EvaluateInDifferentType(Value *V, const Type *Ty, | ||||
7643 | bool isSigned) { | ||||
7644 | if (Constant *C = dyn_cast<Constant>(V)) | ||||
7645 | return ConstantExpr::getIntegerCast(C, Ty, isSigned /*Sext or ZExt*/); | ||||
7646 | |||||
7647 | // Otherwise, it must be an instruction. | ||||
7648 | Instruction *I = cast<Instruction>(V); | ||||
7649 | Instruction *Res = 0; | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7650 | unsigned Opc = I->getOpcode(); |
7651 | switch (Opc) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7652 | case Instruction::Add: |
7653 | case Instruction::Sub: | ||||
Nick Lewycky | c52646a | 2008-01-22 05:08:48 +0000 | [diff] [blame] | 7654 | case Instruction::Mul: |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7655 | case Instruction::And: |
7656 | case Instruction::Or: | ||||
7657 | case Instruction::Xor: | ||||
7658 | case Instruction::AShr: | ||||
7659 | case Instruction::LShr: | ||||
7660 | case Instruction::Shl: { | ||||
7661 | Value *LHS = EvaluateInDifferentType(I->getOperand(0), Ty, isSigned); | ||||
7662 | Value *RHS = EvaluateInDifferentType(I->getOperand(1), Ty, isSigned); | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7663 | Res = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7664 | break; |
7665 | } | ||||
7666 | case Instruction::Trunc: | ||||
7667 | case Instruction::ZExt: | ||||
7668 | case Instruction::SExt: | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7669 | // 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] | 7670 | // just return the source. There's no need to insert it because it is not |
7671 | // new. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7672 | if (I->getOperand(0)->getType() == Ty) |
7673 | return I->getOperand(0); | ||||
7674 | |||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7675 | // 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] | 7676 | Res = CastInst::Create(cast<CastInst>(I)->getOpcode(), I->getOperand(0), |
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7677 | Ty); |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7678 | break; |
Nick Lewycky | 1265a7d | 2008-07-05 21:19:34 +0000 | [diff] [blame] | 7679 | case Instruction::Select: { |
7680 | Value *True = EvaluateInDifferentType(I->getOperand(1), Ty, isSigned); | ||||
7681 | Value *False = EvaluateInDifferentType(I->getOperand(2), Ty, isSigned); | ||||
7682 | Res = SelectInst::Create(I->getOperand(0), True, False); | ||||
7683 | break; | ||||
7684 | } | ||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7685 | case Instruction::PHI: { |
7686 | PHINode *OPN = cast<PHINode>(I); | ||||
7687 | PHINode *NPN = PHINode::Create(Ty); | ||||
7688 | for (unsigned i = 0, e = OPN->getNumIncomingValues(); i != e; ++i) { | ||||
7689 | Value *V =EvaluateInDifferentType(OPN->getIncomingValue(i), Ty, isSigned); | ||||
7690 | NPN->addIncoming(V, OPN->getIncomingBlock(i)); | ||||
7691 | } | ||||
7692 | Res = NPN; | ||||
7693 | break; | ||||
7694 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7695 | default: |
7696 | // TODO: Can handle more cases here. | ||||
7697 | assert(0 && "Unreachable!"); | ||||
7698 | break; | ||||
7699 | } | ||||
7700 | |||||
Chris Lattner | 4200c206 | 2008-06-18 04:00:49 +0000 | [diff] [blame] | 7701 | Res->takeName(I); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7702 | return InsertNewInstBefore(Res, *I); |
7703 | } | ||||
7704 | |||||
7705 | /// @brief Implement the transforms common to all CastInst visitors. | ||||
7706 | Instruction *InstCombiner::commonCastTransforms(CastInst &CI) { | ||||
7707 | Value *Src = CI.getOperand(0); | ||||
7708 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7709 | // Many cases of "cast of a cast" are eliminable. If it's eliminable we just |
7710 | // eliminate it now. | ||||
7711 | if (CastInst *CSrc = dyn_cast<CastInst>(Src)) { // A->B->C cast | ||||
7712 | if (Instruction::CastOps opc = | ||||
7713 | isEliminableCastPair(CSrc, CI.getOpcode(), CI.getType(), TD)) { | ||||
7714 | // The first cast (CSrc) is eliminable so we need to fix up or replace | ||||
7715 | // 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] | 7716 | return CastInst::Create(opc, CSrc->getOperand(0), CI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7717 | } |
7718 | } | ||||
7719 | |||||
7720 | // If we are casting a select then fold the cast into the select | ||||
7721 | if (SelectInst *SI = dyn_cast<SelectInst>(Src)) | ||||
7722 | if (Instruction *NV = FoldOpIntoSelect(CI, SI, this)) | ||||
7723 | return NV; | ||||
7724 | |||||
7725 | // If we are casting a PHI then fold the cast into the PHI | ||||
7726 | if (isa<PHINode>(Src)) | ||||
7727 | if (Instruction *NV = FoldOpIntoPhi(CI)) | ||||
7728 | return NV; | ||||
7729 | |||||
7730 | return 0; | ||||
7731 | } | ||||
7732 | |||||
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7733 | /// FindElementAtOffset - Given a type and a constant offset, determine whether |
7734 | /// or not there is a sequence of GEP indices into the type that will land us at | ||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 7735 | /// the specified offset. If so, fill them into NewIndices and return the |
7736 | /// resultant element type, otherwise return null. | ||||
7737 | static const Type *FindElementAtOffset(const Type *Ty, int64_t Offset, | ||||
7738 | SmallVectorImpl<Value*> &NewIndices, | ||||
7739 | const TargetData *TD) { | ||||
7740 | if (!Ty->isSized()) return 0; | ||||
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7741 | |
7742 | // Start with the index over the outer type. Note that the type size | ||||
7743 | // might be zero (even if the offset isn't zero) if the indexed type | ||||
7744 | // is something like [0 x {int, int}] | ||||
7745 | const Type *IntPtrTy = TD->getIntPtrType(); | ||||
7746 | int64_t FirstIdx = 0; | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 7747 | if (int64_t TySize = TD->getTypePaddedSize(Ty)) { |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7748 | FirstIdx = Offset/TySize; |
Chris Lattner | 0bd6f2b | 2009-01-11 20:41:36 +0000 | [diff] [blame] | 7749 | Offset -= FirstIdx*TySize; |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7750 | |
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7751 | // Handle hosts where % returns negative instead of values [0..TySize). |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7752 | if (Offset < 0) { |
7753 | --FirstIdx; | ||||
7754 | Offset += TySize; | ||||
7755 | assert(Offset >= 0); | ||||
7756 | } | ||||
7757 | assert((uint64_t)Offset < (uint64_t)TySize && "Out of range offset"); | ||||
7758 | } | ||||
7759 | |||||
7760 | NewIndices.push_back(ConstantInt::get(IntPtrTy, FirstIdx)); | ||||
7761 | |||||
7762 | // Index into the types. If we fail, set OrigBase to null. | ||||
7763 | while (Offset) { | ||||
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7764 | // Indexing into tail padding between struct/array elements. |
7765 | if (uint64_t(Offset*8) >= TD->getTypeSizeInBits(Ty)) | ||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 7766 | return 0; |
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7767 | |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7768 | if (const StructType *STy = dyn_cast<StructType>(Ty)) { |
7769 | const StructLayout *SL = TD->getStructLayout(STy); | ||||
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7770 | assert(Offset < (int64_t)SL->getSizeInBytes() && |
7771 | "Offset must stay within the indexed type"); | ||||
7772 | |||||
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7773 | unsigned Elt = SL->getElementContainingOffset(Offset); |
7774 | NewIndices.push_back(ConstantInt::get(Type::Int32Ty, Elt)); | ||||
7775 | |||||
7776 | Offset -= SL->getElementOffset(Elt); | ||||
7777 | Ty = STy->getElementType(Elt); | ||||
Chris Lattner | d35ce6a | 2009-01-11 20:23:52 +0000 | [diff] [blame] | 7778 | } else if (const ArrayType *AT = dyn_cast<ArrayType>(Ty)) { |
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 7779 | uint64_t EltSize = TD->getTypePaddedSize(AT->getElementType()); |
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7780 | assert(EltSize && "Cannot index into a zero-sized array"); |
7781 | NewIndices.push_back(ConstantInt::get(IntPtrTy,Offset/EltSize)); | ||||
7782 | Offset %= EltSize; | ||||
Chris Lattner | d35ce6a | 2009-01-11 20:23:52 +0000 | [diff] [blame] | 7783 | Ty = AT->getElementType(); |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7784 | } else { |
Chris Lattner | ce48c46 | 2009-01-11 20:15:20 +0000 | [diff] [blame] | 7785 | // Otherwise, we can't index into the middle of this atomic type, bail. |
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 7786 | return 0; |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7787 | } |
7788 | } | ||||
7789 | |||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 7790 | return Ty; |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7791 | } |
7792 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7793 | /// @brief Implement the transforms for cast of pointer (bitcast/ptrtoint) |
7794 | Instruction *InstCombiner::commonPointerCastTransforms(CastInst &CI) { | ||||
7795 | Value *Src = CI.getOperand(0); | ||||
7796 | |||||
7797 | if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Src)) { | ||||
7798 | // If casting the result of a getelementptr instruction with no offset, turn | ||||
7799 | // this into a cast of the original pointer! | ||||
7800 | if (GEP->hasAllZeroIndices()) { | ||||
7801 | // Changing the cast operand is usually not a good idea but it is safe | ||||
7802 | // here because the pointer operand is being replaced with another | ||||
7803 | // pointer operand so the opcode doesn't need to change. | ||||
7804 | AddToWorkList(GEP); | ||||
7805 | CI.setOperand(0, GEP->getOperand(0)); | ||||
7806 | return &CI; | ||||
7807 | } | ||||
7808 | |||||
7809 | // If the GEP has a single use, and the base pointer is a bitcast, and the | ||||
7810 | // GEP computes a constant offset, see if we can convert these three | ||||
7811 | // instructions into fewer. This typically happens with unions and other | ||||
7812 | // non-type-safe code. | ||||
7813 | if (GEP->hasOneUse() && isa<BitCastInst>(GEP->getOperand(0))) { | ||||
7814 | if (GEP->hasAllConstantIndices()) { | ||||
7815 | // We are guaranteed to get a constant from EmitGEPOffset. | ||||
7816 | ConstantInt *OffsetV = cast<ConstantInt>(EmitGEPOffset(GEP, CI, *this)); | ||||
7817 | int64_t Offset = OffsetV->getSExtValue(); | ||||
7818 | |||||
7819 | // Get the base pointer input of the bitcast, and the type it points to. | ||||
7820 | Value *OrigBase = cast<BitCastInst>(GEP->getOperand(0))->getOperand(0); | ||||
7821 | const Type *GEPIdxTy = | ||||
7822 | cast<PointerType>(OrigBase->getType())->getElementType(); | ||||
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7823 | SmallVector<Value*, 8> NewIndices; |
7824 | if (FindElementAtOffset(GEPIdxTy, Offset, NewIndices, TD)) { | ||||
7825 | // If we were able to index down into an element, create the GEP | ||||
7826 | // and bitcast the result. This eliminates one bitcast, potentially | ||||
7827 | // two. | ||||
7828 | Instruction *NGEP = GetElementPtrInst::Create(OrigBase, | ||||
7829 | NewIndices.begin(), | ||||
7830 | NewIndices.end(), ""); | ||||
7831 | InsertNewInstBefore(NGEP, CI); | ||||
7832 | NGEP->takeName(GEP); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7833 | |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 7834 | if (isa<BitCastInst>(CI)) |
7835 | return new BitCastInst(NGEP, CI.getType()); | ||||
7836 | assert(isa<PtrToIntInst>(CI)); | ||||
7837 | return new PtrToIntInst(NGEP, CI.getType()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7838 | } |
7839 | } | ||||
7840 | } | ||||
7841 | } | ||||
7842 | |||||
7843 | return commonCastTransforms(CI); | ||||
7844 | } | ||||
7845 | |||||
7846 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7847 | /// Only the TRUNC, ZEXT, SEXT, and BITCAST can both operand and result as |
7848 | /// integer types. This function implements the common transforms for all those | ||||
7849 | /// cases. | ||||
7850 | /// @brief Implement the transforms common to CastInst with integer operands | ||||
7851 | Instruction *InstCombiner::commonIntCastTransforms(CastInst &CI) { | ||||
7852 | if (Instruction *Result = commonCastTransforms(CI)) | ||||
7853 | return Result; | ||||
7854 | |||||
7855 | Value *Src = CI.getOperand(0); | ||||
7856 | const Type *SrcTy = Src->getType(); | ||||
7857 | const Type *DestTy = CI.getType(); | ||||
7858 | uint32_t SrcBitSize = SrcTy->getPrimitiveSizeInBits(); | ||||
7859 | uint32_t DestBitSize = DestTy->getPrimitiveSizeInBits(); | ||||
7860 | |||||
7861 | // See if we can simplify any instructions used by the LHS whose sole | ||||
7862 | // purpose is to compute bits we don't care about. | ||||
Chris Lattner | 676c78e | 2009-01-31 08:15:18 +0000 | [diff] [blame] | 7863 | if (SimplifyDemandedInstructionBits(CI)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7864 | return &CI; |
7865 | |||||
7866 | // If the source isn't an instruction or has more than one use then we | ||||
7867 | // can't do anything more. | ||||
7868 | Instruction *SrcI = dyn_cast<Instruction>(Src); | ||||
7869 | if (!SrcI || !Src->hasOneUse()) | ||||
7870 | return 0; | ||||
7871 | |||||
7872 | // Attempt to propagate the cast into the instruction for int->int casts. | ||||
7873 | int NumCastsRemoved = 0; | ||||
7874 | if (!isa<BitCastInst>(CI) && | ||||
7875 | CanEvaluateInDifferentType(SrcI, cast<IntegerType>(DestTy), | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7876 | CI.getOpcode(), NumCastsRemoved)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7877 | // If this cast is a truncate, evaluting in a different type always |
Chris Lattner | ef70bb8 | 2007-08-02 06:11:14 +0000 | [diff] [blame] | 7878 | // eliminates the cast, so it is always a win. If this is a zero-extension, |
7879 | // we need to do an AND to maintain the clear top-part of the computation, | ||||
7880 | // so we require that the input have eliminated at least one cast. If this | ||||
7881 | // 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] | 7882 | // require that two casts have been eliminated. |
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7883 | bool DoXForm = false; |
7884 | bool JustReplace = false; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7885 | switch (CI.getOpcode()) { |
7886 | default: | ||||
7887 | // All the others use floating point so we shouldn't actually | ||||
7888 | // get here because of the check above. | ||||
7889 | assert(0 && "Unknown cast type"); | ||||
7890 | case Instruction::Trunc: | ||||
7891 | DoXForm = true; | ||||
7892 | break; | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7893 | case Instruction::ZExt: { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7894 | DoXForm = NumCastsRemoved >= 1; |
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7895 | if (!DoXForm && 0) { |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7896 | // If it's unnecessary to issue an AND to clear the high bits, it's |
7897 | // always profitable to do this xform. | ||||
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7898 | Value *TryRes = EvaluateInDifferentType(SrcI, DestTy, false); |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7899 | APInt Mask(APInt::getBitsSet(DestBitSize, SrcBitSize, DestBitSize)); |
7900 | if (MaskedValueIsZero(TryRes, Mask)) | ||||
7901 | return ReplaceInstUsesWith(CI, TryRes); | ||||
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7902 | |
7903 | if (Instruction *TryI = dyn_cast<Instruction>(TryRes)) | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7904 | if (TryI->use_empty()) |
7905 | EraseInstFromFunction(*TryI); | ||||
7906 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7907 | break; |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7908 | } |
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7909 | case Instruction::SExt: { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7910 | DoXForm = NumCastsRemoved >= 2; |
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7911 | if (!DoXForm && !isa<TruncInst>(SrcI) && 0) { |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7912 | // If we do not have to emit the truncate + sext pair, then it's always |
7913 | // profitable to do this xform. | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7914 | // |
7915 | // It's not safe to eliminate the trunc + sext pair if one of the | ||||
7916 | // eliminated cast is a truncate. e.g. | ||||
7917 | // t2 = trunc i32 t1 to i16 | ||||
7918 | // t3 = sext i16 t2 to i32 | ||||
7919 | // != | ||||
7920 | // i32 t1 | ||||
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7921 | Value *TryRes = EvaluateInDifferentType(SrcI, DestTy, true); |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7922 | unsigned NumSignBits = ComputeNumSignBits(TryRes); |
7923 | if (NumSignBits > (DestBitSize - SrcBitSize)) | ||||
7924 | return ReplaceInstUsesWith(CI, TryRes); | ||||
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7925 | |
7926 | if (Instruction *TryI = dyn_cast<Instruction>(TryRes)) | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7927 | if (TryI->use_empty()) |
7928 | EraseInstFromFunction(*TryI); | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7929 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7930 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7931 | } |
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7932 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7933 | |
7934 | if (DoXForm) { | ||||
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7935 | DOUT << "ICE: EvaluateInDifferentType converting expression type to avoid" |
7936 | << " cast: " << CI; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7937 | Value *Res = EvaluateInDifferentType(SrcI, DestTy, |
7938 | CI.getOpcode() == Instruction::SExt); | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7939 | if (JustReplace) |
Chris Lattner | 3c0e6f4 | 2009-01-31 19:05:27 +0000 | [diff] [blame] | 7940 | // Just replace this cast with the result. |
7941 | return ReplaceInstUsesWith(CI, Res); | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7942 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7943 | assert(Res->getType() == DestTy); |
7944 | switch (CI.getOpcode()) { | ||||
7945 | default: assert(0 && "Unknown cast type!"); | ||||
7946 | case Instruction::Trunc: | ||||
7947 | case Instruction::BitCast: | ||||
7948 | // Just replace this cast with the result. | ||||
7949 | return ReplaceInstUsesWith(CI, Res); | ||||
7950 | case Instruction::ZExt: { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7951 | assert(SrcBitSize < DestBitSize && "Not a zext?"); |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7952 | |
7953 | // If the high bits are already zero, just replace this cast with the | ||||
7954 | // result. | ||||
7955 | APInt Mask(APInt::getBitsSet(DestBitSize, SrcBitSize, DestBitSize)); | ||||
7956 | if (MaskedValueIsZero(Res, Mask)) | ||||
7957 | return ReplaceInstUsesWith(CI, Res); | ||||
7958 | |||||
7959 | // We need to emit an AND to clear the high bits. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7960 | Constant *C = ConstantInt::get(APInt::getLowBitsSet(DestBitSize, |
7961 | SrcBitSize)); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 7962 | return BinaryOperator::CreateAnd(Res, C); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7963 | } |
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7964 | case Instruction::SExt: { |
7965 | // If the high bits are already filled with sign bit, just replace this | ||||
7966 | // cast with the result. | ||||
7967 | unsigned NumSignBits = ComputeNumSignBits(Res); | ||||
7968 | if (NumSignBits > (DestBitSize - SrcBitSize)) | ||||
Evan Cheng | 9ca34ab | 2009-01-15 17:01:23 +0000 | [diff] [blame] | 7969 | return ReplaceInstUsesWith(CI, Res); |
7970 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7971 | // 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] | 7972 | return CastInst::Create(Instruction::SExt, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7973 | InsertCastBefore(Instruction::Trunc, Res, Src->getType(), |
7974 | CI), DestTy); | ||||
7975 | } | ||||
Evan Cheng | 814a00c | 2009-01-16 02:11:43 +0000 | [diff] [blame] | 7976 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 7977 | } |
7978 | } | ||||
7979 | |||||
7980 | Value *Op0 = SrcI->getNumOperands() > 0 ? SrcI->getOperand(0) : 0; | ||||
7981 | Value *Op1 = SrcI->getNumOperands() > 1 ? SrcI->getOperand(1) : 0; | ||||
7982 | |||||
7983 | switch (SrcI->getOpcode()) { | ||||
7984 | case Instruction::Add: | ||||
7985 | case Instruction::Mul: | ||||
7986 | case Instruction::And: | ||||
7987 | case Instruction::Or: | ||||
7988 | case Instruction::Xor: | ||||
7989 | // If we are discarding information, rewrite. | ||||
7990 | if (DestBitSize <= SrcBitSize && DestBitSize != 1) { | ||||
7991 | // Don't insert two casts if they cannot be eliminated. We allow | ||||
7992 | // two casts to be inserted if the sizes are the same. This could | ||||
7993 | // only be converting signedness, which is a noop. | ||||
7994 | if (DestBitSize == SrcBitSize || | ||||
7995 | !ValueRequiresCast(CI.getOpcode(), Op1, DestTy,TD) || | ||||
7996 | !ValueRequiresCast(CI.getOpcode(), Op0, DestTy, TD)) { | ||||
7997 | Instruction::CastOps opcode = CI.getOpcode(); | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 7998 | Value *Op0c = InsertCastBefore(opcode, Op0, DestTy, *SrcI); |
7999 | Value *Op1c = InsertCastBefore(opcode, Op1, DestTy, *SrcI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8000 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8001 | cast<BinaryOperator>(SrcI)->getOpcode(), Op0c, Op1c); |
8002 | } | ||||
8003 | } | ||||
8004 | |||||
8005 | // cast (xor bool X, true) to int --> xor (cast bool X to int), 1 | ||||
8006 | if (isa<ZExtInst>(CI) && SrcBitSize == 1 && | ||||
8007 | SrcI->getOpcode() == Instruction::Xor && | ||||
8008 | Op1 == ConstantInt::getTrue() && | ||||
8009 | (!Op0->hasOneUse() || !isa<CmpInst>(Op0))) { | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 8010 | Value *New = InsertCastBefore(Instruction::ZExt, Op0, DestTy, CI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8011 | return BinaryOperator::CreateXor(New, ConstantInt::get(CI.getType(), 1)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8012 | } |
8013 | break; | ||||
8014 | case Instruction::SDiv: | ||||
8015 | case Instruction::UDiv: | ||||
8016 | case Instruction::SRem: | ||||
8017 | case Instruction::URem: | ||||
8018 | // If we are just changing the sign, rewrite. | ||||
8019 | if (DestBitSize == SrcBitSize) { | ||||
8020 | // Don't insert two casts if they cannot be eliminated. We allow | ||||
8021 | // two casts to be inserted if the sizes are the same. This could | ||||
8022 | // only be converting signedness, which is a noop. | ||||
8023 | if (!ValueRequiresCast(CI.getOpcode(), Op1, DestTy, TD) || | ||||
8024 | !ValueRequiresCast(CI.getOpcode(), Op0, DestTy, TD)) { | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 8025 | Value *Op0c = InsertCastBefore(Instruction::BitCast, |
8026 | Op0, DestTy, *SrcI); | ||||
8027 | Value *Op1c = InsertCastBefore(Instruction::BitCast, | ||||
8028 | Op1, DestTy, *SrcI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8029 | return BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8030 | cast<BinaryOperator>(SrcI)->getOpcode(), Op0c, Op1c); |
8031 | } | ||||
8032 | } | ||||
8033 | break; | ||||
8034 | |||||
8035 | case Instruction::Shl: | ||||
8036 | // Allow changing the sign of the source operand. Do not allow | ||||
8037 | // changing the size of the shift, UNLESS the shift amount is a | ||||
8038 | // constant. We must not change variable sized shifts to a smaller | ||||
8039 | // size, because it is undefined to shift more bits out than exist | ||||
8040 | // in the value. | ||||
8041 | if (DestBitSize == SrcBitSize || | ||||
8042 | (DestBitSize < SrcBitSize && isa<Constant>(Op1))) { | ||||
8043 | Instruction::CastOps opcode = (DestBitSize == SrcBitSize ? | ||||
8044 | Instruction::BitCast : Instruction::Trunc); | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 8045 | Value *Op0c = InsertCastBefore(opcode, Op0, DestTy, *SrcI); |
8046 | Value *Op1c = InsertCastBefore(opcode, Op1, DestTy, *SrcI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8047 | return BinaryOperator::CreateShl(Op0c, Op1c); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8048 | } |
8049 | break; | ||||
8050 | case Instruction::AShr: | ||||
8051 | // If this is a signed shr, and if all bits shifted in are about to be | ||||
8052 | // truncated off, turn it into an unsigned shr to allow greater | ||||
8053 | // simplifications. | ||||
8054 | if (DestBitSize < SrcBitSize && | ||||
8055 | isa<ConstantInt>(Op1)) { | ||||
8056 | uint32_t ShiftAmt = cast<ConstantInt>(Op1)->getLimitedValue(SrcBitSize); | ||||
8057 | if (SrcBitSize > ShiftAmt && SrcBitSize-ShiftAmt >= DestBitSize) { | ||||
8058 | // Insert the new logical shift right. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8059 | return BinaryOperator::CreateLShr(Op0, Op1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8060 | } |
8061 | } | ||||
8062 | break; | ||||
8063 | } | ||||
8064 | return 0; | ||||
8065 | } | ||||
8066 | |||||
8067 | Instruction *InstCombiner::visitTrunc(TruncInst &CI) { | ||||
8068 | if (Instruction *Result = commonIntCastTransforms(CI)) | ||||
8069 | return Result; | ||||
8070 | |||||
8071 | Value *Src = CI.getOperand(0); | ||||
8072 | const Type *Ty = CI.getType(); | ||||
8073 | uint32_t DestBitWidth = Ty->getPrimitiveSizeInBits(); | ||||
8074 | uint32_t SrcBitWidth = cast<IntegerType>(Src->getType())->getBitWidth(); | ||||
8075 | |||||
8076 | if (Instruction *SrcI = dyn_cast<Instruction>(Src)) { | ||||
8077 | switch (SrcI->getOpcode()) { | ||||
8078 | default: break; | ||||
8079 | case Instruction::LShr: | ||||
8080 | // We can shrink lshr to something smaller if we know the bits shifted in | ||||
8081 | // are already zeros. | ||||
8082 | if (ConstantInt *ShAmtV = dyn_cast<ConstantInt>(SrcI->getOperand(1))) { | ||||
8083 | uint32_t ShAmt = ShAmtV->getLimitedValue(SrcBitWidth); | ||||
8084 | |||||
8085 | // Get a mask for the bits shifting in. | ||||
8086 | APInt Mask(APInt::getLowBitsSet(SrcBitWidth, ShAmt).shl(DestBitWidth)); | ||||
8087 | Value* SrcIOp0 = SrcI->getOperand(0); | ||||
8088 | if (SrcI->hasOneUse() && MaskedValueIsZero(SrcIOp0, Mask)) { | ||||
8089 | if (ShAmt >= DestBitWidth) // All zeros. | ||||
8090 | return ReplaceInstUsesWith(CI, Constant::getNullValue(Ty)); | ||||
8091 | |||||
8092 | // Okay, we can shrink this. Truncate the input, then return a new | ||||
8093 | // shift. | ||||
8094 | Value *V1 = InsertCastBefore(Instruction::Trunc, SrcIOp0, Ty, CI); | ||||
8095 | Value *V2 = InsertCastBefore(Instruction::Trunc, SrcI->getOperand(1), | ||||
8096 | Ty, CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8097 | return BinaryOperator::CreateLShr(V1, V2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8098 | } |
8099 | } else { // This is a variable shr. | ||||
8100 | |||||
8101 | // Turn 'trunc (lshr X, Y) to bool' into '(X & (1 << Y)) != 0'. This is | ||||
8102 | // more LLVM instructions, but allows '1 << Y' to be hoisted if | ||||
8103 | // loop-invariant and CSE'd. | ||||
8104 | if (CI.getType() == Type::Int1Ty && SrcI->hasOneUse()) { | ||||
8105 | Value *One = ConstantInt::get(SrcI->getType(), 1); | ||||
8106 | |||||
8107 | Value *V = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8108 | BinaryOperator::CreateShl(One, SrcI->getOperand(1), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8109 | "tmp"), CI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8110 | V = InsertNewInstBefore(BinaryOperator::CreateAnd(V, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8111 | SrcI->getOperand(0), |
8112 | "tmp"), CI); | ||||
8113 | Value *Zero = Constant::getNullValue(V->getType()); | ||||
8114 | return new ICmpInst(ICmpInst::ICMP_NE, V, Zero); | ||||
8115 | } | ||||
8116 | } | ||||
8117 | break; | ||||
8118 | } | ||||
8119 | } | ||||
8120 | |||||
8121 | return 0; | ||||
8122 | } | ||||
8123 | |||||
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8124 | /// transformZExtICmp - Transform (zext icmp) to bitwise / integer operations |
8125 | /// in order to eliminate the icmp. | ||||
8126 | Instruction *InstCombiner::transformZExtICmp(ICmpInst *ICI, Instruction &CI, | ||||
8127 | bool DoXform) { | ||||
8128 | // If we are just checking for a icmp eq of a single bit and zext'ing it | ||||
8129 | // to an integer, then shift the bit to the appropriate place and then | ||||
8130 | // cast to integer to avoid the comparison. | ||||
8131 | if (ConstantInt *Op1C = dyn_cast<ConstantInt>(ICI->getOperand(1))) { | ||||
8132 | const APInt &Op1CV = Op1C->getValue(); | ||||
8133 | |||||
8134 | // zext (x <s 0) to i32 --> x>>u31 true if signbit set. | ||||
8135 | // zext (x >s -1) to i32 --> (x>>u31)^1 true if signbit clear. | ||||
8136 | if ((ICI->getPredicate() == ICmpInst::ICMP_SLT && Op1CV == 0) || | ||||
8137 | (ICI->getPredicate() == ICmpInst::ICMP_SGT &&Op1CV.isAllOnesValue())) { | ||||
8138 | if (!DoXform) return ICI; | ||||
8139 | |||||
8140 | Value *In = ICI->getOperand(0); | ||||
8141 | Value *Sh = ConstantInt::get(In->getType(), | ||||
8142 | In->getType()->getPrimitiveSizeInBits()-1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8143 | In = InsertNewInstBefore(BinaryOperator::CreateLShr(In, Sh, |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8144 | In->getName()+".lobit"), |
8145 | CI); | ||||
8146 | if (In->getType() != CI.getType()) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8147 | In = CastInst::CreateIntegerCast(In, CI.getType(), |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8148 | false/*ZExt*/, "tmp", &CI); |
8149 | |||||
8150 | if (ICI->getPredicate() == ICmpInst::ICMP_SGT) { | ||||
8151 | Constant *One = ConstantInt::get(In->getType(), 1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8152 | In = InsertNewInstBefore(BinaryOperator::CreateXor(In, One, |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8153 | In->getName()+".not"), |
8154 | CI); | ||||
8155 | } | ||||
8156 | |||||
8157 | return ReplaceInstUsesWith(CI, In); | ||||
8158 | } | ||||
8159 | |||||
8160 | |||||
8161 | |||||
8162 | // zext (X == 0) to i32 --> X^1 iff X has only the low bit set. | ||||
8163 | // zext (X == 0) to i32 --> (X>>1)^1 iff X has only the 2nd bit set. | ||||
8164 | // zext (X == 1) to i32 --> X iff X has only the low bit set. | ||||
8165 | // zext (X == 2) to i32 --> X>>1 iff X has only the 2nd bit set. | ||||
8166 | // zext (X != 0) to i32 --> X iff X has only the low bit set. | ||||
8167 | // zext (X != 0) to i32 --> X>>1 iff X has only the 2nd bit set. | ||||
8168 | // zext (X != 1) to i32 --> X^1 iff X has only the low bit set. | ||||
8169 | // zext (X != 2) to i32 --> (X>>1)^1 iff X has only the 2nd bit set. | ||||
8170 | if ((Op1CV == 0 || Op1CV.isPowerOf2()) && | ||||
8171 | // This only works for EQ and NE | ||||
8172 | ICI->isEquality()) { | ||||
8173 | // If Op1C some other power of two, convert: | ||||
8174 | uint32_t BitWidth = Op1C->getType()->getBitWidth(); | ||||
8175 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
8176 | APInt TypeMask(APInt::getAllOnesValue(BitWidth)); | ||||
8177 | ComputeMaskedBits(ICI->getOperand(0), TypeMask, KnownZero, KnownOne); | ||||
8178 | |||||
8179 | APInt KnownZeroMask(~KnownZero); | ||||
8180 | if (KnownZeroMask.isPowerOf2()) { // Exactly 1 possible 1? | ||||
8181 | if (!DoXform) return ICI; | ||||
8182 | |||||
8183 | bool isNE = ICI->getPredicate() == ICmpInst::ICMP_NE; | ||||
8184 | if (Op1CV != 0 && (Op1CV != KnownZeroMask)) { | ||||
8185 | // (X&4) == 2 --> false | ||||
8186 | // (X&4) != 2 --> true | ||||
8187 | Constant *Res = ConstantInt::get(Type::Int1Ty, isNE); | ||||
8188 | Res = ConstantExpr::getZExt(Res, CI.getType()); | ||||
8189 | return ReplaceInstUsesWith(CI, Res); | ||||
8190 | } | ||||
8191 | |||||
8192 | uint32_t ShiftAmt = KnownZeroMask.logBase2(); | ||||
8193 | Value *In = ICI->getOperand(0); | ||||
8194 | if (ShiftAmt) { | ||||
8195 | // Perform a logical shr by shiftamt. | ||||
8196 | // Insert the shift to put the result in the low bit. | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8197 | In = InsertNewInstBefore(BinaryOperator::CreateLShr(In, |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8198 | ConstantInt::get(In->getType(), ShiftAmt), |
8199 | In->getName()+".lobit"), CI); | ||||
8200 | } | ||||
8201 | |||||
8202 | if ((Op1CV != 0) == isNE) { // Toggle the low bit. | ||||
8203 | Constant *One = ConstantInt::get(In->getType(), 1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8204 | In = BinaryOperator::CreateXor(In, One, "tmp"); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8205 | InsertNewInstBefore(cast<Instruction>(In), CI); |
8206 | } | ||||
8207 | |||||
8208 | if (CI.getType() == In->getType()) | ||||
8209 | return ReplaceInstUsesWith(CI, In); | ||||
8210 | else | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8211 | return CastInst::CreateIntegerCast(In, CI.getType(), false/*ZExt*/); |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8212 | } |
8213 | } | ||||
8214 | } | ||||
8215 | |||||
8216 | return 0; | ||||
8217 | } | ||||
8218 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8219 | Instruction *InstCombiner::visitZExt(ZExtInst &CI) { |
8220 | // If one of the common conversion will work .. | ||||
8221 | if (Instruction *Result = commonIntCastTransforms(CI)) | ||||
8222 | return Result; | ||||
8223 | |||||
8224 | Value *Src = CI.getOperand(0); | ||||
8225 | |||||
8226 | // If this is a cast of a cast | ||||
8227 | if (CastInst *CSrc = dyn_cast<CastInst>(Src)) { // A->B->C cast | ||||
8228 | // If this is a TRUNC followed by a ZEXT then we are dealing with integral | ||||
8229 | // types and if the sizes are just right we can convert this into a logical | ||||
8230 | // 'and' which will be much cheaper than the pair of casts. | ||||
8231 | if (isa<TruncInst>(CSrc)) { | ||||
8232 | // Get the sizes of the types involved | ||||
8233 | Value *A = CSrc->getOperand(0); | ||||
8234 | uint32_t SrcSize = A->getType()->getPrimitiveSizeInBits(); | ||||
8235 | uint32_t MidSize = CSrc->getType()->getPrimitiveSizeInBits(); | ||||
8236 | uint32_t DstSize = CI.getType()->getPrimitiveSizeInBits(); | ||||
8237 | // If we're actually extending zero bits and the trunc is a no-op | ||||
8238 | if (MidSize < DstSize && SrcSize == DstSize) { | ||||
8239 | // Replace both of the casts with an And of the type mask. | ||||
8240 | APInt AndValue(APInt::getLowBitsSet(SrcSize, MidSize)); | ||||
8241 | Constant *AndConst = ConstantInt::get(AndValue); | ||||
8242 | Instruction *And = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8243 | BinaryOperator::CreateAnd(CSrc->getOperand(0), AndConst); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8244 | // Unfortunately, if the type changed, we need to cast it back. |
8245 | if (And->getType() != CI.getType()) { | ||||
8246 | And->setName(CSrc->getName()+".mask"); | ||||
8247 | InsertNewInstBefore(And, CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8248 | And = CastInst::CreateIntegerCast(And, CI.getType(), false/*ZExt*/); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8249 | } |
8250 | return And; | ||||
8251 | } | ||||
8252 | } | ||||
8253 | } | ||||
8254 | |||||
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8255 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(Src)) |
8256 | return transformZExtICmp(ICI, CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8257 | |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8258 | BinaryOperator *SrcI = dyn_cast<BinaryOperator>(Src); |
8259 | if (SrcI && SrcI->getOpcode() == Instruction::Or) { | ||||
8260 | // zext (or icmp, icmp) --> or (zext icmp), (zext icmp) if at least one | ||||
8261 | // of the (zext icmp) will be transformed. | ||||
8262 | ICmpInst *LHS = dyn_cast<ICmpInst>(SrcI->getOperand(0)); | ||||
8263 | ICmpInst *RHS = dyn_cast<ICmpInst>(SrcI->getOperand(1)); | ||||
8264 | if (LHS && RHS && LHS->hasOneUse() && RHS->hasOneUse() && | ||||
8265 | (transformZExtICmp(LHS, CI, false) || | ||||
8266 | transformZExtICmp(RHS, CI, false))) { | ||||
8267 | Value *LCast = InsertCastBefore(Instruction::ZExt, LHS, CI.getType(), CI); | ||||
8268 | Value *RCast = InsertCastBefore(Instruction::ZExt, RHS, CI.getType(), CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8269 | return BinaryOperator::Create(Instruction::Or, LCast, RCast); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8270 | } |
Evan Cheng | e3779cf | 2008-03-24 00:21:34 +0000 | [diff] [blame] | 8271 | } |
8272 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8273 | return 0; |
8274 | } | ||||
8275 | |||||
8276 | Instruction *InstCombiner::visitSExt(SExtInst &CI) { | ||||
8277 | if (Instruction *I = commonIntCastTransforms(CI)) | ||||
8278 | return I; | ||||
8279 | |||||
8280 | Value *Src = CI.getOperand(0); | ||||
8281 | |||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8282 | // Canonicalize sign-extend from i1 to a select. |
8283 | if (Src->getType() == Type::Int1Ty) | ||||
8284 | return SelectInst::Create(Src, | ||||
8285 | ConstantInt::getAllOnesValue(CI.getType()), | ||||
8286 | Constant::getNullValue(CI.getType())); | ||||
Dan Gohman | f0f1202 | 2008-05-20 21:01:12 +0000 | [diff] [blame] | 8287 | |
8288 | // See if the value being truncated is already sign extended. If so, just | ||||
8289 | // eliminate the trunc/sext pair. | ||||
8290 | if (getOpcode(Src) == Instruction::Trunc) { | ||||
8291 | Value *Op = cast<User>(Src)->getOperand(0); | ||||
8292 | unsigned OpBits = cast<IntegerType>(Op->getType())->getBitWidth(); | ||||
8293 | unsigned MidBits = cast<IntegerType>(Src->getType())->getBitWidth(); | ||||
8294 | unsigned DestBits = cast<IntegerType>(CI.getType())->getBitWidth(); | ||||
8295 | unsigned NumSignBits = ComputeNumSignBits(Op); | ||||
8296 | |||||
8297 | if (OpBits == DestBits) { | ||||
8298 | // Op is i32, Mid is i8, and Dest is i32. If Op has more than 24 sign | ||||
8299 | // bits, it is already ready. | ||||
8300 | if (NumSignBits > DestBits-MidBits) | ||||
8301 | return ReplaceInstUsesWith(CI, Op); | ||||
8302 | } else if (OpBits < DestBits) { | ||||
8303 | // Op is i32, Mid is i8, and Dest is i64. If Op has more than 24 sign | ||||
8304 | // bits, just sext from i32. | ||||
8305 | if (NumSignBits > OpBits-MidBits) | ||||
8306 | return new SExtInst(Op, CI.getType(), "tmp"); | ||||
8307 | } else { | ||||
8308 | // Op is i64, Mid is i8, and Dest is i32. If Op has more than 56 sign | ||||
8309 | // bits, just truncate to i32. | ||||
8310 | if (NumSignBits > OpBits-MidBits) | ||||
8311 | return new TruncInst(Op, CI.getType(), "tmp"); | ||||
8312 | } | ||||
8313 | } | ||||
Chris Lattner | 8a2d059 | 2008-08-06 07:35:52 +0000 | [diff] [blame] | 8314 | |
8315 | // If the input is a shl/ashr pair of a same constant, then this is a sign | ||||
8316 | // extension from a smaller value. If we could trust arbitrary bitwidth | ||||
8317 | // integers, we could turn this into a truncate to the smaller bit and then | ||||
8318 | // use a sext for the whole extension. Since we don't, look deeper and check | ||||
8319 | // for a truncate. If the source and dest are the same type, eliminate the | ||||
8320 | // trunc and extend and just do shifts. For example, turn: | ||||
8321 | // %a = trunc i32 %i to i8 | ||||
8322 | // %b = shl i8 %a, 6 | ||||
8323 | // %c = ashr i8 %b, 6 | ||||
8324 | // %d = sext i8 %c to i32 | ||||
8325 | // into: | ||||
8326 | // %a = shl i32 %i, 30 | ||||
8327 | // %d = ashr i32 %a, 30 | ||||
8328 | Value *A = 0; | ||||
8329 | ConstantInt *BA = 0, *CA = 0; | ||||
8330 | if (match(Src, m_AShr(m_Shl(m_Value(A), m_ConstantInt(BA)), | ||||
8331 | m_ConstantInt(CA))) && | ||||
8332 | BA == CA && isa<TruncInst>(A)) { | ||||
8333 | Value *I = cast<TruncInst>(A)->getOperand(0); | ||||
8334 | if (I->getType() == CI.getType()) { | ||||
8335 | unsigned MidSize = Src->getType()->getPrimitiveSizeInBits(); | ||||
8336 | unsigned SrcDstSize = CI.getType()->getPrimitiveSizeInBits(); | ||||
8337 | unsigned ShAmt = CA->getZExtValue()+SrcDstSize-MidSize; | ||||
8338 | Constant *ShAmtV = ConstantInt::get(CI.getType(), ShAmt); | ||||
8339 | I = InsertNewInstBefore(BinaryOperator::CreateShl(I, ShAmtV, | ||||
8340 | CI.getName()), CI); | ||||
8341 | return BinaryOperator::CreateAShr(I, ShAmtV); | ||||
8342 | } | ||||
8343 | } | ||||
8344 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8345 | return 0; |
8346 | } | ||||
8347 | |||||
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8348 | /// FitsInFPType - Return a Constant* for the specified FP constant if it fits |
8349 | /// in the specified FP type without changing its value. | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 8350 | static Constant *FitsInFPType(ConstantFP *CFP, const fltSemantics &Sem) { |
Dale Johannesen | 6e547b4 | 2008-10-09 23:00:39 +0000 | [diff] [blame] | 8351 | bool losesInfo; |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8352 | APFloat F = CFP->getValueAPF(); |
Dale Johannesen | 6e547b4 | 2008-10-09 23:00:39 +0000 | [diff] [blame] | 8353 | (void)F.convert(Sem, APFloat::rmNearestTiesToEven, &losesInfo); |
8354 | if (!losesInfo) | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 8355 | return ConstantFP::get(F); |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8356 | return 0; |
8357 | } | ||||
8358 | |||||
8359 | /// LookThroughFPExtensions - If this is an fp extension instruction, look | ||||
8360 | /// through it until we get the source value. | ||||
8361 | static Value *LookThroughFPExtensions(Value *V) { | ||||
8362 | if (Instruction *I = dyn_cast<Instruction>(V)) | ||||
8363 | if (I->getOpcode() == Instruction::FPExt) | ||||
8364 | return LookThroughFPExtensions(I->getOperand(0)); | ||||
8365 | |||||
8366 | // If this value is a constant, return the constant in the smallest FP type | ||||
8367 | // that can accurately represent it. This allows us to turn | ||||
8368 | // (float)((double)X+2.0) into x+2.0f. | ||||
8369 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(V)) { | ||||
8370 | if (CFP->getType() == Type::PPC_FP128Ty) | ||||
8371 | return V; // No constant folding of this. | ||||
8372 | // See if the value can be truncated to float and then reextended. | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 8373 | if (Value *V = FitsInFPType(CFP, APFloat::IEEEsingle)) |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8374 | return V; |
8375 | if (CFP->getType() == Type::DoubleTy) | ||||
8376 | return V; // Won't shrink. | ||||
Chris Lattner | 5e0610f | 2008-04-20 00:41:09 +0000 | [diff] [blame] | 8377 | if (Value *V = FitsInFPType(CFP, APFloat::IEEEdouble)) |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8378 | return V; |
8379 | // Don't try to shrink to various long double types. | ||||
8380 | } | ||||
8381 | |||||
8382 | return V; | ||||
8383 | } | ||||
8384 | |||||
8385 | Instruction *InstCombiner::visitFPTrunc(FPTruncInst &CI) { | ||||
8386 | if (Instruction *I = commonCastTransforms(CI)) | ||||
8387 | return I; | ||||
8388 | |||||
8389 | // If we have fptrunc(add (fpextend x), (fpextend y)), where x and y are | ||||
8390 | // smaller than the destination type, we can eliminate the truncate by doing | ||||
8391 | // the add as the smaller type. This applies to add/sub/mul/div as well as | ||||
8392 | // many builtins (sqrt, etc). | ||||
8393 | BinaryOperator *OpI = dyn_cast<BinaryOperator>(CI.getOperand(0)); | ||||
8394 | if (OpI && OpI->hasOneUse()) { | ||||
8395 | switch (OpI->getOpcode()) { | ||||
8396 | default: break; | ||||
8397 | case Instruction::Add: | ||||
8398 | case Instruction::Sub: | ||||
8399 | case Instruction::Mul: | ||||
8400 | case Instruction::FDiv: | ||||
8401 | case Instruction::FRem: | ||||
8402 | const Type *SrcTy = OpI->getType(); | ||||
8403 | Value *LHSTrunc = LookThroughFPExtensions(OpI->getOperand(0)); | ||||
8404 | Value *RHSTrunc = LookThroughFPExtensions(OpI->getOperand(1)); | ||||
8405 | if (LHSTrunc->getType() != SrcTy && | ||||
8406 | RHSTrunc->getType() != SrcTy) { | ||||
8407 | unsigned DstSize = CI.getType()->getPrimitiveSizeInBits(); | ||||
8408 | // If the source types were both smaller than the destination type of | ||||
8409 | // the cast, do this xform. | ||||
8410 | if (LHSTrunc->getType()->getPrimitiveSizeInBits() <= DstSize && | ||||
8411 | RHSTrunc->getType()->getPrimitiveSizeInBits() <= DstSize) { | ||||
8412 | LHSTrunc = InsertCastBefore(Instruction::FPExt, LHSTrunc, | ||||
8413 | CI.getType(), CI); | ||||
8414 | RHSTrunc = InsertCastBefore(Instruction::FPExt, RHSTrunc, | ||||
8415 | CI.getType(), CI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8416 | return BinaryOperator::Create(OpI->getOpcode(), LHSTrunc, RHSTrunc); |
Chris Lattner | df7e840 | 2008-01-27 05:29:54 +0000 | [diff] [blame] | 8417 | } |
8418 | } | ||||
8419 | break; | ||||
8420 | } | ||||
8421 | } | ||||
8422 | return 0; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8423 | } |
8424 | |||||
8425 | Instruction *InstCombiner::visitFPExt(CastInst &CI) { | ||||
8426 | return commonCastTransforms(CI); | ||||
8427 | } | ||||
8428 | |||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8429 | Instruction *InstCombiner::visitFPToUI(FPToUIInst &FI) { |
Chris Lattner | 5f4d691 | 2008-08-06 05:13:06 +0000 | [diff] [blame] | 8430 | Instruction *OpI = dyn_cast<Instruction>(FI.getOperand(0)); |
8431 | if (OpI == 0) | ||||
8432 | return commonCastTransforms(FI); | ||||
8433 | |||||
8434 | // fptoui(uitofp(X)) --> X | ||||
8435 | // fptoui(sitofp(X)) --> X | ||||
8436 | // This is safe if the intermediate type has enough bits in its mantissa to | ||||
8437 | // accurately represent all values of X. For example, do not do this with | ||||
8438 | // i64->float->i64. This is also safe for sitofp case, because any negative | ||||
8439 | // 'X' value would cause an undefined result for the fptoui. | ||||
8440 | if ((isa<UIToFPInst>(OpI) || isa<SIToFPInst>(OpI)) && | ||||
8441 | OpI->getOperand(0)->getType() == FI.getType() && | ||||
8442 | (int)FI.getType()->getPrimitiveSizeInBits() < /*extra bit for sign */ | ||||
8443 | OpI->getType()->getFPMantissaWidth()) | ||||
8444 | return ReplaceInstUsesWith(FI, OpI->getOperand(0)); | ||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8445 | |
8446 | return commonCastTransforms(FI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8447 | } |
8448 | |||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8449 | Instruction *InstCombiner::visitFPToSI(FPToSIInst &FI) { |
Chris Lattner | 5f4d691 | 2008-08-06 05:13:06 +0000 | [diff] [blame] | 8450 | Instruction *OpI = dyn_cast<Instruction>(FI.getOperand(0)); |
8451 | if (OpI == 0) | ||||
8452 | return commonCastTransforms(FI); | ||||
8453 | |||||
8454 | // fptosi(sitofp(X)) --> X | ||||
8455 | // fptosi(uitofp(X)) --> X | ||||
8456 | // This is safe if the intermediate type has enough bits in its mantissa to | ||||
8457 | // accurately represent all values of X. For example, do not do this with | ||||
8458 | // i64->float->i64. This is also safe for sitofp case, because any negative | ||||
8459 | // 'X' value would cause an undefined result for the fptoui. | ||||
8460 | if ((isa<UIToFPInst>(OpI) || isa<SIToFPInst>(OpI)) && | ||||
8461 | OpI->getOperand(0)->getType() == FI.getType() && | ||||
8462 | (int)FI.getType()->getPrimitiveSizeInBits() <= | ||||
8463 | OpI->getType()->getFPMantissaWidth()) | ||||
8464 | return ReplaceInstUsesWith(FI, OpI->getOperand(0)); | ||||
Chris Lattner | deef1a7 | 2008-05-19 20:25:04 +0000 | [diff] [blame] | 8465 | |
8466 | return commonCastTransforms(FI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8467 | } |
8468 | |||||
8469 | Instruction *InstCombiner::visitUIToFP(CastInst &CI) { | ||||
8470 | return commonCastTransforms(CI); | ||||
8471 | } | ||||
8472 | |||||
8473 | Instruction *InstCombiner::visitSIToFP(CastInst &CI) { | ||||
8474 | return commonCastTransforms(CI); | ||||
8475 | } | ||||
8476 | |||||
8477 | Instruction *InstCombiner::visitPtrToInt(CastInst &CI) { | ||||
8478 | return commonPointerCastTransforms(CI); | ||||
8479 | } | ||||
8480 | |||||
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8481 | Instruction *InstCombiner::visitIntToPtr(IntToPtrInst &CI) { |
8482 | if (Instruction *I = commonCastTransforms(CI)) | ||||
8483 | return I; | ||||
8484 | |||||
8485 | const Type *DestPointee = cast<PointerType>(CI.getType())->getElementType(); | ||||
8486 | if (!DestPointee->isSized()) return 0; | ||||
8487 | |||||
8488 | // If this is inttoptr(add (ptrtoint x), cst), try to turn this into a GEP. | ||||
8489 | ConstantInt *Cst; | ||||
8490 | Value *X; | ||||
8491 | if (match(CI.getOperand(0), m_Add(m_Cast<PtrToIntInst>(m_Value(X)), | ||||
8492 | m_ConstantInt(Cst)))) { | ||||
8493 | // If the source and destination operands have the same type, see if this | ||||
8494 | // is a single-index GEP. | ||||
8495 | if (X->getType() == CI.getType()) { | ||||
8496 | // Get the size of the pointee type. | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 8497 | uint64_t Size = TD->getTypePaddedSize(DestPointee); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8498 | |
8499 | // Convert the constant to intptr type. | ||||
8500 | APInt Offset = Cst->getValue(); | ||||
8501 | Offset.sextOrTrunc(TD->getPointerSizeInBits()); | ||||
8502 | |||||
8503 | // If Offset is evenly divisible by Size, we can do this xform. | ||||
8504 | if (Size && !APIntOps::srem(Offset, APInt(Offset.getBitWidth(), Size))){ | ||||
8505 | Offset = APIntOps::sdiv(Offset, APInt(Offset.getBitWidth(), Size)); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8506 | return GetElementPtrInst::Create(X, ConstantInt::get(Offset)); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8507 | } |
8508 | } | ||||
8509 | // TODO: Could handle other cases, e.g. where add is indexing into field of | ||||
8510 | // struct etc. | ||||
8511 | } else if (CI.getOperand(0)->hasOneUse() && | ||||
8512 | match(CI.getOperand(0), m_Add(m_Value(X), m_ConstantInt(Cst)))) { | ||||
8513 | // Otherwise, if this is inttoptr(add x, cst), try to turn this into an | ||||
8514 | // "inttoptr+GEP" instead of "add+intptr". | ||||
8515 | |||||
8516 | // Get the size of the pointee type. | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 8517 | uint64_t Size = TD->getTypePaddedSize(DestPointee); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8518 | |
8519 | // Convert the constant to intptr type. | ||||
8520 | APInt Offset = Cst->getValue(); | ||||
8521 | Offset.sextOrTrunc(TD->getPointerSizeInBits()); | ||||
8522 | |||||
8523 | // If Offset is evenly divisible by Size, we can do this xform. | ||||
8524 | if (Size && !APIntOps::srem(Offset, APInt(Offset.getBitWidth(), Size))){ | ||||
8525 | Offset = APIntOps::sdiv(Offset, APInt(Offset.getBitWidth(), Size)); | ||||
8526 | |||||
8527 | Instruction *P = InsertNewInstBefore(new IntToPtrInst(X, CI.getType(), | ||||
8528 | "tmp"), CI); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8529 | return GetElementPtrInst::Create(P, ConstantInt::get(Offset), "tmp"); |
Chris Lattner | 7c162648 | 2008-01-08 07:23:51 +0000 | [diff] [blame] | 8530 | } |
8531 | } | ||||
8532 | return 0; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8533 | } |
8534 | |||||
8535 | Instruction *InstCombiner::visitBitCast(BitCastInst &CI) { | ||||
8536 | // If the operands are integer typed then apply the integer transforms, | ||||
8537 | // otherwise just apply the common ones. | ||||
8538 | Value *Src = CI.getOperand(0); | ||||
8539 | const Type *SrcTy = Src->getType(); | ||||
8540 | const Type *DestTy = CI.getType(); | ||||
8541 | |||||
8542 | if (SrcTy->isInteger() && DestTy->isInteger()) { | ||||
8543 | if (Instruction *Result = commonIntCastTransforms(CI)) | ||||
8544 | return Result; | ||||
8545 | } else if (isa<PointerType>(SrcTy)) { | ||||
8546 | if (Instruction *I = commonPointerCastTransforms(CI)) | ||||
8547 | return I; | ||||
8548 | } else { | ||||
8549 | if (Instruction *Result = commonCastTransforms(CI)) | ||||
8550 | return Result; | ||||
8551 | } | ||||
8552 | |||||
8553 | |||||
8554 | // Get rid of casts from one type to the same type. These are useless and can | ||||
8555 | // be replaced by the operand. | ||||
8556 | if (DestTy == Src->getType()) | ||||
8557 | return ReplaceInstUsesWith(CI, Src); | ||||
8558 | |||||
8559 | if (const PointerType *DstPTy = dyn_cast<PointerType>(DestTy)) { | ||||
8560 | const PointerType *SrcPTy = cast<PointerType>(SrcTy); | ||||
8561 | const Type *DstElTy = DstPTy->getElementType(); | ||||
8562 | const Type *SrcElTy = SrcPTy->getElementType(); | ||||
8563 | |||||
Nate Begeman | df5b361 | 2008-03-31 00:22:16 +0000 | [diff] [blame] | 8564 | // If the address spaces don't match, don't eliminate the bitcast, which is |
8565 | // required for changing types. | ||||
8566 | if (SrcPTy->getAddressSpace() != DstPTy->getAddressSpace()) | ||||
8567 | return 0; | ||||
8568 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8569 | // If we are casting a malloc or alloca to a pointer to a type of the same |
8570 | // size, rewrite the allocation instruction to allocate the "right" type. | ||||
8571 | if (AllocationInst *AI = dyn_cast<AllocationInst>(Src)) | ||||
8572 | if (Instruction *V = PromoteCastOfAllocation(CI, *AI)) | ||||
8573 | return V; | ||||
8574 | |||||
8575 | // If the source and destination are pointers, and this cast is equivalent | ||||
8576 | // to a getelementptr X, 0, 0, 0... turn it into the appropriate gep. | ||||
8577 | // This can enhance SROA and other transforms that want type-safe pointers. | ||||
8578 | Constant *ZeroUInt = Constant::getNullValue(Type::Int32Ty); | ||||
8579 | unsigned NumZeros = 0; | ||||
8580 | while (SrcElTy != DstElTy && | ||||
8581 | isa<CompositeType>(SrcElTy) && !isa<PointerType>(SrcElTy) && | ||||
8582 | SrcElTy->getNumContainedTypes() /* not "{}" */) { | ||||
8583 | SrcElTy = cast<CompositeType>(SrcElTy)->getTypeAtIndex(ZeroUInt); | ||||
8584 | ++NumZeros; | ||||
8585 | } | ||||
8586 | |||||
8587 | // If we found a path from the src to dest, create the getelementptr now. | ||||
8588 | if (SrcElTy == DstElTy) { | ||||
8589 | SmallVector<Value*, 8> Idxs(NumZeros+1, ZeroUInt); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8590 | return GetElementPtrInst::Create(Src, Idxs.begin(), Idxs.end(), "", |
8591 | ((Instruction*) NULL)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8592 | } |
8593 | } | ||||
8594 | |||||
8595 | if (ShuffleVectorInst *SVI = dyn_cast<ShuffleVectorInst>(Src)) { | ||||
8596 | if (SVI->hasOneUse()) { | ||||
8597 | // Okay, we have (bitconvert (shuffle ..)). Check to see if this is | ||||
8598 | // a bitconvert to a vector with the same # elts. | ||||
8599 | if (isa<VectorType>(DestTy) && | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 8600 | cast<VectorType>(DestTy)->getNumElements() == |
8601 | SVI->getType()->getNumElements() && | ||||
8602 | SVI->getType()->getNumElements() == | ||||
8603 | cast<VectorType>(SVI->getOperand(0)->getType())->getNumElements()) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8604 | CastInst *Tmp; |
8605 | // If either of the operands is a cast from CI.getType(), then | ||||
8606 | // evaluating the shuffle in the casted destination's type will allow | ||||
8607 | // us to eliminate at least one cast. | ||||
8608 | if (((Tmp = dyn_cast<CastInst>(SVI->getOperand(0))) && | ||||
8609 | Tmp->getOperand(0)->getType() == DestTy) || | ||||
8610 | ((Tmp = dyn_cast<CastInst>(SVI->getOperand(1))) && | ||||
8611 | Tmp->getOperand(0)->getType() == DestTy)) { | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 8612 | Value *LHS = InsertCastBefore(Instruction::BitCast, |
8613 | SVI->getOperand(0), DestTy, CI); | ||||
8614 | Value *RHS = InsertCastBefore(Instruction::BitCast, | ||||
8615 | SVI->getOperand(1), DestTy, CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8616 | // Return a new shuffle vector. Use the same element ID's, as we |
8617 | // know the vector types match #elts. | ||||
8618 | return new ShuffleVectorInst(LHS, RHS, SVI->getOperand(2)); | ||||
8619 | } | ||||
8620 | } | ||||
8621 | } | ||||
8622 | } | ||||
8623 | return 0; | ||||
8624 | } | ||||
8625 | |||||
8626 | /// GetSelectFoldableOperands - We want to turn code that looks like this: | ||||
8627 | /// %C = or %A, %B | ||||
8628 | /// %D = select %cond, %C, %A | ||||
8629 | /// into: | ||||
8630 | /// %C = select %cond, %B, 0 | ||||
8631 | /// %D = or %A, %C | ||||
8632 | /// | ||||
8633 | /// Assuming that the specified instruction is an operand to the select, return | ||||
8634 | /// a bitmask indicating which operands of this instruction are foldable if they | ||||
8635 | /// equal the other incoming value of the select. | ||||
8636 | /// | ||||
8637 | static unsigned GetSelectFoldableOperands(Instruction *I) { | ||||
8638 | switch (I->getOpcode()) { | ||||
8639 | case Instruction::Add: | ||||
8640 | case Instruction::Mul: | ||||
8641 | case Instruction::And: | ||||
8642 | case Instruction::Or: | ||||
8643 | case Instruction::Xor: | ||||
8644 | return 3; // Can fold through either operand. | ||||
8645 | case Instruction::Sub: // Can only fold on the amount subtracted. | ||||
8646 | case Instruction::Shl: // Can only fold on the shift amount. | ||||
8647 | case Instruction::LShr: | ||||
8648 | case Instruction::AShr: | ||||
8649 | return 1; | ||||
8650 | default: | ||||
8651 | return 0; // Cannot fold | ||||
8652 | } | ||||
8653 | } | ||||
8654 | |||||
8655 | /// GetSelectFoldableConstant - For the same transformation as the previous | ||||
8656 | /// function, return the identity constant that goes into the select. | ||||
8657 | static Constant *GetSelectFoldableConstant(Instruction *I) { | ||||
8658 | switch (I->getOpcode()) { | ||||
8659 | default: assert(0 && "This cannot happen!"); abort(); | ||||
8660 | case Instruction::Add: | ||||
8661 | case Instruction::Sub: | ||||
8662 | case Instruction::Or: | ||||
8663 | case Instruction::Xor: | ||||
8664 | case Instruction::Shl: | ||||
8665 | case Instruction::LShr: | ||||
8666 | case Instruction::AShr: | ||||
8667 | return Constant::getNullValue(I->getType()); | ||||
8668 | case Instruction::And: | ||||
8669 | return Constant::getAllOnesValue(I->getType()); | ||||
8670 | case Instruction::Mul: | ||||
8671 | return ConstantInt::get(I->getType(), 1); | ||||
8672 | } | ||||
8673 | } | ||||
8674 | |||||
8675 | /// FoldSelectOpOp - Here we have (select c, TI, FI), and we know that TI and FI | ||||
8676 | /// have the same opcode and only one use each. Try to simplify this. | ||||
8677 | Instruction *InstCombiner::FoldSelectOpOp(SelectInst &SI, Instruction *TI, | ||||
8678 | Instruction *FI) { | ||||
8679 | if (TI->getNumOperands() == 1) { | ||||
8680 | // If this is a non-volatile load or a cast from the same type, | ||||
8681 | // merge. | ||||
8682 | if (TI->isCast()) { | ||||
8683 | if (TI->getOperand(0)->getType() != FI->getOperand(0)->getType()) | ||||
8684 | return 0; | ||||
8685 | } else { | ||||
8686 | return 0; // unknown unary op. | ||||
8687 | } | ||||
8688 | |||||
8689 | // Fold this by inserting a select from the input values. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8690 | SelectInst *NewSI = SelectInst::Create(SI.getCondition(), TI->getOperand(0), |
8691 | FI->getOperand(0), SI.getName()+".v"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8692 | InsertNewInstBefore(NewSI, SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8693 | return CastInst::Create(Instruction::CastOps(TI->getOpcode()), NewSI, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8694 | TI->getType()); |
8695 | } | ||||
8696 | |||||
8697 | // Only handle binary operators here. | ||||
8698 | if (!isa<BinaryOperator>(TI)) | ||||
8699 | return 0; | ||||
8700 | |||||
8701 | // Figure out if the operations have any operands in common. | ||||
8702 | Value *MatchOp, *OtherOpT, *OtherOpF; | ||||
8703 | bool MatchIsOpZero; | ||||
8704 | if (TI->getOperand(0) == FI->getOperand(0)) { | ||||
8705 | MatchOp = TI->getOperand(0); | ||||
8706 | OtherOpT = TI->getOperand(1); | ||||
8707 | OtherOpF = FI->getOperand(1); | ||||
8708 | MatchIsOpZero = true; | ||||
8709 | } else if (TI->getOperand(1) == FI->getOperand(1)) { | ||||
8710 | MatchOp = TI->getOperand(1); | ||||
8711 | OtherOpT = TI->getOperand(0); | ||||
8712 | OtherOpF = FI->getOperand(0); | ||||
8713 | MatchIsOpZero = false; | ||||
8714 | } else if (!TI->isCommutative()) { | ||||
8715 | return 0; | ||||
8716 | } else if (TI->getOperand(0) == FI->getOperand(1)) { | ||||
8717 | MatchOp = TI->getOperand(0); | ||||
8718 | OtherOpT = TI->getOperand(1); | ||||
8719 | OtherOpF = FI->getOperand(0); | ||||
8720 | MatchIsOpZero = true; | ||||
8721 | } else if (TI->getOperand(1) == FI->getOperand(0)) { | ||||
8722 | MatchOp = TI->getOperand(1); | ||||
8723 | OtherOpT = TI->getOperand(0); | ||||
8724 | OtherOpF = FI->getOperand(1); | ||||
8725 | MatchIsOpZero = true; | ||||
8726 | } else { | ||||
8727 | return 0; | ||||
8728 | } | ||||
8729 | |||||
8730 | // If we reach here, they do have operations in common. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 8731 | SelectInst *NewSI = SelectInst::Create(SI.getCondition(), OtherOpT, |
8732 | OtherOpF, SI.getName()+".v"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8733 | InsertNewInstBefore(NewSI, SI); |
8734 | |||||
8735 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(TI)) { | ||||
8736 | if (MatchIsOpZero) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8737 | return BinaryOperator::Create(BO->getOpcode(), MatchOp, NewSI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8738 | else |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8739 | return BinaryOperator::Create(BO->getOpcode(), NewSI, MatchOp); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8740 | } |
8741 | assert(0 && "Shouldn't get here"); | ||||
8742 | return 0; | ||||
8743 | } | ||||
8744 | |||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 8745 | /// visitSelectInstWithICmp - Visit a SelectInst that has an |
8746 | /// ICmpInst as its first operand. | ||||
8747 | /// | ||||
8748 | Instruction *InstCombiner::visitSelectInstWithICmp(SelectInst &SI, | ||||
8749 | ICmpInst *ICI) { | ||||
8750 | bool Changed = false; | ||||
8751 | ICmpInst::Predicate Pred = ICI->getPredicate(); | ||||
8752 | Value *CmpLHS = ICI->getOperand(0); | ||||
8753 | Value *CmpRHS = ICI->getOperand(1); | ||||
8754 | Value *TrueVal = SI.getTrueValue(); | ||||
8755 | Value *FalseVal = SI.getFalseValue(); | ||||
8756 | |||||
8757 | // Check cases where the comparison is with a constant that | ||||
8758 | // can be adjusted to fit the min/max idiom. We may edit ICI in | ||||
8759 | // place here, so make sure the select is the only user. | ||||
8760 | if (ICI->hasOneUse()) | ||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8761 | if (ConstantInt *CI = dyn_cast<ConstantInt>(CmpRHS)) { |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 8762 | switch (Pred) { |
8763 | default: break; | ||||
8764 | case ICmpInst::ICMP_ULT: | ||||
8765 | case ICmpInst::ICMP_SLT: { | ||||
8766 | // X < MIN ? T : F --> F | ||||
8767 | if (CI->isMinValue(Pred == ICmpInst::ICMP_SLT)) | ||||
8768 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8769 | // X < C ? X : C-1 --> X > C-1 ? C-1 : X | ||||
8770 | Constant *AdjustedRHS = SubOne(CI); | ||||
8771 | if ((CmpLHS == TrueVal && AdjustedRHS == FalseVal) || | ||||
8772 | (CmpLHS == FalseVal && AdjustedRHS == TrueVal)) { | ||||
8773 | Pred = ICmpInst::getSwappedPredicate(Pred); | ||||
8774 | CmpRHS = AdjustedRHS; | ||||
8775 | std::swap(FalseVal, TrueVal); | ||||
8776 | ICI->setPredicate(Pred); | ||||
8777 | ICI->setOperand(1, CmpRHS); | ||||
8778 | SI.setOperand(1, TrueVal); | ||||
8779 | SI.setOperand(2, FalseVal); | ||||
8780 | Changed = true; | ||||
8781 | } | ||||
8782 | break; | ||||
8783 | } | ||||
8784 | case ICmpInst::ICMP_UGT: | ||||
8785 | case ICmpInst::ICMP_SGT: { | ||||
8786 | // X > MAX ? T : F --> F | ||||
8787 | if (CI->isMaxValue(Pred == ICmpInst::ICMP_SGT)) | ||||
8788 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8789 | // X > C ? X : C+1 --> X < C+1 ? C+1 : X | ||||
8790 | Constant *AdjustedRHS = AddOne(CI); | ||||
8791 | if ((CmpLHS == TrueVal && AdjustedRHS == FalseVal) || | ||||
8792 | (CmpLHS == FalseVal && AdjustedRHS == TrueVal)) { | ||||
8793 | Pred = ICmpInst::getSwappedPredicate(Pred); | ||||
8794 | CmpRHS = AdjustedRHS; | ||||
8795 | std::swap(FalseVal, TrueVal); | ||||
8796 | ICI->setPredicate(Pred); | ||||
8797 | ICI->setOperand(1, CmpRHS); | ||||
8798 | SI.setOperand(1, TrueVal); | ||||
8799 | SI.setOperand(2, FalseVal); | ||||
8800 | Changed = true; | ||||
8801 | } | ||||
8802 | break; | ||||
8803 | } | ||||
8804 | } | ||||
8805 | |||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8806 | // (x <s 0) ? -1 : 0 -> ashr x, 31 -> all ones if signed |
8807 | // (x >s -1) ? -1 : 0 -> ashr x, 31 -> all ones if not signed | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 8808 | CmpInst::Predicate Pred = CmpInst::BAD_ICMP_PREDICATE; |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 8809 | if (match(TrueVal, m_ConstantInt<-1>()) && |
8810 | match(FalseVal, m_ConstantInt<0>())) | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 8811 | Pred = ICI->getPredicate(); |
Chris Lattner | 73c1ddb | 2009-01-05 23:53:12 +0000 | [diff] [blame] | 8812 | else if (match(TrueVal, m_ConstantInt<0>()) && |
8813 | match(FalseVal, m_ConstantInt<-1>())) | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 8814 | Pred = CmpInst::getInversePredicate(ICI->getPredicate()); |
8815 | |||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8816 | if (Pred != CmpInst::BAD_ICMP_PREDICATE) { |
8817 | // If we are just checking for a icmp eq of a single bit and zext'ing it | ||||
8818 | // to an integer, then shift the bit to the appropriate place and then | ||||
8819 | // cast to integer to avoid the comparison. | ||||
8820 | const APInt &Op1CV = CI->getValue(); | ||||
8821 | |||||
8822 | // sext (x <s 0) to i32 --> x>>s31 true if signbit set. | ||||
8823 | // sext (x >s -1) to i32 --> (x>>s31)^-1 true if signbit clear. | ||||
8824 | if ((Pred == ICmpInst::ICMP_SLT && Op1CV == 0) || | ||||
Chris Lattner | 3b87408 | 2008-11-16 05:38:51 +0000 | [diff] [blame] | 8825 | (Pred == ICmpInst::ICMP_SGT && Op1CV.isAllOnesValue())) { |
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8826 | Value *In = ICI->getOperand(0); |
8827 | Value *Sh = ConstantInt::get(In->getType(), | ||||
8828 | In->getType()->getPrimitiveSizeInBits()-1); | ||||
8829 | In = InsertNewInstBefore(BinaryOperator::CreateAShr(In, Sh, | ||||
8830 | In->getName()+".lobit"), | ||||
8831 | *ICI); | ||||
Dan Gohman | 47a6077 | 2008-11-02 00:17:33 +0000 | [diff] [blame] | 8832 | if (In->getType() != SI.getType()) |
8833 | In = CastInst::CreateIntegerCast(In, SI.getType(), | ||||
Dan Gohman | 35b7616 | 2008-10-30 20:40:10 +0000 | [diff] [blame] | 8834 | true/*SExt*/, "tmp", ICI); |
8835 | |||||
8836 | if (Pred == ICmpInst::ICMP_SGT) | ||||
8837 | In = InsertNewInstBefore(BinaryOperator::CreateNot(In, | ||||
8838 | In->getName()+".not"), *ICI); | ||||
8839 | |||||
8840 | return ReplaceInstUsesWith(SI, In); | ||||
8841 | } | ||||
8842 | } | ||||
8843 | } | ||||
8844 | |||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 8845 | if (CmpLHS == TrueVal && CmpRHS == FalseVal) { |
8846 | // Transform (X == Y) ? X : Y -> Y | ||||
8847 | if (Pred == ICmpInst::ICMP_EQ) | ||||
8848 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8849 | // Transform (X != Y) ? X : Y -> X | ||||
8850 | if (Pred == ICmpInst::ICMP_NE) | ||||
8851 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8852 | /// NOTE: if we wanted to, this is where to detect integer MIN/MAX | ||||
8853 | |||||
8854 | } else if (CmpLHS == FalseVal && CmpRHS == TrueVal) { | ||||
8855 | // Transform (X == Y) ? Y : X -> X | ||||
8856 | if (Pred == ICmpInst::ICMP_EQ) | ||||
8857 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8858 | // Transform (X != Y) ? Y : X -> Y | ||||
8859 | if (Pred == ICmpInst::ICMP_NE) | ||||
8860 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8861 | /// NOTE: if we wanted to, this is where to detect integer MIN/MAX | ||||
8862 | } | ||||
8863 | |||||
8864 | /// NOTE: if we wanted to, this is where to detect integer ABS | ||||
8865 | |||||
8866 | return Changed ? &SI : 0; | ||||
8867 | } | ||||
8868 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8869 | Instruction *InstCombiner::visitSelectInst(SelectInst &SI) { |
8870 | Value *CondVal = SI.getCondition(); | ||||
8871 | Value *TrueVal = SI.getTrueValue(); | ||||
8872 | Value *FalseVal = SI.getFalseValue(); | ||||
8873 | |||||
8874 | // select true, X, Y -> X | ||||
8875 | // select false, X, Y -> Y | ||||
8876 | if (ConstantInt *C = dyn_cast<ConstantInt>(CondVal)) | ||||
8877 | return ReplaceInstUsesWith(SI, C->getZExtValue() ? TrueVal : FalseVal); | ||||
8878 | |||||
8879 | // select C, X, X -> X | ||||
8880 | if (TrueVal == FalseVal) | ||||
8881 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8882 | |||||
8883 | if (isa<UndefValue>(TrueVal)) // select C, undef, X -> X | ||||
8884 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8885 | if (isa<UndefValue>(FalseVal)) // select C, X, undef -> X | ||||
8886 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8887 | if (isa<UndefValue>(CondVal)) { // select undef, X, Y -> X or Y | ||||
8888 | if (isa<Constant>(TrueVal)) | ||||
8889 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
8890 | else | ||||
8891 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
8892 | } | ||||
8893 | |||||
8894 | if (SI.getType() == Type::Int1Ty) { | ||||
8895 | if (ConstantInt *C = dyn_cast<ConstantInt>(TrueVal)) { | ||||
8896 | if (C->getZExtValue()) { | ||||
8897 | // Change: A = select B, true, C --> A = or B, C | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8898 | return BinaryOperator::CreateOr(CondVal, FalseVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8899 | } else { |
8900 | // Change: A = select B, false, C --> A = and !B, C | ||||
8901 | Value *NotCond = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8902 | InsertNewInstBefore(BinaryOperator::CreateNot(CondVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8903 | "not."+CondVal->getName()), SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8904 | return BinaryOperator::CreateAnd(NotCond, FalseVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8905 | } |
8906 | } else if (ConstantInt *C = dyn_cast<ConstantInt>(FalseVal)) { | ||||
8907 | if (C->getZExtValue() == false) { | ||||
8908 | // Change: A = select B, C, false --> A = and B, C | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8909 | return BinaryOperator::CreateAnd(CondVal, TrueVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8910 | } else { |
8911 | // Change: A = select B, C, true --> A = or !B, C | ||||
8912 | Value *NotCond = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8913 | InsertNewInstBefore(BinaryOperator::CreateNot(CondVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8914 | "not."+CondVal->getName()), SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8915 | return BinaryOperator::CreateOr(NotCond, TrueVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8916 | } |
8917 | } | ||||
Chris Lattner | 53f85a7 | 2007-11-25 21:27:53 +0000 | [diff] [blame] | 8918 | |
8919 | // select a, b, a -> a&b | ||||
8920 | // select a, a, b -> a|b | ||||
8921 | if (CondVal == TrueVal) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8922 | return BinaryOperator::CreateOr(CondVal, FalseVal); |
Chris Lattner | 53f85a7 | 2007-11-25 21:27:53 +0000 | [diff] [blame] | 8923 | else if (CondVal == FalseVal) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8924 | return BinaryOperator::CreateAnd(CondVal, TrueVal); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8925 | } |
8926 | |||||
8927 | // Selecting between two integer constants? | ||||
8928 | if (ConstantInt *TrueValC = dyn_cast<ConstantInt>(TrueVal)) | ||||
8929 | if (ConstantInt *FalseValC = dyn_cast<ConstantInt>(FalseVal)) { | ||||
8930 | // select C, 1, 0 -> zext C to int | ||||
8931 | if (FalseValC->isZero() && TrueValC->getValue() == 1) { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8932 | return CastInst::Create(Instruction::ZExt, CondVal, SI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8933 | } else if (TrueValC->isZero() && FalseValC->getValue() == 1) { |
8934 | // select C, 0, 1 -> zext !C to int | ||||
8935 | Value *NotCond = | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8936 | InsertNewInstBefore(BinaryOperator::CreateNot(CondVal, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8937 | "not."+CondVal->getName()), SI); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8938 | return CastInst::Create(Instruction::ZExt, NotCond, SI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8939 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8940 | |
8941 | if (ICmpInst *IC = dyn_cast<ICmpInst>(SI.getCondition())) { | ||||
8942 | |||||
8943 | // (x <s 0) ? -1 : 0 -> ashr x, 31 | ||||
8944 | if (TrueValC->isAllOnesValue() && FalseValC->isZero()) | ||||
8945 | if (ConstantInt *CmpCst = dyn_cast<ConstantInt>(IC->getOperand(1))) { | ||||
8946 | if (IC->getPredicate() == ICmpInst::ICMP_SLT && CmpCst->isZero()) { | ||||
8947 | // The comparison constant and the result are not neccessarily the | ||||
8948 | // same width. Make an all-ones value by inserting a AShr. | ||||
8949 | Value *X = IC->getOperand(0); | ||||
8950 | uint32_t Bits = X->getType()->getPrimitiveSizeInBits(); | ||||
8951 | Constant *ShAmt = ConstantInt::get(X->getType(), Bits-1); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8952 | Instruction *SRA = BinaryOperator::Create(Instruction::AShr, X, |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8953 | ShAmt, "ones"); |
8954 | InsertNewInstBefore(SRA, SI); | ||||
Eli Friedman | 722b479 | 2008-11-30 21:09:11 +0000 | [diff] [blame] | 8955 | |
8956 | // Then cast to the appropriate width. | ||||
8957 | return CastInst::CreateIntegerCast(SRA, SI.getType(), true); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8958 | } |
8959 | } | ||||
8960 | |||||
8961 | |||||
8962 | // If one of the constants is zero (we know they can't both be) and we | ||||
8963 | // have an icmp instruction with zero, and we have an 'and' with the | ||||
8964 | // non-constant value, eliminate this whole mess. This corresponds to | ||||
8965 | // cases like this: ((X & 27) ? 27 : 0) | ||||
8966 | if (TrueValC->isZero() || FalseValC->isZero()) | ||||
8967 | if (IC->isEquality() && isa<ConstantInt>(IC->getOperand(1)) && | ||||
8968 | cast<Constant>(IC->getOperand(1))->isNullValue()) | ||||
8969 | if (Instruction *ICA = dyn_cast<Instruction>(IC->getOperand(0))) | ||||
8970 | if (ICA->getOpcode() == Instruction::And && | ||||
8971 | isa<ConstantInt>(ICA->getOperand(1)) && | ||||
8972 | (ICA->getOperand(1) == TrueValC || | ||||
8973 | ICA->getOperand(1) == FalseValC) && | ||||
8974 | isOneBitSet(cast<ConstantInt>(ICA->getOperand(1)))) { | ||||
8975 | // Okay, now we know that everything is set up, we just don't | ||||
8976 | // know whether we have a icmp_ne or icmp_eq and whether the | ||||
8977 | // true or false val is the zero. | ||||
8978 | bool ShouldNotVal = !TrueValC->isZero(); | ||||
8979 | ShouldNotVal ^= IC->getPredicate() == ICmpInst::ICMP_NE; | ||||
8980 | Value *V = ICA; | ||||
8981 | if (ShouldNotVal) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 8982 | V = InsertNewInstBefore(BinaryOperator::Create( |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 8983 | Instruction::Xor, V, ICA->getOperand(1)), SI); |
8984 | return ReplaceInstUsesWith(SI, V); | ||||
8985 | } | ||||
8986 | } | ||||
8987 | } | ||||
8988 | |||||
8989 | // See if we are selecting two values based on a comparison of the two values. | ||||
8990 | if (FCmpInst *FCI = dyn_cast<FCmpInst>(CondVal)) { | ||||
8991 | if (FCI->getOperand(0) == TrueVal && FCI->getOperand(1) == FalseVal) { | ||||
8992 | // Transform (X == Y) ? X : Y -> Y | ||||
Dale Johannesen | 2e1b769 | 2007-10-03 17:45:27 +0000 | [diff] [blame] | 8993 | if (FCI->getPredicate() == FCmpInst::FCMP_OEQ) { |
8994 | // This is not safe in general for floating point: | ||||
8995 | // consider X== -0, Y== +0. | ||||
8996 | // It becomes safe if either operand is a nonzero constant. | ||||
8997 | ConstantFP *CFPt, *CFPf; | ||||
8998 | if (((CFPt = dyn_cast<ConstantFP>(TrueVal)) && | ||||
8999 | !CFPt->getValueAPF().isZero()) || | ||||
9000 | ((CFPf = dyn_cast<ConstantFP>(FalseVal)) && | ||||
9001 | !CFPf->getValueAPF().isZero())) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9002 | return ReplaceInstUsesWith(SI, FalseVal); |
Dale Johannesen | 2e1b769 | 2007-10-03 17:45:27 +0000 | [diff] [blame] | 9003 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9004 | // Transform (X != Y) ? X : Y -> X |
9005 | if (FCI->getPredicate() == FCmpInst::FCMP_ONE) | ||||
9006 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 9007 | // NOTE: if we wanted to, this is where to detect MIN/MAX |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9008 | |
9009 | } else if (FCI->getOperand(0) == FalseVal && FCI->getOperand(1) == TrueVal){ | ||||
9010 | // Transform (X == Y) ? Y : X -> X | ||||
Dale Johannesen | 2e1b769 | 2007-10-03 17:45:27 +0000 | [diff] [blame] | 9011 | if (FCI->getPredicate() == FCmpInst::FCMP_OEQ) { |
9012 | // This is not safe in general for floating point: | ||||
9013 | // consider X== -0, Y== +0. | ||||
9014 | // It becomes safe if either operand is a nonzero constant. | ||||
9015 | ConstantFP *CFPt, *CFPf; | ||||
9016 | if (((CFPt = dyn_cast<ConstantFP>(TrueVal)) && | ||||
9017 | !CFPt->getValueAPF().isZero()) || | ||||
9018 | ((CFPf = dyn_cast<ConstantFP>(FalseVal)) && | ||||
9019 | !CFPf->getValueAPF().isZero())) | ||||
9020 | return ReplaceInstUsesWith(SI, FalseVal); | ||||
9021 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9022 | // Transform (X != Y) ? Y : X -> Y |
9023 | if (FCI->getPredicate() == FCmpInst::FCMP_ONE) | ||||
9024 | return ReplaceInstUsesWith(SI, TrueVal); | ||||
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 9025 | // NOTE: if we wanted to, this is where to detect MIN/MAX |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9026 | } |
Dan Gohman | 58c0963 | 2008-09-16 18:46:06 +0000 | [diff] [blame] | 9027 | // NOTE: if we wanted to, this is where to detect ABS |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9028 | } |
9029 | |||||
9030 | // 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] | 9031 | if (ICmpInst *ICI = dyn_cast<ICmpInst>(CondVal)) |
9032 | if (Instruction *Result = visitSelectInstWithICmp(SI, ICI)) | ||||
9033 | return Result; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9034 | |
9035 | if (Instruction *TI = dyn_cast<Instruction>(TrueVal)) | ||||
9036 | if (Instruction *FI = dyn_cast<Instruction>(FalseVal)) | ||||
9037 | if (TI->hasOneUse() && FI->hasOneUse()) { | ||||
9038 | Instruction *AddOp = 0, *SubOp = 0; | ||||
9039 | |||||
9040 | // Turn (select C, (op X, Y), (op X, Z)) -> (op X, (select C, Y, Z)) | ||||
9041 | if (TI->getOpcode() == FI->getOpcode()) | ||||
9042 | if (Instruction *IV = FoldSelectOpOp(SI, TI, FI)) | ||||
9043 | return IV; | ||||
9044 | |||||
9045 | // Turn select C, (X+Y), (X-Y) --> (X+(select C, Y, (-Y))). This is | ||||
9046 | // even legal for FP. | ||||
9047 | if (TI->getOpcode() == Instruction::Sub && | ||||
9048 | FI->getOpcode() == Instruction::Add) { | ||||
9049 | AddOp = FI; SubOp = TI; | ||||
9050 | } else if (FI->getOpcode() == Instruction::Sub && | ||||
9051 | TI->getOpcode() == Instruction::Add) { | ||||
9052 | AddOp = TI; SubOp = FI; | ||||
9053 | } | ||||
9054 | |||||
9055 | if (AddOp) { | ||||
9056 | Value *OtherAddOp = 0; | ||||
9057 | if (SubOp->getOperand(0) == AddOp->getOperand(0)) { | ||||
9058 | OtherAddOp = AddOp->getOperand(1); | ||||
9059 | } else if (SubOp->getOperand(0) == AddOp->getOperand(1)) { | ||||
9060 | OtherAddOp = AddOp->getOperand(0); | ||||
9061 | } | ||||
9062 | |||||
9063 | if (OtherAddOp) { | ||||
9064 | // So at this point we know we have (Y -> OtherAddOp): | ||||
9065 | // select C, (add X, Y), (sub X, Z) | ||||
9066 | Value *NegVal; // Compute -Z | ||||
9067 | if (Constant *C = dyn_cast<Constant>(SubOp->getOperand(1))) { | ||||
9068 | NegVal = ConstantExpr::getNeg(C); | ||||
9069 | } else { | ||||
9070 | NegVal = InsertNewInstBefore( | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9071 | BinaryOperator::CreateNeg(SubOp->getOperand(1), "tmp"), SI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9072 | } |
9073 | |||||
9074 | Value *NewTrueOp = OtherAddOp; | ||||
9075 | Value *NewFalseOp = NegVal; | ||||
9076 | if (AddOp != TI) | ||||
9077 | std::swap(NewTrueOp, NewFalseOp); | ||||
9078 | Instruction *NewSel = | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9079 | SelectInst::Create(CondVal, NewTrueOp, |
9080 | NewFalseOp, SI.getName() + ".p"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9081 | |
9082 | NewSel = InsertNewInstBefore(NewSel, SI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9083 | return BinaryOperator::CreateAdd(SubOp->getOperand(0), NewSel); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9084 | } |
9085 | } | ||||
9086 | } | ||||
9087 | |||||
9088 | // See if we can fold the select into one of our operands. | ||||
9089 | if (SI.getType()->isInteger()) { | ||||
9090 | // See the comment above GetSelectFoldableOperands for a description of the | ||||
9091 | // transformation we are doing here. | ||||
9092 | if (Instruction *TVI = dyn_cast<Instruction>(TrueVal)) | ||||
9093 | if (TVI->hasOneUse() && TVI->getNumOperands() == 2 && | ||||
9094 | !isa<Constant>(FalseVal)) | ||||
9095 | if (unsigned SFO = GetSelectFoldableOperands(TVI)) { | ||||
9096 | unsigned OpToFold = 0; | ||||
9097 | if ((SFO & 1) && FalseVal == TVI->getOperand(0)) { | ||||
9098 | OpToFold = 1; | ||||
9099 | } else if ((SFO & 2) && FalseVal == TVI->getOperand(1)) { | ||||
9100 | OpToFold = 2; | ||||
9101 | } | ||||
9102 | |||||
9103 | if (OpToFold) { | ||||
9104 | Constant *C = GetSelectFoldableConstant(TVI); | ||||
9105 | Instruction *NewSel = | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9106 | SelectInst::Create(SI.getCondition(), |
9107 | TVI->getOperand(2-OpToFold), C); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9108 | InsertNewInstBefore(NewSel, SI); |
9109 | NewSel->takeName(TVI); | ||||
9110 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(TVI)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9111 | return BinaryOperator::Create(BO->getOpcode(), FalseVal, NewSel); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9112 | else { |
9113 | assert(0 && "Unknown instruction!!"); | ||||
9114 | } | ||||
9115 | } | ||||
9116 | } | ||||
9117 | |||||
9118 | if (Instruction *FVI = dyn_cast<Instruction>(FalseVal)) | ||||
9119 | if (FVI->hasOneUse() && FVI->getNumOperands() == 2 && | ||||
9120 | !isa<Constant>(TrueVal)) | ||||
9121 | if (unsigned SFO = GetSelectFoldableOperands(FVI)) { | ||||
9122 | unsigned OpToFold = 0; | ||||
9123 | if ((SFO & 1) && TrueVal == FVI->getOperand(0)) { | ||||
9124 | OpToFold = 1; | ||||
9125 | } else if ((SFO & 2) && TrueVal == FVI->getOperand(1)) { | ||||
9126 | OpToFold = 2; | ||||
9127 | } | ||||
9128 | |||||
9129 | if (OpToFold) { | ||||
9130 | Constant *C = GetSelectFoldableConstant(FVI); | ||||
9131 | Instruction *NewSel = | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9132 | SelectInst::Create(SI.getCondition(), C, |
9133 | FVI->getOperand(2-OpToFold)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9134 | InsertNewInstBefore(NewSel, SI); |
9135 | NewSel->takeName(FVI); | ||||
9136 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(FVI)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9137 | return BinaryOperator::Create(BO->getOpcode(), TrueVal, NewSel); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9138 | else |
9139 | assert(0 && "Unknown instruction!!"); | ||||
9140 | } | ||||
9141 | } | ||||
9142 | } | ||||
9143 | |||||
9144 | if (BinaryOperator::isNot(CondVal)) { | ||||
9145 | SI.setOperand(0, BinaryOperator::getNotArgument(CondVal)); | ||||
9146 | SI.setOperand(1, FalseVal); | ||||
9147 | SI.setOperand(2, TrueVal); | ||||
9148 | return &SI; | ||||
9149 | } | ||||
9150 | |||||
9151 | return 0; | ||||
9152 | } | ||||
9153 | |||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9154 | /// EnforceKnownAlignment - If the specified pointer points to an object that |
9155 | /// we control, modify the object's alignment to PrefAlign. This isn't | ||||
9156 | /// often possible though. If alignment is important, a more reliable approach | ||||
9157 | /// is to simply align all global variables and allocation instructions to | ||||
9158 | /// their preferred alignment from the beginning. | ||||
9159 | /// | ||||
9160 | static unsigned EnforceKnownAlignment(Value *V, | ||||
9161 | unsigned Align, unsigned PrefAlign) { | ||||
Chris Lattner | 47cf345 | 2007-08-09 19:05:49 +0000 | [diff] [blame] | 9162 | |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9163 | User *U = dyn_cast<User>(V); |
9164 | if (!U) return Align; | ||||
9165 | |||||
9166 | switch (getOpcode(U)) { | ||||
9167 | default: break; | ||||
9168 | case Instruction::BitCast: | ||||
9169 | return EnforceKnownAlignment(U->getOperand(0), Align, PrefAlign); | ||||
9170 | case Instruction::GetElementPtr: { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9171 | // If all indexes are zero, it is just the alignment of the base pointer. |
9172 | bool AllZeroOperands = true; | ||||
Gabor Greif | e92fbe2 | 2008-06-12 21:51:29 +0000 | [diff] [blame] | 9173 | 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] | 9174 | if (!isa<Constant>(*i) || |
9175 | !cast<Constant>(*i)->isNullValue()) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9176 | AllZeroOperands = false; |
9177 | break; | ||||
9178 | } | ||||
Chris Lattner | 47cf345 | 2007-08-09 19:05:49 +0000 | [diff] [blame] | 9179 | |
9180 | if (AllZeroOperands) { | ||||
9181 | // Treat this like a bitcast. | ||||
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9182 | return EnforceKnownAlignment(U->getOperand(0), Align, PrefAlign); |
Chris Lattner | 47cf345 | 2007-08-09 19:05:49 +0000 | [diff] [blame] | 9183 | } |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9184 | break; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9185 | } |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9186 | } |
9187 | |||||
9188 | if (GlobalValue *GV = dyn_cast<GlobalValue>(V)) { | ||||
9189 | // If there is a large requested alignment and we can, bump up the alignment | ||||
9190 | // of the global. | ||||
9191 | if (!GV->isDeclaration()) { | ||||
9192 | GV->setAlignment(PrefAlign); | ||||
9193 | Align = PrefAlign; | ||||
9194 | } | ||||
9195 | } else if (AllocationInst *AI = dyn_cast<AllocationInst>(V)) { | ||||
9196 | // If there is a requested alignment and if this is an alloca, round up. We | ||||
9197 | // don't do this for malloc, because some systems can't respect the request. | ||||
9198 | if (isa<AllocaInst>(AI)) { | ||||
9199 | AI->setAlignment(PrefAlign); | ||||
9200 | Align = PrefAlign; | ||||
9201 | } | ||||
9202 | } | ||||
9203 | |||||
9204 | return Align; | ||||
9205 | } | ||||
9206 | |||||
9207 | /// GetOrEnforceKnownAlignment - If the specified pointer has an alignment that | ||||
9208 | /// we can determine, return it, otherwise return 0. If PrefAlign is specified, | ||||
9209 | /// and it is more than the alignment of the ultimate object, see if we can | ||||
9210 | /// increase the alignment of the ultimate object, making this check succeed. | ||||
9211 | unsigned InstCombiner::GetOrEnforceKnownAlignment(Value *V, | ||||
9212 | unsigned PrefAlign) { | ||||
9213 | unsigned BitWidth = TD ? TD->getTypeSizeInBits(V->getType()) : | ||||
9214 | sizeof(PrefAlign) * CHAR_BIT; | ||||
9215 | APInt Mask = APInt::getAllOnesValue(BitWidth); | ||||
9216 | APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); | ||||
9217 | ComputeMaskedBits(V, Mask, KnownZero, KnownOne); | ||||
9218 | unsigned TrailZ = KnownZero.countTrailingOnes(); | ||||
9219 | unsigned Align = 1u << std::min(BitWidth - 1, TrailZ); | ||||
9220 | |||||
9221 | if (PrefAlign > Align) | ||||
9222 | Align = EnforceKnownAlignment(V, Align, PrefAlign); | ||||
9223 | |||||
9224 | // We don't need to make any adjustment. | ||||
9225 | return Align; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9226 | } |
9227 | |||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9228 | Instruction *InstCombiner::SimplifyMemTransfer(MemIntrinsic *MI) { |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 9229 | unsigned DstAlign = GetOrEnforceKnownAlignment(MI->getOperand(1)); |
9230 | unsigned SrcAlign = GetOrEnforceKnownAlignment(MI->getOperand(2)); | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9231 | unsigned MinAlign = std::min(DstAlign, SrcAlign); |
9232 | unsigned CopyAlign = MI->getAlignment()->getZExtValue(); | ||||
9233 | |||||
9234 | if (CopyAlign < MinAlign) { | ||||
9235 | MI->setAlignment(ConstantInt::get(Type::Int32Ty, MinAlign)); | ||||
9236 | return MI; | ||||
9237 | } | ||||
9238 | |||||
9239 | // If MemCpyInst length is 1/2/4/8 bytes then replace memcpy with | ||||
9240 | // load/store. | ||||
9241 | ConstantInt *MemOpLength = dyn_cast<ConstantInt>(MI->getOperand(3)); | ||||
9242 | if (MemOpLength == 0) return 0; | ||||
9243 | |||||
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9244 | // Source and destination pointer types are always "i8*" for intrinsic. See |
9245 | // if the size is something we can handle with a single primitive load/store. | ||||
9246 | // A single load+store correctly handles overlapping memory in the memmove | ||||
9247 | // case. | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9248 | unsigned Size = MemOpLength->getZExtValue(); |
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 9249 | if (Size == 0) return MI; // Delete this mem transfer. |
9250 | |||||
9251 | if (Size > 8 || (Size&(Size-1))) | ||||
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9252 | return 0; // If not 1/2/4/8 bytes, exit. |
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9253 | |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9254 | // Use an integer load+store unless we can find something better. |
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9255 | Type *NewPtrTy = PointerType::getUnqual(IntegerType::get(Size<<3)); |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9256 | |
9257 | // Memcpy forces the use of i8* for the source and destination. That means | ||||
9258 | // that if you're using memcpy to move one double around, you'll get a cast | ||||
9259 | // from double* to i8*. We'd much rather use a double load+store rather than | ||||
9260 | // an i64 load+store, here because this improves the odds that the source or | ||||
9261 | // dest address will be promotable. See if we can find a better type than the | ||||
9262 | // integer datatype. | ||||
9263 | if (Value *Op = getBitCastOperand(MI->getOperand(1))) { | ||||
9264 | const Type *SrcETy = cast<PointerType>(Op->getType())->getElementType(); | ||||
9265 | if (SrcETy->isSized() && TD->getTypeStoreSize(SrcETy) == Size) { | ||||
9266 | // The SrcETy might be something like {{{double}}} or [1 x double]. Rip | ||||
9267 | // down through these levels if so. | ||||
Dan Gohman | b8e94f6 | 2008-05-23 01:52:21 +0000 | [diff] [blame] | 9268 | while (!SrcETy->isSingleValueType()) { |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9269 | if (const StructType *STy = dyn_cast<StructType>(SrcETy)) { |
9270 | if (STy->getNumElements() == 1) | ||||
9271 | SrcETy = STy->getElementType(0); | ||||
9272 | else | ||||
9273 | break; | ||||
9274 | } else if (const ArrayType *ATy = dyn_cast<ArrayType>(SrcETy)) { | ||||
9275 | if (ATy->getNumElements() == 1) | ||||
9276 | SrcETy = ATy->getElementType(); | ||||
9277 | else | ||||
9278 | break; | ||||
9279 | } else | ||||
9280 | break; | ||||
9281 | } | ||||
9282 | |||||
Dan Gohman | b8e94f6 | 2008-05-23 01:52:21 +0000 | [diff] [blame] | 9283 | if (SrcETy->isSingleValueType()) |
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9284 | NewPtrTy = PointerType::getUnqual(SrcETy); |
9285 | } | ||||
9286 | } | ||||
9287 | |||||
9288 | |||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9289 | // If the memcpy/memmove provides better alignment info than we can |
9290 | // infer, use it. | ||||
9291 | SrcAlign = std::max(SrcAlign, CopyAlign); | ||||
9292 | DstAlign = std::max(DstAlign, CopyAlign); | ||||
9293 | |||||
9294 | Value *Src = InsertBitCastBefore(MI->getOperand(2), NewPtrTy, *MI); | ||||
9295 | Value *Dest = InsertBitCastBefore(MI->getOperand(1), NewPtrTy, *MI); | ||||
Chris Lattner | c669fb6 | 2008-01-14 00:28:35 +0000 | [diff] [blame] | 9296 | Instruction *L = new LoadInst(Src, "tmp", false, SrcAlign); |
9297 | InsertNewInstBefore(L, *MI); | ||||
9298 | InsertNewInstBefore(new StoreInst(L, Dest, false, DstAlign), *MI); | ||||
9299 | |||||
9300 | // Set the size of the copy to 0, it will be deleted on the next iteration. | ||||
9301 | MI->setOperand(3, Constant::getNullValue(MemOpLength->getType())); | ||||
9302 | return MI; | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9303 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9304 | |
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 9305 | Instruction *InstCombiner::SimplifyMemSet(MemSetInst *MI) { |
9306 | unsigned Alignment = GetOrEnforceKnownAlignment(MI->getDest()); | ||||
9307 | if (MI->getAlignment()->getZExtValue() < Alignment) { | ||||
9308 | MI->setAlignment(ConstantInt::get(Type::Int32Ty, Alignment)); | ||||
9309 | return MI; | ||||
9310 | } | ||||
9311 | |||||
9312 | // Extract the length and alignment and fill if they are constant. | ||||
9313 | ConstantInt *LenC = dyn_cast<ConstantInt>(MI->getLength()); | ||||
9314 | ConstantInt *FillC = dyn_cast<ConstantInt>(MI->getValue()); | ||||
9315 | if (!LenC || !FillC || FillC->getType() != Type::Int8Ty) | ||||
9316 | return 0; | ||||
9317 | uint64_t Len = LenC->getZExtValue(); | ||||
9318 | Alignment = MI->getAlignment()->getZExtValue(); | ||||
9319 | |||||
9320 | // If the length is zero, this is a no-op | ||||
9321 | if (Len == 0) return MI; // memset(d,c,0,a) -> noop | ||||
9322 | |||||
9323 | // memset(s,c,n) -> store s, c (for n=1,2,4,8) | ||||
9324 | if (Len <= 8 && isPowerOf2_32((uint32_t)Len)) { | ||||
9325 | const Type *ITy = IntegerType::get(Len*8); // n=1 -> i8. | ||||
9326 | |||||
9327 | Value *Dest = MI->getDest(); | ||||
9328 | Dest = InsertBitCastBefore(Dest, PointerType::getUnqual(ITy), *MI); | ||||
9329 | |||||
9330 | // Alignment 0 is identity for alignment 1 for memset, but not store. | ||||
9331 | if (Alignment == 0) Alignment = 1; | ||||
9332 | |||||
9333 | // Extract the fill value and store. | ||||
9334 | uint64_t Fill = FillC->getZExtValue()*0x0101010101010101ULL; | ||||
9335 | InsertNewInstBefore(new StoreInst(ConstantInt::get(ITy, Fill), Dest, false, | ||||
9336 | Alignment), *MI); | ||||
9337 | |||||
9338 | // Set the size of the copy to 0, it will be deleted on the next iteration. | ||||
9339 | MI->setLength(Constant::getNullValue(LenC->getType())); | ||||
9340 | return MI; | ||||
9341 | } | ||||
9342 | |||||
9343 | return 0; | ||||
9344 | } | ||||
9345 | |||||
9346 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9347 | /// visitCallInst - CallInst simplification. This mostly only handles folding |
9348 | /// of intrinsic instructions. For normal calls, it allows visitCallSite to do | ||||
9349 | /// the heavy lifting. | ||||
9350 | /// | ||||
9351 | Instruction *InstCombiner::visitCallInst(CallInst &CI) { | ||||
9352 | IntrinsicInst *II = dyn_cast<IntrinsicInst>(&CI); | ||||
9353 | if (!II) return visitCallSite(&CI); | ||||
9354 | |||||
9355 | // Intrinsics cannot occur in an invoke, so handle them here instead of in | ||||
9356 | // visitCallSite. | ||||
9357 | if (MemIntrinsic *MI = dyn_cast<MemIntrinsic>(II)) { | ||||
9358 | bool Changed = false; | ||||
9359 | |||||
9360 | // memmove/cpy/set of zero bytes is a noop. | ||||
9361 | if (Constant *NumBytes = dyn_cast<Constant>(MI->getLength())) { | ||||
9362 | if (NumBytes->isNullValue()) return EraseInstFromFunction(CI); | ||||
9363 | |||||
9364 | if (ConstantInt *CI = dyn_cast<ConstantInt>(NumBytes)) | ||||
9365 | if (CI->getZExtValue() == 1) { | ||||
9366 | // Replace the instruction with just byte operations. We would | ||||
9367 | // transform other cases to loads/stores, but we don't know if | ||||
9368 | // alignment is sufficient. | ||||
9369 | } | ||||
9370 | } | ||||
9371 | |||||
9372 | // If we have a memmove and the source operation is a constant global, | ||||
9373 | // then the source and dest pointers can't alias, so we can change this | ||||
9374 | // into a call to memcpy. | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9375 | if (MemMoveInst *MMI = dyn_cast<MemMoveInst>(MI)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9376 | if (GlobalVariable *GVSrc = dyn_cast<GlobalVariable>(MMI->getSource())) |
9377 | if (GVSrc->isConstant()) { | ||||
9378 | Module *M = CI.getParent()->getParent()->getParent(); | ||||
Chris Lattner | 82c2e43 | 2008-11-21 16:42:48 +0000 | [diff] [blame] | 9379 | Intrinsic::ID MemCpyID = Intrinsic::memcpy; |
9380 | const Type *Tys[1]; | ||||
9381 | Tys[0] = CI.getOperand(3)->getType(); | ||||
9382 | CI.setOperand(0, | ||||
9383 | Intrinsic::getDeclaration(M, MemCpyID, Tys, 1)); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9384 | Changed = true; |
9385 | } | ||||
Chris Lattner | 59b27d9 | 2008-05-28 05:30:41 +0000 | [diff] [blame] | 9386 | |
9387 | // memmove(x,x,size) -> noop. | ||||
9388 | if (MMI->getSource() == MMI->getDest()) | ||||
9389 | return EraseInstFromFunction(CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9390 | } |
9391 | |||||
9392 | // If we can determine a pointer alignment that is bigger than currently | ||||
9393 | // set, update the alignment. | ||||
9394 | if (isa<MemCpyInst>(MI) || isa<MemMoveInst>(MI)) { | ||||
Chris Lattner | 00ae513 | 2008-01-13 23:50:23 +0000 | [diff] [blame] | 9395 | if (Instruction *I = SimplifyMemTransfer(MI)) |
9396 | return I; | ||||
Chris Lattner | 5af8a91 | 2008-04-30 06:39:11 +0000 | [diff] [blame] | 9397 | } else if (MemSetInst *MSI = dyn_cast<MemSetInst>(MI)) { |
9398 | if (Instruction *I = SimplifyMemSet(MSI)) | ||||
9399 | return I; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9400 | } |
9401 | |||||
9402 | if (Changed) return II; | ||||
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9403 | } |
9404 | |||||
9405 | switch (II->getIntrinsicID()) { | ||||
9406 | default: break; | ||||
9407 | case Intrinsic::bswap: | ||||
9408 | // bswap(bswap(x)) -> x | ||||
9409 | if (IntrinsicInst *Operand = dyn_cast<IntrinsicInst>(II->getOperand(1))) | ||||
9410 | if (Operand->getIntrinsicID() == Intrinsic::bswap) | ||||
9411 | return ReplaceInstUsesWith(CI, Operand->getOperand(1)); | ||||
9412 | break; | ||||
9413 | case Intrinsic::ppc_altivec_lvx: | ||||
9414 | case Intrinsic::ppc_altivec_lvxl: | ||||
9415 | case Intrinsic::x86_sse_loadu_ps: | ||||
9416 | case Intrinsic::x86_sse2_loadu_pd: | ||||
9417 | case Intrinsic::x86_sse2_loadu_dq: | ||||
9418 | // Turn PPC lvx -> load if the pointer is known aligned. | ||||
9419 | // Turn X86 loadups -> load if the pointer is known aligned. | ||||
9420 | if (GetOrEnforceKnownAlignment(II->getOperand(1), 16) >= 16) { | ||||
9421 | Value *Ptr = InsertBitCastBefore(II->getOperand(1), | ||||
9422 | PointerType::getUnqual(II->getType()), | ||||
9423 | CI); | ||||
9424 | return new LoadInst(Ptr); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9425 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9426 | break; |
9427 | case Intrinsic::ppc_altivec_stvx: | ||||
9428 | case Intrinsic::ppc_altivec_stvxl: | ||||
9429 | // Turn stvx -> store if the pointer is known aligned. | ||||
9430 | if (GetOrEnforceKnownAlignment(II->getOperand(2), 16) >= 16) { | ||||
9431 | const Type *OpPtrTy = | ||||
9432 | PointerType::getUnqual(II->getOperand(1)->getType()); | ||||
9433 | Value *Ptr = InsertBitCastBefore(II->getOperand(2), OpPtrTy, CI); | ||||
9434 | return new StoreInst(II->getOperand(1), Ptr); | ||||
9435 | } | ||||
9436 | break; | ||||
9437 | case Intrinsic::x86_sse_storeu_ps: | ||||
9438 | case Intrinsic::x86_sse2_storeu_pd: | ||||
9439 | case Intrinsic::x86_sse2_storeu_dq: | ||||
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9440 | // Turn X86 storeu -> store if the pointer is known aligned. |
9441 | if (GetOrEnforceKnownAlignment(II->getOperand(1), 16) >= 16) { | ||||
9442 | const Type *OpPtrTy = | ||||
9443 | PointerType::getUnqual(II->getOperand(2)->getType()); | ||||
9444 | Value *Ptr = InsertBitCastBefore(II->getOperand(1), OpPtrTy, CI); | ||||
9445 | return new StoreInst(II->getOperand(2), Ptr); | ||||
9446 | } | ||||
9447 | break; | ||||
9448 | |||||
9449 | case Intrinsic::x86_sse_cvttss2si: { | ||||
9450 | // These intrinsics only demands the 0th element of its input vector. If | ||||
9451 | // we can simplify the input based on that, do so now. | ||||
9452 | uint64_t UndefElts; | ||||
9453 | if (Value *V = SimplifyDemandedVectorElts(II->getOperand(1), 1, | ||||
9454 | UndefElts)) { | ||||
9455 | II->setOperand(1, V); | ||||
9456 | return II; | ||||
9457 | } | ||||
9458 | break; | ||||
9459 | } | ||||
9460 | |||||
9461 | case Intrinsic::ppc_altivec_vperm: | ||||
9462 | // Turn vperm(V1,V2,mask) -> shuffle(V1,V2,mask) if mask is a constant. | ||||
9463 | if (ConstantVector *Mask = dyn_cast<ConstantVector>(II->getOperand(3))) { | ||||
9464 | assert(Mask->getNumOperands() == 16 && "Bad type for intrinsic!"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9465 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9466 | // Check that all of the elements are integer constants or undefs. |
9467 | bool AllEltsOk = true; | ||||
9468 | for (unsigned i = 0; i != 16; ++i) { | ||||
9469 | if (!isa<ConstantInt>(Mask->getOperand(i)) && | ||||
9470 | !isa<UndefValue>(Mask->getOperand(i))) { | ||||
9471 | AllEltsOk = false; | ||||
9472 | break; | ||||
9473 | } | ||||
9474 | } | ||||
9475 | |||||
9476 | if (AllEltsOk) { | ||||
9477 | // Cast the input vectors to byte vectors. | ||||
9478 | Value *Op0 =InsertBitCastBefore(II->getOperand(1),Mask->getType(),CI); | ||||
9479 | Value *Op1 =InsertBitCastBefore(II->getOperand(2),Mask->getType(),CI); | ||||
9480 | Value *Result = UndefValue::get(Op0->getType()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9481 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9482 | // Only extract each element once. |
9483 | Value *ExtractedElts[32]; | ||||
9484 | memset(ExtractedElts, 0, sizeof(ExtractedElts)); | ||||
9485 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9486 | for (unsigned i = 0; i != 16; ++i) { |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9487 | if (isa<UndefValue>(Mask->getOperand(i))) |
9488 | continue; | ||||
9489 | unsigned Idx=cast<ConstantInt>(Mask->getOperand(i))->getZExtValue(); | ||||
9490 | Idx &= 31; // Match the hardware behavior. | ||||
9491 | |||||
9492 | if (ExtractedElts[Idx] == 0) { | ||||
9493 | Instruction *Elt = | ||||
9494 | new ExtractElementInst(Idx < 16 ? Op0 : Op1, Idx&15, "tmp"); | ||||
9495 | InsertNewInstBefore(Elt, CI); | ||||
9496 | ExtractedElts[Idx] = Elt; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9497 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9498 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9499 | // Insert this value into the result vector. |
9500 | Result = InsertElementInst::Create(Result, ExtractedElts[Idx], | ||||
9501 | i, "tmp"); | ||||
9502 | InsertNewInstBefore(cast<Instruction>(Result), CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9503 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9504 | return CastInst::Create(Instruction::BitCast, Result, CI.getType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9505 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9506 | } |
9507 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9508 | |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9509 | case Intrinsic::stackrestore: { |
9510 | // If the save is right next to the restore, remove the restore. This can | ||||
9511 | // happen when variable allocas are DCE'd. | ||||
9512 | if (IntrinsicInst *SS = dyn_cast<IntrinsicInst>(II->getOperand(1))) { | ||||
9513 | if (SS->getIntrinsicID() == Intrinsic::stacksave) { | ||||
9514 | BasicBlock::iterator BI = SS; | ||||
9515 | if (&*++BI == II) | ||||
9516 | return EraseInstFromFunction(CI); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9517 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9518 | } |
9519 | |||||
9520 | // Scan down this block to see if there is another stack restore in the | ||||
9521 | // same block without an intervening call/alloca. | ||||
9522 | BasicBlock::iterator BI = II; | ||||
9523 | TerminatorInst *TI = II->getParent()->getTerminator(); | ||||
9524 | bool CannotRemove = false; | ||||
9525 | for (++BI; &*BI != TI; ++BI) { | ||||
9526 | if (isa<AllocaInst>(BI)) { | ||||
9527 | CannotRemove = true; | ||||
9528 | break; | ||||
9529 | } | ||||
Chris Lattner | a6b477c | 2008-06-25 05:59:28 +0000 | [diff] [blame] | 9530 | if (CallInst *BCI = dyn_cast<CallInst>(BI)) { |
9531 | if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(BCI)) { | ||||
9532 | // If there is a stackrestore below this one, remove this one. | ||||
9533 | if (II->getIntrinsicID() == Intrinsic::stackrestore) | ||||
9534 | return EraseInstFromFunction(CI); | ||||
9535 | // Otherwise, ignore the intrinsic. | ||||
9536 | } else { | ||||
9537 | // If we found a non-intrinsic call, we can't remove the stack | ||||
9538 | // restore. | ||||
Chris Lattner | 416d91c | 2008-02-18 06:12:38 +0000 | [diff] [blame] | 9539 | CannotRemove = true; |
9540 | break; | ||||
9541 | } | ||||
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9542 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9543 | } |
Chris Lattner | 989ba31 | 2008-06-18 04:33:20 +0000 | [diff] [blame] | 9544 | |
9545 | // If the stack restore is in a return/unwind block and if there are no | ||||
9546 | // allocas or calls between the restore and the return, nuke the restore. | ||||
9547 | if (!CannotRemove && (isa<ReturnInst>(TI) || isa<UnwindInst>(TI))) | ||||
9548 | return EraseInstFromFunction(CI); | ||||
9549 | break; | ||||
9550 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9551 | } |
9552 | |||||
9553 | return visitCallSite(II); | ||||
9554 | } | ||||
9555 | |||||
9556 | // InvokeInst simplification | ||||
9557 | // | ||||
9558 | Instruction *InstCombiner::visitInvokeInst(InvokeInst &II) { | ||||
9559 | return visitCallSite(&II); | ||||
9560 | } | ||||
9561 | |||||
Dale Johannesen | 9602183 | 2008-04-25 21:16:07 +0000 | [diff] [blame] | 9562 | /// isSafeToEliminateVarargsCast - If this cast does not affect the value |
9563 | /// 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] | 9564 | static bool isSafeToEliminateVarargsCast(const CallSite CS, |
9565 | const CastInst * const CI, | ||||
9566 | const TargetData * const TD, | ||||
9567 | const int ix) { | ||||
9568 | if (!CI->isLosslessCast()) | ||||
9569 | return false; | ||||
9570 | |||||
9571 | // The size of ByVal arguments is derived from the type, so we | ||||
9572 | // can't change to a type with a different size. If the size were | ||||
9573 | // passed explicitly we could avoid this check. | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9574 | if (!CS.paramHasAttr(ix, Attribute::ByVal)) |
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9575 | return true; |
9576 | |||||
9577 | const Type* SrcTy = | ||||
9578 | cast<PointerType>(CI->getOperand(0)->getType())->getElementType(); | ||||
9579 | const Type* DstTy = cast<PointerType>(CI->getType())->getElementType(); | ||||
9580 | if (!SrcTy->isSized() || !DstTy->isSized()) | ||||
9581 | return false; | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 9582 | if (TD->getTypePaddedSize(SrcTy) != TD->getTypePaddedSize(DstTy)) |
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9583 | return false; |
9584 | return true; | ||||
9585 | } | ||||
9586 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9587 | // visitCallSite - Improvements for call and invoke instructions. |
9588 | // | ||||
9589 | Instruction *InstCombiner::visitCallSite(CallSite CS) { | ||||
9590 | bool Changed = false; | ||||
9591 | |||||
9592 | // If the callee is a constexpr cast of a function, attempt to move the cast | ||||
9593 | // to the arguments of the call/invoke. | ||||
9594 | if (transformConstExprCastCall(CS)) return 0; | ||||
9595 | |||||
9596 | Value *Callee = CS.getCalledValue(); | ||||
9597 | |||||
9598 | if (Function *CalleeF = dyn_cast<Function>(Callee)) | ||||
9599 | if (CalleeF->getCallingConv() != CS.getCallingConv()) { | ||||
9600 | Instruction *OldCall = CS.getInstruction(); | ||||
9601 | // If the call and callee calling conventions don't match, this call must | ||||
9602 | // be unreachable, as the call is undefined. | ||||
9603 | new StoreInst(ConstantInt::getTrue(), | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 9604 | UndefValue::get(PointerType::getUnqual(Type::Int1Ty)), |
9605 | OldCall); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9606 | if (!OldCall->use_empty()) |
9607 | OldCall->replaceAllUsesWith(UndefValue::get(OldCall->getType())); | ||||
9608 | if (isa<CallInst>(OldCall)) // Not worth removing an invoke here. | ||||
9609 | return EraseInstFromFunction(*OldCall); | ||||
9610 | return 0; | ||||
9611 | } | ||||
9612 | |||||
9613 | if (isa<ConstantPointerNull>(Callee) || isa<UndefValue>(Callee)) { | ||||
9614 | // This instruction is not reachable, just remove it. We insert a store to | ||||
9615 | // undef so that we know that this code is not reachable, despite the fact | ||||
9616 | // that we can't modify the CFG here. | ||||
9617 | new StoreInst(ConstantInt::getTrue(), | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 9618 | UndefValue::get(PointerType::getUnqual(Type::Int1Ty)), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9619 | CS.getInstruction()); |
9620 | |||||
9621 | if (!CS.getInstruction()->use_empty()) | ||||
9622 | CS.getInstruction()-> | ||||
9623 | replaceAllUsesWith(UndefValue::get(CS.getInstruction()->getType())); | ||||
9624 | |||||
9625 | if (InvokeInst *II = dyn_cast<InvokeInst>(CS.getInstruction())) { | ||||
9626 | // Don't break the CFG, insert a dummy cond branch. | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9627 | BranchInst::Create(II->getNormalDest(), II->getUnwindDest(), |
9628 | ConstantInt::getTrue(), II); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9629 | } |
9630 | return EraseInstFromFunction(*CS.getInstruction()); | ||||
9631 | } | ||||
9632 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9633 | if (BitCastInst *BC = dyn_cast<BitCastInst>(Callee)) |
9634 | if (IntrinsicInst *In = dyn_cast<IntrinsicInst>(BC->getOperand(0))) | ||||
9635 | if (In->getIntrinsicID() == Intrinsic::init_trampoline) | ||||
9636 | return transformCallThroughTrampoline(CS); | ||||
9637 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9638 | const PointerType *PTy = cast<PointerType>(Callee->getType()); |
9639 | const FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); | ||||
9640 | if (FTy->isVarArg()) { | ||||
Dale Johannesen | 502336c | 2008-04-23 01:03:05 +0000 | [diff] [blame] | 9641 | int ix = FTy->getNumParams() + (isa<InvokeInst>(Callee) ? 3 : 1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9642 | // See if we can optimize any arguments passed through the varargs area of |
9643 | // the call. | ||||
9644 | for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(), | ||||
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9645 | E = CS.arg_end(); I != E; ++I, ++ix) { |
9646 | CastInst *CI = dyn_cast<CastInst>(*I); | ||||
9647 | if (CI && isSafeToEliminateVarargsCast(CS, CI, TD, ix)) { | ||||
9648 | *I = CI->getOperand(0); | ||||
9649 | Changed = true; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9650 | } |
Dale Johannesen | 3561546 | 2008-04-23 18:34:37 +0000 | [diff] [blame] | 9651 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9652 | } |
9653 | |||||
Duncan Sands | 2937e35 | 2007-12-19 21:13:37 +0000 | [diff] [blame] | 9654 | if (isa<InlineAsm>(Callee) && !CS.doesNotThrow()) { |
Duncan Sands | 7868f3c | 2007-12-16 15:51:49 +0000 | [diff] [blame] | 9655 | // Inline asm calls cannot throw - mark them 'nounwind'. |
Duncan Sands | 2937e35 | 2007-12-19 21:13:37 +0000 | [diff] [blame] | 9656 | CS.setDoesNotThrow(); |
Duncan Sands | 7868f3c | 2007-12-16 15:51:49 +0000 | [diff] [blame] | 9657 | Changed = true; |
9658 | } | ||||
9659 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9660 | return Changed ? CS.getInstruction() : 0; |
9661 | } | ||||
9662 | |||||
9663 | // transformConstExprCastCall - If the callee is a constexpr cast of a function, | ||||
9664 | // attempt to move the cast to the arguments of the call/invoke. | ||||
9665 | // | ||||
9666 | bool InstCombiner::transformConstExprCastCall(CallSite CS) { | ||||
9667 | if (!isa<ConstantExpr>(CS.getCalledValue())) return false; | ||||
9668 | ConstantExpr *CE = cast<ConstantExpr>(CS.getCalledValue()); | ||||
9669 | if (CE->getOpcode() != Instruction::BitCast || | ||||
9670 | !isa<Function>(CE->getOperand(0))) | ||||
9671 | return false; | ||||
9672 | Function *Callee = cast<Function>(CE->getOperand(0)); | ||||
9673 | Instruction *Caller = CS.getInstruction(); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9674 | const AttrListPtr &CallerPAL = CS.getAttributes(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9675 | |
9676 | // Okay, this is a cast from a function to a different type. Unless doing so | ||||
9677 | // would cause a type conversion of one of our arguments, change this call to | ||||
9678 | // be a direct call with arguments casted to the appropriate types. | ||||
9679 | // | ||||
9680 | const FunctionType *FT = Callee->getFunctionType(); | ||||
9681 | const Type *OldRetTy = Caller->getType(); | ||||
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9682 | const Type *NewRetTy = FT->getReturnType(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9683 | |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9684 | if (isa<StructType>(NewRetTy)) |
Devang Patel | d091d32 | 2008-03-11 18:04:06 +0000 | [diff] [blame] | 9685 | return false; // TODO: Handle multiple return values. |
9686 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9687 | // Check to see if we are changing the return type... |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9688 | if (OldRetTy != NewRetTy) { |
Bill Wendling | d9644a4 | 2008-05-14 22:45:20 +0000 | [diff] [blame] | 9689 | if (Callee->isDeclaration() && |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9690 | // Conversion is ok if changing from one pointer type to another or from |
9691 | // a pointer to an integer of the same size. | ||||
9692 | !((isa<PointerType>(OldRetTy) || OldRetTy == TD->getIntPtrType()) && | ||||
Duncan Sands | 886cadb | 2008-06-17 15:55:30 +0000 | [diff] [blame] | 9693 | (isa<PointerType>(NewRetTy) || NewRetTy == TD->getIntPtrType()))) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9694 | return false; // Cannot transform this return value. |
9695 | |||||
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9696 | if (!Caller->use_empty() && |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9697 | // void -> non-void is handled specially |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9698 | NewRetTy != Type::VoidTy && !CastInst::isCastable(NewRetTy, OldRetTy)) |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9699 | return false; // Cannot transform this return value. |
9700 | |||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9701 | if (!CallerPAL.isEmpty() && !Caller->use_empty()) { |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9702 | Attributes RAttrs = CallerPAL.getRetAttributes(); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9703 | if (RAttrs & Attribute::typeIncompatible(NewRetTy)) |
Duncan Sands | dbe97dc | 2008-01-07 17:16:06 +0000 | [diff] [blame] | 9704 | return false; // Attribute not compatible with transformed value. |
9705 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9706 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9707 | // If the callsite is an invoke instruction, and the return value is used by |
9708 | // a PHI node in a successor, we cannot change the return type of the call | ||||
9709 | // because there is no place to put the cast instruction (without breaking | ||||
9710 | // the critical edge). Bail out in this case. | ||||
9711 | if (!Caller->use_empty()) | ||||
9712 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) | ||||
9713 | for (Value::use_iterator UI = II->use_begin(), E = II->use_end(); | ||||
9714 | UI != E; ++UI) | ||||
9715 | if (PHINode *PN = dyn_cast<PHINode>(*UI)) | ||||
9716 | if (PN->getParent() == II->getNormalDest() || | ||||
9717 | PN->getParent() == II->getUnwindDest()) | ||||
9718 | return false; | ||||
9719 | } | ||||
9720 | |||||
9721 | unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin()); | ||||
9722 | unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs); | ||||
9723 | |||||
9724 | CallSite::arg_iterator AI = CS.arg_begin(); | ||||
9725 | for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) { | ||||
9726 | const Type *ParamTy = FT->getParamType(i); | ||||
9727 | const Type *ActTy = (*AI)->getType(); | ||||
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9728 | |
9729 | if (!CastInst::isCastable(ActTy, ParamTy)) | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9730 | return false; // Cannot transform this parameter value. |
9731 | |||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9732 | if (CallerPAL.getParamAttributes(i + 1) |
9733 | & Attribute::typeIncompatible(ParamTy)) | ||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9734 | return false; // Attribute not compatible with transformed value. |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9735 | |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9736 | // Converting from one pointer type to another or between a pointer and an |
9737 | // 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] | 9738 | bool isConvertible = ActTy == ParamTy || |
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9739 | ((isa<PointerType>(ParamTy) || ParamTy == TD->getIntPtrType()) && |
9740 | (isa<PointerType>(ActTy) || ActTy == TD->getIntPtrType())); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9741 | if (Callee->isDeclaration() && !isConvertible) return false; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9742 | } |
9743 | |||||
9744 | if (FT->getNumParams() < NumActualArgs && !FT->isVarArg() && | ||||
9745 | Callee->isDeclaration()) | ||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9746 | return false; // Do not delete arguments unless we have a function body. |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9747 | |
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9748 | if (FT->getNumParams() < NumActualArgs && FT->isVarArg() && |
9749 | !CallerPAL.isEmpty()) | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9750 | // 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] | 9751 | // won't be dropping them. Check that these extra arguments have attributes |
9752 | // that are compatible with being a vararg call argument. | ||||
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9753 | for (unsigned i = CallerPAL.getNumSlots(); i; --i) { |
9754 | if (CallerPAL.getSlot(i - 1).Index <= FT->getNumParams()) | ||||
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9755 | break; |
Devang Patel | e480dfa | 2008-09-23 23:03:40 +0000 | [diff] [blame] | 9756 | Attributes PAttrs = CallerPAL.getSlot(i - 1).Attrs; |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9757 | if (PAttrs & Attribute::VarArgsIncompatible) |
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9758 | return false; |
9759 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9760 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9761 | // Okay, we decided that this is a safe thing to do: go ahead and start |
9762 | // inserting cast instructions as necessary... | ||||
9763 | std::vector<Value*> Args; | ||||
9764 | Args.reserve(NumActualArgs); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9765 | SmallVector<AttributeWithIndex, 8> attrVec; |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9766 | attrVec.reserve(NumCommonArgs); |
9767 | |||||
9768 | // Get any return attributes. | ||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9769 | Attributes RAttrs = CallerPAL.getRetAttributes(); |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9770 | |
9771 | // If the return value is not being used, the type may not be compatible | ||||
9772 | // with the existing attributes. Wipe out any problematic attributes. | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9773 | RAttrs &= ~Attribute::typeIncompatible(NewRetTy); |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9774 | |
9775 | // Add the new return attributes. | ||||
9776 | if (RAttrs) | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9777 | attrVec.push_back(AttributeWithIndex::get(0, RAttrs)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9778 | |
9779 | AI = CS.arg_begin(); | ||||
9780 | for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) { | ||||
9781 | const Type *ParamTy = FT->getParamType(i); | ||||
9782 | if ((*AI)->getType() == ParamTy) { | ||||
9783 | Args.push_back(*AI); | ||||
9784 | } else { | ||||
9785 | Instruction::CastOps opcode = CastInst::getCastOpcode(*AI, | ||||
9786 | false, ParamTy, false); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9787 | CastInst *NewCast = CastInst::Create(opcode, *AI, ParamTy, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9788 | Args.push_back(InsertNewInstBefore(NewCast, *Caller)); |
9789 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9790 | |
9791 | // Add any parameter attributes. | ||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9792 | if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9793 | attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9794 | } |
9795 | |||||
9796 | // If the function takes more arguments than the call was taking, add them | ||||
9797 | // now... | ||||
9798 | for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i) | ||||
9799 | Args.push_back(Constant::getNullValue(FT->getParamType(i))); | ||||
9800 | |||||
9801 | // If we are removing arguments to the function, emit an obnoxious warning... | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 9802 | if (FT->getNumParams() < NumActualArgs) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9803 | if (!FT->isVarArg()) { |
9804 | cerr << "WARNING: While resolving call to function '" | ||||
9805 | << Callee->getName() << "' arguments were dropped!\n"; | ||||
9806 | } else { | ||||
9807 | // Add all of the arguments in their promoted form to the arg list... | ||||
9808 | for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) { | ||||
9809 | const Type *PTy = getPromotedType((*AI)->getType()); | ||||
9810 | if (PTy != (*AI)->getType()) { | ||||
9811 | // Must promote to pass through va_arg area! | ||||
9812 | Instruction::CastOps opcode = CastInst::getCastOpcode(*AI, false, | ||||
9813 | PTy, false); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9814 | Instruction *Cast = CastInst::Create(opcode, *AI, PTy, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9815 | InsertNewInstBefore(Cast, *Caller); |
9816 | Args.push_back(Cast); | ||||
9817 | } else { | ||||
9818 | Args.push_back(*AI); | ||||
9819 | } | ||||
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9820 | |
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9821 | // Add any parameter attributes. |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9822 | if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9823 | attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); |
Duncan Sands | 4ced1f8 | 2008-01-13 08:02:44 +0000 | [diff] [blame] | 9824 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9825 | } |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 9826 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9827 | |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9828 | if (Attributes FnAttrs = CallerPAL.getFnAttributes()) |
9829 | attrVec.push_back(AttributeWithIndex::get(~0, FnAttrs)); | ||||
9830 | |||||
Duncan Sands | 7901ce1 | 2008-06-01 07:38:42 +0000 | [diff] [blame] | 9831 | if (NewRetTy == Type::VoidTy) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9832 | Caller->setName(""); // Void type should not have a name. |
9833 | |||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9834 | const AttrListPtr &NewCallerPAL = AttrListPtr::get(attrVec.begin(),attrVec.end()); |
Duncan Sands | c849e66 | 2008-01-06 18:27:01 +0000 | [diff] [blame] | 9835 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9836 | Instruction *NC; |
9837 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9838 | NC = InvokeInst::Create(Callee, II->getNormalDest(), II->getUnwindDest(), |
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 9839 | Args.begin(), Args.end(), |
9840 | Caller->getName(), Caller); | ||||
Reid Spencer | 6b0b09a | 2007-07-30 19:53:57 +0000 | [diff] [blame] | 9841 | cast<InvokeInst>(NC)->setCallingConv(II->getCallingConv()); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9842 | cast<InvokeInst>(NC)->setAttributes(NewCallerPAL); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9843 | } else { |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 9844 | NC = CallInst::Create(Callee, Args.begin(), Args.end(), |
9845 | Caller->getName(), Caller); | ||||
Duncan Sands | f5588dc | 2007-11-27 13:23:08 +0000 | [diff] [blame] | 9846 | CallInst *CI = cast<CallInst>(Caller); |
9847 | if (CI->isTailCall()) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9848 | cast<CallInst>(NC)->setTailCall(); |
Duncan Sands | f5588dc | 2007-11-27 13:23:08 +0000 | [diff] [blame] | 9849 | cast<CallInst>(NC)->setCallingConv(CI->getCallingConv()); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9850 | cast<CallInst>(NC)->setAttributes(NewCallerPAL); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9851 | } |
9852 | |||||
9853 | // Insert a cast of the return type as necessary. | ||||
9854 | Value *NV = NC; | ||||
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9855 | if (OldRetTy != NV->getType() && !Caller->use_empty()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9856 | if (NV->getType() != Type::VoidTy) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9857 | Instruction::CastOps opcode = CastInst::getCastOpcode(NC, false, |
Duncan Sands | 5c48958 | 2008-01-06 10:12:28 +0000 | [diff] [blame] | 9858 | OldRetTy, false); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 9859 | NV = NC = CastInst::Create(opcode, NC, OldRetTy, "tmp"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9860 | |
9861 | // If this is an invoke instruction, we should insert it after the first | ||||
9862 | // non-phi, instruction in the normal successor block. | ||||
9863 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { | ||||
Dan Gohman | 514277c | 2008-05-23 21:05:58 +0000 | [diff] [blame] | 9864 | BasicBlock::iterator I = II->getNormalDest()->getFirstNonPHI(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 9865 | InsertNewInstBefore(NC, *I); |
9866 | } else { | ||||
9867 | // Otherwise, it's a call, just insert cast right after the call instr | ||||
9868 | InsertNewInstBefore(NC, *Caller); | ||||
9869 | } | ||||
9870 | AddUsersToWorkList(*Caller); | ||||
9871 | } else { | ||||
9872 | NV = UndefValue::get(Caller->getType()); | ||||
9873 | } | ||||
9874 | } | ||||
9875 | |||||
9876 | if (Caller->getType() != Type::VoidTy && !Caller->use_empty()) | ||||
9877 | Caller->replaceAllUsesWith(NV); | ||||
9878 | Caller->eraseFromParent(); | ||||
9879 | RemoveFromWorkList(Caller); | ||||
9880 | return true; | ||||
9881 | } | ||||
9882 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9883 | // transformCallThroughTrampoline - Turn a call to a function created by the |
9884 | // init_trampoline intrinsic into a direct call to the underlying function. | ||||
9885 | // | ||||
9886 | Instruction *InstCombiner::transformCallThroughTrampoline(CallSite CS) { | ||||
9887 | Value *Callee = CS.getCalledValue(); | ||||
9888 | const PointerType *PTy = cast<PointerType>(Callee->getType()); | ||||
9889 | const FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9890 | const AttrListPtr &Attrs = CS.getAttributes(); |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9891 | |
9892 | // If the call already has the 'nest' attribute somewhere then give up - | ||||
9893 | // otherwise 'nest' would occur twice after splicing in the chain. | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9894 | if (Attrs.hasAttrSomewhere(Attribute::Nest)) |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9895 | return 0; |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9896 | |
9897 | IntrinsicInst *Tramp = | ||||
9898 | cast<IntrinsicInst>(cast<BitCastInst>(Callee)->getOperand(0)); | ||||
9899 | |||||
Anton Korobeynikov | 48fc88f | 2008-05-07 22:54:15 +0000 | [diff] [blame] | 9900 | Function *NestF = cast<Function>(Tramp->getOperand(2)->stripPointerCasts()); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9901 | const PointerType *NestFPTy = cast<PointerType>(NestF->getType()); |
9902 | const FunctionType *NestFTy = cast<FunctionType>(NestFPTy->getElementType()); | ||||
9903 | |||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9904 | const AttrListPtr &NestAttrs = NestF->getAttributes(); |
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9905 | if (!NestAttrs.isEmpty()) { |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9906 | unsigned NestIdx = 1; |
9907 | const Type *NestTy = 0; | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9908 | Attributes NestAttr = Attribute::None; |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9909 | |
9910 | // Look for a parameter marked with the 'nest' attribute. | ||||
9911 | for (FunctionType::param_iterator I = NestFTy->param_begin(), | ||||
9912 | E = NestFTy->param_end(); I != E; ++NestIdx, ++I) | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9913 | if (NestAttrs.paramHasAttr(NestIdx, Attribute::Nest)) { |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9914 | // Record the parameter type and any other attributes. |
9915 | NestTy = *I; | ||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9916 | NestAttr = NestAttrs.getParamAttributes(NestIdx); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9917 | break; |
9918 | } | ||||
9919 | |||||
9920 | if (NestTy) { | ||||
9921 | Instruction *Caller = CS.getInstruction(); | ||||
9922 | std::vector<Value*> NewArgs; | ||||
9923 | NewArgs.reserve(unsigned(CS.arg_end()-CS.arg_begin())+1); | ||||
9924 | |||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9925 | SmallVector<AttributeWithIndex, 8> NewAttrs; |
Chris Lattner | 1c8733e | 2008-03-12 17:45:29 +0000 | [diff] [blame] | 9926 | NewAttrs.reserve(Attrs.getNumSlots() + 1); |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9927 | |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9928 | // Insert the nest argument into the call argument list, which may |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9929 | // mean appending it. Likewise for attributes. |
9930 | |||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9931 | // Add any result attributes. |
9932 | if (Attributes Attr = Attrs.getRetAttributes()) | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9933 | NewAttrs.push_back(AttributeWithIndex::get(0, Attr)); |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9934 | |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9935 | { |
9936 | unsigned Idx = 1; | ||||
9937 | CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); | ||||
9938 | do { | ||||
9939 | if (Idx == NestIdx) { | ||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9940 | // Add the chain argument and attributes. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9941 | Value *NestVal = Tramp->getOperand(3); |
9942 | if (NestVal->getType() != NestTy) | ||||
9943 | NestVal = new BitCastInst(NestVal, NestTy, "nest", Caller); | ||||
9944 | NewArgs.push_back(NestVal); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9945 | NewAttrs.push_back(AttributeWithIndex::get(NestIdx, NestAttr)); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9946 | } |
9947 | |||||
9948 | if (I == E) | ||||
9949 | break; | ||||
9950 | |||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9951 | // Add the original argument and attributes. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9952 | NewArgs.push_back(*I); |
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9953 | if (Attributes Attr = Attrs.getParamAttributes(Idx)) |
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9954 | NewAttrs.push_back |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 9955 | (AttributeWithIndex::get(Idx + (Idx >= NestIdx), Attr)); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9956 | |
9957 | ++Idx, ++I; | ||||
9958 | } while (1); | ||||
9959 | } | ||||
9960 | |||||
Devang Patel | f2a4a92 | 2008-09-26 22:53:05 +0000 | [diff] [blame] | 9961 | // Add any function attributes. |
9962 | if (Attributes Attr = Attrs.getFnAttributes()) | ||||
9963 | NewAttrs.push_back(AttributeWithIndex::get(~0, Attr)); | ||||
9964 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9965 | // The trampoline may have been bitcast to a bogus type (FTy). |
9966 | // Handle this by synthesizing a new function type, equal to FTy | ||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9967 | // with the chain parameter inserted. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9968 | |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9969 | std::vector<const Type*> NewTypes; |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9970 | NewTypes.reserve(FTy->getNumParams()+1); |
9971 | |||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9972 | // 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] | 9973 | // mean appending it. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9974 | { |
9975 | unsigned Idx = 1; | ||||
9976 | FunctionType::param_iterator I = FTy->param_begin(), | ||||
9977 | E = FTy->param_end(); | ||||
9978 | |||||
9979 | do { | ||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9980 | if (Idx == NestIdx) |
9981 | // Add the chain's type. | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9982 | NewTypes.push_back(NestTy); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9983 | |
9984 | if (I == E) | ||||
9985 | break; | ||||
9986 | |||||
Duncan Sands | 48b8111 | 2008-01-14 19:52:09 +0000 | [diff] [blame] | 9987 | // Add the original type. |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9988 | NewTypes.push_back(*I); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 9989 | |
9990 | ++Idx, ++I; | ||||
9991 | } while (1); | ||||
9992 | } | ||||
9993 | |||||
9994 | // Replace the trampoline call with a direct call. Let the generic | ||||
9995 | // code sort out any function type mismatches. | ||||
9996 | FunctionType *NewFTy = | ||||
Duncan Sands | f5588dc | 2007-11-27 13:23:08 +0000 | [diff] [blame] | 9997 | FunctionType::get(FTy->getReturnType(), NewTypes, FTy->isVarArg()); |
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 9998 | Constant *NewCallee = NestF->getType() == PointerType::getUnqual(NewFTy) ? |
9999 | NestF : ConstantExpr::getBitCast(NestF, PointerType::getUnqual(NewFTy)); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 10000 | const AttrListPtr &NewPAL = AttrListPtr::get(NewAttrs.begin(),NewAttrs.end()); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10001 | |
10002 | Instruction *NewCaller; | ||||
10003 | if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10004 | NewCaller = InvokeInst::Create(NewCallee, |
10005 | II->getNormalDest(), II->getUnwindDest(), | ||||
10006 | NewArgs.begin(), NewArgs.end(), | ||||
10007 | Caller->getName(), Caller); | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10008 | cast<InvokeInst>(NewCaller)->setCallingConv(II->getCallingConv()); |
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 10009 | cast<InvokeInst>(NewCaller)->setAttributes(NewPAL); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10010 | } else { |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10011 | NewCaller = CallInst::Create(NewCallee, NewArgs.begin(), NewArgs.end(), |
10012 | Caller->getName(), Caller); | ||||
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10013 | if (cast<CallInst>(Caller)->isTailCall()) |
10014 | cast<CallInst>(NewCaller)->setTailCall(); | ||||
10015 | cast<CallInst>(NewCaller)-> | ||||
10016 | setCallingConv(cast<CallInst>(Caller)->getCallingConv()); | ||||
Devang Patel | d222f86 | 2008-09-25 21:00:45 +0000 | [diff] [blame] | 10017 | cast<CallInst>(NewCaller)->setAttributes(NewPAL); |
Duncan Sands | 74833f2 | 2007-09-17 10:26:40 +0000 | [diff] [blame] | 10018 | } |
10019 | if (Caller->getType() != Type::VoidTy && !Caller->use_empty()) | ||||
10020 | Caller->replaceAllUsesWith(NewCaller); | ||||
10021 | Caller->eraseFromParent(); | ||||
10022 | RemoveFromWorkList(Caller); | ||||
10023 | return 0; | ||||
10024 | } | ||||
10025 | } | ||||
10026 | |||||
10027 | // Replace the trampoline call with a direct call. Since there is no 'nest' | ||||
10028 | // parameter, there is no need to adjust the argument list. Let the generic | ||||
10029 | // code sort out any function type mismatches. | ||||
10030 | Constant *NewCallee = | ||||
10031 | NestF->getType() == PTy ? NestF : ConstantExpr::getBitCast(NestF, PTy); | ||||
10032 | CS.setCalledFunction(NewCallee); | ||||
10033 | return CS.getInstruction(); | ||||
10034 | } | ||||
10035 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10036 | /// FoldPHIArgBinOpIntoPHI - If we have something like phi [add (a,b), add(c,d)] |
10037 | /// and if a/b/c/d and the add's all have a single use, turn this into two phi's | ||||
10038 | /// and a single binop. | ||||
10039 | Instruction *InstCombiner::FoldPHIArgBinOpIntoPHI(PHINode &PN) { | ||||
10040 | Instruction *FirstInst = cast<Instruction>(PN.getIncomingValue(0)); | ||||
Chris Lattner | 3007801 | 2008-12-01 03:42:51 +0000 | [diff] [blame] | 10041 | assert(isa<BinaryOperator>(FirstInst) || isa<CmpInst>(FirstInst)); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10042 | unsigned Opc = FirstInst->getOpcode(); |
10043 | Value *LHSVal = FirstInst->getOperand(0); | ||||
10044 | Value *RHSVal = FirstInst->getOperand(1); | ||||
10045 | |||||
10046 | const Type *LHSType = LHSVal->getType(); | ||||
10047 | const Type *RHSType = RHSVal->getType(); | ||||
10048 | |||||
10049 | // Scan to see if all operands are the same opcode, all have one use, and all | ||||
10050 | // kill their operands (i.e. the operands have one use). | ||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10051 | for (unsigned i = 1; i != PN.getNumIncomingValues(); ++i) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10052 | Instruction *I = dyn_cast<Instruction>(PN.getIncomingValue(i)); |
10053 | if (!I || I->getOpcode() != Opc || !I->hasOneUse() || | ||||
10054 | // Verify type of the LHS matches so we don't fold cmp's of different | ||||
10055 | // types or GEP's with different index types. | ||||
10056 | I->getOperand(0)->getType() != LHSType || | ||||
10057 | I->getOperand(1)->getType() != RHSType) | ||||
10058 | return 0; | ||||
10059 | |||||
10060 | // If they are CmpInst instructions, check their predicates | ||||
10061 | if (Opc == Instruction::ICmp || Opc == Instruction::FCmp) | ||||
10062 | if (cast<CmpInst>(I)->getPredicate() != | ||||
10063 | cast<CmpInst>(FirstInst)->getPredicate()) | ||||
10064 | return 0; | ||||
10065 | |||||
10066 | // Keep track of which operand needs a phi node. | ||||
10067 | if (I->getOperand(0) != LHSVal) LHSVal = 0; | ||||
10068 | if (I->getOperand(1) != RHSVal) RHSVal = 0; | ||||
10069 | } | ||||
10070 | |||||
Chris Lattner | 3007801 | 2008-12-01 03:42:51 +0000 | [diff] [blame] | 10071 | // Otherwise, this is safe to transform! |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10072 | |
10073 | Value *InLHS = FirstInst->getOperand(0); | ||||
10074 | Value *InRHS = FirstInst->getOperand(1); | ||||
10075 | PHINode *NewLHS = 0, *NewRHS = 0; | ||||
10076 | if (LHSVal == 0) { | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 10077 | NewLHS = PHINode::Create(LHSType, |
10078 | FirstInst->getOperand(0)->getName() + ".pn"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10079 | NewLHS->reserveOperandSpace(PN.getNumOperands()/2); |
10080 | NewLHS->addIncoming(InLHS, PN.getIncomingBlock(0)); | ||||
10081 | InsertNewInstBefore(NewLHS, PN); | ||||
10082 | LHSVal = NewLHS; | ||||
10083 | } | ||||
10084 | |||||
10085 | if (RHSVal == 0) { | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 10086 | NewRHS = PHINode::Create(RHSType, |
10087 | FirstInst->getOperand(1)->getName() + ".pn"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10088 | NewRHS->reserveOperandSpace(PN.getNumOperands()/2); |
10089 | NewRHS->addIncoming(InRHS, PN.getIncomingBlock(0)); | ||||
10090 | InsertNewInstBefore(NewRHS, PN); | ||||
10091 | RHSVal = NewRHS; | ||||
10092 | } | ||||
10093 | |||||
10094 | // Add all operands to the new PHIs. | ||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10095 | if (NewLHS || NewRHS) { |
10096 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
10097 | Instruction *InInst = cast<Instruction>(PN.getIncomingValue(i)); | ||||
10098 | if (NewLHS) { | ||||
10099 | Value *NewInLHS = InInst->getOperand(0); | ||||
10100 | NewLHS->addIncoming(NewInLHS, PN.getIncomingBlock(i)); | ||||
10101 | } | ||||
10102 | if (NewRHS) { | ||||
10103 | Value *NewInRHS = InInst->getOperand(1); | ||||
10104 | NewRHS->addIncoming(NewInRHS, PN.getIncomingBlock(i)); | ||||
10105 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10106 | } |
10107 | } | ||||
10108 | |||||
10109 | if (BinaryOperator *BinOp = dyn_cast<BinaryOperator>(FirstInst)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10110 | return BinaryOperator::Create(BinOp->getOpcode(), LHSVal, RHSVal); |
Chris Lattner | 3007801 | 2008-12-01 03:42:51 +0000 | [diff] [blame] | 10111 | CmpInst *CIOp = cast<CmpInst>(FirstInst); |
10112 | return CmpInst::Create(CIOp->getOpcode(), CIOp->getPredicate(), LHSVal, | ||||
10113 | RHSVal); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10114 | } |
10115 | |||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10116 | Instruction *InstCombiner::FoldPHIArgGEPIntoPHI(PHINode &PN) { |
10117 | GetElementPtrInst *FirstInst =cast<GetElementPtrInst>(PN.getIncomingValue(0)); | ||||
10118 | |||||
10119 | SmallVector<Value*, 16> FixedOperands(FirstInst->op_begin(), | ||||
10120 | FirstInst->op_end()); | ||||
10121 | |||||
10122 | // Scan to see if all operands are the same opcode, all have one use, and all | ||||
10123 | // kill their operands (i.e. the operands have one use). | ||||
10124 | for (unsigned i = 1; i != PN.getNumIncomingValues(); ++i) { | ||||
10125 | GetElementPtrInst *GEP= dyn_cast<GetElementPtrInst>(PN.getIncomingValue(i)); | ||||
10126 | if (!GEP || !GEP->hasOneUse() || GEP->getType() != FirstInst->getType() || | ||||
10127 | GEP->getNumOperands() != FirstInst->getNumOperands()) | ||||
10128 | return 0; | ||||
10129 | |||||
10130 | // Compare the operand lists. | ||||
10131 | for (unsigned op = 0, e = FirstInst->getNumOperands(); op != e; ++op) { | ||||
10132 | if (FirstInst->getOperand(op) == GEP->getOperand(op)) | ||||
10133 | continue; | ||||
10134 | |||||
10135 | // Don't merge two GEPs when two operands differ (introducing phi nodes) | ||||
10136 | // if one of the PHIs has a constant for the index. The index may be | ||||
10137 | // substantially cheaper to compute for the constants, so making it a | ||||
10138 | // variable index could pessimize the path. This also handles the case | ||||
10139 | // for struct indices, which must always be constant. | ||||
10140 | if (isa<ConstantInt>(FirstInst->getOperand(op)) || | ||||
10141 | isa<ConstantInt>(GEP->getOperand(op))) | ||||
10142 | return 0; | ||||
10143 | |||||
10144 | if (FirstInst->getOperand(op)->getType() !=GEP->getOperand(op)->getType()) | ||||
10145 | return 0; | ||||
10146 | FixedOperands[op] = 0; // Needs a PHI. | ||||
10147 | } | ||||
10148 | } | ||||
10149 | |||||
10150 | // Otherwise, this is safe to transform. Insert PHI nodes for each operand | ||||
10151 | // that is variable. | ||||
10152 | SmallVector<PHINode*, 16> OperandPhis(FixedOperands.size()); | ||||
10153 | |||||
10154 | bool HasAnyPHIs = false; | ||||
10155 | for (unsigned i = 0, e = FixedOperands.size(); i != e; ++i) { | ||||
10156 | if (FixedOperands[i]) continue; // operand doesn't need a phi. | ||||
10157 | Value *FirstOp = FirstInst->getOperand(i); | ||||
10158 | PHINode *NewPN = PHINode::Create(FirstOp->getType(), | ||||
10159 | FirstOp->getName()+".pn"); | ||||
10160 | InsertNewInstBefore(NewPN, PN); | ||||
10161 | |||||
10162 | NewPN->reserveOperandSpace(e); | ||||
10163 | NewPN->addIncoming(FirstOp, PN.getIncomingBlock(0)); | ||||
10164 | OperandPhis[i] = NewPN; | ||||
10165 | FixedOperands[i] = NewPN; | ||||
10166 | HasAnyPHIs = true; | ||||
10167 | } | ||||
10168 | |||||
10169 | |||||
10170 | // Add all operands to the new PHIs. | ||||
10171 | if (HasAnyPHIs) { | ||||
10172 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
10173 | GetElementPtrInst *InGEP =cast<GetElementPtrInst>(PN.getIncomingValue(i)); | ||||
10174 | BasicBlock *InBB = PN.getIncomingBlock(i); | ||||
10175 | |||||
10176 | for (unsigned op = 0, e = OperandPhis.size(); op != e; ++op) | ||||
10177 | if (PHINode *OpPhi = OperandPhis[op]) | ||||
10178 | OpPhi->addIncoming(InGEP->getOperand(op), InBB); | ||||
10179 | } | ||||
10180 | } | ||||
10181 | |||||
10182 | Value *Base = FixedOperands[0]; | ||||
10183 | return GetElementPtrInst::Create(Base, FixedOperands.begin()+1, | ||||
10184 | FixedOperands.end()); | ||||
10185 | } | ||||
10186 | |||||
10187 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10188 | /// isSafeToSinkLoad - Return true if we know that it is safe sink the load out |
10189 | /// of the block that defines it. This means that it must be obvious the value | ||||
10190 | /// of the load is not changed from the point of the load to the end of the | ||||
10191 | /// block it is in. | ||||
10192 | /// | ||||
10193 | /// Finally, it is safe, but not profitable, to sink a load targetting a | ||||
10194 | /// non-address-taken alloca. Doing so will cause us to not promote the alloca | ||||
10195 | /// to a register. | ||||
10196 | static bool isSafeToSinkLoad(LoadInst *L) { | ||||
10197 | BasicBlock::iterator BBI = L, E = L->getParent()->end(); | ||||
10198 | |||||
10199 | for (++BBI; BBI != E; ++BBI) | ||||
10200 | if (BBI->mayWriteToMemory()) | ||||
10201 | return false; | ||||
10202 | |||||
10203 | // Check for non-address taken alloca. If not address-taken already, it isn't | ||||
10204 | // profitable to do this xform. | ||||
10205 | if (AllocaInst *AI = dyn_cast<AllocaInst>(L->getOperand(0))) { | ||||
10206 | bool isAddressTaken = false; | ||||
10207 | for (Value::use_iterator UI = AI->use_begin(), E = AI->use_end(); | ||||
10208 | UI != E; ++UI) { | ||||
10209 | if (isa<LoadInst>(UI)) continue; | ||||
10210 | if (StoreInst *SI = dyn_cast<StoreInst>(*UI)) { | ||||
10211 | // If storing TO the alloca, then the address isn't taken. | ||||
10212 | if (SI->getOperand(1) == AI) continue; | ||||
10213 | } | ||||
10214 | isAddressTaken = true; | ||||
10215 | break; | ||||
10216 | } | ||||
10217 | |||||
10218 | if (!isAddressTaken) | ||||
10219 | return false; | ||||
10220 | } | ||||
10221 | |||||
10222 | return true; | ||||
10223 | } | ||||
10224 | |||||
10225 | |||||
10226 | // FoldPHIArgOpIntoPHI - If all operands to a PHI node are the same "unary" | ||||
10227 | // operator and they all are only used by the PHI, PHI together their | ||||
10228 | // inputs, and do the operation once, to the result of the PHI. | ||||
10229 | Instruction *InstCombiner::FoldPHIArgOpIntoPHI(PHINode &PN) { | ||||
10230 | Instruction *FirstInst = cast<Instruction>(PN.getIncomingValue(0)); | ||||
10231 | |||||
10232 | // Scan the instruction, looking for input operations that can be folded away. | ||||
10233 | // If all input operands to the phi are the same instruction (e.g. a cast from | ||||
10234 | // the same type or "+42") we can pull the operation through the PHI, reducing | ||||
10235 | // code size and simplifying code. | ||||
10236 | Constant *ConstantOp = 0; | ||||
10237 | const Type *CastSrcTy = 0; | ||||
10238 | bool isVolatile = false; | ||||
10239 | if (isa<CastInst>(FirstInst)) { | ||||
10240 | CastSrcTy = FirstInst->getOperand(0)->getType(); | ||||
10241 | } else if (isa<BinaryOperator>(FirstInst) || isa<CmpInst>(FirstInst)) { | ||||
10242 | // Can fold binop, compare or shift here if the RHS is a constant, | ||||
10243 | // otherwise call FoldPHIArgBinOpIntoPHI. | ||||
10244 | ConstantOp = dyn_cast<Constant>(FirstInst->getOperand(1)); | ||||
10245 | if (ConstantOp == 0) | ||||
10246 | return FoldPHIArgBinOpIntoPHI(PN); | ||||
10247 | } else if (LoadInst *LI = dyn_cast<LoadInst>(FirstInst)) { | ||||
10248 | isVolatile = LI->isVolatile(); | ||||
10249 | // We can't sink the load if the loaded value could be modified between the | ||||
10250 | // load and the PHI. | ||||
10251 | if (LI->getParent() != PN.getIncomingBlock(0) || | ||||
10252 | !isSafeToSinkLoad(LI)) | ||||
10253 | return 0; | ||||
Chris Lattner | 2d9fdd8 | 2008-07-08 17:18:32 +0000 | [diff] [blame] | 10254 | |
10255 | // If the PHI is of volatile loads and the load block has multiple | ||||
10256 | // successors, sinking it would remove a load of the volatile value from | ||||
10257 | // the path through the other successor. | ||||
10258 | if (isVolatile && | ||||
10259 | LI->getParent()->getTerminator()->getNumSuccessors() != 1) | ||||
10260 | return 0; | ||||
10261 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10262 | } else if (isa<GetElementPtrInst>(FirstInst)) { |
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10263 | return FoldPHIArgGEPIntoPHI(PN); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10264 | } else { |
10265 | return 0; // Cannot fold this operation. | ||||
10266 | } | ||||
10267 | |||||
10268 | // Check to see if all arguments are the same operation. | ||||
10269 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
10270 | if (!isa<Instruction>(PN.getIncomingValue(i))) return 0; | ||||
10271 | Instruction *I = cast<Instruction>(PN.getIncomingValue(i)); | ||||
10272 | if (!I->hasOneUse() || !I->isSameOperationAs(FirstInst)) | ||||
10273 | return 0; | ||||
10274 | if (CastSrcTy) { | ||||
10275 | if (I->getOperand(0)->getType() != CastSrcTy) | ||||
10276 | return 0; // Cast operation must match. | ||||
10277 | } else if (LoadInst *LI = dyn_cast<LoadInst>(I)) { | ||||
10278 | // We can't sink the load if the loaded value could be modified between | ||||
10279 | // the load and the PHI. | ||||
10280 | if (LI->isVolatile() != isVolatile || | ||||
10281 | LI->getParent() != PN.getIncomingBlock(i) || | ||||
10282 | !isSafeToSinkLoad(LI)) | ||||
10283 | return 0; | ||||
Chris Lattner | f786701 | 2008-04-29 17:28:22 +0000 | [diff] [blame] | 10284 | |
Chris Lattner | 2d9fdd8 | 2008-07-08 17:18:32 +0000 | [diff] [blame] | 10285 | // If the PHI is of volatile loads and the load block has multiple |
10286 | // successors, sinking it would remove a load of the volatile value from | ||||
10287 | // the path through the other successor. | ||||
Chris Lattner | f786701 | 2008-04-29 17:28:22 +0000 | [diff] [blame] | 10288 | if (isVolatile && |
10289 | LI->getParent()->getTerminator()->getNumSuccessors() != 1) | ||||
10290 | return 0; | ||||
10291 | |||||
10292 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10293 | } else if (I->getOperand(1) != ConstantOp) { |
10294 | return 0; | ||||
10295 | } | ||||
10296 | } | ||||
10297 | |||||
10298 | // Okay, they are all the same operation. Create a new PHI node of the | ||||
10299 | // 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] | 10300 | PHINode *NewPN = PHINode::Create(FirstInst->getOperand(0)->getType(), |
10301 | PN.getName()+".in"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10302 | NewPN->reserveOperandSpace(PN.getNumOperands()/2); |
10303 | |||||
10304 | Value *InVal = FirstInst->getOperand(0); | ||||
10305 | NewPN->addIncoming(InVal, PN.getIncomingBlock(0)); | ||||
10306 | |||||
10307 | // Add all operands to the new PHI. | ||||
10308 | for (unsigned i = 1, e = PN.getNumIncomingValues(); i != e; ++i) { | ||||
10309 | Value *NewInVal = cast<Instruction>(PN.getIncomingValue(i))->getOperand(0); | ||||
10310 | if (NewInVal != InVal) | ||||
10311 | InVal = 0; | ||||
10312 | NewPN->addIncoming(NewInVal, PN.getIncomingBlock(i)); | ||||
10313 | } | ||||
10314 | |||||
10315 | Value *PhiVal; | ||||
10316 | if (InVal) { | ||||
10317 | // The new PHI unions all of the same values together. This is really | ||||
10318 | // common, so we handle it intelligently here for compile-time speed. | ||||
10319 | PhiVal = InVal; | ||||
10320 | delete NewPN; | ||||
10321 | } else { | ||||
10322 | InsertNewInstBefore(NewPN, PN); | ||||
10323 | PhiVal = NewPN; | ||||
10324 | } | ||||
10325 | |||||
10326 | // Insert and return the new operation. | ||||
10327 | if (CastInst* FirstCI = dyn_cast<CastInst>(FirstInst)) | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10328 | return CastInst::Create(FirstCI->getOpcode(), PhiVal, PN.getType()); |
Chris Lattner | fc984e9 | 2008-04-29 17:13:43 +0000 | [diff] [blame] | 10329 | if (BinaryOperator *BinOp = dyn_cast<BinaryOperator>(FirstInst)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10330 | return BinaryOperator::Create(BinOp->getOpcode(), PhiVal, ConstantOp); |
Chris Lattner | fc984e9 | 2008-04-29 17:13:43 +0000 | [diff] [blame] | 10331 | if (CmpInst *CIOp = dyn_cast<CmpInst>(FirstInst)) |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10332 | return CmpInst::Create(CIOp->getOpcode(), CIOp->getPredicate(), |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10333 | PhiVal, ConstantOp); |
Chris Lattner | fc984e9 | 2008-04-29 17:13:43 +0000 | [diff] [blame] | 10334 | assert(isa<LoadInst>(FirstInst) && "Unknown operation"); |
10335 | |||||
10336 | // If this was a volatile load that we are merging, make sure to loop through | ||||
10337 | // and mark all the input loads as non-volatile. If we don't do this, we will | ||||
10338 | // insert a new volatile load and the old ones will not be deletable. | ||||
10339 | if (isVolatile) | ||||
10340 | for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) | ||||
10341 | cast<LoadInst>(PN.getIncomingValue(i))->setVolatile(false); | ||||
10342 | |||||
10343 | return new LoadInst(PhiVal, "", isVolatile); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10344 | } |
10345 | |||||
10346 | /// DeadPHICycle - Return true if this PHI node is only used by a PHI node cycle | ||||
10347 | /// that is dead. | ||||
10348 | static bool DeadPHICycle(PHINode *PN, | ||||
10349 | SmallPtrSet<PHINode*, 16> &PotentiallyDeadPHIs) { | ||||
10350 | if (PN->use_empty()) return true; | ||||
10351 | if (!PN->hasOneUse()) return false; | ||||
10352 | |||||
10353 | // Remember this node, and if we find the cycle, return. | ||||
10354 | if (!PotentiallyDeadPHIs.insert(PN)) | ||||
10355 | return true; | ||||
Chris Lattner | adf2e34 | 2007-08-28 04:23:55 +0000 | [diff] [blame] | 10356 | |
10357 | // Don't scan crazily complex things. | ||||
10358 | if (PotentiallyDeadPHIs.size() == 16) | ||||
10359 | return false; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10360 | |
10361 | if (PHINode *PU = dyn_cast<PHINode>(PN->use_back())) | ||||
10362 | return DeadPHICycle(PU, PotentiallyDeadPHIs); | ||||
10363 | |||||
10364 | return false; | ||||
10365 | } | ||||
10366 | |||||
Chris Lattner | 27b695d | 2007-11-06 21:52:06 +0000 | [diff] [blame] | 10367 | /// PHIsEqualValue - Return true if this phi node is always equal to |
10368 | /// NonPhiInVal. This happens with mutually cyclic phi nodes like: | ||||
10369 | /// z = some value; x = phi (y, z); y = phi (x, z) | ||||
10370 | static bool PHIsEqualValue(PHINode *PN, Value *NonPhiInVal, | ||||
10371 | SmallPtrSet<PHINode*, 16> &ValueEqualPHIs) { | ||||
10372 | // See if we already saw this PHI node. | ||||
10373 | if (!ValueEqualPHIs.insert(PN)) | ||||
10374 | return true; | ||||
10375 | |||||
10376 | // Don't scan crazily complex things. | ||||
10377 | if (ValueEqualPHIs.size() == 16) | ||||
10378 | return false; | ||||
10379 | |||||
10380 | // Scan the operands to see if they are either phi nodes or are equal to | ||||
10381 | // the value. | ||||
10382 | for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) { | ||||
10383 | Value *Op = PN->getIncomingValue(i); | ||||
10384 | if (PHINode *OpPN = dyn_cast<PHINode>(Op)) { | ||||
10385 | if (!PHIsEqualValue(OpPN, NonPhiInVal, ValueEqualPHIs)) | ||||
10386 | return false; | ||||
10387 | } else if (Op != NonPhiInVal) | ||||
10388 | return false; | ||||
10389 | } | ||||
10390 | |||||
10391 | return true; | ||||
10392 | } | ||||
10393 | |||||
10394 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10395 | // PHINode simplification |
10396 | // | ||||
10397 | Instruction *InstCombiner::visitPHINode(PHINode &PN) { | ||||
10398 | // If LCSSA is around, don't mess with Phi nodes | ||||
10399 | if (MustPreserveLCSSA) return 0; | ||||
10400 | |||||
10401 | if (Value *V = PN.hasConstantValue()) | ||||
10402 | return ReplaceInstUsesWith(PN, V); | ||||
10403 | |||||
10404 | // If all PHI operands are the same operation, pull them through the PHI, | ||||
10405 | // reducing code size. | ||||
10406 | if (isa<Instruction>(PN.getIncomingValue(0)) && | ||||
Chris Lattner | 9e1916e | 2008-12-01 02:34:36 +0000 | [diff] [blame] | 10407 | isa<Instruction>(PN.getIncomingValue(1)) && |
10408 | cast<Instruction>(PN.getIncomingValue(0))->getOpcode() == | ||||
10409 | cast<Instruction>(PN.getIncomingValue(1))->getOpcode() && | ||||
10410 | // FIXME: The hasOneUse check will fail for PHIs that use the value more | ||||
10411 | // than themselves more than once. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10412 | PN.getIncomingValue(0)->hasOneUse()) |
10413 | if (Instruction *Result = FoldPHIArgOpIntoPHI(PN)) | ||||
10414 | return Result; | ||||
10415 | |||||
10416 | // If this is a trivial cycle in the PHI node graph, remove it. Basically, if | ||||
10417 | // this PHI only has a single use (a PHI), and if that PHI only has one use (a | ||||
10418 | // PHI)... break the cycle. | ||||
10419 | if (PN.hasOneUse()) { | ||||
10420 | Instruction *PHIUser = cast<Instruction>(PN.use_back()); | ||||
10421 | if (PHINode *PU = dyn_cast<PHINode>(PHIUser)) { | ||||
10422 | SmallPtrSet<PHINode*, 16> PotentiallyDeadPHIs; | ||||
10423 | PotentiallyDeadPHIs.insert(&PN); | ||||
10424 | if (DeadPHICycle(PU, PotentiallyDeadPHIs)) | ||||
10425 | return ReplaceInstUsesWith(PN, UndefValue::get(PN.getType())); | ||||
10426 | } | ||||
10427 | |||||
10428 | // If this phi has a single use, and if that use just computes a value for | ||||
10429 | // the next iteration of a loop, delete the phi. This occurs with unused | ||||
10430 | // induction variables, e.g. "for (int j = 0; ; ++j);". Detecting this | ||||
10431 | // common case here is good because the only other things that catch this | ||||
10432 | // are induction variable analysis (sometimes) and ADCE, which is only run | ||||
10433 | // late. | ||||
10434 | if (PHIUser->hasOneUse() && | ||||
10435 | (isa<BinaryOperator>(PHIUser) || isa<GetElementPtrInst>(PHIUser)) && | ||||
10436 | PHIUser->use_back() == &PN) { | ||||
10437 | return ReplaceInstUsesWith(PN, UndefValue::get(PN.getType())); | ||||
10438 | } | ||||
10439 | } | ||||
10440 | |||||
Chris Lattner | 27b695d | 2007-11-06 21:52:06 +0000 | [diff] [blame] | 10441 | // We sometimes end up with phi cycles that non-obviously end up being the |
10442 | // same value, for example: | ||||
10443 | // z = some value; x = phi (y, z); y = phi (x, z) | ||||
10444 | // where the phi nodes don't necessarily need to be in the same block. Do a | ||||
10445 | // quick check to see if the PHI node only contains a single non-phi value, if | ||||
10446 | // so, scan to see if the phi cycle is actually equal to that value. | ||||
10447 | { | ||||
10448 | unsigned InValNo = 0, NumOperandVals = PN.getNumIncomingValues(); | ||||
10449 | // Scan for the first non-phi operand. | ||||
10450 | while (InValNo != NumOperandVals && | ||||
10451 | isa<PHINode>(PN.getIncomingValue(InValNo))) | ||||
10452 | ++InValNo; | ||||
10453 | |||||
10454 | if (InValNo != NumOperandVals) { | ||||
10455 | Value *NonPhiInVal = PN.getOperand(InValNo); | ||||
10456 | |||||
10457 | // Scan the rest of the operands to see if there are any conflicts, if so | ||||
10458 | // there is no need to recursively scan other phis. | ||||
10459 | for (++InValNo; InValNo != NumOperandVals; ++InValNo) { | ||||
10460 | Value *OpVal = PN.getIncomingValue(InValNo); | ||||
10461 | if (OpVal != NonPhiInVal && !isa<PHINode>(OpVal)) | ||||
10462 | break; | ||||
10463 | } | ||||
10464 | |||||
10465 | // If we scanned over all operands, then we have one unique value plus | ||||
10466 | // phi values. Scan PHI nodes to see if they all merge in each other or | ||||
10467 | // the value. | ||||
10468 | if (InValNo == NumOperandVals) { | ||||
10469 | SmallPtrSet<PHINode*, 16> ValueEqualPHIs; | ||||
10470 | if (PHIsEqualValue(&PN, NonPhiInVal, ValueEqualPHIs)) | ||||
10471 | return ReplaceInstUsesWith(PN, NonPhiInVal); | ||||
10472 | } | ||||
10473 | } | ||||
10474 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10475 | return 0; |
10476 | } | ||||
10477 | |||||
10478 | static Value *InsertCastToIntPtrTy(Value *V, const Type *DTy, | ||||
10479 | Instruction *InsertPoint, | ||||
10480 | InstCombiner *IC) { | ||||
10481 | unsigned PtrSize = DTy->getPrimitiveSizeInBits(); | ||||
10482 | unsigned VTySize = V->getType()->getPrimitiveSizeInBits(); | ||||
10483 | // We must cast correctly to the pointer type. Ensure that we | ||||
10484 | // sign extend the integer value if it is smaller as this is | ||||
10485 | // used for address computation. | ||||
10486 | Instruction::CastOps opcode = | ||||
10487 | (VTySize < PtrSize ? Instruction::SExt : | ||||
10488 | (VTySize == PtrSize ? Instruction::BitCast : Instruction::Trunc)); | ||||
10489 | return IC->InsertCastBefore(opcode, V, DTy, *InsertPoint); | ||||
10490 | } | ||||
10491 | |||||
10492 | |||||
10493 | Instruction *InstCombiner::visitGetElementPtrInst(GetElementPtrInst &GEP) { | ||||
10494 | Value *PtrOp = GEP.getOperand(0); | ||||
10495 | // Is it 'getelementptr %P, i32 0' or 'getelementptr %P' | ||||
10496 | // If so, eliminate the noop. | ||||
10497 | if (GEP.getNumOperands() == 1) | ||||
10498 | return ReplaceInstUsesWith(GEP, PtrOp); | ||||
10499 | |||||
10500 | if (isa<UndefValue>(GEP.getOperand(0))) | ||||
10501 | return ReplaceInstUsesWith(GEP, UndefValue::get(GEP.getType())); | ||||
10502 | |||||
10503 | bool HasZeroPointerIndex = false; | ||||
10504 | if (Constant *C = dyn_cast<Constant>(GEP.getOperand(1))) | ||||
10505 | HasZeroPointerIndex = C->isNullValue(); | ||||
10506 | |||||
10507 | if (GEP.getNumOperands() == 2 && HasZeroPointerIndex) | ||||
10508 | return ReplaceInstUsesWith(GEP, PtrOp); | ||||
10509 | |||||
10510 | // Eliminate unneeded casts for indices. | ||||
10511 | bool MadeChange = false; | ||||
10512 | |||||
10513 | gep_type_iterator GTI = gep_type_begin(GEP); | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10514 | for (User::op_iterator i = GEP.op_begin() + 1, e = GEP.op_end(); |
10515 | i != e; ++i, ++GTI) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10516 | if (isa<SequentialType>(*GTI)) { |
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10517 | if (CastInst *CI = dyn_cast<CastInst>(*i)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10518 | if (CI->getOpcode() == Instruction::ZExt || |
10519 | CI->getOpcode() == Instruction::SExt) { | ||||
10520 | const Type *SrcTy = CI->getOperand(0)->getType(); | ||||
10521 | // We can eliminate a cast from i32 to i64 iff the target | ||||
10522 | // is a 32-bit pointer target. | ||||
10523 | if (SrcTy->getPrimitiveSizeInBits() >= TD->getPointerSizeInBits()) { | ||||
10524 | MadeChange = true; | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10525 | *i = CI->getOperand(0); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10526 | } |
10527 | } | ||||
10528 | } | ||||
10529 | // 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] | 10530 | // to what we need. If narrower, sign-extend it to what we need. |
10531 | // If the incoming value needs a cast instruction, | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10532 | // insert it. This explicit cast can make subsequent optimizations more |
10533 | // obvious. | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10534 | Value *Op = *i; |
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10535 | if (TD->getTypeSizeInBits(Op->getType()) > TD->getPointerSizeInBits()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10536 | if (Constant *C = dyn_cast<Constant>(Op)) { |
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10537 | *i = ConstantExpr::getTrunc(C, TD->getIntPtrType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10538 | MadeChange = true; |
10539 | } else { | ||||
10540 | Op = InsertCastBefore(Instruction::Trunc, Op, TD->getIntPtrType(), | ||||
10541 | GEP); | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 10542 | *i = Op; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10543 | MadeChange = true; |
10544 | } | ||||
Dan Gohman | 5d639ed | 2008-09-11 23:06:38 +0000 | [diff] [blame] | 10545 | } else if (TD->getTypeSizeInBits(Op->getType()) < TD->getPointerSizeInBits()) { |
10546 | if (Constant *C = dyn_cast<Constant>(Op)) { | ||||
10547 | *i = ConstantExpr::getSExt(C, TD->getIntPtrType()); | ||||
10548 | MadeChange = true; | ||||
10549 | } else { | ||||
10550 | Op = InsertCastBefore(Instruction::SExt, Op, TD->getIntPtrType(), | ||||
10551 | GEP); | ||||
10552 | *i = Op; | ||||
10553 | MadeChange = true; | ||||
10554 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10555 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10556 | } |
10557 | } | ||||
10558 | if (MadeChange) return &GEP; | ||||
10559 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10560 | // Combine Indices - If the source pointer to this getelementptr instruction |
10561 | // is a getelementptr instruction, combine the indices of the two | ||||
10562 | // getelementptr instructions into a single instruction. | ||||
10563 | // | ||||
10564 | SmallVector<Value*, 8> SrcGEPOperands; | ||||
10565 | if (User *Src = dyn_castGetElementPtr(PtrOp)) | ||||
10566 | SrcGEPOperands.append(Src->op_begin(), Src->op_end()); | ||||
10567 | |||||
10568 | if (!SrcGEPOperands.empty()) { | ||||
10569 | // Note that if our source is a gep chain itself that we wait for that | ||||
10570 | // chain to be resolved before we perform this transformation. This | ||||
10571 | // avoids us creating a TON of code in some cases. | ||||
10572 | // | ||||
10573 | if (isa<GetElementPtrInst>(SrcGEPOperands[0]) && | ||||
10574 | cast<Instruction>(SrcGEPOperands[0])->getNumOperands() == 2) | ||||
10575 | return 0; // Wait until our source is folded to completion. | ||||
10576 | |||||
10577 | SmallVector<Value*, 8> Indices; | ||||
10578 | |||||
10579 | // Find out whether the last index in the source GEP is a sequential idx. | ||||
10580 | bool EndsWithSequential = false; | ||||
10581 | for (gep_type_iterator I = gep_type_begin(*cast<User>(PtrOp)), | ||||
10582 | E = gep_type_end(*cast<User>(PtrOp)); I != E; ++I) | ||||
10583 | EndsWithSequential = !isa<StructType>(*I); | ||||
10584 | |||||
10585 | // Can we combine the two pointer arithmetics offsets? | ||||
10586 | if (EndsWithSequential) { | ||||
10587 | // Replace: gep (gep %P, long B), long A, ... | ||||
10588 | // With: T = long A+B; gep %P, T, ... | ||||
10589 | // | ||||
10590 | Value *Sum, *SO1 = SrcGEPOperands.back(), *GO1 = GEP.getOperand(1); | ||||
10591 | if (SO1 == Constant::getNullValue(SO1->getType())) { | ||||
10592 | Sum = GO1; | ||||
10593 | } else if (GO1 == Constant::getNullValue(GO1->getType())) { | ||||
10594 | Sum = SO1; | ||||
10595 | } else { | ||||
10596 | // If they aren't the same type, convert both to an integer of the | ||||
10597 | // target's pointer size. | ||||
10598 | if (SO1->getType() != GO1->getType()) { | ||||
10599 | if (Constant *SO1C = dyn_cast<Constant>(SO1)) { | ||||
10600 | SO1 = ConstantExpr::getIntegerCast(SO1C, GO1->getType(), true); | ||||
10601 | } else if (Constant *GO1C = dyn_cast<Constant>(GO1)) { | ||||
10602 | GO1 = ConstantExpr::getIntegerCast(GO1C, SO1->getType(), true); | ||||
10603 | } else { | ||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 10604 | unsigned PS = TD->getPointerSizeInBits(); |
10605 | if (TD->getTypeSizeInBits(SO1->getType()) == PS) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10606 | // Convert GO1 to SO1's type. |
10607 | GO1 = InsertCastToIntPtrTy(GO1, SO1->getType(), &GEP, this); | ||||
10608 | |||||
Duncan Sands | f99fdc6 | 2007-11-01 20:53:16 +0000 | [diff] [blame] | 10609 | } else if (TD->getTypeSizeInBits(GO1->getType()) == PS) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10610 | // Convert SO1 to GO1's type. |
10611 | SO1 = InsertCastToIntPtrTy(SO1, GO1->getType(), &GEP, this); | ||||
10612 | } else { | ||||
10613 | const Type *PT = TD->getIntPtrType(); | ||||
10614 | SO1 = InsertCastToIntPtrTy(SO1, PT, &GEP, this); | ||||
10615 | GO1 = InsertCastToIntPtrTy(GO1, PT, &GEP, this); | ||||
10616 | } | ||||
10617 | } | ||||
10618 | } | ||||
10619 | if (isa<Constant>(SO1) && isa<Constant>(GO1)) | ||||
10620 | Sum = ConstantExpr::getAdd(cast<Constant>(SO1), cast<Constant>(GO1)); | ||||
10621 | else { | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10622 | Sum = BinaryOperator::CreateAdd(SO1, GO1, PtrOp->getName()+".sum"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10623 | InsertNewInstBefore(cast<Instruction>(Sum), GEP); |
10624 | } | ||||
10625 | } | ||||
10626 | |||||
10627 | // Recycle the GEP we already have if possible. | ||||
10628 | if (SrcGEPOperands.size() == 2) { | ||||
10629 | GEP.setOperand(0, SrcGEPOperands[0]); | ||||
10630 | GEP.setOperand(1, Sum); | ||||
10631 | return &GEP; | ||||
10632 | } else { | ||||
10633 | Indices.insert(Indices.end(), SrcGEPOperands.begin()+1, | ||||
10634 | SrcGEPOperands.end()-1); | ||||
10635 | Indices.push_back(Sum); | ||||
10636 | Indices.insert(Indices.end(), GEP.op_begin()+2, GEP.op_end()); | ||||
10637 | } | ||||
10638 | } else if (isa<Constant>(*GEP.idx_begin()) && | ||||
10639 | cast<Constant>(*GEP.idx_begin())->isNullValue() && | ||||
10640 | SrcGEPOperands.size() != 1) { | ||||
10641 | // Otherwise we can do the fold if the first index of the GEP is a zero | ||||
10642 | Indices.insert(Indices.end(), SrcGEPOperands.begin()+1, | ||||
10643 | SrcGEPOperands.end()); | ||||
10644 | Indices.insert(Indices.end(), GEP.idx_begin()+1, GEP.idx_end()); | ||||
10645 | } | ||||
10646 | |||||
10647 | if (!Indices.empty()) | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10648 | return GetElementPtrInst::Create(SrcGEPOperands[0], Indices.begin(), |
10649 | Indices.end(), GEP.getName()); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10650 | |
10651 | } else if (GlobalValue *GV = dyn_cast<GlobalValue>(PtrOp)) { | ||||
10652 | // GEP of global variable. If all of the indices for this GEP are | ||||
10653 | // constants, we can promote this to a constexpr instead of an instruction. | ||||
10654 | |||||
10655 | // Scan for nonconstants... | ||||
10656 | SmallVector<Constant*, 8> Indices; | ||||
10657 | User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end(); | ||||
10658 | for (; I != E && isa<Constant>(*I); ++I) | ||||
10659 | Indices.push_back(cast<Constant>(*I)); | ||||
10660 | |||||
10661 | if (I == E) { // If they are all constants... | ||||
10662 | Constant *CE = ConstantExpr::getGetElementPtr(GV, | ||||
10663 | &Indices[0],Indices.size()); | ||||
10664 | |||||
10665 | // Replace all uses of the GEP with the new constexpr... | ||||
10666 | return ReplaceInstUsesWith(GEP, CE); | ||||
10667 | } | ||||
10668 | } else if (Value *X = getBitCastOperand(PtrOp)) { // Is the operand a cast? | ||||
10669 | if (!isa<PointerType>(X->getType())) { | ||||
10670 | // Not interesting. Source pointer must be a cast from pointer. | ||||
10671 | } else if (HasZeroPointerIndex) { | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10672 | // transform: GEP (bitcast [10 x i8]* X to [0 x i8]*), i32 0, ... |
10673 | // into : GEP [10 x i8]* X, i32 0, ... | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10674 | // |
10675 | // This occurs when the program declares an array extern like "int X[];" | ||||
10676 | // | ||||
10677 | const PointerType *CPTy = cast<PointerType>(PtrOp->getType()); | ||||
10678 | const PointerType *XTy = cast<PointerType>(X->getType()); | ||||
10679 | if (const ArrayType *XATy = | ||||
10680 | dyn_cast<ArrayType>(XTy->getElementType())) | ||||
10681 | if (const ArrayType *CATy = | ||||
10682 | dyn_cast<ArrayType>(CPTy->getElementType())) | ||||
10683 | if (CATy->getElementType() == XATy->getElementType()) { | ||||
10684 | // At this point, we know that the cast source type is a pointer | ||||
10685 | // to an array of the same type as the destination pointer | ||||
10686 | // array. Because the array type is never stepped over (there | ||||
10687 | // is a leading zero) we can fold the cast into this GEP. | ||||
10688 | GEP.setOperand(0, X); | ||||
10689 | return &GEP; | ||||
10690 | } | ||||
10691 | } else if (GEP.getNumOperands() == 2) { | ||||
10692 | // Transform things like: | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10693 | // %t = getelementptr i32* bitcast ([2 x i32]* %str to i32*), i32 %V |
10694 | // into: %t1 = getelementptr [2 x i32]* %str, i32 0, i32 %V; bitcast | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10695 | const Type *SrcElTy = cast<PointerType>(X->getType())->getElementType(); |
10696 | const Type *ResElTy=cast<PointerType>(PtrOp->getType())->getElementType(); | ||||
10697 | if (isa<ArrayType>(SrcElTy) && | ||||
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 10698 | TD->getTypePaddedSize(cast<ArrayType>(SrcElTy)->getElementType()) == |
10699 | TD->getTypePaddedSize(ResElTy)) { | ||||
David Greene | 393be88 | 2007-09-04 15:46:09 +0000 | [diff] [blame] | 10700 | Value *Idx[2]; |
10701 | Idx[0] = Constant::getNullValue(Type::Int32Ty); | ||||
10702 | Idx[1] = GEP.getOperand(1); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10703 | Value *V = InsertNewInstBefore( |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10704 | GetElementPtrInst::Create(X, Idx, Idx + 2, GEP.getName()), GEP); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10705 | // V and GEP are both pointer types --> BitCast |
10706 | return new BitCastInst(V, GEP.getType()); | ||||
10707 | } | ||||
10708 | |||||
10709 | // Transform things like: | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10710 | // getelementptr i8* bitcast ([100 x double]* X to i8*), i32 %tmp |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10711 | // (where tmp = 8*tmp2) into: |
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10712 | // getelementptr [100 x double]* %arr, i32 0, i32 %tmp2; bitcast |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10713 | |
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10714 | if (isa<ArrayType>(SrcElTy) && ResElTy == Type::Int8Ty) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10715 | uint64_t ArrayEltSize = |
Duncan Sands | d68f13b | 2009-01-12 20:38:59 +0000 | [diff] [blame] | 10716 | TD->getTypePaddedSize(cast<ArrayType>(SrcElTy)->getElementType()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10717 | |
10718 | // Check to see if "tmp" is a scale by a multiple of ArrayEltSize. We | ||||
10719 | // allow either a mul, shift, or constant here. | ||||
10720 | Value *NewIdx = 0; | ||||
10721 | ConstantInt *Scale = 0; | ||||
10722 | if (ArrayEltSize == 1) { | ||||
10723 | NewIdx = GEP.getOperand(1); | ||||
10724 | Scale = ConstantInt::get(NewIdx->getType(), 1); | ||||
10725 | } else if (ConstantInt *CI = dyn_cast<ConstantInt>(GEP.getOperand(1))) { | ||||
10726 | NewIdx = ConstantInt::get(CI->getType(), 1); | ||||
10727 | Scale = CI; | ||||
10728 | } else if (Instruction *Inst =dyn_cast<Instruction>(GEP.getOperand(1))){ | ||||
10729 | if (Inst->getOpcode() == Instruction::Shl && | ||||
10730 | isa<ConstantInt>(Inst->getOperand(1))) { | ||||
10731 | ConstantInt *ShAmt = cast<ConstantInt>(Inst->getOperand(1)); | ||||
10732 | uint32_t ShAmtVal = ShAmt->getLimitedValue(64); | ||||
10733 | Scale = ConstantInt::get(Inst->getType(), 1ULL << ShAmtVal); | ||||
10734 | NewIdx = Inst->getOperand(0); | ||||
10735 | } else if (Inst->getOpcode() == Instruction::Mul && | ||||
10736 | isa<ConstantInt>(Inst->getOperand(1))) { | ||||
10737 | Scale = cast<ConstantInt>(Inst->getOperand(1)); | ||||
10738 | NewIdx = Inst->getOperand(0); | ||||
10739 | } | ||||
10740 | } | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10741 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10742 | // 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] | 10743 | // out, perform the transformation. Note, we don't know whether Scale is |
10744 | // signed or not. We'll use unsigned version of division/modulo | ||||
10745 | // operation after making sure Scale doesn't have the sign bit set. | ||||
10746 | if (Scale && Scale->getSExtValue() >= 0LL && | ||||
10747 | Scale->getZExtValue() % ArrayEltSize == 0) { | ||||
10748 | Scale = ConstantInt::get(Scale->getType(), | ||||
10749 | Scale->getZExtValue() / ArrayEltSize); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10750 | if (Scale->getZExtValue() != 1) { |
10751 | Constant *C = ConstantExpr::getIntegerCast(Scale, NewIdx->getType(), | ||||
Wojciech Matyjewicz | 5b5ab53 | 2007-12-12 15:21:32 +0000 | [diff] [blame] | 10752 | false /*ZExt*/); |
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 10753 | Instruction *Sc = BinaryOperator::CreateMul(NewIdx, C, "idxscale"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10754 | NewIdx = InsertNewInstBefore(Sc, GEP); |
10755 | } | ||||
10756 | |||||
10757 | // Insert the new GEP instruction. | ||||
David Greene | 393be88 | 2007-09-04 15:46:09 +0000 | [diff] [blame] | 10758 | Value *Idx[2]; |
10759 | Idx[0] = Constant::getNullValue(Type::Int32Ty); | ||||
10760 | Idx[1] = NewIdx; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10761 | Instruction *NewGEP = |
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10762 | GetElementPtrInst::Create(X, Idx, Idx + 2, GEP.getName()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10763 | NewGEP = InsertNewInstBefore(NewGEP, GEP); |
10764 | // The NewGEP must be pointer typed, so must the old one -> BitCast | ||||
10765 | return new BitCastInst(NewGEP, GEP.getType()); | ||||
10766 | } | ||||
10767 | } | ||||
10768 | } | ||||
10769 | } | ||||
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10770 | |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 10771 | /// See if we can simplify: |
10772 | /// X = bitcast A to B* | ||||
10773 | /// Y = gep X, <...constant indices...> | ||||
10774 | /// into a gep of the original struct. This is important for SROA and alias | ||||
10775 | /// analysis of unions. If "A" is also a bitcast, wait for A/X to be merged. | ||||
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10776 | if (BitCastInst *BCI = dyn_cast<BitCastInst>(PtrOp)) { |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 10777 | if (!isa<BitCastInst>(BCI->getOperand(0)) && GEP.hasAllConstantIndices()) { |
10778 | // Determine how much the GEP moves the pointer. We are guaranteed to get | ||||
10779 | // a constant back from EmitGEPOffset. | ||||
10780 | ConstantInt *OffsetV = cast<ConstantInt>(EmitGEPOffset(&GEP, GEP, *this)); | ||||
10781 | int64_t Offset = OffsetV->getSExtValue(); | ||||
10782 | |||||
10783 | // If this GEP instruction doesn't move the pointer, just replace the GEP | ||||
10784 | // with a bitcast of the real input to the dest type. | ||||
10785 | if (Offset == 0) { | ||||
10786 | // If the bitcast is of an allocation, and the allocation will be | ||||
10787 | // converted to match the type of the cast, don't touch this. | ||||
10788 | if (isa<AllocationInst>(BCI->getOperand(0))) { | ||||
10789 | // See if the bitcast simplifies, if so, don't nuke this GEP yet. | ||||
10790 | if (Instruction *I = visitBitCast(*BCI)) { | ||||
10791 | if (I != BCI) { | ||||
10792 | I->takeName(BCI); | ||||
10793 | BCI->getParent()->getInstList().insert(BCI, I); | ||||
10794 | ReplaceInstUsesWith(*BCI, I); | ||||
10795 | } | ||||
10796 | return &GEP; | ||||
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10797 | } |
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10798 | } |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 10799 | return new BitCastInst(BCI->getOperand(0), GEP.getType()); |
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10800 | } |
Chris Lattner | 94ccd5f | 2009-01-09 05:44:56 +0000 | [diff] [blame] | 10801 | |
10802 | // Otherwise, if the offset is non-zero, we need to find out if there is a | ||||
10803 | // field at Offset in 'A's type. If so, we can pull the cast through the | ||||
10804 | // GEP. | ||||
10805 | SmallVector<Value*, 8> NewIndices; | ||||
10806 | const Type *InTy = | ||||
10807 | cast<PointerType>(BCI->getOperand(0)->getType())->getElementType(); | ||||
10808 | if (FindElementAtOffset(InTy, Offset, NewIndices, TD)) { | ||||
10809 | Instruction *NGEP = | ||||
10810 | GetElementPtrInst::Create(BCI->getOperand(0), NewIndices.begin(), | ||||
10811 | NewIndices.end()); | ||||
10812 | if (NGEP->getType() == GEP.getType()) return NGEP; | ||||
10813 | InsertNewInstBefore(NGEP, GEP); | ||||
10814 | NGEP->takeName(&GEP); | ||||
10815 | return new BitCastInst(NGEP, GEP.getType()); | ||||
10816 | } | ||||
Chris Lattner | 111ea77 | 2009-01-09 04:53:57 +0000 | [diff] [blame] | 10817 | } |
10818 | } | ||||
10819 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10820 | return 0; |
10821 | } | ||||
10822 | |||||
10823 | Instruction *InstCombiner::visitAllocationInst(AllocationInst &AI) { | ||||
10824 | // 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] | 10825 | if (AI.isArrayAllocation()) { // Check C != 1 |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10826 | if (const ConstantInt *C = dyn_cast<ConstantInt>(AI.getArraySize())) { |
10827 | const Type *NewTy = | ||||
10828 | ArrayType::get(AI.getAllocatedType(), C->getZExtValue()); | ||||
10829 | AllocationInst *New = 0; | ||||
10830 | |||||
10831 | // Create and insert the replacement instruction... | ||||
10832 | if (isa<MallocInst>(AI)) | ||||
10833 | New = new MallocInst(NewTy, 0, AI.getAlignment(), AI.getName()); | ||||
10834 | else { | ||||
10835 | assert(isa<AllocaInst>(AI) && "Unknown type of allocation inst!"); | ||||
10836 | New = new AllocaInst(NewTy, 0, AI.getAlignment(), AI.getName()); | ||||
10837 | } | ||||
10838 | |||||
10839 | InsertNewInstBefore(New, AI); | ||||
10840 | |||||
10841 | // Scan to the end of the allocation instructions, to skip over a block of | ||||
10842 | // allocas if possible... | ||||
10843 | // | ||||
10844 | BasicBlock::iterator It = New; | ||||
10845 | while (isa<AllocationInst>(*It)) ++It; | ||||
10846 | |||||
10847 | // Now that I is pointing to the first non-allocation-inst in the block, | ||||
10848 | // insert our getelementptr instruction... | ||||
10849 | // | ||||
10850 | Value *NullIdx = Constant::getNullValue(Type::Int32Ty); | ||||
David Greene | 393be88 | 2007-09-04 15:46:09 +0000 | [diff] [blame] | 10851 | Value *Idx[2]; |
10852 | Idx[0] = NullIdx; | ||||
10853 | Idx[1] = NullIdx; | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 10854 | Value *V = GetElementPtrInst::Create(New, Idx, Idx + 2, |
10855 | New->getName()+".sub", It); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10856 | |
10857 | // Now make everything use the getelementptr instead of the original | ||||
10858 | // allocation. | ||||
10859 | return ReplaceInstUsesWith(AI, V); | ||||
10860 | } else if (isa<UndefValue>(AI.getArraySize())) { | ||||
10861 | return ReplaceInstUsesWith(AI, Constant::getNullValue(AI.getType())); | ||||
10862 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 10863 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10864 | |
Dan Gohman | 28e78f0 | 2009-01-13 20:18:38 +0000 | [diff] [blame] | 10865 | if (isa<AllocaInst>(AI) && AI.getAllocatedType()->isSized()) { |
10866 | // If alloca'ing a zero byte object, replace the alloca with a null pointer. | ||||
10867 | // Note that we only do this for alloca's, because malloc should allocate and | ||||
10868 | // return a unique pointer, even for a zero byte allocation. | ||||
10869 | if (TD->getTypePaddedSize(AI.getAllocatedType()) == 0) | ||||
10870 | return ReplaceInstUsesWith(AI, Constant::getNullValue(AI.getType())); | ||||
10871 | |||||
10872 | // If the alignment is 0 (unspecified), assign it the preferred alignment. | ||||
10873 | if (AI.getAlignment() == 0) | ||||
10874 | AI.setAlignment(TD->getPrefTypeAlignment(AI.getAllocatedType())); | ||||
10875 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10876 | |
10877 | return 0; | ||||
10878 | } | ||||
10879 | |||||
10880 | Instruction *InstCombiner::visitFreeInst(FreeInst &FI) { | ||||
10881 | Value *Op = FI.getOperand(0); | ||||
10882 | |||||
10883 | // free undef -> unreachable. | ||||
10884 | if (isa<UndefValue>(Op)) { | ||||
10885 | // Insert a new store to null because we cannot modify the CFG here. | ||||
10886 | new StoreInst(ConstantInt::getTrue(), | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 10887 | UndefValue::get(PointerType::getUnqual(Type::Int1Ty)), &FI); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10888 | return EraseInstFromFunction(FI); |
10889 | } | ||||
10890 | |||||
10891 | // If we have 'free null' delete the instruction. This can happen in stl code | ||||
10892 | // when lots of inlining happens. | ||||
10893 | if (isa<ConstantPointerNull>(Op)) | ||||
10894 | return EraseInstFromFunction(FI); | ||||
10895 | |||||
10896 | // Change free <ty>* (cast <ty2>* X to <ty>*) into free <ty2>* X | ||||
10897 | if (BitCastInst *CI = dyn_cast<BitCastInst>(Op)) { | ||||
10898 | FI.setOperand(0, CI->getOperand(0)); | ||||
10899 | return &FI; | ||||
10900 | } | ||||
10901 | |||||
10902 | // Change free (gep X, 0,0,0,0) into free(X) | ||||
10903 | if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(Op)) { | ||||
10904 | if (GEPI->hasAllZeroIndices()) { | ||||
10905 | AddToWorkList(GEPI); | ||||
10906 | FI.setOperand(0, GEPI->getOperand(0)); | ||||
10907 | return &FI; | ||||
10908 | } | ||||
10909 | } | ||||
10910 | |||||
10911 | // Change free(malloc) into nothing, if the malloc has a single use. | ||||
10912 | if (MallocInst *MI = dyn_cast<MallocInst>(Op)) | ||||
10913 | if (MI->hasOneUse()) { | ||||
10914 | EraseInstFromFunction(FI); | ||||
10915 | return EraseInstFromFunction(*MI); | ||||
10916 | } | ||||
10917 | |||||
10918 | return 0; | ||||
10919 | } | ||||
10920 | |||||
10921 | |||||
10922 | /// InstCombineLoadCast - Fold 'load (cast P)' -> cast (load P)' when possible. | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 10923 | static Instruction *InstCombineLoadCast(InstCombiner &IC, LoadInst &LI, |
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 10924 | const TargetData *TD) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10925 | User *CI = cast<User>(LI.getOperand(0)); |
10926 | Value *CastOp = CI->getOperand(0); | ||||
10927 | |||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 10928 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(CI)) { |
10929 | // Instead of loading constant c string, use corresponding integer value | ||||
10930 | // directly if string length is small enough. | ||||
Evan Cheng | 833501d | 2008-06-30 07:31:25 +0000 | [diff] [blame] | 10931 | std::string Str; |
10932 | if (GetConstantStringInfo(CE->getOperand(0), Str) && !Str.empty()) { | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 10933 | unsigned len = Str.length(); |
10934 | const Type *Ty = cast<PointerType>(CE->getType())->getElementType(); | ||||
10935 | unsigned numBits = Ty->getPrimitiveSizeInBits(); | ||||
10936 | // Replace LI with immediate integer store. | ||||
10937 | if ((numBits >> 3) == len + 1) { | ||||
Bill Wendling | 44a36ea | 2008-02-26 10:53:30 +0000 | [diff] [blame] | 10938 | APInt StrVal(numBits, 0); |
10939 | APInt SingleChar(numBits, 0); | ||||
10940 | if (TD->isLittleEndian()) { | ||||
10941 | for (signed i = len-1; i >= 0; i--) { | ||||
10942 | SingleChar = (uint64_t) Str[i]; | ||||
10943 | StrVal = (StrVal << 8) | SingleChar; | ||||
10944 | } | ||||
10945 | } else { | ||||
10946 | for (unsigned i = 0; i < len; i++) { | ||||
10947 | SingleChar = (uint64_t) Str[i]; | ||||
10948 | StrVal = (StrVal << 8) | SingleChar; | ||||
10949 | } | ||||
10950 | // Append NULL at the end. | ||||
10951 | SingleChar = 0; | ||||
10952 | StrVal = (StrVal << 8) | SingleChar; | ||||
10953 | } | ||||
10954 | Value *NL = ConstantInt::get(StrVal); | ||||
10955 | return IC.ReplaceInstUsesWith(LI, NL); | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 10956 | } |
10957 | } | ||||
10958 | } | ||||
10959 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 10960 | const Type *DestPTy = cast<PointerType>(CI->getType())->getElementType(); |
10961 | if (const PointerType *SrcTy = dyn_cast<PointerType>(CastOp->getType())) { | ||||
10962 | const Type *SrcPTy = SrcTy->getElementType(); | ||||
10963 | |||||
10964 | if (DestPTy->isInteger() || isa<PointerType>(DestPTy) || | ||||
10965 | isa<VectorType>(DestPTy)) { | ||||
10966 | // If the source is an array, the code below will not succeed. Check to | ||||
10967 | // see if a trivial 'gep P, 0, 0' will help matters. Only do this for | ||||
10968 | // constants. | ||||
10969 | if (const ArrayType *ASrcTy = dyn_cast<ArrayType>(SrcPTy)) | ||||
10970 | if (Constant *CSrc = dyn_cast<Constant>(CastOp)) | ||||
10971 | if (ASrcTy->getNumElements() != 0) { | ||||
10972 | Value *Idxs[2]; | ||||
10973 | Idxs[0] = Idxs[1] = Constant::getNullValue(Type::Int32Ty); | ||||
10974 | CastOp = ConstantExpr::getGetElementPtr(CSrc, Idxs, 2); | ||||
10975 | SrcTy = cast<PointerType>(CastOp->getType()); | ||||
10976 | SrcPTy = SrcTy->getElementType(); | ||||
10977 | } | ||||
10978 | |||||
10979 | if ((SrcPTy->isInteger() || isa<PointerType>(SrcPTy) || | ||||
10980 | isa<VectorType>(SrcPTy)) && | ||||
10981 | // Do not allow turning this into a load of an integer, which is then | ||||
10982 | // casted to a pointer, this pessimizes pointer analysis a lot. | ||||
10983 | (isa<PointerType>(SrcPTy) == isa<PointerType>(LI.getType())) && | ||||
10984 | IC.getTargetData().getTypeSizeInBits(SrcPTy) == | ||||
10985 | IC.getTargetData().getTypeSizeInBits(DestPTy)) { | ||||
10986 | |||||
10987 | // Okay, we are casting from one integer or pointer type to another of | ||||
10988 | // the same size. Instead of casting the pointer before the load, cast | ||||
10989 | // the result of the loaded value. | ||||
10990 | Value *NewLoad = IC.InsertNewInstBefore(new LoadInst(CastOp, | ||||
10991 | CI->getName(), | ||||
10992 | LI.isVolatile()),LI); | ||||
10993 | // Now cast the result of the load. | ||||
10994 | return new BitCastInst(NewLoad, LI.getType()); | ||||
10995 | } | ||||
10996 | } | ||||
10997 | } | ||||
10998 | return 0; | ||||
10999 | } | ||||
11000 | |||||
11001 | /// isSafeToLoadUnconditionally - Return true if we know that executing a load | ||||
11002 | /// from this value cannot trap. If it is not obviously safe to load from the | ||||
11003 | /// specified pointer, we do a quick local scan of the basic block containing | ||||
11004 | /// ScanFrom, to determine if the address is already accessed. | ||||
11005 | static bool isSafeToLoadUnconditionally(Value *V, Instruction *ScanFrom) { | ||||
Duncan Sands | 9b27dbe | 2007-09-19 10:10:31 +0000 | [diff] [blame] | 11006 | // If it is an alloca it is always safe to load from. |
11007 | if (isa<AllocaInst>(V)) return true; | ||||
11008 | |||||
Duncan Sands | e40a94a | 2007-09-19 10:25:38 +0000 | [diff] [blame] | 11009 | // 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] | 11010 | if (const GlobalValue *GV = dyn_cast<GlobalVariable>(V)) |
Duncan Sands | e40a94a | 2007-09-19 10:25:38 +0000 | [diff] [blame] | 11011 | // Don't try to evaluate aliases. External weak GV can be null. |
Duncan Sands | 9b27dbe | 2007-09-19 10:10:31 +0000 | [diff] [blame] | 11012 | return !isa<GlobalAlias>(GV) && !GV->hasExternalWeakLinkage(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11013 | |
11014 | // Otherwise, be a little bit agressive by scanning the local block where we | ||||
11015 | // want to check to see if the pointer is already being loaded or stored | ||||
11016 | // from/to. If so, the previous load or store would have already trapped, | ||||
11017 | // so there is no harm doing an extra load (also, CSE will later eliminate | ||||
11018 | // the load entirely). | ||||
11019 | BasicBlock::iterator BBI = ScanFrom, E = ScanFrom->getParent()->begin(); | ||||
11020 | |||||
11021 | while (BBI != E) { | ||||
11022 | --BBI; | ||||
11023 | |||||
Chris Lattner | 476983a | 2008-06-20 05:12:56 +0000 | [diff] [blame] | 11024 | // If we see a free or a call (which might do a free) the pointer could be |
11025 | // marked invalid. | ||||
11026 | if (isa<FreeInst>(BBI) || isa<CallInst>(BBI)) | ||||
11027 | return false; | ||||
11028 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11029 | if (LoadInst *LI = dyn_cast<LoadInst>(BBI)) { |
11030 | if (LI->getOperand(0) == V) return true; | ||||
Chris Lattner | 476983a | 2008-06-20 05:12:56 +0000 | [diff] [blame] | 11031 | } else if (StoreInst *SI = dyn_cast<StoreInst>(BBI)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11032 | if (SI->getOperand(1) == V) return true; |
Chris Lattner | 476983a | 2008-06-20 05:12:56 +0000 | [diff] [blame] | 11033 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11034 | |
11035 | } | ||||
11036 | return false; | ||||
11037 | } | ||||
11038 | |||||
11039 | Instruction *InstCombiner::visitLoadInst(LoadInst &LI) { | ||||
11040 | Value *Op = LI.getOperand(0); | ||||
11041 | |||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 11042 | // Attempt to improve the alignment. |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 11043 | unsigned KnownAlign = GetOrEnforceKnownAlignment(Op); |
11044 | if (KnownAlign > | ||||
11045 | (LI.getAlignment() == 0 ? TD->getABITypeAlignment(LI.getType()) : | ||||
11046 | LI.getAlignment())) | ||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 11047 | LI.setAlignment(KnownAlign); |
11048 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11049 | // load (cast X) --> cast (load X) iff safe |
11050 | if (isa<CastInst>(Op)) | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 11051 | if (Instruction *Res = InstCombineLoadCast(*this, LI, TD)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11052 | return Res; |
11053 | |||||
11054 | // None of the following transforms are legal for volatile loads. | ||||
11055 | if (LI.isVolatile()) return 0; | ||||
11056 | |||||
Dan Gohman | 0ff5a1f | 2008-10-15 23:19:35 +0000 | [diff] [blame] | 11057 | // Do really simple store-to-load forwarding and load CSE, to catch cases |
11058 | // where there are several consequtive memory accesses to the same location, | ||||
11059 | // separated by a few arithmetic operations. | ||||
11060 | BasicBlock::iterator BBI = &LI; | ||||
Chris Lattner | 6fd8c80 | 2008-11-27 08:56:30 +0000 | [diff] [blame] | 11061 | if (Value *AvailableVal = FindAvailableLoadedValue(Op, LI.getParent(), BBI,6)) |
11062 | return ReplaceInstUsesWith(LI, AvailableVal); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11063 | |
Christopher Lamb | 2c17539 | 2007-12-29 07:56:53 +0000 | [diff] [blame] | 11064 | if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(Op)) { |
11065 | const Value *GEPI0 = GEPI->getOperand(0); | ||||
11066 | // TODO: Consider a target hook for valid address spaces for this xform. | ||||
11067 | if (isa<ConstantPointerNull>(GEPI0) && | ||||
11068 | cast<PointerType>(GEPI0->getType())->getAddressSpace() == 0) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11069 | // Insert a new store to null instruction before the load to indicate |
11070 | // that this code is not reachable. We do this instead of inserting | ||||
11071 | // an unreachable instruction directly because we cannot modify the | ||||
11072 | // CFG. | ||||
11073 | new StoreInst(UndefValue::get(LI.getType()), | ||||
11074 | Constant::getNullValue(Op->getType()), &LI); | ||||
11075 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
11076 | } | ||||
Christopher Lamb | 2c17539 | 2007-12-29 07:56:53 +0000 | [diff] [blame] | 11077 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11078 | |
11079 | if (Constant *C = dyn_cast<Constant>(Op)) { | ||||
11080 | // load null/undef -> undef | ||||
Christopher Lamb | 2c17539 | 2007-12-29 07:56:53 +0000 | [diff] [blame] | 11081 | // TODO: Consider a target hook for valid address spaces for this xform. |
11082 | if (isa<UndefValue>(C) || (C->isNullValue() && | ||||
11083 | cast<PointerType>(Op->getType())->getAddressSpace() == 0)) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11084 | // Insert a new store to null instruction before the load to indicate that |
11085 | // this code is not reachable. We do this instead of inserting an | ||||
11086 | // unreachable instruction directly because we cannot modify the CFG. | ||||
11087 | new StoreInst(UndefValue::get(LI.getType()), | ||||
11088 | Constant::getNullValue(Op->getType()), &LI); | ||||
11089 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
11090 | } | ||||
11091 | |||||
11092 | // Instcombine load (constant global) into the value loaded. | ||||
11093 | if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op)) | ||||
11094 | if (GV->isConstant() && !GV->isDeclaration()) | ||||
11095 | return ReplaceInstUsesWith(LI, GV->getInitializer()); | ||||
11096 | |||||
11097 | // Instcombine load (constantexpr_GEP global, 0, ...) into the value loaded. | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 11098 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Op)) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11099 | if (CE->getOpcode() == Instruction::GetElementPtr) { |
11100 | if (GlobalVariable *GV = dyn_cast<GlobalVariable>(CE->getOperand(0))) | ||||
11101 | if (GV->isConstant() && !GV->isDeclaration()) | ||||
11102 | if (Constant *V = | ||||
11103 | ConstantFoldLoadThroughGEPConstantExpr(GV->getInitializer(), CE)) | ||||
11104 | return ReplaceInstUsesWith(LI, V); | ||||
11105 | if (CE->getOperand(0)->isNullValue()) { | ||||
11106 | // Insert a new store to null instruction before the load to indicate | ||||
11107 | // that this code is not reachable. We do this instead of inserting | ||||
11108 | // an unreachable instruction directly because we cannot modify the | ||||
11109 | // CFG. | ||||
11110 | new StoreInst(UndefValue::get(LI.getType()), | ||||
11111 | Constant::getNullValue(Op->getType()), &LI); | ||||
11112 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
11113 | } | ||||
11114 | |||||
11115 | } else if (CE->isCast()) { | ||||
Devang Patel | a0f8ea8 | 2007-10-18 19:52:32 +0000 | [diff] [blame] | 11116 | if (Instruction *Res = InstCombineLoadCast(*this, LI, TD)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11117 | return Res; |
11118 | } | ||||
Anton Korobeynikov | 8522e1c | 2008-02-20 11:26:25 +0000 | [diff] [blame] | 11119 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11120 | } |
Chris Lattner | 0270a11 | 2007-08-11 18:48:48 +0000 | [diff] [blame] | 11121 | |
11122 | // If this load comes from anywhere in a constant global, and if the global | ||||
11123 | // is all undef or zero, we know what it loads. | ||||
Duncan Sands | 52fb873 | 2008-10-01 15:25:41 +0000 | [diff] [blame] | 11124 | if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op->getUnderlyingObject())){ |
Chris Lattner | 0270a11 | 2007-08-11 18:48:48 +0000 | [diff] [blame] | 11125 | if (GV->isConstant() && GV->hasInitializer()) { |
11126 | if (GV->getInitializer()->isNullValue()) | ||||
11127 | return ReplaceInstUsesWith(LI, Constant::getNullValue(LI.getType())); | ||||
11128 | else if (isa<UndefValue>(GV->getInitializer())) | ||||
11129 | return ReplaceInstUsesWith(LI, UndefValue::get(LI.getType())); | ||||
11130 | } | ||||
11131 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11132 | |
11133 | if (Op->hasOneUse()) { | ||||
11134 | // Change select and PHI nodes to select values instead of addresses: this | ||||
11135 | // helps alias analysis out a lot, allows many others simplifications, and | ||||
11136 | // exposes redundancy in the code. | ||||
11137 | // | ||||
11138 | // Note that we cannot do the transformation unless we know that the | ||||
11139 | // introduced loads cannot trap! Something like this is valid as long as | ||||
11140 | // the condition is always false: load (select bool %C, int* null, int* %G), | ||||
11141 | // but it would not be valid if we transformed it to load from null | ||||
11142 | // unconditionally. | ||||
11143 | // | ||||
11144 | if (SelectInst *SI = dyn_cast<SelectInst>(Op)) { | ||||
11145 | // load (select (Cond, &V1, &V2)) --> select(Cond, load &V1, load &V2). | ||||
11146 | if (isSafeToLoadUnconditionally(SI->getOperand(1), SI) && | ||||
11147 | isSafeToLoadUnconditionally(SI->getOperand(2), SI)) { | ||||
11148 | Value *V1 = InsertNewInstBefore(new LoadInst(SI->getOperand(1), | ||||
11149 | SI->getOperand(1)->getName()+".val"), LI); | ||||
11150 | Value *V2 = InsertNewInstBefore(new LoadInst(SI->getOperand(2), | ||||
11151 | SI->getOperand(2)->getName()+".val"), LI); | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 11152 | return SelectInst::Create(SI->getCondition(), V1, V2); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11153 | } |
11154 | |||||
11155 | // load (select (cond, null, P)) -> load P | ||||
11156 | if (Constant *C = dyn_cast<Constant>(SI->getOperand(1))) | ||||
11157 | if (C->isNullValue()) { | ||||
11158 | LI.setOperand(0, SI->getOperand(2)); | ||||
11159 | return &LI; | ||||
11160 | } | ||||
11161 | |||||
11162 | // load (select (cond, P, null)) -> load P | ||||
11163 | if (Constant *C = dyn_cast<Constant>(SI->getOperand(2))) | ||||
11164 | if (C->isNullValue()) { | ||||
11165 | LI.setOperand(0, SI->getOperand(1)); | ||||
11166 | return &LI; | ||||
11167 | } | ||||
11168 | } | ||||
11169 | } | ||||
11170 | return 0; | ||||
11171 | } | ||||
11172 | |||||
11173 | /// InstCombineStoreToCast - Fold store V, (cast P) -> store (cast V), P | ||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11174 | /// when possible. This makes it generally easy to do alias analysis and/or |
11175 | /// SROA/mem2reg of the memory object. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11176 | static Instruction *InstCombineStoreToCast(InstCombiner &IC, StoreInst &SI) { |
11177 | User *CI = cast<User>(SI.getOperand(1)); | ||||
11178 | Value *CastOp = CI->getOperand(0); | ||||
11179 | |||||
11180 | const Type *DestPTy = cast<PointerType>(CI->getType())->getElementType(); | ||||
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11181 | const PointerType *SrcTy = dyn_cast<PointerType>(CastOp->getType()); |
11182 | if (SrcTy == 0) return 0; | ||||
11183 | |||||
11184 | const Type *SrcPTy = SrcTy->getElementType(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11185 | |
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11186 | if (!DestPTy->isInteger() && !isa<PointerType>(DestPTy)) |
11187 | return 0; | ||||
11188 | |||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11189 | /// NewGEPIndices - If SrcPTy is an aggregate type, we can emit a "noop gep" |
11190 | /// to its first element. This allows us to handle things like: | ||||
11191 | /// store i32 xxx, (bitcast {foo*, float}* %P to i32*) | ||||
11192 | /// on 32-bit hosts. | ||||
11193 | SmallVector<Value*, 4> NewGEPIndices; | ||||
11194 | |||||
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11195 | // If the source is an array, the code below will not succeed. Check to |
11196 | // see if a trivial 'gep P, 0, 0' will help matters. Only do this for | ||||
11197 | // constants. | ||||
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11198 | if (isa<ArrayType>(SrcPTy) || isa<StructType>(SrcPTy)) { |
11199 | // Index through pointer. | ||||
11200 | Constant *Zero = Constant::getNullValue(Type::Int32Ty); | ||||
11201 | NewGEPIndices.push_back(Zero); | ||||
11202 | |||||
11203 | while (1) { | ||||
11204 | if (const StructType *STy = dyn_cast<StructType>(SrcPTy)) { | ||||
edwin | 7dc0aa3 | 2009-01-24 17:16:04 +0000 | [diff] [blame] | 11205 | if (!STy->getNumElements()) /* Struct can be empty {} */ |
edwin | 07d74e7 | 2009-01-24 11:30:49 +0000 | [diff] [blame] | 11206 | break; |
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11207 | NewGEPIndices.push_back(Zero); |
11208 | SrcPTy = STy->getElementType(0); | ||||
11209 | } else if (const ArrayType *ATy = dyn_cast<ArrayType>(SrcPTy)) { | ||||
11210 | NewGEPIndices.push_back(Zero); | ||||
11211 | SrcPTy = ATy->getElementType(); | ||||
11212 | } else { | ||||
11213 | break; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11214 | } |
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11215 | } |
11216 | |||||
11217 | SrcTy = PointerType::get(SrcPTy, SrcTy->getAddressSpace()); | ||||
11218 | } | ||||
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11219 | |
11220 | if (!SrcPTy->isInteger() && !isa<PointerType>(SrcPTy)) | ||||
11221 | return 0; | ||||
11222 | |||||
Chris Lattner | c73a0d1 | 2009-01-16 20:12:52 +0000 | [diff] [blame] | 11223 | // If the pointers point into different address spaces or if they point to |
11224 | // values with different sizes, we can't do the transformation. | ||||
11225 | if (SrcTy->getAddressSpace() != | ||||
11226 | cast<PointerType>(CI->getType())->getAddressSpace() || | ||||
11227 | IC.getTargetData().getTypeSizeInBits(SrcPTy) != | ||||
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11228 | IC.getTargetData().getTypeSizeInBits(DestPTy)) |
11229 | return 0; | ||||
11230 | |||||
11231 | // Okay, we are casting from one integer or pointer type to another of | ||||
11232 | // the same size. Instead of casting the pointer before | ||||
11233 | // the store, cast the value to be stored. | ||||
11234 | Value *NewCast; | ||||
11235 | Value *SIOp0 = SI.getOperand(0); | ||||
11236 | Instruction::CastOps opcode = Instruction::BitCast; | ||||
11237 | const Type* CastSrcTy = SIOp0->getType(); | ||||
11238 | const Type* CastDstTy = SrcPTy; | ||||
11239 | if (isa<PointerType>(CastDstTy)) { | ||||
11240 | if (CastSrcTy->isInteger()) | ||||
11241 | opcode = Instruction::IntToPtr; | ||||
11242 | } else if (isa<IntegerType>(CastDstTy)) { | ||||
11243 | if (isa<PointerType>(SIOp0->getType())) | ||||
11244 | opcode = Instruction::PtrToInt; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11245 | } |
Chris Lattner | 54dddc7 | 2009-01-24 01:00:13 +0000 | [diff] [blame] | 11246 | |
11247 | // SIOp0 is a pointer to aggregate and this is a store to the first field, | ||||
11248 | // emit a GEP to index into its first field. | ||||
11249 | if (!NewGEPIndices.empty()) { | ||||
11250 | if (Constant *C = dyn_cast<Constant>(CastOp)) | ||||
11251 | CastOp = ConstantExpr::getGetElementPtr(C, &NewGEPIndices[0], | ||||
11252 | NewGEPIndices.size()); | ||||
11253 | else | ||||
11254 | CastOp = IC.InsertNewInstBefore( | ||||
11255 | GetElementPtrInst::Create(CastOp, NewGEPIndices.begin(), | ||||
11256 | NewGEPIndices.end()), SI); | ||||
11257 | } | ||||
11258 | |||||
Chris Lattner | a032c0e | 2009-01-16 20:08:59 +0000 | [diff] [blame] | 11259 | if (Constant *C = dyn_cast<Constant>(SIOp0)) |
11260 | NewCast = ConstantExpr::getCast(opcode, C, CastDstTy); | ||||
11261 | else | ||||
11262 | NewCast = IC.InsertNewInstBefore( | ||||
11263 | CastInst::Create(opcode, SIOp0, CastDstTy, SIOp0->getName()+".c"), | ||||
11264 | SI); | ||||
11265 | return new StoreInst(NewCast, CastOp); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11266 | } |
11267 | |||||
Chris Lattner | 6fd8c80 | 2008-11-27 08:56:30 +0000 | [diff] [blame] | 11268 | /// equivalentAddressValues - Test if A and B will obviously have the same |
11269 | /// value. This includes recognizing that %t0 and %t1 will have the same | ||||
11270 | /// value in code like this: | ||||
11271 | /// %t0 = getelementptr @a, 0, 3 | ||||
11272 | /// store i32 0, i32* %t0 | ||||
11273 | /// %t1 = getelementptr @a, 0, 3 | ||||
11274 | /// %t2 = load i32* %t1 | ||||
11275 | /// | ||||
11276 | static bool equivalentAddressValues(Value *A, Value *B) { | ||||
11277 | // Test if the values are trivially equivalent. | ||||
11278 | if (A == B) return true; | ||||
11279 | |||||
11280 | // Test if the values come form identical arithmetic instructions. | ||||
11281 | if (isa<BinaryOperator>(A) || | ||||
11282 | isa<CastInst>(A) || | ||||
11283 | isa<PHINode>(A) || | ||||
11284 | isa<GetElementPtrInst>(A)) | ||||
11285 | if (Instruction *BI = dyn_cast<Instruction>(B)) | ||||
11286 | if (cast<Instruction>(A)->isIdenticalTo(BI)) | ||||
11287 | return true; | ||||
11288 | |||||
11289 | // Otherwise they may not be equivalent. | ||||
11290 | return false; | ||||
11291 | } | ||||
11292 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11293 | Instruction *InstCombiner::visitStoreInst(StoreInst &SI) { |
11294 | Value *Val = SI.getOperand(0); | ||||
11295 | Value *Ptr = SI.getOperand(1); | ||||
11296 | |||||
11297 | if (isa<UndefValue>(Ptr)) { // store X, undef -> noop (even if volatile) | ||||
11298 | EraseInstFromFunction(SI); | ||||
11299 | ++NumCombined; | ||||
11300 | return 0; | ||||
11301 | } | ||||
11302 | |||||
11303 | // If the RHS is an alloca with a single use, zapify the store, making the | ||||
11304 | // alloca dead. | ||||
Chris Lattner | a02bacc | 2008-04-29 04:58:38 +0000 | [diff] [blame] | 11305 | if (Ptr->hasOneUse() && !SI.isVolatile()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11306 | if (isa<AllocaInst>(Ptr)) { |
11307 | EraseInstFromFunction(SI); | ||||
11308 | ++NumCombined; | ||||
11309 | return 0; | ||||
11310 | } | ||||
11311 | |||||
11312 | if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Ptr)) | ||||
11313 | if (isa<AllocaInst>(GEP->getOperand(0)) && | ||||
11314 | GEP->getOperand(0)->hasOneUse()) { | ||||
11315 | EraseInstFromFunction(SI); | ||||
11316 | ++NumCombined; | ||||
11317 | return 0; | ||||
11318 | } | ||||
11319 | } | ||||
11320 | |||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 11321 | // Attempt to improve the alignment. |
Dan Gohman | 2d648bb | 2008-04-10 18:43:06 +0000 | [diff] [blame] | 11322 | unsigned KnownAlign = GetOrEnforceKnownAlignment(Ptr); |
11323 | if (KnownAlign > | ||||
11324 | (SI.getAlignment() == 0 ? TD->getABITypeAlignment(Val->getType()) : | ||||
11325 | SI.getAlignment())) | ||||
Dan Gohman | 5c4d0e1 | 2007-07-20 16:34:21 +0000 | [diff] [blame] | 11326 | SI.setAlignment(KnownAlign); |
11327 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11328 | // Do really simple DSE, to catch cases where there are several consequtive |
11329 | // stores to the same location, separated by a few arithmetic operations. This | ||||
11330 | // situation often occurs with bitfield accesses. | ||||
11331 | BasicBlock::iterator BBI = &SI; | ||||
11332 | for (unsigned ScanInsts = 6; BBI != SI.getParent()->begin() && ScanInsts; | ||||
11333 | --ScanInsts) { | ||||
11334 | --BBI; | ||||
11335 | |||||
11336 | if (StoreInst *PrevSI = dyn_cast<StoreInst>(BBI)) { | ||||
11337 | // Prev store isn't volatile, and stores to the same location? | ||||
Chris Lattner | 6fd8c80 | 2008-11-27 08:56:30 +0000 | [diff] [blame] | 11338 | if (!PrevSI->isVolatile() &&equivalentAddressValues(PrevSI->getOperand(1), |
11339 | SI.getOperand(1))) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11340 | ++NumDeadStore; |
11341 | ++BBI; | ||||
11342 | EraseInstFromFunction(*PrevSI); | ||||
11343 | continue; | ||||
11344 | } | ||||
11345 | break; | ||||
11346 | } | ||||
11347 | |||||
11348 | // If this is a load, we have to stop. However, if the loaded value is from | ||||
11349 | // the pointer we're loading and is producing the pointer we're storing, | ||||
11350 | // then *this* store is dead (X = load P; store X -> P). | ||||
11351 | if (LoadInst *LI = dyn_cast<LoadInst>(BBI)) { | ||||
Dan Gohman | 0ff5a1f | 2008-10-15 23:19:35 +0000 | [diff] [blame] | 11352 | if (LI == Val && equivalentAddressValues(LI->getOperand(0), Ptr) && |
11353 | !SI.isVolatile()) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11354 | EraseInstFromFunction(SI); |
11355 | ++NumCombined; | ||||
11356 | return 0; | ||||
11357 | } | ||||
11358 | // Otherwise, this is a load from some other location. Stores before it | ||||
11359 | // may not be dead. | ||||
11360 | break; | ||||
11361 | } | ||||
11362 | |||||
11363 | // Don't skip over loads or things that can modify memory. | ||||
Chris Lattner | 8450428 | 2008-05-08 17:20:30 +0000 | [diff] [blame] | 11364 | if (BBI->mayWriteToMemory() || BBI->mayReadFromMemory()) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11365 | break; |
11366 | } | ||||
11367 | |||||
11368 | |||||
11369 | if (SI.isVolatile()) return 0; // Don't hack volatile stores. | ||||
11370 | |||||
11371 | // store X, null -> turns into 'unreachable' in SimplifyCFG | ||||
11372 | if (isa<ConstantPointerNull>(Ptr)) { | ||||
11373 | if (!isa<UndefValue>(Val)) { | ||||
11374 | SI.setOperand(0, UndefValue::get(Val->getType())); | ||||
11375 | if (Instruction *U = dyn_cast<Instruction>(Val)) | ||||
11376 | AddToWorkList(U); // Dropped a use. | ||||
11377 | ++NumCombined; | ||||
11378 | } | ||||
11379 | return 0; // Do not modify these! | ||||
11380 | } | ||||
11381 | |||||
11382 | // store undef, Ptr -> noop | ||||
11383 | if (isa<UndefValue>(Val)) { | ||||
11384 | EraseInstFromFunction(SI); | ||||
11385 | ++NumCombined; | ||||
11386 | return 0; | ||||
11387 | } | ||||
11388 | |||||
11389 | // If the pointer destination is a cast, see if we can fold the cast into the | ||||
11390 | // source instead. | ||||
11391 | if (isa<CastInst>(Ptr)) | ||||
11392 | if (Instruction *Res = InstCombineStoreToCast(*this, SI)) | ||||
11393 | return Res; | ||||
11394 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr)) | ||||
11395 | if (CE->isCast()) | ||||
11396 | if (Instruction *Res = InstCombineStoreToCast(*this, SI)) | ||||
11397 | return Res; | ||||
11398 | |||||
11399 | |||||
11400 | // If this store is the last instruction in the basic block, and if the block | ||||
11401 | // ends with an unconditional branch, try to move it to the successor block. | ||||
11402 | BBI = &SI; ++BBI; | ||||
11403 | if (BranchInst *BI = dyn_cast<BranchInst>(BBI)) | ||||
11404 | if (BI->isUnconditional()) | ||||
11405 | if (SimplifyStoreAtEndOfBlock(SI)) | ||||
11406 | return 0; // xform done! | ||||
11407 | |||||
11408 | return 0; | ||||
11409 | } | ||||
11410 | |||||
11411 | /// SimplifyStoreAtEndOfBlock - Turn things like: | ||||
11412 | /// if () { *P = v1; } else { *P = v2 } | ||||
11413 | /// into a phi node with a store in the successor. | ||||
11414 | /// | ||||
11415 | /// Simplify things like: | ||||
11416 | /// *P = v1; if () { *P = v2; } | ||||
11417 | /// into a phi node with a store in the successor. | ||||
11418 | /// | ||||
11419 | bool InstCombiner::SimplifyStoreAtEndOfBlock(StoreInst &SI) { | ||||
11420 | BasicBlock *StoreBB = SI.getParent(); | ||||
11421 | |||||
11422 | // Check to see if the successor block has exactly two incoming edges. If | ||||
11423 | // so, see if the other predecessor contains a store to the same location. | ||||
11424 | // if so, insert a PHI node (if needed) and move the stores down. | ||||
11425 | BasicBlock *DestBB = StoreBB->getTerminator()->getSuccessor(0); | ||||
11426 | |||||
11427 | // Determine whether Dest has exactly two predecessors and, if so, compute | ||||
11428 | // the other predecessor. | ||||
11429 | pred_iterator PI = pred_begin(DestBB); | ||||
11430 | BasicBlock *OtherBB = 0; | ||||
11431 | if (*PI != StoreBB) | ||||
11432 | OtherBB = *PI; | ||||
11433 | ++PI; | ||||
11434 | if (PI == pred_end(DestBB)) | ||||
11435 | return false; | ||||
11436 | |||||
11437 | if (*PI != StoreBB) { | ||||
11438 | if (OtherBB) | ||||
11439 | return false; | ||||
11440 | OtherBB = *PI; | ||||
11441 | } | ||||
11442 | if (++PI != pred_end(DestBB)) | ||||
11443 | return false; | ||||
Eli Friedman | ab39f9a | 2008-06-13 21:17:49 +0000 | [diff] [blame] | 11444 | |
11445 | // Bail out if all the relevant blocks aren't distinct (this can happen, | ||||
11446 | // for example, if SI is in an infinite loop) | ||||
11447 | if (StoreBB == DestBB || OtherBB == DestBB) | ||||
11448 | return false; | ||||
11449 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11450 | // Verify that the other block ends in a branch and is not otherwise empty. |
11451 | BasicBlock::iterator BBI = OtherBB->getTerminator(); | ||||
11452 | BranchInst *OtherBr = dyn_cast<BranchInst>(BBI); | ||||
11453 | if (!OtherBr || BBI == OtherBB->begin()) | ||||
11454 | return false; | ||||
11455 | |||||
11456 | // If the other block ends in an unconditional branch, check for the 'if then | ||||
11457 | // else' case. there is an instruction before the branch. | ||||
11458 | StoreInst *OtherStore = 0; | ||||
11459 | if (OtherBr->isUnconditional()) { | ||||
11460 | // If this isn't a store, or isn't a store to the same location, bail out. | ||||
11461 | --BBI; | ||||
11462 | OtherStore = dyn_cast<StoreInst>(BBI); | ||||
11463 | if (!OtherStore || OtherStore->getOperand(1) != SI.getOperand(1)) | ||||
11464 | return false; | ||||
11465 | } else { | ||||
11466 | // Otherwise, the other block ended with a conditional branch. If one of the | ||||
11467 | // destinations is StoreBB, then we have the if/then case. | ||||
11468 | if (OtherBr->getSuccessor(0) != StoreBB && | ||||
11469 | OtherBr->getSuccessor(1) != StoreBB) | ||||
11470 | return false; | ||||
11471 | |||||
11472 | // Okay, we know that OtherBr now goes to Dest and StoreBB, so this is an | ||||
11473 | // if/then triangle. See if there is a store to the same ptr as SI that | ||||
11474 | // lives in OtherBB. | ||||
11475 | for (;; --BBI) { | ||||
11476 | // Check to see if we find the matching store. | ||||
11477 | if ((OtherStore = dyn_cast<StoreInst>(BBI))) { | ||||
11478 | if (OtherStore->getOperand(1) != SI.getOperand(1)) | ||||
11479 | return false; | ||||
11480 | break; | ||||
11481 | } | ||||
Eli Friedman | 3a311d5 | 2008-06-13 22:02:12 +0000 | [diff] [blame] | 11482 | // If we find something that may be using or overwriting the stored |
11483 | // value, or if we run out of instructions, we can't do the xform. | ||||
11484 | if (BBI->mayReadFromMemory() || BBI->mayWriteToMemory() || | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11485 | BBI == OtherBB->begin()) |
11486 | return false; | ||||
11487 | } | ||||
11488 | |||||
11489 | // In order to eliminate the store in OtherBr, we have to | ||||
Eli Friedman | 3a311d5 | 2008-06-13 22:02:12 +0000 | [diff] [blame] | 11490 | // make sure nothing reads or overwrites the stored value in |
11491 | // StoreBB. | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11492 | for (BasicBlock::iterator I = StoreBB->begin(); &*I != &SI; ++I) { |
11493 | // FIXME: This should really be AA driven. | ||||
Eli Friedman | 3a311d5 | 2008-06-13 22:02:12 +0000 | [diff] [blame] | 11494 | if (I->mayReadFromMemory() || I->mayWriteToMemory()) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11495 | return false; |
11496 | } | ||||
11497 | } | ||||
11498 | |||||
11499 | // Insert a PHI node now if we need it. | ||||
11500 | Value *MergedVal = OtherStore->getOperand(0); | ||||
11501 | if (MergedVal != SI.getOperand(0)) { | ||||
Gabor Greif | d6da1d0 | 2008-04-06 20:25:17 +0000 | [diff] [blame] | 11502 | PHINode *PN = PHINode::Create(MergedVal->getType(), "storemerge"); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11503 | PN->reserveOperandSpace(2); |
11504 | PN->addIncoming(SI.getOperand(0), SI.getParent()); | ||||
11505 | PN->addIncoming(OtherStore->getOperand(0), OtherBB); | ||||
11506 | MergedVal = InsertNewInstBefore(PN, DestBB->front()); | ||||
11507 | } | ||||
11508 | |||||
11509 | // Advance to a place where it is safe to insert the new store and | ||||
11510 | // insert it. | ||||
Dan Gohman | 514277c | 2008-05-23 21:05:58 +0000 | [diff] [blame] | 11511 | BBI = DestBB->getFirstNonPHI(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11512 | InsertNewInstBefore(new StoreInst(MergedVal, SI.getOperand(1), |
11513 | OtherStore->isVolatile()), *BBI); | ||||
11514 | |||||
11515 | // Nuke the old stores. | ||||
11516 | EraseInstFromFunction(SI); | ||||
11517 | EraseInstFromFunction(*OtherStore); | ||||
11518 | ++NumCombined; | ||||
11519 | return true; | ||||
11520 | } | ||||
11521 | |||||
11522 | |||||
11523 | Instruction *InstCombiner::visitBranchInst(BranchInst &BI) { | ||||
11524 | // Change br (not X), label True, label False to: br X, label False, True | ||||
11525 | Value *X = 0; | ||||
11526 | BasicBlock *TrueDest; | ||||
11527 | BasicBlock *FalseDest; | ||||
11528 | if (match(&BI, m_Br(m_Not(m_Value(X)), TrueDest, FalseDest)) && | ||||
11529 | !isa<Constant>(X)) { | ||||
11530 | // Swap Destinations and condition... | ||||
11531 | BI.setCondition(X); | ||||
11532 | BI.setSuccessor(0, FalseDest); | ||||
11533 | BI.setSuccessor(1, TrueDest); | ||||
11534 | return &BI; | ||||
11535 | } | ||||
11536 | |||||
11537 | // Cannonicalize fcmp_one -> fcmp_oeq | ||||
11538 | FCmpInst::Predicate FPred; Value *Y; | ||||
11539 | if (match(&BI, m_Br(m_FCmp(FPred, m_Value(X), m_Value(Y)), | ||||
11540 | TrueDest, FalseDest))) | ||||
11541 | if ((FPred == FCmpInst::FCMP_ONE || FPred == FCmpInst::FCMP_OLE || | ||||
11542 | FPred == FCmpInst::FCMP_OGE) && BI.getCondition()->hasOneUse()) { | ||||
11543 | FCmpInst *I = cast<FCmpInst>(BI.getCondition()); | ||||
11544 | FCmpInst::Predicate NewPred = FCmpInst::getInversePredicate(FPred); | ||||
11545 | Instruction *NewSCC = new FCmpInst(NewPred, X, Y, "", I); | ||||
11546 | NewSCC->takeName(I); | ||||
11547 | // Swap Destinations and condition... | ||||
11548 | BI.setCondition(NewSCC); | ||||
11549 | BI.setSuccessor(0, FalseDest); | ||||
11550 | BI.setSuccessor(1, TrueDest); | ||||
11551 | RemoveFromWorkList(I); | ||||
11552 | I->eraseFromParent(); | ||||
11553 | AddToWorkList(NewSCC); | ||||
11554 | return &BI; | ||||
11555 | } | ||||
11556 | |||||
11557 | // Cannonicalize icmp_ne -> icmp_eq | ||||
11558 | ICmpInst::Predicate IPred; | ||||
11559 | if (match(&BI, m_Br(m_ICmp(IPred, m_Value(X), m_Value(Y)), | ||||
11560 | TrueDest, FalseDest))) | ||||
11561 | if ((IPred == ICmpInst::ICMP_NE || IPred == ICmpInst::ICMP_ULE || | ||||
11562 | IPred == ICmpInst::ICMP_SLE || IPred == ICmpInst::ICMP_UGE || | ||||
11563 | IPred == ICmpInst::ICMP_SGE) && BI.getCondition()->hasOneUse()) { | ||||
11564 | ICmpInst *I = cast<ICmpInst>(BI.getCondition()); | ||||
11565 | ICmpInst::Predicate NewPred = ICmpInst::getInversePredicate(IPred); | ||||
11566 | Instruction *NewSCC = new ICmpInst(NewPred, X, Y, "", I); | ||||
11567 | NewSCC->takeName(I); | ||||
11568 | // Swap Destinations and condition... | ||||
11569 | BI.setCondition(NewSCC); | ||||
11570 | BI.setSuccessor(0, FalseDest); | ||||
11571 | BI.setSuccessor(1, TrueDest); | ||||
11572 | RemoveFromWorkList(I); | ||||
11573 | I->eraseFromParent();; | ||||
11574 | AddToWorkList(NewSCC); | ||||
11575 | return &BI; | ||||
11576 | } | ||||
11577 | |||||
11578 | return 0; | ||||
11579 | } | ||||
11580 | |||||
11581 | Instruction *InstCombiner::visitSwitchInst(SwitchInst &SI) { | ||||
11582 | Value *Cond = SI.getCondition(); | ||||
11583 | if (Instruction *I = dyn_cast<Instruction>(Cond)) { | ||||
11584 | if (I->getOpcode() == Instruction::Add) | ||||
11585 | if (ConstantInt *AddRHS = dyn_cast<ConstantInt>(I->getOperand(1))) { | ||||
11586 | // change 'switch (X+4) case 1:' into 'switch (X) case -3' | ||||
11587 | for (unsigned i = 2, e = SI.getNumOperands(); i != e; i += 2) | ||||
11588 | SI.setOperand(i,ConstantExpr::getSub(cast<Constant>(SI.getOperand(i)), | ||||
11589 | AddRHS)); | ||||
11590 | SI.setOperand(0, I->getOperand(0)); | ||||
11591 | AddToWorkList(I); | ||||
11592 | return &SI; | ||||
11593 | } | ||||
11594 | } | ||||
11595 | return 0; | ||||
11596 | } | ||||
11597 | |||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 11598 | Instruction *InstCombiner::visitExtractValueInst(ExtractValueInst &EV) { |
Matthijs Kooijman | 45e8eb4 | 2008-07-16 12:55:45 +0000 | [diff] [blame] | 11599 | Value *Agg = EV.getAggregateOperand(); |
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 11600 | |
Matthijs Kooijman | 45e8eb4 | 2008-07-16 12:55:45 +0000 | [diff] [blame] | 11601 | if (!EV.hasIndices()) |
11602 | return ReplaceInstUsesWith(EV, Agg); | ||||
11603 | |||||
11604 | if (Constant *C = dyn_cast<Constant>(Agg)) { | ||||
11605 | if (isa<UndefValue>(C)) | ||||
11606 | return ReplaceInstUsesWith(EV, UndefValue::get(EV.getType())); | ||||
11607 | |||||
11608 | if (isa<ConstantAggregateZero>(C)) | ||||
11609 | return ReplaceInstUsesWith(EV, Constant::getNullValue(EV.getType())); | ||||
11610 | |||||
11611 | if (isa<ConstantArray>(C) || isa<ConstantStruct>(C)) { | ||||
11612 | // Extract the element indexed by the first index out of the constant | ||||
11613 | Value *V = C->getOperand(*EV.idx_begin()); | ||||
11614 | if (EV.getNumIndices() > 1) | ||||
11615 | // Extract the remaining indices out of the constant indexed by the | ||||
11616 | // first index | ||||
11617 | return ExtractValueInst::Create(V, EV.idx_begin() + 1, EV.idx_end()); | ||||
11618 | else | ||||
11619 | return ReplaceInstUsesWith(EV, V); | ||||
11620 | } | ||||
11621 | return 0; // Can't handle other constants | ||||
11622 | } | ||||
11623 | if (InsertValueInst *IV = dyn_cast<InsertValueInst>(Agg)) { | ||||
11624 | // We're extracting from an insertvalue instruction, compare the indices | ||||
11625 | const unsigned *exti, *exte, *insi, *inse; | ||||
11626 | for (exti = EV.idx_begin(), insi = IV->idx_begin(), | ||||
11627 | exte = EV.idx_end(), inse = IV->idx_end(); | ||||
11628 | exti != exte && insi != inse; | ||||
11629 | ++exti, ++insi) { | ||||
11630 | if (*insi != *exti) | ||||
11631 | // The insert and extract both reference distinctly different elements. | ||||
11632 | // This means the extract is not influenced by the insert, and we can | ||||
11633 | // replace the aggregate operand of the extract with the aggregate | ||||
11634 | // operand of the insert. i.e., replace | ||||
11635 | // %I = insertvalue { i32, { i32 } } %A, { i32 } { i32 42 }, 1 | ||||
11636 | // %E = extractvalue { i32, { i32 } } %I, 0 | ||||
11637 | // with | ||||
11638 | // %E = extractvalue { i32, { i32 } } %A, 0 | ||||
11639 | return ExtractValueInst::Create(IV->getAggregateOperand(), | ||||
11640 | EV.idx_begin(), EV.idx_end()); | ||||
11641 | } | ||||
11642 | if (exti == exte && insi == inse) | ||||
11643 | // Both iterators are at the end: Index lists are identical. Replace | ||||
11644 | // %B = insertvalue { i32, { i32 } } %A, i32 42, 1, 0 | ||||
11645 | // %C = extractvalue { i32, { i32 } } %B, 1, 0 | ||||
11646 | // with "i32 42" | ||||
11647 | return ReplaceInstUsesWith(EV, IV->getInsertedValueOperand()); | ||||
11648 | if (exti == exte) { | ||||
11649 | // The extract list is a prefix of the insert list. i.e. replace | ||||
11650 | // %I = insertvalue { i32, { i32 } } %A, i32 42, 1, 0 | ||||
11651 | // %E = extractvalue { i32, { i32 } } %I, 1 | ||||
11652 | // with | ||||
11653 | // %X = extractvalue { i32, { i32 } } %A, 1 | ||||
11654 | // %E = insertvalue { i32 } %X, i32 42, 0 | ||||
11655 | // by switching the order of the insert and extract (though the | ||||
11656 | // insertvalue should be left in, since it may have other uses). | ||||
11657 | Value *NewEV = InsertNewInstBefore( | ||||
11658 | ExtractValueInst::Create(IV->getAggregateOperand(), | ||||
11659 | EV.idx_begin(), EV.idx_end()), | ||||
11660 | EV); | ||||
11661 | return InsertValueInst::Create(NewEV, IV->getInsertedValueOperand(), | ||||
11662 | insi, inse); | ||||
11663 | } | ||||
11664 | if (insi == inse) | ||||
11665 | // The insert list is a prefix of the extract list | ||||
11666 | // We can simply remove the common indices from the extract and make it | ||||
11667 | // operate on the inserted value instead of the insertvalue result. | ||||
11668 | // i.e., replace | ||||
11669 | // %I = insertvalue { i32, { i32 } } %A, { i32 } { i32 42 }, 1 | ||||
11670 | // %E = extractvalue { i32, { i32 } } %I, 1, 0 | ||||
11671 | // with | ||||
11672 | // %E extractvalue { i32 } { i32 42 }, 0 | ||||
11673 | return ExtractValueInst::Create(IV->getInsertedValueOperand(), | ||||
11674 | exti, exte); | ||||
11675 | } | ||||
11676 | // Can't simplify extracts from other values. Note that nested extracts are | ||||
11677 | // already simplified implicitely by the above (extract ( extract (insert) ) | ||||
11678 | // will be translated into extract ( insert ( extract ) ) first and then just | ||||
11679 | // the value inserted, if appropriate). | ||||
Matthijs Kooijman | da9ef70 | 2008-06-11 14:05:05 +0000 | [diff] [blame] | 11680 | return 0; |
11681 | } | ||||
11682 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11683 | /// CheapToScalarize - Return true if the value is cheaper to scalarize than it |
11684 | /// is to leave as a vector operation. | ||||
11685 | static bool CheapToScalarize(Value *V, bool isConstant) { | ||||
11686 | if (isa<ConstantAggregateZero>(V)) | ||||
11687 | return true; | ||||
11688 | if (ConstantVector *C = dyn_cast<ConstantVector>(V)) { | ||||
11689 | if (isConstant) return true; | ||||
11690 | // If all elts are the same, we can extract. | ||||
11691 | Constant *Op0 = C->getOperand(0); | ||||
11692 | for (unsigned i = 1; i < C->getNumOperands(); ++i) | ||||
11693 | if (C->getOperand(i) != Op0) | ||||
11694 | return false; | ||||
11695 | return true; | ||||
11696 | } | ||||
11697 | Instruction *I = dyn_cast<Instruction>(V); | ||||
11698 | if (!I) return false; | ||||
11699 | |||||
11700 | // Insert element gets simplified to the inserted element or is deleted if | ||||
11701 | // this is constant idx extract element and its a constant idx insertelt. | ||||
11702 | if (I->getOpcode() == Instruction::InsertElement && isConstant && | ||||
11703 | isa<ConstantInt>(I->getOperand(2))) | ||||
11704 | return true; | ||||
11705 | if (I->getOpcode() == Instruction::Load && I->hasOneUse()) | ||||
11706 | return true; | ||||
11707 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(I)) | ||||
11708 | if (BO->hasOneUse() && | ||||
11709 | (CheapToScalarize(BO->getOperand(0), isConstant) || | ||||
11710 | CheapToScalarize(BO->getOperand(1), isConstant))) | ||||
11711 | return true; | ||||
11712 | if (CmpInst *CI = dyn_cast<CmpInst>(I)) | ||||
11713 | if (CI->hasOneUse() && | ||||
11714 | (CheapToScalarize(CI->getOperand(0), isConstant) || | ||||
11715 | CheapToScalarize(CI->getOperand(1), isConstant))) | ||||
11716 | return true; | ||||
11717 | |||||
11718 | return false; | ||||
11719 | } | ||||
11720 | |||||
11721 | /// Read and decode a shufflevector mask. | ||||
11722 | /// | ||||
11723 | /// It turns undef elements into values that are larger than the number of | ||||
11724 | /// elements in the input. | ||||
11725 | static std::vector<unsigned> getShuffleMask(const ShuffleVectorInst *SVI) { | ||||
11726 | unsigned NElts = SVI->getType()->getNumElements(); | ||||
11727 | if (isa<ConstantAggregateZero>(SVI->getOperand(2))) | ||||
11728 | return std::vector<unsigned>(NElts, 0); | ||||
11729 | if (isa<UndefValue>(SVI->getOperand(2))) | ||||
11730 | return std::vector<unsigned>(NElts, 2*NElts); | ||||
11731 | |||||
11732 | std::vector<unsigned> Result; | ||||
11733 | const ConstantVector *CP = cast<ConstantVector>(SVI->getOperand(2)); | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 11734 | for (User::const_op_iterator i = CP->op_begin(), e = CP->op_end(); i!=e; ++i) |
11735 | if (isa<UndefValue>(*i)) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11736 | Result.push_back(NElts*2); // undef -> 8 |
11737 | else | ||||
Gabor Greif | 1739600 | 2008-06-12 21:37:33 +0000 | [diff] [blame] | 11738 | Result.push_back(cast<ConstantInt>(*i)->getZExtValue()); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11739 | return Result; |
11740 | } | ||||
11741 | |||||
11742 | /// FindScalarElement - Given a vector and an element number, see if the scalar | ||||
11743 | /// value is already around as a register, for example if it were inserted then | ||||
11744 | /// extracted from the vector. | ||||
11745 | static Value *FindScalarElement(Value *V, unsigned EltNo) { | ||||
11746 | assert(isa<VectorType>(V->getType()) && "Not looking at a vector?"); | ||||
11747 | const VectorType *PTy = cast<VectorType>(V->getType()); | ||||
11748 | unsigned Width = PTy->getNumElements(); | ||||
11749 | if (EltNo >= Width) // Out of range access. | ||||
11750 | return UndefValue::get(PTy->getElementType()); | ||||
11751 | |||||
11752 | if (isa<UndefValue>(V)) | ||||
11753 | return UndefValue::get(PTy->getElementType()); | ||||
11754 | else if (isa<ConstantAggregateZero>(V)) | ||||
11755 | return Constant::getNullValue(PTy->getElementType()); | ||||
11756 | else if (ConstantVector *CP = dyn_cast<ConstantVector>(V)) | ||||
11757 | return CP->getOperand(EltNo); | ||||
11758 | else if (InsertElementInst *III = dyn_cast<InsertElementInst>(V)) { | ||||
11759 | // If this is an insert to a variable element, we don't know what it is. | ||||
11760 | if (!isa<ConstantInt>(III->getOperand(2))) | ||||
11761 | return 0; | ||||
11762 | unsigned IIElt = cast<ConstantInt>(III->getOperand(2))->getZExtValue(); | ||||
11763 | |||||
11764 | // If this is an insert to the element we are looking for, return the | ||||
11765 | // inserted value. | ||||
11766 | if (EltNo == IIElt) | ||||
11767 | return III->getOperand(1); | ||||
11768 | |||||
11769 | // Otherwise, the insertelement doesn't modify the value, recurse on its | ||||
11770 | // vector input. | ||||
11771 | return FindScalarElement(III->getOperand(0), EltNo); | ||||
11772 | } else if (ShuffleVectorInst *SVI = dyn_cast<ShuffleVectorInst>(V)) { | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 11773 | unsigned LHSWidth = |
11774 | cast<VectorType>(SVI->getOperand(0)->getType())->getNumElements(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11775 | unsigned InEl = getShuffleMask(SVI)[EltNo]; |
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 11776 | if (InEl < LHSWidth) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11777 | return FindScalarElement(SVI->getOperand(0), InEl); |
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 11778 | else if (InEl < LHSWidth*2) |
11779 | return FindScalarElement(SVI->getOperand(1), InEl - LHSWidth); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11780 | else |
11781 | return UndefValue::get(PTy->getElementType()); | ||||
11782 | } | ||||
11783 | |||||
11784 | // Otherwise, we don't know. | ||||
11785 | return 0; | ||||
11786 | } | ||||
11787 | |||||
11788 | Instruction *InstCombiner::visitExtractElementInst(ExtractElementInst &EI) { | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11789 | // If vector val is undef, replace extract with scalar undef. |
11790 | if (isa<UndefValue>(EI.getOperand(0))) | ||||
11791 | return ReplaceInstUsesWith(EI, UndefValue::get(EI.getType())); | ||||
11792 | |||||
11793 | // If vector val is constant 0, replace extract with scalar 0. | ||||
11794 | if (isa<ConstantAggregateZero>(EI.getOperand(0))) | ||||
11795 | return ReplaceInstUsesWith(EI, Constant::getNullValue(EI.getType())); | ||||
11796 | |||||
11797 | if (ConstantVector *C = dyn_cast<ConstantVector>(EI.getOperand(0))) { | ||||
Matthijs Kooijman | dd3425f | 2008-06-11 09:00:12 +0000 | [diff] [blame] | 11798 | // If vector val is constant with all elements the same, replace EI with |
11799 | // that element. When the elements are not identical, we cannot replace yet | ||||
11800 | // (we do that below, but only when the index is constant). | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11801 | Constant *op0 = C->getOperand(0); |
11802 | for (unsigned i = 1; i < C->getNumOperands(); ++i) | ||||
11803 | if (C->getOperand(i) != op0) { | ||||
11804 | op0 = 0; | ||||
11805 | break; | ||||
11806 | } | ||||
11807 | if (op0) | ||||
11808 | return ReplaceInstUsesWith(EI, op0); | ||||
11809 | } | ||||
11810 | |||||
11811 | // If extracting a specified index from the vector, see if we can recursively | ||||
11812 | // find a previously computed scalar that was inserted into the vector. | ||||
11813 | if (ConstantInt *IdxC = dyn_cast<ConstantInt>(EI.getOperand(1))) { | ||||
11814 | unsigned IndexVal = IdxC->getZExtValue(); | ||||
11815 | unsigned VectorWidth = | ||||
11816 | cast<VectorType>(EI.getOperand(0)->getType())->getNumElements(); | ||||
11817 | |||||
11818 | // If this is extracting an invalid index, turn this into undef, to avoid | ||||
11819 | // crashing the code below. | ||||
11820 | if (IndexVal >= VectorWidth) | ||||
11821 | return ReplaceInstUsesWith(EI, UndefValue::get(EI.getType())); | ||||
11822 | |||||
11823 | // This instruction only demands the single element from the input vector. | ||||
11824 | // If the input vector has a single use, simplify it based on this use | ||||
11825 | // property. | ||||
11826 | if (EI.getOperand(0)->hasOneUse() && VectorWidth != 1) { | ||||
11827 | uint64_t UndefElts; | ||||
11828 | if (Value *V = SimplifyDemandedVectorElts(EI.getOperand(0), | ||||
11829 | 1 << IndexVal, | ||||
11830 | UndefElts)) { | ||||
11831 | EI.setOperand(0, V); | ||||
11832 | return &EI; | ||||
11833 | } | ||||
11834 | } | ||||
11835 | |||||
11836 | if (Value *Elt = FindScalarElement(EI.getOperand(0), IndexVal)) | ||||
11837 | return ReplaceInstUsesWith(EI, Elt); | ||||
11838 | |||||
11839 | // If the this extractelement is directly using a bitcast from a vector of | ||||
11840 | // the same number of elements, see if we can find the source element from | ||||
11841 | // it. In this case, we will end up needing to bitcast the scalars. | ||||
11842 | if (BitCastInst *BCI = dyn_cast<BitCastInst>(EI.getOperand(0))) { | ||||
11843 | if (const VectorType *VT = | ||||
11844 | dyn_cast<VectorType>(BCI->getOperand(0)->getType())) | ||||
11845 | if (VT->getNumElements() == VectorWidth) | ||||
11846 | if (Value *Elt = FindScalarElement(BCI->getOperand(0), IndexVal)) | ||||
11847 | return new BitCastInst(Elt, EI.getType()); | ||||
11848 | } | ||||
11849 | } | ||||
11850 | |||||
11851 | if (Instruction *I = dyn_cast<Instruction>(EI.getOperand(0))) { | ||||
11852 | if (I->hasOneUse()) { | ||||
11853 | // Push extractelement into predecessor operation if legal and | ||||
11854 | // profitable to do so | ||||
11855 | if (BinaryOperator *BO = dyn_cast<BinaryOperator>(I)) { | ||||
11856 | bool isConstantElt = isa<ConstantInt>(EI.getOperand(1)); | ||||
11857 | if (CheapToScalarize(BO, isConstantElt)) { | ||||
11858 | ExtractElementInst *newEI0 = | ||||
11859 | new ExtractElementInst(BO->getOperand(0), EI.getOperand(1), | ||||
11860 | EI.getName()+".lhs"); | ||||
11861 | ExtractElementInst *newEI1 = | ||||
11862 | new ExtractElementInst(BO->getOperand(1), EI.getOperand(1), | ||||
11863 | EI.getName()+".rhs"); | ||||
11864 | InsertNewInstBefore(newEI0, EI); | ||||
11865 | InsertNewInstBefore(newEI1, EI); | ||||
Gabor Greif | a645dd3 | 2008-05-16 19:29:10 +0000 | [diff] [blame] | 11866 | return BinaryOperator::Create(BO->getOpcode(), newEI0, newEI1); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11867 | } |
11868 | } else if (isa<LoadInst>(I)) { | ||||
Christopher Lamb | bb2f222 | 2007-12-17 01:12:55 +0000 | [diff] [blame] | 11869 | unsigned AS = |
11870 | cast<PointerType>(I->getOperand(0)->getType())->getAddressSpace(); | ||||
Chris Lattner | 13c2d6e | 2008-01-13 22:23:22 +0000 | [diff] [blame] | 11871 | Value *Ptr = InsertBitCastBefore(I->getOperand(0), |
11872 | PointerType::get(EI.getType(), AS),EI); | ||||
Gabor Greif | b91ea9d | 2008-05-15 10:04:30 +0000 | [diff] [blame] | 11873 | GetElementPtrInst *GEP = |
11874 | GetElementPtrInst::Create(Ptr, EI.getOperand(1), I->getName()+".gep"); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11875 | InsertNewInstBefore(GEP, EI); |
11876 | return new LoadInst(GEP); | ||||
11877 | } | ||||
11878 | } | ||||
11879 | if (InsertElementInst *IE = dyn_cast<InsertElementInst>(I)) { | ||||
11880 | // Extracting the inserted element? | ||||
11881 | if (IE->getOperand(2) == EI.getOperand(1)) | ||||
11882 | return ReplaceInstUsesWith(EI, IE->getOperand(1)); | ||||
11883 | // If the inserted and extracted elements are constants, they must not | ||||
11884 | // be the same value, extract from the pre-inserted value instead. | ||||
11885 | if (isa<Constant>(IE->getOperand(2)) && | ||||
11886 | isa<Constant>(EI.getOperand(1))) { | ||||
11887 | AddUsesToWorkList(EI); | ||||
11888 | EI.setOperand(0, IE->getOperand(0)); | ||||
11889 | return &EI; | ||||
11890 | } | ||||
11891 | } else if (ShuffleVectorInst *SVI = dyn_cast<ShuffleVectorInst>(I)) { | ||||
11892 | // If this is extracting an element from a shufflevector, figure out where | ||||
11893 | // it came from and extract from the appropriate input element instead. | ||||
11894 | if (ConstantInt *Elt = dyn_cast<ConstantInt>(EI.getOperand(1))) { | ||||
11895 | unsigned SrcIdx = getShuffleMask(SVI)[Elt->getZExtValue()]; | ||||
11896 | Value *Src; | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 11897 | unsigned LHSWidth = |
11898 | cast<VectorType>(SVI->getOperand(0)->getType())->getNumElements(); | ||||
11899 | |||||
11900 | if (SrcIdx < LHSWidth) | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11901 | Src = SVI->getOperand(0); |
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 11902 | else if (SrcIdx < LHSWidth*2) { |
11903 | SrcIdx -= LHSWidth; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11904 | Src = SVI->getOperand(1); |
11905 | } else { | ||||
11906 | return ReplaceInstUsesWith(EI, UndefValue::get(EI.getType())); | ||||
11907 | } | ||||
11908 | return new ExtractElementInst(Src, SrcIdx); | ||||
11909 | } | ||||
11910 | } | ||||
11911 | } | ||||
11912 | return 0; | ||||
11913 | } | ||||
11914 | |||||
11915 | /// CollectSingleShuffleElements - If V is a shuffle of values that ONLY returns | ||||
11916 | /// elements from either LHS or RHS, return the shuffle mask and true. | ||||
11917 | /// Otherwise, return false. | ||||
11918 | static bool CollectSingleShuffleElements(Value *V, Value *LHS, Value *RHS, | ||||
11919 | std::vector<Constant*> &Mask) { | ||||
11920 | assert(V->getType() == LHS->getType() && V->getType() == RHS->getType() && | ||||
11921 | "Invalid CollectSingleShuffleElements"); | ||||
11922 | unsigned NumElts = cast<VectorType>(V->getType())->getNumElements(); | ||||
11923 | |||||
11924 | if (isa<UndefValue>(V)) { | ||||
11925 | Mask.assign(NumElts, UndefValue::get(Type::Int32Ty)); | ||||
11926 | return true; | ||||
11927 | } else if (V == LHS) { | ||||
11928 | for (unsigned i = 0; i != NumElts; ++i) | ||||
11929 | Mask.push_back(ConstantInt::get(Type::Int32Ty, i)); | ||||
11930 | return true; | ||||
11931 | } else if (V == RHS) { | ||||
11932 | for (unsigned i = 0; i != NumElts; ++i) | ||||
11933 | Mask.push_back(ConstantInt::get(Type::Int32Ty, i+NumElts)); | ||||
11934 | return true; | ||||
11935 | } else if (InsertElementInst *IEI = dyn_cast<InsertElementInst>(V)) { | ||||
11936 | // If this is an insert of an extract from some other vector, include it. | ||||
11937 | Value *VecOp = IEI->getOperand(0); | ||||
11938 | Value *ScalarOp = IEI->getOperand(1); | ||||
11939 | Value *IdxOp = IEI->getOperand(2); | ||||
11940 | |||||
11941 | if (!isa<ConstantInt>(IdxOp)) | ||||
11942 | return false; | ||||
11943 | unsigned InsertedIdx = cast<ConstantInt>(IdxOp)->getZExtValue(); | ||||
11944 | |||||
11945 | if (isa<UndefValue>(ScalarOp)) { // inserting undef into vector. | ||||
11946 | // Okay, we can handle this if the vector we are insertinting into is | ||||
11947 | // transitively ok. | ||||
11948 | if (CollectSingleShuffleElements(VecOp, LHS, RHS, Mask)) { | ||||
11949 | // If so, update the mask to reflect the inserted undef. | ||||
11950 | Mask[InsertedIdx] = UndefValue::get(Type::Int32Ty); | ||||
11951 | return true; | ||||
11952 | } | ||||
11953 | } else if (ExtractElementInst *EI = dyn_cast<ExtractElementInst>(ScalarOp)){ | ||||
11954 | if (isa<ConstantInt>(EI->getOperand(1)) && | ||||
11955 | EI->getOperand(0)->getType() == V->getType()) { | ||||
11956 | unsigned ExtractedIdx = | ||||
11957 | cast<ConstantInt>(EI->getOperand(1))->getZExtValue(); | ||||
11958 | |||||
11959 | // This must be extracting from either LHS or RHS. | ||||
11960 | if (EI->getOperand(0) == LHS || EI->getOperand(0) == RHS) { | ||||
11961 | // Okay, we can handle this if the vector we are insertinting into is | ||||
11962 | // transitively ok. | ||||
11963 | if (CollectSingleShuffleElements(VecOp, LHS, RHS, Mask)) { | ||||
11964 | // If so, update the mask to reflect the inserted value. | ||||
11965 | if (EI->getOperand(0) == LHS) { | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 11966 | Mask[InsertedIdx % NumElts] = |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11967 | ConstantInt::get(Type::Int32Ty, ExtractedIdx); |
11968 | } else { | ||||
11969 | assert(EI->getOperand(0) == RHS); | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 11970 | Mask[InsertedIdx % NumElts] = |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 11971 | ConstantInt::get(Type::Int32Ty, ExtractedIdx+NumElts); |
11972 | |||||
11973 | } | ||||
11974 | return true; | ||||
11975 | } | ||||
11976 | } | ||||
11977 | } | ||||
11978 | } | ||||
11979 | } | ||||
11980 | // TODO: Handle shufflevector here! | ||||
11981 | |||||
11982 | return false; | ||||
11983 | } | ||||
11984 | |||||
11985 | /// CollectShuffleElements - We are building a shuffle of V, using RHS as the | ||||
11986 | /// RHS of the shuffle instruction, if it is not null. Return a shuffle mask | ||||
11987 | /// that computes V and the LHS value of the shuffle. | ||||
11988 | static Value *CollectShuffleElements(Value *V, std::vector<Constant*> &Mask, | ||||
11989 | Value *&RHS) { | ||||
11990 | assert(isa<VectorType>(V->getType()) && | ||||
11991 | (RHS == 0 || V->getType() == RHS->getType()) && | ||||
11992 | "Invalid shuffle!"); | ||||
11993 | unsigned NumElts = cast<VectorType>(V->getType())->getNumElements(); | ||||
11994 | |||||
11995 | if (isa<UndefValue>(V)) { | ||||
11996 | Mask.assign(NumElts, UndefValue::get(Type::Int32Ty)); | ||||
11997 | return V; | ||||
11998 | } else if (isa<ConstantAggregateZero>(V)) { | ||||
11999 | Mask.assign(NumElts, ConstantInt::get(Type::Int32Ty, 0)); | ||||
12000 | return V; | ||||
12001 | } else if (InsertElementInst *IEI = dyn_cast<InsertElementInst>(V)) { | ||||
12002 | // If this is an insert of an extract from some other vector, include it. | ||||
12003 | Value *VecOp = IEI->getOperand(0); | ||||
12004 | Value *ScalarOp = IEI->getOperand(1); | ||||
12005 | Value *IdxOp = IEI->getOperand(2); | ||||
12006 | |||||
12007 | if (ExtractElementInst *EI = dyn_cast<ExtractElementInst>(ScalarOp)) { | ||||
12008 | if (isa<ConstantInt>(EI->getOperand(1)) && isa<ConstantInt>(IdxOp) && | ||||
12009 | EI->getOperand(0)->getType() == V->getType()) { | ||||
12010 | unsigned ExtractedIdx = | ||||
12011 | cast<ConstantInt>(EI->getOperand(1))->getZExtValue(); | ||||
12012 | unsigned InsertedIdx = cast<ConstantInt>(IdxOp)->getZExtValue(); | ||||
12013 | |||||
12014 | // Either the extracted from or inserted into vector must be RHSVec, | ||||
12015 | // otherwise we'd end up with a shuffle of three inputs. | ||||
12016 | if (EI->getOperand(0) == RHS || RHS == 0) { | ||||
12017 | RHS = EI->getOperand(0); | ||||
12018 | Value *V = CollectShuffleElements(VecOp, Mask, RHS); | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 12019 | Mask[InsertedIdx % NumElts] = |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12020 | ConstantInt::get(Type::Int32Ty, NumElts+ExtractedIdx); |
12021 | return V; | ||||
12022 | } | ||||
12023 | |||||
12024 | if (VecOp == RHS) { | ||||
12025 | Value *V = CollectShuffleElements(EI->getOperand(0), Mask, RHS); | ||||
12026 | // Everything but the extracted element is replaced with the RHS. | ||||
12027 | for (unsigned i = 0; i != NumElts; ++i) { | ||||
12028 | if (i != InsertedIdx) | ||||
12029 | Mask[i] = ConstantInt::get(Type::Int32Ty, NumElts+i); | ||||
12030 | } | ||||
12031 | return V; | ||||
12032 | } | ||||
12033 | |||||
12034 | // If this insertelement is a chain that comes from exactly these two | ||||
12035 | // vectors, return the vector and the effective shuffle. | ||||
12036 | if (CollectSingleShuffleElements(IEI, EI->getOperand(0), RHS, Mask)) | ||||
12037 | return EI->getOperand(0); | ||||
12038 | |||||
12039 | } | ||||
12040 | } | ||||
12041 | } | ||||
12042 | // TODO: Handle shufflevector here! | ||||
12043 | |||||
12044 | // Otherwise, can't do anything fancy. Return an identity vector. | ||||
12045 | for (unsigned i = 0; i != NumElts; ++i) | ||||
12046 | Mask.push_back(ConstantInt::get(Type::Int32Ty, i)); | ||||
12047 | return V; | ||||
12048 | } | ||||
12049 | |||||
12050 | Instruction *InstCombiner::visitInsertElementInst(InsertElementInst &IE) { | ||||
12051 | Value *VecOp = IE.getOperand(0); | ||||
12052 | Value *ScalarOp = IE.getOperand(1); | ||||
12053 | Value *IdxOp = IE.getOperand(2); | ||||
12054 | |||||
12055 | // Inserting an undef or into an undefined place, remove this. | ||||
12056 | if (isa<UndefValue>(ScalarOp) || isa<UndefValue>(IdxOp)) | ||||
12057 | ReplaceInstUsesWith(IE, VecOp); | ||||
12058 | |||||
12059 | // If the inserted element was extracted from some other vector, and if the | ||||
12060 | // indexes are constant, try to turn this into a shufflevector operation. | ||||
12061 | if (ExtractElementInst *EI = dyn_cast<ExtractElementInst>(ScalarOp)) { | ||||
12062 | if (isa<ConstantInt>(EI->getOperand(1)) && isa<ConstantInt>(IdxOp) && | ||||
12063 | EI->getOperand(0)->getType() == IE.getType()) { | ||||
12064 | unsigned NumVectorElts = IE.getType()->getNumElements(); | ||||
12065 | unsigned ExtractedIdx = | ||||
12066 | cast<ConstantInt>(EI->getOperand(1))->getZExtValue(); | ||||
12067 | unsigned InsertedIdx = cast<ConstantInt>(IdxOp)->getZExtValue(); | ||||
12068 | |||||
12069 | if (ExtractedIdx >= NumVectorElts) // Out of range extract. | ||||
12070 | return ReplaceInstUsesWith(IE, VecOp); | ||||
12071 | |||||
12072 | if (InsertedIdx >= NumVectorElts) // Out of range insert. | ||||
12073 | return ReplaceInstUsesWith(IE, UndefValue::get(IE.getType())); | ||||
12074 | |||||
12075 | // If we are extracting a value from a vector, then inserting it right | ||||
12076 | // back into the same place, just use the input vector. | ||||
12077 | if (EI->getOperand(0) == VecOp && ExtractedIdx == InsertedIdx) | ||||
12078 | return ReplaceInstUsesWith(IE, VecOp); | ||||
12079 | |||||
12080 | // We could theoretically do this for ANY input. However, doing so could | ||||
12081 | // turn chains of insertelement instructions into a chain of shufflevector | ||||
12082 | // instructions, and right now we do not merge shufflevectors. As such, | ||||
12083 | // only do this in a situation where it is clear that there is benefit. | ||||
12084 | if (isa<UndefValue>(VecOp) || isa<ConstantAggregateZero>(VecOp)) { | ||||
12085 | // Turn this into shuffle(EIOp0, VecOp, Mask). The result has all of | ||||
12086 | // the values of VecOp, except then one read from EIOp0. | ||||
12087 | // Build a new shuffle mask. | ||||
12088 | std::vector<Constant*> Mask; | ||||
12089 | if (isa<UndefValue>(VecOp)) | ||||
12090 | Mask.assign(NumVectorElts, UndefValue::get(Type::Int32Ty)); | ||||
12091 | else { | ||||
12092 | assert(isa<ConstantAggregateZero>(VecOp) && "Unknown thing"); | ||||
12093 | Mask.assign(NumVectorElts, ConstantInt::get(Type::Int32Ty, | ||||
12094 | NumVectorElts)); | ||||
12095 | } | ||||
12096 | Mask[InsertedIdx] = ConstantInt::get(Type::Int32Ty, ExtractedIdx); | ||||
12097 | return new ShuffleVectorInst(EI->getOperand(0), VecOp, | ||||
12098 | ConstantVector::get(Mask)); | ||||
12099 | } | ||||
12100 | |||||
12101 | // If this insertelement isn't used by some other insertelement, turn it | ||||
12102 | // (and any insertelements it points to), into one big shuffle. | ||||
12103 | if (!IE.hasOneUse() || !isa<InsertElementInst>(IE.use_back())) { | ||||
12104 | std::vector<Constant*> Mask; | ||||
12105 | Value *RHS = 0; | ||||
12106 | Value *LHS = CollectShuffleElements(&IE, Mask, RHS); | ||||
12107 | if (RHS == 0) RHS = UndefValue::get(LHS->getType()); | ||||
12108 | // We now have a shuffle of LHS, RHS, Mask. | ||||
12109 | return new ShuffleVectorInst(LHS, RHS, ConstantVector::get(Mask)); | ||||
12110 | } | ||||
12111 | } | ||||
12112 | } | ||||
12113 | |||||
12114 | return 0; | ||||
12115 | } | ||||
12116 | |||||
12117 | |||||
12118 | Instruction *InstCombiner::visitShuffleVectorInst(ShuffleVectorInst &SVI) { | ||||
12119 | Value *LHS = SVI.getOperand(0); | ||||
12120 | Value *RHS = SVI.getOperand(1); | ||||
12121 | std::vector<unsigned> Mask = getShuffleMask(&SVI); | ||||
12122 | |||||
12123 | bool MadeChange = false; | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 12124 | |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12125 | // Undefined shuffle mask -> undefined value. |
12126 | if (isa<UndefValue>(SVI.getOperand(2))) | ||||
12127 | return ReplaceInstUsesWith(SVI, UndefValue::get(SVI.getType())); | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 12128 | |
12129 | uint64_t UndefElts; | ||||
12130 | unsigned VWidth = cast<VectorType>(SVI.getType())->getNumElements(); | ||||
Mon P Wang | bff5d9c | 2008-11-10 04:46:22 +0000 | [diff] [blame] | 12131 | |
12132 | if (VWidth != cast<VectorType>(LHS->getType())->getNumElements()) | ||||
12133 | return 0; | ||||
12134 | |||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 12135 | uint64_t AllOnesEltMask = ~0ULL >> (64-VWidth); |
12136 | if (VWidth <= 64 && | ||||
Dan Gohman | 83b702d | 2008-09-11 22:47:57 +0000 | [diff] [blame] | 12137 | SimplifyDemandedVectorElts(&SVI, AllOnesEltMask, UndefElts)) { |
12138 | LHS = SVI.getOperand(0); | ||||
12139 | RHS = SVI.getOperand(1); | ||||
Dan Gohman | da93bbe | 2008-09-09 18:11:14 +0000 | [diff] [blame] | 12140 | MadeChange = true; |
Dan Gohman | 83b702d | 2008-09-11 22:47:57 +0000 | [diff] [blame] | 12141 | } |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12142 | |
12143 | // Canonicalize shuffle(x ,x,mask) -> shuffle(x, undef,mask') | ||||
12144 | // Canonicalize shuffle(undef,x,mask) -> shuffle(x, undef,mask'). | ||||
12145 | if (LHS == RHS || isa<UndefValue>(LHS)) { | ||||
12146 | if (isa<UndefValue>(LHS) && LHS == RHS) { | ||||
12147 | // shuffle(undef,undef,mask) -> undef. | ||||
12148 | return ReplaceInstUsesWith(SVI, LHS); | ||||
12149 | } | ||||
12150 | |||||
12151 | // Remap any references to RHS to use LHS. | ||||
12152 | std::vector<Constant*> Elts; | ||||
12153 | for (unsigned i = 0, e = Mask.size(); i != e; ++i) { | ||||
12154 | if (Mask[i] >= 2*e) | ||||
12155 | Elts.push_back(UndefValue::get(Type::Int32Ty)); | ||||
12156 | else { | ||||
12157 | if ((Mask[i] >= e && isa<UndefValue>(RHS)) || | ||||
Dan Gohman | bba96b9 | 2008-08-06 18:17:32 +0000 | [diff] [blame] | 12158 | (Mask[i] < e && isa<UndefValue>(LHS))) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12159 | Mask[i] = 2*e; // Turn into undef. |
Dan Gohman | bba96b9 | 2008-08-06 18:17:32 +0000 | [diff] [blame] | 12160 | Elts.push_back(UndefValue::get(Type::Int32Ty)); |
12161 | } else { | ||||
Mon P Wang | 6bf3c59 | 2008-08-20 02:23:25 +0000 | [diff] [blame] | 12162 | Mask[i] = Mask[i] % e; // Force to LHS. |
Dan Gohman | bba96b9 | 2008-08-06 18:17:32 +0000 | [diff] [blame] | 12163 | Elts.push_back(ConstantInt::get(Type::Int32Ty, Mask[i])); |
12164 | } | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12165 | } |
12166 | } | ||||
12167 | SVI.setOperand(0, SVI.getOperand(1)); | ||||
12168 | SVI.setOperand(1, UndefValue::get(RHS->getType())); | ||||
12169 | SVI.setOperand(2, ConstantVector::get(Elts)); | ||||
12170 | LHS = SVI.getOperand(0); | ||||
12171 | RHS = SVI.getOperand(1); | ||||
12172 | MadeChange = true; | ||||
12173 | } | ||||
12174 | |||||
12175 | // Analyze the shuffle, are the LHS or RHS and identity shuffles? | ||||
12176 | bool isLHSID = true, isRHSID = true; | ||||
12177 | |||||
12178 | for (unsigned i = 0, e = Mask.size(); i != e; ++i) { | ||||
12179 | if (Mask[i] >= e*2) continue; // Ignore undef values. | ||||
12180 | // Is this an identity shuffle of the LHS value? | ||||
12181 | isLHSID &= (Mask[i] == i); | ||||
12182 | |||||
12183 | // Is this an identity shuffle of the RHS value? | ||||
12184 | isRHSID &= (Mask[i]-e == i); | ||||
12185 | } | ||||
12186 | |||||
12187 | // Eliminate identity shuffles. | ||||
12188 | if (isLHSID) return ReplaceInstUsesWith(SVI, LHS); | ||||
12189 | if (isRHSID) return ReplaceInstUsesWith(SVI, RHS); | ||||
12190 | |||||
12191 | // If the LHS is a shufflevector itself, see if we can combine it with this | ||||
12192 | // one without producing an unusual shuffle. Here we are really conservative: | ||||
12193 | // we are absolutely afraid of producing a shuffle mask not in the input | ||||
12194 | // program, because the code gen may not be smart enough to turn a merged | ||||
12195 | // shuffle into two specific shuffles: it may produce worse code. As such, | ||||
12196 | // we only merge two shuffles if the result is one of the two input shuffle | ||||
12197 | // masks. In this case, merging the shuffles just removes one instruction, | ||||
12198 | // which we know is safe. This is good for things like turning: | ||||
12199 | // (splat(splat)) -> splat. | ||||
12200 | if (ShuffleVectorInst *LHSSVI = dyn_cast<ShuffleVectorInst>(LHS)) { | ||||
12201 | if (isa<UndefValue>(RHS)) { | ||||
12202 | std::vector<unsigned> LHSMask = getShuffleMask(LHSSVI); | ||||
12203 | |||||
12204 | std::vector<unsigned> NewMask; | ||||
12205 | for (unsigned i = 0, e = Mask.size(); i != e; ++i) | ||||
12206 | if (Mask[i] >= 2*e) | ||||
12207 | NewMask.push_back(2*e); | ||||
12208 | else | ||||
12209 | NewMask.push_back(LHSMask[Mask[i]]); | ||||
12210 | |||||
12211 | // If the result mask is equal to the src shuffle or this shuffle mask, do | ||||
12212 | // the replacement. | ||||
12213 | if (NewMask == LHSMask || NewMask == Mask) { | ||||
wangmp | 496a76d | 2009-01-26 04:39:00 +0000 | [diff] [blame] | 12214 | unsigned LHSInNElts = |
12215 | cast<VectorType>(LHSSVI->getOperand(0)->getType())->getNumElements(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12216 | std::vector<Constant*> Elts; |
12217 | for (unsigned i = 0, e = NewMask.size(); i != e; ++i) { | ||||
wangmp | 496a76d | 2009-01-26 04:39:00 +0000 | [diff] [blame] | 12218 | if (NewMask[i] >= LHSInNElts*2) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12219 | Elts.push_back(UndefValue::get(Type::Int32Ty)); |
12220 | } else { | ||||
12221 | Elts.push_back(ConstantInt::get(Type::Int32Ty, NewMask[i])); | ||||
12222 | } | ||||
12223 | } | ||||
12224 | return new ShuffleVectorInst(LHSSVI->getOperand(0), | ||||
12225 | LHSSVI->getOperand(1), | ||||
12226 | ConstantVector::get(Elts)); | ||||
12227 | } | ||||
12228 | } | ||||
12229 | } | ||||
12230 | |||||
12231 | return MadeChange ? &SVI : 0; | ||||
12232 | } | ||||
12233 | |||||
12234 | |||||
12235 | |||||
12236 | |||||
12237 | /// TryToSinkInstruction - Try to move the specified instruction from its | ||||
12238 | /// current block into the beginning of DestBlock, which can only happen if it's | ||||
12239 | /// safe to move the instruction past all of the instructions between it and the | ||||
12240 | /// end of its block. | ||||
12241 | static bool TryToSinkInstruction(Instruction *I, BasicBlock *DestBlock) { | ||||
12242 | assert(I->hasOneUse() && "Invariants didn't hold!"); | ||||
12243 | |||||
12244 | // Cannot move control-flow-involving, volatile loads, vaarg, etc. | ||||
Chris Lattner | cb19a1c | 2008-05-09 15:07:33 +0000 | [diff] [blame] | 12245 | if (isa<PHINode>(I) || I->mayWriteToMemory() || isa<TerminatorInst>(I)) |
12246 | return false; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12247 | |
12248 | // Do not sink alloca instructions out of the entry block. | ||||
12249 | if (isa<AllocaInst>(I) && I->getParent() == | ||||
12250 | &DestBlock->getParent()->getEntryBlock()) | ||||
12251 | return false; | ||||
12252 | |||||
12253 | // We can only sink load instructions if there is nothing between the load and | ||||
12254 | // the end of block that could change the value. | ||||
Chris Lattner | 0db40a6 | 2008-05-08 17:37:37 +0000 | [diff] [blame] | 12255 | if (I->mayReadFromMemory()) { |
12256 | for (BasicBlock::iterator Scan = I, E = I->getParent()->end(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12257 | Scan != E; ++Scan) |
12258 | if (Scan->mayWriteToMemory()) | ||||
12259 | return false; | ||||
12260 | } | ||||
12261 | |||||
Dan Gohman | 514277c | 2008-05-23 21:05:58 +0000 | [diff] [blame] | 12262 | BasicBlock::iterator InsertPos = DestBlock->getFirstNonPHI(); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12263 | |
12264 | I->moveBefore(InsertPos); | ||||
12265 | ++NumSunkInst; | ||||
12266 | return true; | ||||
12267 | } | ||||
12268 | |||||
12269 | |||||
12270 | /// AddReachableCodeToWorklist - Walk the function in depth-first order, adding | ||||
12271 | /// all reachable code to the worklist. | ||||
12272 | /// | ||||
12273 | /// This has a couple of tricks to make the code faster and more powerful. In | ||||
12274 | /// particular, we constant fold and DCE instructions as we go, to avoid adding | ||||
12275 | /// them to the worklist (this significantly speeds up instcombine on code where | ||||
12276 | /// many instructions are dead or constant). Additionally, if we find a branch | ||||
12277 | /// whose condition is a known constant, we only visit the reachable successors. | ||||
12278 | /// | ||||
12279 | static void AddReachableCodeToWorklist(BasicBlock *BB, | ||||
12280 | SmallPtrSet<BasicBlock*, 64> &Visited, | ||||
12281 | InstCombiner &IC, | ||||
12282 | const TargetData *TD) { | ||||
Chris Lattner | a06291a | 2008-08-15 04:03:01 +0000 | [diff] [blame] | 12283 | SmallVector<BasicBlock*, 256> Worklist; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12284 | Worklist.push_back(BB); |
12285 | |||||
12286 | while (!Worklist.empty()) { | ||||
12287 | BB = Worklist.back(); | ||||
12288 | Worklist.pop_back(); | ||||
12289 | |||||
12290 | // We have now visited this block! If we've already been here, ignore it. | ||||
12291 | if (!Visited.insert(BB)) continue; | ||||
Devang Patel | 794140c | 2008-11-19 18:56:50 +0000 | [diff] [blame] | 12292 | |
12293 | DbgInfoIntrinsic *DBI_Prev = NULL; | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12294 | for (BasicBlock::iterator BBI = BB->begin(), E = BB->end(); BBI != E; ) { |
12295 | Instruction *Inst = BBI++; | ||||
12296 | |||||
12297 | // DCE instruction if trivially dead. | ||||
12298 | if (isInstructionTriviallyDead(Inst)) { | ||||
12299 | ++NumDeadInst; | ||||
12300 | DOUT << "IC: DCE: " << *Inst; | ||||
12301 | Inst->eraseFromParent(); | ||||
12302 | continue; | ||||
12303 | } | ||||
12304 | |||||
12305 | // ConstantProp instruction if trivially constant. | ||||
12306 | if (Constant *C = ConstantFoldInstruction(Inst, TD)) { | ||||
12307 | DOUT << "IC: ConstFold to: " << *C << " from: " << *Inst; | ||||
12308 | Inst->replaceAllUsesWith(C); | ||||
12309 | ++NumConstProp; | ||||
12310 | Inst->eraseFromParent(); | ||||
12311 | continue; | ||||
12312 | } | ||||
Chris Lattner | e0f462d | 2007-07-20 22:06:41 +0000 | [diff] [blame] | 12313 | |
Devang Patel | 794140c | 2008-11-19 18:56:50 +0000 | [diff] [blame] | 12314 | // If there are two consecutive llvm.dbg.stoppoint calls then |
12315 | // it is likely that the optimizer deleted code in between these | ||||
12316 | // two intrinsics. | ||||
12317 | DbgInfoIntrinsic *DBI_Next = dyn_cast<DbgInfoIntrinsic>(Inst); | ||||
12318 | if (DBI_Next) { | ||||
12319 | if (DBI_Prev | ||||
12320 | && DBI_Prev->getIntrinsicID() == llvm::Intrinsic::dbg_stoppoint | ||||
12321 | && DBI_Next->getIntrinsicID() == llvm::Intrinsic::dbg_stoppoint) { | ||||
12322 | IC.RemoveFromWorkList(DBI_Prev); | ||||
12323 | DBI_Prev->eraseFromParent(); | ||||
12324 | } | ||||
12325 | DBI_Prev = DBI_Next; | ||||
12326 | } | ||||
12327 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12328 | IC.AddToWorkList(Inst); |
12329 | } | ||||
12330 | |||||
12331 | // Recursively visit successors. If this is a branch or switch on a | ||||
12332 | // constant, only visit the reachable successor. | ||||
12333 | TerminatorInst *TI = BB->getTerminator(); | ||||
12334 | if (BranchInst *BI = dyn_cast<BranchInst>(TI)) { | ||||
12335 | if (BI->isConditional() && isa<ConstantInt>(BI->getCondition())) { | ||||
12336 | bool CondVal = cast<ConstantInt>(BI->getCondition())->getZExtValue(); | ||||
Nick Lewycky | d551cf1 | 2008-03-09 08:50:23 +0000 | [diff] [blame] | 12337 | BasicBlock *ReachableBB = BI->getSuccessor(!CondVal); |
Nick Lewycky | d8aa33a | 2008-04-25 16:53:59 +0000 | [diff] [blame] | 12338 | Worklist.push_back(ReachableBB); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12339 | continue; |
12340 | } | ||||
12341 | } else if (SwitchInst *SI = dyn_cast<SwitchInst>(TI)) { | ||||
12342 | if (ConstantInt *Cond = dyn_cast<ConstantInt>(SI->getCondition())) { | ||||
12343 | // See if this is an explicit destination. | ||||
12344 | for (unsigned i = 1, e = SI->getNumSuccessors(); i != e; ++i) | ||||
12345 | if (SI->getCaseValue(i) == Cond) { | ||||
Nick Lewycky | d551cf1 | 2008-03-09 08:50:23 +0000 | [diff] [blame] | 12346 | BasicBlock *ReachableBB = SI->getSuccessor(i); |
Nick Lewycky | d8aa33a | 2008-04-25 16:53:59 +0000 | [diff] [blame] | 12347 | Worklist.push_back(ReachableBB); |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12348 | continue; |
12349 | } | ||||
12350 | |||||
12351 | // Otherwise it is the default destination. | ||||
12352 | Worklist.push_back(SI->getSuccessor(0)); | ||||
12353 | continue; | ||||
12354 | } | ||||
12355 | } | ||||
12356 | |||||
12357 | for (unsigned i = 0, e = TI->getNumSuccessors(); i != e; ++i) | ||||
12358 | Worklist.push_back(TI->getSuccessor(i)); | ||||
12359 | } | ||||
12360 | } | ||||
12361 | |||||
12362 | bool InstCombiner::DoOneIteration(Function &F, unsigned Iteration) { | ||||
12363 | bool Changed = false; | ||||
12364 | TD = &getAnalysis<TargetData>(); | ||||
12365 | |||||
12366 | DEBUG(DOUT << "\n\nINSTCOMBINE ITERATION #" << Iteration << " on " | ||||
12367 | << F.getNameStr() << "\n"); | ||||
12368 | |||||
12369 | { | ||||
12370 | // Do a depth-first traversal of the function, populate the worklist with | ||||
12371 | // the reachable instructions. Ignore blocks that are not reachable. Keep | ||||
12372 | // track of which blocks we visit. | ||||
12373 | SmallPtrSet<BasicBlock*, 64> Visited; | ||||
12374 | AddReachableCodeToWorklist(F.begin(), Visited, *this, TD); | ||||
12375 | |||||
12376 | // Do a quick scan over the function. If we find any blocks that are | ||||
12377 | // unreachable, remove any instructions inside of them. This prevents | ||||
12378 | // the instcombine code from having to deal with some bad special cases. | ||||
12379 | for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) | ||||
12380 | if (!Visited.count(BB)) { | ||||
12381 | Instruction *Term = BB->getTerminator(); | ||||
12382 | while (Term != BB->begin()) { // Remove instrs bottom-up | ||||
12383 | BasicBlock::iterator I = Term; --I; | ||||
12384 | |||||
12385 | DOUT << "IC: DCE: " << *I; | ||||
12386 | ++NumDeadInst; | ||||
12387 | |||||
12388 | if (!I->use_empty()) | ||||
12389 | I->replaceAllUsesWith(UndefValue::get(I->getType())); | ||||
12390 | I->eraseFromParent(); | ||||
Chris Lattner | f6d5886 | 2009-01-31 07:04:22 +0000 | [diff] [blame] | 12391 | Changed = true; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12392 | } |
12393 | } | ||||
12394 | } | ||||
12395 | |||||
12396 | while (!Worklist.empty()) { | ||||
12397 | Instruction *I = RemoveOneFromWorkList(); | ||||
12398 | if (I == 0) continue; // skip null values. | ||||
12399 | |||||
12400 | // Check to see if we can DCE the instruction. | ||||
12401 | if (isInstructionTriviallyDead(I)) { | ||||
12402 | // Add operands to the worklist. | ||||
12403 | if (I->getNumOperands() < 4) | ||||
12404 | AddUsesToWorkList(*I); | ||||
12405 | ++NumDeadInst; | ||||
12406 | |||||
12407 | DOUT << "IC: DCE: " << *I; | ||||
12408 | |||||
12409 | I->eraseFromParent(); | ||||
12410 | RemoveFromWorkList(I); | ||||
Chris Lattner | f6d5886 | 2009-01-31 07:04:22 +0000 | [diff] [blame] | 12411 | Changed = true; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12412 | continue; |
12413 | } | ||||
12414 | |||||
12415 | // Instruction isn't dead, see if we can constant propagate it. | ||||
12416 | if (Constant *C = ConstantFoldInstruction(I, TD)) { | ||||
12417 | DOUT << "IC: ConstFold to: " << *C << " from: " << *I; | ||||
12418 | |||||
12419 | // Add operands to the worklist. | ||||
12420 | AddUsesToWorkList(*I); | ||||
12421 | ReplaceInstUsesWith(*I, C); | ||||
12422 | |||||
12423 | ++NumConstProp; | ||||
12424 | I->eraseFromParent(); | ||||
12425 | RemoveFromWorkList(I); | ||||
Chris Lattner | f6d5886 | 2009-01-31 07:04:22 +0000 | [diff] [blame] | 12426 | Changed = true; |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12427 | continue; |
12428 | } | ||||
12429 | |||||
Nick Lewycky | adb6792 | 2008-05-25 20:56:15 +0000 | [diff] [blame] | 12430 | if (TD && I->getType()->getTypeID() == Type::VoidTyID) { |
12431 | // See if we can constant fold its operands. | ||||
Chris Lattner | f6d5886 | 2009-01-31 07:04:22 +0000 | [diff] [blame] | 12432 | for (User::op_iterator i = I->op_begin(), e = I->op_end(); i != e; ++i) |
12433 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(i)) | ||||
Nick Lewycky | adb6792 | 2008-05-25 20:56:15 +0000 | [diff] [blame] | 12434 | if (Constant *NewC = ConstantFoldConstantExpression(CE, TD)) |
Chris Lattner | f6d5886 | 2009-01-31 07:04:22 +0000 | [diff] [blame] | 12435 | if (NewC != CE) { |
12436 | i->set(NewC); | ||||
12437 | Changed = true; | ||||
12438 | } | ||||
Nick Lewycky | adb6792 | 2008-05-25 20:56:15 +0000 | [diff] [blame] | 12439 | } |
12440 | |||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12441 | // 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] | 12442 | if (I->hasOneUse()) { |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12443 | BasicBlock *BB = I->getParent(); |
12444 | BasicBlock *UserParent = cast<Instruction>(I->use_back())->getParent(); | ||||
12445 | if (UserParent != BB) { | ||||
12446 | bool UserIsSuccessor = false; | ||||
12447 | // See if the user is one of our successors. | ||||
12448 | for (succ_iterator SI = succ_begin(BB), E = succ_end(BB); SI != E; ++SI) | ||||
12449 | if (*SI == UserParent) { | ||||
12450 | UserIsSuccessor = true; | ||||
12451 | break; | ||||
12452 | } | ||||
12453 | |||||
12454 | // If the user is one of our immediate successors, and if that successor | ||||
12455 | // only has us as a predecessors (we'd have to split the critical edge | ||||
12456 | // otherwise), we can keep going. | ||||
12457 | if (UserIsSuccessor && !isa<PHINode>(I->use_back()) && | ||||
12458 | next(pred_begin(UserParent)) == pred_end(UserParent)) | ||||
12459 | // Okay, the CFG is simple enough, try to sink this instruction. | ||||
12460 | Changed |= TryToSinkInstruction(I, UserParent); | ||||
12461 | } | ||||
12462 | } | ||||
12463 | |||||
12464 | // Now that we have an instruction, try combining it to simplify it... | ||||
12465 | #ifndef NDEBUG | ||||
12466 | std::string OrigI; | ||||
12467 | #endif | ||||
12468 | DEBUG(std::ostringstream SS; I->print(SS); OrigI = SS.str();); | ||||
12469 | if (Instruction *Result = visit(*I)) { | ||||
12470 | ++NumCombined; | ||||
12471 | // Should we replace the old instruction with a new one? | ||||
12472 | if (Result != I) { | ||||
12473 | DOUT << "IC: Old = " << *I | ||||
12474 | << " New = " << *Result; | ||||
12475 | |||||
12476 | // Everything uses the new instruction now. | ||||
12477 | I->replaceAllUsesWith(Result); | ||||
12478 | |||||
12479 | // Push the new instruction and any users onto the worklist. | ||||
12480 | AddToWorkList(Result); | ||||
12481 | AddUsersToWorkList(*Result); | ||||
12482 | |||||
12483 | // Move the name to the new instruction first. | ||||
12484 | Result->takeName(I); | ||||
12485 | |||||
12486 | // Insert the new instruction into the basic block... | ||||
12487 | BasicBlock *InstParent = I->getParent(); | ||||
12488 | BasicBlock::iterator InsertPos = I; | ||||
12489 | |||||
12490 | if (!isa<PHINode>(Result)) // If combining a PHI, don't insert | ||||
12491 | while (isa<PHINode>(InsertPos)) // middle of a block of PHIs. | ||||
12492 | ++InsertPos; | ||||
12493 | |||||
12494 | InstParent->getInstList().insert(InsertPos, Result); | ||||
12495 | |||||
12496 | // Make sure that we reprocess all operands now that we reduced their | ||||
12497 | // use counts. | ||||
12498 | AddUsesToWorkList(*I); | ||||
12499 | |||||
12500 | // Instructions can end up on the worklist more than once. Make sure | ||||
12501 | // we do not process an instruction that has been deleted. | ||||
12502 | RemoveFromWorkList(I); | ||||
12503 | |||||
12504 | // Erase the old instruction. | ||||
12505 | InstParent->getInstList().erase(I); | ||||
12506 | } else { | ||||
12507 | #ifndef NDEBUG | ||||
12508 | DOUT << "IC: Mod = " << OrigI | ||||
12509 | << " New = " << *I; | ||||
12510 | #endif | ||||
12511 | |||||
12512 | // If the instruction was modified, it's possible that it is now dead. | ||||
12513 | // if so, remove it. | ||||
12514 | if (isInstructionTriviallyDead(I)) { | ||||
12515 | // Make sure we process all operands now that we are reducing their | ||||
12516 | // use counts. | ||||
12517 | AddUsesToWorkList(*I); | ||||
12518 | |||||
12519 | // Instructions may end up in the worklist more than once. Erase all | ||||
12520 | // occurrences of this instruction. | ||||
12521 | RemoveFromWorkList(I); | ||||
12522 | I->eraseFromParent(); | ||||
12523 | } else { | ||||
12524 | AddToWorkList(I); | ||||
12525 | AddUsersToWorkList(*I); | ||||
12526 | } | ||||
12527 | } | ||||
12528 | Changed = true; | ||||
12529 | } | ||||
12530 | } | ||||
12531 | |||||
12532 | assert(WorklistMap.empty() && "Worklist empty, but map not?"); | ||||
Chris Lattner | b933ea6 | 2007-08-05 08:47:58 +0000 | [diff] [blame] | 12533 | |
12534 | // Do an explicit clear, this shrinks the map if needed. | ||||
12535 | WorklistMap.clear(); | ||||
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12536 | return Changed; |
12537 | } | ||||
12538 | |||||
12539 | |||||
12540 | bool InstCombiner::runOnFunction(Function &F) { | ||||
12541 | MustPreserveLCSSA = mustPreserveAnalysisID(LCSSAID); | ||||
12542 | |||||
12543 | bool EverMadeChange = false; | ||||
12544 | |||||
12545 | // Iterate while there is work to do. | ||||
12546 | unsigned Iteration = 0; | ||||
Bill Wendling | d9644a4 | 2008-05-14 22:45:20 +0000 | [diff] [blame] | 12547 | while (DoOneIteration(F, Iteration++)) |
Dan Gohman | f17a25c | 2007-07-18 16:29:46 +0000 | [diff] [blame] | 12548 | EverMadeChange = true; |
12549 | return EverMadeChange; | ||||
12550 | } | ||||
12551 | |||||
12552 | FunctionPass *llvm::createInstructionCombiningPass() { | ||||
12553 | return new InstCombiner(); | ||||
12554 | } |