blob: 52b861da15665b347a9a5a040d20314f7112afb6 [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:
15// %Y = add int 1, %X
16// %Z = add int 1, %Y
17// into:
18// %Z = add int 2, %X
19//
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 Lattner260ab202002-04-18 17:39:14 +0000125
126 // visitInstruction - Specify what to return for unhandled instructions...
Chris Lattner113f4f42002-06-25 16:13:24 +0000127 Instruction *visitInstruction(Instruction &I) { return 0; }
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000128
Chris Lattner970c33a2003-06-19 17:00:31 +0000129 private:
Chris Lattneraec3d942003-10-07 22:32:43 +0000130 Instruction *visitCallSite(CallSite CS);
Chris Lattner970c33a2003-06-19 17:00:31 +0000131 bool transformConstExprCastCall(CallSite CS);
132
Chris Lattner69193f92004-04-05 01:30:19 +0000133 public:
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000134 // InsertNewInstBefore - insert an instruction New before instruction Old
135 // in the program. Add the new instruction to the worklist.
136 //
Chris Lattnere79e8542004-02-23 06:38:22 +0000137 Value *InsertNewInstBefore(Instruction *New, Instruction &Old) {
Chris Lattner65217ff2002-08-23 18:32:43 +0000138 assert(New && New->getParent() == 0 &&
139 "New instruction already inserted into a basic block!");
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000140 BasicBlock *BB = Old.getParent();
141 BB->getInstList().insert(&Old, New); // Insert inst
142 WorkList.push_back(New); // Add to worklist
Chris Lattnere79e8542004-02-23 06:38:22 +0000143 return New;
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000144 }
145
146 // ReplaceInstUsesWith - This method is to be used when an instruction is
147 // found to be dead, replacable with another preexisting expression. Here
148 // we add all uses of I to the worklist, replace all uses of I with the new
149 // value, then return I, so that the inst combiner will know that I was
150 // modified.
151 //
152 Instruction *ReplaceInstUsesWith(Instruction &I, Value *V) {
Chris Lattner51ea1272004-02-28 05:22:00 +0000153 AddUsersToWorkList(I); // Add all modified instrs to worklist
Chris Lattner8953b902004-04-05 02:10:19 +0000154 if (&I != V) {
155 I.replaceAllUsesWith(V);
156 return &I;
157 } else {
158 // If we are replacing the instruction with itself, this must be in a
159 // segment of unreachable code, so just clobber the instruction.
160 I.replaceAllUsesWith(Constant::getNullValue(I.getType()));
161 return &I;
162 }
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000163 }
Chris Lattner51ea1272004-02-28 05:22:00 +0000164
165 // EraseInstFromFunction - When dealing with an instruction that has side
166 // effects or produces a void value, we can't rely on DCE to delete the
167 // instruction. Instead, visit methods should return the value returned by
168 // this function.
169 Instruction *EraseInstFromFunction(Instruction &I) {
170 assert(I.use_empty() && "Cannot erase instruction that is used!");
171 AddUsesToWorkList(I);
172 removeFromWorkList(&I);
173 I.getParent()->getInstList().erase(&I);
174 return 0; // Don't do anything with FI
175 }
176
177
Chris Lattner3ac7c262003-08-13 20:16:26 +0000178 private:
Chris Lattnerdfae8be2003-07-24 17:35:25 +0000179 /// InsertOperandCastBefore - This inserts a cast of V to DestTy before the
180 /// InsertBefore instruction. This is specialized a bit to avoid inserting
181 /// casts that are known to not do anything...
182 ///
183 Value *InsertOperandCastBefore(Value *V, const Type *DestTy,
184 Instruction *InsertBefore);
185
Chris Lattner7fb29e12003-03-11 00:12:48 +0000186 // SimplifyCommutative - This performs a few simplifications for commutative
187 // operators...
188 bool SimplifyCommutative(BinaryOperator &I);
Chris Lattnerba1cb382003-09-19 17:17:26 +0000189
190 Instruction *OptAndOp(Instruction *Op, ConstantIntegral *OpRHS,
191 ConstantIntegral *AndRHS, BinaryOperator &TheAnd);
Chris Lattner260ab202002-04-18 17:39:14 +0000192 };
Chris Lattnerb28b6802002-07-23 18:06:35 +0000193
Chris Lattnerc8b70922002-07-26 21:12:46 +0000194 RegisterOpt<InstCombiner> X("instcombine", "Combine redundant instructions");
Chris Lattner260ab202002-04-18 17:39:14 +0000195}
196
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000197// getComplexity: Assign a complexity or rank value to LLVM Values...
198// 0 -> Constant, 1 -> Other, 2 -> Argument, 2 -> Unary, 3 -> OtherInst
199static unsigned getComplexity(Value *V) {
200 if (isa<Instruction>(V)) {
201 if (BinaryOperator::isNeg(V) || BinaryOperator::isNot(V))
202 return 2;
203 return 3;
204 }
205 if (isa<Argument>(V)) return 2;
206 return isa<Constant>(V) ? 0 : 1;
207}
Chris Lattner260ab202002-04-18 17:39:14 +0000208
Chris Lattner7fb29e12003-03-11 00:12:48 +0000209// isOnlyUse - Return true if this instruction will be deleted if we stop using
210// it.
211static bool isOnlyUse(Value *V) {
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000212 return V->hasOneUse() || isa<Constant>(V);
Chris Lattner7fb29e12003-03-11 00:12:48 +0000213}
214
Chris Lattnere79e8542004-02-23 06:38:22 +0000215// getSignedIntegralType - Given an unsigned integral type, return the signed
216// version of it that has the same size.
217static const Type *getSignedIntegralType(const Type *Ty) {
218 switch (Ty->getPrimitiveID()) {
219 default: assert(0 && "Invalid unsigned integer type!"); abort();
220 case Type::UByteTyID: return Type::SByteTy;
221 case Type::UShortTyID: return Type::ShortTy;
222 case Type::UIntTyID: return Type::IntTy;
223 case Type::ULongTyID: return Type::LongTy;
224 }
225}
226
Chris Lattner92295c52004-03-12 23:53:13 +0000227// getUnsignedIntegralType - Given an signed integral type, return the unsigned
228// version of it that has the same size.
229static const Type *getUnsignedIntegralType(const Type *Ty) {
230 switch (Ty->getPrimitiveID()) {
231 default: assert(0 && "Invalid signed integer type!"); abort();
232 case Type::SByteTyID: return Type::UByteTy;
233 case Type::ShortTyID: return Type::UShortTy;
234 case Type::IntTyID: return Type::UIntTy;
235 case Type::LongTyID: return Type::ULongTy;
236 }
237}
238
Chris Lattnere79e8542004-02-23 06:38:22 +0000239// getPromotedType - Return the specified type promoted as it would be to pass
240// though a va_arg area...
241static const Type *getPromotedType(const Type *Ty) {
242 switch (Ty->getPrimitiveID()) {
243 case Type::SByteTyID:
244 case Type::ShortTyID: return Type::IntTy;
245 case Type::UByteTyID:
246 case Type::UShortTyID: return Type::UIntTy;
247 case Type::FloatTyID: return Type::DoubleTy;
248 default: return Ty;
249 }
250}
251
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000252// SimplifyCommutative - This performs a few simplifications for commutative
253// operators:
Chris Lattner260ab202002-04-18 17:39:14 +0000254//
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000255// 1. Order operands such that they are listed from right (least complex) to
256// left (most complex). This puts constants before unary operators before
257// binary operators.
258//
Chris Lattner7fb29e12003-03-11 00:12:48 +0000259// 2. Transform: (op (op V, C1), C2) ==> (op V, (op C1, C2))
260// 3. Transform: (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2))
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000261//
Chris Lattner7fb29e12003-03-11 00:12:48 +0000262bool InstCombiner::SimplifyCommutative(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000263 bool Changed = false;
264 if (getComplexity(I.getOperand(0)) < getComplexity(I.getOperand(1)))
265 Changed = !I.swapOperands();
266
267 if (!I.isAssociative()) return Changed;
268 Instruction::BinaryOps Opcode = I.getOpcode();
Chris Lattner7fb29e12003-03-11 00:12:48 +0000269 if (BinaryOperator *Op = dyn_cast<BinaryOperator>(I.getOperand(0)))
270 if (Op->getOpcode() == Opcode && isa<Constant>(Op->getOperand(1))) {
271 if (isa<Constant>(I.getOperand(1))) {
Chris Lattner34428442003-05-27 16:40:51 +0000272 Constant *Folded = ConstantExpr::get(I.getOpcode(),
273 cast<Constant>(I.getOperand(1)),
274 cast<Constant>(Op->getOperand(1)));
Chris Lattner7fb29e12003-03-11 00:12:48 +0000275 I.setOperand(0, Op->getOperand(0));
276 I.setOperand(1, Folded);
277 return true;
278 } else if (BinaryOperator *Op1=dyn_cast<BinaryOperator>(I.getOperand(1)))
279 if (Op1->getOpcode() == Opcode && isa<Constant>(Op1->getOperand(1)) &&
280 isOnlyUse(Op) && isOnlyUse(Op1)) {
281 Constant *C1 = cast<Constant>(Op->getOperand(1));
282 Constant *C2 = cast<Constant>(Op1->getOperand(1));
283
284 // Fold (op (op V1, C1), (op V2, C2)) ==> (op (op V1, V2), (op C1,C2))
Chris Lattner34428442003-05-27 16:40:51 +0000285 Constant *Folded = ConstantExpr::get(I.getOpcode(), C1, C2);
Chris Lattner7fb29e12003-03-11 00:12:48 +0000286 Instruction *New = BinaryOperator::create(Opcode, Op->getOperand(0),
287 Op1->getOperand(0),
288 Op1->getName(), &I);
289 WorkList.push_back(New);
290 I.setOperand(0, New);
291 I.setOperand(1, Folded);
292 return true;
293 }
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000294 }
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000295 return Changed;
Chris Lattner260ab202002-04-18 17:39:14 +0000296}
Chris Lattnerca081252001-12-14 16:52:21 +0000297
Chris Lattnerbb74e222003-03-10 23:06:50 +0000298// dyn_castNegVal - Given a 'sub' instruction, return the RHS of the instruction
299// if the LHS is a constant zero (which is the 'negate' form).
Chris Lattner9fa53de2002-05-06 16:49:18 +0000300//
Chris Lattnerbb74e222003-03-10 23:06:50 +0000301static inline Value *dyn_castNegVal(Value *V) {
302 if (BinaryOperator::isNeg(V))
303 return BinaryOperator::getNegArgument(cast<BinaryOperator>(V));
304
Chris Lattner9244df62003-04-30 22:19:10 +0000305 // Constants can be considered to be negated values if they can be folded...
306 if (Constant *C = dyn_cast<Constant>(V))
Chris Lattner34428442003-05-27 16:40:51 +0000307 return ConstantExpr::get(Instruction::Sub,
308 Constant::getNullValue(V->getType()), C);
Chris Lattnerbb74e222003-03-10 23:06:50 +0000309 return 0;
Chris Lattner9fa53de2002-05-06 16:49:18 +0000310}
311
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000312static Constant *NotConstant(Constant *C) {
313 return ConstantExpr::get(Instruction::Xor, C,
314 ConstantIntegral::getAllOnesValue(C->getType()));
315}
316
Chris Lattnerbb74e222003-03-10 23:06:50 +0000317static inline Value *dyn_castNotVal(Value *V) {
318 if (BinaryOperator::isNot(V))
319 return BinaryOperator::getNotArgument(cast<BinaryOperator>(V));
320
321 // Constants can be considered to be not'ed values...
Chris Lattnerdd65d862003-04-30 22:34:06 +0000322 if (ConstantIntegral *C = dyn_cast<ConstantIntegral>(V))
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000323 return NotConstant(C);
Chris Lattnerbb74e222003-03-10 23:06:50 +0000324 return 0;
325}
326
Chris Lattner7fb29e12003-03-11 00:12:48 +0000327// dyn_castFoldableMul - If this value is a multiply that can be folded into
328// other computations (because it has a constant operand), return the
329// non-constant operand of the multiply.
330//
331static inline Value *dyn_castFoldableMul(Value *V) {
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000332 if (V->hasOneUse() && V->getType()->isInteger())
Chris Lattner7fb29e12003-03-11 00:12:48 +0000333 if (Instruction *I = dyn_cast<Instruction>(V))
334 if (I->getOpcode() == Instruction::Mul)
335 if (isa<Constant>(I->getOperand(1)))
336 return I->getOperand(0);
337 return 0;
Chris Lattner3082c5a2003-02-18 19:28:33 +0000338}
Chris Lattner31ae8632002-08-14 17:51:49 +0000339
Chris Lattner7fb29e12003-03-11 00:12:48 +0000340// dyn_castMaskingAnd - If this value is an And instruction masking a value with
341// a constant, return the constant being anded with.
342//
Chris Lattner01d56392003-08-12 19:17:27 +0000343template<class ValueType>
344static inline Constant *dyn_castMaskingAnd(ValueType *V) {
Chris Lattner7fb29e12003-03-11 00:12:48 +0000345 if (Instruction *I = dyn_cast<Instruction>(V))
346 if (I->getOpcode() == Instruction::And)
347 return dyn_cast<Constant>(I->getOperand(1));
348
349 // If this is a constant, it acts just like we were masking with it.
350 return dyn_cast<Constant>(V);
351}
Chris Lattner3082c5a2003-02-18 19:28:33 +0000352
353// Log2 - Calculate the log base 2 for the specified value if it is exactly a
354// power of 2.
355static unsigned Log2(uint64_t Val) {
356 assert(Val > 1 && "Values 0 and 1 should be handled elsewhere!");
357 unsigned Count = 0;
358 while (Val != 1) {
359 if (Val & 1) return 0; // Multiple bits set?
360 Val >>= 1;
361 ++Count;
362 }
363 return Count;
Chris Lattner31ae8632002-08-14 17:51:49 +0000364}
365
Chris Lattnerb8b97502003-08-13 19:01:45 +0000366
367/// AssociativeOpt - Perform an optimization on an associative operator. This
368/// function is designed to check a chain of associative operators for a
369/// potential to apply a certain optimization. Since the optimization may be
370/// applicable if the expression was reassociated, this checks the chain, then
371/// reassociates the expression as necessary to expose the optimization
372/// opportunity. This makes use of a special Functor, which must define
373/// 'shouldApply' and 'apply' methods.
374///
375template<typename Functor>
376Instruction *AssociativeOpt(BinaryOperator &Root, const Functor &F) {
377 unsigned Opcode = Root.getOpcode();
378 Value *LHS = Root.getOperand(0);
379
380 // Quick check, see if the immediate LHS matches...
381 if (F.shouldApply(LHS))
382 return F.apply(Root);
383
384 // Otherwise, if the LHS is not of the same opcode as the root, return.
385 Instruction *LHSI = dyn_cast<Instruction>(LHS);
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000386 while (LHSI && LHSI->getOpcode() == Opcode && LHSI->hasOneUse()) {
Chris Lattnerb8b97502003-08-13 19:01:45 +0000387 // Should we apply this transform to the RHS?
388 bool ShouldApply = F.shouldApply(LHSI->getOperand(1));
389
390 // If not to the RHS, check to see if we should apply to the LHS...
391 if (!ShouldApply && F.shouldApply(LHSI->getOperand(0))) {
392 cast<BinaryOperator>(LHSI)->swapOperands(); // Make the LHS the RHS
393 ShouldApply = true;
394 }
395
396 // If the functor wants to apply the optimization to the RHS of LHSI,
397 // reassociate the expression from ((? op A) op B) to (? op (A op B))
398 if (ShouldApply) {
399 BasicBlock *BB = Root.getParent();
Chris Lattnerb8b97502003-08-13 19:01:45 +0000400
401 // Now all of the instructions are in the current basic block, go ahead
402 // and perform the reassociation.
403 Instruction *TmpLHSI = cast<Instruction>(Root.getOperand(0));
404
405 // First move the selected RHS to the LHS of the root...
406 Root.setOperand(0, LHSI->getOperand(1));
407
408 // Make what used to be the LHS of the root be the user of the root...
409 Value *ExtraOperand = TmpLHSI->getOperand(1);
Chris Lattner284d3b02004-04-16 18:08:07 +0000410 if (&Root == TmpLHSI) {
Chris Lattner8953b902004-04-05 02:10:19 +0000411 Root.replaceAllUsesWith(Constant::getNullValue(TmpLHSI->getType()));
412 return 0;
413 }
Chris Lattner284d3b02004-04-16 18:08:07 +0000414 Root.replaceAllUsesWith(TmpLHSI); // Users now use TmpLHSI
Chris Lattnerb8b97502003-08-13 19:01:45 +0000415 TmpLHSI->setOperand(1, &Root); // TmpLHSI now uses the root
Chris Lattner284d3b02004-04-16 18:08:07 +0000416 TmpLHSI->getParent()->getInstList().remove(TmpLHSI);
417 BasicBlock::iterator ARI = &Root; ++ARI;
418 BB->getInstList().insert(ARI, TmpLHSI); // Move TmpLHSI to after Root
419 ARI = Root;
Chris Lattnerb8b97502003-08-13 19:01:45 +0000420
421 // Now propagate the ExtraOperand down the chain of instructions until we
422 // get to LHSI.
423 while (TmpLHSI != LHSI) {
424 Instruction *NextLHSI = cast<Instruction>(TmpLHSI->getOperand(0));
Chris Lattner284d3b02004-04-16 18:08:07 +0000425 // Move the instruction to immediately before the chain we are
426 // constructing to avoid breaking dominance properties.
427 NextLHSI->getParent()->getInstList().remove(NextLHSI);
428 BB->getInstList().insert(ARI, NextLHSI);
429 ARI = NextLHSI;
430
Chris Lattnerb8b97502003-08-13 19:01:45 +0000431 Value *NextOp = NextLHSI->getOperand(1);
432 NextLHSI->setOperand(1, ExtraOperand);
433 TmpLHSI = NextLHSI;
434 ExtraOperand = NextOp;
435 }
436
437 // Now that the instructions are reassociated, have the functor perform
438 // the transformation...
439 return F.apply(Root);
440 }
441
442 LHSI = dyn_cast<Instruction>(LHSI->getOperand(0));
443 }
444 return 0;
445}
446
447
448// AddRHS - Implements: X + X --> X << 1
449struct AddRHS {
450 Value *RHS;
451 AddRHS(Value *rhs) : RHS(rhs) {}
452 bool shouldApply(Value *LHS) const { return LHS == RHS; }
453 Instruction *apply(BinaryOperator &Add) const {
454 return new ShiftInst(Instruction::Shl, Add.getOperand(0),
455 ConstantInt::get(Type::UByteTy, 1));
456 }
457};
458
459// AddMaskingAnd - Implements (A & C1)+(B & C2) --> (A & C1)|(B & C2)
460// iff C1&C2 == 0
461struct AddMaskingAnd {
462 Constant *C2;
463 AddMaskingAnd(Constant *c) : C2(c) {}
464 bool shouldApply(Value *LHS) const {
465 if (Constant *C1 = dyn_castMaskingAnd(LHS))
466 return ConstantExpr::get(Instruction::And, C1, C2)->isNullValue();
467 return false;
468 }
469 Instruction *apply(BinaryOperator &Add) const {
470 return BinaryOperator::create(Instruction::Or, Add.getOperand(0),
471 Add.getOperand(1));
472 }
473};
474
Chris Lattner183b3362004-04-09 19:05:30 +0000475static Value *FoldOperationIntoSelectOperand(Instruction &BI, Value *SO,
476 InstCombiner *IC) {
477 // Figure out if the constant is the left or the right argument.
478 bool ConstIsRHS = isa<Constant>(BI.getOperand(1));
479 Constant *ConstOperand = cast<Constant>(BI.getOperand(ConstIsRHS));
Chris Lattnerb8b97502003-08-13 19:01:45 +0000480
Chris Lattner183b3362004-04-09 19:05:30 +0000481 if (Constant *SOC = dyn_cast<Constant>(SO)) {
482 if (ConstIsRHS)
483 return ConstantExpr::get(BI.getOpcode(), SOC, ConstOperand);
484 return ConstantExpr::get(BI.getOpcode(), ConstOperand, SOC);
485 }
486
487 Value *Op0 = SO, *Op1 = ConstOperand;
488 if (!ConstIsRHS)
489 std::swap(Op0, Op1);
490 Instruction *New;
491 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(&BI))
492 New = BinaryOperator::create(BO->getOpcode(), Op0, Op1);
493 else if (ShiftInst *SI = dyn_cast<ShiftInst>(&BI))
494 New = new ShiftInst(SI->getOpcode(), Op0, Op1);
Chris Lattnerf9d96652004-04-10 19:15:56 +0000495 else {
Chris Lattner183b3362004-04-09 19:05:30 +0000496 assert(0 && "Unknown binary instruction type!");
Chris Lattnerf9d96652004-04-10 19:15:56 +0000497 abort();
498 }
Chris Lattner183b3362004-04-09 19:05:30 +0000499 return IC->InsertNewInstBefore(New, BI);
500}
501
502// FoldBinOpIntoSelect - Given an instruction with a select as one operand and a
503// constant as the other operand, try to fold the binary operator into the
504// select arguments.
505static Instruction *FoldBinOpIntoSelect(Instruction &BI, SelectInst *SI,
506 InstCombiner *IC) {
507 // Don't modify shared select instructions
508 if (!SI->hasOneUse()) return 0;
509 Value *TV = SI->getOperand(1);
510 Value *FV = SI->getOperand(2);
511
512 if (isa<Constant>(TV) || isa<Constant>(FV)) {
513 Value *SelectTrueVal = FoldOperationIntoSelectOperand(BI, TV, IC);
514 Value *SelectFalseVal = FoldOperationIntoSelectOperand(BI, FV, IC);
515
516 return new SelectInst(SI->getCondition(), SelectTrueVal,
517 SelectFalseVal);
518 }
519 return 0;
520}
Chris Lattnerb8b97502003-08-13 19:01:45 +0000521
Chris Lattner113f4f42002-06-25 16:13:24 +0000522Instruction *InstCombiner::visitAdd(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000523 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +0000524 Value *LHS = I.getOperand(0), *RHS = I.getOperand(1);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000525
Chris Lattnercf4a9962004-04-10 22:01:55 +0000526 if (Constant *RHSC = dyn_cast<Constant>(RHS)) {
527 // X + 0 --> X
528 if (!I.getType()->isFloatingPoint() && // -0 + +0 = +0, so it's not a noop
529 RHSC->isNullValue())
530 return ReplaceInstUsesWith(I, LHS);
531
532 // X + (signbit) --> X ^ signbit
533 if (ConstantInt *CI = dyn_cast<ConstantInt>(RHSC)) {
534 unsigned NumBits = CI->getType()->getPrimitiveSize()*8;
535 uint64_t Val = CI->getRawValue() & (1ULL << NumBits)-1;
536 if (Val == (1ULL << NumBits-1))
537 return BinaryOperator::create(Instruction::Xor, LHS, RHS);
538 }
539 }
Chris Lattner9fa53de2002-05-06 16:49:18 +0000540
Chris Lattnerb8b97502003-08-13 19:01:45 +0000541 // X + X --> X << 1
542 if (I.getType()->isInteger())
543 if (Instruction *Result = AssociativeOpt(I, AddRHS(RHS))) return Result;
Chris Lattnerede3fe02003-08-13 04:18:28 +0000544
Chris Lattner147e9752002-05-08 22:46:53 +0000545 // -A + B --> B - A
Chris Lattnerbb74e222003-03-10 23:06:50 +0000546 if (Value *V = dyn_castNegVal(LHS))
Chris Lattner147e9752002-05-08 22:46:53 +0000547 return BinaryOperator::create(Instruction::Sub, RHS, V);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000548
549 // A + -B --> A - B
Chris Lattnerbb74e222003-03-10 23:06:50 +0000550 if (!isa<Constant>(RHS))
551 if (Value *V = dyn_castNegVal(RHS))
552 return BinaryOperator::create(Instruction::Sub, LHS, V);
Chris Lattner260ab202002-04-18 17:39:14 +0000553
Chris Lattner57c8d992003-02-18 19:57:07 +0000554 // X*C + X --> X * (C+1)
555 if (dyn_castFoldableMul(LHS) == RHS) {
Chris Lattner34428442003-05-27 16:40:51 +0000556 Constant *CP1 =
557 ConstantExpr::get(Instruction::Add,
558 cast<Constant>(cast<Instruction>(LHS)->getOperand(1)),
559 ConstantInt::get(I.getType(), 1));
Chris Lattner57c8d992003-02-18 19:57:07 +0000560 return BinaryOperator::create(Instruction::Mul, RHS, CP1);
561 }
562
563 // X + X*C --> X * (C+1)
564 if (dyn_castFoldableMul(RHS) == LHS) {
Chris Lattner34428442003-05-27 16:40:51 +0000565 Constant *CP1 =
566 ConstantExpr::get(Instruction::Add,
567 cast<Constant>(cast<Instruction>(RHS)->getOperand(1)),
568 ConstantInt::get(I.getType(), 1));
Chris Lattner57c8d992003-02-18 19:57:07 +0000569 return BinaryOperator::create(Instruction::Mul, LHS, CP1);
570 }
571
Chris Lattnerb8b97502003-08-13 19:01:45 +0000572 // (A & C1)+(B & C2) --> (A & C1)|(B & C2) iff C1&C2 == 0
573 if (Constant *C2 = dyn_castMaskingAnd(RHS))
574 if (Instruction *R = AssociativeOpt(I, AddMaskingAnd(C2))) return R;
Chris Lattner7fb29e12003-03-11 00:12:48 +0000575
Chris Lattnerb9cde762003-10-02 15:11:26 +0000576 if (ConstantInt *CRHS = dyn_cast<ConstantInt>(RHS)) {
577 if (Instruction *ILHS = dyn_cast<Instruction>(LHS)) {
578 switch (ILHS->getOpcode()) {
579 case Instruction::Xor:
580 // ~X + C --> (C-1) - X
581 if (ConstantInt *XorRHS = dyn_cast<ConstantInt>(ILHS->getOperand(1)))
582 if (XorRHS->isAllOnesValue())
583 return BinaryOperator::create(Instruction::Sub,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000584 ConstantExpr::get(Instruction::Sub,
585 CRHS, ConstantInt::get(I.getType(), 1)),
Chris Lattnerb9cde762003-10-02 15:11:26 +0000586 ILHS->getOperand(0));
587 break;
Chris Lattner183b3362004-04-09 19:05:30 +0000588 case Instruction::Select:
589 // Try to fold constant add into select arguments.
590 if (Instruction *R = FoldBinOpIntoSelect(I,cast<SelectInst>(ILHS),this))
591 return R;
592
Chris Lattnerb9cde762003-10-02 15:11:26 +0000593 default: break;
594 }
595 }
596 }
597
Chris Lattner113f4f42002-06-25 16:13:24 +0000598 return Changed ? &I : 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000599}
600
Chris Lattnerbdb0ce02003-07-22 21:46:59 +0000601// isSignBit - Return true if the value represented by the constant only has the
602// highest order bit set.
603static bool isSignBit(ConstantInt *CI) {
604 unsigned NumBits = CI->getType()->getPrimitiveSize()*8;
605 return (CI->getRawValue() & ~(-1LL << NumBits)) == (1ULL << (NumBits-1));
606}
607
Chris Lattnerdfae8be2003-07-24 17:35:25 +0000608static unsigned getTypeSizeInBits(const Type *Ty) {
609 return Ty == Type::BoolTy ? 1 : Ty->getPrimitiveSize()*8;
610}
611
Chris Lattner022167f2004-03-13 00:11:49 +0000612/// RemoveNoopCast - Strip off nonconverting casts from the value.
613///
614static Value *RemoveNoopCast(Value *V) {
615 if (CastInst *CI = dyn_cast<CastInst>(V)) {
616 const Type *CTy = CI->getType();
617 const Type *OpTy = CI->getOperand(0)->getType();
618 if (CTy->isInteger() && OpTy->isInteger()) {
619 if (CTy->getPrimitiveSize() == OpTy->getPrimitiveSize())
620 return RemoveNoopCast(CI->getOperand(0));
621 } else if (isa<PointerType>(CTy) && isa<PointerType>(OpTy))
622 return RemoveNoopCast(CI->getOperand(0));
623 }
624 return V;
625}
626
Chris Lattner113f4f42002-06-25 16:13:24 +0000627Instruction *InstCombiner::visitSub(BinaryOperator &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +0000628 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000629
Chris Lattnere6794492002-08-12 21:17:25 +0000630 if (Op0 == Op1) // sub X, X -> 0
631 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner260ab202002-04-18 17:39:14 +0000632
Chris Lattnere6794492002-08-12 21:17:25 +0000633 // If this is a 'B = x-(-A)', change to B = x+A...
Chris Lattnerbb74e222003-03-10 23:06:50 +0000634 if (Value *V = dyn_castNegVal(Op1))
Chris Lattner147e9752002-05-08 22:46:53 +0000635 return BinaryOperator::create(Instruction::Add, Op0, V);
Chris Lattner9fa53de2002-05-06 16:49:18 +0000636
Chris Lattner8f2f5982003-11-05 01:06:05 +0000637 if (ConstantInt *C = dyn_cast<ConstantInt>(Op0)) {
638 // Replace (-1 - A) with (~A)...
Chris Lattner3082c5a2003-02-18 19:28:33 +0000639 if (C->isAllOnesValue())
640 return BinaryOperator::createNot(Op1);
Chris Lattnerad3c4952002-05-09 01:29:19 +0000641
Chris Lattner8f2f5982003-11-05 01:06:05 +0000642 // C - ~X == X + (1+C)
643 if (BinaryOperator::isNot(Op1))
644 return BinaryOperator::create(Instruction::Add,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000645 BinaryOperator::getNotArgument(cast<BinaryOperator>(Op1)),
646 ConstantExpr::get(Instruction::Add, C,
647 ConstantInt::get(I.getType(), 1)));
Chris Lattner92295c52004-03-12 23:53:13 +0000648 // -((uint)X >> 31) -> ((int)X >> 31)
649 // -((int)X >> 31) -> ((uint)X >> 31)
Chris Lattner022167f2004-03-13 00:11:49 +0000650 if (C->isNullValue()) {
651 Value *NoopCastedRHS = RemoveNoopCast(Op1);
652 if (ShiftInst *SI = dyn_cast<ShiftInst>(NoopCastedRHS))
Chris Lattner92295c52004-03-12 23:53:13 +0000653 if (SI->getOpcode() == Instruction::Shr)
654 if (ConstantUInt *CU = dyn_cast<ConstantUInt>(SI->getOperand(1))) {
655 const Type *NewTy;
Chris Lattner022167f2004-03-13 00:11:49 +0000656 if (SI->getType()->isSigned())
657 NewTy = getUnsignedIntegralType(SI->getType());
Chris Lattner92295c52004-03-12 23:53:13 +0000658 else
Chris Lattner022167f2004-03-13 00:11:49 +0000659 NewTy = getSignedIntegralType(SI->getType());
Chris Lattner92295c52004-03-12 23:53:13 +0000660 // Check to see if we are shifting out everything but the sign bit.
Chris Lattner022167f2004-03-13 00:11:49 +0000661 if (CU->getValue() == SI->getType()->getPrimitiveSize()*8-1) {
Chris Lattner92295c52004-03-12 23:53:13 +0000662 // Ok, the transformation is safe. Insert a cast of the incoming
663 // value, then the new shift, then the new cast.
664 Instruction *FirstCast = new CastInst(SI->getOperand(0), NewTy,
665 SI->getOperand(0)->getName());
666 Value *InV = InsertNewInstBefore(FirstCast, I);
667 Instruction *NewShift = new ShiftInst(Instruction::Shr, FirstCast,
668 CU, SI->getName());
Chris Lattner022167f2004-03-13 00:11:49 +0000669 if (NewShift->getType() == I.getType())
670 return NewShift;
671 else {
672 InV = InsertNewInstBefore(NewShift, I);
673 return new CastInst(NewShift, I.getType());
674 }
Chris Lattner92295c52004-03-12 23:53:13 +0000675 }
676 }
Chris Lattner022167f2004-03-13 00:11:49 +0000677 }
Chris Lattner183b3362004-04-09 19:05:30 +0000678
679 // Try to fold constant sub into select arguments.
680 if (SelectInst *SI = dyn_cast<SelectInst>(Op1))
681 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
682 return R;
Chris Lattner8f2f5982003-11-05 01:06:05 +0000683 }
684
Chris Lattner3082c5a2003-02-18 19:28:33 +0000685 if (BinaryOperator *Op1I = dyn_cast<BinaryOperator>(Op1))
Chris Lattnerf95d9b92003-10-15 16:48:29 +0000686 if (Op1I->hasOneUse()) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000687 // Replace (x - (y - z)) with (x + (z - y)) if the (y - z) subexpression
688 // is not used by anyone else...
689 //
Chris Lattnerc2f0aa52004-02-02 20:09:56 +0000690 if (Op1I->getOpcode() == Instruction::Sub &&
691 !Op1I->getType()->isFloatingPoint()) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000692 // Swap the two operands of the subexpr...
693 Value *IIOp0 = Op1I->getOperand(0), *IIOp1 = Op1I->getOperand(1);
694 Op1I->setOperand(0, IIOp1);
695 Op1I->setOperand(1, IIOp0);
696
697 // Create the new top level add instruction...
698 return BinaryOperator::create(Instruction::Add, Op0, Op1);
699 }
700
701 // Replace (A - (A & B)) with (A & ~B) if this is the only use of (A&B)...
702 //
703 if (Op1I->getOpcode() == Instruction::And &&
704 (Op1I->getOperand(0) == Op0 || Op1I->getOperand(1) == Op0)) {
705 Value *OtherOp = Op1I->getOperand(Op1I->getOperand(0) == Op0);
706
707 Instruction *NewNot = BinaryOperator::createNot(OtherOp, "B.not", &I);
708 return BinaryOperator::create(Instruction::And, Op0, NewNot);
709 }
Chris Lattner57c8d992003-02-18 19:57:07 +0000710
711 // X - X*C --> X * (1-C)
712 if (dyn_castFoldableMul(Op1I) == Op0) {
Chris Lattner34428442003-05-27 16:40:51 +0000713 Constant *CP1 =
714 ConstantExpr::get(Instruction::Sub,
715 ConstantInt::get(I.getType(), 1),
716 cast<Constant>(cast<Instruction>(Op1)->getOperand(1)));
Chris Lattner57c8d992003-02-18 19:57:07 +0000717 assert(CP1 && "Couldn't constant fold 1-C?");
718 return BinaryOperator::create(Instruction::Mul, Op0, CP1);
719 }
Chris Lattnerad3c4952002-05-09 01:29:19 +0000720 }
Chris Lattner3082c5a2003-02-18 19:28:33 +0000721
Chris Lattner57c8d992003-02-18 19:57:07 +0000722 // X*C - X --> X * (C-1)
723 if (dyn_castFoldableMul(Op0) == Op1) {
Chris Lattner34428442003-05-27 16:40:51 +0000724 Constant *CP1 =
725 ConstantExpr::get(Instruction::Sub,
726 cast<Constant>(cast<Instruction>(Op0)->getOperand(1)),
727 ConstantInt::get(I.getType(), 1));
Chris Lattner57c8d992003-02-18 19:57:07 +0000728 assert(CP1 && "Couldn't constant fold C - 1?");
729 return BinaryOperator::create(Instruction::Mul, Op1, CP1);
730 }
731
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000732 return 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000733}
734
Chris Lattnere79e8542004-02-23 06:38:22 +0000735/// isSignBitCheck - Given an exploded setcc instruction, return true if it is
736/// really just returns true if the most significant (sign) bit is set.
737static bool isSignBitCheck(unsigned Opcode, Value *LHS, ConstantInt *RHS) {
738 if (RHS->getType()->isSigned()) {
739 // True if source is LHS < 0 or LHS <= -1
740 return Opcode == Instruction::SetLT && RHS->isNullValue() ||
741 Opcode == Instruction::SetLE && RHS->isAllOnesValue();
742 } else {
743 ConstantUInt *RHSC = cast<ConstantUInt>(RHS);
744 // True if source is LHS > 127 or LHS >= 128, where the constants depend on
745 // the size of the integer type.
746 if (Opcode == Instruction::SetGE)
747 return RHSC->getValue() == 1ULL<<(RHS->getType()->getPrimitiveSize()*8-1);
748 if (Opcode == Instruction::SetGT)
749 return RHSC->getValue() ==
750 (1ULL << (RHS->getType()->getPrimitiveSize()*8-1))-1;
751 }
752 return false;
753}
754
Chris Lattner113f4f42002-06-25 16:13:24 +0000755Instruction *InstCombiner::visitMul(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +0000756 bool Changed = SimplifyCommutative(I);
Chris Lattner3082c5a2003-02-18 19:28:33 +0000757 Value *Op0 = I.getOperand(0);
Chris Lattner260ab202002-04-18 17:39:14 +0000758
Chris Lattnere6794492002-08-12 21:17:25 +0000759 // Simplify mul instructions with a constant RHS...
Chris Lattner3082c5a2003-02-18 19:28:33 +0000760 if (Constant *Op1 = dyn_cast<Constant>(I.getOperand(1))) {
761 if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) {
Chris Lattnerede3fe02003-08-13 04:18:28 +0000762
763 // ((X << C1)*C2) == (X * (C2 << C1))
764 if (ShiftInst *SI = dyn_cast<ShiftInst>(Op0))
765 if (SI->getOpcode() == Instruction::Shl)
766 if (Constant *ShOp = dyn_cast<Constant>(SI->getOperand(1)))
767 return BinaryOperator::create(Instruction::Mul, SI->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +0000768 ConstantExpr::get(Instruction::Shl, CI, ShOp));
769
Chris Lattnercce81be2003-09-11 22:24:54 +0000770 if (CI->isNullValue())
771 return ReplaceInstUsesWith(I, Op1); // X * 0 == 0
772 if (CI->equalsInt(1)) // X * 1 == X
773 return ReplaceInstUsesWith(I, Op0);
774 if (CI->isAllOnesValue()) // X * -1 == 0 - X
Chris Lattner35236d82003-06-25 17:09:20 +0000775 return BinaryOperator::createNeg(Op0, I.getName());
Chris Lattner31ba1292002-04-29 22:24:47 +0000776
Chris Lattnercce81be2003-09-11 22:24:54 +0000777 int64_t Val = (int64_t)cast<ConstantInt>(CI)->getRawValue();
Chris Lattner3082c5a2003-02-18 19:28:33 +0000778 if (uint64_t C = Log2(Val)) // Replace X*(2^C) with X << C
779 return new ShiftInst(Instruction::Shl, Op0,
780 ConstantUInt::get(Type::UByteTy, C));
781 } else {
782 ConstantFP *Op1F = cast<ConstantFP>(Op1);
783 if (Op1F->isNullValue())
784 return ReplaceInstUsesWith(I, Op1);
Chris Lattner31ba1292002-04-29 22:24:47 +0000785
Chris Lattner3082c5a2003-02-18 19:28:33 +0000786 // "In IEEE floating point, x*1 is not equivalent to x for nans. However,
787 // ANSI says we can drop signals, so we can do this anyway." (from GCC)
788 if (Op1F->getValue() == 1.0)
789 return ReplaceInstUsesWith(I, Op0); // Eliminate 'mul double %X, 1.0'
790 }
Chris Lattner183b3362004-04-09 19:05:30 +0000791
792 // Try to fold constant mul into select arguments.
793 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
794 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
795 return R;
Chris Lattner260ab202002-04-18 17:39:14 +0000796 }
797
Chris Lattner934a64cf2003-03-10 23:23:04 +0000798 if (Value *Op0v = dyn_castNegVal(Op0)) // -X * -Y = X*Y
799 if (Value *Op1v = dyn_castNegVal(I.getOperand(1)))
800 return BinaryOperator::create(Instruction::Mul, Op0v, Op1v);
801
Chris Lattner2635b522004-02-23 05:39:21 +0000802 // If one of the operands of the multiply is a cast from a boolean value, then
803 // we know the bool is either zero or one, so this is a 'masking' multiply.
804 // See if we can simplify things based on how the boolean was originally
805 // formed.
806 CastInst *BoolCast = 0;
807 if (CastInst *CI = dyn_cast<CastInst>(I.getOperand(0)))
808 if (CI->getOperand(0)->getType() == Type::BoolTy)
809 BoolCast = CI;
810 if (!BoolCast)
811 if (CastInst *CI = dyn_cast<CastInst>(I.getOperand(1)))
812 if (CI->getOperand(0)->getType() == Type::BoolTy)
813 BoolCast = CI;
814 if (BoolCast) {
815 if (SetCondInst *SCI = dyn_cast<SetCondInst>(BoolCast->getOperand(0))) {
816 Value *SCIOp0 = SCI->getOperand(0), *SCIOp1 = SCI->getOperand(1);
817 const Type *SCOpTy = SCIOp0->getType();
818
Chris Lattnere79e8542004-02-23 06:38:22 +0000819 // If the setcc is true iff the sign bit of X is set, then convert this
820 // multiply into a shift/and combination.
821 if (isa<ConstantInt>(SCIOp1) &&
822 isSignBitCheck(SCI->getOpcode(), SCIOp0, cast<ConstantInt>(SCIOp1))) {
Chris Lattner2635b522004-02-23 05:39:21 +0000823 // Shift the X value right to turn it into "all signbits".
824 Constant *Amt = ConstantUInt::get(Type::UByteTy,
825 SCOpTy->getPrimitiveSize()*8-1);
Chris Lattnere79e8542004-02-23 06:38:22 +0000826 if (SCIOp0->getType()->isUnsigned()) {
827 const Type *NewTy = getSignedIntegralType(SCIOp0->getType());
828 SCIOp0 = InsertNewInstBefore(new CastInst(SCIOp0, NewTy,
829 SCIOp0->getName()), I);
830 }
831
832 Value *V =
833 InsertNewInstBefore(new ShiftInst(Instruction::Shr, SCIOp0, Amt,
834 BoolCast->getOperand(0)->getName()+
835 ".mask"), I);
Chris Lattner2635b522004-02-23 05:39:21 +0000836
837 // If the multiply type is not the same as the source type, sign extend
838 // or truncate to the multiply type.
839 if (I.getType() != V->getType())
Chris Lattnere79e8542004-02-23 06:38:22 +0000840 V = InsertNewInstBefore(new CastInst(V, I.getType(), V->getName()),I);
Chris Lattner2635b522004-02-23 05:39:21 +0000841
842 Value *OtherOp = Op0 == BoolCast ? I.getOperand(1) : Op0;
843 return BinaryOperator::create(Instruction::And, V, OtherOp);
844 }
845 }
846 }
847
Chris Lattner113f4f42002-06-25 16:13:24 +0000848 return Changed ? &I : 0;
Chris Lattner260ab202002-04-18 17:39:14 +0000849}
850
Chris Lattner113f4f42002-06-25 16:13:24 +0000851Instruction *InstCombiner::visitDiv(BinaryOperator &I) {
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000852 // div X, 1 == X
Chris Lattner3082c5a2003-02-18 19:28:33 +0000853 if (ConstantInt *RHS = dyn_cast<ConstantInt>(I.getOperand(1))) {
Chris Lattnere6794492002-08-12 21:17:25 +0000854 if (RHS->equalsInt(1))
855 return ReplaceInstUsesWith(I, I.getOperand(0));
Chris Lattner3082c5a2003-02-18 19:28:33 +0000856
857 // Check to see if this is an unsigned division with an exact power of 2,
858 // if so, convert to a right shift.
859 if (ConstantUInt *C = dyn_cast<ConstantUInt>(RHS))
860 if (uint64_t Val = C->getValue()) // Don't break X / 0
861 if (uint64_t C = Log2(Val))
862 return new ShiftInst(Instruction::Shr, I.getOperand(0),
863 ConstantUInt::get(Type::UByteTy, C));
864 }
865
866 // 0 / X == 0, we don't need to preserve faults!
867 if (ConstantInt *LHS = dyn_cast<ConstantInt>(I.getOperand(0)))
868 if (LHS->equalsInt(0))
869 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
870
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000871 return 0;
872}
873
874
Chris Lattner113f4f42002-06-25 16:13:24 +0000875Instruction *InstCombiner::visitRem(BinaryOperator &I) {
Chris Lattner3082c5a2003-02-18 19:28:33 +0000876 if (ConstantInt *RHS = dyn_cast<ConstantInt>(I.getOperand(1))) {
877 if (RHS->equalsInt(1)) // X % 1 == 0
878 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner56b50512004-03-26 16:11:24 +0000879 if (RHS->isAllOnesValue()) // X % -1 == 0
880 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner3082c5a2003-02-18 19:28:33 +0000881
882 // Check to see if this is an unsigned remainder with an exact power of 2,
883 // if so, convert to a bitwise and.
884 if (ConstantUInt *C = dyn_cast<ConstantUInt>(RHS))
885 if (uint64_t Val = C->getValue()) // Don't break X % 0 (divide by zero)
886 if (Log2(Val))
887 return BinaryOperator::create(Instruction::And, I.getOperand(0),
888 ConstantUInt::get(I.getType(), Val-1));
889 }
890
891 // 0 % X == 0, we don't need to preserve faults!
892 if (ConstantInt *LHS = dyn_cast<ConstantInt>(I.getOperand(0)))
893 if (LHS->equalsInt(0))
Chris Lattnere6794492002-08-12 21:17:25 +0000894 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
895
Chris Lattnerf4cdbf32002-05-06 16:14:14 +0000896 return 0;
897}
898
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000899// isMaxValueMinusOne - return true if this is Max-1
Chris Lattnere6794492002-08-12 21:17:25 +0000900static bool isMaxValueMinusOne(const ConstantInt *C) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000901 if (const ConstantUInt *CU = dyn_cast<ConstantUInt>(C)) {
902 // Calculate -1 casted to the right type...
903 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
904 uint64_t Val = ~0ULL; // All ones
905 Val >>= 64-TypeBits; // Shift out unwanted 1 bits...
906 return CU->getValue() == Val-1;
907 }
908
909 const ConstantSInt *CS = cast<ConstantSInt>(C);
910
911 // Calculate 0111111111..11111
912 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
913 int64_t Val = INT64_MAX; // All ones
914 Val >>= 64-TypeBits; // Shift out unwanted 1 bits...
915 return CS->getValue() == Val-1;
916}
917
918// isMinValuePlusOne - return true if this is Min+1
Chris Lattnere6794492002-08-12 21:17:25 +0000919static bool isMinValuePlusOne(const ConstantInt *C) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +0000920 if (const ConstantUInt *CU = dyn_cast<ConstantUInt>(C))
921 return CU->getValue() == 1;
922
923 const ConstantSInt *CS = cast<ConstantSInt>(C);
924
925 // Calculate 1111111111000000000000
926 unsigned TypeBits = C->getType()->getPrimitiveSize()*8;
927 int64_t Val = -1; // All ones
928 Val <<= TypeBits-1; // Shift over to the right spot
929 return CS->getValue() == Val+1;
930}
931
Chris Lattner3ac7c262003-08-13 20:16:26 +0000932/// getSetCondCode - Encode a setcc opcode into a three bit mask. These bits
933/// are carefully arranged to allow folding of expressions such as:
934///
935/// (A < B) | (A > B) --> (A != B)
936///
937/// Bit value '4' represents that the comparison is true if A > B, bit value '2'
938/// represents that the comparison is true if A == B, and bit value '1' is true
939/// if A < B.
940///
941static unsigned getSetCondCode(const SetCondInst *SCI) {
942 switch (SCI->getOpcode()) {
943 // False -> 0
944 case Instruction::SetGT: return 1;
945 case Instruction::SetEQ: return 2;
946 case Instruction::SetGE: return 3;
947 case Instruction::SetLT: return 4;
948 case Instruction::SetNE: return 5;
949 case Instruction::SetLE: return 6;
950 // True -> 7
951 default:
952 assert(0 && "Invalid SetCC opcode!");
953 return 0;
954 }
955}
956
957/// getSetCCValue - This is the complement of getSetCondCode, which turns an
958/// opcode and two operands into either a constant true or false, or a brand new
959/// SetCC instruction.
960static Value *getSetCCValue(unsigned Opcode, Value *LHS, Value *RHS) {
961 switch (Opcode) {
962 case 0: return ConstantBool::False;
963 case 1: return new SetCondInst(Instruction::SetGT, LHS, RHS);
964 case 2: return new SetCondInst(Instruction::SetEQ, LHS, RHS);
965 case 3: return new SetCondInst(Instruction::SetGE, LHS, RHS);
966 case 4: return new SetCondInst(Instruction::SetLT, LHS, RHS);
967 case 5: return new SetCondInst(Instruction::SetNE, LHS, RHS);
968 case 6: return new SetCondInst(Instruction::SetLE, LHS, RHS);
969 case 7: return ConstantBool::True;
970 default: assert(0 && "Illegal SetCCCode!"); return 0;
971 }
972}
973
974// FoldSetCCLogical - Implements (setcc1 A, B) & (setcc2 A, B) --> (setcc3 A, B)
975struct FoldSetCCLogical {
976 InstCombiner &IC;
977 Value *LHS, *RHS;
978 FoldSetCCLogical(InstCombiner &ic, SetCondInst *SCI)
979 : IC(ic), LHS(SCI->getOperand(0)), RHS(SCI->getOperand(1)) {}
980 bool shouldApply(Value *V) const {
981 if (SetCondInst *SCI = dyn_cast<SetCondInst>(V))
982 return (SCI->getOperand(0) == LHS && SCI->getOperand(1) == RHS ||
983 SCI->getOperand(0) == RHS && SCI->getOperand(1) == LHS);
984 return false;
985 }
986 Instruction *apply(BinaryOperator &Log) const {
987 SetCondInst *SCI = cast<SetCondInst>(Log.getOperand(0));
988 if (SCI->getOperand(0) != LHS) {
989 assert(SCI->getOperand(1) == LHS);
990 SCI->swapOperands(); // Swap the LHS and RHS of the SetCC
991 }
992
993 unsigned LHSCode = getSetCondCode(SCI);
994 unsigned RHSCode = getSetCondCode(cast<SetCondInst>(Log.getOperand(1)));
995 unsigned Code;
996 switch (Log.getOpcode()) {
997 case Instruction::And: Code = LHSCode & RHSCode; break;
998 case Instruction::Or: Code = LHSCode | RHSCode; break;
999 case Instruction::Xor: Code = LHSCode ^ RHSCode; break;
Chris Lattner2caaaba2003-09-22 20:33:34 +00001000 default: assert(0 && "Illegal logical opcode!"); return 0;
Chris Lattner3ac7c262003-08-13 20:16:26 +00001001 }
1002
1003 Value *RV = getSetCCValue(Code, LHS, RHS);
1004 if (Instruction *I = dyn_cast<Instruction>(RV))
1005 return I;
1006 // Otherwise, it's a constant boolean value...
1007 return IC.ReplaceInstUsesWith(Log, RV);
1008 }
1009};
1010
1011
Chris Lattnerba1cb382003-09-19 17:17:26 +00001012// OptAndOp - This handles expressions of the form ((val OP C1) & C2). Where
1013// the Op parameter is 'OP', OpRHS is 'C1', and AndRHS is 'C2'. Op is
1014// guaranteed to be either a shift instruction or a binary operator.
1015Instruction *InstCombiner::OptAndOp(Instruction *Op,
1016 ConstantIntegral *OpRHS,
1017 ConstantIntegral *AndRHS,
1018 BinaryOperator &TheAnd) {
1019 Value *X = Op->getOperand(0);
Chris Lattnerfcf21a72004-01-12 19:47:05 +00001020 Constant *Together = 0;
1021 if (!isa<ShiftInst>(Op))
1022 Together = ConstantExpr::get(Instruction::And, AndRHS, OpRHS);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001023
Chris Lattnerba1cb382003-09-19 17:17:26 +00001024 switch (Op->getOpcode()) {
1025 case Instruction::Xor:
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001026 if (Together->isNullValue()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001027 // (X ^ C1) & C2 --> (X & C2) iff (C1&C2) == 0
1028 return BinaryOperator::create(Instruction::And, X, AndRHS);
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001029 } else if (Op->hasOneUse()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001030 // (X ^ C1) & C2 --> (X & C2) ^ (C1&C2)
1031 std::string OpName = Op->getName(); Op->setName("");
1032 Instruction *And = BinaryOperator::create(Instruction::And,
1033 X, AndRHS, OpName);
1034 InsertNewInstBefore(And, TheAnd);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001035 return BinaryOperator::create(Instruction::Xor, And, Together);
Chris Lattnerba1cb382003-09-19 17:17:26 +00001036 }
1037 break;
1038 case Instruction::Or:
1039 // (X | C1) & C2 --> X & C2 iff C1 & C1 == 0
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001040 if (Together->isNullValue())
Chris Lattnerba1cb382003-09-19 17:17:26 +00001041 return BinaryOperator::create(Instruction::And, X, AndRHS);
1042 else {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001043 if (Together == AndRHS) // (X | C) & C --> C
1044 return ReplaceInstUsesWith(TheAnd, AndRHS);
1045
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001046 if (Op->hasOneUse() && Together != OpRHS) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001047 // (X | C1) & C2 --> (X | (C1&C2)) & C2
1048 std::string Op0Name = Op->getName(); Op->setName("");
1049 Instruction *Or = BinaryOperator::create(Instruction::Or, X,
1050 Together, Op0Name);
1051 InsertNewInstBefore(Or, TheAnd);
1052 return BinaryOperator::create(Instruction::And, Or, AndRHS);
1053 }
1054 }
1055 break;
1056 case Instruction::Add:
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001057 if (Op->hasOneUse()) {
Chris Lattnerba1cb382003-09-19 17:17:26 +00001058 // Adding a one to a single bit bit-field should be turned into an XOR
1059 // of the bit. First thing to check is to see if this AND is with a
1060 // single bit constant.
1061 unsigned long long AndRHSV = cast<ConstantInt>(AndRHS)->getRawValue();
1062
1063 // Clear bits that are not part of the constant.
1064 AndRHSV &= (1ULL << AndRHS->getType()->getPrimitiveSize()*8)-1;
1065
1066 // If there is only one bit set...
1067 if ((AndRHSV & (AndRHSV-1)) == 0) {
1068 // Ok, at this point, we know that we are masking the result of the
1069 // ADD down to exactly one bit. If the constant we are adding has
1070 // no bits set below this bit, then we can eliminate the ADD.
1071 unsigned long long AddRHS = cast<ConstantInt>(OpRHS)->getRawValue();
1072
1073 // Check to see if any bits below the one bit set in AndRHSV are set.
1074 if ((AddRHS & (AndRHSV-1)) == 0) {
1075 // If not, the only thing that can effect the output of the AND is
1076 // the bit specified by AndRHSV. If that bit is set, the effect of
1077 // the XOR is to toggle the bit. If it is clear, then the ADD has
1078 // no effect.
1079 if ((AddRHS & AndRHSV) == 0) { // Bit is not set, noop
1080 TheAnd.setOperand(0, X);
1081 return &TheAnd;
1082 } else {
1083 std::string Name = Op->getName(); Op->setName("");
1084 // Pull the XOR out of the AND.
1085 Instruction *NewAnd =
1086 BinaryOperator::create(Instruction::And, X, AndRHS, Name);
1087 InsertNewInstBefore(NewAnd, TheAnd);
1088 return BinaryOperator::create(Instruction::Xor, NewAnd, AndRHS);
1089 }
1090 }
1091 }
1092 }
1093 break;
Chris Lattner2da29172003-09-19 19:05:02 +00001094
1095 case Instruction::Shl: {
1096 // We know that the AND will not produce any of the bits shifted in, so if
1097 // the anded constant includes them, clear them now!
1098 //
1099 Constant *AllOne = ConstantIntegral::getAllOnesValue(AndRHS->getType());
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001100 Constant *CI = ConstantExpr::get(Instruction::And, AndRHS,
1101 ConstantExpr::get(Instruction::Shl, AllOne, OpRHS));
Chris Lattner2da29172003-09-19 19:05:02 +00001102 if (CI != AndRHS) {
1103 TheAnd.setOperand(1, CI);
1104 return &TheAnd;
1105 }
1106 break;
1107 }
1108 case Instruction::Shr:
1109 // We know that the AND will not produce any of the bits shifted in, so if
1110 // the anded constant includes them, clear them now! This only applies to
1111 // unsigned shifts, because a signed shr may bring in set bits!
1112 //
1113 if (AndRHS->getType()->isUnsigned()) {
1114 Constant *AllOne = ConstantIntegral::getAllOnesValue(AndRHS->getType());
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001115 Constant *CI = ConstantExpr::get(Instruction::And, AndRHS,
1116 ConstantExpr::get(Instruction::Shr, AllOne, OpRHS));
Chris Lattner2da29172003-09-19 19:05:02 +00001117 if (CI != AndRHS) {
1118 TheAnd.setOperand(1, CI);
1119 return &TheAnd;
1120 }
1121 }
1122 break;
Chris Lattnerba1cb382003-09-19 17:17:26 +00001123 }
1124 return 0;
1125}
1126
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001127
Chris Lattner113f4f42002-06-25 16:13:24 +00001128Instruction *InstCombiner::visitAnd(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001129 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00001130 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001131
1132 // and X, X = X and X, 0 == 0
Chris Lattnere6794492002-08-12 21:17:25 +00001133 if (Op0 == Op1 || Op1 == Constant::getNullValue(I.getType()))
1134 return ReplaceInstUsesWith(I, Op1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001135
1136 // and X, -1 == X
Chris Lattner49b47ae2003-07-23 17:57:01 +00001137 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattnere6794492002-08-12 21:17:25 +00001138 if (RHS->isAllOnesValue())
1139 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001140
Chris Lattnerba1cb382003-09-19 17:17:26 +00001141 // Optimize a variety of ((val OP C1) & C2) combinations...
1142 if (isa<BinaryOperator>(Op0) || isa<ShiftInst>(Op0)) {
1143 Instruction *Op0I = cast<Instruction>(Op0);
Chris Lattner33217db2003-07-23 19:36:21 +00001144 Value *X = Op0I->getOperand(0);
Chris Lattner16464b32003-07-23 19:25:52 +00001145 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1)))
Chris Lattnerba1cb382003-09-19 17:17:26 +00001146 if (Instruction *Res = OptAndOp(Op0I, Op0CI, RHS, I))
1147 return Res;
Chris Lattner33217db2003-07-23 19:36:21 +00001148 }
Chris Lattner183b3362004-04-09 19:05:30 +00001149
1150 // Try to fold constant and into select arguments.
1151 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1152 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1153 return R;
Chris Lattner49b47ae2003-07-23 17:57:01 +00001154 }
1155
Chris Lattnerbb74e222003-03-10 23:06:50 +00001156 Value *Op0NotVal = dyn_castNotVal(Op0);
1157 Value *Op1NotVal = dyn_castNotVal(Op1);
Chris Lattner3082c5a2003-02-18 19:28:33 +00001158
1159 // (~A & ~B) == (~(A | B)) - Demorgan's Law
Chris Lattnerbb74e222003-03-10 23:06:50 +00001160 if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) {
Chris Lattner3082c5a2003-02-18 19:28:33 +00001161 Instruction *Or = BinaryOperator::create(Instruction::Or, Op0NotVal,
Chris Lattner49b47ae2003-07-23 17:57:01 +00001162 Op1NotVal,I.getName()+".demorgan");
1163 InsertNewInstBefore(Or, I);
Chris Lattner3082c5a2003-02-18 19:28:33 +00001164 return BinaryOperator::createNot(Or);
1165 }
1166
1167 if (Op0NotVal == Op1 || Op1NotVal == Op0) // A & ~A == ~A & A == 0
1168 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattner65217ff2002-08-23 18:32:43 +00001169
Chris Lattner3ac7c262003-08-13 20:16:26 +00001170 // (setcc1 A, B) & (setcc2 A, B) --> (setcc3 A, B)
1171 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1172 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1173 return R;
1174
Chris Lattner113f4f42002-06-25 16:13:24 +00001175 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001176}
1177
1178
1179
Chris Lattner113f4f42002-06-25 16:13:24 +00001180Instruction *InstCombiner::visitOr(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001181 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00001182 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001183
1184 // or X, X = X or X, 0 == X
Chris Lattnere6794492002-08-12 21:17:25 +00001185 if (Op0 == Op1 || Op1 == Constant::getNullValue(I.getType()))
1186 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001187
1188 // or X, -1 == -1
Chris Lattner8f0d1562003-07-23 18:29:44 +00001189 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattnere6794492002-08-12 21:17:25 +00001190 if (RHS->isAllOnesValue())
1191 return ReplaceInstUsesWith(I, Op1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001192
Chris Lattner8f0d1562003-07-23 18:29:44 +00001193 if (Instruction *Op0I = dyn_cast<Instruction>(Op0)) {
1194 // (X & C1) | C2 --> (X | C2) & (C1|C2)
1195 if (Op0I->getOpcode() == Instruction::And && isOnlyUse(Op0))
1196 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) {
1197 std::string Op0Name = Op0I->getName(); Op0I->setName("");
1198 Instruction *Or = BinaryOperator::create(Instruction::Or,
1199 Op0I->getOperand(0), RHS,
1200 Op0Name);
1201 InsertNewInstBefore(Or, I);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001202 return BinaryOperator::create(Instruction::And, Or,
1203 ConstantExpr::get(Instruction::Or, RHS, Op0CI));
Chris Lattner8f0d1562003-07-23 18:29:44 +00001204 }
1205
1206 // (X ^ C1) | C2 --> (X | C2) ^ (C1&~C2)
1207 if (Op0I->getOpcode() == Instruction::Xor && isOnlyUse(Op0))
1208 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1))) {
1209 std::string Op0Name = Op0I->getName(); Op0I->setName("");
1210 Instruction *Or = BinaryOperator::create(Instruction::Or,
1211 Op0I->getOperand(0), RHS,
1212 Op0Name);
1213 InsertNewInstBefore(Or, I);
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001214 return BinaryOperator::create(Instruction::Xor, Or,
1215 ConstantExpr::get(Instruction::And, Op0CI,
1216 NotConstant(RHS)));
Chris Lattner8f0d1562003-07-23 18:29:44 +00001217 }
1218 }
Chris Lattner183b3362004-04-09 19:05:30 +00001219
1220 // Try to fold constant and into select arguments.
1221 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1222 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1223 return R;
Chris Lattner8f0d1562003-07-23 18:29:44 +00001224 }
1225
Chris Lattner812aab72003-08-12 19:11:07 +00001226 // (A & C1)|(A & C2) == A & (C1|C2)
Chris Lattner01d56392003-08-12 19:17:27 +00001227 if (Instruction *LHS = dyn_cast<BinaryOperator>(Op0))
1228 if (Instruction *RHS = dyn_cast<BinaryOperator>(Op1))
1229 if (LHS->getOperand(0) == RHS->getOperand(0))
1230 if (Constant *C0 = dyn_castMaskingAnd(LHS))
1231 if (Constant *C1 = dyn_castMaskingAnd(RHS))
1232 return BinaryOperator::create(Instruction::And, LHS->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001233 ConstantExpr::get(Instruction::Or, C0, C1));
Chris Lattner812aab72003-08-12 19:11:07 +00001234
Chris Lattner3e327a42003-03-10 23:13:59 +00001235 Value *Op0NotVal = dyn_castNotVal(Op0);
1236 Value *Op1NotVal = dyn_castNotVal(Op1);
Chris Lattner3082c5a2003-02-18 19:28:33 +00001237
Chris Lattner3e327a42003-03-10 23:13:59 +00001238 if (Op1 == Op0NotVal) // ~A | A == -1
1239 return ReplaceInstUsesWith(I,
1240 ConstantIntegral::getAllOnesValue(I.getType()));
1241
1242 if (Op0 == Op1NotVal) // A | ~A == -1
1243 return ReplaceInstUsesWith(I,
1244 ConstantIntegral::getAllOnesValue(I.getType()));
1245
1246 // (~A | ~B) == (~(A & B)) - Demorgan's Law
1247 if (Op0NotVal && Op1NotVal && isOnlyUse(Op0) && isOnlyUse(Op1)) {
1248 Instruction *And = BinaryOperator::create(Instruction::And, Op0NotVal,
1249 Op1NotVal,I.getName()+".demorgan",
1250 &I);
1251 WorkList.push_back(And);
1252 return BinaryOperator::createNot(And);
1253 }
Chris Lattner3082c5a2003-02-18 19:28:33 +00001254
Chris Lattner3ac7c262003-08-13 20:16:26 +00001255 // (setcc1 A, B) | (setcc2 A, B) --> (setcc3 A, B)
1256 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1257 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1258 return R;
1259
Chris Lattner113f4f42002-06-25 16:13:24 +00001260 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001261}
1262
Chris Lattnerc2076352004-02-16 01:20:27 +00001263// XorSelf - Implements: X ^ X --> 0
1264struct XorSelf {
1265 Value *RHS;
1266 XorSelf(Value *rhs) : RHS(rhs) {}
1267 bool shouldApply(Value *LHS) const { return LHS == RHS; }
1268 Instruction *apply(BinaryOperator &Xor) const {
1269 return &Xor;
1270 }
1271};
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001272
1273
Chris Lattner113f4f42002-06-25 16:13:24 +00001274Instruction *InstCombiner::visitXor(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001275 bool Changed = SimplifyCommutative(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00001276 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001277
Chris Lattnerc2076352004-02-16 01:20:27 +00001278 // xor X, X = 0, even if X is nested in a sequence of Xor's.
1279 if (Instruction *Result = AssociativeOpt(I, XorSelf(Op1))) {
1280 assert(Result == &I && "AssociativeOpt didn't work?");
Chris Lattnere6794492002-08-12 21:17:25 +00001281 return ReplaceInstUsesWith(I, Constant::getNullValue(I.getType()));
Chris Lattnerc2076352004-02-16 01:20:27 +00001282 }
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001283
Chris Lattner97638592003-07-23 21:37:07 +00001284 if (ConstantIntegral *RHS = dyn_cast<ConstantIntegral>(Op1)) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001285 // xor X, 0 == X
Chris Lattner97638592003-07-23 21:37:07 +00001286 if (RHS->isNullValue())
Chris Lattnere6794492002-08-12 21:17:25 +00001287 return ReplaceInstUsesWith(I, Op0);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001288
Chris Lattner97638592003-07-23 21:37:07 +00001289 if (BinaryOperator *Op0I = dyn_cast<BinaryOperator>(Op0)) {
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001290 // xor (setcc A, B), true = not (setcc A, B) = setncc A, B
Chris Lattner97638592003-07-23 21:37:07 +00001291 if (SetCondInst *SCI = dyn_cast<SetCondInst>(Op0I))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001292 if (RHS == ConstantBool::True && SCI->hasOneUse())
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001293 return new SetCondInst(SCI->getInverseCondition(),
1294 SCI->getOperand(0), SCI->getOperand(1));
Chris Lattnere5806662003-11-04 23:50:51 +00001295
Chris Lattner8f2f5982003-11-05 01:06:05 +00001296 // ~(c-X) == X-c-1 == X+(-c-1)
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001297 if (Op0I->getOpcode() == Instruction::Sub && RHS->isAllOnesValue())
1298 if (Constant *Op0I0C = dyn_cast<Constant>(Op0I->getOperand(0))) {
1299 Constant *NegOp0I0C = ConstantExpr::get(Instruction::Sub,
1300 Constant::getNullValue(Op0I0C->getType()), Op0I0C);
1301 Constant *ConstantRHS = ConstantExpr::get(Instruction::Sub, NegOp0I0C,
1302 ConstantInt::get(I.getType(), 1));
1303 return BinaryOperator::create(Instruction::Add, Op0I->getOperand(1),
1304 ConstantRHS);
1305 }
Chris Lattner97638592003-07-23 21:37:07 +00001306
1307 if (ConstantInt *Op0CI = dyn_cast<ConstantInt>(Op0I->getOperand(1)))
Chris Lattnere5806662003-11-04 23:50:51 +00001308 switch (Op0I->getOpcode()) {
1309 case Instruction::Add:
Chris Lattner0f68fa62003-11-04 23:37:10 +00001310 // ~(X-c) --> (-c-1)-X
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001311 if (RHS->isAllOnesValue()) {
1312 Constant *NegOp0CI = ConstantExpr::get(Instruction::Sub,
1313 Constant::getNullValue(Op0CI->getType()), Op0CI);
Chris Lattner0f68fa62003-11-04 23:37:10 +00001314 return BinaryOperator::create(Instruction::Sub,
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001315 ConstantExpr::get(Instruction::Sub, NegOp0CI,
1316 ConstantInt::get(I.getType(), 1)),
Chris Lattner0f68fa62003-11-04 23:37:10 +00001317 Op0I->getOperand(0));
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001318 }
Chris Lattnere5806662003-11-04 23:50:51 +00001319 break;
1320 case Instruction::And:
Chris Lattner97638592003-07-23 21:37:07 +00001321 // (X & C1) ^ C2 --> (X & C1) | C2 iff (C1&C2) == 0
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001322 if (ConstantExpr::get(Instruction::And, RHS, Op0CI)->isNullValue())
Chris Lattner97638592003-07-23 21:37:07 +00001323 return BinaryOperator::create(Instruction::Or, Op0, RHS);
Chris Lattnere5806662003-11-04 23:50:51 +00001324 break;
1325 case Instruction::Or:
Chris Lattner97638592003-07-23 21:37:07 +00001326 // (X | C1) ^ C2 --> (X | C1) & ~C2 iff (C1&C2) == C2
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001327 if (ConstantExpr::get(Instruction::And, RHS, Op0CI) == RHS)
1328 return BinaryOperator::create(Instruction::And, Op0,
1329 NotConstant(RHS));
Chris Lattnere5806662003-11-04 23:50:51 +00001330 break;
1331 default: break;
Chris Lattner97638592003-07-23 21:37:07 +00001332 }
Chris Lattnerb8d6e402002-08-20 18:24:26 +00001333 }
Chris Lattner183b3362004-04-09 19:05:30 +00001334
1335 // Try to fold constant and into select arguments.
1336 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1337 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1338 return R;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001339 }
1340
Chris Lattnerbb74e222003-03-10 23:06:50 +00001341 if (Value *X = dyn_castNotVal(Op0)) // ~A ^ A == -1
Chris Lattner3082c5a2003-02-18 19:28:33 +00001342 if (X == Op1)
1343 return ReplaceInstUsesWith(I,
1344 ConstantIntegral::getAllOnesValue(I.getType()));
1345
Chris Lattnerbb74e222003-03-10 23:06:50 +00001346 if (Value *X = dyn_castNotVal(Op1)) // A ^ ~A == -1
Chris Lattner3082c5a2003-02-18 19:28:33 +00001347 if (X == Op0)
1348 return ReplaceInstUsesWith(I,
1349 ConstantIntegral::getAllOnesValue(I.getType()));
1350
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001351 if (Instruction *Op1I = dyn_cast<Instruction>(Op1))
Chris Lattnerb36d9082004-02-16 03:54:20 +00001352 if (Op1I->getOpcode() == Instruction::Or) {
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001353 if (Op1I->getOperand(0) == Op0) { // B^(B|A) == (A|B)^B
1354 cast<BinaryOperator>(Op1I)->swapOperands();
1355 I.swapOperands();
1356 std::swap(Op0, Op1);
1357 } else if (Op1I->getOperand(1) == Op0) { // B^(A|B) == (A|B)^B
1358 I.swapOperands();
1359 std::swap(Op0, Op1);
Chris Lattnerb36d9082004-02-16 03:54:20 +00001360 }
1361 } else if (Op1I->getOpcode() == Instruction::Xor) {
1362 if (Op0 == Op1I->getOperand(0)) // A^(A^B) == B
1363 return ReplaceInstUsesWith(I, Op1I->getOperand(1));
1364 else if (Op0 == Op1I->getOperand(1)) // A^(B^A) == B
1365 return ReplaceInstUsesWith(I, Op1I->getOperand(0));
1366 }
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001367
1368 if (Instruction *Op0I = dyn_cast<Instruction>(Op0))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001369 if (Op0I->getOpcode() == Instruction::Or && Op0I->hasOneUse()) {
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001370 if (Op0I->getOperand(0) == Op1) // (B|A)^B == (A|B)^B
1371 cast<BinaryOperator>(Op0I)->swapOperands();
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001372 if (Op0I->getOperand(1) == Op1) { // (A|B)^B == A & ~B
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001373 Value *NotB = BinaryOperator::createNot(Op1, Op1->getName()+".not", &I);
1374 WorkList.push_back(cast<Instruction>(NotB));
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001375 return BinaryOperator::create(Instruction::And, Op0I->getOperand(0),
1376 NotB);
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001377 }
Chris Lattnerb36d9082004-02-16 03:54:20 +00001378 } else if (Op0I->getOpcode() == Instruction::Xor) {
1379 if (Op1 == Op0I->getOperand(0)) // (A^B)^A == B
1380 return ReplaceInstUsesWith(I, Op0I->getOperand(1));
1381 else if (Op1 == Op0I->getOperand(1)) // (B^A)^A == B
1382 return ReplaceInstUsesWith(I, Op0I->getOperand(0));
Chris Lattner1bbb7b62003-03-10 18:24:17 +00001383 }
1384
Chris Lattner7fb29e12003-03-11 00:12:48 +00001385 // (A & C1)^(B & C2) -> (A & C1)|(B & C2) iff C1^C2 == 0
1386 if (Constant *C1 = dyn_castMaskingAnd(Op0))
1387 if (Constant *C2 = dyn_castMaskingAnd(Op1))
Chris Lattner34428442003-05-27 16:40:51 +00001388 if (ConstantExpr::get(Instruction::And, C1, C2)->isNullValue())
Chris Lattner7fb29e12003-03-11 00:12:48 +00001389 return BinaryOperator::create(Instruction::Or, Op0, Op1);
1390
Chris Lattner3ac7c262003-08-13 20:16:26 +00001391 // (setcc1 A, B) ^ (setcc2 A, B) --> (setcc3 A, B)
1392 if (SetCondInst *RHS = dyn_cast<SetCondInst>(I.getOperand(1)))
1393 if (Instruction *R = AssociativeOpt(I, FoldSetCCLogical(*this, RHS)))
1394 return R;
1395
Chris Lattner113f4f42002-06-25 16:13:24 +00001396 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001397}
1398
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001399// AddOne, SubOne - Add or subtract a constant one from an integer constant...
1400static Constant *AddOne(ConstantInt *C) {
Chris Lattner34428442003-05-27 16:40:51 +00001401 Constant *Result = ConstantExpr::get(Instruction::Add, C,
1402 ConstantInt::get(C->getType(), 1));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001403 assert(Result && "Constant folding integer addition failed!");
1404 return Result;
1405}
1406static Constant *SubOne(ConstantInt *C) {
Chris Lattner34428442003-05-27 16:40:51 +00001407 Constant *Result = ConstantExpr::get(Instruction::Sub, C,
1408 ConstantInt::get(C->getType(), 1));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001409 assert(Result && "Constant folding integer addition failed!");
1410 return Result;
1411}
1412
Chris Lattner1fc23f32002-05-09 20:11:54 +00001413// isTrueWhenEqual - Return true if the specified setcondinst instruction is
1414// true when both operands are equal...
1415//
Chris Lattner113f4f42002-06-25 16:13:24 +00001416static bool isTrueWhenEqual(Instruction &I) {
1417 return I.getOpcode() == Instruction::SetEQ ||
1418 I.getOpcode() == Instruction::SetGE ||
1419 I.getOpcode() == Instruction::SetLE;
Chris Lattner1fc23f32002-05-09 20:11:54 +00001420}
1421
Chris Lattner113f4f42002-06-25 16:13:24 +00001422Instruction *InstCombiner::visitSetCondInst(BinaryOperator &I) {
Chris Lattnerdcf240a2003-03-10 21:43:22 +00001423 bool Changed = SimplifyCommutative(I);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001424 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
1425 const Type *Ty = Op0->getType();
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001426
1427 // setcc X, X
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001428 if (Op0 == Op1)
1429 return ReplaceInstUsesWith(I, ConstantBool::get(isTrueWhenEqual(I)));
Chris Lattner1fc23f32002-05-09 20:11:54 +00001430
Chris Lattnerd07283a2003-08-13 05:38:46 +00001431 // setcc <global/alloca*>, 0 - Global/Stack value addresses are never null!
1432 if (isa<ConstantPointerNull>(Op1) &&
1433 (isa<GlobalValue>(Op0) || isa<AllocaInst>(Op0)))
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001434 return ReplaceInstUsesWith(I, ConstantBool::get(!isTrueWhenEqual(I)));
1435
Chris Lattnerd07283a2003-08-13 05:38:46 +00001436
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001437 // setcc's with boolean values can always be turned into bitwise operations
1438 if (Ty == Type::BoolTy) {
1439 // If this is <, >, or !=, we can change this into a simple xor instruction
1440 if (!isTrueWhenEqual(I))
Chris Lattner16930792003-11-03 04:25:02 +00001441 return BinaryOperator::create(Instruction::Xor, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001442
1443 // Otherwise we need to make a temporary intermediate instruction and insert
1444 // it into the instruction stream. This is what we are after:
1445 //
1446 // seteq bool %A, %B -> ~(A^B)
1447 // setle bool %A, %B -> ~A | B
1448 // setge bool %A, %B -> A | ~B
1449 //
1450 if (I.getOpcode() == Instruction::SetEQ) { // seteq case
1451 Instruction *Xor = BinaryOperator::create(Instruction::Xor, Op0, Op1,
1452 I.getName()+"tmp");
1453 InsertNewInstBefore(Xor, I);
Chris Lattner16930792003-11-03 04:25:02 +00001454 return BinaryOperator::createNot(Xor);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001455 }
1456
1457 // Handle the setXe cases...
1458 assert(I.getOpcode() == Instruction::SetGE ||
1459 I.getOpcode() == Instruction::SetLE);
1460
1461 if (I.getOpcode() == Instruction::SetGE)
1462 std::swap(Op0, Op1); // Change setge -> setle
1463
1464 // Now we just have the SetLE case.
Chris Lattner31ae8632002-08-14 17:51:49 +00001465 Instruction *Not = BinaryOperator::createNot(Op0, I.getName()+"tmp");
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001466 InsertNewInstBefore(Not, I);
Chris Lattner16930792003-11-03 04:25:02 +00001467 return BinaryOperator::create(Instruction::Or, Not, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001468 }
1469
1470 // Check to see if we are doing one of many comparisons against constant
1471 // integers at the end of their ranges...
1472 //
1473 if (ConstantInt *CI = dyn_cast<ConstantInt>(Op1)) {
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001474 // Simplify seteq and setne instructions...
1475 if (I.getOpcode() == Instruction::SetEQ ||
1476 I.getOpcode() == Instruction::SetNE) {
1477 bool isSetNE = I.getOpcode() == Instruction::SetNE;
1478
Chris Lattnercfbce7c2003-07-23 17:26:36 +00001479 // If the first operand is (and|or|xor) with a constant, and the second
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001480 // operand is a constant, simplify a bit.
Chris Lattnerc992add2003-08-13 05:33:12 +00001481 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0)) {
1482 switch (BO->getOpcode()) {
1483 case Instruction::Add:
1484 if (CI->isNullValue()) {
1485 // Replace ((add A, B) != 0) with (A != -B) if A or B is
1486 // efficiently invertible, or if the add has just this one use.
1487 Value *BOp0 = BO->getOperand(0), *BOp1 = BO->getOperand(1);
1488 if (Value *NegVal = dyn_castNegVal(BOp1))
1489 return new SetCondInst(I.getOpcode(), BOp0, NegVal);
1490 else if (Value *NegVal = dyn_castNegVal(BOp0))
1491 return new SetCondInst(I.getOpcode(), NegVal, BOp1);
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001492 else if (BO->hasOneUse()) {
Chris Lattnerc992add2003-08-13 05:33:12 +00001493 Instruction *Neg = BinaryOperator::createNeg(BOp1, BO->getName());
1494 BO->setName("");
1495 InsertNewInstBefore(Neg, I);
1496 return new SetCondInst(I.getOpcode(), BOp0, Neg);
1497 }
1498 }
1499 break;
1500 case Instruction::Xor:
1501 // For the xor case, we can xor two constants together, eliminating
1502 // the explicit xor.
1503 if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1)))
1504 return BinaryOperator::create(I.getOpcode(), BO->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001505 ConstantExpr::get(Instruction::Xor, CI, BOC));
Chris Lattnerc992add2003-08-13 05:33:12 +00001506
1507 // FALLTHROUGH
1508 case Instruction::Sub:
1509 // Replace (([sub|xor] A, B) != 0) with (A != B)
1510 if (CI->isNullValue())
1511 return new SetCondInst(I.getOpcode(), BO->getOperand(0),
1512 BO->getOperand(1));
1513 break;
1514
1515 case Instruction::Or:
1516 // If bits are being or'd in that are not present in the constant we
1517 // are comparing against, then the comparison could never succeed!
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001518 if (Constant *BOC = dyn_cast<Constant>(BO->getOperand(1))) {
1519 Constant *NotCI = NotConstant(CI);
1520 if (!ConstantExpr::get(Instruction::And, BOC, NotCI)->isNullValue())
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001521 return ReplaceInstUsesWith(I, ConstantBool::get(isSetNE));
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001522 }
Chris Lattnerc992add2003-08-13 05:33:12 +00001523 break;
1524
1525 case Instruction::And:
1526 if (ConstantInt *BOC = dyn_cast<ConstantInt>(BO->getOperand(1))) {
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001527 // If bits are being compared against that are and'd out, then the
1528 // comparison can never succeed!
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001529 if (!ConstantExpr::get(Instruction::And, CI,
1530 NotConstant(BOC))->isNullValue())
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001531 return ReplaceInstUsesWith(I, ConstantBool::get(isSetNE));
Chris Lattnerc992add2003-08-13 05:33:12 +00001532
1533 // Replace (and X, (1 << size(X)-1) != 0) with x < 0, converting X
1534 // to be a signed value as appropriate.
1535 if (isSignBit(BOC)) {
1536 Value *X = BO->getOperand(0);
1537 // If 'X' is not signed, insert a cast now...
1538 if (!BOC->getType()->isSigned()) {
Chris Lattnere79e8542004-02-23 06:38:22 +00001539 const Type *DestTy = getSignedIntegralType(BOC->getType());
Chris Lattnerc992add2003-08-13 05:33:12 +00001540 CastInst *NewCI = new CastInst(X,DestTy,X->getName()+".signed");
1541 InsertNewInstBefore(NewCI, I);
1542 X = NewCI;
1543 }
1544 return new SetCondInst(isSetNE ? Instruction::SetLT :
1545 Instruction::SetGE, X,
1546 Constant::getNullValue(X->getType()));
1547 }
Chris Lattnerd492a0b2003-07-23 17:02:11 +00001548 }
Chris Lattnerc992add2003-08-13 05:33:12 +00001549 default: break;
1550 }
1551 }
Chris Lattner2b55ea32004-02-23 07:16:20 +00001552 } else { // Not a SetEQ/SetNE
1553 // If the LHS is a cast from an integral value of the same size,
1554 if (CastInst *Cast = dyn_cast<CastInst>(Op0)) {
1555 Value *CastOp = Cast->getOperand(0);
1556 const Type *SrcTy = CastOp->getType();
1557 unsigned SrcTySize = SrcTy->getPrimitiveSize();
1558 if (SrcTy != Cast->getType() && SrcTy->isInteger() &&
1559 SrcTySize == Cast->getType()->getPrimitiveSize()) {
1560 assert((SrcTy->isSigned() ^ Cast->getType()->isSigned()) &&
1561 "Source and destination signednesses should differ!");
1562 if (Cast->getType()->isSigned()) {
1563 // If this is a signed comparison, check for comparisons in the
1564 // vicinity of zero.
1565 if (I.getOpcode() == Instruction::SetLT && CI->isNullValue())
1566 // X < 0 => x > 127
1567 return BinaryOperator::create(Instruction::SetGT, CastOp,
1568 ConstantUInt::get(SrcTy, (1ULL << (SrcTySize*8-1))-1));
1569 else if (I.getOpcode() == Instruction::SetGT &&
1570 cast<ConstantSInt>(CI)->getValue() == -1)
1571 // X > -1 => x < 128
Chris Lattnerc40b9d72004-02-23 21:46:42 +00001572 return BinaryOperator::create(Instruction::SetLT, CastOp,
Chris Lattner2b55ea32004-02-23 07:16:20 +00001573 ConstantUInt::get(SrcTy, 1ULL << (SrcTySize*8-1)));
1574 } else {
1575 ConstantUInt *CUI = cast<ConstantUInt>(CI);
1576 if (I.getOpcode() == Instruction::SetLT &&
1577 CUI->getValue() == 1ULL << (SrcTySize*8-1))
1578 // X < 128 => X > -1
1579 return BinaryOperator::create(Instruction::SetGT, CastOp,
1580 ConstantSInt::get(SrcTy, -1));
1581 else if (I.getOpcode() == Instruction::SetGT &&
1582 CUI->getValue() == (1ULL << (SrcTySize*8-1))-1)
1583 // X > 127 => X < 0
1584 return BinaryOperator::create(Instruction::SetLT, CastOp,
1585 Constant::getNullValue(SrcTy));
1586 }
1587 }
1588 }
Chris Lattnere967b342003-06-04 05:10:11 +00001589 }
Chris Lattner791ac1a2003-06-01 03:35:25 +00001590
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001591 // Check to see if we are comparing against the minimum or maximum value...
Chris Lattnere6794492002-08-12 21:17:25 +00001592 if (CI->isMinValue()) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001593 if (I.getOpcode() == Instruction::SetLT) // A < MIN -> FALSE
1594 return ReplaceInstUsesWith(I, ConstantBool::False);
1595 if (I.getOpcode() == Instruction::SetGE) // A >= MIN -> TRUE
1596 return ReplaceInstUsesWith(I, ConstantBool::True);
1597 if (I.getOpcode() == Instruction::SetLE) // A <= MIN -> A == MIN
Chris Lattner16930792003-11-03 04:25:02 +00001598 return BinaryOperator::create(Instruction::SetEQ, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001599 if (I.getOpcode() == Instruction::SetGT) // A > MIN -> A != MIN
Chris Lattner16930792003-11-03 04:25:02 +00001600 return BinaryOperator::create(Instruction::SetNE, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001601
Chris Lattnere6794492002-08-12 21:17:25 +00001602 } else if (CI->isMaxValue()) {
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001603 if (I.getOpcode() == Instruction::SetGT) // A > MAX -> FALSE
1604 return ReplaceInstUsesWith(I, ConstantBool::False);
1605 if (I.getOpcode() == Instruction::SetLE) // A <= MAX -> TRUE
1606 return ReplaceInstUsesWith(I, ConstantBool::True);
1607 if (I.getOpcode() == Instruction::SetGE) // A >= MAX -> A == MAX
Chris Lattner16930792003-11-03 04:25:02 +00001608 return BinaryOperator::create(Instruction::SetEQ, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001609 if (I.getOpcode() == Instruction::SetLT) // A < MAX -> A != MAX
Chris Lattner16930792003-11-03 04:25:02 +00001610 return BinaryOperator::create(Instruction::SetNE, Op0, Op1);
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001611
1612 // Comparing against a value really close to min or max?
1613 } else if (isMinValuePlusOne(CI)) {
1614 if (I.getOpcode() == Instruction::SetLT) // A < MIN+1 -> A == MIN
Chris Lattner16930792003-11-03 04:25:02 +00001615 return BinaryOperator::create(Instruction::SetEQ, Op0, SubOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001616 if (I.getOpcode() == Instruction::SetGE) // A >= MIN-1 -> A != MIN
Chris Lattner16930792003-11-03 04:25:02 +00001617 return BinaryOperator::create(Instruction::SetNE, Op0, SubOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001618
1619 } else if (isMaxValueMinusOne(CI)) {
1620 if (I.getOpcode() == Instruction::SetGT) // A > MAX-1 -> A == MAX
Chris Lattner16930792003-11-03 04:25:02 +00001621 return BinaryOperator::create(Instruction::SetEQ, Op0, AddOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001622 if (I.getOpcode() == Instruction::SetLE) // A <= MAX-1 -> A != MAX
Chris Lattner16930792003-11-03 04:25:02 +00001623 return BinaryOperator::create(Instruction::SetNE, Op0, AddOne(CI));
Chris Lattner6d14f2a2002-08-09 23:47:40 +00001624 }
Chris Lattner59611142004-02-23 05:47:48 +00001625
1626 // If we still have a setle or setge instruction, turn it into the
1627 // appropriate setlt or setgt instruction. Since the border cases have
1628 // already been handled above, this requires little checking.
1629 //
1630 if (I.getOpcode() == Instruction::SetLE)
1631 return BinaryOperator::create(Instruction::SetLT, Op0, AddOne(CI));
1632 if (I.getOpcode() == Instruction::SetGE)
1633 return BinaryOperator::create(Instruction::SetGT, Op0, SubOne(CI));
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001634 }
1635
Chris Lattner16930792003-11-03 04:25:02 +00001636 // Test to see if the operands of the setcc are casted versions of other
1637 // values. If the cast can be stripped off both arguments, we do so now.
Chris Lattner6444c372003-11-03 05:17:03 +00001638 if (CastInst *CI = dyn_cast<CastInst>(Op0)) {
1639 Value *CastOp0 = CI->getOperand(0);
1640 if (CastOp0->getType()->isLosslesslyConvertibleTo(CI->getType()) &&
Chris Lattner7d2a5392004-03-13 23:54:27 +00001641 (isa<Constant>(Op1) || isa<CastInst>(Op1)) &&
Chris Lattner16930792003-11-03 04:25:02 +00001642 (I.getOpcode() == Instruction::SetEQ ||
1643 I.getOpcode() == Instruction::SetNE)) {
1644 // We keep moving the cast from the left operand over to the right
1645 // operand, where it can often be eliminated completely.
Chris Lattner6444c372003-11-03 05:17:03 +00001646 Op0 = CastOp0;
Chris Lattner16930792003-11-03 04:25:02 +00001647
1648 // If operand #1 is a cast instruction, see if we can eliminate it as
1649 // well.
Chris Lattner6444c372003-11-03 05:17:03 +00001650 if (CastInst *CI2 = dyn_cast<CastInst>(Op1))
1651 if (CI2->getOperand(0)->getType()->isLosslesslyConvertibleTo(
Chris Lattner16930792003-11-03 04:25:02 +00001652 Op0->getType()))
Chris Lattner6444c372003-11-03 05:17:03 +00001653 Op1 = CI2->getOperand(0);
Chris Lattner16930792003-11-03 04:25:02 +00001654
1655 // If Op1 is a constant, we can fold the cast into the constant.
1656 if (Op1->getType() != Op0->getType())
1657 if (Constant *Op1C = dyn_cast<Constant>(Op1)) {
1658 Op1 = ConstantExpr::getCast(Op1C, Op0->getType());
1659 } else {
1660 // Otherwise, cast the RHS right before the setcc
1661 Op1 = new CastInst(Op1, Op0->getType(), Op1->getName());
1662 InsertNewInstBefore(cast<Instruction>(Op1), I);
1663 }
1664 return BinaryOperator::create(I.getOpcode(), Op0, Op1);
1665 }
1666
Chris Lattner6444c372003-11-03 05:17:03 +00001667 // Handle the special case of: setcc (cast bool to X), <cst>
1668 // This comes up when you have code like
1669 // int X = A < B;
1670 // if (X) ...
1671 // For generality, we handle any zero-extension of any operand comparison
1672 // with a constant.
1673 if (ConstantInt *ConstantRHS = dyn_cast<ConstantInt>(Op1)) {
1674 const Type *SrcTy = CastOp0->getType();
1675 const Type *DestTy = Op0->getType();
1676 if (SrcTy->getPrimitiveSize() < DestTy->getPrimitiveSize() &&
1677 (SrcTy->isUnsigned() || SrcTy == Type::BoolTy)) {
1678 // Ok, we have an expansion of operand 0 into a new type. Get the
1679 // constant value, masink off bits which are not set in the RHS. These
1680 // could be set if the destination value is signed.
1681 uint64_t ConstVal = ConstantRHS->getRawValue();
1682 ConstVal &= (1ULL << DestTy->getPrimitiveSize()*8)-1;
1683
1684 // If the constant we are comparing it with has high bits set, which
1685 // don't exist in the original value, the values could never be equal,
1686 // because the source would be zero extended.
1687 unsigned SrcBits =
1688 SrcTy == Type::BoolTy ? 1 : SrcTy->getPrimitiveSize()*8;
Chris Lattner7c94d112003-11-05 17:31:36 +00001689 bool HasSignBit = ConstVal & (1ULL << (DestTy->getPrimitiveSize()*8-1));
1690 if (ConstVal & ~((1ULL << SrcBits)-1)) {
Chris Lattner6444c372003-11-03 05:17:03 +00001691 switch (I.getOpcode()) {
1692 default: assert(0 && "Unknown comparison type!");
1693 case Instruction::SetEQ:
1694 return ReplaceInstUsesWith(I, ConstantBool::False);
1695 case Instruction::SetNE:
1696 return ReplaceInstUsesWith(I, ConstantBool::True);
1697 case Instruction::SetLT:
1698 case Instruction::SetLE:
1699 if (DestTy->isSigned() && HasSignBit)
1700 return ReplaceInstUsesWith(I, ConstantBool::False);
1701 return ReplaceInstUsesWith(I, ConstantBool::True);
1702 case Instruction::SetGT:
1703 case Instruction::SetGE:
1704 if (DestTy->isSigned() && HasSignBit)
1705 return ReplaceInstUsesWith(I, ConstantBool::True);
1706 return ReplaceInstUsesWith(I, ConstantBool::False);
1707 }
1708 }
1709
1710 // Otherwise, we can replace the setcc with a setcc of the smaller
1711 // operand value.
1712 Op1 = ConstantExpr::getCast(cast<Constant>(Op1), SrcTy);
1713 return BinaryOperator::create(I.getOpcode(), CastOp0, Op1);
1714 }
1715 }
1716 }
Chris Lattner113f4f42002-06-25 16:13:24 +00001717 return Changed ? &I : 0;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001718}
1719
1720
1721
Chris Lattnere8d6c602003-03-10 19:16:08 +00001722Instruction *InstCombiner::visitShiftInst(ShiftInst &I) {
Chris Lattner113f4f42002-06-25 16:13:24 +00001723 assert(I.getOperand(1)->getType() == Type::UByteTy);
1724 Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001725 bool isLeftShift = I.getOpcode() == Instruction::Shl;
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001726
1727 // shl X, 0 == X and shr X, 0 == X
1728 // shl 0, X == 0 and shr 0, X == 0
1729 if (Op1 == Constant::getNullValue(Type::UByteTy) ||
Chris Lattnere6794492002-08-12 21:17:25 +00001730 Op0 == Constant::getNullValue(Op0->getType()))
1731 return ReplaceInstUsesWith(I, Op0);
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001732
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001733 // shr int -1, X = -1 (for any arithmetic shift rights of ~0)
1734 if (!isLeftShift)
1735 if (ConstantSInt *CSI = dyn_cast<ConstantSInt>(Op0))
1736 if (CSI->isAllOnesValue())
1737 return ReplaceInstUsesWith(I, CSI);
1738
Chris Lattner183b3362004-04-09 19:05:30 +00001739 // Try to fold constant and into select arguments.
1740 if (isa<Constant>(Op0))
1741 if (SelectInst *SI = dyn_cast<SelectInst>(Op1))
1742 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1743 return R;
1744
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001745 if (ConstantUInt *CUI = dyn_cast<ConstantUInt>(Op1)) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001746 // shl uint X, 32 = 0 and shr ubyte Y, 9 = 0, ... just don't eliminate shr
1747 // of a signed value.
1748 //
Chris Lattnere8d6c602003-03-10 19:16:08 +00001749 unsigned TypeBits = Op0->getType()->getPrimitiveSize()*8;
Chris Lattnerf5ce2542004-02-23 20:30:06 +00001750 if (CUI->getValue() >= TypeBits) {
1751 if (!Op0->getType()->isSigned() || isLeftShift)
1752 return ReplaceInstUsesWith(I, Constant::getNullValue(Op0->getType()));
1753 else {
1754 I.setOperand(1, ConstantUInt::get(Type::UByteTy, TypeBits-1));
1755 return &I;
1756 }
1757 }
Chris Lattner55f3d942002-09-10 23:04:09 +00001758
Chris Lattnerede3fe02003-08-13 04:18:28 +00001759 // ((X*C1) << C2) == (X * (C1 << C2))
1760 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op0))
1761 if (BO->getOpcode() == Instruction::Mul && isLeftShift)
1762 if (Constant *BOOp = dyn_cast<Constant>(BO->getOperand(1)))
1763 return BinaryOperator::create(Instruction::Mul, BO->getOperand(0),
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001764 ConstantExpr::get(Instruction::Shl, BOOp, CUI));
Chris Lattnerede3fe02003-08-13 04:18:28 +00001765
Chris Lattner183b3362004-04-09 19:05:30 +00001766 // Try to fold constant and into select arguments.
1767 if (SelectInst *SI = dyn_cast<SelectInst>(Op0))
1768 if (Instruction *R = FoldBinOpIntoSelect(I, SI, this))
1769 return R;
Chris Lattnerede3fe02003-08-13 04:18:28 +00001770
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001771 // If the operand is an bitwise operator with a constant RHS, and the
1772 // shift is the only use, we can pull it out of the shift.
Chris Lattnerf95d9b92003-10-15 16:48:29 +00001773 if (Op0->hasOneUse())
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001774 if (BinaryOperator *Op0BO = dyn_cast<BinaryOperator>(Op0))
1775 if (ConstantInt *Op0C = dyn_cast<ConstantInt>(Op0BO->getOperand(1))) {
1776 bool isValid = true; // Valid only for And, Or, Xor
1777 bool highBitSet = false; // Transform if high bit of constant set?
1778
1779 switch (Op0BO->getOpcode()) {
1780 default: isValid = false; break; // Do not perform transform!
1781 case Instruction::Or:
1782 case Instruction::Xor:
1783 highBitSet = false;
1784 break;
1785 case Instruction::And:
1786 highBitSet = true;
1787 break;
1788 }
1789
1790 // If this is a signed shift right, and the high bit is modified
1791 // by the logical operation, do not perform the transformation.
1792 // The highBitSet boolean indicates the value of the high bit of
1793 // the constant which would cause it to be modified for this
1794 // operation.
1795 //
1796 if (isValid && !isLeftShift && !I.getType()->isUnsigned()) {
1797 uint64_t Val = Op0C->getRawValue();
1798 isValid = ((Val & (1 << (TypeBits-1))) != 0) == highBitSet;
1799 }
1800
1801 if (isValid) {
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001802 Constant *NewRHS = ConstantExpr::get(I.getOpcode(), Op0C, CUI);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001803
1804 Instruction *NewShift =
1805 new ShiftInst(I.getOpcode(), Op0BO->getOperand(0), CUI,
1806 Op0BO->getName());
1807 Op0BO->setName("");
1808 InsertNewInstBefore(NewShift, I);
1809
1810 return BinaryOperator::create(Op0BO->getOpcode(), NewShift,
1811 NewRHS);
1812 }
1813 }
1814
Chris Lattner3204d4e2003-07-24 17:52:58 +00001815 // If this is a shift of a shift, see if we can fold the two together...
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001816 if (ShiftInst *Op0SI = dyn_cast<ShiftInst>(Op0))
Chris Lattnerab780df2003-07-24 18:38:56 +00001817 if (ConstantUInt *ShiftAmt1C =
1818 dyn_cast<ConstantUInt>(Op0SI->getOperand(1))) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001819 unsigned ShiftAmt1 = ShiftAmt1C->getValue();
1820 unsigned ShiftAmt2 = CUI->getValue();
1821
1822 // Check for (A << c1) << c2 and (A >> c1) >> c2
1823 if (I.getOpcode() == Op0SI->getOpcode()) {
1824 unsigned Amt = ShiftAmt1+ShiftAmt2; // Fold into one big shift...
Chris Lattnerf5ce2542004-02-23 20:30:06 +00001825 if (Op0->getType()->getPrimitiveSize()*8 < Amt)
1826 Amt = Op0->getType()->getPrimitiveSize()*8;
Chris Lattner3204d4e2003-07-24 17:52:58 +00001827 return new ShiftInst(I.getOpcode(), Op0SI->getOperand(0),
1828 ConstantUInt::get(Type::UByteTy, Amt));
1829 }
1830
Chris Lattnerab780df2003-07-24 18:38:56 +00001831 // Check for (A << c1) >> c2 or visaversa. If we are dealing with
1832 // signed types, we can only support the (A >> c1) << c2 configuration,
1833 // because it can not turn an arbitrary bit of A into a sign bit.
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001834 if (I.getType()->isUnsigned() || isLeftShift) {
Chris Lattner3204d4e2003-07-24 17:52:58 +00001835 // Calculate bitmask for what gets shifted off the edge...
1836 Constant *C = ConstantIntegral::getAllOnesValue(I.getType());
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001837 if (isLeftShift)
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001838 C = ConstantExpr::get(Instruction::Shl, C, ShiftAmt1C);
Chris Lattnerdeaa0dd2003-08-12 21:53:41 +00001839 else
Chris Lattnerc1e7cc02004-01-12 19:35:11 +00001840 C = ConstantExpr::get(Instruction::Shr, C, ShiftAmt1C);
Chris Lattner3204d4e2003-07-24 17:52:58 +00001841
1842 Instruction *Mask =
1843 BinaryOperator::create(Instruction::And, Op0SI->getOperand(0),
1844 C, Op0SI->getOperand(0)->getName()+".mask");
1845 InsertNewInstBefore(Mask, I);
1846
1847 // Figure out what flavor of shift we should use...
1848 if (ShiftAmt1 == ShiftAmt2)
1849 return ReplaceInstUsesWith(I, Mask); // (A << c) >> c === A & c2
1850 else if (ShiftAmt1 < ShiftAmt2) {
1851 return new ShiftInst(I.getOpcode(), Mask,
1852 ConstantUInt::get(Type::UByteTy, ShiftAmt2-ShiftAmt1));
1853 } else {
1854 return new ShiftInst(Op0SI->getOpcode(), Mask,
1855 ConstantUInt::get(Type::UByteTy, ShiftAmt1-ShiftAmt2));
1856 }
1857 }
1858 }
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001859 }
Chris Lattner2e0fb392002-10-08 16:16:40 +00001860
Chris Lattnerf4cdbf32002-05-06 16:14:14 +00001861 return 0;
1862}
1863
1864
Chris Lattner48a44f72002-05-02 17:06:02 +00001865// isEliminableCastOfCast - Return true if it is valid to eliminate the CI
1866// instruction.
1867//
Chris Lattnerdfae8be2003-07-24 17:35:25 +00001868static inline bool isEliminableCastOfCast(const Type *SrcTy, const Type *MidTy,
1869 const Type *DstTy) {
Chris Lattner48a44f72002-05-02 17:06:02 +00001870
Chris Lattner650b6da2002-08-02 20:00:25 +00001871 // It is legal to eliminate the instruction if casting A->B->A if the sizes
1872 // are identical and the bits don't get reinterpreted (for example
Chris Lattner0bb75912002-08-14 23:21:10 +00001873 // int->float->int would not be allowed)
Misha Brukmane5838c42003-05-20 18:45:36 +00001874 if (SrcTy == DstTy && SrcTy->isLosslesslyConvertibleTo(MidTy))
Chris Lattner650b6da2002-08-02 20:00:25 +00001875 return true;
Chris Lattner48a44f72002-05-02 17:06:02 +00001876
1877 // Allow free casting and conversion of sizes as long as the sign doesn't
1878 // change...
Chris Lattnerb0b412e2002-09-03 01:08:28 +00001879 if (SrcTy->isIntegral() && MidTy->isIntegral() && DstTy->isIntegral()) {
Chris Lattner650b6da2002-08-02 20:00:25 +00001880 unsigned SrcSize = SrcTy->getPrimitiveSize();
1881 unsigned MidSize = MidTy->getPrimitiveSize();
1882 unsigned DstSize = DstTy->getPrimitiveSize();
Chris Lattner650b6da2002-08-02 20:00:25 +00001883
Chris Lattner3732aca2002-08-15 16:15:25 +00001884 // Cases where we are monotonically decreasing the size of the type are
1885 // always ok, regardless of what sign changes are going on.
1886 //
Chris Lattner0bb75912002-08-14 23:21:10 +00001887 if (SrcSize >= MidSize && MidSize >= DstSize)
Chris Lattner650b6da2002-08-02 20:00:25 +00001888 return true;
Chris Lattner3732aca2002-08-15 16:15:25 +00001889
Chris Lattner555518c2002-09-23 23:39:43 +00001890 // Cases where the source and destination type are the same, but the middle
1891 // type is bigger are noops.
1892 //
1893 if (SrcSize == DstSize && MidSize > SrcSize)
1894 return true;
1895
Chris Lattner3732aca2002-08-15 16:15:25 +00001896 // If we are monotonically growing, things are more complex.
1897 //
1898 if (SrcSize <= MidSize && MidSize <= DstSize) {
1899 // We have eight combinations of signedness to worry about. Here's the
1900 // table:
1901 static const int SignTable[8] = {
1902 // CODE, SrcSigned, MidSigned, DstSigned, Comment
1903 1, // U U U Always ok
1904 1, // U U S Always ok
1905 3, // U S U Ok iff SrcSize != MidSize
1906 3, // U S S Ok iff SrcSize != MidSize
1907 0, // S U U Never ok
1908 2, // S U S Ok iff MidSize == DstSize
1909 1, // S S U Always ok
1910 1, // S S S Always ok
1911 };
1912
1913 // Choose an action based on the current entry of the signtable that this
1914 // cast of cast refers to...
1915 unsigned Row = SrcTy->isSigned()*4+MidTy->isSigned()*2+DstTy->isSigned();
1916 switch (SignTable[Row]) {
1917 case 0: return false; // Never ok
1918 case 1: return true; // Always ok
1919 case 2: return MidSize == DstSize; // Ok iff MidSize == DstSize
1920 case 3: // Ok iff SrcSize != MidSize
1921 return SrcSize != MidSize || SrcTy == Type::BoolTy;
1922 default: assert(0 && "Bad entry in sign table!");
1923 }
Chris Lattner3732aca2002-08-15 16:15:25 +00001924 }
Chris Lattner650b6da2002-08-02 20:00:25 +00001925 }
Chris Lattner48a44f72002-05-02 17:06:02 +00001926
1927 // Otherwise, we cannot succeed. Specifically we do not want to allow things
1928 // like: short -> ushort -> uint, because this can create wrong results if
1929 // the input short is negative!
1930 //
1931 return false;
1932}
1933
Chris Lattnerdfae8be2003-07-24 17:35:25 +00001934static bool ValueRequiresCast(const Value *V, const Type *Ty) {
1935 if (V->getType() == Ty || isa<Constant>(V)) return false;
1936 if (const CastInst *CI = dyn_cast<CastInst>(V))
1937 if (isEliminableCastOfCast(CI->getOperand(0)->getType(), CI->getType(), Ty))
1938 return false;
1939 return true;
1940}
1941
1942/// InsertOperandCastBefore - This inserts a cast of V to DestTy before the
1943/// InsertBefore instruction. This is specialized a bit to avoid inserting
1944/// casts that are known to not do anything...
1945///
1946Value *InstCombiner::InsertOperandCastBefore(Value *V, const Type *DestTy,
1947 Instruction *InsertBefore) {
1948 if (V->getType() == DestTy) return V;
1949 if (Constant *C = dyn_cast<Constant>(V))
1950 return ConstantExpr::getCast(C, DestTy);
1951
1952 CastInst *CI = new CastInst(V, DestTy, V->getName());
1953 InsertNewInstBefore(CI, *InsertBefore);
1954 return CI;
1955}
Chris Lattner48a44f72002-05-02 17:06:02 +00001956
1957// CastInst simplification
Chris Lattner260ab202002-04-18 17:39:14 +00001958//
Chris Lattner113f4f42002-06-25 16:13:24 +00001959Instruction *InstCombiner::visitCastInst(CastInst &CI) {
Chris Lattner55d4bda2003-06-23 21:59:52 +00001960 Value *Src = CI.getOperand(0);
1961
Chris Lattner48a44f72002-05-02 17:06:02 +00001962 // If the user is casting a value to the same type, eliminate this cast
1963 // instruction...
Chris Lattner55d4bda2003-06-23 21:59:52 +00001964 if (CI.getType() == Src->getType())
1965 return ReplaceInstUsesWith(CI, Src);
Chris Lattner48a44f72002-05-02 17:06:02 +00001966
Chris Lattner48a44f72002-05-02 17:06:02 +00001967 // If casting the result of another cast instruction, try to eliminate this
1968 // one!
1969 //
Chris Lattner55d4bda2003-06-23 21:59:52 +00001970 if (CastInst *CSrc = dyn_cast<CastInst>(Src)) {
Chris Lattnerdfae8be2003-07-24 17:35:25 +00001971 if (isEliminableCastOfCast(CSrc->getOperand(0)->getType(),
1972 CSrc->getType(), CI.getType())) {
Chris Lattner48a44f72002-05-02 17:06:02 +00001973 // This instruction now refers directly to the cast's src operand. This
1974 // has a good chance of making CSrc dead.
Chris Lattner113f4f42002-06-25 16:13:24 +00001975 CI.setOperand(0, CSrc->getOperand(0));
1976 return &CI;
Chris Lattner48a44f72002-05-02 17:06:02 +00001977 }
1978
Chris Lattner650b6da2002-08-02 20:00:25 +00001979 // If this is an A->B->A cast, and we are dealing with integral types, try
1980 // to convert this into a logical 'and' instruction.
1981 //
1982 if (CSrc->getOperand(0)->getType() == CI.getType() &&
Chris Lattnerb0b412e2002-09-03 01:08:28 +00001983 CI.getType()->isInteger() && CSrc->getType()->isInteger() &&
Chris Lattner650b6da2002-08-02 20:00:25 +00001984 CI.getType()->isUnsigned() && CSrc->getType()->isUnsigned() &&
1985 CSrc->getType()->getPrimitiveSize() < CI.getType()->getPrimitiveSize()){
1986 assert(CSrc->getType() != Type::ULongTy &&
1987 "Cannot have type bigger than ulong!");
Chris Lattner196897c2003-05-26 23:41:32 +00001988 uint64_t AndValue = (1ULL << CSrc->getType()->getPrimitiveSize()*8)-1;
Chris Lattner650b6da2002-08-02 20:00:25 +00001989 Constant *AndOp = ConstantUInt::get(CI.getType(), AndValue);
1990 return BinaryOperator::create(Instruction::And, CSrc->getOperand(0),
1991 AndOp);
1992 }
1993 }
1994
Chris Lattnerd0d51602003-06-21 23:12:02 +00001995 // If casting the result of a getelementptr instruction with no offset, turn
1996 // this into a cast of the original pointer!
1997 //
Chris Lattner55d4bda2003-06-23 21:59:52 +00001998 if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Src)) {
Chris Lattnerd0d51602003-06-21 23:12:02 +00001999 bool AllZeroOperands = true;
2000 for (unsigned i = 1, e = GEP->getNumOperands(); i != e; ++i)
2001 if (!isa<Constant>(GEP->getOperand(i)) ||
2002 !cast<Constant>(GEP->getOperand(i))->isNullValue()) {
2003 AllZeroOperands = false;
2004 break;
2005 }
2006 if (AllZeroOperands) {
2007 CI.setOperand(0, GEP->getOperand(0));
2008 return &CI;
2009 }
2010 }
2011
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002012 // If we are casting a malloc or alloca to a pointer to a type of the same
2013 // size, rewrite the allocation instruction to allocate the "right" type.
2014 //
2015 if (AllocationInst *AI = dyn_cast<AllocationInst>(Src))
Chris Lattnerd4d987d2003-11-02 06:54:48 +00002016 if (AI->hasOneUse() && !AI->isArrayAllocation())
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002017 if (const PointerType *PTy = dyn_cast<PointerType>(CI.getType())) {
2018 // Get the type really allocated and the type casted to...
2019 const Type *AllocElTy = AI->getAllocatedType();
2020 unsigned AllocElTySize = TD->getTypeSize(AllocElTy);
2021 const Type *CastElTy = PTy->getElementType();
2022 unsigned CastElTySize = TD->getTypeSize(CastElTy);
Chris Lattner7c94d112003-11-05 17:31:36 +00002023
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002024 // If the allocation is for an even multiple of the cast type size
Chris Lattneraf789322003-11-03 01:29:41 +00002025 if (CastElTySize && (AllocElTySize % CastElTySize == 0)) {
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002026 Value *Amt = ConstantUInt::get(Type::UIntTy,
2027 AllocElTySize/CastElTySize);
2028 std::string Name = AI->getName(); AI->setName("");
2029 AllocationInst *New;
2030 if (isa<MallocInst>(AI))
2031 New = new MallocInst(CastElTy, Amt, Name);
2032 else
2033 New = new AllocaInst(CastElTy, Amt, Name);
2034 InsertNewInstBefore(New, CI);
2035 return ReplaceInstUsesWith(CI, New);
2036 }
2037 }
2038
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002039 // If the source value is an instruction with only this use, we can attempt to
2040 // propagate the cast into the instruction. Also, only handle integral types
2041 // for now.
2042 if (Instruction *SrcI = dyn_cast<Instruction>(Src))
Chris Lattnerf95d9b92003-10-15 16:48:29 +00002043 if (SrcI->hasOneUse() && Src->getType()->isIntegral() &&
Chris Lattnerdfae8be2003-07-24 17:35:25 +00002044 CI.getType()->isInteger()) { // Don't mess with casts to bool here
2045 const Type *DestTy = CI.getType();
2046 unsigned SrcBitSize = getTypeSizeInBits(Src->getType());
2047 unsigned DestBitSize = getTypeSizeInBits(DestTy);
2048
2049 Value *Op0 = SrcI->getNumOperands() > 0 ? SrcI->getOperand(0) : 0;
2050 Value *Op1 = SrcI->getNumOperands() > 1 ? SrcI->getOperand(1) : 0;
2051
2052 switch (SrcI->getOpcode()) {
2053 case Instruction::Add:
2054 case Instruction::Mul:
2055 case Instruction::And:
2056 case Instruction::Or:
2057 case Instruction::Xor:
2058 // If we are discarding information, or just changing the sign, rewrite.
2059 if (DestBitSize <= SrcBitSize && DestBitSize != 1) {
2060 // Don't insert two casts if they cannot be eliminated. We allow two
2061 // casts to be inserted if the sizes are the same. This could only be
2062 // converting signedness, which is a noop.
2063 if (DestBitSize == SrcBitSize || !ValueRequiresCast(Op1, DestTy) ||
2064 !ValueRequiresCast(Op0, DestTy)) {
2065 Value *Op0c = InsertOperandCastBefore(Op0, DestTy, SrcI);
2066 Value *Op1c = InsertOperandCastBefore(Op1, DestTy, SrcI);
2067 return BinaryOperator::create(cast<BinaryOperator>(SrcI)
2068 ->getOpcode(), Op0c, Op1c);
2069 }
2070 }
2071 break;
2072 case Instruction::Shl:
2073 // Allow changing the sign of the source operand. Do not allow changing
2074 // the size of the shift, UNLESS the shift amount is a constant. We
2075 // mush not change variable sized shifts to a smaller size, because it
2076 // is undefined to shift more bits out than exist in the value.
2077 if (DestBitSize == SrcBitSize ||
2078 (DestBitSize < SrcBitSize && isa<Constant>(Op1))) {
2079 Value *Op0c = InsertOperandCastBefore(Op0, DestTy, SrcI);
2080 return new ShiftInst(Instruction::Shl, Op0c, Op1);
2081 }
2082 break;
2083 }
2084 }
2085
Chris Lattner260ab202002-04-18 17:39:14 +00002086 return 0;
Chris Lattnerca081252001-12-14 16:52:21 +00002087}
2088
Chris Lattner56e4d3d2004-04-09 23:46:01 +00002089/// GetSelectFoldableOperands - We want to turn code that looks like this:
2090/// %C = or %A, %B
2091/// %D = select %cond, %C, %A
2092/// into:
2093/// %C = select %cond, %B, 0
2094/// %D = or %A, %C
2095///
2096/// Assuming that the specified instruction is an operand to the select, return
2097/// a bitmask indicating which operands of this instruction are foldable if they
2098/// equal the other incoming value of the select.
2099///
2100static unsigned GetSelectFoldableOperands(Instruction *I) {
2101 switch (I->getOpcode()) {
2102 case Instruction::Add:
2103 case Instruction::Mul:
2104 case Instruction::And:
2105 case Instruction::Or:
2106 case Instruction::Xor:
2107 return 3; // Can fold through either operand.
2108 case Instruction::Sub: // Can only fold on the amount subtracted.
2109 case Instruction::Shl: // Can only fold on the shift amount.
2110 case Instruction::Shr:
2111 return 1;
2112 default:
2113 return 0; // Cannot fold
2114 }
2115}
2116
2117/// GetSelectFoldableConstant - For the same transformation as the previous
2118/// function, return the identity constant that goes into the select.
2119static Constant *GetSelectFoldableConstant(Instruction *I) {
2120 switch (I->getOpcode()) {
2121 default: assert(0 && "This cannot happen!"); abort();
2122 case Instruction::Add:
2123 case Instruction::Sub:
2124 case Instruction::Or:
2125 case Instruction::Xor:
2126 return Constant::getNullValue(I->getType());
2127 case Instruction::Shl:
2128 case Instruction::Shr:
2129 return Constant::getNullValue(Type::UByteTy);
2130 case Instruction::And:
2131 return ConstantInt::getAllOnesValue(I->getType());
2132 case Instruction::Mul:
2133 return ConstantInt::get(I->getType(), 1);
2134 }
2135}
2136
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002137Instruction *InstCombiner::visitSelectInst(SelectInst &SI) {
Chris Lattner533bc492004-03-30 19:37:13 +00002138 Value *CondVal = SI.getCondition();
2139 Value *TrueVal = SI.getTrueValue();
2140 Value *FalseVal = SI.getFalseValue();
2141
2142 // select true, X, Y -> X
2143 // select false, X, Y -> Y
2144 if (ConstantBool *C = dyn_cast<ConstantBool>(CondVal))
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002145 if (C == ConstantBool::True)
Chris Lattner533bc492004-03-30 19:37:13 +00002146 return ReplaceInstUsesWith(SI, TrueVal);
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002147 else {
2148 assert(C == ConstantBool::False);
Chris Lattner533bc492004-03-30 19:37:13 +00002149 return ReplaceInstUsesWith(SI, FalseVal);
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002150 }
Chris Lattner533bc492004-03-30 19:37:13 +00002151
2152 // select C, X, X -> X
2153 if (TrueVal == FalseVal)
2154 return ReplaceInstUsesWith(SI, TrueVal);
2155
Chris Lattner1c631e82004-04-08 04:43:23 +00002156 if (SI.getType() == Type::BoolTy)
2157 if (ConstantBool *C = dyn_cast<ConstantBool>(TrueVal)) {
2158 if (C == ConstantBool::True) {
2159 // Change: A = select B, true, C --> A = or B, C
2160 return BinaryOperator::create(Instruction::Or, CondVal, FalseVal);
2161 } else {
2162 // Change: A = select B, false, C --> A = and !B, C
2163 Value *NotCond =
2164 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
2165 "not."+CondVal->getName()), SI);
2166 return BinaryOperator::create(Instruction::And, NotCond, FalseVal);
2167 }
2168 } else if (ConstantBool *C = dyn_cast<ConstantBool>(FalseVal)) {
2169 if (C == ConstantBool::False) {
2170 // Change: A = select B, C, false --> A = and B, C
2171 return BinaryOperator::create(Instruction::And, CondVal, TrueVal);
2172 } else {
2173 // Change: A = select B, C, true --> A = or !B, C
2174 Value *NotCond =
2175 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
2176 "not."+CondVal->getName()), SI);
2177 return BinaryOperator::create(Instruction::Or, NotCond, TrueVal);
2178 }
2179 }
2180
Chris Lattner183b3362004-04-09 19:05:30 +00002181 // Selecting between two integer constants?
2182 if (ConstantInt *TrueValC = dyn_cast<ConstantInt>(TrueVal))
2183 if (ConstantInt *FalseValC = dyn_cast<ConstantInt>(FalseVal)) {
2184 // select C, 1, 0 -> cast C to int
2185 if (FalseValC->isNullValue() && TrueValC->getRawValue() == 1) {
2186 return new CastInst(CondVal, SI.getType());
2187 } else if (TrueValC->isNullValue() && FalseValC->getRawValue() == 1) {
2188 // select C, 0, 1 -> cast !C to int
2189 Value *NotCond =
2190 InsertNewInstBefore(BinaryOperator::createNot(CondVal,
Chris Lattnercf7baf32004-04-09 18:19:44 +00002191 "not."+CondVal->getName()), SI);
Chris Lattner183b3362004-04-09 19:05:30 +00002192 return new CastInst(NotCond, SI.getType());
Chris Lattnercf7baf32004-04-09 18:19:44 +00002193 }
Chris Lattner533bc492004-03-30 19:37:13 +00002194 }
Chris Lattner623fba12004-04-10 22:21:27 +00002195
2196 // See if we are selecting two values based on a comparison of the two values.
2197 if (SetCondInst *SCI = dyn_cast<SetCondInst>(CondVal)) {
2198 if (SCI->getOperand(0) == TrueVal && SCI->getOperand(1) == FalseVal) {
2199 // Transform (X == Y) ? X : Y -> Y
2200 if (SCI->getOpcode() == Instruction::SetEQ)
2201 return ReplaceInstUsesWith(SI, FalseVal);
2202 // Transform (X != Y) ? X : Y -> X
2203 if (SCI->getOpcode() == Instruction::SetNE)
2204 return ReplaceInstUsesWith(SI, TrueVal);
2205 // NOTE: if we wanted to, this is where to detect MIN/MAX/ABS/etc.
2206
2207 } else if (SCI->getOperand(0) == FalseVal && SCI->getOperand(1) == TrueVal){
2208 // Transform (X == Y) ? Y : X -> X
2209 if (SCI->getOpcode() == Instruction::SetEQ)
Chris Lattner24cf0202004-04-11 01:39:19 +00002210 return ReplaceInstUsesWith(SI, FalseVal);
Chris Lattner623fba12004-04-10 22:21:27 +00002211 // Transform (X != Y) ? Y : X -> Y
2212 if (SCI->getOpcode() == Instruction::SetNE)
Chris Lattner24cf0202004-04-11 01:39:19 +00002213 return ReplaceInstUsesWith(SI, TrueVal);
Chris Lattner623fba12004-04-10 22:21:27 +00002214 // NOTE: if we wanted to, this is where to detect MIN/MAX/ABS/etc.
2215 }
2216 }
Chris Lattner1c631e82004-04-08 04:43:23 +00002217
Chris Lattner56e4d3d2004-04-09 23:46:01 +00002218 // See if we can fold the select into one of our operands.
2219 if (SI.getType()->isInteger()) {
2220 // See the comment above GetSelectFoldableOperands for a description of the
2221 // transformation we are doing here.
2222 if (Instruction *TVI = dyn_cast<Instruction>(TrueVal))
2223 if (TVI->hasOneUse() && TVI->getNumOperands() == 2 &&
2224 !isa<Constant>(FalseVal))
2225 if (unsigned SFO = GetSelectFoldableOperands(TVI)) {
2226 unsigned OpToFold = 0;
2227 if ((SFO & 1) && FalseVal == TVI->getOperand(0)) {
2228 OpToFold = 1;
2229 } else if ((SFO & 2) && FalseVal == TVI->getOperand(1)) {
2230 OpToFold = 2;
2231 }
2232
2233 if (OpToFold) {
2234 Constant *C = GetSelectFoldableConstant(TVI);
2235 std::string Name = TVI->getName(); TVI->setName("");
2236 Instruction *NewSel =
2237 new SelectInst(SI.getCondition(), TVI->getOperand(2-OpToFold), C,
2238 Name);
2239 InsertNewInstBefore(NewSel, SI);
2240 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(TVI))
2241 return BinaryOperator::create(BO->getOpcode(), FalseVal, NewSel);
2242 else if (ShiftInst *SI = dyn_cast<ShiftInst>(TVI))
2243 return new ShiftInst(SI->getOpcode(), FalseVal, NewSel);
2244 else {
2245 assert(0 && "Unknown instruction!!");
2246 }
2247 }
2248 }
2249
2250 if (Instruction *FVI = dyn_cast<Instruction>(FalseVal))
2251 if (FVI->hasOneUse() && FVI->getNumOperands() == 2 &&
2252 !isa<Constant>(TrueVal))
2253 if (unsigned SFO = GetSelectFoldableOperands(FVI)) {
2254 unsigned OpToFold = 0;
2255 if ((SFO & 1) && TrueVal == FVI->getOperand(0)) {
2256 OpToFold = 1;
2257 } else if ((SFO & 2) && TrueVal == FVI->getOperand(1)) {
2258 OpToFold = 2;
2259 }
2260
2261 if (OpToFold) {
2262 Constant *C = GetSelectFoldableConstant(FVI);
2263 std::string Name = FVI->getName(); FVI->setName("");
2264 Instruction *NewSel =
2265 new SelectInst(SI.getCondition(), C, FVI->getOperand(2-OpToFold),
2266 Name);
2267 InsertNewInstBefore(NewSel, SI);
2268 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(FVI))
2269 return BinaryOperator::create(BO->getOpcode(), TrueVal, NewSel);
2270 else if (ShiftInst *SI = dyn_cast<ShiftInst>(FVI))
2271 return new ShiftInst(SI->getOpcode(), TrueVal, NewSel);
2272 else {
2273 assert(0 && "Unknown instruction!!");
2274 }
2275 }
2276 }
2277 }
Chris Lattnerb909e8b2004-03-12 05:52:32 +00002278 return 0;
2279}
2280
2281
Chris Lattner970c33a2003-06-19 17:00:31 +00002282// CallInst simplification
2283//
2284Instruction *InstCombiner::visitCallInst(CallInst &CI) {
Chris Lattner51ea1272004-02-28 05:22:00 +00002285 // Intrinsics cannot occur in an invoke, so handle them here instead of in
2286 // visitCallSite.
2287 if (Function *F = CI.getCalledFunction())
2288 switch (F->getIntrinsicID()) {
2289 case Intrinsic::memmove:
2290 case Intrinsic::memcpy:
2291 case Intrinsic::memset:
2292 // memmove/cpy/set of zero bytes is a noop.
2293 if (Constant *NumBytes = dyn_cast<Constant>(CI.getOperand(3))) {
2294 if (NumBytes->isNullValue())
2295 return EraseInstFromFunction(CI);
2296 }
2297 break;
2298 default:
2299 break;
2300 }
2301
Chris Lattneraec3d942003-10-07 22:32:43 +00002302 return visitCallSite(&CI);
Chris Lattner970c33a2003-06-19 17:00:31 +00002303}
2304
2305// InvokeInst simplification
2306//
2307Instruction *InstCombiner::visitInvokeInst(InvokeInst &II) {
Chris Lattneraec3d942003-10-07 22:32:43 +00002308 return visitCallSite(&II);
Chris Lattner970c33a2003-06-19 17:00:31 +00002309}
2310
Chris Lattneraec3d942003-10-07 22:32:43 +00002311// visitCallSite - Improvements for call and invoke instructions.
2312//
2313Instruction *InstCombiner::visitCallSite(CallSite CS) {
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002314 bool Changed = false;
2315
2316 // If the callee is a constexpr cast of a function, attempt to move the cast
2317 // to the arguments of the call/invoke.
Chris Lattneraec3d942003-10-07 22:32:43 +00002318 if (transformConstExprCastCall(CS)) return 0;
2319
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002320 Value *Callee = CS.getCalledValue();
2321 const PointerType *PTy = cast<PointerType>(Callee->getType());
2322 const FunctionType *FTy = cast<FunctionType>(PTy->getElementType());
2323 if (FTy->isVarArg()) {
2324 // See if we can optimize any arguments passed through the varargs area of
2325 // the call.
2326 for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(),
2327 E = CS.arg_end(); I != E; ++I)
2328 if (CastInst *CI = dyn_cast<CastInst>(*I)) {
2329 // If this cast does not effect the value passed through the varargs
2330 // area, we can eliminate the use of the cast.
2331 Value *Op = CI->getOperand(0);
2332 if (CI->getType()->isLosslesslyConvertibleTo(Op->getType())) {
2333 *I = Op;
2334 Changed = true;
2335 }
2336 }
2337 }
Chris Lattneraec3d942003-10-07 22:32:43 +00002338
Chris Lattner75b4d1d2003-10-07 22:54:13 +00002339 return Changed ? CS.getInstruction() : 0;
Chris Lattneraec3d942003-10-07 22:32:43 +00002340}
2341
Chris Lattner970c33a2003-06-19 17:00:31 +00002342// transformConstExprCastCall - If the callee is a constexpr cast of a function,
2343// attempt to move the cast to the arguments of the call/invoke.
2344//
2345bool InstCombiner::transformConstExprCastCall(CallSite CS) {
2346 if (!isa<ConstantExpr>(CS.getCalledValue())) return false;
2347 ConstantExpr *CE = cast<ConstantExpr>(CS.getCalledValue());
2348 if (CE->getOpcode() != Instruction::Cast ||
2349 !isa<ConstantPointerRef>(CE->getOperand(0)))
2350 return false;
2351 ConstantPointerRef *CPR = cast<ConstantPointerRef>(CE->getOperand(0));
2352 if (!isa<Function>(CPR->getValue())) return false;
2353 Function *Callee = cast<Function>(CPR->getValue());
2354 Instruction *Caller = CS.getInstruction();
2355
2356 // Okay, this is a cast from a function to a different type. Unless doing so
2357 // would cause a type conversion of one of our arguments, change this call to
2358 // be a direct call with arguments casted to the appropriate types.
2359 //
2360 const FunctionType *FT = Callee->getFunctionType();
2361 const Type *OldRetTy = Caller->getType();
2362
Chris Lattner1f7942f2004-01-14 06:06:08 +00002363 // Check to see if we are changing the return type...
2364 if (OldRetTy != FT->getReturnType()) {
2365 if (Callee->isExternal() &&
2366 !OldRetTy->isLosslesslyConvertibleTo(FT->getReturnType()) &&
2367 !Caller->use_empty())
2368 return false; // Cannot transform this return value...
2369
2370 // If the callsite is an invoke instruction, and the return value is used by
2371 // a PHI node in a successor, we cannot change the return type of the call
2372 // because there is no place to put the cast instruction (without breaking
2373 // the critical edge). Bail out in this case.
2374 if (!Caller->use_empty())
2375 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller))
2376 for (Value::use_iterator UI = II->use_begin(), E = II->use_end();
2377 UI != E; ++UI)
2378 if (PHINode *PN = dyn_cast<PHINode>(*UI))
2379 if (PN->getParent() == II->getNormalDest() ||
Chris Lattnerfae8ab32004-02-08 21:44:31 +00002380 PN->getParent() == II->getUnwindDest())
Chris Lattner1f7942f2004-01-14 06:06:08 +00002381 return false;
2382 }
Chris Lattner970c33a2003-06-19 17:00:31 +00002383
2384 unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin());
2385 unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs);
2386
2387 CallSite::arg_iterator AI = CS.arg_begin();
2388 for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) {
2389 const Type *ParamTy = FT->getParamType(i);
2390 bool isConvertible = (*AI)->getType()->isLosslesslyConvertibleTo(ParamTy);
2391 if (Callee->isExternal() && !isConvertible) return false;
2392 }
2393
2394 if (FT->getNumParams() < NumActualArgs && !FT->isVarArg() &&
2395 Callee->isExternal())
2396 return false; // Do not delete arguments unless we have a function body...
2397
2398 // Okay, we decided that this is a safe thing to do: go ahead and start
2399 // inserting cast instructions as necessary...
2400 std::vector<Value*> Args;
2401 Args.reserve(NumActualArgs);
2402
2403 AI = CS.arg_begin();
2404 for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) {
2405 const Type *ParamTy = FT->getParamType(i);
2406 if ((*AI)->getType() == ParamTy) {
2407 Args.push_back(*AI);
2408 } else {
Chris Lattner1c631e82004-04-08 04:43:23 +00002409 Args.push_back(InsertNewInstBefore(new CastInst(*AI, ParamTy, "tmp"),
2410 *Caller));
Chris Lattner970c33a2003-06-19 17:00:31 +00002411 }
2412 }
2413
2414 // If the function takes more arguments than the call was taking, add them
2415 // now...
2416 for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i)
2417 Args.push_back(Constant::getNullValue(FT->getParamType(i)));
2418
2419 // If we are removing arguments to the function, emit an obnoxious warning...
2420 if (FT->getNumParams() < NumActualArgs)
2421 if (!FT->isVarArg()) {
2422 std::cerr << "WARNING: While resolving call to function '"
2423 << Callee->getName() << "' arguments were dropped!\n";
2424 } else {
2425 // Add all of the arguments in their promoted form to the arg list...
2426 for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) {
2427 const Type *PTy = getPromotedType((*AI)->getType());
2428 if (PTy != (*AI)->getType()) {
2429 // Must promote to pass through va_arg area!
2430 Instruction *Cast = new CastInst(*AI, PTy, "tmp");
2431 InsertNewInstBefore(Cast, *Caller);
2432 Args.push_back(Cast);
2433 } else {
2434 Args.push_back(*AI);
2435 }
2436 }
2437 }
2438
2439 if (FT->getReturnType() == Type::VoidTy)
2440 Caller->setName(""); // Void type should not have a name...
2441
2442 Instruction *NC;
2443 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) {
Chris Lattnerfae8ab32004-02-08 21:44:31 +00002444 NC = new InvokeInst(Callee, II->getNormalDest(), II->getUnwindDest(),
Chris Lattner970c33a2003-06-19 17:00:31 +00002445 Args, Caller->getName(), Caller);
2446 } else {
2447 NC = new CallInst(Callee, Args, Caller->getName(), Caller);
2448 }
2449
2450 // Insert a cast of the return type as necessary...
2451 Value *NV = NC;
2452 if (Caller->getType() != NV->getType() && !Caller->use_empty()) {
2453 if (NV->getType() != Type::VoidTy) {
2454 NV = NC = new CastInst(NC, Caller->getType(), "tmp");
Chris Lattner686767f2003-10-30 00:46:41 +00002455
2456 // If this is an invoke instruction, we should insert it after the first
2457 // non-phi, instruction in the normal successor block.
2458 if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) {
2459 BasicBlock::iterator I = II->getNormalDest()->begin();
2460 while (isa<PHINode>(I)) ++I;
2461 InsertNewInstBefore(NC, *I);
2462 } else {
2463 // Otherwise, it's a call, just insert cast right after the call instr
2464 InsertNewInstBefore(NC, *Caller);
2465 }
Chris Lattner51ea1272004-02-28 05:22:00 +00002466 AddUsersToWorkList(*Caller);
Chris Lattner970c33a2003-06-19 17:00:31 +00002467 } else {
2468 NV = Constant::getNullValue(Caller->getType());
2469 }
2470 }
2471
2472 if (Caller->getType() != Type::VoidTy && !Caller->use_empty())
2473 Caller->replaceAllUsesWith(NV);
2474 Caller->getParent()->getInstList().erase(Caller);
2475 removeFromWorkList(Caller);
2476 return true;
2477}
2478
2479
Chris Lattner48a44f72002-05-02 17:06:02 +00002480
Chris Lattnerbbbdd852002-05-06 18:06:38 +00002481// PHINode simplification
2482//
Chris Lattner113f4f42002-06-25 16:13:24 +00002483Instruction *InstCombiner::visitPHINode(PHINode &PN) {
Chris Lattner91daeb52003-12-19 05:58:40 +00002484 if (Value *V = hasConstantValue(&PN))
2485 return ReplaceInstUsesWith(PN, V);
Chris Lattner4db2d222004-02-16 05:07:08 +00002486
2487 // If the only user of this instruction is a cast instruction, and all of the
2488 // incoming values are constants, change this PHI to merge together the casted
2489 // constants.
2490 if (PN.hasOneUse())
2491 if (CastInst *CI = dyn_cast<CastInst>(PN.use_back()))
2492 if (CI->getType() != PN.getType()) { // noop casts will be folded
2493 bool AllConstant = true;
2494 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i)
2495 if (!isa<Constant>(PN.getIncomingValue(i))) {
2496 AllConstant = false;
2497 break;
2498 }
2499 if (AllConstant) {
2500 // Make a new PHI with all casted values.
2501 PHINode *New = new PHINode(CI->getType(), PN.getName(), &PN);
2502 for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) {
2503 Constant *OldArg = cast<Constant>(PN.getIncomingValue(i));
2504 New->addIncoming(ConstantExpr::getCast(OldArg, New->getType()),
2505 PN.getIncomingBlock(i));
2506 }
2507
2508 // Update the cast instruction.
2509 CI->setOperand(0, New);
2510 WorkList.push_back(CI); // revisit the cast instruction to fold.
2511 WorkList.push_back(New); // Make sure to revisit the new Phi
2512 return &PN; // PN is now dead!
2513 }
2514 }
Chris Lattner91daeb52003-12-19 05:58:40 +00002515 return 0;
Chris Lattnerbbbdd852002-05-06 18:06:38 +00002516}
2517
Chris Lattner69193f92004-04-05 01:30:19 +00002518static Value *InsertSignExtendToPtrTy(Value *V, const Type *DTy,
2519 Instruction *InsertPoint,
2520 InstCombiner *IC) {
2521 unsigned PS = IC->getTargetData().getPointerSize();
2522 const Type *VTy = V->getType();
2523 Instruction *Cast;
2524 if (!VTy->isSigned() && VTy->getPrimitiveSize() < PS)
2525 // We must insert a cast to ensure we sign-extend.
2526 V = IC->InsertNewInstBefore(new CastInst(V, VTy->getSignedVersion(),
2527 V->getName()), *InsertPoint);
2528 return IC->InsertNewInstBefore(new CastInst(V, DTy, V->getName()),
2529 *InsertPoint);
2530}
2531
Chris Lattner48a44f72002-05-02 17:06:02 +00002532
Chris Lattner113f4f42002-06-25 16:13:24 +00002533Instruction *InstCombiner::visitGetElementPtrInst(GetElementPtrInst &GEP) {
Chris Lattner471bd762003-05-22 19:07:21 +00002534 // Is it 'getelementptr %P, long 0' or 'getelementptr %P'
Chris Lattner113f4f42002-06-25 16:13:24 +00002535 // If so, eliminate the noop.
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002536 if (GEP.getNumOperands() == 1)
2537 return ReplaceInstUsesWith(GEP, GEP.getOperand(0));
2538
2539 bool HasZeroPointerIndex = false;
2540 if (Constant *C = dyn_cast<Constant>(GEP.getOperand(1)))
2541 HasZeroPointerIndex = C->isNullValue();
2542
2543 if (GEP.getNumOperands() == 2 && HasZeroPointerIndex)
Chris Lattnere6794492002-08-12 21:17:25 +00002544 return ReplaceInstUsesWith(GEP, GEP.getOperand(0));
Chris Lattner48a44f72002-05-02 17:06:02 +00002545
Chris Lattner69193f92004-04-05 01:30:19 +00002546 // Eliminate unneeded casts for indices.
2547 bool MadeChange = false;
Chris Lattner2b2412d2004-04-07 18:38:20 +00002548 gep_type_iterator GTI = gep_type_begin(GEP);
2549 for (unsigned i = 1, e = GEP.getNumOperands(); i != e; ++i, ++GTI)
2550 if (isa<SequentialType>(*GTI)) {
2551 if (CastInst *CI = dyn_cast<CastInst>(GEP.getOperand(i))) {
2552 Value *Src = CI->getOperand(0);
2553 const Type *SrcTy = Src->getType();
2554 const Type *DestTy = CI->getType();
2555 if (Src->getType()->isInteger()) {
2556 if (SrcTy->getPrimitiveSize() == DestTy->getPrimitiveSize()) {
2557 // We can always eliminate a cast from ulong or long to the other.
2558 // We can always eliminate a cast from uint to int or the other on
2559 // 32-bit pointer platforms.
2560 if (DestTy->getPrimitiveSize() >= TD->getPointerSize()) {
2561 MadeChange = true;
2562 GEP.setOperand(i, Src);
2563 }
2564 } else if (SrcTy->getPrimitiveSize() < DestTy->getPrimitiveSize() &&
2565 SrcTy->getPrimitiveSize() == 4) {
2566 // We can always eliminate a cast from int to [u]long. We can
2567 // eliminate a cast from uint to [u]long iff the target is a 32-bit
2568 // pointer target.
2569 if (SrcTy->isSigned() ||
2570 SrcTy->getPrimitiveSize() >= TD->getPointerSize()) {
2571 MadeChange = true;
2572 GEP.setOperand(i, Src);
2573 }
Chris Lattner69193f92004-04-05 01:30:19 +00002574 }
2575 }
2576 }
Chris Lattner2b2412d2004-04-07 18:38:20 +00002577 // If we are using a wider index than needed for this platform, shrink it
2578 // to what we need. If the incoming value needs a cast instruction,
2579 // insert it. This explicit cast can make subsequent optimizations more
2580 // obvious.
2581 Value *Op = GEP.getOperand(i);
2582 if (Op->getType()->getPrimitiveSize() > TD->getPointerSize())
Chris Lattner1e9ac1a2004-04-17 18:16:10 +00002583 if (Constant *C = dyn_cast<Constant>(Op)) {
2584 GEP.setOperand(i, ConstantExpr::getCast(C, TD->getIntPtrType()));
2585 MadeChange = true;
2586 } else {
Chris Lattner2b2412d2004-04-07 18:38:20 +00002587 Op = InsertNewInstBefore(new CastInst(Op, TD->getIntPtrType(),
2588 Op->getName()), GEP);
2589 GEP.setOperand(i, Op);
2590 MadeChange = true;
2591 }
Chris Lattner69193f92004-04-05 01:30:19 +00002592 }
2593 if (MadeChange) return &GEP;
2594
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002595 // Combine Indices - If the source pointer to this getelementptr instruction
2596 // is a getelementptr instruction, combine the indices of the two
2597 // getelementptr instructions into a single instruction.
2598 //
Chris Lattner57c67b02004-03-25 22:59:29 +00002599 std::vector<Value*> SrcGEPOperands;
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002600 if (GetElementPtrInst *Src = dyn_cast<GetElementPtrInst>(GEP.getOperand(0))) {
Chris Lattner57c67b02004-03-25 22:59:29 +00002601 SrcGEPOperands.assign(Src->op_begin(), Src->op_end());
2602 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(GEP.getOperand(0))) {
2603 if (CE->getOpcode() == Instruction::GetElementPtr)
2604 SrcGEPOperands.assign(CE->op_begin(), CE->op_end());
2605 }
2606
2607 if (!SrcGEPOperands.empty()) {
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002608 std::vector<Value *> Indices;
Chris Lattnerca081252001-12-14 16:52:21 +00002609
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002610 // Can we combine the two pointer arithmetics offsets?
Chris Lattner57c67b02004-03-25 22:59:29 +00002611 if (SrcGEPOperands.size() == 2 && isa<Constant>(SrcGEPOperands[1]) &&
Chris Lattner471bd762003-05-22 19:07:21 +00002612 isa<Constant>(GEP.getOperand(1))) {
Chris Lattner69193f92004-04-05 01:30:19 +00002613 Constant *SGC = cast<Constant>(SrcGEPOperands[1]);
2614 Constant *GC = cast<Constant>(GEP.getOperand(1));
2615 if (SGC->getType() != GC->getType()) {
2616 SGC = ConstantExpr::getSignExtend(SGC, Type::LongTy);
2617 GC = ConstantExpr::getSignExtend(GC, Type::LongTy);
2618 }
2619
Chris Lattner235af562003-03-05 22:33:14 +00002620 // Replace: gep (gep %P, long C1), long C2, ...
2621 // With: gep %P, long (C1+C2), ...
Chris Lattner57c67b02004-03-25 22:59:29 +00002622 GEP.setOperand(0, SrcGEPOperands[0]);
Chris Lattner69193f92004-04-05 01:30:19 +00002623 GEP.setOperand(1, ConstantExpr::getAdd(SGC, GC));
Chris Lattner57c67b02004-03-25 22:59:29 +00002624 if (Instruction *I = dyn_cast<Instruction>(GEP.getOperand(0)))
2625 AddUsersToWorkList(*I); // Reduce use count of Src
Chris Lattner235af562003-03-05 22:33:14 +00002626 return &GEP;
Chris Lattner57c67b02004-03-25 22:59:29 +00002627 } else if (SrcGEPOperands.size() == 2) {
Chris Lattner235af562003-03-05 22:33:14 +00002628 // Replace: gep (gep %P, long B), long A, ...
2629 // With: T = long A+B; gep %P, T, ...
2630 //
Chris Lattnerae739ae2004-02-23 21:46:58 +00002631 // Note that if our source is a gep chain itself that we wait for that
2632 // chain to be resolved before we perform this transformation. This
2633 // avoids us creating a TON of code in some cases.
2634 //
Chris Lattner57c67b02004-03-25 22:59:29 +00002635 if (isa<GetElementPtrInst>(SrcGEPOperands[0]) &&
2636 cast<Instruction>(SrcGEPOperands[0])->getNumOperands() == 2)
Chris Lattnerae739ae2004-02-23 21:46:58 +00002637 return 0; // Wait until our source is folded to completion.
2638
Chris Lattner69193f92004-04-05 01:30:19 +00002639 Value *Sum, *SO1 = SrcGEPOperands[1], *GO1 = GEP.getOperand(1);
2640 if (SO1 == Constant::getNullValue(SO1->getType())) {
2641 Sum = GO1;
2642 } else if (GO1 == Constant::getNullValue(GO1->getType())) {
2643 Sum = SO1;
2644 } else {
2645 // If they aren't the same type, convert both to an integer of the
2646 // target's pointer size.
2647 if (SO1->getType() != GO1->getType()) {
2648 if (Constant *SO1C = dyn_cast<Constant>(SO1)) {
2649 SO1 = ConstantExpr::getCast(SO1C, GO1->getType());
2650 } else if (Constant *GO1C = dyn_cast<Constant>(GO1)) {
2651 GO1 = ConstantExpr::getCast(GO1C, SO1->getType());
2652 } else {
2653 unsigned PS = TD->getPointerSize();
2654 Instruction *Cast;
2655 if (SO1->getType()->getPrimitiveSize() == PS) {
2656 // Convert GO1 to SO1's type.
2657 GO1 = InsertSignExtendToPtrTy(GO1, SO1->getType(), &GEP, this);
2658
2659 } else if (GO1->getType()->getPrimitiveSize() == PS) {
2660 // Convert SO1 to GO1's type.
2661 SO1 = InsertSignExtendToPtrTy(SO1, GO1->getType(), &GEP, this);
2662 } else {
2663 const Type *PT = TD->getIntPtrType();
2664 SO1 = InsertSignExtendToPtrTy(SO1, PT, &GEP, this);
2665 GO1 = InsertSignExtendToPtrTy(GO1, PT, &GEP, this);
2666 }
2667 }
2668 }
2669 Sum = BinaryOperator::create(Instruction::Add, SO1, GO1,
2670 GEP.getOperand(0)->getName()+".sum", &GEP);
Chris Lattner4d1fcf12004-04-05 16:02:41 +00002671 WorkList.push_back(cast<Instruction>(Sum));
Chris Lattner69193f92004-04-05 01:30:19 +00002672 }
Chris Lattner57c67b02004-03-25 22:59:29 +00002673 GEP.setOperand(0, SrcGEPOperands[0]);
Chris Lattner235af562003-03-05 22:33:14 +00002674 GEP.setOperand(1, Sum);
Chris Lattner235af562003-03-05 22:33:14 +00002675 return &GEP;
Chris Lattner69193f92004-04-05 01:30:19 +00002676 } else if (isa<Constant>(*GEP.idx_begin()) &&
2677 cast<Constant>(*GEP.idx_begin())->isNullValue() &&
Chris Lattner57c67b02004-03-25 22:59:29 +00002678 SrcGEPOperands.size() != 1) {
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002679 // Otherwise we can do the fold if the first index of the GEP is a zero
Chris Lattner57c67b02004-03-25 22:59:29 +00002680 Indices.insert(Indices.end(), SrcGEPOperands.begin()+1,
2681 SrcGEPOperands.end());
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002682 Indices.insert(Indices.end(), GEP.idx_begin()+1, GEP.idx_end());
Chris Lattner69193f92004-04-05 01:30:19 +00002683 } else if (SrcGEPOperands.back() ==
2684 Constant::getNullValue(SrcGEPOperands.back()->getType())) {
2685 // We have to check to make sure this really is an ARRAY index we are
2686 // ending up with, not a struct index.
2687 generic_gep_type_iterator<std::vector<Value*>::iterator>
2688 GTI = gep_type_begin(SrcGEPOperands[0]->getType(),
2689 SrcGEPOperands.begin()+1, SrcGEPOperands.end());
2690 std::advance(GTI, SrcGEPOperands.size()-2);
2691 if (isa<SequentialType>(*GTI)) {
2692 // If the src gep ends with a constant array index, merge this get into
2693 // it, even if we have a non-zero array index.
2694 Indices.insert(Indices.end(), SrcGEPOperands.begin()+1,
2695 SrcGEPOperands.end()-1);
2696 Indices.insert(Indices.end(), GEP.idx_begin(), GEP.idx_end());
2697 }
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002698 }
2699
2700 if (!Indices.empty())
Chris Lattner57c67b02004-03-25 22:59:29 +00002701 return new GetElementPtrInst(SrcGEPOperands[0], Indices, GEP.getName());
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002702
2703 } else if (GlobalValue *GV = dyn_cast<GlobalValue>(GEP.getOperand(0))) {
2704 // GEP of global variable. If all of the indices for this GEP are
2705 // constants, we can promote this to a constexpr instead of an instruction.
2706
2707 // Scan for nonconstants...
2708 std::vector<Constant*> Indices;
2709 User::op_iterator I = GEP.idx_begin(), E = GEP.idx_end();
2710 for (; I != E && isa<Constant>(*I); ++I)
2711 Indices.push_back(cast<Constant>(*I));
2712
2713 if (I == E) { // If they are all constants...
Chris Lattner46b3d302003-04-16 22:40:51 +00002714 Constant *CE =
Chris Lattnerc59af1d2002-08-17 22:21:59 +00002715 ConstantExpr::getGetElementPtr(ConstantPointerRef::get(GV), Indices);
2716
2717 // Replace all uses of the GEP with the new constexpr...
2718 return ReplaceInstUsesWith(GEP, CE);
2719 }
Chris Lattner8d0bacb2004-02-22 05:25:17 +00002720 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(GEP.getOperand(0))) {
2721 if (CE->getOpcode() == Instruction::Cast) {
2722 if (HasZeroPointerIndex) {
2723 // transform: GEP (cast [10 x ubyte]* X to [0 x ubyte]*), long 0, ...
2724 // into : GEP [10 x ubyte]* X, long 0, ...
2725 //
2726 // This occurs when the program declares an array extern like "int X[];"
2727 //
2728 Constant *X = CE->getOperand(0);
2729 const PointerType *CPTy = cast<PointerType>(CE->getType());
2730 if (const PointerType *XTy = dyn_cast<PointerType>(X->getType()))
2731 if (const ArrayType *XATy =
2732 dyn_cast<ArrayType>(XTy->getElementType()))
2733 if (const ArrayType *CATy =
2734 dyn_cast<ArrayType>(CPTy->getElementType()))
2735 if (CATy->getElementType() == XATy->getElementType()) {
2736 // At this point, we know that the cast source type is a pointer
2737 // to an array of the same type as the destination pointer
2738 // array. Because the array type is never stepped over (there
2739 // is a leading zero) we can fold the cast into this GEP.
2740 GEP.setOperand(0, X);
2741 return &GEP;
2742 }
2743 }
2744 }
Chris Lattnerca081252001-12-14 16:52:21 +00002745 }
2746
Chris Lattnerca081252001-12-14 16:52:21 +00002747 return 0;
2748}
2749
Chris Lattner1085bdf2002-11-04 16:18:53 +00002750Instruction *InstCombiner::visitAllocationInst(AllocationInst &AI) {
2751 // Convert: malloc Ty, C - where C is a constant != 1 into: malloc [C x Ty], 1
2752 if (AI.isArrayAllocation()) // Check C != 1
2753 if (const ConstantUInt *C = dyn_cast<ConstantUInt>(AI.getArraySize())) {
2754 const Type *NewTy = ArrayType::get(AI.getAllocatedType(), C->getValue());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002755 AllocationInst *New = 0;
Chris Lattner1085bdf2002-11-04 16:18:53 +00002756
2757 // Create and insert the replacement instruction...
2758 if (isa<MallocInst>(AI))
Chris Lattnerabb77c92004-03-19 06:08:10 +00002759 New = new MallocInst(NewTy, 0, AI.getName());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002760 else {
2761 assert(isa<AllocaInst>(AI) && "Unknown type of allocation inst!");
Chris Lattnerabb77c92004-03-19 06:08:10 +00002762 New = new AllocaInst(NewTy, 0, AI.getName());
Chris Lattnera2620ac2002-11-09 00:49:43 +00002763 }
Chris Lattnerabb77c92004-03-19 06:08:10 +00002764
2765 InsertNewInstBefore(New, AI);
Chris Lattner1085bdf2002-11-04 16:18:53 +00002766
2767 // Scan to the end of the allocation instructions, to skip over a block of
2768 // allocas if possible...
2769 //
2770 BasicBlock::iterator It = New;
2771 while (isa<AllocationInst>(*It)) ++It;
2772
2773 // Now that I is pointing to the first non-allocation-inst in the block,
2774 // insert our getelementptr instruction...
2775 //
Chris Lattner69193f92004-04-05 01:30:19 +00002776 std::vector<Value*> Idx(2, Constant::getNullValue(Type::IntTy));
Chris Lattner1085bdf2002-11-04 16:18:53 +00002777 Value *V = new GetElementPtrInst(New, Idx, New->getName()+".sub", It);
2778
2779 // Now make everything use the getelementptr instead of the original
2780 // allocation.
Chris Lattnerabb77c92004-03-19 06:08:10 +00002781 return ReplaceInstUsesWith(AI, V);
Chris Lattner1085bdf2002-11-04 16:18:53 +00002782 }
Chris Lattnerabb77c92004-03-19 06:08:10 +00002783
2784 // If alloca'ing a zero byte object, replace the alloca with a null pointer.
2785 // Note that we only do this for alloca's, because malloc should allocate and
2786 // return a unique pointer, even for a zero byte allocation.
2787 if (isa<AllocaInst>(AI) && TD->getTypeSize(AI.getAllocatedType()) == 0)
2788 return ReplaceInstUsesWith(AI, Constant::getNullValue(AI.getType()));
2789
Chris Lattner1085bdf2002-11-04 16:18:53 +00002790 return 0;
2791}
2792
Chris Lattner8427bff2003-12-07 01:24:23 +00002793Instruction *InstCombiner::visitFreeInst(FreeInst &FI) {
2794 Value *Op = FI.getOperand(0);
2795
2796 // Change free <ty>* (cast <ty2>* X to <ty>*) into free <ty2>* X
2797 if (CastInst *CI = dyn_cast<CastInst>(Op))
2798 if (isa<PointerType>(CI->getOperand(0)->getType())) {
2799 FI.setOperand(0, CI->getOperand(0));
2800 return &FI;
2801 }
2802
Chris Lattnerf3a36602004-02-28 04:57:37 +00002803 // If we have 'free null' delete the instruction. This can happen in stl code
2804 // when lots of inlining happens.
Chris Lattner51ea1272004-02-28 05:22:00 +00002805 if (isa<ConstantPointerNull>(Op))
2806 return EraseInstFromFunction(FI);
Chris Lattnerf3a36602004-02-28 04:57:37 +00002807
Chris Lattner8427bff2003-12-07 01:24:23 +00002808 return 0;
2809}
2810
2811
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002812/// GetGEPGlobalInitializer - Given a constant, and a getelementptr
2813/// constantexpr, return the constant value being addressed by the constant
2814/// expression, or null if something is funny.
2815///
2816static Constant *GetGEPGlobalInitializer(Constant *C, ConstantExpr *CE) {
Chris Lattner69193f92004-04-05 01:30:19 +00002817 if (CE->getOperand(1) != Constant::getNullValue(CE->getOperand(1)->getType()))
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002818 return 0; // Do not allow stepping over the value!
2819
2820 // Loop over all of the operands, tracking down which value we are
2821 // addressing...
2822 for (unsigned i = 2, e = CE->getNumOperands(); i != e; ++i)
2823 if (ConstantUInt *CU = dyn_cast<ConstantUInt>(CE->getOperand(i))) {
Chris Lattner76b2ff42004-02-15 05:55:15 +00002824 ConstantStruct *CS = dyn_cast<ConstantStruct>(C);
2825 if (CS == 0) return 0;
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002826 if (CU->getValue() >= CS->getValues().size()) return 0;
2827 C = cast<Constant>(CS->getValues()[CU->getValue()]);
2828 } else if (ConstantSInt *CS = dyn_cast<ConstantSInt>(CE->getOperand(i))) {
Chris Lattner76b2ff42004-02-15 05:55:15 +00002829 ConstantArray *CA = dyn_cast<ConstantArray>(C);
2830 if (CA == 0) return 0;
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002831 if ((uint64_t)CS->getValue() >= CA->getValues().size()) return 0;
2832 C = cast<Constant>(CA->getValues()[CS->getValue()]);
2833 } else
2834 return 0;
2835 return C;
2836}
2837
2838Instruction *InstCombiner::visitLoadInst(LoadInst &LI) {
2839 Value *Op = LI.getOperand(0);
Chris Lattner7e8af382004-01-12 04:13:56 +00002840 if (LI.isVolatile()) return 0;
2841
Chris Lattner6679e462004-04-14 03:28:36 +00002842 if (Constant *C = dyn_cast<Constant>(Op))
2843 if (C->isNullValue()) // load null -> 0
2844 return ReplaceInstUsesWith(LI, Constant::getNullValue(LI.getType()));
2845 else if (ConstantPointerRef *CPR = dyn_cast<ConstantPointerRef>(C))
2846 Op = CPR->getValue();
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002847
2848 // Instcombine load (constant global) into the value loaded...
2849 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op))
Chris Lattnerbdb0ce02003-07-22 21:46:59 +00002850 if (GV->isConstant() && !GV->isExternal())
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002851 return ReplaceInstUsesWith(LI, GV->getInitializer());
2852
2853 // Instcombine load (constantexpr_GEP global, 0, ...) into the value loaded...
2854 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Op))
2855 if (CE->getOpcode() == Instruction::GetElementPtr)
2856 if (ConstantPointerRef *G=dyn_cast<ConstantPointerRef>(CE->getOperand(0)))
2857 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(G->getValue()))
Chris Lattnerbdb0ce02003-07-22 21:46:59 +00002858 if (GV->isConstant() && !GV->isExternal())
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002859 if (Constant *V = GetGEPGlobalInitializer(GV->getInitializer(), CE))
2860 return ReplaceInstUsesWith(LI, V);
Chris Lattnere228ee52004-04-08 20:39:49 +00002861
2862 // load (cast X) --> cast (load X) iff safe
2863 if (CastInst *CI = dyn_cast<CastInst>(Op)) {
2864 const Type *DestPTy = cast<PointerType>(CI->getType())->getElementType();
2865 if (const PointerType *SrcTy =
2866 dyn_cast<PointerType>(CI->getOperand(0)->getType())) {
2867 const Type *SrcPTy = SrcTy->getElementType();
2868 if (TD->getTypeSize(SrcPTy) == TD->getTypeSize(DestPTy) &&
2869 (SrcPTy->isInteger() || isa<PointerType>(SrcPTy)) &&
2870 (DestPTy->isInteger() || isa<PointerType>(DestPTy))) {
2871 // Okay, we are casting from one integer or pointer type to another of
2872 // the same size. Instead of casting the pointer before the load, cast
2873 // the result of the loaded value.
2874 Value *NewLoad = InsertNewInstBefore(new LoadInst(CI->getOperand(0),
2875 CI->getName()), LI);
2876 // Now cast the result of the load.
2877 return new CastInst(NewLoad, LI.getType());
2878 }
2879 }
2880 }
2881
Chris Lattner0f1d8a32003-06-26 05:06:25 +00002882 return 0;
2883}
2884
2885
Chris Lattner9eef8a72003-06-04 04:46:00 +00002886Instruction *InstCombiner::visitBranchInst(BranchInst &BI) {
2887 // Change br (not X), label True, label False to: br X, label False, True
Chris Lattner4f7acca2004-02-27 06:27:46 +00002888 if (BI.isConditional() && !isa<Constant>(BI.getCondition())) {
Chris Lattnere967b342003-06-04 05:10:11 +00002889 if (Value *V = dyn_castNotVal(BI.getCondition())) {
2890 BasicBlock *TrueDest = BI.getSuccessor(0);
2891 BasicBlock *FalseDest = BI.getSuccessor(1);
2892 // Swap Destinations and condition...
2893 BI.setCondition(V);
2894 BI.setSuccessor(0, FalseDest);
2895 BI.setSuccessor(1, TrueDest);
2896 return &BI;
Chris Lattner4f7acca2004-02-27 06:27:46 +00002897 } else if (SetCondInst *I = dyn_cast<SetCondInst>(BI.getCondition())) {
2898 // Cannonicalize setne -> seteq
2899 if ((I->getOpcode() == Instruction::SetNE ||
2900 I->getOpcode() == Instruction::SetLE ||
2901 I->getOpcode() == Instruction::SetGE) && I->hasOneUse()) {
2902 std::string Name = I->getName(); I->setName("");
2903 Instruction::BinaryOps NewOpcode =
2904 SetCondInst::getInverseCondition(I->getOpcode());
2905 Value *NewSCC = BinaryOperator::create(NewOpcode, I->getOperand(0),
2906 I->getOperand(1), Name, I);
2907 BasicBlock *TrueDest = BI.getSuccessor(0);
2908 BasicBlock *FalseDest = BI.getSuccessor(1);
2909 // Swap Destinations and condition...
2910 BI.setCondition(NewSCC);
2911 BI.setSuccessor(0, FalseDest);
2912 BI.setSuccessor(1, TrueDest);
2913 removeFromWorkList(I);
2914 I->getParent()->getInstList().erase(I);
2915 WorkList.push_back(cast<Instruction>(NewSCC));
2916 return &BI;
2917 }
Chris Lattnere967b342003-06-04 05:10:11 +00002918 }
Chris Lattner4f7acca2004-02-27 06:27:46 +00002919 }
Chris Lattner9eef8a72003-06-04 04:46:00 +00002920 return 0;
2921}
Chris Lattner1085bdf2002-11-04 16:18:53 +00002922
Chris Lattnerca081252001-12-14 16:52:21 +00002923
Chris Lattner99f48c62002-09-02 04:59:56 +00002924void InstCombiner::removeFromWorkList(Instruction *I) {
2925 WorkList.erase(std::remove(WorkList.begin(), WorkList.end(), I),
2926 WorkList.end());
2927}
2928
Chris Lattner113f4f42002-06-25 16:13:24 +00002929bool InstCombiner::runOnFunction(Function &F) {
Chris Lattner260ab202002-04-18 17:39:14 +00002930 bool Changed = false;
Chris Lattnerf4ad1652003-11-02 05:57:39 +00002931 TD = &getAnalysis<TargetData>();
Chris Lattnerca081252001-12-14 16:52:21 +00002932
Chris Lattner260ab202002-04-18 17:39:14 +00002933 WorkList.insert(WorkList.end(), inst_begin(F), inst_end(F));
Chris Lattnerca081252001-12-14 16:52:21 +00002934
2935 while (!WorkList.empty()) {
2936 Instruction *I = WorkList.back(); // Get an instruction from the worklist
2937 WorkList.pop_back();
2938
Misha Brukman632df282002-10-29 23:06:16 +00002939 // Check to see if we can DCE or ConstantPropagate the instruction...
Chris Lattner99f48c62002-09-02 04:59:56 +00002940 // Check to see if we can DIE the instruction...
2941 if (isInstructionTriviallyDead(I)) {
2942 // Add operands to the worklist...
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002943 if (I->getNumOperands() < 4)
Chris Lattner51ea1272004-02-28 05:22:00 +00002944 AddUsesToWorkList(*I);
Chris Lattner99f48c62002-09-02 04:59:56 +00002945 ++NumDeadInst;
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002946
2947 I->getParent()->getInstList().erase(I);
2948 removeFromWorkList(I);
2949 continue;
2950 }
Chris Lattner99f48c62002-09-02 04:59:56 +00002951
Misha Brukman632df282002-10-29 23:06:16 +00002952 // Instruction isn't dead, see if we can constant propagate it...
Chris Lattner99f48c62002-09-02 04:59:56 +00002953 if (Constant *C = ConstantFoldInstruction(I)) {
2954 // Add operands to the worklist...
Chris Lattner51ea1272004-02-28 05:22:00 +00002955 AddUsesToWorkList(*I);
Chris Lattnerc6509f42002-12-05 22:41:53 +00002956 ReplaceInstUsesWith(*I, C);
2957
Chris Lattner99f48c62002-09-02 04:59:56 +00002958 ++NumConstProp;
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002959 I->getParent()->getInstList().erase(I);
Chris Lattner800aaaf2003-10-07 15:17:02 +00002960 removeFromWorkList(I);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002961 continue;
Chris Lattner99f48c62002-09-02 04:59:56 +00002962 }
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002963
Chris Lattner57c67b02004-03-25 22:59:29 +00002964 // Check to see if any of the operands of this instruction are a
2965 // ConstantPointerRef. Since they sneak in all over the place and inhibit
2966 // optimization, we want to strip them out unconditionally!
2967 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
2968 if (ConstantPointerRef *CPR =
2969 dyn_cast<ConstantPointerRef>(I->getOperand(i))) {
2970 I->setOperand(i, CPR->getValue());
2971 Changed = true;
2972 }
2973
Chris Lattnerca081252001-12-14 16:52:21 +00002974 // Now that we have an instruction, try combining it to simplify it...
Chris Lattnerae7a0d32002-08-02 19:29:35 +00002975 if (Instruction *Result = visit(*I)) {
Chris Lattner0b18c1d2002-05-10 15:38:35 +00002976 ++NumCombined;
Chris Lattner260ab202002-04-18 17:39:14 +00002977 // Should we replace the old instruction with a new one?
Chris Lattner053c0932002-05-14 15:24:07 +00002978 if (Result != I) {
Chris Lattner7d2a5392004-03-13 23:54:27 +00002979 DEBUG(std::cerr << "IC: Old = " << *I
2980 << " New = " << *Result);
2981
Chris Lattner053c0932002-05-14 15:24:07 +00002982 // Instructions can end up on the worklist more than once. Make sure
2983 // we do not process an instruction that has been deleted.
Chris Lattner99f48c62002-09-02 04:59:56 +00002984 removeFromWorkList(I);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002985
2986 // Move the name to the new instruction first...
2987 std::string OldName = I->getName(); I->setName("");
Chris Lattner950fc782003-10-07 22:58:41 +00002988 Result->setName(OldName);
Chris Lattnere8ed4ef2003-10-06 17:11:01 +00002989
2990 // Insert the new instruction into the basic block...
2991 BasicBlock *InstParent = I->getParent();
2992 InstParent->getInstList().insert(I, Result);
2993
2994 // Everything uses the new instruction now...
2995 I->replaceAllUsesWith(Result);
2996
2997 // Erase the old instruction.
2998 InstParent->getInstList().erase(I);
Chris Lattner113f4f42002-06-25 16:13:24 +00002999 } else {
Chris Lattner7d2a5392004-03-13 23:54:27 +00003000 DEBUG(std::cerr << "IC: MOD = " << *I);
3001
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003002 BasicBlock::iterator II = I;
3003
3004 // If the instruction was modified, it's possible that it is now dead.
3005 // if so, remove it.
3006 if (dceInstruction(II)) {
3007 // Instructions may end up in the worklist more than once. Erase them
3008 // all.
Chris Lattner99f48c62002-09-02 04:59:56 +00003009 removeFromWorkList(I);
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003010 Result = 0;
3011 }
Chris Lattner053c0932002-05-14 15:24:07 +00003012 }
Chris Lattner260ab202002-04-18 17:39:14 +00003013
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003014 if (Result) {
3015 WorkList.push_back(Result);
Chris Lattner51ea1272004-02-28 05:22:00 +00003016 AddUsersToWorkList(*Result);
Chris Lattnerae7a0d32002-08-02 19:29:35 +00003017 }
Chris Lattner260ab202002-04-18 17:39:14 +00003018 Changed = true;
Chris Lattnerca081252001-12-14 16:52:21 +00003019 }
3020 }
3021
Chris Lattner260ab202002-04-18 17:39:14 +00003022 return Changed;
Chris Lattner04805fa2002-02-26 21:46:54 +00003023}
3024
Chris Lattner8427bff2003-12-07 01:24:23 +00003025Pass *llvm::createInstructionCombiningPass() {
Chris Lattner260ab202002-04-18 17:39:14 +00003026 return new InstCombiner();
Chris Lattner04805fa2002-02-26 21:46:54 +00003027}
Brian Gaeke960707c2003-11-11 22:41:34 +00003028