blob: 649f9cb9607fe6a260f7c558dffbcb9a74b94e5c [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));
Bob Wilsonb29d7d22010-02-28 05:34:05 +000064static cl::opt<bool> EnableFullLoadPRE("enable-full-load-pre", cl::init(false));
Owen Andersonaa0b6342008-06-19 19:57:25 +000065
Owen Anderson1ad2cb72007-07-24 17:55:58 +000066//===----------------------------------------------------------------------===//
67// ValueTable Class
68//===----------------------------------------------------------------------===//
69
70/// This class holds the mapping between values and value numbers. It is used
71/// as an efficient mechanism to determine the expression-wise equivalence of
72/// two values.
73namespace {
Chris Lattner3e8b6632009-09-02 06:11:42 +000074 struct Expression {
Owen Andersona81e2412010-01-17 19:33:27 +000075 enum ExpressionOpcode {
76 ADD = Instruction::Add,
77 FADD = Instruction::FAdd,
78 SUB = Instruction::Sub,
79 FSUB = Instruction::FSub,
80 MUL = Instruction::Mul,
81 FMUL = Instruction::FMul,
82 UDIV = Instruction::UDiv,
83 SDIV = Instruction::SDiv,
84 FDIV = Instruction::FDiv,
85 UREM = Instruction::URem,
86 SREM = Instruction::SRem,
87 FREM = Instruction::FRem,
88 SHL = Instruction::Shl,
89 LSHR = Instruction::LShr,
90 ASHR = Instruction::AShr,
91 AND = Instruction::And,
92 OR = Instruction::Or,
93 XOR = Instruction::Xor,
94 TRUNC = Instruction::Trunc,
95 ZEXT = Instruction::ZExt,
96 SEXT = Instruction::SExt,
97 FPTOUI = Instruction::FPToUI,
98 FPTOSI = Instruction::FPToSI,
99 UITOFP = Instruction::UIToFP,
100 SITOFP = Instruction::SIToFP,
101 FPTRUNC = Instruction::FPTrunc,
102 FPEXT = Instruction::FPExt,
103 PTRTOINT = Instruction::PtrToInt,
104 INTTOPTR = Instruction::IntToPtr,
105 BITCAST = Instruction::BitCast,
106 ICMPEQ, ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
107 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
108 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
109 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
110 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
111 SHUFFLE, SELECT, GEP, CALL, CONSTANT,
112 INSERTVALUE, EXTRACTVALUE, EMPTY, TOMBSTONE };
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000113
114 ExpressionOpcode opcode;
115 const Type* type;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000116 SmallVector<uint32_t, 4> varargs;
Chris Lattnerb2412a82009-09-21 02:42:51 +0000117 Value *function;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000118
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000119 Expression() { }
120 Expression(ExpressionOpcode o) : opcode(o) { }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000121
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000122 bool operator==(const Expression &other) const {
123 if (opcode != other.opcode)
124 return false;
125 else if (opcode == EMPTY || opcode == TOMBSTONE)
126 return true;
127 else if (type != other.type)
128 return false;
Owen Andersonb388ca92007-10-18 19:39:33 +0000129 else if (function != other.function)
130 return false;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000131 else {
132 if (varargs.size() != other.varargs.size())
133 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000134
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000135 for (size_t i = 0; i < varargs.size(); ++i)
136 if (varargs[i] != other.varargs[i])
137 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000138
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000139 return true;
140 }
141 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000142
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000143 bool operator!=(const Expression &other) const {
Bill Wendling75f02ee2008-12-22 22:16:31 +0000144 return !(*this == other);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000145 }
146 };
Daniel Dunbara279bc32009-09-20 02:20:51 +0000147
Chris Lattner3e8b6632009-09-02 06:11:42 +0000148 class ValueTable {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000149 private:
150 DenseMap<Value*, uint32_t> valueNumbering;
151 DenseMap<Expression, uint32_t> expressionNumbering;
Owen Andersona472c4a2008-05-12 20:15:55 +0000152 AliasAnalysis* AA;
153 MemoryDependenceAnalysis* MD;
154 DominatorTree* DT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000155
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000156 uint32_t nextValueNumber;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000157
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000158 Expression::ExpressionOpcode getOpcode(CmpInst* C);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000159 Expression create_expression(BinaryOperator* BO);
160 Expression create_expression(CmpInst* C);
161 Expression create_expression(ShuffleVectorInst* V);
162 Expression create_expression(ExtractElementInst* C);
163 Expression create_expression(InsertElementInst* V);
164 Expression create_expression(SelectInst* V);
165 Expression create_expression(CastInst* C);
166 Expression create_expression(GetElementPtrInst* G);
Owen Andersonb388ca92007-10-18 19:39:33 +0000167 Expression create_expression(CallInst* C);
Owen Anderson3b3f58c2008-05-13 08:17:22 +0000168 Expression create_expression(Constant* C);
Owen Andersond41ed4e2009-10-19 22:14:22 +0000169 Expression create_expression(ExtractValueInst* C);
170 Expression create_expression(InsertValueInst* C);
171
172 uint32_t lookup_or_add_call(CallInst* C);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000173 public:
Dan Gohmane63c4a22009-04-01 16:37:47 +0000174 ValueTable() : nextValueNumber(1) { }
Chris Lattnerb2412a82009-09-21 02:42:51 +0000175 uint32_t lookup_or_add(Value *V);
176 uint32_t lookup(Value *V) const;
177 void add(Value *V, uint32_t num);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000178 void clear();
Chris Lattnerb2412a82009-09-21 02:42:51 +0000179 void erase(Value *v);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000180 unsigned size();
Owen Andersona472c4a2008-05-12 20:15:55 +0000181 void setAliasAnalysis(AliasAnalysis* A) { AA = A; }
Chris Lattner663e4412008-12-01 00:40:32 +0000182 AliasAnalysis *getAliasAnalysis() const { return AA; }
Owen Andersona472c4a2008-05-12 20:15:55 +0000183 void setMemDep(MemoryDependenceAnalysis* M) { MD = M; }
184 void setDomTree(DominatorTree* D) { DT = D; }
Owen Anderson0ae33ef2008-07-03 17:44:33 +0000185 uint32_t getNextUnusedValueNumber() { return nextValueNumber; }
Bill Wendling246dbbb2008-12-22 21:36:08 +0000186 void verifyRemoved(const Value *) const;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000187 };
188}
189
190namespace llvm {
Chris Lattner76c1b972007-09-17 18:34:04 +0000191template <> struct DenseMapInfo<Expression> {
Owen Anderson830db6a2007-08-02 18:16:06 +0000192 static inline Expression getEmptyKey() {
193 return Expression(Expression::EMPTY);
194 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000195
Owen Anderson830db6a2007-08-02 18:16:06 +0000196 static inline Expression getTombstoneKey() {
197 return Expression(Expression::TOMBSTONE);
198 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000199
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000200 static unsigned getHashValue(const Expression e) {
201 unsigned hash = e.opcode;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000202
Anton Korobeynikov07e6e562008-02-20 11:26:25 +0000203 hash = ((unsigned)((uintptr_t)e.type >> 4) ^
Owen Andersond41ed4e2009-10-19 22:14:22 +0000204 (unsigned)((uintptr_t)e.type >> 9));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000205
Owen Anderson830db6a2007-08-02 18:16:06 +0000206 for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
207 E = e.varargs.end(); I != E; ++I)
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000208 hash = *I + hash * 37;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000209
Anton Korobeynikov07e6e562008-02-20 11:26:25 +0000210 hash = ((unsigned)((uintptr_t)e.function >> 4) ^
211 (unsigned)((uintptr_t)e.function >> 9)) +
212 hash * 37;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000213
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000214 return hash;
215 }
Chris Lattner76c1b972007-09-17 18:34:04 +0000216 static bool isEqual(const Expression &LHS, const Expression &RHS) {
217 return LHS == RHS;
218 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000219};
Chris Lattner4bbf4ee2009-12-15 07:26:43 +0000220
221template <>
222struct isPodLike<Expression> { static const bool value = true; };
223
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000224}
225
226//===----------------------------------------------------------------------===//
227// ValueTable Internal Functions
228//===----------------------------------------------------------------------===//
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000229
230Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000231 if (isa<ICmpInst>(C)) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000232 switch (C->getPredicate()) {
Chris Lattner88365bb2008-03-21 21:14:38 +0000233 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000234 llvm_unreachable("Comparison with unknown predicate?");
Chris Lattner88365bb2008-03-21 21:14:38 +0000235 case ICmpInst::ICMP_EQ: return Expression::ICMPEQ;
236 case ICmpInst::ICMP_NE: return Expression::ICMPNE;
237 case ICmpInst::ICMP_UGT: return Expression::ICMPUGT;
238 case ICmpInst::ICMP_UGE: return Expression::ICMPUGE;
239 case ICmpInst::ICMP_ULT: return Expression::ICMPULT;
240 case ICmpInst::ICMP_ULE: return Expression::ICMPULE;
241 case ICmpInst::ICMP_SGT: return Expression::ICMPSGT;
242 case ICmpInst::ICMP_SGE: return Expression::ICMPSGE;
243 case ICmpInst::ICMP_SLT: return Expression::ICMPSLT;
244 case ICmpInst::ICMP_SLE: return Expression::ICMPSLE;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000245 }
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000246 } else {
247 switch (C->getPredicate()) {
248 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000249 llvm_unreachable("Comparison with unknown predicate?");
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000250 case FCmpInst::FCMP_OEQ: return Expression::FCMPOEQ;
251 case FCmpInst::FCMP_OGT: return Expression::FCMPOGT;
252 case FCmpInst::FCMP_OGE: return Expression::FCMPOGE;
253 case FCmpInst::FCMP_OLT: return Expression::FCMPOLT;
254 case FCmpInst::FCMP_OLE: return Expression::FCMPOLE;
255 case FCmpInst::FCMP_ONE: return Expression::FCMPONE;
256 case FCmpInst::FCMP_ORD: return Expression::FCMPORD;
257 case FCmpInst::FCMP_UNO: return Expression::FCMPUNO;
258 case FCmpInst::FCMP_UEQ: return Expression::FCMPUEQ;
259 case FCmpInst::FCMP_UGT: return Expression::FCMPUGT;
260 case FCmpInst::FCMP_UGE: return Expression::FCMPUGE;
261 case FCmpInst::FCMP_ULT: return Expression::FCMPULT;
262 case FCmpInst::FCMP_ULE: return Expression::FCMPULE;
263 case FCmpInst::FCMP_UNE: return Expression::FCMPUNE;
264 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000265 }
266}
267
Owen Andersonb388ca92007-10-18 19:39:33 +0000268Expression ValueTable::create_expression(CallInst* C) {
269 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000270
Owen Andersonb388ca92007-10-18 19:39:33 +0000271 e.type = C->getType();
Owen Andersonb388ca92007-10-18 19:39:33 +0000272 e.function = C->getCalledFunction();
273 e.opcode = Expression::CALL;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000274
Eric Christopher551754c2010-04-16 23:37:20 +0000275 for (CallInst::op_iterator I = C->op_begin()+1, E = C->op_end();
Owen Andersonb388ca92007-10-18 19:39:33 +0000276 I != E; ++I)
Owen Anderson8f46c782008-04-11 05:11:49 +0000277 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000278
Owen Andersonb388ca92007-10-18 19:39:33 +0000279 return e;
280}
281
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000282Expression ValueTable::create_expression(BinaryOperator* BO) {
283 Expression e;
Owen Andersond41ed4e2009-10-19 22:14:22 +0000284 e.varargs.push_back(lookup_or_add(BO->getOperand(0)));
285 e.varargs.push_back(lookup_or_add(BO->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000286 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000287 e.type = BO->getType();
Owen Andersona81e2412010-01-17 19:33:27 +0000288 e.opcode = static_cast<Expression::ExpressionOpcode>(BO->getOpcode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000289
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000290 return e;
291}
292
293Expression ValueTable::create_expression(CmpInst* C) {
294 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000295
Owen Andersond41ed4e2009-10-19 22:14:22 +0000296 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
297 e.varargs.push_back(lookup_or_add(C->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000298 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000299 e.type = C->getType();
300 e.opcode = getOpcode(C);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000301
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000302 return e;
303}
304
305Expression ValueTable::create_expression(CastInst* C) {
306 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000307
Owen Andersond41ed4e2009-10-19 22:14:22 +0000308 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000309 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000310 e.type = C->getType();
Owen Andersona81e2412010-01-17 19:33:27 +0000311 e.opcode = static_cast<Expression::ExpressionOpcode>(C->getOpcode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000312
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000313 return e;
314}
315
316Expression ValueTable::create_expression(ShuffleVectorInst* S) {
317 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000318
Owen Andersond41ed4e2009-10-19 22:14:22 +0000319 e.varargs.push_back(lookup_or_add(S->getOperand(0)));
320 e.varargs.push_back(lookup_or_add(S->getOperand(1)));
321 e.varargs.push_back(lookup_or_add(S->getOperand(2)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000322 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000323 e.type = S->getType();
324 e.opcode = Expression::SHUFFLE;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000325
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000326 return e;
327}
328
329Expression ValueTable::create_expression(ExtractElementInst* E) {
330 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000331
Owen Andersond41ed4e2009-10-19 22:14:22 +0000332 e.varargs.push_back(lookup_or_add(E->getOperand(0)));
333 e.varargs.push_back(lookup_or_add(E->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000334 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000335 e.type = E->getType();
336 e.opcode = Expression::EXTRACT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000337
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000338 return e;
339}
340
341Expression ValueTable::create_expression(InsertElementInst* I) {
342 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000343
Owen Andersond41ed4e2009-10-19 22:14:22 +0000344 e.varargs.push_back(lookup_or_add(I->getOperand(0)));
345 e.varargs.push_back(lookup_or_add(I->getOperand(1)));
346 e.varargs.push_back(lookup_or_add(I->getOperand(2)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000347 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000348 e.type = I->getType();
349 e.opcode = Expression::INSERT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000350
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000351 return e;
352}
353
354Expression ValueTable::create_expression(SelectInst* I) {
355 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000356
Owen Andersond41ed4e2009-10-19 22:14:22 +0000357 e.varargs.push_back(lookup_or_add(I->getCondition()));
358 e.varargs.push_back(lookup_or_add(I->getTrueValue()));
359 e.varargs.push_back(lookup_or_add(I->getFalseValue()));
Owen Andersonb388ca92007-10-18 19:39:33 +0000360 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000361 e.type = I->getType();
362 e.opcode = Expression::SELECT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000363
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000364 return e;
365}
366
367Expression ValueTable::create_expression(GetElementPtrInst* G) {
368 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000369
Owen Andersond41ed4e2009-10-19 22:14:22 +0000370 e.varargs.push_back(lookup_or_add(G->getPointerOperand()));
Owen Andersonb388ca92007-10-18 19:39:33 +0000371 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000372 e.type = G->getType();
373 e.opcode = Expression::GEP;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000374
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000375 for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
376 I != E; ++I)
Owen Anderson8f46c782008-04-11 05:11:49 +0000377 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000378
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000379 return e;
380}
381
Owen Andersond41ed4e2009-10-19 22:14:22 +0000382Expression ValueTable::create_expression(ExtractValueInst* E) {
383 Expression e;
384
385 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
386 for (ExtractValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
387 II != IE; ++II)
388 e.varargs.push_back(*II);
389 e.function = 0;
390 e.type = E->getType();
391 e.opcode = Expression::EXTRACTVALUE;
392
393 return e;
394}
395
396Expression ValueTable::create_expression(InsertValueInst* E) {
397 Expression e;
398
399 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
400 e.varargs.push_back(lookup_or_add(E->getInsertedValueOperand()));
401 for (InsertValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
402 II != IE; ++II)
403 e.varargs.push_back(*II);
404 e.function = 0;
405 e.type = E->getType();
406 e.opcode = Expression::INSERTVALUE;
407
408 return e;
409}
410
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000411//===----------------------------------------------------------------------===//
412// ValueTable External Functions
413//===----------------------------------------------------------------------===//
414
Owen Andersonb2303722008-06-18 21:41:49 +0000415/// add - Insert a value into the table with a specified value number.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000416void ValueTable::add(Value *V, uint32_t num) {
Owen Andersonb2303722008-06-18 21:41:49 +0000417 valueNumbering.insert(std::make_pair(V, num));
418}
419
Owen Andersond41ed4e2009-10-19 22:14:22 +0000420uint32_t ValueTable::lookup_or_add_call(CallInst* C) {
421 if (AA->doesNotAccessMemory(C)) {
422 Expression exp = create_expression(C);
423 uint32_t& e = expressionNumbering[exp];
424 if (!e) e = nextValueNumber++;
425 valueNumbering[C] = e;
426 return e;
427 } else if (AA->onlyReadsMemory(C)) {
428 Expression exp = create_expression(C);
429 uint32_t& e = expressionNumbering[exp];
430 if (!e) {
431 e = nextValueNumber++;
432 valueNumbering[C] = e;
433 return e;
434 }
Dan Gohman4ec01b22009-11-14 02:27:51 +0000435 if (!MD) {
436 e = nextValueNumber++;
437 valueNumbering[C] = e;
438 return e;
439 }
Owen Andersond41ed4e2009-10-19 22:14:22 +0000440
441 MemDepResult local_dep = MD->getDependency(C);
442
443 if (!local_dep.isDef() && !local_dep.isNonLocal()) {
444 valueNumbering[C] = nextValueNumber;
445 return nextValueNumber++;
446 }
447
448 if (local_dep.isDef()) {
449 CallInst* local_cdep = cast<CallInst>(local_dep.getInst());
450
451 if (local_cdep->getNumOperands() != C->getNumOperands()) {
452 valueNumbering[C] = nextValueNumber;
453 return nextValueNumber++;
454 }
455
Eric Christopher551754c2010-04-16 23:37:20 +0000456 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000457 uint32_t c_vn = lookup_or_add(C->getOperand(i));
458 uint32_t cd_vn = lookup_or_add(local_cdep->getOperand(i));
459 if (c_vn != cd_vn) {
460 valueNumbering[C] = nextValueNumber;
461 return nextValueNumber++;
462 }
463 }
464
465 uint32_t v = lookup_or_add(local_cdep);
466 valueNumbering[C] = v;
467 return v;
468 }
469
470 // Non-local case.
471 const MemoryDependenceAnalysis::NonLocalDepInfo &deps =
472 MD->getNonLocalCallDependency(CallSite(C));
473 // FIXME: call/call dependencies for readonly calls should return def, not
474 // clobber! Move the checking logic to MemDep!
475 CallInst* cdep = 0;
476
477 // Check to see if we have a single dominating call instruction that is
478 // identical to C.
479 for (unsigned i = 0, e = deps.size(); i != e; ++i) {
Chris Lattnere18b9712009-12-09 07:08:01 +0000480 const NonLocalDepEntry *I = &deps[i];
Owen Andersond41ed4e2009-10-19 22:14:22 +0000481 // Ignore non-local dependencies.
Chris Lattnere18b9712009-12-09 07:08:01 +0000482 if (I->getResult().isNonLocal())
Owen Andersond41ed4e2009-10-19 22:14:22 +0000483 continue;
484
485 // We don't handle non-depedencies. If we already have a call, reject
486 // instruction dependencies.
Chris Lattnere18b9712009-12-09 07:08:01 +0000487 if (I->getResult().isClobber() || cdep != 0) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000488 cdep = 0;
489 break;
490 }
491
Chris Lattnere18b9712009-12-09 07:08:01 +0000492 CallInst *NonLocalDepCall = dyn_cast<CallInst>(I->getResult().getInst());
Owen Andersond41ed4e2009-10-19 22:14:22 +0000493 // FIXME: All duplicated with non-local case.
Chris Lattnere18b9712009-12-09 07:08:01 +0000494 if (NonLocalDepCall && DT->properlyDominates(I->getBB(), C->getParent())){
Owen Andersond41ed4e2009-10-19 22:14:22 +0000495 cdep = NonLocalDepCall;
496 continue;
497 }
498
499 cdep = 0;
500 break;
501 }
502
503 if (!cdep) {
504 valueNumbering[C] = nextValueNumber;
505 return nextValueNumber++;
506 }
507
508 if (cdep->getNumOperands() != C->getNumOperands()) {
509 valueNumbering[C] = nextValueNumber;
510 return nextValueNumber++;
511 }
Eric Christopher551754c2010-04-16 23:37:20 +0000512 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000513 uint32_t c_vn = lookup_or_add(C->getOperand(i));
514 uint32_t cd_vn = lookup_or_add(cdep->getOperand(i));
515 if (c_vn != cd_vn) {
516 valueNumbering[C] = nextValueNumber;
517 return nextValueNumber++;
518 }
519 }
520
521 uint32_t v = lookup_or_add(cdep);
522 valueNumbering[C] = v;
523 return v;
524
525 } else {
526 valueNumbering[C] = nextValueNumber;
527 return nextValueNumber++;
528 }
529}
530
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000531/// lookup_or_add - Returns the value number for the specified value, assigning
532/// it a new number if it did not have one before.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000533uint32_t ValueTable::lookup_or_add(Value *V) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000534 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
535 if (VI != valueNumbering.end())
536 return VI->second;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000537
Owen Andersond41ed4e2009-10-19 22:14:22 +0000538 if (!isa<Instruction>(V)) {
Owen Anderson158d86e2009-10-19 21:14:57 +0000539 valueNumbering[V] = nextValueNumber;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000540 return nextValueNumber++;
541 }
Owen Andersond41ed4e2009-10-19 22:14:22 +0000542
543 Instruction* I = cast<Instruction>(V);
544 Expression exp;
545 switch (I->getOpcode()) {
546 case Instruction::Call:
547 return lookup_or_add_call(cast<CallInst>(I));
548 case Instruction::Add:
549 case Instruction::FAdd:
550 case Instruction::Sub:
551 case Instruction::FSub:
552 case Instruction::Mul:
553 case Instruction::FMul:
554 case Instruction::UDiv:
555 case Instruction::SDiv:
556 case Instruction::FDiv:
557 case Instruction::URem:
558 case Instruction::SRem:
559 case Instruction::FRem:
560 case Instruction::Shl:
561 case Instruction::LShr:
562 case Instruction::AShr:
563 case Instruction::And:
564 case Instruction::Or :
565 case Instruction::Xor:
566 exp = create_expression(cast<BinaryOperator>(I));
567 break;
568 case Instruction::ICmp:
569 case Instruction::FCmp:
570 exp = create_expression(cast<CmpInst>(I));
571 break;
572 case Instruction::Trunc:
573 case Instruction::ZExt:
574 case Instruction::SExt:
575 case Instruction::FPToUI:
576 case Instruction::FPToSI:
577 case Instruction::UIToFP:
578 case Instruction::SIToFP:
579 case Instruction::FPTrunc:
580 case Instruction::FPExt:
581 case Instruction::PtrToInt:
582 case Instruction::IntToPtr:
583 case Instruction::BitCast:
584 exp = create_expression(cast<CastInst>(I));
585 break;
586 case Instruction::Select:
587 exp = create_expression(cast<SelectInst>(I));
588 break;
589 case Instruction::ExtractElement:
590 exp = create_expression(cast<ExtractElementInst>(I));
591 break;
592 case Instruction::InsertElement:
593 exp = create_expression(cast<InsertElementInst>(I));
594 break;
595 case Instruction::ShuffleVector:
596 exp = create_expression(cast<ShuffleVectorInst>(I));
597 break;
598 case Instruction::ExtractValue:
599 exp = create_expression(cast<ExtractValueInst>(I));
600 break;
601 case Instruction::InsertValue:
602 exp = create_expression(cast<InsertValueInst>(I));
603 break;
604 case Instruction::GetElementPtr:
605 exp = create_expression(cast<GetElementPtrInst>(I));
606 break;
607 default:
608 valueNumbering[V] = nextValueNumber;
609 return nextValueNumber++;
610 }
611
612 uint32_t& e = expressionNumbering[exp];
613 if (!e) e = nextValueNumber++;
614 valueNumbering[V] = e;
615 return e;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000616}
617
618/// lookup - Returns the value number of the specified value. Fails if
619/// the value has not yet been numbered.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000620uint32_t ValueTable::lookup(Value *V) const {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000621 DenseMap<Value*, uint32_t>::const_iterator VI = valueNumbering.find(V);
Chris Lattner88365bb2008-03-21 21:14:38 +0000622 assert(VI != valueNumbering.end() && "Value not numbered?");
623 return VI->second;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000624}
625
626/// clear - Remove all entries from the ValueTable
627void ValueTable::clear() {
628 valueNumbering.clear();
629 expressionNumbering.clear();
630 nextValueNumber = 1;
631}
632
Owen Andersonbf7d0bc2007-07-31 23:27:13 +0000633/// erase - Remove a value from the value numbering
Chris Lattnerb2412a82009-09-21 02:42:51 +0000634void ValueTable::erase(Value *V) {
Owen Andersonbf7d0bc2007-07-31 23:27:13 +0000635 valueNumbering.erase(V);
636}
637
Bill Wendling246dbbb2008-12-22 21:36:08 +0000638/// verifyRemoved - Verify that the value is removed from all internal data
639/// structures.
640void ValueTable::verifyRemoved(const Value *V) const {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000641 for (DenseMap<Value*, uint32_t>::const_iterator
Bill Wendling246dbbb2008-12-22 21:36:08 +0000642 I = valueNumbering.begin(), E = valueNumbering.end(); I != E; ++I) {
643 assert(I->first != V && "Inst still occurs in value numbering map!");
644 }
645}
646
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000647//===----------------------------------------------------------------------===//
Bill Wendling30788b82008-12-22 22:32:22 +0000648// GVN Pass
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000649//===----------------------------------------------------------------------===//
650
651namespace {
Chris Lattner3e8b6632009-09-02 06:11:42 +0000652 struct ValueNumberScope {
Owen Anderson6fafe842008-06-20 01:15:47 +0000653 ValueNumberScope* parent;
654 DenseMap<uint32_t, Value*> table;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000655
Owen Anderson6fafe842008-06-20 01:15:47 +0000656 ValueNumberScope(ValueNumberScope* p) : parent(p) { }
657 };
658}
659
660namespace {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000661
Chris Lattner3e8b6632009-09-02 06:11:42 +0000662 class GVN : public FunctionPass {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000663 bool runOnFunction(Function &F);
664 public:
665 static char ID; // Pass identification, replacement for typeid
Bob Wilsonb29d7d22010-02-28 05:34:05 +0000666 explicit GVN(bool noloads = false)
667 : FunctionPass(&ID), NoLoads(noloads), MD(0) { }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000668
669 private:
Dan Gohman4ec01b22009-11-14 02:27:51 +0000670 bool NoLoads;
Chris Lattner663e4412008-12-01 00:40:32 +0000671 MemoryDependenceAnalysis *MD;
672 DominatorTree *DT;
673
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000674 ValueTable VN;
Owen Anderson6fafe842008-06-20 01:15:47 +0000675 DenseMap<BasicBlock*, ValueNumberScope*> localAvail;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000676
Bob Wilson484d4a32010-02-16 19:51:59 +0000677 // List of critical edges to be split between iterations.
678 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> toSplit;
679
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000680 // This transformation requires dominator postdominator info
681 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000682 AU.addRequired<DominatorTree>();
Dan Gohman4ec01b22009-11-14 02:27:51 +0000683 if (!NoLoads)
684 AU.addRequired<MemoryDependenceAnalysis>();
Owen Andersonb388ca92007-10-18 19:39:33 +0000685 AU.addRequired<AliasAnalysis>();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000686
Owen Andersonb70a5712008-06-23 17:49:45 +0000687 AU.addPreserved<DominatorTree>();
Owen Andersonb388ca92007-10-18 19:39:33 +0000688 AU.addPreserved<AliasAnalysis>();
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000689 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000690
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000691 // Helper fuctions
692 // FIXME: eliminate or document these better
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000693 bool processLoad(LoadInst* L,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000694 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000695 bool processInstruction(Instruction *I,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000696 SmallVectorImpl<Instruction*> &toErase);
Owen Anderson830db6a2007-08-02 18:16:06 +0000697 bool processNonLocalLoad(LoadInst* L,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000698 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000699 bool processBlock(BasicBlock *BB);
Owen Andersonb2303722008-06-18 21:41:49 +0000700 void dump(DenseMap<uint32_t, Value*>& d);
Owen Anderson3e75a422007-08-14 18:04:11 +0000701 bool iterateOnFunction(Function &F);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000702 Value *CollapsePhi(PHINode* p);
Owen Andersonb2303722008-06-18 21:41:49 +0000703 bool performPRE(Function& F);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000704 Value *lookupNumber(BasicBlock *BB, uint32_t num);
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +0000705 void cleanupGlobalSets();
Bill Wendling246dbbb2008-12-22 21:36:08 +0000706 void verifyRemoved(const Instruction *I) const;
Bob Wilson484d4a32010-02-16 19:51:59 +0000707 bool splitCriticalEdges();
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000708 };
Daniel Dunbara279bc32009-09-20 02:20:51 +0000709
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000710 char GVN::ID = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000711}
712
713// createGVNPass - The public interface to this file...
Bob Wilsonb29d7d22010-02-28 05:34:05 +0000714FunctionPass *llvm::createGVNPass(bool NoLoads) {
715 return new GVN(NoLoads);
Dan Gohman4ec01b22009-11-14 02:27:51 +0000716}
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000717
718static RegisterPass<GVN> X("gvn",
719 "Global Value Numbering");
720
Owen Andersonb2303722008-06-18 21:41:49 +0000721void GVN::dump(DenseMap<uint32_t, Value*>& d) {
Dan Gohmanad12b262009-12-18 03:25:51 +0000722 errs() << "{\n";
Owen Andersonb2303722008-06-18 21:41:49 +0000723 for (DenseMap<uint32_t, Value*>::iterator I = d.begin(),
Owen Anderson0cd32032007-07-25 19:57:03 +0000724 E = d.end(); I != E; ++I) {
Dan Gohmanad12b262009-12-18 03:25:51 +0000725 errs() << I->first << "\n";
Owen Anderson0cd32032007-07-25 19:57:03 +0000726 I->second->dump();
727 }
Dan Gohmanad12b262009-12-18 03:25:51 +0000728 errs() << "}\n";
Owen Anderson0cd32032007-07-25 19:57:03 +0000729}
730
Chris Lattnerb2412a82009-09-21 02:42:51 +0000731static bool isSafeReplacement(PHINode* p, Instruction *inst) {
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000732 if (!isa<PHINode>(inst))
733 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000734
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000735 for (Instruction::use_iterator UI = p->use_begin(), E = p->use_end();
736 UI != E; ++UI)
737 if (PHINode* use_phi = dyn_cast<PHINode>(UI))
738 if (use_phi->getParent() == inst->getParent())
739 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000740
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000741 return true;
742}
743
Chris Lattnerb2412a82009-09-21 02:42:51 +0000744Value *GVN::CollapsePhi(PHINode *PN) {
745 Value *ConstVal = PN->hasConstantValue(DT);
746 if (!ConstVal) return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000747
Chris Lattnerb2412a82009-09-21 02:42:51 +0000748 Instruction *Inst = dyn_cast<Instruction>(ConstVal);
749 if (!Inst)
750 return ConstVal;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000751
Chris Lattnerb2412a82009-09-21 02:42:51 +0000752 if (DT->dominates(Inst, PN))
753 if (isSafeReplacement(PN, Inst))
754 return Inst;
Owen Anderson1defe2d2007-08-16 22:51:56 +0000755 return 0;
756}
Owen Anderson0cd32032007-07-25 19:57:03 +0000757
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000758/// IsValueFullyAvailableInBlock - Return true if we can prove that the value
759/// we're analyzing is fully available in the specified block. As we go, keep
Chris Lattner72bc70d2008-12-05 07:49:08 +0000760/// track of which blocks we know are fully alive in FullyAvailableBlocks. This
761/// map is actually a tri-state map with the following values:
762/// 0) we know the block *is not* fully available.
763/// 1) we know the block *is* fully available.
764/// 2) we do not know whether the block is fully available or not, but we are
765/// currently speculating that it will be.
766/// 3) we are speculating for this block and have used that to speculate for
767/// other blocks.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000768static bool IsValueFullyAvailableInBlock(BasicBlock *BB,
Chris Lattner72bc70d2008-12-05 07:49:08 +0000769 DenseMap<BasicBlock*, char> &FullyAvailableBlocks) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000770 // Optimistically assume that the block is fully available and check to see
771 // if we already know about this block in one lookup.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000772 std::pair<DenseMap<BasicBlock*, char>::iterator, char> IV =
Chris Lattner72bc70d2008-12-05 07:49:08 +0000773 FullyAvailableBlocks.insert(std::make_pair(BB, 2));
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000774
775 // If the entry already existed for this block, return the precomputed value.
Chris Lattner72bc70d2008-12-05 07:49:08 +0000776 if (!IV.second) {
777 // If this is a speculative "available" value, mark it as being used for
778 // speculation of other blocks.
779 if (IV.first->second == 2)
780 IV.first->second = 3;
781 return IV.first->second != 0;
782 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000783
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000784 // Otherwise, see if it is fully available in all predecessors.
785 pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000786
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000787 // If this block has no predecessors, it isn't live-in here.
788 if (PI == PE)
Chris Lattner72bc70d2008-12-05 07:49:08 +0000789 goto SpeculationFailure;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000790
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000791 for (; PI != PE; ++PI)
792 // If the value isn't fully available in one of our predecessors, then it
793 // isn't fully available in this block either. Undo our previous
794 // optimistic assumption and bail out.
795 if (!IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
Chris Lattner72bc70d2008-12-05 07:49:08 +0000796 goto SpeculationFailure;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000797
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000798 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000799
Chris Lattner72bc70d2008-12-05 07:49:08 +0000800// SpeculationFailure - If we get here, we found out that this is not, after
801// all, a fully-available block. We have a problem if we speculated on this and
802// used the speculation to mark other blocks as available.
803SpeculationFailure:
804 char &BBVal = FullyAvailableBlocks[BB];
Daniel Dunbara279bc32009-09-20 02:20:51 +0000805
Chris Lattner72bc70d2008-12-05 07:49:08 +0000806 // If we didn't speculate on this, just return with it set to false.
807 if (BBVal == 2) {
808 BBVal = 0;
809 return false;
810 }
811
812 // If we did speculate on this value, we could have blocks set to 1 that are
813 // incorrect. Walk the (transitive) successors of this block and mark them as
814 // 0 if set to one.
815 SmallVector<BasicBlock*, 32> BBWorklist;
816 BBWorklist.push_back(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000817
Dan Gohman321a8132010-01-05 16:27:25 +0000818 do {
Chris Lattner72bc70d2008-12-05 07:49:08 +0000819 BasicBlock *Entry = BBWorklist.pop_back_val();
820 // Note that this sets blocks to 0 (unavailable) if they happen to not
821 // already be in FullyAvailableBlocks. This is safe.
822 char &EntryVal = FullyAvailableBlocks[Entry];
823 if (EntryVal == 0) continue; // Already unavailable.
824
825 // Mark as unavailable.
826 EntryVal = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000827
Chris Lattner72bc70d2008-12-05 07:49:08 +0000828 for (succ_iterator I = succ_begin(Entry), E = succ_end(Entry); I != E; ++I)
829 BBWorklist.push_back(*I);
Dan Gohman321a8132010-01-05 16:27:25 +0000830 } while (!BBWorklist.empty());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000831
Chris Lattner72bc70d2008-12-05 07:49:08 +0000832 return false;
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000833}
834
Chris Lattner771a5422009-09-20 20:09:34 +0000835
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000836/// CanCoerceMustAliasedValueToLoad - Return true if
837/// CoerceAvailableValueToLoadType will succeed.
838static bool CanCoerceMustAliasedValueToLoad(Value *StoredVal,
839 const Type *LoadTy,
840 const TargetData &TD) {
841 // If the loaded or stored value is an first class array or struct, don't try
842 // to transform them. We need to be able to bitcast to integer.
Duncan Sands1df98592010-02-16 11:11:14 +0000843 if (LoadTy->isStructTy() || LoadTy->isArrayTy() ||
844 StoredVal->getType()->isStructTy() ||
845 StoredVal->getType()->isArrayTy())
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000846 return false;
847
848 // The store has to be at least as big as the load.
849 if (TD.getTypeSizeInBits(StoredVal->getType()) <
850 TD.getTypeSizeInBits(LoadTy))
851 return false;
852
853 return true;
854}
855
856
Chris Lattner771a5422009-09-20 20:09:34 +0000857/// CoerceAvailableValueToLoadType - If we saw a store of a value to memory, and
858/// then a load from a must-aliased pointer of a different type, try to coerce
859/// the stored value. LoadedTy is the type of the load we want to replace and
860/// InsertPt is the place to insert new instructions.
861///
862/// If we can't do it, return null.
863static Value *CoerceAvailableValueToLoadType(Value *StoredVal,
864 const Type *LoadedTy,
865 Instruction *InsertPt,
866 const TargetData &TD) {
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000867 if (!CanCoerceMustAliasedValueToLoad(StoredVal, LoadedTy, TD))
868 return 0;
869
Chris Lattner771a5422009-09-20 20:09:34 +0000870 const Type *StoredValTy = StoredVal->getType();
871
Chris Lattner7944c212010-05-08 20:01:44 +0000872 uint64_t StoreSize = TD.getTypeStoreSizeInBits(StoredValTy);
Chris Lattner771a5422009-09-20 20:09:34 +0000873 uint64_t LoadSize = TD.getTypeSizeInBits(LoadedTy);
874
875 // If the store and reload are the same size, we can always reuse it.
876 if (StoreSize == LoadSize) {
Duncan Sands1df98592010-02-16 11:11:14 +0000877 if (StoredValTy->isPointerTy() && LoadedTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000878 // Pointer to Pointer -> use bitcast.
879 return new BitCastInst(StoredVal, LoadedTy, "", InsertPt);
880 }
881
882 // Convert source pointers to integers, which can be bitcast.
Duncan Sands1df98592010-02-16 11:11:14 +0000883 if (StoredValTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000884 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
885 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
886 }
887
888 const Type *TypeToCastTo = LoadedTy;
Duncan Sands1df98592010-02-16 11:11:14 +0000889 if (TypeToCastTo->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000890 TypeToCastTo = TD.getIntPtrType(StoredValTy->getContext());
891
892 if (StoredValTy != TypeToCastTo)
893 StoredVal = new BitCastInst(StoredVal, TypeToCastTo, "", InsertPt);
894
895 // Cast to pointer if the load needs a pointer type.
Duncan Sands1df98592010-02-16 11:11:14 +0000896 if (LoadedTy->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000897 StoredVal = new IntToPtrInst(StoredVal, LoadedTy, "", InsertPt);
898
899 return StoredVal;
900 }
901
902 // If the loaded value is smaller than the available value, then we can
903 // extract out a piece from it. If the available value is too small, then we
904 // can't do anything.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000905 assert(StoreSize >= LoadSize && "CanCoerceMustAliasedValueToLoad fail");
Chris Lattner771a5422009-09-20 20:09:34 +0000906
907 // Convert source pointers to integers, which can be manipulated.
Duncan Sands1df98592010-02-16 11:11:14 +0000908 if (StoredValTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000909 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
910 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
911 }
912
913 // Convert vectors and fp to integer, which can be manipulated.
Duncan Sands1df98592010-02-16 11:11:14 +0000914 if (!StoredValTy->isIntegerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000915 StoredValTy = IntegerType::get(StoredValTy->getContext(), StoreSize);
916 StoredVal = new BitCastInst(StoredVal, StoredValTy, "", InsertPt);
917 }
918
919 // If this is a big-endian system, we need to shift the value down to the low
920 // bits so that a truncate will work.
921 if (TD.isBigEndian()) {
922 Constant *Val = ConstantInt::get(StoredVal->getType(), StoreSize-LoadSize);
923 StoredVal = BinaryOperator::CreateLShr(StoredVal, Val, "tmp", InsertPt);
924 }
925
926 // Truncate the integer to the right size now.
927 const Type *NewIntTy = IntegerType::get(StoredValTy->getContext(), LoadSize);
928 StoredVal = new TruncInst(StoredVal, NewIntTy, "trunc", InsertPt);
929
930 if (LoadedTy == NewIntTy)
931 return StoredVal;
932
933 // If the result is a pointer, inttoptr.
Duncan Sands1df98592010-02-16 11:11:14 +0000934 if (LoadedTy->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000935 return new IntToPtrInst(StoredVal, LoadedTy, "inttoptr", InsertPt);
936
937 // Otherwise, bitcast.
938 return new BitCastInst(StoredVal, LoadedTy, "bitcast", InsertPt);
939}
940
Chris Lattnerca749402009-09-21 06:24:16 +0000941/// GetBaseWithConstantOffset - Analyze the specified pointer to see if it can
942/// be expressed as a base pointer plus a constant offset. Return the base and
943/// offset to the caller.
944static Value *GetBaseWithConstantOffset(Value *Ptr, int64_t &Offset,
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000945 const TargetData &TD) {
Chris Lattnerca749402009-09-21 06:24:16 +0000946 Operator *PtrOp = dyn_cast<Operator>(Ptr);
947 if (PtrOp == 0) return Ptr;
948
949 // Just look through bitcasts.
950 if (PtrOp->getOpcode() == Instruction::BitCast)
951 return GetBaseWithConstantOffset(PtrOp->getOperand(0), Offset, TD);
952
953 // If this is a GEP with constant indices, we can look through it.
954 GEPOperator *GEP = dyn_cast<GEPOperator>(PtrOp);
955 if (GEP == 0 || !GEP->hasAllConstantIndices()) return Ptr;
956
957 gep_type_iterator GTI = gep_type_begin(GEP);
958 for (User::op_iterator I = GEP->idx_begin(), E = GEP->idx_end(); I != E;
959 ++I, ++GTI) {
960 ConstantInt *OpC = cast<ConstantInt>(*I);
961 if (OpC->isZero()) continue;
962
963 // Handle a struct and array indices which add their offset to the pointer.
964 if (const StructType *STy = dyn_cast<StructType>(*GTI)) {
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000965 Offset += TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue());
Chris Lattnerca749402009-09-21 06:24:16 +0000966 } else {
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000967 uint64_t Size = TD.getTypeAllocSize(GTI.getIndexedType());
Chris Lattnerca749402009-09-21 06:24:16 +0000968 Offset += OpC->getSExtValue()*Size;
969 }
970 }
971
972 // Re-sign extend from the pointer size if needed to get overflow edge cases
973 // right.
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000974 unsigned PtrSize = TD.getPointerSizeInBits();
Chris Lattnerca749402009-09-21 06:24:16 +0000975 if (PtrSize < 64)
976 Offset = (Offset << (64-PtrSize)) >> (64-PtrSize);
977
978 return GetBaseWithConstantOffset(GEP->getPointerOperand(), Offset, TD);
979}
980
981
Chris Lattnerfaf815b2009-12-06 01:57:02 +0000982/// AnalyzeLoadFromClobberingWrite - This function is called when we have a
983/// memdep query of a load that ends up being a clobbering memory write (store,
984/// memset, memcpy, memmove). This means that the write *may* provide bits used
985/// by the load but we can't be sure because the pointers don't mustalias.
986///
987/// Check this case to see if there is anything more we can do before we give
988/// up. This returns -1 if we have to give up, or a byte number in the stored
989/// value of the piece that feeds the load.
Chris Lattner03f17da2009-12-09 07:34:10 +0000990static int AnalyzeLoadFromClobberingWrite(const Type *LoadTy, Value *LoadPtr,
991 Value *WritePtr,
Chris Lattnerfaf815b2009-12-06 01:57:02 +0000992 uint64_t WriteSizeInBits,
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000993 const TargetData &TD) {
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000994 // If the loaded or stored value is an first class array or struct, don't try
995 // to transform them. We need to be able to bitcast to integer.
Duncan Sands1df98592010-02-16 11:11:14 +0000996 if (LoadTy->isStructTy() || LoadTy->isArrayTy())
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000997 return -1;
998
Chris Lattnerca749402009-09-21 06:24:16 +0000999 int64_t StoreOffset = 0, LoadOffset = 0;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001000 Value *StoreBase = GetBaseWithConstantOffset(WritePtr, StoreOffset, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001001 Value *LoadBase =
Chris Lattner03f17da2009-12-09 07:34:10 +00001002 GetBaseWithConstantOffset(LoadPtr, LoadOffset, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001003 if (StoreBase != LoadBase)
1004 return -1;
1005
1006 // If the load and store are to the exact same address, they should have been
1007 // a must alias. AA must have gotten confused.
Chris Lattner219d7742010-03-25 05:58:19 +00001008 // FIXME: Study to see if/when this happens. One case is forwarding a memset
1009 // to a load from the base of the memset.
Chris Lattnerca749402009-09-21 06:24:16 +00001010#if 0
Chris Lattner219d7742010-03-25 05:58:19 +00001011 if (LoadOffset == StoreOffset) {
David Greenebf7f78e2010-01-05 01:27:17 +00001012 dbgs() << "STORE/LOAD DEP WITH COMMON POINTER MISSED:\n"
Chris Lattnerca749402009-09-21 06:24:16 +00001013 << "Base = " << *StoreBase << "\n"
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001014 << "Store Ptr = " << *WritePtr << "\n"
1015 << "Store Offs = " << StoreOffset << "\n"
Chris Lattnerb6760b42009-12-10 00:04:46 +00001016 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattnerb3f927f2009-12-09 02:41:54 +00001017 abort();
Chris Lattnerca749402009-09-21 06:24:16 +00001018 }
Chris Lattner219d7742010-03-25 05:58:19 +00001019#endif
Chris Lattnerca749402009-09-21 06:24:16 +00001020
1021 // If the load and store don't overlap at all, the store doesn't provide
1022 // anything to the load. In this case, they really don't alias at all, AA
1023 // must have gotten confused.
1024 // FIXME: Investigate cases where this bails out, e.g. rdar://7238614. Then
1025 // remove this check, as it is duplicated with what we have below.
Chris Lattner03f17da2009-12-09 07:34:10 +00001026 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy);
Chris Lattnerca749402009-09-21 06:24:16 +00001027
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001028 if ((WriteSizeInBits & 7) | (LoadSize & 7))
Chris Lattnerca749402009-09-21 06:24:16 +00001029 return -1;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001030 uint64_t StoreSize = WriteSizeInBits >> 3; // Convert to bytes.
Chris Lattnerca749402009-09-21 06:24:16 +00001031 LoadSize >>= 3;
1032
1033
1034 bool isAAFailure = false;
Chris Lattner219d7742010-03-25 05:58:19 +00001035 if (StoreOffset < LoadOffset)
Chris Lattnerca749402009-09-21 06:24:16 +00001036 isAAFailure = StoreOffset+int64_t(StoreSize) <= LoadOffset;
Chris Lattner219d7742010-03-25 05:58:19 +00001037 else
Chris Lattnerca749402009-09-21 06:24:16 +00001038 isAAFailure = LoadOffset+int64_t(LoadSize) <= StoreOffset;
Chris Lattner219d7742010-03-25 05:58:19 +00001039
Chris Lattnerca749402009-09-21 06:24:16 +00001040 if (isAAFailure) {
1041#if 0
David Greenebf7f78e2010-01-05 01:27:17 +00001042 dbgs() << "STORE LOAD DEP WITH COMMON BASE:\n"
Chris Lattnerca749402009-09-21 06:24:16 +00001043 << "Base = " << *StoreBase << "\n"
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001044 << "Store Ptr = " << *WritePtr << "\n"
1045 << "Store Offs = " << StoreOffset << "\n"
Chris Lattnerb6760b42009-12-10 00:04:46 +00001046 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattnerb3f927f2009-12-09 02:41:54 +00001047 abort();
Chris Lattnerca749402009-09-21 06:24:16 +00001048#endif
1049 return -1;
1050 }
1051
1052 // If the Load isn't completely contained within the stored bits, we don't
1053 // have all the bits to feed it. We could do something crazy in the future
1054 // (issue a smaller load then merge the bits in) but this seems unlikely to be
1055 // valuable.
1056 if (StoreOffset > LoadOffset ||
1057 StoreOffset+StoreSize < LoadOffset+LoadSize)
1058 return -1;
1059
1060 // Okay, we can do this transformation. Return the number of bytes into the
1061 // store that the load is.
1062 return LoadOffset-StoreOffset;
1063}
1064
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001065/// AnalyzeLoadFromClobberingStore - This function is called when we have a
1066/// memdep query of a load that ends up being a clobbering store.
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001067static int AnalyzeLoadFromClobberingStore(const Type *LoadTy, Value *LoadPtr,
1068 StoreInst *DepSI,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001069 const TargetData &TD) {
1070 // Cannot handle reading from store of first-class aggregate yet.
Duncan Sands1df98592010-02-16 11:11:14 +00001071 if (DepSI->getOperand(0)->getType()->isStructTy() ||
1072 DepSI->getOperand(0)->getType()->isArrayTy())
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001073 return -1;
1074
1075 Value *StorePtr = DepSI->getPointerOperand();
Chris Lattner16f244e2009-12-10 00:11:45 +00001076 uint64_t StoreSize = TD.getTypeSizeInBits(DepSI->getOperand(0)->getType());
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001077 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
Chris Lattner03f17da2009-12-09 07:34:10 +00001078 StorePtr, StoreSize, TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001079}
1080
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001081static int AnalyzeLoadFromClobberingMemInst(const Type *LoadTy, Value *LoadPtr,
1082 MemIntrinsic *MI,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001083 const TargetData &TD) {
1084 // If the mem operation is a non-constant size, we can't handle it.
1085 ConstantInt *SizeCst = dyn_cast<ConstantInt>(MI->getLength());
1086 if (SizeCst == 0) return -1;
1087 uint64_t MemSizeInBits = SizeCst->getZExtValue()*8;
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001088
1089 // If this is memset, we just need to see if the offset is valid in the size
1090 // of the memset..
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001091 if (MI->getIntrinsicID() == Intrinsic::memset)
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001092 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr, MI->getDest(),
1093 MemSizeInBits, TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001094
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001095 // If we have a memcpy/memmove, the only case we can handle is if this is a
1096 // copy from constant memory. In that case, we can read directly from the
1097 // constant memory.
1098 MemTransferInst *MTI = cast<MemTransferInst>(MI);
1099
1100 Constant *Src = dyn_cast<Constant>(MTI->getSource());
1101 if (Src == 0) return -1;
1102
1103 GlobalVariable *GV = dyn_cast<GlobalVariable>(Src->getUnderlyingObject());
1104 if (GV == 0 || !GV->isConstant()) return -1;
1105
1106 // See if the access is within the bounds of the transfer.
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001107 int Offset = AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
1108 MI->getDest(), MemSizeInBits, TD);
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001109 if (Offset == -1)
1110 return Offset;
1111
1112 // Otherwise, see if we can constant fold a load from the constant with the
1113 // offset applied as appropriate.
1114 Src = ConstantExpr::getBitCast(Src,
1115 llvm::Type::getInt8PtrTy(Src->getContext()));
1116 Constant *OffsetCst =
1117 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1118 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001119 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001120 if (ConstantFoldLoadFromConstPtr(Src, &TD))
1121 return Offset;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001122 return -1;
1123}
1124
Chris Lattnerca749402009-09-21 06:24:16 +00001125
1126/// GetStoreValueForLoad - This function is called when we have a
1127/// memdep query of a load that ends up being a clobbering store. This means
1128/// that the store *may* provide bits used by the load but we can't be sure
1129/// because the pointers don't mustalias. Check this case to see if there is
1130/// anything more we can do before we give up.
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001131static Value *GetStoreValueForLoad(Value *SrcVal, unsigned Offset,
1132 const Type *LoadTy,
1133 Instruction *InsertPt, const TargetData &TD){
Chris Lattnerca749402009-09-21 06:24:16 +00001134 LLVMContext &Ctx = SrcVal->getType()->getContext();
1135
Chris Lattner7944c212010-05-08 20:01:44 +00001136 uint64_t StoreSize = (TD.getTypeSizeInBits(SrcVal->getType()) + 7) / 8;
1137 uint64_t LoadSize = (TD.getTypeSizeInBits(LoadTy) + 7) / 8;
Chris Lattnerca749402009-09-21 06:24:16 +00001138
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001139 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
Chris Lattnerca749402009-09-21 06:24:16 +00001140
1141 // Compute which bits of the stored value are being used by the load. Convert
1142 // to an integer type to start with.
Duncan Sands1df98592010-02-16 11:11:14 +00001143 if (SrcVal->getType()->isPointerTy())
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001144 SrcVal = Builder.CreatePtrToInt(SrcVal, TD.getIntPtrType(Ctx), "tmp");
Duncan Sands1df98592010-02-16 11:11:14 +00001145 if (!SrcVal->getType()->isIntegerTy())
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001146 SrcVal = Builder.CreateBitCast(SrcVal, IntegerType::get(Ctx, StoreSize*8),
1147 "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001148
1149 // Shift the bits to the least significant depending on endianness.
1150 unsigned ShiftAmt;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001151 if (TD.isLittleEndian())
Chris Lattnerca749402009-09-21 06:24:16 +00001152 ShiftAmt = Offset*8;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001153 else
Chris Lattner19ad7842009-09-21 17:55:47 +00001154 ShiftAmt = (StoreSize-LoadSize-Offset)*8;
Chris Lattnerca749402009-09-21 06:24:16 +00001155
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001156 if (ShiftAmt)
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001157 SrcVal = Builder.CreateLShr(SrcVal, ShiftAmt, "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001158
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001159 if (LoadSize != StoreSize)
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001160 SrcVal = Builder.CreateTrunc(SrcVal, IntegerType::get(Ctx, LoadSize*8),
1161 "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001162
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001163 return CoerceAvailableValueToLoadType(SrcVal, LoadTy, InsertPt, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001164}
1165
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001166/// GetMemInstValueForLoad - This function is called when we have a
1167/// memdep query of a load that ends up being a clobbering mem intrinsic.
1168static Value *GetMemInstValueForLoad(MemIntrinsic *SrcInst, unsigned Offset,
1169 const Type *LoadTy, Instruction *InsertPt,
1170 const TargetData &TD){
1171 LLVMContext &Ctx = LoadTy->getContext();
1172 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
1173
1174 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
1175
1176 // We know that this method is only called when the mem transfer fully
1177 // provides the bits for the load.
1178 if (MemSetInst *MSI = dyn_cast<MemSetInst>(SrcInst)) {
1179 // memset(P, 'x', 1234) -> splat('x'), even if x is a variable, and
1180 // independently of what the offset is.
1181 Value *Val = MSI->getValue();
1182 if (LoadSize != 1)
1183 Val = Builder.CreateZExt(Val, IntegerType::get(Ctx, LoadSize*8));
1184
1185 Value *OneElt = Val;
1186
1187 // Splat the value out to the right number of bits.
1188 for (unsigned NumBytesSet = 1; NumBytesSet != LoadSize; ) {
1189 // If we can double the number of bytes set, do it.
1190 if (NumBytesSet*2 <= LoadSize) {
1191 Value *ShVal = Builder.CreateShl(Val, NumBytesSet*8);
1192 Val = Builder.CreateOr(Val, ShVal);
1193 NumBytesSet <<= 1;
1194 continue;
1195 }
1196
1197 // Otherwise insert one byte at a time.
1198 Value *ShVal = Builder.CreateShl(Val, 1*8);
1199 Val = Builder.CreateOr(OneElt, ShVal);
1200 ++NumBytesSet;
1201 }
1202
1203 return CoerceAvailableValueToLoadType(Val, LoadTy, InsertPt, TD);
1204 }
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001205
1206 // Otherwise, this is a memcpy/memmove from a constant global.
1207 MemTransferInst *MTI = cast<MemTransferInst>(SrcInst);
1208 Constant *Src = cast<Constant>(MTI->getSource());
1209
1210 // Otherwise, see if we can constant fold a load from the constant with the
1211 // offset applied as appropriate.
1212 Src = ConstantExpr::getBitCast(Src,
1213 llvm::Type::getInt8PtrTy(Src->getContext()));
1214 Constant *OffsetCst =
1215 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1216 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
1217 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
1218 return ConstantFoldLoadFromConstPtr(Src, &TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001219}
1220
Dan Gohmanb3579832010-04-15 17:08:50 +00001221namespace {
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001222
Chris Lattner87913512009-09-21 06:30:24 +00001223struct AvailableValueInBlock {
1224 /// BB - The basic block in question.
1225 BasicBlock *BB;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001226 enum ValType {
1227 SimpleVal, // A simple offsetted value that is accessed.
1228 MemIntrin // A memory intrinsic which is loaded from.
1229 };
1230
Chris Lattner87913512009-09-21 06:30:24 +00001231 /// V - The value that is live out of the block.
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001232 PointerIntPair<Value *, 1, ValType> Val;
1233
1234 /// Offset - The byte offset in Val that is interesting for the load query.
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001235 unsigned Offset;
Chris Lattner87913512009-09-21 06:30:24 +00001236
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001237 static AvailableValueInBlock get(BasicBlock *BB, Value *V,
1238 unsigned Offset = 0) {
Chris Lattner87913512009-09-21 06:30:24 +00001239 AvailableValueInBlock Res;
1240 Res.BB = BB;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001241 Res.Val.setPointer(V);
1242 Res.Val.setInt(SimpleVal);
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001243 Res.Offset = Offset;
Chris Lattner87913512009-09-21 06:30:24 +00001244 return Res;
1245 }
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001246
1247 static AvailableValueInBlock getMI(BasicBlock *BB, MemIntrinsic *MI,
1248 unsigned Offset = 0) {
1249 AvailableValueInBlock Res;
1250 Res.BB = BB;
1251 Res.Val.setPointer(MI);
1252 Res.Val.setInt(MemIntrin);
1253 Res.Offset = Offset;
1254 return Res;
1255 }
1256
1257 bool isSimpleValue() const { return Val.getInt() == SimpleVal; }
1258 Value *getSimpleValue() const {
1259 assert(isSimpleValue() && "Wrong accessor");
1260 return Val.getPointer();
1261 }
1262
1263 MemIntrinsic *getMemIntrinValue() const {
1264 assert(!isSimpleValue() && "Wrong accessor");
1265 return cast<MemIntrinsic>(Val.getPointer());
1266 }
Chris Lattner5362c542009-12-21 23:04:33 +00001267
1268 /// MaterializeAdjustedValue - Emit code into this block to adjust the value
1269 /// defined here to the specified type. This handles various coercion cases.
1270 Value *MaterializeAdjustedValue(const Type *LoadTy,
1271 const TargetData *TD) const {
1272 Value *Res;
1273 if (isSimpleValue()) {
1274 Res = getSimpleValue();
1275 if (Res->getType() != LoadTy) {
1276 assert(TD && "Need target data to handle type mismatch case");
1277 Res = GetStoreValueForLoad(Res, Offset, LoadTy, BB->getTerminator(),
1278 *TD);
1279
1280 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\nOffset: " << Offset << " "
1281 << *getSimpleValue() << '\n'
1282 << *Res << '\n' << "\n\n\n");
1283 }
1284 } else {
1285 Res = GetMemInstValueForLoad(getMemIntrinValue(), Offset,
1286 LoadTy, BB->getTerminator(), *TD);
1287 DEBUG(errs() << "GVN COERCED NONLOCAL MEM INTRIN:\nOffset: " << Offset
1288 << " " << *getMemIntrinValue() << '\n'
1289 << *Res << '\n' << "\n\n\n");
1290 }
1291 return Res;
1292 }
Chris Lattner87913512009-09-21 06:30:24 +00001293};
1294
Dan Gohmanb3579832010-04-15 17:08:50 +00001295}
1296
Chris Lattnera09fbf02009-10-10 23:50:30 +00001297/// ConstructSSAForLoadSet - Given a set of loads specified by ValuesPerBlock,
1298/// construct SSA form, allowing us to eliminate LI. This returns the value
1299/// that should be used at LI's definition site.
1300static Value *ConstructSSAForLoadSet(LoadInst *LI,
1301 SmallVectorImpl<AvailableValueInBlock> &ValuesPerBlock,
1302 const TargetData *TD,
Chris Lattnerd2191e52009-12-21 23:15:48 +00001303 const DominatorTree &DT,
Chris Lattnera09fbf02009-10-10 23:50:30 +00001304 AliasAnalysis *AA) {
Chris Lattnerd2191e52009-12-21 23:15:48 +00001305 // Check for the fully redundant, dominating load case. In this case, we can
1306 // just use the dominating value directly.
1307 if (ValuesPerBlock.size() == 1 &&
1308 DT.properlyDominates(ValuesPerBlock[0].BB, LI->getParent()))
1309 return ValuesPerBlock[0].MaterializeAdjustedValue(LI->getType(), TD);
1310
1311 // Otherwise, we have to construct SSA form.
Chris Lattnera09fbf02009-10-10 23:50:30 +00001312 SmallVector<PHINode*, 8> NewPHIs;
1313 SSAUpdater SSAUpdate(&NewPHIs);
1314 SSAUpdate.Initialize(LI);
1315
1316 const Type *LoadTy = LI->getType();
1317
Chris Lattner771a5422009-09-20 20:09:34 +00001318 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001319 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1320 BasicBlock *BB = AV.BB;
Chris Lattner771a5422009-09-20 20:09:34 +00001321
Chris Lattnera09fbf02009-10-10 23:50:30 +00001322 if (SSAUpdate.HasValueForBlock(BB))
1323 continue;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001324
Chris Lattner5362c542009-12-21 23:04:33 +00001325 SSAUpdate.AddAvailableValue(BB, AV.MaterializeAdjustedValue(LoadTy, TD));
Chris Lattner771a5422009-09-20 20:09:34 +00001326 }
Chris Lattnera09fbf02009-10-10 23:50:30 +00001327
1328 // Perform PHI construction.
1329 Value *V = SSAUpdate.GetValueInMiddleOfBlock(LI->getParent());
1330
1331 // If new PHI nodes were created, notify alias analysis.
Duncan Sands1df98592010-02-16 11:11:14 +00001332 if (V->getType()->isPointerTy())
Chris Lattnera09fbf02009-10-10 23:50:30 +00001333 for (unsigned i = 0, e = NewPHIs.size(); i != e; ++i)
1334 AA->copyValue(LI, NewPHIs[i]);
1335
1336 return V;
Chris Lattner771a5422009-09-20 20:09:34 +00001337}
1338
Gabor Greifea3eec92010-04-09 10:57:00 +00001339static bool isLifetimeStart(const Instruction *Inst) {
1340 if (const IntrinsicInst* II = dyn_cast<IntrinsicInst>(Inst))
Owen Anderson9ff5a232009-12-02 07:35:19 +00001341 return II->getIntrinsicID() == Intrinsic::lifetime_start;
Chris Lattner720e7902009-12-02 06:44:58 +00001342 return false;
1343}
1344
Owen Anderson62bc33c2007-08-16 22:02:55 +00001345/// processNonLocalLoad - Attempt to eliminate a load whose dependencies are
1346/// non-local by performing PHI construction.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001347bool GVN::processNonLocalLoad(LoadInst *LI,
Chris Lattner8e1e95c2008-03-21 22:01:16 +00001348 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001349 // Find the non-local dependencies of the load.
Chris Lattner0ee443d2009-12-22 04:25:02 +00001350 SmallVector<NonLocalDepResult, 64> Deps;
Chris Lattner91bcf642008-12-09 19:25:07 +00001351 MD->getNonLocalPointerDependency(LI->getOperand(0), true, LI->getParent(),
1352 Deps);
David Greenebf7f78e2010-01-05 01:27:17 +00001353 //DEBUG(dbgs() << "INVESTIGATING NONLOCAL LOAD: "
Dan Gohman2a298992009-07-31 20:24:18 +00001354 // << Deps.size() << *LI << '\n');
Daniel Dunbara279bc32009-09-20 02:20:51 +00001355
Owen Anderson516eb1c2008-08-26 22:07:42 +00001356 // If we had to process more than one hundred blocks to find the
1357 // dependencies, this load isn't worth worrying about. Optimizing
1358 // it will be too expensive.
Chris Lattner91bcf642008-12-09 19:25:07 +00001359 if (Deps.size() > 100)
Owen Anderson516eb1c2008-08-26 22:07:42 +00001360 return false;
Chris Lattner5f4f84b2008-12-18 00:51:32 +00001361
1362 // If we had a phi translation failure, we'll have a single entry which is a
1363 // clobber in the current block. Reject this early.
Chris Lattnere18b9712009-12-09 07:08:01 +00001364 if (Deps.size() == 1 && Deps[0].getResult().isClobber()) {
Torok Edwin4306b1a2009-06-17 18:48:18 +00001365 DEBUG(
David Greenebf7f78e2010-01-05 01:27:17 +00001366 dbgs() << "GVN: non-local load ";
1367 WriteAsOperand(dbgs(), LI);
1368 dbgs() << " is clobbered by " << *Deps[0].getResult().getInst() << '\n';
Torok Edwin4306b1a2009-06-17 18:48:18 +00001369 );
Chris Lattner5f4f84b2008-12-18 00:51:32 +00001370 return false;
Torok Edwin4306b1a2009-06-17 18:48:18 +00001371 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001372
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001373 // Filter out useless results (non-locals, etc). Keep track of the blocks
1374 // where we have a value available in repl, also keep track of whether we see
1375 // dependencies that produce an unknown value for the load (such as a call
1376 // that could potentially clobber the load).
Chris Lattner87913512009-09-21 06:30:24 +00001377 SmallVector<AvailableValueInBlock, 16> ValuesPerBlock;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001378 SmallVector<BasicBlock*, 16> UnavailableBlocks;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001379
Chris Lattner771a5422009-09-20 20:09:34 +00001380 const TargetData *TD = 0;
1381
Chris Lattner91bcf642008-12-09 19:25:07 +00001382 for (unsigned i = 0, e = Deps.size(); i != e; ++i) {
Chris Lattnere18b9712009-12-09 07:08:01 +00001383 BasicBlock *DepBB = Deps[i].getBB();
1384 MemDepResult DepInfo = Deps[i].getResult();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001385
Chris Lattnerb51deb92008-12-05 21:04:20 +00001386 if (DepInfo.isClobber()) {
Chris Lattneraf064ae2009-12-09 18:21:46 +00001387 // The address being loaded in this non-local block may not be the same as
1388 // the pointer operand of the load if PHI translation occurs. Make sure
1389 // to consider the right address.
1390 Value *Address = Deps[i].getAddress();
1391
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001392 // If the dependence is to a store that writes to a superset of the bits
1393 // read by the load, we can extract the bits we need for the load from the
1394 // stored value.
1395 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInfo.getInst())) {
1396 if (TD == 0)
1397 TD = getAnalysisIfAvailable<TargetData>();
Chris Lattneraf064ae2009-12-09 18:21:46 +00001398 if (TD && Address) {
1399 int Offset = AnalyzeLoadFromClobberingStore(LI->getType(), Address,
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001400 DepSI, *TD);
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001401 if (Offset != -1) {
1402 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1403 DepSI->getOperand(0),
1404 Offset));
1405 continue;
1406 }
1407 }
1408 }
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001409
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001410 // If the clobbering value is a memset/memcpy/memmove, see if we can
1411 // forward a value on from it.
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001412 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(DepInfo.getInst())) {
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001413 if (TD == 0)
1414 TD = getAnalysisIfAvailable<TargetData>();
Chris Lattneraf064ae2009-12-09 18:21:46 +00001415 if (TD && Address) {
1416 int Offset = AnalyzeLoadFromClobberingMemInst(LI->getType(), Address,
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001417 DepMI, *TD);
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001418 if (Offset != -1) {
1419 ValuesPerBlock.push_back(AvailableValueInBlock::getMI(DepBB, DepMI,
1420 Offset));
1421 continue;
1422 }
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001423 }
1424 }
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001425
Chris Lattnerb51deb92008-12-05 21:04:20 +00001426 UnavailableBlocks.push_back(DepBB);
1427 continue;
1428 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001429
Chris Lattnerb51deb92008-12-05 21:04:20 +00001430 Instruction *DepInst = DepInfo.getInst();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001431
Chris Lattnerb51deb92008-12-05 21:04:20 +00001432 // Loading the allocation -> undef.
Chris Lattner720e7902009-12-02 06:44:58 +00001433 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst) ||
Owen Anderson9ff5a232009-12-02 07:35:19 +00001434 // Loading immediately after lifetime begin -> undef.
1435 isLifetimeStart(DepInst)) {
Chris Lattner87913512009-09-21 06:30:24 +00001436 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1437 UndefValue::get(LI->getType())));
Chris Lattnerbf145d62008-12-01 01:15:42 +00001438 continue;
1439 }
Owen Andersonb62f7922009-10-28 07:05:35 +00001440
Chris Lattner87913512009-09-21 06:30:24 +00001441 if (StoreInst *S = dyn_cast<StoreInst>(DepInst)) {
Daniel Dunbara279bc32009-09-20 02:20:51 +00001442 // Reject loads and stores that are to the same address but are of
Chris Lattner771a5422009-09-20 20:09:34 +00001443 // different types if we have to.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001444 if (S->getOperand(0)->getType() != LI->getType()) {
Chris Lattner771a5422009-09-20 20:09:34 +00001445 if (TD == 0)
1446 TD = getAnalysisIfAvailable<TargetData>();
1447
1448 // If the stored value is larger or equal to the loaded value, we can
1449 // reuse it.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001450 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(S->getOperand(0),
1451 LI->getType(), *TD)) {
Chris Lattner771a5422009-09-20 20:09:34 +00001452 UnavailableBlocks.push_back(DepBB);
1453 continue;
1454 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001455 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001456
Chris Lattner87913512009-09-21 06:30:24 +00001457 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1458 S->getOperand(0)));
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001459 continue;
1460 }
1461
1462 if (LoadInst *LD = dyn_cast<LoadInst>(DepInst)) {
Chris Lattner771a5422009-09-20 20:09:34 +00001463 // If the types mismatch and we can't handle it, reject reuse of the load.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001464 if (LD->getType() != LI->getType()) {
Chris Lattner771a5422009-09-20 20:09:34 +00001465 if (TD == 0)
1466 TD = getAnalysisIfAvailable<TargetData>();
1467
1468 // If the stored value is larger or equal to the loaded value, we can
1469 // reuse it.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001470 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(LD, LI->getType(),*TD)){
Chris Lattner771a5422009-09-20 20:09:34 +00001471 UnavailableBlocks.push_back(DepBB);
1472 continue;
1473 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001474 }
Chris Lattner87913512009-09-21 06:30:24 +00001475 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB, LD));
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001476 continue;
Owen Anderson0cd32032007-07-25 19:57:03 +00001477 }
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001478
1479 UnavailableBlocks.push_back(DepBB);
1480 continue;
Chris Lattner88365bb2008-03-21 21:14:38 +00001481 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001482
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001483 // If we have no predecessors that produce a known value for this load, exit
1484 // early.
1485 if (ValuesPerBlock.empty()) return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001486
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001487 // If all of the instructions we depend on produce a known value for this
1488 // load, then it is fully redundant and we can use PHI insertion to compute
1489 // its value. Insert PHIs and remove the fully redundant value now.
1490 if (UnavailableBlocks.empty()) {
David Greenebf7f78e2010-01-05 01:27:17 +00001491 DEBUG(dbgs() << "GVN REMOVING NONLOCAL LOAD: " << *LI << '\n');
Chris Lattner771a5422009-09-20 20:09:34 +00001492
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001493 // Perform PHI construction.
Chris Lattnerd2191e52009-12-21 23:15:48 +00001494 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnera09fbf02009-10-10 23:50:30 +00001495 VN.getAliasAnalysis());
Chris Lattner771a5422009-09-20 20:09:34 +00001496 LI->replaceAllUsesWith(V);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001497
Chris Lattner771a5422009-09-20 20:09:34 +00001498 if (isa<PHINode>(V))
1499 V->takeName(LI);
Duncan Sands1df98592010-02-16 11:11:14 +00001500 if (V->getType()->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +00001501 MD->invalidateCachedPointerInfo(V);
Bob Wilson74175c22010-02-22 21:39:41 +00001502 VN.erase(LI);
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001503 toErase.push_back(LI);
Dan Gohmanfe601042010-06-22 15:08:57 +00001504 ++NumGVNLoad;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001505 return true;
1506 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001507
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001508 if (!EnablePRE || !EnableLoadPRE)
1509 return false;
1510
1511 // Okay, we have *some* definitions of the value. This means that the value
1512 // is available in some of our (transitive) predecessors. Lets think about
1513 // doing PRE of this load. This will involve inserting a new load into the
1514 // predecessor when it's not available. We could do this in general, but
1515 // prefer to not increase code size. As such, we only do this when we know
1516 // that we only have to insert *one* load (which means we're basically moving
1517 // the load, not inserting a new one).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001518
Owen Anderson88554df2009-05-31 09:03:40 +00001519 SmallPtrSet<BasicBlock *, 4> Blockers;
1520 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1521 Blockers.insert(UnavailableBlocks[i]);
1522
1523 // Lets find first basic block with more than one predecessor. Walk backwards
1524 // through predecessors if needed.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001525 BasicBlock *LoadBB = LI->getParent();
Owen Anderson88554df2009-05-31 09:03:40 +00001526 BasicBlock *TmpBB = LoadBB;
1527
1528 bool isSinglePred = false;
Dale Johannesen42c3f552009-06-17 20:48:23 +00001529 bool allSingleSucc = true;
Owen Anderson88554df2009-05-31 09:03:40 +00001530 while (TmpBB->getSinglePredecessor()) {
1531 isSinglePred = true;
1532 TmpBB = TmpBB->getSinglePredecessor();
Owen Anderson88554df2009-05-31 09:03:40 +00001533 if (TmpBB == LoadBB) // Infinite (unreachable) loop.
1534 return false;
1535 if (Blockers.count(TmpBB))
1536 return false;
Dale Johannesen42c3f552009-06-17 20:48:23 +00001537 if (TmpBB->getTerminator()->getNumSuccessors() != 1)
1538 allSingleSucc = false;
Owen Anderson88554df2009-05-31 09:03:40 +00001539 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001540
Owen Anderson88554df2009-05-31 09:03:40 +00001541 assert(TmpBB);
1542 LoadBB = TmpBB;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001543
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001544 // If we have a repl set with LI itself in it, this means we have a loop where
1545 // at least one of the values is LI. Since this means that we won't be able
1546 // to eliminate LI even if we insert uses in the other predecessors, we will
1547 // end up increasing code size. Reject this by scanning for LI.
Bob Wilson3bd19d72010-03-02 00:09:29 +00001548 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
1549 if (ValuesPerBlock[i].isSimpleValue() &&
1550 ValuesPerBlock[i].getSimpleValue() == LI) {
1551 // Skip cases where LI is the only definition, even for EnableFullLoadPRE.
1552 if (!EnableFullLoadPRE || e == 1)
Bob Wilson6cad4172010-02-01 21:17:14 +00001553 return false;
Bob Wilson3bd19d72010-03-02 00:09:29 +00001554 }
Bob Wilson6cad4172010-02-01 21:17:14 +00001555 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001556
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001557 // FIXME: It is extremely unclear what this loop is doing, other than
1558 // artificially restricting loadpre.
Owen Anderson88554df2009-05-31 09:03:40 +00001559 if (isSinglePred) {
1560 bool isHot = false;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001561 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
1562 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1563 if (AV.isSimpleValue())
Daniel Dunbara279bc32009-09-20 02:20:51 +00001564 // "Hot" Instruction is in some loop (because it dominates its dep.
1565 // instruction).
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001566 if (Instruction *I = dyn_cast<Instruction>(AV.getSimpleValue()))
1567 if (DT->dominates(LI, I)) {
1568 isHot = true;
1569 break;
1570 }
1571 }
Owen Anderson88554df2009-05-31 09:03:40 +00001572
1573 // We are interested only in "hot" instructions. We don't want to do any
1574 // mis-optimizations here.
1575 if (!isHot)
1576 return false;
1577 }
1578
Bob Wilson6cad4172010-02-01 21:17:14 +00001579 // Check to see how many predecessors have the loaded value fully
1580 // available.
1581 DenseMap<BasicBlock*, Value*> PredLoads;
Chris Lattner72bc70d2008-12-05 07:49:08 +00001582 DenseMap<BasicBlock*, char> FullyAvailableBlocks;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001583 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner87913512009-09-21 06:30:24 +00001584 FullyAvailableBlocks[ValuesPerBlock[i].BB] = true;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001585 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1586 FullyAvailableBlocks[UnavailableBlocks[i]] = false;
1587
Bob Wilson34414a62010-05-04 20:03:21 +00001588 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> NeedToSplit;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001589 for (pred_iterator PI = pred_begin(LoadBB), E = pred_end(LoadBB);
1590 PI != E; ++PI) {
Bob Wilson6cad4172010-02-01 21:17:14 +00001591 BasicBlock *Pred = *PI;
1592 if (IsValueFullyAvailableInBlock(Pred, FullyAvailableBlocks)) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001593 continue;
Bob Wilson6cad4172010-02-01 21:17:14 +00001594 }
1595 PredLoads[Pred] = 0;
Bob Wilson484d4a32010-02-16 19:51:59 +00001596
Bob Wilson6cad4172010-02-01 21:17:14 +00001597 if (Pred->getTerminator()->getNumSuccessors() != 1) {
Bob Wilson484d4a32010-02-16 19:51:59 +00001598 if (isa<IndirectBrInst>(Pred->getTerminator())) {
1599 DEBUG(dbgs() << "COULD NOT PRE LOAD BECAUSE OF INDBR CRITICAL EDGE '"
1600 << Pred->getName() << "': " << *LI << '\n');
1601 return false;
1602 }
Bob Wilsonae23daf2010-02-16 21:06:42 +00001603 unsigned SuccNum = GetSuccessorNumber(Pred, LoadBB);
Bob Wilson34414a62010-05-04 20:03:21 +00001604 NeedToSplit.push_back(std::make_pair(Pred->getTerminator(), SuccNum));
Bob Wilson6cad4172010-02-01 21:17:14 +00001605 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001606 }
Bob Wilson34414a62010-05-04 20:03:21 +00001607 if (!NeedToSplit.empty()) {
Bob Wilsonbc786532010-05-05 20:44:15 +00001608 toSplit.append(NeedToSplit.begin(), NeedToSplit.end());
Bob Wilson70704972010-03-01 23:37:32 +00001609 return false;
Bob Wilson34414a62010-05-04 20:03:21 +00001610 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001611
Bob Wilson6cad4172010-02-01 21:17:14 +00001612 // Decide whether PRE is profitable for this load.
1613 unsigned NumUnavailablePreds = PredLoads.size();
1614 assert(NumUnavailablePreds != 0 &&
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001615 "Fully available value should be eliminated above!");
Bob Wilson6cad4172010-02-01 21:17:14 +00001616 if (!EnableFullLoadPRE) {
1617 // If this load is unavailable in multiple predecessors, reject it.
1618 // FIXME: If we could restructure the CFG, we could make a common pred with
1619 // all the preds that don't have an available LI and insert a new load into
1620 // that one block.
1621 if (NumUnavailablePreds != 1)
1622 return false;
Owen Andersona37226a2007-08-07 23:12:31 +00001623 }
Bob Wilson6cad4172010-02-01 21:17:14 +00001624
1625 // Check if the load can safely be moved to all the unavailable predecessors.
1626 bool CanDoPRE = true;
Chris Lattnerdd696052009-11-28 15:39:14 +00001627 SmallVector<Instruction*, 8> NewInsts;
Bob Wilson6cad4172010-02-01 21:17:14 +00001628 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1629 E = PredLoads.end(); I != E; ++I) {
1630 BasicBlock *UnavailablePred = I->first;
1631
1632 // Do PHI translation to get its value in the predecessor if necessary. The
1633 // returned pointer (if non-null) is guaranteed to dominate UnavailablePred.
1634
1635 // If all preds have a single successor, then we know it is safe to insert
1636 // the load on the pred (?!?), so we can insert code to materialize the
1637 // pointer if it is not available.
1638 PHITransAddr Address(LI->getOperand(0), TD);
1639 Value *LoadPtr = 0;
1640 if (allSingleSucc) {
1641 LoadPtr = Address.PHITranslateWithInsertion(LoadBB, UnavailablePred,
1642 *DT, NewInsts);
1643 } else {
Daniel Dunbar6d8f2ca2010-02-24 08:48:04 +00001644 Address.PHITranslateValue(LoadBB, UnavailablePred, DT);
Bob Wilson6cad4172010-02-01 21:17:14 +00001645 LoadPtr = Address.getAddr();
Bob Wilson6cad4172010-02-01 21:17:14 +00001646 }
1647
1648 // If we couldn't find or insert a computation of this phi translated value,
1649 // we fail PRE.
1650 if (LoadPtr == 0) {
1651 DEBUG(dbgs() << "COULDN'T INSERT PHI TRANSLATED VALUE OF: "
1652 << *LI->getOperand(0) << "\n");
1653 CanDoPRE = false;
1654 break;
1655 }
1656
1657 // Make sure it is valid to move this load here. We have to watch out for:
1658 // @1 = getelementptr (i8* p, ...
1659 // test p and branch if == 0
1660 // load @1
1661 // It is valid to have the getelementptr before the test, even if p can be 0,
1662 // as getelementptr only does address arithmetic.
1663 // If we are not pushing the value through any multiple-successor blocks
1664 // we do not have this case. Otherwise, check that the load is safe to
1665 // put anywhere; this can be improved, but should be conservatively safe.
1666 if (!allSingleSucc &&
1667 // FIXME: REEVALUTE THIS.
1668 !isSafeToLoadUnconditionally(LoadPtr,
1669 UnavailablePred->getTerminator(),
1670 LI->getAlignment(), TD)) {
1671 CanDoPRE = false;
1672 break;
1673 }
1674
1675 I->second = LoadPtr;
Chris Lattner05e15f82009-12-09 01:59:31 +00001676 }
1677
Bob Wilson6cad4172010-02-01 21:17:14 +00001678 if (!CanDoPRE) {
1679 while (!NewInsts.empty())
1680 NewInsts.pop_back_val()->eraseFromParent();
Dale Johannesen42c3f552009-06-17 20:48:23 +00001681 return false;
Chris Lattner0c264b12009-11-28 16:08:18 +00001682 }
Dale Johannesen42c3f552009-06-17 20:48:23 +00001683
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001684 // Okay, we can eliminate this load by inserting a reload in the predecessor
1685 // and using PHI construction to get the value in the other predecessors, do
1686 // it.
David Greenebf7f78e2010-01-05 01:27:17 +00001687 DEBUG(dbgs() << "GVN REMOVING PRE LOAD: " << *LI << '\n');
Chris Lattner0c264b12009-11-28 16:08:18 +00001688 DEBUG(if (!NewInsts.empty())
David Greenebf7f78e2010-01-05 01:27:17 +00001689 dbgs() << "INSERTED " << NewInsts.size() << " INSTS: "
Chris Lattner0c264b12009-11-28 16:08:18 +00001690 << *NewInsts.back() << '\n');
1691
Bob Wilson6cad4172010-02-01 21:17:14 +00001692 // Assign value numbers to the new instructions.
1693 for (unsigned i = 0, e = NewInsts.size(); i != e; ++i) {
1694 // FIXME: We really _ought_ to insert these value numbers into their
1695 // parent's availability map. However, in doing so, we risk getting into
1696 // ordering issues. If a block hasn't been processed yet, we would be
1697 // marking a value as AVAIL-IN, which isn't what we intend.
1698 VN.lookup_or_add(NewInsts[i]);
1699 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001700
Bob Wilson6cad4172010-02-01 21:17:14 +00001701 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1702 E = PredLoads.end(); I != E; ++I) {
1703 BasicBlock *UnavailablePred = I->first;
1704 Value *LoadPtr = I->second;
1705
1706 Value *NewLoad = new LoadInst(LoadPtr, LI->getName()+".pre", false,
1707 LI->getAlignment(),
1708 UnavailablePred->getTerminator());
1709
1710 // Add the newly created load.
1711 ValuesPerBlock.push_back(AvailableValueInBlock::get(UnavailablePred,
1712 NewLoad));
Bob Wilson188f4282010-02-23 05:55:00 +00001713 MD->invalidateCachedPointerInfo(LoadPtr);
1714 DEBUG(dbgs() << "GVN INSERTED " << *NewLoad << '\n');
Bob Wilson6cad4172010-02-01 21:17:14 +00001715 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001716
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001717 // Perform PHI construction.
Chris Lattnerd2191e52009-12-21 23:15:48 +00001718 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnera09fbf02009-10-10 23:50:30 +00001719 VN.getAliasAnalysis());
Chris Lattner771a5422009-09-20 20:09:34 +00001720 LI->replaceAllUsesWith(V);
1721 if (isa<PHINode>(V))
1722 V->takeName(LI);
Duncan Sands1df98592010-02-16 11:11:14 +00001723 if (V->getType()->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +00001724 MD->invalidateCachedPointerInfo(V);
Bob Wilson74175c22010-02-22 21:39:41 +00001725 VN.erase(LI);
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001726 toErase.push_back(LI);
Dan Gohmanfe601042010-06-22 15:08:57 +00001727 ++NumPRELoad;
Owen Anderson0cd32032007-07-25 19:57:03 +00001728 return true;
1729}
1730
Owen Anderson62bc33c2007-08-16 22:02:55 +00001731/// processLoad - Attempt to eliminate a load, first by eliminating it
1732/// locally, and then attempting non-local elimination if that fails.
Chris Lattnerb51deb92008-12-05 21:04:20 +00001733bool GVN::processLoad(LoadInst *L, SmallVectorImpl<Instruction*> &toErase) {
Dan Gohman4ec01b22009-11-14 02:27:51 +00001734 if (!MD)
1735 return false;
1736
Chris Lattnerb51deb92008-12-05 21:04:20 +00001737 if (L->isVolatile())
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001738 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001739
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001740 // ... to a pointer that has been loaded from before...
Chris Lattnerb2412a82009-09-21 02:42:51 +00001741 MemDepResult Dep = MD->getDependency(L);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001742
Chris Lattnerb51deb92008-12-05 21:04:20 +00001743 // If the value isn't available, don't do anything!
Chris Lattnerb2412a82009-09-21 02:42:51 +00001744 if (Dep.isClobber()) {
Chris Lattnereed919b2009-09-21 05:57:11 +00001745 // Check to see if we have something like this:
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001746 // store i32 123, i32* %P
1747 // %A = bitcast i32* %P to i8*
1748 // %B = gep i8* %A, i32 1
1749 // %C = load i8* %B
1750 //
1751 // We could do that by recognizing if the clobber instructions are obviously
1752 // a common base + constant offset, and if the previous store (or memset)
1753 // completely covers this load. This sort of thing can happen in bitfield
1754 // access code.
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001755 Value *AvailVal = 0;
Chris Lattnereed919b2009-09-21 05:57:11 +00001756 if (StoreInst *DepSI = dyn_cast<StoreInst>(Dep.getInst()))
Chris Lattner1ce08292009-09-21 06:22:46 +00001757 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001758 int Offset = AnalyzeLoadFromClobberingStore(L->getType(),
1759 L->getPointerOperand(),
1760 DepSI, *TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001761 if (Offset != -1)
1762 AvailVal = GetStoreValueForLoad(DepSI->getOperand(0), Offset,
1763 L->getType(), L, *TD);
Chris Lattner1ce08292009-09-21 06:22:46 +00001764 }
Chris Lattnereed919b2009-09-21 05:57:11 +00001765
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001766 // If the clobbering value is a memset/memcpy/memmove, see if we can forward
1767 // a value on from it.
1768 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(Dep.getInst())) {
1769 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001770 int Offset = AnalyzeLoadFromClobberingMemInst(L->getType(),
1771 L->getPointerOperand(),
1772 DepMI, *TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001773 if (Offset != -1)
1774 AvailVal = GetMemInstValueForLoad(DepMI, Offset, L->getType(), L,*TD);
1775 }
1776 }
1777
1778 if (AvailVal) {
David Greenebf7f78e2010-01-05 01:27:17 +00001779 DEBUG(dbgs() << "GVN COERCED INST:\n" << *Dep.getInst() << '\n'
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001780 << *AvailVal << '\n' << *L << "\n\n\n");
1781
1782 // Replace the load!
1783 L->replaceAllUsesWith(AvailVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001784 if (AvailVal->getType()->isPointerTy())
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001785 MD->invalidateCachedPointerInfo(AvailVal);
Bob Wilson74175c22010-02-22 21:39:41 +00001786 VN.erase(L);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001787 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001788 ++NumGVNLoad;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001789 return true;
1790 }
1791
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001792 DEBUG(
1793 // fast print dep, using operator<< on instruction would be too slow
David Greenebf7f78e2010-01-05 01:27:17 +00001794 dbgs() << "GVN: load ";
1795 WriteAsOperand(dbgs(), L);
Chris Lattnerb2412a82009-09-21 02:42:51 +00001796 Instruction *I = Dep.getInst();
David Greenebf7f78e2010-01-05 01:27:17 +00001797 dbgs() << " is clobbered by " << *I << '\n';
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001798 );
Chris Lattnerb51deb92008-12-05 21:04:20 +00001799 return false;
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001800 }
Chris Lattnerb51deb92008-12-05 21:04:20 +00001801
1802 // If it is defined in another block, try harder.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001803 if (Dep.isNonLocal())
Chris Lattnerb51deb92008-12-05 21:04:20 +00001804 return processNonLocalLoad(L, toErase);
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001805
Chris Lattnerb2412a82009-09-21 02:42:51 +00001806 Instruction *DepInst = Dep.getInst();
Chris Lattnerb51deb92008-12-05 21:04:20 +00001807 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInst)) {
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001808 Value *StoredVal = DepSI->getOperand(0);
1809
1810 // The store and load are to a must-aliased pointer, but they may not
1811 // actually have the same type. See if we know how to reuse the stored
1812 // value (depending on its type).
1813 const TargetData *TD = 0;
Chris Lattnera52fce42009-10-21 04:11:19 +00001814 if (StoredVal->getType() != L->getType()) {
1815 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1816 StoredVal = CoerceAvailableValueToLoadType(StoredVal, L->getType(),
1817 L, *TD);
1818 if (StoredVal == 0)
1819 return false;
1820
David Greenebf7f78e2010-01-05 01:27:17 +00001821 DEBUG(dbgs() << "GVN COERCED STORE:\n" << *DepSI << '\n' << *StoredVal
Chris Lattnera52fce42009-10-21 04:11:19 +00001822 << '\n' << *L << "\n\n\n");
1823 }
1824 else
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001825 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001826 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001827
Chris Lattnerb51deb92008-12-05 21:04:20 +00001828 // Remove it!
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001829 L->replaceAllUsesWith(StoredVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001830 if (StoredVal->getType()->isPointerTy())
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001831 MD->invalidateCachedPointerInfo(StoredVal);
Bob Wilson74175c22010-02-22 21:39:41 +00001832 VN.erase(L);
Chris Lattnerb51deb92008-12-05 21:04:20 +00001833 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001834 ++NumGVNLoad;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001835 return true;
1836 }
1837
1838 if (LoadInst *DepLI = dyn_cast<LoadInst>(DepInst)) {
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001839 Value *AvailableVal = DepLI;
1840
1841 // The loads are of a must-aliased pointer, but they may not actually have
1842 // the same type. See if we know how to reuse the previously loaded value
1843 // (depending on its type).
1844 const TargetData *TD = 0;
Chris Lattnera52fce42009-10-21 04:11:19 +00001845 if (DepLI->getType() != L->getType()) {
1846 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1847 AvailableVal = CoerceAvailableValueToLoadType(DepLI, L->getType(), L,*TD);
1848 if (AvailableVal == 0)
1849 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001850
David Greenebf7f78e2010-01-05 01:27:17 +00001851 DEBUG(dbgs() << "GVN COERCED LOAD:\n" << *DepLI << "\n" << *AvailableVal
Chris Lattnera52fce42009-10-21 04:11:19 +00001852 << "\n" << *L << "\n\n\n");
1853 }
1854 else
1855 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001856 }
1857
Chris Lattnerb51deb92008-12-05 21:04:20 +00001858 // Remove it!
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001859 L->replaceAllUsesWith(AvailableVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001860 if (DepLI->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001861 MD->invalidateCachedPointerInfo(DepLI);
Bob Wilson74175c22010-02-22 21:39:41 +00001862 VN.erase(L);
Chris Lattnerb51deb92008-12-05 21:04:20 +00001863 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001864 ++NumGVNLoad;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001865 return true;
1866 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001867
Chris Lattner237a8282008-11-30 01:39:32 +00001868 // If this load really doesn't depend on anything, then we must be loading an
1869 // undef value. This can happen when loading for a fresh allocation with no
1870 // intervening stores, for example.
Victor Hernandez7b929da2009-10-23 21:09:37 +00001871 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst)) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001872 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson74175c22010-02-22 21:39:41 +00001873 VN.erase(L);
Chris Lattner237a8282008-11-30 01:39:32 +00001874 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001875 ++NumGVNLoad;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001876 return true;
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001877 }
Owen Andersonb62f7922009-10-28 07:05:35 +00001878
Owen Anderson9ff5a232009-12-02 07:35:19 +00001879 // If this load occurs either right after a lifetime begin,
Owen Andersonb62f7922009-10-28 07:05:35 +00001880 // then the loaded value is undefined.
1881 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(DepInst)) {
Owen Anderson9ff5a232009-12-02 07:35:19 +00001882 if (II->getIntrinsicID() == Intrinsic::lifetime_start) {
Owen Andersonb62f7922009-10-28 07:05:35 +00001883 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson74175c22010-02-22 21:39:41 +00001884 VN.erase(L);
Owen Andersonb62f7922009-10-28 07:05:35 +00001885 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001886 ++NumGVNLoad;
Owen Andersonb62f7922009-10-28 07:05:35 +00001887 return true;
1888 }
1889 }
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001890
Chris Lattnerb51deb92008-12-05 21:04:20 +00001891 return false;
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001892}
1893
Chris Lattnerb2412a82009-09-21 02:42:51 +00001894Value *GVN::lookupNumber(BasicBlock *BB, uint32_t num) {
Owen Andersonb70a5712008-06-23 17:49:45 +00001895 DenseMap<BasicBlock*, ValueNumberScope*>::iterator I = localAvail.find(BB);
1896 if (I == localAvail.end())
1897 return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001898
Chris Lattnerb2412a82009-09-21 02:42:51 +00001899 ValueNumberScope *Locals = I->second;
1900 while (Locals) {
1901 DenseMap<uint32_t, Value*>::iterator I = Locals->table.find(num);
1902 if (I != Locals->table.end())
Owen Anderson6fafe842008-06-20 01:15:47 +00001903 return I->second;
Chris Lattnerb2412a82009-09-21 02:42:51 +00001904 Locals = Locals->parent;
Owen Anderson6fafe842008-06-20 01:15:47 +00001905 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001906
Owen Anderson6fafe842008-06-20 01:15:47 +00001907 return 0;
1908}
1909
Owen Anderson255dafc2008-12-15 02:03:00 +00001910
Owen Anderson36057c72007-08-14 18:16:29 +00001911/// processInstruction - When calculating availability, handle an instruction
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001912/// by inserting it into the appropriate sets
Owen Andersonaf4240a2008-06-12 19:25:32 +00001913bool GVN::processInstruction(Instruction *I,
Chris Lattner8e1e95c2008-03-21 22:01:16 +00001914 SmallVectorImpl<Instruction*> &toErase) {
Devang Patelbe905e22010-02-11 00:20:49 +00001915 // Ignore dbg info intrinsics.
1916 if (isa<DbgInfoIntrinsic>(I))
1917 return false;
1918
Chris Lattnerb2412a82009-09-21 02:42:51 +00001919 if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
1920 bool Changed = processLoad(LI, toErase);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001921
Chris Lattnerb2412a82009-09-21 02:42:51 +00001922 if (!Changed) {
1923 unsigned Num = VN.lookup_or_add(LI);
1924 localAvail[I->getParent()]->table.insert(std::make_pair(Num, LI));
Owen Andersonb2303722008-06-18 21:41:49 +00001925 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001926
Chris Lattnerb2412a82009-09-21 02:42:51 +00001927 return Changed;
Owen Andersonb2303722008-06-18 21:41:49 +00001928 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001929
Chris Lattnerb2412a82009-09-21 02:42:51 +00001930 uint32_t NextNum = VN.getNextUnusedValueNumber();
1931 unsigned Num = VN.lookup_or_add(I);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001932
Chris Lattnerb2412a82009-09-21 02:42:51 +00001933 if (BranchInst *BI = dyn_cast<BranchInst>(I)) {
1934 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001935
Owen Andersone8a290f2009-04-01 23:53:49 +00001936 if (!BI->isConditional() || isa<Constant>(BI->getCondition()))
1937 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001938
Chris Lattnerb2412a82009-09-21 02:42:51 +00001939 Value *BranchCond = BI->getCondition();
1940 uint32_t CondVN = VN.lookup_or_add(BranchCond);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001941
Chris Lattnerb2412a82009-09-21 02:42:51 +00001942 BasicBlock *TrueSucc = BI->getSuccessor(0);
1943 BasicBlock *FalseSucc = BI->getSuccessor(1);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001944
Chris Lattnerb2412a82009-09-21 02:42:51 +00001945 if (TrueSucc->getSinglePredecessor())
1946 localAvail[TrueSucc]->table[CondVN] =
1947 ConstantInt::getTrue(TrueSucc->getContext());
1948 if (FalseSucc->getSinglePredecessor())
1949 localAvail[FalseSucc]->table[CondVN] =
1950 ConstantInt::getFalse(TrueSucc->getContext());
Owen Andersone8a290f2009-04-01 23:53:49 +00001951
1952 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001953
Owen Andersone5ffa902008-04-07 09:59:07 +00001954 // Allocations are always uniquely numbered, so we can save time and memory
Daniel Dunbara279bc32009-09-20 02:20:51 +00001955 // by fast failing them.
Victor Hernandez7b929da2009-10-23 21:09:37 +00001956 } else if (isa<AllocaInst>(I) || isa<TerminatorInst>(I)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001957 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Andersone5ffa902008-04-07 09:59:07 +00001958 return false;
Owen Andersonb2303722008-06-18 21:41:49 +00001959 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001960
Owen Anderson62bc33c2007-08-16 22:02:55 +00001961 // Collapse PHI nodes
Owen Anderson31f49672007-08-14 18:33:27 +00001962 if (PHINode* p = dyn_cast<PHINode>(I)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001963 Value *constVal = CollapsePhi(p);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001964
Owen Anderson31f49672007-08-14 18:33:27 +00001965 if (constVal) {
Owen Anderson1defe2d2007-08-16 22:51:56 +00001966 p->replaceAllUsesWith(constVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001967 if (MD && constVal->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001968 MD->invalidateCachedPointerInfo(constVal);
Owen Andersonae53c932008-12-23 00:49:51 +00001969 VN.erase(p);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001970
Owen Anderson1defe2d2007-08-16 22:51:56 +00001971 toErase.push_back(p);
Owen Andersonb2303722008-06-18 21:41:49 +00001972 } else {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001973 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson31f49672007-08-14 18:33:27 +00001974 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001975
Owen Anderson0ae33ef2008-07-03 17:44:33 +00001976 // If the number we were assigned was a brand new VN, then we don't
1977 // need to do a lookup to see if the number already exists
1978 // somewhere in the domtree: it can't!
Chris Lattnerb2412a82009-09-21 02:42:51 +00001979 } else if (Num == NextNum) {
1980 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001981
Owen Anderson255dafc2008-12-15 02:03:00 +00001982 // Perform fast-path value-number based elimination of values inherited from
1983 // dominators.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001984 } else if (Value *repl = lookupNumber(I->getParent(), Num)) {
Owen Anderson5fc4aba2007-12-08 01:37:09 +00001985 // Remove it!
Owen Andersonbf7d0bc2007-07-31 23:27:13 +00001986 VN.erase(I);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001987 I->replaceAllUsesWith(repl);
Duncan Sands1df98592010-02-16 11:11:14 +00001988 if (MD && repl->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001989 MD->invalidateCachedPointerInfo(repl);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001990 toErase.push_back(I);
1991 return true;
Owen Anderson255dafc2008-12-15 02:03:00 +00001992
Owen Anderson0ae33ef2008-07-03 17:44:33 +00001993 } else {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001994 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001995 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001996
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001997 return false;
1998}
1999
Bill Wendling30788b82008-12-22 22:32:22 +00002000/// runOnFunction - This is the main transformation entry point for a function.
Owen Anderson3e75a422007-08-14 18:04:11 +00002001bool GVN::runOnFunction(Function& F) {
Dan Gohman4ec01b22009-11-14 02:27:51 +00002002 if (!NoLoads)
2003 MD = &getAnalysis<MemoryDependenceAnalysis>();
Chris Lattner663e4412008-12-01 00:40:32 +00002004 DT = &getAnalysis<DominatorTree>();
Owen Andersona472c4a2008-05-12 20:15:55 +00002005 VN.setAliasAnalysis(&getAnalysis<AliasAnalysis>());
Chris Lattner663e4412008-12-01 00:40:32 +00002006 VN.setMemDep(MD);
2007 VN.setDomTree(DT);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002008
Chris Lattnerb2412a82009-09-21 02:42:51 +00002009 bool Changed = false;
2010 bool ShouldContinue = true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002011
Owen Anderson5d0af032008-07-16 17:52:31 +00002012 // Merge unconditional branches, allowing PRE to catch more
2013 // optimization opportunities.
2014 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002015 BasicBlock *BB = FI;
Owen Anderson5d0af032008-07-16 17:52:31 +00002016 ++FI;
Owen Andersonb31b06d2008-07-17 00:01:40 +00002017 bool removedBlock = MergeBlockIntoPredecessor(BB, this);
Dan Gohmanfe601042010-06-22 15:08:57 +00002018 if (removedBlock) ++NumGVNBlocks;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002019
Chris Lattnerb2412a82009-09-21 02:42:51 +00002020 Changed |= removedBlock;
Owen Anderson5d0af032008-07-16 17:52:31 +00002021 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002022
Chris Lattnerae199312008-12-09 19:21:47 +00002023 unsigned Iteration = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002024
Chris Lattnerb2412a82009-09-21 02:42:51 +00002025 while (ShouldContinue) {
David Greenebf7f78e2010-01-05 01:27:17 +00002026 DEBUG(dbgs() << "GVN iteration: " << Iteration << "\n");
Chris Lattnerb2412a82009-09-21 02:42:51 +00002027 ShouldContinue = iterateOnFunction(F);
Bob Wilson484d4a32010-02-16 19:51:59 +00002028 if (splitCriticalEdges())
2029 ShouldContinue = true;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002030 Changed |= ShouldContinue;
Chris Lattnerae199312008-12-09 19:21:47 +00002031 ++Iteration;
Owen Anderson3e75a422007-08-14 18:04:11 +00002032 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002033
Owen Andersone98c54c2008-07-18 18:03:38 +00002034 if (EnablePRE) {
Owen Anderson0c7f91c2008-09-03 23:06:07 +00002035 bool PREChanged = true;
2036 while (PREChanged) {
2037 PREChanged = performPRE(F);
Chris Lattnerb2412a82009-09-21 02:42:51 +00002038 Changed |= PREChanged;
Owen Anderson0c7f91c2008-09-03 23:06:07 +00002039 }
Owen Andersone98c54c2008-07-18 18:03:38 +00002040 }
Chris Lattnerae199312008-12-09 19:21:47 +00002041 // FIXME: Should perform GVN again after PRE does something. PRE can move
2042 // computations into blocks where they become fully redundant. Note that
2043 // we can't do this until PRE's critical edge splitting updates memdep.
2044 // Actually, when this happens, we should just fully integrate PRE into GVN.
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002045
2046 cleanupGlobalSets();
2047
Chris Lattnerb2412a82009-09-21 02:42:51 +00002048 return Changed;
Owen Anderson3e75a422007-08-14 18:04:11 +00002049}
2050
2051
Chris Lattnerb2412a82009-09-21 02:42:51 +00002052bool GVN::processBlock(BasicBlock *BB) {
Chris Lattnerae199312008-12-09 19:21:47 +00002053 // FIXME: Kill off toErase by doing erasing eagerly in a helper function (and
2054 // incrementing BI before processing an instruction).
Owen Andersonaf4240a2008-06-12 19:25:32 +00002055 SmallVector<Instruction*, 8> toErase;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002056 bool ChangedFunction = false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002057
Owen Andersonaf4240a2008-06-12 19:25:32 +00002058 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
2059 BI != BE;) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002060 ChangedFunction |= processInstruction(BI, toErase);
Owen Andersonaf4240a2008-06-12 19:25:32 +00002061 if (toErase.empty()) {
2062 ++BI;
2063 continue;
2064 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002065
Owen Andersonaf4240a2008-06-12 19:25:32 +00002066 // If we need some instructions deleted, do it now.
2067 NumGVNInstr += toErase.size();
Daniel Dunbara279bc32009-09-20 02:20:51 +00002068
Owen Andersonaf4240a2008-06-12 19:25:32 +00002069 // Avoid iterator invalidation.
2070 bool AtStart = BI == BB->begin();
2071 if (!AtStart)
2072 --BI;
2073
2074 for (SmallVector<Instruction*, 4>::iterator I = toErase.begin(),
Chris Lattner663e4412008-12-01 00:40:32 +00002075 E = toErase.end(); I != E; ++I) {
David Greenebf7f78e2010-01-05 01:27:17 +00002076 DEBUG(dbgs() << "GVN removed: " << **I << '\n');
Dan Gohman4ec01b22009-11-14 02:27:51 +00002077 if (MD) MD->removeInstruction(*I);
Owen Andersonaf4240a2008-06-12 19:25:32 +00002078 (*I)->eraseFromParent();
Bill Wendlingec40d502008-12-22 21:57:30 +00002079 DEBUG(verifyRemoved(*I));
Chris Lattner663e4412008-12-01 00:40:32 +00002080 }
Chris Lattnerae199312008-12-09 19:21:47 +00002081 toErase.clear();
Owen Andersonaf4240a2008-06-12 19:25:32 +00002082
2083 if (AtStart)
2084 BI = BB->begin();
2085 else
2086 ++BI;
Owen Andersonaf4240a2008-06-12 19:25:32 +00002087 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002088
Chris Lattnerb2412a82009-09-21 02:42:51 +00002089 return ChangedFunction;
Owen Andersonaf4240a2008-06-12 19:25:32 +00002090}
2091
Owen Andersonb2303722008-06-18 21:41:49 +00002092/// performPRE - Perform a purely local form of PRE that looks for diamond
2093/// control flow patterns and attempts to perform simple PRE at the join point.
Chris Lattnerfb6e7012009-10-31 22:11:15 +00002094bool GVN::performPRE(Function &F) {
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002095 bool Changed = false;
Chris Lattner09713792008-12-01 07:29:03 +00002096 DenseMap<BasicBlock*, Value*> predMap;
Owen Andersonb2303722008-06-18 21:41:49 +00002097 for (df_iterator<BasicBlock*> DI = df_begin(&F.getEntryBlock()),
2098 DE = df_end(&F.getEntryBlock()); DI != DE; ++DI) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002099 BasicBlock *CurrentBlock = *DI;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002100
Owen Andersonb2303722008-06-18 21:41:49 +00002101 // Nothing to PRE in the entry block.
2102 if (CurrentBlock == &F.getEntryBlock()) continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002103
Owen Andersonb2303722008-06-18 21:41:49 +00002104 for (BasicBlock::iterator BI = CurrentBlock->begin(),
2105 BE = CurrentBlock->end(); BI != BE; ) {
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002106 Instruction *CurInst = BI++;
Duncan Sands7af1c782009-05-06 06:49:50 +00002107
Victor Hernandez7b929da2009-10-23 21:09:37 +00002108 if (isa<AllocaInst>(CurInst) ||
Victor Hernandez83d63912009-09-18 22:35:49 +00002109 isa<TerminatorInst>(CurInst) || isa<PHINode>(CurInst) ||
Devang Patel9674d152009-10-14 17:29:00 +00002110 CurInst->getType()->isVoidTy() ||
Duncan Sands7af1c782009-05-06 06:49:50 +00002111 CurInst->mayReadFromMemory() || CurInst->mayHaveSideEffects() ||
John Criswell090c0a22009-03-10 15:04:53 +00002112 isa<DbgInfoIntrinsic>(CurInst))
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002113 continue;
Duncan Sands7af1c782009-05-06 06:49:50 +00002114
Chris Lattnerb2412a82009-09-21 02:42:51 +00002115 uint32_t ValNo = VN.lookup(CurInst);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002116
Owen Andersonb2303722008-06-18 21:41:49 +00002117 // Look for the predecessors for PRE opportunities. We're
2118 // only trying to solve the basic diamond case, where
2119 // a value is computed in the successor and one predecessor,
2120 // but not the other. We also explicitly disallow cases
2121 // where the successor is its own predecessor, because they're
2122 // more complicated to get right.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002123 unsigned NumWith = 0;
2124 unsigned NumWithout = 0;
2125 BasicBlock *PREPred = 0;
Chris Lattner09713792008-12-01 07:29:03 +00002126 predMap.clear();
2127
Owen Andersonb2303722008-06-18 21:41:49 +00002128 for (pred_iterator PI = pred_begin(CurrentBlock),
2129 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
2130 // We're not interested in PRE where the block is its
Bob Wilsone7b635f2010-02-03 00:33:21 +00002131 // own predecessor, or in blocks with predecessors
Owen Anderson6fafe842008-06-20 01:15:47 +00002132 // that are not reachable.
2133 if (*PI == CurrentBlock) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002134 NumWithout = 2;
Owen Anderson6fafe842008-06-20 01:15:47 +00002135 break;
2136 } else if (!localAvail.count(*PI)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002137 NumWithout = 2;
Owen Anderson6fafe842008-06-20 01:15:47 +00002138 break;
2139 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002140
2141 DenseMap<uint32_t, Value*>::iterator predV =
Chris Lattnerb2412a82009-09-21 02:42:51 +00002142 localAvail[*PI]->table.find(ValNo);
Owen Anderson6fafe842008-06-20 01:15:47 +00002143 if (predV == localAvail[*PI]->table.end()) {
Owen Andersonb2303722008-06-18 21:41:49 +00002144 PREPred = *PI;
Dan Gohmanfe601042010-06-22 15:08:57 +00002145 ++NumWithout;
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002146 } else if (predV->second == CurInst) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002147 NumWithout = 2;
Owen Andersonb2303722008-06-18 21:41:49 +00002148 } else {
Owen Anderson6fafe842008-06-20 01:15:47 +00002149 predMap[*PI] = predV->second;
Dan Gohmanfe601042010-06-22 15:08:57 +00002150 ++NumWith;
Owen Andersonb2303722008-06-18 21:41:49 +00002151 }
2152 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002153
Owen Andersonb2303722008-06-18 21:41:49 +00002154 // Don't do PRE when it might increase code size, i.e. when
2155 // we would need to insert instructions in more than one pred.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002156 if (NumWithout != 1 || NumWith == 0)
Owen Andersonb2303722008-06-18 21:41:49 +00002157 continue;
Chris Lattnerfb6e7012009-10-31 22:11:15 +00002158
2159 // Don't do PRE across indirect branch.
2160 if (isa<IndirectBrInst>(PREPred->getTerminator()))
2161 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002162
Owen Anderson5c274ee2008-06-19 19:54:19 +00002163 // We can't do PRE safely on a critical edge, so instead we schedule
2164 // the edge to be split and perform the PRE the next time we iterate
2165 // on the function.
Bob Wilsonae23daf2010-02-16 21:06:42 +00002166 unsigned SuccNum = GetSuccessorNumber(PREPred, CurrentBlock);
Chris Lattnerb2412a82009-09-21 02:42:51 +00002167 if (isCriticalEdge(PREPred->getTerminator(), SuccNum)) {
2168 toSplit.push_back(std::make_pair(PREPred->getTerminator(), SuccNum));
Owen Anderson5c274ee2008-06-19 19:54:19 +00002169 continue;
2170 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002171
Bob Wilsone7b635f2010-02-03 00:33:21 +00002172 // Instantiate the expression in the predecessor that lacked it.
Owen Andersonb2303722008-06-18 21:41:49 +00002173 // Because we are going top-down through the block, all value numbers
2174 // will be available in the predecessor by the time we need them. Any
Bob Wilsone7b635f2010-02-03 00:33:21 +00002175 // that weren't originally present will have been instantiated earlier
Owen Andersonb2303722008-06-18 21:41:49 +00002176 // in this loop.
Nick Lewycky67760642009-09-27 07:38:41 +00002177 Instruction *PREInstr = CurInst->clone();
Owen Andersonb2303722008-06-18 21:41:49 +00002178 bool success = true;
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002179 for (unsigned i = 0, e = CurInst->getNumOperands(); i != e; ++i) {
2180 Value *Op = PREInstr->getOperand(i);
2181 if (isa<Argument>(Op) || isa<Constant>(Op) || isa<GlobalValue>(Op))
2182 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002183
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002184 if (Value *V = lookupNumber(PREPred, VN.lookup(Op))) {
2185 PREInstr->setOperand(i, V);
2186 } else {
2187 success = false;
2188 break;
Owen Andersonc45996b2008-07-11 20:05:13 +00002189 }
Owen Andersonb2303722008-06-18 21:41:49 +00002190 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002191
Owen Andersonb2303722008-06-18 21:41:49 +00002192 // Fail out if we encounter an operand that is not available in
Daniel Dunbara279bc32009-09-20 02:20:51 +00002193 // the PRE predecessor. This is typically because of loads which
Owen Andersonb2303722008-06-18 21:41:49 +00002194 // are not value numbered precisely.
2195 if (!success) {
2196 delete PREInstr;
Bill Wendling70ded192008-12-22 22:14:07 +00002197 DEBUG(verifyRemoved(PREInstr));
Owen Andersonb2303722008-06-18 21:41:49 +00002198 continue;
2199 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002200
Owen Andersonb2303722008-06-18 21:41:49 +00002201 PREInstr->insertBefore(PREPred->getTerminator());
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002202 PREInstr->setName(CurInst->getName() + ".pre");
Owen Anderson6fafe842008-06-20 01:15:47 +00002203 predMap[PREPred] = PREInstr;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002204 VN.add(PREInstr, ValNo);
Dan Gohmanfe601042010-06-22 15:08:57 +00002205 ++NumGVNPRE;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002206
Owen Andersonb2303722008-06-18 21:41:49 +00002207 // Update the availability map to include the new instruction.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002208 localAvail[PREPred]->table.insert(std::make_pair(ValNo, PREInstr));
Daniel Dunbara279bc32009-09-20 02:20:51 +00002209
Owen Andersonb2303722008-06-18 21:41:49 +00002210 // Create a PHI to make the value available in this block.
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002211 PHINode* Phi = PHINode::Create(CurInst->getType(),
2212 CurInst->getName() + ".pre-phi",
Owen Andersonb2303722008-06-18 21:41:49 +00002213 CurrentBlock->begin());
2214 for (pred_iterator PI = pred_begin(CurrentBlock),
2215 PE = pred_end(CurrentBlock); PI != PE; ++PI)
Owen Anderson6fafe842008-06-20 01:15:47 +00002216 Phi->addIncoming(predMap[*PI], *PI);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002217
Chris Lattnerb2412a82009-09-21 02:42:51 +00002218 VN.add(Phi, ValNo);
2219 localAvail[CurrentBlock]->table[ValNo] = Phi;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002220
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002221 CurInst->replaceAllUsesWith(Phi);
Duncan Sands1df98592010-02-16 11:11:14 +00002222 if (MD && Phi->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00002223 MD->invalidateCachedPointerInfo(Phi);
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002224 VN.erase(CurInst);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002225
David Greenebf7f78e2010-01-05 01:27:17 +00002226 DEBUG(dbgs() << "GVN PRE removed: " << *CurInst << '\n');
Dan Gohman4ec01b22009-11-14 02:27:51 +00002227 if (MD) MD->removeInstruction(CurInst);
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002228 CurInst->eraseFromParent();
Bill Wendlingec40d502008-12-22 21:57:30 +00002229 DEBUG(verifyRemoved(CurInst));
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002230 Changed = true;
Owen Andersonb2303722008-06-18 21:41:49 +00002231 }
2232 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002233
Bob Wilson484d4a32010-02-16 19:51:59 +00002234 if (splitCriticalEdges())
2235 Changed = true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002236
Bob Wilson484d4a32010-02-16 19:51:59 +00002237 return Changed;
2238}
2239
2240/// splitCriticalEdges - Split critical edges found during the previous
2241/// iteration that may enable further optimization.
2242bool GVN::splitCriticalEdges() {
2243 if (toSplit.empty())
2244 return false;
2245 do {
2246 std::pair<TerminatorInst*, unsigned> Edge = toSplit.pop_back_val();
2247 SplitCriticalEdge(Edge.first, Edge.second, this);
2248 } while (!toSplit.empty());
Evan Cheng19d417c2010-03-01 22:23:12 +00002249 if (MD) MD->invalidateCachedPredecessors();
Bob Wilson484d4a32010-02-16 19:51:59 +00002250 return true;
Owen Andersonb2303722008-06-18 21:41:49 +00002251}
2252
Bill Wendling30788b82008-12-22 22:32:22 +00002253/// iterateOnFunction - Executes one iteration of GVN
Owen Anderson3e75a422007-08-14 18:04:11 +00002254bool GVN::iterateOnFunction(Function &F) {
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002255 cleanupGlobalSets();
Chris Lattner2e607012008-03-21 21:33:23 +00002256
Owen Andersone8a290f2009-04-01 23:53:49 +00002257 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2258 DE = df_end(DT->getRootNode()); DI != DE; ++DI) {
2259 if (DI->getIDom())
2260 localAvail[DI->getBlock()] =
2261 new ValueNumberScope(localAvail[DI->getIDom()->getBlock()]);
2262 else
2263 localAvail[DI->getBlock()] = new ValueNumberScope(0);
2264 }
2265
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002266 // Top-down walk of the dominator tree
Chris Lattnerb2412a82009-09-21 02:42:51 +00002267 bool Changed = false;
Owen Andersonc34d1122008-12-15 03:52:17 +00002268#if 0
2269 // Needed for value numbering with phi construction to work.
Owen Anderson255dafc2008-12-15 02:03:00 +00002270 ReversePostOrderTraversal<Function*> RPOT(&F);
2271 for (ReversePostOrderTraversal<Function*>::rpo_iterator RI = RPOT.begin(),
2272 RE = RPOT.end(); RI != RE; ++RI)
Chris Lattnerb2412a82009-09-21 02:42:51 +00002273 Changed |= processBlock(*RI);
Owen Andersonc34d1122008-12-15 03:52:17 +00002274#else
2275 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2276 DE = df_end(DT->getRootNode()); DI != DE; ++DI)
Chris Lattnerb2412a82009-09-21 02:42:51 +00002277 Changed |= processBlock(DI->getBlock());
Owen Andersonc34d1122008-12-15 03:52:17 +00002278#endif
2279
Chris Lattnerb2412a82009-09-21 02:42:51 +00002280 return Changed;
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002281}
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002282
2283void GVN::cleanupGlobalSets() {
2284 VN.clear();
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002285
2286 for (DenseMap<BasicBlock*, ValueNumberScope*>::iterator
2287 I = localAvail.begin(), E = localAvail.end(); I != E; ++I)
2288 delete I->second;
2289 localAvail.clear();
2290}
Bill Wendling246dbbb2008-12-22 21:36:08 +00002291
2292/// verifyRemoved - Verify that the specified instruction does not occur in our
2293/// internal data structures.
Bill Wendling6d463f22008-12-22 22:28:56 +00002294void GVN::verifyRemoved(const Instruction *Inst) const {
2295 VN.verifyRemoved(Inst);
Bill Wendling70ded192008-12-22 22:14:07 +00002296
Bill Wendling6d463f22008-12-22 22:28:56 +00002297 // Walk through the value number scope to make sure the instruction isn't
2298 // ferreted away in it.
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +00002299 for (DenseMap<BasicBlock*, ValueNumberScope*>::const_iterator
Bill Wendling6d463f22008-12-22 22:28:56 +00002300 I = localAvail.begin(), E = localAvail.end(); I != E; ++I) {
2301 const ValueNumberScope *VNS = I->second;
2302
2303 while (VNS) {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +00002304 for (DenseMap<uint32_t, Value*>::const_iterator
Bill Wendling6d463f22008-12-22 22:28:56 +00002305 II = VNS->table.begin(), IE = VNS->table.end(); II != IE; ++II) {
2306 assert(II->second != Inst && "Inst still in value numbering scope!");
2307 }
2308
2309 VNS = VNS->parent;
Bill Wendling70ded192008-12-22 22:14:07 +00002310 }
2311 }
Bill Wendling246dbbb2008-12-22 21:36:08 +00002312}