blob: c540d6fd2c03e22ad21f39f470d7025472febdfd [file] [log] [blame]
Chris Lattner9f3c25a2009-11-09 22:57:59 +00001//===- InstructionSimplify.cpp - Fold instruction operands ----------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements routines for folding instructions into simpler forms
11// that do not require creating new instructions. For example, this does
12// constant folding, and can handle identities like (X&0)->0.
13//
14//===----------------------------------------------------------------------===//
15
16#include "llvm/Analysis/InstructionSimplify.h"
17#include "llvm/Analysis/ConstantFolding.h"
Duncan Sands18450092010-11-16 12:16:38 +000018#include "llvm/Analysis/Dominators.h"
Chris Lattnerd06094f2009-11-10 00:55:12 +000019#include "llvm/Support/PatternMatch.h"
Duncan Sands18450092010-11-16 12:16:38 +000020#include "llvm/Support/ValueHandle.h"
Chris Lattner9f3c25a2009-11-09 22:57:59 +000021using namespace llvm;
Chris Lattnerd06094f2009-11-10 00:55:12 +000022using namespace llvm::PatternMatch;
Chris Lattner9f3c25a2009-11-09 22:57:59 +000023
Duncan Sands18450092010-11-16 12:16:38 +000024#define RecursionLimit 3
Duncan Sandsa74a58c2010-11-10 18:23:01 +000025
26static Value *SimplifyBinOp(unsigned, Value *, Value *, const TargetData *,
Duncan Sands18450092010-11-16 12:16:38 +000027 const DominatorTree *, unsigned);
Duncan Sandsa74a58c2010-11-10 18:23:01 +000028static Value *SimplifyCmpInst(unsigned, Value *, Value *, const TargetData *,
Duncan Sands18450092010-11-16 12:16:38 +000029 const DominatorTree *, unsigned);
30
31/// ValueDominatesPHI - Does the given value dominate the specified phi node?
32static bool ValueDominatesPHI(Value *V, PHINode *P, const DominatorTree *DT) {
33 Instruction *I = dyn_cast<Instruction>(V);
34 if (!I)
35 // Arguments and constants dominate all instructions.
36 return true;
37
38 // If we have a DominatorTree then do a precise test.
39 if (DT)
40 return DT->dominates(I, P);
41
42 // Otherwise, if the instruction is in the entry block, and is not an invoke,
43 // then it obviously dominates all phi nodes.
44 if (I->getParent() == &I->getParent()->getParent()->getEntryBlock() &&
45 !isa<InvokeInst>(I))
46 return true;
47
48 return false;
49}
Duncan Sandsa74a58c2010-11-10 18:23:01 +000050
Duncan Sandsb2cbdc32010-11-10 13:00:08 +000051/// ThreadBinOpOverSelect - In the case of a binary operation with a select
52/// instruction as an operand, try to simplify the binop by seeing whether
53/// evaluating it on both branches of the select results in the same value.
54/// Returns the common value if so, otherwise returns null.
55static Value *ThreadBinOpOverSelect(unsigned Opcode, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +000056 const TargetData *TD,
57 const DominatorTree *DT,
58 unsigned MaxRecurse) {
Duncan Sandsb2cbdc32010-11-10 13:00:08 +000059 SelectInst *SI;
60 if (isa<SelectInst>(LHS)) {
61 SI = cast<SelectInst>(LHS);
62 } else {
63 assert(isa<SelectInst>(RHS) && "No select instruction operand!");
64 SI = cast<SelectInst>(RHS);
65 }
66
67 // Evaluate the BinOp on the true and false branches of the select.
68 Value *TV;
69 Value *FV;
70 if (SI == LHS) {
Duncan Sands18450092010-11-16 12:16:38 +000071 TV = SimplifyBinOp(Opcode, SI->getTrueValue(), RHS, TD, DT, MaxRecurse);
72 FV = SimplifyBinOp(Opcode, SI->getFalseValue(), RHS, TD, DT, MaxRecurse);
Duncan Sandsb2cbdc32010-11-10 13:00:08 +000073 } else {
Duncan Sands18450092010-11-16 12:16:38 +000074 TV = SimplifyBinOp(Opcode, LHS, SI->getTrueValue(), TD, DT, MaxRecurse);
75 FV = SimplifyBinOp(Opcode, LHS, SI->getFalseValue(), TD, DT, MaxRecurse);
Duncan Sandsb2cbdc32010-11-10 13:00:08 +000076 }
77
78 // If they simplified to the same value, then return the common value.
79 // If they both failed to simplify then return null.
80 if (TV == FV)
81 return TV;
82
83 // If one branch simplified to undef, return the other one.
84 if (TV && isa<UndefValue>(TV))
85 return FV;
86 if (FV && isa<UndefValue>(FV))
87 return TV;
88
89 // If applying the operation did not change the true and false select values,
90 // then the result of the binop is the select itself.
91 if (TV == SI->getTrueValue() && FV == SI->getFalseValue())
92 return SI;
93
94 // If one branch simplified and the other did not, and the simplified
95 // value is equal to the unsimplified one, return the simplified value.
96 // For example, select (cond, X, X & Z) & Z -> X & Z.
97 if ((FV && !TV) || (TV && !FV)) {
98 // Check that the simplified value has the form "X op Y" where "op" is the
99 // same as the original operation.
100 Instruction *Simplified = dyn_cast<Instruction>(FV ? FV : TV);
101 if (Simplified && Simplified->getOpcode() == Opcode) {
102 // The value that didn't simplify is "UnsimplifiedLHS op UnsimplifiedRHS".
103 // We already know that "op" is the same as for the simplified value. See
104 // if the operands match too. If so, return the simplified value.
105 Value *UnsimplifiedBranch = FV ? SI->getTrueValue() : SI->getFalseValue();
106 Value *UnsimplifiedLHS = SI == LHS ? UnsimplifiedBranch : LHS;
107 Value *UnsimplifiedRHS = SI == LHS ? RHS : UnsimplifiedBranch;
108 if (Simplified->getOperand(0) == UnsimplifiedLHS &&
109 Simplified->getOperand(1) == UnsimplifiedRHS)
110 return Simplified;
111 if (Simplified->isCommutative() &&
112 Simplified->getOperand(1) == UnsimplifiedLHS &&
113 Simplified->getOperand(0) == UnsimplifiedRHS)
114 return Simplified;
115 }
116 }
117
118 return 0;
119}
120
121/// ThreadCmpOverSelect - In the case of a comparison with a select instruction,
122/// try to simplify the comparison by seeing whether both branches of the select
123/// result in the same value. Returns the common value if so, otherwise returns
124/// null.
125static Value *ThreadCmpOverSelect(CmpInst::Predicate Pred, Value *LHS,
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000126 Value *RHS, const TargetData *TD,
Duncan Sands18450092010-11-16 12:16:38 +0000127 const DominatorTree *DT,
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000128 unsigned MaxRecurse) {
Duncan Sandsb2cbdc32010-11-10 13:00:08 +0000129 // Make sure the select is on the LHS.
130 if (!isa<SelectInst>(LHS)) {
131 std::swap(LHS, RHS);
132 Pred = CmpInst::getSwappedPredicate(Pred);
133 }
134 assert(isa<SelectInst>(LHS) && "Not comparing with a select instruction!");
135 SelectInst *SI = cast<SelectInst>(LHS);
136
137 // Now that we have "cmp select(cond, TV, FV), RHS", analyse it.
138 // Does "cmp TV, RHS" simplify?
Duncan Sands18450092010-11-16 12:16:38 +0000139 if (Value *TCmp = SimplifyCmpInst(Pred, SI->getTrueValue(), RHS, TD, DT,
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000140 MaxRecurse))
Duncan Sandsb2cbdc32010-11-10 13:00:08 +0000141 // It does! Does "cmp FV, RHS" simplify?
Duncan Sands18450092010-11-16 12:16:38 +0000142 if (Value *FCmp = SimplifyCmpInst(Pred, SI->getFalseValue(), RHS, TD, DT,
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000143 MaxRecurse))
Duncan Sandsb2cbdc32010-11-10 13:00:08 +0000144 // It does! If they simplified to the same value, then use it as the
145 // result of the original comparison.
146 if (TCmp == FCmp)
147 return TCmp;
148 return 0;
149}
150
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000151/// ThreadBinOpOverPHI - In the case of a binary operation with an operand that
152/// is a PHI instruction, try to simplify the binop by seeing whether evaluating
153/// it on the incoming phi values yields the same result for every value. If so
154/// returns the common value, otherwise returns null.
155static Value *ThreadBinOpOverPHI(unsigned Opcode, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +0000156 const TargetData *TD, const DominatorTree *DT,
157 unsigned MaxRecurse) {
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000158 PHINode *PI;
159 if (isa<PHINode>(LHS)) {
160 PI = cast<PHINode>(LHS);
Duncan Sands18450092010-11-16 12:16:38 +0000161 // Bail out if RHS and the phi may be mutually interdependent due to a loop.
162 if (!ValueDominatesPHI(RHS, PI, DT))
163 return 0;
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000164 } else {
165 assert(isa<PHINode>(RHS) && "No PHI instruction operand!");
166 PI = cast<PHINode>(RHS);
Duncan Sands18450092010-11-16 12:16:38 +0000167 // Bail out if LHS and the phi may be mutually interdependent due to a loop.
168 if (!ValueDominatesPHI(LHS, PI, DT))
169 return 0;
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000170 }
171
172 // Evaluate the BinOp on the incoming phi values.
173 Value *CommonValue = 0;
174 for (unsigned i = 0, e = PI->getNumIncomingValues(); i != e; ++i) {
Duncan Sands55200892010-11-15 17:52:45 +0000175 Value *Incoming = PI->getIncomingValue(i);
Duncan Sandsff103412010-11-17 04:30:22 +0000176 // If the incoming value is the phi node itself, it can safely be skipped.
Duncan Sands55200892010-11-15 17:52:45 +0000177 if (Incoming == PI) continue;
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000178 Value *V = PI == LHS ?
Duncan Sands18450092010-11-16 12:16:38 +0000179 SimplifyBinOp(Opcode, Incoming, RHS, TD, DT, MaxRecurse) :
180 SimplifyBinOp(Opcode, LHS, Incoming, TD, DT, MaxRecurse);
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000181 // If the operation failed to simplify, or simplified to a different value
182 // to previously, then give up.
183 if (!V || (CommonValue && V != CommonValue))
184 return 0;
185 CommonValue = V;
186 }
187
188 return CommonValue;
189}
190
191/// ThreadCmpOverPHI - In the case of a comparison with a PHI instruction, try
192/// try to simplify the comparison by seeing whether comparing with all of the
193/// incoming phi values yields the same result every time. If so returns the
194/// common result, otherwise returns null.
195static Value *ThreadCmpOverPHI(CmpInst::Predicate Pred, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +0000196 const TargetData *TD, const DominatorTree *DT,
197 unsigned MaxRecurse) {
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000198 // Make sure the phi is on the LHS.
199 if (!isa<PHINode>(LHS)) {
200 std::swap(LHS, RHS);
201 Pred = CmpInst::getSwappedPredicate(Pred);
202 }
203 assert(isa<PHINode>(LHS) && "Not comparing with a phi instruction!");
204 PHINode *PI = cast<PHINode>(LHS);
205
Duncan Sands18450092010-11-16 12:16:38 +0000206 // Bail out if RHS and the phi may be mutually interdependent due to a loop.
207 if (!ValueDominatesPHI(RHS, PI, DT))
208 return 0;
209
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000210 // Evaluate the BinOp on the incoming phi values.
211 Value *CommonValue = 0;
212 for (unsigned i = 0, e = PI->getNumIncomingValues(); i != e; ++i) {
Duncan Sands55200892010-11-15 17:52:45 +0000213 Value *Incoming = PI->getIncomingValue(i);
Duncan Sandsff103412010-11-17 04:30:22 +0000214 // If the incoming value is the phi node itself, it can safely be skipped.
Duncan Sands55200892010-11-15 17:52:45 +0000215 if (Incoming == PI) continue;
Duncan Sands18450092010-11-16 12:16:38 +0000216 Value *V = SimplifyCmpInst(Pred, Incoming, RHS, TD, DT, MaxRecurse);
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000217 // If the operation failed to simplify, or simplified to a different value
218 // to previously, then give up.
219 if (!V || (CommonValue && V != CommonValue))
220 return 0;
221 CommonValue = V;
222 }
223
224 return CommonValue;
225}
226
Chris Lattner8aee8ef2009-11-27 17:42:22 +0000227/// SimplifyAddInst - Given operands for an Add, see if we can
228/// fold the result. If not, this returns null.
229Value *llvm::SimplifyAddInst(Value *Op0, Value *Op1, bool isNSW, bool isNUW,
Duncan Sands18450092010-11-16 12:16:38 +0000230 const TargetData *TD, const DominatorTree *) {
Chris Lattner8aee8ef2009-11-27 17:42:22 +0000231 if (Constant *CLHS = dyn_cast<Constant>(Op0)) {
232 if (Constant *CRHS = dyn_cast<Constant>(Op1)) {
233 Constant *Ops[] = { CLHS, CRHS };
234 return ConstantFoldInstOperands(Instruction::Add, CLHS->getType(),
235 Ops, 2, TD);
236 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000237
Chris Lattner8aee8ef2009-11-27 17:42:22 +0000238 // Canonicalize the constant to the RHS.
239 std::swap(Op0, Op1);
240 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000241
Chris Lattner8aee8ef2009-11-27 17:42:22 +0000242 if (Constant *Op1C = dyn_cast<Constant>(Op1)) {
243 // X + undef -> undef
244 if (isa<UndefValue>(Op1C))
245 return Op1C;
Duncan Sands12a86f52010-11-14 11:23:23 +0000246
Chris Lattner8aee8ef2009-11-27 17:42:22 +0000247 // X + 0 --> X
248 if (Op1C->isNullValue())
249 return Op0;
250 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000251
Chris Lattner8aee8ef2009-11-27 17:42:22 +0000252 // FIXME: Could pull several more out of instcombine.
253 return 0;
254}
255
Chris Lattnerd06094f2009-11-10 00:55:12 +0000256/// SimplifyAndInst - Given operands for an And, see if we can
Chris Lattner9f3c25a2009-11-09 22:57:59 +0000257/// fold the result. If not, this returns null.
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000258static Value *SimplifyAndInst(Value *Op0, Value *Op1, const TargetData *TD,
Duncan Sands18450092010-11-16 12:16:38 +0000259 const DominatorTree *DT, unsigned MaxRecurse) {
Chris Lattnerd06094f2009-11-10 00:55:12 +0000260 if (Constant *CLHS = dyn_cast<Constant>(Op0)) {
261 if (Constant *CRHS = dyn_cast<Constant>(Op1)) {
262 Constant *Ops[] = { CLHS, CRHS };
263 return ConstantFoldInstOperands(Instruction::And, CLHS->getType(),
264 Ops, 2, TD);
265 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000266
Chris Lattnerd06094f2009-11-10 00:55:12 +0000267 // Canonicalize the constant to the RHS.
268 std::swap(Op0, Op1);
269 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000270
Chris Lattnerd06094f2009-11-10 00:55:12 +0000271 // X & undef -> 0
272 if (isa<UndefValue>(Op1))
273 return Constant::getNullValue(Op0->getType());
Duncan Sands12a86f52010-11-14 11:23:23 +0000274
Chris Lattnerd06094f2009-11-10 00:55:12 +0000275 // X & X = X
276 if (Op0 == Op1)
277 return Op0;
Duncan Sands12a86f52010-11-14 11:23:23 +0000278
Chris Lattnerd06094f2009-11-10 00:55:12 +0000279 // X & <0,0> = <0,0>
280 if (isa<ConstantAggregateZero>(Op1))
281 return Op1;
Duncan Sands12a86f52010-11-14 11:23:23 +0000282
Chris Lattnerd06094f2009-11-10 00:55:12 +0000283 // X & <-1,-1> = X
284 if (ConstantVector *CP = dyn_cast<ConstantVector>(Op1))
285 if (CP->isAllOnesValue())
286 return Op0;
Duncan Sands12a86f52010-11-14 11:23:23 +0000287
Chris Lattnerd06094f2009-11-10 00:55:12 +0000288 if (ConstantInt *Op1CI = dyn_cast<ConstantInt>(Op1)) {
289 // X & 0 = 0
290 if (Op1CI->isZero())
291 return Op1CI;
292 // X & -1 = X
293 if (Op1CI->isAllOnesValue())
294 return Op0;
295 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000296
Chris Lattnerd06094f2009-11-10 00:55:12 +0000297 // A & ~A = ~A & A = 0
298 Value *A, *B;
Chris Lattner70ce6d02009-11-10 02:04:54 +0000299 if ((match(Op0, m_Not(m_Value(A))) && A == Op1) ||
300 (match(Op1, m_Not(m_Value(A))) && A == Op0))
Chris Lattnerd06094f2009-11-10 00:55:12 +0000301 return Constant::getNullValue(Op0->getType());
Duncan Sands12a86f52010-11-14 11:23:23 +0000302
Chris Lattnerd06094f2009-11-10 00:55:12 +0000303 // (A | ?) & A = A
304 if (match(Op0, m_Or(m_Value(A), m_Value(B))) &&
305 (A == Op1 || B == Op1))
306 return Op1;
Duncan Sands12a86f52010-11-14 11:23:23 +0000307
Chris Lattnerd06094f2009-11-10 00:55:12 +0000308 // A & (A | ?) = A
309 if (match(Op1, m_Or(m_Value(A), m_Value(B))) &&
310 (A == Op0 || B == Op0))
311 return Op0;
Duncan Sands12a86f52010-11-14 11:23:23 +0000312
Benjamin Kramer6844c8e2010-09-10 22:39:55 +0000313 // (A & B) & A -> A & B
314 if (match(Op0, m_And(m_Value(A), m_Value(B))) &&
315 (A == Op1 || B == Op1))
316 return Op0;
317
318 // A & (A & B) -> A & B
319 if (match(Op1, m_And(m_Value(A), m_Value(B))) &&
320 (A == Op0 || B == Op0))
321 return Op1;
322
Duncan Sandsb2cbdc32010-11-10 13:00:08 +0000323 // If the operation is with the result of a select instruction, check whether
324 // operating on either branch of the select always yields the same value.
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000325 if (MaxRecurse && (isa<SelectInst>(Op0) || isa<SelectInst>(Op1)))
Duncan Sands18450092010-11-16 12:16:38 +0000326 if (Value *V = ThreadBinOpOverSelect(Instruction::And, Op0, Op1, TD, DT,
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000327 MaxRecurse-1))
328 return V;
329
330 // If the operation is with the result of a phi instruction, check whether
331 // operating on all incoming values of the phi always yields the same value.
332 if (MaxRecurse && (isa<PHINode>(Op0) || isa<PHINode>(Op1)))
Duncan Sands18450092010-11-16 12:16:38 +0000333 if (Value *V = ThreadBinOpOverPHI(Instruction::And, Op0, Op1, TD, DT,
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000334 MaxRecurse-1))
Duncan Sandsb2cbdc32010-11-10 13:00:08 +0000335 return V;
336
Chris Lattner9f3c25a2009-11-09 22:57:59 +0000337 return 0;
338}
339
Duncan Sands18450092010-11-16 12:16:38 +0000340Value *llvm::SimplifyAndInst(Value *Op0, Value *Op1, const TargetData *TD,
341 const DominatorTree *DT) {
342 return ::SimplifyAndInst(Op0, Op1, TD, DT, RecursionLimit);
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000343}
344
Chris Lattnerd06094f2009-11-10 00:55:12 +0000345/// SimplifyOrInst - Given operands for an Or, see if we can
346/// fold the result. If not, this returns null.
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000347static Value *SimplifyOrInst(Value *Op0, Value *Op1, const TargetData *TD,
Duncan Sands18450092010-11-16 12:16:38 +0000348 const DominatorTree *DT, unsigned MaxRecurse) {
Chris Lattnerd06094f2009-11-10 00:55:12 +0000349 if (Constant *CLHS = dyn_cast<Constant>(Op0)) {
350 if (Constant *CRHS = dyn_cast<Constant>(Op1)) {
351 Constant *Ops[] = { CLHS, CRHS };
352 return ConstantFoldInstOperands(Instruction::Or, CLHS->getType(),
353 Ops, 2, TD);
354 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000355
Chris Lattnerd06094f2009-11-10 00:55:12 +0000356 // Canonicalize the constant to the RHS.
357 std::swap(Op0, Op1);
358 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000359
Chris Lattnerd06094f2009-11-10 00:55:12 +0000360 // X | undef -> -1
361 if (isa<UndefValue>(Op1))
362 return Constant::getAllOnesValue(Op0->getType());
Duncan Sands12a86f52010-11-14 11:23:23 +0000363
Chris Lattnerd06094f2009-11-10 00:55:12 +0000364 // X | X = X
365 if (Op0 == Op1)
366 return Op0;
367
368 // X | <0,0> = X
369 if (isa<ConstantAggregateZero>(Op1))
370 return Op0;
Duncan Sands12a86f52010-11-14 11:23:23 +0000371
Chris Lattnerd06094f2009-11-10 00:55:12 +0000372 // X | <-1,-1> = <-1,-1>
373 if (ConstantVector *CP = dyn_cast<ConstantVector>(Op1))
Duncan Sands12a86f52010-11-14 11:23:23 +0000374 if (CP->isAllOnesValue())
Chris Lattnerd06094f2009-11-10 00:55:12 +0000375 return Op1;
Duncan Sands12a86f52010-11-14 11:23:23 +0000376
Chris Lattnerd06094f2009-11-10 00:55:12 +0000377 if (ConstantInt *Op1CI = dyn_cast<ConstantInt>(Op1)) {
378 // X | 0 = X
379 if (Op1CI->isZero())
380 return Op0;
381 // X | -1 = -1
382 if (Op1CI->isAllOnesValue())
383 return Op1CI;
384 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000385
Chris Lattnerd06094f2009-11-10 00:55:12 +0000386 // A | ~A = ~A | A = -1
387 Value *A, *B;
Chris Lattner70ce6d02009-11-10 02:04:54 +0000388 if ((match(Op0, m_Not(m_Value(A))) && A == Op1) ||
389 (match(Op1, m_Not(m_Value(A))) && A == Op0))
Chris Lattnerd06094f2009-11-10 00:55:12 +0000390 return Constant::getAllOnesValue(Op0->getType());
Duncan Sands12a86f52010-11-14 11:23:23 +0000391
Chris Lattnerd06094f2009-11-10 00:55:12 +0000392 // (A & ?) | A = A
393 if (match(Op0, m_And(m_Value(A), m_Value(B))) &&
394 (A == Op1 || B == Op1))
395 return Op1;
Duncan Sands12a86f52010-11-14 11:23:23 +0000396
Chris Lattnerd06094f2009-11-10 00:55:12 +0000397 // A | (A & ?) = A
398 if (match(Op1, m_And(m_Value(A), m_Value(B))) &&
399 (A == Op0 || B == Op0))
400 return Op0;
Duncan Sands12a86f52010-11-14 11:23:23 +0000401
Benjamin Kramer6844c8e2010-09-10 22:39:55 +0000402 // (A | B) | A -> A | B
403 if (match(Op0, m_Or(m_Value(A), m_Value(B))) &&
404 (A == Op1 || B == Op1))
405 return Op0;
406
407 // A | (A | B) -> A | B
408 if (match(Op1, m_Or(m_Value(A), m_Value(B))) &&
409 (A == Op0 || B == Op0))
410 return Op1;
411
Duncan Sandsb2cbdc32010-11-10 13:00:08 +0000412 // If the operation is with the result of a select instruction, check whether
413 // operating on either branch of the select always yields the same value.
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000414 if (MaxRecurse && (isa<SelectInst>(Op0) || isa<SelectInst>(Op1)))
Duncan Sands18450092010-11-16 12:16:38 +0000415 if (Value *V = ThreadBinOpOverSelect(Instruction::Or, Op0, Op1, TD, DT,
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000416 MaxRecurse-1))
417 return V;
418
419 // If the operation is with the result of a phi instruction, check whether
420 // operating on all incoming values of the phi always yields the same value.
421 if (MaxRecurse && (isa<PHINode>(Op0) || isa<PHINode>(Op1)))
Duncan Sands18450092010-11-16 12:16:38 +0000422 if (Value *V = ThreadBinOpOverPHI(Instruction::Or, Op0, Op1, TD, DT,
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000423 MaxRecurse-1))
Duncan Sandsb2cbdc32010-11-10 13:00:08 +0000424 return V;
425
Chris Lattnerd06094f2009-11-10 00:55:12 +0000426 return 0;
427}
428
Duncan Sands18450092010-11-16 12:16:38 +0000429Value *llvm::SimplifyOrInst(Value *Op0, Value *Op1, const TargetData *TD,
430 const DominatorTree *DT) {
431 return ::SimplifyOrInst(Op0, Op1, TD, DT, RecursionLimit);
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000432}
Chris Lattnerd06094f2009-11-10 00:55:12 +0000433
Chris Lattner210c5d42009-11-09 23:55:12 +0000434static const Type *GetCompareTy(Value *Op) {
435 return CmpInst::makeCmpResultType(Op->getType());
436}
437
Chris Lattner9dbb4292009-11-09 23:28:39 +0000438/// SimplifyICmpInst - Given operands for an ICmpInst, see if we can
439/// fold the result. If not, this returns null.
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000440static Value *SimplifyICmpInst(unsigned Predicate, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +0000441 const TargetData *TD, const DominatorTree *DT,
442 unsigned MaxRecurse) {
Chris Lattner9f3c25a2009-11-09 22:57:59 +0000443 CmpInst::Predicate Pred = (CmpInst::Predicate)Predicate;
Chris Lattner9dbb4292009-11-09 23:28:39 +0000444 assert(CmpInst::isIntPredicate(Pred) && "Not an integer compare!");
Duncan Sands12a86f52010-11-14 11:23:23 +0000445
Chris Lattnerd06094f2009-11-10 00:55:12 +0000446 if (Constant *CLHS = dyn_cast<Constant>(LHS)) {
Chris Lattner8f73dea2009-11-09 23:06:58 +0000447 if (Constant *CRHS = dyn_cast<Constant>(RHS))
448 return ConstantFoldCompareInstOperands(Pred, CLHS, CRHS, TD);
Chris Lattnerd06094f2009-11-10 00:55:12 +0000449
450 // If we have a constant, make sure it is on the RHS.
451 std::swap(LHS, RHS);
452 Pred = CmpInst::getSwappedPredicate(Pred);
453 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000454
Chris Lattner210c5d42009-11-09 23:55:12 +0000455 // ITy - This is the return type of the compare we're considering.
456 const Type *ITy = GetCompareTy(LHS);
Duncan Sands12a86f52010-11-14 11:23:23 +0000457
Chris Lattner210c5d42009-11-09 23:55:12 +0000458 // icmp X, X -> true/false
Chris Lattnerc8e14b32010-03-03 19:46:03 +0000459 // X icmp undef -> true/false. For example, icmp ugt %X, undef -> false
460 // because X could be 0.
461 if (LHS == RHS || isa<UndefValue>(RHS))
Chris Lattner210c5d42009-11-09 23:55:12 +0000462 return ConstantInt::get(ITy, CmpInst::isTrueWhenEqual(Pred));
Duncan Sands12a86f52010-11-14 11:23:23 +0000463
Chris Lattner210c5d42009-11-09 23:55:12 +0000464 // icmp <global/alloca*/null>, <global/alloca*/null> - Global/Stack value
465 // addresses never equal each other! We already know that Op0 != Op1.
Duncan Sands12a86f52010-11-14 11:23:23 +0000466 if ((isa<GlobalValue>(LHS) || isa<AllocaInst>(LHS) ||
Chris Lattner210c5d42009-11-09 23:55:12 +0000467 isa<ConstantPointerNull>(LHS)) &&
Duncan Sands12a86f52010-11-14 11:23:23 +0000468 (isa<GlobalValue>(RHS) || isa<AllocaInst>(RHS) ||
Chris Lattner210c5d42009-11-09 23:55:12 +0000469 isa<ConstantPointerNull>(RHS)))
470 return ConstantInt::get(ITy, CmpInst::isFalseWhenEqual(Pred));
Duncan Sands12a86f52010-11-14 11:23:23 +0000471
Chris Lattner210c5d42009-11-09 23:55:12 +0000472 // See if we are doing a comparison with a constant.
473 if (ConstantInt *CI = dyn_cast<ConstantInt>(RHS)) {
474 // If we have an icmp le or icmp ge instruction, turn it into the
475 // appropriate icmp lt or icmp gt instruction. This allows us to rely on
476 // them being folded in the code below.
477 switch (Pred) {
478 default: break;
479 case ICmpInst::ICMP_ULE:
480 if (CI->isMaxValue(false)) // A <=u MAX -> TRUE
481 return ConstantInt::getTrue(CI->getContext());
482 break;
483 case ICmpInst::ICMP_SLE:
484 if (CI->isMaxValue(true)) // A <=s MAX -> TRUE
485 return ConstantInt::getTrue(CI->getContext());
486 break;
487 case ICmpInst::ICMP_UGE:
488 if (CI->isMinValue(false)) // A >=u MIN -> TRUE
489 return ConstantInt::getTrue(CI->getContext());
490 break;
491 case ICmpInst::ICMP_SGE:
492 if (CI->isMinValue(true)) // A >=s MIN -> TRUE
493 return ConstantInt::getTrue(CI->getContext());
494 break;
495 }
Chris Lattner210c5d42009-11-09 23:55:12 +0000496 }
Duncan Sands1ac7c992010-11-07 16:12:23 +0000497
498 // If the comparison is with the result of a select instruction, check whether
499 // comparing with either branch of the select always yields the same value.
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000500 if (MaxRecurse && (isa<SelectInst>(LHS) || isa<SelectInst>(RHS)))
Duncan Sands18450092010-11-16 12:16:38 +0000501 if (Value *V = ThreadCmpOverSelect(Pred, LHS, RHS, TD, DT, MaxRecurse-1))
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000502 return V;
503
504 // If the comparison is with the result of a phi instruction, check whether
505 // doing the compare with each incoming phi value yields a common result.
506 if (MaxRecurse && (isa<PHINode>(LHS) || isa<PHINode>(RHS)))
Duncan Sands18450092010-11-16 12:16:38 +0000507 if (Value *V = ThreadCmpOverPHI(Pred, LHS, RHS, TD, DT, MaxRecurse-1))
Duncan Sands3bbb0cc2010-11-09 17:25:51 +0000508 return V;
Duncan Sands1ac7c992010-11-07 16:12:23 +0000509
Chris Lattner9f3c25a2009-11-09 22:57:59 +0000510 return 0;
511}
512
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000513Value *llvm::SimplifyICmpInst(unsigned Predicate, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +0000514 const TargetData *TD, const DominatorTree *DT) {
515 return ::SimplifyICmpInst(Predicate, LHS, RHS, TD, DT, RecursionLimit);
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000516}
517
Chris Lattner9dbb4292009-11-09 23:28:39 +0000518/// SimplifyFCmpInst - Given operands for an FCmpInst, see if we can
519/// fold the result. If not, this returns null.
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000520static Value *SimplifyFCmpInst(unsigned Predicate, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +0000521 const TargetData *TD, const DominatorTree *DT,
522 unsigned MaxRecurse) {
Chris Lattner9dbb4292009-11-09 23:28:39 +0000523 CmpInst::Predicate Pred = (CmpInst::Predicate)Predicate;
524 assert(CmpInst::isFPPredicate(Pred) && "Not an FP compare!");
525
Chris Lattnerd06094f2009-11-10 00:55:12 +0000526 if (Constant *CLHS = dyn_cast<Constant>(LHS)) {
Chris Lattner9dbb4292009-11-09 23:28:39 +0000527 if (Constant *CRHS = dyn_cast<Constant>(RHS))
528 return ConstantFoldCompareInstOperands(Pred, CLHS, CRHS, TD);
Duncan Sands12a86f52010-11-14 11:23:23 +0000529
Chris Lattnerd06094f2009-11-10 00:55:12 +0000530 // If we have a constant, make sure it is on the RHS.
531 std::swap(LHS, RHS);
532 Pred = CmpInst::getSwappedPredicate(Pred);
533 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000534
Chris Lattner210c5d42009-11-09 23:55:12 +0000535 // Fold trivial predicates.
536 if (Pred == FCmpInst::FCMP_FALSE)
537 return ConstantInt::get(GetCompareTy(LHS), 0);
538 if (Pred == FCmpInst::FCMP_TRUE)
539 return ConstantInt::get(GetCompareTy(LHS), 1);
540
Chris Lattner210c5d42009-11-09 23:55:12 +0000541 if (isa<UndefValue>(RHS)) // fcmp pred X, undef -> undef
542 return UndefValue::get(GetCompareTy(LHS));
543
544 // fcmp x,x -> true/false. Not all compares are foldable.
545 if (LHS == RHS) {
546 if (CmpInst::isTrueWhenEqual(Pred))
547 return ConstantInt::get(GetCompareTy(LHS), 1);
548 if (CmpInst::isFalseWhenEqual(Pred))
549 return ConstantInt::get(GetCompareTy(LHS), 0);
550 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000551
Chris Lattner210c5d42009-11-09 23:55:12 +0000552 // Handle fcmp with constant RHS
553 if (Constant *RHSC = dyn_cast<Constant>(RHS)) {
554 // If the constant is a nan, see if we can fold the comparison based on it.
555 if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHSC)) {
556 if (CFP->getValueAPF().isNaN()) {
557 if (FCmpInst::isOrdered(Pred)) // True "if ordered and foo"
558 return ConstantInt::getFalse(CFP->getContext());
559 assert(FCmpInst::isUnordered(Pred) &&
560 "Comparison must be either ordered or unordered!");
561 // True if unordered.
562 return ConstantInt::getTrue(CFP->getContext());
563 }
Dan Gohman6b617a72010-02-22 04:06:03 +0000564 // Check whether the constant is an infinity.
565 if (CFP->getValueAPF().isInfinity()) {
566 if (CFP->getValueAPF().isNegative()) {
567 switch (Pred) {
568 case FCmpInst::FCMP_OLT:
569 // No value is ordered and less than negative infinity.
570 return ConstantInt::getFalse(CFP->getContext());
571 case FCmpInst::FCMP_UGE:
572 // All values are unordered with or at least negative infinity.
573 return ConstantInt::getTrue(CFP->getContext());
574 default:
575 break;
576 }
577 } else {
578 switch (Pred) {
579 case FCmpInst::FCMP_OGT:
580 // No value is ordered and greater than infinity.
581 return ConstantInt::getFalse(CFP->getContext());
582 case FCmpInst::FCMP_ULE:
583 // All values are unordered with and at most infinity.
584 return ConstantInt::getTrue(CFP->getContext());
585 default:
586 break;
587 }
588 }
589 }
Chris Lattner210c5d42009-11-09 23:55:12 +0000590 }
591 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000592
Duncan Sands92826de2010-11-07 16:46:25 +0000593 // If the comparison is with the result of a select instruction, check whether
594 // comparing with either branch of the select always yields the same value.
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000595 if (MaxRecurse && (isa<SelectInst>(LHS) || isa<SelectInst>(RHS)))
Duncan Sands18450092010-11-16 12:16:38 +0000596 if (Value *V = ThreadCmpOverSelect(Pred, LHS, RHS, TD, DT, MaxRecurse-1))
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000597 return V;
598
599 // If the comparison is with the result of a phi instruction, check whether
600 // doing the compare with each incoming phi value yields a common result.
601 if (MaxRecurse && (isa<PHINode>(LHS) || isa<PHINode>(RHS)))
Duncan Sands18450092010-11-16 12:16:38 +0000602 if (Value *V = ThreadCmpOverPHI(Pred, LHS, RHS, TD, DT, MaxRecurse-1))
Duncan Sands3bbb0cc2010-11-09 17:25:51 +0000603 return V;
Duncan Sands92826de2010-11-07 16:46:25 +0000604
Chris Lattner9dbb4292009-11-09 23:28:39 +0000605 return 0;
606}
607
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000608Value *llvm::SimplifyFCmpInst(unsigned Predicate, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +0000609 const TargetData *TD, const DominatorTree *DT) {
610 return ::SimplifyFCmpInst(Predicate, LHS, RHS, TD, DT, RecursionLimit);
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000611}
612
Chris Lattner04754262010-04-20 05:32:14 +0000613/// SimplifySelectInst - Given operands for a SelectInst, see if we can fold
614/// the result. If not, this returns null.
615Value *llvm::SimplifySelectInst(Value *CondVal, Value *TrueVal, Value *FalseVal,
Duncan Sands18450092010-11-16 12:16:38 +0000616 const TargetData *TD, const DominatorTree *) {
Chris Lattner04754262010-04-20 05:32:14 +0000617 // select true, X, Y -> X
618 // select false, X, Y -> Y
619 if (ConstantInt *CB = dyn_cast<ConstantInt>(CondVal))
620 return CB->getZExtValue() ? TrueVal : FalseVal;
Duncan Sands12a86f52010-11-14 11:23:23 +0000621
Chris Lattner04754262010-04-20 05:32:14 +0000622 // select C, X, X -> X
623 if (TrueVal == FalseVal)
624 return TrueVal;
Duncan Sands12a86f52010-11-14 11:23:23 +0000625
Chris Lattner04754262010-04-20 05:32:14 +0000626 if (isa<UndefValue>(TrueVal)) // select C, undef, X -> X
627 return FalseVal;
628 if (isa<UndefValue>(FalseVal)) // select C, X, undef -> X
629 return TrueVal;
630 if (isa<UndefValue>(CondVal)) { // select undef, X, Y -> X or Y
631 if (isa<Constant>(TrueVal))
632 return TrueVal;
633 return FalseVal;
634 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000635
Chris Lattner04754262010-04-20 05:32:14 +0000636 return 0;
637}
638
Chris Lattnerc514c1f2009-11-27 00:29:05 +0000639/// SimplifyGEPInst - Given operands for an GetElementPtrInst, see if we can
640/// fold the result. If not, this returns null.
641Value *llvm::SimplifyGEPInst(Value *const *Ops, unsigned NumOps,
Duncan Sands18450092010-11-16 12:16:38 +0000642 const TargetData *TD, const DominatorTree *) {
Chris Lattnerc514c1f2009-11-27 00:29:05 +0000643 // getelementptr P -> P.
644 if (NumOps == 1)
645 return Ops[0];
646
647 // TODO.
648 //if (isa<UndefValue>(Ops[0]))
649 // return UndefValue::get(GEP.getType());
650
651 // getelementptr P, 0 -> P.
652 if (NumOps == 2)
653 if (ConstantInt *C = dyn_cast<ConstantInt>(Ops[1]))
654 if (C->isZero())
655 return Ops[0];
Duncan Sands12a86f52010-11-14 11:23:23 +0000656
Chris Lattnerc514c1f2009-11-27 00:29:05 +0000657 // Check to see if this is constant foldable.
658 for (unsigned i = 0; i != NumOps; ++i)
659 if (!isa<Constant>(Ops[i]))
660 return 0;
Duncan Sands12a86f52010-11-14 11:23:23 +0000661
Chris Lattnerc514c1f2009-11-27 00:29:05 +0000662 return ConstantExpr::getGetElementPtr(cast<Constant>(Ops[0]),
663 (Constant *const*)Ops+1, NumOps-1);
664}
665
Duncan Sandsff103412010-11-17 04:30:22 +0000666/// SimplifyPHINode - See if we can fold the given phi. If not, returns null.
667static Value *SimplifyPHINode(PHINode *PN, const DominatorTree *DT) {
668 // If all of the PHI's incoming values are the same then replace the PHI node
669 // with the common value.
670 Value *CommonValue = 0;
671 bool HasUndefInput = false;
672 for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) {
673 Value *Incoming = PN->getIncomingValue(i);
674 // If the incoming value is the phi node itself, it can safely be skipped.
675 if (Incoming == PN) continue;
676 if (isa<UndefValue>(Incoming)) {
677 // Remember that we saw an undef value, but otherwise ignore them.
678 HasUndefInput = true;
679 continue;
680 }
681 if (CommonValue && Incoming != CommonValue)
682 return 0; // Not the same, bail out.
683 CommonValue = Incoming;
684 }
685
686 // If CommonValue is null then all of the incoming values were either undef or
687 // equal to the phi node itself.
688 if (!CommonValue)
689 return UndefValue::get(PN->getType());
690
691 // If we have a PHI node like phi(X, undef, X), where X is defined by some
692 // instruction, we cannot return X as the result of the PHI node unless it
693 // dominates the PHI block.
694 if (HasUndefInput)
695 return ValueDominatesPHI(CommonValue, PN, DT) ? CommonValue : 0;
696
697 return CommonValue;
698}
699
Chris Lattnerc514c1f2009-11-27 00:29:05 +0000700
Chris Lattnerd06094f2009-11-10 00:55:12 +0000701//=== Helper functions for higher up the class hierarchy.
Chris Lattner9dbb4292009-11-09 23:28:39 +0000702
Chris Lattnerd06094f2009-11-10 00:55:12 +0000703/// SimplifyBinOp - Given operands for a BinaryOperator, see if we can
704/// fold the result. If not, this returns null.
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000705static Value *SimplifyBinOp(unsigned Opcode, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +0000706 const TargetData *TD, const DominatorTree *DT,
707 unsigned MaxRecurse) {
Chris Lattnerd06094f2009-11-10 00:55:12 +0000708 switch (Opcode) {
Duncan Sands18450092010-11-16 12:16:38 +0000709 case Instruction::And: return SimplifyAndInst(LHS, RHS, TD, DT, MaxRecurse);
710 case Instruction::Or: return SimplifyOrInst(LHS, RHS, TD, DT, MaxRecurse);
Chris Lattnerd06094f2009-11-10 00:55:12 +0000711 default:
712 if (Constant *CLHS = dyn_cast<Constant>(LHS))
713 if (Constant *CRHS = dyn_cast<Constant>(RHS)) {
714 Constant *COps[] = {CLHS, CRHS};
715 return ConstantFoldInstOperands(Opcode, LHS->getType(), COps, 2, TD);
716 }
Duncan Sandsb2cbdc32010-11-10 13:00:08 +0000717
718 // If the operation is with the result of a select instruction, check whether
719 // operating on either branch of the select always yields the same value.
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000720 if (MaxRecurse && (isa<SelectInst>(LHS) || isa<SelectInst>(RHS)))
Duncan Sands18450092010-11-16 12:16:38 +0000721 if (Value *V = ThreadBinOpOverSelect(Opcode, LHS, RHS, TD, DT,
722 MaxRecurse-1))
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000723 return V;
724
725 // If the operation is with the result of a phi instruction, check whether
726 // operating on all incoming values of the phi always yields the same value.
727 if (MaxRecurse && (isa<PHINode>(LHS) || isa<PHINode>(RHS)))
Duncan Sands18450092010-11-16 12:16:38 +0000728 if (Value *V = ThreadBinOpOverPHI(Opcode, LHS, RHS, TD, DT, MaxRecurse-1))
Duncan Sandsb2cbdc32010-11-10 13:00:08 +0000729 return V;
730
Chris Lattnerd06094f2009-11-10 00:55:12 +0000731 return 0;
732 }
733}
Chris Lattner9dbb4292009-11-09 23:28:39 +0000734
Duncan Sands12a86f52010-11-14 11:23:23 +0000735Value *llvm::SimplifyBinOp(unsigned Opcode, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +0000736 const TargetData *TD, const DominatorTree *DT) {
737 return ::SimplifyBinOp(Opcode, LHS, RHS, TD, DT, RecursionLimit);
Chris Lattner9dbb4292009-11-09 23:28:39 +0000738}
739
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000740/// SimplifyCmpInst - Given operands for a CmpInst, see if we can
741/// fold the result.
742static Value *SimplifyCmpInst(unsigned Predicate, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +0000743 const TargetData *TD, const DominatorTree *DT,
744 unsigned MaxRecurse) {
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000745 if (CmpInst::isIntPredicate((CmpInst::Predicate)Predicate))
Duncan Sands18450092010-11-16 12:16:38 +0000746 return SimplifyICmpInst(Predicate, LHS, RHS, TD, DT, MaxRecurse);
747 return SimplifyFCmpInst(Predicate, LHS, RHS, TD, DT, MaxRecurse);
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000748}
749
750Value *llvm::SimplifyCmpInst(unsigned Predicate, Value *LHS, Value *RHS,
Duncan Sands18450092010-11-16 12:16:38 +0000751 const TargetData *TD, const DominatorTree *DT) {
752 return ::SimplifyCmpInst(Predicate, LHS, RHS, TD, DT, RecursionLimit);
Duncan Sandsa74a58c2010-11-10 18:23:01 +0000753}
Chris Lattnere3453782009-11-10 01:08:51 +0000754
755/// SimplifyInstruction - See if we can compute a simplified version of this
756/// instruction. If not, this returns null.
Duncan Sandseff05812010-11-14 18:36:10 +0000757Value *llvm::SimplifyInstruction(Instruction *I, const TargetData *TD,
758 const DominatorTree *DT) {
Chris Lattnere3453782009-11-10 01:08:51 +0000759 switch (I->getOpcode()) {
760 default:
761 return ConstantFoldInstruction(I, TD);
Chris Lattner8aee8ef2009-11-27 17:42:22 +0000762 case Instruction::Add:
Owen Anderson4e282de2010-09-16 20:51:41 +0000763 return SimplifyAddInst(I->getOperand(0), I->getOperand(1),
764 cast<BinaryOperator>(I)->hasNoSignedWrap(),
Duncan Sands18450092010-11-16 12:16:38 +0000765 cast<BinaryOperator>(I)->hasNoUnsignedWrap(),
766 TD, DT);
Chris Lattnere3453782009-11-10 01:08:51 +0000767 case Instruction::And:
Duncan Sands18450092010-11-16 12:16:38 +0000768 return SimplifyAndInst(I->getOperand(0), I->getOperand(1), TD, DT);
Chris Lattnere3453782009-11-10 01:08:51 +0000769 case Instruction::Or:
Duncan Sands18450092010-11-16 12:16:38 +0000770 return SimplifyOrInst(I->getOperand(0), I->getOperand(1), TD, DT);
Chris Lattnere3453782009-11-10 01:08:51 +0000771 case Instruction::ICmp:
Owen Anderson4e282de2010-09-16 20:51:41 +0000772 return SimplifyICmpInst(cast<ICmpInst>(I)->getPredicate(),
Duncan Sands18450092010-11-16 12:16:38 +0000773 I->getOperand(0), I->getOperand(1), TD, DT);
Chris Lattnere3453782009-11-10 01:08:51 +0000774 case Instruction::FCmp:
Owen Anderson4e282de2010-09-16 20:51:41 +0000775 return SimplifyFCmpInst(cast<FCmpInst>(I)->getPredicate(),
Duncan Sands18450092010-11-16 12:16:38 +0000776 I->getOperand(0), I->getOperand(1), TD, DT);
Chris Lattner04754262010-04-20 05:32:14 +0000777 case Instruction::Select:
Owen Anderson4e282de2010-09-16 20:51:41 +0000778 return SimplifySelectInst(I->getOperand(0), I->getOperand(1),
Duncan Sands18450092010-11-16 12:16:38 +0000779 I->getOperand(2), TD, DT);
Chris Lattnerc514c1f2009-11-27 00:29:05 +0000780 case Instruction::GetElementPtr: {
781 SmallVector<Value*, 8> Ops(I->op_begin(), I->op_end());
Duncan Sands18450092010-11-16 12:16:38 +0000782 return SimplifyGEPInst(&Ops[0], Ops.size(), TD, DT);
Chris Lattnerc514c1f2009-11-27 00:29:05 +0000783 }
Duncan Sandscd6636c2010-11-14 13:30:18 +0000784 case Instruction::PHI:
Duncan Sandsff103412010-11-17 04:30:22 +0000785 return SimplifyPHINode(cast<PHINode>(I), DT);
Chris Lattnere3453782009-11-10 01:08:51 +0000786 }
787}
788
Chris Lattner40d8c282009-11-10 22:26:15 +0000789/// ReplaceAndSimplifyAllUses - Perform From->replaceAllUsesWith(To) and then
790/// delete the From instruction. In addition to a basic RAUW, this does a
791/// recursive simplification of the newly formed instructions. This catches
792/// things where one simplification exposes other opportunities. This only
793/// simplifies and deletes scalar operations, it does not change the CFG.
794///
795void llvm::ReplaceAndSimplifyAllUses(Instruction *From, Value *To,
Duncan Sandseff05812010-11-14 18:36:10 +0000796 const TargetData *TD,
797 const DominatorTree *DT) {
Chris Lattner40d8c282009-11-10 22:26:15 +0000798 assert(From != To && "ReplaceAndSimplifyAllUses(X,X) is not valid!");
Duncan Sands12a86f52010-11-14 11:23:23 +0000799
Chris Lattnerd2bfe542010-07-15 06:36:08 +0000800 // FromHandle/ToHandle - This keeps a WeakVH on the from/to values so that
801 // we can know if it gets deleted out from under us or replaced in a
802 // recursive simplification.
Chris Lattner40d8c282009-11-10 22:26:15 +0000803 WeakVH FromHandle(From);
Chris Lattnerd2bfe542010-07-15 06:36:08 +0000804 WeakVH ToHandle(To);
Duncan Sands12a86f52010-11-14 11:23:23 +0000805
Chris Lattner40d8c282009-11-10 22:26:15 +0000806 while (!From->use_empty()) {
807 // Update the instruction to use the new value.
Chris Lattnerd2bfe542010-07-15 06:36:08 +0000808 Use &TheUse = From->use_begin().getUse();
809 Instruction *User = cast<Instruction>(TheUse.getUser());
810 TheUse = To;
811
812 // Check to see if the instruction can be folded due to the operand
813 // replacement. For example changing (or X, Y) into (or X, -1) can replace
814 // the 'or' with -1.
815 Value *SimplifiedVal;
816 {
817 // Sanity check to make sure 'User' doesn't dangle across
818 // SimplifyInstruction.
819 AssertingVH<> UserHandle(User);
Duncan Sands12a86f52010-11-14 11:23:23 +0000820
Duncan Sandseff05812010-11-14 18:36:10 +0000821 SimplifiedVal = SimplifyInstruction(User, TD, DT);
Chris Lattnerd2bfe542010-07-15 06:36:08 +0000822 if (SimplifiedVal == 0) continue;
Chris Lattner40d8c282009-11-10 22:26:15 +0000823 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000824
Chris Lattnerd2bfe542010-07-15 06:36:08 +0000825 // Recursively simplify this user to the new value.
Duncan Sandseff05812010-11-14 18:36:10 +0000826 ReplaceAndSimplifyAllUses(User, SimplifiedVal, TD, DT);
Chris Lattnerd2bfe542010-07-15 06:36:08 +0000827 From = dyn_cast_or_null<Instruction>((Value*)FromHandle);
828 To = ToHandle;
Duncan Sands12a86f52010-11-14 11:23:23 +0000829
Chris Lattnerd2bfe542010-07-15 06:36:08 +0000830 assert(ToHandle && "To value deleted by recursive simplification?");
Duncan Sands12a86f52010-11-14 11:23:23 +0000831
Chris Lattnerd2bfe542010-07-15 06:36:08 +0000832 // If the recursive simplification ended up revisiting and deleting
833 // 'From' then we're done.
834 if (From == 0)
835 return;
Chris Lattner40d8c282009-11-10 22:26:15 +0000836 }
Duncan Sands12a86f52010-11-14 11:23:23 +0000837
Chris Lattnerd2bfe542010-07-15 06:36:08 +0000838 // If 'From' has value handles referring to it, do a real RAUW to update them.
839 From->replaceAllUsesWith(To);
Duncan Sands12a86f52010-11-14 11:23:23 +0000840
Chris Lattner40d8c282009-11-10 22:26:15 +0000841 From->eraseFromParent();
842}