blob: 927a3ac8a233861f0ec7202e2ca060efe9e39c1c [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"
Owen Anderson45537912007-07-26 18:26:51 +000023#include "llvm/Function.h"
Devang Patelc64bc162009-03-06 02:59:27 +000024#include "llvm/IntrinsicInst.h"
Owen Andersond672ecb2009-07-03 00:17:18 +000025#include "llvm/LLVMContext.h"
Chris Lattnereed919b2009-09-21 05:57:11 +000026#include "llvm/Operator.h"
Owen Anderson45537912007-07-26 18:26:51 +000027#include "llvm/Value.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000028#include "llvm/ADT/DenseMap.h"
29#include "llvm/ADT/DepthFirstIterator.h"
Owen Anderson255dafc2008-12-15 02:03:00 +000030#include "llvm/ADT/PostOrderIterator.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000031#include "llvm/ADT/SmallPtrSet.h"
32#include "llvm/ADT/SmallVector.h"
33#include "llvm/ADT/Statistic.h"
Owen Andersonb388ca92007-10-18 19:39:33 +000034#include "llvm/Analysis/AliasAnalysis.h"
Chris Lattnerbc9a28d2009-12-06 05:29:56 +000035#include "llvm/Analysis/ConstantFolding.h"
36#include "llvm/Analysis/Dominators.h"
Victor Hernandezf006b182009-10-27 20:05:49 +000037#include "llvm/Analysis/MemoryBuiltins.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000038#include "llvm/Analysis/MemoryDependenceAnalysis.h"
Chris Lattner05e15f82009-12-09 01:59:31 +000039#include "llvm/Analysis/PHITransAddr.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000040#include "llvm/Support/CFG.h"
Owen Andersonaa0b6342008-06-19 19:57:25 +000041#include "llvm/Support/CommandLine.h"
Chris Lattner9f8a6a72008-03-29 04:36:18 +000042#include "llvm/Support/Debug.h"
Torok Edwinc25e7582009-07-11 20:10:48 +000043#include "llvm/Support/ErrorHandling.h"
Chris Lattnereed919b2009-09-21 05:57:11 +000044#include "llvm/Support/GetElementPtrTypeIterator.h"
Chris Lattnerfaf815b2009-12-06 01:57:02 +000045#include "llvm/Support/IRBuilder.h"
Daniel Dunbarce63ffb2009-07-25 00:23:56 +000046#include "llvm/Support/raw_ostream.h"
Chris Lattnerbb6495c2009-09-20 19:03:47 +000047#include "llvm/Target/TargetData.h"
Owen Anderson5c274ee2008-06-19 19:54:19 +000048#include "llvm/Transforms/Utils/BasicBlockUtils.h"
Dale Johannesen42c3f552009-06-17 20:48:23 +000049#include "llvm/Transforms/Utils/Local.h"
Chris Lattnera09fbf02009-10-10 23:50:30 +000050#include "llvm/Transforms/Utils/SSAUpdater.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000051using namespace llvm;
52
Bill Wendling70ded192008-12-22 22:14:07 +000053STATISTIC(NumGVNInstr, "Number of instructions deleted");
54STATISTIC(NumGVNLoad, "Number of loads deleted");
55STATISTIC(NumGVNPRE, "Number of instructions PRE'd");
Owen Anderson961edc82008-07-15 16:28:06 +000056STATISTIC(NumGVNBlocks, "Number of blocks merged");
Bill Wendling70ded192008-12-22 22:14:07 +000057STATISTIC(NumPRELoad, "Number of loads PRE'd");
Chris Lattnerd27290d2008-03-22 04:13:49 +000058
Evan Cheng88d11c02008-06-20 01:01:07 +000059static cl::opt<bool> EnablePRE("enable-pre",
Owen Andersonc2b856e2008-07-17 19:41:00 +000060 cl::init(true), cl::Hidden);
Dan Gohmanc915c952009-06-15 18:30:15 +000061static cl::opt<bool> EnableLoadPRE("enable-load-pre", cl::init(true));
Owen Andersonaa0b6342008-06-19 19:57:25 +000062
Owen Anderson1ad2cb72007-07-24 17:55:58 +000063//===----------------------------------------------------------------------===//
64// ValueTable Class
65//===----------------------------------------------------------------------===//
66
67/// This class holds the mapping between values and value numbers. It is used
68/// as an efficient mechanism to determine the expression-wise equivalence of
69/// two values.
70namespace {
Chris Lattner3e8b6632009-09-02 06:11:42 +000071 struct Expression {
Dan Gohmanae3a0be2009-06-04 22:49:04 +000072 enum ExpressionOpcode { ADD, FADD, SUB, FSUB, MUL, FMUL,
73 UDIV, SDIV, FDIV, UREM, SREM,
Daniel Dunbara279bc32009-09-20 02:20:51 +000074 FREM, SHL, LSHR, ASHR, AND, OR, XOR, ICMPEQ,
75 ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
76 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
77 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
78 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
Owen Anderson1ad2cb72007-07-24 17:55:58 +000079 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
80 SHUFFLE, SELECT, TRUNC, ZEXT, SEXT, FPTOUI,
Daniel Dunbara279bc32009-09-20 02:20:51 +000081 FPTOSI, UITOFP, SITOFP, FPTRUNC, FPEXT,
Owen Anderson3b3f58c2008-05-13 08:17:22 +000082 PTRTOINT, INTTOPTR, BITCAST, GEP, CALL, CONSTANT,
Owen Andersond41ed4e2009-10-19 22:14:22 +000083 INSERTVALUE, EXTRACTVALUE, EMPTY, TOMBSTONE };
Owen Anderson1ad2cb72007-07-24 17:55:58 +000084
85 ExpressionOpcode opcode;
86 const Type* type;
Owen Anderson1ad2cb72007-07-24 17:55:58 +000087 SmallVector<uint32_t, 4> varargs;
Chris Lattnerb2412a82009-09-21 02:42:51 +000088 Value *function;
Daniel Dunbara279bc32009-09-20 02:20:51 +000089
Owen Anderson1ad2cb72007-07-24 17:55:58 +000090 Expression() { }
91 Expression(ExpressionOpcode o) : opcode(o) { }
Daniel Dunbara279bc32009-09-20 02:20:51 +000092
Owen Anderson1ad2cb72007-07-24 17:55:58 +000093 bool operator==(const Expression &other) const {
94 if (opcode != other.opcode)
95 return false;
96 else if (opcode == EMPTY || opcode == TOMBSTONE)
97 return true;
98 else if (type != other.type)
99 return false;
Owen Andersonb388ca92007-10-18 19:39:33 +0000100 else if (function != other.function)
101 return false;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000102 else {
103 if (varargs.size() != other.varargs.size())
104 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000105
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000106 for (size_t i = 0; i < varargs.size(); ++i)
107 if (varargs[i] != other.varargs[i])
108 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000109
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000110 return true;
111 }
112 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000113
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000114 bool operator!=(const Expression &other) const {
Bill Wendling75f02ee2008-12-22 22:16:31 +0000115 return !(*this == other);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000116 }
117 };
Daniel Dunbara279bc32009-09-20 02:20:51 +0000118
Chris Lattner3e8b6632009-09-02 06:11:42 +0000119 class ValueTable {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000120 private:
121 DenseMap<Value*, uint32_t> valueNumbering;
122 DenseMap<Expression, uint32_t> expressionNumbering;
Owen Andersona472c4a2008-05-12 20:15:55 +0000123 AliasAnalysis* AA;
124 MemoryDependenceAnalysis* MD;
125 DominatorTree* DT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000126
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000127 uint32_t nextValueNumber;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000128
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000129 Expression::ExpressionOpcode getOpcode(BinaryOperator* BO);
130 Expression::ExpressionOpcode getOpcode(CmpInst* C);
131 Expression::ExpressionOpcode getOpcode(CastInst* C);
132 Expression create_expression(BinaryOperator* BO);
133 Expression create_expression(CmpInst* C);
134 Expression create_expression(ShuffleVectorInst* V);
135 Expression create_expression(ExtractElementInst* C);
136 Expression create_expression(InsertElementInst* V);
137 Expression create_expression(SelectInst* V);
138 Expression create_expression(CastInst* C);
139 Expression create_expression(GetElementPtrInst* G);
Owen Andersonb388ca92007-10-18 19:39:33 +0000140 Expression create_expression(CallInst* C);
Owen Anderson3b3f58c2008-05-13 08:17:22 +0000141 Expression create_expression(Constant* C);
Owen Andersond41ed4e2009-10-19 22:14:22 +0000142 Expression create_expression(ExtractValueInst* C);
143 Expression create_expression(InsertValueInst* C);
144
145 uint32_t lookup_or_add_call(CallInst* C);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000146 public:
Dan Gohmane63c4a22009-04-01 16:37:47 +0000147 ValueTable() : nextValueNumber(1) { }
Chris Lattnerb2412a82009-09-21 02:42:51 +0000148 uint32_t lookup_or_add(Value *V);
149 uint32_t lookup(Value *V) const;
150 void add(Value *V, uint32_t num);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000151 void clear();
Chris Lattnerb2412a82009-09-21 02:42:51 +0000152 void erase(Value *v);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000153 unsigned size();
Owen Andersona472c4a2008-05-12 20:15:55 +0000154 void setAliasAnalysis(AliasAnalysis* A) { AA = A; }
Chris Lattner663e4412008-12-01 00:40:32 +0000155 AliasAnalysis *getAliasAnalysis() const { return AA; }
Owen Andersona472c4a2008-05-12 20:15:55 +0000156 void setMemDep(MemoryDependenceAnalysis* M) { MD = M; }
157 void setDomTree(DominatorTree* D) { DT = D; }
Owen Anderson0ae33ef2008-07-03 17:44:33 +0000158 uint32_t getNextUnusedValueNumber() { return nextValueNumber; }
Bill Wendling246dbbb2008-12-22 21:36:08 +0000159 void verifyRemoved(const Value *) const;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000160 };
161}
162
163namespace llvm {
Chris Lattner76c1b972007-09-17 18:34:04 +0000164template <> struct DenseMapInfo<Expression> {
Owen Anderson830db6a2007-08-02 18:16:06 +0000165 static inline Expression getEmptyKey() {
166 return Expression(Expression::EMPTY);
167 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000168
Owen Anderson830db6a2007-08-02 18:16:06 +0000169 static inline Expression getTombstoneKey() {
170 return Expression(Expression::TOMBSTONE);
171 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000172
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000173 static unsigned getHashValue(const Expression e) {
174 unsigned hash = e.opcode;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000175
Anton Korobeynikov07e6e562008-02-20 11:26:25 +0000176 hash = ((unsigned)((uintptr_t)e.type >> 4) ^
Owen Andersond41ed4e2009-10-19 22:14:22 +0000177 (unsigned)((uintptr_t)e.type >> 9));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000178
Owen Anderson830db6a2007-08-02 18:16:06 +0000179 for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
180 E = e.varargs.end(); I != E; ++I)
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000181 hash = *I + hash * 37;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000182
Anton Korobeynikov07e6e562008-02-20 11:26:25 +0000183 hash = ((unsigned)((uintptr_t)e.function >> 4) ^
184 (unsigned)((uintptr_t)e.function >> 9)) +
185 hash * 37;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000186
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000187 return hash;
188 }
Chris Lattner76c1b972007-09-17 18:34:04 +0000189 static bool isEqual(const Expression &LHS, const Expression &RHS) {
190 return LHS == RHS;
191 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000192};
Chris Lattner4bbf4ee2009-12-15 07:26:43 +0000193
194template <>
195struct isPodLike<Expression> { static const bool value = true; };
196
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000197}
198
199//===----------------------------------------------------------------------===//
200// ValueTable Internal Functions
201//===----------------------------------------------------------------------===//
Chris Lattner88365bb2008-03-21 21:14:38 +0000202Expression::ExpressionOpcode ValueTable::getOpcode(BinaryOperator* BO) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000203 switch(BO->getOpcode()) {
Chris Lattner88365bb2008-03-21 21:14:38 +0000204 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000205 llvm_unreachable("Binary operator with unknown opcode?");
Chris Lattner88365bb2008-03-21 21:14:38 +0000206 case Instruction::Add: return Expression::ADD;
Dan Gohmanae3a0be2009-06-04 22:49:04 +0000207 case Instruction::FAdd: return Expression::FADD;
Chris Lattner88365bb2008-03-21 21:14:38 +0000208 case Instruction::Sub: return Expression::SUB;
Dan Gohmanae3a0be2009-06-04 22:49:04 +0000209 case Instruction::FSub: return Expression::FSUB;
Chris Lattner88365bb2008-03-21 21:14:38 +0000210 case Instruction::Mul: return Expression::MUL;
Dan Gohmanae3a0be2009-06-04 22:49:04 +0000211 case Instruction::FMul: return Expression::FMUL;
Chris Lattner88365bb2008-03-21 21:14:38 +0000212 case Instruction::UDiv: return Expression::UDIV;
213 case Instruction::SDiv: return Expression::SDIV;
214 case Instruction::FDiv: return Expression::FDIV;
215 case Instruction::URem: return Expression::UREM;
216 case Instruction::SRem: return Expression::SREM;
217 case Instruction::FRem: return Expression::FREM;
218 case Instruction::Shl: return Expression::SHL;
219 case Instruction::LShr: return Expression::LSHR;
220 case Instruction::AShr: return Expression::ASHR;
221 case Instruction::And: return Expression::AND;
222 case Instruction::Or: return Expression::OR;
223 case Instruction::Xor: return Expression::XOR;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000224 }
225}
226
227Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000228 if (isa<ICmpInst>(C)) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000229 switch (C->getPredicate()) {
Chris Lattner88365bb2008-03-21 21:14:38 +0000230 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000231 llvm_unreachable("Comparison with unknown predicate?");
Chris Lattner88365bb2008-03-21 21:14:38 +0000232 case ICmpInst::ICMP_EQ: return Expression::ICMPEQ;
233 case ICmpInst::ICMP_NE: return Expression::ICMPNE;
234 case ICmpInst::ICMP_UGT: return Expression::ICMPUGT;
235 case ICmpInst::ICMP_UGE: return Expression::ICMPUGE;
236 case ICmpInst::ICMP_ULT: return Expression::ICMPULT;
237 case ICmpInst::ICMP_ULE: return Expression::ICMPULE;
238 case ICmpInst::ICMP_SGT: return Expression::ICMPSGT;
239 case ICmpInst::ICMP_SGE: return Expression::ICMPSGE;
240 case ICmpInst::ICMP_SLT: return Expression::ICMPSLT;
241 case ICmpInst::ICMP_SLE: return Expression::ICMPSLE;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000242 }
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000243 } else {
244 switch (C->getPredicate()) {
245 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000246 llvm_unreachable("Comparison with unknown predicate?");
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000247 case FCmpInst::FCMP_OEQ: return Expression::FCMPOEQ;
248 case FCmpInst::FCMP_OGT: return Expression::FCMPOGT;
249 case FCmpInst::FCMP_OGE: return Expression::FCMPOGE;
250 case FCmpInst::FCMP_OLT: return Expression::FCMPOLT;
251 case FCmpInst::FCMP_OLE: return Expression::FCMPOLE;
252 case FCmpInst::FCMP_ONE: return Expression::FCMPONE;
253 case FCmpInst::FCMP_ORD: return Expression::FCMPORD;
254 case FCmpInst::FCMP_UNO: return Expression::FCMPUNO;
255 case FCmpInst::FCMP_UEQ: return Expression::FCMPUEQ;
256 case FCmpInst::FCMP_UGT: return Expression::FCMPUGT;
257 case FCmpInst::FCMP_UGE: return Expression::FCMPUGE;
258 case FCmpInst::FCMP_ULT: return Expression::FCMPULT;
259 case FCmpInst::FCMP_ULE: return Expression::FCMPULE;
260 case FCmpInst::FCMP_UNE: return Expression::FCMPUNE;
261 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000262 }
263}
264
Chris Lattner88365bb2008-03-21 21:14:38 +0000265Expression::ExpressionOpcode ValueTable::getOpcode(CastInst* C) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000266 switch(C->getOpcode()) {
Chris Lattner88365bb2008-03-21 21:14:38 +0000267 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000268 llvm_unreachable("Cast operator with unknown opcode?");
Chris Lattner88365bb2008-03-21 21:14:38 +0000269 case Instruction::Trunc: return Expression::TRUNC;
270 case Instruction::ZExt: return Expression::ZEXT;
271 case Instruction::SExt: return Expression::SEXT;
272 case Instruction::FPToUI: return Expression::FPTOUI;
273 case Instruction::FPToSI: return Expression::FPTOSI;
274 case Instruction::UIToFP: return Expression::UITOFP;
275 case Instruction::SIToFP: return Expression::SITOFP;
276 case Instruction::FPTrunc: return Expression::FPTRUNC;
277 case Instruction::FPExt: return Expression::FPEXT;
278 case Instruction::PtrToInt: return Expression::PTRTOINT;
279 case Instruction::IntToPtr: return Expression::INTTOPTR;
280 case Instruction::BitCast: return Expression::BITCAST;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000281 }
282}
283
Owen Andersonb388ca92007-10-18 19:39:33 +0000284Expression ValueTable::create_expression(CallInst* C) {
285 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000286
Owen Andersonb388ca92007-10-18 19:39:33 +0000287 e.type = C->getType();
Owen Andersonb388ca92007-10-18 19:39:33 +0000288 e.function = C->getCalledFunction();
289 e.opcode = Expression::CALL;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000290
Owen Andersonb388ca92007-10-18 19:39:33 +0000291 for (CallInst::op_iterator I = C->op_begin()+1, E = C->op_end();
292 I != E; ++I)
Owen Anderson8f46c782008-04-11 05:11:49 +0000293 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000294
Owen Andersonb388ca92007-10-18 19:39:33 +0000295 return e;
296}
297
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000298Expression ValueTable::create_expression(BinaryOperator* BO) {
299 Expression e;
Owen Andersond41ed4e2009-10-19 22:14:22 +0000300 e.varargs.push_back(lookup_or_add(BO->getOperand(0)));
301 e.varargs.push_back(lookup_or_add(BO->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000302 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000303 e.type = BO->getType();
304 e.opcode = getOpcode(BO);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000305
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000306 return e;
307}
308
309Expression ValueTable::create_expression(CmpInst* C) {
310 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000311
Owen Andersond41ed4e2009-10-19 22:14:22 +0000312 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
313 e.varargs.push_back(lookup_or_add(C->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000314 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000315 e.type = C->getType();
316 e.opcode = getOpcode(C);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000317
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000318 return e;
319}
320
321Expression ValueTable::create_expression(CastInst* C) {
322 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000323
Owen Andersond41ed4e2009-10-19 22:14:22 +0000324 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000325 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000326 e.type = C->getType();
327 e.opcode = getOpcode(C);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000328
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000329 return e;
330}
331
332Expression ValueTable::create_expression(ShuffleVectorInst* S) {
333 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000334
Owen Andersond41ed4e2009-10-19 22:14:22 +0000335 e.varargs.push_back(lookup_or_add(S->getOperand(0)));
336 e.varargs.push_back(lookup_or_add(S->getOperand(1)));
337 e.varargs.push_back(lookup_or_add(S->getOperand(2)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000338 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000339 e.type = S->getType();
340 e.opcode = Expression::SHUFFLE;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000341
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000342 return e;
343}
344
345Expression ValueTable::create_expression(ExtractElementInst* E) {
346 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000347
Owen Andersond41ed4e2009-10-19 22:14:22 +0000348 e.varargs.push_back(lookup_or_add(E->getOperand(0)));
349 e.varargs.push_back(lookup_or_add(E->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000350 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000351 e.type = E->getType();
352 e.opcode = Expression::EXTRACT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000353
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000354 return e;
355}
356
357Expression ValueTable::create_expression(InsertElementInst* I) {
358 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000359
Owen Andersond41ed4e2009-10-19 22:14:22 +0000360 e.varargs.push_back(lookup_or_add(I->getOperand(0)));
361 e.varargs.push_back(lookup_or_add(I->getOperand(1)));
362 e.varargs.push_back(lookup_or_add(I->getOperand(2)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000363 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000364 e.type = I->getType();
365 e.opcode = Expression::INSERT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000366
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000367 return e;
368}
369
370Expression ValueTable::create_expression(SelectInst* I) {
371 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000372
Owen Andersond41ed4e2009-10-19 22:14:22 +0000373 e.varargs.push_back(lookup_or_add(I->getCondition()));
374 e.varargs.push_back(lookup_or_add(I->getTrueValue()));
375 e.varargs.push_back(lookup_or_add(I->getFalseValue()));
Owen Andersonb388ca92007-10-18 19:39:33 +0000376 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000377 e.type = I->getType();
378 e.opcode = Expression::SELECT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000379
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000380 return e;
381}
382
383Expression ValueTable::create_expression(GetElementPtrInst* G) {
384 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000385
Owen Andersond41ed4e2009-10-19 22:14:22 +0000386 e.varargs.push_back(lookup_or_add(G->getPointerOperand()));
Owen Andersonb388ca92007-10-18 19:39:33 +0000387 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000388 e.type = G->getType();
389 e.opcode = Expression::GEP;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000390
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000391 for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
392 I != E; ++I)
Owen Anderson8f46c782008-04-11 05:11:49 +0000393 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000394
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000395 return e;
396}
397
Owen Andersond41ed4e2009-10-19 22:14:22 +0000398Expression ValueTable::create_expression(ExtractValueInst* E) {
399 Expression e;
400
401 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
402 for (ExtractValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
403 II != IE; ++II)
404 e.varargs.push_back(*II);
405 e.function = 0;
406 e.type = E->getType();
407 e.opcode = Expression::EXTRACTVALUE;
408
409 return e;
410}
411
412Expression ValueTable::create_expression(InsertValueInst* E) {
413 Expression e;
414
415 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
416 e.varargs.push_back(lookup_or_add(E->getInsertedValueOperand()));
417 for (InsertValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
418 II != IE; ++II)
419 e.varargs.push_back(*II);
420 e.function = 0;
421 e.type = E->getType();
422 e.opcode = Expression::INSERTVALUE;
423
424 return e;
425}
426
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000427//===----------------------------------------------------------------------===//
428// ValueTable External Functions
429//===----------------------------------------------------------------------===//
430
Owen Andersonb2303722008-06-18 21:41:49 +0000431/// add - Insert a value into the table with a specified value number.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000432void ValueTable::add(Value *V, uint32_t num) {
Owen Andersonb2303722008-06-18 21:41:49 +0000433 valueNumbering.insert(std::make_pair(V, num));
434}
435
Owen Andersond41ed4e2009-10-19 22:14:22 +0000436uint32_t ValueTable::lookup_or_add_call(CallInst* C) {
437 if (AA->doesNotAccessMemory(C)) {
438 Expression exp = create_expression(C);
439 uint32_t& e = expressionNumbering[exp];
440 if (!e) e = nextValueNumber++;
441 valueNumbering[C] = e;
442 return e;
443 } else if (AA->onlyReadsMemory(C)) {
444 Expression exp = create_expression(C);
445 uint32_t& e = expressionNumbering[exp];
446 if (!e) {
447 e = nextValueNumber++;
448 valueNumbering[C] = e;
449 return e;
450 }
Dan Gohman4ec01b22009-11-14 02:27:51 +0000451 if (!MD) {
452 e = nextValueNumber++;
453 valueNumbering[C] = e;
454 return e;
455 }
Owen Andersond41ed4e2009-10-19 22:14:22 +0000456
457 MemDepResult local_dep = MD->getDependency(C);
458
459 if (!local_dep.isDef() && !local_dep.isNonLocal()) {
460 valueNumbering[C] = nextValueNumber;
461 return nextValueNumber++;
462 }
463
464 if (local_dep.isDef()) {
465 CallInst* local_cdep = cast<CallInst>(local_dep.getInst());
466
467 if (local_cdep->getNumOperands() != C->getNumOperands()) {
468 valueNumbering[C] = nextValueNumber;
469 return nextValueNumber++;
470 }
471
472 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
473 uint32_t c_vn = lookup_or_add(C->getOperand(i));
474 uint32_t cd_vn = lookup_or_add(local_cdep->getOperand(i));
475 if (c_vn != cd_vn) {
476 valueNumbering[C] = nextValueNumber;
477 return nextValueNumber++;
478 }
479 }
480
481 uint32_t v = lookup_or_add(local_cdep);
482 valueNumbering[C] = v;
483 return v;
484 }
485
486 // Non-local case.
487 const MemoryDependenceAnalysis::NonLocalDepInfo &deps =
488 MD->getNonLocalCallDependency(CallSite(C));
489 // FIXME: call/call dependencies for readonly calls should return def, not
490 // clobber! Move the checking logic to MemDep!
491 CallInst* cdep = 0;
492
493 // Check to see if we have a single dominating call instruction that is
494 // identical to C.
495 for (unsigned i = 0, e = deps.size(); i != e; ++i) {
Chris Lattnere18b9712009-12-09 07:08:01 +0000496 const NonLocalDepEntry *I = &deps[i];
Owen Andersond41ed4e2009-10-19 22:14:22 +0000497 // Ignore non-local dependencies.
Chris Lattnere18b9712009-12-09 07:08:01 +0000498 if (I->getResult().isNonLocal())
Owen Andersond41ed4e2009-10-19 22:14:22 +0000499 continue;
500
501 // We don't handle non-depedencies. If we already have a call, reject
502 // instruction dependencies.
Chris Lattnere18b9712009-12-09 07:08:01 +0000503 if (I->getResult().isClobber() || cdep != 0) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000504 cdep = 0;
505 break;
506 }
507
Chris Lattnere18b9712009-12-09 07:08:01 +0000508 CallInst *NonLocalDepCall = dyn_cast<CallInst>(I->getResult().getInst());
Owen Andersond41ed4e2009-10-19 22:14:22 +0000509 // FIXME: All duplicated with non-local case.
Chris Lattnere18b9712009-12-09 07:08:01 +0000510 if (NonLocalDepCall && DT->properlyDominates(I->getBB(), C->getParent())){
Owen Andersond41ed4e2009-10-19 22:14:22 +0000511 cdep = NonLocalDepCall;
512 continue;
513 }
514
515 cdep = 0;
516 break;
517 }
518
519 if (!cdep) {
520 valueNumbering[C] = nextValueNumber;
521 return nextValueNumber++;
522 }
523
524 if (cdep->getNumOperands() != C->getNumOperands()) {
525 valueNumbering[C] = nextValueNumber;
526 return nextValueNumber++;
527 }
528 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
529 uint32_t c_vn = lookup_or_add(C->getOperand(i));
530 uint32_t cd_vn = lookup_or_add(cdep->getOperand(i));
531 if (c_vn != cd_vn) {
532 valueNumbering[C] = nextValueNumber;
533 return nextValueNumber++;
534 }
535 }
536
537 uint32_t v = lookup_or_add(cdep);
538 valueNumbering[C] = v;
539 return v;
540
541 } else {
542 valueNumbering[C] = nextValueNumber;
543 return nextValueNumber++;
544 }
545}
546
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000547/// lookup_or_add - Returns the value number for the specified value, assigning
548/// it a new number if it did not have one before.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000549uint32_t ValueTable::lookup_or_add(Value *V) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000550 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
551 if (VI != valueNumbering.end())
552 return VI->second;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000553
Owen Andersond41ed4e2009-10-19 22:14:22 +0000554 if (!isa<Instruction>(V)) {
Owen Anderson158d86e2009-10-19 21:14:57 +0000555 valueNumbering[V] = nextValueNumber;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000556 return nextValueNumber++;
557 }
Owen Andersond41ed4e2009-10-19 22:14:22 +0000558
559 Instruction* I = cast<Instruction>(V);
560 Expression exp;
561 switch (I->getOpcode()) {
562 case Instruction::Call:
563 return lookup_or_add_call(cast<CallInst>(I));
564 case Instruction::Add:
565 case Instruction::FAdd:
566 case Instruction::Sub:
567 case Instruction::FSub:
568 case Instruction::Mul:
569 case Instruction::FMul:
570 case Instruction::UDiv:
571 case Instruction::SDiv:
572 case Instruction::FDiv:
573 case Instruction::URem:
574 case Instruction::SRem:
575 case Instruction::FRem:
576 case Instruction::Shl:
577 case Instruction::LShr:
578 case Instruction::AShr:
579 case Instruction::And:
580 case Instruction::Or :
581 case Instruction::Xor:
582 exp = create_expression(cast<BinaryOperator>(I));
583 break;
584 case Instruction::ICmp:
585 case Instruction::FCmp:
586 exp = create_expression(cast<CmpInst>(I));
587 break;
588 case Instruction::Trunc:
589 case Instruction::ZExt:
590 case Instruction::SExt:
591 case Instruction::FPToUI:
592 case Instruction::FPToSI:
593 case Instruction::UIToFP:
594 case Instruction::SIToFP:
595 case Instruction::FPTrunc:
596 case Instruction::FPExt:
597 case Instruction::PtrToInt:
598 case Instruction::IntToPtr:
599 case Instruction::BitCast:
600 exp = create_expression(cast<CastInst>(I));
601 break;
602 case Instruction::Select:
603 exp = create_expression(cast<SelectInst>(I));
604 break;
605 case Instruction::ExtractElement:
606 exp = create_expression(cast<ExtractElementInst>(I));
607 break;
608 case Instruction::InsertElement:
609 exp = create_expression(cast<InsertElementInst>(I));
610 break;
611 case Instruction::ShuffleVector:
612 exp = create_expression(cast<ShuffleVectorInst>(I));
613 break;
614 case Instruction::ExtractValue:
615 exp = create_expression(cast<ExtractValueInst>(I));
616 break;
617 case Instruction::InsertValue:
618 exp = create_expression(cast<InsertValueInst>(I));
619 break;
620 case Instruction::GetElementPtr:
621 exp = create_expression(cast<GetElementPtrInst>(I));
622 break;
623 default:
624 valueNumbering[V] = nextValueNumber;
625 return nextValueNumber++;
626 }
627
628 uint32_t& e = expressionNumbering[exp];
629 if (!e) e = nextValueNumber++;
630 valueNumbering[V] = e;
631 return e;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000632}
633
634/// lookup - Returns the value number of the specified value. Fails if
635/// the value has not yet been numbered.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000636uint32_t ValueTable::lookup(Value *V) const {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000637 DenseMap<Value*, uint32_t>::const_iterator VI = valueNumbering.find(V);
Chris Lattner88365bb2008-03-21 21:14:38 +0000638 assert(VI != valueNumbering.end() && "Value not numbered?");
639 return VI->second;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000640}
641
642/// clear - Remove all entries from the ValueTable
643void ValueTable::clear() {
644 valueNumbering.clear();
645 expressionNumbering.clear();
646 nextValueNumber = 1;
647}
648
Owen Andersonbf7d0bc2007-07-31 23:27:13 +0000649/// erase - Remove a value from the value numbering
Chris Lattnerb2412a82009-09-21 02:42:51 +0000650void ValueTable::erase(Value *V) {
Owen Andersonbf7d0bc2007-07-31 23:27:13 +0000651 valueNumbering.erase(V);
652}
653
Bill Wendling246dbbb2008-12-22 21:36:08 +0000654/// verifyRemoved - Verify that the value is removed from all internal data
655/// structures.
656void ValueTable::verifyRemoved(const Value *V) const {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000657 for (DenseMap<Value*, uint32_t>::const_iterator
Bill Wendling246dbbb2008-12-22 21:36:08 +0000658 I = valueNumbering.begin(), E = valueNumbering.end(); I != E; ++I) {
659 assert(I->first != V && "Inst still occurs in value numbering map!");
660 }
661}
662
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000663//===----------------------------------------------------------------------===//
Bill Wendling30788b82008-12-22 22:32:22 +0000664// GVN Pass
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000665//===----------------------------------------------------------------------===//
666
667namespace {
Chris Lattner3e8b6632009-09-02 06:11:42 +0000668 struct ValueNumberScope {
Owen Anderson6fafe842008-06-20 01:15:47 +0000669 ValueNumberScope* parent;
670 DenseMap<uint32_t, Value*> table;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000671
Owen Anderson6fafe842008-06-20 01:15:47 +0000672 ValueNumberScope(ValueNumberScope* p) : parent(p) { }
673 };
674}
675
676namespace {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000677
Chris Lattner3e8b6632009-09-02 06:11:42 +0000678 class GVN : public FunctionPass {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000679 bool runOnFunction(Function &F);
680 public:
681 static char ID; // Pass identification, replacement for typeid
Dan Gohman4ec01b22009-11-14 02:27:51 +0000682 explicit GVN(bool nopre = false, bool noloads = false)
683 : FunctionPass(&ID), NoPRE(nopre), NoLoads(noloads), MD(0) { }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000684
685 private:
Evan Cheng2f192c22009-10-30 20:12:24 +0000686 bool NoPRE;
Dan Gohman4ec01b22009-11-14 02:27:51 +0000687 bool NoLoads;
Chris Lattner663e4412008-12-01 00:40:32 +0000688 MemoryDependenceAnalysis *MD;
689 DominatorTree *DT;
690
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000691 ValueTable VN;
Owen Anderson6fafe842008-06-20 01:15:47 +0000692 DenseMap<BasicBlock*, ValueNumberScope*> localAvail;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000693
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000694 // This transformation requires dominator postdominator info
695 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000696 AU.addRequired<DominatorTree>();
Dan Gohman4ec01b22009-11-14 02:27:51 +0000697 if (!NoLoads)
698 AU.addRequired<MemoryDependenceAnalysis>();
Owen Andersonb388ca92007-10-18 19:39:33 +0000699 AU.addRequired<AliasAnalysis>();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000700
Owen Andersonb70a5712008-06-23 17:49:45 +0000701 AU.addPreserved<DominatorTree>();
Owen Andersonb388ca92007-10-18 19:39:33 +0000702 AU.addPreserved<AliasAnalysis>();
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000703 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000704
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000705 // Helper fuctions
706 // FIXME: eliminate or document these better
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000707 bool processLoad(LoadInst* L,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000708 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000709 bool processInstruction(Instruction *I,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000710 SmallVectorImpl<Instruction*> &toErase);
Owen Anderson830db6a2007-08-02 18:16:06 +0000711 bool processNonLocalLoad(LoadInst* L,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000712 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000713 bool processBlock(BasicBlock *BB);
Owen Andersonb2303722008-06-18 21:41:49 +0000714 void dump(DenseMap<uint32_t, Value*>& d);
Owen Anderson3e75a422007-08-14 18:04:11 +0000715 bool iterateOnFunction(Function &F);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000716 Value *CollapsePhi(PHINode* p);
Owen Andersonb2303722008-06-18 21:41:49 +0000717 bool performPRE(Function& F);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000718 Value *lookupNumber(BasicBlock *BB, uint32_t num);
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +0000719 void cleanupGlobalSets();
Bill Wendling246dbbb2008-12-22 21:36:08 +0000720 void verifyRemoved(const Instruction *I) const;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000721 };
Daniel Dunbara279bc32009-09-20 02:20:51 +0000722
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000723 char GVN::ID = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000724}
725
726// createGVNPass - The public interface to this file...
Dan Gohman4ec01b22009-11-14 02:27:51 +0000727FunctionPass *llvm::createGVNPass(bool NoPRE, bool NoLoads) {
728 return new GVN(NoPRE, NoLoads);
729}
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000730
731static RegisterPass<GVN> X("gvn",
732 "Global Value Numbering");
733
Owen Andersonb2303722008-06-18 21:41:49 +0000734void GVN::dump(DenseMap<uint32_t, Value*>& d) {
Dan Gohmanad12b262009-12-18 03:25:51 +0000735 errs() << "{\n";
Owen Andersonb2303722008-06-18 21:41:49 +0000736 for (DenseMap<uint32_t, Value*>::iterator I = d.begin(),
Owen Anderson0cd32032007-07-25 19:57:03 +0000737 E = d.end(); I != E; ++I) {
Dan Gohmanad12b262009-12-18 03:25:51 +0000738 errs() << I->first << "\n";
Owen Anderson0cd32032007-07-25 19:57:03 +0000739 I->second->dump();
740 }
Dan Gohmanad12b262009-12-18 03:25:51 +0000741 errs() << "}\n";
Owen Anderson0cd32032007-07-25 19:57:03 +0000742}
743
Chris Lattnerb2412a82009-09-21 02:42:51 +0000744static bool isSafeReplacement(PHINode* p, Instruction *inst) {
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000745 if (!isa<PHINode>(inst))
746 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000747
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000748 for (Instruction::use_iterator UI = p->use_begin(), E = p->use_end();
749 UI != E; ++UI)
750 if (PHINode* use_phi = dyn_cast<PHINode>(UI))
751 if (use_phi->getParent() == inst->getParent())
752 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000753
Owen Anderson4eebf0b2009-08-26 22:55:11 +0000754 return true;
755}
756
Chris Lattnerb2412a82009-09-21 02:42:51 +0000757Value *GVN::CollapsePhi(PHINode *PN) {
758 Value *ConstVal = PN->hasConstantValue(DT);
759 if (!ConstVal) return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000760
Chris Lattnerb2412a82009-09-21 02:42:51 +0000761 Instruction *Inst = dyn_cast<Instruction>(ConstVal);
762 if (!Inst)
763 return ConstVal;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000764
Chris Lattnerb2412a82009-09-21 02:42:51 +0000765 if (DT->dominates(Inst, PN))
766 if (isSafeReplacement(PN, Inst))
767 return Inst;
Owen Anderson1defe2d2007-08-16 22:51:56 +0000768 return 0;
769}
Owen Anderson0cd32032007-07-25 19:57:03 +0000770
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000771/// IsValueFullyAvailableInBlock - Return true if we can prove that the value
772/// we're analyzing is fully available in the specified block. As we go, keep
Chris Lattner72bc70d2008-12-05 07:49:08 +0000773/// track of which blocks we know are fully alive in FullyAvailableBlocks. This
774/// map is actually a tri-state map with the following values:
775/// 0) we know the block *is not* fully available.
776/// 1) we know the block *is* fully available.
777/// 2) we do not know whether the block is fully available or not, but we are
778/// currently speculating that it will be.
779/// 3) we are speculating for this block and have used that to speculate for
780/// other blocks.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000781static bool IsValueFullyAvailableInBlock(BasicBlock *BB,
Chris Lattner72bc70d2008-12-05 07:49:08 +0000782 DenseMap<BasicBlock*, char> &FullyAvailableBlocks) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000783 // Optimistically assume that the block is fully available and check to see
784 // if we already know about this block in one lookup.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000785 std::pair<DenseMap<BasicBlock*, char>::iterator, char> IV =
Chris Lattner72bc70d2008-12-05 07:49:08 +0000786 FullyAvailableBlocks.insert(std::make_pair(BB, 2));
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000787
788 // If the entry already existed for this block, return the precomputed value.
Chris Lattner72bc70d2008-12-05 07:49:08 +0000789 if (!IV.second) {
790 // If this is a speculative "available" value, mark it as being used for
791 // speculation of other blocks.
792 if (IV.first->second == 2)
793 IV.first->second = 3;
794 return IV.first->second != 0;
795 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000796
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000797 // Otherwise, see if it is fully available in all predecessors.
798 pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000799
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000800 // If this block has no predecessors, it isn't live-in here.
801 if (PI == PE)
Chris Lattner72bc70d2008-12-05 07:49:08 +0000802 goto SpeculationFailure;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000803
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000804 for (; PI != PE; ++PI)
805 // If the value isn't fully available in one of our predecessors, then it
806 // isn't fully available in this block either. Undo our previous
807 // optimistic assumption and bail out.
808 if (!IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
Chris Lattner72bc70d2008-12-05 07:49:08 +0000809 goto SpeculationFailure;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000810
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000811 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000812
Chris Lattner72bc70d2008-12-05 07:49:08 +0000813// SpeculationFailure - If we get here, we found out that this is not, after
814// all, a fully-available block. We have a problem if we speculated on this and
815// used the speculation to mark other blocks as available.
816SpeculationFailure:
817 char &BBVal = FullyAvailableBlocks[BB];
Daniel Dunbara279bc32009-09-20 02:20:51 +0000818
Chris Lattner72bc70d2008-12-05 07:49:08 +0000819 // If we didn't speculate on this, just return with it set to false.
820 if (BBVal == 2) {
821 BBVal = 0;
822 return false;
823 }
824
825 // If we did speculate on this value, we could have blocks set to 1 that are
826 // incorrect. Walk the (transitive) successors of this block and mark them as
827 // 0 if set to one.
828 SmallVector<BasicBlock*, 32> BBWorklist;
829 BBWorklist.push_back(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000830
Chris Lattner72bc70d2008-12-05 07:49:08 +0000831 while (!BBWorklist.empty()) {
832 BasicBlock *Entry = BBWorklist.pop_back_val();
833 // Note that this sets blocks to 0 (unavailable) if they happen to not
834 // already be in FullyAvailableBlocks. This is safe.
835 char &EntryVal = FullyAvailableBlocks[Entry];
836 if (EntryVal == 0) continue; // Already unavailable.
837
838 // Mark as unavailable.
839 EntryVal = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000840
Chris Lattner72bc70d2008-12-05 07:49:08 +0000841 for (succ_iterator I = succ_begin(Entry), E = succ_end(Entry); I != E; ++I)
842 BBWorklist.push_back(*I);
843 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000844
Chris Lattner72bc70d2008-12-05 07:49:08 +0000845 return false;
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000846}
847
Chris Lattner771a5422009-09-20 20:09:34 +0000848
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000849/// CanCoerceMustAliasedValueToLoad - Return true if
850/// CoerceAvailableValueToLoadType will succeed.
851static bool CanCoerceMustAliasedValueToLoad(Value *StoredVal,
852 const Type *LoadTy,
853 const TargetData &TD) {
854 // If the loaded or stored value is an first class array or struct, don't try
855 // to transform them. We need to be able to bitcast to integer.
856 if (isa<StructType>(LoadTy) || isa<ArrayType>(LoadTy) ||
857 isa<StructType>(StoredVal->getType()) ||
858 isa<ArrayType>(StoredVal->getType()))
859 return false;
860
861 // The store has to be at least as big as the load.
862 if (TD.getTypeSizeInBits(StoredVal->getType()) <
863 TD.getTypeSizeInBits(LoadTy))
864 return false;
865
866 return true;
867}
868
869
Chris Lattner771a5422009-09-20 20:09:34 +0000870/// CoerceAvailableValueToLoadType - If we saw a store of a value to memory, and
871/// then a load from a must-aliased pointer of a different type, try to coerce
872/// the stored value. LoadedTy is the type of the load we want to replace and
873/// InsertPt is the place to insert new instructions.
874///
875/// If we can't do it, return null.
876static Value *CoerceAvailableValueToLoadType(Value *StoredVal,
877 const Type *LoadedTy,
878 Instruction *InsertPt,
879 const TargetData &TD) {
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000880 if (!CanCoerceMustAliasedValueToLoad(StoredVal, LoadedTy, TD))
881 return 0;
882
Chris Lattner771a5422009-09-20 20:09:34 +0000883 const Type *StoredValTy = StoredVal->getType();
884
885 uint64_t StoreSize = TD.getTypeSizeInBits(StoredValTy);
886 uint64_t LoadSize = TD.getTypeSizeInBits(LoadedTy);
887
888 // If the store and reload are the same size, we can always reuse it.
889 if (StoreSize == LoadSize) {
890 if (isa<PointerType>(StoredValTy) && isa<PointerType>(LoadedTy)) {
891 // Pointer to Pointer -> use bitcast.
892 return new BitCastInst(StoredVal, LoadedTy, "", InsertPt);
893 }
894
895 // Convert source pointers to integers, which can be bitcast.
896 if (isa<PointerType>(StoredValTy)) {
897 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
898 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
899 }
900
901 const Type *TypeToCastTo = LoadedTy;
902 if (isa<PointerType>(TypeToCastTo))
903 TypeToCastTo = TD.getIntPtrType(StoredValTy->getContext());
904
905 if (StoredValTy != TypeToCastTo)
906 StoredVal = new BitCastInst(StoredVal, TypeToCastTo, "", InsertPt);
907
908 // Cast to pointer if the load needs a pointer type.
909 if (isa<PointerType>(LoadedTy))
910 StoredVal = new IntToPtrInst(StoredVal, LoadedTy, "", InsertPt);
911
912 return StoredVal;
913 }
914
915 // If the loaded value is smaller than the available value, then we can
916 // extract out a piece from it. If the available value is too small, then we
917 // can't do anything.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000918 assert(StoreSize >= LoadSize && "CanCoerceMustAliasedValueToLoad fail");
Chris Lattner771a5422009-09-20 20:09:34 +0000919
920 // Convert source pointers to integers, which can be manipulated.
921 if (isa<PointerType>(StoredValTy)) {
922 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
923 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
924 }
925
926 // Convert vectors and fp to integer, which can be manipulated.
927 if (!isa<IntegerType>(StoredValTy)) {
928 StoredValTy = IntegerType::get(StoredValTy->getContext(), StoreSize);
929 StoredVal = new BitCastInst(StoredVal, StoredValTy, "", InsertPt);
930 }
931
932 // If this is a big-endian system, we need to shift the value down to the low
933 // bits so that a truncate will work.
934 if (TD.isBigEndian()) {
935 Constant *Val = ConstantInt::get(StoredVal->getType(), StoreSize-LoadSize);
936 StoredVal = BinaryOperator::CreateLShr(StoredVal, Val, "tmp", InsertPt);
937 }
938
939 // Truncate the integer to the right size now.
940 const Type *NewIntTy = IntegerType::get(StoredValTy->getContext(), LoadSize);
941 StoredVal = new TruncInst(StoredVal, NewIntTy, "trunc", InsertPt);
942
943 if (LoadedTy == NewIntTy)
944 return StoredVal;
945
946 // If the result is a pointer, inttoptr.
947 if (isa<PointerType>(LoadedTy))
948 return new IntToPtrInst(StoredVal, LoadedTy, "inttoptr", InsertPt);
949
950 // Otherwise, bitcast.
951 return new BitCastInst(StoredVal, LoadedTy, "bitcast", InsertPt);
952}
953
Chris Lattnerca749402009-09-21 06:24:16 +0000954/// GetBaseWithConstantOffset - Analyze the specified pointer to see if it can
955/// be expressed as a base pointer plus a constant offset. Return the base and
956/// offset to the caller.
957static Value *GetBaseWithConstantOffset(Value *Ptr, int64_t &Offset,
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000958 const TargetData &TD) {
Chris Lattnerca749402009-09-21 06:24:16 +0000959 Operator *PtrOp = dyn_cast<Operator>(Ptr);
960 if (PtrOp == 0) return Ptr;
961
962 // Just look through bitcasts.
963 if (PtrOp->getOpcode() == Instruction::BitCast)
964 return GetBaseWithConstantOffset(PtrOp->getOperand(0), Offset, TD);
965
966 // If this is a GEP with constant indices, we can look through it.
967 GEPOperator *GEP = dyn_cast<GEPOperator>(PtrOp);
968 if (GEP == 0 || !GEP->hasAllConstantIndices()) return Ptr;
969
970 gep_type_iterator GTI = gep_type_begin(GEP);
971 for (User::op_iterator I = GEP->idx_begin(), E = GEP->idx_end(); I != E;
972 ++I, ++GTI) {
973 ConstantInt *OpC = cast<ConstantInt>(*I);
974 if (OpC->isZero()) continue;
975
976 // Handle a struct and array indices which add their offset to the pointer.
977 if (const StructType *STy = dyn_cast<StructType>(*GTI)) {
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000978 Offset += TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue());
Chris Lattnerca749402009-09-21 06:24:16 +0000979 } else {
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000980 uint64_t Size = TD.getTypeAllocSize(GTI.getIndexedType());
Chris Lattnerca749402009-09-21 06:24:16 +0000981 Offset += OpC->getSExtValue()*Size;
982 }
983 }
984
985 // Re-sign extend from the pointer size if needed to get overflow edge cases
986 // right.
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000987 unsigned PtrSize = TD.getPointerSizeInBits();
Chris Lattnerca749402009-09-21 06:24:16 +0000988 if (PtrSize < 64)
989 Offset = (Offset << (64-PtrSize)) >> (64-PtrSize);
990
991 return GetBaseWithConstantOffset(GEP->getPointerOperand(), Offset, TD);
992}
993
994
Chris Lattnerfaf815b2009-12-06 01:57:02 +0000995/// AnalyzeLoadFromClobberingWrite - This function is called when we have a
996/// memdep query of a load that ends up being a clobbering memory write (store,
997/// memset, memcpy, memmove). This means that the write *may* provide bits used
998/// by the load but we can't be sure because the pointers don't mustalias.
999///
1000/// Check this case to see if there is anything more we can do before we give
1001/// up. This returns -1 if we have to give up, or a byte number in the stored
1002/// value of the piece that feeds the load.
Chris Lattner03f17da2009-12-09 07:34:10 +00001003static int AnalyzeLoadFromClobberingWrite(const Type *LoadTy, Value *LoadPtr,
1004 Value *WritePtr,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001005 uint64_t WriteSizeInBits,
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001006 const TargetData &TD) {
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001007 // If the loaded or stored value is an first class array or struct, don't try
1008 // to transform them. We need to be able to bitcast to integer.
Chris Lattner03f17da2009-12-09 07:34:10 +00001009 if (isa<StructType>(LoadTy) || isa<ArrayType>(LoadTy))
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001010 return -1;
1011
Chris Lattnerca749402009-09-21 06:24:16 +00001012 int64_t StoreOffset = 0, LoadOffset = 0;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001013 Value *StoreBase = GetBaseWithConstantOffset(WritePtr, StoreOffset, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001014 Value *LoadBase =
Chris Lattner03f17da2009-12-09 07:34:10 +00001015 GetBaseWithConstantOffset(LoadPtr, LoadOffset, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001016 if (StoreBase != LoadBase)
1017 return -1;
1018
1019 // If the load and store are to the exact same address, they should have been
1020 // a must alias. AA must have gotten confused.
1021 // FIXME: Study to see if/when this happens.
1022 if (LoadOffset == StoreOffset) {
1023#if 0
1024 errs() << "STORE/LOAD DEP WITH COMMON POINTER MISSED:\n"
1025 << "Base = " << *StoreBase << "\n"
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001026 << "Store Ptr = " << *WritePtr << "\n"
1027 << "Store Offs = " << StoreOffset << "\n"
Chris Lattnerb6760b42009-12-10 00:04:46 +00001028 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattnerb3f927f2009-12-09 02:41:54 +00001029 abort();
Chris Lattnerca749402009-09-21 06:24:16 +00001030#endif
1031 return -1;
1032 }
1033
1034 // If the load and store don't overlap at all, the store doesn't provide
1035 // anything to the load. In this case, they really don't alias at all, AA
1036 // must have gotten confused.
1037 // FIXME: Investigate cases where this bails out, e.g. rdar://7238614. Then
1038 // remove this check, as it is duplicated with what we have below.
Chris Lattner03f17da2009-12-09 07:34:10 +00001039 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy);
Chris Lattnerca749402009-09-21 06:24:16 +00001040
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001041 if ((WriteSizeInBits & 7) | (LoadSize & 7))
Chris Lattnerca749402009-09-21 06:24:16 +00001042 return -1;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001043 uint64_t StoreSize = WriteSizeInBits >> 3; // Convert to bytes.
Chris Lattnerca749402009-09-21 06:24:16 +00001044 LoadSize >>= 3;
1045
1046
1047 bool isAAFailure = false;
1048 if (StoreOffset < LoadOffset) {
1049 isAAFailure = StoreOffset+int64_t(StoreSize) <= LoadOffset;
1050 } else {
1051 isAAFailure = LoadOffset+int64_t(LoadSize) <= StoreOffset;
1052 }
1053 if (isAAFailure) {
1054#if 0
1055 errs() << "STORE LOAD DEP WITH COMMON BASE:\n"
1056 << "Base = " << *StoreBase << "\n"
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001057 << "Store Ptr = " << *WritePtr << "\n"
1058 << "Store Offs = " << StoreOffset << "\n"
Chris Lattnerb6760b42009-12-10 00:04:46 +00001059 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattnerb3f927f2009-12-09 02:41:54 +00001060 abort();
Chris Lattnerca749402009-09-21 06:24:16 +00001061#endif
1062 return -1;
1063 }
1064
1065 // If the Load isn't completely contained within the stored bits, we don't
1066 // have all the bits to feed it. We could do something crazy in the future
1067 // (issue a smaller load then merge the bits in) but this seems unlikely to be
1068 // valuable.
1069 if (StoreOffset > LoadOffset ||
1070 StoreOffset+StoreSize < LoadOffset+LoadSize)
1071 return -1;
1072
1073 // Okay, we can do this transformation. Return the number of bytes into the
1074 // store that the load is.
1075 return LoadOffset-StoreOffset;
1076}
1077
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001078/// AnalyzeLoadFromClobberingStore - This function is called when we have a
1079/// memdep query of a load that ends up being a clobbering store.
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001080static int AnalyzeLoadFromClobberingStore(const Type *LoadTy, Value *LoadPtr,
1081 StoreInst *DepSI,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001082 const TargetData &TD) {
1083 // Cannot handle reading from store of first-class aggregate yet.
1084 if (isa<StructType>(DepSI->getOperand(0)->getType()) ||
1085 isa<ArrayType>(DepSI->getOperand(0)->getType()))
1086 return -1;
1087
1088 Value *StorePtr = DepSI->getPointerOperand();
Chris Lattner16f244e2009-12-10 00:11:45 +00001089 uint64_t StoreSize = TD.getTypeSizeInBits(DepSI->getOperand(0)->getType());
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001090 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
Chris Lattner03f17da2009-12-09 07:34:10 +00001091 StorePtr, StoreSize, TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001092}
1093
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001094static int AnalyzeLoadFromClobberingMemInst(const Type *LoadTy, Value *LoadPtr,
1095 MemIntrinsic *MI,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001096 const TargetData &TD) {
1097 // If the mem operation is a non-constant size, we can't handle it.
1098 ConstantInt *SizeCst = dyn_cast<ConstantInt>(MI->getLength());
1099 if (SizeCst == 0) return -1;
1100 uint64_t MemSizeInBits = SizeCst->getZExtValue()*8;
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001101
1102 // If this is memset, we just need to see if the offset is valid in the size
1103 // of the memset..
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001104 if (MI->getIntrinsicID() == Intrinsic::memset)
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001105 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr, MI->getDest(),
1106 MemSizeInBits, TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001107
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001108 // If we have a memcpy/memmove, the only case we can handle is if this is a
1109 // copy from constant memory. In that case, we can read directly from the
1110 // constant memory.
1111 MemTransferInst *MTI = cast<MemTransferInst>(MI);
1112
1113 Constant *Src = dyn_cast<Constant>(MTI->getSource());
1114 if (Src == 0) return -1;
1115
1116 GlobalVariable *GV = dyn_cast<GlobalVariable>(Src->getUnderlyingObject());
1117 if (GV == 0 || !GV->isConstant()) return -1;
1118
1119 // See if the access is within the bounds of the transfer.
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001120 int Offset = AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
1121 MI->getDest(), MemSizeInBits, TD);
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001122 if (Offset == -1)
1123 return Offset;
1124
1125 // Otherwise, see if we can constant fold a load from the constant with the
1126 // offset applied as appropriate.
1127 Src = ConstantExpr::getBitCast(Src,
1128 llvm::Type::getInt8PtrTy(Src->getContext()));
1129 Constant *OffsetCst =
1130 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1131 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001132 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001133 if (ConstantFoldLoadFromConstPtr(Src, &TD))
1134 return Offset;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001135 return -1;
1136}
1137
Chris Lattnerca749402009-09-21 06:24:16 +00001138
1139/// GetStoreValueForLoad - This function is called when we have a
1140/// memdep query of a load that ends up being a clobbering store. This means
1141/// that the store *may* provide bits used by the load but we can't be sure
1142/// because the pointers don't mustalias. Check this case to see if there is
1143/// anything more we can do before we give up.
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001144static Value *GetStoreValueForLoad(Value *SrcVal, unsigned Offset,
1145 const Type *LoadTy,
1146 Instruction *InsertPt, const TargetData &TD){
Chris Lattnerca749402009-09-21 06:24:16 +00001147 LLVMContext &Ctx = SrcVal->getType()->getContext();
1148
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001149 uint64_t StoreSize = TD.getTypeSizeInBits(SrcVal->getType())/8;
1150 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
Chris Lattnerca749402009-09-21 06:24:16 +00001151
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001152 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
Chris Lattnerca749402009-09-21 06:24:16 +00001153
1154 // Compute which bits of the stored value are being used by the load. Convert
1155 // to an integer type to start with.
1156 if (isa<PointerType>(SrcVal->getType()))
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001157 SrcVal = Builder.CreatePtrToInt(SrcVal, TD.getIntPtrType(Ctx), "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001158 if (!isa<IntegerType>(SrcVal->getType()))
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001159 SrcVal = Builder.CreateBitCast(SrcVal, IntegerType::get(Ctx, StoreSize*8),
1160 "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001161
1162 // Shift the bits to the least significant depending on endianness.
1163 unsigned ShiftAmt;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001164 if (TD.isLittleEndian())
Chris Lattnerca749402009-09-21 06:24:16 +00001165 ShiftAmt = Offset*8;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001166 else
Chris Lattner19ad7842009-09-21 17:55:47 +00001167 ShiftAmt = (StoreSize-LoadSize-Offset)*8;
Chris Lattnerca749402009-09-21 06:24:16 +00001168
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001169 if (ShiftAmt)
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001170 SrcVal = Builder.CreateLShr(SrcVal, ShiftAmt, "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001171
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001172 if (LoadSize != StoreSize)
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001173 SrcVal = Builder.CreateTrunc(SrcVal, IntegerType::get(Ctx, LoadSize*8),
1174 "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001175
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001176 return CoerceAvailableValueToLoadType(SrcVal, LoadTy, InsertPt, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001177}
1178
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001179/// GetMemInstValueForLoad - This function is called when we have a
1180/// memdep query of a load that ends up being a clobbering mem intrinsic.
1181static Value *GetMemInstValueForLoad(MemIntrinsic *SrcInst, unsigned Offset,
1182 const Type *LoadTy, Instruction *InsertPt,
1183 const TargetData &TD){
1184 LLVMContext &Ctx = LoadTy->getContext();
1185 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
1186
1187 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
1188
1189 // We know that this method is only called when the mem transfer fully
1190 // provides the bits for the load.
1191 if (MemSetInst *MSI = dyn_cast<MemSetInst>(SrcInst)) {
1192 // memset(P, 'x', 1234) -> splat('x'), even if x is a variable, and
1193 // independently of what the offset is.
1194 Value *Val = MSI->getValue();
1195 if (LoadSize != 1)
1196 Val = Builder.CreateZExt(Val, IntegerType::get(Ctx, LoadSize*8));
1197
1198 Value *OneElt = Val;
1199
1200 // Splat the value out to the right number of bits.
1201 for (unsigned NumBytesSet = 1; NumBytesSet != LoadSize; ) {
1202 // If we can double the number of bytes set, do it.
1203 if (NumBytesSet*2 <= LoadSize) {
1204 Value *ShVal = Builder.CreateShl(Val, NumBytesSet*8);
1205 Val = Builder.CreateOr(Val, ShVal);
1206 NumBytesSet <<= 1;
1207 continue;
1208 }
1209
1210 // Otherwise insert one byte at a time.
1211 Value *ShVal = Builder.CreateShl(Val, 1*8);
1212 Val = Builder.CreateOr(OneElt, ShVal);
1213 ++NumBytesSet;
1214 }
1215
1216 return CoerceAvailableValueToLoadType(Val, LoadTy, InsertPt, TD);
1217 }
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001218
1219 // Otherwise, this is a memcpy/memmove from a constant global.
1220 MemTransferInst *MTI = cast<MemTransferInst>(SrcInst);
1221 Constant *Src = cast<Constant>(MTI->getSource());
1222
1223 // Otherwise, see if we can constant fold a load from the constant with the
1224 // offset applied as appropriate.
1225 Src = ConstantExpr::getBitCast(Src,
1226 llvm::Type::getInt8PtrTy(Src->getContext()));
1227 Constant *OffsetCst =
1228 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1229 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
1230 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
1231 return ConstantFoldLoadFromConstPtr(Src, &TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001232}
1233
1234
1235
Chris Lattner87913512009-09-21 06:30:24 +00001236struct AvailableValueInBlock {
1237 /// BB - The basic block in question.
1238 BasicBlock *BB;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001239 enum ValType {
1240 SimpleVal, // A simple offsetted value that is accessed.
1241 MemIntrin // A memory intrinsic which is loaded from.
1242 };
1243
Chris Lattner87913512009-09-21 06:30:24 +00001244 /// V - The value that is live out of the block.
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001245 PointerIntPair<Value *, 1, ValType> Val;
1246
1247 /// Offset - The byte offset in Val that is interesting for the load query.
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001248 unsigned Offset;
Chris Lattner87913512009-09-21 06:30:24 +00001249
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001250 static AvailableValueInBlock get(BasicBlock *BB, Value *V,
1251 unsigned Offset = 0) {
Chris Lattner87913512009-09-21 06:30:24 +00001252 AvailableValueInBlock Res;
1253 Res.BB = BB;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001254 Res.Val.setPointer(V);
1255 Res.Val.setInt(SimpleVal);
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001256 Res.Offset = Offset;
Chris Lattner87913512009-09-21 06:30:24 +00001257 return Res;
1258 }
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001259
1260 static AvailableValueInBlock getMI(BasicBlock *BB, MemIntrinsic *MI,
1261 unsigned Offset = 0) {
1262 AvailableValueInBlock Res;
1263 Res.BB = BB;
1264 Res.Val.setPointer(MI);
1265 Res.Val.setInt(MemIntrin);
1266 Res.Offset = Offset;
1267 return Res;
1268 }
1269
1270 bool isSimpleValue() const { return Val.getInt() == SimpleVal; }
1271 Value *getSimpleValue() const {
1272 assert(isSimpleValue() && "Wrong accessor");
1273 return Val.getPointer();
1274 }
1275
1276 MemIntrinsic *getMemIntrinValue() const {
1277 assert(!isSimpleValue() && "Wrong accessor");
1278 return cast<MemIntrinsic>(Val.getPointer());
1279 }
Chris Lattner5362c542009-12-21 23:04:33 +00001280
1281 /// MaterializeAdjustedValue - Emit code into this block to adjust the value
1282 /// defined here to the specified type. This handles various coercion cases.
1283 Value *MaterializeAdjustedValue(const Type *LoadTy,
1284 const TargetData *TD) const {
1285 Value *Res;
1286 if (isSimpleValue()) {
1287 Res = getSimpleValue();
1288 if (Res->getType() != LoadTy) {
1289 assert(TD && "Need target data to handle type mismatch case");
1290 Res = GetStoreValueForLoad(Res, Offset, LoadTy, BB->getTerminator(),
1291 *TD);
1292
1293 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\nOffset: " << Offset << " "
1294 << *getSimpleValue() << '\n'
1295 << *Res << '\n' << "\n\n\n");
1296 }
1297 } else {
1298 Res = GetMemInstValueForLoad(getMemIntrinValue(), Offset,
1299 LoadTy, BB->getTerminator(), *TD);
1300 DEBUG(errs() << "GVN COERCED NONLOCAL MEM INTRIN:\nOffset: " << Offset
1301 << " " << *getMemIntrinValue() << '\n'
1302 << *Res << '\n' << "\n\n\n");
1303 }
1304 return Res;
1305 }
Chris Lattner87913512009-09-21 06:30:24 +00001306};
1307
Chris Lattnera09fbf02009-10-10 23:50:30 +00001308/// ConstructSSAForLoadSet - Given a set of loads specified by ValuesPerBlock,
1309/// construct SSA form, allowing us to eliminate LI. This returns the value
1310/// that should be used at LI's definition site.
1311static Value *ConstructSSAForLoadSet(LoadInst *LI,
1312 SmallVectorImpl<AvailableValueInBlock> &ValuesPerBlock,
1313 const TargetData *TD,
1314 AliasAnalysis *AA) {
1315 SmallVector<PHINode*, 8> NewPHIs;
1316 SSAUpdater SSAUpdate(&NewPHIs);
1317 SSAUpdate.Initialize(LI);
1318
1319 const Type *LoadTy = LI->getType();
1320
Chris Lattner771a5422009-09-20 20:09:34 +00001321 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001322 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1323 BasicBlock *BB = AV.BB;
Chris Lattner771a5422009-09-20 20:09:34 +00001324
Chris Lattnera09fbf02009-10-10 23:50:30 +00001325 if (SSAUpdate.HasValueForBlock(BB))
1326 continue;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001327
Chris Lattner5362c542009-12-21 23:04:33 +00001328 SSAUpdate.AddAvailableValue(BB, AV.MaterializeAdjustedValue(LoadTy, TD));
Chris Lattner771a5422009-09-20 20:09:34 +00001329 }
Chris Lattnera09fbf02009-10-10 23:50:30 +00001330
1331 // Perform PHI construction.
1332 Value *V = SSAUpdate.GetValueInMiddleOfBlock(LI->getParent());
1333
1334 // If new PHI nodes were created, notify alias analysis.
1335 if (isa<PointerType>(V->getType()))
1336 for (unsigned i = 0, e = NewPHIs.size(); i != e; ++i)
1337 AA->copyValue(LI, NewPHIs[i]);
1338
1339 return V;
Chris Lattner771a5422009-09-20 20:09:34 +00001340}
1341
Owen Anderson9ff5a232009-12-02 07:35:19 +00001342static bool isLifetimeStart(Instruction *Inst) {
Chris Lattner720e7902009-12-02 06:44:58 +00001343 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(Inst))
Owen Anderson9ff5a232009-12-02 07:35:19 +00001344 return II->getIntrinsicID() == Intrinsic::lifetime_start;
Chris Lattner720e7902009-12-02 06:44:58 +00001345 return false;
1346}
1347
Owen Anderson62bc33c2007-08-16 22:02:55 +00001348/// processNonLocalLoad - Attempt to eliminate a load whose dependencies are
1349/// non-local by performing PHI construction.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001350bool GVN::processNonLocalLoad(LoadInst *LI,
Chris Lattner8e1e95c2008-03-21 22:01:16 +00001351 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001352 // Find the non-local dependencies of the load.
Chris Lattnere18b9712009-12-09 07:08:01 +00001353 SmallVector<NonLocalDepEntry, 64> Deps;
Chris Lattner91bcf642008-12-09 19:25:07 +00001354 MD->getNonLocalPointerDependency(LI->getOperand(0), true, LI->getParent(),
1355 Deps);
Dan Gohman2a298992009-07-31 20:24:18 +00001356 //DEBUG(errs() << "INVESTIGATING NONLOCAL LOAD: "
1357 // << Deps.size() << *LI << '\n');
Daniel Dunbara279bc32009-09-20 02:20:51 +00001358
Owen Anderson516eb1c2008-08-26 22:07:42 +00001359 // If we had to process more than one hundred blocks to find the
1360 // dependencies, this load isn't worth worrying about. Optimizing
1361 // it will be too expensive.
Chris Lattner91bcf642008-12-09 19:25:07 +00001362 if (Deps.size() > 100)
Owen Anderson516eb1c2008-08-26 22:07:42 +00001363 return false;
Chris Lattner5f4f84b2008-12-18 00:51:32 +00001364
1365 // If we had a phi translation failure, we'll have a single entry which is a
1366 // clobber in the current block. Reject this early.
Chris Lattnere18b9712009-12-09 07:08:01 +00001367 if (Deps.size() == 1 && Deps[0].getResult().isClobber()) {
Torok Edwin4306b1a2009-06-17 18:48:18 +00001368 DEBUG(
Dan Gohmanfd87a542009-07-25 01:43:01 +00001369 errs() << "GVN: non-local load ";
1370 WriteAsOperand(errs(), LI);
Chris Lattnere18b9712009-12-09 07:08:01 +00001371 errs() << " is clobbered by " << *Deps[0].getResult().getInst() << '\n';
Torok Edwin4306b1a2009-06-17 18:48:18 +00001372 );
Chris Lattner5f4f84b2008-12-18 00:51:32 +00001373 return false;
Torok Edwin4306b1a2009-06-17 18:48:18 +00001374 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001375
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001376 // Filter out useless results (non-locals, etc). Keep track of the blocks
1377 // where we have a value available in repl, also keep track of whether we see
1378 // dependencies that produce an unknown value for the load (such as a call
1379 // that could potentially clobber the load).
Chris Lattner87913512009-09-21 06:30:24 +00001380 SmallVector<AvailableValueInBlock, 16> ValuesPerBlock;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001381 SmallVector<BasicBlock*, 16> UnavailableBlocks;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001382
Chris Lattner771a5422009-09-20 20:09:34 +00001383 const TargetData *TD = 0;
1384
Chris Lattner91bcf642008-12-09 19:25:07 +00001385 for (unsigned i = 0, e = Deps.size(); i != e; ++i) {
Chris Lattnere18b9712009-12-09 07:08:01 +00001386 BasicBlock *DepBB = Deps[i].getBB();
1387 MemDepResult DepInfo = Deps[i].getResult();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001388
Chris Lattnerb51deb92008-12-05 21:04:20 +00001389 if (DepInfo.isClobber()) {
Chris Lattneraf064ae2009-12-09 18:21:46 +00001390 // The address being loaded in this non-local block may not be the same as
1391 // the pointer operand of the load if PHI translation occurs. Make sure
1392 // to consider the right address.
1393 Value *Address = Deps[i].getAddress();
1394
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001395 // If the dependence is to a store that writes to a superset of the bits
1396 // read by the load, we can extract the bits we need for the load from the
1397 // stored value.
1398 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInfo.getInst())) {
1399 if (TD == 0)
1400 TD = getAnalysisIfAvailable<TargetData>();
Chris Lattneraf064ae2009-12-09 18:21:46 +00001401 if (TD && Address) {
1402 int Offset = AnalyzeLoadFromClobberingStore(LI->getType(), Address,
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001403 DepSI, *TD);
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001404 if (Offset != -1) {
1405 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1406 DepSI->getOperand(0),
1407 Offset));
1408 continue;
1409 }
1410 }
1411 }
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001412
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001413 // If the clobbering value is a memset/memcpy/memmove, see if we can
1414 // forward a value on from it.
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001415 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(DepInfo.getInst())) {
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001416 if (TD == 0)
1417 TD = getAnalysisIfAvailable<TargetData>();
Chris Lattneraf064ae2009-12-09 18:21:46 +00001418 if (TD && Address) {
1419 int Offset = AnalyzeLoadFromClobberingMemInst(LI->getType(), Address,
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001420 DepMI, *TD);
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001421 if (Offset != -1) {
1422 ValuesPerBlock.push_back(AvailableValueInBlock::getMI(DepBB, DepMI,
1423 Offset));
1424 continue;
1425 }
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001426 }
1427 }
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001428
Chris Lattnerb51deb92008-12-05 21:04:20 +00001429 UnavailableBlocks.push_back(DepBB);
1430 continue;
1431 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001432
Chris Lattnerb51deb92008-12-05 21:04:20 +00001433 Instruction *DepInst = DepInfo.getInst();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001434
Chris Lattnerb51deb92008-12-05 21:04:20 +00001435 // Loading the allocation -> undef.
Chris Lattner720e7902009-12-02 06:44:58 +00001436 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst) ||
Owen Anderson9ff5a232009-12-02 07:35:19 +00001437 // Loading immediately after lifetime begin -> undef.
1438 isLifetimeStart(DepInst)) {
Chris Lattner87913512009-09-21 06:30:24 +00001439 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1440 UndefValue::get(LI->getType())));
Chris Lattnerbf145d62008-12-01 01:15:42 +00001441 continue;
1442 }
Owen Andersonb62f7922009-10-28 07:05:35 +00001443
Chris Lattner87913512009-09-21 06:30:24 +00001444 if (StoreInst *S = dyn_cast<StoreInst>(DepInst)) {
Daniel Dunbara279bc32009-09-20 02:20:51 +00001445 // Reject loads and stores that are to the same address but are of
Chris Lattner771a5422009-09-20 20:09:34 +00001446 // different types if we have to.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001447 if (S->getOperand(0)->getType() != LI->getType()) {
Chris Lattner771a5422009-09-20 20:09:34 +00001448 if (TD == 0)
1449 TD = getAnalysisIfAvailable<TargetData>();
1450
1451 // If the stored value is larger or equal to the loaded value, we can
1452 // reuse it.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001453 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(S->getOperand(0),
1454 LI->getType(), *TD)) {
Chris Lattner771a5422009-09-20 20:09:34 +00001455 UnavailableBlocks.push_back(DepBB);
1456 continue;
1457 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001458 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001459
Chris Lattner87913512009-09-21 06:30:24 +00001460 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1461 S->getOperand(0)));
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001462 continue;
1463 }
1464
1465 if (LoadInst *LD = dyn_cast<LoadInst>(DepInst)) {
Chris Lattner771a5422009-09-20 20:09:34 +00001466 // If the types mismatch and we can't handle it, reject reuse of the load.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001467 if (LD->getType() != LI->getType()) {
Chris Lattner771a5422009-09-20 20:09:34 +00001468 if (TD == 0)
1469 TD = getAnalysisIfAvailable<TargetData>();
1470
1471 // If the stored value is larger or equal to the loaded value, we can
1472 // reuse it.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001473 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(LD, LI->getType(),*TD)){
Chris Lattner771a5422009-09-20 20:09:34 +00001474 UnavailableBlocks.push_back(DepBB);
1475 continue;
1476 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001477 }
Chris Lattner87913512009-09-21 06:30:24 +00001478 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB, LD));
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001479 continue;
Owen Anderson0cd32032007-07-25 19:57:03 +00001480 }
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001481
1482 UnavailableBlocks.push_back(DepBB);
1483 continue;
Chris Lattner88365bb2008-03-21 21:14:38 +00001484 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001485
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001486 // If we have no predecessors that produce a known value for this load, exit
1487 // early.
1488 if (ValuesPerBlock.empty()) return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001489
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001490 // If all of the instructions we depend on produce a known value for this
1491 // load, then it is fully redundant and we can use PHI insertion to compute
1492 // its value. Insert PHIs and remove the fully redundant value now.
1493 if (UnavailableBlocks.empty()) {
Dan Gohman2a298992009-07-31 20:24:18 +00001494 DEBUG(errs() << "GVN REMOVING NONLOCAL LOAD: " << *LI << '\n');
Chris Lattner771a5422009-09-20 20:09:34 +00001495
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001496 // Perform PHI construction.
Chris Lattnera09fbf02009-10-10 23:50:30 +00001497 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD,
1498 VN.getAliasAnalysis());
Chris Lattner771a5422009-09-20 20:09:34 +00001499 LI->replaceAllUsesWith(V);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001500
Chris Lattner771a5422009-09-20 20:09:34 +00001501 if (isa<PHINode>(V))
1502 V->takeName(LI);
1503 if (isa<PointerType>(V->getType()))
1504 MD->invalidateCachedPointerInfo(V);
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001505 toErase.push_back(LI);
1506 NumGVNLoad++;
1507 return true;
1508 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001509
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001510 if (!EnablePRE || !EnableLoadPRE)
1511 return false;
1512
1513 // Okay, we have *some* definitions of the value. This means that the value
1514 // is available in some of our (transitive) predecessors. Lets think about
1515 // doing PRE of this load. This will involve inserting a new load into the
1516 // predecessor when it's not available. We could do this in general, but
1517 // prefer to not increase code size. As such, we only do this when we know
1518 // that we only have to insert *one* load (which means we're basically moving
1519 // the load, not inserting a new one).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001520
Owen Anderson88554df2009-05-31 09:03:40 +00001521 SmallPtrSet<BasicBlock *, 4> Blockers;
1522 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1523 Blockers.insert(UnavailableBlocks[i]);
1524
1525 // Lets find first basic block with more than one predecessor. Walk backwards
1526 // through predecessors if needed.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001527 BasicBlock *LoadBB = LI->getParent();
Owen Anderson88554df2009-05-31 09:03:40 +00001528 BasicBlock *TmpBB = LoadBB;
1529
1530 bool isSinglePred = false;
Dale Johannesen42c3f552009-06-17 20:48:23 +00001531 bool allSingleSucc = true;
Owen Anderson88554df2009-05-31 09:03:40 +00001532 while (TmpBB->getSinglePredecessor()) {
1533 isSinglePred = true;
1534 TmpBB = TmpBB->getSinglePredecessor();
1535 if (!TmpBB) // If haven't found any, bail now.
1536 return false;
1537 if (TmpBB == LoadBB) // Infinite (unreachable) loop.
1538 return false;
1539 if (Blockers.count(TmpBB))
1540 return false;
Dale Johannesen42c3f552009-06-17 20:48:23 +00001541 if (TmpBB->getTerminator()->getNumSuccessors() != 1)
1542 allSingleSucc = false;
Owen Anderson88554df2009-05-31 09:03:40 +00001543 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001544
Owen Anderson88554df2009-05-31 09:03:40 +00001545 assert(TmpBB);
1546 LoadBB = TmpBB;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001547
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001548 // If we have a repl set with LI itself in it, this means we have a loop where
1549 // at least one of the values is LI. Since this means that we won't be able
1550 // to eliminate LI even if we insert uses in the other predecessors, we will
1551 // end up increasing code size. Reject this by scanning for LI.
1552 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001553 if (ValuesPerBlock[i].isSimpleValue() &&
1554 ValuesPerBlock[i].getSimpleValue() == LI)
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001555 return false;
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
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001579 // Okay, we have some hope :). Check to see if the loaded value is fully
1580 // available in all but one predecessor.
1581 // FIXME: If we could restructure the CFG, we could make a common pred with
1582 // all the preds that don't have an available LI and insert a new load into
1583 // that one block.
1584 BasicBlock *UnavailablePred = 0;
1585
Chris Lattner72bc70d2008-12-05 07:49:08 +00001586 DenseMap<BasicBlock*, char> FullyAvailableBlocks;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001587 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner87913512009-09-21 06:30:24 +00001588 FullyAvailableBlocks[ValuesPerBlock[i].BB] = true;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001589 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1590 FullyAvailableBlocks[UnavailableBlocks[i]] = false;
1591
1592 for (pred_iterator PI = pred_begin(LoadBB), E = pred_end(LoadBB);
1593 PI != E; ++PI) {
1594 if (IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
1595 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001596
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001597 // If this load is not available in multiple predecessors, reject it.
1598 if (UnavailablePred && UnavailablePred != *PI)
1599 return false;
1600 UnavailablePred = *PI;
1601 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001602
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001603 assert(UnavailablePred != 0 &&
1604 "Fully available value should be eliminated above!");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001605
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001606 // We don't currently handle critical edges :(
1607 if (UnavailablePred->getTerminator()->getNumSuccessors() != 1) {
Daniel Dunbarce63ffb2009-07-25 00:23:56 +00001608 DEBUG(errs() << "COULD NOT PRE LOAD BECAUSE OF CRITICAL EDGE '"
Dan Gohman2a298992009-07-31 20:24:18 +00001609 << UnavailablePred->getName() << "': " << *LI << '\n');
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001610 return false;
Owen Andersona37226a2007-08-07 23:12:31 +00001611 }
Chris Lattner616613d2009-11-27 08:25:10 +00001612
Chris Lattner6f7b2102009-11-27 22:05:15 +00001613 // Do PHI translation to get its value in the predecessor if necessary. The
1614 // returned pointer (if non-null) is guaranteed to dominate UnavailablePred.
1615 //
Chris Lattnerdd696052009-11-28 15:39:14 +00001616 SmallVector<Instruction*, 8> NewInsts;
Chris Lattner971fd572009-11-27 22:50:07 +00001617
Chris Lattner0c264b12009-11-28 16:08:18 +00001618 // If all preds have a single successor, then we know it is safe to insert the
1619 // load on the pred (?!?), so we can insert code to materialize the pointer if
1620 // it is not available.
Chris Lattner05e15f82009-12-09 01:59:31 +00001621 PHITransAddr Address(LI->getOperand(0), TD);
1622 Value *LoadPtr = 0;
Chris Lattner0c264b12009-11-28 16:08:18 +00001623 if (allSingleSucc) {
Chris Lattner05e15f82009-12-09 01:59:31 +00001624 LoadPtr = Address.PHITranslateWithInsertion(LoadBB, UnavailablePred,
1625 *DT, NewInsts);
Chris Lattner0c264b12009-11-28 16:08:18 +00001626 } else {
Chris Lattner05e15f82009-12-09 01:59:31 +00001627 Address.PHITranslateValue(LoadBB, UnavailablePred);
1628 LoadPtr = Address.getAddr();
1629
1630 // Make sure the value is live in the predecessor.
1631 if (Instruction *Inst = dyn_cast_or_null<Instruction>(LoadPtr))
1632 if (!DT->dominates(Inst->getParent(), UnavailablePred))
1633 LoadPtr = 0;
1634 }
1635
1636 // If we couldn't find or insert a computation of this phi translated value,
1637 // we fail PRE.
1638 if (LoadPtr == 0) {
1639 assert(NewInsts.empty() && "Shouldn't insert insts on failure");
1640 DEBUG(errs() << "COULDN'T INSERT PHI TRANSLATED VALUE OF: "
1641 << *LI->getOperand(0) << "\n");
1642 return false;
Chris Lattner0c264b12009-11-28 16:08:18 +00001643 }
Owen Andersonc9f20272009-12-03 03:43:29 +00001644
1645 // Assign value numbers to these new instructions.
Chris Lattner05e15f82009-12-09 01:59:31 +00001646 for (unsigned i = 0, e = NewInsts.size(); i != e; ++i) {
Owen Andersonc9f20272009-12-03 03:43:29 +00001647 // FIXME: We really _ought_ to insert these value numbers into their
1648 // parent's availability map. However, in doing so, we risk getting into
1649 // ordering issues. If a block hasn't been processed yet, we would be
1650 // marking a value as AVAIL-IN, which isn't what we intend.
Chris Lattner05e15f82009-12-09 01:59:31 +00001651 VN.lookup_or_add(NewInsts[i]);
Chris Lattner616613d2009-11-27 08:25:10 +00001652 }
1653
Dale Johannesen42c3f552009-06-17 20:48:23 +00001654 // Make sure it is valid to move this load here. We have to watch out for:
1655 // @1 = getelementptr (i8* p, ...
1656 // test p and branch if == 0
1657 // load @1
1658 // It is valid to have the getelementptr before the test, even if p can be 0,
1659 // as getelementptr only does address arithmetic.
1660 // If we are not pushing the value through any multiple-successor blocks
1661 // we do not have this case. Otherwise, check that the load is safe to
1662 // put anywhere; this can be improved, but should be conservatively safe.
1663 if (!allSingleSucc &&
Chris Lattnerdd696052009-11-28 15:39:14 +00001664 // FIXME: REEVALUTE THIS.
Chris Lattner0c264b12009-11-28 16:08:18 +00001665 !isSafeToLoadUnconditionally(LoadPtr, UnavailablePred->getTerminator())) {
1666 assert(NewInsts.empty() && "Should not have inserted instructions");
Dale Johannesen42c3f552009-06-17 20:48:23 +00001667 return false;
Chris Lattner0c264b12009-11-28 16:08:18 +00001668 }
Dale Johannesen42c3f552009-06-17 20:48:23 +00001669
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001670 // Okay, we can eliminate this load by inserting a reload in the predecessor
1671 // and using PHI construction to get the value in the other predecessors, do
1672 // it.
Dan Gohman2a298992009-07-31 20:24:18 +00001673 DEBUG(errs() << "GVN REMOVING PRE LOAD: " << *LI << '\n');
Chris Lattner0c264b12009-11-28 16:08:18 +00001674 DEBUG(if (!NewInsts.empty())
1675 errs() << "INSERTED " << NewInsts.size() << " INSTS: "
1676 << *NewInsts.back() << '\n');
1677
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001678 Value *NewLoad = new LoadInst(LoadPtr, LI->getName()+".pre", false,
1679 LI->getAlignment(),
1680 UnavailablePred->getTerminator());
Daniel Dunbara279bc32009-09-20 02:20:51 +00001681
Chris Lattnera09fbf02009-10-10 23:50:30 +00001682 // Add the newly created load.
1683 ValuesPerBlock.push_back(AvailableValueInBlock::get(UnavailablePred,NewLoad));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001684
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001685 // Perform PHI construction.
Chris Lattnera09fbf02009-10-10 23:50:30 +00001686 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD,
1687 VN.getAliasAnalysis());
Chris Lattner771a5422009-09-20 20:09:34 +00001688 LI->replaceAllUsesWith(V);
1689 if (isa<PHINode>(V))
1690 V->takeName(LI);
1691 if (isa<PointerType>(V->getType()))
1692 MD->invalidateCachedPointerInfo(V);
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001693 toErase.push_back(LI);
1694 NumPRELoad++;
Owen Anderson0cd32032007-07-25 19:57:03 +00001695 return true;
1696}
1697
Owen Anderson62bc33c2007-08-16 22:02:55 +00001698/// processLoad - Attempt to eliminate a load, first by eliminating it
1699/// locally, and then attempting non-local elimination if that fails.
Chris Lattnerb51deb92008-12-05 21:04:20 +00001700bool GVN::processLoad(LoadInst *L, SmallVectorImpl<Instruction*> &toErase) {
Dan Gohman4ec01b22009-11-14 02:27:51 +00001701 if (!MD)
1702 return false;
1703
Chris Lattnerb51deb92008-12-05 21:04:20 +00001704 if (L->isVolatile())
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001705 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001706
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001707 // ... to a pointer that has been loaded from before...
Chris Lattnerb2412a82009-09-21 02:42:51 +00001708 MemDepResult Dep = MD->getDependency(L);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001709
Chris Lattnerb51deb92008-12-05 21:04:20 +00001710 // If the value isn't available, don't do anything!
Chris Lattnerb2412a82009-09-21 02:42:51 +00001711 if (Dep.isClobber()) {
Chris Lattnereed919b2009-09-21 05:57:11 +00001712 // Check to see if we have something like this:
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001713 // store i32 123, i32* %P
1714 // %A = bitcast i32* %P to i8*
1715 // %B = gep i8* %A, i32 1
1716 // %C = load i8* %B
1717 //
1718 // We could do that by recognizing if the clobber instructions are obviously
1719 // a common base + constant offset, and if the previous store (or memset)
1720 // completely covers this load. This sort of thing can happen in bitfield
1721 // access code.
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001722 Value *AvailVal = 0;
Chris Lattnereed919b2009-09-21 05:57:11 +00001723 if (StoreInst *DepSI = dyn_cast<StoreInst>(Dep.getInst()))
Chris Lattner1ce08292009-09-21 06:22:46 +00001724 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001725 int Offset = AnalyzeLoadFromClobberingStore(L->getType(),
1726 L->getPointerOperand(),
1727 DepSI, *TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001728 if (Offset != -1)
1729 AvailVal = GetStoreValueForLoad(DepSI->getOperand(0), Offset,
1730 L->getType(), L, *TD);
Chris Lattner1ce08292009-09-21 06:22:46 +00001731 }
Chris Lattnereed919b2009-09-21 05:57:11 +00001732
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001733 // If the clobbering value is a memset/memcpy/memmove, see if we can forward
1734 // a value on from it.
1735 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(Dep.getInst())) {
1736 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001737 int Offset = AnalyzeLoadFromClobberingMemInst(L->getType(),
1738 L->getPointerOperand(),
1739 DepMI, *TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001740 if (Offset != -1)
1741 AvailVal = GetMemInstValueForLoad(DepMI, Offset, L->getType(), L,*TD);
1742 }
1743 }
1744
1745 if (AvailVal) {
1746 DEBUG(errs() << "GVN COERCED INST:\n" << *Dep.getInst() << '\n'
1747 << *AvailVal << '\n' << *L << "\n\n\n");
1748
1749 // Replace the load!
1750 L->replaceAllUsesWith(AvailVal);
1751 if (isa<PointerType>(AvailVal->getType()))
1752 MD->invalidateCachedPointerInfo(AvailVal);
1753 toErase.push_back(L);
1754 NumGVNLoad++;
1755 return true;
1756 }
1757
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001758 DEBUG(
1759 // fast print dep, using operator<< on instruction would be too slow
Dan Gohmanfd87a542009-07-25 01:43:01 +00001760 errs() << "GVN: load ";
1761 WriteAsOperand(errs(), L);
Chris Lattnerb2412a82009-09-21 02:42:51 +00001762 Instruction *I = Dep.getInst();
Dan Gohman2a298992009-07-31 20:24:18 +00001763 errs() << " is clobbered by " << *I << '\n';
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001764 );
Chris Lattnerb51deb92008-12-05 21:04:20 +00001765 return false;
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001766 }
Chris Lattnerb51deb92008-12-05 21:04:20 +00001767
1768 // If it is defined in another block, try harder.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001769 if (Dep.isNonLocal())
Chris Lattnerb51deb92008-12-05 21:04:20 +00001770 return processNonLocalLoad(L, toErase);
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001771
Chris Lattnerb2412a82009-09-21 02:42:51 +00001772 Instruction *DepInst = Dep.getInst();
Chris Lattnerb51deb92008-12-05 21:04:20 +00001773 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInst)) {
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001774 Value *StoredVal = DepSI->getOperand(0);
1775
1776 // The store and load are to a must-aliased pointer, but they may not
1777 // actually have the same type. See if we know how to reuse the stored
1778 // value (depending on its type).
1779 const TargetData *TD = 0;
Chris Lattnera52fce42009-10-21 04:11:19 +00001780 if (StoredVal->getType() != L->getType()) {
1781 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1782 StoredVal = CoerceAvailableValueToLoadType(StoredVal, L->getType(),
1783 L, *TD);
1784 if (StoredVal == 0)
1785 return false;
1786
1787 DEBUG(errs() << "GVN COERCED STORE:\n" << *DepSI << '\n' << *StoredVal
1788 << '\n' << *L << "\n\n\n");
1789 }
1790 else
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001791 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001792 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001793
Chris Lattnerb51deb92008-12-05 21:04:20 +00001794 // Remove it!
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001795 L->replaceAllUsesWith(StoredVal);
1796 if (isa<PointerType>(StoredVal->getType()))
1797 MD->invalidateCachedPointerInfo(StoredVal);
Chris Lattnerb51deb92008-12-05 21:04:20 +00001798 toErase.push_back(L);
1799 NumGVNLoad++;
1800 return true;
1801 }
1802
1803 if (LoadInst *DepLI = dyn_cast<LoadInst>(DepInst)) {
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001804 Value *AvailableVal = DepLI;
1805
1806 // The loads are of a must-aliased pointer, but they may not actually have
1807 // the same type. See if we know how to reuse the previously loaded value
1808 // (depending on its type).
1809 const TargetData *TD = 0;
Chris Lattnera52fce42009-10-21 04:11:19 +00001810 if (DepLI->getType() != L->getType()) {
1811 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1812 AvailableVal = CoerceAvailableValueToLoadType(DepLI, L->getType(), L,*TD);
1813 if (AvailableVal == 0)
1814 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001815
Chris Lattnera52fce42009-10-21 04:11:19 +00001816 DEBUG(errs() << "GVN COERCED LOAD:\n" << *DepLI << "\n" << *AvailableVal
1817 << "\n" << *L << "\n\n\n");
1818 }
1819 else
1820 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001821 }
1822
Chris Lattnerb51deb92008-12-05 21:04:20 +00001823 // Remove it!
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001824 L->replaceAllUsesWith(AvailableVal);
Chris Lattnerbc99be12008-12-09 22:06:23 +00001825 if (isa<PointerType>(DepLI->getType()))
1826 MD->invalidateCachedPointerInfo(DepLI);
Chris Lattnerb51deb92008-12-05 21:04:20 +00001827 toErase.push_back(L);
1828 NumGVNLoad++;
1829 return true;
1830 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001831
Chris Lattner237a8282008-11-30 01:39:32 +00001832 // If this load really doesn't depend on anything, then we must be loading an
1833 // undef value. This can happen when loading for a fresh allocation with no
1834 // intervening stores, for example.
Victor Hernandez7b929da2009-10-23 21:09:37 +00001835 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst)) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001836 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Chris Lattner237a8282008-11-30 01:39:32 +00001837 toErase.push_back(L);
Chris Lattner237a8282008-11-30 01:39:32 +00001838 NumGVNLoad++;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001839 return true;
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001840 }
Owen Andersonb62f7922009-10-28 07:05:35 +00001841
Owen Anderson9ff5a232009-12-02 07:35:19 +00001842 // If this load occurs either right after a lifetime begin,
Owen Andersonb62f7922009-10-28 07:05:35 +00001843 // then the loaded value is undefined.
1844 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(DepInst)) {
Owen Anderson9ff5a232009-12-02 07:35:19 +00001845 if (II->getIntrinsicID() == Intrinsic::lifetime_start) {
Owen Andersonb62f7922009-10-28 07:05:35 +00001846 L->replaceAllUsesWith(UndefValue::get(L->getType()));
1847 toErase.push_back(L);
1848 NumGVNLoad++;
1849 return true;
1850 }
1851 }
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001852
Chris Lattnerb51deb92008-12-05 21:04:20 +00001853 return false;
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001854}
1855
Chris Lattnerb2412a82009-09-21 02:42:51 +00001856Value *GVN::lookupNumber(BasicBlock *BB, uint32_t num) {
Owen Andersonb70a5712008-06-23 17:49:45 +00001857 DenseMap<BasicBlock*, ValueNumberScope*>::iterator I = localAvail.find(BB);
1858 if (I == localAvail.end())
1859 return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001860
Chris Lattnerb2412a82009-09-21 02:42:51 +00001861 ValueNumberScope *Locals = I->second;
1862 while (Locals) {
1863 DenseMap<uint32_t, Value*>::iterator I = Locals->table.find(num);
1864 if (I != Locals->table.end())
Owen Anderson6fafe842008-06-20 01:15:47 +00001865 return I->second;
Chris Lattnerb2412a82009-09-21 02:42:51 +00001866 Locals = Locals->parent;
Owen Anderson6fafe842008-06-20 01:15:47 +00001867 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001868
Owen Anderson6fafe842008-06-20 01:15:47 +00001869 return 0;
1870}
1871
Owen Anderson255dafc2008-12-15 02:03:00 +00001872
Owen Anderson36057c72007-08-14 18:16:29 +00001873/// processInstruction - When calculating availability, handle an instruction
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001874/// by inserting it into the appropriate sets
Owen Andersonaf4240a2008-06-12 19:25:32 +00001875bool GVN::processInstruction(Instruction *I,
Chris Lattner8e1e95c2008-03-21 22:01:16 +00001876 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001877 if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
1878 bool Changed = processLoad(LI, toErase);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001879
Chris Lattnerb2412a82009-09-21 02:42:51 +00001880 if (!Changed) {
1881 unsigned Num = VN.lookup_or_add(LI);
1882 localAvail[I->getParent()]->table.insert(std::make_pair(Num, LI));
Owen Andersonb2303722008-06-18 21:41:49 +00001883 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001884
Chris Lattnerb2412a82009-09-21 02:42:51 +00001885 return Changed;
Owen Andersonb2303722008-06-18 21:41:49 +00001886 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001887
Chris Lattnerb2412a82009-09-21 02:42:51 +00001888 uint32_t NextNum = VN.getNextUnusedValueNumber();
1889 unsigned Num = VN.lookup_or_add(I);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001890
Chris Lattnerb2412a82009-09-21 02:42:51 +00001891 if (BranchInst *BI = dyn_cast<BranchInst>(I)) {
1892 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001893
Owen Andersone8a290f2009-04-01 23:53:49 +00001894 if (!BI->isConditional() || isa<Constant>(BI->getCondition()))
1895 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001896
Chris Lattnerb2412a82009-09-21 02:42:51 +00001897 Value *BranchCond = BI->getCondition();
1898 uint32_t CondVN = VN.lookup_or_add(BranchCond);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001899
Chris Lattnerb2412a82009-09-21 02:42:51 +00001900 BasicBlock *TrueSucc = BI->getSuccessor(0);
1901 BasicBlock *FalseSucc = BI->getSuccessor(1);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001902
Chris Lattnerb2412a82009-09-21 02:42:51 +00001903 if (TrueSucc->getSinglePredecessor())
1904 localAvail[TrueSucc]->table[CondVN] =
1905 ConstantInt::getTrue(TrueSucc->getContext());
1906 if (FalseSucc->getSinglePredecessor())
1907 localAvail[FalseSucc]->table[CondVN] =
1908 ConstantInt::getFalse(TrueSucc->getContext());
Owen Andersone8a290f2009-04-01 23:53:49 +00001909
1910 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001911
Owen Andersone5ffa902008-04-07 09:59:07 +00001912 // Allocations are always uniquely numbered, so we can save time and memory
Daniel Dunbara279bc32009-09-20 02:20:51 +00001913 // by fast failing them.
Victor Hernandez7b929da2009-10-23 21:09:37 +00001914 } else if (isa<AllocaInst>(I) || isa<TerminatorInst>(I)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001915 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Andersone5ffa902008-04-07 09:59:07 +00001916 return false;
Owen Andersonb2303722008-06-18 21:41:49 +00001917 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001918
Owen Anderson62bc33c2007-08-16 22:02:55 +00001919 // Collapse PHI nodes
Owen Anderson31f49672007-08-14 18:33:27 +00001920 if (PHINode* p = dyn_cast<PHINode>(I)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001921 Value *constVal = CollapsePhi(p);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001922
Owen Anderson31f49672007-08-14 18:33:27 +00001923 if (constVal) {
Owen Anderson1defe2d2007-08-16 22:51:56 +00001924 p->replaceAllUsesWith(constVal);
Dan Gohman4ec01b22009-11-14 02:27:51 +00001925 if (MD && isa<PointerType>(constVal->getType()))
Chris Lattnerbc99be12008-12-09 22:06:23 +00001926 MD->invalidateCachedPointerInfo(constVal);
Owen Andersonae53c932008-12-23 00:49:51 +00001927 VN.erase(p);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001928
Owen Anderson1defe2d2007-08-16 22:51:56 +00001929 toErase.push_back(p);
Owen Andersonb2303722008-06-18 21:41:49 +00001930 } else {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001931 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson31f49672007-08-14 18:33:27 +00001932 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001933
Owen Anderson0ae33ef2008-07-03 17:44:33 +00001934 // If the number we were assigned was a brand new VN, then we don't
1935 // need to do a lookup to see if the number already exists
1936 // somewhere in the domtree: it can't!
Chris Lattnerb2412a82009-09-21 02:42:51 +00001937 } else if (Num == NextNum) {
1938 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001939
Owen Anderson255dafc2008-12-15 02:03:00 +00001940 // Perform fast-path value-number based elimination of values inherited from
1941 // dominators.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001942 } else if (Value *repl = lookupNumber(I->getParent(), Num)) {
Owen Anderson5fc4aba2007-12-08 01:37:09 +00001943 // Remove it!
Owen Andersonbf7d0bc2007-07-31 23:27:13 +00001944 VN.erase(I);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001945 I->replaceAllUsesWith(repl);
Dan Gohman4ec01b22009-11-14 02:27:51 +00001946 if (MD && isa<PointerType>(repl->getType()))
Chris Lattnerbc99be12008-12-09 22:06:23 +00001947 MD->invalidateCachedPointerInfo(repl);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001948 toErase.push_back(I);
1949 return true;
Owen Anderson255dafc2008-12-15 02:03:00 +00001950
Owen Anderson0ae33ef2008-07-03 17:44:33 +00001951 } else {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001952 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001953 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001954
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001955 return false;
1956}
1957
Bill Wendling30788b82008-12-22 22:32:22 +00001958/// runOnFunction - This is the main transformation entry point for a function.
Owen Anderson3e75a422007-08-14 18:04:11 +00001959bool GVN::runOnFunction(Function& F) {
Dan Gohman4ec01b22009-11-14 02:27:51 +00001960 if (!NoLoads)
1961 MD = &getAnalysis<MemoryDependenceAnalysis>();
Chris Lattner663e4412008-12-01 00:40:32 +00001962 DT = &getAnalysis<DominatorTree>();
Owen Andersona472c4a2008-05-12 20:15:55 +00001963 VN.setAliasAnalysis(&getAnalysis<AliasAnalysis>());
Chris Lattner663e4412008-12-01 00:40:32 +00001964 VN.setMemDep(MD);
1965 VN.setDomTree(DT);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001966
Chris Lattnerb2412a82009-09-21 02:42:51 +00001967 bool Changed = false;
1968 bool ShouldContinue = true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001969
Owen Anderson5d0af032008-07-16 17:52:31 +00001970 // Merge unconditional branches, allowing PRE to catch more
1971 // optimization opportunities.
1972 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00001973 BasicBlock *BB = FI;
Owen Anderson5d0af032008-07-16 17:52:31 +00001974 ++FI;
Owen Andersonb31b06d2008-07-17 00:01:40 +00001975 bool removedBlock = MergeBlockIntoPredecessor(BB, this);
1976 if (removedBlock) NumGVNBlocks++;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001977
Chris Lattnerb2412a82009-09-21 02:42:51 +00001978 Changed |= removedBlock;
Owen Anderson5d0af032008-07-16 17:52:31 +00001979 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001980
Chris Lattnerae199312008-12-09 19:21:47 +00001981 unsigned Iteration = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001982
Chris Lattnerb2412a82009-09-21 02:42:51 +00001983 while (ShouldContinue) {
Dan Gohmanfd87a542009-07-25 01:43:01 +00001984 DEBUG(errs() << "GVN iteration: " << Iteration << "\n");
Chris Lattnerb2412a82009-09-21 02:42:51 +00001985 ShouldContinue = iterateOnFunction(F);
1986 Changed |= ShouldContinue;
Chris Lattnerae199312008-12-09 19:21:47 +00001987 ++Iteration;
Owen Anderson3e75a422007-08-14 18:04:11 +00001988 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001989
Owen Andersone98c54c2008-07-18 18:03:38 +00001990 if (EnablePRE) {
Owen Anderson0c7f91c2008-09-03 23:06:07 +00001991 bool PREChanged = true;
1992 while (PREChanged) {
1993 PREChanged = performPRE(F);
Chris Lattnerb2412a82009-09-21 02:42:51 +00001994 Changed |= PREChanged;
Owen Anderson0c7f91c2008-09-03 23:06:07 +00001995 }
Owen Andersone98c54c2008-07-18 18:03:38 +00001996 }
Chris Lattnerae199312008-12-09 19:21:47 +00001997 // FIXME: Should perform GVN again after PRE does something. PRE can move
1998 // computations into blocks where they become fully redundant. Note that
1999 // we can't do this until PRE's critical edge splitting updates memdep.
2000 // Actually, when this happens, we should just fully integrate PRE into GVN.
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002001
2002 cleanupGlobalSets();
2003
Chris Lattnerb2412a82009-09-21 02:42:51 +00002004 return Changed;
Owen Anderson3e75a422007-08-14 18:04:11 +00002005}
2006
2007
Chris Lattnerb2412a82009-09-21 02:42:51 +00002008bool GVN::processBlock(BasicBlock *BB) {
Chris Lattnerae199312008-12-09 19:21:47 +00002009 // FIXME: Kill off toErase by doing erasing eagerly in a helper function (and
2010 // incrementing BI before processing an instruction).
Owen Andersonaf4240a2008-06-12 19:25:32 +00002011 SmallVector<Instruction*, 8> toErase;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002012 bool ChangedFunction = false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002013
Owen Andersonaf4240a2008-06-12 19:25:32 +00002014 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
2015 BI != BE;) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002016 ChangedFunction |= processInstruction(BI, toErase);
Owen Andersonaf4240a2008-06-12 19:25:32 +00002017 if (toErase.empty()) {
2018 ++BI;
2019 continue;
2020 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002021
Owen Andersonaf4240a2008-06-12 19:25:32 +00002022 // If we need some instructions deleted, do it now.
2023 NumGVNInstr += toErase.size();
Daniel Dunbara279bc32009-09-20 02:20:51 +00002024
Owen Andersonaf4240a2008-06-12 19:25:32 +00002025 // Avoid iterator invalidation.
2026 bool AtStart = BI == BB->begin();
2027 if (!AtStart)
2028 --BI;
2029
2030 for (SmallVector<Instruction*, 4>::iterator I = toErase.begin(),
Chris Lattner663e4412008-12-01 00:40:32 +00002031 E = toErase.end(); I != E; ++I) {
Dan Gohman2a298992009-07-31 20:24:18 +00002032 DEBUG(errs() << "GVN removed: " << **I << '\n');
Dan Gohman4ec01b22009-11-14 02:27:51 +00002033 if (MD) MD->removeInstruction(*I);
Owen Andersonaf4240a2008-06-12 19:25:32 +00002034 (*I)->eraseFromParent();
Bill Wendlingec40d502008-12-22 21:57:30 +00002035 DEBUG(verifyRemoved(*I));
Chris Lattner663e4412008-12-01 00:40:32 +00002036 }
Chris Lattnerae199312008-12-09 19:21:47 +00002037 toErase.clear();
Owen Andersonaf4240a2008-06-12 19:25:32 +00002038
2039 if (AtStart)
2040 BI = BB->begin();
2041 else
2042 ++BI;
Owen Andersonaf4240a2008-06-12 19:25:32 +00002043 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002044
Chris Lattnerb2412a82009-09-21 02:42:51 +00002045 return ChangedFunction;
Owen Andersonaf4240a2008-06-12 19:25:32 +00002046}
2047
Owen Andersonb2303722008-06-18 21:41:49 +00002048/// performPRE - Perform a purely local form of PRE that looks for diamond
2049/// control flow patterns and attempts to perform simple PRE at the join point.
Chris Lattnerfb6e7012009-10-31 22:11:15 +00002050bool GVN::performPRE(Function &F) {
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002051 bool Changed = false;
Owen Anderson5c274ee2008-06-19 19:54:19 +00002052 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> toSplit;
Chris Lattner09713792008-12-01 07:29:03 +00002053 DenseMap<BasicBlock*, Value*> predMap;
Owen Andersonb2303722008-06-18 21:41:49 +00002054 for (df_iterator<BasicBlock*> DI = df_begin(&F.getEntryBlock()),
2055 DE = df_end(&F.getEntryBlock()); DI != DE; ++DI) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002056 BasicBlock *CurrentBlock = *DI;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002057
Owen Andersonb2303722008-06-18 21:41:49 +00002058 // Nothing to PRE in the entry block.
2059 if (CurrentBlock == &F.getEntryBlock()) continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002060
Owen Andersonb2303722008-06-18 21:41:49 +00002061 for (BasicBlock::iterator BI = CurrentBlock->begin(),
2062 BE = CurrentBlock->end(); BI != BE; ) {
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002063 Instruction *CurInst = BI++;
Duncan Sands7af1c782009-05-06 06:49:50 +00002064
Victor Hernandez7b929da2009-10-23 21:09:37 +00002065 if (isa<AllocaInst>(CurInst) ||
Victor Hernandez83d63912009-09-18 22:35:49 +00002066 isa<TerminatorInst>(CurInst) || isa<PHINode>(CurInst) ||
Devang Patel9674d152009-10-14 17:29:00 +00002067 CurInst->getType()->isVoidTy() ||
Duncan Sands7af1c782009-05-06 06:49:50 +00002068 CurInst->mayReadFromMemory() || CurInst->mayHaveSideEffects() ||
John Criswell090c0a22009-03-10 15:04:53 +00002069 isa<DbgInfoIntrinsic>(CurInst))
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002070 continue;
Duncan Sands7af1c782009-05-06 06:49:50 +00002071
Chris Lattnerb2412a82009-09-21 02:42:51 +00002072 uint32_t ValNo = VN.lookup(CurInst);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002073
Owen Andersonb2303722008-06-18 21:41:49 +00002074 // Look for the predecessors for PRE opportunities. We're
2075 // only trying to solve the basic diamond case, where
2076 // a value is computed in the successor and one predecessor,
2077 // but not the other. We also explicitly disallow cases
2078 // where the successor is its own predecessor, because they're
2079 // more complicated to get right.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002080 unsigned NumWith = 0;
2081 unsigned NumWithout = 0;
2082 BasicBlock *PREPred = 0;
Chris Lattner09713792008-12-01 07:29:03 +00002083 predMap.clear();
2084
Owen Andersonb2303722008-06-18 21:41:49 +00002085 for (pred_iterator PI = pred_begin(CurrentBlock),
2086 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
2087 // We're not interested in PRE where the block is its
Owen Anderson6fafe842008-06-20 01:15:47 +00002088 // own predecessor, on in blocks with predecessors
2089 // that are not reachable.
2090 if (*PI == CurrentBlock) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002091 NumWithout = 2;
Owen Anderson6fafe842008-06-20 01:15:47 +00002092 break;
2093 } else if (!localAvail.count(*PI)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002094 NumWithout = 2;
Owen Anderson6fafe842008-06-20 01:15:47 +00002095 break;
2096 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002097
2098 DenseMap<uint32_t, Value*>::iterator predV =
Chris Lattnerb2412a82009-09-21 02:42:51 +00002099 localAvail[*PI]->table.find(ValNo);
Owen Anderson6fafe842008-06-20 01:15:47 +00002100 if (predV == localAvail[*PI]->table.end()) {
Owen Andersonb2303722008-06-18 21:41:49 +00002101 PREPred = *PI;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002102 NumWithout++;
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002103 } else if (predV->second == CurInst) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002104 NumWithout = 2;
Owen Andersonb2303722008-06-18 21:41:49 +00002105 } else {
Owen Anderson6fafe842008-06-20 01:15:47 +00002106 predMap[*PI] = predV->second;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002107 NumWith++;
Owen Andersonb2303722008-06-18 21:41:49 +00002108 }
2109 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002110
Owen Andersonb2303722008-06-18 21:41:49 +00002111 // Don't do PRE when it might increase code size, i.e. when
2112 // we would need to insert instructions in more than one pred.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002113 if (NumWithout != 1 || NumWith == 0)
Owen Andersonb2303722008-06-18 21:41:49 +00002114 continue;
Chris Lattnerfb6e7012009-10-31 22:11:15 +00002115
2116 // Don't do PRE across indirect branch.
2117 if (isa<IndirectBrInst>(PREPred->getTerminator()))
2118 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002119
Owen Anderson5c274ee2008-06-19 19:54:19 +00002120 // We can't do PRE safely on a critical edge, so instead we schedule
2121 // the edge to be split and perform the PRE the next time we iterate
2122 // on the function.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002123 unsigned SuccNum = 0;
Owen Anderson5c274ee2008-06-19 19:54:19 +00002124 for (unsigned i = 0, e = PREPred->getTerminator()->getNumSuccessors();
2125 i != e; ++i)
Owen Anderson0c7f91c2008-09-03 23:06:07 +00002126 if (PREPred->getTerminator()->getSuccessor(i) == CurrentBlock) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002127 SuccNum = i;
Owen Anderson5c274ee2008-06-19 19:54:19 +00002128 break;
2129 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002130
Chris Lattnerb2412a82009-09-21 02:42:51 +00002131 if (isCriticalEdge(PREPred->getTerminator(), SuccNum)) {
2132 toSplit.push_back(std::make_pair(PREPred->getTerminator(), SuccNum));
Owen Anderson5c274ee2008-06-19 19:54:19 +00002133 continue;
2134 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002135
Owen Andersonb2303722008-06-18 21:41:49 +00002136 // Instantiate the expression the in predecessor that lacked it.
2137 // Because we are going top-down through the block, all value numbers
2138 // will be available in the predecessor by the time we need them. Any
2139 // that weren't original present will have been instantiated earlier
2140 // in this loop.
Nick Lewycky67760642009-09-27 07:38:41 +00002141 Instruction *PREInstr = CurInst->clone();
Owen Andersonb2303722008-06-18 21:41:49 +00002142 bool success = true;
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002143 for (unsigned i = 0, e = CurInst->getNumOperands(); i != e; ++i) {
2144 Value *Op = PREInstr->getOperand(i);
2145 if (isa<Argument>(Op) || isa<Constant>(Op) || isa<GlobalValue>(Op))
2146 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002147
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002148 if (Value *V = lookupNumber(PREPred, VN.lookup(Op))) {
2149 PREInstr->setOperand(i, V);
2150 } else {
2151 success = false;
2152 break;
Owen Andersonc45996b2008-07-11 20:05:13 +00002153 }
Owen Andersonb2303722008-06-18 21:41:49 +00002154 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002155
Owen Andersonb2303722008-06-18 21:41:49 +00002156 // Fail out if we encounter an operand that is not available in
Daniel Dunbara279bc32009-09-20 02:20:51 +00002157 // the PRE predecessor. This is typically because of loads which
Owen Andersonb2303722008-06-18 21:41:49 +00002158 // are not value numbered precisely.
2159 if (!success) {
2160 delete PREInstr;
Bill Wendling70ded192008-12-22 22:14:07 +00002161 DEBUG(verifyRemoved(PREInstr));
Owen Andersonb2303722008-06-18 21:41:49 +00002162 continue;
2163 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002164
Owen Andersonb2303722008-06-18 21:41:49 +00002165 PREInstr->insertBefore(PREPred->getTerminator());
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002166 PREInstr->setName(CurInst->getName() + ".pre");
Owen Anderson6fafe842008-06-20 01:15:47 +00002167 predMap[PREPred] = PREInstr;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002168 VN.add(PREInstr, ValNo);
Owen Andersonb2303722008-06-18 21:41:49 +00002169 NumGVNPRE++;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002170
Owen Andersonb2303722008-06-18 21:41:49 +00002171 // Update the availability map to include the new instruction.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002172 localAvail[PREPred]->table.insert(std::make_pair(ValNo, PREInstr));
Daniel Dunbara279bc32009-09-20 02:20:51 +00002173
Owen Andersonb2303722008-06-18 21:41:49 +00002174 // Create a PHI to make the value available in this block.
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002175 PHINode* Phi = PHINode::Create(CurInst->getType(),
2176 CurInst->getName() + ".pre-phi",
Owen Andersonb2303722008-06-18 21:41:49 +00002177 CurrentBlock->begin());
2178 for (pred_iterator PI = pred_begin(CurrentBlock),
2179 PE = pred_end(CurrentBlock); PI != PE; ++PI)
Owen Anderson6fafe842008-06-20 01:15:47 +00002180 Phi->addIncoming(predMap[*PI], *PI);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002181
Chris Lattnerb2412a82009-09-21 02:42:51 +00002182 VN.add(Phi, ValNo);
2183 localAvail[CurrentBlock]->table[ValNo] = Phi;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002184
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002185 CurInst->replaceAllUsesWith(Phi);
Dan Gohman4ec01b22009-11-14 02:27:51 +00002186 if (MD && isa<PointerType>(Phi->getType()))
Chris Lattnerbc99be12008-12-09 22:06:23 +00002187 MD->invalidateCachedPointerInfo(Phi);
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002188 VN.erase(CurInst);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002189
Dan Gohman2a298992009-07-31 20:24:18 +00002190 DEBUG(errs() << "GVN PRE removed: " << *CurInst << '\n');
Dan Gohman4ec01b22009-11-14 02:27:51 +00002191 if (MD) MD->removeInstruction(CurInst);
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002192 CurInst->eraseFromParent();
Bill Wendlingec40d502008-12-22 21:57:30 +00002193 DEBUG(verifyRemoved(CurInst));
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002194 Changed = true;
Owen Andersonb2303722008-06-18 21:41:49 +00002195 }
2196 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002197
Owen Anderson5c274ee2008-06-19 19:54:19 +00002198 for (SmallVector<std::pair<TerminatorInst*, unsigned>, 4>::iterator
Anton Korobeynikov64b53562008-12-05 19:38:49 +00002199 I = toSplit.begin(), E = toSplit.end(); I != E; ++I)
Owen Anderson5c274ee2008-06-19 19:54:19 +00002200 SplitCriticalEdge(I->first, I->second, this);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002201
Anton Korobeynikov64b53562008-12-05 19:38:49 +00002202 return Changed || toSplit.size();
Owen Andersonb2303722008-06-18 21:41:49 +00002203}
2204
Bill Wendling30788b82008-12-22 22:32:22 +00002205/// iterateOnFunction - Executes one iteration of GVN
Owen Anderson3e75a422007-08-14 18:04:11 +00002206bool GVN::iterateOnFunction(Function &F) {
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002207 cleanupGlobalSets();
Chris Lattner2e607012008-03-21 21:33:23 +00002208
Owen Andersone8a290f2009-04-01 23:53:49 +00002209 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2210 DE = df_end(DT->getRootNode()); DI != DE; ++DI) {
2211 if (DI->getIDom())
2212 localAvail[DI->getBlock()] =
2213 new ValueNumberScope(localAvail[DI->getIDom()->getBlock()]);
2214 else
2215 localAvail[DI->getBlock()] = new ValueNumberScope(0);
2216 }
2217
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002218 // Top-down walk of the dominator tree
Chris Lattnerb2412a82009-09-21 02:42:51 +00002219 bool Changed = false;
Owen Andersonc34d1122008-12-15 03:52:17 +00002220#if 0
2221 // Needed for value numbering with phi construction to work.
Owen Anderson255dafc2008-12-15 02:03:00 +00002222 ReversePostOrderTraversal<Function*> RPOT(&F);
2223 for (ReversePostOrderTraversal<Function*>::rpo_iterator RI = RPOT.begin(),
2224 RE = RPOT.end(); RI != RE; ++RI)
Chris Lattnerb2412a82009-09-21 02:42:51 +00002225 Changed |= processBlock(*RI);
Owen Andersonc34d1122008-12-15 03:52:17 +00002226#else
2227 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2228 DE = df_end(DT->getRootNode()); DI != DE; ++DI)
Chris Lattnerb2412a82009-09-21 02:42:51 +00002229 Changed |= processBlock(DI->getBlock());
Owen Andersonc34d1122008-12-15 03:52:17 +00002230#endif
2231
Chris Lattnerb2412a82009-09-21 02:42:51 +00002232 return Changed;
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002233}
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002234
2235void GVN::cleanupGlobalSets() {
2236 VN.clear();
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002237
2238 for (DenseMap<BasicBlock*, ValueNumberScope*>::iterator
2239 I = localAvail.begin(), E = localAvail.end(); I != E; ++I)
2240 delete I->second;
2241 localAvail.clear();
2242}
Bill Wendling246dbbb2008-12-22 21:36:08 +00002243
2244/// verifyRemoved - Verify that the specified instruction does not occur in our
2245/// internal data structures.
Bill Wendling6d463f22008-12-22 22:28:56 +00002246void GVN::verifyRemoved(const Instruction *Inst) const {
2247 VN.verifyRemoved(Inst);
Bill Wendling70ded192008-12-22 22:14:07 +00002248
Bill Wendling6d463f22008-12-22 22:28:56 +00002249 // Walk through the value number scope to make sure the instruction isn't
2250 // ferreted away in it.
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +00002251 for (DenseMap<BasicBlock*, ValueNumberScope*>::const_iterator
Bill Wendling6d463f22008-12-22 22:28:56 +00002252 I = localAvail.begin(), E = localAvail.end(); I != E; ++I) {
2253 const ValueNumberScope *VNS = I->second;
2254
2255 while (VNS) {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +00002256 for (DenseMap<uint32_t, Value*>::const_iterator
Bill Wendling6d463f22008-12-22 22:28:56 +00002257 II = VNS->table.begin(), IE = VNS->table.end(); II != IE; ++II) {
2258 assert(II->second != Inst && "Inst still in value numbering scope!");
2259 }
2260
2261 VNS = VNS->parent;
Bill Wendling70ded192008-12-22 22:14:07 +00002262 }
2263 }
Bill Wendling246dbbb2008-12-22 21:36:08 +00002264}