blob: dc75342b4050302d98e463da3df5c5e3691cd30b [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"
Dan Gohmandd9344f2010-05-28 16:19:17 +000038#include "llvm/Analysis/Loads.h"
Victor Hernandezf006b182009-10-27 20:05:49 +000039#include "llvm/Analysis/MemoryBuiltins.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000040#include "llvm/Analysis/MemoryDependenceAnalysis.h"
Chris Lattner05e15f82009-12-09 01:59:31 +000041#include "llvm/Analysis/PHITransAddr.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000042#include "llvm/Support/CFG.h"
Owen Andersonaa0b6342008-06-19 19:57:25 +000043#include "llvm/Support/CommandLine.h"
Chris Lattner9f8a6a72008-03-29 04:36:18 +000044#include "llvm/Support/Debug.h"
Torok Edwinc25e7582009-07-11 20:10:48 +000045#include "llvm/Support/ErrorHandling.h"
Chris Lattnereed919b2009-09-21 05:57:11 +000046#include "llvm/Support/GetElementPtrTypeIterator.h"
Chris Lattnerfaf815b2009-12-06 01:57:02 +000047#include "llvm/Support/IRBuilder.h"
Daniel Dunbarce63ffb2009-07-25 00:23:56 +000048#include "llvm/Support/raw_ostream.h"
Chris Lattnerbb6495c2009-09-20 19:03:47 +000049#include "llvm/Target/TargetData.h"
Owen Anderson5c274ee2008-06-19 19:54:19 +000050#include "llvm/Transforms/Utils/BasicBlockUtils.h"
Dale Johannesen42c3f552009-06-17 20:48:23 +000051#include "llvm/Transforms/Utils/Local.h"
Chris Lattnera09fbf02009-10-10 23:50:30 +000052#include "llvm/Transforms/Utils/SSAUpdater.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000053using namespace llvm;
54
Bill Wendling70ded192008-12-22 22:14:07 +000055STATISTIC(NumGVNInstr, "Number of instructions deleted");
56STATISTIC(NumGVNLoad, "Number of loads deleted");
57STATISTIC(NumGVNPRE, "Number of instructions PRE'd");
Owen Anderson961edc82008-07-15 16:28:06 +000058STATISTIC(NumGVNBlocks, "Number of blocks merged");
Bill Wendling70ded192008-12-22 22:14:07 +000059STATISTIC(NumPRELoad, "Number of loads PRE'd");
Chris Lattnerd27290d2008-03-22 04:13:49 +000060
Evan Cheng88d11c02008-06-20 01:01:07 +000061static cl::opt<bool> EnablePRE("enable-pre",
Owen Andersonc2b856e2008-07-17 19:41:00 +000062 cl::init(true), cl::Hidden);
Dan Gohmanc915c952009-06-15 18:30:15 +000063static cl::opt<bool> EnableLoadPRE("enable-load-pre", cl::init(true));
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
Chris Lattner17aa6802010-09-04 18:12:00 +0000142 /*bool operator!=(const Expression &other) const {
Bill Wendling75f02ee2008-12-22 22:16:31 +0000143 return !(*this == other);
Chris Lattner17aa6802010-09-04 18:12:00 +0000144 }*/
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000145 };
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 Andersond41ed4e2009-10-19 22:14:22 +0000167 Expression create_expression(ExtractValueInst* C);
168 Expression create_expression(InsertValueInst* C);
169
170 uint32_t lookup_or_add_call(CallInst* C);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000171 public:
Dan Gohmane63c4a22009-04-01 16:37:47 +0000172 ValueTable() : nextValueNumber(1) { }
Chris Lattnerb2412a82009-09-21 02:42:51 +0000173 uint32_t lookup_or_add(Value *V);
174 uint32_t lookup(Value *V) const;
175 void add(Value *V, uint32_t num);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000176 void clear();
Chris Lattnerb2412a82009-09-21 02:42:51 +0000177 void erase(Value *v);
Owen Andersona472c4a2008-05-12 20:15:55 +0000178 void setAliasAnalysis(AliasAnalysis* A) { AA = A; }
Chris Lattner663e4412008-12-01 00:40:32 +0000179 AliasAnalysis *getAliasAnalysis() const { return AA; }
Owen Andersona472c4a2008-05-12 20:15:55 +0000180 void setMemDep(MemoryDependenceAnalysis* M) { MD = M; }
181 void setDomTree(DominatorTree* D) { DT = D; }
Owen Anderson0ae33ef2008-07-03 17:44:33 +0000182 uint32_t getNextUnusedValueNumber() { return nextValueNumber; }
Bill Wendling246dbbb2008-12-22 21:36:08 +0000183 void verifyRemoved(const Value *) const;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000184 };
185}
186
187namespace llvm {
Chris Lattner76c1b972007-09-17 18:34:04 +0000188template <> struct DenseMapInfo<Expression> {
Owen Anderson830db6a2007-08-02 18:16:06 +0000189 static inline Expression getEmptyKey() {
190 return Expression(Expression::EMPTY);
191 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000192
Owen Anderson830db6a2007-08-02 18:16:06 +0000193 static inline Expression getTombstoneKey() {
194 return Expression(Expression::TOMBSTONE);
195 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000196
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000197 static unsigned getHashValue(const Expression e) {
198 unsigned hash = e.opcode;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000199
Anton Korobeynikov07e6e562008-02-20 11:26:25 +0000200 hash = ((unsigned)((uintptr_t)e.type >> 4) ^
Owen Andersond41ed4e2009-10-19 22:14:22 +0000201 (unsigned)((uintptr_t)e.type >> 9));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000202
Owen Anderson830db6a2007-08-02 18:16:06 +0000203 for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
204 E = e.varargs.end(); I != E; ++I)
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000205 hash = *I + hash * 37;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000206
Anton Korobeynikov07e6e562008-02-20 11:26:25 +0000207 hash = ((unsigned)((uintptr_t)e.function >> 4) ^
208 (unsigned)((uintptr_t)e.function >> 9)) +
209 hash * 37;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000210
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000211 return hash;
212 }
Chris Lattner76c1b972007-09-17 18:34:04 +0000213 static bool isEqual(const Expression &LHS, const Expression &RHS) {
214 return LHS == RHS;
215 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000216};
Chris Lattner4bbf4ee2009-12-15 07:26:43 +0000217
218template <>
219struct isPodLike<Expression> { static const bool value = true; };
220
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000221}
222
223//===----------------------------------------------------------------------===//
224// ValueTable Internal Functions
225//===----------------------------------------------------------------------===//
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000226
227Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000228 if (isa<ICmpInst>(C)) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000229 switch (C->getPredicate()) {
Chris Lattner88365bb2008-03-21 21:14:38 +0000230 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000231 llvm_unreachable("Comparison with unknown predicate?");
Chris Lattner88365bb2008-03-21 21:14:38 +0000232 case ICmpInst::ICMP_EQ: return Expression::ICMPEQ;
233 case ICmpInst::ICMP_NE: return Expression::ICMPNE;
234 case ICmpInst::ICMP_UGT: return Expression::ICMPUGT;
235 case ICmpInst::ICMP_UGE: return Expression::ICMPUGE;
236 case ICmpInst::ICMP_ULT: return Expression::ICMPULT;
237 case ICmpInst::ICMP_ULE: return Expression::ICMPULE;
238 case ICmpInst::ICMP_SGT: return Expression::ICMPSGT;
239 case ICmpInst::ICMP_SGE: return Expression::ICMPSGE;
240 case ICmpInst::ICMP_SLT: return Expression::ICMPSLT;
241 case ICmpInst::ICMP_SLE: return Expression::ICMPSLE;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000242 }
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000243 } else {
244 switch (C->getPredicate()) {
245 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000246 llvm_unreachable("Comparison with unknown predicate?");
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000247 case FCmpInst::FCMP_OEQ: return Expression::FCMPOEQ;
248 case FCmpInst::FCMP_OGT: return Expression::FCMPOGT;
249 case FCmpInst::FCMP_OGE: return Expression::FCMPOGE;
250 case FCmpInst::FCMP_OLT: return Expression::FCMPOLT;
251 case FCmpInst::FCMP_OLE: return Expression::FCMPOLE;
252 case FCmpInst::FCMP_ONE: return Expression::FCMPONE;
253 case FCmpInst::FCMP_ORD: return Expression::FCMPORD;
254 case FCmpInst::FCMP_UNO: return Expression::FCMPUNO;
255 case FCmpInst::FCMP_UEQ: return Expression::FCMPUEQ;
256 case FCmpInst::FCMP_UGT: return Expression::FCMPUGT;
257 case FCmpInst::FCMP_UGE: return Expression::FCMPUGE;
258 case FCmpInst::FCMP_ULT: return Expression::FCMPULT;
259 case FCmpInst::FCMP_ULE: return Expression::FCMPULE;
260 case FCmpInst::FCMP_UNE: return Expression::FCMPUNE;
261 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000262 }
263}
264
Owen Andersonb388ca92007-10-18 19:39:33 +0000265Expression ValueTable::create_expression(CallInst* C) {
266 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000267
Owen Andersonb388ca92007-10-18 19:39:33 +0000268 e.type = C->getType();
Owen Andersonb388ca92007-10-18 19:39:33 +0000269 e.function = C->getCalledFunction();
270 e.opcode = Expression::CALL;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000271
Gabor Greif0a14be02010-06-24 10:04:07 +0000272 CallSite CS(C);
273 for (CallInst::op_iterator I = CS.arg_begin(), E = CS.arg_end();
Owen Andersonb388ca92007-10-18 19:39:33 +0000274 I != E; ++I)
Owen Anderson8f46c782008-04-11 05:11:49 +0000275 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000276
Owen Andersonb388ca92007-10-18 19:39:33 +0000277 return e;
278}
279
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000280Expression ValueTable::create_expression(BinaryOperator* BO) {
281 Expression e;
Owen Andersond41ed4e2009-10-19 22:14:22 +0000282 e.varargs.push_back(lookup_or_add(BO->getOperand(0)));
283 e.varargs.push_back(lookup_or_add(BO->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000284 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000285 e.type = BO->getType();
Owen Andersona81e2412010-01-17 19:33:27 +0000286 e.opcode = static_cast<Expression::ExpressionOpcode>(BO->getOpcode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000287
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000288 return e;
289}
290
291Expression ValueTable::create_expression(CmpInst* C) {
292 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000293
Owen Andersond41ed4e2009-10-19 22:14:22 +0000294 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
295 e.varargs.push_back(lookup_or_add(C->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000296 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000297 e.type = C->getType();
298 e.opcode = getOpcode(C);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000299
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000300 return e;
301}
302
303Expression ValueTable::create_expression(CastInst* C) {
304 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000305
Owen Andersond41ed4e2009-10-19 22:14:22 +0000306 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000307 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000308 e.type = C->getType();
Owen Andersona81e2412010-01-17 19:33:27 +0000309 e.opcode = static_cast<Expression::ExpressionOpcode>(C->getOpcode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000310
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000311 return e;
312}
313
314Expression ValueTable::create_expression(ShuffleVectorInst* S) {
315 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000316
Owen Andersond41ed4e2009-10-19 22:14:22 +0000317 e.varargs.push_back(lookup_or_add(S->getOperand(0)));
318 e.varargs.push_back(lookup_or_add(S->getOperand(1)));
319 e.varargs.push_back(lookup_or_add(S->getOperand(2)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000320 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000321 e.type = S->getType();
322 e.opcode = Expression::SHUFFLE;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000323
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000324 return e;
325}
326
327Expression ValueTable::create_expression(ExtractElementInst* E) {
328 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000329
Owen Andersond41ed4e2009-10-19 22:14:22 +0000330 e.varargs.push_back(lookup_or_add(E->getOperand(0)));
331 e.varargs.push_back(lookup_or_add(E->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000332 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000333 e.type = E->getType();
334 e.opcode = Expression::EXTRACT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000335
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000336 return e;
337}
338
339Expression ValueTable::create_expression(InsertElementInst* I) {
340 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000341
Owen Andersond41ed4e2009-10-19 22:14:22 +0000342 e.varargs.push_back(lookup_or_add(I->getOperand(0)));
343 e.varargs.push_back(lookup_or_add(I->getOperand(1)));
344 e.varargs.push_back(lookup_or_add(I->getOperand(2)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000345 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000346 e.type = I->getType();
347 e.opcode = Expression::INSERT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000348
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000349 return e;
350}
351
352Expression ValueTable::create_expression(SelectInst* I) {
353 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000354
Owen Andersond41ed4e2009-10-19 22:14:22 +0000355 e.varargs.push_back(lookup_or_add(I->getCondition()));
356 e.varargs.push_back(lookup_or_add(I->getTrueValue()));
357 e.varargs.push_back(lookup_or_add(I->getFalseValue()));
Owen Andersonb388ca92007-10-18 19:39:33 +0000358 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000359 e.type = I->getType();
360 e.opcode = Expression::SELECT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000361
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000362 return e;
363}
364
365Expression ValueTable::create_expression(GetElementPtrInst* G) {
366 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000367
Owen Andersond41ed4e2009-10-19 22:14:22 +0000368 e.varargs.push_back(lookup_or_add(G->getPointerOperand()));
Owen Andersonb388ca92007-10-18 19:39:33 +0000369 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000370 e.type = G->getType();
371 e.opcode = Expression::GEP;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000372
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000373 for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
374 I != E; ++I)
Owen Anderson8f46c782008-04-11 05:11:49 +0000375 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000376
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000377 return e;
378}
379
Owen Andersond41ed4e2009-10-19 22:14:22 +0000380Expression ValueTable::create_expression(ExtractValueInst* E) {
381 Expression e;
382
383 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
384 for (ExtractValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
385 II != IE; ++II)
386 e.varargs.push_back(*II);
387 e.function = 0;
388 e.type = E->getType();
389 e.opcode = Expression::EXTRACTVALUE;
390
391 return e;
392}
393
394Expression ValueTable::create_expression(InsertValueInst* E) {
395 Expression e;
396
397 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
398 e.varargs.push_back(lookup_or_add(E->getInsertedValueOperand()));
399 for (InsertValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
400 II != IE; ++II)
401 e.varargs.push_back(*II);
402 e.function = 0;
403 e.type = E->getType();
404 e.opcode = Expression::INSERTVALUE;
405
406 return e;
407}
408
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000409//===----------------------------------------------------------------------===//
410// ValueTable External Functions
411//===----------------------------------------------------------------------===//
412
Owen Andersonb2303722008-06-18 21:41:49 +0000413/// add - Insert a value into the table with a specified value number.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000414void ValueTable::add(Value *V, uint32_t num) {
Owen Andersonb2303722008-06-18 21:41:49 +0000415 valueNumbering.insert(std::make_pair(V, num));
416}
417
Owen Andersond41ed4e2009-10-19 22:14:22 +0000418uint32_t ValueTable::lookup_or_add_call(CallInst* C) {
419 if (AA->doesNotAccessMemory(C)) {
420 Expression exp = create_expression(C);
421 uint32_t& e = expressionNumbering[exp];
422 if (!e) e = nextValueNumber++;
423 valueNumbering[C] = e;
424 return e;
425 } else if (AA->onlyReadsMemory(C)) {
426 Expression exp = create_expression(C);
427 uint32_t& e = expressionNumbering[exp];
428 if (!e) {
429 e = nextValueNumber++;
430 valueNumbering[C] = e;
431 return e;
432 }
Dan Gohman4ec01b22009-11-14 02:27:51 +0000433 if (!MD) {
434 e = nextValueNumber++;
435 valueNumbering[C] = e;
436 return e;
437 }
Owen Andersond41ed4e2009-10-19 22:14:22 +0000438
439 MemDepResult local_dep = MD->getDependency(C);
440
441 if (!local_dep.isDef() && !local_dep.isNonLocal()) {
442 valueNumbering[C] = nextValueNumber;
443 return nextValueNumber++;
444 }
445
446 if (local_dep.isDef()) {
447 CallInst* local_cdep = cast<CallInst>(local_dep.getInst());
448
Gabor Greif237e1da2010-06-30 09:17:53 +0000449 if (local_cdep->getNumArgOperands() != C->getNumArgOperands()) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000450 valueNumbering[C] = nextValueNumber;
451 return nextValueNumber++;
452 }
453
Gabor Greifd883a9d2010-06-24 10:17:17 +0000454 for (unsigned i = 0, e = C->getNumArgOperands(); i < e; ++i) {
455 uint32_t c_vn = lookup_or_add(C->getArgOperand(i));
456 uint32_t cd_vn = lookup_or_add(local_cdep->getArgOperand(i));
Owen Andersond41ed4e2009-10-19 22:14:22 +0000457 if (c_vn != cd_vn) {
458 valueNumbering[C] = nextValueNumber;
459 return nextValueNumber++;
460 }
461 }
462
463 uint32_t v = lookup_or_add(local_cdep);
464 valueNumbering[C] = v;
465 return v;
466 }
467
468 // Non-local case.
469 const MemoryDependenceAnalysis::NonLocalDepInfo &deps =
470 MD->getNonLocalCallDependency(CallSite(C));
471 // FIXME: call/call dependencies for readonly calls should return def, not
472 // clobber! Move the checking logic to MemDep!
473 CallInst* cdep = 0;
474
475 // Check to see if we have a single dominating call instruction that is
476 // identical to C.
477 for (unsigned i = 0, e = deps.size(); i != e; ++i) {
Chris Lattnere18b9712009-12-09 07:08:01 +0000478 const NonLocalDepEntry *I = &deps[i];
Owen Andersond41ed4e2009-10-19 22:14:22 +0000479 // Ignore non-local dependencies.
Chris Lattnere18b9712009-12-09 07:08:01 +0000480 if (I->getResult().isNonLocal())
Owen Andersond41ed4e2009-10-19 22:14:22 +0000481 continue;
482
483 // We don't handle non-depedencies. If we already have a call, reject
484 // instruction dependencies.
Chris Lattnere18b9712009-12-09 07:08:01 +0000485 if (I->getResult().isClobber() || cdep != 0) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000486 cdep = 0;
487 break;
488 }
489
Chris Lattnere18b9712009-12-09 07:08:01 +0000490 CallInst *NonLocalDepCall = dyn_cast<CallInst>(I->getResult().getInst());
Owen Andersond41ed4e2009-10-19 22:14:22 +0000491 // FIXME: All duplicated with non-local case.
Chris Lattnere18b9712009-12-09 07:08:01 +0000492 if (NonLocalDepCall && DT->properlyDominates(I->getBB(), C->getParent())){
Owen Andersond41ed4e2009-10-19 22:14:22 +0000493 cdep = NonLocalDepCall;
494 continue;
495 }
496
497 cdep = 0;
498 break;
499 }
500
501 if (!cdep) {
502 valueNumbering[C] = nextValueNumber;
503 return nextValueNumber++;
504 }
505
Gabor Greif237e1da2010-06-30 09:17:53 +0000506 if (cdep->getNumArgOperands() != C->getNumArgOperands()) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000507 valueNumbering[C] = nextValueNumber;
508 return nextValueNumber++;
509 }
Gabor Greifd883a9d2010-06-24 10:17:17 +0000510 for (unsigned i = 0, e = C->getNumArgOperands(); i < e; ++i) {
511 uint32_t c_vn = lookup_or_add(C->getArgOperand(i));
512 uint32_t cd_vn = lookup_or_add(cdep->getArgOperand(i));
Owen Andersond41ed4e2009-10-19 22:14:22 +0000513 if (c_vn != cd_vn) {
514 valueNumbering[C] = nextValueNumber;
515 return nextValueNumber++;
516 }
517 }
518
519 uint32_t v = lookup_or_add(cdep);
520 valueNumbering[C] = v;
521 return v;
522
523 } else {
524 valueNumbering[C] = nextValueNumber;
525 return nextValueNumber++;
526 }
527}
528
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000529/// lookup_or_add - Returns the value number for the specified value, assigning
530/// it a new number if it did not have one before.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000531uint32_t ValueTable::lookup_or_add(Value *V) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000532 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
533 if (VI != valueNumbering.end())
534 return VI->second;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000535
Owen Andersond41ed4e2009-10-19 22:14:22 +0000536 if (!isa<Instruction>(V)) {
Owen Anderson158d86e2009-10-19 21:14:57 +0000537 valueNumbering[V] = nextValueNumber;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000538 return nextValueNumber++;
539 }
Owen Andersond41ed4e2009-10-19 22:14:22 +0000540
541 Instruction* I = cast<Instruction>(V);
542 Expression exp;
543 switch (I->getOpcode()) {
544 case Instruction::Call:
545 return lookup_or_add_call(cast<CallInst>(I));
546 case Instruction::Add:
547 case Instruction::FAdd:
548 case Instruction::Sub:
549 case Instruction::FSub:
550 case Instruction::Mul:
551 case Instruction::FMul:
552 case Instruction::UDiv:
553 case Instruction::SDiv:
554 case Instruction::FDiv:
555 case Instruction::URem:
556 case Instruction::SRem:
557 case Instruction::FRem:
558 case Instruction::Shl:
559 case Instruction::LShr:
560 case Instruction::AShr:
561 case Instruction::And:
562 case Instruction::Or :
563 case Instruction::Xor:
564 exp = create_expression(cast<BinaryOperator>(I));
565 break;
566 case Instruction::ICmp:
567 case Instruction::FCmp:
568 exp = create_expression(cast<CmpInst>(I));
569 break;
570 case Instruction::Trunc:
571 case Instruction::ZExt:
572 case Instruction::SExt:
573 case Instruction::FPToUI:
574 case Instruction::FPToSI:
575 case Instruction::UIToFP:
576 case Instruction::SIToFP:
577 case Instruction::FPTrunc:
578 case Instruction::FPExt:
579 case Instruction::PtrToInt:
580 case Instruction::IntToPtr:
581 case Instruction::BitCast:
582 exp = create_expression(cast<CastInst>(I));
583 break;
584 case Instruction::Select:
585 exp = create_expression(cast<SelectInst>(I));
586 break;
587 case Instruction::ExtractElement:
588 exp = create_expression(cast<ExtractElementInst>(I));
589 break;
590 case Instruction::InsertElement:
591 exp = create_expression(cast<InsertElementInst>(I));
592 break;
593 case Instruction::ShuffleVector:
594 exp = create_expression(cast<ShuffleVectorInst>(I));
595 break;
596 case Instruction::ExtractValue:
597 exp = create_expression(cast<ExtractValueInst>(I));
598 break;
599 case Instruction::InsertValue:
600 exp = create_expression(cast<InsertValueInst>(I));
601 break;
602 case Instruction::GetElementPtr:
603 exp = create_expression(cast<GetElementPtrInst>(I));
604 break;
605 default:
606 valueNumbering[V] = nextValueNumber;
607 return nextValueNumber++;
608 }
609
610 uint32_t& e = expressionNumbering[exp];
611 if (!e) e = nextValueNumber++;
612 valueNumbering[V] = e;
613 return e;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000614}
615
616/// lookup - Returns the value number of the specified value. Fails if
617/// the value has not yet been numbered.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000618uint32_t ValueTable::lookup(Value *V) const {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000619 DenseMap<Value*, uint32_t>::const_iterator VI = valueNumbering.find(V);
Chris Lattner88365bb2008-03-21 21:14:38 +0000620 assert(VI != valueNumbering.end() && "Value not numbered?");
621 return VI->second;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000622}
623
624/// clear - Remove all entries from the ValueTable
625void ValueTable::clear() {
626 valueNumbering.clear();
627 expressionNumbering.clear();
628 nextValueNumber = 1;
629}
630
Owen Andersonbf7d0bc2007-07-31 23:27:13 +0000631/// erase - Remove a value from the value numbering
Chris Lattnerb2412a82009-09-21 02:42:51 +0000632void ValueTable::erase(Value *V) {
Owen Andersonbf7d0bc2007-07-31 23:27:13 +0000633 valueNumbering.erase(V);
634}
635
Bill Wendling246dbbb2008-12-22 21:36:08 +0000636/// verifyRemoved - Verify that the value is removed from all internal data
637/// structures.
638void ValueTable::verifyRemoved(const Value *V) const {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000639 for (DenseMap<Value*, uint32_t>::const_iterator
Bill Wendling246dbbb2008-12-22 21:36:08 +0000640 I = valueNumbering.begin(), E = valueNumbering.end(); I != E; ++I) {
641 assert(I->first != V && "Inst still occurs in value numbering map!");
642 }
643}
644
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000645//===----------------------------------------------------------------------===//
Bill Wendling30788b82008-12-22 22:32:22 +0000646// GVN Pass
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000647//===----------------------------------------------------------------------===//
648
649namespace {
Chris Lattner3e8b6632009-09-02 06:11:42 +0000650 struct ValueNumberScope {
Owen Anderson6fafe842008-06-20 01:15:47 +0000651 ValueNumberScope* parent;
652 DenseMap<uint32_t, Value*> table;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000653
Owen Anderson6fafe842008-06-20 01:15:47 +0000654 ValueNumberScope(ValueNumberScope* p) : parent(p) { }
655 };
656}
657
658namespace {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000659
Chris Lattner3e8b6632009-09-02 06:11:42 +0000660 class GVN : public FunctionPass {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000661 bool runOnFunction(Function &F);
662 public:
663 static char ID; // Pass identification, replacement for typeid
Bob Wilsonb29d7d22010-02-28 05:34:05 +0000664 explicit GVN(bool noloads = false)
Owen Anderson90c579d2010-08-06 18:33:48 +0000665 : FunctionPass(ID), NoLoads(noloads), MD(0) { }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000666
667 private:
Dan Gohman4ec01b22009-11-14 02:27:51 +0000668 bool NoLoads;
Chris Lattner663e4412008-12-01 00:40:32 +0000669 MemoryDependenceAnalysis *MD;
670 DominatorTree *DT;
671
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000672 ValueTable VN;
Owen Anderson6fafe842008-06-20 01:15:47 +0000673 DenseMap<BasicBlock*, ValueNumberScope*> localAvail;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000674
Bob Wilson484d4a32010-02-16 19:51:59 +0000675 // List of critical edges to be split between iterations.
676 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> toSplit;
677
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000678 // This transformation requires dominator postdominator info
679 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000680 AU.addRequired<DominatorTree>();
Dan Gohman4ec01b22009-11-14 02:27:51 +0000681 if (!NoLoads)
682 AU.addRequired<MemoryDependenceAnalysis>();
Owen Andersonb388ca92007-10-18 19:39:33 +0000683 AU.addRequired<AliasAnalysis>();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000684
Owen Andersonb70a5712008-06-23 17:49:45 +0000685 AU.addPreserved<DominatorTree>();
Owen Andersonb388ca92007-10-18 19:39:33 +0000686 AU.addPreserved<AliasAnalysis>();
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000687 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000688
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000689 // Helper fuctions
690 // FIXME: eliminate or document these better
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000691 bool processLoad(LoadInst* L,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000692 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000693 bool processInstruction(Instruction *I,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000694 SmallVectorImpl<Instruction*> &toErase);
Owen Anderson830db6a2007-08-02 18:16:06 +0000695 bool processNonLocalLoad(LoadInst* L,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000696 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000697 bool processBlock(BasicBlock *BB);
Owen Andersonb2303722008-06-18 21:41:49 +0000698 void dump(DenseMap<uint32_t, Value*>& d);
Owen Anderson3e75a422007-08-14 18:04:11 +0000699 bool iterateOnFunction(Function &F);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000700 Value *CollapsePhi(PHINode* p);
Owen Andersonb2303722008-06-18 21:41:49 +0000701 bool performPRE(Function& F);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000702 Value *lookupNumber(BasicBlock *BB, uint32_t num);
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +0000703 void cleanupGlobalSets();
Bill Wendling246dbbb2008-12-22 21:36:08 +0000704 void verifyRemoved(const Instruction *I) const;
Bob Wilson484d4a32010-02-16 19:51:59 +0000705 bool splitCriticalEdges();
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000706 };
Daniel Dunbara279bc32009-09-20 02:20:51 +0000707
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000708 char GVN::ID = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000709}
710
711// createGVNPass - The public interface to this file...
Bob Wilsonb29d7d22010-02-28 05:34:05 +0000712FunctionPass *llvm::createGVNPass(bool NoLoads) {
713 return new GVN(NoLoads);
Dan Gohman4ec01b22009-11-14 02:27:51 +0000714}
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000715
Owen Andersond13db2c2010-07-21 22:09:45 +0000716INITIALIZE_PASS(GVN, "gvn", "Global Value Numbering", false, false);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000717
Owen Andersonb2303722008-06-18 21:41:49 +0000718void GVN::dump(DenseMap<uint32_t, Value*>& d) {
Dan Gohmanad12b262009-12-18 03:25:51 +0000719 errs() << "{\n";
Owen Andersonb2303722008-06-18 21:41:49 +0000720 for (DenseMap<uint32_t, Value*>::iterator I = d.begin(),
Owen Anderson0cd32032007-07-25 19:57:03 +0000721 E = d.end(); I != E; ++I) {
Dan Gohmanad12b262009-12-18 03:25:51 +0000722 errs() << I->first << "\n";
Owen Anderson0cd32032007-07-25 19:57:03 +0000723 I->second->dump();
724 }
Dan Gohmanad12b262009-12-18 03:25:51 +0000725 errs() << "}\n";
Owen Anderson0cd32032007-07-25 19:57:03 +0000726}
727
Chris Lattnerb2412a82009-09-21 02:42:51 +0000728static bool isSafeReplacement(PHINode* p, Instruction *inst) {
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000729 if (!isa<PHINode>(inst))
730 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000731
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000732 for (Instruction::use_iterator UI = p->use_begin(), E = p->use_end();
733 UI != E; ++UI)
Gabor Greif96f1d8e2010-07-22 13:36:47 +0000734 if (PHINode* use_phi = dyn_cast<PHINode>(*UI))
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000735 if (use_phi->getParent() == inst->getParent())
736 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000737
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000738 return true;
739}
740
Chris Lattnerb2412a82009-09-21 02:42:51 +0000741Value *GVN::CollapsePhi(PHINode *PN) {
742 Value *ConstVal = PN->hasConstantValue(DT);
743 if (!ConstVal) return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000744
Chris Lattnerb2412a82009-09-21 02:42:51 +0000745 Instruction *Inst = dyn_cast<Instruction>(ConstVal);
746 if (!Inst)
747 return ConstVal;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000748
Chris Lattnerb2412a82009-09-21 02:42:51 +0000749 if (DT->dominates(Inst, PN))
750 if (isSafeReplacement(PN, Inst))
751 return Inst;
Owen Anderson1defe2d2007-08-16 22:51:56 +0000752 return 0;
753}
Owen Anderson0cd32032007-07-25 19:57:03 +0000754
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000755/// IsValueFullyAvailableInBlock - Return true if we can prove that the value
756/// we're analyzing is fully available in the specified block. As we go, keep
Chris Lattner72bc70d2008-12-05 07:49:08 +0000757/// track of which blocks we know are fully alive in FullyAvailableBlocks. This
758/// map is actually a tri-state map with the following values:
759/// 0) we know the block *is not* fully available.
760/// 1) we know the block *is* fully available.
761/// 2) we do not know whether the block is fully available or not, but we are
762/// currently speculating that it will be.
763/// 3) we are speculating for this block and have used that to speculate for
764/// other blocks.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000765static bool IsValueFullyAvailableInBlock(BasicBlock *BB,
Chris Lattner72bc70d2008-12-05 07:49:08 +0000766 DenseMap<BasicBlock*, char> &FullyAvailableBlocks) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000767 // Optimistically assume that the block is fully available and check to see
768 // if we already know about this block in one lookup.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000769 std::pair<DenseMap<BasicBlock*, char>::iterator, char> IV =
Chris Lattner72bc70d2008-12-05 07:49:08 +0000770 FullyAvailableBlocks.insert(std::make_pair(BB, 2));
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000771
772 // If the entry already existed for this block, return the precomputed value.
Chris Lattner72bc70d2008-12-05 07:49:08 +0000773 if (!IV.second) {
774 // If this is a speculative "available" value, mark it as being used for
775 // speculation of other blocks.
776 if (IV.first->second == 2)
777 IV.first->second = 3;
778 return IV.first->second != 0;
779 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000780
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000781 // Otherwise, see if it is fully available in all predecessors.
782 pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000783
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000784 // If this block has no predecessors, it isn't live-in here.
785 if (PI == PE)
Chris Lattner72bc70d2008-12-05 07:49:08 +0000786 goto SpeculationFailure;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000787
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000788 for (; PI != PE; ++PI)
789 // If the value isn't fully available in one of our predecessors, then it
790 // isn't fully available in this block either. Undo our previous
791 // optimistic assumption and bail out.
792 if (!IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
Chris Lattner72bc70d2008-12-05 07:49:08 +0000793 goto SpeculationFailure;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000794
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000795 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000796
Chris Lattner72bc70d2008-12-05 07:49:08 +0000797// SpeculationFailure - If we get here, we found out that this is not, after
798// all, a fully-available block. We have a problem if we speculated on this and
799// used the speculation to mark other blocks as available.
800SpeculationFailure:
801 char &BBVal = FullyAvailableBlocks[BB];
Daniel Dunbara279bc32009-09-20 02:20:51 +0000802
Chris Lattner72bc70d2008-12-05 07:49:08 +0000803 // If we didn't speculate on this, just return with it set to false.
804 if (BBVal == 2) {
805 BBVal = 0;
806 return false;
807 }
808
809 // If we did speculate on this value, we could have blocks set to 1 that are
810 // incorrect. Walk the (transitive) successors of this block and mark them as
811 // 0 if set to one.
812 SmallVector<BasicBlock*, 32> BBWorklist;
813 BBWorklist.push_back(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000814
Dan Gohman321a8132010-01-05 16:27:25 +0000815 do {
Chris Lattner72bc70d2008-12-05 07:49:08 +0000816 BasicBlock *Entry = BBWorklist.pop_back_val();
817 // Note that this sets blocks to 0 (unavailable) if they happen to not
818 // already be in FullyAvailableBlocks. This is safe.
819 char &EntryVal = FullyAvailableBlocks[Entry];
820 if (EntryVal == 0) continue; // Already unavailable.
821
822 // Mark as unavailable.
823 EntryVal = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000824
Chris Lattner72bc70d2008-12-05 07:49:08 +0000825 for (succ_iterator I = succ_begin(Entry), E = succ_end(Entry); I != E; ++I)
826 BBWorklist.push_back(*I);
Dan Gohman321a8132010-01-05 16:27:25 +0000827 } while (!BBWorklist.empty());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000828
Chris Lattner72bc70d2008-12-05 07:49:08 +0000829 return false;
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000830}
831
Chris Lattner771a5422009-09-20 20:09:34 +0000832
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000833/// CanCoerceMustAliasedValueToLoad - Return true if
834/// CoerceAvailableValueToLoadType will succeed.
835static bool CanCoerceMustAliasedValueToLoad(Value *StoredVal,
836 const Type *LoadTy,
837 const TargetData &TD) {
838 // If the loaded or stored value is an first class array or struct, don't try
839 // to transform them. We need to be able to bitcast to integer.
Duncan Sands1df98592010-02-16 11:11:14 +0000840 if (LoadTy->isStructTy() || LoadTy->isArrayTy() ||
841 StoredVal->getType()->isStructTy() ||
842 StoredVal->getType()->isArrayTy())
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000843 return false;
844
845 // The store has to be at least as big as the load.
846 if (TD.getTypeSizeInBits(StoredVal->getType()) <
847 TD.getTypeSizeInBits(LoadTy))
848 return false;
849
850 return true;
851}
852
853
Chris Lattner771a5422009-09-20 20:09:34 +0000854/// CoerceAvailableValueToLoadType - If we saw a store of a value to memory, and
855/// then a load from a must-aliased pointer of a different type, try to coerce
856/// the stored value. LoadedTy is the type of the load we want to replace and
857/// InsertPt is the place to insert new instructions.
858///
859/// If we can't do it, return null.
860static Value *CoerceAvailableValueToLoadType(Value *StoredVal,
861 const Type *LoadedTy,
862 Instruction *InsertPt,
863 const TargetData &TD) {
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000864 if (!CanCoerceMustAliasedValueToLoad(StoredVal, LoadedTy, TD))
865 return 0;
866
Chris Lattner771a5422009-09-20 20:09:34 +0000867 const Type *StoredValTy = StoredVal->getType();
868
Chris Lattner7944c212010-05-08 20:01:44 +0000869 uint64_t StoreSize = TD.getTypeStoreSizeInBits(StoredValTy);
Chris Lattner771a5422009-09-20 20:09:34 +0000870 uint64_t LoadSize = TD.getTypeSizeInBits(LoadedTy);
871
872 // If the store and reload are the same size, we can always reuse it.
873 if (StoreSize == LoadSize) {
Duncan Sands1df98592010-02-16 11:11:14 +0000874 if (StoredValTy->isPointerTy() && LoadedTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000875 // Pointer to Pointer -> use bitcast.
876 return new BitCastInst(StoredVal, LoadedTy, "", InsertPt);
877 }
878
879 // Convert source pointers to integers, which can be bitcast.
Duncan Sands1df98592010-02-16 11:11:14 +0000880 if (StoredValTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000881 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
882 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
883 }
884
885 const Type *TypeToCastTo = LoadedTy;
Duncan Sands1df98592010-02-16 11:11:14 +0000886 if (TypeToCastTo->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000887 TypeToCastTo = TD.getIntPtrType(StoredValTy->getContext());
888
889 if (StoredValTy != TypeToCastTo)
890 StoredVal = new BitCastInst(StoredVal, TypeToCastTo, "", InsertPt);
891
892 // Cast to pointer if the load needs a pointer type.
Duncan Sands1df98592010-02-16 11:11:14 +0000893 if (LoadedTy->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000894 StoredVal = new IntToPtrInst(StoredVal, LoadedTy, "", InsertPt);
895
896 return StoredVal;
897 }
898
899 // If the loaded value is smaller than the available value, then we can
900 // extract out a piece from it. If the available value is too small, then we
901 // can't do anything.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000902 assert(StoreSize >= LoadSize && "CanCoerceMustAliasedValueToLoad fail");
Chris Lattner771a5422009-09-20 20:09:34 +0000903
904 // Convert source pointers to integers, which can be manipulated.
Duncan Sands1df98592010-02-16 11:11:14 +0000905 if (StoredValTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000906 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
907 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
908 }
909
910 // Convert vectors and fp to integer, which can be manipulated.
Duncan Sands1df98592010-02-16 11:11:14 +0000911 if (!StoredValTy->isIntegerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000912 StoredValTy = IntegerType::get(StoredValTy->getContext(), StoreSize);
913 StoredVal = new BitCastInst(StoredVal, StoredValTy, "", InsertPt);
914 }
915
916 // If this is a big-endian system, we need to shift the value down to the low
917 // bits so that a truncate will work.
918 if (TD.isBigEndian()) {
919 Constant *Val = ConstantInt::get(StoredVal->getType(), StoreSize-LoadSize);
920 StoredVal = BinaryOperator::CreateLShr(StoredVal, Val, "tmp", InsertPt);
921 }
922
923 // Truncate the integer to the right size now.
924 const Type *NewIntTy = IntegerType::get(StoredValTy->getContext(), LoadSize);
925 StoredVal = new TruncInst(StoredVal, NewIntTy, "trunc", InsertPt);
926
927 if (LoadedTy == NewIntTy)
928 return StoredVal;
929
930 // If the result is a pointer, inttoptr.
Duncan Sands1df98592010-02-16 11:11:14 +0000931 if (LoadedTy->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000932 return new IntToPtrInst(StoredVal, LoadedTy, "inttoptr", InsertPt);
933
934 // Otherwise, bitcast.
935 return new BitCastInst(StoredVal, LoadedTy, "bitcast", InsertPt);
936}
937
Chris Lattnerca749402009-09-21 06:24:16 +0000938/// GetBaseWithConstantOffset - Analyze the specified pointer to see if it can
939/// be expressed as a base pointer plus a constant offset. Return the base and
940/// offset to the caller.
941static Value *GetBaseWithConstantOffset(Value *Ptr, int64_t &Offset,
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000942 const TargetData &TD) {
Chris Lattnerca749402009-09-21 06:24:16 +0000943 Operator *PtrOp = dyn_cast<Operator>(Ptr);
944 if (PtrOp == 0) return Ptr;
945
946 // Just look through bitcasts.
947 if (PtrOp->getOpcode() == Instruction::BitCast)
948 return GetBaseWithConstantOffset(PtrOp->getOperand(0), Offset, TD);
949
950 // If this is a GEP with constant indices, we can look through it.
951 GEPOperator *GEP = dyn_cast<GEPOperator>(PtrOp);
952 if (GEP == 0 || !GEP->hasAllConstantIndices()) return Ptr;
953
954 gep_type_iterator GTI = gep_type_begin(GEP);
955 for (User::op_iterator I = GEP->idx_begin(), E = GEP->idx_end(); I != E;
956 ++I, ++GTI) {
957 ConstantInt *OpC = cast<ConstantInt>(*I);
958 if (OpC->isZero()) continue;
959
960 // Handle a struct and array indices which add their offset to the pointer.
961 if (const StructType *STy = dyn_cast<StructType>(*GTI)) {
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000962 Offset += TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue());
Chris Lattnerca749402009-09-21 06:24:16 +0000963 } else {
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000964 uint64_t Size = TD.getTypeAllocSize(GTI.getIndexedType());
Chris Lattnerca749402009-09-21 06:24:16 +0000965 Offset += OpC->getSExtValue()*Size;
966 }
967 }
968
969 // Re-sign extend from the pointer size if needed to get overflow edge cases
970 // right.
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000971 unsigned PtrSize = TD.getPointerSizeInBits();
Chris Lattnerca749402009-09-21 06:24:16 +0000972 if (PtrSize < 64)
973 Offset = (Offset << (64-PtrSize)) >> (64-PtrSize);
974
975 return GetBaseWithConstantOffset(GEP->getPointerOperand(), Offset, TD);
976}
977
978
Chris Lattnerfaf815b2009-12-06 01:57:02 +0000979/// AnalyzeLoadFromClobberingWrite - This function is called when we have a
980/// memdep query of a load that ends up being a clobbering memory write (store,
981/// memset, memcpy, memmove). This means that the write *may* provide bits used
982/// by the load but we can't be sure because the pointers don't mustalias.
983///
984/// Check this case to see if there is anything more we can do before we give
985/// up. This returns -1 if we have to give up, or a byte number in the stored
986/// value of the piece that feeds the load.
Chris Lattner03f17da2009-12-09 07:34:10 +0000987static int AnalyzeLoadFromClobberingWrite(const Type *LoadTy, Value *LoadPtr,
988 Value *WritePtr,
Chris Lattnerfaf815b2009-12-06 01:57:02 +0000989 uint64_t WriteSizeInBits,
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000990 const TargetData &TD) {
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000991 // If the loaded or stored value is an first class array or struct, don't try
992 // to transform them. We need to be able to bitcast to integer.
Duncan Sands1df98592010-02-16 11:11:14 +0000993 if (LoadTy->isStructTy() || LoadTy->isArrayTy())
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000994 return -1;
995
Chris Lattnerca749402009-09-21 06:24:16 +0000996 int64_t StoreOffset = 0, LoadOffset = 0;
Chris Lattnerfaf815b2009-12-06 01:57:02 +0000997 Value *StoreBase = GetBaseWithConstantOffset(WritePtr, StoreOffset, TD);
Chris Lattnerca749402009-09-21 06:24:16 +0000998 Value *LoadBase =
Chris Lattner03f17da2009-12-09 07:34:10 +0000999 GetBaseWithConstantOffset(LoadPtr, LoadOffset, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001000 if (StoreBase != LoadBase)
1001 return -1;
1002
1003 // If the load and store are to the exact same address, they should have been
1004 // a must alias. AA must have gotten confused.
Chris Lattner219d7742010-03-25 05:58:19 +00001005 // FIXME: Study to see if/when this happens. One case is forwarding a memset
1006 // to a load from the base of the memset.
Chris Lattnerca749402009-09-21 06:24:16 +00001007#if 0
Chris Lattner219d7742010-03-25 05:58:19 +00001008 if (LoadOffset == StoreOffset) {
David Greenebf7f78e2010-01-05 01:27:17 +00001009 dbgs() << "STORE/LOAD DEP WITH COMMON POINTER MISSED:\n"
Chris Lattnerca749402009-09-21 06:24:16 +00001010 << "Base = " << *StoreBase << "\n"
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001011 << "Store Ptr = " << *WritePtr << "\n"
1012 << "Store Offs = " << StoreOffset << "\n"
Chris Lattnerb6760b42009-12-10 00:04:46 +00001013 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattnerb3f927f2009-12-09 02:41:54 +00001014 abort();
Chris Lattnerca749402009-09-21 06:24:16 +00001015 }
Chris Lattner219d7742010-03-25 05:58:19 +00001016#endif
Chris Lattnerca749402009-09-21 06:24:16 +00001017
1018 // If the load and store don't overlap at all, the store doesn't provide
1019 // anything to the load. In this case, they really don't alias at all, AA
1020 // must have gotten confused.
1021 // FIXME: Investigate cases where this bails out, e.g. rdar://7238614. Then
1022 // remove this check, as it is duplicated with what we have below.
Chris Lattner03f17da2009-12-09 07:34:10 +00001023 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy);
Chris Lattnerca749402009-09-21 06:24:16 +00001024
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001025 if ((WriteSizeInBits & 7) | (LoadSize & 7))
Chris Lattnerca749402009-09-21 06:24:16 +00001026 return -1;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001027 uint64_t StoreSize = WriteSizeInBits >> 3; // Convert to bytes.
Chris Lattnerca749402009-09-21 06:24:16 +00001028 LoadSize >>= 3;
1029
1030
1031 bool isAAFailure = false;
Chris Lattner219d7742010-03-25 05:58:19 +00001032 if (StoreOffset < LoadOffset)
Chris Lattnerca749402009-09-21 06:24:16 +00001033 isAAFailure = StoreOffset+int64_t(StoreSize) <= LoadOffset;
Chris Lattner219d7742010-03-25 05:58:19 +00001034 else
Chris Lattnerca749402009-09-21 06:24:16 +00001035 isAAFailure = LoadOffset+int64_t(LoadSize) <= StoreOffset;
Chris Lattner219d7742010-03-25 05:58:19 +00001036
Chris Lattnerca749402009-09-21 06:24:16 +00001037 if (isAAFailure) {
1038#if 0
David Greenebf7f78e2010-01-05 01:27:17 +00001039 dbgs() << "STORE LOAD DEP WITH COMMON BASE:\n"
Chris Lattnerca749402009-09-21 06:24:16 +00001040 << "Base = " << *StoreBase << "\n"
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001041 << "Store Ptr = " << *WritePtr << "\n"
1042 << "Store Offs = " << StoreOffset << "\n"
Chris Lattnerb6760b42009-12-10 00:04:46 +00001043 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattnerb3f927f2009-12-09 02:41:54 +00001044 abort();
Chris Lattnerca749402009-09-21 06:24:16 +00001045#endif
1046 return -1;
1047 }
1048
1049 // If the Load isn't completely contained within the stored bits, we don't
1050 // have all the bits to feed it. We could do something crazy in the future
1051 // (issue a smaller load then merge the bits in) but this seems unlikely to be
1052 // valuable.
1053 if (StoreOffset > LoadOffset ||
1054 StoreOffset+StoreSize < LoadOffset+LoadSize)
1055 return -1;
1056
1057 // Okay, we can do this transformation. Return the number of bytes into the
1058 // store that the load is.
1059 return LoadOffset-StoreOffset;
1060}
1061
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001062/// AnalyzeLoadFromClobberingStore - This function is called when we have a
1063/// memdep query of a load that ends up being a clobbering store.
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001064static int AnalyzeLoadFromClobberingStore(const Type *LoadTy, Value *LoadPtr,
1065 StoreInst *DepSI,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001066 const TargetData &TD) {
1067 // Cannot handle reading from store of first-class aggregate yet.
Duncan Sands1df98592010-02-16 11:11:14 +00001068 if (DepSI->getOperand(0)->getType()->isStructTy() ||
1069 DepSI->getOperand(0)->getType()->isArrayTy())
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001070 return -1;
1071
1072 Value *StorePtr = DepSI->getPointerOperand();
Chris Lattner16f244e2009-12-10 00:11:45 +00001073 uint64_t StoreSize = TD.getTypeSizeInBits(DepSI->getOperand(0)->getType());
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001074 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
Chris Lattner03f17da2009-12-09 07:34:10 +00001075 StorePtr, StoreSize, TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001076}
1077
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001078static int AnalyzeLoadFromClobberingMemInst(const Type *LoadTy, Value *LoadPtr,
1079 MemIntrinsic *MI,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001080 const TargetData &TD) {
1081 // If the mem operation is a non-constant size, we can't handle it.
1082 ConstantInt *SizeCst = dyn_cast<ConstantInt>(MI->getLength());
1083 if (SizeCst == 0) return -1;
1084 uint64_t MemSizeInBits = SizeCst->getZExtValue()*8;
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001085
1086 // If this is memset, we just need to see if the offset is valid in the size
1087 // of the memset..
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001088 if (MI->getIntrinsicID() == Intrinsic::memset)
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001089 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr, MI->getDest(),
1090 MemSizeInBits, TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001091
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001092 // If we have a memcpy/memmove, the only case we can handle is if this is a
1093 // copy from constant memory. In that case, we can read directly from the
1094 // constant memory.
1095 MemTransferInst *MTI = cast<MemTransferInst>(MI);
1096
1097 Constant *Src = dyn_cast<Constant>(MTI->getSource());
1098 if (Src == 0) return -1;
1099
1100 GlobalVariable *GV = dyn_cast<GlobalVariable>(Src->getUnderlyingObject());
1101 if (GV == 0 || !GV->isConstant()) return -1;
1102
1103 // See if the access is within the bounds of the transfer.
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001104 int Offset = AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
1105 MI->getDest(), MemSizeInBits, TD);
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001106 if (Offset == -1)
1107 return Offset;
1108
1109 // Otherwise, see if we can constant fold a load from the constant with the
1110 // offset applied as appropriate.
1111 Src = ConstantExpr::getBitCast(Src,
1112 llvm::Type::getInt8PtrTy(Src->getContext()));
1113 Constant *OffsetCst =
1114 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1115 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001116 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001117 if (ConstantFoldLoadFromConstPtr(Src, &TD))
1118 return Offset;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001119 return -1;
1120}
1121
Chris Lattnerca749402009-09-21 06:24:16 +00001122
1123/// GetStoreValueForLoad - This function is called when we have a
1124/// memdep query of a load that ends up being a clobbering store. This means
1125/// that the store *may* provide bits used by the load but we can't be sure
1126/// because the pointers don't mustalias. Check this case to see if there is
1127/// anything more we can do before we give up.
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001128static Value *GetStoreValueForLoad(Value *SrcVal, unsigned Offset,
1129 const Type *LoadTy,
1130 Instruction *InsertPt, const TargetData &TD){
Chris Lattnerca749402009-09-21 06:24:16 +00001131 LLVMContext &Ctx = SrcVal->getType()->getContext();
1132
Chris Lattner7944c212010-05-08 20:01:44 +00001133 uint64_t StoreSize = (TD.getTypeSizeInBits(SrcVal->getType()) + 7) / 8;
1134 uint64_t LoadSize = (TD.getTypeSizeInBits(LoadTy) + 7) / 8;
Chris Lattnerca749402009-09-21 06:24:16 +00001135
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001136 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
Chris Lattnerca749402009-09-21 06:24:16 +00001137
1138 // Compute which bits of the stored value are being used by the load. Convert
1139 // to an integer type to start with.
Duncan Sands1df98592010-02-16 11:11:14 +00001140 if (SrcVal->getType()->isPointerTy())
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001141 SrcVal = Builder.CreatePtrToInt(SrcVal, TD.getIntPtrType(Ctx), "tmp");
Duncan Sands1df98592010-02-16 11:11:14 +00001142 if (!SrcVal->getType()->isIntegerTy())
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001143 SrcVal = Builder.CreateBitCast(SrcVal, IntegerType::get(Ctx, StoreSize*8),
1144 "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001145
1146 // Shift the bits to the least significant depending on endianness.
1147 unsigned ShiftAmt;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001148 if (TD.isLittleEndian())
Chris Lattnerca749402009-09-21 06:24:16 +00001149 ShiftAmt = Offset*8;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001150 else
Chris Lattner19ad7842009-09-21 17:55:47 +00001151 ShiftAmt = (StoreSize-LoadSize-Offset)*8;
Chris Lattnerca749402009-09-21 06:24:16 +00001152
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001153 if (ShiftAmt)
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001154 SrcVal = Builder.CreateLShr(SrcVal, ShiftAmt, "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001155
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001156 if (LoadSize != StoreSize)
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001157 SrcVal = Builder.CreateTrunc(SrcVal, IntegerType::get(Ctx, LoadSize*8),
1158 "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001159
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001160 return CoerceAvailableValueToLoadType(SrcVal, LoadTy, InsertPt, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001161}
1162
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001163/// GetMemInstValueForLoad - This function is called when we have a
1164/// memdep query of a load that ends up being a clobbering mem intrinsic.
1165static Value *GetMemInstValueForLoad(MemIntrinsic *SrcInst, unsigned Offset,
1166 const Type *LoadTy, Instruction *InsertPt,
1167 const TargetData &TD){
1168 LLVMContext &Ctx = LoadTy->getContext();
1169 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
1170
1171 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
1172
1173 // We know that this method is only called when the mem transfer fully
1174 // provides the bits for the load.
1175 if (MemSetInst *MSI = dyn_cast<MemSetInst>(SrcInst)) {
1176 // memset(P, 'x', 1234) -> splat('x'), even if x is a variable, and
1177 // independently of what the offset is.
1178 Value *Val = MSI->getValue();
1179 if (LoadSize != 1)
1180 Val = Builder.CreateZExt(Val, IntegerType::get(Ctx, LoadSize*8));
1181
1182 Value *OneElt = Val;
1183
1184 // Splat the value out to the right number of bits.
1185 for (unsigned NumBytesSet = 1; NumBytesSet != LoadSize; ) {
1186 // If we can double the number of bytes set, do it.
1187 if (NumBytesSet*2 <= LoadSize) {
1188 Value *ShVal = Builder.CreateShl(Val, NumBytesSet*8);
1189 Val = Builder.CreateOr(Val, ShVal);
1190 NumBytesSet <<= 1;
1191 continue;
1192 }
1193
1194 // Otherwise insert one byte at a time.
1195 Value *ShVal = Builder.CreateShl(Val, 1*8);
1196 Val = Builder.CreateOr(OneElt, ShVal);
1197 ++NumBytesSet;
1198 }
1199
1200 return CoerceAvailableValueToLoadType(Val, LoadTy, InsertPt, TD);
1201 }
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001202
1203 // Otherwise, this is a memcpy/memmove from a constant global.
1204 MemTransferInst *MTI = cast<MemTransferInst>(SrcInst);
1205 Constant *Src = cast<Constant>(MTI->getSource());
1206
1207 // Otherwise, see if we can constant fold a load from the constant with the
1208 // offset applied as appropriate.
1209 Src = ConstantExpr::getBitCast(Src,
1210 llvm::Type::getInt8PtrTy(Src->getContext()));
1211 Constant *OffsetCst =
1212 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1213 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
1214 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
1215 return ConstantFoldLoadFromConstPtr(Src, &TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001216}
1217
Dan Gohmanb3579832010-04-15 17:08:50 +00001218namespace {
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001219
Chris Lattner87913512009-09-21 06:30:24 +00001220struct AvailableValueInBlock {
1221 /// BB - The basic block in question.
1222 BasicBlock *BB;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001223 enum ValType {
1224 SimpleVal, // A simple offsetted value that is accessed.
1225 MemIntrin // A memory intrinsic which is loaded from.
1226 };
1227
Chris Lattner87913512009-09-21 06:30:24 +00001228 /// V - The value that is live out of the block.
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001229 PointerIntPair<Value *, 1, ValType> Val;
1230
1231 /// Offset - The byte offset in Val that is interesting for the load query.
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001232 unsigned Offset;
Chris Lattner87913512009-09-21 06:30:24 +00001233
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001234 static AvailableValueInBlock get(BasicBlock *BB, Value *V,
1235 unsigned Offset = 0) {
Chris Lattner87913512009-09-21 06:30:24 +00001236 AvailableValueInBlock Res;
1237 Res.BB = BB;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001238 Res.Val.setPointer(V);
1239 Res.Val.setInt(SimpleVal);
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001240 Res.Offset = Offset;
Chris Lattner87913512009-09-21 06:30:24 +00001241 return Res;
1242 }
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001243
1244 static AvailableValueInBlock getMI(BasicBlock *BB, MemIntrinsic *MI,
1245 unsigned Offset = 0) {
1246 AvailableValueInBlock Res;
1247 Res.BB = BB;
1248 Res.Val.setPointer(MI);
1249 Res.Val.setInt(MemIntrin);
1250 Res.Offset = Offset;
1251 return Res;
1252 }
1253
1254 bool isSimpleValue() const { return Val.getInt() == SimpleVal; }
1255 Value *getSimpleValue() const {
1256 assert(isSimpleValue() && "Wrong accessor");
1257 return Val.getPointer();
1258 }
1259
1260 MemIntrinsic *getMemIntrinValue() const {
1261 assert(!isSimpleValue() && "Wrong accessor");
1262 return cast<MemIntrinsic>(Val.getPointer());
1263 }
Chris Lattner5362c542009-12-21 23:04:33 +00001264
1265 /// MaterializeAdjustedValue - Emit code into this block to adjust the value
1266 /// defined here to the specified type. This handles various coercion cases.
1267 Value *MaterializeAdjustedValue(const Type *LoadTy,
1268 const TargetData *TD) const {
1269 Value *Res;
1270 if (isSimpleValue()) {
1271 Res = getSimpleValue();
1272 if (Res->getType() != LoadTy) {
1273 assert(TD && "Need target data to handle type mismatch case");
1274 Res = GetStoreValueForLoad(Res, Offset, LoadTy, BB->getTerminator(),
1275 *TD);
1276
1277 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\nOffset: " << Offset << " "
1278 << *getSimpleValue() << '\n'
1279 << *Res << '\n' << "\n\n\n");
1280 }
1281 } else {
1282 Res = GetMemInstValueForLoad(getMemIntrinValue(), Offset,
1283 LoadTy, BB->getTerminator(), *TD);
1284 DEBUG(errs() << "GVN COERCED NONLOCAL MEM INTRIN:\nOffset: " << Offset
1285 << " " << *getMemIntrinValue() << '\n'
1286 << *Res << '\n' << "\n\n\n");
1287 }
1288 return Res;
1289 }
Chris Lattner87913512009-09-21 06:30:24 +00001290};
1291
Dan Gohmanb3579832010-04-15 17:08:50 +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);
Duncan Sandsfc6e29d2010-09-02 08:14:03 +00001311 SSAUpdate.Initialize(LI->getType(), LI->getName());
Chris Lattnera09fbf02009-10-10 23:50:30 +00001312
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
Gabor Greifea3eec92010-04-09 10:57:00 +00001336static bool isLifetimeStart(const Instruction *Inst) {
1337 if (const 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);
Dan Gohmanfe601042010-06-22 15:08:57 +00001501 ++NumGVNLoad;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001502 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;
Owen Andersonb0ba0f42010-09-25 05:26:18 +00001534
1535 // If any of these blocks has more than one successor (i.e. if the edge we
1536 // just traversed was critical), then there are other paths through this
1537 // block along which the load may not be anticipated. Hoisting the load
1538 // above this block would be adding the load to execution paths along
1539 // which it was not previously executed.
Dale Johannesen42c3f552009-06-17 20:48:23 +00001540 if (TmpBB->getTerminator()->getNumSuccessors() != 1)
Owen Andersonb0ba0f42010-09-25 05:26:18 +00001541 return false;
Owen Anderson88554df2009-05-31 09:03:40 +00001542 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001543
Owen Anderson88554df2009-05-31 09:03:40 +00001544 assert(TmpBB);
1545 LoadBB = TmpBB;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001546
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001547 // FIXME: It is extremely unclear what this loop is doing, other than
1548 // artificially restricting loadpre.
Owen Anderson88554df2009-05-31 09:03:40 +00001549 if (isSinglePred) {
1550 bool isHot = false;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001551 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
1552 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1553 if (AV.isSimpleValue())
Daniel Dunbara279bc32009-09-20 02:20:51 +00001554 // "Hot" Instruction is in some loop (because it dominates its dep.
1555 // instruction).
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001556 if (Instruction *I = dyn_cast<Instruction>(AV.getSimpleValue()))
1557 if (DT->dominates(LI, I)) {
1558 isHot = true;
1559 break;
1560 }
1561 }
Owen Anderson88554df2009-05-31 09:03:40 +00001562
1563 // We are interested only in "hot" instructions. We don't want to do any
1564 // mis-optimizations here.
1565 if (!isHot)
1566 return false;
1567 }
1568
Bob Wilson6cad4172010-02-01 21:17:14 +00001569 // Check to see how many predecessors have the loaded value fully
1570 // available.
1571 DenseMap<BasicBlock*, Value*> PredLoads;
Chris Lattner72bc70d2008-12-05 07:49:08 +00001572 DenseMap<BasicBlock*, char> FullyAvailableBlocks;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001573 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner87913512009-09-21 06:30:24 +00001574 FullyAvailableBlocks[ValuesPerBlock[i].BB] = true;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001575 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1576 FullyAvailableBlocks[UnavailableBlocks[i]] = false;
1577
Bob Wilson34414a62010-05-04 20:03:21 +00001578 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> NeedToSplit;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001579 for (pred_iterator PI = pred_begin(LoadBB), E = pred_end(LoadBB);
1580 PI != E; ++PI) {
Bob Wilson6cad4172010-02-01 21:17:14 +00001581 BasicBlock *Pred = *PI;
1582 if (IsValueFullyAvailableInBlock(Pred, FullyAvailableBlocks)) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001583 continue;
Bob Wilson6cad4172010-02-01 21:17:14 +00001584 }
1585 PredLoads[Pred] = 0;
Bob Wilson484d4a32010-02-16 19:51:59 +00001586
Bob Wilson6cad4172010-02-01 21:17:14 +00001587 if (Pred->getTerminator()->getNumSuccessors() != 1) {
Bob Wilson484d4a32010-02-16 19:51:59 +00001588 if (isa<IndirectBrInst>(Pred->getTerminator())) {
1589 DEBUG(dbgs() << "COULD NOT PRE LOAD BECAUSE OF INDBR CRITICAL EDGE '"
1590 << Pred->getName() << "': " << *LI << '\n');
1591 return false;
1592 }
Bob Wilsonae23daf2010-02-16 21:06:42 +00001593 unsigned SuccNum = GetSuccessorNumber(Pred, LoadBB);
Bob Wilson34414a62010-05-04 20:03:21 +00001594 NeedToSplit.push_back(std::make_pair(Pred->getTerminator(), SuccNum));
Bob Wilson6cad4172010-02-01 21:17:14 +00001595 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001596 }
Bob Wilson34414a62010-05-04 20:03:21 +00001597 if (!NeedToSplit.empty()) {
Bob Wilsonbc786532010-05-05 20:44:15 +00001598 toSplit.append(NeedToSplit.begin(), NeedToSplit.end());
Bob Wilson70704972010-03-01 23:37:32 +00001599 return false;
Bob Wilson34414a62010-05-04 20:03:21 +00001600 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001601
Bob Wilson6cad4172010-02-01 21:17:14 +00001602 // Decide whether PRE is profitable for this load.
1603 unsigned NumUnavailablePreds = PredLoads.size();
1604 assert(NumUnavailablePreds != 0 &&
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001605 "Fully available value should be eliminated above!");
Owen Anderson7267e142010-10-01 20:02:55 +00001606
1607 // If this load is unavailable in multiple predecessors, reject it.
1608 // FIXME: If we could restructure the CFG, we could make a common pred with
1609 // all the preds that don't have an available LI and insert a new load into
1610 // that one block.
1611 if (NumUnavailablePreds != 1)
Bob Wilson6cad4172010-02-01 21:17:14 +00001612 return false;
Bob Wilson6cad4172010-02-01 21:17:14 +00001613
1614 // Check if the load can safely be moved to all the unavailable predecessors.
1615 bool CanDoPRE = true;
Chris Lattnerdd696052009-11-28 15:39:14 +00001616 SmallVector<Instruction*, 8> NewInsts;
Bob Wilson6cad4172010-02-01 21:17:14 +00001617 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1618 E = PredLoads.end(); I != E; ++I) {
1619 BasicBlock *UnavailablePred = I->first;
1620
1621 // Do PHI translation to get its value in the predecessor if necessary. The
1622 // returned pointer (if non-null) is guaranteed to dominate UnavailablePred.
1623
1624 // If all preds have a single successor, then we know it is safe to insert
1625 // the load on the pred (?!?), so we can insert code to materialize the
1626 // pointer if it is not available.
1627 PHITransAddr Address(LI->getOperand(0), TD);
1628 Value *LoadPtr = 0;
1629 if (allSingleSucc) {
1630 LoadPtr = Address.PHITranslateWithInsertion(LoadBB, UnavailablePred,
1631 *DT, NewInsts);
1632 } else {
Daniel Dunbar6d8f2ca2010-02-24 08:48:04 +00001633 Address.PHITranslateValue(LoadBB, UnavailablePred, DT);
Bob Wilson6cad4172010-02-01 21:17:14 +00001634 LoadPtr = Address.getAddr();
Bob Wilson6cad4172010-02-01 21:17:14 +00001635 }
1636
1637 // If we couldn't find or insert a computation of this phi translated value,
1638 // we fail PRE.
1639 if (LoadPtr == 0) {
1640 DEBUG(dbgs() << "COULDN'T INSERT PHI TRANSLATED VALUE OF: "
1641 << *LI->getOperand(0) << "\n");
1642 CanDoPRE = false;
1643 break;
1644 }
1645
1646 // Make sure it is valid to move this load here. We have to watch out for:
1647 // @1 = getelementptr (i8* p, ...
1648 // test p and branch if == 0
1649 // load @1
1650 // It is valid to have the getelementptr before the test, even if p can be 0,
1651 // as getelementptr only does address arithmetic.
1652 // If we are not pushing the value through any multiple-successor blocks
1653 // we do not have this case. Otherwise, check that the load is safe to
1654 // put anywhere; this can be improved, but should be conservatively safe.
1655 if (!allSingleSucc &&
1656 // FIXME: REEVALUTE THIS.
1657 !isSafeToLoadUnconditionally(LoadPtr,
1658 UnavailablePred->getTerminator(),
1659 LI->getAlignment(), TD)) {
1660 CanDoPRE = false;
1661 break;
1662 }
1663
1664 I->second = LoadPtr;
Chris Lattner05e15f82009-12-09 01:59:31 +00001665 }
1666
Bob Wilson6cad4172010-02-01 21:17:14 +00001667 if (!CanDoPRE) {
1668 while (!NewInsts.empty())
1669 NewInsts.pop_back_val()->eraseFromParent();
Dale Johannesen42c3f552009-06-17 20:48:23 +00001670 return false;
Chris Lattner0c264b12009-11-28 16:08:18 +00001671 }
Dale Johannesen42c3f552009-06-17 20:48:23 +00001672
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001673 // Okay, we can eliminate this load by inserting a reload in the predecessor
1674 // and using PHI construction to get the value in the other predecessors, do
1675 // it.
David Greenebf7f78e2010-01-05 01:27:17 +00001676 DEBUG(dbgs() << "GVN REMOVING PRE LOAD: " << *LI << '\n');
Chris Lattner0c264b12009-11-28 16:08:18 +00001677 DEBUG(if (!NewInsts.empty())
David Greenebf7f78e2010-01-05 01:27:17 +00001678 dbgs() << "INSERTED " << NewInsts.size() << " INSTS: "
Chris Lattner0c264b12009-11-28 16:08:18 +00001679 << *NewInsts.back() << '\n');
1680
Bob Wilson6cad4172010-02-01 21:17:14 +00001681 // Assign value numbers to the new instructions.
1682 for (unsigned i = 0, e = NewInsts.size(); i != e; ++i) {
1683 // FIXME: We really _ought_ to insert these value numbers into their
1684 // parent's availability map. However, in doing so, we risk getting into
1685 // ordering issues. If a block hasn't been processed yet, we would be
1686 // marking a value as AVAIL-IN, which isn't what we intend.
1687 VN.lookup_or_add(NewInsts[i]);
1688 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001689
Bob Wilson6cad4172010-02-01 21:17:14 +00001690 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1691 E = PredLoads.end(); I != E; ++I) {
1692 BasicBlock *UnavailablePred = I->first;
1693 Value *LoadPtr = I->second;
1694
1695 Value *NewLoad = new LoadInst(LoadPtr, LI->getName()+".pre", false,
1696 LI->getAlignment(),
1697 UnavailablePred->getTerminator());
1698
1699 // Add the newly created load.
1700 ValuesPerBlock.push_back(AvailableValueInBlock::get(UnavailablePred,
1701 NewLoad));
Bob Wilson188f4282010-02-23 05:55:00 +00001702 MD->invalidateCachedPointerInfo(LoadPtr);
1703 DEBUG(dbgs() << "GVN INSERTED " << *NewLoad << '\n');
Bob Wilson6cad4172010-02-01 21:17:14 +00001704 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001705
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001706 // Perform PHI construction.
Chris Lattnerd2191e52009-12-21 23:15:48 +00001707 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnera09fbf02009-10-10 23:50:30 +00001708 VN.getAliasAnalysis());
Chris Lattner771a5422009-09-20 20:09:34 +00001709 LI->replaceAllUsesWith(V);
1710 if (isa<PHINode>(V))
1711 V->takeName(LI);
Duncan Sands1df98592010-02-16 11:11:14 +00001712 if (V->getType()->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +00001713 MD->invalidateCachedPointerInfo(V);
Bob Wilson74175c22010-02-22 21:39:41 +00001714 VN.erase(LI);
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001715 toErase.push_back(LI);
Dan Gohmanfe601042010-06-22 15:08:57 +00001716 ++NumPRELoad;
Owen Anderson0cd32032007-07-25 19:57:03 +00001717 return true;
1718}
1719
Owen Anderson62bc33c2007-08-16 22:02:55 +00001720/// processLoad - Attempt to eliminate a load, first by eliminating it
1721/// locally, and then attempting non-local elimination if that fails.
Chris Lattnerb51deb92008-12-05 21:04:20 +00001722bool GVN::processLoad(LoadInst *L, SmallVectorImpl<Instruction*> &toErase) {
Dan Gohman4ec01b22009-11-14 02:27:51 +00001723 if (!MD)
1724 return false;
1725
Chris Lattnerb51deb92008-12-05 21:04:20 +00001726 if (L->isVolatile())
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001727 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001728
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001729 // ... to a pointer that has been loaded from before...
Chris Lattnerb2412a82009-09-21 02:42:51 +00001730 MemDepResult Dep = MD->getDependency(L);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001731
Chris Lattnerb51deb92008-12-05 21:04:20 +00001732 // If the value isn't available, don't do anything!
Chris Lattnerb2412a82009-09-21 02:42:51 +00001733 if (Dep.isClobber()) {
Chris Lattnereed919b2009-09-21 05:57:11 +00001734 // Check to see if we have something like this:
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001735 // store i32 123, i32* %P
1736 // %A = bitcast i32* %P to i8*
1737 // %B = gep i8* %A, i32 1
1738 // %C = load i8* %B
1739 //
1740 // We could do that by recognizing if the clobber instructions are obviously
1741 // a common base + constant offset, and if the previous store (or memset)
1742 // completely covers this load. This sort of thing can happen in bitfield
1743 // access code.
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001744 Value *AvailVal = 0;
Chris Lattnereed919b2009-09-21 05:57:11 +00001745 if (StoreInst *DepSI = dyn_cast<StoreInst>(Dep.getInst()))
Chris Lattner1ce08292009-09-21 06:22:46 +00001746 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001747 int Offset = AnalyzeLoadFromClobberingStore(L->getType(),
1748 L->getPointerOperand(),
1749 DepSI, *TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001750 if (Offset != -1)
1751 AvailVal = GetStoreValueForLoad(DepSI->getOperand(0), Offset,
1752 L->getType(), L, *TD);
Chris Lattner1ce08292009-09-21 06:22:46 +00001753 }
Chris Lattnereed919b2009-09-21 05:57:11 +00001754
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001755 // If the clobbering value is a memset/memcpy/memmove, see if we can forward
1756 // a value on from it.
1757 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(Dep.getInst())) {
1758 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001759 int Offset = AnalyzeLoadFromClobberingMemInst(L->getType(),
1760 L->getPointerOperand(),
1761 DepMI, *TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001762 if (Offset != -1)
1763 AvailVal = GetMemInstValueForLoad(DepMI, Offset, L->getType(), L,*TD);
1764 }
1765 }
1766
1767 if (AvailVal) {
David Greenebf7f78e2010-01-05 01:27:17 +00001768 DEBUG(dbgs() << "GVN COERCED INST:\n" << *Dep.getInst() << '\n'
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001769 << *AvailVal << '\n' << *L << "\n\n\n");
1770
1771 // Replace the load!
1772 L->replaceAllUsesWith(AvailVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001773 if (AvailVal->getType()->isPointerTy())
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001774 MD->invalidateCachedPointerInfo(AvailVal);
Bob Wilson74175c22010-02-22 21:39:41 +00001775 VN.erase(L);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001776 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001777 ++NumGVNLoad;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001778 return true;
1779 }
1780
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001781 DEBUG(
1782 // fast print dep, using operator<< on instruction would be too slow
David Greenebf7f78e2010-01-05 01:27:17 +00001783 dbgs() << "GVN: load ";
1784 WriteAsOperand(dbgs(), L);
Chris Lattnerb2412a82009-09-21 02:42:51 +00001785 Instruction *I = Dep.getInst();
David Greenebf7f78e2010-01-05 01:27:17 +00001786 dbgs() << " is clobbered by " << *I << '\n';
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001787 );
Chris Lattnerb51deb92008-12-05 21:04:20 +00001788 return false;
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001789 }
Chris Lattnerb51deb92008-12-05 21:04:20 +00001790
1791 // If it is defined in another block, try harder.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001792 if (Dep.isNonLocal())
Chris Lattnerb51deb92008-12-05 21:04:20 +00001793 return processNonLocalLoad(L, toErase);
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001794
Chris Lattnerb2412a82009-09-21 02:42:51 +00001795 Instruction *DepInst = Dep.getInst();
Chris Lattnerb51deb92008-12-05 21:04:20 +00001796 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInst)) {
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001797 Value *StoredVal = DepSI->getOperand(0);
1798
1799 // The store and load are to a must-aliased pointer, but they may not
1800 // actually have the same type. See if we know how to reuse the stored
1801 // value (depending on its type).
1802 const TargetData *TD = 0;
Chris Lattnera52fce42009-10-21 04:11:19 +00001803 if (StoredVal->getType() != L->getType()) {
1804 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1805 StoredVal = CoerceAvailableValueToLoadType(StoredVal, L->getType(),
1806 L, *TD);
1807 if (StoredVal == 0)
1808 return false;
1809
David Greenebf7f78e2010-01-05 01:27:17 +00001810 DEBUG(dbgs() << "GVN COERCED STORE:\n" << *DepSI << '\n' << *StoredVal
Chris Lattnera52fce42009-10-21 04:11:19 +00001811 << '\n' << *L << "\n\n\n");
1812 }
1813 else
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001814 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001815 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001816
Chris Lattnerb51deb92008-12-05 21:04:20 +00001817 // Remove it!
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001818 L->replaceAllUsesWith(StoredVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001819 if (StoredVal->getType()->isPointerTy())
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001820 MD->invalidateCachedPointerInfo(StoredVal);
Bob Wilson74175c22010-02-22 21:39:41 +00001821 VN.erase(L);
Chris Lattnerb51deb92008-12-05 21:04:20 +00001822 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001823 ++NumGVNLoad;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001824 return true;
1825 }
1826
1827 if (LoadInst *DepLI = dyn_cast<LoadInst>(DepInst)) {
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001828 Value *AvailableVal = DepLI;
1829
1830 // The loads are of a must-aliased pointer, but they may not actually have
1831 // the same type. See if we know how to reuse the previously loaded value
1832 // (depending on its type).
1833 const TargetData *TD = 0;
Chris Lattnera52fce42009-10-21 04:11:19 +00001834 if (DepLI->getType() != L->getType()) {
1835 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1836 AvailableVal = CoerceAvailableValueToLoadType(DepLI, L->getType(), L,*TD);
1837 if (AvailableVal == 0)
1838 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001839
David Greenebf7f78e2010-01-05 01:27:17 +00001840 DEBUG(dbgs() << "GVN COERCED LOAD:\n" << *DepLI << "\n" << *AvailableVal
Chris Lattnera52fce42009-10-21 04:11:19 +00001841 << "\n" << *L << "\n\n\n");
1842 }
1843 else
1844 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001845 }
1846
Chris Lattnerb51deb92008-12-05 21:04:20 +00001847 // Remove it!
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001848 L->replaceAllUsesWith(AvailableVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001849 if (DepLI->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001850 MD->invalidateCachedPointerInfo(DepLI);
Bob Wilson74175c22010-02-22 21:39:41 +00001851 VN.erase(L);
Chris Lattnerb51deb92008-12-05 21:04:20 +00001852 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001853 ++NumGVNLoad;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001854 return true;
1855 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001856
Chris Lattner237a8282008-11-30 01:39:32 +00001857 // If this load really doesn't depend on anything, then we must be loading an
1858 // undef value. This can happen when loading for a fresh allocation with no
1859 // intervening stores, for example.
Victor Hernandez7b929da2009-10-23 21:09:37 +00001860 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst)) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001861 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson74175c22010-02-22 21:39:41 +00001862 VN.erase(L);
Chris Lattner237a8282008-11-30 01:39:32 +00001863 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001864 ++NumGVNLoad;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001865 return true;
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001866 }
Owen Andersonb62f7922009-10-28 07:05:35 +00001867
Owen Anderson9ff5a232009-12-02 07:35:19 +00001868 // If this load occurs either right after a lifetime begin,
Owen Andersonb62f7922009-10-28 07:05:35 +00001869 // then the loaded value is undefined.
1870 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(DepInst)) {
Owen Anderson9ff5a232009-12-02 07:35:19 +00001871 if (II->getIntrinsicID() == Intrinsic::lifetime_start) {
Owen Andersonb62f7922009-10-28 07:05:35 +00001872 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson74175c22010-02-22 21:39:41 +00001873 VN.erase(L);
Owen Andersonb62f7922009-10-28 07:05:35 +00001874 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001875 ++NumGVNLoad;
Owen Andersonb62f7922009-10-28 07:05:35 +00001876 return true;
1877 }
1878 }
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001879
Chris Lattnerb51deb92008-12-05 21:04:20 +00001880 return false;
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001881}
1882
Chris Lattnerb2412a82009-09-21 02:42:51 +00001883Value *GVN::lookupNumber(BasicBlock *BB, uint32_t num) {
Owen Andersonb70a5712008-06-23 17:49:45 +00001884 DenseMap<BasicBlock*, ValueNumberScope*>::iterator I = localAvail.find(BB);
1885 if (I == localAvail.end())
1886 return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001887
Chris Lattnerb2412a82009-09-21 02:42:51 +00001888 ValueNumberScope *Locals = I->second;
1889 while (Locals) {
1890 DenseMap<uint32_t, Value*>::iterator I = Locals->table.find(num);
1891 if (I != Locals->table.end())
Owen Anderson6fafe842008-06-20 01:15:47 +00001892 return I->second;
Chris Lattnerb2412a82009-09-21 02:42:51 +00001893 Locals = Locals->parent;
Owen Anderson6fafe842008-06-20 01:15:47 +00001894 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001895
Owen Anderson6fafe842008-06-20 01:15:47 +00001896 return 0;
1897}
1898
Owen Anderson255dafc2008-12-15 02:03:00 +00001899
Owen Anderson36057c72007-08-14 18:16:29 +00001900/// processInstruction - When calculating availability, handle an instruction
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001901/// by inserting it into the appropriate sets
Owen Andersonaf4240a2008-06-12 19:25:32 +00001902bool GVN::processInstruction(Instruction *I,
Chris Lattner8e1e95c2008-03-21 22:01:16 +00001903 SmallVectorImpl<Instruction*> &toErase) {
Devang Patelbe905e22010-02-11 00:20:49 +00001904 // Ignore dbg info intrinsics.
1905 if (isa<DbgInfoIntrinsic>(I))
1906 return false;
1907
Chris Lattnerb2412a82009-09-21 02:42:51 +00001908 if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
1909 bool Changed = processLoad(LI, toErase);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001910
Chris Lattnerb2412a82009-09-21 02:42:51 +00001911 if (!Changed) {
1912 unsigned Num = VN.lookup_or_add(LI);
1913 localAvail[I->getParent()]->table.insert(std::make_pair(Num, LI));
Owen Andersonb2303722008-06-18 21:41:49 +00001914 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001915
Chris Lattnerb2412a82009-09-21 02:42:51 +00001916 return Changed;
Owen Andersonb2303722008-06-18 21:41:49 +00001917 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001918
Chris Lattnerb2412a82009-09-21 02:42:51 +00001919 uint32_t NextNum = VN.getNextUnusedValueNumber();
1920 unsigned Num = VN.lookup_or_add(I);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001921
Chris Lattnerb2412a82009-09-21 02:42:51 +00001922 if (BranchInst *BI = dyn_cast<BranchInst>(I)) {
1923 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001924
Owen Andersone8a290f2009-04-01 23:53:49 +00001925 if (!BI->isConditional() || isa<Constant>(BI->getCondition()))
1926 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001927
Chris Lattnerb2412a82009-09-21 02:42:51 +00001928 Value *BranchCond = BI->getCondition();
1929 uint32_t CondVN = VN.lookup_or_add(BranchCond);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001930
Chris Lattnerb2412a82009-09-21 02:42:51 +00001931 BasicBlock *TrueSucc = BI->getSuccessor(0);
1932 BasicBlock *FalseSucc = BI->getSuccessor(1);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001933
Chris Lattnerb2412a82009-09-21 02:42:51 +00001934 if (TrueSucc->getSinglePredecessor())
1935 localAvail[TrueSucc]->table[CondVN] =
1936 ConstantInt::getTrue(TrueSucc->getContext());
1937 if (FalseSucc->getSinglePredecessor())
1938 localAvail[FalseSucc]->table[CondVN] =
1939 ConstantInt::getFalse(TrueSucc->getContext());
Owen Andersone8a290f2009-04-01 23:53:49 +00001940
1941 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001942
Owen Andersone5ffa902008-04-07 09:59:07 +00001943 // Allocations are always uniquely numbered, so we can save time and memory
Daniel Dunbara279bc32009-09-20 02:20:51 +00001944 // by fast failing them.
Victor Hernandez7b929da2009-10-23 21:09:37 +00001945 } else if (isa<AllocaInst>(I) || isa<TerminatorInst>(I)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001946 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Andersone5ffa902008-04-07 09:59:07 +00001947 return false;
Owen Andersonb2303722008-06-18 21:41:49 +00001948 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001949
Owen Anderson62bc33c2007-08-16 22:02:55 +00001950 // Collapse PHI nodes
Owen Anderson31f49672007-08-14 18:33:27 +00001951 if (PHINode* p = dyn_cast<PHINode>(I)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001952 Value *constVal = CollapsePhi(p);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001953
Owen Anderson31f49672007-08-14 18:33:27 +00001954 if (constVal) {
Owen Anderson1defe2d2007-08-16 22:51:56 +00001955 p->replaceAllUsesWith(constVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001956 if (MD && constVal->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001957 MD->invalidateCachedPointerInfo(constVal);
Owen Andersonae53c932008-12-23 00:49:51 +00001958 VN.erase(p);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001959
Owen Anderson1defe2d2007-08-16 22:51:56 +00001960 toErase.push_back(p);
Owen Andersonb2303722008-06-18 21:41:49 +00001961 } else {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001962 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson31f49672007-08-14 18:33:27 +00001963 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001964
Owen Anderson0ae33ef2008-07-03 17:44:33 +00001965 // If the number we were assigned was a brand new VN, then we don't
1966 // need to do a lookup to see if the number already exists
1967 // somewhere in the domtree: it can't!
Chris Lattnerb2412a82009-09-21 02:42:51 +00001968 } else if (Num == NextNum) {
1969 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001970
Owen Anderson255dafc2008-12-15 02:03:00 +00001971 // Perform fast-path value-number based elimination of values inherited from
1972 // dominators.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001973 } else if (Value *repl = lookupNumber(I->getParent(), Num)) {
Owen Anderson5fc4aba2007-12-08 01:37:09 +00001974 // Remove it!
Owen Andersonbf7d0bc2007-07-31 23:27:13 +00001975 VN.erase(I);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001976 I->replaceAllUsesWith(repl);
Duncan Sands1df98592010-02-16 11:11:14 +00001977 if (MD && repl->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001978 MD->invalidateCachedPointerInfo(repl);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001979 toErase.push_back(I);
1980 return true;
Owen Anderson255dafc2008-12-15 02:03:00 +00001981
Owen Anderson0ae33ef2008-07-03 17:44:33 +00001982 } else {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001983 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001984 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001985
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001986 return false;
1987}
1988
Bill Wendling30788b82008-12-22 22:32:22 +00001989/// runOnFunction - This is the main transformation entry point for a function.
Owen Anderson3e75a422007-08-14 18:04:11 +00001990bool GVN::runOnFunction(Function& F) {
Dan Gohman4ec01b22009-11-14 02:27:51 +00001991 if (!NoLoads)
1992 MD = &getAnalysis<MemoryDependenceAnalysis>();
Chris Lattner663e4412008-12-01 00:40:32 +00001993 DT = &getAnalysis<DominatorTree>();
Owen Andersona472c4a2008-05-12 20:15:55 +00001994 VN.setAliasAnalysis(&getAnalysis<AliasAnalysis>());
Chris Lattner663e4412008-12-01 00:40:32 +00001995 VN.setMemDep(MD);
1996 VN.setDomTree(DT);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001997
Chris Lattnerb2412a82009-09-21 02:42:51 +00001998 bool Changed = false;
1999 bool ShouldContinue = true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002000
Owen Anderson5d0af032008-07-16 17:52:31 +00002001 // Merge unconditional branches, allowing PRE to catch more
2002 // optimization opportunities.
2003 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002004 BasicBlock *BB = FI;
Owen Anderson5d0af032008-07-16 17:52:31 +00002005 ++FI;
Owen Andersonb31b06d2008-07-17 00:01:40 +00002006 bool removedBlock = MergeBlockIntoPredecessor(BB, this);
Dan Gohmanfe601042010-06-22 15:08:57 +00002007 if (removedBlock) ++NumGVNBlocks;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002008
Chris Lattnerb2412a82009-09-21 02:42:51 +00002009 Changed |= removedBlock;
Owen Anderson5d0af032008-07-16 17:52:31 +00002010 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002011
Chris Lattnerae199312008-12-09 19:21:47 +00002012 unsigned Iteration = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002013
Chris Lattnerb2412a82009-09-21 02:42:51 +00002014 while (ShouldContinue) {
David Greenebf7f78e2010-01-05 01:27:17 +00002015 DEBUG(dbgs() << "GVN iteration: " << Iteration << "\n");
Chris Lattnerb2412a82009-09-21 02:42:51 +00002016 ShouldContinue = iterateOnFunction(F);
Bob Wilson484d4a32010-02-16 19:51:59 +00002017 if (splitCriticalEdges())
2018 ShouldContinue = true;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002019 Changed |= ShouldContinue;
Chris Lattnerae199312008-12-09 19:21:47 +00002020 ++Iteration;
Owen Anderson3e75a422007-08-14 18:04:11 +00002021 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002022
Owen Andersone98c54c2008-07-18 18:03:38 +00002023 if (EnablePRE) {
Owen Anderson0c7f91c2008-09-03 23:06:07 +00002024 bool PREChanged = true;
2025 while (PREChanged) {
2026 PREChanged = performPRE(F);
Chris Lattnerb2412a82009-09-21 02:42:51 +00002027 Changed |= PREChanged;
Owen Anderson0c7f91c2008-09-03 23:06:07 +00002028 }
Owen Andersone98c54c2008-07-18 18:03:38 +00002029 }
Chris Lattnerae199312008-12-09 19:21:47 +00002030 // FIXME: Should perform GVN again after PRE does something. PRE can move
2031 // computations into blocks where they become fully redundant. Note that
2032 // we can't do this until PRE's critical edge splitting updates memdep.
2033 // Actually, when this happens, we should just fully integrate PRE into GVN.
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002034
2035 cleanupGlobalSets();
2036
Chris Lattnerb2412a82009-09-21 02:42:51 +00002037 return Changed;
Owen Anderson3e75a422007-08-14 18:04:11 +00002038}
2039
2040
Chris Lattnerb2412a82009-09-21 02:42:51 +00002041bool GVN::processBlock(BasicBlock *BB) {
Chris Lattnerae199312008-12-09 19:21:47 +00002042 // FIXME: Kill off toErase by doing erasing eagerly in a helper function (and
2043 // incrementing BI before processing an instruction).
Owen Andersonaf4240a2008-06-12 19:25:32 +00002044 SmallVector<Instruction*, 8> toErase;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002045 bool ChangedFunction = false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002046
Owen Andersonaf4240a2008-06-12 19:25:32 +00002047 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
2048 BI != BE;) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002049 ChangedFunction |= processInstruction(BI, toErase);
Owen Andersonaf4240a2008-06-12 19:25:32 +00002050 if (toErase.empty()) {
2051 ++BI;
2052 continue;
2053 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002054
Owen Andersonaf4240a2008-06-12 19:25:32 +00002055 // If we need some instructions deleted, do it now.
2056 NumGVNInstr += toErase.size();
Daniel Dunbara279bc32009-09-20 02:20:51 +00002057
Owen Andersonaf4240a2008-06-12 19:25:32 +00002058 // Avoid iterator invalidation.
2059 bool AtStart = BI == BB->begin();
2060 if (!AtStart)
2061 --BI;
2062
2063 for (SmallVector<Instruction*, 4>::iterator I = toErase.begin(),
Chris Lattner663e4412008-12-01 00:40:32 +00002064 E = toErase.end(); I != E; ++I) {
David Greenebf7f78e2010-01-05 01:27:17 +00002065 DEBUG(dbgs() << "GVN removed: " << **I << '\n');
Dan Gohman4ec01b22009-11-14 02:27:51 +00002066 if (MD) MD->removeInstruction(*I);
Owen Andersonaf4240a2008-06-12 19:25:32 +00002067 (*I)->eraseFromParent();
Bill Wendlingec40d502008-12-22 21:57:30 +00002068 DEBUG(verifyRemoved(*I));
Chris Lattner663e4412008-12-01 00:40:32 +00002069 }
Chris Lattnerae199312008-12-09 19:21:47 +00002070 toErase.clear();
Owen Andersonaf4240a2008-06-12 19:25:32 +00002071
2072 if (AtStart)
2073 BI = BB->begin();
2074 else
2075 ++BI;
Owen Andersonaf4240a2008-06-12 19:25:32 +00002076 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002077
Chris Lattnerb2412a82009-09-21 02:42:51 +00002078 return ChangedFunction;
Owen Andersonaf4240a2008-06-12 19:25:32 +00002079}
2080
Owen Andersonb2303722008-06-18 21:41:49 +00002081/// performPRE - Perform a purely local form of PRE that looks for diamond
2082/// control flow patterns and attempts to perform simple PRE at the join point.
Chris Lattnerfb6e7012009-10-31 22:11:15 +00002083bool GVN::performPRE(Function &F) {
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002084 bool Changed = false;
Chris Lattner09713792008-12-01 07:29:03 +00002085 DenseMap<BasicBlock*, Value*> predMap;
Owen Andersonb2303722008-06-18 21:41:49 +00002086 for (df_iterator<BasicBlock*> DI = df_begin(&F.getEntryBlock()),
2087 DE = df_end(&F.getEntryBlock()); DI != DE; ++DI) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002088 BasicBlock *CurrentBlock = *DI;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002089
Owen Andersonb2303722008-06-18 21:41:49 +00002090 // Nothing to PRE in the entry block.
2091 if (CurrentBlock == &F.getEntryBlock()) continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002092
Owen Andersonb2303722008-06-18 21:41:49 +00002093 for (BasicBlock::iterator BI = CurrentBlock->begin(),
2094 BE = CurrentBlock->end(); BI != BE; ) {
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002095 Instruction *CurInst = BI++;
Duncan Sands7af1c782009-05-06 06:49:50 +00002096
Victor Hernandez7b929da2009-10-23 21:09:37 +00002097 if (isa<AllocaInst>(CurInst) ||
Victor Hernandez83d63912009-09-18 22:35:49 +00002098 isa<TerminatorInst>(CurInst) || isa<PHINode>(CurInst) ||
Devang Patel9674d152009-10-14 17:29:00 +00002099 CurInst->getType()->isVoidTy() ||
Duncan Sands7af1c782009-05-06 06:49:50 +00002100 CurInst->mayReadFromMemory() || CurInst->mayHaveSideEffects() ||
John Criswell090c0a22009-03-10 15:04:53 +00002101 isa<DbgInfoIntrinsic>(CurInst))
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002102 continue;
Owen Anderson5015b342010-08-07 00:20:35 +00002103
2104 // We don't currently value number ANY inline asm calls.
2105 if (CallInst *CallI = dyn_cast<CallInst>(CurInst))
2106 if (CallI->isInlineAsm())
2107 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) {
Gabor Greif08149852010-07-09 14:36:49 +00002124 BasicBlock *P = *PI;
Owen Andersonb2303722008-06-18 21:41:49 +00002125 // We're not interested in PRE where the block is its
Bob Wilsone7b635f2010-02-03 00:33:21 +00002126 // own predecessor, or in blocks with predecessors
Owen Anderson6fafe842008-06-20 01:15:47 +00002127 // that are not reachable.
Gabor Greif08149852010-07-09 14:36:49 +00002128 if (P == CurrentBlock) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002129 NumWithout = 2;
Owen Anderson6fafe842008-06-20 01:15:47 +00002130 break;
Gabor Greif08149852010-07-09 14:36:49 +00002131 } else if (!localAvail.count(P)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002132 NumWithout = 2;
Owen Anderson6fafe842008-06-20 01:15:47 +00002133 break;
2134 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002135
2136 DenseMap<uint32_t, Value*>::iterator predV =
Gabor Greif08149852010-07-09 14:36:49 +00002137 localAvail[P]->table.find(ValNo);
2138 if (predV == localAvail[P]->table.end()) {
2139 PREPred = P;
Dan Gohmanfe601042010-06-22 15:08:57 +00002140 ++NumWithout;
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002141 } else if (predV->second == CurInst) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002142 NumWithout = 2;
Owen Andersonb2303722008-06-18 21:41:49 +00002143 } else {
Gabor Greif08149852010-07-09 14:36:49 +00002144 predMap[P] = predV->second;
Dan Gohmanfe601042010-06-22 15:08:57 +00002145 ++NumWith;
Owen Andersonb2303722008-06-18 21:41:49 +00002146 }
2147 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002148
Owen Andersonb2303722008-06-18 21:41:49 +00002149 // Don't do PRE when it might increase code size, i.e. when
2150 // we would need to insert instructions in more than one pred.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002151 if (NumWithout != 1 || NumWith == 0)
Owen Andersonb2303722008-06-18 21:41:49 +00002152 continue;
Chris Lattnerfb6e7012009-10-31 22:11:15 +00002153
2154 // Don't do PRE across indirect branch.
2155 if (isa<IndirectBrInst>(PREPred->getTerminator()))
2156 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002157
Owen Anderson5c274ee2008-06-19 19:54:19 +00002158 // We can't do PRE safely on a critical edge, so instead we schedule
2159 // the edge to be split and perform the PRE the next time we iterate
2160 // on the function.
Bob Wilsonae23daf2010-02-16 21:06:42 +00002161 unsigned SuccNum = GetSuccessorNumber(PREPred, CurrentBlock);
Chris Lattnerb2412a82009-09-21 02:42:51 +00002162 if (isCriticalEdge(PREPred->getTerminator(), SuccNum)) {
2163 toSplit.push_back(std::make_pair(PREPred->getTerminator(), SuccNum));
Owen Anderson5c274ee2008-06-19 19:54:19 +00002164 continue;
2165 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002166
Bob Wilsone7b635f2010-02-03 00:33:21 +00002167 // Instantiate the expression in the predecessor that lacked it.
Owen Andersonb2303722008-06-18 21:41:49 +00002168 // Because we are going top-down through the block, all value numbers
2169 // will be available in the predecessor by the time we need them. Any
Bob Wilsone7b635f2010-02-03 00:33:21 +00002170 // that weren't originally present will have been instantiated earlier
Owen Andersonb2303722008-06-18 21:41:49 +00002171 // in this loop.
Nick Lewycky67760642009-09-27 07:38:41 +00002172 Instruction *PREInstr = CurInst->clone();
Owen Andersonb2303722008-06-18 21:41:49 +00002173 bool success = true;
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002174 for (unsigned i = 0, e = CurInst->getNumOperands(); i != e; ++i) {
2175 Value *Op = PREInstr->getOperand(i);
2176 if (isa<Argument>(Op) || isa<Constant>(Op) || isa<GlobalValue>(Op))
2177 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002178
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002179 if (Value *V = lookupNumber(PREPred, VN.lookup(Op))) {
2180 PREInstr->setOperand(i, V);
2181 } else {
2182 success = false;
2183 break;
Owen Andersonc45996b2008-07-11 20:05:13 +00002184 }
Owen Andersonb2303722008-06-18 21:41:49 +00002185 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002186
Owen Andersonb2303722008-06-18 21:41:49 +00002187 // Fail out if we encounter an operand that is not available in
Daniel Dunbara279bc32009-09-20 02:20:51 +00002188 // the PRE predecessor. This is typically because of loads which
Owen Andersonb2303722008-06-18 21:41:49 +00002189 // are not value numbered precisely.
2190 if (!success) {
2191 delete PREInstr;
Bill Wendling70ded192008-12-22 22:14:07 +00002192 DEBUG(verifyRemoved(PREInstr));
Owen Andersonb2303722008-06-18 21:41:49 +00002193 continue;
2194 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002195
Owen Andersonb2303722008-06-18 21:41:49 +00002196 PREInstr->insertBefore(PREPred->getTerminator());
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002197 PREInstr->setName(CurInst->getName() + ".pre");
Owen Anderson6fafe842008-06-20 01:15:47 +00002198 predMap[PREPred] = PREInstr;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002199 VN.add(PREInstr, ValNo);
Dan Gohmanfe601042010-06-22 15:08:57 +00002200 ++NumGVNPRE;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002201
Owen Andersonb2303722008-06-18 21:41:49 +00002202 // Update the availability map to include the new instruction.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002203 localAvail[PREPred]->table.insert(std::make_pair(ValNo, PREInstr));
Daniel Dunbara279bc32009-09-20 02:20:51 +00002204
Owen Andersonb2303722008-06-18 21:41:49 +00002205 // Create a PHI to make the value available in this block.
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002206 PHINode* Phi = PHINode::Create(CurInst->getType(),
2207 CurInst->getName() + ".pre-phi",
Owen Andersonb2303722008-06-18 21:41:49 +00002208 CurrentBlock->begin());
2209 for (pred_iterator PI = pred_begin(CurrentBlock),
Gabor Greif1d3ae022010-07-09 14:48:08 +00002210 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
2211 BasicBlock *P = *PI;
2212 Phi->addIncoming(predMap[P], P);
2213 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002214
Chris Lattnerb2412a82009-09-21 02:42:51 +00002215 VN.add(Phi, ValNo);
2216 localAvail[CurrentBlock]->table[ValNo] = Phi;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002217
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002218 CurInst->replaceAllUsesWith(Phi);
Duncan Sands1df98592010-02-16 11:11:14 +00002219 if (MD && Phi->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00002220 MD->invalidateCachedPointerInfo(Phi);
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002221 VN.erase(CurInst);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002222
David Greenebf7f78e2010-01-05 01:27:17 +00002223 DEBUG(dbgs() << "GVN PRE removed: " << *CurInst << '\n');
Dan Gohman4ec01b22009-11-14 02:27:51 +00002224 if (MD) MD->removeInstruction(CurInst);
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002225 CurInst->eraseFromParent();
Bill Wendlingec40d502008-12-22 21:57:30 +00002226 DEBUG(verifyRemoved(CurInst));
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002227 Changed = true;
Owen Andersonb2303722008-06-18 21:41:49 +00002228 }
2229 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002230
Bob Wilson484d4a32010-02-16 19:51:59 +00002231 if (splitCriticalEdges())
2232 Changed = true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002233
Bob Wilson484d4a32010-02-16 19:51:59 +00002234 return Changed;
2235}
2236
2237/// splitCriticalEdges - Split critical edges found during the previous
2238/// iteration that may enable further optimization.
2239bool GVN::splitCriticalEdges() {
2240 if (toSplit.empty())
2241 return false;
2242 do {
2243 std::pair<TerminatorInst*, unsigned> Edge = toSplit.pop_back_val();
2244 SplitCriticalEdge(Edge.first, Edge.second, this);
2245 } while (!toSplit.empty());
Evan Cheng19d417c2010-03-01 22:23:12 +00002246 if (MD) MD->invalidateCachedPredecessors();
Bob Wilson484d4a32010-02-16 19:51:59 +00002247 return true;
Owen Andersonb2303722008-06-18 21:41:49 +00002248}
2249
Bill Wendling30788b82008-12-22 22:32:22 +00002250/// iterateOnFunction - Executes one iteration of GVN
Owen Anderson3e75a422007-08-14 18:04:11 +00002251bool GVN::iterateOnFunction(Function &F) {
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002252 cleanupGlobalSets();
Chris Lattner2e607012008-03-21 21:33:23 +00002253
Owen Andersone8a290f2009-04-01 23:53:49 +00002254 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2255 DE = df_end(DT->getRootNode()); DI != DE; ++DI) {
2256 if (DI->getIDom())
2257 localAvail[DI->getBlock()] =
2258 new ValueNumberScope(localAvail[DI->getIDom()->getBlock()]);
2259 else
2260 localAvail[DI->getBlock()] = new ValueNumberScope(0);
2261 }
2262
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002263 // Top-down walk of the dominator tree
Chris Lattnerb2412a82009-09-21 02:42:51 +00002264 bool Changed = false;
Owen Andersonc34d1122008-12-15 03:52:17 +00002265#if 0
2266 // Needed for value numbering with phi construction to work.
Owen Anderson255dafc2008-12-15 02:03:00 +00002267 ReversePostOrderTraversal<Function*> RPOT(&F);
2268 for (ReversePostOrderTraversal<Function*>::rpo_iterator RI = RPOT.begin(),
2269 RE = RPOT.end(); RI != RE; ++RI)
Chris Lattnerb2412a82009-09-21 02:42:51 +00002270 Changed |= processBlock(*RI);
Owen Andersonc34d1122008-12-15 03:52:17 +00002271#else
2272 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2273 DE = df_end(DT->getRootNode()); DI != DE; ++DI)
Chris Lattnerb2412a82009-09-21 02:42:51 +00002274 Changed |= processBlock(DI->getBlock());
Owen Andersonc34d1122008-12-15 03:52:17 +00002275#endif
2276
Chris Lattnerb2412a82009-09-21 02:42:51 +00002277 return Changed;
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002278}
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002279
2280void GVN::cleanupGlobalSets() {
2281 VN.clear();
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002282
2283 for (DenseMap<BasicBlock*, ValueNumberScope*>::iterator
2284 I = localAvail.begin(), E = localAvail.end(); I != E; ++I)
2285 delete I->second;
2286 localAvail.clear();
2287}
Bill Wendling246dbbb2008-12-22 21:36:08 +00002288
2289/// verifyRemoved - Verify that the specified instruction does not occur in our
2290/// internal data structures.
Bill Wendling6d463f22008-12-22 22:28:56 +00002291void GVN::verifyRemoved(const Instruction *Inst) const {
2292 VN.verifyRemoved(Inst);
Bill Wendling70ded192008-12-22 22:14:07 +00002293
Bill Wendling6d463f22008-12-22 22:28:56 +00002294 // Walk through the value number scope to make sure the instruction isn't
2295 // ferreted away in it.
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +00002296 for (DenseMap<BasicBlock*, ValueNumberScope*>::const_iterator
Bill Wendling6d463f22008-12-22 22:28:56 +00002297 I = localAvail.begin(), E = localAvail.end(); I != E; ++I) {
2298 const ValueNumberScope *VNS = I->second;
2299
2300 while (VNS) {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +00002301 for (DenseMap<uint32_t, Value*>::const_iterator
Bill Wendling6d463f22008-12-22 22:28:56 +00002302 II = VNS->table.begin(), IE = VNS->table.end(); II != IE; ++II) {
2303 assert(II->second != Inst && "Inst still in value numbering scope!");
2304 }
2305
2306 VNS = VNS->parent;
Bill Wendling70ded192008-12-22 22:14:07 +00002307 }
2308 }
Bill Wendling246dbbb2008-12-22 21:36:08 +00002309}