blob: 376222ffe2786b455539f8933e922e45f246c2cf [file] [log] [blame]
Chris Lattnere6794492002-08-12 21:17:25 +00001//===- InstructionCombining.cpp - Combine multiple instructions -----------===//
John Criswell482202a2003-10-20 19:43:21 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by the LLVM research group and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
Chris Lattnerca081252001-12-14 16:52:21 +00009//
10// InstructionCombining - Combine instructions to form fewer, simple
Chris Lattner99f48c62002-09-02 04:59:56 +000011// instructions. This pass does not modify the CFG This pass is where algebraic
12// simplification happens.
Chris Lattnerca081252001-12-14 16:52:21 +000013//
14// This pass combines things like:
Chris Lattnerdd1a86d2004-05-04 15:19:33 +000015// %Y = add int %X, 1
16// %Z = add int %Y, 1
Chris Lattnerca081252001-12-14 16:52:21 +000017// into:
Chris Lattnerdd1a86d2004-05-04 15:19:33 +000018// %Z = add int %X, 2
Chris Lattnerca081252001-12-14 16:52:21 +000019//
20// This is a simple worklist driven algorithm.
21//
Chris Lattner216c7b82003-09-10 05:29:43 +000022// This pass guarantees that the following canonicalizations are performed on
Chris Lattnerbfb1d032003-07-23 21:41:57 +000023// the program:
24// 1. If a binary operator has a constant operand, it is moved to the RHS
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +000025// 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.
Chris Lattnerbfb1d032003-07-23 21:41:57 +000027// 3. SetCC instructions are converted from <,>,<=,>= to ==,!= if possible
28// 4. All SetCC instructions on boolean values are replaced with logical ops
Chris Lattnerede3fe02003-08-13 04:18:28 +000029// 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.
Chris Lattnerbfb1d032003-07-23 21:41:57 +000032// N. This list is incomplete
33//
Chris Lattnerca081252001-12-14 16:52:21 +000034//===----------------------------------------------------------------------===//
35
Chris Lattner7d2a5392004-03-13 23:54:27 +000036#define DEBUG_TYPE "instcombine"
Chris Lattnerb4cfa7f2002-05-07 20:03:00 +000037#include "llvm/Transforms/Scalar.h"
Chris Lattner471bd762003-05-22 19:07:21 +000038#include "llvm/Instructions.h"
Chris Lattner51ea1272004-02-28 05:22:00 +000039#include "llvm/Intrinsics.h"
Chris Lattner04805fa2002-02-26 21:46:54 +000040#include "llvm/Pass.h"
Chris Lattner34428442003-05-27 16:40:51 +000041#include "llvm/Constants.h"
Chris Lattner1085bdf2002-11-04 16:18:53 +000042#include "llvm/DerivedTypes.h"
Chris Lattner0f1d8a32003-06-26 05:06:25 +000043#include "llvm/GlobalVariable.h"
Chris Lattnerf4ad1652003-11-02 05:57:39 +000044#include "llvm/Target/TargetData.h"
45#include "llvm/Transforms/Utils/BasicBlockUtils.h"
46#include "llvm/Transforms/Utils/Local.h"
Chris Lattner69193f92004-04-05 01:30:19 +000047#include "llvm/Support/CallSite.h"
48#include "llvm/Support/GetElementPtrTypeIterator.h"
Chris Lattner60a65912002-02-12 21:07:25 +000049#include "llvm/Support/InstIterator.h"
Chris Lattner260ab202002-04-18 17:39:14 +000050#include "llvm/Support/InstVisitor.h"
Chris Lattner7d2a5392004-03-13 23:54:27 +000051#include "Support/Debug.h"
Chris Lattnerbf3a0992002-10-01 22:38:41 +000052#include "Support/Statistic.h"
Chris Lattner053c0932002-05-14 15:24:07 +000053#include <algorithm>
Chris Lattner8427bff2003-12-07 01:24:23 +000054using namespace llvm;
Brian Gaeke960707c2003-11-11 22:41:34 +000055
Chris Lattner260ab202002-04-18 17:39:14 +000056namespace {
Chris Lattnerbf3a0992002-10-01 22:38:41 +000057 Statistic<> NumCombined ("instcombine", "Number of insts combined");
58 Statistic<> NumConstProp("instcombine", "Number of constant folds");
59 Statistic<> NumDeadInst ("instcombine", "Number of dead inst eliminated");
60
Chris Lattnerc8e66542002-04-27 06:56:12 +000061 class InstCombiner : public FunctionPass,
Chris Lattner260ab202002-04-18 17:39:14 +000062 public InstVisitor<InstCombiner, Instruction*> {
63 // Worklist of all of the instructions that need to be simplified.
64 std::vector<Instruction*> WorkList;
Chris Lattnerf4ad1652003-11-02 05:57:39 +000065 TargetData *TD;
Chris Lattner260ab202002-04-18 17:39:14 +000066
Chris Lattner51ea1272004-02-28 05:22:00 +000067 /// AddUsersToWorkList - When an instruction is simplified, add all users of
68 /// the instruction to the work lists because they might get more simplified
69 /// now.
70 ///
71 void AddUsersToWorkList(Instruction &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +000072 for (Value::use_iterator UI = I.use_begin(), UE = I.use_end();
Chris Lattner260ab202002-04-18 17:39:14 +000073 UI != UE; ++UI)
74 WorkList.push_back(cast<Instruction>(*UI));
75 }
76
Chris Lattner51ea1272004-02-28 05:22:00 +000077 /// AddUsesToWorkList - When an instruction is simplified, add operands to
78 /// the work lists because they might get more simplified now.
79 ///
80 void AddUsesToWorkList(Instruction &I) {
81 for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i)
82 if (Instruction *Op = dyn_cast<Instruction>(I.getOperand(i)))
83 WorkList.push_back(Op);
84 }
85
Chris Lattner99f48c62002-09-02 04:59:56 +000086 // removeFromWorkList - remove all instances of I from the worklist.
87 void removeFromWorkList(Instruction *I);
Chris Lattner260ab202002-04-18 17:39:14 +000088 public:
Chris Lattner113f4f42002-06-25 16:13:24 +000089 virtual bool runOnFunction(Function &F);
Chris Lattner260ab202002-04-18 17:39:14 +000090
Chris Lattnerf12cc842002-04-28 21:27:06 +000091 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Chris Lattnerf4ad1652003-11-02 05:57:39 +000092 AU.addRequired<TargetData>();
Chris Lattner820d9712002-10-21 20:00:28 +000093 AU.setPreservesCFG();
Chris Lattnerf12cc842002-04-28 21:27:06 +000094 }
95
Chris Lattner69193f92004-04-05 01:30:19 +000096 TargetData &getTargetData() const { return *TD; }
97
Chris Lattner260ab202002-04-18 17:39:14 +000098 // Visitation implementation - Implement instruction combining for different
99 // instruction types. The semantics are as follows:
100 // Return Value:
101 // null - No change was made
Chris Lattnere6794492002-08-12 21:17:25 +0000102 // I - Change was made, I is still valid, I may be dead though
Chris Lattner260ab202002-04-18 17:39:14 +0000103 // otherwise - Change was made, replace I with returned instruction
104 //
Chris Lattner113f4f42002-06-25 16:13:24 +0000105 Instruction *visitAdd(BinaryOperator &I);
106 Instruction *visitSub(BinaryOperator &I);
107 Instruction *visitMul(BinaryOperator &I);
108 Instruction *visitDiv(BinaryOperator &I);
109 Instruction *visitRem(BinaryOperator &I);
110 Instruction *visitAnd(BinaryOperator &I);
111 Instruction *visitOr (BinaryOperator &I);
112 Instruction *visitXor(BinaryOperator &I);
113 Instruction *visitSetCondInst(BinaryOperator &I);
Chris Lattnere8d6c602003-03-10 19:16:08 +0000114 Instruction *visitShiftInst(ShiftInst &I);
Chris Lattner113f4f42002-06-25 16:13:24 +0000115 Instruction *visitCastInst(CastInst &CI);
Chris Lattnerb909e8b2004-03-12 05:52:32 +0000116 Instruction *visitSelectInst(SelectInst &CI);
Chris Lattner970c33a2003-06-19 17:00:31 +0000117 Instruction *visitCallInst(CallInst &CI);
118 Instruction *visitInvokeInst(InvokeInst &II);
Chris Lattner113f4f42002-06-25 16:13:24 +0000119 Instruction *visitPHINode(PHINode &PN);
120 Instruction *visitGetElementPtrInst(GetElementPtrInst &GEP);
Chris Lattner1085bdf2002-11-04 16:18:53 +0000121 Instruction *visitAllocationInst(AllocationInst &AI);
Chris Lattner8427bff2003-12-07 01:24:23 +0000122 Instruction *visitFreeInst(FreeInst &FI);
Chris Lattner0f1d8a32003-06-26 05:06:25 +0000123 Instruction *visitLoadInst(LoadInst &LI);
Chris Lattner9eef8a72003-06-04 04:46:00 +0000124 Instruction *visitBranchInst(BranchInst &BI);
Chris Lattner4c9c20a2004-07-03 00:26:11 +0000125 Instruction *visitSwitchInst(SwitchInst &SI);
Chris Lattner260ab202002-04-18 17:39:14 +0000126
127 // visitInstruction - Specify what to return for unhandled instructions...
Chris Lattner113f4f42002-06-25 16:13:24 +0000128 Instruction *visitInstruction(Instruction &I) { return 0; }
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000129
Chris Lattner970c33a2003-06-19 17:00:31 +0000130 private:
Chris Lattneraec3d942003-10-07 22:32:43 +0000131 Instruction *visitCallSite(CallSite CS);
Chris Lattner970c33a2003-06-19 17:00:31 +0000132 bool transformConstExprCastCall(CallSite CS);
133
Chris Lattner69193f92004-04-05 01:30:19 +0000134 public:
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000135 // InsertNewInstBefore - insert an instruction New before instruction Old
136 // in the program. Add the new instruction to the worklist.
137 //
Chris Lattnere79e8542004-02-23 06:38:22 +0000138 Value *InsertNewInstBefore(Instruction *New, Instruction &Old) {
Chris Lattner65217ff2002-08-23 18:32:43 +0000139 assert(New && New->getParent() == 0 &&
140 "New instruction already inserted into a basic block!");
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000141 BasicBlock *BB = Old.getParent();
142 BB->getInstList().insert(&Old, New); // Insert inst
143 WorkList.push_back(New); // Add to worklist
Chris Lattnere79e8542004-02-23 06:38:22 +0000144 return New;
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000145 }
146
147 // ReplaceInstUsesWith - This method is to be used when an instruction is
148 // found to be dead, replacable with another preexisting expression. Here
149 // we add all uses of I to the worklist, replace all uses of I with the new
150 // value, then return I, so that the inst combiner will know that I was
151 // modified.
152 //
153 Instruction *ReplaceInstUsesWith(Instruction &I, Value *V) {
Chris Lattner51ea1272004-02-28 05:22:00 +0000154 AddUsersToWorkList(I); // Add all modified instrs to worklist
Chris Lattner8953b902004-04-05 02:10:19 +0000155 if (&I != V) {
156 I.replaceAllUsesWith(V);
157 return &I;
158 } else {
159 // If we are replacing the instruction with itself, this must be in a
160 // segment of unreachable code, so just clobber the instruction.
161 I.replaceAllUsesWith(Constant::getNullValue(I.getType()));
162 return &I;
163 }
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000164 }
Chris Lattner51ea1272004-02-28 05:22:00 +0000165
166 // EraseInstFromFunction - When dealing with an instruction that has side
167 // effects or produces a void value, we can't rely on DCE to delete the
168 // instruction. Instead, visit methods should return the value returned by
169 // this function.
170 Instruction *EraseInstFromFunction(Instruction &I) {
171 assert(I.use_empty() && "Cannot erase instruction that is used!");
172 AddUsesToWorkList(I);
173 removeFromWorkList(&I);
174 I.getParent()->getInstList().erase(&I);
175 return 0; // Don't do anything with FI
176 }
177
178
Chris Lattner3ac7c262003-08-13 20:16:26 +0000179 private:
Chris Lattnerdfae8be2003-07-24 17:35:25 +0000180 /// InsertOperandCastBefore - This inserts a cast of V to DestTy before the
181 /// InsertBefore instruction. This is specialized a bit to avoid inserting
182 /// casts that are known to not do anything...
183 ///
184 Value *InsertOperandCastBefore(Value *V, const Type *DestTy,
185 Instruction *InsertBefore);
186
Chris Lattner7fb29e12003-03-11 00:12:48 +0000187 // SimplifyCommutative - This performs a few simplifications for commutative
188 // operators...
189 bool SimplifyCommutative(BinaryOperator &I);
Chris Lattnerba1cb382003-09-19 17:17:26 +0000190
191 Instruction *OptAndOp(Instruction *Op, ConstantIntegral *OpRHS,
192 ConstantIntegral *AndRHS, BinaryOperator &TheAnd);
Chris Lattner260ab202002-04-18 17:39:14 +0000193 };
Chris Lattnerb28b6802002-07-23 18:06:35 +0000194
Chris Lattnerc8b70922002-07-26 21:12:46 +0000195 RegisterOpt<InstCombiner> X("instcombine", "Combine redundant instructions");
Chris Lattner260ab202002-04-18 17:39:14 +0000196}
197
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000198// getComplexity: Assign a complexity or rank value to LLVM Values...
199// 0 -> Constant, 1 -> Other, 2 -> Argument, 2 -> Unary, 3 -> OtherInst
200static unsigned getComplexity(Value *V) {
201 if (isa<Instruction>(V)) {
202 if (BinaryOperator::isNeg(V) || BinaryOperator::isNot(V))
203 return 2;
204 return 3;
205 }
206 if (isa<Argument>(V)) return 2;
207 return isa<Constant>(V) ? 0 : 1;
208}
Chris Lattner260ab202002-04-18 17:39:14 +0000209
Chris Lattner7fb29e12003-03-11 00:12:48 +0000210// isOnlyUse - Return true if this instruction will be deleted if we stop using
211// it.
212static bool isOnlyUse(Value *V) {
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000213 return V->hasOneUse() || isa<Constant>(V);
Chris Lattner7fb29e12003-03-11 00:12:48 +0000214}
215
Chris Lattnere79e8542004-02-23 06:38:22 +0000216// getPromotedType - Return the specified type promoted as it would be to pass
217// though a va_arg area...
218static const Type *getPromotedType(const Type *Ty) {
Chris Lattner97bfcea2004-06-17 18:16:02 +0000219 switch (Ty->getTypeID()) {
Chris Lattnere79e8542004-02-23 06:38:22 +0000220 case Type::SByteTyID:
221 case Type::ShortTyID: return Type::IntTy;
222 case Type::UByteTyID:
223 case Type::UShortTyID: return Type::UIntTy;
224 case Type::FloatTyID: return Type::DoubleTy;
225 default: return Ty;
226 }
227}
228
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000229// SimplifyCommutative - This performs a few simplifications for commutative
230// operators:
Chris Lattner260ab202002-04-18 17:39:14 +0000231//
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000232// 1. Order operands such that they are listed from right (least complex) to
233// left (most complex). This puts constants before unary operators before
234// binary operators.
235//
Chris Lattner7fb29e12003-03-11 00:12:48 +0000236// 2. Transform: (op (op V, C1), C2) ==> (op V, (op C1, C2))
237// 3. Transform: (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2))
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000238//
Chris Lattner7fb29e12003-03-11 00:12:48 +0000239bool InstCombiner::SimplifyCommutative(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000240 bool Changed = false;
241 if (getComplexity(I.getOperand(0)) < getComplexity(I.getOperand(1)))
242 Changed = !I.swapOperands();
243
244 if (!I.isAssociative()) return Changed;
245 Instruction::BinaryOps Opcode = I.getOpcode();
Chris Lattner7fb29e12003-03-11 00:12:48 +0000246 if (BinaryOperator *Op = dyn_cast<BinaryOperator>(I.getOperand(0)))
247 if (Op->getOpcode() == Opcode && isa<Constant>(Op->getOperand(1))) {
248 if (isa<Constant>(I.getOperand(1))) {
Chris Lattner34428442003-05-27 16:40:51 +0000249 Constant *Folded = ConstantExpr::get(I.getOpcode(),
250 cast<Constant>(I.getOperand(1)),
251 cast<Constant>(Op->getOperand(1)));
Chris Lattner7fb29e12003-03-11 00:12:48 +0000252 I.setOperand(0, Op->getOperand(0));
253 I.setOperand(1, Folded);
254 return true;
255 } else if (BinaryOperator *Op1=dyn_cast<BinaryOperator>(I.getOperand(1)))
256 if (Op1->getOpcode() == Opcode && isa<Constant>(Op1->getOperand(1)) &&
257 isOnlyUse(Op) && isOnlyUse(Op1)) {
258 Constant *C1 = cast<Constant>(Op->getOperand(1));
259 Constant *C2 = cast<Constant>(Op1->getOperand(1));
260
261 // Fold (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2))
Chris Lattner34428442003-05-27 16:40:51 +0000262 Constant *Folded = ConstantExpr::get(I.getOpcode(), C1, C2);
Chris Lattner7fb29e12003-03-11 00:12:48 +0000263 Instruction *New = BinaryOperator::create(Opcode, Op->getOperand(0),
264 Op1->getOperand(0),
265 Op1->getName(), &I);
266 WorkList.push_back(New);
267 I.setOperand(0, New);
268 I.setOperand(1, Folded);
269 return true;
270 }
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000271 }
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000272 return Changed;
Chris Lattner260ab202002-04-18 17:39:14 +0000273}
Chris Lattnerca081252001-12-14 16:52:21 +0000274
Chris Lattnerbb74e222003-03-10 23:06:50 +0000275// dyn_castNegVal - Given a 'sub' instruction, return the RHS of the instruction
276// if the LHS is a constant zero (which is the 'negate' form).
Chris Lattner9fa53de2002-05-06 16:49:18 +0000277//
Chris Lattnerbb74e222003-03-10 23:06:50 +0000278static inline Value *dyn_castNegVal(Value *V) {
279 if (BinaryOperator::isNeg(V))
280 return BinaryOperator::getNegArgument(cast<BinaryOperator>(V));
281
Chris Lattner9244df62003-04-30 22:19:10 +0000282 // Constants can be considered to be negated values if they can be folded...
283 if (Constant *C = dyn_cast<Constant>(V))
Chris Lattnerc8e7e292004-06-10 02:12:35 +0000284 return ConstantExpr::getNeg(C);
Chris Lattnerbb74e222003-03-10 23:06:50 +0000285 return 0;
Chris Lattner9fa53de2002-05-06 16:49:18 +0000286}
287
Chris Lattnerbb74e222003-03-10 23:06:50 +0000288static inline Value *dyn_castNotVal(Value *V) {
289 if (BinaryOperator::isNot(V))
290 return BinaryOperator::getNotArgument(cast<BinaryOperator>(V));
291
292 // Constants can be considered to be not'ed values...
Chris Lattnerdd65d862003-04-30 22:34:06 +0000293 if (ConstantIntegral *C = dyn_cast<ConstantIntegral>(V))
Chris Lattnerc8e7e292004-06-10 02:12:35 +0000294 return ConstantExpr::getNot(C);
Chris Lattnerbb74e222003-03-10 23:06:50 +0000295 return 0;
296}
297
Chris Lattner7fb29e12003-03-11 00:12:48 +0000298// dyn_castFoldableMul - If this value is a multiply that can be folded into
299// other computations (because it has a constant operand), return the
300// non-constant operand of the multiply.
301//
302static inline Value *dyn_castFoldableMul(Value *V) {
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000303 if (V->hasOneUse() && V->getType()->isInteger())
Chris Lattner7fb29e12003-03-11 00:12:48 +0000304 if (Instruction *I = dyn_cast<Instruction>(V))
305 if (I->getOpcode() == Instruction::Mul)
306 if (isa<Constant>(I->getOperand(1)))
307 return I->getOperand(0);
308 return 0;
Chris Lattner3082c5a2003-02-18 19:28:33 +0000309}
Chris Lattner31ae8632002-08-14 17:51:49 +0000310
Chris Lattner7fb29e12003-03-11 00:12:48 +0000311// dyn_castMaskingAnd - If this value is an And instruction masking a value with
312// a constant, return the constant being anded with.
313//
Chris Lattner01d56392003-08-12 19:17:27 +0000314template<class ValueType>
315static inline Constant *dyn_castMaskingAnd(ValueType *V) {
Chris Lattner7fb29e12003-03-11 00:12:48 +0000316 if (Instruction *I = dyn_cast<Instruction>(V))
317 if (I->getOpcode() == Instruction::And)
318 return dyn_cast<Constant>(I->getOperand(1));
319
320 // If this is a constant, it acts just like we were masking with it.
321 return dyn_cast<Constant>(V);
322}
Chris Lattner3082c5a2003-02-18 19:28:33 +0000323
324// Log2 - Calculate the log base 2 for the specified value if it is exactly a
325// power of 2.
326static unsigned Log2(uint64_t Val) {
327 assert(Val > 1 && "Values 0 and 1 should be handled elsewhere!");
328 unsigned Count = 0;
329 while (Val != 1) {
330 if (Val & 1) return 0; // Multiple bits set?
331 Val >>= 1;
332 ++Count;
333 }
334 return Count;
Chris Lattner31ae8632002-08-14 17:51:49 +0000335}
336
Chris Lattnerb8b97502003-08-13 19:01:45 +0000337
338/// AssociativeOpt - Perform an optimization on an associative operator. This
339/// function is designed to check a chain of associative operators for a
340/// potential to apply a certain optimization. Since the optimization may be
341/// applicable if the expression was reassociated, this checks the chain, then
342/// reassociates the expression as necessary to expose the optimization
343/// opportunity. This makes use of a special Functor, which must define
344/// 'shouldApply' and 'apply' methods.
345///
346template<typename Functor>
347Instruction *AssociativeOpt(BinaryOperator &Root, const Functor &F) {
348 unsigned Opcode = Root.getOpcode();
349 Value *LHS = Root.getOperand(0);
350
351 // Quick check, see if the immediate LHS matches...
352 if (F.shouldApply(LHS))
353 return F.apply(Root);
354
355 // Otherwise, if the LHS is not of the same opcode as the root, return.
356 Instruction *LHSI = dyn_cast<Instruction>(LHS);
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000357 while (LHSI && LHSI->getOpcode() == Opcode && LHSI->hasOneUse()) {
Chris Lattnerb8b97502003-08-13 19:01:45 +0000358 // Should we apply this transform to the RHS?
359 bool ShouldApply = F.shouldApply(LHSI->getOperand(1));
360
361 // If not to the RHS, check to see if we should apply to the LHS...
362 if (!ShouldApply && F.shouldApply(LHSI->getOperand(0))) {
363 cast<BinaryOperator>(LHSI)->swapOperands(); // Make the LHS the RHS
364 ShouldApply = true;
365 }
366
367 // If the functor wants to apply the optimization to the RHS of LHSI,
368 // reassociate the expression from ((? op A) op B) to (? op (A op B))
369 if (ShouldApply) {
370 BasicBlock *BB = Root.getParent();
Chris Lattnerb8b97502003-08-13 19:01:45 +0000371
372 // Now all of the instructions are in the current basic block, go ahead
373 // and perform the reassociation.
374 Instruction *TmpLHSI = cast<Instruction>(Root.getOperand(0));
375
376 // First move the selected RHS to the LHS of the root...
377 Root.setOperand(0, LHSI->getOperand(1));
378
379 // Make what used to be the LHS of the root be the user of the root...
380 Value *ExtraOperand = TmpLHSI->getOperand(1);
Chris Lattner284d3b02004-04-16 18:08:07 +0000381 if (&Root == TmpLHSI) {
Chris Lattner8953b902004-04-05 02:10:19 +0000382 Root.replaceAllUsesWith(Constant::getNullValue(TmpLHSI->getType()));
383 return 0;
384 }
Chris Lattner284d3b02004-04-16 18:08:07 +0000385 Root.replaceAllUsesWith(TmpLHSI); // Users now use TmpLHSI
Chris Lattnerb8b97502003-08-13 19:01:45 +0000386 TmpLHSI->setOperand(1, &Root); // TmpLHSI now uses the root
Chris Lattner284d3b02004-04-16 18:08:07 +0000387 TmpLHSI->getParent()->getInstList().remove(TmpLHSI);
388 BasicBlock::iterator ARI = &Root; ++ARI;
389 BB->getInstList().insert(ARI, TmpLHSI); // Move TmpLHSI to after Root
390 ARI = Root;
Chris Lattnerb8b97502003-08-13 19:01:45 +0000391
392 // Now propagate the ExtraOperand down the chain of instructions until we
393 // get to LHSI.
394 while (TmpLHSI != LHSI) {
395 Instruction *NextLHSI = cast<Instruction>(TmpLHSI->getOperand(0));
Chris Lattner284d3b02004-04-16 18:08:07 +0000396 // Move the instruction to immediately before the chain we are
397 // constructing to avoid breaking dominance properties.
398 NextLHSI->getParent()->getInstList().remove(NextLHSI);
399 BB->getInstList().insert(ARI, NextLHSI);
400 ARI = NextLHSI;
401
Chris Lattnerb8b97502003-08-13 19:01:45 +0000402 Value *NextOp = NextLHSI->getOperand(1);
403 NextLHSI->setOperand(1, ExtraOperand);
404 TmpLHSI = NextLHSI;
405 ExtraOperand = NextOp;
406 }
407
408 // Now that the instructions are reassociated, have the functor perform
409 // the transformation...
410 return F.apply(Root);
411 }
412
413 LHSI = dyn_cast<Instruction>(LHSI->getOperand(0));
414 }
415 return 0;
416}
417
418
419// AddRHS - Implements: X + X --> X << 1
420struct AddRHS {
421 Value *RHS;
422 AddRHS(Value *rhs) : RHS(rhs) {}
423 bool shouldApply(Value *LHS) const { return LHS == RHS; }
424 Instruction *apply(BinaryOperator &Add) const {
425 return new ShiftInst(Instruction::Shl, Add.getOperand(0),
426 ConstantInt::get(Type::UByteTy, 1));
427 }
428};
429
430// AddMaskingAnd - Implements (A & C1)+(B & C2) --> (A & C1)|(B & C2)
431// iff C1&C2 == 0
432struct AddMaskingAnd {
433 Constant *C2;
434 AddMaskingAnd(Constant *c) : C2(c) {}
435 bool shouldApply(Value *LHS) const {
436 if (Constant *C1 = dyn_castMaskingAnd(LHS))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000437 return ConstantExpr::getAnd(C1, C2)->isNullValue();
Chris Lattnerb8b97502003-08-13 19:01:45 +0000438 return false;
439 }
440 Instruction *apply(BinaryOperator &Add) const {
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000441 return BinaryOperator::createOr(Add.getOperand(0), Add.getOperand(1));
Chris Lattnerb8b97502003-08-13 19:01:45 +0000442 }
443};
444
Chris Lattner183b3362004-04-09 19:05:30 +0000445static Value *FoldOperationIntoSelectOperand(Instruction &BI, Value *SO,
446 InstCombiner *IC) {
447 // Figure out if the constant is the left or the right argument.
448 bool ConstIsRHS = isa<Constant>(BI.getOperand(1));
449 Constant *ConstOperand = cast<Constant>(BI.getOperand(ConstIsRHS));
Chris Lattnerb8b97502003-08-13 19:01:45 +0000450
Chris Lattner183b3362004-04-09 19:05:30 +0000451 if (Constant *SOC = dyn_cast<Constant>(SO)) {
452 if (ConstIsRHS)
453 return ConstantExpr::get(BI.getOpcode(), SOC, ConstOperand);
454 return ConstantExpr::get(BI.getOpcode(), ConstOperand, SOC);
455 }
456
457 Value *Op0 = SO, *Op1 = ConstOperand;
458 if (!ConstIsRHS)
459 std::swap(Op0, Op1);
460 Instruction *New;
461 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(&BI))
462 New = BinaryOperator::create(BO->getOpcode(), Op0, Op1);
463 else if (ShiftInst *SI = dyn_cast<ShiftInst>(&BI))
464 New = new ShiftInst(SI->getOpcode(), Op0, Op1);
Chris Lattnerf9d96652004-04-10 19:15:56 +0000465 else {
Chris Lattner183b3362004-04-09 19:05:30 +0000466 assert(0 && "Unknown binary instruction type!");
Chris Lattnerf9d96652004-04-10 19:15:56 +0000467 abort();
468 }
Chris Lattner183b3362004-04-09 19:05:30 +0000469 return IC->InsertNewInstBefore(New, BI);
470}
471
472// FoldBinOpIntoSelect - Given an instruction with a select as one operand and a
473// constant as the other operand, try to fold the binary operator into the
474// select arguments.
475static Instruction *FoldBinOpIntoSelect(Instruction &BI, SelectInst *SI,
476 InstCombiner *IC) {
477 // Don't modify shared select instructions
478 if (!SI->hasOneUse()) return 0;
479 Value *TV = SI->getOperand(1);
480 Value *FV = SI->getOperand(2);
481
482 if (isa<Constant>(TV) || isa<Constant>(FV)) {
483 Value *SelectTrueVal = FoldOperationIntoSelectOperand(BI, TV, IC);
484 Value *SelectFalseVal = FoldOperationIntoSelectOperand(BI, FV, IC);
485
486 return new SelectInst(SI->getCondition(), SelectTrueVal,
487 SelectFalseVal);
488 }
489 return 0;
490}
Chris Lattnerb8b97502003-08-13 19:01:45 +0000491
Chris Lattner113f4f42002-06-25 16:13:24 +0000492Instruction *InstCombiner::visitAdd(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000493 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +0000494 Value *LHS = I.getOperand(0), *RHS = I.getOperand(1);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000495
Chris Lattnercf4a9962004-04-10 22:01:55 +0000496 if (Constant *RHSC = dyn_cast<Constant>(RHS)) {
497 // X + 0 --> X
498 if (!I.getType()->isFloatingPoint() && // -0 + +0 = +0, so it's not a noop
499 RHSC->isNullValue())
500 return ReplaceInstUsesWith(I, LHS);
501
502 // X + (signbit) --> X ^ signbit
503 if (ConstantInt *CI = dyn_cast<ConstantInt>(RHSC)) {
504 unsigned NumBits = CI->getType()->getPrimitiveSize()*8;
505 uint64_t Val = CI->getRawValue() & (1ULL << NumBits)-1;
506 if (Val == (1ULL << NumBits-1))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000507 return BinaryOperator::createXor(LHS, RHS);
Chris Lattnercf4a9962004-04-10 22:01:55 +0000508 }
509 }
Chris Lattner9fa53de2002-05-06 16:49:18 +0000510
Chris Lattnerb8b97502003-08-13 19:01:45 +0000511 // X + X --> X << 1
512 if (I.getType()->isInteger())
513 if (Instruction *Result = AssociativeOpt(I, AddRHS(RHS))) return Result;
Chris Lattnerede3fe02003-08-13 04:18:28 +0000514
Chris Lattner147e9752002-05-08 22:46:53 +0000515 // -A + B --> B - A
Chris Lattnerbb74e222003-03-10 23:06:50 +0000516 if (Value *V = dyn_castNegVal(LHS))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000517 return BinaryOperator::createSub(RHS, V);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000518
519 // A + -B --> A - B
Chris Lattnerbb74e222003-03-10 23:06:50 +0000520 if (!isa<Constant>(RHS))
521 if (Value *V = dyn_castNegVal(RHS))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000522 return BinaryOperator::createSub(LHS, V);
Chris Lattner260ab202002-04-18 17:39:14 +0000523
Chris Lattner57c8d992003-02-18 19:57:07 +0000524 // X*C + X --> X * (C+1)
525 if (dyn_castFoldableMul(LHS) == RHS) {
Chris Lattner34428442003-05-27 16:40:51 +0000526 Constant *CP1 =
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000527 ConstantExpr::getAdd(
Chris Lattner34428442003-05-27 16:40:51 +0000528 cast<Constant>(cast<Instruction>(LHS)->getOperand(1)),
529 ConstantInt::get(I.getType(), 1));
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000530 return BinaryOperator::createMul(RHS, CP1);
Chris Lattner57c8d992003-02-18 19:57:07 +0000531 }
532
533 // X + X*C --> X * (C+1)
534 if (dyn_castFoldableMul(RHS) == LHS) {
Chris Lattner34428442003-05-27 16:40:51 +0000535 Constant *CP1 =
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000536 ConstantExpr::getAdd(
Chris Lattner34428442003-05-27 16:40:51 +0000537 cast<Constant>(cast<Instruction>(RHS)->getOperand(1)),
538 ConstantInt::get(I.getType(), 1));
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000539 return BinaryOperator::createMul(LHS, CP1);
Chris Lattner57c8d992003-02-18 19:57:07 +0000540 }
541
Chris Lattnerb8b97502003-08-13 19:01:45 +0000542 // (A & C1)+(B & C2) --> (A & C1)|(B & C2) iff C1&C2 == 0
543 if (Constant *C2 = dyn_castMaskingAnd(RHS))
544 if (Instruction *R = AssociativeOpt(I, AddMaskingAnd(C2))) return R;
Chris Lattner7fb29e12003-03-11 00:12:48 +0000545
Chris Lattnerb9cde762003-10-02 15:11:26 +0000546 if (ConstantInt *CRHS = dyn_cast<ConstantInt>(RHS)) {
547 if (Instruction *ILHS = dyn_cast<Instruction>(LHS)) {
548 switch (ILHS->getOpcode()) {
549 case Instruction::Xor:
550 // ~X + C --> (C-1) - X
551 if (ConstantInt *XorRHS = dyn_cast<ConstantInt>(ILHS->getOperand(1)))
552 if (XorRHS->isAllOnesValue())
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000553 return BinaryOperator::createSub(ConstantExpr::getSub(CRHS,
554 ConstantInt::get(I.getType(), 1)),
Chris Lattnerb9cde762003-10-02 15:11:26 +0000555 ILHS->getOperand(0));
556 break;
Chris Lattner183b3362004-04-09 19:05:30 +0000557 case Instruction::Select:
558 // Try to fold constant add into select arguments.
559 if (Instruction *R = FoldBinOpIntoSelect(I,cast<SelectInst>(ILHS),this))
560 return R;
561
Chris Lattnerb9cde762003-10-02 15:11:26 +0000562 default: break;
563 }
564 }
565 }
566
Chris Lattner113f4f42002-06-25 16:13:24 +0000567 return Changed ? &I : 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000568}
569
Chris Lattnerbdb0ce02003-07-22 21:46:59 +0000570// isSignBit - Return true if the value represented by the constant only has the
571// highest order bit set.
572static bool isSignBit(ConstantInt *CI) {
573 unsigned NumBits = CI->getType()->getPrimitiveSize()*8;
574 return (CI->getRawValue() & ~(-1LL << NumBits)) == (1ULL << (NumBits-1));
575}
576
Chris Lattnerdfae8be2003-07-24 17:35:25 +0000577static unsigned getTypeSizeInBits(const Type *Ty) {
578 return Ty == Type::BoolTy ? 1 : Ty->getPrimitiveSize()*8;
579}
580
Chris Lattner022167f2004-03-13 00:11:49 +0000581/// RemoveNoopCast - Strip off nonconverting casts from the value.
582///
583static Value *RemoveNoopCast(Value *V) {
584 if (CastInst *CI = dyn_cast<CastInst>(V)) {
585 const Type *CTy = CI->getType();
586 const Type *OpTy = CI->getOperand(0)->getType();
587 if (CTy->isInteger() && OpTy->isInteger()) {
588 if (CTy->getPrimitiveSize() == OpTy->getPrimitiveSize())
589 return RemoveNoopCast(CI->getOperand(0));
590 } else if (isa<PointerType>(CTy) && isa<PointerType>(OpTy))
591 return RemoveNoopCast(CI->getOperand(0));
592 }
593 return V;
594}
595
Chris Lattner113f4f42002-06-25 16:13:24 +0000596Instruction *InstCombiner::visitSub(BinaryOperator &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +0000597 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000598
Chris Lattnere6794492002-08-12 21:17:25 +0000599 if (Op0 == Op1) // sub X, X -> 0
600 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner260ab202002-04-18 17:39:14 +0000601
Chris Lattnere6794492002-08-12 21:17:25 +0000602 // If this is a 'B = x-(-A)', change to B = x+A...
Chris Lattnerbb74e222003-03-10 23:06:50 +0000603 if (Value *V = dyn_castNegVal(Op1))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000604 return BinaryOperator::createAdd(Op0, V);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000605
Chris Lattner8f2f5982003-11-05 01:06:05 +0000606 if (ConstantInt *C = dyn_cast<ConstantInt>(Op0)) {
607 // Replace (-1 - A) with (~A)...
Chris Lattner3082c5a2003-02-18 19:28:33 +0000608 if (C->isAllOnesValue())
609 return BinaryOperator::createNot(Op1);
Chris Lattnerad3c4952002-05-09 01:29:19 +0000610
Chris Lattner8f2f5982003-11-05 01:06:05 +0000611 // C - ~X == X + (1+C)
612 if (BinaryOperator::isNot(Op1))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000613 return BinaryOperator::createAdd(
614 BinaryOperator::getNotArgument(cast<BinaryOperator>(Op1)),
615 ConstantExpr::getAdd(C, ConstantInt::get(I.getType(), 1)));
Chris Lattner92295c52004-03-12 23:53:13 +0000616 // -((uint)X >> 31) -> ((int)X >> 31)
617 // -((int)X >> 31) -> ((uint)X >> 31)
Chris Lattner022167f2004-03-13 00:11:49 +0000618 if (C->isNullValue()) {
619 Value *NoopCastedRHS = RemoveNoopCast(Op1);
620 if (ShiftInst *SI = dyn_cast<ShiftInst>(NoopCastedRHS))
Chris Lattner92295c52004-03-12 23:53:13 +0000621 if (SI->getOpcode() == Instruction::Shr)
622 if (ConstantUInt *CU = dyn_cast<ConstantUInt>(SI->getOperand(1))) {
623 const Type *NewTy;
Chris Lattner022167f2004-03-13 00:11:49 +0000624 if (SI->getType()->isSigned())
Chris Lattner97bfcea2004-06-17 18:16:02 +0000625 NewTy = SI->getType()->getUnsignedVersion();
Chris Lattner92295c52004-03-12 23:53:13 +0000626 else
Chris Lattner97bfcea2004-06-17 18:16:02 +0000627 NewTy = SI->getType()->getSignedVersion();
Chris Lattner92295c52004-03-12 23:53:13 +0000628 // Check to see if we are shifting out everything but the sign bit.
Chris Lattner022167f2004-03-13 00:11:49 +0000629 if (CU->getValue() == SI->getType()->getPrimitiveSize()*8-1) {
Chris Lattner92295c52004-03-12 23:53:13 +0000630 // Ok, the transformation is safe. Insert a cast of the incoming
631 // value, then the new shift, then the new cast.
632 Instruction *FirstCast = new CastInst(SI->getOperand(0), NewTy,
633 SI->getOperand(0)->getName());
634 Value *InV = InsertNewInstBefore(FirstCast, I);
635 Instruction *NewShift = new ShiftInst(Instruction::Shr, FirstCast,
636 CU, SI->getName());
Chris Lattner022167f2004-03-13 00:11:49 +0000637 if (NewShift->getType() == I.getType())
638 return NewShift;
639 else {
640 InV = InsertNewInstBefore(NewShift, I);
641 return new CastInst(NewShift, I.getType());
642 }
Chris Lattner92295c52004-03-12 23:53:13 +0000643 }
644 }
Chris Lattner022167f2004-03-13 00:11:49 +0000645 }
Chris Lattner183b3362004-04-09 19:05:30 +0000646
647 // Try to fold constant sub into select arguments.
648 if (SelectInst *SI = dyn_cast<SelectInst>(Op1))
649 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
650 return R;
Chris Lattner8f2f5982003-11-05 01:06:05 +0000651 }
652
Chris Lattner3082c5a2003-02-18 19:28:33 +0000653 if (BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1))
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000654 if (Op1I->hasOneUse()) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000655 // Replace (x - (y - z)) with (x + (z - y)) if the (y - z) subexpression
656 // is not used by anyone else...
657 //
Chris Lattnerc2f0aa52004-02-02 20:09:56 +0000658 if (Op1I->getOpcode() == Instruction::Sub &&
659 !Op1I->getType()->isFloatingPoint()) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000660 // Swap the two operands of the subexpr...
661 Value *IIOp0 = Op1I->getOperand(0), *IIOp1 = Op1I->getOperand(1);
662 Op1I->setOperand(0, IIOp1);
663 Op1I->setOperand(1, IIOp0);
664
665 // Create the new top level add instruction...
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000666 return BinaryOperator::createAdd(Op0, Op1);
Chris Lattner3082c5a2003-02-18 19:28:33 +0000667 }
668
669 // Replace (A - (A & B)) with (A & ~B) if this is the only use of (A&B)...
670 //
671 if (Op1I->getOpcode() == Instruction::And &&
672 (Op1I->getOperand(0) == Op0 || Op1I->getOperand(1) == Op0)) {
673 Value *OtherOp = Op1I->getOperand(Op1I->getOperand(0) == Op0);
674
Chris Lattner396dbfe2004-06-09 05:08:07 +0000675 Value *NewNot =
676 InsertNewInstBefore(BinaryOperator::createNot(OtherOp, "B.not"), I);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000677 return BinaryOperator::createAnd(Op0, NewNot);
Chris Lattner3082c5a2003-02-18 19:28:33 +0000678 }
Chris Lattner57c8d992003-02-18 19:57:07 +0000679
680 // X - X*C --> X * (1-C)
681 if (dyn_castFoldableMul(Op1I) == Op0) {
Chris Lattner34428442003-05-27 16:40:51 +0000682 Constant *CP1 =
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000683 ConstantExpr::getSub(ConstantInt::get(I.getType(), 1),
Chris Lattner34428442003-05-27 16:40:51 +0000684 cast<Constant>(cast<Instruction>(Op1)->getOperand(1)));
Chris Lattner57c8d992003-02-18 19:57:07 +0000685 assert(CP1 && "Couldn't constant fold 1-C?");
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000686 return BinaryOperator::createMul(Op0, CP1);
Chris Lattner57c8d992003-02-18 19:57:07 +0000687 }
Chris Lattnerad3c4952002-05-09 01:29:19 +0000688 }
Chris Lattner3082c5a2003-02-18 19:28:33 +0000689
Chris Lattner57c8d992003-02-18 19:57:07 +0000690 // X*C - X --> X * (C-1)
691 if (dyn_castFoldableMul(Op0) == Op1) {
Chris Lattner34428442003-05-27 16:40:51 +0000692 Constant *CP1 =
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000693 ConstantExpr::getSub(cast<Constant>(cast<Instruction>(Op0)->getOperand(1)),
Chris Lattner34428442003-05-27 16:40:51 +0000694 ConstantInt::get(I.getType(), 1));
Chris Lattner57c8d992003-02-18 19:57:07 +0000695 assert(CP1 && "Couldn't constant fold C - 1?");
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000696 return BinaryOperator::createMul(Op1, CP1);
Chris Lattner57c8d992003-02-18 19:57:07 +0000697 }
698
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000699 return 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000700}
701
Chris Lattnere79e8542004-02-23 06:38:22 +0000702/// isSignBitCheck - Given an exploded setcc instruction, return true if it is
703/// really just returns true if the most significant (sign) bit is set.
704static bool isSignBitCheck(unsigned Opcode, Value *LHS, ConstantInt *RHS) {
705 if (RHS->getType()->isSigned()) {
706 // True if source is LHS < 0 or LHS <= -1
707 return Opcode == Instruction::SetLT && RHS->isNullValue() ||
708 Opcode == Instruction::SetLE && RHS->isAllOnesValue();
709 } else {
710 ConstantUInt *RHSC = cast<ConstantUInt>(RHS);
711 // True if source is LHS > 127 or LHS >= 128, where the constants depend on
712 // the size of the integer type.
713 if (Opcode == Instruction::SetGE)
714 return RHSC->getValue() == 1ULL<<(RHS->getType()->getPrimitiveSize()*8-1);
715 if (Opcode == Instruction::SetGT)
716 return RHSC->getValue() ==
717 (1ULL << (RHS->getType()->getPrimitiveSize()*8-1))-1;
718 }
719 return false;
720}
721
Chris Lattner113f4f42002-06-25 16:13:24 +0000722Instruction *InstCombiner::visitMul(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000723 bool Changed = SimplifyCommutative(I);
Chris Lattner3082c5a2003-02-18 19:28:33 +0000724 Value *Op0 = I.getOperand(0);
Chris Lattner260ab202002-04-18 17:39:14 +0000725
Chris Lattnere6794492002-08-12 21:17:25 +0000726 // Simplify mul instructions with a constant RHS...
Chris Lattner3082c5a2003-02-18 19:28:33 +0000727 if (Constant *Op1 = dyn_cast<Constant>(I.getOperand(1))) {
728 if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) {
Chris Lattnerede3fe02003-08-13 04:18:28 +0000729
730 // ((X << C1)*C2) == (X * (C2 << C1))
731 if (ShiftInst *SI = dyn_cast<ShiftInst>(Op0))
732 if (SI->getOpcode() == Instruction::Shl)
733 if (Constant *ShOp = dyn_cast<Constant>(SI->getOperand(1)))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000734 return BinaryOperator::createMul(SI->getOperand(0),
735 ConstantExpr::getShl(CI, ShOp));
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000736
Chris Lattnercce81be2003-09-11 22:24:54 +0000737 if (CI->isNullValue())
738 return ReplaceInstUsesWith(I, Op1); // X * 0 == 0
739 if (CI->equalsInt(1)) // X * 1 == X
740 return ReplaceInstUsesWith(I, Op0);
741 if (CI->isAllOnesValue()) // X * -1 == 0 - X
Chris Lattner35236d82003-06-25 17:09:20 +0000742 return BinaryOperator::createNeg(Op0, I.getName());
Chris Lattner31ba1292002-04-29 22:24:47 +0000743
Chris Lattnercce81be2003-09-11 22:24:54 +0000744 int64_t Val = (int64_t)cast<ConstantInt>(CI)->getRawValue();
Chris Lattner3082c5a2003-02-18 19:28:33 +0000745 if (uint64_t C = Log2(Val)) // Replace X*(2^C) with X << C
746 return new ShiftInst(Instruction::Shl, Op0,
747 ConstantUInt::get(Type::UByteTy, C));
748 } else {
749 ConstantFP *Op1F = cast<ConstantFP>(Op1);
750 if (Op1F->isNullValue())
751 return ReplaceInstUsesWith(I, Op1);
Chris Lattner31ba1292002-04-29 22:24:47 +0000752
Chris Lattner3082c5a2003-02-18 19:28:33 +0000753 // "In IEEE floating point, x*1 is not equivalent to x for nans. However,
754 // ANSI says we can drop signals, so we can do this anyway." (from GCC)
755 if (Op1F->getValue() == 1.0)
756 return ReplaceInstUsesWith(I, Op0); // Eliminate 'mul double %X, 1.0'
757 }
Chris Lattner183b3362004-04-09 19:05:30 +0000758
759 // Try to fold constant mul into select arguments.
760 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
761 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
762 return R;
Chris Lattner260ab202002-04-18 17:39:14 +0000763 }
764
Chris Lattner934a64cf2003-03-10 23:23:04 +0000765 if (Value *Op0v = dyn_castNegVal(Op0)) // -X * -Y = X*Y
766 if (Value *Op1v = dyn_castNegVal(I.getOperand(1)))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000767 return BinaryOperator::createMul(Op0v, Op1v);
Chris Lattner934a64cf2003-03-10 23:23:04 +0000768
Chris Lattner2635b522004-02-23 05:39:21 +0000769 // If one of the operands of the multiply is a cast from a boolean value, then
770 // we know the bool is either zero or one, so this is a 'masking' multiply.
771 // See if we can simplify things based on how the boolean was originally
772 // formed.
773 CastInst *BoolCast = 0;
774 if (CastInst *CI = dyn_cast<CastInst>(I.getOperand(0)))
775 if (CI->getOperand(0)->getType() == Type::BoolTy)
776 BoolCast = CI;
777 if (!BoolCast)
778 if (CastInst *CI = dyn_cast<CastInst>(I.getOperand(1)))
779 if (CI->getOperand(0)->getType() == Type::BoolTy)
780 BoolCast = CI;
781 if (BoolCast) {
782 if (SetCondInst *SCI = dyn_cast<SetCondInst>(BoolCast->getOperand(0))) {
783 Value *SCIOp0 = SCI->getOperand(0), *SCIOp1 = SCI->getOperand(1);
784 const Type *SCOpTy = SCIOp0->getType();
785
Chris Lattnere79e8542004-02-23 06:38:22 +0000786 // If the setcc is true iff the sign bit of X is set, then convert this
787 // multiply into a shift/and combination.
788 if (isa<ConstantInt>(SCIOp1) &&
789 isSignBitCheck(SCI->getOpcode(), SCIOp0, cast<ConstantInt>(SCIOp1))) {
Chris Lattner2635b522004-02-23 05:39:21 +0000790 // Shift the X value right to turn it into "all signbits".
791 Constant *Amt = ConstantUInt::get(Type::UByteTy,
792 SCOpTy->getPrimitiveSize()*8-1);
Chris Lattnere79e8542004-02-23 06:38:22 +0000793 if (SCIOp0->getType()->isUnsigned()) {
Chris Lattner97bfcea2004-06-17 18:16:02 +0000794 const Type *NewTy = SCIOp0->getType()->getSignedVersion();
Chris Lattnere79e8542004-02-23 06:38:22 +0000795 SCIOp0 = InsertNewInstBefore(new CastInst(SCIOp0, NewTy,
796 SCIOp0->getName()), I);
797 }
798
799 Value *V =
800 InsertNewInstBefore(new ShiftInst(Instruction::Shr, SCIOp0, Amt,
801 BoolCast->getOperand(0)->getName()+
802 ".mask"), I);
Chris Lattner2635b522004-02-23 05:39:21 +0000803
804 // If the multiply type is not the same as the source type, sign extend
805 // or truncate to the multiply type.
806 if (I.getType() != V->getType())
Chris Lattnere79e8542004-02-23 06:38:22 +0000807 V = InsertNewInstBefore(new CastInst(V, I.getType(), V->getName()),I);
Chris Lattner2635b522004-02-23 05:39:21 +0000808
809 Value *OtherOp = Op0 == BoolCast ? I.getOperand(1) : Op0;
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000810 return BinaryOperator::createAnd(V, OtherOp);
Chris Lattner2635b522004-02-23 05:39:21 +0000811 }
812 }
813 }
814
Chris Lattner113f4f42002-06-25 16:13:24 +0000815 return Changed ? &I : 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000816}
817
Chris Lattner113f4f42002-06-25 16:13:24 +0000818Instruction *InstCombiner::visitDiv(BinaryOperator &I) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000819 if (ConstantInt *RHS = dyn_cast<ConstantInt>(I.getOperand(1))) {
Chris Lattnere20c3342004-04-26 14:01:59 +0000820 // div X, 1 == X
Chris Lattnere6794492002-08-12 21:17:25 +0000821 if (RHS->equalsInt(1))
822 return ReplaceInstUsesWith(I, I.getOperand(0));
Chris Lattner3082c5a2003-02-18 19:28:33 +0000823
Chris Lattnere20c3342004-04-26 14:01:59 +0000824 // div X, -1 == -X
825 if (RHS->isAllOnesValue())
826 return BinaryOperator::createNeg(I.getOperand(0));
827
Chris Lattner3082c5a2003-02-18 19:28:33 +0000828 // Check to see if this is an unsigned division with an exact power of 2,
829 // if so, convert to a right shift.
830 if (ConstantUInt *C = dyn_cast<ConstantUInt>(RHS))
831 if (uint64_t Val = C->getValue()) // Don't break X / 0
832 if (uint64_t C = Log2(Val))
833 return new ShiftInst(Instruction::Shr, I.getOperand(0),
834 ConstantUInt::get(Type::UByteTy, C));
835 }
836
837 // 0 / X == 0, we don't need to preserve faults!
838 if (ConstantInt *LHS = dyn_cast<ConstantInt>(I.getOperand(0)))
839 if (LHS->equalsInt(0))
840 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
841
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000842 return 0;
843}
844
845
Chris Lattner113f4f42002-06-25 16:13:24 +0000846Instruction *InstCombiner::visitRem(BinaryOperator &I) {
Chris Lattner7fd5f072004-07-06 07:01:22 +0000847 if (I.getType()->isSigned())
848 if (Value *RHSNeg = dyn_castNegVal(I.getOperand(1)))
Chris Lattner98c6bdf2004-07-06 07:11:42 +0000849 if (!isa<ConstantSInt>(RHSNeg) ||
850 cast<ConstantSInt>(RHSNeg)->getValue() >= 0) {
Chris Lattner7fd5f072004-07-06 07:01:22 +0000851 // X % -Y -> X % Y
852 AddUsesToWorkList(I);
853 I.setOperand(1, RHSNeg);
854 return &I;
855 }
856
Chris Lattner3082c5a2003-02-18 19:28:33 +0000857 if (ConstantInt *RHS = dyn_cast<ConstantInt>(I.getOperand(1))) {
858 if (RHS->equalsInt(1)) // X % 1 == 0
859 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
860
861 // Check to see if this is an unsigned remainder with an exact power of 2,
862 // if so, convert to a bitwise and.
863 if (ConstantUInt *C = dyn_cast<ConstantUInt>(RHS))
864 if (uint64_t Val = C->getValue()) // Don't break X % 0 (divide by zero)
Chris Lattnerd9e58132004-05-07 15:35:56 +0000865 if (!(Val & (Val-1))) // Power of 2
Chris Lattnerdf20a4d2004-06-10 02:07:29 +0000866 return BinaryOperator::createAnd(I.getOperand(0),
Chris Lattner3082c5a2003-02-18 19:28:33 +0000867 ConstantUInt::get(I.getType(), Val-1));
868 }
869
870 // 0 % X == 0, we don't need to preserve faults!
871 if (ConstantInt *LHS = dyn_cast<ConstantInt>(I.getOperand(0)))
872 if (LHS->equalsInt(0))
Chris Lattnere6794492002-08-12 21:17:25 +0000873 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
874
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000875 return 0;
876}
877
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000878// isMaxValueMinusOne - return true if this is Max-1
Chris Lattnere6794492002-08-12 21:17:25 +0000879static bool isMaxValueMinusOne(const ConstantInt *C) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000880 if (const ConstantUInt *CU = dyn_cast<ConstantUInt>(C)) {
881 // Calculate -1 casted to the right type...
882 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
883 uint64_t Val = ~0ULL; // All ones
884 Val >>= 64-TypeBits; // Shift out unwanted 1 bits...
885 return CU->getValue() == Val-1;
886 }
887
888 const ConstantSInt *CS = cast<ConstantSInt>(C);
889
890 // Calculate 0111111111..11111
891 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
892 int64_t Val = INT64_MAX; // All ones
893 Val >>= 64-TypeBits; // Shift out unwanted 1 bits...
894 return CS->getValue() == Val-1;
895}
896
897// isMinValuePlusOne - return true if this is Min+1
Chris Lattnere6794492002-08-12 21:17:25 +0000898static bool isMinValuePlusOne(const ConstantInt *C) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000899 if (const ConstantUInt *CU = dyn_cast<ConstantUInt>(C))
900 return CU->getValue() == 1;
901
902 const ConstantSInt *CS = cast<ConstantSInt>(C);
903
904 // Calculate 1111111111000000000000
905 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
906 int64_t Val = -1; // All ones
907 Val <<= TypeBits-1; // Shift over to the right spot
908 return CS->getValue() == Val+1;
909}
910
Chris Lattner35167c32004-06-09 07:59:58 +0000911// isOneBitSet - Return true if there is exactly one bit set in the specified
912// constant.
913static bool isOneBitSet(const ConstantInt *CI) {
914 uint64_t V = CI->getRawValue();
915 return V && (V & (V-1)) == 0;
916}
917
Chris Lattner3ac7c262003-08-13 20:16:26 +0000918/// getSetCondCode - Encode a setcc opcode into a three bit mask. These bits
919/// are carefully arranged to allow folding of expressions such as:
920///
921/// (A < B) | (A > B) --> (A != B)
922///
923/// Bit value '4' represents that the comparison is true if A > B, bit value '2'
924/// represents that the comparison is true if A == B, and bit value '1' is true
925/// if A < B.
926///
927static unsigned getSetCondCode(const SetCondInst *SCI) {
928 switch (SCI->getOpcode()) {
929 // False -> 0
930 case Instruction::SetGT: return 1;
931 case Instruction::SetEQ: return 2;
932 case Instruction::SetGE: return 3;
933 case Instruction::SetLT: return 4;
934 case Instruction::SetNE: return 5;
935 case Instruction::SetLE: return 6;
936 // True -> 7
937 default:
938 assert(0 && "Invalid SetCC opcode!");
939 return 0;
940 }
941}
942
943/// getSetCCValue - This is the complement of getSetCondCode, which turns an
944/// opcode and two operands into either a constant true or false, or a brand new
945/// SetCC instruction.
946static Value *getSetCCValue(unsigned Opcode, Value *LHS, Value *RHS) {
947 switch (Opcode) {
948 case 0: return ConstantBool::False;
949 case 1: return new SetCondInst(Instruction::SetGT, LHS, RHS);
950 case 2: return new SetCondInst(Instruction::SetEQ, LHS, RHS);
951 case 3: return new SetCondInst(Instruction::SetGE, LHS, RHS);
952 case 4: return new SetCondInst(Instruction::SetLT, LHS, RHS);
953 case 5: return new SetCondInst(Instruction::SetNE, LHS, RHS);
954 case 6: return new SetCondInst(Instruction::SetLE, LHS, RHS);
955 case 7: return ConstantBool::True;
956 default: assert(0 && "Illegal SetCCCode!"); return 0;
957 }
958}
959
960// FoldSetCCLogical - Implements (setcc1 A, B) & (setcc2 A, B) --> (setcc3 A, B)
961struct FoldSetCCLogical {
962 InstCombiner &IC;
963 Value *LHS, *RHS;
964 FoldSetCCLogical(InstCombiner &ic, SetCondInst *SCI)
965 : IC(ic), LHS(SCI->getOperand(0)), RHS(SCI->getOperand(1)) {}
966 bool shouldApply(Value *V) const {
967 if (SetCondInst *SCI = dyn_cast<SetCondInst>(V))
968 return (SCI->getOperand(0) == LHS && SCI->getOperand(1) == RHS ||
969 SCI->getOperand(0) == RHS && SCI->getOperand(1) == LHS);
970 return false;
971 }
972 Instruction *apply(BinaryOperator &Log) const {
973 SetCondInst *SCI = cast<SetCondInst>(Log.getOperand(0));
974 if (SCI->getOperand(0) != LHS) {
975 assert(SCI->getOperand(1) == LHS);
976 SCI->swapOperands(); // Swap the LHS and RHS of the SetCC
977 }
978
979 unsigned LHSCode = getSetCondCode(SCI);
980 unsigned RHSCode = getSetCondCode(cast<SetCondInst>(Log.getOperand(1)));
981 unsigned Code;
982 switch (Log.getOpcode()) {
983 case Instruction::And: Code = LHSCode & RHSCode; break;
984 case Instruction::Or: Code = LHSCode | RHSCode; break;
985 case Instruction::Xor: Code = LHSCode ^ RHSCode; break;
Chris Lattner2caaaba2003-09-22 20:33:34 +0000986 default: assert(0 && "Illegal logical opcode!"); return 0;
Chris Lattner3ac7c262003-08-13 20:16:26 +0000987 }
988
989 Value *RV = getSetCCValue(Code, LHS, RHS);
990 if (Instruction *I = dyn_cast<Instruction>(RV))
991 return I;
992 // Otherwise, it's a constant boolean value...
993 return IC.ReplaceInstUsesWith(Log, RV);
994 }
995};
996
997
Chris Lattnerba1cb382003-09-19 17:17:26 +0000998// OptAndOp - This handles expressions of the form ((val OP C1) & C2). Where
999// the Op parameter is 'OP', OpRHS is 'C1', and AndRHS is 'C2'. Op is
1000// guaranteed to be either a shift instruction or a binary operator.
1001Instruction *InstCombiner::OptAndOp(Instruction *Op,
1002 ConstantIntegral *OpRHS,
1003 ConstantIntegral *AndRHS,
1004 BinaryOperator &TheAnd) {
1005 Value *X = Op->getOperand(0);
Chris Lattnerfcf21a72004-01-12 19:47:05 +00001006 Constant *Together = 0;
1007 if (!isa<ShiftInst>(Op))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001008 Together = ConstantExpr::getAnd(AndRHS, OpRHS);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001009
Chris Lattnerba1cb382003-09-19 17:17:26 +00001010 switch (Op->getOpcode()) {
1011 case Instruction::Xor:
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001012 if (Together->isNullValue()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001013 // (X ^ C1) & C2 --> (X & C2) iff (C1&C2) == 0
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001014 return BinaryOperator::createAnd(X, AndRHS);
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001015 } else if (Op->hasOneUse()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001016 // (X ^ C1) & C2 --> (X & C2) ^ (C1&C2)
1017 std::string OpName = Op->getName(); Op->setName("");
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001018 Instruction *And = BinaryOperator::createAnd(X, AndRHS, OpName);
Chris Lattnerba1cb382003-09-19 17:17:26 +00001019 InsertNewInstBefore(And, TheAnd);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001020 return BinaryOperator::createXor(And, Together);
Chris Lattnerba1cb382003-09-19 17:17:26 +00001021 }
1022 break;
1023 case Instruction::Or:
1024 // (X | C1) & C2 --> X & C2 iff C1 & C1 == 0
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001025 if (Together->isNullValue())
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001026 return BinaryOperator::createAnd(X, AndRHS);
Chris Lattnerba1cb382003-09-19 17:17:26 +00001027 else {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001028 if (Together == AndRHS) // (X | C) & C --> C
1029 return ReplaceInstUsesWith(TheAnd, AndRHS);
1030
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001031 if (Op->hasOneUse() && Together != OpRHS) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001032 // (X | C1) & C2 --> (X | (C1&C2)) & C2
1033 std::string Op0Name = Op->getName(); Op->setName("");
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001034 Instruction *Or = BinaryOperator::createOr(X, Together, Op0Name);
Chris Lattnerba1cb382003-09-19 17:17:26 +00001035 InsertNewInstBefore(Or, TheAnd);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001036 return BinaryOperator::createAnd(Or, AndRHS);
Chris Lattnerba1cb382003-09-19 17:17:26 +00001037 }
1038 }
1039 break;
1040 case Instruction::Add:
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001041 if (Op->hasOneUse()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001042 // Adding a one to a single bit bit-field should be turned into an XOR
1043 // of the bit. First thing to check is to see if this AND is with a
1044 // single bit constant.
Chris Lattner35167c32004-06-09 07:59:58 +00001045 uint64_t AndRHSV = cast<ConstantInt>(AndRHS)->getRawValue();
Chris Lattnerba1cb382003-09-19 17:17:26 +00001046
1047 // Clear bits that are not part of the constant.
1048 AndRHSV &= (1ULL << AndRHS->getType()->getPrimitiveSize()*8)-1;
1049
1050 // If there is only one bit set...
Chris Lattner35167c32004-06-09 07:59:58 +00001051 if (isOneBitSet(cast<ConstantInt>(AndRHS))) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001052 // Ok, at this point, we know that we are masking the result of the
1053 // ADD down to exactly one bit. If the constant we are adding has
1054 // no bits set below this bit, then we can eliminate the ADD.
Chris Lattner35167c32004-06-09 07:59:58 +00001055 uint64_t AddRHS = cast<ConstantInt>(OpRHS)->getRawValue();
Chris Lattnerba1cb382003-09-19 17:17:26 +00001056
1057 // Check to see if any bits below the one bit set in AndRHSV are set.
1058 if ((AddRHS & (AndRHSV-1)) == 0) {
1059 // If not, the only thing that can effect the output of the AND is
1060 // the bit specified by AndRHSV. If that bit is set, the effect of
1061 // the XOR is to toggle the bit. If it is clear, then the ADD has
1062 // no effect.
1063 if ((AddRHS & AndRHSV) == 0) { // Bit is not set, noop
1064 TheAnd.setOperand(0, X);
1065 return &TheAnd;
1066 } else {
1067 std::string Name = Op->getName(); Op->setName("");
1068 // Pull the XOR out of the AND.
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001069 Instruction *NewAnd = BinaryOperator::createAnd(X, AndRHS, Name);
Chris Lattnerba1cb382003-09-19 17:17:26 +00001070 InsertNewInstBefore(NewAnd, TheAnd);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001071 return BinaryOperator::createXor(NewAnd, AndRHS);
Chris Lattnerba1cb382003-09-19 17:17:26 +00001072 }
1073 }
1074 }
1075 }
1076 break;
Chris Lattner2da29172003-09-19 19:05:02 +00001077
1078 case Instruction::Shl: {
1079 // We know that the AND will not produce any of the bits shifted in, so if
1080 // the anded constant includes them, clear them now!
1081 //
1082 Constant *AllOne = ConstantIntegral::getAllOnesValue(AndRHS->getType());
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001083 Constant *CI = ConstantExpr::getAnd(AndRHS,
1084 ConstantExpr::getShl(AllOne, OpRHS));
Chris Lattner2da29172003-09-19 19:05:02 +00001085 if (CI != AndRHS) {
1086 TheAnd.setOperand(1, CI);
1087 return &TheAnd;
1088 }
1089 break;
1090 }
1091 case Instruction::Shr:
1092 // We know that the AND will not produce any of the bits shifted in, so if
1093 // the anded constant includes them, clear them now! This only applies to
1094 // unsigned shifts, because a signed shr may bring in set bits!
1095 //
1096 if (AndRHS->getType()->isUnsigned()) {
1097 Constant *AllOne = ConstantIntegral::getAllOnesValue(AndRHS->getType());
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001098 Constant *CI = ConstantExpr::getAnd(AndRHS,
1099 ConstantExpr::getShr(AllOne, OpRHS));
Chris Lattner2da29172003-09-19 19:05:02 +00001100 if (CI != AndRHS) {
1101 TheAnd.setOperand(1, CI);
1102 return &TheAnd;
1103 }
1104 }
1105 break;
Chris Lattnerba1cb382003-09-19 17:17:26 +00001106 }
1107 return 0;
1108}
1109
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001110
Chris Lattner113f4f42002-06-25 16:13:24 +00001111Instruction *InstCombiner::visitAnd(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001112 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00001113 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001114
1115 // and X, X = X and X, 0 == 0
Chris Lattnere6794492002-08-12 21:17:25 +00001116 if (Op0 == Op1 || Op1 == Constant::getNullValue(I.getType()))
1117 return ReplaceInstUsesWith(I, Op1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001118
1119 // and X, -1 == X
Chris Lattner49b47ae2003-07-23 17:57:01 +00001120 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattnere6794492002-08-12 21:17:25 +00001121 if (RHS->isAllOnesValue())
1122 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001123
Chris Lattnerba1cb382003-09-19 17:17:26 +00001124 // Optimize a variety of ((val OP C1) & C2) combinations...
1125 if (isa<BinaryOperator>(Op0) || isa<ShiftInst>(Op0)) {
1126 Instruction *Op0I = cast<Instruction>(Op0);
Chris Lattner33217db2003-07-23 19:36:21 +00001127 Value *X = Op0I->getOperand(0);
Chris Lattner16464b32003-07-23 19:25:52 +00001128 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1)))
Chris Lattnerba1cb382003-09-19 17:17:26 +00001129 if (Instruction *Res = OptAndOp(Op0I, Op0CI, RHS, I))
1130 return Res;
Chris Lattner33217db2003-07-23 19:36:21 +00001131 }
Chris Lattner183b3362004-04-09 19:05:30 +00001132
1133 // Try to fold constant and into select arguments.
1134 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1135 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1136 return R;
Chris Lattner49b47ae2003-07-23 17:57:01 +00001137 }
1138
Chris Lattnerbb74e222003-03-10 23:06:50 +00001139 Value *Op0NotVal = dyn_castNotVal(Op0);
1140 Value *Op1NotVal = dyn_castNotVal(Op1);
Chris Lattner3082c5a2003-02-18 19:28:33 +00001141
Chris Lattner023a4832004-06-18 06:07:51 +00001142 if (Op0NotVal == Op1 || Op1NotVal == Op0) // A & ~A == ~A & A == 0
1143 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
1144
Chris Lattner3082c5a2003-02-18 19:28:33 +00001145 // (~A & ~B) == (~(A | B)) - Demorgan's Law
Chris Lattnerbb74e222003-03-10 23:06:50 +00001146 if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) {
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001147 Instruction *Or = BinaryOperator::createOr(Op0NotVal, Op1NotVal,
1148 I.getName()+".demorgan");
Chris Lattner49b47ae2003-07-23 17:57:01 +00001149 InsertNewInstBefore(Or, I);
Chris Lattner3082c5a2003-02-18 19:28:33 +00001150 return BinaryOperator::createNot(Or);
1151 }
1152
Chris Lattner3ac7c262003-08-13 20:16:26 +00001153 // (setcc1 A, B) & (setcc2 A, B) --> (setcc3 A, B)
1154 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1155 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1156 return R;
1157
Chris Lattner113f4f42002-06-25 16:13:24 +00001158 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001159}
1160
1161
1162
Chris Lattner113f4f42002-06-25 16:13:24 +00001163Instruction *InstCombiner::visitOr(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001164 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00001165 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001166
1167 // or X, X = X or X, 0 == X
Chris Lattnere6794492002-08-12 21:17:25 +00001168 if (Op0 == Op1 || Op1 == Constant::getNullValue(I.getType()))
1169 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001170
1171 // or X, -1 == -1
Chris Lattner8f0d1562003-07-23 18:29:44 +00001172 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattnere6794492002-08-12 21:17:25 +00001173 if (RHS->isAllOnesValue())
1174 return ReplaceInstUsesWith(I, Op1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001175
Chris Lattner8f0d1562003-07-23 18:29:44 +00001176 if (Instruction *Op0I = dyn_cast<Instruction>(Op0)) {
1177 // (X & C1) | C2 --> (X | C2) & (C1|C2)
1178 if (Op0I->getOpcode() == Instruction::And && isOnlyUse(Op0))
1179 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) {
1180 std::string Op0Name = Op0I->getName(); Op0I->setName("");
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001181 Instruction *Or = BinaryOperator::createOr(Op0I->getOperand(0), RHS,
1182 Op0Name);
Chris Lattner8f0d1562003-07-23 18:29:44 +00001183 InsertNewInstBefore(Or, I);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001184 return BinaryOperator::createAnd(Or, ConstantExpr::getOr(RHS, Op0CI));
Chris Lattner8f0d1562003-07-23 18:29:44 +00001185 }
1186
1187 // (X ^ C1) | C2 --> (X | C2) ^ (C1&~C2)
1188 if (Op0I->getOpcode() == Instruction::Xor && isOnlyUse(Op0))
1189 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) {
1190 std::string Op0Name = Op0I->getName(); Op0I->setName("");
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001191 Instruction *Or = BinaryOperator::createOr(Op0I->getOperand(0), RHS,
1192 Op0Name);
Chris Lattner8f0d1562003-07-23 18:29:44 +00001193 InsertNewInstBefore(Or, I);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001194 return BinaryOperator::createXor(Or,
Chris Lattnerc8e7e292004-06-10 02:12:35 +00001195 ConstantExpr::getAnd(Op0CI,
1196 ConstantExpr::getNot(RHS)));
Chris Lattner8f0d1562003-07-23 18:29:44 +00001197 }
1198 }
Chris Lattner183b3362004-04-09 19:05:30 +00001199
1200 // Try to fold constant and into select arguments.
1201 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1202 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1203 return R;
Chris Lattner8f0d1562003-07-23 18:29:44 +00001204 }
1205
Chris Lattner812aab72003-08-12 19:11:07 +00001206 // (A & C1)|(A & C2) == A & (C1|C2)
Chris Lattner01d56392003-08-12 19:17:27 +00001207 if (Instruction *LHS = dyn_cast<BinaryOperator>(Op0))
1208 if (Instruction *RHS = dyn_cast<BinaryOperator>(Op1))
1209 if (LHS->getOperand(0) == RHS->getOperand(0))
1210 if (Constant *C0 = dyn_castMaskingAnd(LHS))
1211 if (Constant *C1 = dyn_castMaskingAnd(RHS))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001212 return BinaryOperator::createAnd(LHS->getOperand(0),
1213 ConstantExpr::getOr(C0, C1));
Chris Lattner812aab72003-08-12 19:11:07 +00001214
Chris Lattner3e327a42003-03-10 23:13:59 +00001215 Value *Op0NotVal = dyn_castNotVal(Op0);
1216 Value *Op1NotVal = dyn_castNotVal(Op1);
Chris Lattner3082c5a2003-02-18 19:28:33 +00001217
Chris Lattner3e327a42003-03-10 23:13:59 +00001218 if (Op1 == Op0NotVal) // ~A | A == -1
1219 return ReplaceInstUsesWith(I,
1220 ConstantIntegral::getAllOnesValue(I.getType()));
1221
1222 if (Op0 == Op1NotVal) // A | ~A == -1
1223 return ReplaceInstUsesWith(I,
1224 ConstantIntegral::getAllOnesValue(I.getType()));
1225
1226 // (~A | ~B) == (~(A & B)) - Demorgan's Law
1227 if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) {
Chris Lattner396dbfe2004-06-09 05:08:07 +00001228 Value *And = InsertNewInstBefore(
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001229 BinaryOperator::createAnd(Op0NotVal,
1230 Op1NotVal,I.getName()+".demorgan"),I);
Chris Lattner3e327a42003-03-10 23:13:59 +00001231 return BinaryOperator::createNot(And);
1232 }
Chris Lattner3082c5a2003-02-18 19:28:33 +00001233
Chris Lattner3ac7c262003-08-13 20:16:26 +00001234 // (setcc1 A, B) | (setcc2 A, B) --> (setcc3 A, B)
1235 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1236 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1237 return R;
1238
Chris Lattner113f4f42002-06-25 16:13:24 +00001239 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001240}
1241
Chris Lattnerc2076352004-02-16 01:20:27 +00001242// XorSelf - Implements: X ^ X --> 0
1243struct XorSelf {
1244 Value *RHS;
1245 XorSelf(Value *rhs) : RHS(rhs) {}
1246 bool shouldApply(Value *LHS) const { return LHS == RHS; }
1247 Instruction *apply(BinaryOperator &Xor) const {
1248 return &Xor;
1249 }
1250};
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001251
1252
Chris Lattner113f4f42002-06-25 16:13:24 +00001253Instruction *InstCombiner::visitXor(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001254 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00001255 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001256
Chris Lattnerc2076352004-02-16 01:20:27 +00001257 // xor X, X = 0, even if X is nested in a sequence of Xor's.
1258 if (Instruction *Result = AssociativeOpt(I, XorSelf(Op1))) {
1259 assert(Result == &I && "AssociativeOpt didn't work?");
Chris Lattnere6794492002-08-12 21:17:25 +00001260 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattnerc2076352004-02-16 01:20:27 +00001261 }
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001262
Chris Lattner97638592003-07-23 21:37:07 +00001263 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001264 // xor X, 0 == X
Chris Lattner97638592003-07-23 21:37:07 +00001265 if (RHS->isNullValue())
Chris Lattnere6794492002-08-12 21:17:25 +00001266 return ReplaceInstUsesWith(I, Op0);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001267
Chris Lattner97638592003-07-23 21:37:07 +00001268 if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) {
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001269 // xor (setcc A, B), true = not (setcc A, B) = setncc A, B
Chris Lattner97638592003-07-23 21:37:07 +00001270 if (SetCondInst *SCI = dyn_cast<SetCondInst>(Op0I))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001271 if (RHS == ConstantBool::True && SCI->hasOneUse())
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001272 return new SetCondInst(SCI->getInverseCondition(),
1273 SCI->getOperand(0), SCI->getOperand(1));
Chris Lattnere5806662003-11-04 23:50:51 +00001274
Chris Lattner8f2f5982003-11-05 01:06:05 +00001275 // ~(c-X) == X-c-1 == X+(-c-1)
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001276 if (Op0I->getOpcode() == Instruction::Sub && RHS->isAllOnesValue())
1277 if (Constant *Op0I0C = dyn_cast<Constant>(Op0I->getOperand(0))) {
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001278 Constant *NegOp0I0C = ConstantExpr::getNeg(Op0I0C);
1279 Constant *ConstantRHS = ConstantExpr::getSub(NegOp0I0C,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001280 ConstantInt::get(I.getType(), 1));
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001281 return BinaryOperator::createAdd(Op0I->getOperand(1), ConstantRHS);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001282 }
Chris Lattner023a4832004-06-18 06:07:51 +00001283
1284 // ~(~X & Y) --> (X | ~Y)
1285 if (Op0I->getOpcode() == Instruction::And && RHS->isAllOnesValue()) {
1286 if (dyn_castNotVal(Op0I->getOperand(1))) Op0I->swapOperands();
1287 if (Value *Op0NotVal = dyn_castNotVal(Op0I->getOperand(0))) {
1288 Instruction *NotY =
1289 BinaryOperator::createNot(Op0I->getOperand(1),
1290 Op0I->getOperand(1)->getName()+".not");
1291 InsertNewInstBefore(NotY, I);
1292 return BinaryOperator::createOr(Op0NotVal, NotY);
1293 }
1294 }
Chris Lattner97638592003-07-23 21:37:07 +00001295
1296 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1)))
Chris Lattnere5806662003-11-04 23:50:51 +00001297 switch (Op0I->getOpcode()) {
1298 case Instruction::Add:
Chris Lattner0f68fa62003-11-04 23:37:10 +00001299 // ~(X-c) --> (-c-1)-X
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001300 if (RHS->isAllOnesValue()) {
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001301 Constant *NegOp0CI = ConstantExpr::getNeg(Op0CI);
1302 return BinaryOperator::createSub(
1303 ConstantExpr::getSub(NegOp0CI,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001304 ConstantInt::get(I.getType(), 1)),
Chris Lattner0f68fa62003-11-04 23:37:10 +00001305 Op0I->getOperand(0));
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001306 }
Chris Lattnere5806662003-11-04 23:50:51 +00001307 break;
1308 case Instruction::And:
Chris Lattner97638592003-07-23 21:37:07 +00001309 // (X & C1) ^ C2 --> (X & C1) | C2 iff (C1&C2) == 0
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001310 if (ConstantExpr::getAnd(RHS, Op0CI)->isNullValue())
1311 return BinaryOperator::createOr(Op0, RHS);
Chris Lattnere5806662003-11-04 23:50:51 +00001312 break;
1313 case Instruction::Or:
Chris Lattner97638592003-07-23 21:37:07 +00001314 // (X | C1) ^ C2 --> (X | C1) & ~C2 iff (C1&C2) == C2
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001315 if (ConstantExpr::getAnd(RHS, Op0CI) == RHS)
Chris Lattnerc8e7e292004-06-10 02:12:35 +00001316 return BinaryOperator::createAnd(Op0, ConstantExpr::getNot(RHS));
Chris Lattnere5806662003-11-04 23:50:51 +00001317 break;
1318 default: break;
Chris Lattner97638592003-07-23 21:37:07 +00001319 }
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001320 }
Chris Lattner183b3362004-04-09 19:05:30 +00001321
1322 // Try to fold constant and into select arguments.
1323 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1324 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1325 return R;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001326 }
1327
Chris Lattnerbb74e222003-03-10 23:06:50 +00001328 if (Value *X = dyn_castNotVal(Op0)) // ~A ^ A == -1
Chris Lattner3082c5a2003-02-18 19:28:33 +00001329 if (X == Op1)
1330 return ReplaceInstUsesWith(I,
1331 ConstantIntegral::getAllOnesValue(I.getType()));
1332
Chris Lattnerbb74e222003-03-10 23:06:50 +00001333 if (Value *X = dyn_castNotVal(Op1)) // A ^ ~A == -1
Chris Lattner3082c5a2003-02-18 19:28:33 +00001334 if (X == Op0)
1335 return ReplaceInstUsesWith(I,
1336 ConstantIntegral::getAllOnesValue(I.getType()));
1337
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001338 if (Instruction *Op1I = dyn_cast<Instruction>(Op1))
Chris Lattnerb36d9082004-02-16 03:54:20 +00001339 if (Op1I->getOpcode() == Instruction::Or) {
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001340 if (Op1I->getOperand(0) == Op0) { // B^(B|A) == (A|B)^B
1341 cast<BinaryOperator>(Op1I)->swapOperands();
1342 I.swapOperands();
1343 std::swap(Op0, Op1);
1344 } else if (Op1I->getOperand(1) == Op0) { // B^(A|B) == (A|B)^B
1345 I.swapOperands();
1346 std::swap(Op0, Op1);
Chris Lattnerb36d9082004-02-16 03:54:20 +00001347 }
1348 } else if (Op1I->getOpcode() == Instruction::Xor) {
1349 if (Op0 == Op1I->getOperand(0)) // A^(A^B) == B
1350 return ReplaceInstUsesWith(I, Op1I->getOperand(1));
1351 else if (Op0 == Op1I->getOperand(1)) // A^(B^A) == B
1352 return ReplaceInstUsesWith(I, Op1I->getOperand(0));
1353 }
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001354
1355 if (Instruction *Op0I = dyn_cast<Instruction>(Op0))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001356 if (Op0I->getOpcode() == Instruction::Or && Op0I->hasOneUse()) {
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001357 if (Op0I->getOperand(0) == Op1) // (B|A)^B == (A|B)^B
1358 cast<BinaryOperator>(Op0I)->swapOperands();
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001359 if (Op0I->getOperand(1) == Op1) { // (A|B)^B == A & ~B
Chris Lattner396dbfe2004-06-09 05:08:07 +00001360 Value *NotB = InsertNewInstBefore(BinaryOperator::createNot(Op1,
1361 Op1->getName()+".not"), I);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001362 return BinaryOperator::createAnd(Op0I->getOperand(0), NotB);
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001363 }
Chris Lattnerb36d9082004-02-16 03:54:20 +00001364 } else if (Op0I->getOpcode() == Instruction::Xor) {
1365 if (Op1 == Op0I->getOperand(0)) // (A^B)^A == B
1366 return ReplaceInstUsesWith(I, Op0I->getOperand(1));
1367 else if (Op1 == Op0I->getOperand(1)) // (B^A)^A == B
1368 return ReplaceInstUsesWith(I, Op0I->getOperand(0));
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001369 }
1370
Chris Lattner7fb29e12003-03-11 00:12:48 +00001371 // (A & C1)^(B & C2) -> (A & C1)|(B & C2) iff C1^C2 == 0
1372 if (Constant *C1 = dyn_castMaskingAnd(Op0))
1373 if (Constant *C2 = dyn_castMaskingAnd(Op1))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001374 if (ConstantExpr::getAnd(C1, C2)->isNullValue())
1375 return BinaryOperator::createOr(Op0, Op1);
Chris Lattner7fb29e12003-03-11 00:12:48 +00001376
Chris Lattner3ac7c262003-08-13 20:16:26 +00001377 // (setcc1 A, B) ^ (setcc2 A, B) --> (setcc3 A, B)
1378 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1379 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1380 return R;
1381
Chris Lattner113f4f42002-06-25 16:13:24 +00001382 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001383}
1384
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001385// AddOne, SubOne - Add or subtract a constant one from an integer constant...
1386static Constant *AddOne(ConstantInt *C) {
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001387 return ConstantExpr::getAdd(C, ConstantInt::get(C->getType(), 1));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001388}
1389static Constant *SubOne(ConstantInt *C) {
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001390 return ConstantExpr::getSub(C, ConstantInt::get(C->getType(), 1));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001391}
1392
Chris Lattner1fc23f32002-05-09 20:11:54 +00001393// isTrueWhenEqual - Return true if the specified setcondinst instruction is
1394// true when both operands are equal...
1395//
Chris Lattner113f4f42002-06-25 16:13:24 +00001396static bool isTrueWhenEqual(Instruction &I) {
1397 return I.getOpcode() == Instruction::SetEQ ||
1398 I.getOpcode() == Instruction::SetGE ||
1399 I.getOpcode() == Instruction::SetLE;
Chris Lattner1fc23f32002-05-09 20:11:54 +00001400}
1401
Chris Lattner113f4f42002-06-25 16:13:24 +00001402Instruction *InstCombiner::visitSetCondInst(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001403 bool Changed = SimplifyCommutative(I);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001404 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
1405 const Type *Ty = Op0->getType();
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001406
1407 // setcc X, X
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001408 if (Op0 == Op1)
1409 return ReplaceInstUsesWith(I, ConstantBool::get(isTrueWhenEqual(I)));
Chris Lattner1fc23f32002-05-09 20:11:54 +00001410
Chris Lattnerd07283a2003-08-13 05:38:46 +00001411 // setcc <global/alloca*>, 0 - Global/Stack value addresses are never null!
1412 if (isa<ConstantPointerNull>(Op1) &&
1413 (isa<GlobalValue>(Op0) || isa<AllocaInst>(Op0)))
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001414 return ReplaceInstUsesWith(I, ConstantBool::get(!isTrueWhenEqual(I)));
1415
Chris Lattnerd07283a2003-08-13 05:38:46 +00001416
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001417 // setcc's with boolean values can always be turned into bitwise operations
1418 if (Ty == Type::BoolTy) {
1419 // If this is <, >, or !=, we can change this into a simple xor instruction
1420 if (!isTrueWhenEqual(I))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001421 return BinaryOperator::createXor(Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001422
1423 // Otherwise we need to make a temporary intermediate instruction and insert
1424 // it into the instruction stream. This is what we are after:
1425 //
1426 // seteq bool %A, %B -> ~(A^B)
1427 // setle bool %A, %B -> ~A | B
1428 // setge bool %A, %B -> A | ~B
1429 //
1430 if (I.getOpcode() == Instruction::SetEQ) { // seteq case
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001431 Instruction *Xor = BinaryOperator::createXor(Op0, Op1, I.getName()+"tmp");
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001432 InsertNewInstBefore(Xor, I);
Chris Lattner16930792003-11-03 04:25:02 +00001433 return BinaryOperator::createNot(Xor);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001434 }
1435
1436 // Handle the setXe cases...
1437 assert(I.getOpcode() == Instruction::SetGE ||
1438 I.getOpcode() == Instruction::SetLE);
1439
1440 if (I.getOpcode() == Instruction::SetGE)
1441 std::swap(Op0, Op1); // Change setge -> setle
1442
1443 // Now we just have the SetLE case.
Chris Lattner31ae8632002-08-14 17:51:49 +00001444 Instruction *Not = BinaryOperator::createNot(Op0, I.getName()+"tmp");
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001445 InsertNewInstBefore(Not, I);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001446 return BinaryOperator::createOr(Not, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001447 }
1448
Chris Lattner2dd01742004-06-09 04:24:29 +00001449 // See if we are doing a comparison between a constant and an instruction that
1450 // can be folded into the comparison.
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001451 if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) {
Chris Lattnere1e10e12004-05-25 06:32:08 +00001452 if (Instruction *LHSI = dyn_cast<Instruction>(Op0))
Chris Lattner2dd01742004-06-09 04:24:29 +00001453 if (LHSI->hasOneUse())
Chris Lattner35167c32004-06-09 07:59:58 +00001454 switch (LHSI->getOpcode()) {
1455 case Instruction::And:
1456 if (isa<ConstantInt>(LHSI->getOperand(1))) {
1457
Chris Lattnere1e10e12004-05-25 06:32:08 +00001458
Chris Lattner35167c32004-06-09 07:59:58 +00001459 // If this is: (X >> C1) & C2 != C3 (where any shift and any compare
1460 // could exist), turn it into (X & (C2 << C1)) != (C3 << C1). This
1461 // happens a LOT in code produced by the C front-end, for bitfield
1462 // access.
1463 if (LHSI->getOperand(0)->hasOneUse())
1464 if (ShiftInst *Shift = dyn_cast<ShiftInst>(LHSI->getOperand(0)))
1465 if (ConstantUInt *ShAmt =
1466 dyn_cast<ConstantUInt>(Shift->getOperand(1))) {
1467 ConstantInt *AndCST = cast<ConstantInt>(LHSI->getOperand(1));
1468
1469 // We can fold this as long as we can't shift unknown bits
1470 // into the mask. This can only happen with signed shift
1471 // rights, as they sign-extend.
1472 const Type *Ty = Shift->getType();
1473 if (Shift->getOpcode() != Instruction::Shr ||
1474 Shift->getType()->isUnsigned() ||
1475 // To test for the bad case of the signed shr, see if any
1476 // of the bits shifted in could be tested after the mask.
1477 ConstantExpr::getAnd(ConstantExpr::getShl(ConstantInt::getAllOnesValue(Ty), ConstantUInt::get(Type::UByteTy, Ty->getPrimitiveSize()*8-ShAmt->getValue())), AndCST)->isNullValue()) {
1478 unsigned ShiftOp = Shift->getOpcode() == Instruction::Shl
1479 ? Instruction::Shr : Instruction::Shl;
1480 I.setOperand(1, ConstantExpr::get(ShiftOp, CI, ShAmt));
1481 LHSI->setOperand(1,ConstantExpr::get(ShiftOp,AndCST,ShAmt));
1482 LHSI->setOperand(0, Shift->getOperand(0));
1483 WorkList.push_back(Shift); // Shift is probably dead.
1484 AddUsesToWorkList(I);
1485 return &I;
1486 }
Chris Lattner2dd01742004-06-09 04:24:29 +00001487 }
Chris Lattner35167c32004-06-09 07:59:58 +00001488 }
1489 break;
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001490 case Instruction::Div:
1491 if (0 && isa<ConstantInt>(LHSI->getOperand(1))) {
1492 std::cerr << "COULD FOLD: " << *LHSI;
1493 std::cerr << "COULD FOLD: " << I << "\n";
1494 }
1495 break;
Chris Lattner35167c32004-06-09 07:59:58 +00001496 case Instruction::Select:
Chris Lattner2dd01742004-06-09 04:24:29 +00001497 // If either operand of the select is a constant, we can fold the
1498 // comparison into the select arms, which will cause one to be
1499 // constant folded and the select turned into a bitwise or.
1500 Value *Op1 = 0, *Op2 = 0;
Chris Lattner35167c32004-06-09 07:59:58 +00001501 if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(1))) {
Chris Lattner2dd01742004-06-09 04:24:29 +00001502 // Fold the known value into the constant operand.
1503 Op1 = ConstantExpr::get(I.getOpcode(), C, CI);
1504 // Insert a new SetCC of the other select operand.
1505 Op2 = InsertNewInstBefore(new SetCondInst(I.getOpcode(),
Chris Lattner35167c32004-06-09 07:59:58 +00001506 LHSI->getOperand(2), CI,
Chris Lattner2dd01742004-06-09 04:24:29 +00001507 I.getName()), I);
Chris Lattner35167c32004-06-09 07:59:58 +00001508 } else if (Constant *C = dyn_cast<Constant>(LHSI->getOperand(2))) {
Chris Lattner2dd01742004-06-09 04:24:29 +00001509 // Fold the known value into the constant operand.
1510 Op2 = ConstantExpr::get(I.getOpcode(), C, CI);
1511 // Insert a new SetCC of the other select operand.
1512 Op1 = InsertNewInstBefore(new SetCondInst(I.getOpcode(),
Chris Lattner35167c32004-06-09 07:59:58 +00001513 LHSI->getOperand(1), CI,
Chris Lattner2dd01742004-06-09 04:24:29 +00001514 I.getName()), I);
1515 }
1516
1517 if (Op1)
Chris Lattner35167c32004-06-09 07:59:58 +00001518 return new SelectInst(LHSI->getOperand(0), Op1, Op2);
1519 break;
Chris Lattner2dd01742004-06-09 04:24:29 +00001520 }
Chris Lattnere1e10e12004-05-25 06:32:08 +00001521
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001522 // Simplify seteq and setne instructions...
1523 if (I.getOpcode() == Instruction::SetEQ ||
1524 I.getOpcode() == Instruction::SetNE) {
1525 bool isSetNE = I.getOpcode() == Instruction::SetNE;
1526
Chris Lattnercfbce7c2003-07-23 17:26:36 +00001527 // If the first operand is (and|or|xor) with a constant, and the second
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001528 // operand is a constant, simplify a bit.
Chris Lattnerc992add2003-08-13 05:33:12 +00001529 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0)) {
1530 switch (BO->getOpcode()) {
Chris Lattner23b47b62004-07-06 07:38:18 +00001531 case Instruction::Rem:
1532 // If we have a signed (X % (2^c)) == 0, turn it into an unsigned one.
1533 if (CI->isNullValue() && isa<ConstantSInt>(BO->getOperand(1)) &&
1534 BO->hasOneUse() &&
1535 cast<ConstantSInt>(BO->getOperand(1))->getValue() > 1)
1536 if (unsigned L2 =
1537 Log2(cast<ConstantSInt>(BO->getOperand(1))->getValue())) {
1538 const Type *UTy = BO->getType()->getUnsignedVersion();
1539 Value *NewX = InsertNewInstBefore(new CastInst(BO->getOperand(0),
1540 UTy, "tmp"), I);
1541 Constant *RHSCst = ConstantUInt::get(UTy, 1ULL << L2);
1542 Value *NewRem =InsertNewInstBefore(BinaryOperator::createRem(NewX,
1543 RHSCst, BO->getName()), I);
1544 return BinaryOperator::create(I.getOpcode(), NewRem,
1545 Constant::getNullValue(UTy));
1546 }
1547 break;
1548
Chris Lattnerc992add2003-08-13 05:33:12 +00001549 case Instruction::Add:
Chris Lattner6e079362004-06-27 22:51:36 +00001550 // Replace ((add A, B) != C) with (A != C-B) if B & C are constants.
1551 if (ConstantInt *BOp1C = dyn_cast<ConstantInt>(BO->getOperand(1))) {
1552 return new SetCondInst(I.getOpcode(), BO->getOperand(0),
1553 ConstantExpr::getSub(CI, BOp1C));
1554 } else if (CI->isNullValue()) {
Chris Lattnerc992add2003-08-13 05:33:12 +00001555 // Replace ((add A, B) != 0) with (A != -B) if A or B is
1556 // efficiently invertible, or if the add has just this one use.
1557 Value *BOp0 = BO->getOperand(0), *BOp1 = BO->getOperand(1);
Chris Lattner6e079362004-06-27 22:51:36 +00001558
Chris Lattnerc992add2003-08-13 05:33:12 +00001559 if (Value *NegVal = dyn_castNegVal(BOp1))
1560 return new SetCondInst(I.getOpcode(), BOp0, NegVal);
1561 else if (Value *NegVal = dyn_castNegVal(BOp0))
1562 return new SetCondInst(I.getOpcode(), NegVal, BOp1);
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001563 else if (BO->hasOneUse()) {
Chris Lattnerc992add2003-08-13 05:33:12 +00001564 Instruction *Neg = BinaryOperator::createNeg(BOp1, BO->getName());
1565 BO->setName("");
1566 InsertNewInstBefore(Neg, I);
1567 return new SetCondInst(I.getOpcode(), BOp0, Neg);
1568 }
1569 }
1570 break;
1571 case Instruction::Xor:
1572 // For the xor case, we can xor two constants together, eliminating
1573 // the explicit xor.
1574 if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1)))
1575 return BinaryOperator::create(I.getOpcode(), BO->getOperand(0),
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001576 ConstantExpr::getXor(CI, BOC));
Chris Lattnerc992add2003-08-13 05:33:12 +00001577
1578 // FALLTHROUGH
1579 case Instruction::Sub:
1580 // Replace (([sub|xor] A, B) != 0) with (A != B)
1581 if (CI->isNullValue())
1582 return new SetCondInst(I.getOpcode(), BO->getOperand(0),
1583 BO->getOperand(1));
1584 break;
1585
1586 case Instruction::Or:
1587 // If bits are being or'd in that are not present in the constant we
1588 // are comparing against, then the comparison could never succeed!
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001589 if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1))) {
Chris Lattnerc8e7e292004-06-10 02:12:35 +00001590 Constant *NotCI = ConstantExpr::getNot(CI);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001591 if (!ConstantExpr::getAnd(BOC, NotCI)->isNullValue())
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001592 return ReplaceInstUsesWith(I, ConstantBool::get(isSetNE));
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001593 }
Chris Lattnerc992add2003-08-13 05:33:12 +00001594 break;
1595
1596 case Instruction::And:
1597 if (ConstantInt *BOC = dyn_cast<ConstantInt>(BO->getOperand(1))) {
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001598 // If bits are being compared against that are and'd out, then the
1599 // comparison can never succeed!
Chris Lattnerc8e7e292004-06-10 02:12:35 +00001600 if (!ConstantExpr::getAnd(CI,
1601 ConstantExpr::getNot(BOC))->isNullValue())
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001602 return ReplaceInstUsesWith(I, ConstantBool::get(isSetNE));
Chris Lattnerc992add2003-08-13 05:33:12 +00001603
Chris Lattner35167c32004-06-09 07:59:58 +00001604 // If we have ((X & C) == C), turn it into ((X & C) != 0).
Chris Lattneree59d4b2004-06-10 02:33:20 +00001605 if (CI == BOC && isOneBitSet(CI))
Chris Lattner35167c32004-06-09 07:59:58 +00001606 return new SetCondInst(isSetNE ? Instruction::SetEQ :
1607 Instruction::SetNE, Op0,
1608 Constant::getNullValue(CI->getType()));
Chris Lattner35167c32004-06-09 07:59:58 +00001609
Chris Lattnerc992add2003-08-13 05:33:12 +00001610 // Replace (and X, (1 << size(X)-1) != 0) with x < 0, converting X
1611 // to be a signed value as appropriate.
1612 if (isSignBit(BOC)) {
1613 Value *X = BO->getOperand(0);
1614 // If 'X' is not signed, insert a cast now...
1615 if (!BOC->getType()->isSigned()) {
Chris Lattner97bfcea2004-06-17 18:16:02 +00001616 const Type *DestTy = BOC->getType()->getSignedVersion();
Chris Lattnerc992add2003-08-13 05:33:12 +00001617 CastInst *NewCI = new CastInst(X,DestTy,X->getName()+".signed");
1618 InsertNewInstBefore(NewCI, I);
1619 X = NewCI;
1620 }
1621 return new SetCondInst(isSetNE ? Instruction::SetLT :
1622 Instruction::SetGE, X,
1623 Constant::getNullValue(X->getType()));
1624 }
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001625 }
Chris Lattnerc992add2003-08-13 05:33:12 +00001626 default: break;
1627 }
1628 }
Chris Lattner2b55ea32004-02-23 07:16:20 +00001629 } else { // Not a SetEQ/SetNE
1630 // If the LHS is a cast from an integral value of the same size,
1631 if (CastInst *Cast = dyn_cast<CastInst>(Op0)) {
1632 Value *CastOp = Cast->getOperand(0);
1633 const Type *SrcTy = CastOp->getType();
1634 unsigned SrcTySize = SrcTy->getPrimitiveSize();
1635 if (SrcTy != Cast->getType() && SrcTy->isInteger() &&
1636 SrcTySize == Cast->getType()->getPrimitiveSize()) {
1637 assert((SrcTy->isSigned() ^ Cast->getType()->isSigned()) &&
1638 "Source and destination signednesses should differ!");
1639 if (Cast->getType()->isSigned()) {
1640 // If this is a signed comparison, check for comparisons in the
1641 // vicinity of zero.
1642 if (I.getOpcode() == Instruction::SetLT && CI->isNullValue())
1643 // X < 0 => x > 127
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001644 return BinaryOperator::createSetGT(CastOp,
Chris Lattner2b55ea32004-02-23 07:16:20 +00001645 ConstantUInt::get(SrcTy, (1ULL << (SrcTySize*8-1))-1));
1646 else if (I.getOpcode() == Instruction::SetGT &&
1647 cast<ConstantSInt>(CI)->getValue() == -1)
1648 // X > -1 => x < 128
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001649 return BinaryOperator::createSetLT(CastOp,
Chris Lattner2b55ea32004-02-23 07:16:20 +00001650 ConstantUInt::get(SrcTy, 1ULL << (SrcTySize*8-1)));
1651 } else {
1652 ConstantUInt *CUI = cast<ConstantUInt>(CI);
1653 if (I.getOpcode() == Instruction::SetLT &&
1654 CUI->getValue() == 1ULL << (SrcTySize*8-1))
1655 // X < 128 => X > -1
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001656 return BinaryOperator::createSetGT(CastOp,
1657 ConstantSInt::get(SrcTy, -1));
Chris Lattner2b55ea32004-02-23 07:16:20 +00001658 else if (I.getOpcode() == Instruction::SetGT &&
1659 CUI->getValue() == (1ULL << (SrcTySize*8-1))-1)
1660 // X > 127 => X < 0
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001661 return BinaryOperator::createSetLT(CastOp,
1662 Constant::getNullValue(SrcTy));
Chris Lattner2b55ea32004-02-23 07:16:20 +00001663 }
1664 }
1665 }
Chris Lattnere967b342003-06-04 05:10:11 +00001666 }
Chris Lattner791ac1a2003-06-01 03:35:25 +00001667
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001668 // Check to see if we are comparing against the minimum or maximum value...
Chris Lattnere6794492002-08-12 21:17:25 +00001669 if (CI->isMinValue()) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001670 if (I.getOpcode() == Instruction::SetLT) // A < MIN -> FALSE
1671 return ReplaceInstUsesWith(I, ConstantBool::False);
1672 if (I.getOpcode() == Instruction::SetGE) // A >= MIN -> TRUE
1673 return ReplaceInstUsesWith(I, ConstantBool::True);
1674 if (I.getOpcode() == Instruction::SetLE) // A <= MIN -> A == MIN
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001675 return BinaryOperator::createSetEQ(Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001676 if (I.getOpcode() == Instruction::SetGT) // A > MIN -> A != MIN
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001677 return BinaryOperator::createSetNE(Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001678
Chris Lattnere6794492002-08-12 21:17:25 +00001679 } else if (CI->isMaxValue()) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001680 if (I.getOpcode() == Instruction::SetGT) // A > MAX -> FALSE
1681 return ReplaceInstUsesWith(I, ConstantBool::False);
1682 if (I.getOpcode() == Instruction::SetLE) // A <= MAX -> TRUE
1683 return ReplaceInstUsesWith(I, ConstantBool::True);
1684 if (I.getOpcode() == Instruction::SetGE) // A >= MAX -> A == MAX
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001685 return BinaryOperator::createSetEQ(Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001686 if (I.getOpcode() == Instruction::SetLT) // A < MAX -> A != MAX
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001687 return BinaryOperator::createSetNE(Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001688
1689 // Comparing against a value really close to min or max?
1690 } else if (isMinValuePlusOne(CI)) {
1691 if (I.getOpcode() == Instruction::SetLT) // A < MIN+1 -> A == MIN
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001692 return BinaryOperator::createSetEQ(Op0, SubOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001693 if (I.getOpcode() == Instruction::SetGE) // A >= MIN-1 -> A != MIN
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001694 return BinaryOperator::createSetNE(Op0, SubOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001695
1696 } else if (isMaxValueMinusOne(CI)) {
1697 if (I.getOpcode() == Instruction::SetGT) // A > MAX-1 -> A == MAX
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001698 return BinaryOperator::createSetEQ(Op0, AddOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001699 if (I.getOpcode() == Instruction::SetLE) // A <= MAX-1 -> A != MAX
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001700 return BinaryOperator::createSetNE(Op0, AddOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001701 }
Chris Lattner59611142004-02-23 05:47:48 +00001702
1703 // If we still have a setle or setge instruction, turn it into the
1704 // appropriate setlt or setgt instruction. Since the border cases have
1705 // already been handled above, this requires little checking.
1706 //
1707 if (I.getOpcode() == Instruction::SetLE)
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001708 return BinaryOperator::createSetLT(Op0, AddOne(CI));
Chris Lattner59611142004-02-23 05:47:48 +00001709 if (I.getOpcode() == Instruction::SetGE)
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001710 return BinaryOperator::createSetGT(Op0, SubOne(CI));
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001711 }
1712
Chris Lattner16930792003-11-03 04:25:02 +00001713 // Test to see if the operands of the setcc are casted versions of other
1714 // values. If the cast can be stripped off both arguments, we do so now.
Chris Lattner6444c372003-11-03 05:17:03 +00001715 if (CastInst *CI = dyn_cast<CastInst>(Op0)) {
1716 Value *CastOp0 = CI->getOperand(0);
1717 if (CastOp0->getType()->isLosslesslyConvertibleTo(CI->getType()) &&
Chris Lattner7d2a5392004-03-13 23:54:27 +00001718 (isa<Constant>(Op1) || isa<CastInst>(Op1)) &&
Chris Lattner16930792003-11-03 04:25:02 +00001719 (I.getOpcode() == Instruction::SetEQ ||
1720 I.getOpcode() == Instruction::SetNE)) {
1721 // We keep moving the cast from the left operand over to the right
1722 // operand, where it can often be eliminated completely.
Chris Lattner6444c372003-11-03 05:17:03 +00001723 Op0 = CastOp0;
Chris Lattner16930792003-11-03 04:25:02 +00001724
1725 // If operand #1 is a cast instruction, see if we can eliminate it as
1726 // well.
Chris Lattner6444c372003-11-03 05:17:03 +00001727 if (CastInst *CI2 = dyn_cast<CastInst>(Op1))
1728 if (CI2->getOperand(0)->getType()->isLosslesslyConvertibleTo(
Chris Lattner16930792003-11-03 04:25:02 +00001729 Op0->getType()))
Chris Lattner6444c372003-11-03 05:17:03 +00001730 Op1 = CI2->getOperand(0);
Chris Lattner16930792003-11-03 04:25:02 +00001731
1732 // If Op1 is a constant, we can fold the cast into the constant.
1733 if (Op1->getType() != Op0->getType())
1734 if (Constant *Op1C = dyn_cast<Constant>(Op1)) {
1735 Op1 = ConstantExpr::getCast(Op1C, Op0->getType());
1736 } else {
1737 // Otherwise, cast the RHS right before the setcc
1738 Op1 = new CastInst(Op1, Op0->getType(), Op1->getName());
1739 InsertNewInstBefore(cast<Instruction>(Op1), I);
1740 }
1741 return BinaryOperator::create(I.getOpcode(), Op0, Op1);
1742 }
1743
Chris Lattner6444c372003-11-03 05:17:03 +00001744 // Handle the special case of: setcc (cast bool to X), <cst>
1745 // This comes up when you have code like
1746 // int X = A < B;
1747 // if (X) ...
1748 // For generality, we handle any zero-extension of any operand comparison
1749 // with a constant.
1750 if (ConstantInt *ConstantRHS = dyn_cast<ConstantInt>(Op1)) {
1751 const Type *SrcTy = CastOp0->getType();
1752 const Type *DestTy = Op0->getType();
1753 if (SrcTy->getPrimitiveSize() < DestTy->getPrimitiveSize() &&
1754 (SrcTy->isUnsigned() || SrcTy == Type::BoolTy)) {
1755 // Ok, we have an expansion of operand 0 into a new type. Get the
1756 // constant value, masink off bits which are not set in the RHS. These
1757 // could be set if the destination value is signed.
1758 uint64_t ConstVal = ConstantRHS->getRawValue();
1759 ConstVal &= (1ULL << DestTy->getPrimitiveSize()*8)-1;
1760
1761 // If the constant we are comparing it with has high bits set, which
1762 // don't exist in the original value, the values could never be equal,
1763 // because the source would be zero extended.
1764 unsigned SrcBits =
1765 SrcTy == Type::BoolTy ? 1 : SrcTy->getPrimitiveSize()*8;
Chris Lattner7c94d112003-11-05 17:31:36 +00001766 bool HasSignBit = ConstVal & (1ULL << (DestTy->getPrimitiveSize()*8-1));
1767 if (ConstVal & ~((1ULL << SrcBits)-1)) {
Chris Lattner6444c372003-11-03 05:17:03 +00001768 switch (I.getOpcode()) {
1769 default: assert(0 && "Unknown comparison type!");
1770 case Instruction::SetEQ:
1771 return ReplaceInstUsesWith(I, ConstantBool::False);
1772 case Instruction::SetNE:
1773 return ReplaceInstUsesWith(I, ConstantBool::True);
1774 case Instruction::SetLT:
1775 case Instruction::SetLE:
1776 if (DestTy->isSigned() && HasSignBit)
1777 return ReplaceInstUsesWith(I, ConstantBool::False);
1778 return ReplaceInstUsesWith(I, ConstantBool::True);
1779 case Instruction::SetGT:
1780 case Instruction::SetGE:
1781 if (DestTy->isSigned() && HasSignBit)
1782 return ReplaceInstUsesWith(I, ConstantBool::True);
1783 return ReplaceInstUsesWith(I, ConstantBool::False);
1784 }
1785 }
1786
1787 // Otherwise, we can replace the setcc with a setcc of the smaller
1788 // operand value.
1789 Op1 = ConstantExpr::getCast(cast<Constant>(Op1), SrcTy);
1790 return BinaryOperator::create(I.getOpcode(), CastOp0, Op1);
1791 }
1792 }
1793 }
Chris Lattner113f4f42002-06-25 16:13:24 +00001794 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001795}
1796
1797
1798
Chris Lattnere8d6c602003-03-10 19:16:08 +00001799Instruction *InstCombiner::visitShiftInst(ShiftInst &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +00001800 assert(I.getOperand(1)->getType() == Type::UByteTy);
1801 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001802 bool isLeftShift = I.getOpcode() == Instruction::Shl;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001803
1804 // shl X, 0 == X and shr X, 0 == X
1805 // shl 0, X == 0 and shr 0, X == 0
1806 if (Op1 == Constant::getNullValue(Type::UByteTy) ||
Chris Lattnere6794492002-08-12 21:17:25 +00001807 Op0 == Constant::getNullValue(Op0->getType()))
1808 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001809
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001810 // shr int -1, X = -1 (for any arithmetic shift rights of ~0)
1811 if (!isLeftShift)
1812 if (ConstantSInt *CSI = dyn_cast<ConstantSInt>(Op0))
1813 if (CSI->isAllOnesValue())
1814 return ReplaceInstUsesWith(I, CSI);
1815
Chris Lattner183b3362004-04-09 19:05:30 +00001816 // Try to fold constant and into select arguments.
1817 if (isa<Constant>(Op0))
1818 if (SelectInst *SI = dyn_cast<SelectInst>(Op1))
1819 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1820 return R;
1821
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001822 if (ConstantUInt *CUI = dyn_cast<ConstantUInt>(Op1)) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001823 // shl uint X, 32 = 0 and shr ubyte Y, 9 = 0, ... just don't eliminate shr
1824 // of a signed value.
1825 //
Chris Lattnere8d6c602003-03-10 19:16:08 +00001826 unsigned TypeBits = Op0->getType()->getPrimitiveSize()*8;
Chris Lattnerf5ce2542004-02-23 20:30:06 +00001827 if (CUI->getValue() >= TypeBits) {
1828 if (!Op0->getType()->isSigned() || isLeftShift)
1829 return ReplaceInstUsesWith(I, Constant::getNullValue(Op0->getType()));
1830 else {
1831 I.setOperand(1, ConstantUInt::get(Type::UByteTy, TypeBits-1));
1832 return &I;
1833 }
1834 }
Chris Lattner55f3d942002-09-10 23:04:09 +00001835
Chris Lattnerede3fe02003-08-13 04:18:28 +00001836 // ((X*C1) << C2) == (X * (C1 << C2))
1837 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0))
1838 if (BO->getOpcode() == Instruction::Mul && isLeftShift)
1839 if (Constant *BOOp = dyn_cast<Constant>(BO->getOperand(1)))
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001840 return BinaryOperator::createMul(BO->getOperand(0),
1841 ConstantExpr::getShl(BOOp, CUI));
Chris Lattnerede3fe02003-08-13 04:18:28 +00001842
Chris Lattner183b3362004-04-09 19:05:30 +00001843 // Try to fold constant and into select arguments.
1844 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1845 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1846 return R;
Chris Lattnerede3fe02003-08-13 04:18:28 +00001847
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001848 // If the operand is an bitwise operator with a constant RHS, and the
1849 // shift is the only use, we can pull it out of the shift.
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001850 if (Op0->hasOneUse())
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001851 if (BinaryOperator *Op0BO = dyn_cast<BinaryOperator>(Op0))
1852 if (ConstantInt *Op0C = dyn_cast<ConstantInt>(Op0BO->getOperand(1))) {
1853 bool isValid = true; // Valid only for And, Or, Xor
1854 bool highBitSet = false; // Transform if high bit of constant set?
1855
1856 switch (Op0BO->getOpcode()) {
1857 default: isValid = false; break; // Do not perform transform!
1858 case Instruction::Or:
1859 case Instruction::Xor:
1860 highBitSet = false;
1861 break;
1862 case Instruction::And:
1863 highBitSet = true;
1864 break;
1865 }
1866
1867 // If this is a signed shift right, and the high bit is modified
1868 // by the logical operation, do not perform the transformation.
1869 // The highBitSet boolean indicates the value of the high bit of
1870 // the constant which would cause it to be modified for this
1871 // operation.
1872 //
1873 if (isValid && !isLeftShift && !I.getType()->isUnsigned()) {
1874 uint64_t Val = Op0C->getRawValue();
1875 isValid = ((Val & (1 << (TypeBits-1))) != 0) == highBitSet;
1876 }
1877
1878 if (isValid) {
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001879 Constant *NewRHS = ConstantExpr::get(I.getOpcode(), Op0C, CUI);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001880
1881 Instruction *NewShift =
1882 new ShiftInst(I.getOpcode(), Op0BO->getOperand(0), CUI,
1883 Op0BO->getName());
1884 Op0BO->setName("");
1885 InsertNewInstBefore(NewShift, I);
1886
1887 return BinaryOperator::create(Op0BO->getOpcode(), NewShift,
1888 NewRHS);
1889 }
1890 }
1891
Chris Lattner3204d4e2003-07-24 17:52:58 +00001892 // If this is a shift of a shift, see if we can fold the two together...
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001893 if (ShiftInst *Op0SI = dyn_cast<ShiftInst>(Op0))
Chris Lattnerab780df2003-07-24 18:38:56 +00001894 if (ConstantUInt *ShiftAmt1C =
1895 dyn_cast<ConstantUInt>(Op0SI->getOperand(1))) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001896 unsigned ShiftAmt1 = ShiftAmt1C->getValue();
1897 unsigned ShiftAmt2 = CUI->getValue();
1898
1899 // Check for (A << c1) << c2 and (A >> c1) >> c2
1900 if (I.getOpcode() == Op0SI->getOpcode()) {
1901 unsigned Amt = ShiftAmt1+ShiftAmt2; // Fold into one big shift...
Chris Lattnerf5ce2542004-02-23 20:30:06 +00001902 if (Op0->getType()->getPrimitiveSize()*8 < Amt)
1903 Amt = Op0->getType()->getPrimitiveSize()*8;
Chris Lattner3204d4e2003-07-24 17:52:58 +00001904 return new ShiftInst(I.getOpcode(), Op0SI->getOperand(0),
1905 ConstantUInt::get(Type::UByteTy, Amt));
1906 }
1907
Chris Lattnerab780df2003-07-24 18:38:56 +00001908 // Check for (A << c1) >> c2 or visaversa. If we are dealing with
1909 // signed types, we can only support the (A >> c1) << c2 configuration,
1910 // because it can not turn an arbitrary bit of A into a sign bit.
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001911 if (I.getType()->isUnsigned() || isLeftShift) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001912 // Calculate bitmask for what gets shifted off the edge...
1913 Constant *C = ConstantIntegral::getAllOnesValue(I.getType());
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001914 if (isLeftShift)
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001915 C = ConstantExpr::getShl(C, ShiftAmt1C);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001916 else
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001917 C = ConstantExpr::getShr(C, ShiftAmt1C);
Chris Lattner3204d4e2003-07-24 17:52:58 +00001918
1919 Instruction *Mask =
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00001920 BinaryOperator::createAnd(Op0SI->getOperand(0), C,
1921 Op0SI->getOperand(0)->getName()+".mask");
Chris Lattner3204d4e2003-07-24 17:52:58 +00001922 InsertNewInstBefore(Mask, I);
1923
1924 // Figure out what flavor of shift we should use...
1925 if (ShiftAmt1 == ShiftAmt2)
1926 return ReplaceInstUsesWith(I, Mask); // (A << c) >> c === A & c2
1927 else if (ShiftAmt1 < ShiftAmt2) {
1928 return new ShiftInst(I.getOpcode(), Mask,
1929 ConstantUInt::get(Type::UByteTy, ShiftAmt2-ShiftAmt1));
1930 } else {
1931 return new ShiftInst(Op0SI->getOpcode(), Mask,
1932 ConstantUInt::get(Type::UByteTy, ShiftAmt1-ShiftAmt2));
1933 }
1934 }
1935 }
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001936 }
Chris Lattner2e0fb392002-10-08 16:16:40 +00001937
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001938 return 0;
1939}
1940
Chris Lattner4e2dbc62004-07-20 00:59:32 +00001941enum CastType {
1942 Noop = 0,
1943 Truncate = 1,
1944 Signext = 2,
1945 Zeroext = 3
1946};
1947
1948/// getCastType - In the future, we will split the cast instruction into these
1949/// various types. Until then, we have to do the analysis here.
1950static CastType getCastType(const Type *Src, const Type *Dest) {
1951 assert(Src->isIntegral() && Dest->isIntegral() &&
1952 "Only works on integral types!");
1953 unsigned SrcSize = Src->getPrimitiveSize()*8;
1954 if (Src == Type::BoolTy) SrcSize = 1;
1955 unsigned DestSize = Dest->getPrimitiveSize()*8;
1956 if (Dest == Type::BoolTy) DestSize = 1;
1957
1958 if (SrcSize == DestSize) return Noop;
1959 if (SrcSize > DestSize) return Truncate;
1960 if (Src->isSigned()) return Signext;
1961 return Zeroext;
1962}
1963
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001964
Chris Lattner48a44f72002-05-02 17:06:02 +00001965// isEliminableCastOfCast - Return true if it is valid to eliminate the CI
1966// instruction.
1967//
Chris Lattnerdfae8be2003-07-24 17:35:25 +00001968static inline bool isEliminableCastOfCast(const Type *SrcTy, const Type *MidTy,
Chris Lattner11ffd592004-07-20 05:21:00 +00001969 const Type *DstTy, TargetData *TD) {
Chris Lattner48a44f72002-05-02 17:06:02 +00001970
Chris Lattner650b6da2002-08-02 20:00:25 +00001971 // It is legal to eliminate the instruction if casting A->B->A if the sizes
1972 // are identical and the bits don't get reinterpreted (for example
Chris Lattner0bb75912002-08-14 23:21:10 +00001973 // int->float->int would not be allowed)
Misha Brukmane5838c42003-05-20 18:45:36 +00001974 if (SrcTy == DstTy && SrcTy->isLosslesslyConvertibleTo(MidTy))
Chris Lattner650b6da2002-08-02 20:00:25 +00001975 return true;
Chris Lattner48a44f72002-05-02 17:06:02 +00001976
Chris Lattner11ffd592004-07-20 05:21:00 +00001977 // If the source and destination types are pointer types, and the intermediate
1978 // is an integer type bigger than a pointer, eliminate the casts.
1979 if (isa<PointerType>(SrcTy) && isa<PointerType>(DstTy)) {
1980 if (isa<PointerType>(MidTy)) return true;
1981
1982 if (MidTy->isInteger() && MidTy->getPrimitiveSize() >= TD->getPointerSize())
1983 return true;
1984 }
1985
Chris Lattner48a44f72002-05-02 17:06:02 +00001986 // Allow free casting and conversion of sizes as long as the sign doesn't
1987 // change...
Chris Lattnerb0b412e2002-09-03 01:08:28 +00001988 if (SrcTy->isIntegral() && MidTy->isIntegral() && DstTy->isIntegral()) {
Chris Lattner4e2dbc62004-07-20 00:59:32 +00001989 CastType FirstCast = getCastType(SrcTy, MidTy);
1990 CastType SecondCast = getCastType(MidTy, DstTy);
Chris Lattner650b6da2002-08-02 20:00:25 +00001991
Chris Lattner4e2dbc62004-07-20 00:59:32 +00001992 // Capture the effect of these two casts. If the result is a legal cast,
1993 // the CastType is stored here, otherwise a special code is used.
1994 static const unsigned CastResult[] = {
1995 // First cast is noop
1996 0, 1, 2, 3,
1997 // First cast is a truncate
1998 1, 1, 4, 4, // trunc->extend is not safe to eliminate
1999 // First cast is a sign ext
2000 2, 5, 2, 4, // signext->trunc always ok, signext->zeroext never ok
2001 // First cast is a zero ext
2002 3, 5, 3, 3, // zeroext->trunc always ok
2003 };
2004
2005 unsigned Result = CastResult[FirstCast*4+SecondCast];
2006 switch (Result) {
2007 default: assert(0 && "Illegal table value!");
2008 case 0:
2009 case 1:
2010 case 2:
2011 case 3:
2012 // FIXME: in the future, when LLVM has explicit sign/zeroextends and
2013 // truncates, we could eliminate more casts.
2014 return (unsigned)getCastType(SrcTy, DstTy) == Result;
2015 case 4:
2016 return false; // Not possible to eliminate this here.
2017 case 5:
2018 // Sign or zero extend followed by truncate is always ok
Chris Lattner650b6da2002-08-02 20:00:25 +00002019 return true;
Chris Lattner3732aca2002-08-15 16:15:25 +00002020 }
Chris Lattner650b6da2002-08-02 20:00:25 +00002021 }
Chris Lattner48a44f72002-05-02 17:06:02 +00002022 return false;
2023}
2024
Chris Lattner11ffd592004-07-20 05:21:00 +00002025static bool ValueRequiresCast(const Value *V, const Type *Ty, TargetData *TD) {
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002026 if (V->getType() == Ty || isa<Constant>(V)) return false;
2027 if (const CastInst *CI = dyn_cast<CastInst>(V))
Chris Lattner11ffd592004-07-20 05:21:00 +00002028 if (isEliminableCastOfCast(CI->getOperand(0)->getType(), CI->getType(), Ty,
2029 TD))
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002030 return false;
2031 return true;
2032}
2033
2034/// InsertOperandCastBefore - This inserts a cast of V to DestTy before the
2035/// InsertBefore instruction. This is specialized a bit to avoid inserting
2036/// casts that are known to not do anything...
2037///
2038Value *InstCombiner::InsertOperandCastBefore(Value *V, const Type *DestTy,
2039 Instruction *InsertBefore) {
2040 if (V->getType() == DestTy) return V;
2041 if (Constant *C = dyn_cast<Constant>(V))
2042 return ConstantExpr::getCast(C, DestTy);
2043
2044 CastInst *CI = new CastInst(V, DestTy, V->getName());
2045 InsertNewInstBefore(CI, *InsertBefore);
2046 return CI;
2047}
Chris Lattner48a44f72002-05-02 17:06:02 +00002048
2049// CastInst simplification
Chris Lattner260ab202002-04-18 17:39:14 +00002050//
Chris Lattner113f4f42002-06-25 16:13:24 +00002051Instruction *InstCombiner::visitCastInst(CastInst &CI) {
Chris Lattner55d4bda2003-06-23 21:59:52 +00002052 Value *Src = CI.getOperand(0);
2053
Chris Lattner48a44f72002-05-02 17:06:02 +00002054 // If the user is casting a value to the same type, eliminate this cast
2055 // instruction...
Chris Lattner55d4bda2003-06-23 21:59:52 +00002056 if (CI.getType() == Src->getType())
2057 return ReplaceInstUsesWith(CI, Src);
Chris Lattner48a44f72002-05-02 17:06:02 +00002058
Chris Lattner48a44f72002-05-02 17:06:02 +00002059 // If casting the result of another cast instruction, try to eliminate this
2060 // one!
2061 //
Chris Lattner55d4bda2003-06-23 21:59:52 +00002062 if (CastInst *CSrc = dyn_cast<CastInst>(Src)) {
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002063 if (isEliminableCastOfCast(CSrc->getOperand(0)->getType(),
Chris Lattner11ffd592004-07-20 05:21:00 +00002064 CSrc->getType(), CI.getType(), TD)) {
Chris Lattner48a44f72002-05-02 17:06:02 +00002065 // This instruction now refers directly to the cast's src operand. This
2066 // has a good chance of making CSrc dead.
Chris Lattner113f4f42002-06-25 16:13:24 +00002067 CI.setOperand(0, CSrc->getOperand(0));
2068 return &CI;
Chris Lattner48a44f72002-05-02 17:06:02 +00002069 }
2070
Chris Lattner650b6da2002-08-02 20:00:25 +00002071 // If this is an A->B->A cast, and we are dealing with integral types, try
2072 // to convert this into a logical 'and' instruction.
2073 //
2074 if (CSrc->getOperand(0)->getType() == CI.getType() &&
Chris Lattnerb0b412e2002-09-03 01:08:28 +00002075 CI.getType()->isInteger() && CSrc->getType()->isInteger() &&
Chris Lattner650b6da2002-08-02 20:00:25 +00002076 CI.getType()->isUnsigned() && CSrc->getType()->isUnsigned() &&
2077 CSrc->getType()->getPrimitiveSize() < CI.getType()->getPrimitiveSize()){
2078 assert(CSrc->getType() != Type::ULongTy &&
2079 "Cannot have type bigger than ulong!");
Chris Lattner196897c2003-05-26 23:41:32 +00002080 uint64_t AndValue = (1ULL << CSrc->getType()->getPrimitiveSize()*8)-1;
Chris Lattner650b6da2002-08-02 20:00:25 +00002081 Constant *AndOp = ConstantUInt::get(CI.getType(), AndValue);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002082 return BinaryOperator::createAnd(CSrc->getOperand(0), AndOp);
Chris Lattner650b6da2002-08-02 20:00:25 +00002083 }
2084 }
2085
Chris Lattner03841652004-05-25 04:29:21 +00002086 // If this is a cast to bool, turn it into the appropriate setne instruction.
2087 if (CI.getType() == Type::BoolTy)
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002088 return BinaryOperator::createSetNE(CI.getOperand(0),
Chris Lattner03841652004-05-25 04:29:21 +00002089 Constant::getNullValue(CI.getOperand(0)->getType()));
2090
Chris Lattnerd0d51602003-06-21 23:12:02 +00002091 // If casting the result of a getelementptr instruction with no offset, turn
2092 // this into a cast of the original pointer!
2093 //
Chris Lattner55d4bda2003-06-23 21:59:52 +00002094 if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Src)) {
Chris Lattnerd0d51602003-06-21 23:12:02 +00002095 bool AllZeroOperands = true;
2096 for (unsigned i = 1, e = GEP->getNumOperands(); i != e; ++i)
2097 if (!isa<Constant>(GEP->getOperand(i)) ||
2098 !cast<Constant>(GEP->getOperand(i))->isNullValue()) {
2099 AllZeroOperands = false;
2100 break;
2101 }
2102 if (AllZeroOperands) {
2103 CI.setOperand(0, GEP->getOperand(0));
2104 return &CI;
2105 }
2106 }
2107
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002108 // If we are casting a malloc or alloca to a pointer to a type of the same
2109 // size, rewrite the allocation instruction to allocate the "right" type.
2110 //
2111 if (AllocationInst *AI = dyn_cast<AllocationInst>(Src))
Chris Lattnerd4d987d2003-11-02 06:54:48 +00002112 if (AI->hasOneUse() && !AI->isArrayAllocation())
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002113 if (const PointerType *PTy = dyn_cast<PointerType>(CI.getType())) {
2114 // Get the type really allocated and the type casted to...
2115 const Type *AllocElTy = AI->getAllocatedType();
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002116 const Type *CastElTy = PTy->getElementType();
Chris Lattner9eb9ccd2004-07-06 19:28:42 +00002117 if (AllocElTy->isSized() && CastElTy->isSized()) {
2118 unsigned AllocElTySize = TD->getTypeSize(AllocElTy);
2119 unsigned CastElTySize = TD->getTypeSize(CastElTy);
Chris Lattner7c94d112003-11-05 17:31:36 +00002120
Chris Lattner9eb9ccd2004-07-06 19:28:42 +00002121 // If the allocation is for an even multiple of the cast type size
2122 if (CastElTySize && (AllocElTySize % CastElTySize == 0)) {
2123 Value *Amt = ConstantUInt::get(Type::UIntTy,
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002124 AllocElTySize/CastElTySize);
Chris Lattner9eb9ccd2004-07-06 19:28:42 +00002125 std::string Name = AI->getName(); AI->setName("");
2126 AllocationInst *New;
2127 if (isa<MallocInst>(AI))
2128 New = new MallocInst(CastElTy, Amt, Name);
2129 else
2130 New = new AllocaInst(CastElTy, Amt, Name);
2131 InsertNewInstBefore(New, *AI);
2132 return ReplaceInstUsesWith(CI, New);
2133 }
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002134 }
2135 }
2136
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002137 // If the source value is an instruction with only this use, we can attempt to
2138 // propagate the cast into the instruction. Also, only handle integral types
2139 // for now.
2140 if (Instruction *SrcI = dyn_cast<Instruction>(Src))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00002141 if (SrcI->hasOneUse() && Src->getType()->isIntegral() &&
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002142 CI.getType()->isInteger()) { // Don't mess with casts to bool here
2143 const Type *DestTy = CI.getType();
2144 unsigned SrcBitSize = getTypeSizeInBits(Src->getType());
2145 unsigned DestBitSize = getTypeSizeInBits(DestTy);
2146
2147 Value *Op0 = SrcI->getNumOperands() > 0 ? SrcI->getOperand(0) : 0;
2148 Value *Op1 = SrcI->getNumOperands() > 1 ? SrcI->getOperand(1) : 0;
2149
2150 switch (SrcI->getOpcode()) {
2151 case Instruction::Add:
2152 case Instruction::Mul:
2153 case Instruction::And:
2154 case Instruction::Or:
2155 case Instruction::Xor:
2156 // If we are discarding information, or just changing the sign, rewrite.
2157 if (DestBitSize <= SrcBitSize && DestBitSize != 1) {
2158 // Don't insert two casts if they cannot be eliminated. We allow two
2159 // casts to be inserted if the sizes are the same. This could only be
2160 // converting signedness, which is a noop.
Chris Lattner11ffd592004-07-20 05:21:00 +00002161 if (DestBitSize == SrcBitSize || !ValueRequiresCast(Op1, DestTy,TD) ||
2162 !ValueRequiresCast(Op0, DestTy, TD)) {
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002163 Value *Op0c = InsertOperandCastBefore(Op0, DestTy, SrcI);
2164 Value *Op1c = InsertOperandCastBefore(Op1, DestTy, SrcI);
2165 return BinaryOperator::create(cast<BinaryOperator>(SrcI)
2166 ->getOpcode(), Op0c, Op1c);
2167 }
2168 }
2169 break;
2170 case Instruction::Shl:
2171 // Allow changing the sign of the source operand. Do not allow changing
2172 // the size of the shift, UNLESS the shift amount is a constant. We
2173 // mush not change variable sized shifts to a smaller size, because it
2174 // is undefined to shift more bits out than exist in the value.
2175 if (DestBitSize == SrcBitSize ||
2176 (DestBitSize < SrcBitSize && isa<Constant>(Op1))) {
2177 Value *Op0c = InsertOperandCastBefore(Op0, DestTy, SrcI);
2178 return new ShiftInst(Instruction::Shl, Op0c, Op1);
2179 }
2180 break;
2181 }
2182 }
2183
Chris Lattner260ab202002-04-18 17:39:14 +00002184 return 0;
Chris Lattnerca081252001-12-14 16:52:21 +00002185}
2186
Chris Lattner56e4d3d2004-04-09 23:46:01 +00002187/// GetSelectFoldableOperands - We want to turn code that looks like this:
2188/// %C = or %A, %B
2189/// %D = select %cond, %C, %A
2190/// into:
2191/// %C = select %cond, %B, 0
2192/// %D = or %A, %C
2193///
2194/// Assuming that the specified instruction is an operand to the select, return
2195/// a bitmask indicating which operands of this instruction are foldable if they
2196/// equal the other incoming value of the select.
2197///
2198static unsigned GetSelectFoldableOperands(Instruction *I) {
2199 switch (I->getOpcode()) {
2200 case Instruction::Add:
2201 case Instruction::Mul:
2202 case Instruction::And:
2203 case Instruction::Or:
2204 case Instruction::Xor:
2205 return 3; // Can fold through either operand.
2206 case Instruction::Sub: // Can only fold on the amount subtracted.
2207 case Instruction::Shl: // Can only fold on the shift amount.
2208 case Instruction::Shr:
2209 return 1;
2210 default:
2211 return 0; // Cannot fold
2212 }
2213}
2214
2215/// GetSelectFoldableConstant - For the same transformation as the previous
2216/// function, return the identity constant that goes into the select.
2217static Constant *GetSelectFoldableConstant(Instruction *I) {
2218 switch (I->getOpcode()) {
2219 default: assert(0 && "This cannot happen!"); abort();
2220 case Instruction::Add:
2221 case Instruction::Sub:
2222 case Instruction::Or:
2223 case Instruction::Xor:
2224 return Constant::getNullValue(I->getType());
2225 case Instruction::Shl:
2226 case Instruction::Shr:
2227 return Constant::getNullValue(Type::UByteTy);
2228 case Instruction::And:
2229 return ConstantInt::getAllOnesValue(I->getType());
2230 case Instruction::Mul:
2231 return ConstantInt::get(I->getType(), 1);
2232 }
2233}
2234
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002235Instruction *InstCombiner::visitSelectInst(SelectInst &SI) {
Chris Lattner533bc492004-03-30 19:37:13 +00002236 Value *CondVal = SI.getCondition();
2237 Value *TrueVal = SI.getTrueValue();
2238 Value *FalseVal = SI.getFalseValue();
2239
2240 // select true, X, Y -> X
2241 // select false, X, Y -> Y
2242 if (ConstantBool *C = dyn_cast<ConstantBool>(CondVal))
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002243 if (C == ConstantBool::True)
Chris Lattner533bc492004-03-30 19:37:13 +00002244 return ReplaceInstUsesWith(SI, TrueVal);
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002245 else {
2246 assert(C == ConstantBool::False);
Chris Lattner533bc492004-03-30 19:37:13 +00002247 return ReplaceInstUsesWith(SI, FalseVal);
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002248 }
Chris Lattner533bc492004-03-30 19:37:13 +00002249
2250 // select C, X, X -> X
2251 if (TrueVal == FalseVal)
2252 return ReplaceInstUsesWith(SI, TrueVal);
2253
Chris Lattner1c631e82004-04-08 04:43:23 +00002254 if (SI.getType() == Type::BoolTy)
2255 if (ConstantBool *C = dyn_cast<ConstantBool>(TrueVal)) {
2256 if (C == ConstantBool::True) {
2257 // Change: A = select B, true, C --> A = or B, C
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002258 return BinaryOperator::createOr(CondVal, FalseVal);
Chris Lattner1c631e82004-04-08 04:43:23 +00002259 } else {
2260 // Change: A = select B, false, C --> A = and !B, C
2261 Value *NotCond =
2262 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
2263 "not."+CondVal->getName()), SI);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002264 return BinaryOperator::createAnd(NotCond, FalseVal);
Chris Lattner1c631e82004-04-08 04:43:23 +00002265 }
2266 } else if (ConstantBool *C = dyn_cast<ConstantBool>(FalseVal)) {
2267 if (C == ConstantBool::False) {
2268 // Change: A = select B, C, false --> A = and B, C
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002269 return BinaryOperator::createAnd(CondVal, TrueVal);
Chris Lattner1c631e82004-04-08 04:43:23 +00002270 } else {
2271 // Change: A = select B, C, true --> A = or !B, C
2272 Value *NotCond =
2273 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
2274 "not."+CondVal->getName()), SI);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002275 return BinaryOperator::createOr(NotCond, TrueVal);
Chris Lattner1c631e82004-04-08 04:43:23 +00002276 }
2277 }
2278
Chris Lattner183b3362004-04-09 19:05:30 +00002279 // Selecting between two integer constants?
2280 if (ConstantInt *TrueValC = dyn_cast<ConstantInt>(TrueVal))
2281 if (ConstantInt *FalseValC = dyn_cast<ConstantInt>(FalseVal)) {
2282 // select C, 1, 0 -> cast C to int
2283 if (FalseValC->isNullValue() && TrueValC->getRawValue() == 1) {
2284 return new CastInst(CondVal, SI.getType());
2285 } else if (TrueValC->isNullValue() && FalseValC->getRawValue() == 1) {
2286 // select C, 0, 1 -> cast !C to int
2287 Value *NotCond =
2288 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
Chris Lattnercf7baf32004-04-09 18:19:44 +00002289 "not."+CondVal->getName()), SI);
Chris Lattner183b3362004-04-09 19:05:30 +00002290 return new CastInst(NotCond, SI.getType());
Chris Lattnercf7baf32004-04-09 18:19:44 +00002291 }
Chris Lattner35167c32004-06-09 07:59:58 +00002292
2293 // If one of the constants is zero (we know they can't both be) and we
2294 // have a setcc instruction with zero, and we have an 'and' with the
2295 // non-constant value, eliminate this whole mess. This corresponds to
2296 // cases like this: ((X & 27) ? 27 : 0)
2297 if (TrueValC->isNullValue() || FalseValC->isNullValue())
2298 if (Instruction *IC = dyn_cast<Instruction>(SI.getCondition()))
2299 if ((IC->getOpcode() == Instruction::SetEQ ||
2300 IC->getOpcode() == Instruction::SetNE) &&
2301 isa<ConstantInt>(IC->getOperand(1)) &&
2302 cast<Constant>(IC->getOperand(1))->isNullValue())
2303 if (Instruction *ICA = dyn_cast<Instruction>(IC->getOperand(0)))
2304 if (ICA->getOpcode() == Instruction::And &&
2305 isa<ConstantInt>(ICA->getOperand(1)) &&
2306 (ICA->getOperand(1) == TrueValC ||
2307 ICA->getOperand(1) == FalseValC) &&
2308 isOneBitSet(cast<ConstantInt>(ICA->getOperand(1)))) {
2309 // Okay, now we know that everything is set up, we just don't
2310 // know whether we have a setne or seteq and whether the true or
2311 // false val is the zero.
2312 bool ShouldNotVal = !TrueValC->isNullValue();
2313 ShouldNotVal ^= IC->getOpcode() == Instruction::SetNE;
2314 Value *V = ICA;
2315 if (ShouldNotVal)
2316 V = InsertNewInstBefore(BinaryOperator::create(
2317 Instruction::Xor, V, ICA->getOperand(1)), SI);
2318 return ReplaceInstUsesWith(SI, V);
2319 }
Chris Lattner533bc492004-03-30 19:37:13 +00002320 }
Chris Lattner623fba12004-04-10 22:21:27 +00002321
2322 // See if we are selecting two values based on a comparison of the two values.
2323 if (SetCondInst *SCI = dyn_cast<SetCondInst>(CondVal)) {
2324 if (SCI->getOperand(0) == TrueVal && SCI->getOperand(1) == FalseVal) {
2325 // Transform (X == Y) ? X : Y -> Y
2326 if (SCI->getOpcode() == Instruction::SetEQ)
2327 return ReplaceInstUsesWith(SI, FalseVal);
2328 // Transform (X != Y) ? X : Y -> X
2329 if (SCI->getOpcode() == Instruction::SetNE)
2330 return ReplaceInstUsesWith(SI, TrueVal);
2331 // NOTE: if we wanted to, this is where to detect MIN/MAX/ABS/etc.
2332
2333 } else if (SCI->getOperand(0) == FalseVal && SCI->getOperand(1) == TrueVal){
2334 // Transform (X == Y) ? Y : X -> X
2335 if (SCI->getOpcode() == Instruction::SetEQ)
Chris Lattner24cf0202004-04-11 01:39:19 +00002336 return ReplaceInstUsesWith(SI, FalseVal);
Chris Lattner623fba12004-04-10 22:21:27 +00002337 // Transform (X != Y) ? Y : X -> Y
2338 if (SCI->getOpcode() == Instruction::SetNE)
Chris Lattner24cf0202004-04-11 01:39:19 +00002339 return ReplaceInstUsesWith(SI, TrueVal);
Chris Lattner623fba12004-04-10 22:21:27 +00002340 // NOTE: if we wanted to, this is where to detect MIN/MAX/ABS/etc.
2341 }
2342 }
Chris Lattner1c631e82004-04-08 04:43:23 +00002343
Chris Lattner56e4d3d2004-04-09 23:46:01 +00002344 // See if we can fold the select into one of our operands.
2345 if (SI.getType()->isInteger()) {
2346 // See the comment above GetSelectFoldableOperands for a description of the
2347 // transformation we are doing here.
2348 if (Instruction *TVI = dyn_cast<Instruction>(TrueVal))
2349 if (TVI->hasOneUse() && TVI->getNumOperands() == 2 &&
2350 !isa<Constant>(FalseVal))
2351 if (unsigned SFO = GetSelectFoldableOperands(TVI)) {
2352 unsigned OpToFold = 0;
2353 if ((SFO & 1) && FalseVal == TVI->getOperand(0)) {
2354 OpToFold = 1;
2355 } else if ((SFO & 2) && FalseVal == TVI->getOperand(1)) {
2356 OpToFold = 2;
2357 }
2358
2359 if (OpToFold) {
2360 Constant *C = GetSelectFoldableConstant(TVI);
2361 std::string Name = TVI->getName(); TVI->setName("");
2362 Instruction *NewSel =
2363 new SelectInst(SI.getCondition(), TVI->getOperand(2-OpToFold), C,
2364 Name);
2365 InsertNewInstBefore(NewSel, SI);
2366 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(TVI))
2367 return BinaryOperator::create(BO->getOpcode(), FalseVal, NewSel);
2368 else if (ShiftInst *SI = dyn_cast<ShiftInst>(TVI))
2369 return new ShiftInst(SI->getOpcode(), FalseVal, NewSel);
2370 else {
2371 assert(0 && "Unknown instruction!!");
2372 }
2373 }
2374 }
2375
2376 if (Instruction *FVI = dyn_cast<Instruction>(FalseVal))
2377 if (FVI->hasOneUse() && FVI->getNumOperands() == 2 &&
2378 !isa<Constant>(TrueVal))
2379 if (unsigned SFO = GetSelectFoldableOperands(FVI)) {
2380 unsigned OpToFold = 0;
2381 if ((SFO & 1) && TrueVal == FVI->getOperand(0)) {
2382 OpToFold = 1;
2383 } else if ((SFO & 2) && TrueVal == FVI->getOperand(1)) {
2384 OpToFold = 2;
2385 }
2386
2387 if (OpToFold) {
2388 Constant *C = GetSelectFoldableConstant(FVI);
2389 std::string Name = FVI->getName(); FVI->setName("");
2390 Instruction *NewSel =
2391 new SelectInst(SI.getCondition(), C, FVI->getOperand(2-OpToFold),
2392 Name);
2393 InsertNewInstBefore(NewSel, SI);
2394 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(FVI))
2395 return BinaryOperator::create(BO->getOpcode(), TrueVal, NewSel);
2396 else if (ShiftInst *SI = dyn_cast<ShiftInst>(FVI))
2397 return new ShiftInst(SI->getOpcode(), TrueVal, NewSel);
2398 else {
2399 assert(0 && "Unknown instruction!!");
2400 }
2401 }
2402 }
2403 }
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002404 return 0;
2405}
2406
2407
Chris Lattner970c33a2003-06-19 17:00:31 +00002408// CallInst simplification
2409//
2410Instruction *InstCombiner::visitCallInst(CallInst &CI) {
Chris Lattner51ea1272004-02-28 05:22:00 +00002411 // Intrinsics cannot occur in an invoke, so handle them here instead of in
2412 // visitCallSite.
2413 if (Function *F = CI.getCalledFunction())
2414 switch (F->getIntrinsicID()) {
2415 case Intrinsic::memmove:
2416 case Intrinsic::memcpy:
2417 case Intrinsic::memset:
2418 // memmove/cpy/set of zero bytes is a noop.
2419 if (Constant *NumBytes = dyn_cast<Constant>(CI.getOperand(3))) {
2420 if (NumBytes->isNullValue())
2421 return EraseInstFromFunction(CI);
2422 }
2423 break;
2424 default:
2425 break;
2426 }
2427
Chris Lattneraec3d942003-10-07 22:32:43 +00002428 return visitCallSite(&CI);
Chris Lattner970c33a2003-06-19 17:00:31 +00002429}
2430
2431// InvokeInst simplification
2432//
2433Instruction *InstCombiner::visitInvokeInst(InvokeInst &II) {
Chris Lattneraec3d942003-10-07 22:32:43 +00002434 return visitCallSite(&II);
Chris Lattner970c33a2003-06-19 17:00:31 +00002435}
2436
Chris Lattneraec3d942003-10-07 22:32:43 +00002437// visitCallSite - Improvements for call and invoke instructions.
2438//
2439Instruction *InstCombiner::visitCallSite(CallSite CS) {
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002440 bool Changed = false;
2441
2442 // If the callee is a constexpr cast of a function, attempt to move the cast
2443 // to the arguments of the call/invoke.
Chris Lattneraec3d942003-10-07 22:32:43 +00002444 if (transformConstExprCastCall(CS)) return 0;
2445
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002446 Value *Callee = CS.getCalledValue();
2447 const PointerType *PTy = cast<PointerType>(Callee->getType());
2448 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
2449 if (FTy->isVarArg()) {
2450 // See if we can optimize any arguments passed through the varargs area of
2451 // the call.
2452 for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(),
2453 E = CS.arg_end(); I != E; ++I)
2454 if (CastInst *CI = dyn_cast<CastInst>(*I)) {
2455 // If this cast does not effect the value passed through the varargs
2456 // area, we can eliminate the use of the cast.
2457 Value *Op = CI->getOperand(0);
2458 if (CI->getType()->isLosslesslyConvertibleTo(Op->getType())) {
2459 *I = Op;
2460 Changed = true;
2461 }
2462 }
2463 }
Chris Lattneraec3d942003-10-07 22:32:43 +00002464
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002465 return Changed ? CS.getInstruction() : 0;
Chris Lattneraec3d942003-10-07 22:32:43 +00002466}
2467
Chris Lattner970c33a2003-06-19 17:00:31 +00002468// transformConstExprCastCall - If the callee is a constexpr cast of a function,
2469// attempt to move the cast to the arguments of the call/invoke.
2470//
2471bool InstCombiner::transformConstExprCastCall(CallSite CS) {
2472 if (!isa<ConstantExpr>(CS.getCalledValue())) return false;
2473 ConstantExpr *CE = cast<ConstantExpr>(CS.getCalledValue());
Chris Lattnerf3edc492004-07-18 18:59:44 +00002474 if (CE->getOpcode() != Instruction::Cast || !isa<Function>(CE->getOperand(0)))
Chris Lattner970c33a2003-06-19 17:00:31 +00002475 return false;
Reid Spencer87436872004-07-18 00:38:32 +00002476 Function *Callee = cast<Function>(CE->getOperand(0));
Chris Lattner970c33a2003-06-19 17:00:31 +00002477 Instruction *Caller = CS.getInstruction();
2478
2479 // Okay, this is a cast from a function to a different type. Unless doing so
2480 // would cause a type conversion of one of our arguments, change this call to
2481 // be a direct call with arguments casted to the appropriate types.
2482 //
2483 const FunctionType *FT = Callee->getFunctionType();
2484 const Type *OldRetTy = Caller->getType();
2485
Chris Lattner1f7942f2004-01-14 06:06:08 +00002486 // Check to see if we are changing the return type...
2487 if (OldRetTy != FT->getReturnType()) {
2488 if (Callee->isExternal() &&
2489 !OldRetTy->isLosslesslyConvertibleTo(FT->getReturnType()) &&
2490 !Caller->use_empty())
2491 return false; // Cannot transform this return value...
2492
2493 // If the callsite is an invoke instruction, and the return value is used by
2494 // a PHI node in a successor, we cannot change the return type of the call
2495 // because there is no place to put the cast instruction (without breaking
2496 // the critical edge). Bail out in this case.
2497 if (!Caller->use_empty())
2498 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller))
2499 for (Value::use_iterator UI = II->use_begin(), E = II->use_end();
2500 UI != E; ++UI)
2501 if (PHINode *PN = dyn_cast<PHINode>(*UI))
2502 if (PN->getParent() == II->getNormalDest() ||
Chris Lattnerfae8ab32004-02-08 21:44:31 +00002503 PN->getParent() == II->getUnwindDest())
Chris Lattner1f7942f2004-01-14 06:06:08 +00002504 return false;
2505 }
Chris Lattner970c33a2003-06-19 17:00:31 +00002506
2507 unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin());
2508 unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs);
2509
2510 CallSite::arg_iterator AI = CS.arg_begin();
2511 for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) {
2512 const Type *ParamTy = FT->getParamType(i);
2513 bool isConvertible = (*AI)->getType()->isLosslesslyConvertibleTo(ParamTy);
2514 if (Callee->isExternal() && !isConvertible) return false;
2515 }
2516
2517 if (FT->getNumParams() < NumActualArgs && !FT->isVarArg() &&
2518 Callee->isExternal())
2519 return false; // Do not delete arguments unless we have a function body...
2520
2521 // Okay, we decided that this is a safe thing to do: go ahead and start
2522 // inserting cast instructions as necessary...
2523 std::vector<Value*> Args;
2524 Args.reserve(NumActualArgs);
2525
2526 AI = CS.arg_begin();
2527 for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) {
2528 const Type *ParamTy = FT->getParamType(i);
2529 if ((*AI)->getType() == ParamTy) {
2530 Args.push_back(*AI);
2531 } else {
Chris Lattner1c631e82004-04-08 04:43:23 +00002532 Args.push_back(InsertNewInstBefore(new CastInst(*AI, ParamTy, "tmp"),
2533 *Caller));
Chris Lattner970c33a2003-06-19 17:00:31 +00002534 }
2535 }
2536
2537 // If the function takes more arguments than the call was taking, add them
2538 // now...
2539 for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i)
2540 Args.push_back(Constant::getNullValue(FT->getParamType(i)));
2541
2542 // If we are removing arguments to the function, emit an obnoxious warning...
2543 if (FT->getNumParams() < NumActualArgs)
2544 if (!FT->isVarArg()) {
2545 std::cerr << "WARNING: While resolving call to function '"
2546 << Callee->getName() << "' arguments were dropped!\n";
2547 } else {
2548 // Add all of the arguments in their promoted form to the arg list...
2549 for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) {
2550 const Type *PTy = getPromotedType((*AI)->getType());
2551 if (PTy != (*AI)->getType()) {
2552 // Must promote to pass through va_arg area!
2553 Instruction *Cast = new CastInst(*AI, PTy, "tmp");
2554 InsertNewInstBefore(Cast, *Caller);
2555 Args.push_back(Cast);
2556 } else {
2557 Args.push_back(*AI);
2558 }
2559 }
2560 }
2561
2562 if (FT->getReturnType() == Type::VoidTy)
2563 Caller->setName(""); // Void type should not have a name...
2564
2565 Instruction *NC;
2566 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) {
Chris Lattnerfae8ab32004-02-08 21:44:31 +00002567 NC = new InvokeInst(Callee, II->getNormalDest(), II->getUnwindDest(),
Chris Lattner970c33a2003-06-19 17:00:31 +00002568 Args, Caller->getName(), Caller);
2569 } else {
2570 NC = new CallInst(Callee, Args, Caller->getName(), Caller);
2571 }
2572
2573 // Insert a cast of the return type as necessary...
2574 Value *NV = NC;
2575 if (Caller->getType() != NV->getType() && !Caller->use_empty()) {
2576 if (NV->getType() != Type::VoidTy) {
2577 NV = NC = new CastInst(NC, Caller->getType(), "tmp");
Chris Lattner686767f2003-10-30 00:46:41 +00002578
2579 // If this is an invoke instruction, we should insert it after the first
2580 // non-phi, instruction in the normal successor block.
2581 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) {
2582 BasicBlock::iterator I = II->getNormalDest()->begin();
2583 while (isa<PHINode>(I)) ++I;
2584 InsertNewInstBefore(NC, *I);
2585 } else {
2586 // Otherwise, it's a call, just insert cast right after the call instr
2587 InsertNewInstBefore(NC, *Caller);
2588 }
Chris Lattner51ea1272004-02-28 05:22:00 +00002589 AddUsersToWorkList(*Caller);
Chris Lattner970c33a2003-06-19 17:00:31 +00002590 } else {
2591 NV = Constant::getNullValue(Caller->getType());
2592 }
2593 }
2594
2595 if (Caller->getType() != Type::VoidTy && !Caller->use_empty())
2596 Caller->replaceAllUsesWith(NV);
2597 Caller->getParent()->getInstList().erase(Caller);
2598 removeFromWorkList(Caller);
2599 return true;
2600}
2601
2602
Chris Lattner48a44f72002-05-02 17:06:02 +00002603
Chris Lattnerbbbdd852002-05-06 18:06:38 +00002604// PHINode simplification
2605//
Chris Lattner113f4f42002-06-25 16:13:24 +00002606Instruction *InstCombiner::visitPHINode(PHINode &PN) {
Chris Lattner91daeb52003-12-19 05:58:40 +00002607 if (Value *V = hasConstantValue(&PN))
2608 return ReplaceInstUsesWith(PN, V);
Chris Lattner4db2d222004-02-16 05:07:08 +00002609
2610 // If the only user of this instruction is a cast instruction, and all of the
2611 // incoming values are constants, change this PHI to merge together the casted
2612 // constants.
2613 if (PN.hasOneUse())
2614 if (CastInst *CI = dyn_cast<CastInst>(PN.use_back()))
2615 if (CI->getType() != PN.getType()) { // noop casts will be folded
2616 bool AllConstant = true;
2617 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i)
2618 if (!isa<Constant>(PN.getIncomingValue(i))) {
2619 AllConstant = false;
2620 break;
2621 }
2622 if (AllConstant) {
2623 // Make a new PHI with all casted values.
2624 PHINode *New = new PHINode(CI->getType(), PN.getName(), &PN);
2625 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) {
2626 Constant *OldArg = cast<Constant>(PN.getIncomingValue(i));
2627 New->addIncoming(ConstantExpr::getCast(OldArg, New->getType()),
2628 PN.getIncomingBlock(i));
2629 }
2630
2631 // Update the cast instruction.
2632 CI->setOperand(0, New);
2633 WorkList.push_back(CI); // revisit the cast instruction to fold.
2634 WorkList.push_back(New); // Make sure to revisit the new Phi
2635 return &PN; // PN is now dead!
2636 }
2637 }
Chris Lattner91daeb52003-12-19 05:58:40 +00002638 return 0;
Chris Lattnerbbbdd852002-05-06 18:06:38 +00002639}
2640
Chris Lattner69193f92004-04-05 01:30:19 +00002641static Value *InsertSignExtendToPtrTy(Value *V, const Type *DTy,
2642 Instruction *InsertPoint,
2643 InstCombiner *IC) {
2644 unsigned PS = IC->getTargetData().getPointerSize();
2645 const Type *VTy = V->getType();
2646 Instruction *Cast;
2647 if (!VTy->isSigned() && VTy->getPrimitiveSize() < PS)
2648 // We must insert a cast to ensure we sign-extend.
2649 V = IC->InsertNewInstBefore(new CastInst(V, VTy->getSignedVersion(),
2650 V->getName()), *InsertPoint);
2651 return IC->InsertNewInstBefore(new CastInst(V, DTy, V->getName()),
2652 *InsertPoint);
2653}
2654
Chris Lattner48a44f72002-05-02 17:06:02 +00002655
Chris Lattner113f4f42002-06-25 16:13:24 +00002656Instruction *InstCombiner::visitGetElementPtrInst(GetElementPtrInst &GEP) {
Chris Lattner5f667a62004-05-07 22:09:22 +00002657 Value *PtrOp = GEP.getOperand(0);
Chris Lattner471bd762003-05-22 19:07:21 +00002658 // Is it 'getelementptr %P, long 0' or 'getelementptr %P'
Chris Lattner113f4f42002-06-25 16:13:24 +00002659 // If so, eliminate the noop.
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002660 if (GEP.getNumOperands() == 1)
Chris Lattner5f667a62004-05-07 22:09:22 +00002661 return ReplaceInstUsesWith(GEP, PtrOp);
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002662
2663 bool HasZeroPointerIndex = false;
2664 if (Constant *C = dyn_cast<Constant>(GEP.getOperand(1)))
2665 HasZeroPointerIndex = C->isNullValue();
2666
2667 if (GEP.getNumOperands() == 2 && HasZeroPointerIndex)
Chris Lattner5f667a62004-05-07 22:09:22 +00002668 return ReplaceInstUsesWith(GEP, PtrOp);
Chris Lattner48a44f72002-05-02 17:06:02 +00002669
Chris Lattner69193f92004-04-05 01:30:19 +00002670 // Eliminate unneeded casts for indices.
2671 bool MadeChange = false;
Chris Lattner2b2412d2004-04-07 18:38:20 +00002672 gep_type_iterator GTI = gep_type_begin(GEP);
2673 for (unsigned i = 1, e = GEP.getNumOperands(); i != e; ++i, ++GTI)
2674 if (isa<SequentialType>(*GTI)) {
2675 if (CastInst *CI = dyn_cast<CastInst>(GEP.getOperand(i))) {
2676 Value *Src = CI->getOperand(0);
2677 const Type *SrcTy = Src->getType();
2678 const Type *DestTy = CI->getType();
2679 if (Src->getType()->isInteger()) {
2680 if (SrcTy->getPrimitiveSize() == DestTy->getPrimitiveSize()) {
2681 // We can always eliminate a cast from ulong or long to the other.
2682 // We can always eliminate a cast from uint to int or the other on
2683 // 32-bit pointer platforms.
2684 if (DestTy->getPrimitiveSize() >= TD->getPointerSize()) {
2685 MadeChange = true;
2686 GEP.setOperand(i, Src);
2687 }
2688 } else if (SrcTy->getPrimitiveSize() < DestTy->getPrimitiveSize() &&
2689 SrcTy->getPrimitiveSize() == 4) {
2690 // We can always eliminate a cast from int to [u]long. We can
2691 // eliminate a cast from uint to [u]long iff the target is a 32-bit
2692 // pointer target.
2693 if (SrcTy->isSigned() ||
2694 SrcTy->getPrimitiveSize() >= TD->getPointerSize()) {
2695 MadeChange = true;
2696 GEP.setOperand(i, Src);
2697 }
Chris Lattner69193f92004-04-05 01:30:19 +00002698 }
2699 }
2700 }
Chris Lattner2b2412d2004-04-07 18:38:20 +00002701 // If we are using a wider index than needed for this platform, shrink it
2702 // to what we need. If the incoming value needs a cast instruction,
2703 // insert it. This explicit cast can make subsequent optimizations more
2704 // obvious.
2705 Value *Op = GEP.getOperand(i);
2706 if (Op->getType()->getPrimitiveSize() > TD->getPointerSize())
Chris Lattner1e9ac1a2004-04-17 18:16:10 +00002707 if (Constant *C = dyn_cast<Constant>(Op)) {
Chris Lattner44d0b952004-07-20 01:48:15 +00002708 GEP.setOperand(i, ConstantExpr::getCast(C,
2709 TD->getIntPtrType()->getSignedVersion()));
Chris Lattner1e9ac1a2004-04-17 18:16:10 +00002710 MadeChange = true;
2711 } else {
Chris Lattner2b2412d2004-04-07 18:38:20 +00002712 Op = InsertNewInstBefore(new CastInst(Op, TD->getIntPtrType(),
2713 Op->getName()), GEP);
2714 GEP.setOperand(i, Op);
2715 MadeChange = true;
2716 }
Chris Lattner44d0b952004-07-20 01:48:15 +00002717
2718 // If this is a constant idx, make sure to canonicalize it to be a signed
2719 // operand, otherwise CSE and other optimizations are pessimized.
2720 if (ConstantUInt *CUI = dyn_cast<ConstantUInt>(Op)) {
2721 GEP.setOperand(i, ConstantExpr::getCast(CUI,
2722 CUI->getType()->getSignedVersion()));
2723 MadeChange = true;
2724 }
Chris Lattner69193f92004-04-05 01:30:19 +00002725 }
2726 if (MadeChange) return &GEP;
2727
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002728 // Combine Indices - If the source pointer to this getelementptr instruction
2729 // is a getelementptr instruction, combine the indices of the two
2730 // getelementptr instructions into a single instruction.
2731 //
Chris Lattner57c67b02004-03-25 22:59:29 +00002732 std::vector<Value*> SrcGEPOperands;
Chris Lattner5f667a62004-05-07 22:09:22 +00002733 if (GetElementPtrInst *Src = dyn_cast<GetElementPtrInst>(PtrOp)) {
Chris Lattner57c67b02004-03-25 22:59:29 +00002734 SrcGEPOperands.assign(Src->op_begin(), Src->op_end());
Chris Lattner5f667a62004-05-07 22:09:22 +00002735 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(PtrOp)) {
Chris Lattner57c67b02004-03-25 22:59:29 +00002736 if (CE->getOpcode() == Instruction::GetElementPtr)
2737 SrcGEPOperands.assign(CE->op_begin(), CE->op_end());
2738 }
2739
2740 if (!SrcGEPOperands.empty()) {
Chris Lattner5f667a62004-05-07 22:09:22 +00002741 // Note that if our source is a gep chain itself that we wait for that
2742 // chain to be resolved before we perform this transformation. This
2743 // avoids us creating a TON of code in some cases.
2744 //
2745 if (isa<GetElementPtrInst>(SrcGEPOperands[0]) &&
2746 cast<Instruction>(SrcGEPOperands[0])->getNumOperands() == 2)
2747 return 0; // Wait until our source is folded to completion.
2748
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002749 std::vector<Value *> Indices;
Chris Lattner5f667a62004-05-07 22:09:22 +00002750
2751 // Find out whether the last index in the source GEP is a sequential idx.
2752 bool EndsWithSequential = false;
2753 for (gep_type_iterator I = gep_type_begin(*cast<User>(PtrOp)),
2754 E = gep_type_end(*cast<User>(PtrOp)); I != E; ++I)
Chris Lattner8ec5f882004-05-08 22:41:42 +00002755 EndsWithSequential = !isa<StructType>(*I);
Chris Lattnerca081252001-12-14 16:52:21 +00002756
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002757 // Can we combine the two pointer arithmetics offsets?
Chris Lattner5f667a62004-05-07 22:09:22 +00002758 if (EndsWithSequential) {
Chris Lattner235af562003-03-05 22:33:14 +00002759 // Replace: gep (gep %P, long B), long A, ...
2760 // With: T = long A+B; gep %P, T, ...
2761 //
Chris Lattner5f667a62004-05-07 22:09:22 +00002762 Value *Sum, *SO1 = SrcGEPOperands.back(), *GO1 = GEP.getOperand(1);
Chris Lattner69193f92004-04-05 01:30:19 +00002763 if (SO1 == Constant::getNullValue(SO1->getType())) {
2764 Sum = GO1;
2765 } else if (GO1 == Constant::getNullValue(GO1->getType())) {
2766 Sum = SO1;
2767 } else {
2768 // If they aren't the same type, convert both to an integer of the
2769 // target's pointer size.
2770 if (SO1->getType() != GO1->getType()) {
2771 if (Constant *SO1C = dyn_cast<Constant>(SO1)) {
2772 SO1 = ConstantExpr::getCast(SO1C, GO1->getType());
2773 } else if (Constant *GO1C = dyn_cast<Constant>(GO1)) {
2774 GO1 = ConstantExpr::getCast(GO1C, SO1->getType());
2775 } else {
2776 unsigned PS = TD->getPointerSize();
2777 Instruction *Cast;
2778 if (SO1->getType()->getPrimitiveSize() == PS) {
2779 // Convert GO1 to SO1's type.
2780 GO1 = InsertSignExtendToPtrTy(GO1, SO1->getType(), &GEP, this);
2781
2782 } else if (GO1->getType()->getPrimitiveSize() == PS) {
2783 // Convert SO1 to GO1's type.
2784 SO1 = InsertSignExtendToPtrTy(SO1, GO1->getType(), &GEP, this);
2785 } else {
2786 const Type *PT = TD->getIntPtrType();
2787 SO1 = InsertSignExtendToPtrTy(SO1, PT, &GEP, this);
2788 GO1 = InsertSignExtendToPtrTy(GO1, PT, &GEP, this);
2789 }
2790 }
2791 }
Chris Lattner5f667a62004-05-07 22:09:22 +00002792 if (isa<Constant>(SO1) && isa<Constant>(GO1))
2793 Sum = ConstantExpr::getAdd(cast<Constant>(SO1), cast<Constant>(GO1));
2794 else {
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002795 Sum = BinaryOperator::createAdd(SO1, GO1, PtrOp->getName()+".sum");
2796 InsertNewInstBefore(cast<Instruction>(Sum), GEP);
Chris Lattner5f667a62004-05-07 22:09:22 +00002797 }
Chris Lattner69193f92004-04-05 01:30:19 +00002798 }
Chris Lattner5f667a62004-05-07 22:09:22 +00002799
2800 // Recycle the GEP we already have if possible.
2801 if (SrcGEPOperands.size() == 2) {
2802 GEP.setOperand(0, SrcGEPOperands[0]);
2803 GEP.setOperand(1, Sum);
2804 return &GEP;
2805 } else {
2806 Indices.insert(Indices.end(), SrcGEPOperands.begin()+1,
2807 SrcGEPOperands.end()-1);
2808 Indices.push_back(Sum);
2809 Indices.insert(Indices.end(), GEP.op_begin()+2, GEP.op_end());
2810 }
Chris Lattner69193f92004-04-05 01:30:19 +00002811 } else if (isa<Constant>(*GEP.idx_begin()) &&
2812 cast<Constant>(*GEP.idx_begin())->isNullValue() &&
Chris Lattner57c67b02004-03-25 22:59:29 +00002813 SrcGEPOperands.size() != 1) {
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002814 // Otherwise we can do the fold if the first index of the GEP is a zero
Chris Lattner57c67b02004-03-25 22:59:29 +00002815 Indices.insert(Indices.end(), SrcGEPOperands.begin()+1,
2816 SrcGEPOperands.end());
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002817 Indices.insert(Indices.end(), GEP.idx_begin()+1, GEP.idx_end());
2818 }
2819
2820 if (!Indices.empty())
Chris Lattner57c67b02004-03-25 22:59:29 +00002821 return new GetElementPtrInst(SrcGEPOperands[0], Indices, GEP.getName());
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002822
Chris Lattner5f667a62004-05-07 22:09:22 +00002823 } else if (GlobalValue *GV = dyn_cast<GlobalValue>(PtrOp)) {
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002824 // GEP of global variable. If all of the indices for this GEP are
2825 // constants, we can promote this to a constexpr instead of an instruction.
2826
2827 // Scan for nonconstants...
2828 std::vector<Constant*> Indices;
2829 User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end();
2830 for (; I != E && isa<Constant>(*I); ++I)
2831 Indices.push_back(cast<Constant>(*I));
2832
2833 if (I == E) { // If they are all constants...
Chris Lattnerf3edc492004-07-18 18:59:44 +00002834 Constant *CE = ConstantExpr::getGetElementPtr(GV, Indices);
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002835
2836 // Replace all uses of the GEP with the new constexpr...
2837 return ReplaceInstUsesWith(GEP, CE);
2838 }
Chris Lattner5f667a62004-05-07 22:09:22 +00002839 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(PtrOp)) {
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002840 if (CE->getOpcode() == Instruction::Cast) {
2841 if (HasZeroPointerIndex) {
2842 // transform: GEP (cast [10 x ubyte]* X to [0 x ubyte]*), long 0, ...
2843 // into : GEP [10 x ubyte]* X, long 0, ...
2844 //
2845 // This occurs when the program declares an array extern like "int X[];"
2846 //
2847 Constant *X = CE->getOperand(0);
2848 const PointerType *CPTy = cast<PointerType>(CE->getType());
2849 if (const PointerType *XTy = dyn_cast<PointerType>(X->getType()))
2850 if (const ArrayType *XATy =
2851 dyn_cast<ArrayType>(XTy->getElementType()))
2852 if (const ArrayType *CATy =
2853 dyn_cast<ArrayType>(CPTy->getElementType()))
2854 if (CATy->getElementType() == XATy->getElementType()) {
2855 // At this point, we know that the cast source type is a pointer
2856 // to an array of the same type as the destination pointer
2857 // array. Because the array type is never stepped over (there
2858 // is a leading zero) we can fold the cast into this GEP.
2859 GEP.setOperand(0, X);
2860 return &GEP;
2861 }
2862 }
2863 }
Chris Lattnerca081252001-12-14 16:52:21 +00002864 }
2865
Chris Lattnerca081252001-12-14 16:52:21 +00002866 return 0;
2867}
2868
Chris Lattner1085bdf2002-11-04 16:18:53 +00002869Instruction *InstCombiner::visitAllocationInst(AllocationInst &AI) {
2870 // Convert: malloc Ty, C - where C is a constant != 1 into: malloc [C x Ty], 1
2871 if (AI.isArrayAllocation()) // Check C != 1
2872 if (const ConstantUInt *C = dyn_cast<ConstantUInt>(AI.getArraySize())) {
2873 const Type *NewTy = ArrayType::get(AI.getAllocatedType(), C->getValue());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002874 AllocationInst *New = 0;
Chris Lattner1085bdf2002-11-04 16:18:53 +00002875
2876 // Create and insert the replacement instruction...
2877 if (isa<MallocInst>(AI))
Chris Lattnerabb77c92004-03-19 06:08:10 +00002878 New = new MallocInst(NewTy, 0, AI.getName());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002879 else {
2880 assert(isa<AllocaInst>(AI) && "Unknown type of allocation inst!");
Chris Lattnerabb77c92004-03-19 06:08:10 +00002881 New = new AllocaInst(NewTy, 0, AI.getName());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002882 }
Chris Lattnerabb77c92004-03-19 06:08:10 +00002883
2884 InsertNewInstBefore(New, AI);
Chris Lattner1085bdf2002-11-04 16:18:53 +00002885
2886 // Scan to the end of the allocation instructions, to skip over a block of
2887 // allocas if possible...
2888 //
2889 BasicBlock::iterator It = New;
2890 while (isa<AllocationInst>(*It)) ++It;
2891
2892 // Now that I is pointing to the first non-allocation-inst in the block,
2893 // insert our getelementptr instruction...
2894 //
Chris Lattner69193f92004-04-05 01:30:19 +00002895 std::vector<Value*> Idx(2, Constant::getNullValue(Type::IntTy));
Chris Lattner1085bdf2002-11-04 16:18:53 +00002896 Value *V = new GetElementPtrInst(New, Idx, New->getName()+".sub", It);
2897
2898 // Now make everything use the getelementptr instead of the original
2899 // allocation.
Chris Lattnerabb77c92004-03-19 06:08:10 +00002900 return ReplaceInstUsesWith(AI, V);
Chris Lattner1085bdf2002-11-04 16:18:53 +00002901 }
Chris Lattnerabb77c92004-03-19 06:08:10 +00002902
2903 // If alloca'ing a zero byte object, replace the alloca with a null pointer.
2904 // Note that we only do this for alloca's, because malloc should allocate and
2905 // return a unique pointer, even for a zero byte allocation.
Chris Lattner49df6ce2004-07-02 22:55:47 +00002906 if (isa<AllocaInst>(AI) && AI.getAllocatedType()->isSized() &&
2907 TD->getTypeSize(AI.getAllocatedType()) == 0)
Chris Lattnerabb77c92004-03-19 06:08:10 +00002908 return ReplaceInstUsesWith(AI, Constant::getNullValue(AI.getType()));
2909
Chris Lattner1085bdf2002-11-04 16:18:53 +00002910 return 0;
2911}
2912
Chris Lattner8427bff2003-12-07 01:24:23 +00002913Instruction *InstCombiner::visitFreeInst(FreeInst &FI) {
2914 Value *Op = FI.getOperand(0);
2915
2916 // Change free <ty>* (cast <ty2>* X to <ty>*) into free <ty2>* X
2917 if (CastInst *CI = dyn_cast<CastInst>(Op))
2918 if (isa<PointerType>(CI->getOperand(0)->getType())) {
2919 FI.setOperand(0, CI->getOperand(0));
2920 return &FI;
2921 }
2922
Chris Lattnerf3a36602004-02-28 04:57:37 +00002923 // If we have 'free null' delete the instruction. This can happen in stl code
2924 // when lots of inlining happens.
Chris Lattner51ea1272004-02-28 05:22:00 +00002925 if (isa<ConstantPointerNull>(Op))
2926 return EraseInstFromFunction(FI);
Chris Lattnerf3a36602004-02-28 04:57:37 +00002927
Chris Lattner8427bff2003-12-07 01:24:23 +00002928 return 0;
2929}
2930
2931
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002932/// GetGEPGlobalInitializer - Given a constant, and a getelementptr
2933/// constantexpr, return the constant value being addressed by the constant
2934/// expression, or null if something is funny.
2935///
2936static Constant *GetGEPGlobalInitializer(Constant *C, ConstantExpr *CE) {
Chris Lattner69193f92004-04-05 01:30:19 +00002937 if (CE->getOperand(1) != Constant::getNullValue(CE->getOperand(1)->getType()))
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002938 return 0; // Do not allow stepping over the value!
2939
2940 // Loop over all of the operands, tracking down which value we are
2941 // addressing...
Chris Lattnered79d8a2004-05-27 17:30:27 +00002942 gep_type_iterator I = gep_type_begin(CE), E = gep_type_end(CE);
2943 for (++I; I != E; ++I)
2944 if (const StructType *STy = dyn_cast<StructType>(*I)) {
2945 ConstantUInt *CU = cast<ConstantUInt>(I.getOperand());
2946 assert(CU->getValue() < STy->getNumElements() &&
2947 "Struct index out of range!");
2948 if (ConstantStruct *CS = dyn_cast<ConstantStruct>(C)) {
2949 C = cast<Constant>(CS->getValues()[CU->getValue()]);
2950 } else if (isa<ConstantAggregateZero>(C)) {
2951 C = Constant::getNullValue(STy->getElementType(CU->getValue()));
2952 } else {
2953 return 0;
2954 }
2955 } else if (ConstantInt *CI = dyn_cast<ConstantInt>(I.getOperand())) {
2956 const ArrayType *ATy = cast<ArrayType>(*I);
2957 if ((uint64_t)CI->getRawValue() >= ATy->getNumElements()) return 0;
2958 if (ConstantArray *CA = dyn_cast<ConstantArray>(C))
2959 C = cast<Constant>(CA->getValues()[CI->getRawValue()]);
2960 else if (isa<ConstantAggregateZero>(C))
2961 C = Constant::getNullValue(ATy->getElementType());
2962 else
2963 return 0;
2964 } else {
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002965 return 0;
Chris Lattnered79d8a2004-05-27 17:30:27 +00002966 }
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002967 return C;
2968}
2969
Chris Lattner35e24772004-07-13 01:49:43 +00002970static Instruction *InstCombineLoadCast(InstCombiner &IC, LoadInst &LI) {
2971 User *CI = cast<User>(LI.getOperand(0));
2972
2973 const Type *DestPTy = cast<PointerType>(CI->getType())->getElementType();
2974 if (const PointerType *SrcTy =
2975 dyn_cast<PointerType>(CI->getOperand(0)->getType())) {
2976 const Type *SrcPTy = SrcTy->getElementType();
2977 if (SrcPTy->isSized() && DestPTy->isSized() &&
2978 IC.getTargetData().getTypeSize(SrcPTy) ==
2979 IC.getTargetData().getTypeSize(DestPTy) &&
2980 (SrcPTy->isInteger() || isa<PointerType>(SrcPTy)) &&
2981 (DestPTy->isInteger() || isa<PointerType>(DestPTy))) {
2982 // Okay, we are casting from one integer or pointer type to another of
2983 // the same size. Instead of casting the pointer before the load, cast
2984 // the result of the loaded value.
2985 Value *NewLoad = IC.InsertNewInstBefore(new LoadInst(CI->getOperand(0),
2986 CI->getName()), LI);
2987 // Now cast the result of the load.
2988 return new CastInst(NewLoad, LI.getType());
2989 }
2990 }
2991 return 0;
2992}
2993
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002994Instruction *InstCombiner::visitLoadInst(LoadInst &LI) {
2995 Value *Op = LI.getOperand(0);
Chris Lattner7e8af382004-01-12 04:13:56 +00002996 if (LI.isVolatile()) return 0;
2997
Chris Lattner6679e462004-04-14 03:28:36 +00002998 if (Constant *C = dyn_cast<Constant>(Op))
2999 if (C->isNullValue()) // load null -> 0
3000 return ReplaceInstUsesWith(LI, Constant::getNullValue(LI.getType()));
Chris Lattner0f1d8a32003-06-26 05:06:25 +00003001
3002 // Instcombine load (constant global) into the value loaded...
3003 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op))
Chris Lattnerbdb0ce02003-07-22 21:46:59 +00003004 if (GV->isConstant() && !GV->isExternal())
Chris Lattner0f1d8a32003-06-26 05:06:25 +00003005 return ReplaceInstUsesWith(LI, GV->getInitializer());
3006
3007 // Instcombine load (constantexpr_GEP global, 0, ...) into the value loaded...
3008 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Op))
Chris Lattner35e24772004-07-13 01:49:43 +00003009 if (CE->getOpcode() == Instruction::GetElementPtr) {
Reid Spencer87436872004-07-18 00:38:32 +00003010 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(CE->getOperand(0)))
3011 if (GV->isConstant() && !GV->isExternal())
3012 if (Constant *V = GetGEPGlobalInitializer(GV->getInitializer(), CE))
3013 return ReplaceInstUsesWith(LI, V);
Chris Lattner35e24772004-07-13 01:49:43 +00003014 } else if (CE->getOpcode() == Instruction::Cast) {
3015 if (Instruction *Res = InstCombineLoadCast(*this, LI))
3016 return Res;
3017 }
Chris Lattnere228ee52004-04-08 20:39:49 +00003018
3019 // load (cast X) --> cast (load X) iff safe
Chris Lattner35e24772004-07-13 01:49:43 +00003020 if (CastInst *CI = dyn_cast<CastInst>(Op))
3021 if (Instruction *Res = InstCombineLoadCast(*this, LI))
3022 return Res;
Chris Lattnere228ee52004-04-08 20:39:49 +00003023
Chris Lattner0f1d8a32003-06-26 05:06:25 +00003024 return 0;
3025}
3026
3027
Chris Lattner9eef8a72003-06-04 04:46:00 +00003028Instruction *InstCombiner::visitBranchInst(BranchInst &BI) {
3029 // Change br (not X), label True, label False to: br X, label False, True
Chris Lattner4f7acca2004-02-27 06:27:46 +00003030 if (BI.isConditional() && !isa<Constant>(BI.getCondition())) {
Chris Lattnere967b342003-06-04 05:10:11 +00003031 if (Value *V = dyn_castNotVal(BI.getCondition())) {
3032 BasicBlock *TrueDest = BI.getSuccessor(0);
3033 BasicBlock *FalseDest = BI.getSuccessor(1);
3034 // Swap Destinations and condition...
3035 BI.setCondition(V);
3036 BI.setSuccessor(0, FalseDest);
3037 BI.setSuccessor(1, TrueDest);
3038 return &BI;
Chris Lattner4f7acca2004-02-27 06:27:46 +00003039 } else if (SetCondInst *I = dyn_cast<SetCondInst>(BI.getCondition())) {
3040 // Cannonicalize setne -> seteq
3041 if ((I->getOpcode() == Instruction::SetNE ||
3042 I->getOpcode() == Instruction::SetLE ||
3043 I->getOpcode() == Instruction::SetGE) && I->hasOneUse()) {
3044 std::string Name = I->getName(); I->setName("");
3045 Instruction::BinaryOps NewOpcode =
3046 SetCondInst::getInverseCondition(I->getOpcode());
3047 Value *NewSCC = BinaryOperator::create(NewOpcode, I->getOperand(0),
3048 I->getOperand(1), Name, I);
3049 BasicBlock *TrueDest = BI.getSuccessor(0);
3050 BasicBlock *FalseDest = BI.getSuccessor(1);
3051 // Swap Destinations and condition...
3052 BI.setCondition(NewSCC);
3053 BI.setSuccessor(0, FalseDest);
3054 BI.setSuccessor(1, TrueDest);
3055 removeFromWorkList(I);
3056 I->getParent()->getInstList().erase(I);
3057 WorkList.push_back(cast<Instruction>(NewSCC));
3058 return &BI;
3059 }
Chris Lattnere967b342003-06-04 05:10:11 +00003060 }
Chris Lattner4f7acca2004-02-27 06:27:46 +00003061 }
Chris Lattner9eef8a72003-06-04 04:46:00 +00003062 return 0;
3063}
Chris Lattner1085bdf2002-11-04 16:18:53 +00003064
Chris Lattner4c9c20a2004-07-03 00:26:11 +00003065Instruction *InstCombiner::visitSwitchInst(SwitchInst &SI) {
3066 Value *Cond = SI.getCondition();
3067 if (Instruction *I = dyn_cast<Instruction>(Cond)) {
3068 if (I->getOpcode() == Instruction::Add)
3069 if (ConstantInt *AddRHS = dyn_cast<ConstantInt>(I->getOperand(1))) {
3070 // change 'switch (X+4) case 1:' into 'switch (X) case -3'
3071 for (unsigned i = 2, e = SI.getNumOperands(); i != e; i += 2)
3072 SI.setOperand(i, ConstantExpr::getSub(cast<Constant>(SI.getOperand(i)),
3073 AddRHS));
3074 SI.setOperand(0, I->getOperand(0));
3075 WorkList.push_back(I);
3076 return &SI;
3077 }
3078 }
3079 return 0;
3080}
3081
Chris Lattnerca081252001-12-14 16:52:21 +00003082
Chris Lattner99f48c62002-09-02 04:59:56 +00003083void InstCombiner::removeFromWorkList(Instruction *I) {
3084 WorkList.erase(std::remove(WorkList.begin(), WorkList.end(), I),
3085 WorkList.end());
3086}
3087
Chris Lattner113f4f42002-06-25 16:13:24 +00003088bool InstCombiner::runOnFunction(Function &F) {
Chris Lattner260ab202002-04-18 17:39:14 +00003089 bool Changed = false;
Chris Lattnerf4ad1652003-11-02 05:57:39 +00003090 TD = &getAnalysis<TargetData>();
Chris Lattnerca081252001-12-14 16:52:21 +00003091
Chris Lattnerb643a9e2004-05-01 23:19:52 +00003092 for (inst_iterator i = inst_begin(F), e = inst_end(F); i != e; ++i)
3093 WorkList.push_back(&*i);
Chris Lattner2d3a7a62004-04-27 15:13:33 +00003094
Chris Lattnerca081252001-12-14 16:52:21 +00003095
3096 while (!WorkList.empty()) {
3097 Instruction *I = WorkList.back(); // Get an instruction from the worklist
3098 WorkList.pop_back();
3099
Misha Brukman632df282002-10-29 23:06:16 +00003100 // Check to see if we can DCE or ConstantPropagate the instruction...
Chris Lattner99f48c62002-09-02 04:59:56 +00003101 // Check to see if we can DIE the instruction...
3102 if (isInstructionTriviallyDead(I)) {
3103 // Add operands to the worklist...
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003104 if (I->getNumOperands() < 4)
Chris Lattner51ea1272004-02-28 05:22:00 +00003105 AddUsesToWorkList(*I);
Chris Lattner99f48c62002-09-02 04:59:56 +00003106 ++NumDeadInst;
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003107
3108 I->getParent()->getInstList().erase(I);
3109 removeFromWorkList(I);
3110 continue;
3111 }
Chris Lattner99f48c62002-09-02 04:59:56 +00003112
Misha Brukman632df282002-10-29 23:06:16 +00003113 // Instruction isn't dead, see if we can constant propagate it...
Chris Lattner99f48c62002-09-02 04:59:56 +00003114 if (Constant *C = ConstantFoldInstruction(I)) {
3115 // Add operands to the worklist...
Chris Lattner51ea1272004-02-28 05:22:00 +00003116 AddUsesToWorkList(*I);
Chris Lattnerc6509f42002-12-05 22:41:53 +00003117 ReplaceInstUsesWith(*I, C);
3118
Chris Lattner99f48c62002-09-02 04:59:56 +00003119 ++NumConstProp;
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003120 I->getParent()->getInstList().erase(I);
Chris Lattner800aaaf2003-10-07 15:17:02 +00003121 removeFromWorkList(I);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003122 continue;
Chris Lattner99f48c62002-09-02 04:59:56 +00003123 }
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003124
Chris Lattnerca081252001-12-14 16:52:21 +00003125 // Now that we have an instruction, try combining it to simplify it...
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003126 if (Instruction *Result = visit(*I)) {
Chris Lattner0b18c1d2002-05-10 15:38:35 +00003127 ++NumCombined;
Chris Lattner260ab202002-04-18 17:39:14 +00003128 // Should we replace the old instruction with a new one?
Chris Lattner053c0932002-05-14 15:24:07 +00003129 if (Result != I) {
Chris Lattner7d2a5392004-03-13 23:54:27 +00003130 DEBUG(std::cerr << "IC: Old = " << *I
3131 << " New = " << *Result);
3132
Chris Lattner396dbfe2004-06-09 05:08:07 +00003133 // Everything uses the new instruction now.
3134 I->replaceAllUsesWith(Result);
3135
3136 // Push the new instruction and any users onto the worklist.
3137 WorkList.push_back(Result);
3138 AddUsersToWorkList(*Result);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003139
3140 // Move the name to the new instruction first...
3141 std::string OldName = I->getName(); I->setName("");
Chris Lattner950fc782003-10-07 22:58:41 +00003142 Result->setName(OldName);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003143
3144 // Insert the new instruction into the basic block...
3145 BasicBlock *InstParent = I->getParent();
3146 InstParent->getInstList().insert(I, Result);
3147
Chris Lattner63d75af2004-05-01 23:27:23 +00003148 // Make sure that we reprocess all operands now that we reduced their
3149 // use counts.
Chris Lattnerb643a9e2004-05-01 23:19:52 +00003150 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
3151 if (Instruction *OpI = dyn_cast<Instruction>(I->getOperand(i)))
3152 WorkList.push_back(OpI);
3153
Chris Lattner396dbfe2004-06-09 05:08:07 +00003154 // Instructions can end up on the worklist more than once. Make sure
3155 // we do not process an instruction that has been deleted.
3156 removeFromWorkList(I);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003157
3158 // Erase the old instruction.
3159 InstParent->getInstList().erase(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00003160 } else {
Chris Lattner7d2a5392004-03-13 23:54:27 +00003161 DEBUG(std::cerr << "IC: MOD = " << *I);
3162
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003163 // If the instruction was modified, it's possible that it is now dead.
3164 // if so, remove it.
Chris Lattner63d75af2004-05-01 23:27:23 +00003165 if (isInstructionTriviallyDead(I)) {
3166 // Make sure we process all operands now that we are reducing their
3167 // use counts.
3168 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
3169 if (Instruction *OpI = dyn_cast<Instruction>(I->getOperand(i)))
3170 WorkList.push_back(OpI);
3171
3172 // Instructions may end up in the worklist more than once. Erase all
3173 // occurrances of this instruction.
Chris Lattner99f48c62002-09-02 04:59:56 +00003174 removeFromWorkList(I);
Chris Lattner63d75af2004-05-01 23:27:23 +00003175 I->getParent()->getInstList().erase(I);
Chris Lattner396dbfe2004-06-09 05:08:07 +00003176 } else {
3177 WorkList.push_back(Result);
3178 AddUsersToWorkList(*Result);
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003179 }
Chris Lattner053c0932002-05-14 15:24:07 +00003180 }
Chris Lattner260ab202002-04-18 17:39:14 +00003181 Changed = true;
Chris Lattnerca081252001-12-14 16:52:21 +00003182 }
3183 }
3184
Chris Lattner260ab202002-04-18 17:39:14 +00003185 return Changed;
Chris Lattner04805fa2002-02-26 21:46:54 +00003186}
3187
Chris Lattner8427bff2003-12-07 01:24:23 +00003188Pass *llvm::createInstructionCombiningPass() {
Chris Lattner260ab202002-04-18 17:39:14 +00003189 return new InstCombiner();
Chris Lattner04805fa2002-02-26 21:46:54 +00003190}
Brian Gaeke960707c2003-11-11 22:41:34 +00003191