blob: a4c071602491a5412b5dd42ee1f384c15aeb19a6 [file] [log] [blame]
Chris Lattner72bc70d2008-12-05 07:49:08 +00001//===- GVN.cpp - Eliminate redundant values and loads ---------------------===//
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Owen Anderson1ad2cb72007-07-24 17:55:58 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This pass performs global value numbering to eliminate fully redundant
11// instructions. It also performs simple dead load elimination.
12//
John Criswell090c0a22009-03-10 15:04:53 +000013// Note that this pass does the value numbering itself; it does not use the
Matthijs Kooijman845f5242008-06-05 07:55:49 +000014// ValueNumbering analysis passes.
15//
Owen Anderson1ad2cb72007-07-24 17:55:58 +000016//===----------------------------------------------------------------------===//
17
18#define DEBUG_TYPE "gvn"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000019#include "llvm/Transforms/Scalar.h"
Owen Anderson0cd32032007-07-25 19:57:03 +000020#include "llvm/BasicBlock.h"
Owen Anderson45537912007-07-26 18:26:51 +000021#include "llvm/Constants.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000022#include "llvm/DerivedTypes.h"
Chris Lattnera53cfd12009-12-28 21:28:46 +000023#include "llvm/GlobalVariable.h"
Owen Anderson45537912007-07-26 18:26:51 +000024#include "llvm/Function.h"
Devang Patelc64bc162009-03-06 02:59:27 +000025#include "llvm/IntrinsicInst.h"
Owen Andersond672ecb2009-07-03 00:17:18 +000026#include "llvm/LLVMContext.h"
Chris Lattnereed919b2009-09-21 05:57:11 +000027#include "llvm/Operator.h"
Owen Anderson45537912007-07-26 18:26:51 +000028#include "llvm/Value.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000029#include "llvm/ADT/DenseMap.h"
30#include "llvm/ADT/DepthFirstIterator.h"
Owen Anderson255dafc2008-12-15 02:03:00 +000031#include "llvm/ADT/PostOrderIterator.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000032#include "llvm/ADT/SmallPtrSet.h"
33#include "llvm/ADT/SmallVector.h"
34#include "llvm/ADT/Statistic.h"
Owen Andersonb388ca92007-10-18 19:39:33 +000035#include "llvm/Analysis/AliasAnalysis.h"
Chris Lattnerbc9a28d2009-12-06 05:29:56 +000036#include "llvm/Analysis/ConstantFolding.h"
37#include "llvm/Analysis/Dominators.h"
Victor Hernandezf006b182009-10-27 20:05:49 +000038#include "llvm/Analysis/MemoryBuiltins.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000039#include "llvm/Analysis/MemoryDependenceAnalysis.h"
Chris Lattner05e15f82009-12-09 01:59:31 +000040#include "llvm/Analysis/PHITransAddr.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000041#include "llvm/Support/CFG.h"
Owen Andersonaa0b6342008-06-19 19:57:25 +000042#include "llvm/Support/CommandLine.h"
Chris Lattner9f8a6a72008-03-29 04:36:18 +000043#include "llvm/Support/Debug.h"
Torok Edwinc25e7582009-07-11 20:10:48 +000044#include "llvm/Support/ErrorHandling.h"
Chris Lattnereed919b2009-09-21 05:57:11 +000045#include "llvm/Support/GetElementPtrTypeIterator.h"
Chris Lattnerfaf815b2009-12-06 01:57:02 +000046#include "llvm/Support/IRBuilder.h"
Daniel Dunbarce63ffb2009-07-25 00:23:56 +000047#include "llvm/Support/raw_ostream.h"
Chris Lattnerbb6495c2009-09-20 19:03:47 +000048#include "llvm/Target/TargetData.h"
Owen Anderson5c274ee2008-06-19 19:54:19 +000049#include "llvm/Transforms/Utils/BasicBlockUtils.h"
Dale Johannesen42c3f552009-06-17 20:48:23 +000050#include "llvm/Transforms/Utils/Local.h"
Chris Lattnera09fbf02009-10-10 23:50:30 +000051#include "llvm/Transforms/Utils/SSAUpdater.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000052using namespace llvm;
53
Bill Wendling70ded192008-12-22 22:14:07 +000054STATISTIC(NumGVNInstr, "Number of instructions deleted");
55STATISTIC(NumGVNLoad, "Number of loads deleted");
56STATISTIC(NumGVNPRE, "Number of instructions PRE'd");
Owen Anderson961edc82008-07-15 16:28:06 +000057STATISTIC(NumGVNBlocks, "Number of blocks merged");
Bill Wendling70ded192008-12-22 22:14:07 +000058STATISTIC(NumPRELoad, "Number of loads PRE'd");
Chris Lattnerd27290d2008-03-22 04:13:49 +000059
Evan Cheng88d11c02008-06-20 01:01:07 +000060static cl::opt<bool> EnablePRE("enable-pre",
Owen Andersonc2b856e2008-07-17 19:41:00 +000061 cl::init(true), cl::Hidden);
Dan Gohmanc915c952009-06-15 18:30:15 +000062static cl::opt<bool> EnableLoadPRE("enable-load-pre", cl::init(true));
Bob Wilsonb29d7d22010-02-28 05:34:05 +000063static cl::opt<bool> EnableFullLoadPRE("enable-full-load-pre", cl::init(false));
Owen Andersonaa0b6342008-06-19 19:57:25 +000064
Owen Anderson1ad2cb72007-07-24 17:55:58 +000065//===----------------------------------------------------------------------===//
66// ValueTable Class
67//===----------------------------------------------------------------------===//
68
69/// This class holds the mapping between values and value numbers. It is used
70/// as an efficient mechanism to determine the expression-wise equivalence of
71/// two values.
72namespace {
Chris Lattner3e8b6632009-09-02 06:11:42 +000073 struct Expression {
Owen Andersona81e2412010-01-17 19:33:27 +000074 enum ExpressionOpcode {
75 ADD = Instruction::Add,
76 FADD = Instruction::FAdd,
77 SUB = Instruction::Sub,
78 FSUB = Instruction::FSub,
79 MUL = Instruction::Mul,
80 FMUL = Instruction::FMul,
81 UDIV = Instruction::UDiv,
82 SDIV = Instruction::SDiv,
83 FDIV = Instruction::FDiv,
84 UREM = Instruction::URem,
85 SREM = Instruction::SRem,
86 FREM = Instruction::FRem,
87 SHL = Instruction::Shl,
88 LSHR = Instruction::LShr,
89 ASHR = Instruction::AShr,
90 AND = Instruction::And,
91 OR = Instruction::Or,
92 XOR = Instruction::Xor,
93 TRUNC = Instruction::Trunc,
94 ZEXT = Instruction::ZExt,
95 SEXT = Instruction::SExt,
96 FPTOUI = Instruction::FPToUI,
97 FPTOSI = Instruction::FPToSI,
98 UITOFP = Instruction::UIToFP,
99 SITOFP = Instruction::SIToFP,
100 FPTRUNC = Instruction::FPTrunc,
101 FPEXT = Instruction::FPExt,
102 PTRTOINT = Instruction::PtrToInt,
103 INTTOPTR = Instruction::IntToPtr,
104 BITCAST = Instruction::BitCast,
105 ICMPEQ, ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
106 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
107 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
108 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
109 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
110 SHUFFLE, SELECT, GEP, CALL, CONSTANT,
111 INSERTVALUE, EXTRACTVALUE, EMPTY, TOMBSTONE };
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000112
113 ExpressionOpcode opcode;
114 const Type* type;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000115 SmallVector<uint32_t, 4> varargs;
Chris Lattnerb2412a82009-09-21 02:42:51 +0000116 Value *function;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000117
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000118 Expression() { }
119 Expression(ExpressionOpcode o) : opcode(o) { }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000120
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000121 bool operator==(const Expression &other) const {
122 if (opcode != other.opcode)
123 return false;
124 else if (opcode == EMPTY || opcode == TOMBSTONE)
125 return true;
126 else if (type != other.type)
127 return false;
Owen Andersonb388ca92007-10-18 19:39:33 +0000128 else if (function != other.function)
129 return false;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000130 else {
131 if (varargs.size() != other.varargs.size())
132 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000133
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000134 for (size_t i = 0; i < varargs.size(); ++i)
135 if (varargs[i] != other.varargs[i])
136 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000137
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000138 return true;
139 }
140 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000141
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000142 bool operator!=(const Expression &other) const {
Bill Wendling75f02ee2008-12-22 22:16:31 +0000143 return !(*this == other);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000144 }
145 };
Daniel Dunbara279bc32009-09-20 02:20:51 +0000146
Chris Lattner3e8b6632009-09-02 06:11:42 +0000147 class ValueTable {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000148 private:
149 DenseMap<Value*, uint32_t> valueNumbering;
150 DenseMap<Expression, uint32_t> expressionNumbering;
Owen Andersona472c4a2008-05-12 20:15:55 +0000151 AliasAnalysis* AA;
152 MemoryDependenceAnalysis* MD;
153 DominatorTree* DT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000154
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000155 uint32_t nextValueNumber;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000156
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000157 Expression::ExpressionOpcode getOpcode(CmpInst* C);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000158 Expression create_expression(BinaryOperator* BO);
159 Expression create_expression(CmpInst* C);
160 Expression create_expression(ShuffleVectorInst* V);
161 Expression create_expression(ExtractElementInst* C);
162 Expression create_expression(InsertElementInst* V);
163 Expression create_expression(SelectInst* V);
164 Expression create_expression(CastInst* C);
165 Expression create_expression(GetElementPtrInst* G);
Owen Andersonb388ca92007-10-18 19:39:33 +0000166 Expression create_expression(CallInst* C);
Owen Anderson3b3f58c2008-05-13 08:17:22 +0000167 Expression create_expression(Constant* C);
Owen Andersond41ed4e2009-10-19 22:14:22 +0000168 Expression create_expression(ExtractValueInst* C);
169 Expression create_expression(InsertValueInst* C);
170
171 uint32_t lookup_or_add_call(CallInst* C);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000172 public:
Dan Gohmane63c4a22009-04-01 16:37:47 +0000173 ValueTable() : nextValueNumber(1) { }
Chris Lattnerb2412a82009-09-21 02:42:51 +0000174 uint32_t lookup_or_add(Value *V);
175 uint32_t lookup(Value *V) const;
176 void add(Value *V, uint32_t num);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000177 void clear();
Chris Lattnerb2412a82009-09-21 02:42:51 +0000178 void erase(Value *v);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000179 unsigned size();
Owen Andersona472c4a2008-05-12 20:15:55 +0000180 void setAliasAnalysis(AliasAnalysis* A) { AA = A; }
Chris Lattner663e4412008-12-01 00:40:32 +0000181 AliasAnalysis *getAliasAnalysis() const { return AA; }
Owen Andersona472c4a2008-05-12 20:15:55 +0000182 void setMemDep(MemoryDependenceAnalysis* M) { MD = M; }
183 void setDomTree(DominatorTree* D) { DT = D; }
Owen Anderson0ae33ef2008-07-03 17:44:33 +0000184 uint32_t getNextUnusedValueNumber() { return nextValueNumber; }
Bill Wendling246dbbb2008-12-22 21:36:08 +0000185 void verifyRemoved(const Value *) const;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000186 };
187}
188
189namespace llvm {
Chris Lattner76c1b972007-09-17 18:34:04 +0000190template <> struct DenseMapInfo<Expression> {
Owen Anderson830db6a2007-08-02 18:16:06 +0000191 static inline Expression getEmptyKey() {
192 return Expression(Expression::EMPTY);
193 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000194
Owen Anderson830db6a2007-08-02 18:16:06 +0000195 static inline Expression getTombstoneKey() {
196 return Expression(Expression::TOMBSTONE);
197 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000198
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000199 static unsigned getHashValue(const Expression e) {
200 unsigned hash = e.opcode;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000201
Anton Korobeynikov07e6e562008-02-20 11:26:25 +0000202 hash = ((unsigned)((uintptr_t)e.type >> 4) ^
Owen Andersond41ed4e2009-10-19 22:14:22 +0000203 (unsigned)((uintptr_t)e.type >> 9));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000204
Owen Anderson830db6a2007-08-02 18:16:06 +0000205 for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
206 E = e.varargs.end(); I != E; ++I)
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000207 hash = *I + hash * 37;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000208
Anton Korobeynikov07e6e562008-02-20 11:26:25 +0000209 hash = ((unsigned)((uintptr_t)e.function >> 4) ^
210 (unsigned)((uintptr_t)e.function >> 9)) +
211 hash * 37;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000212
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000213 return hash;
214 }
Chris Lattner76c1b972007-09-17 18:34:04 +0000215 static bool isEqual(const Expression &LHS, const Expression &RHS) {
216 return LHS == RHS;
217 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000218};
Chris Lattner4bbf4ee2009-12-15 07:26:43 +0000219
220template <>
221struct isPodLike<Expression> { static const bool value = true; };
222
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000223}
224
225//===----------------------------------------------------------------------===//
226// ValueTable Internal Functions
227//===----------------------------------------------------------------------===//
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000228
229Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000230 if (isa<ICmpInst>(C)) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000231 switch (C->getPredicate()) {
Chris Lattner88365bb2008-03-21 21:14:38 +0000232 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000233 llvm_unreachable("Comparison with unknown predicate?");
Chris Lattner88365bb2008-03-21 21:14:38 +0000234 case ICmpInst::ICMP_EQ: return Expression::ICMPEQ;
235 case ICmpInst::ICMP_NE: return Expression::ICMPNE;
236 case ICmpInst::ICMP_UGT: return Expression::ICMPUGT;
237 case ICmpInst::ICMP_UGE: return Expression::ICMPUGE;
238 case ICmpInst::ICMP_ULT: return Expression::ICMPULT;
239 case ICmpInst::ICMP_ULE: return Expression::ICMPULE;
240 case ICmpInst::ICMP_SGT: return Expression::ICMPSGT;
241 case ICmpInst::ICMP_SGE: return Expression::ICMPSGE;
242 case ICmpInst::ICMP_SLT: return Expression::ICMPSLT;
243 case ICmpInst::ICMP_SLE: return Expression::ICMPSLE;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000244 }
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000245 } else {
246 switch (C->getPredicate()) {
247 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000248 llvm_unreachable("Comparison with unknown predicate?");
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000249 case FCmpInst::FCMP_OEQ: return Expression::FCMPOEQ;
250 case FCmpInst::FCMP_OGT: return Expression::FCMPOGT;
251 case FCmpInst::FCMP_OGE: return Expression::FCMPOGE;
252 case FCmpInst::FCMP_OLT: return Expression::FCMPOLT;
253 case FCmpInst::FCMP_OLE: return Expression::FCMPOLE;
254 case FCmpInst::FCMP_ONE: return Expression::FCMPONE;
255 case FCmpInst::FCMP_ORD: return Expression::FCMPORD;
256 case FCmpInst::FCMP_UNO: return Expression::FCMPUNO;
257 case FCmpInst::FCMP_UEQ: return Expression::FCMPUEQ;
258 case FCmpInst::FCMP_UGT: return Expression::FCMPUGT;
259 case FCmpInst::FCMP_UGE: return Expression::FCMPUGE;
260 case FCmpInst::FCMP_ULT: return Expression::FCMPULT;
261 case FCmpInst::FCMP_ULE: return Expression::FCMPULE;
262 case FCmpInst::FCMP_UNE: return Expression::FCMPUNE;
263 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000264 }
265}
266
Owen Andersonb388ca92007-10-18 19:39:33 +0000267Expression ValueTable::create_expression(CallInst* C) {
268 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000269
Owen Andersonb388ca92007-10-18 19:39:33 +0000270 e.type = C->getType();
Owen Andersonb388ca92007-10-18 19:39:33 +0000271 e.function = C->getCalledFunction();
272 e.opcode = Expression::CALL;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000273
Owen Andersonb388ca92007-10-18 19:39:33 +0000274 for (CallInst::op_iterator I = C->op_begin()+1, E = C->op_end();
275 I != E; ++I)
Owen Anderson8f46c782008-04-11 05:11:49 +0000276 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000277
Owen Andersonb388ca92007-10-18 19:39:33 +0000278 return e;
279}
280
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000281Expression ValueTable::create_expression(BinaryOperator* BO) {
282 Expression e;
Owen Andersond41ed4e2009-10-19 22:14:22 +0000283 e.varargs.push_back(lookup_or_add(BO->getOperand(0)));
284 e.varargs.push_back(lookup_or_add(BO->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000285 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000286 e.type = BO->getType();
Owen Andersona81e2412010-01-17 19:33:27 +0000287 e.opcode = static_cast<Expression::ExpressionOpcode>(BO->getOpcode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000288
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000289 return e;
290}
291
292Expression ValueTable::create_expression(CmpInst* C) {
293 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000294
Owen Andersond41ed4e2009-10-19 22:14:22 +0000295 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
296 e.varargs.push_back(lookup_or_add(C->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000297 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000298 e.type = C->getType();
299 e.opcode = getOpcode(C);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000300
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000301 return e;
302}
303
304Expression ValueTable::create_expression(CastInst* C) {
305 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000306
Owen Andersond41ed4e2009-10-19 22:14:22 +0000307 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000308 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000309 e.type = C->getType();
Owen Andersona81e2412010-01-17 19:33:27 +0000310 e.opcode = static_cast<Expression::ExpressionOpcode>(C->getOpcode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000311
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000312 return e;
313}
314
315Expression ValueTable::create_expression(ShuffleVectorInst* S) {
316 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000317
Owen Andersond41ed4e2009-10-19 22:14:22 +0000318 e.varargs.push_back(lookup_or_add(S->getOperand(0)));
319 e.varargs.push_back(lookup_or_add(S->getOperand(1)));
320 e.varargs.push_back(lookup_or_add(S->getOperand(2)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000321 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000322 e.type = S->getType();
323 e.opcode = Expression::SHUFFLE;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000324
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000325 return e;
326}
327
328Expression ValueTable::create_expression(ExtractElementInst* E) {
329 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000330
Owen Andersond41ed4e2009-10-19 22:14:22 +0000331 e.varargs.push_back(lookup_or_add(E->getOperand(0)));
332 e.varargs.push_back(lookup_or_add(E->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000333 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000334 e.type = E->getType();
335 e.opcode = Expression::EXTRACT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000336
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000337 return e;
338}
339
340Expression ValueTable::create_expression(InsertElementInst* I) {
341 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000342
Owen Andersond41ed4e2009-10-19 22:14:22 +0000343 e.varargs.push_back(lookup_or_add(I->getOperand(0)));
344 e.varargs.push_back(lookup_or_add(I->getOperand(1)));
345 e.varargs.push_back(lookup_or_add(I->getOperand(2)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000346 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000347 e.type = I->getType();
348 e.opcode = Expression::INSERT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000349
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000350 return e;
351}
352
353Expression ValueTable::create_expression(SelectInst* I) {
354 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000355
Owen Andersond41ed4e2009-10-19 22:14:22 +0000356 e.varargs.push_back(lookup_or_add(I->getCondition()));
357 e.varargs.push_back(lookup_or_add(I->getTrueValue()));
358 e.varargs.push_back(lookup_or_add(I->getFalseValue()));
Owen Andersonb388ca92007-10-18 19:39:33 +0000359 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000360 e.type = I->getType();
361 e.opcode = Expression::SELECT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000362
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000363 return e;
364}
365
366Expression ValueTable::create_expression(GetElementPtrInst* G) {
367 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000368
Owen Andersond41ed4e2009-10-19 22:14:22 +0000369 e.varargs.push_back(lookup_or_add(G->getPointerOperand()));
Owen Andersonb388ca92007-10-18 19:39:33 +0000370 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000371 e.type = G->getType();
372 e.opcode = Expression::GEP;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000373
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000374 for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
375 I != E; ++I)
Owen Anderson8f46c782008-04-11 05:11:49 +0000376 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000377
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000378 return e;
379}
380
Owen Andersond41ed4e2009-10-19 22:14:22 +0000381Expression ValueTable::create_expression(ExtractValueInst* E) {
382 Expression e;
383
384 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
385 for (ExtractValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
386 II != IE; ++II)
387 e.varargs.push_back(*II);
388 e.function = 0;
389 e.type = E->getType();
390 e.opcode = Expression::EXTRACTVALUE;
391
392 return e;
393}
394
395Expression ValueTable::create_expression(InsertValueInst* E) {
396 Expression e;
397
398 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
399 e.varargs.push_back(lookup_or_add(E->getInsertedValueOperand()));
400 for (InsertValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
401 II != IE; ++II)
402 e.varargs.push_back(*II);
403 e.function = 0;
404 e.type = E->getType();
405 e.opcode = Expression::INSERTVALUE;
406
407 return e;
408}
409
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000410//===----------------------------------------------------------------------===//
411// ValueTable External Functions
412//===----------------------------------------------------------------------===//
413
Owen Andersonb2303722008-06-18 21:41:49 +0000414/// add - Insert a value into the table with a specified value number.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000415void ValueTable::add(Value *V, uint32_t num) {
Owen Andersonb2303722008-06-18 21:41:49 +0000416 valueNumbering.insert(std::make_pair(V, num));
417}
418
Owen Andersond41ed4e2009-10-19 22:14:22 +0000419uint32_t ValueTable::lookup_or_add_call(CallInst* C) {
420 if (AA->doesNotAccessMemory(C)) {
421 Expression exp = create_expression(C);
422 uint32_t& e = expressionNumbering[exp];
423 if (!e) e = nextValueNumber++;
424 valueNumbering[C] = e;
425 return e;
426 } else if (AA->onlyReadsMemory(C)) {
427 Expression exp = create_expression(C);
428 uint32_t& e = expressionNumbering[exp];
429 if (!e) {
430 e = nextValueNumber++;
431 valueNumbering[C] = e;
432 return e;
433 }
Dan Gohman4ec01b22009-11-14 02:27:51 +0000434 if (!MD) {
435 e = nextValueNumber++;
436 valueNumbering[C] = e;
437 return e;
438 }
Owen Andersond41ed4e2009-10-19 22:14:22 +0000439
440 MemDepResult local_dep = MD->getDependency(C);
441
442 if (!local_dep.isDef() && !local_dep.isNonLocal()) {
443 valueNumbering[C] = nextValueNumber;
444 return nextValueNumber++;
445 }
446
447 if (local_dep.isDef()) {
448 CallInst* local_cdep = cast<CallInst>(local_dep.getInst());
449
450 if (local_cdep->getNumOperands() != C->getNumOperands()) {
451 valueNumbering[C] = nextValueNumber;
452 return nextValueNumber++;
453 }
454
455 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
456 uint32_t c_vn = lookup_or_add(C->getOperand(i));
457 uint32_t cd_vn = lookup_or_add(local_cdep->getOperand(i));
458 if (c_vn != cd_vn) {
459 valueNumbering[C] = nextValueNumber;
460 return nextValueNumber++;
461 }
462 }
463
464 uint32_t v = lookup_or_add(local_cdep);
465 valueNumbering[C] = v;
466 return v;
467 }
468
469 // Non-local case.
470 const MemoryDependenceAnalysis::NonLocalDepInfo &deps =
471 MD->getNonLocalCallDependency(CallSite(C));
472 // FIXME: call/call dependencies for readonly calls should return def, not
473 // clobber! Move the checking logic to MemDep!
474 CallInst* cdep = 0;
475
476 // Check to see if we have a single dominating call instruction that is
477 // identical to C.
478 for (unsigned i = 0, e = deps.size(); i != e; ++i) {
Chris Lattnere18b9712009-12-09 07:08:01 +0000479 const NonLocalDepEntry *I = &deps[i];
Owen Andersond41ed4e2009-10-19 22:14:22 +0000480 // Ignore non-local dependencies.
Chris Lattnere18b9712009-12-09 07:08:01 +0000481 if (I->getResult().isNonLocal())
Owen Andersond41ed4e2009-10-19 22:14:22 +0000482 continue;
483
484 // We don't handle non-depedencies. If we already have a call, reject
485 // instruction dependencies.
Chris Lattnere18b9712009-12-09 07:08:01 +0000486 if (I->getResult().isClobber() || cdep != 0) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000487 cdep = 0;
488 break;
489 }
490
Chris Lattnere18b9712009-12-09 07:08:01 +0000491 CallInst *NonLocalDepCall = dyn_cast<CallInst>(I->getResult().getInst());
Owen Andersond41ed4e2009-10-19 22:14:22 +0000492 // FIXME: All duplicated with non-local case.
Chris Lattnere18b9712009-12-09 07:08:01 +0000493 if (NonLocalDepCall && DT->properlyDominates(I->getBB(), C->getParent())){
Owen Andersond41ed4e2009-10-19 22:14:22 +0000494 cdep = NonLocalDepCall;
495 continue;
496 }
497
498 cdep = 0;
499 break;
500 }
501
502 if (!cdep) {
503 valueNumbering[C] = nextValueNumber;
504 return nextValueNumber++;
505 }
506
507 if (cdep->getNumOperands() != C->getNumOperands()) {
508 valueNumbering[C] = nextValueNumber;
509 return nextValueNumber++;
510 }
511 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
512 uint32_t c_vn = lookup_or_add(C->getOperand(i));
513 uint32_t cd_vn = lookup_or_add(cdep->getOperand(i));
514 if (c_vn != cd_vn) {
515 valueNumbering[C] = nextValueNumber;
516 return nextValueNumber++;
517 }
518 }
519
520 uint32_t v = lookup_or_add(cdep);
521 valueNumbering[C] = v;
522 return v;
523
524 } else {
525 valueNumbering[C] = nextValueNumber;
526 return nextValueNumber++;
527 }
528}
529
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000530/// lookup_or_add - Returns the value number for the specified value, assigning
531/// it a new number if it did not have one before.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000532uint32_t ValueTable::lookup_or_add(Value *V) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000533 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
534 if (VI != valueNumbering.end())
535 return VI->second;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000536
Owen Andersond41ed4e2009-10-19 22:14:22 +0000537 if (!isa<Instruction>(V)) {
Owen Anderson158d86e2009-10-19 21:14:57 +0000538 valueNumbering[V] = nextValueNumber;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000539 return nextValueNumber++;
540 }
Owen Andersond41ed4e2009-10-19 22:14:22 +0000541
542 Instruction* I = cast<Instruction>(V);
543 Expression exp;
544 switch (I->getOpcode()) {
545 case Instruction::Call:
546 return lookup_or_add_call(cast<CallInst>(I));
547 case Instruction::Add:
548 case Instruction::FAdd:
549 case Instruction::Sub:
550 case Instruction::FSub:
551 case Instruction::Mul:
552 case Instruction::FMul:
553 case Instruction::UDiv:
554 case Instruction::SDiv:
555 case Instruction::FDiv:
556 case Instruction::URem:
557 case Instruction::SRem:
558 case Instruction::FRem:
559 case Instruction::Shl:
560 case Instruction::LShr:
561 case Instruction::AShr:
562 case Instruction::And:
563 case Instruction::Or :
564 case Instruction::Xor:
565 exp = create_expression(cast<BinaryOperator>(I));
566 break;
567 case Instruction::ICmp:
568 case Instruction::FCmp:
569 exp = create_expression(cast<CmpInst>(I));
570 break;
571 case Instruction::Trunc:
572 case Instruction::ZExt:
573 case Instruction::SExt:
574 case Instruction::FPToUI:
575 case Instruction::FPToSI:
576 case Instruction::UIToFP:
577 case Instruction::SIToFP:
578 case Instruction::FPTrunc:
579 case Instruction::FPExt:
580 case Instruction::PtrToInt:
581 case Instruction::IntToPtr:
582 case Instruction::BitCast:
583 exp = create_expression(cast<CastInst>(I));
584 break;
585 case Instruction::Select:
586 exp = create_expression(cast<SelectInst>(I));
587 break;
588 case Instruction::ExtractElement:
589 exp = create_expression(cast<ExtractElementInst>(I));
590 break;
591 case Instruction::InsertElement:
592 exp = create_expression(cast<InsertElementInst>(I));
593 break;
594 case Instruction::ShuffleVector:
595 exp = create_expression(cast<ShuffleVectorInst>(I));
596 break;
597 case Instruction::ExtractValue:
598 exp = create_expression(cast<ExtractValueInst>(I));
599 break;
600 case Instruction::InsertValue:
601 exp = create_expression(cast<InsertValueInst>(I));
602 break;
603 case Instruction::GetElementPtr:
604 exp = create_expression(cast<GetElementPtrInst>(I));
605 break;
606 default:
607 valueNumbering[V] = nextValueNumber;
608 return nextValueNumber++;
609 }
610
611 uint32_t& e = expressionNumbering[exp];
612 if (!e) e = nextValueNumber++;
613 valueNumbering[V] = e;
614 return e;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000615}
616
617/// lookup - Returns the value number of the specified value. Fails if
618/// the value has not yet been numbered.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000619uint32_t ValueTable::lookup(Value *V) const {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000620 DenseMap<Value*, uint32_t>::const_iterator VI = valueNumbering.find(V);
Chris Lattner88365bb2008-03-21 21:14:38 +0000621 assert(VI != valueNumbering.end() && "Value not numbered?");
622 return VI->second;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000623}
624
625/// clear - Remove all entries from the ValueTable
626void ValueTable::clear() {
627 valueNumbering.clear();
628 expressionNumbering.clear();
629 nextValueNumber = 1;
630}
631
Owen Andersonbf7d0bc2007-07-31 23:27:13 +0000632/// erase - Remove a value from the value numbering
Chris Lattnerb2412a82009-09-21 02:42:51 +0000633void ValueTable::erase(Value *V) {
Owen Andersonbf7d0bc2007-07-31 23:27:13 +0000634 valueNumbering.erase(V);
635}
636
Bill Wendling246dbbb2008-12-22 21:36:08 +0000637/// verifyRemoved - Verify that the value is removed from all internal data
638/// structures.
639void ValueTable::verifyRemoved(const Value *V) const {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000640 for (DenseMap<Value*, uint32_t>::const_iterator
Bill Wendling246dbbb2008-12-22 21:36:08 +0000641 I = valueNumbering.begin(), E = valueNumbering.end(); I != E; ++I) {
642 assert(I->first != V && "Inst still occurs in value numbering map!");
643 }
644}
645
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000646//===----------------------------------------------------------------------===//
Bill Wendling30788b82008-12-22 22:32:22 +0000647// GVN Pass
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000648//===----------------------------------------------------------------------===//
649
650namespace {
Chris Lattner3e8b6632009-09-02 06:11:42 +0000651 struct ValueNumberScope {
Owen Anderson6fafe842008-06-20 01:15:47 +0000652 ValueNumberScope* parent;
653 DenseMap<uint32_t, Value*> table;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000654
Owen Anderson6fafe842008-06-20 01:15:47 +0000655 ValueNumberScope(ValueNumberScope* p) : parent(p) { }
656 };
657}
658
659namespace {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000660
Chris Lattner3e8b6632009-09-02 06:11:42 +0000661 class GVN : public FunctionPass {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000662 bool runOnFunction(Function &F);
663 public:
664 static char ID; // Pass identification, replacement for typeid
Bob Wilsonb29d7d22010-02-28 05:34:05 +0000665 explicit GVN(bool noloads = false)
666 : FunctionPass(&ID), NoLoads(noloads), MD(0) { }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000667
668 private:
Dan Gohman4ec01b22009-11-14 02:27:51 +0000669 bool NoLoads;
Chris Lattner663e4412008-12-01 00:40:32 +0000670 MemoryDependenceAnalysis *MD;
671 DominatorTree *DT;
672
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000673 ValueTable VN;
Owen Anderson6fafe842008-06-20 01:15:47 +0000674 DenseMap<BasicBlock*, ValueNumberScope*> localAvail;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000675
Bob Wilson484d4a32010-02-16 19:51:59 +0000676 // List of critical edges to be split between iterations.
677 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> toSplit;
678
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000679 // This transformation requires dominator postdominator info
680 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000681 AU.addRequired<DominatorTree>();
Dan Gohman4ec01b22009-11-14 02:27:51 +0000682 if (!NoLoads)
683 AU.addRequired<MemoryDependenceAnalysis>();
Owen Andersonb388ca92007-10-18 19:39:33 +0000684 AU.addRequired<AliasAnalysis>();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000685
Owen Andersonb70a5712008-06-23 17:49:45 +0000686 AU.addPreserved<DominatorTree>();
Owen Andersonb388ca92007-10-18 19:39:33 +0000687 AU.addPreserved<AliasAnalysis>();
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000688 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000689
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000690 // Helper fuctions
691 // FIXME: eliminate or document these better
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000692 bool processLoad(LoadInst* L,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000693 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000694 bool processInstruction(Instruction *I,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000695 SmallVectorImpl<Instruction*> &toErase);
Owen Anderson830db6a2007-08-02 18:16:06 +0000696 bool processNonLocalLoad(LoadInst* L,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000697 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000698 bool processBlock(BasicBlock *BB);
Owen Andersonb2303722008-06-18 21:41:49 +0000699 void dump(DenseMap<uint32_t, Value*>& d);
Owen Anderson3e75a422007-08-14 18:04:11 +0000700 bool iterateOnFunction(Function &F);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000701 Value *CollapsePhi(PHINode* p);
Owen Andersonb2303722008-06-18 21:41:49 +0000702 bool performPRE(Function& F);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000703 Value *lookupNumber(BasicBlock *BB, uint32_t num);
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +0000704 void cleanupGlobalSets();
Bill Wendling246dbbb2008-12-22 21:36:08 +0000705 void verifyRemoved(const Instruction *I) const;
Bob Wilson484d4a32010-02-16 19:51:59 +0000706 bool splitCriticalEdges();
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000707 };
Daniel Dunbara279bc32009-09-20 02:20:51 +0000708
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000709 char GVN::ID = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000710}
711
712// createGVNPass - The public interface to this file...
Bob Wilsonb29d7d22010-02-28 05:34:05 +0000713FunctionPass *llvm::createGVNPass(bool NoLoads) {
714 return new GVN(NoLoads);
Dan Gohman4ec01b22009-11-14 02:27:51 +0000715}
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000716
717static RegisterPass<GVN> X("gvn",
718 "Global Value Numbering");
719
Owen Andersonb2303722008-06-18 21:41:49 +0000720void GVN::dump(DenseMap<uint32_t, Value*>& d) {
Dan Gohmanad12b262009-12-18 03:25:51 +0000721 errs() << "{\n";
Owen Andersonb2303722008-06-18 21:41:49 +0000722 for (DenseMap<uint32_t, Value*>::iterator I = d.begin(),
Owen Anderson0cd32032007-07-25 19:57:03 +0000723 E = d.end(); I != E; ++I) {
Dan Gohmanad12b262009-12-18 03:25:51 +0000724 errs() << I->first << "\n";
Owen Anderson0cd32032007-07-25 19:57:03 +0000725 I->second->dump();
726 }
Dan Gohmanad12b262009-12-18 03:25:51 +0000727 errs() << "}\n";
Owen Anderson0cd32032007-07-25 19:57:03 +0000728}
729
Chris Lattnerb2412a82009-09-21 02:42:51 +0000730static bool isSafeReplacement(PHINode* p, Instruction *inst) {
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000731 if (!isa<PHINode>(inst))
732 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000733
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000734 for (Instruction::use_iterator UI = p->use_begin(), E = p->use_end();
735 UI != E; ++UI)
736 if (PHINode* use_phi = dyn_cast<PHINode>(UI))
737 if (use_phi->getParent() == inst->getParent())
738 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000739
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000740 return true;
741}
742
Chris Lattnerb2412a82009-09-21 02:42:51 +0000743Value *GVN::CollapsePhi(PHINode *PN) {
744 Value *ConstVal = PN->hasConstantValue(DT);
745 if (!ConstVal) return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000746
Chris Lattnerb2412a82009-09-21 02:42:51 +0000747 Instruction *Inst = dyn_cast<Instruction>(ConstVal);
748 if (!Inst)
749 return ConstVal;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000750
Chris Lattnerb2412a82009-09-21 02:42:51 +0000751 if (DT->dominates(Inst, PN))
752 if (isSafeReplacement(PN, Inst))
753 return Inst;
Owen Anderson1defe2d2007-08-16 22:51:56 +0000754 return 0;
755}
Owen Anderson0cd32032007-07-25 19:57:03 +0000756
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000757/// IsValueFullyAvailableInBlock - Return true if we can prove that the value
758/// we're analyzing is fully available in the specified block. As we go, keep
Chris Lattner72bc70d2008-12-05 07:49:08 +0000759/// track of which blocks we know are fully alive in FullyAvailableBlocks. This
760/// map is actually a tri-state map with the following values:
761/// 0) we know the block *is not* fully available.
762/// 1) we know the block *is* fully available.
763/// 2) we do not know whether the block is fully available or not, but we are
764/// currently speculating that it will be.
765/// 3) we are speculating for this block and have used that to speculate for
766/// other blocks.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000767static bool IsValueFullyAvailableInBlock(BasicBlock *BB,
Chris Lattner72bc70d2008-12-05 07:49:08 +0000768 DenseMap<BasicBlock*, char> &FullyAvailableBlocks) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000769 // Optimistically assume that the block is fully available and check to see
770 // if we already know about this block in one lookup.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000771 std::pair<DenseMap<BasicBlock*, char>::iterator, char> IV =
Chris Lattner72bc70d2008-12-05 07:49:08 +0000772 FullyAvailableBlocks.insert(std::make_pair(BB, 2));
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000773
774 // If the entry already existed for this block, return the precomputed value.
Chris Lattner72bc70d2008-12-05 07:49:08 +0000775 if (!IV.second) {
776 // If this is a speculative "available" value, mark it as being used for
777 // speculation of other blocks.
778 if (IV.first->second == 2)
779 IV.first->second = 3;
780 return IV.first->second != 0;
781 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000782
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000783 // Otherwise, see if it is fully available in all predecessors.
784 pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000785
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000786 // If this block has no predecessors, it isn't live-in here.
787 if (PI == PE)
Chris Lattner72bc70d2008-12-05 07:49:08 +0000788 goto SpeculationFailure;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000789
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000790 for (; PI != PE; ++PI)
791 // If the value isn't fully available in one of our predecessors, then it
792 // isn't fully available in this block either. Undo our previous
793 // optimistic assumption and bail out.
794 if (!IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
Chris Lattner72bc70d2008-12-05 07:49:08 +0000795 goto SpeculationFailure;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000796
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000797 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000798
Chris Lattner72bc70d2008-12-05 07:49:08 +0000799// SpeculationFailure - If we get here, we found out that this is not, after
800// all, a fully-available block. We have a problem if we speculated on this and
801// used the speculation to mark other blocks as available.
802SpeculationFailure:
803 char &BBVal = FullyAvailableBlocks[BB];
Daniel Dunbara279bc32009-09-20 02:20:51 +0000804
Chris Lattner72bc70d2008-12-05 07:49:08 +0000805 // If we didn't speculate on this, just return with it set to false.
806 if (BBVal == 2) {
807 BBVal = 0;
808 return false;
809 }
810
811 // If we did speculate on this value, we could have blocks set to 1 that are
812 // incorrect. Walk the (transitive) successors of this block and mark them as
813 // 0 if set to one.
814 SmallVector<BasicBlock*, 32> BBWorklist;
815 BBWorklist.push_back(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000816
Dan Gohman321a8132010-01-05 16:27:25 +0000817 do {
Chris Lattner72bc70d2008-12-05 07:49:08 +0000818 BasicBlock *Entry = BBWorklist.pop_back_val();
819 // Note that this sets blocks to 0 (unavailable) if they happen to not
820 // already be in FullyAvailableBlocks. This is safe.
821 char &EntryVal = FullyAvailableBlocks[Entry];
822 if (EntryVal == 0) continue; // Already unavailable.
823
824 // Mark as unavailable.
825 EntryVal = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000826
Chris Lattner72bc70d2008-12-05 07:49:08 +0000827 for (succ_iterator I = succ_begin(Entry), E = succ_end(Entry); I != E; ++I)
828 BBWorklist.push_back(*I);
Dan Gohman321a8132010-01-05 16:27:25 +0000829 } while (!BBWorklist.empty());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000830
Chris Lattner72bc70d2008-12-05 07:49:08 +0000831 return false;
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000832}
833
Chris Lattner771a5422009-09-20 20:09:34 +0000834
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000835/// CanCoerceMustAliasedValueToLoad - Return true if
836/// CoerceAvailableValueToLoadType will succeed.
837static bool CanCoerceMustAliasedValueToLoad(Value *StoredVal,
838 const Type *LoadTy,
839 const TargetData &TD) {
840 // If the loaded or stored value is an first class array or struct, don't try
841 // to transform them. We need to be able to bitcast to integer.
Duncan Sands1df98592010-02-16 11:11:14 +0000842 if (LoadTy->isStructTy() || LoadTy->isArrayTy() ||
843 StoredVal->getType()->isStructTy() ||
844 StoredVal->getType()->isArrayTy())
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000845 return false;
846
847 // The store has to be at least as big as the load.
848 if (TD.getTypeSizeInBits(StoredVal->getType()) <
849 TD.getTypeSizeInBits(LoadTy))
850 return false;
851
852 return true;
853}
854
855
Chris Lattner771a5422009-09-20 20:09:34 +0000856/// CoerceAvailableValueToLoadType - If we saw a store of a value to memory, and
857/// then a load from a must-aliased pointer of a different type, try to coerce
858/// the stored value. LoadedTy is the type of the load we want to replace and
859/// InsertPt is the place to insert new instructions.
860///
861/// If we can't do it, return null.
862static Value *CoerceAvailableValueToLoadType(Value *StoredVal,
863 const Type *LoadedTy,
864 Instruction *InsertPt,
865 const TargetData &TD) {
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000866 if (!CanCoerceMustAliasedValueToLoad(StoredVal, LoadedTy, TD))
867 return 0;
868
Chris Lattner771a5422009-09-20 20:09:34 +0000869 const Type *StoredValTy = StoredVal->getType();
870
871 uint64_t StoreSize = TD.getTypeSizeInBits(StoredValTy);
872 uint64_t LoadSize = TD.getTypeSizeInBits(LoadedTy);
873
874 // If the store and reload are the same size, we can always reuse it.
875 if (StoreSize == LoadSize) {
Duncan Sands1df98592010-02-16 11:11:14 +0000876 if (StoredValTy->isPointerTy() && LoadedTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000877 // Pointer to Pointer -> use bitcast.
878 return new BitCastInst(StoredVal, LoadedTy, "", InsertPt);
879 }
880
881 // Convert source pointers to integers, which can be bitcast.
Duncan Sands1df98592010-02-16 11:11:14 +0000882 if (StoredValTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000883 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
884 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
885 }
886
887 const Type *TypeToCastTo = LoadedTy;
Duncan Sands1df98592010-02-16 11:11:14 +0000888 if (TypeToCastTo->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000889 TypeToCastTo = TD.getIntPtrType(StoredValTy->getContext());
890
891 if (StoredValTy != TypeToCastTo)
892 StoredVal = new BitCastInst(StoredVal, TypeToCastTo, "", InsertPt);
893
894 // Cast to pointer if the load needs a pointer type.
Duncan Sands1df98592010-02-16 11:11:14 +0000895 if (LoadedTy->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000896 StoredVal = new IntToPtrInst(StoredVal, LoadedTy, "", InsertPt);
897
898 return StoredVal;
899 }
900
901 // If the loaded value is smaller than the available value, then we can
902 // extract out a piece from it. If the available value is too small, then we
903 // can't do anything.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000904 assert(StoreSize >= LoadSize && "CanCoerceMustAliasedValueToLoad fail");
Chris Lattner771a5422009-09-20 20:09:34 +0000905
906 // Convert source pointers to integers, which can be manipulated.
Duncan Sands1df98592010-02-16 11:11:14 +0000907 if (StoredValTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000908 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
909 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
910 }
911
912 // Convert vectors and fp to integer, which can be manipulated.
Duncan Sands1df98592010-02-16 11:11:14 +0000913 if (!StoredValTy->isIntegerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000914 StoredValTy = IntegerType::get(StoredValTy->getContext(), StoreSize);
915 StoredVal = new BitCastInst(StoredVal, StoredValTy, "", InsertPt);
916 }
917
918 // If this is a big-endian system, we need to shift the value down to the low
919 // bits so that a truncate will work.
920 if (TD.isBigEndian()) {
921 Constant *Val = ConstantInt::get(StoredVal->getType(), StoreSize-LoadSize);
922 StoredVal = BinaryOperator::CreateLShr(StoredVal, Val, "tmp", InsertPt);
923 }
924
925 // Truncate the integer to the right size now.
926 const Type *NewIntTy = IntegerType::get(StoredValTy->getContext(), LoadSize);
927 StoredVal = new TruncInst(StoredVal, NewIntTy, "trunc", InsertPt);
928
929 if (LoadedTy == NewIntTy)
930 return StoredVal;
931
932 // If the result is a pointer, inttoptr.
Duncan Sands1df98592010-02-16 11:11:14 +0000933 if (LoadedTy->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000934 return new IntToPtrInst(StoredVal, LoadedTy, "inttoptr", InsertPt);
935
936 // Otherwise, bitcast.
937 return new BitCastInst(StoredVal, LoadedTy, "bitcast", InsertPt);
938}
939
Chris Lattnerca749402009-09-21 06:24:16 +0000940/// GetBaseWithConstantOffset - Analyze the specified pointer to see if it can
941/// be expressed as a base pointer plus a constant offset. Return the base and
942/// offset to the caller.
943static Value *GetBaseWithConstantOffset(Value *Ptr, int64_t &Offset,
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000944 const TargetData &TD) {
Chris Lattnerca749402009-09-21 06:24:16 +0000945 Operator *PtrOp = dyn_cast<Operator>(Ptr);
946 if (PtrOp == 0) return Ptr;
947
948 // Just look through bitcasts.
949 if (PtrOp->getOpcode() == Instruction::BitCast)
950 return GetBaseWithConstantOffset(PtrOp->getOperand(0), Offset, TD);
951
952 // If this is a GEP with constant indices, we can look through it.
953 GEPOperator *GEP = dyn_cast<GEPOperator>(PtrOp);
954 if (GEP == 0 || !GEP->hasAllConstantIndices()) return Ptr;
955
956 gep_type_iterator GTI = gep_type_begin(GEP);
957 for (User::op_iterator I = GEP->idx_begin(), E = GEP->idx_end(); I != E;
958 ++I, ++GTI) {
959 ConstantInt *OpC = cast<ConstantInt>(*I);
960 if (OpC->isZero()) continue;
961
962 // Handle a struct and array indices which add their offset to the pointer.
963 if (const StructType *STy = dyn_cast<StructType>(*GTI)) {
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000964 Offset += TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue());
Chris Lattnerca749402009-09-21 06:24:16 +0000965 } else {
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000966 uint64_t Size = TD.getTypeAllocSize(GTI.getIndexedType());
Chris Lattnerca749402009-09-21 06:24:16 +0000967 Offset += OpC->getSExtValue()*Size;
968 }
969 }
970
971 // Re-sign extend from the pointer size if needed to get overflow edge cases
972 // right.
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000973 unsigned PtrSize = TD.getPointerSizeInBits();
Chris Lattnerca749402009-09-21 06:24:16 +0000974 if (PtrSize < 64)
975 Offset = (Offset << (64-PtrSize)) >> (64-PtrSize);
976
977 return GetBaseWithConstantOffset(GEP->getPointerOperand(), Offset, TD);
978}
979
980
Chris Lattnerfaf815b2009-12-06 01:57:02 +0000981/// AnalyzeLoadFromClobberingWrite - This function is called when we have a
982/// memdep query of a load that ends up being a clobbering memory write (store,
983/// memset, memcpy, memmove). This means that the write *may* provide bits used
984/// by the load but we can't be sure because the pointers don't mustalias.
985///
986/// Check this case to see if there is anything more we can do before we give
987/// up. This returns -1 if we have to give up, or a byte number in the stored
988/// value of the piece that feeds the load.
Chris Lattner03f17da2009-12-09 07:34:10 +0000989static int AnalyzeLoadFromClobberingWrite(const Type *LoadTy, Value *LoadPtr,
990 Value *WritePtr,
Chris Lattnerfaf815b2009-12-06 01:57:02 +0000991 uint64_t WriteSizeInBits,
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000992 const TargetData &TD) {
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000993 // If the loaded or stored value is an first class array or struct, don't try
994 // to transform them. We need to be able to bitcast to integer.
Duncan Sands1df98592010-02-16 11:11:14 +0000995 if (LoadTy->isStructTy() || LoadTy->isArrayTy())
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000996 return -1;
997
Chris Lattnerca749402009-09-21 06:24:16 +0000998 int64_t StoreOffset = 0, LoadOffset = 0;
Chris Lattnerfaf815b2009-12-06 01:57:02 +0000999 Value *StoreBase = GetBaseWithConstantOffset(WritePtr, StoreOffset, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001000 Value *LoadBase =
Chris Lattner03f17da2009-12-09 07:34:10 +00001001 GetBaseWithConstantOffset(LoadPtr, LoadOffset, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001002 if (StoreBase != LoadBase)
1003 return -1;
1004
1005 // If the load and store are to the exact same address, they should have been
1006 // a must alias. AA must have gotten confused.
1007 // FIXME: Study to see if/when this happens.
1008 if (LoadOffset == StoreOffset) {
1009#if 0
David Greenebf7f78e2010-01-05 01:27:17 +00001010 dbgs() << "STORE/LOAD DEP WITH COMMON POINTER MISSED:\n"
Chris Lattnerca749402009-09-21 06:24:16 +00001011 << "Base = " << *StoreBase << "\n"
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001012 << "Store Ptr = " << *WritePtr << "\n"
1013 << "Store Offs = " << StoreOffset << "\n"
Chris Lattnerb6760b42009-12-10 00:04:46 +00001014 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattnerb3f927f2009-12-09 02:41:54 +00001015 abort();
Chris Lattnerca749402009-09-21 06:24:16 +00001016#endif
1017 return -1;
1018 }
1019
1020 // If the load and store don't overlap at all, the store doesn't provide
1021 // anything to the load. In this case, they really don't alias at all, AA
1022 // must have gotten confused.
1023 // FIXME: Investigate cases where this bails out, e.g. rdar://7238614. Then
1024 // remove this check, as it is duplicated with what we have below.
Chris Lattner03f17da2009-12-09 07:34:10 +00001025 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy);
Chris Lattnerca749402009-09-21 06:24:16 +00001026
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001027 if ((WriteSizeInBits & 7) | (LoadSize & 7))
Chris Lattnerca749402009-09-21 06:24:16 +00001028 return -1;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001029 uint64_t StoreSize = WriteSizeInBits >> 3; // Convert to bytes.
Chris Lattnerca749402009-09-21 06:24:16 +00001030 LoadSize >>= 3;
1031
1032
1033 bool isAAFailure = false;
1034 if (StoreOffset < LoadOffset) {
1035 isAAFailure = StoreOffset+int64_t(StoreSize) <= LoadOffset;
1036 } else {
1037 isAAFailure = LoadOffset+int64_t(LoadSize) <= StoreOffset;
1038 }
1039 if (isAAFailure) {
1040#if 0
David Greenebf7f78e2010-01-05 01:27:17 +00001041 dbgs() << "STORE LOAD DEP WITH COMMON BASE:\n"
Chris Lattnerca749402009-09-21 06:24:16 +00001042 << "Base = " << *StoreBase << "\n"
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001043 << "Store Ptr = " << *WritePtr << "\n"
1044 << "Store Offs = " << StoreOffset << "\n"
Chris Lattnerb6760b42009-12-10 00:04:46 +00001045 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattnerb3f927f2009-12-09 02:41:54 +00001046 abort();
Chris Lattnerca749402009-09-21 06:24:16 +00001047#endif
1048 return -1;
1049 }
1050
1051 // If the Load isn't completely contained within the stored bits, we don't
1052 // have all the bits to feed it. We could do something crazy in the future
1053 // (issue a smaller load then merge the bits in) but this seems unlikely to be
1054 // valuable.
1055 if (StoreOffset > LoadOffset ||
1056 StoreOffset+StoreSize < LoadOffset+LoadSize)
1057 return -1;
1058
1059 // Okay, we can do this transformation. Return the number of bytes into the
1060 // store that the load is.
1061 return LoadOffset-StoreOffset;
1062}
1063
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001064/// AnalyzeLoadFromClobberingStore - This function is called when we have a
1065/// memdep query of a load that ends up being a clobbering store.
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001066static int AnalyzeLoadFromClobberingStore(const Type *LoadTy, Value *LoadPtr,
1067 StoreInst *DepSI,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001068 const TargetData &TD) {
1069 // Cannot handle reading from store of first-class aggregate yet.
Duncan Sands1df98592010-02-16 11:11:14 +00001070 if (DepSI->getOperand(0)->getType()->isStructTy() ||
1071 DepSI->getOperand(0)->getType()->isArrayTy())
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001072 return -1;
1073
1074 Value *StorePtr = DepSI->getPointerOperand();
Chris Lattner16f244e2009-12-10 00:11:45 +00001075 uint64_t StoreSize = TD.getTypeSizeInBits(DepSI->getOperand(0)->getType());
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001076 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
Chris Lattner03f17da2009-12-09 07:34:10 +00001077 StorePtr, StoreSize, TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001078}
1079
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001080static int AnalyzeLoadFromClobberingMemInst(const Type *LoadTy, Value *LoadPtr,
1081 MemIntrinsic *MI,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001082 const TargetData &TD) {
1083 // If the mem operation is a non-constant size, we can't handle it.
1084 ConstantInt *SizeCst = dyn_cast<ConstantInt>(MI->getLength());
1085 if (SizeCst == 0) return -1;
1086 uint64_t MemSizeInBits = SizeCst->getZExtValue()*8;
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001087
1088 // If this is memset, we just need to see if the offset is valid in the size
1089 // of the memset..
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001090 if (MI->getIntrinsicID() == Intrinsic::memset)
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001091 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr, MI->getDest(),
1092 MemSizeInBits, TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001093
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001094 // If we have a memcpy/memmove, the only case we can handle is if this is a
1095 // copy from constant memory. In that case, we can read directly from the
1096 // constant memory.
1097 MemTransferInst *MTI = cast<MemTransferInst>(MI);
1098
1099 Constant *Src = dyn_cast<Constant>(MTI->getSource());
1100 if (Src == 0) return -1;
1101
1102 GlobalVariable *GV = dyn_cast<GlobalVariable>(Src->getUnderlyingObject());
1103 if (GV == 0 || !GV->isConstant()) return -1;
1104
1105 // See if the access is within the bounds of the transfer.
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001106 int Offset = AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
1107 MI->getDest(), MemSizeInBits, TD);
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001108 if (Offset == -1)
1109 return Offset;
1110
1111 // Otherwise, see if we can constant fold a load from the constant with the
1112 // offset applied as appropriate.
1113 Src = ConstantExpr::getBitCast(Src,
1114 llvm::Type::getInt8PtrTy(Src->getContext()));
1115 Constant *OffsetCst =
1116 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1117 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001118 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001119 if (ConstantFoldLoadFromConstPtr(Src, &TD))
1120 return Offset;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001121 return -1;
1122}
1123
Chris Lattnerca749402009-09-21 06:24:16 +00001124
1125/// GetStoreValueForLoad - This function is called when we have a
1126/// memdep query of a load that ends up being a clobbering store. This means
1127/// that the store *may* provide bits used by the load but we can't be sure
1128/// because the pointers don't mustalias. Check this case to see if there is
1129/// anything more we can do before we give up.
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001130static Value *GetStoreValueForLoad(Value *SrcVal, unsigned Offset,
1131 const Type *LoadTy,
1132 Instruction *InsertPt, const TargetData &TD){
Chris Lattnerca749402009-09-21 06:24:16 +00001133 LLVMContext &Ctx = SrcVal->getType()->getContext();
1134
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001135 uint64_t StoreSize = TD.getTypeSizeInBits(SrcVal->getType())/8;
1136 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
Chris Lattnerca749402009-09-21 06:24:16 +00001137
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001138 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
Chris Lattnerca749402009-09-21 06:24:16 +00001139
1140 // Compute which bits of the stored value are being used by the load. Convert
1141 // to an integer type to start with.
Duncan Sands1df98592010-02-16 11:11:14 +00001142 if (SrcVal->getType()->isPointerTy())
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001143 SrcVal = Builder.CreatePtrToInt(SrcVal, TD.getIntPtrType(Ctx), "tmp");
Duncan Sands1df98592010-02-16 11:11:14 +00001144 if (!SrcVal->getType()->isIntegerTy())
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001145 SrcVal = Builder.CreateBitCast(SrcVal, IntegerType::get(Ctx, StoreSize*8),
1146 "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001147
1148 // Shift the bits to the least significant depending on endianness.
1149 unsigned ShiftAmt;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001150 if (TD.isLittleEndian())
Chris Lattnerca749402009-09-21 06:24:16 +00001151 ShiftAmt = Offset*8;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001152 else
Chris Lattner19ad7842009-09-21 17:55:47 +00001153 ShiftAmt = (StoreSize-LoadSize-Offset)*8;
Chris Lattnerca749402009-09-21 06:24:16 +00001154
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001155 if (ShiftAmt)
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001156 SrcVal = Builder.CreateLShr(SrcVal, ShiftAmt, "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001157
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001158 if (LoadSize != StoreSize)
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001159 SrcVal = Builder.CreateTrunc(SrcVal, IntegerType::get(Ctx, LoadSize*8),
1160 "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001161
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001162 return CoerceAvailableValueToLoadType(SrcVal, LoadTy, InsertPt, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001163}
1164
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001165/// GetMemInstValueForLoad - This function is called when we have a
1166/// memdep query of a load that ends up being a clobbering mem intrinsic.
1167static Value *GetMemInstValueForLoad(MemIntrinsic *SrcInst, unsigned Offset,
1168 const Type *LoadTy, Instruction *InsertPt,
1169 const TargetData &TD){
1170 LLVMContext &Ctx = LoadTy->getContext();
1171 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
1172
1173 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
1174
1175 // We know that this method is only called when the mem transfer fully
1176 // provides the bits for the load.
1177 if (MemSetInst *MSI = dyn_cast<MemSetInst>(SrcInst)) {
1178 // memset(P, 'x', 1234) -> splat('x'), even if x is a variable, and
1179 // independently of what the offset is.
1180 Value *Val = MSI->getValue();
1181 if (LoadSize != 1)
1182 Val = Builder.CreateZExt(Val, IntegerType::get(Ctx, LoadSize*8));
1183
1184 Value *OneElt = Val;
1185
1186 // Splat the value out to the right number of bits.
1187 for (unsigned NumBytesSet = 1; NumBytesSet != LoadSize; ) {
1188 // If we can double the number of bytes set, do it.
1189 if (NumBytesSet*2 <= LoadSize) {
1190 Value *ShVal = Builder.CreateShl(Val, NumBytesSet*8);
1191 Val = Builder.CreateOr(Val, ShVal);
1192 NumBytesSet <<= 1;
1193 continue;
1194 }
1195
1196 // Otherwise insert one byte at a time.
1197 Value *ShVal = Builder.CreateShl(Val, 1*8);
1198 Val = Builder.CreateOr(OneElt, ShVal);
1199 ++NumBytesSet;
1200 }
1201
1202 return CoerceAvailableValueToLoadType(Val, LoadTy, InsertPt, TD);
1203 }
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001204
1205 // Otherwise, this is a memcpy/memmove from a constant global.
1206 MemTransferInst *MTI = cast<MemTransferInst>(SrcInst);
1207 Constant *Src = cast<Constant>(MTI->getSource());
1208
1209 // Otherwise, see if we can constant fold a load from the constant with the
1210 // offset applied as appropriate.
1211 Src = ConstantExpr::getBitCast(Src,
1212 llvm::Type::getInt8PtrTy(Src->getContext()));
1213 Constant *OffsetCst =
1214 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1215 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
1216 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
1217 return ConstantFoldLoadFromConstPtr(Src, &TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001218}
1219
1220
1221
Chris Lattner87913512009-09-21 06:30:24 +00001222struct AvailableValueInBlock {
1223 /// BB - The basic block in question.
1224 BasicBlock *BB;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001225 enum ValType {
1226 SimpleVal, // A simple offsetted value that is accessed.
1227 MemIntrin // A memory intrinsic which is loaded from.
1228 };
1229
Chris Lattner87913512009-09-21 06:30:24 +00001230 /// V - The value that is live out of the block.
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001231 PointerIntPair<Value *, 1, ValType> Val;
1232
1233 /// Offset - The byte offset in Val that is interesting for the load query.
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001234 unsigned Offset;
Chris Lattner87913512009-09-21 06:30:24 +00001235
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001236 static AvailableValueInBlock get(BasicBlock *BB, Value *V,
1237 unsigned Offset = 0) {
Chris Lattner87913512009-09-21 06:30:24 +00001238 AvailableValueInBlock Res;
1239 Res.BB = BB;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001240 Res.Val.setPointer(V);
1241 Res.Val.setInt(SimpleVal);
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001242 Res.Offset = Offset;
Chris Lattner87913512009-09-21 06:30:24 +00001243 return Res;
1244 }
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001245
1246 static AvailableValueInBlock getMI(BasicBlock *BB, MemIntrinsic *MI,
1247 unsigned Offset = 0) {
1248 AvailableValueInBlock Res;
1249 Res.BB = BB;
1250 Res.Val.setPointer(MI);
1251 Res.Val.setInt(MemIntrin);
1252 Res.Offset = Offset;
1253 return Res;
1254 }
1255
1256 bool isSimpleValue() const { return Val.getInt() == SimpleVal; }
1257 Value *getSimpleValue() const {
1258 assert(isSimpleValue() && "Wrong accessor");
1259 return Val.getPointer();
1260 }
1261
1262 MemIntrinsic *getMemIntrinValue() const {
1263 assert(!isSimpleValue() && "Wrong accessor");
1264 return cast<MemIntrinsic>(Val.getPointer());
1265 }
Chris Lattner5362c542009-12-21 23:04:33 +00001266
1267 /// MaterializeAdjustedValue - Emit code into this block to adjust the value
1268 /// defined here to the specified type. This handles various coercion cases.
1269 Value *MaterializeAdjustedValue(const Type *LoadTy,
1270 const TargetData *TD) const {
1271 Value *Res;
1272 if (isSimpleValue()) {
1273 Res = getSimpleValue();
1274 if (Res->getType() != LoadTy) {
1275 assert(TD && "Need target data to handle type mismatch case");
1276 Res = GetStoreValueForLoad(Res, Offset, LoadTy, BB->getTerminator(),
1277 *TD);
1278
1279 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\nOffset: " << Offset << " "
1280 << *getSimpleValue() << '\n'
1281 << *Res << '\n' << "\n\n\n");
1282 }
1283 } else {
1284 Res = GetMemInstValueForLoad(getMemIntrinValue(), Offset,
1285 LoadTy, BB->getTerminator(), *TD);
1286 DEBUG(errs() << "GVN COERCED NONLOCAL MEM INTRIN:\nOffset: " << Offset
1287 << " " << *getMemIntrinValue() << '\n'
1288 << *Res << '\n' << "\n\n\n");
1289 }
1290 return Res;
1291 }
Chris Lattner87913512009-09-21 06:30:24 +00001292};
1293
Chris Lattnera09fbf02009-10-10 23:50:30 +00001294/// ConstructSSAForLoadSet - Given a set of loads specified by ValuesPerBlock,
1295/// construct SSA form, allowing us to eliminate LI. This returns the value
1296/// that should be used at LI's definition site.
1297static Value *ConstructSSAForLoadSet(LoadInst *LI,
1298 SmallVectorImpl<AvailableValueInBlock> &ValuesPerBlock,
1299 const TargetData *TD,
Chris Lattnerd2191e52009-12-21 23:15:48 +00001300 const DominatorTree &DT,
Chris Lattnera09fbf02009-10-10 23:50:30 +00001301 AliasAnalysis *AA) {
Chris Lattnerd2191e52009-12-21 23:15:48 +00001302 // Check for the fully redundant, dominating load case. In this case, we can
1303 // just use the dominating value directly.
1304 if (ValuesPerBlock.size() == 1 &&
1305 DT.properlyDominates(ValuesPerBlock[0].BB, LI->getParent()))
1306 return ValuesPerBlock[0].MaterializeAdjustedValue(LI->getType(), TD);
1307
1308 // Otherwise, we have to construct SSA form.
Chris Lattnera09fbf02009-10-10 23:50:30 +00001309 SmallVector<PHINode*, 8> NewPHIs;
1310 SSAUpdater SSAUpdate(&NewPHIs);
1311 SSAUpdate.Initialize(LI);
1312
1313 const Type *LoadTy = LI->getType();
1314
Chris Lattner771a5422009-09-20 20:09:34 +00001315 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001316 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1317 BasicBlock *BB = AV.BB;
Chris Lattner771a5422009-09-20 20:09:34 +00001318
Chris Lattnera09fbf02009-10-10 23:50:30 +00001319 if (SSAUpdate.HasValueForBlock(BB))
1320 continue;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001321
Chris Lattner5362c542009-12-21 23:04:33 +00001322 SSAUpdate.AddAvailableValue(BB, AV.MaterializeAdjustedValue(LoadTy, TD));
Chris Lattner771a5422009-09-20 20:09:34 +00001323 }
Chris Lattnera09fbf02009-10-10 23:50:30 +00001324
1325 // Perform PHI construction.
1326 Value *V = SSAUpdate.GetValueInMiddleOfBlock(LI->getParent());
1327
1328 // If new PHI nodes were created, notify alias analysis.
Duncan Sands1df98592010-02-16 11:11:14 +00001329 if (V->getType()->isPointerTy())
Chris Lattnera09fbf02009-10-10 23:50:30 +00001330 for (unsigned i = 0, e = NewPHIs.size(); i != e; ++i)
1331 AA->copyValue(LI, NewPHIs[i]);
1332
1333 return V;
Chris Lattner771a5422009-09-20 20:09:34 +00001334}
1335
Owen Anderson9ff5a232009-12-02 07:35:19 +00001336static bool isLifetimeStart(Instruction *Inst) {
Chris Lattner720e7902009-12-02 06:44:58 +00001337 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(Inst))
Owen Anderson9ff5a232009-12-02 07:35:19 +00001338 return II->getIntrinsicID() == Intrinsic::lifetime_start;
Chris Lattner720e7902009-12-02 06:44:58 +00001339 return false;
1340}
1341
Owen Anderson62bc33c2007-08-16 22:02:55 +00001342/// processNonLocalLoad - Attempt to eliminate a load whose dependencies are
1343/// non-local by performing PHI construction.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001344bool GVN::processNonLocalLoad(LoadInst *LI,
Chris Lattner8e1e95c2008-03-21 22:01:16 +00001345 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001346 // Find the non-local dependencies of the load.
Chris Lattner0ee443d2009-12-22 04:25:02 +00001347 SmallVector<NonLocalDepResult, 64> Deps;
Chris Lattner91bcf642008-12-09 19:25:07 +00001348 MD->getNonLocalPointerDependency(LI->getOperand(0), true, LI->getParent(),
1349 Deps);
David Greenebf7f78e2010-01-05 01:27:17 +00001350 //DEBUG(dbgs() << "INVESTIGATING NONLOCAL LOAD: "
Dan Gohman2a298992009-07-31 20:24:18 +00001351 // << Deps.size() << *LI << '\n');
Daniel Dunbara279bc32009-09-20 02:20:51 +00001352
Owen Anderson516eb1c2008-08-26 22:07:42 +00001353 // If we had to process more than one hundred blocks to find the
1354 // dependencies, this load isn't worth worrying about. Optimizing
1355 // it will be too expensive.
Chris Lattner91bcf642008-12-09 19:25:07 +00001356 if (Deps.size() > 100)
Owen Anderson516eb1c2008-08-26 22:07:42 +00001357 return false;
Chris Lattner5f4f84b2008-12-18 00:51:32 +00001358
1359 // If we had a phi translation failure, we'll have a single entry which is a
1360 // clobber in the current block. Reject this early.
Chris Lattnere18b9712009-12-09 07:08:01 +00001361 if (Deps.size() == 1 && Deps[0].getResult().isClobber()) {
Torok Edwin4306b1a2009-06-17 18:48:18 +00001362 DEBUG(
David Greenebf7f78e2010-01-05 01:27:17 +00001363 dbgs() << "GVN: non-local load ";
1364 WriteAsOperand(dbgs(), LI);
1365 dbgs() << " is clobbered by " << *Deps[0].getResult().getInst() << '\n';
Torok Edwin4306b1a2009-06-17 18:48:18 +00001366 );
Chris Lattner5f4f84b2008-12-18 00:51:32 +00001367 return false;
Torok Edwin4306b1a2009-06-17 18:48:18 +00001368 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001369
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001370 // Filter out useless results (non-locals, etc). Keep track of the blocks
1371 // where we have a value available in repl, also keep track of whether we see
1372 // dependencies that produce an unknown value for the load (such as a call
1373 // that could potentially clobber the load).
Chris Lattner87913512009-09-21 06:30:24 +00001374 SmallVector<AvailableValueInBlock, 16> ValuesPerBlock;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001375 SmallVector<BasicBlock*, 16> UnavailableBlocks;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001376
Chris Lattner771a5422009-09-20 20:09:34 +00001377 const TargetData *TD = 0;
1378
Chris Lattner91bcf642008-12-09 19:25:07 +00001379 for (unsigned i = 0, e = Deps.size(); i != e; ++i) {
Chris Lattnere18b9712009-12-09 07:08:01 +00001380 BasicBlock *DepBB = Deps[i].getBB();
1381 MemDepResult DepInfo = Deps[i].getResult();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001382
Chris Lattnerb51deb92008-12-05 21:04:20 +00001383 if (DepInfo.isClobber()) {
Chris Lattneraf064ae2009-12-09 18:21:46 +00001384 // The address being loaded in this non-local block may not be the same as
1385 // the pointer operand of the load if PHI translation occurs. Make sure
1386 // to consider the right address.
1387 Value *Address = Deps[i].getAddress();
1388
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001389 // If the dependence is to a store that writes to a superset of the bits
1390 // read by the load, we can extract the bits we need for the load from the
1391 // stored value.
1392 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInfo.getInst())) {
1393 if (TD == 0)
1394 TD = getAnalysisIfAvailable<TargetData>();
Chris Lattneraf064ae2009-12-09 18:21:46 +00001395 if (TD && Address) {
1396 int Offset = AnalyzeLoadFromClobberingStore(LI->getType(), Address,
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001397 DepSI, *TD);
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001398 if (Offset != -1) {
1399 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1400 DepSI->getOperand(0),
1401 Offset));
1402 continue;
1403 }
1404 }
1405 }
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001406
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001407 // If the clobbering value is a memset/memcpy/memmove, see if we can
1408 // forward a value on from it.
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001409 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(DepInfo.getInst())) {
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001410 if (TD == 0)
1411 TD = getAnalysisIfAvailable<TargetData>();
Chris Lattneraf064ae2009-12-09 18:21:46 +00001412 if (TD && Address) {
1413 int Offset = AnalyzeLoadFromClobberingMemInst(LI->getType(), Address,
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001414 DepMI, *TD);
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001415 if (Offset != -1) {
1416 ValuesPerBlock.push_back(AvailableValueInBlock::getMI(DepBB, DepMI,
1417 Offset));
1418 continue;
1419 }
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001420 }
1421 }
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001422
Chris Lattnerb51deb92008-12-05 21:04:20 +00001423 UnavailableBlocks.push_back(DepBB);
1424 continue;
1425 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001426
Chris Lattnerb51deb92008-12-05 21:04:20 +00001427 Instruction *DepInst = DepInfo.getInst();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001428
Chris Lattnerb51deb92008-12-05 21:04:20 +00001429 // Loading the allocation -> undef.
Chris Lattner720e7902009-12-02 06:44:58 +00001430 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst) ||
Owen Anderson9ff5a232009-12-02 07:35:19 +00001431 // Loading immediately after lifetime begin -> undef.
1432 isLifetimeStart(DepInst)) {
Chris Lattner87913512009-09-21 06:30:24 +00001433 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1434 UndefValue::get(LI->getType())));
Chris Lattnerbf145d62008-12-01 01:15:42 +00001435 continue;
1436 }
Owen Andersonb62f7922009-10-28 07:05:35 +00001437
Chris Lattner87913512009-09-21 06:30:24 +00001438 if (StoreInst *S = dyn_cast<StoreInst>(DepInst)) {
Daniel Dunbara279bc32009-09-20 02:20:51 +00001439 // Reject loads and stores that are to the same address but are of
Chris Lattner771a5422009-09-20 20:09:34 +00001440 // different types if we have to.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001441 if (S->getOperand(0)->getType() != LI->getType()) {
Chris Lattner771a5422009-09-20 20:09:34 +00001442 if (TD == 0)
1443 TD = getAnalysisIfAvailable<TargetData>();
1444
1445 // If the stored value is larger or equal to the loaded value, we can
1446 // reuse it.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001447 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(S->getOperand(0),
1448 LI->getType(), *TD)) {
Chris Lattner771a5422009-09-20 20:09:34 +00001449 UnavailableBlocks.push_back(DepBB);
1450 continue;
1451 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001452 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001453
Chris Lattner87913512009-09-21 06:30:24 +00001454 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1455 S->getOperand(0)));
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001456 continue;
1457 }
1458
1459 if (LoadInst *LD = dyn_cast<LoadInst>(DepInst)) {
Chris Lattner771a5422009-09-20 20:09:34 +00001460 // If the types mismatch and we can't handle it, reject reuse of the load.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001461 if (LD->getType() != LI->getType()) {
Chris Lattner771a5422009-09-20 20:09:34 +00001462 if (TD == 0)
1463 TD = getAnalysisIfAvailable<TargetData>();
1464
1465 // If the stored value is larger or equal to the loaded value, we can
1466 // reuse it.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001467 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(LD, LI->getType(),*TD)){
Chris Lattner771a5422009-09-20 20:09:34 +00001468 UnavailableBlocks.push_back(DepBB);
1469 continue;
1470 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001471 }
Chris Lattner87913512009-09-21 06:30:24 +00001472 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB, LD));
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001473 continue;
Owen Anderson0cd32032007-07-25 19:57:03 +00001474 }
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001475
1476 UnavailableBlocks.push_back(DepBB);
1477 continue;
Chris Lattner88365bb2008-03-21 21:14:38 +00001478 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001479
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001480 // If we have no predecessors that produce a known value for this load, exit
1481 // early.
1482 if (ValuesPerBlock.empty()) return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001483
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001484 // If all of the instructions we depend on produce a known value for this
1485 // load, then it is fully redundant and we can use PHI insertion to compute
1486 // its value. Insert PHIs and remove the fully redundant value now.
1487 if (UnavailableBlocks.empty()) {
David Greenebf7f78e2010-01-05 01:27:17 +00001488 DEBUG(dbgs() << "GVN REMOVING NONLOCAL LOAD: " << *LI << '\n');
Chris Lattner771a5422009-09-20 20:09:34 +00001489
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001490 // Perform PHI construction.
Chris Lattnerd2191e52009-12-21 23:15:48 +00001491 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnera09fbf02009-10-10 23:50:30 +00001492 VN.getAliasAnalysis());
Chris Lattner771a5422009-09-20 20:09:34 +00001493 LI->replaceAllUsesWith(V);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001494
Chris Lattner771a5422009-09-20 20:09:34 +00001495 if (isa<PHINode>(V))
1496 V->takeName(LI);
Duncan Sands1df98592010-02-16 11:11:14 +00001497 if (V->getType()->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +00001498 MD->invalidateCachedPointerInfo(V);
Bob Wilson74175c22010-02-22 21:39:41 +00001499 VN.erase(LI);
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001500 toErase.push_back(LI);
1501 NumGVNLoad++;
1502 return true;
1503 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001504
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001505 if (!EnablePRE || !EnableLoadPRE)
1506 return false;
1507
1508 // Okay, we have *some* definitions of the value. This means that the value
1509 // is available in some of our (transitive) predecessors. Lets think about
1510 // doing PRE of this load. This will involve inserting a new load into the
1511 // predecessor when it's not available. We could do this in general, but
1512 // prefer to not increase code size. As such, we only do this when we know
1513 // that we only have to insert *one* load (which means we're basically moving
1514 // the load, not inserting a new one).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001515
Owen Anderson88554df2009-05-31 09:03:40 +00001516 SmallPtrSet<BasicBlock *, 4> Blockers;
1517 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1518 Blockers.insert(UnavailableBlocks[i]);
1519
1520 // Lets find first basic block with more than one predecessor. Walk backwards
1521 // through predecessors if needed.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001522 BasicBlock *LoadBB = LI->getParent();
Owen Anderson88554df2009-05-31 09:03:40 +00001523 BasicBlock *TmpBB = LoadBB;
1524
1525 bool isSinglePred = false;
Dale Johannesen42c3f552009-06-17 20:48:23 +00001526 bool allSingleSucc = true;
Owen Anderson88554df2009-05-31 09:03:40 +00001527 while (TmpBB->getSinglePredecessor()) {
1528 isSinglePred = true;
1529 TmpBB = TmpBB->getSinglePredecessor();
Owen Anderson88554df2009-05-31 09:03:40 +00001530 if (TmpBB == LoadBB) // Infinite (unreachable) loop.
1531 return false;
1532 if (Blockers.count(TmpBB))
1533 return false;
Dale Johannesen42c3f552009-06-17 20:48:23 +00001534 if (TmpBB->getTerminator()->getNumSuccessors() != 1)
1535 allSingleSucc = false;
Owen Anderson88554df2009-05-31 09:03:40 +00001536 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001537
Owen Anderson88554df2009-05-31 09:03:40 +00001538 assert(TmpBB);
1539 LoadBB = TmpBB;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001540
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001541 // If we have a repl set with LI itself in it, this means we have a loop where
1542 // at least one of the values is LI. Since this means that we won't be able
1543 // to eliminate LI even if we insert uses in the other predecessors, we will
1544 // end up increasing code size. Reject this by scanning for LI.
Bob Wilson6cad4172010-02-01 21:17:14 +00001545 if (!EnableFullLoadPRE) {
1546 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
1547 if (ValuesPerBlock[i].isSimpleValue() &&
1548 ValuesPerBlock[i].getSimpleValue() == LI)
1549 return false;
1550 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001551
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001552 // FIXME: It is extremely unclear what this loop is doing, other than
1553 // artificially restricting loadpre.
Owen Anderson88554df2009-05-31 09:03:40 +00001554 if (isSinglePred) {
1555 bool isHot = false;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001556 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
1557 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1558 if (AV.isSimpleValue())
Daniel Dunbara279bc32009-09-20 02:20:51 +00001559 // "Hot" Instruction is in some loop (because it dominates its dep.
1560 // instruction).
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001561 if (Instruction *I = dyn_cast<Instruction>(AV.getSimpleValue()))
1562 if (DT->dominates(LI, I)) {
1563 isHot = true;
1564 break;
1565 }
1566 }
Owen Anderson88554df2009-05-31 09:03:40 +00001567
1568 // We are interested only in "hot" instructions. We don't want to do any
1569 // mis-optimizations here.
1570 if (!isHot)
1571 return false;
1572 }
1573
Bob Wilson6cad4172010-02-01 21:17:14 +00001574 // Check to see how many predecessors have the loaded value fully
1575 // available.
1576 DenseMap<BasicBlock*, Value*> PredLoads;
Chris Lattner72bc70d2008-12-05 07:49:08 +00001577 DenseMap<BasicBlock*, char> FullyAvailableBlocks;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001578 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner87913512009-09-21 06:30:24 +00001579 FullyAvailableBlocks[ValuesPerBlock[i].BB] = true;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001580 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1581 FullyAvailableBlocks[UnavailableBlocks[i]] = false;
1582
Bob Wilson70704972010-03-01 23:37:32 +00001583 bool NeedToSplitEdges = false;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001584 for (pred_iterator PI = pred_begin(LoadBB), E = pred_end(LoadBB);
1585 PI != E; ++PI) {
Bob Wilson6cad4172010-02-01 21:17:14 +00001586 BasicBlock *Pred = *PI;
1587 if (IsValueFullyAvailableInBlock(Pred, FullyAvailableBlocks)) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001588 continue;
Bob Wilson6cad4172010-02-01 21:17:14 +00001589 }
1590 PredLoads[Pred] = 0;
Bob Wilson484d4a32010-02-16 19:51:59 +00001591
Bob Wilson6cad4172010-02-01 21:17:14 +00001592 if (Pred->getTerminator()->getNumSuccessors() != 1) {
Bob Wilson484d4a32010-02-16 19:51:59 +00001593 if (isa<IndirectBrInst>(Pred->getTerminator())) {
1594 DEBUG(dbgs() << "COULD NOT PRE LOAD BECAUSE OF INDBR CRITICAL EDGE '"
1595 << Pred->getName() << "': " << *LI << '\n');
1596 return false;
1597 }
Bob Wilsonae23daf2010-02-16 21:06:42 +00001598 unsigned SuccNum = GetSuccessorNumber(Pred, LoadBB);
Bob Wilson484d4a32010-02-16 19:51:59 +00001599 toSplit.push_back(std::make_pair(Pred->getTerminator(), SuccNum));
Bob Wilson70704972010-03-01 23:37:32 +00001600 NeedToSplitEdges = true;
Bob Wilson6cad4172010-02-01 21:17:14 +00001601 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001602 }
Bob Wilson70704972010-03-01 23:37:32 +00001603 if (NeedToSplitEdges)
1604 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001605
Bob Wilson6cad4172010-02-01 21:17:14 +00001606 // Decide whether PRE is profitable for this load.
1607 unsigned NumUnavailablePreds = PredLoads.size();
1608 assert(NumUnavailablePreds != 0 &&
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001609 "Fully available value should be eliminated above!");
Bob Wilson6cad4172010-02-01 21:17:14 +00001610 if (!EnableFullLoadPRE) {
1611 // If this load is unavailable in multiple predecessors, reject it.
1612 // FIXME: If we could restructure the CFG, we could make a common pred with
1613 // all the preds that don't have an available LI and insert a new load into
1614 // that one block.
1615 if (NumUnavailablePreds != 1)
1616 return false;
Owen Andersona37226a2007-08-07 23:12:31 +00001617 }
Bob Wilson6cad4172010-02-01 21:17:14 +00001618
1619 // Check if the load can safely be moved to all the unavailable predecessors.
1620 bool CanDoPRE = true;
Chris Lattnerdd696052009-11-28 15:39:14 +00001621 SmallVector<Instruction*, 8> NewInsts;
Bob Wilson6cad4172010-02-01 21:17:14 +00001622 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1623 E = PredLoads.end(); I != E; ++I) {
1624 BasicBlock *UnavailablePred = I->first;
1625
1626 // Do PHI translation to get its value in the predecessor if necessary. The
1627 // returned pointer (if non-null) is guaranteed to dominate UnavailablePred.
1628
1629 // If all preds have a single successor, then we know it is safe to insert
1630 // the load on the pred (?!?), so we can insert code to materialize the
1631 // pointer if it is not available.
1632 PHITransAddr Address(LI->getOperand(0), TD);
1633 Value *LoadPtr = 0;
1634 if (allSingleSucc) {
1635 LoadPtr = Address.PHITranslateWithInsertion(LoadBB, UnavailablePred,
1636 *DT, NewInsts);
1637 } else {
Daniel Dunbar6d8f2ca2010-02-24 08:48:04 +00001638 Address.PHITranslateValue(LoadBB, UnavailablePred, DT);
Bob Wilson6cad4172010-02-01 21:17:14 +00001639 LoadPtr = Address.getAddr();
Bob Wilson6cad4172010-02-01 21:17:14 +00001640 }
1641
1642 // If we couldn't find or insert a computation of this phi translated value,
1643 // we fail PRE.
1644 if (LoadPtr == 0) {
1645 DEBUG(dbgs() << "COULDN'T INSERT PHI TRANSLATED VALUE OF: "
1646 << *LI->getOperand(0) << "\n");
1647 CanDoPRE = false;
1648 break;
1649 }
1650
1651 // Make sure it is valid to move this load here. We have to watch out for:
1652 // @1 = getelementptr (i8* p, ...
1653 // test p and branch if == 0
1654 // load @1
1655 // It is valid to have the getelementptr before the test, even if p can be 0,
1656 // as getelementptr only does address arithmetic.
1657 // If we are not pushing the value through any multiple-successor blocks
1658 // we do not have this case. Otherwise, check that the load is safe to
1659 // put anywhere; this can be improved, but should be conservatively safe.
1660 if (!allSingleSucc &&
1661 // FIXME: REEVALUTE THIS.
1662 !isSafeToLoadUnconditionally(LoadPtr,
1663 UnavailablePred->getTerminator(),
1664 LI->getAlignment(), TD)) {
1665 CanDoPRE = false;
1666 break;
1667 }
1668
1669 I->second = LoadPtr;
Chris Lattner05e15f82009-12-09 01:59:31 +00001670 }
1671
Bob Wilson6cad4172010-02-01 21:17:14 +00001672 if (!CanDoPRE) {
1673 while (!NewInsts.empty())
1674 NewInsts.pop_back_val()->eraseFromParent();
Dale Johannesen42c3f552009-06-17 20:48:23 +00001675 return false;
Chris Lattner0c264b12009-11-28 16:08:18 +00001676 }
Dale Johannesen42c3f552009-06-17 20:48:23 +00001677
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001678 // Okay, we can eliminate this load by inserting a reload in the predecessor
1679 // and using PHI construction to get the value in the other predecessors, do
1680 // it.
David Greenebf7f78e2010-01-05 01:27:17 +00001681 DEBUG(dbgs() << "GVN REMOVING PRE LOAD: " << *LI << '\n');
Chris Lattner0c264b12009-11-28 16:08:18 +00001682 DEBUG(if (!NewInsts.empty())
David Greenebf7f78e2010-01-05 01:27:17 +00001683 dbgs() << "INSERTED " << NewInsts.size() << " INSTS: "
Chris Lattner0c264b12009-11-28 16:08:18 +00001684 << *NewInsts.back() << '\n');
1685
Bob Wilson6cad4172010-02-01 21:17:14 +00001686 // Assign value numbers to the new instructions.
1687 for (unsigned i = 0, e = NewInsts.size(); i != e; ++i) {
1688 // FIXME: We really _ought_ to insert these value numbers into their
1689 // parent's availability map. However, in doing so, we risk getting into
1690 // ordering issues. If a block hasn't been processed yet, we would be
1691 // marking a value as AVAIL-IN, which isn't what we intend.
1692 VN.lookup_or_add(NewInsts[i]);
1693 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001694
Bob Wilson6cad4172010-02-01 21:17:14 +00001695 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1696 E = PredLoads.end(); I != E; ++I) {
1697 BasicBlock *UnavailablePred = I->first;
1698 Value *LoadPtr = I->second;
1699
1700 Value *NewLoad = new LoadInst(LoadPtr, LI->getName()+".pre", false,
1701 LI->getAlignment(),
1702 UnavailablePred->getTerminator());
1703
1704 // Add the newly created load.
1705 ValuesPerBlock.push_back(AvailableValueInBlock::get(UnavailablePred,
1706 NewLoad));
Bob Wilson188f4282010-02-23 05:55:00 +00001707 MD->invalidateCachedPointerInfo(LoadPtr);
1708 DEBUG(dbgs() << "GVN INSERTED " << *NewLoad << '\n');
Bob Wilson6cad4172010-02-01 21:17:14 +00001709 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001710
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001711 // Perform PHI construction.
Chris Lattnerd2191e52009-12-21 23:15:48 +00001712 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnera09fbf02009-10-10 23:50:30 +00001713 VN.getAliasAnalysis());
Chris Lattner771a5422009-09-20 20:09:34 +00001714 LI->replaceAllUsesWith(V);
1715 if (isa<PHINode>(V))
1716 V->takeName(LI);
Duncan Sands1df98592010-02-16 11:11:14 +00001717 if (V->getType()->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +00001718 MD->invalidateCachedPointerInfo(V);
Bob Wilson74175c22010-02-22 21:39:41 +00001719 VN.erase(LI);
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001720 toErase.push_back(LI);
1721 NumPRELoad++;
Owen Anderson0cd32032007-07-25 19:57:03 +00001722 return true;
1723}
1724
Owen Anderson62bc33c2007-08-16 22:02:55 +00001725/// processLoad - Attempt to eliminate a load, first by eliminating it
1726/// locally, and then attempting non-local elimination if that fails.
Chris Lattnerb51deb92008-12-05 21:04:20 +00001727bool GVN::processLoad(LoadInst *L, SmallVectorImpl<Instruction*> &toErase) {
Dan Gohman4ec01b22009-11-14 02:27:51 +00001728 if (!MD)
1729 return false;
1730
Chris Lattnerb51deb92008-12-05 21:04:20 +00001731 if (L->isVolatile())
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001732 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001733
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001734 // ... to a pointer that has been loaded from before...
Chris Lattnerb2412a82009-09-21 02:42:51 +00001735 MemDepResult Dep = MD->getDependency(L);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001736
Chris Lattnerb51deb92008-12-05 21:04:20 +00001737 // If the value isn't available, don't do anything!
Chris Lattnerb2412a82009-09-21 02:42:51 +00001738 if (Dep.isClobber()) {
Chris Lattnereed919b2009-09-21 05:57:11 +00001739 // Check to see if we have something like this:
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001740 // store i32 123, i32* %P
1741 // %A = bitcast i32* %P to i8*
1742 // %B = gep i8* %A, i32 1
1743 // %C = load i8* %B
1744 //
1745 // We could do that by recognizing if the clobber instructions are obviously
1746 // a common base + constant offset, and if the previous store (or memset)
1747 // completely covers this load. This sort of thing can happen in bitfield
1748 // access code.
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001749 Value *AvailVal = 0;
Chris Lattnereed919b2009-09-21 05:57:11 +00001750 if (StoreInst *DepSI = dyn_cast<StoreInst>(Dep.getInst()))
Chris Lattner1ce08292009-09-21 06:22:46 +00001751 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001752 int Offset = AnalyzeLoadFromClobberingStore(L->getType(),
1753 L->getPointerOperand(),
1754 DepSI, *TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001755 if (Offset != -1)
1756 AvailVal = GetStoreValueForLoad(DepSI->getOperand(0), Offset,
1757 L->getType(), L, *TD);
Chris Lattner1ce08292009-09-21 06:22:46 +00001758 }
Chris Lattnereed919b2009-09-21 05:57:11 +00001759
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001760 // If the clobbering value is a memset/memcpy/memmove, see if we can forward
1761 // a value on from it.
1762 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(Dep.getInst())) {
1763 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001764 int Offset = AnalyzeLoadFromClobberingMemInst(L->getType(),
1765 L->getPointerOperand(),
1766 DepMI, *TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001767 if (Offset != -1)
1768 AvailVal = GetMemInstValueForLoad(DepMI, Offset, L->getType(), L,*TD);
1769 }
1770 }
1771
1772 if (AvailVal) {
David Greenebf7f78e2010-01-05 01:27:17 +00001773 DEBUG(dbgs() << "GVN COERCED INST:\n" << *Dep.getInst() << '\n'
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001774 << *AvailVal << '\n' << *L << "\n\n\n");
1775
1776 // Replace the load!
1777 L->replaceAllUsesWith(AvailVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001778 if (AvailVal->getType()->isPointerTy())
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001779 MD->invalidateCachedPointerInfo(AvailVal);
Bob Wilson74175c22010-02-22 21:39:41 +00001780 VN.erase(L);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001781 toErase.push_back(L);
1782 NumGVNLoad++;
1783 return true;
1784 }
1785
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001786 DEBUG(
1787 // fast print dep, using operator<< on instruction would be too slow
David Greenebf7f78e2010-01-05 01:27:17 +00001788 dbgs() << "GVN: load ";
1789 WriteAsOperand(dbgs(), L);
Chris Lattnerb2412a82009-09-21 02:42:51 +00001790 Instruction *I = Dep.getInst();
David Greenebf7f78e2010-01-05 01:27:17 +00001791 dbgs() << " is clobbered by " << *I << '\n';
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001792 );
Chris Lattnerb51deb92008-12-05 21:04:20 +00001793 return false;
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001794 }
Chris Lattnerb51deb92008-12-05 21:04:20 +00001795
1796 // If it is defined in another block, try harder.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001797 if (Dep.isNonLocal())
Chris Lattnerb51deb92008-12-05 21:04:20 +00001798 return processNonLocalLoad(L, toErase);
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001799
Chris Lattnerb2412a82009-09-21 02:42:51 +00001800 Instruction *DepInst = Dep.getInst();
Chris Lattnerb51deb92008-12-05 21:04:20 +00001801 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInst)) {
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001802 Value *StoredVal = DepSI->getOperand(0);
1803
1804 // The store and load are to a must-aliased pointer, but they may not
1805 // actually have the same type. See if we know how to reuse the stored
1806 // value (depending on its type).
1807 const TargetData *TD = 0;
Chris Lattnera52fce42009-10-21 04:11:19 +00001808 if (StoredVal->getType() != L->getType()) {
1809 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1810 StoredVal = CoerceAvailableValueToLoadType(StoredVal, L->getType(),
1811 L, *TD);
1812 if (StoredVal == 0)
1813 return false;
1814
David Greenebf7f78e2010-01-05 01:27:17 +00001815 DEBUG(dbgs() << "GVN COERCED STORE:\n" << *DepSI << '\n' << *StoredVal
Chris Lattnera52fce42009-10-21 04:11:19 +00001816 << '\n' << *L << "\n\n\n");
1817 }
1818 else
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001819 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001820 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001821
Chris Lattnerb51deb92008-12-05 21:04:20 +00001822 // Remove it!
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001823 L->replaceAllUsesWith(StoredVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001824 if (StoredVal->getType()->isPointerTy())
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001825 MD->invalidateCachedPointerInfo(StoredVal);
Bob Wilson74175c22010-02-22 21:39:41 +00001826 VN.erase(L);
Chris Lattnerb51deb92008-12-05 21:04:20 +00001827 toErase.push_back(L);
1828 NumGVNLoad++;
1829 return true;
1830 }
1831
1832 if (LoadInst *DepLI = dyn_cast<LoadInst>(DepInst)) {
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001833 Value *AvailableVal = DepLI;
1834
1835 // The loads are of a must-aliased pointer, but they may not actually have
1836 // the same type. See if we know how to reuse the previously loaded value
1837 // (depending on its type).
1838 const TargetData *TD = 0;
Chris Lattnera52fce42009-10-21 04:11:19 +00001839 if (DepLI->getType() != L->getType()) {
1840 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1841 AvailableVal = CoerceAvailableValueToLoadType(DepLI, L->getType(), L,*TD);
1842 if (AvailableVal == 0)
1843 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001844
David Greenebf7f78e2010-01-05 01:27:17 +00001845 DEBUG(dbgs() << "GVN COERCED LOAD:\n" << *DepLI << "\n" << *AvailableVal
Chris Lattnera52fce42009-10-21 04:11:19 +00001846 << "\n" << *L << "\n\n\n");
1847 }
1848 else
1849 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001850 }
1851
Chris Lattnerb51deb92008-12-05 21:04:20 +00001852 // Remove it!
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001853 L->replaceAllUsesWith(AvailableVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001854 if (DepLI->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001855 MD->invalidateCachedPointerInfo(DepLI);
Bob Wilson74175c22010-02-22 21:39:41 +00001856 VN.erase(L);
Chris Lattnerb51deb92008-12-05 21:04:20 +00001857 toErase.push_back(L);
1858 NumGVNLoad++;
1859 return true;
1860 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001861
Chris Lattner237a8282008-11-30 01:39:32 +00001862 // If this load really doesn't depend on anything, then we must be loading an
1863 // undef value. This can happen when loading for a fresh allocation with no
1864 // intervening stores, for example.
Victor Hernandez7b929da2009-10-23 21:09:37 +00001865 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst)) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001866 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson74175c22010-02-22 21:39:41 +00001867 VN.erase(L);
Chris Lattner237a8282008-11-30 01:39:32 +00001868 toErase.push_back(L);
Chris Lattner237a8282008-11-30 01:39:32 +00001869 NumGVNLoad++;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001870 return true;
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001871 }
Owen Andersonb62f7922009-10-28 07:05:35 +00001872
Owen Anderson9ff5a232009-12-02 07:35:19 +00001873 // If this load occurs either right after a lifetime begin,
Owen Andersonb62f7922009-10-28 07:05:35 +00001874 // then the loaded value is undefined.
1875 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(DepInst)) {
Owen Anderson9ff5a232009-12-02 07:35:19 +00001876 if (II->getIntrinsicID() == Intrinsic::lifetime_start) {
Owen Andersonb62f7922009-10-28 07:05:35 +00001877 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson74175c22010-02-22 21:39:41 +00001878 VN.erase(L);
Owen Andersonb62f7922009-10-28 07:05:35 +00001879 toErase.push_back(L);
1880 NumGVNLoad++;
1881 return true;
1882 }
1883 }
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001884
Chris Lattnerb51deb92008-12-05 21:04:20 +00001885 return false;
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001886}
1887
Chris Lattnerb2412a82009-09-21 02:42:51 +00001888Value *GVN::lookupNumber(BasicBlock *BB, uint32_t num) {
Owen Andersonb70a5712008-06-23 17:49:45 +00001889 DenseMap<BasicBlock*, ValueNumberScope*>::iterator I = localAvail.find(BB);
1890 if (I == localAvail.end())
1891 return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001892
Chris Lattnerb2412a82009-09-21 02:42:51 +00001893 ValueNumberScope *Locals = I->second;
1894 while (Locals) {
1895 DenseMap<uint32_t, Value*>::iterator I = Locals->table.find(num);
1896 if (I != Locals->table.end())
Owen Anderson6fafe842008-06-20 01:15:47 +00001897 return I->second;
Chris Lattnerb2412a82009-09-21 02:42:51 +00001898 Locals = Locals->parent;
Owen Anderson6fafe842008-06-20 01:15:47 +00001899 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001900
Owen Anderson6fafe842008-06-20 01:15:47 +00001901 return 0;
1902}
1903
Owen Anderson255dafc2008-12-15 02:03:00 +00001904
Owen Anderson36057c72007-08-14 18:16:29 +00001905/// processInstruction - When calculating availability, handle an instruction
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001906/// by inserting it into the appropriate sets
Owen Andersonaf4240a2008-06-12 19:25:32 +00001907bool GVN::processInstruction(Instruction *I,
Chris Lattner8e1e95c2008-03-21 22:01:16 +00001908 SmallVectorImpl<Instruction*> &toErase) {
Devang Patelbe905e22010-02-11 00:20:49 +00001909 // Ignore dbg info intrinsics.
1910 if (isa<DbgInfoIntrinsic>(I))
1911 return false;
1912
Chris Lattnerb2412a82009-09-21 02:42:51 +00001913 if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
1914 bool Changed = processLoad(LI, toErase);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001915
Chris Lattnerb2412a82009-09-21 02:42:51 +00001916 if (!Changed) {
1917 unsigned Num = VN.lookup_or_add(LI);
1918 localAvail[I->getParent()]->table.insert(std::make_pair(Num, LI));
Owen Andersonb2303722008-06-18 21:41:49 +00001919 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001920
Chris Lattnerb2412a82009-09-21 02:42:51 +00001921 return Changed;
Owen Andersonb2303722008-06-18 21:41:49 +00001922 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001923
Chris Lattnerb2412a82009-09-21 02:42:51 +00001924 uint32_t NextNum = VN.getNextUnusedValueNumber();
1925 unsigned Num = VN.lookup_or_add(I);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001926
Chris Lattnerb2412a82009-09-21 02:42:51 +00001927 if (BranchInst *BI = dyn_cast<BranchInst>(I)) {
1928 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001929
Owen Andersone8a290f2009-04-01 23:53:49 +00001930 if (!BI->isConditional() || isa<Constant>(BI->getCondition()))
1931 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001932
Chris Lattnerb2412a82009-09-21 02:42:51 +00001933 Value *BranchCond = BI->getCondition();
1934 uint32_t CondVN = VN.lookup_or_add(BranchCond);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001935
Chris Lattnerb2412a82009-09-21 02:42:51 +00001936 BasicBlock *TrueSucc = BI->getSuccessor(0);
1937 BasicBlock *FalseSucc = BI->getSuccessor(1);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001938
Chris Lattnerb2412a82009-09-21 02:42:51 +00001939 if (TrueSucc->getSinglePredecessor())
1940 localAvail[TrueSucc]->table[CondVN] =
1941 ConstantInt::getTrue(TrueSucc->getContext());
1942 if (FalseSucc->getSinglePredecessor())
1943 localAvail[FalseSucc]->table[CondVN] =
1944 ConstantInt::getFalse(TrueSucc->getContext());
Owen Andersone8a290f2009-04-01 23:53:49 +00001945
1946 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001947
Owen Andersone5ffa902008-04-07 09:59:07 +00001948 // Allocations are always uniquely numbered, so we can save time and memory
Daniel Dunbara279bc32009-09-20 02:20:51 +00001949 // by fast failing them.
Victor Hernandez7b929da2009-10-23 21:09:37 +00001950 } else if (isa<AllocaInst>(I) || isa<TerminatorInst>(I)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001951 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Andersone5ffa902008-04-07 09:59:07 +00001952 return false;
Owen Andersonb2303722008-06-18 21:41:49 +00001953 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001954
Owen Anderson62bc33c2007-08-16 22:02:55 +00001955 // Collapse PHI nodes
Owen Anderson31f49672007-08-14 18:33:27 +00001956 if (PHINode* p = dyn_cast<PHINode>(I)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001957 Value *constVal = CollapsePhi(p);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001958
Owen Anderson31f49672007-08-14 18:33:27 +00001959 if (constVal) {
Owen Anderson1defe2d2007-08-16 22:51:56 +00001960 p->replaceAllUsesWith(constVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001961 if (MD && constVal->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001962 MD->invalidateCachedPointerInfo(constVal);
Owen Andersonae53c932008-12-23 00:49:51 +00001963 VN.erase(p);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001964
Owen Anderson1defe2d2007-08-16 22:51:56 +00001965 toErase.push_back(p);
Owen Andersonb2303722008-06-18 21:41:49 +00001966 } else {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001967 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson31f49672007-08-14 18:33:27 +00001968 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001969
Owen Anderson0ae33ef2008-07-03 17:44:33 +00001970 // If the number we were assigned was a brand new VN, then we don't
1971 // need to do a lookup to see if the number already exists
1972 // somewhere in the domtree: it can't!
Chris Lattnerb2412a82009-09-21 02:42:51 +00001973 } else if (Num == NextNum) {
1974 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001975
Owen Anderson255dafc2008-12-15 02:03:00 +00001976 // Perform fast-path value-number based elimination of values inherited from
1977 // dominators.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001978 } else if (Value *repl = lookupNumber(I->getParent(), Num)) {
Owen Anderson5fc4aba2007-12-08 01:37:09 +00001979 // Remove it!
Owen Andersonbf7d0bc2007-07-31 23:27:13 +00001980 VN.erase(I);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001981 I->replaceAllUsesWith(repl);
Duncan Sands1df98592010-02-16 11:11:14 +00001982 if (MD && repl->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001983 MD->invalidateCachedPointerInfo(repl);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001984 toErase.push_back(I);
1985 return true;
Owen Anderson255dafc2008-12-15 02:03:00 +00001986
Owen Anderson0ae33ef2008-07-03 17:44:33 +00001987 } else {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001988 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001989 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001990
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001991 return false;
1992}
1993
Bill Wendling30788b82008-12-22 22:32:22 +00001994/// runOnFunction - This is the main transformation entry point for a function.
Owen Anderson3e75a422007-08-14 18:04:11 +00001995bool GVN::runOnFunction(Function& F) {
Dan Gohman4ec01b22009-11-14 02:27:51 +00001996 if (!NoLoads)
1997 MD = &getAnalysis<MemoryDependenceAnalysis>();
Chris Lattner663e4412008-12-01 00:40:32 +00001998 DT = &getAnalysis<DominatorTree>();
Owen Andersona472c4a2008-05-12 20:15:55 +00001999 VN.setAliasAnalysis(&getAnalysis<AliasAnalysis>());
Chris Lattner663e4412008-12-01 00:40:32 +00002000 VN.setMemDep(MD);
2001 VN.setDomTree(DT);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002002
Chris Lattnerb2412a82009-09-21 02:42:51 +00002003 bool Changed = false;
2004 bool ShouldContinue = true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002005
Owen Anderson5d0af032008-07-16 17:52:31 +00002006 // Merge unconditional branches, allowing PRE to catch more
2007 // optimization opportunities.
2008 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002009 BasicBlock *BB = FI;
Owen Anderson5d0af032008-07-16 17:52:31 +00002010 ++FI;
Owen Andersonb31b06d2008-07-17 00:01:40 +00002011 bool removedBlock = MergeBlockIntoPredecessor(BB, this);
2012 if (removedBlock) NumGVNBlocks++;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002013
Chris Lattnerb2412a82009-09-21 02:42:51 +00002014 Changed |= removedBlock;
Owen Anderson5d0af032008-07-16 17:52:31 +00002015 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002016
Chris Lattnerae199312008-12-09 19:21:47 +00002017 unsigned Iteration = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002018
Chris Lattnerb2412a82009-09-21 02:42:51 +00002019 while (ShouldContinue) {
David Greenebf7f78e2010-01-05 01:27:17 +00002020 DEBUG(dbgs() << "GVN iteration: " << Iteration << "\n");
Chris Lattnerb2412a82009-09-21 02:42:51 +00002021 ShouldContinue = iterateOnFunction(F);
Bob Wilson484d4a32010-02-16 19:51:59 +00002022 if (splitCriticalEdges())
2023 ShouldContinue = true;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002024 Changed |= ShouldContinue;
Chris Lattnerae199312008-12-09 19:21:47 +00002025 ++Iteration;
Owen Anderson3e75a422007-08-14 18:04:11 +00002026 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002027
Owen Andersone98c54c2008-07-18 18:03:38 +00002028 if (EnablePRE) {
Owen Anderson0c7f91c2008-09-03 23:06:07 +00002029 bool PREChanged = true;
2030 while (PREChanged) {
2031 PREChanged = performPRE(F);
Chris Lattnerb2412a82009-09-21 02:42:51 +00002032 Changed |= PREChanged;
Owen Anderson0c7f91c2008-09-03 23:06:07 +00002033 }
Owen Andersone98c54c2008-07-18 18:03:38 +00002034 }
Chris Lattnerae199312008-12-09 19:21:47 +00002035 // FIXME: Should perform GVN again after PRE does something. PRE can move
2036 // computations into blocks where they become fully redundant. Note that
2037 // we can't do this until PRE's critical edge splitting updates memdep.
2038 // Actually, when this happens, we should just fully integrate PRE into GVN.
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002039
2040 cleanupGlobalSets();
2041
Chris Lattnerb2412a82009-09-21 02:42:51 +00002042 return Changed;
Owen Anderson3e75a422007-08-14 18:04:11 +00002043}
2044
2045
Chris Lattnerb2412a82009-09-21 02:42:51 +00002046bool GVN::processBlock(BasicBlock *BB) {
Chris Lattnerae199312008-12-09 19:21:47 +00002047 // FIXME: Kill off toErase by doing erasing eagerly in a helper function (and
2048 // incrementing BI before processing an instruction).
Owen Andersonaf4240a2008-06-12 19:25:32 +00002049 SmallVector<Instruction*, 8> toErase;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002050 bool ChangedFunction = false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002051
Owen Andersonaf4240a2008-06-12 19:25:32 +00002052 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
2053 BI != BE;) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002054 ChangedFunction |= processInstruction(BI, toErase);
Owen Andersonaf4240a2008-06-12 19:25:32 +00002055 if (toErase.empty()) {
2056 ++BI;
2057 continue;
2058 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002059
Owen Andersonaf4240a2008-06-12 19:25:32 +00002060 // If we need some instructions deleted, do it now.
2061 NumGVNInstr += toErase.size();
Daniel Dunbara279bc32009-09-20 02:20:51 +00002062
Owen Andersonaf4240a2008-06-12 19:25:32 +00002063 // Avoid iterator invalidation.
2064 bool AtStart = BI == BB->begin();
2065 if (!AtStart)
2066 --BI;
2067
2068 for (SmallVector<Instruction*, 4>::iterator I = toErase.begin(),
Chris Lattner663e4412008-12-01 00:40:32 +00002069 E = toErase.end(); I != E; ++I) {
David Greenebf7f78e2010-01-05 01:27:17 +00002070 DEBUG(dbgs() << "GVN removed: " << **I << '\n');
Dan Gohman4ec01b22009-11-14 02:27:51 +00002071 if (MD) MD->removeInstruction(*I);
Owen Andersonaf4240a2008-06-12 19:25:32 +00002072 (*I)->eraseFromParent();
Bill Wendlingec40d502008-12-22 21:57:30 +00002073 DEBUG(verifyRemoved(*I));
Chris Lattner663e4412008-12-01 00:40:32 +00002074 }
Chris Lattnerae199312008-12-09 19:21:47 +00002075 toErase.clear();
Owen Andersonaf4240a2008-06-12 19:25:32 +00002076
2077 if (AtStart)
2078 BI = BB->begin();
2079 else
2080 ++BI;
Owen Andersonaf4240a2008-06-12 19:25:32 +00002081 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002082
Chris Lattnerb2412a82009-09-21 02:42:51 +00002083 return ChangedFunction;
Owen Andersonaf4240a2008-06-12 19:25:32 +00002084}
2085
Owen Andersonb2303722008-06-18 21:41:49 +00002086/// performPRE - Perform a purely local form of PRE that looks for diamond
2087/// control flow patterns and attempts to perform simple PRE at the join point.
Chris Lattnerfb6e7012009-10-31 22:11:15 +00002088bool GVN::performPRE(Function &F) {
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002089 bool Changed = false;
Chris Lattner09713792008-12-01 07:29:03 +00002090 DenseMap<BasicBlock*, Value*> predMap;
Owen Andersonb2303722008-06-18 21:41:49 +00002091 for (df_iterator<BasicBlock*> DI = df_begin(&F.getEntryBlock()),
2092 DE = df_end(&F.getEntryBlock()); DI != DE; ++DI) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002093 BasicBlock *CurrentBlock = *DI;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002094
Owen Andersonb2303722008-06-18 21:41:49 +00002095 // Nothing to PRE in the entry block.
2096 if (CurrentBlock == &F.getEntryBlock()) continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002097
Owen Andersonb2303722008-06-18 21:41:49 +00002098 for (BasicBlock::iterator BI = CurrentBlock->begin(),
2099 BE = CurrentBlock->end(); BI != BE; ) {
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002100 Instruction *CurInst = BI++;
Duncan Sands7af1c782009-05-06 06:49:50 +00002101
Victor Hernandez7b929da2009-10-23 21:09:37 +00002102 if (isa<AllocaInst>(CurInst) ||
Victor Hernandez83d63912009-09-18 22:35:49 +00002103 isa<TerminatorInst>(CurInst) || isa<PHINode>(CurInst) ||
Devang Patel9674d152009-10-14 17:29:00 +00002104 CurInst->getType()->isVoidTy() ||
Duncan Sands7af1c782009-05-06 06:49:50 +00002105 CurInst->mayReadFromMemory() || CurInst->mayHaveSideEffects() ||
John Criswell090c0a22009-03-10 15:04:53 +00002106 isa<DbgInfoIntrinsic>(CurInst))
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002107 continue;
Duncan Sands7af1c782009-05-06 06:49:50 +00002108
Chris Lattnerb2412a82009-09-21 02:42:51 +00002109 uint32_t ValNo = VN.lookup(CurInst);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002110
Owen Andersonb2303722008-06-18 21:41:49 +00002111 // Look for the predecessors for PRE opportunities. We're
2112 // only trying to solve the basic diamond case, where
2113 // a value is computed in the successor and one predecessor,
2114 // but not the other. We also explicitly disallow cases
2115 // where the successor is its own predecessor, because they're
2116 // more complicated to get right.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002117 unsigned NumWith = 0;
2118 unsigned NumWithout = 0;
2119 BasicBlock *PREPred = 0;
Chris Lattner09713792008-12-01 07:29:03 +00002120 predMap.clear();
2121
Owen Andersonb2303722008-06-18 21:41:49 +00002122 for (pred_iterator PI = pred_begin(CurrentBlock),
2123 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
2124 // We're not interested in PRE where the block is its
Bob Wilsone7b635f2010-02-03 00:33:21 +00002125 // own predecessor, or in blocks with predecessors
Owen Anderson6fafe842008-06-20 01:15:47 +00002126 // that are not reachable.
2127 if (*PI == CurrentBlock) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002128 NumWithout = 2;
Owen Anderson6fafe842008-06-20 01:15:47 +00002129 break;
2130 } else if (!localAvail.count(*PI)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002131 NumWithout = 2;
Owen Anderson6fafe842008-06-20 01:15:47 +00002132 break;
2133 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002134
2135 DenseMap<uint32_t, Value*>::iterator predV =
Chris Lattnerb2412a82009-09-21 02:42:51 +00002136 localAvail[*PI]->table.find(ValNo);
Owen Anderson6fafe842008-06-20 01:15:47 +00002137 if (predV == localAvail[*PI]->table.end()) {
Owen Andersonb2303722008-06-18 21:41:49 +00002138 PREPred = *PI;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002139 NumWithout++;
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002140 } else if (predV->second == CurInst) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002141 NumWithout = 2;
Owen Andersonb2303722008-06-18 21:41:49 +00002142 } else {
Owen Anderson6fafe842008-06-20 01:15:47 +00002143 predMap[*PI] = predV->second;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002144 NumWith++;
Owen Andersonb2303722008-06-18 21:41:49 +00002145 }
2146 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002147
Owen Andersonb2303722008-06-18 21:41:49 +00002148 // Don't do PRE when it might increase code size, i.e. when
2149 // we would need to insert instructions in more than one pred.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002150 if (NumWithout != 1 || NumWith == 0)
Owen Andersonb2303722008-06-18 21:41:49 +00002151 continue;
Chris Lattnerfb6e7012009-10-31 22:11:15 +00002152
2153 // Don't do PRE across indirect branch.
2154 if (isa<IndirectBrInst>(PREPred->getTerminator()))
2155 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002156
Owen Anderson5c274ee2008-06-19 19:54:19 +00002157 // We can't do PRE safely on a critical edge, so instead we schedule
2158 // the edge to be split and perform the PRE the next time we iterate
2159 // on the function.
Bob Wilsonae23daf2010-02-16 21:06:42 +00002160 unsigned SuccNum = GetSuccessorNumber(PREPred, CurrentBlock);
Chris Lattnerb2412a82009-09-21 02:42:51 +00002161 if (isCriticalEdge(PREPred->getTerminator(), SuccNum)) {
2162 toSplit.push_back(std::make_pair(PREPred->getTerminator(), SuccNum));
Owen Anderson5c274ee2008-06-19 19:54:19 +00002163 continue;
2164 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002165
Bob Wilsone7b635f2010-02-03 00:33:21 +00002166 // Instantiate the expression in the predecessor that lacked it.
Owen Andersonb2303722008-06-18 21:41:49 +00002167 // Because we are going top-down through the block, all value numbers
2168 // will be available in the predecessor by the time we need them. Any
Bob Wilsone7b635f2010-02-03 00:33:21 +00002169 // that weren't originally present will have been instantiated earlier
Owen Andersonb2303722008-06-18 21:41:49 +00002170 // in this loop.
Nick Lewycky67760642009-09-27 07:38:41 +00002171 Instruction *PREInstr = CurInst->clone();
Owen Andersonb2303722008-06-18 21:41:49 +00002172 bool success = true;
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002173 for (unsigned i = 0, e = CurInst->getNumOperands(); i != e; ++i) {
2174 Value *Op = PREInstr->getOperand(i);
2175 if (isa<Argument>(Op) || isa<Constant>(Op) || isa<GlobalValue>(Op))
2176 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002177
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002178 if (Value *V = lookupNumber(PREPred, VN.lookup(Op))) {
2179 PREInstr->setOperand(i, V);
2180 } else {
2181 success = false;
2182 break;
Owen Andersonc45996b2008-07-11 20:05:13 +00002183 }
Owen Andersonb2303722008-06-18 21:41:49 +00002184 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002185
Owen Andersonb2303722008-06-18 21:41:49 +00002186 // Fail out if we encounter an operand that is not available in
Daniel Dunbara279bc32009-09-20 02:20:51 +00002187 // the PRE predecessor. This is typically because of loads which
Owen Andersonb2303722008-06-18 21:41:49 +00002188 // are not value numbered precisely.
2189 if (!success) {
2190 delete PREInstr;
Bill Wendling70ded192008-12-22 22:14:07 +00002191 DEBUG(verifyRemoved(PREInstr));
Owen Andersonb2303722008-06-18 21:41:49 +00002192 continue;
2193 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002194
Owen Andersonb2303722008-06-18 21:41:49 +00002195 PREInstr->insertBefore(PREPred->getTerminator());
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002196 PREInstr->setName(CurInst->getName() + ".pre");
Owen Anderson6fafe842008-06-20 01:15:47 +00002197 predMap[PREPred] = PREInstr;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002198 VN.add(PREInstr, ValNo);
Owen Andersonb2303722008-06-18 21:41:49 +00002199 NumGVNPRE++;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002200
Owen Andersonb2303722008-06-18 21:41:49 +00002201 // Update the availability map to include the new instruction.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002202 localAvail[PREPred]->table.insert(std::make_pair(ValNo, PREInstr));
Daniel Dunbara279bc32009-09-20 02:20:51 +00002203
Owen Andersonb2303722008-06-18 21:41:49 +00002204 // Create a PHI to make the value available in this block.
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002205 PHINode* Phi = PHINode::Create(CurInst->getType(),
2206 CurInst->getName() + ".pre-phi",
Owen Andersonb2303722008-06-18 21:41:49 +00002207 CurrentBlock->begin());
2208 for (pred_iterator PI = pred_begin(CurrentBlock),
2209 PE = pred_end(CurrentBlock); PI != PE; ++PI)
Owen Anderson6fafe842008-06-20 01:15:47 +00002210 Phi->addIncoming(predMap[*PI], *PI);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002211
Chris Lattnerb2412a82009-09-21 02:42:51 +00002212 VN.add(Phi, ValNo);
2213 localAvail[CurrentBlock]->table[ValNo] = Phi;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002214
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002215 CurInst->replaceAllUsesWith(Phi);
Duncan Sands1df98592010-02-16 11:11:14 +00002216 if (MD && Phi->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00002217 MD->invalidateCachedPointerInfo(Phi);
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002218 VN.erase(CurInst);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002219
David Greenebf7f78e2010-01-05 01:27:17 +00002220 DEBUG(dbgs() << "GVN PRE removed: " << *CurInst << '\n');
Dan Gohman4ec01b22009-11-14 02:27:51 +00002221 if (MD) MD->removeInstruction(CurInst);
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002222 CurInst->eraseFromParent();
Bill Wendlingec40d502008-12-22 21:57:30 +00002223 DEBUG(verifyRemoved(CurInst));
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002224 Changed = true;
Owen Andersonb2303722008-06-18 21:41:49 +00002225 }
2226 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002227
Bob Wilson484d4a32010-02-16 19:51:59 +00002228 if (splitCriticalEdges())
2229 Changed = true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002230
Bob Wilson484d4a32010-02-16 19:51:59 +00002231 return Changed;
2232}
2233
2234/// splitCriticalEdges - Split critical edges found during the previous
2235/// iteration that may enable further optimization.
2236bool GVN::splitCriticalEdges() {
2237 if (toSplit.empty())
2238 return false;
2239 do {
2240 std::pair<TerminatorInst*, unsigned> Edge = toSplit.pop_back_val();
2241 SplitCriticalEdge(Edge.first, Edge.second, this);
2242 } while (!toSplit.empty());
Evan Cheng19d417c2010-03-01 22:23:12 +00002243 if (MD) MD->invalidateCachedPredecessors();
Bob Wilson484d4a32010-02-16 19:51:59 +00002244 return true;
Owen Andersonb2303722008-06-18 21:41:49 +00002245}
2246
Bill Wendling30788b82008-12-22 22:32:22 +00002247/// iterateOnFunction - Executes one iteration of GVN
Owen Anderson3e75a422007-08-14 18:04:11 +00002248bool GVN::iterateOnFunction(Function &F) {
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002249 cleanupGlobalSets();
Chris Lattner2e607012008-03-21 21:33:23 +00002250
Owen Andersone8a290f2009-04-01 23:53:49 +00002251 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2252 DE = df_end(DT->getRootNode()); DI != DE; ++DI) {
2253 if (DI->getIDom())
2254 localAvail[DI->getBlock()] =
2255 new ValueNumberScope(localAvail[DI->getIDom()->getBlock()]);
2256 else
2257 localAvail[DI->getBlock()] = new ValueNumberScope(0);
2258 }
2259
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002260 // Top-down walk of the dominator tree
Chris Lattnerb2412a82009-09-21 02:42:51 +00002261 bool Changed = false;
Owen Andersonc34d1122008-12-15 03:52:17 +00002262#if 0
2263 // Needed for value numbering with phi construction to work.
Owen Anderson255dafc2008-12-15 02:03:00 +00002264 ReversePostOrderTraversal<Function*> RPOT(&F);
2265 for (ReversePostOrderTraversal<Function*>::rpo_iterator RI = RPOT.begin(),
2266 RE = RPOT.end(); RI != RE; ++RI)
Chris Lattnerb2412a82009-09-21 02:42:51 +00002267 Changed |= processBlock(*RI);
Owen Andersonc34d1122008-12-15 03:52:17 +00002268#else
2269 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2270 DE = df_end(DT->getRootNode()); DI != DE; ++DI)
Chris Lattnerb2412a82009-09-21 02:42:51 +00002271 Changed |= processBlock(DI->getBlock());
Owen Andersonc34d1122008-12-15 03:52:17 +00002272#endif
2273
Chris Lattnerb2412a82009-09-21 02:42:51 +00002274 return Changed;
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002275}
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002276
2277void GVN::cleanupGlobalSets() {
2278 VN.clear();
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002279
2280 for (DenseMap<BasicBlock*, ValueNumberScope*>::iterator
2281 I = localAvail.begin(), E = localAvail.end(); I != E; ++I)
2282 delete I->second;
2283 localAvail.clear();
2284}
Bill Wendling246dbbb2008-12-22 21:36:08 +00002285
2286/// verifyRemoved - Verify that the specified instruction does not occur in our
2287/// internal data structures.
Bill Wendling6d463f22008-12-22 22:28:56 +00002288void GVN::verifyRemoved(const Instruction *Inst) const {
2289 VN.verifyRemoved(Inst);
Bill Wendling70ded192008-12-22 22:14:07 +00002290
Bill Wendling6d463f22008-12-22 22:28:56 +00002291 // Walk through the value number scope to make sure the instruction isn't
2292 // ferreted away in it.
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +00002293 for (DenseMap<BasicBlock*, ValueNumberScope*>::const_iterator
Bill Wendling6d463f22008-12-22 22:28:56 +00002294 I = localAvail.begin(), E = localAvail.end(); I != E; ++I) {
2295 const ValueNumberScope *VNS = I->second;
2296
2297 while (VNS) {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +00002298 for (DenseMap<uint32_t, Value*>::const_iterator
Bill Wendling6d463f22008-12-22 22:28:56 +00002299 II = VNS->table.begin(), IE = VNS->table.end(); II != IE; ++II) {
2300 assert(II->second != Inst && "Inst still in value numbering scope!");
2301 }
2302
2303 VNS = VNS->parent;
Bill Wendling70ded192008-12-22 22:14:07 +00002304 }
2305 }
Bill Wendling246dbbb2008-12-22 21:36:08 +00002306}