blob: 6d35db4d6f7c36fd674353c4a8066ab892a21fcc [file] [log] [blame]
Chris Lattner72bc70d2008-12-05 07:49:08 +00001//===- GVN.cpp - Eliminate redundant values and loads ---------------------===//
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Owen Anderson1ad2cb72007-07-24 17:55:58 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This pass performs global value numbering to eliminate fully redundant
11// instructions. It also performs simple dead load elimination.
12//
John Criswell090c0a22009-03-10 15:04:53 +000013// Note that this pass does the value numbering itself; it does not use the
Matthijs Kooijman845f5242008-06-05 07:55:49 +000014// ValueNumbering analysis passes.
15//
Owen Anderson1ad2cb72007-07-24 17:55:58 +000016//===----------------------------------------------------------------------===//
17
18#define DEBUG_TYPE "gvn"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000019#include "llvm/Transforms/Scalar.h"
Owen Anderson0cd32032007-07-25 19:57:03 +000020#include "llvm/BasicBlock.h"
Owen Anderson45537912007-07-26 18:26:51 +000021#include "llvm/Constants.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000022#include "llvm/DerivedTypes.h"
Chris Lattnera53cfd12009-12-28 21:28:46 +000023#include "llvm/GlobalVariable.h"
Owen Anderson45537912007-07-26 18:26:51 +000024#include "llvm/Function.h"
Devang Patelc64bc162009-03-06 02:59:27 +000025#include "llvm/IntrinsicInst.h"
Owen Andersond672ecb2009-07-03 00:17:18 +000026#include "llvm/LLVMContext.h"
Chris Lattnereed919b2009-09-21 05:57:11 +000027#include "llvm/Operator.h"
Owen Anderson45537912007-07-26 18:26:51 +000028#include "llvm/Value.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000029#include "llvm/ADT/DenseMap.h"
30#include "llvm/ADT/DepthFirstIterator.h"
Owen Anderson255dafc2008-12-15 02:03:00 +000031#include "llvm/ADT/PostOrderIterator.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000032#include "llvm/ADT/SmallPtrSet.h"
33#include "llvm/ADT/SmallVector.h"
34#include "llvm/ADT/Statistic.h"
Owen Andersonb388ca92007-10-18 19:39:33 +000035#include "llvm/Analysis/AliasAnalysis.h"
Chris Lattnerbc9a28d2009-12-06 05:29:56 +000036#include "llvm/Analysis/ConstantFolding.h"
37#include "llvm/Analysis/Dominators.h"
Duncan Sands88c3df72010-11-12 21:10:24 +000038#include "llvm/Analysis/InstructionSimplify.h"
Dan Gohmandd9344f2010-05-28 16:19:17 +000039#include "llvm/Analysis/Loads.h"
Victor Hernandezf006b182009-10-27 20:05:49 +000040#include "llvm/Analysis/MemoryBuiltins.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000041#include "llvm/Analysis/MemoryDependenceAnalysis.h"
Chris Lattner05e15f82009-12-09 01:59:31 +000042#include "llvm/Analysis/PHITransAddr.h"
Owen Andersona04a0642010-11-18 18:32:40 +000043#include "llvm/Support/Allocator.h"
Owen Anderson1ad2cb72007-07-24 17:55:58 +000044#include "llvm/Support/CFG.h"
Owen Andersonaa0b6342008-06-19 19:57:25 +000045#include "llvm/Support/CommandLine.h"
Chris Lattner9f8a6a72008-03-29 04:36:18 +000046#include "llvm/Support/Debug.h"
Torok Edwinc25e7582009-07-11 20:10:48 +000047#include "llvm/Support/ErrorHandling.h"
Chris Lattnereed919b2009-09-21 05:57:11 +000048#include "llvm/Support/GetElementPtrTypeIterator.h"
Chris Lattnerfaf815b2009-12-06 01:57:02 +000049#include "llvm/Support/IRBuilder.h"
Daniel Dunbarce63ffb2009-07-25 00:23:56 +000050#include "llvm/Support/raw_ostream.h"
Chris Lattnerbb6495c2009-09-20 19:03:47 +000051#include "llvm/Target/TargetData.h"
Owen Anderson5c274ee2008-06-19 19:54:19 +000052#include "llvm/Transforms/Utils/BasicBlockUtils.h"
Dale Johannesen42c3f552009-06-17 20:48:23 +000053#include "llvm/Transforms/Utils/Local.h"
Chris Lattnera09fbf02009-10-10 23:50:30 +000054#include "llvm/Transforms/Utils/SSAUpdater.h"
Owen Andersona04a0642010-11-18 18:32:40 +000055#include <list>
Owen Anderson1ad2cb72007-07-24 17:55:58 +000056using namespace llvm;
57
Bill Wendling70ded192008-12-22 22:14:07 +000058STATISTIC(NumGVNInstr, "Number of instructions deleted");
59STATISTIC(NumGVNLoad, "Number of loads deleted");
60STATISTIC(NumGVNPRE, "Number of instructions PRE'd");
Owen Anderson961edc82008-07-15 16:28:06 +000061STATISTIC(NumGVNBlocks, "Number of blocks merged");
Bill Wendling70ded192008-12-22 22:14:07 +000062STATISTIC(NumPRELoad, "Number of loads PRE'd");
Chris Lattnerd27290d2008-03-22 04:13:49 +000063
Evan Cheng88d11c02008-06-20 01:01:07 +000064static cl::opt<bool> EnablePRE("enable-pre",
Owen Andersonc2b856e2008-07-17 19:41:00 +000065 cl::init(true), cl::Hidden);
Dan Gohmanc915c952009-06-15 18:30:15 +000066static cl::opt<bool> EnableLoadPRE("enable-load-pre", cl::init(true));
Owen Andersonaa0b6342008-06-19 19:57:25 +000067
Owen Anderson1ad2cb72007-07-24 17:55:58 +000068//===----------------------------------------------------------------------===//
69// ValueTable Class
70//===----------------------------------------------------------------------===//
71
72/// This class holds the mapping between values and value numbers. It is used
73/// as an efficient mechanism to determine the expression-wise equivalence of
74/// two values.
75namespace {
Chris Lattner3e8b6632009-09-02 06:11:42 +000076 struct Expression {
Owen Andersona81e2412010-01-17 19:33:27 +000077 enum ExpressionOpcode {
78 ADD = Instruction::Add,
79 FADD = Instruction::FAdd,
80 SUB = Instruction::Sub,
81 FSUB = Instruction::FSub,
82 MUL = Instruction::Mul,
83 FMUL = Instruction::FMul,
84 UDIV = Instruction::UDiv,
85 SDIV = Instruction::SDiv,
86 FDIV = Instruction::FDiv,
87 UREM = Instruction::URem,
88 SREM = Instruction::SRem,
89 FREM = Instruction::FRem,
90 SHL = Instruction::Shl,
91 LSHR = Instruction::LShr,
92 ASHR = Instruction::AShr,
93 AND = Instruction::And,
94 OR = Instruction::Or,
95 XOR = Instruction::Xor,
96 TRUNC = Instruction::Trunc,
97 ZEXT = Instruction::ZExt,
98 SEXT = Instruction::SExt,
99 FPTOUI = Instruction::FPToUI,
100 FPTOSI = Instruction::FPToSI,
101 UITOFP = Instruction::UIToFP,
102 SITOFP = Instruction::SIToFP,
103 FPTRUNC = Instruction::FPTrunc,
104 FPEXT = Instruction::FPExt,
105 PTRTOINT = Instruction::PtrToInt,
106 INTTOPTR = Instruction::IntToPtr,
107 BITCAST = Instruction::BitCast,
108 ICMPEQ, ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
109 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
110 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
111 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
112 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
113 SHUFFLE, SELECT, GEP, CALL, CONSTANT,
114 INSERTVALUE, EXTRACTVALUE, EMPTY, TOMBSTONE };
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000115
116 ExpressionOpcode opcode;
117 const Type* type;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000118 SmallVector<uint32_t, 4> varargs;
Chris Lattnerb2412a82009-09-21 02:42:51 +0000119 Value *function;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000120
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000121 Expression() { }
122 Expression(ExpressionOpcode o) : opcode(o) { }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000123
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000124 bool operator==(const Expression &other) const {
125 if (opcode != other.opcode)
126 return false;
127 else if (opcode == EMPTY || opcode == TOMBSTONE)
128 return true;
129 else if (type != other.type)
130 return false;
Owen Andersonb388ca92007-10-18 19:39:33 +0000131 else if (function != other.function)
132 return false;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000133 else {
134 if (varargs.size() != other.varargs.size())
135 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000136
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000137 for (size_t i = 0; i < varargs.size(); ++i)
138 if (varargs[i] != other.varargs[i])
139 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000140
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000141 return true;
142 }
143 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000144
Chris Lattner17aa6802010-09-04 18:12:00 +0000145 /*bool operator!=(const Expression &other) const {
Bill Wendling75f02ee2008-12-22 22:16:31 +0000146 return !(*this == other);
Chris Lattner17aa6802010-09-04 18:12:00 +0000147 }*/
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000148 };
Daniel Dunbara279bc32009-09-20 02:20:51 +0000149
Chris Lattner3e8b6632009-09-02 06:11:42 +0000150 class ValueTable {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000151 private:
152 DenseMap<Value*, uint32_t> valueNumbering;
153 DenseMap<Expression, uint32_t> expressionNumbering;
Owen Andersona472c4a2008-05-12 20:15:55 +0000154 AliasAnalysis* AA;
155 MemoryDependenceAnalysis* MD;
156 DominatorTree* DT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000157
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000158 uint32_t nextValueNumber;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000159
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000160 Expression::ExpressionOpcode getOpcode(CmpInst* C);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000161 Expression create_expression(BinaryOperator* BO);
162 Expression create_expression(CmpInst* C);
163 Expression create_expression(ShuffleVectorInst* V);
164 Expression create_expression(ExtractElementInst* C);
165 Expression create_expression(InsertElementInst* V);
166 Expression create_expression(SelectInst* V);
167 Expression create_expression(CastInst* C);
168 Expression create_expression(GetElementPtrInst* G);
Owen Andersonb388ca92007-10-18 19:39:33 +0000169 Expression create_expression(CallInst* C);
Owen Andersond41ed4e2009-10-19 22:14:22 +0000170 Expression create_expression(ExtractValueInst* C);
171 Expression create_expression(InsertValueInst* C);
172
173 uint32_t lookup_or_add_call(CallInst* C);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000174 public:
Dan Gohmane63c4a22009-04-01 16:37:47 +0000175 ValueTable() : nextValueNumber(1) { }
Chris Lattnerb2412a82009-09-21 02:42:51 +0000176 uint32_t lookup_or_add(Value *V);
177 uint32_t lookup(Value *V) const;
178 void add(Value *V, uint32_t num);
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000179 void clear();
Chris Lattnerb2412a82009-09-21 02:42:51 +0000180 void erase(Value *v);
Owen Andersona472c4a2008-05-12 20:15:55 +0000181 void setAliasAnalysis(AliasAnalysis* A) { AA = A; }
Chris Lattner663e4412008-12-01 00:40:32 +0000182 AliasAnalysis *getAliasAnalysis() const { return AA; }
Owen Andersona472c4a2008-05-12 20:15:55 +0000183 void setMemDep(MemoryDependenceAnalysis* M) { MD = M; }
184 void setDomTree(DominatorTree* D) { DT = D; }
Owen Anderson0ae33ef2008-07-03 17:44:33 +0000185 uint32_t getNextUnusedValueNumber() { return nextValueNumber; }
Bill Wendling246dbbb2008-12-22 21:36:08 +0000186 void verifyRemoved(const Value *) const;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000187 };
188}
189
190namespace llvm {
Chris Lattner76c1b972007-09-17 18:34:04 +0000191template <> struct DenseMapInfo<Expression> {
Owen Anderson830db6a2007-08-02 18:16:06 +0000192 static inline Expression getEmptyKey() {
193 return Expression(Expression::EMPTY);
194 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000195
Owen Anderson830db6a2007-08-02 18:16:06 +0000196 static inline Expression getTombstoneKey() {
197 return Expression(Expression::TOMBSTONE);
198 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000199
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000200 static unsigned getHashValue(const Expression e) {
201 unsigned hash = e.opcode;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000202
Anton Korobeynikov07e6e562008-02-20 11:26:25 +0000203 hash = ((unsigned)((uintptr_t)e.type >> 4) ^
Owen Andersond41ed4e2009-10-19 22:14:22 +0000204 (unsigned)((uintptr_t)e.type >> 9));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000205
Owen Anderson830db6a2007-08-02 18:16:06 +0000206 for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
207 E = e.varargs.end(); I != E; ++I)
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000208 hash = *I + hash * 37;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000209
Anton Korobeynikov07e6e562008-02-20 11:26:25 +0000210 hash = ((unsigned)((uintptr_t)e.function >> 4) ^
211 (unsigned)((uintptr_t)e.function >> 9)) +
212 hash * 37;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000213
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000214 return hash;
215 }
Chris Lattner76c1b972007-09-17 18:34:04 +0000216 static bool isEqual(const Expression &LHS, const Expression &RHS) {
217 return LHS == RHS;
218 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000219};
Chris Lattner4bbf4ee2009-12-15 07:26:43 +0000220
221template <>
222struct isPodLike<Expression> { static const bool value = true; };
223
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000224}
225
226//===----------------------------------------------------------------------===//
227// ValueTable Internal Functions
228//===----------------------------------------------------------------------===//
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000229
230Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000231 if (isa<ICmpInst>(C)) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000232 switch (C->getPredicate()) {
Chris Lattner88365bb2008-03-21 21:14:38 +0000233 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000234 llvm_unreachable("Comparison with unknown predicate?");
Chris Lattner88365bb2008-03-21 21:14:38 +0000235 case ICmpInst::ICMP_EQ: return Expression::ICMPEQ;
236 case ICmpInst::ICMP_NE: return Expression::ICMPNE;
237 case ICmpInst::ICMP_UGT: return Expression::ICMPUGT;
238 case ICmpInst::ICMP_UGE: return Expression::ICMPUGE;
239 case ICmpInst::ICMP_ULT: return Expression::ICMPULT;
240 case ICmpInst::ICMP_ULE: return Expression::ICMPULE;
241 case ICmpInst::ICMP_SGT: return Expression::ICMPSGT;
242 case ICmpInst::ICMP_SGE: return Expression::ICMPSGE;
243 case ICmpInst::ICMP_SLT: return Expression::ICMPSLT;
244 case ICmpInst::ICMP_SLE: return Expression::ICMPSLE;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000245 }
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000246 } else {
247 switch (C->getPredicate()) {
248 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinc23197a2009-07-14 16:55:14 +0000249 llvm_unreachable("Comparison with unknown predicate?");
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +0000250 case FCmpInst::FCMP_OEQ: return Expression::FCMPOEQ;
251 case FCmpInst::FCMP_OGT: return Expression::FCMPOGT;
252 case FCmpInst::FCMP_OGE: return Expression::FCMPOGE;
253 case FCmpInst::FCMP_OLT: return Expression::FCMPOLT;
254 case FCmpInst::FCMP_OLE: return Expression::FCMPOLE;
255 case FCmpInst::FCMP_ONE: return Expression::FCMPONE;
256 case FCmpInst::FCMP_ORD: return Expression::FCMPORD;
257 case FCmpInst::FCMP_UNO: return Expression::FCMPUNO;
258 case FCmpInst::FCMP_UEQ: return Expression::FCMPUEQ;
259 case FCmpInst::FCMP_UGT: return Expression::FCMPUGT;
260 case FCmpInst::FCMP_UGE: return Expression::FCMPUGE;
261 case FCmpInst::FCMP_ULT: return Expression::FCMPULT;
262 case FCmpInst::FCMP_ULE: return Expression::FCMPULE;
263 case FCmpInst::FCMP_UNE: return Expression::FCMPUNE;
264 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000265 }
266}
267
Owen Andersonb388ca92007-10-18 19:39:33 +0000268Expression ValueTable::create_expression(CallInst* C) {
269 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000270
Owen Andersonb388ca92007-10-18 19:39:33 +0000271 e.type = C->getType();
Owen Andersonb388ca92007-10-18 19:39:33 +0000272 e.function = C->getCalledFunction();
273 e.opcode = Expression::CALL;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000274
Gabor Greif0a14be02010-06-24 10:04:07 +0000275 CallSite CS(C);
276 for (CallInst::op_iterator I = CS.arg_begin(), E = CS.arg_end();
Owen Andersonb388ca92007-10-18 19:39:33 +0000277 I != E; ++I)
Owen Anderson8f46c782008-04-11 05:11:49 +0000278 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000279
Owen Andersonb388ca92007-10-18 19:39:33 +0000280 return e;
281}
282
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000283Expression ValueTable::create_expression(BinaryOperator* BO) {
284 Expression e;
Owen Andersond41ed4e2009-10-19 22:14:22 +0000285 e.varargs.push_back(lookup_or_add(BO->getOperand(0)));
286 e.varargs.push_back(lookup_or_add(BO->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000287 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000288 e.type = BO->getType();
Owen Andersona81e2412010-01-17 19:33:27 +0000289 e.opcode = static_cast<Expression::ExpressionOpcode>(BO->getOpcode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000290
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000291 return e;
292}
293
294Expression ValueTable::create_expression(CmpInst* C) {
295 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000296
Owen Andersond41ed4e2009-10-19 22:14:22 +0000297 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
298 e.varargs.push_back(lookup_or_add(C->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000299 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000300 e.type = C->getType();
301 e.opcode = getOpcode(C);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000302
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000303 return e;
304}
305
306Expression ValueTable::create_expression(CastInst* C) {
307 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000308
Owen Andersond41ed4e2009-10-19 22:14:22 +0000309 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000310 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000311 e.type = C->getType();
Owen Andersona81e2412010-01-17 19:33:27 +0000312 e.opcode = static_cast<Expression::ExpressionOpcode>(C->getOpcode());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000313
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000314 return e;
315}
316
317Expression ValueTable::create_expression(ShuffleVectorInst* S) {
318 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000319
Owen Andersond41ed4e2009-10-19 22:14:22 +0000320 e.varargs.push_back(lookup_or_add(S->getOperand(0)));
321 e.varargs.push_back(lookup_or_add(S->getOperand(1)));
322 e.varargs.push_back(lookup_or_add(S->getOperand(2)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000323 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000324 e.type = S->getType();
325 e.opcode = Expression::SHUFFLE;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000326
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000327 return e;
328}
329
330Expression ValueTable::create_expression(ExtractElementInst* E) {
331 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000332
Owen Andersond41ed4e2009-10-19 22:14:22 +0000333 e.varargs.push_back(lookup_or_add(E->getOperand(0)));
334 e.varargs.push_back(lookup_or_add(E->getOperand(1)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000335 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000336 e.type = E->getType();
337 e.opcode = Expression::EXTRACT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000338
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000339 return e;
340}
341
342Expression ValueTable::create_expression(InsertElementInst* I) {
343 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000344
Owen Andersond41ed4e2009-10-19 22:14:22 +0000345 e.varargs.push_back(lookup_or_add(I->getOperand(0)));
346 e.varargs.push_back(lookup_or_add(I->getOperand(1)));
347 e.varargs.push_back(lookup_or_add(I->getOperand(2)));
Owen Andersonb388ca92007-10-18 19:39:33 +0000348 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000349 e.type = I->getType();
350 e.opcode = Expression::INSERT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000351
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000352 return e;
353}
354
355Expression ValueTable::create_expression(SelectInst* I) {
356 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000357
Owen Andersond41ed4e2009-10-19 22:14:22 +0000358 e.varargs.push_back(lookup_or_add(I->getCondition()));
359 e.varargs.push_back(lookup_or_add(I->getTrueValue()));
360 e.varargs.push_back(lookup_or_add(I->getFalseValue()));
Owen Andersonb388ca92007-10-18 19:39:33 +0000361 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000362 e.type = I->getType();
363 e.opcode = Expression::SELECT;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000364
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000365 return e;
366}
367
368Expression ValueTable::create_expression(GetElementPtrInst* G) {
369 Expression e;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000370
Owen Andersond41ed4e2009-10-19 22:14:22 +0000371 e.varargs.push_back(lookup_or_add(G->getPointerOperand()));
Owen Andersonb388ca92007-10-18 19:39:33 +0000372 e.function = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000373 e.type = G->getType();
374 e.opcode = Expression::GEP;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000375
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000376 for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
377 I != E; ++I)
Owen Anderson8f46c782008-04-11 05:11:49 +0000378 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000379
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000380 return e;
381}
382
Owen Andersond41ed4e2009-10-19 22:14:22 +0000383Expression ValueTable::create_expression(ExtractValueInst* E) {
384 Expression e;
385
386 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
387 for (ExtractValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
388 II != IE; ++II)
389 e.varargs.push_back(*II);
390 e.function = 0;
391 e.type = E->getType();
392 e.opcode = Expression::EXTRACTVALUE;
393
394 return e;
395}
396
397Expression ValueTable::create_expression(InsertValueInst* E) {
398 Expression e;
399
400 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
401 e.varargs.push_back(lookup_or_add(E->getInsertedValueOperand()));
402 for (InsertValueInst::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::INSERTVALUE;
408
409 return e;
410}
411
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000412//===----------------------------------------------------------------------===//
413// ValueTable External Functions
414//===----------------------------------------------------------------------===//
415
Owen Andersonb2303722008-06-18 21:41:49 +0000416/// add - Insert a value into the table with a specified value number.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000417void ValueTable::add(Value *V, uint32_t num) {
Owen Andersonb2303722008-06-18 21:41:49 +0000418 valueNumbering.insert(std::make_pair(V, num));
419}
420
Owen Andersond41ed4e2009-10-19 22:14:22 +0000421uint32_t ValueTable::lookup_or_add_call(CallInst* C) {
422 if (AA->doesNotAccessMemory(C)) {
423 Expression exp = create_expression(C);
424 uint32_t& e = expressionNumbering[exp];
425 if (!e) e = nextValueNumber++;
426 valueNumbering[C] = e;
427 return e;
428 } else if (AA->onlyReadsMemory(C)) {
429 Expression exp = create_expression(C);
430 uint32_t& e = expressionNumbering[exp];
431 if (!e) {
432 e = nextValueNumber++;
433 valueNumbering[C] = e;
434 return e;
435 }
Dan Gohman4ec01b22009-11-14 02:27:51 +0000436 if (!MD) {
437 e = nextValueNumber++;
438 valueNumbering[C] = e;
439 return e;
440 }
Owen Andersond41ed4e2009-10-19 22:14:22 +0000441
442 MemDepResult local_dep = MD->getDependency(C);
443
444 if (!local_dep.isDef() && !local_dep.isNonLocal()) {
445 valueNumbering[C] = nextValueNumber;
446 return nextValueNumber++;
447 }
448
449 if (local_dep.isDef()) {
450 CallInst* local_cdep = cast<CallInst>(local_dep.getInst());
451
Gabor Greif237e1da2010-06-30 09:17:53 +0000452 if (local_cdep->getNumArgOperands() != C->getNumArgOperands()) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000453 valueNumbering[C] = nextValueNumber;
454 return nextValueNumber++;
455 }
456
Gabor Greifd883a9d2010-06-24 10:17:17 +0000457 for (unsigned i = 0, e = C->getNumArgOperands(); i < e; ++i) {
458 uint32_t c_vn = lookup_or_add(C->getArgOperand(i));
459 uint32_t cd_vn = lookup_or_add(local_cdep->getArgOperand(i));
Owen Andersond41ed4e2009-10-19 22:14:22 +0000460 if (c_vn != cd_vn) {
461 valueNumbering[C] = nextValueNumber;
462 return nextValueNumber++;
463 }
464 }
465
466 uint32_t v = lookup_or_add(local_cdep);
467 valueNumbering[C] = v;
468 return v;
469 }
470
471 // Non-local case.
472 const MemoryDependenceAnalysis::NonLocalDepInfo &deps =
473 MD->getNonLocalCallDependency(CallSite(C));
474 // FIXME: call/call dependencies for readonly calls should return def, not
475 // clobber! Move the checking logic to MemDep!
476 CallInst* cdep = 0;
477
478 // Check to see if we have a single dominating call instruction that is
479 // identical to C.
480 for (unsigned i = 0, e = deps.size(); i != e; ++i) {
Chris Lattnere18b9712009-12-09 07:08:01 +0000481 const NonLocalDepEntry *I = &deps[i];
Owen Andersond41ed4e2009-10-19 22:14:22 +0000482 // Ignore non-local dependencies.
Chris Lattnere18b9712009-12-09 07:08:01 +0000483 if (I->getResult().isNonLocal())
Owen Andersond41ed4e2009-10-19 22:14:22 +0000484 continue;
485
486 // We don't handle non-depedencies. If we already have a call, reject
487 // instruction dependencies.
Chris Lattnere18b9712009-12-09 07:08:01 +0000488 if (I->getResult().isClobber() || cdep != 0) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000489 cdep = 0;
490 break;
491 }
492
Chris Lattnere18b9712009-12-09 07:08:01 +0000493 CallInst *NonLocalDepCall = dyn_cast<CallInst>(I->getResult().getInst());
Owen Andersond41ed4e2009-10-19 22:14:22 +0000494 // FIXME: All duplicated with non-local case.
Chris Lattnere18b9712009-12-09 07:08:01 +0000495 if (NonLocalDepCall && DT->properlyDominates(I->getBB(), C->getParent())){
Owen Andersond41ed4e2009-10-19 22:14:22 +0000496 cdep = NonLocalDepCall;
497 continue;
498 }
499
500 cdep = 0;
501 break;
502 }
503
504 if (!cdep) {
505 valueNumbering[C] = nextValueNumber;
506 return nextValueNumber++;
507 }
508
Gabor Greif237e1da2010-06-30 09:17:53 +0000509 if (cdep->getNumArgOperands() != C->getNumArgOperands()) {
Owen Andersond41ed4e2009-10-19 22:14:22 +0000510 valueNumbering[C] = nextValueNumber;
511 return nextValueNumber++;
512 }
Gabor Greifd883a9d2010-06-24 10:17:17 +0000513 for (unsigned i = 0, e = C->getNumArgOperands(); i < e; ++i) {
514 uint32_t c_vn = lookup_or_add(C->getArgOperand(i));
515 uint32_t cd_vn = lookup_or_add(cdep->getArgOperand(i));
Owen Andersond41ed4e2009-10-19 22:14:22 +0000516 if (c_vn != cd_vn) {
517 valueNumbering[C] = nextValueNumber;
518 return nextValueNumber++;
519 }
520 }
521
522 uint32_t v = lookup_or_add(cdep);
523 valueNumbering[C] = v;
524 return v;
525
526 } else {
527 valueNumbering[C] = nextValueNumber;
528 return nextValueNumber++;
529 }
530}
531
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000532/// lookup_or_add - Returns the value number for the specified value, assigning
533/// it a new number if it did not have one before.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000534uint32_t ValueTable::lookup_or_add(Value *V) {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000535 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
536 if (VI != valueNumbering.end())
537 return VI->second;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000538
Owen Andersond41ed4e2009-10-19 22:14:22 +0000539 if (!isa<Instruction>(V)) {
Owen Anderson158d86e2009-10-19 21:14:57 +0000540 valueNumbering[V] = nextValueNumber;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000541 return nextValueNumber++;
542 }
Owen Andersond41ed4e2009-10-19 22:14:22 +0000543
544 Instruction* I = cast<Instruction>(V);
545 Expression exp;
546 switch (I->getOpcode()) {
547 case Instruction::Call:
548 return lookup_or_add_call(cast<CallInst>(I));
549 case Instruction::Add:
550 case Instruction::FAdd:
551 case Instruction::Sub:
552 case Instruction::FSub:
553 case Instruction::Mul:
554 case Instruction::FMul:
555 case Instruction::UDiv:
556 case Instruction::SDiv:
557 case Instruction::FDiv:
558 case Instruction::URem:
559 case Instruction::SRem:
560 case Instruction::FRem:
561 case Instruction::Shl:
562 case Instruction::LShr:
563 case Instruction::AShr:
564 case Instruction::And:
565 case Instruction::Or :
566 case Instruction::Xor:
567 exp = create_expression(cast<BinaryOperator>(I));
568 break;
569 case Instruction::ICmp:
570 case Instruction::FCmp:
571 exp = create_expression(cast<CmpInst>(I));
572 break;
573 case Instruction::Trunc:
574 case Instruction::ZExt:
575 case Instruction::SExt:
576 case Instruction::FPToUI:
577 case Instruction::FPToSI:
578 case Instruction::UIToFP:
579 case Instruction::SIToFP:
580 case Instruction::FPTrunc:
581 case Instruction::FPExt:
582 case Instruction::PtrToInt:
583 case Instruction::IntToPtr:
584 case Instruction::BitCast:
585 exp = create_expression(cast<CastInst>(I));
586 break;
587 case Instruction::Select:
588 exp = create_expression(cast<SelectInst>(I));
589 break;
590 case Instruction::ExtractElement:
591 exp = create_expression(cast<ExtractElementInst>(I));
592 break;
593 case Instruction::InsertElement:
594 exp = create_expression(cast<InsertElementInst>(I));
595 break;
596 case Instruction::ShuffleVector:
597 exp = create_expression(cast<ShuffleVectorInst>(I));
598 break;
599 case Instruction::ExtractValue:
600 exp = create_expression(cast<ExtractValueInst>(I));
601 break;
602 case Instruction::InsertValue:
603 exp = create_expression(cast<InsertValueInst>(I));
604 break;
605 case Instruction::GetElementPtr:
606 exp = create_expression(cast<GetElementPtrInst>(I));
607 break;
608 default:
609 valueNumbering[V] = nextValueNumber;
610 return nextValueNumber++;
611 }
612
613 uint32_t& e = expressionNumbering[exp];
614 if (!e) e = nextValueNumber++;
615 valueNumbering[V] = e;
616 return e;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000617}
618
619/// lookup - Returns the value number of the specified value. Fails if
620/// the value has not yet been numbered.
Chris Lattnerb2412a82009-09-21 02:42:51 +0000621uint32_t ValueTable::lookup(Value *V) const {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000622 DenseMap<Value*, uint32_t>::const_iterator VI = valueNumbering.find(V);
Chris Lattner88365bb2008-03-21 21:14:38 +0000623 assert(VI != valueNumbering.end() && "Value not numbered?");
624 return VI->second;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000625}
626
627/// clear - Remove all entries from the ValueTable
628void ValueTable::clear() {
629 valueNumbering.clear();
630 expressionNumbering.clear();
631 nextValueNumber = 1;
632}
633
Owen Andersonbf7d0bc2007-07-31 23:27:13 +0000634/// erase - Remove a value from the value numbering
Chris Lattnerb2412a82009-09-21 02:42:51 +0000635void ValueTable::erase(Value *V) {
Owen Andersonbf7d0bc2007-07-31 23:27:13 +0000636 valueNumbering.erase(V);
637}
638
Bill Wendling246dbbb2008-12-22 21:36:08 +0000639/// verifyRemoved - Verify that the value is removed from all internal data
640/// structures.
641void ValueTable::verifyRemoved(const Value *V) const {
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000642 for (DenseMap<Value*, uint32_t>::const_iterator
Bill Wendling246dbbb2008-12-22 21:36:08 +0000643 I = valueNumbering.begin(), E = valueNumbering.end(); I != E; ++I) {
644 assert(I->first != V && "Inst still occurs in value numbering map!");
645 }
646}
647
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000648//===----------------------------------------------------------------------===//
Bill Wendling30788b82008-12-22 22:32:22 +0000649// GVN Pass
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000650//===----------------------------------------------------------------------===//
651
652namespace {
Chris Lattner3e8b6632009-09-02 06:11:42 +0000653 struct ValueNumberScope {
Owen Anderson6fafe842008-06-20 01:15:47 +0000654 ValueNumberScope* parent;
655 DenseMap<uint32_t, Value*> table;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000656
Owen Anderson6fafe842008-06-20 01:15:47 +0000657 ValueNumberScope(ValueNumberScope* p) : parent(p) { }
658 };
659}
660
661namespace {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000662
Chris Lattner3e8b6632009-09-02 06:11:42 +0000663 class GVN : public FunctionPass {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000664 bool runOnFunction(Function &F);
665 public:
666 static char ID; // Pass identification, replacement for typeid
Bob Wilsonb29d7d22010-02-28 05:34:05 +0000667 explicit GVN(bool noloads = false)
Owen Anderson081c34b2010-10-19 17:21:58 +0000668 : FunctionPass(ID), NoLoads(noloads), MD(0) {
669 initializeGVNPass(*PassRegistry::getPassRegistry());
670 }
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000671
672 private:
Dan Gohman4ec01b22009-11-14 02:27:51 +0000673 bool NoLoads;
Chris Lattner663e4412008-12-01 00:40:32 +0000674 MemoryDependenceAnalysis *MD;
675 DominatorTree *DT;
Duncan Sands88c3df72010-11-12 21:10:24 +0000676 const TargetData* TD;
Chris Lattner663e4412008-12-01 00:40:32 +0000677
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000678 ValueTable VN;
Owen Andersona04a0642010-11-18 18:32:40 +0000679
Owen Anderson68c26392010-11-19 22:48:40 +0000680 /// NumberTable - A mapping from value numers to lists of Value*'s that
681 /// have that value number. Use lookupNumber to query it.
Owen Andersona04a0642010-11-18 18:32:40 +0000682 DenseMap<uint32_t, std::pair<Value*, void*> > NumberTable;
683 BumpPtrAllocator TableAllocator;
Owen Anderson68c26392010-11-19 22:48:40 +0000684
685 /// insert_table - Push a new Value to the NumberTable onto the list for
686 /// its value number.
Owen Andersona04a0642010-11-18 18:32:40 +0000687 void insert_table(uint32_t N, Value *V) {
688 std::pair<Value*, void*>& Curr = NumberTable[N];
689 if (!Curr.first) {
690 Curr.first = V;
691 return;
692 }
693
694 std::pair<Value*, void*>* Node =
695 TableAllocator.Allocate<std::pair<Value*, void*> >();
696 Node->first = V;
697 Node->second = Curr.second;
698 Curr.second = Node;
699 }
700
Owen Anderson68c26392010-11-19 22:48:40 +0000701 /// erase_table - Scan the list of values corresponding to a given value
702 /// number, and remove the given value if encountered.
Owen Andersona04a0642010-11-18 18:32:40 +0000703 void erase_table(uint32_t N, Value *V) {
704 std::pair<Value*, void*>* Prev = 0;
705 std::pair<Value*, void*>* Curr = &NumberTable[N];
706
707 while (Curr->first != V) {
708 Prev = Curr;
709 Curr = static_cast<std::pair<Value*, void*>*>(Curr->second);
710 }
711
712 if (Prev) {
713 Prev->second = Curr->second;
714 } else {
715 if (!Curr->second) {
716 Curr->first = 0;
717 } else {
718 std::pair<Value*, void*>* Next =
719 static_cast<std::pair<Value*, void*>*>(Curr->second);
720 Curr->first = Next->first;
721 Curr->second = Next->second;
722 }
723 }
724 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000725
Bob Wilson484d4a32010-02-16 19:51:59 +0000726 // List of critical edges to be split between iterations.
727 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> toSplit;
728
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000729 // This transformation requires dominator postdominator info
730 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000731 AU.addRequired<DominatorTree>();
Dan Gohman4ec01b22009-11-14 02:27:51 +0000732 if (!NoLoads)
733 AU.addRequired<MemoryDependenceAnalysis>();
Owen Andersonb388ca92007-10-18 19:39:33 +0000734 AU.addRequired<AliasAnalysis>();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000735
Owen Andersonb70a5712008-06-23 17:49:45 +0000736 AU.addPreserved<DominatorTree>();
Owen Andersonb388ca92007-10-18 19:39:33 +0000737 AU.addPreserved<AliasAnalysis>();
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000738 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000739
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000740 // Helper fuctions
741 // FIXME: eliminate or document these better
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000742 bool processLoad(LoadInst* L,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000743 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000744 bool processInstruction(Instruction *I,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000745 SmallVectorImpl<Instruction*> &toErase);
Owen Anderson830db6a2007-08-02 18:16:06 +0000746 bool processNonLocalLoad(LoadInst* L,
Chris Lattner8e1e95c2008-03-21 22:01:16 +0000747 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000748 bool processBlock(BasicBlock *BB);
Owen Andersonb2303722008-06-18 21:41:49 +0000749 void dump(DenseMap<uint32_t, Value*>& d);
Owen Anderson3e75a422007-08-14 18:04:11 +0000750 bool iterateOnFunction(Function &F);
Owen Andersonb2303722008-06-18 21:41:49 +0000751 bool performPRE(Function& F);
Chris Lattnerb2412a82009-09-21 02:42:51 +0000752 Value *lookupNumber(BasicBlock *BB, uint32_t num);
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +0000753 void cleanupGlobalSets();
Bill Wendling246dbbb2008-12-22 21:36:08 +0000754 void verifyRemoved(const Instruction *I) const;
Bob Wilson484d4a32010-02-16 19:51:59 +0000755 bool splitCriticalEdges();
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000756 };
Daniel Dunbara279bc32009-09-20 02:20:51 +0000757
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000758 char GVN::ID = 0;
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000759}
760
761// createGVNPass - The public interface to this file...
Bob Wilsonb29d7d22010-02-28 05:34:05 +0000762FunctionPass *llvm::createGVNPass(bool NoLoads) {
763 return new GVN(NoLoads);
Dan Gohman4ec01b22009-11-14 02:27:51 +0000764}
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000765
Owen Anderson2ab36d32010-10-12 19:48:12 +0000766INITIALIZE_PASS_BEGIN(GVN, "gvn", "Global Value Numbering", false, false)
767INITIALIZE_PASS_DEPENDENCY(MemoryDependenceAnalysis)
768INITIALIZE_PASS_DEPENDENCY(DominatorTree)
769INITIALIZE_AG_DEPENDENCY(AliasAnalysis)
770INITIALIZE_PASS_END(GVN, "gvn", "Global Value Numbering", false, false)
Owen Anderson1ad2cb72007-07-24 17:55:58 +0000771
Owen Andersonb2303722008-06-18 21:41:49 +0000772void GVN::dump(DenseMap<uint32_t, Value*>& d) {
Dan Gohmanad12b262009-12-18 03:25:51 +0000773 errs() << "{\n";
Owen Andersonb2303722008-06-18 21:41:49 +0000774 for (DenseMap<uint32_t, Value*>::iterator I = d.begin(),
Owen Anderson0cd32032007-07-25 19:57:03 +0000775 E = d.end(); I != E; ++I) {
Dan Gohmanad12b262009-12-18 03:25:51 +0000776 errs() << I->first << "\n";
Owen Anderson0cd32032007-07-25 19:57:03 +0000777 I->second->dump();
778 }
Dan Gohmanad12b262009-12-18 03:25:51 +0000779 errs() << "}\n";
Owen Anderson0cd32032007-07-25 19:57:03 +0000780}
781
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000782/// IsValueFullyAvailableInBlock - Return true if we can prove that the value
783/// we're analyzing is fully available in the specified block. As we go, keep
Chris Lattner72bc70d2008-12-05 07:49:08 +0000784/// track of which blocks we know are fully alive in FullyAvailableBlocks. This
785/// map is actually a tri-state map with the following values:
786/// 0) we know the block *is not* fully available.
787/// 1) we know the block *is* fully available.
788/// 2) we do not know whether the block is fully available or not, but we are
789/// currently speculating that it will be.
790/// 3) we are speculating for this block and have used that to speculate for
791/// other blocks.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000792static bool IsValueFullyAvailableInBlock(BasicBlock *BB,
Chris Lattner72bc70d2008-12-05 07:49:08 +0000793 DenseMap<BasicBlock*, char> &FullyAvailableBlocks) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000794 // Optimistically assume that the block is fully available and check to see
795 // if we already know about this block in one lookup.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000796 std::pair<DenseMap<BasicBlock*, char>::iterator, char> IV =
Chris Lattner72bc70d2008-12-05 07:49:08 +0000797 FullyAvailableBlocks.insert(std::make_pair(BB, 2));
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000798
799 // If the entry already existed for this block, return the precomputed value.
Chris Lattner72bc70d2008-12-05 07:49:08 +0000800 if (!IV.second) {
801 // If this is a speculative "available" value, mark it as being used for
802 // speculation of other blocks.
803 if (IV.first->second == 2)
804 IV.first->second = 3;
805 return IV.first->second != 0;
806 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000807
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000808 // Otherwise, see if it is fully available in all predecessors.
809 pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000810
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000811 // If this block has no predecessors, it isn't live-in here.
812 if (PI == PE)
Chris Lattner72bc70d2008-12-05 07:49:08 +0000813 goto SpeculationFailure;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000814
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000815 for (; PI != PE; ++PI)
816 // If the value isn't fully available in one of our predecessors, then it
817 // isn't fully available in this block either. Undo our previous
818 // optimistic assumption and bail out.
819 if (!IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
Chris Lattner72bc70d2008-12-05 07:49:08 +0000820 goto SpeculationFailure;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000821
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000822 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000823
Chris Lattner72bc70d2008-12-05 07:49:08 +0000824// SpeculationFailure - If we get here, we found out that this is not, after
825// all, a fully-available block. We have a problem if we speculated on this and
826// used the speculation to mark other blocks as available.
827SpeculationFailure:
828 char &BBVal = FullyAvailableBlocks[BB];
Daniel Dunbara279bc32009-09-20 02:20:51 +0000829
Chris Lattner72bc70d2008-12-05 07:49:08 +0000830 // If we didn't speculate on this, just return with it set to false.
831 if (BBVal == 2) {
832 BBVal = 0;
833 return false;
834 }
835
836 // If we did speculate on this value, we could have blocks set to 1 that are
837 // incorrect. Walk the (transitive) successors of this block and mark them as
838 // 0 if set to one.
839 SmallVector<BasicBlock*, 32> BBWorklist;
840 BBWorklist.push_back(BB);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000841
Dan Gohman321a8132010-01-05 16:27:25 +0000842 do {
Chris Lattner72bc70d2008-12-05 07:49:08 +0000843 BasicBlock *Entry = BBWorklist.pop_back_val();
844 // Note that this sets blocks to 0 (unavailable) if they happen to not
845 // already be in FullyAvailableBlocks. This is safe.
846 char &EntryVal = FullyAvailableBlocks[Entry];
847 if (EntryVal == 0) continue; // Already unavailable.
848
849 // Mark as unavailable.
850 EntryVal = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000851
Chris Lattner72bc70d2008-12-05 07:49:08 +0000852 for (succ_iterator I = succ_begin(Entry), E = succ_end(Entry); I != E; ++I)
853 BBWorklist.push_back(*I);
Dan Gohman321a8132010-01-05 16:27:25 +0000854 } while (!BBWorklist.empty());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000855
Chris Lattner72bc70d2008-12-05 07:49:08 +0000856 return false;
Chris Lattnerc89c6a92008-12-02 08:16:11 +0000857}
858
Chris Lattner771a5422009-09-20 20:09:34 +0000859
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000860/// CanCoerceMustAliasedValueToLoad - Return true if
861/// CoerceAvailableValueToLoadType will succeed.
862static bool CanCoerceMustAliasedValueToLoad(Value *StoredVal,
863 const Type *LoadTy,
864 const TargetData &TD) {
865 // If the loaded or stored value is an first class array or struct, don't try
866 // to transform them. We need to be able to bitcast to integer.
Duncan Sands1df98592010-02-16 11:11:14 +0000867 if (LoadTy->isStructTy() || LoadTy->isArrayTy() ||
868 StoredVal->getType()->isStructTy() ||
869 StoredVal->getType()->isArrayTy())
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000870 return false;
871
872 // The store has to be at least as big as the load.
873 if (TD.getTypeSizeInBits(StoredVal->getType()) <
874 TD.getTypeSizeInBits(LoadTy))
875 return false;
876
877 return true;
878}
879
880
Chris Lattner771a5422009-09-20 20:09:34 +0000881/// CoerceAvailableValueToLoadType - If we saw a store of a value to memory, and
882/// then a load from a must-aliased pointer of a different type, try to coerce
883/// the stored value. LoadedTy is the type of the load we want to replace and
884/// InsertPt is the place to insert new instructions.
885///
886/// If we can't do it, return null.
887static Value *CoerceAvailableValueToLoadType(Value *StoredVal,
888 const Type *LoadedTy,
889 Instruction *InsertPt,
890 const TargetData &TD) {
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000891 if (!CanCoerceMustAliasedValueToLoad(StoredVal, LoadedTy, TD))
892 return 0;
893
Chris Lattner771a5422009-09-20 20:09:34 +0000894 const Type *StoredValTy = StoredVal->getType();
895
Chris Lattner7944c212010-05-08 20:01:44 +0000896 uint64_t StoreSize = TD.getTypeStoreSizeInBits(StoredValTy);
Chris Lattner771a5422009-09-20 20:09:34 +0000897 uint64_t LoadSize = TD.getTypeSizeInBits(LoadedTy);
898
899 // If the store and reload are the same size, we can always reuse it.
900 if (StoreSize == LoadSize) {
Duncan Sands1df98592010-02-16 11:11:14 +0000901 if (StoredValTy->isPointerTy() && LoadedTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000902 // Pointer to Pointer -> use bitcast.
903 return new BitCastInst(StoredVal, LoadedTy, "", InsertPt);
904 }
905
906 // Convert source pointers to integers, which can be bitcast.
Duncan Sands1df98592010-02-16 11:11:14 +0000907 if (StoredValTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000908 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
909 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
910 }
911
912 const Type *TypeToCastTo = LoadedTy;
Duncan Sands1df98592010-02-16 11:11:14 +0000913 if (TypeToCastTo->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000914 TypeToCastTo = TD.getIntPtrType(StoredValTy->getContext());
915
916 if (StoredValTy != TypeToCastTo)
917 StoredVal = new BitCastInst(StoredVal, TypeToCastTo, "", InsertPt);
918
919 // Cast to pointer if the load needs a pointer type.
Duncan Sands1df98592010-02-16 11:11:14 +0000920 if (LoadedTy->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000921 StoredVal = new IntToPtrInst(StoredVal, LoadedTy, "", InsertPt);
922
923 return StoredVal;
924 }
925
926 // If the loaded value is smaller than the available value, then we can
927 // extract out a piece from it. If the available value is too small, then we
928 // can't do anything.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +0000929 assert(StoreSize >= LoadSize && "CanCoerceMustAliasedValueToLoad fail");
Chris Lattner771a5422009-09-20 20:09:34 +0000930
931 // Convert source pointers to integers, which can be manipulated.
Duncan Sands1df98592010-02-16 11:11:14 +0000932 if (StoredValTy->isPointerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000933 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
934 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
935 }
936
937 // Convert vectors and fp to integer, which can be manipulated.
Duncan Sands1df98592010-02-16 11:11:14 +0000938 if (!StoredValTy->isIntegerTy()) {
Chris Lattner771a5422009-09-20 20:09:34 +0000939 StoredValTy = IntegerType::get(StoredValTy->getContext(), StoreSize);
940 StoredVal = new BitCastInst(StoredVal, StoredValTy, "", InsertPt);
941 }
942
943 // If this is a big-endian system, we need to shift the value down to the low
944 // bits so that a truncate will work.
945 if (TD.isBigEndian()) {
946 Constant *Val = ConstantInt::get(StoredVal->getType(), StoreSize-LoadSize);
947 StoredVal = BinaryOperator::CreateLShr(StoredVal, Val, "tmp", InsertPt);
948 }
949
950 // Truncate the integer to the right size now.
951 const Type *NewIntTy = IntegerType::get(StoredValTy->getContext(), LoadSize);
952 StoredVal = new TruncInst(StoredVal, NewIntTy, "trunc", InsertPt);
953
954 if (LoadedTy == NewIntTy)
955 return StoredVal;
956
957 // If the result is a pointer, inttoptr.
Duncan Sands1df98592010-02-16 11:11:14 +0000958 if (LoadedTy->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +0000959 return new IntToPtrInst(StoredVal, LoadedTy, "inttoptr", InsertPt);
960
961 // Otherwise, bitcast.
962 return new BitCastInst(StoredVal, LoadedTy, "bitcast", InsertPt);
963}
964
Chris Lattnerca749402009-09-21 06:24:16 +0000965/// GetBaseWithConstantOffset - Analyze the specified pointer to see if it can
966/// be expressed as a base pointer plus a constant offset. Return the base and
967/// offset to the caller.
968static Value *GetBaseWithConstantOffset(Value *Ptr, int64_t &Offset,
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000969 const TargetData &TD) {
Chris Lattnerca749402009-09-21 06:24:16 +0000970 Operator *PtrOp = dyn_cast<Operator>(Ptr);
971 if (PtrOp == 0) return Ptr;
972
973 // Just look through bitcasts.
974 if (PtrOp->getOpcode() == Instruction::BitCast)
975 return GetBaseWithConstantOffset(PtrOp->getOperand(0), Offset, TD);
976
977 // If this is a GEP with constant indices, we can look through it.
978 GEPOperator *GEP = dyn_cast<GEPOperator>(PtrOp);
979 if (GEP == 0 || !GEP->hasAllConstantIndices()) return Ptr;
980
981 gep_type_iterator GTI = gep_type_begin(GEP);
982 for (User::op_iterator I = GEP->idx_begin(), E = GEP->idx_end(); I != E;
983 ++I, ++GTI) {
984 ConstantInt *OpC = cast<ConstantInt>(*I);
985 if (OpC->isZero()) continue;
986
987 // Handle a struct and array indices which add their offset to the pointer.
988 if (const StructType *STy = dyn_cast<StructType>(*GTI)) {
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000989 Offset += TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue());
Chris Lattnerca749402009-09-21 06:24:16 +0000990 } else {
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000991 uint64_t Size = TD.getTypeAllocSize(GTI.getIndexedType());
Chris Lattnerca749402009-09-21 06:24:16 +0000992 Offset += OpC->getSExtValue()*Size;
993 }
994 }
995
996 // Re-sign extend from the pointer size if needed to get overflow edge cases
997 // right.
Chris Lattner4fbd14e2009-09-21 06:48:08 +0000998 unsigned PtrSize = TD.getPointerSizeInBits();
Chris Lattnerca749402009-09-21 06:24:16 +0000999 if (PtrSize < 64)
1000 Offset = (Offset << (64-PtrSize)) >> (64-PtrSize);
1001
1002 return GetBaseWithConstantOffset(GEP->getPointerOperand(), Offset, TD);
1003}
1004
1005
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001006/// AnalyzeLoadFromClobberingWrite - This function is called when we have a
1007/// memdep query of a load that ends up being a clobbering memory write (store,
1008/// memset, memcpy, memmove). This means that the write *may* provide bits used
1009/// by the load but we can't be sure because the pointers don't mustalias.
1010///
1011/// Check this case to see if there is anything more we can do before we give
1012/// up. This returns -1 if we have to give up, or a byte number in the stored
1013/// value of the piece that feeds the load.
Chris Lattner03f17da2009-12-09 07:34:10 +00001014static int AnalyzeLoadFromClobberingWrite(const Type *LoadTy, Value *LoadPtr,
1015 Value *WritePtr,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001016 uint64_t WriteSizeInBits,
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001017 const TargetData &TD) {
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001018 // If the loaded or stored value is an first class array or struct, don't try
1019 // to transform them. We need to be able to bitcast to integer.
Duncan Sands1df98592010-02-16 11:11:14 +00001020 if (LoadTy->isStructTy() || LoadTy->isArrayTy())
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001021 return -1;
1022
Chris Lattnerca749402009-09-21 06:24:16 +00001023 int64_t StoreOffset = 0, LoadOffset = 0;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001024 Value *StoreBase = GetBaseWithConstantOffset(WritePtr, StoreOffset, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001025 Value *LoadBase =
Chris Lattner03f17da2009-12-09 07:34:10 +00001026 GetBaseWithConstantOffset(LoadPtr, LoadOffset, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001027 if (StoreBase != LoadBase)
1028 return -1;
1029
1030 // If the load and store are to the exact same address, they should have been
1031 // a must alias. AA must have gotten confused.
Chris Lattner219d7742010-03-25 05:58:19 +00001032 // FIXME: Study to see if/when this happens. One case is forwarding a memset
1033 // to a load from the base of the memset.
Chris Lattnerca749402009-09-21 06:24:16 +00001034#if 0
Chris Lattner219d7742010-03-25 05:58:19 +00001035 if (LoadOffset == StoreOffset) {
David Greenebf7f78e2010-01-05 01:27:17 +00001036 dbgs() << "STORE/LOAD DEP WITH COMMON POINTER MISSED:\n"
Chris Lattnerca749402009-09-21 06:24:16 +00001037 << "Base = " << *StoreBase << "\n"
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001038 << "Store Ptr = " << *WritePtr << "\n"
1039 << "Store Offs = " << StoreOffset << "\n"
Chris Lattnerb6760b42009-12-10 00:04:46 +00001040 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattnerb3f927f2009-12-09 02:41:54 +00001041 abort();
Chris Lattnerca749402009-09-21 06:24:16 +00001042 }
Chris Lattner219d7742010-03-25 05:58:19 +00001043#endif
Chris Lattnerca749402009-09-21 06:24:16 +00001044
1045 // If the load and store don't overlap at all, the store doesn't provide
1046 // anything to the load. In this case, they really don't alias at all, AA
1047 // must have gotten confused.
1048 // FIXME: Investigate cases where this bails out, e.g. rdar://7238614. Then
1049 // remove this check, as it is duplicated with what we have below.
Chris Lattner03f17da2009-12-09 07:34:10 +00001050 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy);
Chris Lattnerca749402009-09-21 06:24:16 +00001051
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001052 if ((WriteSizeInBits & 7) | (LoadSize & 7))
Chris Lattnerca749402009-09-21 06:24:16 +00001053 return -1;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001054 uint64_t StoreSize = WriteSizeInBits >> 3; // Convert to bytes.
Chris Lattnerca749402009-09-21 06:24:16 +00001055 LoadSize >>= 3;
1056
1057
1058 bool isAAFailure = false;
Chris Lattner219d7742010-03-25 05:58:19 +00001059 if (StoreOffset < LoadOffset)
Chris Lattnerca749402009-09-21 06:24:16 +00001060 isAAFailure = StoreOffset+int64_t(StoreSize) <= LoadOffset;
Chris Lattner219d7742010-03-25 05:58:19 +00001061 else
Chris Lattnerca749402009-09-21 06:24:16 +00001062 isAAFailure = LoadOffset+int64_t(LoadSize) <= StoreOffset;
Chris Lattner219d7742010-03-25 05:58:19 +00001063
Chris Lattnerca749402009-09-21 06:24:16 +00001064 if (isAAFailure) {
1065#if 0
David Greenebf7f78e2010-01-05 01:27:17 +00001066 dbgs() << "STORE LOAD DEP WITH COMMON BASE:\n"
Chris Lattnerca749402009-09-21 06:24:16 +00001067 << "Base = " << *StoreBase << "\n"
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001068 << "Store Ptr = " << *WritePtr << "\n"
1069 << "Store Offs = " << StoreOffset << "\n"
Chris Lattnerb6760b42009-12-10 00:04:46 +00001070 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattnerb3f927f2009-12-09 02:41:54 +00001071 abort();
Chris Lattnerca749402009-09-21 06:24:16 +00001072#endif
1073 return -1;
1074 }
1075
1076 // If the Load isn't completely contained within the stored bits, we don't
1077 // have all the bits to feed it. We could do something crazy in the future
1078 // (issue a smaller load then merge the bits in) but this seems unlikely to be
1079 // valuable.
1080 if (StoreOffset > LoadOffset ||
1081 StoreOffset+StoreSize < LoadOffset+LoadSize)
1082 return -1;
1083
1084 // Okay, we can do this transformation. Return the number of bytes into the
1085 // store that the load is.
1086 return LoadOffset-StoreOffset;
1087}
1088
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001089/// AnalyzeLoadFromClobberingStore - This function is called when we have a
1090/// memdep query of a load that ends up being a clobbering store.
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001091static int AnalyzeLoadFromClobberingStore(const Type *LoadTy, Value *LoadPtr,
1092 StoreInst *DepSI,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001093 const TargetData &TD) {
1094 // Cannot handle reading from store of first-class aggregate yet.
Dan Gohman3355c4e2010-11-10 19:03:33 +00001095 if (DepSI->getValueOperand()->getType()->isStructTy() ||
1096 DepSI->getValueOperand()->getType()->isArrayTy())
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001097 return -1;
1098
1099 Value *StorePtr = DepSI->getPointerOperand();
Dan Gohman3355c4e2010-11-10 19:03:33 +00001100 uint64_t StoreSize =TD.getTypeSizeInBits(DepSI->getValueOperand()->getType());
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001101 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
Chris Lattner03f17da2009-12-09 07:34:10 +00001102 StorePtr, StoreSize, TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001103}
1104
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001105static int AnalyzeLoadFromClobberingMemInst(const Type *LoadTy, Value *LoadPtr,
1106 MemIntrinsic *MI,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001107 const TargetData &TD) {
1108 // If the mem operation is a non-constant size, we can't handle it.
1109 ConstantInt *SizeCst = dyn_cast<ConstantInt>(MI->getLength());
1110 if (SizeCst == 0) return -1;
1111 uint64_t MemSizeInBits = SizeCst->getZExtValue()*8;
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001112
1113 // If this is memset, we just need to see if the offset is valid in the size
1114 // of the memset..
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001115 if (MI->getIntrinsicID() == Intrinsic::memset)
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001116 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr, MI->getDest(),
1117 MemSizeInBits, TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001118
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001119 // If we have a memcpy/memmove, the only case we can handle is if this is a
1120 // copy from constant memory. In that case, we can read directly from the
1121 // constant memory.
1122 MemTransferInst *MTI = cast<MemTransferInst>(MI);
1123
1124 Constant *Src = dyn_cast<Constant>(MTI->getSource());
1125 if (Src == 0) return -1;
1126
1127 GlobalVariable *GV = dyn_cast<GlobalVariable>(Src->getUnderlyingObject());
1128 if (GV == 0 || !GV->isConstant()) return -1;
1129
1130 // See if the access is within the bounds of the transfer.
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001131 int Offset = AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
1132 MI->getDest(), MemSizeInBits, TD);
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001133 if (Offset == -1)
1134 return Offset;
1135
1136 // Otherwise, see if we can constant fold a load from the constant with the
1137 // offset applied as appropriate.
1138 Src = ConstantExpr::getBitCast(Src,
1139 llvm::Type::getInt8PtrTy(Src->getContext()));
1140 Constant *OffsetCst =
1141 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1142 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001143 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001144 if (ConstantFoldLoadFromConstPtr(Src, &TD))
1145 return Offset;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001146 return -1;
1147}
1148
Chris Lattnerca749402009-09-21 06:24:16 +00001149
1150/// GetStoreValueForLoad - This function is called when we have a
1151/// memdep query of a load that ends up being a clobbering store. This means
1152/// that the store *may* provide bits used by the load but we can't be sure
1153/// because the pointers don't mustalias. Check this case to see if there is
1154/// anything more we can do before we give up.
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001155static Value *GetStoreValueForLoad(Value *SrcVal, unsigned Offset,
1156 const Type *LoadTy,
1157 Instruction *InsertPt, const TargetData &TD){
Chris Lattnerca749402009-09-21 06:24:16 +00001158 LLVMContext &Ctx = SrcVal->getType()->getContext();
1159
Chris Lattner7944c212010-05-08 20:01:44 +00001160 uint64_t StoreSize = (TD.getTypeSizeInBits(SrcVal->getType()) + 7) / 8;
1161 uint64_t LoadSize = (TD.getTypeSizeInBits(LoadTy) + 7) / 8;
Chris Lattnerca749402009-09-21 06:24:16 +00001162
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001163 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
Chris Lattnerca749402009-09-21 06:24:16 +00001164
1165 // Compute which bits of the stored value are being used by the load. Convert
1166 // to an integer type to start with.
Duncan Sands1df98592010-02-16 11:11:14 +00001167 if (SrcVal->getType()->isPointerTy())
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001168 SrcVal = Builder.CreatePtrToInt(SrcVal, TD.getIntPtrType(Ctx), "tmp");
Duncan Sands1df98592010-02-16 11:11:14 +00001169 if (!SrcVal->getType()->isIntegerTy())
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001170 SrcVal = Builder.CreateBitCast(SrcVal, IntegerType::get(Ctx, StoreSize*8),
1171 "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001172
1173 // Shift the bits to the least significant depending on endianness.
1174 unsigned ShiftAmt;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001175 if (TD.isLittleEndian())
Chris Lattnerca749402009-09-21 06:24:16 +00001176 ShiftAmt = Offset*8;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001177 else
Chris Lattner19ad7842009-09-21 17:55:47 +00001178 ShiftAmt = (StoreSize-LoadSize-Offset)*8;
Chris Lattnerca749402009-09-21 06:24:16 +00001179
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001180 if (ShiftAmt)
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001181 SrcVal = Builder.CreateLShr(SrcVal, ShiftAmt, "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001182
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001183 if (LoadSize != StoreSize)
Chris Lattnerb2c6ae82009-12-09 18:13:28 +00001184 SrcVal = Builder.CreateTrunc(SrcVal, IntegerType::get(Ctx, LoadSize*8),
1185 "tmp");
Chris Lattnerca749402009-09-21 06:24:16 +00001186
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001187 return CoerceAvailableValueToLoadType(SrcVal, LoadTy, InsertPt, TD);
Chris Lattnerca749402009-09-21 06:24:16 +00001188}
1189
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001190/// GetMemInstValueForLoad - This function is called when we have a
1191/// memdep query of a load that ends up being a clobbering mem intrinsic.
1192static Value *GetMemInstValueForLoad(MemIntrinsic *SrcInst, unsigned Offset,
1193 const Type *LoadTy, Instruction *InsertPt,
1194 const TargetData &TD){
1195 LLVMContext &Ctx = LoadTy->getContext();
1196 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
1197
1198 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
1199
1200 // We know that this method is only called when the mem transfer fully
1201 // provides the bits for the load.
1202 if (MemSetInst *MSI = dyn_cast<MemSetInst>(SrcInst)) {
1203 // memset(P, 'x', 1234) -> splat('x'), even if x is a variable, and
1204 // independently of what the offset is.
1205 Value *Val = MSI->getValue();
1206 if (LoadSize != 1)
1207 Val = Builder.CreateZExt(Val, IntegerType::get(Ctx, LoadSize*8));
1208
1209 Value *OneElt = Val;
1210
1211 // Splat the value out to the right number of bits.
1212 for (unsigned NumBytesSet = 1; NumBytesSet != LoadSize; ) {
1213 // If we can double the number of bytes set, do it.
1214 if (NumBytesSet*2 <= LoadSize) {
1215 Value *ShVal = Builder.CreateShl(Val, NumBytesSet*8);
1216 Val = Builder.CreateOr(Val, ShVal);
1217 NumBytesSet <<= 1;
1218 continue;
1219 }
1220
1221 // Otherwise insert one byte at a time.
1222 Value *ShVal = Builder.CreateShl(Val, 1*8);
1223 Val = Builder.CreateOr(OneElt, ShVal);
1224 ++NumBytesSet;
1225 }
1226
1227 return CoerceAvailableValueToLoadType(Val, LoadTy, InsertPt, TD);
1228 }
Chris Lattnerbc9a28d2009-12-06 05:29:56 +00001229
1230 // Otherwise, this is a memcpy/memmove from a constant global.
1231 MemTransferInst *MTI = cast<MemTransferInst>(SrcInst);
1232 Constant *Src = cast<Constant>(MTI->getSource());
1233
1234 // Otherwise, see if we can constant fold a load from the constant with the
1235 // offset applied as appropriate.
1236 Src = ConstantExpr::getBitCast(Src,
1237 llvm::Type::getInt8PtrTy(Src->getContext()));
1238 Constant *OffsetCst =
1239 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1240 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
1241 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
1242 return ConstantFoldLoadFromConstPtr(Src, &TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001243}
1244
Dan Gohmanb3579832010-04-15 17:08:50 +00001245namespace {
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001246
Chris Lattner87913512009-09-21 06:30:24 +00001247struct AvailableValueInBlock {
1248 /// BB - The basic block in question.
1249 BasicBlock *BB;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001250 enum ValType {
1251 SimpleVal, // A simple offsetted value that is accessed.
1252 MemIntrin // A memory intrinsic which is loaded from.
1253 };
1254
Chris Lattner87913512009-09-21 06:30:24 +00001255 /// V - The value that is live out of the block.
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001256 PointerIntPair<Value *, 1, ValType> Val;
1257
1258 /// Offset - The byte offset in Val that is interesting for the load query.
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001259 unsigned Offset;
Chris Lattner87913512009-09-21 06:30:24 +00001260
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001261 static AvailableValueInBlock get(BasicBlock *BB, Value *V,
1262 unsigned Offset = 0) {
Chris Lattner87913512009-09-21 06:30:24 +00001263 AvailableValueInBlock Res;
1264 Res.BB = BB;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001265 Res.Val.setPointer(V);
1266 Res.Val.setInt(SimpleVal);
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001267 Res.Offset = Offset;
Chris Lattner87913512009-09-21 06:30:24 +00001268 return Res;
1269 }
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001270
1271 static AvailableValueInBlock getMI(BasicBlock *BB, MemIntrinsic *MI,
1272 unsigned Offset = 0) {
1273 AvailableValueInBlock Res;
1274 Res.BB = BB;
1275 Res.Val.setPointer(MI);
1276 Res.Val.setInt(MemIntrin);
1277 Res.Offset = Offset;
1278 return Res;
1279 }
1280
1281 bool isSimpleValue() const { return Val.getInt() == SimpleVal; }
1282 Value *getSimpleValue() const {
1283 assert(isSimpleValue() && "Wrong accessor");
1284 return Val.getPointer();
1285 }
1286
1287 MemIntrinsic *getMemIntrinValue() const {
1288 assert(!isSimpleValue() && "Wrong accessor");
1289 return cast<MemIntrinsic>(Val.getPointer());
1290 }
Chris Lattner5362c542009-12-21 23:04:33 +00001291
1292 /// MaterializeAdjustedValue - Emit code into this block to adjust the value
1293 /// defined here to the specified type. This handles various coercion cases.
1294 Value *MaterializeAdjustedValue(const Type *LoadTy,
1295 const TargetData *TD) const {
1296 Value *Res;
1297 if (isSimpleValue()) {
1298 Res = getSimpleValue();
1299 if (Res->getType() != LoadTy) {
1300 assert(TD && "Need target data to handle type mismatch case");
1301 Res = GetStoreValueForLoad(Res, Offset, LoadTy, BB->getTerminator(),
1302 *TD);
1303
1304 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\nOffset: " << Offset << " "
1305 << *getSimpleValue() << '\n'
1306 << *Res << '\n' << "\n\n\n");
1307 }
1308 } else {
1309 Res = GetMemInstValueForLoad(getMemIntrinValue(), Offset,
1310 LoadTy, BB->getTerminator(), *TD);
1311 DEBUG(errs() << "GVN COERCED NONLOCAL MEM INTRIN:\nOffset: " << Offset
1312 << " " << *getMemIntrinValue() << '\n'
1313 << *Res << '\n' << "\n\n\n");
1314 }
1315 return Res;
1316 }
Chris Lattner87913512009-09-21 06:30:24 +00001317};
1318
Dan Gohmanb3579832010-04-15 17:08:50 +00001319}
1320
Chris Lattnera09fbf02009-10-10 23:50:30 +00001321/// ConstructSSAForLoadSet - Given a set of loads specified by ValuesPerBlock,
1322/// construct SSA form, allowing us to eliminate LI. This returns the value
1323/// that should be used at LI's definition site.
1324static Value *ConstructSSAForLoadSet(LoadInst *LI,
1325 SmallVectorImpl<AvailableValueInBlock> &ValuesPerBlock,
1326 const TargetData *TD,
Chris Lattnerd2191e52009-12-21 23:15:48 +00001327 const DominatorTree &DT,
Chris Lattnera09fbf02009-10-10 23:50:30 +00001328 AliasAnalysis *AA) {
Chris Lattnerd2191e52009-12-21 23:15:48 +00001329 // Check for the fully redundant, dominating load case. In this case, we can
1330 // just use the dominating value directly.
1331 if (ValuesPerBlock.size() == 1 &&
1332 DT.properlyDominates(ValuesPerBlock[0].BB, LI->getParent()))
1333 return ValuesPerBlock[0].MaterializeAdjustedValue(LI->getType(), TD);
1334
1335 // Otherwise, we have to construct SSA form.
Chris Lattnera09fbf02009-10-10 23:50:30 +00001336 SmallVector<PHINode*, 8> NewPHIs;
1337 SSAUpdater SSAUpdate(&NewPHIs);
Duncan Sandsfc6e29d2010-09-02 08:14:03 +00001338 SSAUpdate.Initialize(LI->getType(), LI->getName());
Chris Lattnera09fbf02009-10-10 23:50:30 +00001339
1340 const Type *LoadTy = LI->getType();
1341
Chris Lattner771a5422009-09-20 20:09:34 +00001342 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001343 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1344 BasicBlock *BB = AV.BB;
Chris Lattner771a5422009-09-20 20:09:34 +00001345
Chris Lattnera09fbf02009-10-10 23:50:30 +00001346 if (SSAUpdate.HasValueForBlock(BB))
1347 continue;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001348
Chris Lattner5362c542009-12-21 23:04:33 +00001349 SSAUpdate.AddAvailableValue(BB, AV.MaterializeAdjustedValue(LoadTy, TD));
Chris Lattner771a5422009-09-20 20:09:34 +00001350 }
Chris Lattnera09fbf02009-10-10 23:50:30 +00001351
1352 // Perform PHI construction.
1353 Value *V = SSAUpdate.GetValueInMiddleOfBlock(LI->getParent());
1354
1355 // If new PHI nodes were created, notify alias analysis.
Duncan Sands1df98592010-02-16 11:11:14 +00001356 if (V->getType()->isPointerTy())
Chris Lattnera09fbf02009-10-10 23:50:30 +00001357 for (unsigned i = 0, e = NewPHIs.size(); i != e; ++i)
1358 AA->copyValue(LI, NewPHIs[i]);
1359
1360 return V;
Chris Lattner771a5422009-09-20 20:09:34 +00001361}
1362
Gabor Greifea3eec92010-04-09 10:57:00 +00001363static bool isLifetimeStart(const Instruction *Inst) {
1364 if (const IntrinsicInst* II = dyn_cast<IntrinsicInst>(Inst))
Owen Anderson9ff5a232009-12-02 07:35:19 +00001365 return II->getIntrinsicID() == Intrinsic::lifetime_start;
Chris Lattner720e7902009-12-02 06:44:58 +00001366 return false;
1367}
1368
Owen Anderson62bc33c2007-08-16 22:02:55 +00001369/// processNonLocalLoad - Attempt to eliminate a load whose dependencies are
1370/// non-local by performing PHI construction.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001371bool GVN::processNonLocalLoad(LoadInst *LI,
Chris Lattner8e1e95c2008-03-21 22:01:16 +00001372 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001373 // Find the non-local dependencies of the load.
Chris Lattner0ee443d2009-12-22 04:25:02 +00001374 SmallVector<NonLocalDepResult, 64> Deps;
Dan Gohman6d8eb152010-11-11 21:50:19 +00001375 AliasAnalysis::Location Loc = VN.getAliasAnalysis()->getLocation(LI);
1376 MD->getNonLocalPointerDependency(Loc, true, LI->getParent(), Deps);
David Greenebf7f78e2010-01-05 01:27:17 +00001377 //DEBUG(dbgs() << "INVESTIGATING NONLOCAL LOAD: "
Dan Gohman2a298992009-07-31 20:24:18 +00001378 // << Deps.size() << *LI << '\n');
Daniel Dunbara279bc32009-09-20 02:20:51 +00001379
Owen Anderson516eb1c2008-08-26 22:07:42 +00001380 // If we had to process more than one hundred blocks to find the
1381 // dependencies, this load isn't worth worrying about. Optimizing
1382 // it will be too expensive.
Chris Lattner91bcf642008-12-09 19:25:07 +00001383 if (Deps.size() > 100)
Owen Anderson516eb1c2008-08-26 22:07:42 +00001384 return false;
Chris Lattner5f4f84b2008-12-18 00:51:32 +00001385
1386 // If we had a phi translation failure, we'll have a single entry which is a
1387 // clobber in the current block. Reject this early.
Chris Lattnere18b9712009-12-09 07:08:01 +00001388 if (Deps.size() == 1 && Deps[0].getResult().isClobber()) {
Torok Edwin4306b1a2009-06-17 18:48:18 +00001389 DEBUG(
David Greenebf7f78e2010-01-05 01:27:17 +00001390 dbgs() << "GVN: non-local load ";
1391 WriteAsOperand(dbgs(), LI);
1392 dbgs() << " is clobbered by " << *Deps[0].getResult().getInst() << '\n';
Torok Edwin4306b1a2009-06-17 18:48:18 +00001393 );
Chris Lattner5f4f84b2008-12-18 00:51:32 +00001394 return false;
Torok Edwin4306b1a2009-06-17 18:48:18 +00001395 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001396
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001397 // Filter out useless results (non-locals, etc). Keep track of the blocks
1398 // where we have a value available in repl, also keep track of whether we see
1399 // dependencies that produce an unknown value for the load (such as a call
1400 // that could potentially clobber the load).
Chris Lattner87913512009-09-21 06:30:24 +00001401 SmallVector<AvailableValueInBlock, 16> ValuesPerBlock;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001402 SmallVector<BasicBlock*, 16> UnavailableBlocks;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001403
Chris Lattner91bcf642008-12-09 19:25:07 +00001404 for (unsigned i = 0, e = Deps.size(); i != e; ++i) {
Chris Lattnere18b9712009-12-09 07:08:01 +00001405 BasicBlock *DepBB = Deps[i].getBB();
1406 MemDepResult DepInfo = Deps[i].getResult();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001407
Chris Lattnerb51deb92008-12-05 21:04:20 +00001408 if (DepInfo.isClobber()) {
Chris Lattneraf064ae2009-12-09 18:21:46 +00001409 // The address being loaded in this non-local block may not be the same as
1410 // the pointer operand of the load if PHI translation occurs. Make sure
1411 // to consider the right address.
1412 Value *Address = Deps[i].getAddress();
1413
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001414 // If the dependence is to a store that writes to a superset of the bits
1415 // read by the load, we can extract the bits we need for the load from the
1416 // stored value.
1417 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInfo.getInst())) {
Chris Lattneraf064ae2009-12-09 18:21:46 +00001418 if (TD && Address) {
1419 int Offset = AnalyzeLoadFromClobberingStore(LI->getType(), Address,
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001420 DepSI, *TD);
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001421 if (Offset != -1) {
1422 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
Dan Gohman3355c4e2010-11-10 19:03:33 +00001423 DepSI->getValueOperand(),
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001424 Offset));
1425 continue;
1426 }
1427 }
1428 }
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001429
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001430 // If the clobbering value is a memset/memcpy/memmove, see if we can
1431 // forward a value on from it.
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001432 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(DepInfo.getInst())) {
Chris Lattneraf064ae2009-12-09 18:21:46 +00001433 if (TD && Address) {
1434 int Offset = AnalyzeLoadFromClobberingMemInst(LI->getType(), Address,
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001435 DepMI, *TD);
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001436 if (Offset != -1) {
1437 ValuesPerBlock.push_back(AvailableValueInBlock::getMI(DepBB, DepMI,
1438 Offset));
1439 continue;
1440 }
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001441 }
1442 }
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001443
Chris Lattnerb51deb92008-12-05 21:04:20 +00001444 UnavailableBlocks.push_back(DepBB);
1445 continue;
1446 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001447
Chris Lattnerb51deb92008-12-05 21:04:20 +00001448 Instruction *DepInst = DepInfo.getInst();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001449
Chris Lattnerb51deb92008-12-05 21:04:20 +00001450 // Loading the allocation -> undef.
Chris Lattner720e7902009-12-02 06:44:58 +00001451 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst) ||
Owen Anderson9ff5a232009-12-02 07:35:19 +00001452 // Loading immediately after lifetime begin -> undef.
1453 isLifetimeStart(DepInst)) {
Chris Lattner87913512009-09-21 06:30:24 +00001454 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1455 UndefValue::get(LI->getType())));
Chris Lattnerbf145d62008-12-01 01:15:42 +00001456 continue;
1457 }
Owen Andersonb62f7922009-10-28 07:05:35 +00001458
Chris Lattner87913512009-09-21 06:30:24 +00001459 if (StoreInst *S = dyn_cast<StoreInst>(DepInst)) {
Daniel Dunbara279bc32009-09-20 02:20:51 +00001460 // Reject loads and stores that are to the same address but are of
Chris Lattner771a5422009-09-20 20:09:34 +00001461 // different types if we have to.
Dan Gohman3355c4e2010-11-10 19:03:33 +00001462 if (S->getValueOperand()->getType() != LI->getType()) {
Chris Lattner771a5422009-09-20 20:09:34 +00001463 // If the stored value is larger or equal to the loaded value, we can
1464 // reuse it.
Dan Gohman3355c4e2010-11-10 19:03:33 +00001465 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(S->getValueOperand(),
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001466 LI->getType(), *TD)) {
Chris Lattner771a5422009-09-20 20:09:34 +00001467 UnavailableBlocks.push_back(DepBB);
1468 continue;
1469 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001470 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001471
Chris Lattner87913512009-09-21 06:30:24 +00001472 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
Dan Gohman3355c4e2010-11-10 19:03:33 +00001473 S->getValueOperand()));
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001474 continue;
1475 }
1476
1477 if (LoadInst *LD = dyn_cast<LoadInst>(DepInst)) {
Chris Lattner771a5422009-09-20 20:09:34 +00001478 // If the types mismatch and we can't handle it, reject reuse of the load.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001479 if (LD->getType() != LI->getType()) {
Chris Lattner771a5422009-09-20 20:09:34 +00001480 // If the stored value is larger or equal to the loaded value, we can
1481 // reuse it.
Chris Lattner8b2bc3d2009-09-21 17:24:04 +00001482 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(LD, LI->getType(),*TD)){
Chris Lattner771a5422009-09-20 20:09:34 +00001483 UnavailableBlocks.push_back(DepBB);
1484 continue;
1485 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001486 }
Chris Lattner87913512009-09-21 06:30:24 +00001487 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB, LD));
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001488 continue;
Owen Anderson0cd32032007-07-25 19:57:03 +00001489 }
Chris Lattner4fbd14e2009-09-21 06:48:08 +00001490
1491 UnavailableBlocks.push_back(DepBB);
1492 continue;
Chris Lattner88365bb2008-03-21 21:14:38 +00001493 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001494
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001495 // If we have no predecessors that produce a known value for this load, exit
1496 // early.
1497 if (ValuesPerBlock.empty()) return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001498
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001499 // If all of the instructions we depend on produce a known value for this
1500 // load, then it is fully redundant and we can use PHI insertion to compute
1501 // its value. Insert PHIs and remove the fully redundant value now.
1502 if (UnavailableBlocks.empty()) {
David Greenebf7f78e2010-01-05 01:27:17 +00001503 DEBUG(dbgs() << "GVN REMOVING NONLOCAL LOAD: " << *LI << '\n');
Chris Lattner771a5422009-09-20 20:09:34 +00001504
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001505 // Perform PHI construction.
Chris Lattnerd2191e52009-12-21 23:15:48 +00001506 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnera09fbf02009-10-10 23:50:30 +00001507 VN.getAliasAnalysis());
Chris Lattner771a5422009-09-20 20:09:34 +00001508 LI->replaceAllUsesWith(V);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001509
Chris Lattner771a5422009-09-20 20:09:34 +00001510 if (isa<PHINode>(V))
1511 V->takeName(LI);
Duncan Sands1df98592010-02-16 11:11:14 +00001512 if (V->getType()->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +00001513 MD->invalidateCachedPointerInfo(V);
Bob Wilson74175c22010-02-22 21:39:41 +00001514 VN.erase(LI);
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001515 toErase.push_back(LI);
Dan Gohmanfe601042010-06-22 15:08:57 +00001516 ++NumGVNLoad;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001517 return true;
1518 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001519
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001520 if (!EnablePRE || !EnableLoadPRE)
1521 return false;
1522
1523 // Okay, we have *some* definitions of the value. This means that the value
1524 // is available in some of our (transitive) predecessors. Lets think about
1525 // doing PRE of this load. This will involve inserting a new load into the
1526 // predecessor when it's not available. We could do this in general, but
1527 // prefer to not increase code size. As such, we only do this when we know
1528 // that we only have to insert *one* load (which means we're basically moving
1529 // the load, not inserting a new one).
Daniel Dunbara279bc32009-09-20 02:20:51 +00001530
Owen Anderson88554df2009-05-31 09:03:40 +00001531 SmallPtrSet<BasicBlock *, 4> Blockers;
1532 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1533 Blockers.insert(UnavailableBlocks[i]);
1534
1535 // Lets find first basic block with more than one predecessor. Walk backwards
1536 // through predecessors if needed.
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001537 BasicBlock *LoadBB = LI->getParent();
Owen Anderson88554df2009-05-31 09:03:40 +00001538 BasicBlock *TmpBB = LoadBB;
1539
1540 bool isSinglePred = false;
Dale Johannesen42c3f552009-06-17 20:48:23 +00001541 bool allSingleSucc = true;
Owen Anderson88554df2009-05-31 09:03:40 +00001542 while (TmpBB->getSinglePredecessor()) {
1543 isSinglePred = true;
1544 TmpBB = TmpBB->getSinglePredecessor();
Owen Anderson88554df2009-05-31 09:03:40 +00001545 if (TmpBB == LoadBB) // Infinite (unreachable) loop.
1546 return false;
1547 if (Blockers.count(TmpBB))
1548 return false;
Owen Andersonb0ba0f42010-09-25 05:26:18 +00001549
1550 // If any of these blocks has more than one successor (i.e. if the edge we
1551 // just traversed was critical), then there are other paths through this
1552 // block along which the load may not be anticipated. Hoisting the load
1553 // above this block would be adding the load to execution paths along
1554 // which it was not previously executed.
Dale Johannesen42c3f552009-06-17 20:48:23 +00001555 if (TmpBB->getTerminator()->getNumSuccessors() != 1)
Owen Andersonb0ba0f42010-09-25 05:26:18 +00001556 return false;
Owen Anderson88554df2009-05-31 09:03:40 +00001557 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001558
Owen Anderson88554df2009-05-31 09:03:40 +00001559 assert(TmpBB);
1560 LoadBB = TmpBB;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001561
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001562 // FIXME: It is extremely unclear what this loop is doing, other than
1563 // artificially restricting loadpre.
Owen Anderson88554df2009-05-31 09:03:40 +00001564 if (isSinglePred) {
1565 bool isHot = false;
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001566 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
1567 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1568 if (AV.isSimpleValue())
Daniel Dunbara279bc32009-09-20 02:20:51 +00001569 // "Hot" Instruction is in some loop (because it dominates its dep.
1570 // instruction).
Chris Lattnercb9cbc42009-12-06 04:54:31 +00001571 if (Instruction *I = dyn_cast<Instruction>(AV.getSimpleValue()))
1572 if (DT->dominates(LI, I)) {
1573 isHot = true;
1574 break;
1575 }
1576 }
Owen Anderson88554df2009-05-31 09:03:40 +00001577
1578 // We are interested only in "hot" instructions. We don't want to do any
1579 // mis-optimizations here.
1580 if (!isHot)
1581 return false;
1582 }
1583
Bob Wilson6cad4172010-02-01 21:17:14 +00001584 // Check to see how many predecessors have the loaded value fully
1585 // available.
1586 DenseMap<BasicBlock*, Value*> PredLoads;
Chris Lattner72bc70d2008-12-05 07:49:08 +00001587 DenseMap<BasicBlock*, char> FullyAvailableBlocks;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001588 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner87913512009-09-21 06:30:24 +00001589 FullyAvailableBlocks[ValuesPerBlock[i].BB] = true;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001590 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1591 FullyAvailableBlocks[UnavailableBlocks[i]] = false;
1592
Bob Wilson34414a62010-05-04 20:03:21 +00001593 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> NeedToSplit;
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001594 for (pred_iterator PI = pred_begin(LoadBB), E = pred_end(LoadBB);
1595 PI != E; ++PI) {
Bob Wilson6cad4172010-02-01 21:17:14 +00001596 BasicBlock *Pred = *PI;
1597 if (IsValueFullyAvailableInBlock(Pred, FullyAvailableBlocks)) {
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001598 continue;
Bob Wilson6cad4172010-02-01 21:17:14 +00001599 }
1600 PredLoads[Pred] = 0;
Bob Wilson484d4a32010-02-16 19:51:59 +00001601
Bob Wilson6cad4172010-02-01 21:17:14 +00001602 if (Pred->getTerminator()->getNumSuccessors() != 1) {
Bob Wilson484d4a32010-02-16 19:51:59 +00001603 if (isa<IndirectBrInst>(Pred->getTerminator())) {
1604 DEBUG(dbgs() << "COULD NOT PRE LOAD BECAUSE OF INDBR CRITICAL EDGE '"
1605 << Pred->getName() << "': " << *LI << '\n');
1606 return false;
1607 }
Bob Wilsonae23daf2010-02-16 21:06:42 +00001608 unsigned SuccNum = GetSuccessorNumber(Pred, LoadBB);
Bob Wilson34414a62010-05-04 20:03:21 +00001609 NeedToSplit.push_back(std::make_pair(Pred->getTerminator(), SuccNum));
Bob Wilson6cad4172010-02-01 21:17:14 +00001610 }
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001611 }
Bob Wilson34414a62010-05-04 20:03:21 +00001612 if (!NeedToSplit.empty()) {
Bob Wilsonbc786532010-05-05 20:44:15 +00001613 toSplit.append(NeedToSplit.begin(), NeedToSplit.end());
Bob Wilson70704972010-03-01 23:37:32 +00001614 return false;
Bob Wilson34414a62010-05-04 20:03:21 +00001615 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001616
Bob Wilson6cad4172010-02-01 21:17:14 +00001617 // Decide whether PRE is profitable for this load.
1618 unsigned NumUnavailablePreds = PredLoads.size();
1619 assert(NumUnavailablePreds != 0 &&
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001620 "Fully available value should be eliminated above!");
Owen Anderson7267e142010-10-01 20:02:55 +00001621
1622 // If this load is unavailable in multiple predecessors, reject it.
1623 // FIXME: If we could restructure the CFG, we could make a common pred with
1624 // all the preds that don't have an available LI and insert a new load into
1625 // that one block.
1626 if (NumUnavailablePreds != 1)
Bob Wilson6cad4172010-02-01 21:17:14 +00001627 return false;
Bob Wilson6cad4172010-02-01 21:17:14 +00001628
1629 // Check if the load can safely be moved to all the unavailable predecessors.
1630 bool CanDoPRE = true;
Chris Lattnerdd696052009-11-28 15:39:14 +00001631 SmallVector<Instruction*, 8> NewInsts;
Bob Wilson6cad4172010-02-01 21:17:14 +00001632 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1633 E = PredLoads.end(); I != E; ++I) {
1634 BasicBlock *UnavailablePred = I->first;
1635
1636 // Do PHI translation to get its value in the predecessor if necessary. The
1637 // returned pointer (if non-null) is guaranteed to dominate UnavailablePred.
1638
1639 // If all preds have a single successor, then we know it is safe to insert
1640 // the load on the pred (?!?), so we can insert code to materialize the
1641 // pointer if it is not available.
Dan Gohman3355c4e2010-11-10 19:03:33 +00001642 PHITransAddr Address(LI->getPointerOperand(), TD);
Bob Wilson6cad4172010-02-01 21:17:14 +00001643 Value *LoadPtr = 0;
1644 if (allSingleSucc) {
1645 LoadPtr = Address.PHITranslateWithInsertion(LoadBB, UnavailablePred,
1646 *DT, NewInsts);
1647 } else {
Daniel Dunbar6d8f2ca2010-02-24 08:48:04 +00001648 Address.PHITranslateValue(LoadBB, UnavailablePred, DT);
Bob Wilson6cad4172010-02-01 21:17:14 +00001649 LoadPtr = Address.getAddr();
Bob Wilson6cad4172010-02-01 21:17:14 +00001650 }
1651
1652 // If we couldn't find or insert a computation of this phi translated value,
1653 // we fail PRE.
1654 if (LoadPtr == 0) {
1655 DEBUG(dbgs() << "COULDN'T INSERT PHI TRANSLATED VALUE OF: "
Dan Gohman3355c4e2010-11-10 19:03:33 +00001656 << *LI->getPointerOperand() << "\n");
Bob Wilson6cad4172010-02-01 21:17:14 +00001657 CanDoPRE = false;
1658 break;
1659 }
1660
1661 // Make sure it is valid to move this load here. We have to watch out for:
1662 // @1 = getelementptr (i8* p, ...
1663 // test p and branch if == 0
1664 // load @1
1665 // It is valid to have the getelementptr before the test, even if p can be 0,
1666 // as getelementptr only does address arithmetic.
1667 // If we are not pushing the value through any multiple-successor blocks
1668 // we do not have this case. Otherwise, check that the load is safe to
1669 // put anywhere; this can be improved, but should be conservatively safe.
1670 if (!allSingleSucc &&
1671 // FIXME: REEVALUTE THIS.
1672 !isSafeToLoadUnconditionally(LoadPtr,
1673 UnavailablePred->getTerminator(),
1674 LI->getAlignment(), TD)) {
1675 CanDoPRE = false;
1676 break;
1677 }
1678
1679 I->second = LoadPtr;
Chris Lattner05e15f82009-12-09 01:59:31 +00001680 }
1681
Bob Wilson6cad4172010-02-01 21:17:14 +00001682 if (!CanDoPRE) {
1683 while (!NewInsts.empty())
1684 NewInsts.pop_back_val()->eraseFromParent();
Dale Johannesen42c3f552009-06-17 20:48:23 +00001685 return false;
Chris Lattner0c264b12009-11-28 16:08:18 +00001686 }
Dale Johannesen42c3f552009-06-17 20:48:23 +00001687
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001688 // Okay, we can eliminate this load by inserting a reload in the predecessor
1689 // and using PHI construction to get the value in the other predecessors, do
1690 // it.
David Greenebf7f78e2010-01-05 01:27:17 +00001691 DEBUG(dbgs() << "GVN REMOVING PRE LOAD: " << *LI << '\n');
Chris Lattner0c264b12009-11-28 16:08:18 +00001692 DEBUG(if (!NewInsts.empty())
David Greenebf7f78e2010-01-05 01:27:17 +00001693 dbgs() << "INSERTED " << NewInsts.size() << " INSTS: "
Chris Lattner0c264b12009-11-28 16:08:18 +00001694 << *NewInsts.back() << '\n');
1695
Bob Wilson6cad4172010-02-01 21:17:14 +00001696 // Assign value numbers to the new instructions.
1697 for (unsigned i = 0, e = NewInsts.size(); i != e; ++i) {
1698 // FIXME: We really _ought_ to insert these value numbers into their
1699 // parent's availability map. However, in doing so, we risk getting into
1700 // ordering issues. If a block hasn't been processed yet, we would be
1701 // marking a value as AVAIL-IN, which isn't what we intend.
1702 VN.lookup_or_add(NewInsts[i]);
1703 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001704
Bob Wilson6cad4172010-02-01 21:17:14 +00001705 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1706 E = PredLoads.end(); I != E; ++I) {
1707 BasicBlock *UnavailablePred = I->first;
1708 Value *LoadPtr = I->second;
1709
1710 Value *NewLoad = new LoadInst(LoadPtr, LI->getName()+".pre", false,
1711 LI->getAlignment(),
1712 UnavailablePred->getTerminator());
1713
1714 // Add the newly created load.
1715 ValuesPerBlock.push_back(AvailableValueInBlock::get(UnavailablePred,
1716 NewLoad));
Bob Wilson188f4282010-02-23 05:55:00 +00001717 MD->invalidateCachedPointerInfo(LoadPtr);
1718 DEBUG(dbgs() << "GVN INSERTED " << *NewLoad << '\n');
Bob Wilson6cad4172010-02-01 21:17:14 +00001719 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001720
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001721 // Perform PHI construction.
Chris Lattnerd2191e52009-12-21 23:15:48 +00001722 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnera09fbf02009-10-10 23:50:30 +00001723 VN.getAliasAnalysis());
Chris Lattner771a5422009-09-20 20:09:34 +00001724 LI->replaceAllUsesWith(V);
1725 if (isa<PHINode>(V))
1726 V->takeName(LI);
Duncan Sands1df98592010-02-16 11:11:14 +00001727 if (V->getType()->isPointerTy())
Chris Lattner771a5422009-09-20 20:09:34 +00001728 MD->invalidateCachedPointerInfo(V);
Bob Wilson74175c22010-02-22 21:39:41 +00001729 VN.erase(LI);
Chris Lattnerc89c6a92008-12-02 08:16:11 +00001730 toErase.push_back(LI);
Dan Gohmanfe601042010-06-22 15:08:57 +00001731 ++NumPRELoad;
Owen Anderson0cd32032007-07-25 19:57:03 +00001732 return true;
1733}
1734
Owen Anderson62bc33c2007-08-16 22:02:55 +00001735/// processLoad - Attempt to eliminate a load, first by eliminating it
1736/// locally, and then attempting non-local elimination if that fails.
Chris Lattnerb51deb92008-12-05 21:04:20 +00001737bool GVN::processLoad(LoadInst *L, SmallVectorImpl<Instruction*> &toErase) {
Dan Gohman4ec01b22009-11-14 02:27:51 +00001738 if (!MD)
1739 return false;
1740
Chris Lattnerb51deb92008-12-05 21:04:20 +00001741 if (L->isVolatile())
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001742 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001743
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001744 // ... to a pointer that has been loaded from before...
Chris Lattnerb2412a82009-09-21 02:42:51 +00001745 MemDepResult Dep = MD->getDependency(L);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001746
Chris Lattnerb51deb92008-12-05 21:04:20 +00001747 // If the value isn't available, don't do anything!
Chris Lattnerb2412a82009-09-21 02:42:51 +00001748 if (Dep.isClobber()) {
Chris Lattnereed919b2009-09-21 05:57:11 +00001749 // Check to see if we have something like this:
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001750 // store i32 123, i32* %P
1751 // %A = bitcast i32* %P to i8*
1752 // %B = gep i8* %A, i32 1
1753 // %C = load i8* %B
1754 //
1755 // We could do that by recognizing if the clobber instructions are obviously
1756 // a common base + constant offset, and if the previous store (or memset)
1757 // completely covers this load. This sort of thing can happen in bitfield
1758 // access code.
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001759 Value *AvailVal = 0;
Chris Lattnereed919b2009-09-21 05:57:11 +00001760 if (StoreInst *DepSI = dyn_cast<StoreInst>(Dep.getInst()))
Duncan Sands88c3df72010-11-12 21:10:24 +00001761 if (TD) {
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001762 int Offset = AnalyzeLoadFromClobberingStore(L->getType(),
1763 L->getPointerOperand(),
1764 DepSI, *TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001765 if (Offset != -1)
Dan Gohman3355c4e2010-11-10 19:03:33 +00001766 AvailVal = GetStoreValueForLoad(DepSI->getValueOperand(), Offset,
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001767 L->getType(), L, *TD);
Chris Lattner1ce08292009-09-21 06:22:46 +00001768 }
Chris Lattnereed919b2009-09-21 05:57:11 +00001769
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001770 // If the clobbering value is a memset/memcpy/memmove, see if we can forward
1771 // a value on from it.
1772 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(Dep.getInst())) {
Duncan Sands88c3df72010-11-12 21:10:24 +00001773 if (TD) {
Chris Lattner4ca70fe2009-12-09 07:37:07 +00001774 int Offset = AnalyzeLoadFromClobberingMemInst(L->getType(),
1775 L->getPointerOperand(),
1776 DepMI, *TD);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001777 if (Offset != -1)
1778 AvailVal = GetMemInstValueForLoad(DepMI, Offset, L->getType(), L,*TD);
1779 }
1780 }
1781
1782 if (AvailVal) {
David Greenebf7f78e2010-01-05 01:27:17 +00001783 DEBUG(dbgs() << "GVN COERCED INST:\n" << *Dep.getInst() << '\n'
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001784 << *AvailVal << '\n' << *L << "\n\n\n");
1785
1786 // Replace the load!
1787 L->replaceAllUsesWith(AvailVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001788 if (AvailVal->getType()->isPointerTy())
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001789 MD->invalidateCachedPointerInfo(AvailVal);
Bob Wilson74175c22010-02-22 21:39:41 +00001790 VN.erase(L);
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001791 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001792 ++NumGVNLoad;
Chris Lattnerfaf815b2009-12-06 01:57:02 +00001793 return true;
1794 }
1795
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001796 DEBUG(
1797 // fast print dep, using operator<< on instruction would be too slow
David Greenebf7f78e2010-01-05 01:27:17 +00001798 dbgs() << "GVN: load ";
1799 WriteAsOperand(dbgs(), L);
Chris Lattnerb2412a82009-09-21 02:42:51 +00001800 Instruction *I = Dep.getInst();
David Greenebf7f78e2010-01-05 01:27:17 +00001801 dbgs() << " is clobbered by " << *I << '\n';
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001802 );
Chris Lattnerb51deb92008-12-05 21:04:20 +00001803 return false;
Torok Edwin3f3c6d42009-05-29 09:46:03 +00001804 }
Chris Lattnerb51deb92008-12-05 21:04:20 +00001805
1806 // If it is defined in another block, try harder.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001807 if (Dep.isNonLocal())
Chris Lattnerb51deb92008-12-05 21:04:20 +00001808 return processNonLocalLoad(L, toErase);
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001809
Chris Lattnerb2412a82009-09-21 02:42:51 +00001810 Instruction *DepInst = Dep.getInst();
Chris Lattnerb51deb92008-12-05 21:04:20 +00001811 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInst)) {
Dan Gohman3355c4e2010-11-10 19:03:33 +00001812 Value *StoredVal = DepSI->getValueOperand();
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001813
1814 // The store and load are to a must-aliased pointer, but they may not
1815 // actually have the same type. See if we know how to reuse the stored
1816 // value (depending on its type).
Chris Lattnera52fce42009-10-21 04:11:19 +00001817 if (StoredVal->getType() != L->getType()) {
Duncan Sands88c3df72010-11-12 21:10:24 +00001818 if (TD) {
Chris Lattnera52fce42009-10-21 04:11:19 +00001819 StoredVal = CoerceAvailableValueToLoadType(StoredVal, L->getType(),
1820 L, *TD);
1821 if (StoredVal == 0)
1822 return false;
1823
David Greenebf7f78e2010-01-05 01:27:17 +00001824 DEBUG(dbgs() << "GVN COERCED STORE:\n" << *DepSI << '\n' << *StoredVal
Chris Lattnera52fce42009-10-21 04:11:19 +00001825 << '\n' << *L << "\n\n\n");
1826 }
1827 else
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001828 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001829 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001830
Chris Lattnerb51deb92008-12-05 21:04:20 +00001831 // Remove it!
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001832 L->replaceAllUsesWith(StoredVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001833 if (StoredVal->getType()->isPointerTy())
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001834 MD->invalidateCachedPointerInfo(StoredVal);
Bob Wilson74175c22010-02-22 21:39:41 +00001835 VN.erase(L);
Chris Lattnerb51deb92008-12-05 21:04:20 +00001836 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001837 ++NumGVNLoad;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001838 return true;
1839 }
1840
1841 if (LoadInst *DepLI = dyn_cast<LoadInst>(DepInst)) {
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001842 Value *AvailableVal = DepLI;
1843
1844 // The loads are of a must-aliased pointer, but they may not actually have
1845 // the same type. See if we know how to reuse the previously loaded value
1846 // (depending on its type).
Chris Lattnera52fce42009-10-21 04:11:19 +00001847 if (DepLI->getType() != L->getType()) {
Duncan Sands88c3df72010-11-12 21:10:24 +00001848 if (TD) {
Chris Lattnera52fce42009-10-21 04:11:19 +00001849 AvailableVal = CoerceAvailableValueToLoadType(DepLI, L->getType(), L,*TD);
1850 if (AvailableVal == 0)
1851 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001852
David Greenebf7f78e2010-01-05 01:27:17 +00001853 DEBUG(dbgs() << "GVN COERCED LOAD:\n" << *DepLI << "\n" << *AvailableVal
Chris Lattnera52fce42009-10-21 04:11:19 +00001854 << "\n" << *L << "\n\n\n");
1855 }
1856 else
1857 return false;
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001858 }
1859
Chris Lattnerb51deb92008-12-05 21:04:20 +00001860 // Remove it!
Chris Lattnerbb6495c2009-09-20 19:03:47 +00001861 L->replaceAllUsesWith(AvailableVal);
Duncan Sands1df98592010-02-16 11:11:14 +00001862 if (DepLI->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001863 MD->invalidateCachedPointerInfo(DepLI);
Bob Wilson74175c22010-02-22 21:39:41 +00001864 VN.erase(L);
Chris Lattnerb51deb92008-12-05 21:04:20 +00001865 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001866 ++NumGVNLoad;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001867 return true;
1868 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001869
Chris Lattner237a8282008-11-30 01:39:32 +00001870 // If this load really doesn't depend on anything, then we must be loading an
1871 // undef value. This can happen when loading for a fresh allocation with no
1872 // intervening stores, for example.
Victor Hernandez7b929da2009-10-23 21:09:37 +00001873 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst)) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001874 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson74175c22010-02-22 21:39:41 +00001875 VN.erase(L);
Chris Lattner237a8282008-11-30 01:39:32 +00001876 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001877 ++NumGVNLoad;
Chris Lattnerb51deb92008-12-05 21:04:20 +00001878 return true;
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001879 }
Owen Andersonb62f7922009-10-28 07:05:35 +00001880
Owen Anderson9ff5a232009-12-02 07:35:19 +00001881 // If this load occurs either right after a lifetime begin,
Owen Andersonb62f7922009-10-28 07:05:35 +00001882 // then the loaded value is undefined.
1883 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(DepInst)) {
Owen Anderson9ff5a232009-12-02 07:35:19 +00001884 if (II->getIntrinsicID() == Intrinsic::lifetime_start) {
Owen Andersonb62f7922009-10-28 07:05:35 +00001885 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson74175c22010-02-22 21:39:41 +00001886 VN.erase(L);
Owen Andersonb62f7922009-10-28 07:05:35 +00001887 toErase.push_back(L);
Dan Gohmanfe601042010-06-22 15:08:57 +00001888 ++NumGVNLoad;
Owen Andersonb62f7922009-10-28 07:05:35 +00001889 return true;
1890 }
1891 }
Eli Friedmanb6c36e42008-02-12 12:08:14 +00001892
Chris Lattnerb51deb92008-12-05 21:04:20 +00001893 return false;
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001894}
1895
Owen Anderson68c26392010-11-19 22:48:40 +00001896// lookupNumber - In order to find a leader for a given value number at a
1897// specific basic block, we first obtain the list of all Values for that number,
1898// and then scan the list to find one whose block dominates the block in
1899// question. This is fast because dominator tree queries consist of only
1900// a few comparisons of DFS numbers.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001901Value *GVN::lookupNumber(BasicBlock *BB, uint32_t num) {
Owen Andersona04a0642010-11-18 18:32:40 +00001902 std::pair<Value*, void*> Vals = NumberTable[num];
1903 if (!Vals.first) return 0;
1904 Instruction *Inst = dyn_cast<Instruction>(Vals.first);
1905 if (!Inst) return Vals.first;
1906 BasicBlock *Parent = Inst->getParent();
1907 if (DT->dominates(Parent, BB))
1908 return Inst;
1909
1910 std::pair<Value*, void*>* Next =
1911 static_cast<std::pair<Value*, void*>*>(Vals.second);
1912 while (Next) {
1913 Instruction *CurrInst = dyn_cast<Instruction>(Next->first);
1914 if (!CurrInst) return Next->first;
1915
1916 BasicBlock *Parent = CurrInst->getParent();
1917 if (DT->dominates(Parent, BB))
1918 return CurrInst;
1919
1920 Next = static_cast<std::pair<Value*, void*>*>(Next->second);
Owen Anderson6fafe842008-06-20 01:15:47 +00001921 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001922
Owen Anderson6fafe842008-06-20 01:15:47 +00001923 return 0;
1924}
1925
Owen Anderson255dafc2008-12-15 02:03:00 +00001926
Owen Anderson36057c72007-08-14 18:16:29 +00001927/// processInstruction - When calculating availability, handle an instruction
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001928/// by inserting it into the appropriate sets
Owen Andersonaf4240a2008-06-12 19:25:32 +00001929bool GVN::processInstruction(Instruction *I,
Chris Lattner8e1e95c2008-03-21 22:01:16 +00001930 SmallVectorImpl<Instruction*> &toErase) {
Devang Patelbe905e22010-02-11 00:20:49 +00001931 // Ignore dbg info intrinsics.
1932 if (isa<DbgInfoIntrinsic>(I))
1933 return false;
1934
Duncan Sands88c3df72010-11-12 21:10:24 +00001935 // If the instruction can be easily simplified then do so now in preference
1936 // to value numbering it. Value numbering often exposes redundancies, for
1937 // example if it determines that %y is equal to %x then the instruction
1938 // "%z = and i32 %x, %y" becomes "%z = and i32 %x, %x" which we now simplify.
Duncan Sandseff05812010-11-14 18:36:10 +00001939 if (Value *V = SimplifyInstruction(I, TD, DT)) {
Duncan Sands88c3df72010-11-12 21:10:24 +00001940 I->replaceAllUsesWith(V);
1941 if (MD && V->getType()->isPointerTy())
1942 MD->invalidateCachedPointerInfo(V);
1943 VN.erase(I);
1944 toErase.push_back(I);
1945 return true;
1946 }
1947
Chris Lattnerb2412a82009-09-21 02:42:51 +00001948 if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
1949 bool Changed = processLoad(LI, toErase);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001950
Chris Lattnerb2412a82009-09-21 02:42:51 +00001951 if (!Changed) {
1952 unsigned Num = VN.lookup_or_add(LI);
Owen Andersona04a0642010-11-18 18:32:40 +00001953 insert_table(Num, LI);
Owen Andersonb2303722008-06-18 21:41:49 +00001954 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001955
Chris Lattnerb2412a82009-09-21 02:42:51 +00001956 return Changed;
Owen Andersonb2303722008-06-18 21:41:49 +00001957 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001958
Chris Lattnerb2412a82009-09-21 02:42:51 +00001959 uint32_t NextNum = VN.getNextUnusedValueNumber();
1960 unsigned Num = VN.lookup_or_add(I);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001961
Owen Andersone5ffa902008-04-07 09:59:07 +00001962 // Allocations are always uniquely numbered, so we can save time and memory
Daniel Dunbara279bc32009-09-20 02:20:51 +00001963 // by fast failing them.
Owen Andersona04a0642010-11-18 18:32:40 +00001964 if (isa<AllocaInst>(I) || isa<TerminatorInst>(I)) {
1965 insert_table(Num, I);
Owen Andersone5ffa902008-04-07 09:59:07 +00001966 return false;
Owen Andersonb2303722008-06-18 21:41:49 +00001967 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001968
Duncan Sandsd32c4a12010-11-17 04:05:21 +00001969 if (isa<PHINode>(I)) {
Owen Andersona04a0642010-11-18 18:32:40 +00001970 insert_table(Num, I);
1971
Owen Anderson0ae33ef2008-07-03 17:44:33 +00001972 // If the number we were assigned was a brand new VN, then we don't
1973 // need to do a lookup to see if the number already exists
1974 // somewhere in the domtree: it can't!
Chris Lattnerb2412a82009-09-21 02:42:51 +00001975 } else if (Num == NextNum) {
Owen Andersona04a0642010-11-18 18:32:40 +00001976 insert_table(Num, I);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001977
Owen Anderson255dafc2008-12-15 02:03:00 +00001978 // Perform fast-path value-number based elimination of values inherited from
1979 // dominators.
Chris Lattnerb2412a82009-09-21 02:42:51 +00001980 } else if (Value *repl = lookupNumber(I->getParent(), Num)) {
Owen Anderson5fc4aba2007-12-08 01:37:09 +00001981 // Remove it!
Owen Andersonbf7d0bc2007-07-31 23:27:13 +00001982 VN.erase(I);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001983 I->replaceAllUsesWith(repl);
Duncan Sands1df98592010-02-16 11:11:14 +00001984 if (MD && repl->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00001985 MD->invalidateCachedPointerInfo(repl);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001986 toErase.push_back(I);
1987 return true;
Owen Anderson255dafc2008-12-15 02:03:00 +00001988
Owen Anderson0ae33ef2008-07-03 17:44:33 +00001989 } else {
Owen Andersona04a0642010-11-18 18:32:40 +00001990 insert_table(Num, I);
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001991 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001992
Owen Anderson1ad2cb72007-07-24 17:55:58 +00001993 return false;
1994}
1995
Bill Wendling30788b82008-12-22 22:32:22 +00001996/// runOnFunction - This is the main transformation entry point for a function.
Owen Anderson3e75a422007-08-14 18:04:11 +00001997bool GVN::runOnFunction(Function& F) {
Dan Gohman4ec01b22009-11-14 02:27:51 +00001998 if (!NoLoads)
1999 MD = &getAnalysis<MemoryDependenceAnalysis>();
Chris Lattner663e4412008-12-01 00:40:32 +00002000 DT = &getAnalysis<DominatorTree>();
Duncan Sands88c3df72010-11-12 21:10:24 +00002001 TD = getAnalysisIfAvailable<TargetData>();
Owen Andersona472c4a2008-05-12 20:15:55 +00002002 VN.setAliasAnalysis(&getAnalysis<AliasAnalysis>());
Chris Lattner663e4412008-12-01 00:40:32 +00002003 VN.setMemDep(MD);
2004 VN.setDomTree(DT);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002005
Chris Lattnerb2412a82009-09-21 02:42:51 +00002006 bool Changed = false;
2007 bool ShouldContinue = true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002008
Owen Anderson5d0af032008-07-16 17:52:31 +00002009 // Merge unconditional branches, allowing PRE to catch more
2010 // optimization opportunities.
2011 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002012 BasicBlock *BB = FI;
Owen Anderson5d0af032008-07-16 17:52:31 +00002013 ++FI;
Owen Andersonb31b06d2008-07-17 00:01:40 +00002014 bool removedBlock = MergeBlockIntoPredecessor(BB, this);
Dan Gohmanfe601042010-06-22 15:08:57 +00002015 if (removedBlock) ++NumGVNBlocks;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002016
Chris Lattnerb2412a82009-09-21 02:42:51 +00002017 Changed |= removedBlock;
Owen Anderson5d0af032008-07-16 17:52:31 +00002018 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002019
Chris Lattnerae199312008-12-09 19:21:47 +00002020 unsigned Iteration = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002021
Chris Lattnerb2412a82009-09-21 02:42:51 +00002022 while (ShouldContinue) {
David Greenebf7f78e2010-01-05 01:27:17 +00002023 DEBUG(dbgs() << "GVN iteration: " << Iteration << "\n");
Chris Lattnerb2412a82009-09-21 02:42:51 +00002024 ShouldContinue = iterateOnFunction(F);
Bob Wilson484d4a32010-02-16 19:51:59 +00002025 if (splitCriticalEdges())
2026 ShouldContinue = true;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002027 Changed |= ShouldContinue;
Chris Lattnerae199312008-12-09 19:21:47 +00002028 ++Iteration;
Owen Anderson3e75a422007-08-14 18:04:11 +00002029 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002030
Owen Andersone98c54c2008-07-18 18:03:38 +00002031 if (EnablePRE) {
Owen Anderson0c7f91c2008-09-03 23:06:07 +00002032 bool PREChanged = true;
2033 while (PREChanged) {
2034 PREChanged = performPRE(F);
Chris Lattnerb2412a82009-09-21 02:42:51 +00002035 Changed |= PREChanged;
Owen Anderson0c7f91c2008-09-03 23:06:07 +00002036 }
Owen Andersone98c54c2008-07-18 18:03:38 +00002037 }
Chris Lattnerae199312008-12-09 19:21:47 +00002038 // FIXME: Should perform GVN again after PRE does something. PRE can move
2039 // computations into blocks where they become fully redundant. Note that
2040 // we can't do this until PRE's critical edge splitting updates memdep.
2041 // Actually, when this happens, we should just fully integrate PRE into GVN.
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002042
2043 cleanupGlobalSets();
2044
Chris Lattnerb2412a82009-09-21 02:42:51 +00002045 return Changed;
Owen Anderson3e75a422007-08-14 18:04:11 +00002046}
2047
2048
Chris Lattnerb2412a82009-09-21 02:42:51 +00002049bool GVN::processBlock(BasicBlock *BB) {
Chris Lattnerae199312008-12-09 19:21:47 +00002050 // FIXME: Kill off toErase by doing erasing eagerly in a helper function (and
2051 // incrementing BI before processing an instruction).
Owen Andersonaf4240a2008-06-12 19:25:32 +00002052 SmallVector<Instruction*, 8> toErase;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002053 bool ChangedFunction = false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002054
Owen Andersonaf4240a2008-06-12 19:25:32 +00002055 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
2056 BI != BE;) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002057 ChangedFunction |= processInstruction(BI, toErase);
Owen Andersonaf4240a2008-06-12 19:25:32 +00002058 if (toErase.empty()) {
2059 ++BI;
2060 continue;
2061 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002062
Owen Andersonaf4240a2008-06-12 19:25:32 +00002063 // If we need some instructions deleted, do it now.
2064 NumGVNInstr += toErase.size();
Daniel Dunbara279bc32009-09-20 02:20:51 +00002065
Owen Andersonaf4240a2008-06-12 19:25:32 +00002066 // Avoid iterator invalidation.
2067 bool AtStart = BI == BB->begin();
2068 if (!AtStart)
2069 --BI;
2070
2071 for (SmallVector<Instruction*, 4>::iterator I = toErase.begin(),
Chris Lattner663e4412008-12-01 00:40:32 +00002072 E = toErase.end(); I != E; ++I) {
David Greenebf7f78e2010-01-05 01:27:17 +00002073 DEBUG(dbgs() << "GVN removed: " << **I << '\n');
Dan Gohman4ec01b22009-11-14 02:27:51 +00002074 if (MD) MD->removeInstruction(*I);
Owen Andersonaf4240a2008-06-12 19:25:32 +00002075 (*I)->eraseFromParent();
Bill Wendlingec40d502008-12-22 21:57:30 +00002076 DEBUG(verifyRemoved(*I));
Chris Lattner663e4412008-12-01 00:40:32 +00002077 }
Chris Lattnerae199312008-12-09 19:21:47 +00002078 toErase.clear();
Owen Andersonaf4240a2008-06-12 19:25:32 +00002079
2080 if (AtStart)
2081 BI = BB->begin();
2082 else
2083 ++BI;
Owen Andersonaf4240a2008-06-12 19:25:32 +00002084 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002085
Chris Lattnerb2412a82009-09-21 02:42:51 +00002086 return ChangedFunction;
Owen Andersonaf4240a2008-06-12 19:25:32 +00002087}
2088
Owen Andersonb2303722008-06-18 21:41:49 +00002089/// performPRE - Perform a purely local form of PRE that looks for diamond
2090/// control flow patterns and attempts to perform simple PRE at the join point.
Chris Lattnerfb6e7012009-10-31 22:11:15 +00002091bool GVN::performPRE(Function &F) {
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002092 bool Changed = false;
Chris Lattner09713792008-12-01 07:29:03 +00002093 DenseMap<BasicBlock*, Value*> predMap;
Owen Andersonb2303722008-06-18 21:41:49 +00002094 for (df_iterator<BasicBlock*> DI = df_begin(&F.getEntryBlock()),
2095 DE = df_end(&F.getEntryBlock()); DI != DE; ++DI) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002096 BasicBlock *CurrentBlock = *DI;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002097
Owen Andersonb2303722008-06-18 21:41:49 +00002098 // Nothing to PRE in the entry block.
2099 if (CurrentBlock == &F.getEntryBlock()) continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002100
Owen Andersonb2303722008-06-18 21:41:49 +00002101 for (BasicBlock::iterator BI = CurrentBlock->begin(),
2102 BE = CurrentBlock->end(); BI != BE; ) {
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002103 Instruction *CurInst = BI++;
Duncan Sands7af1c782009-05-06 06:49:50 +00002104
Victor Hernandez7b929da2009-10-23 21:09:37 +00002105 if (isa<AllocaInst>(CurInst) ||
Victor Hernandez83d63912009-09-18 22:35:49 +00002106 isa<TerminatorInst>(CurInst) || isa<PHINode>(CurInst) ||
Devang Patel9674d152009-10-14 17:29:00 +00002107 CurInst->getType()->isVoidTy() ||
Duncan Sands7af1c782009-05-06 06:49:50 +00002108 CurInst->mayReadFromMemory() || CurInst->mayHaveSideEffects() ||
John Criswell090c0a22009-03-10 15:04:53 +00002109 isa<DbgInfoIntrinsic>(CurInst))
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002110 continue;
Owen Anderson5015b342010-08-07 00:20:35 +00002111
2112 // We don't currently value number ANY inline asm calls.
2113 if (CallInst *CallI = dyn_cast<CallInst>(CurInst))
2114 if (CallI->isInlineAsm())
2115 continue;
Duncan Sands7af1c782009-05-06 06:49:50 +00002116
Chris Lattnerb2412a82009-09-21 02:42:51 +00002117 uint32_t ValNo = VN.lookup(CurInst);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002118
Owen Andersonb2303722008-06-18 21:41:49 +00002119 // Look for the predecessors for PRE opportunities. We're
2120 // only trying to solve the basic diamond case, where
2121 // a value is computed in the successor and one predecessor,
2122 // but not the other. We also explicitly disallow cases
2123 // where the successor is its own predecessor, because they're
2124 // more complicated to get right.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002125 unsigned NumWith = 0;
2126 unsigned NumWithout = 0;
2127 BasicBlock *PREPred = 0;
Chris Lattner09713792008-12-01 07:29:03 +00002128 predMap.clear();
2129
Owen Andersonb2303722008-06-18 21:41:49 +00002130 for (pred_iterator PI = pred_begin(CurrentBlock),
2131 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
Gabor Greif08149852010-07-09 14:36:49 +00002132 BasicBlock *P = *PI;
Owen Andersonb2303722008-06-18 21:41:49 +00002133 // We're not interested in PRE where the block is its
Bob Wilsone7b635f2010-02-03 00:33:21 +00002134 // own predecessor, or in blocks with predecessors
Owen Anderson6fafe842008-06-20 01:15:47 +00002135 // that are not reachable.
Gabor Greif08149852010-07-09 14:36:49 +00002136 if (P == CurrentBlock) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002137 NumWithout = 2;
Owen Anderson6fafe842008-06-20 01:15:47 +00002138 break;
Owen Andersona04a0642010-11-18 18:32:40 +00002139 } else if (!DT->dominates(&F.getEntryBlock(), P)) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002140 NumWithout = 2;
Owen Anderson6fafe842008-06-20 01:15:47 +00002141 break;
2142 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002143
Owen Andersona04a0642010-11-18 18:32:40 +00002144 Value* predV = lookupNumber(P, ValNo);
2145 if (predV == 0) {
Gabor Greif08149852010-07-09 14:36:49 +00002146 PREPred = P;
Dan Gohmanfe601042010-06-22 15:08:57 +00002147 ++NumWithout;
Owen Andersona04a0642010-11-18 18:32:40 +00002148 } else if (predV == CurInst) {
Chris Lattnerb2412a82009-09-21 02:42:51 +00002149 NumWithout = 2;
Owen Andersonb2303722008-06-18 21:41:49 +00002150 } else {
Owen Andersona04a0642010-11-18 18:32:40 +00002151 predMap[P] = predV;
Dan Gohmanfe601042010-06-22 15:08:57 +00002152 ++NumWith;
Owen Andersonb2303722008-06-18 21:41:49 +00002153 }
2154 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002155
Owen Andersonb2303722008-06-18 21:41:49 +00002156 // Don't do PRE when it might increase code size, i.e. when
2157 // we would need to insert instructions in more than one pred.
Chris Lattnerb2412a82009-09-21 02:42:51 +00002158 if (NumWithout != 1 || NumWith == 0)
Owen Andersonb2303722008-06-18 21:41:49 +00002159 continue;
Chris Lattnerfb6e7012009-10-31 22:11:15 +00002160
2161 // Don't do PRE across indirect branch.
2162 if (isa<IndirectBrInst>(PREPred->getTerminator()))
2163 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002164
Owen Anderson5c274ee2008-06-19 19:54:19 +00002165 // We can't do PRE safely on a critical edge, so instead we schedule
2166 // the edge to be split and perform the PRE the next time we iterate
2167 // on the function.
Bob Wilsonae23daf2010-02-16 21:06:42 +00002168 unsigned SuccNum = GetSuccessorNumber(PREPred, CurrentBlock);
Chris Lattnerb2412a82009-09-21 02:42:51 +00002169 if (isCriticalEdge(PREPred->getTerminator(), SuccNum)) {
2170 toSplit.push_back(std::make_pair(PREPred->getTerminator(), SuccNum));
Owen Anderson5c274ee2008-06-19 19:54:19 +00002171 continue;
2172 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002173
Bob Wilsone7b635f2010-02-03 00:33:21 +00002174 // Instantiate the expression in the predecessor that lacked it.
Owen Andersonb2303722008-06-18 21:41:49 +00002175 // Because we are going top-down through the block, all value numbers
2176 // will be available in the predecessor by the time we need them. Any
Bob Wilsone7b635f2010-02-03 00:33:21 +00002177 // that weren't originally present will have been instantiated earlier
Owen Andersonb2303722008-06-18 21:41:49 +00002178 // in this loop.
Nick Lewycky67760642009-09-27 07:38:41 +00002179 Instruction *PREInstr = CurInst->clone();
Owen Andersonb2303722008-06-18 21:41:49 +00002180 bool success = true;
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002181 for (unsigned i = 0, e = CurInst->getNumOperands(); i != e; ++i) {
2182 Value *Op = PREInstr->getOperand(i);
2183 if (isa<Argument>(Op) || isa<Constant>(Op) || isa<GlobalValue>(Op))
2184 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002185
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002186 if (Value *V = lookupNumber(PREPred, VN.lookup(Op))) {
2187 PREInstr->setOperand(i, V);
2188 } else {
2189 success = false;
2190 break;
Owen Andersonc45996b2008-07-11 20:05:13 +00002191 }
Owen Andersonb2303722008-06-18 21:41:49 +00002192 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002193
Owen Andersonb2303722008-06-18 21:41:49 +00002194 // Fail out if we encounter an operand that is not available in
Daniel Dunbara279bc32009-09-20 02:20:51 +00002195 // the PRE predecessor. This is typically because of loads which
Owen Andersonb2303722008-06-18 21:41:49 +00002196 // are not value numbered precisely.
2197 if (!success) {
2198 delete PREInstr;
Bill Wendling70ded192008-12-22 22:14:07 +00002199 DEBUG(verifyRemoved(PREInstr));
Owen Andersonb2303722008-06-18 21:41:49 +00002200 continue;
2201 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002202
Owen Andersonb2303722008-06-18 21:41:49 +00002203 PREInstr->insertBefore(PREPred->getTerminator());
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002204 PREInstr->setName(CurInst->getName() + ".pre");
Owen Anderson6fafe842008-06-20 01:15:47 +00002205 predMap[PREPred] = PREInstr;
Chris Lattnerb2412a82009-09-21 02:42:51 +00002206 VN.add(PREInstr, ValNo);
Dan Gohmanfe601042010-06-22 15:08:57 +00002207 ++NumGVNPRE;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002208
Owen Andersonb2303722008-06-18 21:41:49 +00002209 // Update the availability map to include the new instruction.
Owen Andersona04a0642010-11-18 18:32:40 +00002210 insert_table(ValNo, PREInstr);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002211
Owen Andersonb2303722008-06-18 21:41:49 +00002212 // Create a PHI to make the value available in this block.
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002213 PHINode* Phi = PHINode::Create(CurInst->getType(),
2214 CurInst->getName() + ".pre-phi",
Owen Andersonb2303722008-06-18 21:41:49 +00002215 CurrentBlock->begin());
2216 for (pred_iterator PI = pred_begin(CurrentBlock),
Gabor Greif1d3ae022010-07-09 14:48:08 +00002217 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
2218 BasicBlock *P = *PI;
2219 Phi->addIncoming(predMap[P], P);
2220 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002221
Chris Lattnerb2412a82009-09-21 02:42:51 +00002222 VN.add(Phi, ValNo);
Owen Andersona04a0642010-11-18 18:32:40 +00002223 insert_table(ValNo, Phi);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002224
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002225 CurInst->replaceAllUsesWith(Phi);
Duncan Sands1df98592010-02-16 11:11:14 +00002226 if (MD && Phi->getType()->isPointerTy())
Chris Lattnerbc99be12008-12-09 22:06:23 +00002227 MD->invalidateCachedPointerInfo(Phi);
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002228 VN.erase(CurInst);
Owen Andersona04a0642010-11-18 18:32:40 +00002229 erase_table(ValNo, CurInst);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002230
David Greenebf7f78e2010-01-05 01:27:17 +00002231 DEBUG(dbgs() << "GVN PRE removed: " << *CurInst << '\n');
Dan Gohman4ec01b22009-11-14 02:27:51 +00002232 if (MD) MD->removeInstruction(CurInst);
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002233 CurInst->eraseFromParent();
Bill Wendlingec40d502008-12-22 21:57:30 +00002234 DEBUG(verifyRemoved(CurInst));
Chris Lattnerd0f5bfc2008-12-01 07:35:54 +00002235 Changed = true;
Owen Andersonb2303722008-06-18 21:41:49 +00002236 }
2237 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002238
Bob Wilson484d4a32010-02-16 19:51:59 +00002239 if (splitCriticalEdges())
2240 Changed = true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002241
Bob Wilson484d4a32010-02-16 19:51:59 +00002242 return Changed;
2243}
2244
2245/// splitCriticalEdges - Split critical edges found during the previous
2246/// iteration that may enable further optimization.
2247bool GVN::splitCriticalEdges() {
2248 if (toSplit.empty())
2249 return false;
2250 do {
2251 std::pair<TerminatorInst*, unsigned> Edge = toSplit.pop_back_val();
2252 SplitCriticalEdge(Edge.first, Edge.second, this);
2253 } while (!toSplit.empty());
Evan Cheng19d417c2010-03-01 22:23:12 +00002254 if (MD) MD->invalidateCachedPredecessors();
Bob Wilson484d4a32010-02-16 19:51:59 +00002255 return true;
Owen Andersonb2303722008-06-18 21:41:49 +00002256}
2257
Bill Wendling30788b82008-12-22 22:32:22 +00002258/// iterateOnFunction - Executes one iteration of GVN
Owen Anderson3e75a422007-08-14 18:04:11 +00002259bool GVN::iterateOnFunction(Function &F) {
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002260 cleanupGlobalSets();
Owen Andersona04a0642010-11-18 18:32:40 +00002261
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002262 // Top-down walk of the dominator tree
Chris Lattnerb2412a82009-09-21 02:42:51 +00002263 bool Changed = false;
Owen Andersonc34d1122008-12-15 03:52:17 +00002264#if 0
2265 // Needed for value numbering with phi construction to work.
Owen Anderson255dafc2008-12-15 02:03:00 +00002266 ReversePostOrderTraversal<Function*> RPOT(&F);
2267 for (ReversePostOrderTraversal<Function*>::rpo_iterator RI = RPOT.begin(),
2268 RE = RPOT.end(); RI != RE; ++RI)
Chris Lattnerb2412a82009-09-21 02:42:51 +00002269 Changed |= processBlock(*RI);
Owen Andersonc34d1122008-12-15 03:52:17 +00002270#else
2271 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2272 DE = df_end(DT->getRootNode()); DI != DE; ++DI)
Chris Lattnerb2412a82009-09-21 02:42:51 +00002273 Changed |= processBlock(DI->getBlock());
Owen Andersonc34d1122008-12-15 03:52:17 +00002274#endif
2275
Chris Lattnerb2412a82009-09-21 02:42:51 +00002276 return Changed;
Owen Anderson1ad2cb72007-07-24 17:55:58 +00002277}
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002278
2279void GVN::cleanupGlobalSets() {
2280 VN.clear();
Owen Andersona04a0642010-11-18 18:32:40 +00002281 NumberTable.clear();
2282 TableAllocator.Reset();
Nuno Lopes7cdd9ee2008-10-10 16:25:50 +00002283}
Bill Wendling246dbbb2008-12-22 21:36:08 +00002284
2285/// verifyRemoved - Verify that the specified instruction does not occur in our
2286/// internal data structures.
Bill Wendling6d463f22008-12-22 22:28:56 +00002287void GVN::verifyRemoved(const Instruction *Inst) const {
2288 VN.verifyRemoved(Inst);
Bill Wendling70ded192008-12-22 22:14:07 +00002289
Bill Wendling6d463f22008-12-22 22:28:56 +00002290 // Walk through the value number scope to make sure the instruction isn't
2291 // ferreted away in it.
Owen Andersona04a0642010-11-18 18:32:40 +00002292 for (DenseMap<uint32_t, std::pair<Value*, void*> >::const_iterator
2293 I = NumberTable.begin(), E = NumberTable.end(); I != E; ++I) {
2294 std::pair<Value*, void*> const * Node = &I->second;
2295 assert(Node->first != Inst && "Inst still in value numbering scope!");
2296
2297 while (Node->second) {
2298 Node = static_cast<std::pair<Value*, void*>*>(Node->second);
2299 assert(Node->first != Inst && "Inst still in value numbering scope!");
Bill Wendling70ded192008-12-22 22:14:07 +00002300 }
2301 }
Bill Wendling246dbbb2008-12-22 21:36:08 +00002302}