blob: 6b993750a3fc613ed1283b38801ca100b097352c [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,
1969 const Type *DstTy) {
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
1977 // Allow free casting and conversion of sizes as long as the sign doesn't
1978 // change...
Chris Lattnerb0b412e2002-09-03 01:08:28 +00001979 if (SrcTy->isIntegral() && MidTy->isIntegral() && DstTy->isIntegral()) {
Chris Lattner4e2dbc62004-07-20 00:59:32 +00001980 CastType FirstCast = getCastType(SrcTy, MidTy);
1981 CastType SecondCast = getCastType(MidTy, DstTy);
Chris Lattner650b6da2002-08-02 20:00:25 +00001982
Chris Lattner4e2dbc62004-07-20 00:59:32 +00001983 // Capture the effect of these two casts. If the result is a legal cast,
1984 // the CastType is stored here, otherwise a special code is used.
1985 static const unsigned CastResult[] = {
1986 // First cast is noop
1987 0, 1, 2, 3,
1988 // First cast is a truncate
1989 1, 1, 4, 4, // trunc->extend is not safe to eliminate
1990 // First cast is a sign ext
1991 2, 5, 2, 4, // signext->trunc always ok, signext->zeroext never ok
1992 // First cast is a zero ext
1993 3, 5, 3, 3, // zeroext->trunc always ok
1994 };
1995
1996 unsigned Result = CastResult[FirstCast*4+SecondCast];
1997 switch (Result) {
1998 default: assert(0 && "Illegal table value!");
1999 case 0:
2000 case 1:
2001 case 2:
2002 case 3:
2003 // FIXME: in the future, when LLVM has explicit sign/zeroextends and
2004 // truncates, we could eliminate more casts.
2005 return (unsigned)getCastType(SrcTy, DstTy) == Result;
2006 case 4:
2007 return false; // Not possible to eliminate this here.
2008 case 5:
2009 // Sign or zero extend followed by truncate is always ok
Chris Lattner650b6da2002-08-02 20:00:25 +00002010 return true;
Chris Lattner3732aca2002-08-15 16:15:25 +00002011 }
Chris Lattner650b6da2002-08-02 20:00:25 +00002012 }
Chris Lattner48a44f72002-05-02 17:06:02 +00002013
2014 // Otherwise, we cannot succeed. Specifically we do not want to allow things
2015 // like: short -> ushort -> uint, because this can create wrong results if
2016 // the input short is negative!
2017 //
2018 return false;
2019}
2020
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002021static bool ValueRequiresCast(const Value *V, const Type *Ty) {
2022 if (V->getType() == Ty || isa<Constant>(V)) return false;
2023 if (const CastInst *CI = dyn_cast<CastInst>(V))
2024 if (isEliminableCastOfCast(CI->getOperand(0)->getType(), CI->getType(), Ty))
2025 return false;
2026 return true;
2027}
2028
2029/// InsertOperandCastBefore - This inserts a cast of V to DestTy before the
2030/// InsertBefore instruction. This is specialized a bit to avoid inserting
2031/// casts that are known to not do anything...
2032///
2033Value *InstCombiner::InsertOperandCastBefore(Value *V, const Type *DestTy,
2034 Instruction *InsertBefore) {
2035 if (V->getType() == DestTy) return V;
2036 if (Constant *C = dyn_cast<Constant>(V))
2037 return ConstantExpr::getCast(C, DestTy);
2038
2039 CastInst *CI = new CastInst(V, DestTy, V->getName());
2040 InsertNewInstBefore(CI, *InsertBefore);
2041 return CI;
2042}
Chris Lattner48a44f72002-05-02 17:06:02 +00002043
2044// CastInst simplification
Chris Lattner260ab202002-04-18 17:39:14 +00002045//
Chris Lattner113f4f42002-06-25 16:13:24 +00002046Instruction *InstCombiner::visitCastInst(CastInst &CI) {
Chris Lattner55d4bda2003-06-23 21:59:52 +00002047 Value *Src = CI.getOperand(0);
2048
Chris Lattner48a44f72002-05-02 17:06:02 +00002049 // If the user is casting a value to the same type, eliminate this cast
2050 // instruction...
Chris Lattner55d4bda2003-06-23 21:59:52 +00002051 if (CI.getType() == Src->getType())
2052 return ReplaceInstUsesWith(CI, Src);
Chris Lattner48a44f72002-05-02 17:06:02 +00002053
Chris Lattner48a44f72002-05-02 17:06:02 +00002054 // If casting the result of another cast instruction, try to eliminate this
2055 // one!
2056 //
Chris Lattner55d4bda2003-06-23 21:59:52 +00002057 if (CastInst *CSrc = dyn_cast<CastInst>(Src)) {
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002058 if (isEliminableCastOfCast(CSrc->getOperand(0)->getType(),
2059 CSrc->getType(), CI.getType())) {
Chris Lattner48a44f72002-05-02 17:06:02 +00002060 // This instruction now refers directly to the cast's src operand. This
2061 // has a good chance of making CSrc dead.
Chris Lattner113f4f42002-06-25 16:13:24 +00002062 CI.setOperand(0, CSrc->getOperand(0));
2063 return &CI;
Chris Lattner48a44f72002-05-02 17:06:02 +00002064 }
2065
Chris Lattner650b6da2002-08-02 20:00:25 +00002066 // If this is an A->B->A cast, and we are dealing with integral types, try
2067 // to convert this into a logical 'and' instruction.
2068 //
2069 if (CSrc->getOperand(0)->getType() == CI.getType() &&
Chris Lattnerb0b412e2002-09-03 01:08:28 +00002070 CI.getType()->isInteger() && CSrc->getType()->isInteger() &&
Chris Lattner650b6da2002-08-02 20:00:25 +00002071 CI.getType()->isUnsigned() && CSrc->getType()->isUnsigned() &&
2072 CSrc->getType()->getPrimitiveSize() < CI.getType()->getPrimitiveSize()){
2073 assert(CSrc->getType() != Type::ULongTy &&
2074 "Cannot have type bigger than ulong!");
Chris Lattner196897c2003-05-26 23:41:32 +00002075 uint64_t AndValue = (1ULL << CSrc->getType()->getPrimitiveSize()*8)-1;
Chris Lattner650b6da2002-08-02 20:00:25 +00002076 Constant *AndOp = ConstantUInt::get(CI.getType(), AndValue);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002077 return BinaryOperator::createAnd(CSrc->getOperand(0), AndOp);
Chris Lattner650b6da2002-08-02 20:00:25 +00002078 }
2079 }
2080
Chris Lattner03841652004-05-25 04:29:21 +00002081 // If this is a cast to bool, turn it into the appropriate setne instruction.
2082 if (CI.getType() == Type::BoolTy)
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002083 return BinaryOperator::createSetNE(CI.getOperand(0),
Chris Lattner03841652004-05-25 04:29:21 +00002084 Constant::getNullValue(CI.getOperand(0)->getType()));
2085
Chris Lattnerd0d51602003-06-21 23:12:02 +00002086 // If casting the result of a getelementptr instruction with no offset, turn
2087 // this into a cast of the original pointer!
2088 //
Chris Lattner55d4bda2003-06-23 21:59:52 +00002089 if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Src)) {
Chris Lattnerd0d51602003-06-21 23:12:02 +00002090 bool AllZeroOperands = true;
2091 for (unsigned i = 1, e = GEP->getNumOperands(); i != e; ++i)
2092 if (!isa<Constant>(GEP->getOperand(i)) ||
2093 !cast<Constant>(GEP->getOperand(i))->isNullValue()) {
2094 AllZeroOperands = false;
2095 break;
2096 }
2097 if (AllZeroOperands) {
2098 CI.setOperand(0, GEP->getOperand(0));
2099 return &CI;
2100 }
2101 }
2102
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002103 // If we are casting a malloc or alloca to a pointer to a type of the same
2104 // size, rewrite the allocation instruction to allocate the "right" type.
2105 //
2106 if (AllocationInst *AI = dyn_cast<AllocationInst>(Src))
Chris Lattnerd4d987d2003-11-02 06:54:48 +00002107 if (AI->hasOneUse() && !AI->isArrayAllocation())
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002108 if (const PointerType *PTy = dyn_cast<PointerType>(CI.getType())) {
2109 // Get the type really allocated and the type casted to...
2110 const Type *AllocElTy = AI->getAllocatedType();
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002111 const Type *CastElTy = PTy->getElementType();
Chris Lattner9eb9ccd2004-07-06 19:28:42 +00002112 if (AllocElTy->isSized() && CastElTy->isSized()) {
2113 unsigned AllocElTySize = TD->getTypeSize(AllocElTy);
2114 unsigned CastElTySize = TD->getTypeSize(CastElTy);
Chris Lattner7c94d112003-11-05 17:31:36 +00002115
Chris Lattner9eb9ccd2004-07-06 19:28:42 +00002116 // If the allocation is for an even multiple of the cast type size
2117 if (CastElTySize && (AllocElTySize % CastElTySize == 0)) {
2118 Value *Amt = ConstantUInt::get(Type::UIntTy,
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002119 AllocElTySize/CastElTySize);
Chris Lattner9eb9ccd2004-07-06 19:28:42 +00002120 std::string Name = AI->getName(); AI->setName("");
2121 AllocationInst *New;
2122 if (isa<MallocInst>(AI))
2123 New = new MallocInst(CastElTy, Amt, Name);
2124 else
2125 New = new AllocaInst(CastElTy, Amt, Name);
2126 InsertNewInstBefore(New, *AI);
2127 return ReplaceInstUsesWith(CI, New);
2128 }
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002129 }
2130 }
2131
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002132 // If the source value is an instruction with only this use, we can attempt to
2133 // propagate the cast into the instruction. Also, only handle integral types
2134 // for now.
2135 if (Instruction *SrcI = dyn_cast<Instruction>(Src))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00002136 if (SrcI->hasOneUse() && Src->getType()->isIntegral() &&
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002137 CI.getType()->isInteger()) { // Don't mess with casts to bool here
2138 const Type *DestTy = CI.getType();
2139 unsigned SrcBitSize = getTypeSizeInBits(Src->getType());
2140 unsigned DestBitSize = getTypeSizeInBits(DestTy);
2141
2142 Value *Op0 = SrcI->getNumOperands() > 0 ? SrcI->getOperand(0) : 0;
2143 Value *Op1 = SrcI->getNumOperands() > 1 ? SrcI->getOperand(1) : 0;
2144
2145 switch (SrcI->getOpcode()) {
2146 case Instruction::Add:
2147 case Instruction::Mul:
2148 case Instruction::And:
2149 case Instruction::Or:
2150 case Instruction::Xor:
2151 // If we are discarding information, or just changing the sign, rewrite.
2152 if (DestBitSize <= SrcBitSize && DestBitSize != 1) {
2153 // Don't insert two casts if they cannot be eliminated. We allow two
2154 // casts to be inserted if the sizes are the same. This could only be
2155 // converting signedness, which is a noop.
2156 if (DestBitSize == SrcBitSize || !ValueRequiresCast(Op1, DestTy) ||
2157 !ValueRequiresCast(Op0, DestTy)) {
2158 Value *Op0c = InsertOperandCastBefore(Op0, DestTy, SrcI);
2159 Value *Op1c = InsertOperandCastBefore(Op1, DestTy, SrcI);
2160 return BinaryOperator::create(cast<BinaryOperator>(SrcI)
2161 ->getOpcode(), Op0c, Op1c);
2162 }
2163 }
2164 break;
2165 case Instruction::Shl:
2166 // Allow changing the sign of the source operand. Do not allow changing
2167 // the size of the shift, UNLESS the shift amount is a constant. We
2168 // mush not change variable sized shifts to a smaller size, because it
2169 // is undefined to shift more bits out than exist in the value.
2170 if (DestBitSize == SrcBitSize ||
2171 (DestBitSize < SrcBitSize && isa<Constant>(Op1))) {
2172 Value *Op0c = InsertOperandCastBefore(Op0, DestTy, SrcI);
2173 return new ShiftInst(Instruction::Shl, Op0c, Op1);
2174 }
2175 break;
2176 }
2177 }
2178
Chris Lattner260ab202002-04-18 17:39:14 +00002179 return 0;
Chris Lattnerca081252001-12-14 16:52:21 +00002180}
2181
Chris Lattner56e4d3d2004-04-09 23:46:01 +00002182/// GetSelectFoldableOperands - We want to turn code that looks like this:
2183/// %C = or %A, %B
2184/// %D = select %cond, %C, %A
2185/// into:
2186/// %C = select %cond, %B, 0
2187/// %D = or %A, %C
2188///
2189/// Assuming that the specified instruction is an operand to the select, return
2190/// a bitmask indicating which operands of this instruction are foldable if they
2191/// equal the other incoming value of the select.
2192///
2193static unsigned GetSelectFoldableOperands(Instruction *I) {
2194 switch (I->getOpcode()) {
2195 case Instruction::Add:
2196 case Instruction::Mul:
2197 case Instruction::And:
2198 case Instruction::Or:
2199 case Instruction::Xor:
2200 return 3; // Can fold through either operand.
2201 case Instruction::Sub: // Can only fold on the amount subtracted.
2202 case Instruction::Shl: // Can only fold on the shift amount.
2203 case Instruction::Shr:
2204 return 1;
2205 default:
2206 return 0; // Cannot fold
2207 }
2208}
2209
2210/// GetSelectFoldableConstant - For the same transformation as the previous
2211/// function, return the identity constant that goes into the select.
2212static Constant *GetSelectFoldableConstant(Instruction *I) {
2213 switch (I->getOpcode()) {
2214 default: assert(0 && "This cannot happen!"); abort();
2215 case Instruction::Add:
2216 case Instruction::Sub:
2217 case Instruction::Or:
2218 case Instruction::Xor:
2219 return Constant::getNullValue(I->getType());
2220 case Instruction::Shl:
2221 case Instruction::Shr:
2222 return Constant::getNullValue(Type::UByteTy);
2223 case Instruction::And:
2224 return ConstantInt::getAllOnesValue(I->getType());
2225 case Instruction::Mul:
2226 return ConstantInt::get(I->getType(), 1);
2227 }
2228}
2229
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002230Instruction *InstCombiner::visitSelectInst(SelectInst &SI) {
Chris Lattner533bc492004-03-30 19:37:13 +00002231 Value *CondVal = SI.getCondition();
2232 Value *TrueVal = SI.getTrueValue();
2233 Value *FalseVal = SI.getFalseValue();
2234
2235 // select true, X, Y -> X
2236 // select false, X, Y -> Y
2237 if (ConstantBool *C = dyn_cast<ConstantBool>(CondVal))
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002238 if (C == ConstantBool::True)
Chris Lattner533bc492004-03-30 19:37:13 +00002239 return ReplaceInstUsesWith(SI, TrueVal);
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002240 else {
2241 assert(C == ConstantBool::False);
Chris Lattner533bc492004-03-30 19:37:13 +00002242 return ReplaceInstUsesWith(SI, FalseVal);
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002243 }
Chris Lattner533bc492004-03-30 19:37:13 +00002244
2245 // select C, X, X -> X
2246 if (TrueVal == FalseVal)
2247 return ReplaceInstUsesWith(SI, TrueVal);
2248
Chris Lattner1c631e82004-04-08 04:43:23 +00002249 if (SI.getType() == Type::BoolTy)
2250 if (ConstantBool *C = dyn_cast<ConstantBool>(TrueVal)) {
2251 if (C == ConstantBool::True) {
2252 // Change: A = select B, true, C --> A = or B, C
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002253 return BinaryOperator::createOr(CondVal, FalseVal);
Chris Lattner1c631e82004-04-08 04:43:23 +00002254 } else {
2255 // Change: A = select B, false, C --> A = and !B, C
2256 Value *NotCond =
2257 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
2258 "not."+CondVal->getName()), SI);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002259 return BinaryOperator::createAnd(NotCond, FalseVal);
Chris Lattner1c631e82004-04-08 04:43:23 +00002260 }
2261 } else if (ConstantBool *C = dyn_cast<ConstantBool>(FalseVal)) {
2262 if (C == ConstantBool::False) {
2263 // Change: A = select B, C, false --> A = and B, C
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002264 return BinaryOperator::createAnd(CondVal, TrueVal);
Chris Lattner1c631e82004-04-08 04:43:23 +00002265 } else {
2266 // Change: A = select B, C, true --> A = or !B, C
2267 Value *NotCond =
2268 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
2269 "not."+CondVal->getName()), SI);
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002270 return BinaryOperator::createOr(NotCond, TrueVal);
Chris Lattner1c631e82004-04-08 04:43:23 +00002271 }
2272 }
2273
Chris Lattner183b3362004-04-09 19:05:30 +00002274 // Selecting between two integer constants?
2275 if (ConstantInt *TrueValC = dyn_cast<ConstantInt>(TrueVal))
2276 if (ConstantInt *FalseValC = dyn_cast<ConstantInt>(FalseVal)) {
2277 // select C, 1, 0 -> cast C to int
2278 if (FalseValC->isNullValue() && TrueValC->getRawValue() == 1) {
2279 return new CastInst(CondVal, SI.getType());
2280 } else if (TrueValC->isNullValue() && FalseValC->getRawValue() == 1) {
2281 // select C, 0, 1 -> cast !C to int
2282 Value *NotCond =
2283 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
Chris Lattnercf7baf32004-04-09 18:19:44 +00002284 "not."+CondVal->getName()), SI);
Chris Lattner183b3362004-04-09 19:05:30 +00002285 return new CastInst(NotCond, SI.getType());
Chris Lattnercf7baf32004-04-09 18:19:44 +00002286 }
Chris Lattner35167c32004-06-09 07:59:58 +00002287
2288 // If one of the constants is zero (we know they can't both be) and we
2289 // have a setcc instruction with zero, and we have an 'and' with the
2290 // non-constant value, eliminate this whole mess. This corresponds to
2291 // cases like this: ((X & 27) ? 27 : 0)
2292 if (TrueValC->isNullValue() || FalseValC->isNullValue())
2293 if (Instruction *IC = dyn_cast<Instruction>(SI.getCondition()))
2294 if ((IC->getOpcode() == Instruction::SetEQ ||
2295 IC->getOpcode() == Instruction::SetNE) &&
2296 isa<ConstantInt>(IC->getOperand(1)) &&
2297 cast<Constant>(IC->getOperand(1))->isNullValue())
2298 if (Instruction *ICA = dyn_cast<Instruction>(IC->getOperand(0)))
2299 if (ICA->getOpcode() == Instruction::And &&
2300 isa<ConstantInt>(ICA->getOperand(1)) &&
2301 (ICA->getOperand(1) == TrueValC ||
2302 ICA->getOperand(1) == FalseValC) &&
2303 isOneBitSet(cast<ConstantInt>(ICA->getOperand(1)))) {
2304 // Okay, now we know that everything is set up, we just don't
2305 // know whether we have a setne or seteq and whether the true or
2306 // false val is the zero.
2307 bool ShouldNotVal = !TrueValC->isNullValue();
2308 ShouldNotVal ^= IC->getOpcode() == Instruction::SetNE;
2309 Value *V = ICA;
2310 if (ShouldNotVal)
2311 V = InsertNewInstBefore(BinaryOperator::create(
2312 Instruction::Xor, V, ICA->getOperand(1)), SI);
2313 return ReplaceInstUsesWith(SI, V);
2314 }
Chris Lattner533bc492004-03-30 19:37:13 +00002315 }
Chris Lattner623fba12004-04-10 22:21:27 +00002316
2317 // See if we are selecting two values based on a comparison of the two values.
2318 if (SetCondInst *SCI = dyn_cast<SetCondInst>(CondVal)) {
2319 if (SCI->getOperand(0) == TrueVal && SCI->getOperand(1) == FalseVal) {
2320 // Transform (X == Y) ? X : Y -> Y
2321 if (SCI->getOpcode() == Instruction::SetEQ)
2322 return ReplaceInstUsesWith(SI, FalseVal);
2323 // Transform (X != Y) ? X : Y -> X
2324 if (SCI->getOpcode() == Instruction::SetNE)
2325 return ReplaceInstUsesWith(SI, TrueVal);
2326 // NOTE: if we wanted to, this is where to detect MIN/MAX/ABS/etc.
2327
2328 } else if (SCI->getOperand(0) == FalseVal && SCI->getOperand(1) == TrueVal){
2329 // Transform (X == Y) ? Y : X -> X
2330 if (SCI->getOpcode() == Instruction::SetEQ)
Chris Lattner24cf0202004-04-11 01:39:19 +00002331 return ReplaceInstUsesWith(SI, FalseVal);
Chris Lattner623fba12004-04-10 22:21:27 +00002332 // Transform (X != Y) ? Y : X -> Y
2333 if (SCI->getOpcode() == Instruction::SetNE)
Chris Lattner24cf0202004-04-11 01:39:19 +00002334 return ReplaceInstUsesWith(SI, TrueVal);
Chris Lattner623fba12004-04-10 22:21:27 +00002335 // NOTE: if we wanted to, this is where to detect MIN/MAX/ABS/etc.
2336 }
2337 }
Chris Lattner1c631e82004-04-08 04:43:23 +00002338
Chris Lattner56e4d3d2004-04-09 23:46:01 +00002339 // See if we can fold the select into one of our operands.
2340 if (SI.getType()->isInteger()) {
2341 // See the comment above GetSelectFoldableOperands for a description of the
2342 // transformation we are doing here.
2343 if (Instruction *TVI = dyn_cast<Instruction>(TrueVal))
2344 if (TVI->hasOneUse() && TVI->getNumOperands() == 2 &&
2345 !isa<Constant>(FalseVal))
2346 if (unsigned SFO = GetSelectFoldableOperands(TVI)) {
2347 unsigned OpToFold = 0;
2348 if ((SFO & 1) && FalseVal == TVI->getOperand(0)) {
2349 OpToFold = 1;
2350 } else if ((SFO & 2) && FalseVal == TVI->getOperand(1)) {
2351 OpToFold = 2;
2352 }
2353
2354 if (OpToFold) {
2355 Constant *C = GetSelectFoldableConstant(TVI);
2356 std::string Name = TVI->getName(); TVI->setName("");
2357 Instruction *NewSel =
2358 new SelectInst(SI.getCondition(), TVI->getOperand(2-OpToFold), C,
2359 Name);
2360 InsertNewInstBefore(NewSel, SI);
2361 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(TVI))
2362 return BinaryOperator::create(BO->getOpcode(), FalseVal, NewSel);
2363 else if (ShiftInst *SI = dyn_cast<ShiftInst>(TVI))
2364 return new ShiftInst(SI->getOpcode(), FalseVal, NewSel);
2365 else {
2366 assert(0 && "Unknown instruction!!");
2367 }
2368 }
2369 }
2370
2371 if (Instruction *FVI = dyn_cast<Instruction>(FalseVal))
2372 if (FVI->hasOneUse() && FVI->getNumOperands() == 2 &&
2373 !isa<Constant>(TrueVal))
2374 if (unsigned SFO = GetSelectFoldableOperands(FVI)) {
2375 unsigned OpToFold = 0;
2376 if ((SFO & 1) && TrueVal == FVI->getOperand(0)) {
2377 OpToFold = 1;
2378 } else if ((SFO & 2) && TrueVal == FVI->getOperand(1)) {
2379 OpToFold = 2;
2380 }
2381
2382 if (OpToFold) {
2383 Constant *C = GetSelectFoldableConstant(FVI);
2384 std::string Name = FVI->getName(); FVI->setName("");
2385 Instruction *NewSel =
2386 new SelectInst(SI.getCondition(), C, FVI->getOperand(2-OpToFold),
2387 Name);
2388 InsertNewInstBefore(NewSel, SI);
2389 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(FVI))
2390 return BinaryOperator::create(BO->getOpcode(), TrueVal, NewSel);
2391 else if (ShiftInst *SI = dyn_cast<ShiftInst>(FVI))
2392 return new ShiftInst(SI->getOpcode(), TrueVal, NewSel);
2393 else {
2394 assert(0 && "Unknown instruction!!");
2395 }
2396 }
2397 }
2398 }
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002399 return 0;
2400}
2401
2402
Chris Lattner970c33a2003-06-19 17:00:31 +00002403// CallInst simplification
2404//
2405Instruction *InstCombiner::visitCallInst(CallInst &CI) {
Chris Lattner51ea1272004-02-28 05:22:00 +00002406 // Intrinsics cannot occur in an invoke, so handle them here instead of in
2407 // visitCallSite.
2408 if (Function *F = CI.getCalledFunction())
2409 switch (F->getIntrinsicID()) {
2410 case Intrinsic::memmove:
2411 case Intrinsic::memcpy:
2412 case Intrinsic::memset:
2413 // memmove/cpy/set of zero bytes is a noop.
2414 if (Constant *NumBytes = dyn_cast<Constant>(CI.getOperand(3))) {
2415 if (NumBytes->isNullValue())
2416 return EraseInstFromFunction(CI);
2417 }
2418 break;
2419 default:
2420 break;
2421 }
2422
Chris Lattneraec3d942003-10-07 22:32:43 +00002423 return visitCallSite(&CI);
Chris Lattner970c33a2003-06-19 17:00:31 +00002424}
2425
2426// InvokeInst simplification
2427//
2428Instruction *InstCombiner::visitInvokeInst(InvokeInst &II) {
Chris Lattneraec3d942003-10-07 22:32:43 +00002429 return visitCallSite(&II);
Chris Lattner970c33a2003-06-19 17:00:31 +00002430}
2431
Chris Lattneraec3d942003-10-07 22:32:43 +00002432// visitCallSite - Improvements for call and invoke instructions.
2433//
2434Instruction *InstCombiner::visitCallSite(CallSite CS) {
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002435 bool Changed = false;
2436
2437 // If the callee is a constexpr cast of a function, attempt to move the cast
2438 // to the arguments of the call/invoke.
Chris Lattneraec3d942003-10-07 22:32:43 +00002439 if (transformConstExprCastCall(CS)) return 0;
2440
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002441 Value *Callee = CS.getCalledValue();
2442 const PointerType *PTy = cast<PointerType>(Callee->getType());
2443 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
2444 if (FTy->isVarArg()) {
2445 // See if we can optimize any arguments passed through the varargs area of
2446 // the call.
2447 for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(),
2448 E = CS.arg_end(); I != E; ++I)
2449 if (CastInst *CI = dyn_cast<CastInst>(*I)) {
2450 // If this cast does not effect the value passed through the varargs
2451 // area, we can eliminate the use of the cast.
2452 Value *Op = CI->getOperand(0);
2453 if (CI->getType()->isLosslesslyConvertibleTo(Op->getType())) {
2454 *I = Op;
2455 Changed = true;
2456 }
2457 }
2458 }
Chris Lattneraec3d942003-10-07 22:32:43 +00002459
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002460 return Changed ? CS.getInstruction() : 0;
Chris Lattneraec3d942003-10-07 22:32:43 +00002461}
2462
Chris Lattner970c33a2003-06-19 17:00:31 +00002463// transformConstExprCastCall - If the callee is a constexpr cast of a function,
2464// attempt to move the cast to the arguments of the call/invoke.
2465//
2466bool InstCombiner::transformConstExprCastCall(CallSite CS) {
2467 if (!isa<ConstantExpr>(CS.getCalledValue())) return false;
2468 ConstantExpr *CE = cast<ConstantExpr>(CS.getCalledValue());
Chris Lattnerf3edc492004-07-18 18:59:44 +00002469 if (CE->getOpcode() != Instruction::Cast || !isa<Function>(CE->getOperand(0)))
Chris Lattner970c33a2003-06-19 17:00:31 +00002470 return false;
Reid Spencer87436872004-07-18 00:38:32 +00002471 Function *Callee = cast<Function>(CE->getOperand(0));
Chris Lattner970c33a2003-06-19 17:00:31 +00002472 Instruction *Caller = CS.getInstruction();
2473
2474 // Okay, this is a cast from a function to a different type. Unless doing so
2475 // would cause a type conversion of one of our arguments, change this call to
2476 // be a direct call with arguments casted to the appropriate types.
2477 //
2478 const FunctionType *FT = Callee->getFunctionType();
2479 const Type *OldRetTy = Caller->getType();
2480
Chris Lattner1f7942f2004-01-14 06:06:08 +00002481 // Check to see if we are changing the return type...
2482 if (OldRetTy != FT->getReturnType()) {
2483 if (Callee->isExternal() &&
2484 !OldRetTy->isLosslesslyConvertibleTo(FT->getReturnType()) &&
2485 !Caller->use_empty())
2486 return false; // Cannot transform this return value...
2487
2488 // If the callsite is an invoke instruction, and the return value is used by
2489 // a PHI node in a successor, we cannot change the return type of the call
2490 // because there is no place to put the cast instruction (without breaking
2491 // the critical edge). Bail out in this case.
2492 if (!Caller->use_empty())
2493 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller))
2494 for (Value::use_iterator UI = II->use_begin(), E = II->use_end();
2495 UI != E; ++UI)
2496 if (PHINode *PN = dyn_cast<PHINode>(*UI))
2497 if (PN->getParent() == II->getNormalDest() ||
Chris Lattnerfae8ab32004-02-08 21:44:31 +00002498 PN->getParent() == II->getUnwindDest())
Chris Lattner1f7942f2004-01-14 06:06:08 +00002499 return false;
2500 }
Chris Lattner970c33a2003-06-19 17:00:31 +00002501
2502 unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin());
2503 unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs);
2504
2505 CallSite::arg_iterator AI = CS.arg_begin();
2506 for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) {
2507 const Type *ParamTy = FT->getParamType(i);
2508 bool isConvertible = (*AI)->getType()->isLosslesslyConvertibleTo(ParamTy);
2509 if (Callee->isExternal() && !isConvertible) return false;
2510 }
2511
2512 if (FT->getNumParams() < NumActualArgs && !FT->isVarArg() &&
2513 Callee->isExternal())
2514 return false; // Do not delete arguments unless we have a function body...
2515
2516 // Okay, we decided that this is a safe thing to do: go ahead and start
2517 // inserting cast instructions as necessary...
2518 std::vector<Value*> Args;
2519 Args.reserve(NumActualArgs);
2520
2521 AI = CS.arg_begin();
2522 for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) {
2523 const Type *ParamTy = FT->getParamType(i);
2524 if ((*AI)->getType() == ParamTy) {
2525 Args.push_back(*AI);
2526 } else {
Chris Lattner1c631e82004-04-08 04:43:23 +00002527 Args.push_back(InsertNewInstBefore(new CastInst(*AI, ParamTy, "tmp"),
2528 *Caller));
Chris Lattner970c33a2003-06-19 17:00:31 +00002529 }
2530 }
2531
2532 // If the function takes more arguments than the call was taking, add them
2533 // now...
2534 for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i)
2535 Args.push_back(Constant::getNullValue(FT->getParamType(i)));
2536
2537 // If we are removing arguments to the function, emit an obnoxious warning...
2538 if (FT->getNumParams() < NumActualArgs)
2539 if (!FT->isVarArg()) {
2540 std::cerr << "WARNING: While resolving call to function '"
2541 << Callee->getName() << "' arguments were dropped!\n";
2542 } else {
2543 // Add all of the arguments in their promoted form to the arg list...
2544 for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) {
2545 const Type *PTy = getPromotedType((*AI)->getType());
2546 if (PTy != (*AI)->getType()) {
2547 // Must promote to pass through va_arg area!
2548 Instruction *Cast = new CastInst(*AI, PTy, "tmp");
2549 InsertNewInstBefore(Cast, *Caller);
2550 Args.push_back(Cast);
2551 } else {
2552 Args.push_back(*AI);
2553 }
2554 }
2555 }
2556
2557 if (FT->getReturnType() == Type::VoidTy)
2558 Caller->setName(""); // Void type should not have a name...
2559
2560 Instruction *NC;
2561 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) {
Chris Lattnerfae8ab32004-02-08 21:44:31 +00002562 NC = new InvokeInst(Callee, II->getNormalDest(), II->getUnwindDest(),
Chris Lattner970c33a2003-06-19 17:00:31 +00002563 Args, Caller->getName(), Caller);
2564 } else {
2565 NC = new CallInst(Callee, Args, Caller->getName(), Caller);
2566 }
2567
2568 // Insert a cast of the return type as necessary...
2569 Value *NV = NC;
2570 if (Caller->getType() != NV->getType() && !Caller->use_empty()) {
2571 if (NV->getType() != Type::VoidTy) {
2572 NV = NC = new CastInst(NC, Caller->getType(), "tmp");
Chris Lattner686767f2003-10-30 00:46:41 +00002573
2574 // If this is an invoke instruction, we should insert it after the first
2575 // non-phi, instruction in the normal successor block.
2576 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) {
2577 BasicBlock::iterator I = II->getNormalDest()->begin();
2578 while (isa<PHINode>(I)) ++I;
2579 InsertNewInstBefore(NC, *I);
2580 } else {
2581 // Otherwise, it's a call, just insert cast right after the call instr
2582 InsertNewInstBefore(NC, *Caller);
2583 }
Chris Lattner51ea1272004-02-28 05:22:00 +00002584 AddUsersToWorkList(*Caller);
Chris Lattner970c33a2003-06-19 17:00:31 +00002585 } else {
2586 NV = Constant::getNullValue(Caller->getType());
2587 }
2588 }
2589
2590 if (Caller->getType() != Type::VoidTy && !Caller->use_empty())
2591 Caller->replaceAllUsesWith(NV);
2592 Caller->getParent()->getInstList().erase(Caller);
2593 removeFromWorkList(Caller);
2594 return true;
2595}
2596
2597
Chris Lattner48a44f72002-05-02 17:06:02 +00002598
Chris Lattnerbbbdd852002-05-06 18:06:38 +00002599// PHINode simplification
2600//
Chris Lattner113f4f42002-06-25 16:13:24 +00002601Instruction *InstCombiner::visitPHINode(PHINode &PN) {
Chris Lattner91daeb52003-12-19 05:58:40 +00002602 if (Value *V = hasConstantValue(&PN))
2603 return ReplaceInstUsesWith(PN, V);
Chris Lattner4db2d222004-02-16 05:07:08 +00002604
2605 // If the only user of this instruction is a cast instruction, and all of the
2606 // incoming values are constants, change this PHI to merge together the casted
2607 // constants.
2608 if (PN.hasOneUse())
2609 if (CastInst *CI = dyn_cast<CastInst>(PN.use_back()))
2610 if (CI->getType() != PN.getType()) { // noop casts will be folded
2611 bool AllConstant = true;
2612 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i)
2613 if (!isa<Constant>(PN.getIncomingValue(i))) {
2614 AllConstant = false;
2615 break;
2616 }
2617 if (AllConstant) {
2618 // Make a new PHI with all casted values.
2619 PHINode *New = new PHINode(CI->getType(), PN.getName(), &PN);
2620 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) {
2621 Constant *OldArg = cast<Constant>(PN.getIncomingValue(i));
2622 New->addIncoming(ConstantExpr::getCast(OldArg, New->getType()),
2623 PN.getIncomingBlock(i));
2624 }
2625
2626 // Update the cast instruction.
2627 CI->setOperand(0, New);
2628 WorkList.push_back(CI); // revisit the cast instruction to fold.
2629 WorkList.push_back(New); // Make sure to revisit the new Phi
2630 return &PN; // PN is now dead!
2631 }
2632 }
Chris Lattner91daeb52003-12-19 05:58:40 +00002633 return 0;
Chris Lattnerbbbdd852002-05-06 18:06:38 +00002634}
2635
Chris Lattner69193f92004-04-05 01:30:19 +00002636static Value *InsertSignExtendToPtrTy(Value *V, const Type *DTy,
2637 Instruction *InsertPoint,
2638 InstCombiner *IC) {
2639 unsigned PS = IC->getTargetData().getPointerSize();
2640 const Type *VTy = V->getType();
2641 Instruction *Cast;
2642 if (!VTy->isSigned() && VTy->getPrimitiveSize() < PS)
2643 // We must insert a cast to ensure we sign-extend.
2644 V = IC->InsertNewInstBefore(new CastInst(V, VTy->getSignedVersion(),
2645 V->getName()), *InsertPoint);
2646 return IC->InsertNewInstBefore(new CastInst(V, DTy, V->getName()),
2647 *InsertPoint);
2648}
2649
Chris Lattner48a44f72002-05-02 17:06:02 +00002650
Chris Lattner113f4f42002-06-25 16:13:24 +00002651Instruction *InstCombiner::visitGetElementPtrInst(GetElementPtrInst &GEP) {
Chris Lattner5f667a62004-05-07 22:09:22 +00002652 Value *PtrOp = GEP.getOperand(0);
Chris Lattner471bd762003-05-22 19:07:21 +00002653 // Is it 'getelementptr %P, long 0' or 'getelementptr %P'
Chris Lattner113f4f42002-06-25 16:13:24 +00002654 // If so, eliminate the noop.
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002655 if (GEP.getNumOperands() == 1)
Chris Lattner5f667a62004-05-07 22:09:22 +00002656 return ReplaceInstUsesWith(GEP, PtrOp);
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002657
2658 bool HasZeroPointerIndex = false;
2659 if (Constant *C = dyn_cast<Constant>(GEP.getOperand(1)))
2660 HasZeroPointerIndex = C->isNullValue();
2661
2662 if (GEP.getNumOperands() == 2 && HasZeroPointerIndex)
Chris Lattner5f667a62004-05-07 22:09:22 +00002663 return ReplaceInstUsesWith(GEP, PtrOp);
Chris Lattner48a44f72002-05-02 17:06:02 +00002664
Chris Lattner69193f92004-04-05 01:30:19 +00002665 // Eliminate unneeded casts for indices.
2666 bool MadeChange = false;
Chris Lattner2b2412d2004-04-07 18:38:20 +00002667 gep_type_iterator GTI = gep_type_begin(GEP);
2668 for (unsigned i = 1, e = GEP.getNumOperands(); i != e; ++i, ++GTI)
2669 if (isa<SequentialType>(*GTI)) {
2670 if (CastInst *CI = dyn_cast<CastInst>(GEP.getOperand(i))) {
2671 Value *Src = CI->getOperand(0);
2672 const Type *SrcTy = Src->getType();
2673 const Type *DestTy = CI->getType();
2674 if (Src->getType()->isInteger()) {
2675 if (SrcTy->getPrimitiveSize() == DestTy->getPrimitiveSize()) {
2676 // We can always eliminate a cast from ulong or long to the other.
2677 // We can always eliminate a cast from uint to int or the other on
2678 // 32-bit pointer platforms.
2679 if (DestTy->getPrimitiveSize() >= TD->getPointerSize()) {
2680 MadeChange = true;
2681 GEP.setOperand(i, Src);
2682 }
2683 } else if (SrcTy->getPrimitiveSize() < DestTy->getPrimitiveSize() &&
2684 SrcTy->getPrimitiveSize() == 4) {
2685 // We can always eliminate a cast from int to [u]long. We can
2686 // eliminate a cast from uint to [u]long iff the target is a 32-bit
2687 // pointer target.
2688 if (SrcTy->isSigned() ||
2689 SrcTy->getPrimitiveSize() >= TD->getPointerSize()) {
2690 MadeChange = true;
2691 GEP.setOperand(i, Src);
2692 }
Chris Lattner69193f92004-04-05 01:30:19 +00002693 }
2694 }
2695 }
Chris Lattner2b2412d2004-04-07 18:38:20 +00002696 // If we are using a wider index than needed for this platform, shrink it
2697 // to what we need. If the incoming value needs a cast instruction,
2698 // insert it. This explicit cast can make subsequent optimizations more
2699 // obvious.
2700 Value *Op = GEP.getOperand(i);
2701 if (Op->getType()->getPrimitiveSize() > TD->getPointerSize())
Chris Lattner1e9ac1a2004-04-17 18:16:10 +00002702 if (Constant *C = dyn_cast<Constant>(Op)) {
2703 GEP.setOperand(i, ConstantExpr::getCast(C, TD->getIntPtrType()));
2704 MadeChange = true;
2705 } else {
Chris Lattner2b2412d2004-04-07 18:38:20 +00002706 Op = InsertNewInstBefore(new CastInst(Op, TD->getIntPtrType(),
2707 Op->getName()), GEP);
2708 GEP.setOperand(i, Op);
2709 MadeChange = true;
2710 }
Chris Lattner69193f92004-04-05 01:30:19 +00002711 }
2712 if (MadeChange) return &GEP;
2713
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002714 // Combine Indices - If the source pointer to this getelementptr instruction
2715 // is a getelementptr instruction, combine the indices of the two
2716 // getelementptr instructions into a single instruction.
2717 //
Chris Lattner57c67b02004-03-25 22:59:29 +00002718 std::vector<Value*> SrcGEPOperands;
Chris Lattner5f667a62004-05-07 22:09:22 +00002719 if (GetElementPtrInst *Src = dyn_cast<GetElementPtrInst>(PtrOp)) {
Chris Lattner57c67b02004-03-25 22:59:29 +00002720 SrcGEPOperands.assign(Src->op_begin(), Src->op_end());
Chris Lattner5f667a62004-05-07 22:09:22 +00002721 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(PtrOp)) {
Chris Lattner57c67b02004-03-25 22:59:29 +00002722 if (CE->getOpcode() == Instruction::GetElementPtr)
2723 SrcGEPOperands.assign(CE->op_begin(), CE->op_end());
2724 }
2725
2726 if (!SrcGEPOperands.empty()) {
Chris Lattner5f667a62004-05-07 22:09:22 +00002727 // Note that if our source is a gep chain itself that we wait for that
2728 // chain to be resolved before we perform this transformation. This
2729 // avoids us creating a TON of code in some cases.
2730 //
2731 if (isa<GetElementPtrInst>(SrcGEPOperands[0]) &&
2732 cast<Instruction>(SrcGEPOperands[0])->getNumOperands() == 2)
2733 return 0; // Wait until our source is folded to completion.
2734
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002735 std::vector<Value *> Indices;
Chris Lattner5f667a62004-05-07 22:09:22 +00002736
2737 // Find out whether the last index in the source GEP is a sequential idx.
2738 bool EndsWithSequential = false;
2739 for (gep_type_iterator I = gep_type_begin(*cast<User>(PtrOp)),
2740 E = gep_type_end(*cast<User>(PtrOp)); I != E; ++I)
Chris Lattner8ec5f882004-05-08 22:41:42 +00002741 EndsWithSequential = !isa<StructType>(*I);
Chris Lattnerca081252001-12-14 16:52:21 +00002742
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002743 // Can we combine the two pointer arithmetics offsets?
Chris Lattner5f667a62004-05-07 22:09:22 +00002744 if (EndsWithSequential) {
Chris Lattner235af562003-03-05 22:33:14 +00002745 // Replace: gep (gep %P, long B), long A, ...
2746 // With: T = long A+B; gep %P, T, ...
2747 //
Chris Lattner5f667a62004-05-07 22:09:22 +00002748 Value *Sum, *SO1 = SrcGEPOperands.back(), *GO1 = GEP.getOperand(1);
Chris Lattner69193f92004-04-05 01:30:19 +00002749 if (SO1 == Constant::getNullValue(SO1->getType())) {
2750 Sum = GO1;
2751 } else if (GO1 == Constant::getNullValue(GO1->getType())) {
2752 Sum = SO1;
2753 } else {
2754 // If they aren't the same type, convert both to an integer of the
2755 // target's pointer size.
2756 if (SO1->getType() != GO1->getType()) {
2757 if (Constant *SO1C = dyn_cast<Constant>(SO1)) {
2758 SO1 = ConstantExpr::getCast(SO1C, GO1->getType());
2759 } else if (Constant *GO1C = dyn_cast<Constant>(GO1)) {
2760 GO1 = ConstantExpr::getCast(GO1C, SO1->getType());
2761 } else {
2762 unsigned PS = TD->getPointerSize();
2763 Instruction *Cast;
2764 if (SO1->getType()->getPrimitiveSize() == PS) {
2765 // Convert GO1 to SO1's type.
2766 GO1 = InsertSignExtendToPtrTy(GO1, SO1->getType(), &GEP, this);
2767
2768 } else if (GO1->getType()->getPrimitiveSize() == PS) {
2769 // Convert SO1 to GO1's type.
2770 SO1 = InsertSignExtendToPtrTy(SO1, GO1->getType(), &GEP, this);
2771 } else {
2772 const Type *PT = TD->getIntPtrType();
2773 SO1 = InsertSignExtendToPtrTy(SO1, PT, &GEP, this);
2774 GO1 = InsertSignExtendToPtrTy(GO1, PT, &GEP, this);
2775 }
2776 }
2777 }
Chris Lattner5f667a62004-05-07 22:09:22 +00002778 if (isa<Constant>(SO1) && isa<Constant>(GO1))
2779 Sum = ConstantExpr::getAdd(cast<Constant>(SO1), cast<Constant>(GO1));
2780 else {
Chris Lattnerdf20a4d2004-06-10 02:07:29 +00002781 Sum = BinaryOperator::createAdd(SO1, GO1, PtrOp->getName()+".sum");
2782 InsertNewInstBefore(cast<Instruction>(Sum), GEP);
Chris Lattner5f667a62004-05-07 22:09:22 +00002783 }
Chris Lattner69193f92004-04-05 01:30:19 +00002784 }
Chris Lattner5f667a62004-05-07 22:09:22 +00002785
2786 // Recycle the GEP we already have if possible.
2787 if (SrcGEPOperands.size() == 2) {
2788 GEP.setOperand(0, SrcGEPOperands[0]);
2789 GEP.setOperand(1, Sum);
2790 return &GEP;
2791 } else {
2792 Indices.insert(Indices.end(), SrcGEPOperands.begin()+1,
2793 SrcGEPOperands.end()-1);
2794 Indices.push_back(Sum);
2795 Indices.insert(Indices.end(), GEP.op_begin()+2, GEP.op_end());
2796 }
Chris Lattner69193f92004-04-05 01:30:19 +00002797 } else if (isa<Constant>(*GEP.idx_begin()) &&
2798 cast<Constant>(*GEP.idx_begin())->isNullValue() &&
Chris Lattner57c67b02004-03-25 22:59:29 +00002799 SrcGEPOperands.size() != 1) {
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002800 // Otherwise we can do the fold if the first index of the GEP is a zero
Chris Lattner57c67b02004-03-25 22:59:29 +00002801 Indices.insert(Indices.end(), SrcGEPOperands.begin()+1,
2802 SrcGEPOperands.end());
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002803 Indices.insert(Indices.end(), GEP.idx_begin()+1, GEP.idx_end());
2804 }
2805
2806 if (!Indices.empty())
Chris Lattner57c67b02004-03-25 22:59:29 +00002807 return new GetElementPtrInst(SrcGEPOperands[0], Indices, GEP.getName());
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002808
Chris Lattner5f667a62004-05-07 22:09:22 +00002809 } else if (GlobalValue *GV = dyn_cast<GlobalValue>(PtrOp)) {
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002810 // GEP of global variable. If all of the indices for this GEP are
2811 // constants, we can promote this to a constexpr instead of an instruction.
2812
2813 // Scan for nonconstants...
2814 std::vector<Constant*> Indices;
2815 User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end();
2816 for (; I != E && isa<Constant>(*I); ++I)
2817 Indices.push_back(cast<Constant>(*I));
2818
2819 if (I == E) { // If they are all constants...
Chris Lattnerf3edc492004-07-18 18:59:44 +00002820 Constant *CE = ConstantExpr::getGetElementPtr(GV, Indices);
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002821
2822 // Replace all uses of the GEP with the new constexpr...
2823 return ReplaceInstUsesWith(GEP, CE);
2824 }
Chris Lattner5f667a62004-05-07 22:09:22 +00002825 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(PtrOp)) {
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002826 if (CE->getOpcode() == Instruction::Cast) {
2827 if (HasZeroPointerIndex) {
2828 // transform: GEP (cast [10 x ubyte]* X to [0 x ubyte]*), long 0, ...
2829 // into : GEP [10 x ubyte]* X, long 0, ...
2830 //
2831 // This occurs when the program declares an array extern like "int X[];"
2832 //
2833 Constant *X = CE->getOperand(0);
2834 const PointerType *CPTy = cast<PointerType>(CE->getType());
2835 if (const PointerType *XTy = dyn_cast<PointerType>(X->getType()))
2836 if (const ArrayType *XATy =
2837 dyn_cast<ArrayType>(XTy->getElementType()))
2838 if (const ArrayType *CATy =
2839 dyn_cast<ArrayType>(CPTy->getElementType()))
2840 if (CATy->getElementType() == XATy->getElementType()) {
2841 // At this point, we know that the cast source type is a pointer
2842 // to an array of the same type as the destination pointer
2843 // array. Because the array type is never stepped over (there
2844 // is a leading zero) we can fold the cast into this GEP.
2845 GEP.setOperand(0, X);
2846 return &GEP;
2847 }
2848 }
2849 }
Chris Lattnerca081252001-12-14 16:52:21 +00002850 }
2851
Chris Lattnerca081252001-12-14 16:52:21 +00002852 return 0;
2853}
2854
Chris Lattner1085bdf2002-11-04 16:18:53 +00002855Instruction *InstCombiner::visitAllocationInst(AllocationInst &AI) {
2856 // Convert: malloc Ty, C - where C is a constant != 1 into: malloc [C x Ty], 1
2857 if (AI.isArrayAllocation()) // Check C != 1
2858 if (const ConstantUInt *C = dyn_cast<ConstantUInt>(AI.getArraySize())) {
2859 const Type *NewTy = ArrayType::get(AI.getAllocatedType(), C->getValue());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002860 AllocationInst *New = 0;
Chris Lattner1085bdf2002-11-04 16:18:53 +00002861
2862 // Create and insert the replacement instruction...
2863 if (isa<MallocInst>(AI))
Chris Lattnerabb77c92004-03-19 06:08:10 +00002864 New = new MallocInst(NewTy, 0, AI.getName());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002865 else {
2866 assert(isa<AllocaInst>(AI) && "Unknown type of allocation inst!");
Chris Lattnerabb77c92004-03-19 06:08:10 +00002867 New = new AllocaInst(NewTy, 0, AI.getName());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002868 }
Chris Lattnerabb77c92004-03-19 06:08:10 +00002869
2870 InsertNewInstBefore(New, AI);
Chris Lattner1085bdf2002-11-04 16:18:53 +00002871
2872 // Scan to the end of the allocation instructions, to skip over a block of
2873 // allocas if possible...
2874 //
2875 BasicBlock::iterator It = New;
2876 while (isa<AllocationInst>(*It)) ++It;
2877
2878 // Now that I is pointing to the first non-allocation-inst in the block,
2879 // insert our getelementptr instruction...
2880 //
Chris Lattner69193f92004-04-05 01:30:19 +00002881 std::vector<Value*> Idx(2, Constant::getNullValue(Type::IntTy));
Chris Lattner1085bdf2002-11-04 16:18:53 +00002882 Value *V = new GetElementPtrInst(New, Idx, New->getName()+".sub", It);
2883
2884 // Now make everything use the getelementptr instead of the original
2885 // allocation.
Chris Lattnerabb77c92004-03-19 06:08:10 +00002886 return ReplaceInstUsesWith(AI, V);
Chris Lattner1085bdf2002-11-04 16:18:53 +00002887 }
Chris Lattnerabb77c92004-03-19 06:08:10 +00002888
2889 // If alloca'ing a zero byte object, replace the alloca with a null pointer.
2890 // Note that we only do this for alloca's, because malloc should allocate and
2891 // return a unique pointer, even for a zero byte allocation.
Chris Lattner49df6ce2004-07-02 22:55:47 +00002892 if (isa<AllocaInst>(AI) && AI.getAllocatedType()->isSized() &&
2893 TD->getTypeSize(AI.getAllocatedType()) == 0)
Chris Lattnerabb77c92004-03-19 06:08:10 +00002894 return ReplaceInstUsesWith(AI, Constant::getNullValue(AI.getType()));
2895
Chris Lattner1085bdf2002-11-04 16:18:53 +00002896 return 0;
2897}
2898
Chris Lattner8427bff2003-12-07 01:24:23 +00002899Instruction *InstCombiner::visitFreeInst(FreeInst &FI) {
2900 Value *Op = FI.getOperand(0);
2901
2902 // Change free <ty>* (cast <ty2>* X to <ty>*) into free <ty2>* X
2903 if (CastInst *CI = dyn_cast<CastInst>(Op))
2904 if (isa<PointerType>(CI->getOperand(0)->getType())) {
2905 FI.setOperand(0, CI->getOperand(0));
2906 return &FI;
2907 }
2908
Chris Lattnerf3a36602004-02-28 04:57:37 +00002909 // If we have 'free null' delete the instruction. This can happen in stl code
2910 // when lots of inlining happens.
Chris Lattner51ea1272004-02-28 05:22:00 +00002911 if (isa<ConstantPointerNull>(Op))
2912 return EraseInstFromFunction(FI);
Chris Lattnerf3a36602004-02-28 04:57:37 +00002913
Chris Lattner8427bff2003-12-07 01:24:23 +00002914 return 0;
2915}
2916
2917
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002918/// GetGEPGlobalInitializer - Given a constant, and a getelementptr
2919/// constantexpr, return the constant value being addressed by the constant
2920/// expression, or null if something is funny.
2921///
2922static Constant *GetGEPGlobalInitializer(Constant *C, ConstantExpr *CE) {
Chris Lattner69193f92004-04-05 01:30:19 +00002923 if (CE->getOperand(1) != Constant::getNullValue(CE->getOperand(1)->getType()))
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002924 return 0; // Do not allow stepping over the value!
2925
2926 // Loop over all of the operands, tracking down which value we are
2927 // addressing...
Chris Lattnered79d8a2004-05-27 17:30:27 +00002928 gep_type_iterator I = gep_type_begin(CE), E = gep_type_end(CE);
2929 for (++I; I != E; ++I)
2930 if (const StructType *STy = dyn_cast<StructType>(*I)) {
2931 ConstantUInt *CU = cast<ConstantUInt>(I.getOperand());
2932 assert(CU->getValue() < STy->getNumElements() &&
2933 "Struct index out of range!");
2934 if (ConstantStruct *CS = dyn_cast<ConstantStruct>(C)) {
2935 C = cast<Constant>(CS->getValues()[CU->getValue()]);
2936 } else if (isa<ConstantAggregateZero>(C)) {
2937 C = Constant::getNullValue(STy->getElementType(CU->getValue()));
2938 } else {
2939 return 0;
2940 }
2941 } else if (ConstantInt *CI = dyn_cast<ConstantInt>(I.getOperand())) {
2942 const ArrayType *ATy = cast<ArrayType>(*I);
2943 if ((uint64_t)CI->getRawValue() >= ATy->getNumElements()) return 0;
2944 if (ConstantArray *CA = dyn_cast<ConstantArray>(C))
2945 C = cast<Constant>(CA->getValues()[CI->getRawValue()]);
2946 else if (isa<ConstantAggregateZero>(C))
2947 C = Constant::getNullValue(ATy->getElementType());
2948 else
2949 return 0;
2950 } else {
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002951 return 0;
Chris Lattnered79d8a2004-05-27 17:30:27 +00002952 }
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002953 return C;
2954}
2955
Chris Lattner35e24772004-07-13 01:49:43 +00002956static Instruction *InstCombineLoadCast(InstCombiner &IC, LoadInst &LI) {
2957 User *CI = cast<User>(LI.getOperand(0));
2958
2959 const Type *DestPTy = cast<PointerType>(CI->getType())->getElementType();
2960 if (const PointerType *SrcTy =
2961 dyn_cast<PointerType>(CI->getOperand(0)->getType())) {
2962 const Type *SrcPTy = SrcTy->getElementType();
2963 if (SrcPTy->isSized() && DestPTy->isSized() &&
2964 IC.getTargetData().getTypeSize(SrcPTy) ==
2965 IC.getTargetData().getTypeSize(DestPTy) &&
2966 (SrcPTy->isInteger() || isa<PointerType>(SrcPTy)) &&
2967 (DestPTy->isInteger() || isa<PointerType>(DestPTy))) {
2968 // Okay, we are casting from one integer or pointer type to another of
2969 // the same size. Instead of casting the pointer before the load, cast
2970 // the result of the loaded value.
2971 Value *NewLoad = IC.InsertNewInstBefore(new LoadInst(CI->getOperand(0),
2972 CI->getName()), LI);
2973 // Now cast the result of the load.
2974 return new CastInst(NewLoad, LI.getType());
2975 }
2976 }
2977 return 0;
2978}
2979
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002980Instruction *InstCombiner::visitLoadInst(LoadInst &LI) {
2981 Value *Op = LI.getOperand(0);
Chris Lattner7e8af382004-01-12 04:13:56 +00002982 if (LI.isVolatile()) return 0;
2983
Chris Lattner6679e462004-04-14 03:28:36 +00002984 if (Constant *C = dyn_cast<Constant>(Op))
2985 if (C->isNullValue()) // load null -> 0
2986 return ReplaceInstUsesWith(LI, Constant::getNullValue(LI.getType()));
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002987
2988 // Instcombine load (constant global) into the value loaded...
2989 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op))
Chris Lattnerbdb0ce02003-07-22 21:46:59 +00002990 if (GV->isConstant() && !GV->isExternal())
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002991 return ReplaceInstUsesWith(LI, GV->getInitializer());
2992
2993 // Instcombine load (constantexpr_GEP global, 0, ...) into the value loaded...
2994 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Op))
Chris Lattner35e24772004-07-13 01:49:43 +00002995 if (CE->getOpcode() == Instruction::GetElementPtr) {
Reid Spencer87436872004-07-18 00:38:32 +00002996 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(CE->getOperand(0)))
2997 if (GV->isConstant() && !GV->isExternal())
2998 if (Constant *V = GetGEPGlobalInitializer(GV->getInitializer(), CE))
2999 return ReplaceInstUsesWith(LI, V);
Chris Lattner35e24772004-07-13 01:49:43 +00003000 } else if (CE->getOpcode() == Instruction::Cast) {
3001 if (Instruction *Res = InstCombineLoadCast(*this, LI))
3002 return Res;
3003 }
Chris Lattnere228ee52004-04-08 20:39:49 +00003004
3005 // load (cast X) --> cast (load X) iff safe
Chris Lattner35e24772004-07-13 01:49:43 +00003006 if (CastInst *CI = dyn_cast<CastInst>(Op))
3007 if (Instruction *Res = InstCombineLoadCast(*this, LI))
3008 return Res;
Chris Lattnere228ee52004-04-08 20:39:49 +00003009
Chris Lattner0f1d8a32003-06-26 05:06:25 +00003010 return 0;
3011}
3012
3013
Chris Lattner9eef8a72003-06-04 04:46:00 +00003014Instruction *InstCombiner::visitBranchInst(BranchInst &BI) {
3015 // Change br (not X), label True, label False to: br X, label False, True
Chris Lattner4f7acca2004-02-27 06:27:46 +00003016 if (BI.isConditional() && !isa<Constant>(BI.getCondition())) {
Chris Lattnere967b342003-06-04 05:10:11 +00003017 if (Value *V = dyn_castNotVal(BI.getCondition())) {
3018 BasicBlock *TrueDest = BI.getSuccessor(0);
3019 BasicBlock *FalseDest = BI.getSuccessor(1);
3020 // Swap Destinations and condition...
3021 BI.setCondition(V);
3022 BI.setSuccessor(0, FalseDest);
3023 BI.setSuccessor(1, TrueDest);
3024 return &BI;
Chris Lattner4f7acca2004-02-27 06:27:46 +00003025 } else if (SetCondInst *I = dyn_cast<SetCondInst>(BI.getCondition())) {
3026 // Cannonicalize setne -> seteq
3027 if ((I->getOpcode() == Instruction::SetNE ||
3028 I->getOpcode() == Instruction::SetLE ||
3029 I->getOpcode() == Instruction::SetGE) && I->hasOneUse()) {
3030 std::string Name = I->getName(); I->setName("");
3031 Instruction::BinaryOps NewOpcode =
3032 SetCondInst::getInverseCondition(I->getOpcode());
3033 Value *NewSCC = BinaryOperator::create(NewOpcode, I->getOperand(0),
3034 I->getOperand(1), Name, I);
3035 BasicBlock *TrueDest = BI.getSuccessor(0);
3036 BasicBlock *FalseDest = BI.getSuccessor(1);
3037 // Swap Destinations and condition...
3038 BI.setCondition(NewSCC);
3039 BI.setSuccessor(0, FalseDest);
3040 BI.setSuccessor(1, TrueDest);
3041 removeFromWorkList(I);
3042 I->getParent()->getInstList().erase(I);
3043 WorkList.push_back(cast<Instruction>(NewSCC));
3044 return &BI;
3045 }
Chris Lattnere967b342003-06-04 05:10:11 +00003046 }
Chris Lattner4f7acca2004-02-27 06:27:46 +00003047 }
Chris Lattner9eef8a72003-06-04 04:46:00 +00003048 return 0;
3049}
Chris Lattner1085bdf2002-11-04 16:18:53 +00003050
Chris Lattner4c9c20a2004-07-03 00:26:11 +00003051Instruction *InstCombiner::visitSwitchInst(SwitchInst &SI) {
3052 Value *Cond = SI.getCondition();
3053 if (Instruction *I = dyn_cast<Instruction>(Cond)) {
3054 if (I->getOpcode() == Instruction::Add)
3055 if (ConstantInt *AddRHS = dyn_cast<ConstantInt>(I->getOperand(1))) {
3056 // change 'switch (X+4) case 1:' into 'switch (X) case -3'
3057 for (unsigned i = 2, e = SI.getNumOperands(); i != e; i += 2)
3058 SI.setOperand(i, ConstantExpr::getSub(cast<Constant>(SI.getOperand(i)),
3059 AddRHS));
3060 SI.setOperand(0, I->getOperand(0));
3061 WorkList.push_back(I);
3062 return &SI;
3063 }
3064 }
3065 return 0;
3066}
3067
Chris Lattnerca081252001-12-14 16:52:21 +00003068
Chris Lattner99f48c62002-09-02 04:59:56 +00003069void InstCombiner::removeFromWorkList(Instruction *I) {
3070 WorkList.erase(std::remove(WorkList.begin(), WorkList.end(), I),
3071 WorkList.end());
3072}
3073
Chris Lattner113f4f42002-06-25 16:13:24 +00003074bool InstCombiner::runOnFunction(Function &F) {
Chris Lattner260ab202002-04-18 17:39:14 +00003075 bool Changed = false;
Chris Lattnerf4ad1652003-11-02 05:57:39 +00003076 TD = &getAnalysis<TargetData>();
Chris Lattnerca081252001-12-14 16:52:21 +00003077
Chris Lattnerb643a9e2004-05-01 23:19:52 +00003078 for (inst_iterator i = inst_begin(F), e = inst_end(F); i != e; ++i)
3079 WorkList.push_back(&*i);
Chris Lattner2d3a7a62004-04-27 15:13:33 +00003080
Chris Lattnerca081252001-12-14 16:52:21 +00003081
3082 while (!WorkList.empty()) {
3083 Instruction *I = WorkList.back(); // Get an instruction from the worklist
3084 WorkList.pop_back();
3085
Misha Brukman632df282002-10-29 23:06:16 +00003086 // Check to see if we can DCE or ConstantPropagate the instruction...
Chris Lattner99f48c62002-09-02 04:59:56 +00003087 // Check to see if we can DIE the instruction...
3088 if (isInstructionTriviallyDead(I)) {
3089 // Add operands to the worklist...
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003090 if (I->getNumOperands() < 4)
Chris Lattner51ea1272004-02-28 05:22:00 +00003091 AddUsesToWorkList(*I);
Chris Lattner99f48c62002-09-02 04:59:56 +00003092 ++NumDeadInst;
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003093
3094 I->getParent()->getInstList().erase(I);
3095 removeFromWorkList(I);
3096 continue;
3097 }
Chris Lattner99f48c62002-09-02 04:59:56 +00003098
Misha Brukman632df282002-10-29 23:06:16 +00003099 // Instruction isn't dead, see if we can constant propagate it...
Chris Lattner99f48c62002-09-02 04:59:56 +00003100 if (Constant *C = ConstantFoldInstruction(I)) {
3101 // Add operands to the worklist...
Chris Lattner51ea1272004-02-28 05:22:00 +00003102 AddUsesToWorkList(*I);
Chris Lattnerc6509f42002-12-05 22:41:53 +00003103 ReplaceInstUsesWith(*I, C);
3104
Chris Lattner99f48c62002-09-02 04:59:56 +00003105 ++NumConstProp;
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003106 I->getParent()->getInstList().erase(I);
Chris Lattner800aaaf2003-10-07 15:17:02 +00003107 removeFromWorkList(I);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003108 continue;
Chris Lattner99f48c62002-09-02 04:59:56 +00003109 }
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003110
Chris Lattnerca081252001-12-14 16:52:21 +00003111 // Now that we have an instruction, try combining it to simplify it...
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003112 if (Instruction *Result = visit(*I)) {
Chris Lattner0b18c1d2002-05-10 15:38:35 +00003113 ++NumCombined;
Chris Lattner260ab202002-04-18 17:39:14 +00003114 // Should we replace the old instruction with a new one?
Chris Lattner053c0932002-05-14 15:24:07 +00003115 if (Result != I) {
Chris Lattner7d2a5392004-03-13 23:54:27 +00003116 DEBUG(std::cerr << "IC: Old = " << *I
3117 << " New = " << *Result);
3118
Chris Lattner396dbfe2004-06-09 05:08:07 +00003119 // Everything uses the new instruction now.
3120 I->replaceAllUsesWith(Result);
3121
3122 // Push the new instruction and any users onto the worklist.
3123 WorkList.push_back(Result);
3124 AddUsersToWorkList(*Result);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003125
3126 // Move the name to the new instruction first...
3127 std::string OldName = I->getName(); I->setName("");
Chris Lattner950fc782003-10-07 22:58:41 +00003128 Result->setName(OldName);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003129
3130 // Insert the new instruction into the basic block...
3131 BasicBlock *InstParent = I->getParent();
3132 InstParent->getInstList().insert(I, Result);
3133
Chris Lattner63d75af2004-05-01 23:27:23 +00003134 // Make sure that we reprocess all operands now that we reduced their
3135 // use counts.
Chris Lattnerb643a9e2004-05-01 23:19:52 +00003136 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
3137 if (Instruction *OpI = dyn_cast<Instruction>(I->getOperand(i)))
3138 WorkList.push_back(OpI);
3139
Chris Lattner396dbfe2004-06-09 05:08:07 +00003140 // Instructions can end up on the worklist more than once. Make sure
3141 // we do not process an instruction that has been deleted.
3142 removeFromWorkList(I);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00003143
3144 // Erase the old instruction.
3145 InstParent->getInstList().erase(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00003146 } else {
Chris Lattner7d2a5392004-03-13 23:54:27 +00003147 DEBUG(std::cerr << "IC: MOD = " << *I);
3148
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003149 // If the instruction was modified, it's possible that it is now dead.
3150 // if so, remove it.
Chris Lattner63d75af2004-05-01 23:27:23 +00003151 if (isInstructionTriviallyDead(I)) {
3152 // Make sure we process all operands now that we are reducing their
3153 // use counts.
3154 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
3155 if (Instruction *OpI = dyn_cast<Instruction>(I->getOperand(i)))
3156 WorkList.push_back(OpI);
3157
3158 // Instructions may end up in the worklist more than once. Erase all
3159 // occurrances of this instruction.
Chris Lattner99f48c62002-09-02 04:59:56 +00003160 removeFromWorkList(I);
Chris Lattner63d75af2004-05-01 23:27:23 +00003161 I->getParent()->getInstList().erase(I);
Chris Lattner396dbfe2004-06-09 05:08:07 +00003162 } else {
3163 WorkList.push_back(Result);
3164 AddUsersToWorkList(*Result);
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003165 }
Chris Lattner053c0932002-05-14 15:24:07 +00003166 }
Chris Lattner260ab202002-04-18 17:39:14 +00003167 Changed = true;
Chris Lattnerca081252001-12-14 16:52:21 +00003168 }
3169 }
3170
Chris Lattner260ab202002-04-18 17:39:14 +00003171 return Changed;
Chris Lattner04805fa2002-02-26 21:46:54 +00003172}
3173
Chris Lattner8427bff2003-12-07 01:24:23 +00003174Pass *llvm::createInstructionCombiningPass() {
Chris Lattner260ab202002-04-18 17:39:14 +00003175 return new InstCombiner();
Chris Lattner04805fa2002-02-26 21:46:54 +00003176}
Brian Gaeke960707c2003-11-11 22:41:34 +00003177