blob: aa6d453087f10e10f361219a600425254324dde0 [file] [log] [blame]
Chris Lattnerd2a653a2008-12-05 07:49:08 +00001//===- GVN.cpp - Eliminate redundant values and loads ---------------------===//
Owen Andersonab6ec2e2007-07-24 17:55:58 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattnerf3ebc3f2007-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 Andersonab6ec2e2007-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 Criswell073e4d12009-03-10 15:04:53 +000013// Note that this pass does the value numbering itself; it does not use the
Matthijs Kooijman5afc2742008-06-05 07:55:49 +000014// ValueNumbering analysis passes.
15//
Owen Andersonab6ec2e2007-07-24 17:55:58 +000016//===----------------------------------------------------------------------===//
17
18#define DEBUG_TYPE "gvn"
Owen Andersonab6ec2e2007-07-24 17:55:58 +000019#include "llvm/Transforms/Scalar.h"
Owen Andersondbf23cc2007-07-26 18:26:51 +000020#include "llvm/Constants.h"
Owen Andersonab6ec2e2007-07-24 17:55:58 +000021#include "llvm/DerivedTypes.h"
Chris Lattner17079fc2009-12-28 21:28:46 +000022#include "llvm/GlobalVariable.h"
Owen Andersondbf23cc2007-07-26 18:26:51 +000023#include "llvm/Function.h"
Devang Patele8c6d312009-03-06 02:59:27 +000024#include "llvm/IntrinsicInst.h"
Dan Gohman4467aa52010-12-15 23:53:55 +000025#include "llvm/LLVMContext.h"
Chris Lattner0a9616d2009-09-21 05:57:11 +000026#include "llvm/Operator.h"
Owen Anderson09b83ba2007-10-18 19:39:33 +000027#include "llvm/Analysis/AliasAnalysis.h"
Chris Lattner778cb922009-12-06 05:29:56 +000028#include "llvm/Analysis/ConstantFolding.h"
29#include "llvm/Analysis/Dominators.h"
Duncan Sands246b71c2010-11-12 21:10:24 +000030#include "llvm/Analysis/InstructionSimplify.h"
Dan Gohman826bdf82010-05-28 16:19:17 +000031#include "llvm/Analysis/Loads.h"
Victor Hernandezf390e042009-10-27 20:05:49 +000032#include "llvm/Analysis/MemoryBuiltins.h"
Owen Andersonab6ec2e2007-07-24 17:55:58 +000033#include "llvm/Analysis/MemoryDependenceAnalysis.h"
Chris Lattner972e6d82009-12-09 01:59:31 +000034#include "llvm/Analysis/PHITransAddr.h"
Chris Lattnere28618d2010-11-30 22:25:26 +000035#include "llvm/Analysis/ValueTracking.h"
Chris Lattnerbf0aa922011-01-02 22:09:33 +000036#include "llvm/Assembly/Writer.h"
Chris Lattnere28618d2010-11-30 22:25:26 +000037#include "llvm/Target/TargetData.h"
38#include "llvm/Transforms/Utils/BasicBlockUtils.h"
39#include "llvm/Transforms/Utils/Local.h"
40#include "llvm/Transforms/Utils/SSAUpdater.h"
41#include "llvm/ADT/DenseMap.h"
42#include "llvm/ADT/DepthFirstIterator.h"
43#include "llvm/ADT/PostOrderIterator.h"
44#include "llvm/ADT/SmallPtrSet.h"
45#include "llvm/ADT/Statistic.h"
Owen Andersonc21c1002010-11-18 18:32:40 +000046#include "llvm/Support/Allocator.h"
Owen Andersonab6ec2e2007-07-24 17:55:58 +000047#include "llvm/Support/CFG.h"
Owen Andersone780d662008-06-19 19:57:25 +000048#include "llvm/Support/CommandLine.h"
Chris Lattnerd528b212008-03-29 04:36:18 +000049#include "llvm/Support/Debug.h"
Torok Edwin56d06592009-07-11 20:10:48 +000050#include "llvm/Support/ErrorHandling.h"
Chris Lattner0a9616d2009-09-21 05:57:11 +000051#include "llvm/Support/GetElementPtrTypeIterator.h"
Chris Lattner42376062009-12-06 01:57:02 +000052#include "llvm/Support/IRBuilder.h"
Owen Andersonc21c1002010-11-18 18:32:40 +000053#include <list>
Owen Andersonab6ec2e2007-07-24 17:55:58 +000054using namespace llvm;
55
Bill Wendling3c793442008-12-22 22:14:07 +000056STATISTIC(NumGVNInstr, "Number of instructions deleted");
57STATISTIC(NumGVNLoad, "Number of loads deleted");
58STATISTIC(NumGVNPRE, "Number of instructions PRE'd");
Owen Anderson53d546e2008-07-15 16:28:06 +000059STATISTIC(NumGVNBlocks, "Number of blocks merged");
Bill Wendling3c793442008-12-22 22:14:07 +000060STATISTIC(NumPRELoad, "Number of loads PRE'd");
Chris Lattner168be762008-03-22 04:13:49 +000061
Evan Cheng9598f932008-06-20 01:01:07 +000062static cl::opt<bool> EnablePRE("enable-pre",
Owen Andersonaddbe3e2008-07-17 19:41:00 +000063 cl::init(true), cl::Hidden);
Dan Gohmana8f8a852009-06-15 18:30:15 +000064static cl::opt<bool> EnableLoadPRE("enable-load-pre", cl::init(true));
Owen Andersone780d662008-06-19 19:57:25 +000065
Owen Andersonab6ec2e2007-07-24 17:55:58 +000066//===----------------------------------------------------------------------===//
67// ValueTable Class
68//===----------------------------------------------------------------------===//
69
70/// This class holds the mapping between values and value numbers. It is used
71/// as an efficient mechanism to determine the expression-wise equivalence of
72/// two values.
73namespace {
Chris Lattner2dd09db2009-09-02 06:11:42 +000074 struct Expression {
Owen Andersoncdea3572010-01-17 19:33:27 +000075 enum ExpressionOpcode {
76 ADD = Instruction::Add,
77 FADD = Instruction::FAdd,
78 SUB = Instruction::Sub,
79 FSUB = Instruction::FSub,
80 MUL = Instruction::Mul,
81 FMUL = Instruction::FMul,
82 UDIV = Instruction::UDiv,
83 SDIV = Instruction::SDiv,
84 FDIV = Instruction::FDiv,
85 UREM = Instruction::URem,
86 SREM = Instruction::SRem,
87 FREM = Instruction::FRem,
88 SHL = Instruction::Shl,
89 LSHR = Instruction::LShr,
90 ASHR = Instruction::AShr,
91 AND = Instruction::And,
92 OR = Instruction::Or,
93 XOR = Instruction::Xor,
94 TRUNC = Instruction::Trunc,
95 ZEXT = Instruction::ZExt,
96 SEXT = Instruction::SExt,
97 FPTOUI = Instruction::FPToUI,
98 FPTOSI = Instruction::FPToSI,
99 UITOFP = Instruction::UIToFP,
100 SITOFP = Instruction::SIToFP,
101 FPTRUNC = Instruction::FPTrunc,
102 FPEXT = Instruction::FPExt,
103 PTRTOINT = Instruction::PtrToInt,
104 INTTOPTR = Instruction::IntToPtr,
105 BITCAST = Instruction::BitCast,
106 ICMPEQ, ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
107 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
108 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
109 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
110 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
111 SHUFFLE, SELECT, GEP, CALL, CONSTANT,
112 INSERTVALUE, EXTRACTVALUE, EMPTY, TOMBSTONE };
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000113
114 ExpressionOpcode opcode;
115 const Type* type;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000116 SmallVector<uint32_t, 4> varargs;
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000117 Value *function;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000118
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000119 Expression() { }
120 Expression(ExpressionOpcode o) : opcode(o) { }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000121
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000122 bool operator==(const Expression &other) const {
123 if (opcode != other.opcode)
124 return false;
125 else if (opcode == EMPTY || opcode == TOMBSTONE)
126 return true;
127 else if (type != other.type)
128 return false;
Owen Anderson09b83ba2007-10-18 19:39:33 +0000129 else if (function != other.function)
130 return false;
Benjamin Kramer43493c02010-12-21 21:30:19 +0000131 else if (varargs != other.varargs)
132 return false;
133 return true;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000134 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000135
Chris Lattner65b48b52010-09-04 18:12:00 +0000136 /*bool operator!=(const Expression &other) const {
Bill Wendling86f01cb2008-12-22 22:16:31 +0000137 return !(*this == other);
Chris Lattner65b48b52010-09-04 18:12:00 +0000138 }*/
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000139 };
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000140
Chris Lattner2dd09db2009-09-02 06:11:42 +0000141 class ValueTable {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000142 private:
143 DenseMap<Value*, uint32_t> valueNumbering;
144 DenseMap<Expression, uint32_t> expressionNumbering;
Owen Andersonf7928602008-05-12 20:15:55 +0000145 AliasAnalysis* AA;
146 MemoryDependenceAnalysis* MD;
147 DominatorTree* DT;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000148
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000149 uint32_t nextValueNumber;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000150
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000151 Expression::ExpressionOpcode getOpcode(CmpInst* C);
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000152 Expression create_expression(BinaryOperator* BO);
153 Expression create_expression(CmpInst* C);
154 Expression create_expression(ShuffleVectorInst* V);
155 Expression create_expression(ExtractElementInst* C);
156 Expression create_expression(InsertElementInst* V);
157 Expression create_expression(SelectInst* V);
158 Expression create_expression(CastInst* C);
159 Expression create_expression(GetElementPtrInst* G);
Owen Anderson09b83ba2007-10-18 19:39:33 +0000160 Expression create_expression(CallInst* C);
Owen Anderson168ad692009-10-19 22:14:22 +0000161 Expression create_expression(ExtractValueInst* C);
162 Expression create_expression(InsertValueInst* C);
163
164 uint32_t lookup_or_add_call(CallInst* C);
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000165 public:
Dan Gohmanc4971722009-04-01 16:37:47 +0000166 ValueTable() : nextValueNumber(1) { }
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000167 uint32_t lookup_or_add(Value *V);
168 uint32_t lookup(Value *V) const;
169 void add(Value *V, uint32_t num);
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000170 void clear();
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000171 void erase(Value *v);
Owen Andersonf7928602008-05-12 20:15:55 +0000172 void setAliasAnalysis(AliasAnalysis* A) { AA = A; }
Chris Lattner8541ede2008-12-01 00:40:32 +0000173 AliasAnalysis *getAliasAnalysis() const { return AA; }
Owen Andersonf7928602008-05-12 20:15:55 +0000174 void setMemDep(MemoryDependenceAnalysis* M) { MD = M; }
175 void setDomTree(DominatorTree* D) { DT = D; }
Owen Anderson3ea90a72008-07-03 17:44:33 +0000176 uint32_t getNextUnusedValueNumber() { return nextValueNumber; }
Bill Wendling6b18a392008-12-22 21:36:08 +0000177 void verifyRemoved(const Value *) const;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000178 };
179}
180
181namespace llvm {
Chris Lattner0625bd62007-09-17 18:34:04 +0000182template <> struct DenseMapInfo<Expression> {
Owen Anderson9699a6e2007-08-02 18:16:06 +0000183 static inline Expression getEmptyKey() {
184 return Expression(Expression::EMPTY);
185 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000186
Owen Anderson9699a6e2007-08-02 18:16:06 +0000187 static inline Expression getTombstoneKey() {
188 return Expression(Expression::TOMBSTONE);
189 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000190
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000191 static unsigned getHashValue(const Expression e) {
192 unsigned hash = e.opcode;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000193
Anton Korobeynikov1bfd1212008-02-20 11:26:25 +0000194 hash = ((unsigned)((uintptr_t)e.type >> 4) ^
Owen Anderson168ad692009-10-19 22:14:22 +0000195 (unsigned)((uintptr_t)e.type >> 9));
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000196
Owen Anderson9699a6e2007-08-02 18:16:06 +0000197 for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
198 E = e.varargs.end(); I != E; ++I)
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000199 hash = *I + hash * 37;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000200
Anton Korobeynikov1bfd1212008-02-20 11:26:25 +0000201 hash = ((unsigned)((uintptr_t)e.function >> 4) ^
202 (unsigned)((uintptr_t)e.function >> 9)) +
203 hash * 37;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000204
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000205 return hash;
206 }
Chris Lattner0625bd62007-09-17 18:34:04 +0000207 static bool isEqual(const Expression &LHS, const Expression &RHS) {
208 return LHS == RHS;
209 }
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000210};
Chris Lattner45d040b2009-12-15 07:26:43 +0000211
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000212}
213
214//===----------------------------------------------------------------------===//
215// ValueTable Internal Functions
216//===----------------------------------------------------------------------===//
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000217
218Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
Nick Lewyckya21d3da2009-07-08 03:04:38 +0000219 if (isa<ICmpInst>(C)) {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000220 switch (C->getPredicate()) {
Chris Lattner2876a642008-03-21 21:14:38 +0000221 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinfbcc6632009-07-14 16:55:14 +0000222 llvm_unreachable("Comparison with unknown predicate?");
Chris Lattner2876a642008-03-21 21:14:38 +0000223 case ICmpInst::ICMP_EQ: return Expression::ICMPEQ;
224 case ICmpInst::ICMP_NE: return Expression::ICMPNE;
225 case ICmpInst::ICMP_UGT: return Expression::ICMPUGT;
226 case ICmpInst::ICMP_UGE: return Expression::ICMPUGE;
227 case ICmpInst::ICMP_ULT: return Expression::ICMPULT;
228 case ICmpInst::ICMP_ULE: return Expression::ICMPULE;
229 case ICmpInst::ICMP_SGT: return Expression::ICMPSGT;
230 case ICmpInst::ICMP_SGE: return Expression::ICMPSGE;
231 case ICmpInst::ICMP_SLT: return Expression::ICMPSLT;
232 case ICmpInst::ICMP_SLE: return Expression::ICMPSLE;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000233 }
Nick Lewyckya21d3da2009-07-08 03:04:38 +0000234 } else {
235 switch (C->getPredicate()) {
236 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinfbcc6632009-07-14 16:55:14 +0000237 llvm_unreachable("Comparison with unknown predicate?");
Nick Lewyckya21d3da2009-07-08 03:04:38 +0000238 case FCmpInst::FCMP_OEQ: return Expression::FCMPOEQ;
239 case FCmpInst::FCMP_OGT: return Expression::FCMPOGT;
240 case FCmpInst::FCMP_OGE: return Expression::FCMPOGE;
241 case FCmpInst::FCMP_OLT: return Expression::FCMPOLT;
242 case FCmpInst::FCMP_OLE: return Expression::FCMPOLE;
243 case FCmpInst::FCMP_ONE: return Expression::FCMPONE;
244 case FCmpInst::FCMP_ORD: return Expression::FCMPORD;
245 case FCmpInst::FCMP_UNO: return Expression::FCMPUNO;
246 case FCmpInst::FCMP_UEQ: return Expression::FCMPUEQ;
247 case FCmpInst::FCMP_UGT: return Expression::FCMPUGT;
248 case FCmpInst::FCMP_UGE: return Expression::FCMPUGE;
249 case FCmpInst::FCMP_ULT: return Expression::FCMPULT;
250 case FCmpInst::FCMP_ULE: return Expression::FCMPULE;
251 case FCmpInst::FCMP_UNE: return Expression::FCMPUNE;
252 }
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000253 }
254}
255
Owen Anderson09b83ba2007-10-18 19:39:33 +0000256Expression ValueTable::create_expression(CallInst* C) {
257 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000258
Owen Anderson09b83ba2007-10-18 19:39:33 +0000259 e.type = C->getType();
Owen Anderson09b83ba2007-10-18 19:39:33 +0000260 e.function = C->getCalledFunction();
261 e.opcode = Expression::CALL;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000262
Gabor Greif5bcaa552010-06-24 10:04:07 +0000263 CallSite CS(C);
264 for (CallInst::op_iterator I = CS.arg_begin(), E = CS.arg_end();
Owen Anderson09b83ba2007-10-18 19:39:33 +0000265 I != E; ++I)
Owen Anderson1e73f292008-04-11 05:11:49 +0000266 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000267
Owen Anderson09b83ba2007-10-18 19:39:33 +0000268 return e;
269}
270
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000271Expression ValueTable::create_expression(BinaryOperator* BO) {
272 Expression e;
Owen Anderson168ad692009-10-19 22:14:22 +0000273 e.varargs.push_back(lookup_or_add(BO->getOperand(0)));
274 e.varargs.push_back(lookup_or_add(BO->getOperand(1)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000275 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000276 e.type = BO->getType();
Owen Andersoncdea3572010-01-17 19:33:27 +0000277 e.opcode = static_cast<Expression::ExpressionOpcode>(BO->getOpcode());
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000278
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000279 return e;
280}
281
282Expression ValueTable::create_expression(CmpInst* C) {
283 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000284
Owen Anderson168ad692009-10-19 22:14:22 +0000285 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
286 e.varargs.push_back(lookup_or_add(C->getOperand(1)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000287 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000288 e.type = C->getType();
289 e.opcode = getOpcode(C);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000290
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000291 return e;
292}
293
294Expression ValueTable::create_expression(CastInst* C) {
295 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000296
Owen Anderson168ad692009-10-19 22:14:22 +0000297 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000298 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000299 e.type = C->getType();
Owen Andersoncdea3572010-01-17 19:33:27 +0000300 e.opcode = static_cast<Expression::ExpressionOpcode>(C->getOpcode());
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000301
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000302 return e;
303}
304
305Expression ValueTable::create_expression(ShuffleVectorInst* S) {
306 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000307
Owen Anderson168ad692009-10-19 22:14:22 +0000308 e.varargs.push_back(lookup_or_add(S->getOperand(0)));
309 e.varargs.push_back(lookup_or_add(S->getOperand(1)));
310 e.varargs.push_back(lookup_or_add(S->getOperand(2)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000311 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000312 e.type = S->getType();
313 e.opcode = Expression::SHUFFLE;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000314
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000315 return e;
316}
317
318Expression ValueTable::create_expression(ExtractElementInst* E) {
319 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000320
Owen Anderson168ad692009-10-19 22:14:22 +0000321 e.varargs.push_back(lookup_or_add(E->getOperand(0)));
322 e.varargs.push_back(lookup_or_add(E->getOperand(1)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000323 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000324 e.type = E->getType();
325 e.opcode = Expression::EXTRACT;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000326
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000327 return e;
328}
329
330Expression ValueTable::create_expression(InsertElementInst* I) {
331 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000332
Owen Anderson168ad692009-10-19 22:14:22 +0000333 e.varargs.push_back(lookup_or_add(I->getOperand(0)));
334 e.varargs.push_back(lookup_or_add(I->getOperand(1)));
335 e.varargs.push_back(lookup_or_add(I->getOperand(2)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000336 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000337 e.type = I->getType();
338 e.opcode = Expression::INSERT;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000339
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000340 return e;
341}
342
343Expression ValueTable::create_expression(SelectInst* I) {
344 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000345
Owen Anderson168ad692009-10-19 22:14:22 +0000346 e.varargs.push_back(lookup_or_add(I->getCondition()));
347 e.varargs.push_back(lookup_or_add(I->getTrueValue()));
348 e.varargs.push_back(lookup_or_add(I->getFalseValue()));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000349 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000350 e.type = I->getType();
351 e.opcode = Expression::SELECT;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000352
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000353 return e;
354}
355
356Expression ValueTable::create_expression(GetElementPtrInst* G) {
357 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000358
Owen Anderson168ad692009-10-19 22:14:22 +0000359 e.varargs.push_back(lookup_or_add(G->getPointerOperand()));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000360 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000361 e.type = G->getType();
362 e.opcode = Expression::GEP;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000363
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000364 for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
365 I != E; ++I)
Owen Anderson1e73f292008-04-11 05:11:49 +0000366 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000367
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000368 return e;
369}
370
Owen Anderson168ad692009-10-19 22:14:22 +0000371Expression ValueTable::create_expression(ExtractValueInst* E) {
372 Expression e;
373
374 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
375 for (ExtractValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
376 II != IE; ++II)
377 e.varargs.push_back(*II);
378 e.function = 0;
379 e.type = E->getType();
380 e.opcode = Expression::EXTRACTVALUE;
381
382 return e;
383}
384
385Expression ValueTable::create_expression(InsertValueInst* E) {
386 Expression e;
387
388 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
389 e.varargs.push_back(lookup_or_add(E->getInsertedValueOperand()));
390 for (InsertValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
391 II != IE; ++II)
392 e.varargs.push_back(*II);
393 e.function = 0;
394 e.type = E->getType();
395 e.opcode = Expression::INSERTVALUE;
396
397 return e;
398}
399
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000400//===----------------------------------------------------------------------===//
401// ValueTable External Functions
402//===----------------------------------------------------------------------===//
403
Owen Anderson6a903bc2008-06-18 21:41:49 +0000404/// add - Insert a value into the table with a specified value number.
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000405void ValueTable::add(Value *V, uint32_t num) {
Owen Anderson6a903bc2008-06-18 21:41:49 +0000406 valueNumbering.insert(std::make_pair(V, num));
407}
408
Owen Anderson168ad692009-10-19 22:14:22 +0000409uint32_t ValueTable::lookup_or_add_call(CallInst* C) {
410 if (AA->doesNotAccessMemory(C)) {
411 Expression exp = create_expression(C);
412 uint32_t& e = expressionNumbering[exp];
413 if (!e) e = nextValueNumber++;
414 valueNumbering[C] = e;
415 return e;
416 } else if (AA->onlyReadsMemory(C)) {
417 Expression exp = create_expression(C);
418 uint32_t& e = expressionNumbering[exp];
419 if (!e) {
420 e = nextValueNumber++;
421 valueNumbering[C] = e;
422 return e;
423 }
Dan Gohman81132462009-11-14 02:27:51 +0000424 if (!MD) {
425 e = nextValueNumber++;
426 valueNumbering[C] = e;
427 return e;
428 }
Owen Anderson168ad692009-10-19 22:14:22 +0000429
430 MemDepResult local_dep = MD->getDependency(C);
431
432 if (!local_dep.isDef() && !local_dep.isNonLocal()) {
433 valueNumbering[C] = nextValueNumber;
434 return nextValueNumber++;
435 }
436
437 if (local_dep.isDef()) {
438 CallInst* local_cdep = cast<CallInst>(local_dep.getInst());
439
Gabor Greiff628ecd2010-06-30 09:17:53 +0000440 if (local_cdep->getNumArgOperands() != C->getNumArgOperands()) {
Owen Anderson168ad692009-10-19 22:14:22 +0000441 valueNumbering[C] = nextValueNumber;
442 return nextValueNumber++;
443 }
444
Gabor Greif2d958d42010-06-24 10:17:17 +0000445 for (unsigned i = 0, e = C->getNumArgOperands(); i < e; ++i) {
446 uint32_t c_vn = lookup_or_add(C->getArgOperand(i));
447 uint32_t cd_vn = lookup_or_add(local_cdep->getArgOperand(i));
Owen Anderson168ad692009-10-19 22:14:22 +0000448 if (c_vn != cd_vn) {
449 valueNumbering[C] = nextValueNumber;
450 return nextValueNumber++;
451 }
452 }
453
454 uint32_t v = lookup_or_add(local_cdep);
455 valueNumbering[C] = v;
456 return v;
457 }
458
459 // Non-local case.
460 const MemoryDependenceAnalysis::NonLocalDepInfo &deps =
461 MD->getNonLocalCallDependency(CallSite(C));
462 // FIXME: call/call dependencies for readonly calls should return def, not
463 // clobber! Move the checking logic to MemDep!
464 CallInst* cdep = 0;
465
466 // Check to see if we have a single dominating call instruction that is
467 // identical to C.
468 for (unsigned i = 0, e = deps.size(); i != e; ++i) {
Chris Lattner0c315472009-12-09 07:08:01 +0000469 const NonLocalDepEntry *I = &deps[i];
Owen Anderson168ad692009-10-19 22:14:22 +0000470 // Ignore non-local dependencies.
Chris Lattner0c315472009-12-09 07:08:01 +0000471 if (I->getResult().isNonLocal())
Owen Anderson168ad692009-10-19 22:14:22 +0000472 continue;
473
474 // We don't handle non-depedencies. If we already have a call, reject
475 // instruction dependencies.
Chris Lattner0c315472009-12-09 07:08:01 +0000476 if (I->getResult().isClobber() || cdep != 0) {
Owen Anderson168ad692009-10-19 22:14:22 +0000477 cdep = 0;
478 break;
479 }
480
Chris Lattner0c315472009-12-09 07:08:01 +0000481 CallInst *NonLocalDepCall = dyn_cast<CallInst>(I->getResult().getInst());
Owen Anderson168ad692009-10-19 22:14:22 +0000482 // FIXME: All duplicated with non-local case.
Chris Lattner0c315472009-12-09 07:08:01 +0000483 if (NonLocalDepCall && DT->properlyDominates(I->getBB(), C->getParent())){
Owen Anderson168ad692009-10-19 22:14:22 +0000484 cdep = NonLocalDepCall;
485 continue;
486 }
487
488 cdep = 0;
489 break;
490 }
491
492 if (!cdep) {
493 valueNumbering[C] = nextValueNumber;
494 return nextValueNumber++;
495 }
496
Gabor Greiff628ecd2010-06-30 09:17:53 +0000497 if (cdep->getNumArgOperands() != C->getNumArgOperands()) {
Owen Anderson168ad692009-10-19 22:14:22 +0000498 valueNumbering[C] = nextValueNumber;
499 return nextValueNumber++;
500 }
Gabor Greif2d958d42010-06-24 10:17:17 +0000501 for (unsigned i = 0, e = C->getNumArgOperands(); i < e; ++i) {
502 uint32_t c_vn = lookup_or_add(C->getArgOperand(i));
503 uint32_t cd_vn = lookup_or_add(cdep->getArgOperand(i));
Owen Anderson168ad692009-10-19 22:14:22 +0000504 if (c_vn != cd_vn) {
505 valueNumbering[C] = nextValueNumber;
506 return nextValueNumber++;
507 }
508 }
509
510 uint32_t v = lookup_or_add(cdep);
511 valueNumbering[C] = v;
512 return v;
513
514 } else {
515 valueNumbering[C] = nextValueNumber;
516 return nextValueNumber++;
517 }
518}
519
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000520/// lookup_or_add - Returns the value number for the specified value, assigning
521/// it a new number if it did not have one before.
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000522uint32_t ValueTable::lookup_or_add(Value *V) {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000523 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
524 if (VI != valueNumbering.end())
525 return VI->second;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000526
Owen Anderson168ad692009-10-19 22:14:22 +0000527 if (!isa<Instruction>(V)) {
Owen Anderson1059b5b2009-10-19 21:14:57 +0000528 valueNumbering[V] = nextValueNumber;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000529 return nextValueNumber++;
530 }
Owen Anderson168ad692009-10-19 22:14:22 +0000531
532 Instruction* I = cast<Instruction>(V);
533 Expression exp;
534 switch (I->getOpcode()) {
535 case Instruction::Call:
536 return lookup_or_add_call(cast<CallInst>(I));
537 case Instruction::Add:
538 case Instruction::FAdd:
539 case Instruction::Sub:
540 case Instruction::FSub:
541 case Instruction::Mul:
542 case Instruction::FMul:
543 case Instruction::UDiv:
544 case Instruction::SDiv:
545 case Instruction::FDiv:
546 case Instruction::URem:
547 case Instruction::SRem:
548 case Instruction::FRem:
549 case Instruction::Shl:
550 case Instruction::LShr:
551 case Instruction::AShr:
552 case Instruction::And:
553 case Instruction::Or :
554 case Instruction::Xor:
555 exp = create_expression(cast<BinaryOperator>(I));
556 break;
557 case Instruction::ICmp:
558 case Instruction::FCmp:
559 exp = create_expression(cast<CmpInst>(I));
560 break;
561 case Instruction::Trunc:
562 case Instruction::ZExt:
563 case Instruction::SExt:
564 case Instruction::FPToUI:
565 case Instruction::FPToSI:
566 case Instruction::UIToFP:
567 case Instruction::SIToFP:
568 case Instruction::FPTrunc:
569 case Instruction::FPExt:
570 case Instruction::PtrToInt:
571 case Instruction::IntToPtr:
572 case Instruction::BitCast:
573 exp = create_expression(cast<CastInst>(I));
574 break;
575 case Instruction::Select:
576 exp = create_expression(cast<SelectInst>(I));
577 break;
578 case Instruction::ExtractElement:
579 exp = create_expression(cast<ExtractElementInst>(I));
580 break;
581 case Instruction::InsertElement:
582 exp = create_expression(cast<InsertElementInst>(I));
583 break;
584 case Instruction::ShuffleVector:
585 exp = create_expression(cast<ShuffleVectorInst>(I));
586 break;
587 case Instruction::ExtractValue:
588 exp = create_expression(cast<ExtractValueInst>(I));
589 break;
590 case Instruction::InsertValue:
591 exp = create_expression(cast<InsertValueInst>(I));
592 break;
593 case Instruction::GetElementPtr:
594 exp = create_expression(cast<GetElementPtrInst>(I));
595 break;
596 default:
597 valueNumbering[V] = nextValueNumber;
598 return nextValueNumber++;
599 }
600
601 uint32_t& e = expressionNumbering[exp];
602 if (!e) e = nextValueNumber++;
603 valueNumbering[V] = e;
604 return e;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000605}
606
607/// lookup - Returns the value number of the specified value. Fails if
608/// the value has not yet been numbered.
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000609uint32_t ValueTable::lookup(Value *V) const {
Jeffrey Yasskinb40d3f72009-11-10 01:02:17 +0000610 DenseMap<Value*, uint32_t>::const_iterator VI = valueNumbering.find(V);
Chris Lattner2876a642008-03-21 21:14:38 +0000611 assert(VI != valueNumbering.end() && "Value not numbered?");
612 return VI->second;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000613}
614
615/// clear - Remove all entries from the ValueTable
616void ValueTable::clear() {
617 valueNumbering.clear();
618 expressionNumbering.clear();
619 nextValueNumber = 1;
620}
621
Owen Anderson10ffa862007-07-31 23:27:13 +0000622/// erase - Remove a value from the value numbering
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000623void ValueTable::erase(Value *V) {
Owen Anderson10ffa862007-07-31 23:27:13 +0000624 valueNumbering.erase(V);
625}
626
Bill Wendling6b18a392008-12-22 21:36:08 +0000627/// verifyRemoved - Verify that the value is removed from all internal data
628/// structures.
629void ValueTable::verifyRemoved(const Value *V) const {
Jeffrey Yasskinb40d3f72009-11-10 01:02:17 +0000630 for (DenseMap<Value*, uint32_t>::const_iterator
Bill Wendling6b18a392008-12-22 21:36:08 +0000631 I = valueNumbering.begin(), E = valueNumbering.end(); I != E; ++I) {
632 assert(I->first != V && "Inst still occurs in value numbering map!");
633 }
634}
635
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000636//===----------------------------------------------------------------------===//
Bill Wendling456e8852008-12-22 22:32:22 +0000637// GVN Pass
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000638//===----------------------------------------------------------------------===//
639
640namespace {
641
Chris Lattner2dd09db2009-09-02 06:11:42 +0000642 class GVN : public FunctionPass {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000643 bool runOnFunction(Function &F);
644 public:
645 static char ID; // Pass identification, replacement for typeid
Bob Wilson11361662010-02-28 05:34:05 +0000646 explicit GVN(bool noloads = false)
Owen Anderson6c18d1a2010-10-19 17:21:58 +0000647 : FunctionPass(ID), NoLoads(noloads), MD(0) {
648 initializeGVNPass(*PassRegistry::getPassRegistry());
649 }
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000650
651 private:
Dan Gohman81132462009-11-14 02:27:51 +0000652 bool NoLoads;
Chris Lattner8541ede2008-12-01 00:40:32 +0000653 MemoryDependenceAnalysis *MD;
654 DominatorTree *DT;
Duncan Sands246b71c2010-11-12 21:10:24 +0000655 const TargetData* TD;
Chris Lattner8541ede2008-12-01 00:40:32 +0000656
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000657 ValueTable VN;
Owen Andersonc21c1002010-11-18 18:32:40 +0000658
Owen Andersonea326db2010-11-19 22:48:40 +0000659 /// NumberTable - A mapping from value numers to lists of Value*'s that
660 /// have that value number. Use lookupNumber to query it.
Owen Anderson5ab8d4b2010-12-21 23:54:34 +0000661 struct NumberTableEntry {
662 Value *Val;
663 BasicBlock *BB;
664 NumberTableEntry *Next;
665 };
666 DenseMap<uint32_t, NumberTableEntry> NumberTable;
Owen Andersonc21c1002010-11-18 18:32:40 +0000667 BumpPtrAllocator TableAllocator;
Owen Andersonea326db2010-11-19 22:48:40 +0000668
669 /// insert_table - Push a new Value to the NumberTable onto the list for
670 /// its value number.
Owen Anderson5ab8d4b2010-12-21 23:54:34 +0000671 void insert_table(uint32_t N, Value *V, BasicBlock *BB) {
672 NumberTableEntry& Curr = NumberTable[N];
673 if (!Curr.Val) {
674 Curr.Val = V;
675 Curr.BB = BB;
Owen Andersonc21c1002010-11-18 18:32:40 +0000676 return;
677 }
678
Owen Anderson5ab8d4b2010-12-21 23:54:34 +0000679 NumberTableEntry* Node = TableAllocator.Allocate<NumberTableEntry>();
680 Node->Val = V;
681 Node->BB = BB;
682 Node->Next = Curr.Next;
683 Curr.Next = Node;
Owen Andersonc21c1002010-11-18 18:32:40 +0000684 }
685
Owen Andersonea326db2010-11-19 22:48:40 +0000686 /// erase_table - Scan the list of values corresponding to a given value
687 /// number, and remove the given value if encountered.
Owen Anderson5ab8d4b2010-12-21 23:54:34 +0000688 void erase_table(uint32_t N, Value *V, BasicBlock *BB) {
689 NumberTableEntry* Prev = 0;
690 NumberTableEntry* Curr = &NumberTable[N];
Owen Andersonc21c1002010-11-18 18:32:40 +0000691
Owen Anderson5ab8d4b2010-12-21 23:54:34 +0000692 while (Curr->Val != V || Curr->BB != BB) {
Owen Andersonc21c1002010-11-18 18:32:40 +0000693 Prev = Curr;
Owen Anderson5ab8d4b2010-12-21 23:54:34 +0000694 Curr = Curr->Next;
Owen Andersonc21c1002010-11-18 18:32:40 +0000695 }
696
697 if (Prev) {
Owen Anderson5ab8d4b2010-12-21 23:54:34 +0000698 Prev->Next = Curr->Next;
Owen Andersonc21c1002010-11-18 18:32:40 +0000699 } else {
Owen Anderson5ab8d4b2010-12-21 23:54:34 +0000700 if (!Curr->Next) {
701 Curr->Val = 0;
702 Curr->BB = 0;
Owen Andersonc21c1002010-11-18 18:32:40 +0000703 } else {
Owen Anderson5ab8d4b2010-12-21 23:54:34 +0000704 NumberTableEntry* Next = Curr->Next;
705 Curr->Val = Next->Val;
706 Curr->BB = Next->BB;
Owen Andersonc21c1002010-11-18 18:32:40 +0000707 }
708 }
709 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000710
Bob Wilson92cdb6e2010-02-16 19:51:59 +0000711 // List of critical edges to be split between iterations.
712 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> toSplit;
713
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000714 // This transformation requires dominator postdominator info
715 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000716 AU.addRequired<DominatorTree>();
Dan Gohman81132462009-11-14 02:27:51 +0000717 if (!NoLoads)
718 AU.addRequired<MemoryDependenceAnalysis>();
Owen Anderson09b83ba2007-10-18 19:39:33 +0000719 AU.addRequired<AliasAnalysis>();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000720
Owen Anderson54e02192008-06-23 17:49:45 +0000721 AU.addPreserved<DominatorTree>();
Owen Anderson09b83ba2007-10-18 19:39:33 +0000722 AU.addPreserved<AliasAnalysis>();
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000723 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000724
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000725 // Helper fuctions
726 // FIXME: eliminate or document these better
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000727 bool processLoad(LoadInst* L,
Chris Lattner804209d2008-03-21 22:01:16 +0000728 SmallVectorImpl<Instruction*> &toErase);
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000729 bool processInstruction(Instruction *I,
Chris Lattner804209d2008-03-21 22:01:16 +0000730 SmallVectorImpl<Instruction*> &toErase);
Owen Anderson9699a6e2007-08-02 18:16:06 +0000731 bool processNonLocalLoad(LoadInst* L,
Chris Lattner804209d2008-03-21 22:01:16 +0000732 SmallVectorImpl<Instruction*> &toErase);
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000733 bool processBlock(BasicBlock *BB);
Owen Anderson6a903bc2008-06-18 21:41:49 +0000734 void dump(DenseMap<uint32_t, Value*>& d);
Owen Anderson676070d2007-08-14 18:04:11 +0000735 bool iterateOnFunction(Function &F);
Owen Anderson6a903bc2008-06-18 21:41:49 +0000736 bool performPRE(Function& F);
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000737 Value *lookupNumber(BasicBlock *BB, uint32_t num);
Nuno Lopese3127f32008-10-10 16:25:50 +0000738 void cleanupGlobalSets();
Bill Wendling6b18a392008-12-22 21:36:08 +0000739 void verifyRemoved(const Instruction *I) const;
Bob Wilson92cdb6e2010-02-16 19:51:59 +0000740 bool splitCriticalEdges();
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000741 };
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000742
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000743 char GVN::ID = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000744}
745
746// createGVNPass - The public interface to this file...
Bob Wilson11361662010-02-28 05:34:05 +0000747FunctionPass *llvm::createGVNPass(bool NoLoads) {
748 return new GVN(NoLoads);
Dan Gohman81132462009-11-14 02:27:51 +0000749}
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000750
Owen Anderson8ac477f2010-10-12 19:48:12 +0000751INITIALIZE_PASS_BEGIN(GVN, "gvn", "Global Value Numbering", false, false)
752INITIALIZE_PASS_DEPENDENCY(MemoryDependenceAnalysis)
753INITIALIZE_PASS_DEPENDENCY(DominatorTree)
754INITIALIZE_AG_DEPENDENCY(AliasAnalysis)
755INITIALIZE_PASS_END(GVN, "gvn", "Global Value Numbering", false, false)
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000756
Owen Anderson6a903bc2008-06-18 21:41:49 +0000757void GVN::dump(DenseMap<uint32_t, Value*>& d) {
Dan Gohman57e80862009-12-18 03:25:51 +0000758 errs() << "{\n";
Owen Anderson6a903bc2008-06-18 21:41:49 +0000759 for (DenseMap<uint32_t, Value*>::iterator I = d.begin(),
Owen Anderson5e5599b2007-07-25 19:57:03 +0000760 E = d.end(); I != E; ++I) {
Dan Gohman57e80862009-12-18 03:25:51 +0000761 errs() << I->first << "\n";
Owen Anderson5e5599b2007-07-25 19:57:03 +0000762 I->second->dump();
763 }
Dan Gohman57e80862009-12-18 03:25:51 +0000764 errs() << "}\n";
Owen Anderson5e5599b2007-07-25 19:57:03 +0000765}
766
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000767/// IsValueFullyAvailableInBlock - Return true if we can prove that the value
768/// we're analyzing is fully available in the specified block. As we go, keep
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000769/// track of which blocks we know are fully alive in FullyAvailableBlocks. This
770/// map is actually a tri-state map with the following values:
771/// 0) we know the block *is not* fully available.
772/// 1) we know the block *is* fully available.
773/// 2) we do not know whether the block is fully available or not, but we are
774/// currently speculating that it will be.
775/// 3) we are speculating for this block and have used that to speculate for
776/// other blocks.
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000777static bool IsValueFullyAvailableInBlock(BasicBlock *BB,
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000778 DenseMap<BasicBlock*, char> &FullyAvailableBlocks) {
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000779 // Optimistically assume that the block is fully available and check to see
780 // if we already know about this block in one lookup.
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000781 std::pair<DenseMap<BasicBlock*, char>::iterator, char> IV =
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000782 FullyAvailableBlocks.insert(std::make_pair(BB, 2));
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000783
784 // If the entry already existed for this block, return the precomputed value.
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000785 if (!IV.second) {
786 // If this is a speculative "available" value, mark it as being used for
787 // speculation of other blocks.
788 if (IV.first->second == 2)
789 IV.first->second = 3;
790 return IV.first->second != 0;
791 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000792
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000793 // Otherwise, see if it is fully available in all predecessors.
794 pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000795
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000796 // If this block has no predecessors, it isn't live-in here.
797 if (PI == PE)
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000798 goto SpeculationFailure;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000799
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000800 for (; PI != PE; ++PI)
801 // If the value isn't fully available in one of our predecessors, then it
802 // isn't fully available in this block either. Undo our previous
803 // optimistic assumption and bail out.
804 if (!IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000805 goto SpeculationFailure;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000806
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000807 return true;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000808
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000809// SpeculationFailure - If we get here, we found out that this is not, after
810// all, a fully-available block. We have a problem if we speculated on this and
811// used the speculation to mark other blocks as available.
812SpeculationFailure:
813 char &BBVal = FullyAvailableBlocks[BB];
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000814
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000815 // If we didn't speculate on this, just return with it set to false.
816 if (BBVal == 2) {
817 BBVal = 0;
818 return false;
819 }
820
821 // If we did speculate on this value, we could have blocks set to 1 that are
822 // incorrect. Walk the (transitive) successors of this block and mark them as
823 // 0 if set to one.
824 SmallVector<BasicBlock*, 32> BBWorklist;
825 BBWorklist.push_back(BB);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000826
Dan Gohman28943872010-01-05 16:27:25 +0000827 do {
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000828 BasicBlock *Entry = BBWorklist.pop_back_val();
829 // Note that this sets blocks to 0 (unavailable) if they happen to not
830 // already be in FullyAvailableBlocks. This is safe.
831 char &EntryVal = FullyAvailableBlocks[Entry];
832 if (EntryVal == 0) continue; // Already unavailable.
833
834 // Mark as unavailable.
835 EntryVal = 0;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000836
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000837 for (succ_iterator I = succ_begin(Entry), E = succ_end(Entry); I != E; ++I)
838 BBWorklist.push_back(*I);
Dan Gohman28943872010-01-05 16:27:25 +0000839 } while (!BBWorklist.empty());
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000840
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000841 return false;
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000842}
843
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000844
Chris Lattner9045f232009-09-21 17:24:04 +0000845/// CanCoerceMustAliasedValueToLoad - Return true if
846/// CoerceAvailableValueToLoadType will succeed.
847static bool CanCoerceMustAliasedValueToLoad(Value *StoredVal,
848 const Type *LoadTy,
849 const TargetData &TD) {
850 // If the loaded or stored value is an first class array or struct, don't try
851 // to transform them. We need to be able to bitcast to integer.
Duncan Sands19d0b472010-02-16 11:11:14 +0000852 if (LoadTy->isStructTy() || LoadTy->isArrayTy() ||
853 StoredVal->getType()->isStructTy() ||
854 StoredVal->getType()->isArrayTy())
Chris Lattner9045f232009-09-21 17:24:04 +0000855 return false;
856
857 // The store has to be at least as big as the load.
858 if (TD.getTypeSizeInBits(StoredVal->getType()) <
859 TD.getTypeSizeInBits(LoadTy))
860 return false;
861
862 return true;
863}
864
865
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000866/// CoerceAvailableValueToLoadType - If we saw a store of a value to memory, and
867/// then a load from a must-aliased pointer of a different type, try to coerce
868/// the stored value. LoadedTy is the type of the load we want to replace and
869/// InsertPt is the place to insert new instructions.
870///
871/// If we can't do it, return null.
872static Value *CoerceAvailableValueToLoadType(Value *StoredVal,
873 const Type *LoadedTy,
874 Instruction *InsertPt,
875 const TargetData &TD) {
Chris Lattner9045f232009-09-21 17:24:04 +0000876 if (!CanCoerceMustAliasedValueToLoad(StoredVal, LoadedTy, TD))
877 return 0;
878
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000879 const Type *StoredValTy = StoredVal->getType();
880
Chris Lattner5a62d6e2010-05-08 20:01:44 +0000881 uint64_t StoreSize = TD.getTypeStoreSizeInBits(StoredValTy);
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000882 uint64_t LoadSize = TD.getTypeSizeInBits(LoadedTy);
883
884 // If the store and reload are the same size, we can always reuse it.
885 if (StoreSize == LoadSize) {
Duncan Sands19d0b472010-02-16 11:11:14 +0000886 if (StoredValTy->isPointerTy() && LoadedTy->isPointerTy()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000887 // Pointer to Pointer -> use bitcast.
888 return new BitCastInst(StoredVal, LoadedTy, "", InsertPt);
889 }
890
891 // Convert source pointers to integers, which can be bitcast.
Duncan Sands19d0b472010-02-16 11:11:14 +0000892 if (StoredValTy->isPointerTy()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000893 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
894 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
895 }
896
897 const Type *TypeToCastTo = LoadedTy;
Duncan Sands19d0b472010-02-16 11:11:14 +0000898 if (TypeToCastTo->isPointerTy())
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000899 TypeToCastTo = TD.getIntPtrType(StoredValTy->getContext());
900
901 if (StoredValTy != TypeToCastTo)
902 StoredVal = new BitCastInst(StoredVal, TypeToCastTo, "", InsertPt);
903
904 // Cast to pointer if the load needs a pointer type.
Duncan Sands19d0b472010-02-16 11:11:14 +0000905 if (LoadedTy->isPointerTy())
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000906 StoredVal = new IntToPtrInst(StoredVal, LoadedTy, "", InsertPt);
907
908 return StoredVal;
909 }
910
911 // If the loaded value is smaller than the available value, then we can
912 // extract out a piece from it. If the available value is too small, then we
913 // can't do anything.
Chris Lattner9045f232009-09-21 17:24:04 +0000914 assert(StoreSize >= LoadSize && "CanCoerceMustAliasedValueToLoad fail");
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000915
916 // Convert source pointers to integers, which can be manipulated.
Duncan Sands19d0b472010-02-16 11:11:14 +0000917 if (StoredValTy->isPointerTy()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000918 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
919 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
920 }
921
922 // Convert vectors and fp to integer, which can be manipulated.
Duncan Sands19d0b472010-02-16 11:11:14 +0000923 if (!StoredValTy->isIntegerTy()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000924 StoredValTy = IntegerType::get(StoredValTy->getContext(), StoreSize);
925 StoredVal = new BitCastInst(StoredVal, StoredValTy, "", InsertPt);
926 }
927
928 // If this is a big-endian system, we need to shift the value down to the low
929 // bits so that a truncate will work.
930 if (TD.isBigEndian()) {
931 Constant *Val = ConstantInt::get(StoredVal->getType(), StoreSize-LoadSize);
932 StoredVal = BinaryOperator::CreateLShr(StoredVal, Val, "tmp", InsertPt);
933 }
934
935 // Truncate the integer to the right size now.
936 const Type *NewIntTy = IntegerType::get(StoredValTy->getContext(), LoadSize);
937 StoredVal = new TruncInst(StoredVal, NewIntTy, "trunc", InsertPt);
938
939 if (LoadedTy == NewIntTy)
940 return StoredVal;
941
942 // If the result is a pointer, inttoptr.
Duncan Sands19d0b472010-02-16 11:11:14 +0000943 if (LoadedTy->isPointerTy())
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000944 return new IntToPtrInst(StoredVal, LoadedTy, "inttoptr", InsertPt);
945
946 // Otherwise, bitcast.
947 return new BitCastInst(StoredVal, LoadedTy, "bitcast", InsertPt);
948}
949
Chris Lattner42376062009-12-06 01:57:02 +0000950/// AnalyzeLoadFromClobberingWrite - This function is called when we have a
951/// memdep query of a load that ends up being a clobbering memory write (store,
952/// memset, memcpy, memmove). This means that the write *may* provide bits used
953/// by the load but we can't be sure because the pointers don't mustalias.
954///
955/// Check this case to see if there is anything more we can do before we give
956/// up. This returns -1 if we have to give up, or a byte number in the stored
957/// value of the piece that feeds the load.
Chris Lattner0def8612009-12-09 07:34:10 +0000958static int AnalyzeLoadFromClobberingWrite(const Type *LoadTy, Value *LoadPtr,
959 Value *WritePtr,
Chris Lattner42376062009-12-06 01:57:02 +0000960 uint64_t WriteSizeInBits,
Chris Lattner4d8af2f2009-09-21 06:48:08 +0000961 const TargetData &TD) {
Chris Lattner9045f232009-09-21 17:24:04 +0000962 // If the loaded or stored value is an first class array or struct, don't try
963 // to transform them. We need to be able to bitcast to integer.
Duncan Sands19d0b472010-02-16 11:11:14 +0000964 if (LoadTy->isStructTy() || LoadTy->isArrayTy())
Chris Lattner9045f232009-09-21 17:24:04 +0000965 return -1;
966
Chris Lattnerd28f9082009-09-21 06:24:16 +0000967 int64_t StoreOffset = 0, LoadOffset = 0;
Chris Lattnere28618d2010-11-30 22:25:26 +0000968 Value *StoreBase = GetPointerBaseWithConstantOffset(WritePtr, StoreOffset,TD);
969 Value *LoadBase = GetPointerBaseWithConstantOffset(LoadPtr, LoadOffset, TD);
Chris Lattnerd28f9082009-09-21 06:24:16 +0000970 if (StoreBase != LoadBase)
971 return -1;
972
973 // If the load and store are to the exact same address, they should have been
974 // a must alias. AA must have gotten confused.
Chris Lattner05638042010-03-25 05:58:19 +0000975 // FIXME: Study to see if/when this happens. One case is forwarding a memset
976 // to a load from the base of the memset.
Chris Lattnerd28f9082009-09-21 06:24:16 +0000977#if 0
Chris Lattner05638042010-03-25 05:58:19 +0000978 if (LoadOffset == StoreOffset) {
David Greene2e6efc42010-01-05 01:27:17 +0000979 dbgs() << "STORE/LOAD DEP WITH COMMON POINTER MISSED:\n"
Chris Lattnerd28f9082009-09-21 06:24:16 +0000980 << "Base = " << *StoreBase << "\n"
Chris Lattner42376062009-12-06 01:57:02 +0000981 << "Store Ptr = " << *WritePtr << "\n"
982 << "Store Offs = " << StoreOffset << "\n"
Chris Lattner3ddf8042009-12-10 00:04:46 +0000983 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattner946b58d2009-12-09 02:41:54 +0000984 abort();
Chris Lattnerd28f9082009-09-21 06:24:16 +0000985 }
Chris Lattner05638042010-03-25 05:58:19 +0000986#endif
Chris Lattnerd28f9082009-09-21 06:24:16 +0000987
988 // If the load and store don't overlap at all, the store doesn't provide
989 // anything to the load. In this case, they really don't alias at all, AA
990 // must have gotten confused.
Chris Lattner0def8612009-12-09 07:34:10 +0000991 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy);
Chris Lattnerd28f9082009-09-21 06:24:16 +0000992
Chris Lattner42376062009-12-06 01:57:02 +0000993 if ((WriteSizeInBits & 7) | (LoadSize & 7))
Chris Lattnerd28f9082009-09-21 06:24:16 +0000994 return -1;
Chris Lattner42376062009-12-06 01:57:02 +0000995 uint64_t StoreSize = WriteSizeInBits >> 3; // Convert to bytes.
Chris Lattnerd28f9082009-09-21 06:24:16 +0000996 LoadSize >>= 3;
997
998
999 bool isAAFailure = false;
Chris Lattner05638042010-03-25 05:58:19 +00001000 if (StoreOffset < LoadOffset)
Chris Lattnerd28f9082009-09-21 06:24:16 +00001001 isAAFailure = StoreOffset+int64_t(StoreSize) <= LoadOffset;
Chris Lattner05638042010-03-25 05:58:19 +00001002 else
Chris Lattnerd28f9082009-09-21 06:24:16 +00001003 isAAFailure = LoadOffset+int64_t(LoadSize) <= StoreOffset;
Chris Lattner05638042010-03-25 05:58:19 +00001004
Chris Lattnerd28f9082009-09-21 06:24:16 +00001005 if (isAAFailure) {
1006#if 0
David Greene2e6efc42010-01-05 01:27:17 +00001007 dbgs() << "STORE LOAD DEP WITH COMMON BASE:\n"
Chris Lattnerd28f9082009-09-21 06:24:16 +00001008 << "Base = " << *StoreBase << "\n"
Chris Lattner42376062009-12-06 01:57:02 +00001009 << "Store Ptr = " << *WritePtr << "\n"
1010 << "Store Offs = " << StoreOffset << "\n"
Chris Lattner3ddf8042009-12-10 00:04:46 +00001011 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattner946b58d2009-12-09 02:41:54 +00001012 abort();
Chris Lattnerd28f9082009-09-21 06:24:16 +00001013#endif
1014 return -1;
1015 }
1016
1017 // If the Load isn't completely contained within the stored bits, we don't
1018 // have all the bits to feed it. We could do something crazy in the future
1019 // (issue a smaller load then merge the bits in) but this seems unlikely to be
1020 // valuable.
1021 if (StoreOffset > LoadOffset ||
1022 StoreOffset+StoreSize < LoadOffset+LoadSize)
1023 return -1;
1024
1025 // Okay, we can do this transformation. Return the number of bytes into the
1026 // store that the load is.
1027 return LoadOffset-StoreOffset;
1028}
1029
Chris Lattner42376062009-12-06 01:57:02 +00001030/// AnalyzeLoadFromClobberingStore - This function is called when we have a
1031/// memdep query of a load that ends up being a clobbering store.
Chris Lattner07df9ef2009-12-09 07:37:07 +00001032static int AnalyzeLoadFromClobberingStore(const Type *LoadTy, Value *LoadPtr,
1033 StoreInst *DepSI,
Chris Lattner42376062009-12-06 01:57:02 +00001034 const TargetData &TD) {
1035 // Cannot handle reading from store of first-class aggregate yet.
Dan Gohmand2099112010-11-10 19:03:33 +00001036 if (DepSI->getValueOperand()->getType()->isStructTy() ||
1037 DepSI->getValueOperand()->getType()->isArrayTy())
Chris Lattner42376062009-12-06 01:57:02 +00001038 return -1;
1039
1040 Value *StorePtr = DepSI->getPointerOperand();
Dan Gohmand2099112010-11-10 19:03:33 +00001041 uint64_t StoreSize =TD.getTypeSizeInBits(DepSI->getValueOperand()->getType());
Chris Lattner07df9ef2009-12-09 07:37:07 +00001042 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
Chris Lattner0def8612009-12-09 07:34:10 +00001043 StorePtr, StoreSize, TD);
Chris Lattner42376062009-12-06 01:57:02 +00001044}
1045
Chris Lattner07df9ef2009-12-09 07:37:07 +00001046static int AnalyzeLoadFromClobberingMemInst(const Type *LoadTy, Value *LoadPtr,
1047 MemIntrinsic *MI,
Chris Lattner42376062009-12-06 01:57:02 +00001048 const TargetData &TD) {
1049 // If the mem operation is a non-constant size, we can't handle it.
1050 ConstantInt *SizeCst = dyn_cast<ConstantInt>(MI->getLength());
1051 if (SizeCst == 0) return -1;
1052 uint64_t MemSizeInBits = SizeCst->getZExtValue()*8;
Chris Lattner778cb922009-12-06 05:29:56 +00001053
1054 // If this is memset, we just need to see if the offset is valid in the size
1055 // of the memset..
Chris Lattner42376062009-12-06 01:57:02 +00001056 if (MI->getIntrinsicID() == Intrinsic::memset)
Chris Lattner07df9ef2009-12-09 07:37:07 +00001057 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr, MI->getDest(),
1058 MemSizeInBits, TD);
Chris Lattner42376062009-12-06 01:57:02 +00001059
Chris Lattner778cb922009-12-06 05:29:56 +00001060 // If we have a memcpy/memmove, the only case we can handle is if this is a
1061 // copy from constant memory. In that case, we can read directly from the
1062 // constant memory.
1063 MemTransferInst *MTI = cast<MemTransferInst>(MI);
1064
1065 Constant *Src = dyn_cast<Constant>(MTI->getSource());
1066 if (Src == 0) return -1;
1067
Dan Gohmana4fcd242010-12-15 20:02:24 +00001068 GlobalVariable *GV = dyn_cast<GlobalVariable>(GetUnderlyingObject(Src));
Chris Lattner778cb922009-12-06 05:29:56 +00001069 if (GV == 0 || !GV->isConstant()) return -1;
1070
1071 // See if the access is within the bounds of the transfer.
Chris Lattner07df9ef2009-12-09 07:37:07 +00001072 int Offset = AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
1073 MI->getDest(), MemSizeInBits, TD);
Chris Lattner778cb922009-12-06 05:29:56 +00001074 if (Offset == -1)
1075 return Offset;
1076
1077 // Otherwise, see if we can constant fold a load from the constant with the
1078 // offset applied as appropriate.
1079 Src = ConstantExpr::getBitCast(Src,
1080 llvm::Type::getInt8PtrTy(Src->getContext()));
1081 Constant *OffsetCst =
1082 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1083 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
Chris Lattner07df9ef2009-12-09 07:37:07 +00001084 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
Chris Lattner778cb922009-12-06 05:29:56 +00001085 if (ConstantFoldLoadFromConstPtr(Src, &TD))
1086 return Offset;
Chris Lattner42376062009-12-06 01:57:02 +00001087 return -1;
1088}
1089
Chris Lattnerd28f9082009-09-21 06:24:16 +00001090
1091/// GetStoreValueForLoad - This function is called when we have a
1092/// memdep query of a load that ends up being a clobbering store. This means
1093/// that the store *may* provide bits used by the load but we can't be sure
1094/// because the pointers don't mustalias. Check this case to see if there is
1095/// anything more we can do before we give up.
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001096static Value *GetStoreValueForLoad(Value *SrcVal, unsigned Offset,
1097 const Type *LoadTy,
1098 Instruction *InsertPt, const TargetData &TD){
Chris Lattnerd28f9082009-09-21 06:24:16 +00001099 LLVMContext &Ctx = SrcVal->getType()->getContext();
1100
Chris Lattner5a62d6e2010-05-08 20:01:44 +00001101 uint64_t StoreSize = (TD.getTypeSizeInBits(SrcVal->getType()) + 7) / 8;
1102 uint64_t LoadSize = (TD.getTypeSizeInBits(LoadTy) + 7) / 8;
Chris Lattnerd28f9082009-09-21 06:24:16 +00001103
Chris Lattnerf8ba1252009-12-09 18:13:28 +00001104 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
Chris Lattnerd28f9082009-09-21 06:24:16 +00001105
1106 // Compute which bits of the stored value are being used by the load. Convert
1107 // to an integer type to start with.
Duncan Sands19d0b472010-02-16 11:11:14 +00001108 if (SrcVal->getType()->isPointerTy())
Chris Lattnerf8ba1252009-12-09 18:13:28 +00001109 SrcVal = Builder.CreatePtrToInt(SrcVal, TD.getIntPtrType(Ctx), "tmp");
Duncan Sands19d0b472010-02-16 11:11:14 +00001110 if (!SrcVal->getType()->isIntegerTy())
Chris Lattnerf8ba1252009-12-09 18:13:28 +00001111 SrcVal = Builder.CreateBitCast(SrcVal, IntegerType::get(Ctx, StoreSize*8),
1112 "tmp");
Chris Lattnerd28f9082009-09-21 06:24:16 +00001113
1114 // Shift the bits to the least significant depending on endianness.
1115 unsigned ShiftAmt;
Chris Lattner42376062009-12-06 01:57:02 +00001116 if (TD.isLittleEndian())
Chris Lattnerd28f9082009-09-21 06:24:16 +00001117 ShiftAmt = Offset*8;
Chris Lattner42376062009-12-06 01:57:02 +00001118 else
Chris Lattner24705382009-09-21 17:55:47 +00001119 ShiftAmt = (StoreSize-LoadSize-Offset)*8;
Chris Lattnerd28f9082009-09-21 06:24:16 +00001120
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001121 if (ShiftAmt)
Chris Lattnerf8ba1252009-12-09 18:13:28 +00001122 SrcVal = Builder.CreateLShr(SrcVal, ShiftAmt, "tmp");
Chris Lattnerd28f9082009-09-21 06:24:16 +00001123
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001124 if (LoadSize != StoreSize)
Chris Lattnerf8ba1252009-12-09 18:13:28 +00001125 SrcVal = Builder.CreateTrunc(SrcVal, IntegerType::get(Ctx, LoadSize*8),
1126 "tmp");
Chris Lattnerd28f9082009-09-21 06:24:16 +00001127
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001128 return CoerceAvailableValueToLoadType(SrcVal, LoadTy, InsertPt, TD);
Chris Lattnerd28f9082009-09-21 06:24:16 +00001129}
1130
Chris Lattner42376062009-12-06 01:57:02 +00001131/// GetMemInstValueForLoad - This function is called when we have a
1132/// memdep query of a load that ends up being a clobbering mem intrinsic.
1133static Value *GetMemInstValueForLoad(MemIntrinsic *SrcInst, unsigned Offset,
1134 const Type *LoadTy, Instruction *InsertPt,
1135 const TargetData &TD){
1136 LLVMContext &Ctx = LoadTy->getContext();
1137 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
1138
1139 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
1140
1141 // We know that this method is only called when the mem transfer fully
1142 // provides the bits for the load.
1143 if (MemSetInst *MSI = dyn_cast<MemSetInst>(SrcInst)) {
1144 // memset(P, 'x', 1234) -> splat('x'), even if x is a variable, and
1145 // independently of what the offset is.
1146 Value *Val = MSI->getValue();
1147 if (LoadSize != 1)
1148 Val = Builder.CreateZExt(Val, IntegerType::get(Ctx, LoadSize*8));
1149
1150 Value *OneElt = Val;
1151
1152 // Splat the value out to the right number of bits.
1153 for (unsigned NumBytesSet = 1; NumBytesSet != LoadSize; ) {
1154 // If we can double the number of bytes set, do it.
1155 if (NumBytesSet*2 <= LoadSize) {
1156 Value *ShVal = Builder.CreateShl(Val, NumBytesSet*8);
1157 Val = Builder.CreateOr(Val, ShVal);
1158 NumBytesSet <<= 1;
1159 continue;
1160 }
1161
1162 // Otherwise insert one byte at a time.
1163 Value *ShVal = Builder.CreateShl(Val, 1*8);
1164 Val = Builder.CreateOr(OneElt, ShVal);
1165 ++NumBytesSet;
1166 }
1167
1168 return CoerceAvailableValueToLoadType(Val, LoadTy, InsertPt, TD);
1169 }
Chris Lattner778cb922009-12-06 05:29:56 +00001170
1171 // Otherwise, this is a memcpy/memmove from a constant global.
1172 MemTransferInst *MTI = cast<MemTransferInst>(SrcInst);
1173 Constant *Src = cast<Constant>(MTI->getSource());
1174
1175 // Otherwise, see if we can constant fold a load from the constant with the
1176 // offset applied as appropriate.
1177 Src = ConstantExpr::getBitCast(Src,
1178 llvm::Type::getInt8PtrTy(Src->getContext()));
1179 Constant *OffsetCst =
1180 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1181 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
1182 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
1183 return ConstantFoldLoadFromConstPtr(Src, &TD);
Chris Lattner42376062009-12-06 01:57:02 +00001184}
1185
Dan Gohmanb29cda92010-04-15 17:08:50 +00001186namespace {
Chris Lattner42376062009-12-06 01:57:02 +00001187
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001188struct AvailableValueInBlock {
1189 /// BB - The basic block in question.
1190 BasicBlock *BB;
Chris Lattner93236ba2009-12-06 04:54:31 +00001191 enum ValType {
1192 SimpleVal, // A simple offsetted value that is accessed.
1193 MemIntrin // A memory intrinsic which is loaded from.
1194 };
1195
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001196 /// V - The value that is live out of the block.
Chris Lattner93236ba2009-12-06 04:54:31 +00001197 PointerIntPair<Value *, 1, ValType> Val;
1198
1199 /// Offset - The byte offset in Val that is interesting for the load query.
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001200 unsigned Offset;
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001201
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001202 static AvailableValueInBlock get(BasicBlock *BB, Value *V,
1203 unsigned Offset = 0) {
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001204 AvailableValueInBlock Res;
1205 Res.BB = BB;
Chris Lattner93236ba2009-12-06 04:54:31 +00001206 Res.Val.setPointer(V);
1207 Res.Val.setInt(SimpleVal);
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001208 Res.Offset = Offset;
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001209 return Res;
1210 }
Chris Lattner93236ba2009-12-06 04:54:31 +00001211
1212 static AvailableValueInBlock getMI(BasicBlock *BB, MemIntrinsic *MI,
1213 unsigned Offset = 0) {
1214 AvailableValueInBlock Res;
1215 Res.BB = BB;
1216 Res.Val.setPointer(MI);
1217 Res.Val.setInt(MemIntrin);
1218 Res.Offset = Offset;
1219 return Res;
1220 }
1221
1222 bool isSimpleValue() const { return Val.getInt() == SimpleVal; }
1223 Value *getSimpleValue() const {
1224 assert(isSimpleValue() && "Wrong accessor");
1225 return Val.getPointer();
1226 }
1227
1228 MemIntrinsic *getMemIntrinValue() const {
1229 assert(!isSimpleValue() && "Wrong accessor");
1230 return cast<MemIntrinsic>(Val.getPointer());
1231 }
Chris Lattner927b0ac2009-12-21 23:04:33 +00001232
1233 /// MaterializeAdjustedValue - Emit code into this block to adjust the value
1234 /// defined here to the specified type. This handles various coercion cases.
1235 Value *MaterializeAdjustedValue(const Type *LoadTy,
1236 const TargetData *TD) const {
1237 Value *Res;
1238 if (isSimpleValue()) {
1239 Res = getSimpleValue();
1240 if (Res->getType() != LoadTy) {
1241 assert(TD && "Need target data to handle type mismatch case");
1242 Res = GetStoreValueForLoad(Res, Offset, LoadTy, BB->getTerminator(),
1243 *TD);
1244
1245 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\nOffset: " << Offset << " "
1246 << *getSimpleValue() << '\n'
1247 << *Res << '\n' << "\n\n\n");
1248 }
1249 } else {
1250 Res = GetMemInstValueForLoad(getMemIntrinValue(), Offset,
1251 LoadTy, BB->getTerminator(), *TD);
1252 DEBUG(errs() << "GVN COERCED NONLOCAL MEM INTRIN:\nOffset: " << Offset
1253 << " " << *getMemIntrinValue() << '\n'
1254 << *Res << '\n' << "\n\n\n");
1255 }
1256 return Res;
1257 }
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001258};
1259
Dan Gohmanb29cda92010-04-15 17:08:50 +00001260}
1261
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001262/// ConstructSSAForLoadSet - Given a set of loads specified by ValuesPerBlock,
1263/// construct SSA form, allowing us to eliminate LI. This returns the value
1264/// that should be used at LI's definition site.
1265static Value *ConstructSSAForLoadSet(LoadInst *LI,
1266 SmallVectorImpl<AvailableValueInBlock> &ValuesPerBlock,
1267 const TargetData *TD,
Chris Lattnerbf200182009-12-21 23:15:48 +00001268 const DominatorTree &DT,
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001269 AliasAnalysis *AA) {
Chris Lattnerbf200182009-12-21 23:15:48 +00001270 // Check for the fully redundant, dominating load case. In this case, we can
1271 // just use the dominating value directly.
1272 if (ValuesPerBlock.size() == 1 &&
1273 DT.properlyDominates(ValuesPerBlock[0].BB, LI->getParent()))
1274 return ValuesPerBlock[0].MaterializeAdjustedValue(LI->getType(), TD);
1275
1276 // Otherwise, we have to construct SSA form.
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001277 SmallVector<PHINode*, 8> NewPHIs;
1278 SSAUpdater SSAUpdate(&NewPHIs);
Duncan Sands67781492010-09-02 08:14:03 +00001279 SSAUpdate.Initialize(LI->getType(), LI->getName());
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001280
1281 const Type *LoadTy = LI->getType();
1282
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001283 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
Chris Lattner93236ba2009-12-06 04:54:31 +00001284 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1285 BasicBlock *BB = AV.BB;
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001286
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001287 if (SSAUpdate.HasValueForBlock(BB))
1288 continue;
Chris Lattner93236ba2009-12-06 04:54:31 +00001289
Chris Lattner927b0ac2009-12-21 23:04:33 +00001290 SSAUpdate.AddAvailableValue(BB, AV.MaterializeAdjustedValue(LoadTy, TD));
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001291 }
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001292
1293 // Perform PHI construction.
1294 Value *V = SSAUpdate.GetValueInMiddleOfBlock(LI->getParent());
1295
1296 // If new PHI nodes were created, notify alias analysis.
Duncan Sands19d0b472010-02-16 11:11:14 +00001297 if (V->getType()->isPointerTy())
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001298 for (unsigned i = 0, e = NewPHIs.size(); i != e; ++i)
1299 AA->copyValue(LI, NewPHIs[i]);
1300
1301 return V;
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001302}
1303
Gabor Greifce6dd882010-04-09 10:57:00 +00001304static bool isLifetimeStart(const Instruction *Inst) {
1305 if (const IntrinsicInst* II = dyn_cast<IntrinsicInst>(Inst))
Owen Andersonb9878ee2009-12-02 07:35:19 +00001306 return II->getIntrinsicID() == Intrinsic::lifetime_start;
Chris Lattnerc4680252009-12-02 06:44:58 +00001307 return false;
1308}
1309
Owen Anderson221a4362007-08-16 22:02:55 +00001310/// processNonLocalLoad - Attempt to eliminate a load whose dependencies are
1311/// non-local by performing PHI construction.
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001312bool GVN::processNonLocalLoad(LoadInst *LI,
Chris Lattner804209d2008-03-21 22:01:16 +00001313 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001314 // Find the non-local dependencies of the load.
Chris Lattner9b7d99e2009-12-22 04:25:02 +00001315 SmallVector<NonLocalDepResult, 64> Deps;
Dan Gohman65316d62010-11-11 21:50:19 +00001316 AliasAnalysis::Location Loc = VN.getAliasAnalysis()->getLocation(LI);
1317 MD->getNonLocalPointerDependency(Loc, true, LI->getParent(), Deps);
David Greene2e6efc42010-01-05 01:27:17 +00001318 //DEBUG(dbgs() << "INVESTIGATING NONLOCAL LOAD: "
Dan Gohmanef3ef7f2009-07-31 20:24:18 +00001319 // << Deps.size() << *LI << '\n');
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001320
Owen Andersonb39e0de2008-08-26 22:07:42 +00001321 // If we had to process more than one hundred blocks to find the
1322 // dependencies, this load isn't worth worrying about. Optimizing
1323 // it will be too expensive.
Chris Lattnerb6fc4b82008-12-09 19:25:07 +00001324 if (Deps.size() > 100)
Owen Andersonb39e0de2008-08-26 22:07:42 +00001325 return false;
Chris Lattnerb6372932008-12-18 00:51:32 +00001326
1327 // If we had a phi translation failure, we'll have a single entry which is a
1328 // clobber in the current block. Reject this early.
Chris Lattner0c315472009-12-09 07:08:01 +00001329 if (Deps.size() == 1 && Deps[0].getResult().isClobber()) {
Torok Edwinba93ea72009-06-17 18:48:18 +00001330 DEBUG(
David Greene2e6efc42010-01-05 01:27:17 +00001331 dbgs() << "GVN: non-local load ";
1332 WriteAsOperand(dbgs(), LI);
1333 dbgs() << " is clobbered by " << *Deps[0].getResult().getInst() << '\n';
Torok Edwinba93ea72009-06-17 18:48:18 +00001334 );
Chris Lattnerb6372932008-12-18 00:51:32 +00001335 return false;
Torok Edwinba93ea72009-06-17 18:48:18 +00001336 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001337
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001338 // Filter out useless results (non-locals, etc). Keep track of the blocks
1339 // where we have a value available in repl, also keep track of whether we see
1340 // dependencies that produce an unknown value for the load (such as a call
1341 // that could potentially clobber the load).
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001342 SmallVector<AvailableValueInBlock, 16> ValuesPerBlock;
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001343 SmallVector<BasicBlock*, 16> UnavailableBlocks;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001344
Chris Lattnerb6fc4b82008-12-09 19:25:07 +00001345 for (unsigned i = 0, e = Deps.size(); i != e; ++i) {
Chris Lattner0c315472009-12-09 07:08:01 +00001346 BasicBlock *DepBB = Deps[i].getBB();
1347 MemDepResult DepInfo = Deps[i].getResult();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001348
Chris Lattner0e3d6332008-12-05 21:04:20 +00001349 if (DepInfo.isClobber()) {
Chris Lattnerca5f9cb2009-12-09 18:21:46 +00001350 // The address being loaded in this non-local block may not be the same as
1351 // the pointer operand of the load if PHI translation occurs. Make sure
1352 // to consider the right address.
1353 Value *Address = Deps[i].getAddress();
1354
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001355 // If the dependence is to a store that writes to a superset of the bits
1356 // read by the load, we can extract the bits we need for the load from the
1357 // stored value.
1358 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInfo.getInst())) {
Chris Lattnerca5f9cb2009-12-09 18:21:46 +00001359 if (TD && Address) {
1360 int Offset = AnalyzeLoadFromClobberingStore(LI->getType(), Address,
Chris Lattner07df9ef2009-12-09 07:37:07 +00001361 DepSI, *TD);
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001362 if (Offset != -1) {
1363 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
Dan Gohmand2099112010-11-10 19:03:33 +00001364 DepSI->getValueOperand(),
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001365 Offset));
1366 continue;
1367 }
1368 }
1369 }
Chris Lattner42376062009-12-06 01:57:02 +00001370
Chris Lattner42376062009-12-06 01:57:02 +00001371 // If the clobbering value is a memset/memcpy/memmove, see if we can
1372 // forward a value on from it.
Chris Lattner93236ba2009-12-06 04:54:31 +00001373 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(DepInfo.getInst())) {
Chris Lattnerca5f9cb2009-12-09 18:21:46 +00001374 if (TD && Address) {
1375 int Offset = AnalyzeLoadFromClobberingMemInst(LI->getType(), Address,
Chris Lattner07df9ef2009-12-09 07:37:07 +00001376 DepMI, *TD);
Chris Lattner93236ba2009-12-06 04:54:31 +00001377 if (Offset != -1) {
1378 ValuesPerBlock.push_back(AvailableValueInBlock::getMI(DepBB, DepMI,
1379 Offset));
1380 continue;
1381 }
Chris Lattner42376062009-12-06 01:57:02 +00001382 }
1383 }
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001384
Chris Lattner0e3d6332008-12-05 21:04:20 +00001385 UnavailableBlocks.push_back(DepBB);
1386 continue;
1387 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001388
Chris Lattner0e3d6332008-12-05 21:04:20 +00001389 Instruction *DepInst = DepInfo.getInst();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001390
Chris Lattner0e3d6332008-12-05 21:04:20 +00001391 // Loading the allocation -> undef.
Chris Lattnerc4680252009-12-02 06:44:58 +00001392 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst) ||
Owen Andersonb9878ee2009-12-02 07:35:19 +00001393 // Loading immediately after lifetime begin -> undef.
1394 isLifetimeStart(DepInst)) {
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001395 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1396 UndefValue::get(LI->getType())));
Chris Lattner7e61daf2008-12-01 01:15:42 +00001397 continue;
1398 }
Owen Anderson2b2bd282009-10-28 07:05:35 +00001399
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001400 if (StoreInst *S = dyn_cast<StoreInst>(DepInst)) {
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001401 // Reject loads and stores that are to the same address but are of
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001402 // different types if we have to.
Dan Gohmand2099112010-11-10 19:03:33 +00001403 if (S->getValueOperand()->getType() != LI->getType()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001404 // If the stored value is larger or equal to the loaded value, we can
1405 // reuse it.
Dan Gohmand2099112010-11-10 19:03:33 +00001406 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(S->getValueOperand(),
Chris Lattner9045f232009-09-21 17:24:04 +00001407 LI->getType(), *TD)) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001408 UnavailableBlocks.push_back(DepBB);
1409 continue;
1410 }
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001411 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001412
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001413 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
Dan Gohmand2099112010-11-10 19:03:33 +00001414 S->getValueOperand()));
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001415 continue;
1416 }
1417
1418 if (LoadInst *LD = dyn_cast<LoadInst>(DepInst)) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001419 // If the types mismatch and we can't handle it, reject reuse of the load.
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001420 if (LD->getType() != LI->getType()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001421 // If the stored value is larger or equal to the loaded value, we can
1422 // reuse it.
Chris Lattner9045f232009-09-21 17:24:04 +00001423 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(LD, LI->getType(),*TD)){
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001424 UnavailableBlocks.push_back(DepBB);
1425 continue;
1426 }
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001427 }
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001428 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB, LD));
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001429 continue;
Owen Anderson5e5599b2007-07-25 19:57:03 +00001430 }
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001431
1432 UnavailableBlocks.push_back(DepBB);
1433 continue;
Chris Lattner2876a642008-03-21 21:14:38 +00001434 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001435
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001436 // If we have no predecessors that produce a known value for this load, exit
1437 // early.
1438 if (ValuesPerBlock.empty()) return false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001439
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001440 // If all of the instructions we depend on produce a known value for this
1441 // load, then it is fully redundant and we can use PHI insertion to compute
1442 // its value. Insert PHIs and remove the fully redundant value now.
1443 if (UnavailableBlocks.empty()) {
David Greene2e6efc42010-01-05 01:27:17 +00001444 DEBUG(dbgs() << "GVN REMOVING NONLOCAL LOAD: " << *LI << '\n');
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001445
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001446 // Perform PHI construction.
Chris Lattnerbf200182009-12-21 23:15:48 +00001447 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001448 VN.getAliasAnalysis());
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001449 LI->replaceAllUsesWith(V);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001450
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001451 if (isa<PHINode>(V))
1452 V->takeName(LI);
Duncan Sands19d0b472010-02-16 11:11:14 +00001453 if (V->getType()->isPointerTy())
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001454 MD->invalidateCachedPointerInfo(V);
Bob Wilson1da90412010-02-22 21:39:41 +00001455 VN.erase(LI);
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001456 toErase.push_back(LI);
Dan Gohmand2d1ae12010-06-22 15:08:57 +00001457 ++NumGVNLoad;
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001458 return true;
1459 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001460
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001461 if (!EnablePRE || !EnableLoadPRE)
1462 return false;
1463
1464 // Okay, we have *some* definitions of the value. This means that the value
1465 // is available in some of our (transitive) predecessors. Lets think about
1466 // doing PRE of this load. This will involve inserting a new load into the
1467 // predecessor when it's not available. We could do this in general, but
1468 // prefer to not increase code size. As such, we only do this when we know
1469 // that we only have to insert *one* load (which means we're basically moving
1470 // the load, not inserting a new one).
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001471
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001472 SmallPtrSet<BasicBlock *, 4> Blockers;
1473 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1474 Blockers.insert(UnavailableBlocks[i]);
1475
1476 // Lets find first basic block with more than one predecessor. Walk backwards
1477 // through predecessors if needed.
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001478 BasicBlock *LoadBB = LI->getParent();
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001479 BasicBlock *TmpBB = LoadBB;
1480
1481 bool isSinglePred = false;
Dale Johannesen81b64632009-06-17 20:48:23 +00001482 bool allSingleSucc = true;
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001483 while (TmpBB->getSinglePredecessor()) {
1484 isSinglePred = true;
1485 TmpBB = TmpBB->getSinglePredecessor();
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001486 if (TmpBB == LoadBB) // Infinite (unreachable) loop.
1487 return false;
1488 if (Blockers.count(TmpBB))
1489 return false;
Owen Andersonb590a922010-09-25 05:26:18 +00001490
1491 // If any of these blocks has more than one successor (i.e. if the edge we
1492 // just traversed was critical), then there are other paths through this
1493 // block along which the load may not be anticipated. Hoisting the load
1494 // above this block would be adding the load to execution paths along
1495 // which it was not previously executed.
Dale Johannesen81b64632009-06-17 20:48:23 +00001496 if (TmpBB->getTerminator()->getNumSuccessors() != 1)
Owen Andersonb590a922010-09-25 05:26:18 +00001497 return false;
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001498 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001499
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001500 assert(TmpBB);
1501 LoadBB = TmpBB;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001502
Chris Lattner93236ba2009-12-06 04:54:31 +00001503 // FIXME: It is extremely unclear what this loop is doing, other than
1504 // artificially restricting loadpre.
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001505 if (isSinglePred) {
1506 bool isHot = false;
Chris Lattner93236ba2009-12-06 04:54:31 +00001507 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
1508 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1509 if (AV.isSimpleValue())
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001510 // "Hot" Instruction is in some loop (because it dominates its dep.
1511 // instruction).
Chris Lattner93236ba2009-12-06 04:54:31 +00001512 if (Instruction *I = dyn_cast<Instruction>(AV.getSimpleValue()))
1513 if (DT->dominates(LI, I)) {
1514 isHot = true;
1515 break;
1516 }
1517 }
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001518
1519 // We are interested only in "hot" instructions. We don't want to do any
1520 // mis-optimizations here.
1521 if (!isHot)
1522 return false;
1523 }
1524
Bob Wilsond517b522010-02-01 21:17:14 +00001525 // Check to see how many predecessors have the loaded value fully
1526 // available.
1527 DenseMap<BasicBlock*, Value*> PredLoads;
Chris Lattnerd2a653a2008-12-05 07:49:08 +00001528 DenseMap<BasicBlock*, char> FullyAvailableBlocks;
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001529 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001530 FullyAvailableBlocks[ValuesPerBlock[i].BB] = true;
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001531 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1532 FullyAvailableBlocks[UnavailableBlocks[i]] = false;
1533
Bob Wilsona2fda8b2010-05-04 20:03:21 +00001534 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> NeedToSplit;
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001535 for (pred_iterator PI = pred_begin(LoadBB), E = pred_end(LoadBB);
1536 PI != E; ++PI) {
Bob Wilsond517b522010-02-01 21:17:14 +00001537 BasicBlock *Pred = *PI;
1538 if (IsValueFullyAvailableInBlock(Pred, FullyAvailableBlocks)) {
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001539 continue;
Bob Wilsond517b522010-02-01 21:17:14 +00001540 }
1541 PredLoads[Pred] = 0;
Bob Wilson92cdb6e2010-02-16 19:51:59 +00001542
Bob Wilsond517b522010-02-01 21:17:14 +00001543 if (Pred->getTerminator()->getNumSuccessors() != 1) {
Bob Wilson92cdb6e2010-02-16 19:51:59 +00001544 if (isa<IndirectBrInst>(Pred->getTerminator())) {
1545 DEBUG(dbgs() << "COULD NOT PRE LOAD BECAUSE OF INDBR CRITICAL EDGE '"
1546 << Pred->getName() << "': " << *LI << '\n');
1547 return false;
1548 }
Bob Wilsonaff96b22010-02-16 21:06:42 +00001549 unsigned SuccNum = GetSuccessorNumber(Pred, LoadBB);
Bob Wilsona2fda8b2010-05-04 20:03:21 +00001550 NeedToSplit.push_back(std::make_pair(Pred->getTerminator(), SuccNum));
Bob Wilsond517b522010-02-01 21:17:14 +00001551 }
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001552 }
Bob Wilsona2fda8b2010-05-04 20:03:21 +00001553 if (!NeedToSplit.empty()) {
Bob Wilson0c8b29b2010-05-05 20:44:15 +00001554 toSplit.append(NeedToSplit.begin(), NeedToSplit.end());
Bob Wilson892432b2010-03-01 23:37:32 +00001555 return false;
Bob Wilsona2fda8b2010-05-04 20:03:21 +00001556 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001557
Bob Wilsond517b522010-02-01 21:17:14 +00001558 // Decide whether PRE is profitable for this load.
1559 unsigned NumUnavailablePreds = PredLoads.size();
1560 assert(NumUnavailablePreds != 0 &&
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001561 "Fully available value should be eliminated above!");
Owen Anderson13a642d2010-10-01 20:02:55 +00001562
1563 // If this load is unavailable in multiple predecessors, reject it.
1564 // FIXME: If we could restructure the CFG, we could make a common pred with
1565 // all the preds that don't have an available LI and insert a new load into
1566 // that one block.
1567 if (NumUnavailablePreds != 1)
Bob Wilsond517b522010-02-01 21:17:14 +00001568 return false;
Bob Wilsond517b522010-02-01 21:17:14 +00001569
1570 // Check if the load can safely be moved to all the unavailable predecessors.
1571 bool CanDoPRE = true;
Chris Lattner44da5bd2009-11-28 15:39:14 +00001572 SmallVector<Instruction*, 8> NewInsts;
Bob Wilsond517b522010-02-01 21:17:14 +00001573 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1574 E = PredLoads.end(); I != E; ++I) {
1575 BasicBlock *UnavailablePred = I->first;
1576
1577 // Do PHI translation to get its value in the predecessor if necessary. The
1578 // returned pointer (if non-null) is guaranteed to dominate UnavailablePred.
1579
1580 // If all preds have a single successor, then we know it is safe to insert
1581 // the load on the pred (?!?), so we can insert code to materialize the
1582 // pointer if it is not available.
Dan Gohmand2099112010-11-10 19:03:33 +00001583 PHITransAddr Address(LI->getPointerOperand(), TD);
Bob Wilsond517b522010-02-01 21:17:14 +00001584 Value *LoadPtr = 0;
1585 if (allSingleSucc) {
1586 LoadPtr = Address.PHITranslateWithInsertion(LoadBB, UnavailablePred,
1587 *DT, NewInsts);
1588 } else {
Daniel Dunbar693ea892010-02-24 08:48:04 +00001589 Address.PHITranslateValue(LoadBB, UnavailablePred, DT);
Bob Wilsond517b522010-02-01 21:17:14 +00001590 LoadPtr = Address.getAddr();
Bob Wilsond517b522010-02-01 21:17:14 +00001591 }
1592
1593 // If we couldn't find or insert a computation of this phi translated value,
1594 // we fail PRE.
1595 if (LoadPtr == 0) {
1596 DEBUG(dbgs() << "COULDN'T INSERT PHI TRANSLATED VALUE OF: "
Dan Gohmand2099112010-11-10 19:03:33 +00001597 << *LI->getPointerOperand() << "\n");
Bob Wilsond517b522010-02-01 21:17:14 +00001598 CanDoPRE = false;
1599 break;
1600 }
1601
1602 // Make sure it is valid to move this load here. We have to watch out for:
1603 // @1 = getelementptr (i8* p, ...
1604 // test p and branch if == 0
1605 // load @1
1606 // It is valid to have the getelementptr before the test, even if p can be 0,
1607 // as getelementptr only does address arithmetic.
1608 // If we are not pushing the value through any multiple-successor blocks
1609 // we do not have this case. Otherwise, check that the load is safe to
1610 // put anywhere; this can be improved, but should be conservatively safe.
1611 if (!allSingleSucc &&
1612 // FIXME: REEVALUTE THIS.
1613 !isSafeToLoadUnconditionally(LoadPtr,
1614 UnavailablePred->getTerminator(),
1615 LI->getAlignment(), TD)) {
1616 CanDoPRE = false;
1617 break;
1618 }
1619
1620 I->second = LoadPtr;
Chris Lattner972e6d82009-12-09 01:59:31 +00001621 }
1622
Bob Wilsond517b522010-02-01 21:17:14 +00001623 if (!CanDoPRE) {
1624 while (!NewInsts.empty())
1625 NewInsts.pop_back_val()->eraseFromParent();
Dale Johannesen81b64632009-06-17 20:48:23 +00001626 return false;
Chris Lattner32140312009-11-28 16:08:18 +00001627 }
Dale Johannesen81b64632009-06-17 20:48:23 +00001628
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001629 // Okay, we can eliminate this load by inserting a reload in the predecessor
1630 // and using PHI construction to get the value in the other predecessors, do
1631 // it.
David Greene2e6efc42010-01-05 01:27:17 +00001632 DEBUG(dbgs() << "GVN REMOVING PRE LOAD: " << *LI << '\n');
Chris Lattner32140312009-11-28 16:08:18 +00001633 DEBUG(if (!NewInsts.empty())
David Greene2e6efc42010-01-05 01:27:17 +00001634 dbgs() << "INSERTED " << NewInsts.size() << " INSTS: "
Chris Lattner32140312009-11-28 16:08:18 +00001635 << *NewInsts.back() << '\n');
1636
Bob Wilsond517b522010-02-01 21:17:14 +00001637 // Assign value numbers to the new instructions.
1638 for (unsigned i = 0, e = NewInsts.size(); i != e; ++i) {
1639 // FIXME: We really _ought_ to insert these value numbers into their
1640 // parent's availability map. However, in doing so, we risk getting into
1641 // ordering issues. If a block hasn't been processed yet, we would be
1642 // marking a value as AVAIL-IN, which isn't what we intend.
1643 VN.lookup_or_add(NewInsts[i]);
1644 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001645
Bob Wilsond517b522010-02-01 21:17:14 +00001646 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1647 E = PredLoads.end(); I != E; ++I) {
1648 BasicBlock *UnavailablePred = I->first;
1649 Value *LoadPtr = I->second;
1650
Dan Gohman4467aa52010-12-15 23:53:55 +00001651 Instruction *NewLoad = new LoadInst(LoadPtr, LI->getName()+".pre", false,
1652 LI->getAlignment(),
1653 UnavailablePred->getTerminator());
1654
1655 // Transfer the old load's TBAA tag to the new load.
1656 if (MDNode *Tag = LI->getMetadata(LLVMContext::MD_tbaa))
1657 NewLoad->setMetadata(LLVMContext::MD_tbaa, Tag);
Bob Wilsond517b522010-02-01 21:17:14 +00001658
1659 // Add the newly created load.
1660 ValuesPerBlock.push_back(AvailableValueInBlock::get(UnavailablePred,
1661 NewLoad));
Bob Wilson923261b2010-02-23 05:55:00 +00001662 MD->invalidateCachedPointerInfo(LoadPtr);
1663 DEBUG(dbgs() << "GVN INSERTED " << *NewLoad << '\n');
Bob Wilsond517b522010-02-01 21:17:14 +00001664 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001665
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001666 // Perform PHI construction.
Chris Lattnerbf200182009-12-21 23:15:48 +00001667 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001668 VN.getAliasAnalysis());
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001669 LI->replaceAllUsesWith(V);
1670 if (isa<PHINode>(V))
1671 V->takeName(LI);
Duncan Sands19d0b472010-02-16 11:11:14 +00001672 if (V->getType()->isPointerTy())
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001673 MD->invalidateCachedPointerInfo(V);
Bob Wilson1da90412010-02-22 21:39:41 +00001674 VN.erase(LI);
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001675 toErase.push_back(LI);
Dan Gohmand2d1ae12010-06-22 15:08:57 +00001676 ++NumPRELoad;
Owen Anderson5e5599b2007-07-25 19:57:03 +00001677 return true;
1678}
1679
Owen Anderson221a4362007-08-16 22:02:55 +00001680/// processLoad - Attempt to eliminate a load, first by eliminating it
1681/// locally, and then attempting non-local elimination if that fails.
Chris Lattner0e3d6332008-12-05 21:04:20 +00001682bool GVN::processLoad(LoadInst *L, SmallVectorImpl<Instruction*> &toErase) {
Dan Gohman81132462009-11-14 02:27:51 +00001683 if (!MD)
1684 return false;
1685
Chris Lattner0e3d6332008-12-05 21:04:20 +00001686 if (L->isVolatile())
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001687 return false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001688
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001689 // ... to a pointer that has been loaded from before...
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001690 MemDepResult Dep = MD->getDependency(L);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001691
Chris Lattner0e3d6332008-12-05 21:04:20 +00001692 // If the value isn't available, don't do anything!
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001693 if (Dep.isClobber()) {
Chris Lattner0a9616d2009-09-21 05:57:11 +00001694 // Check to see if we have something like this:
Chris Lattner1dd48c32009-09-20 19:03:47 +00001695 // store i32 123, i32* %P
1696 // %A = bitcast i32* %P to i8*
1697 // %B = gep i8* %A, i32 1
1698 // %C = load i8* %B
1699 //
1700 // We could do that by recognizing if the clobber instructions are obviously
1701 // a common base + constant offset, and if the previous store (or memset)
1702 // completely covers this load. This sort of thing can happen in bitfield
1703 // access code.
Chris Lattner42376062009-12-06 01:57:02 +00001704 Value *AvailVal = 0;
Chris Lattner0a9616d2009-09-21 05:57:11 +00001705 if (StoreInst *DepSI = dyn_cast<StoreInst>(Dep.getInst()))
Duncan Sands246b71c2010-11-12 21:10:24 +00001706 if (TD) {
Chris Lattner07df9ef2009-12-09 07:37:07 +00001707 int Offset = AnalyzeLoadFromClobberingStore(L->getType(),
1708 L->getPointerOperand(),
1709 DepSI, *TD);
Chris Lattner42376062009-12-06 01:57:02 +00001710 if (Offset != -1)
Dan Gohmand2099112010-11-10 19:03:33 +00001711 AvailVal = GetStoreValueForLoad(DepSI->getValueOperand(), Offset,
Chris Lattner42376062009-12-06 01:57:02 +00001712 L->getType(), L, *TD);
Chris Lattner9d7fb292009-09-21 06:22:46 +00001713 }
Chris Lattner0a9616d2009-09-21 05:57:11 +00001714
Chris Lattner42376062009-12-06 01:57:02 +00001715 // If the clobbering value is a memset/memcpy/memmove, see if we can forward
1716 // a value on from it.
1717 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(Dep.getInst())) {
Duncan Sands246b71c2010-11-12 21:10:24 +00001718 if (TD) {
Chris Lattner07df9ef2009-12-09 07:37:07 +00001719 int Offset = AnalyzeLoadFromClobberingMemInst(L->getType(),
1720 L->getPointerOperand(),
1721 DepMI, *TD);
Chris Lattner42376062009-12-06 01:57:02 +00001722 if (Offset != -1)
1723 AvailVal = GetMemInstValueForLoad(DepMI, Offset, L->getType(), L,*TD);
1724 }
1725 }
1726
1727 if (AvailVal) {
David Greene2e6efc42010-01-05 01:27:17 +00001728 DEBUG(dbgs() << "GVN COERCED INST:\n" << *Dep.getInst() << '\n'
Chris Lattner42376062009-12-06 01:57:02 +00001729 << *AvailVal << '\n' << *L << "\n\n\n");
1730
1731 // Replace the load!
1732 L->replaceAllUsesWith(AvailVal);
Duncan Sands19d0b472010-02-16 11:11:14 +00001733 if (AvailVal->getType()->isPointerTy())
Chris Lattner42376062009-12-06 01:57:02 +00001734 MD->invalidateCachedPointerInfo(AvailVal);
Bob Wilson1da90412010-02-22 21:39:41 +00001735 VN.erase(L);
Chris Lattner42376062009-12-06 01:57:02 +00001736 toErase.push_back(L);
Dan Gohmand2d1ae12010-06-22 15:08:57 +00001737 ++NumGVNLoad;
Chris Lattner42376062009-12-06 01:57:02 +00001738 return true;
1739 }
1740
Torok Edwin72070282009-05-29 09:46:03 +00001741 DEBUG(
1742 // fast print dep, using operator<< on instruction would be too slow
David Greene2e6efc42010-01-05 01:27:17 +00001743 dbgs() << "GVN: load ";
1744 WriteAsOperand(dbgs(), L);
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001745 Instruction *I = Dep.getInst();
David Greene2e6efc42010-01-05 01:27:17 +00001746 dbgs() << " is clobbered by " << *I << '\n';
Torok Edwin72070282009-05-29 09:46:03 +00001747 );
Chris Lattner0e3d6332008-12-05 21:04:20 +00001748 return false;
Torok Edwin72070282009-05-29 09:46:03 +00001749 }
Chris Lattner0e3d6332008-12-05 21:04:20 +00001750
1751 // If it is defined in another block, try harder.
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001752 if (Dep.isNonLocal())
Chris Lattner0e3d6332008-12-05 21:04:20 +00001753 return processNonLocalLoad(L, toErase);
Eli Friedman716c10c2008-02-12 12:08:14 +00001754
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001755 Instruction *DepInst = Dep.getInst();
Chris Lattner0e3d6332008-12-05 21:04:20 +00001756 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInst)) {
Dan Gohmand2099112010-11-10 19:03:33 +00001757 Value *StoredVal = DepSI->getValueOperand();
Chris Lattner1dd48c32009-09-20 19:03:47 +00001758
1759 // The store and load are to a must-aliased pointer, but they may not
1760 // actually have the same type. See if we know how to reuse the stored
1761 // value (depending on its type).
Chris Lattner8ed7bef2009-10-21 04:11:19 +00001762 if (StoredVal->getType() != L->getType()) {
Duncan Sands246b71c2010-11-12 21:10:24 +00001763 if (TD) {
Chris Lattner8ed7bef2009-10-21 04:11:19 +00001764 StoredVal = CoerceAvailableValueToLoadType(StoredVal, L->getType(),
1765 L, *TD);
1766 if (StoredVal == 0)
1767 return false;
1768
David Greene2e6efc42010-01-05 01:27:17 +00001769 DEBUG(dbgs() << "GVN COERCED STORE:\n" << *DepSI << '\n' << *StoredVal
Chris Lattner8ed7bef2009-10-21 04:11:19 +00001770 << '\n' << *L << "\n\n\n");
1771 }
1772 else
Chris Lattner1dd48c32009-09-20 19:03:47 +00001773 return false;
Chris Lattner1dd48c32009-09-20 19:03:47 +00001774 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001775
Chris Lattner0e3d6332008-12-05 21:04:20 +00001776 // Remove it!
Chris Lattner1dd48c32009-09-20 19:03:47 +00001777 L->replaceAllUsesWith(StoredVal);
Duncan Sands19d0b472010-02-16 11:11:14 +00001778 if (StoredVal->getType()->isPointerTy())
Chris Lattner1dd48c32009-09-20 19:03:47 +00001779 MD->invalidateCachedPointerInfo(StoredVal);
Bob Wilson1da90412010-02-22 21:39:41 +00001780 VN.erase(L);
Chris Lattner0e3d6332008-12-05 21:04:20 +00001781 toErase.push_back(L);
Dan Gohmand2d1ae12010-06-22 15:08:57 +00001782 ++NumGVNLoad;
Chris Lattner0e3d6332008-12-05 21:04:20 +00001783 return true;
1784 }
1785
1786 if (LoadInst *DepLI = dyn_cast<LoadInst>(DepInst)) {
Chris Lattner1dd48c32009-09-20 19:03:47 +00001787 Value *AvailableVal = DepLI;
1788
1789 // The loads are of a must-aliased pointer, but they may not actually have
1790 // the same type. See if we know how to reuse the previously loaded value
1791 // (depending on its type).
Chris Lattner8ed7bef2009-10-21 04:11:19 +00001792 if (DepLI->getType() != L->getType()) {
Duncan Sands246b71c2010-11-12 21:10:24 +00001793 if (TD) {
Chris Lattner8ed7bef2009-10-21 04:11:19 +00001794 AvailableVal = CoerceAvailableValueToLoadType(DepLI, L->getType(), L,*TD);
1795 if (AvailableVal == 0)
1796 return false;
Chris Lattner1dd48c32009-09-20 19:03:47 +00001797
David Greene2e6efc42010-01-05 01:27:17 +00001798 DEBUG(dbgs() << "GVN COERCED LOAD:\n" << *DepLI << "\n" << *AvailableVal
Chris Lattner8ed7bef2009-10-21 04:11:19 +00001799 << "\n" << *L << "\n\n\n");
1800 }
1801 else
1802 return false;
Chris Lattner1dd48c32009-09-20 19:03:47 +00001803 }
1804
Chris Lattner0e3d6332008-12-05 21:04:20 +00001805 // Remove it!
Chris Lattner1dd48c32009-09-20 19:03:47 +00001806 L->replaceAllUsesWith(AvailableVal);
Duncan Sands19d0b472010-02-16 11:11:14 +00001807 if (DepLI->getType()->isPointerTy())
Chris Lattnerfa9f99a2008-12-09 22:06:23 +00001808 MD->invalidateCachedPointerInfo(DepLI);
Bob Wilson1da90412010-02-22 21:39:41 +00001809 VN.erase(L);
Chris Lattner0e3d6332008-12-05 21:04:20 +00001810 toErase.push_back(L);
Dan Gohmand2d1ae12010-06-22 15:08:57 +00001811 ++NumGVNLoad;
Chris Lattner0e3d6332008-12-05 21:04:20 +00001812 return true;
1813 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001814
Chris Lattner3ff6d012008-11-30 01:39:32 +00001815 // If this load really doesn't depend on anything, then we must be loading an
1816 // undef value. This can happen when loading for a fresh allocation with no
1817 // intervening stores, for example.
Victor Hernandez8acf2952009-10-23 21:09:37 +00001818 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst)) {
Owen Andersonb292b8c2009-07-30 23:03:37 +00001819 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson1da90412010-02-22 21:39:41 +00001820 VN.erase(L);
Chris Lattner3ff6d012008-11-30 01:39:32 +00001821 toErase.push_back(L);
Dan Gohmand2d1ae12010-06-22 15:08:57 +00001822 ++NumGVNLoad;
Chris Lattner0e3d6332008-12-05 21:04:20 +00001823 return true;
Eli Friedman716c10c2008-02-12 12:08:14 +00001824 }
Owen Anderson2b2bd282009-10-28 07:05:35 +00001825
Owen Andersonb9878ee2009-12-02 07:35:19 +00001826 // If this load occurs either right after a lifetime begin,
Owen Anderson2b2bd282009-10-28 07:05:35 +00001827 // then the loaded value is undefined.
1828 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(DepInst)) {
Owen Andersonb9878ee2009-12-02 07:35:19 +00001829 if (II->getIntrinsicID() == Intrinsic::lifetime_start) {
Owen Anderson2b2bd282009-10-28 07:05:35 +00001830 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson1da90412010-02-22 21:39:41 +00001831 VN.erase(L);
Owen Anderson2b2bd282009-10-28 07:05:35 +00001832 toErase.push_back(L);
Dan Gohmand2d1ae12010-06-22 15:08:57 +00001833 ++NumGVNLoad;
Owen Anderson2b2bd282009-10-28 07:05:35 +00001834 return true;
1835 }
1836 }
Eli Friedman716c10c2008-02-12 12:08:14 +00001837
Chris Lattner0e3d6332008-12-05 21:04:20 +00001838 return false;
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001839}
1840
Owen Andersonea326db2010-11-19 22:48:40 +00001841// lookupNumber - In order to find a leader for a given value number at a
1842// specific basic block, we first obtain the list of all Values for that number,
1843// and then scan the list to find one whose block dominates the block in
1844// question. This is fast because dominator tree queries consist of only
1845// a few comparisons of DFS numbers.
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001846Value *GVN::lookupNumber(BasicBlock *BB, uint32_t num) {
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00001847 NumberTableEntry Vals = NumberTable[num];
1848 if (!Vals.Val) return 0;
Owen Andersonc21c1002010-11-18 18:32:40 +00001849
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00001850 Value *Val = 0;
1851 if (DT->dominates(Vals.BB, BB)) {
1852 Val = Vals.Val;
1853 if (isa<Constant>(Val)) return Val;
1854 }
1855
1856 NumberTableEntry* Next = Vals.Next;
Owen Andersonc21c1002010-11-18 18:32:40 +00001857 while (Next) {
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00001858 if (DT->dominates(Next->BB, BB)) {
1859 if (isa<Constant>(Next->Val)) return Next->Val;
1860 if (!Val) Val = Next->Val;
1861 }
Owen Andersonc21c1002010-11-18 18:32:40 +00001862
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00001863 Next = Next->Next;
Owen Anderson1b3ea962008-06-20 01:15:47 +00001864 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001865
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00001866 return Val;
Owen Anderson1b3ea962008-06-20 01:15:47 +00001867}
1868
Owen Andersonbfe133e2008-12-15 02:03:00 +00001869
Owen Anderson398602a2007-08-14 18:16:29 +00001870/// processInstruction - When calculating availability, handle an instruction
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001871/// by inserting it into the appropriate sets
Owen Andersonaccdca12008-06-12 19:25:32 +00001872bool GVN::processInstruction(Instruction *I,
Chris Lattner804209d2008-03-21 22:01:16 +00001873 SmallVectorImpl<Instruction*> &toErase) {
Devang Patel03936a12010-02-11 00:20:49 +00001874 // Ignore dbg info intrinsics.
1875 if (isa<DbgInfoIntrinsic>(I))
1876 return false;
1877
Duncan Sands246b71c2010-11-12 21:10:24 +00001878 // If the instruction can be easily simplified then do so now in preference
1879 // to value numbering it. Value numbering often exposes redundancies, for
1880 // example if it determines that %y is equal to %x then the instruction
1881 // "%z = and i32 %x, %y" becomes "%z = and i32 %x, %x" which we now simplify.
Duncan Sandsb99f39b2010-11-14 18:36:10 +00001882 if (Value *V = SimplifyInstruction(I, TD, DT)) {
Duncan Sands246b71c2010-11-12 21:10:24 +00001883 I->replaceAllUsesWith(V);
1884 if (MD && V->getType()->isPointerTy())
1885 MD->invalidateCachedPointerInfo(V);
1886 VN.erase(I);
1887 toErase.push_back(I);
1888 return true;
1889 }
1890
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001891 if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
1892 bool Changed = processLoad(LI, toErase);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001893
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001894 if (!Changed) {
1895 unsigned Num = VN.lookup_or_add(LI);
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00001896 insert_table(Num, LI, LI->getParent());
Owen Anderson6a903bc2008-06-18 21:41:49 +00001897 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001898
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001899 return Changed;
Owen Anderson6a903bc2008-06-18 21:41:49 +00001900 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001901
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001902 uint32_t NextNum = VN.getNextUnusedValueNumber();
1903 unsigned Num = VN.lookup_or_add(I);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001904
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00001905 // For conditions branches, we can perform simple conditional propagation on
1906 // the condition value itself.
1907 if (BranchInst *BI = dyn_cast<BranchInst>(I)) {
1908 insert_table(Num, I, I->getParent());
1909
1910 if (!BI->isConditional() || isa<Constant>(BI->getCondition()))
1911 return false;
1912
1913 Value *BranchCond = BI->getCondition();
1914 uint32_t CondVN = VN.lookup_or_add(BranchCond);
1915
1916 BasicBlock *TrueSucc = BI->getSuccessor(0);
1917 BasicBlock *FalseSucc = BI->getSuccessor(1);
1918
1919 if (TrueSucc->getSinglePredecessor())
1920 insert_table(CondVN,
1921 ConstantInt::getTrue(TrueSucc->getContext()),
1922 TrueSucc);
1923 if (FalseSucc->getSinglePredecessor())
1924 insert_table(CondVN,
1925 ConstantInt::getFalse(TrueSucc->getContext()),
1926 FalseSucc);
1927
1928 return false;
1929 }
1930
Owen Anderson0c1e6342008-04-07 09:59:07 +00001931 // Allocations are always uniquely numbered, so we can save time and memory
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001932 // by fast failing them.
Chris Lattnerb6252a32010-12-19 20:24:28 +00001933 if (isa<AllocaInst>(I) || isa<TerminatorInst>(I) || isa<PHINode>(I)) {
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00001934 insert_table(Num, I, I->getParent());
Owen Anderson0c1e6342008-04-07 09:59:07 +00001935 return false;
Owen Anderson6a903bc2008-06-18 21:41:49 +00001936 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001937
Owen Anderson3ea90a72008-07-03 17:44:33 +00001938 // If the number we were assigned was a brand new VN, then we don't
1939 // need to do a lookup to see if the number already exists
1940 // somewhere in the domtree: it can't!
Chris Lattnerb6252a32010-12-19 20:24:28 +00001941 if (Num == NextNum) {
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00001942 insert_table(Num, I, I->getParent());
Chris Lattnerb6252a32010-12-19 20:24:28 +00001943 return false;
1944 }
1945
Owen Andersonbfe133e2008-12-15 02:03:00 +00001946 // Perform fast-path value-number based elimination of values inherited from
1947 // dominators.
Chris Lattnerb6252a32010-12-19 20:24:28 +00001948 Value *repl = lookupNumber(I->getParent(), Num);
1949 if (repl == 0) {
1950 // Failure, just remember this instance for future use.
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00001951 insert_table(Num, I, I->getParent());
Chris Lattnerb6252a32010-12-19 20:24:28 +00001952 return false;
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001953 }
Chris Lattnerb6252a32010-12-19 20:24:28 +00001954
1955 // Remove it!
1956 VN.erase(I);
1957 I->replaceAllUsesWith(repl);
1958 if (MD && repl->getType()->isPointerTy())
1959 MD->invalidateCachedPointerInfo(repl);
1960 toErase.push_back(I);
1961 return true;
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001962}
1963
Bill Wendling456e8852008-12-22 22:32:22 +00001964/// runOnFunction - This is the main transformation entry point for a function.
Owen Anderson676070d2007-08-14 18:04:11 +00001965bool GVN::runOnFunction(Function& F) {
Dan Gohman81132462009-11-14 02:27:51 +00001966 if (!NoLoads)
1967 MD = &getAnalysis<MemoryDependenceAnalysis>();
Chris Lattner8541ede2008-12-01 00:40:32 +00001968 DT = &getAnalysis<DominatorTree>();
Duncan Sands246b71c2010-11-12 21:10:24 +00001969 TD = getAnalysisIfAvailable<TargetData>();
Owen Andersonf7928602008-05-12 20:15:55 +00001970 VN.setAliasAnalysis(&getAnalysis<AliasAnalysis>());
Chris Lattner8541ede2008-12-01 00:40:32 +00001971 VN.setMemDep(MD);
1972 VN.setDomTree(DT);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001973
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001974 bool Changed = false;
1975 bool ShouldContinue = true;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001976
Owen Andersonac310962008-07-16 17:52:31 +00001977 // Merge unconditional branches, allowing PRE to catch more
1978 // optimization opportunities.
1979 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001980 BasicBlock *BB = FI;
Owen Andersonac310962008-07-16 17:52:31 +00001981 ++FI;
Owen Andersonc0623812008-07-17 00:01:40 +00001982 bool removedBlock = MergeBlockIntoPredecessor(BB, this);
Dan Gohmand2d1ae12010-06-22 15:08:57 +00001983 if (removedBlock) ++NumGVNBlocks;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001984
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001985 Changed |= removedBlock;
Owen Andersonac310962008-07-16 17:52:31 +00001986 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001987
Chris Lattner0a5a8d52008-12-09 19:21:47 +00001988 unsigned Iteration = 0;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001989
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001990 while (ShouldContinue) {
David Greene2e6efc42010-01-05 01:27:17 +00001991 DEBUG(dbgs() << "GVN iteration: " << Iteration << "\n");
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001992 ShouldContinue = iterateOnFunction(F);
Bob Wilson92cdb6e2010-02-16 19:51:59 +00001993 if (splitCriticalEdges())
1994 ShouldContinue = true;
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001995 Changed |= ShouldContinue;
Chris Lattner0a5a8d52008-12-09 19:21:47 +00001996 ++Iteration;
Owen Anderson676070d2007-08-14 18:04:11 +00001997 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001998
Owen Anderson04a6e0b2008-07-18 18:03:38 +00001999 if (EnablePRE) {
Owen Anderson2fbfb702008-09-03 23:06:07 +00002000 bool PREChanged = true;
2001 while (PREChanged) {
2002 PREChanged = performPRE(F);
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002003 Changed |= PREChanged;
Owen Anderson2fbfb702008-09-03 23:06:07 +00002004 }
Owen Anderson04a6e0b2008-07-18 18:03:38 +00002005 }
Chris Lattner0a5a8d52008-12-09 19:21:47 +00002006 // FIXME: Should perform GVN again after PRE does something. PRE can move
2007 // computations into blocks where they become fully redundant. Note that
2008 // we can't do this until PRE's critical edge splitting updates memdep.
2009 // Actually, when this happens, we should just fully integrate PRE into GVN.
Nuno Lopese3127f32008-10-10 16:25:50 +00002010
2011 cleanupGlobalSets();
2012
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002013 return Changed;
Owen Anderson676070d2007-08-14 18:04:11 +00002014}
2015
2016
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002017bool GVN::processBlock(BasicBlock *BB) {
Chris Lattner0a5a8d52008-12-09 19:21:47 +00002018 // FIXME: Kill off toErase by doing erasing eagerly in a helper function (and
2019 // incrementing BI before processing an instruction).
Owen Andersonaccdca12008-06-12 19:25:32 +00002020 SmallVector<Instruction*, 8> toErase;
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002021 bool ChangedFunction = false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002022
Owen Andersonaccdca12008-06-12 19:25:32 +00002023 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
2024 BI != BE;) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002025 ChangedFunction |= processInstruction(BI, toErase);
Owen Andersonaccdca12008-06-12 19:25:32 +00002026 if (toErase.empty()) {
2027 ++BI;
2028 continue;
2029 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002030
Owen Andersonaccdca12008-06-12 19:25:32 +00002031 // If we need some instructions deleted, do it now.
2032 NumGVNInstr += toErase.size();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002033
Owen Andersonaccdca12008-06-12 19:25:32 +00002034 // Avoid iterator invalidation.
2035 bool AtStart = BI == BB->begin();
2036 if (!AtStart)
2037 --BI;
2038
2039 for (SmallVector<Instruction*, 4>::iterator I = toErase.begin(),
Chris Lattner8541ede2008-12-01 00:40:32 +00002040 E = toErase.end(); I != E; ++I) {
David Greene2e6efc42010-01-05 01:27:17 +00002041 DEBUG(dbgs() << "GVN removed: " << **I << '\n');
Dan Gohman81132462009-11-14 02:27:51 +00002042 if (MD) MD->removeInstruction(*I);
Owen Andersonaccdca12008-06-12 19:25:32 +00002043 (*I)->eraseFromParent();
Bill Wendlingebb6a542008-12-22 21:57:30 +00002044 DEBUG(verifyRemoved(*I));
Chris Lattner8541ede2008-12-01 00:40:32 +00002045 }
Chris Lattner0a5a8d52008-12-09 19:21:47 +00002046 toErase.clear();
Owen Andersonaccdca12008-06-12 19:25:32 +00002047
2048 if (AtStart)
2049 BI = BB->begin();
2050 else
2051 ++BI;
Owen Andersonaccdca12008-06-12 19:25:32 +00002052 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002053
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002054 return ChangedFunction;
Owen Andersonaccdca12008-06-12 19:25:32 +00002055}
2056
Owen Anderson6a903bc2008-06-18 21:41:49 +00002057/// performPRE - Perform a purely local form of PRE that looks for diamond
2058/// control flow patterns and attempts to perform simple PRE at the join point.
Chris Lattnera546dcf2009-10-31 22:11:15 +00002059bool GVN::performPRE(Function &F) {
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002060 bool Changed = false;
Chris Lattnerf00aae42008-12-01 07:29:03 +00002061 DenseMap<BasicBlock*, Value*> predMap;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002062 for (df_iterator<BasicBlock*> DI = df_begin(&F.getEntryBlock()),
2063 DE = df_end(&F.getEntryBlock()); DI != DE; ++DI) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002064 BasicBlock *CurrentBlock = *DI;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002065
Owen Anderson6a903bc2008-06-18 21:41:49 +00002066 // Nothing to PRE in the entry block.
2067 if (CurrentBlock == &F.getEntryBlock()) continue;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002068
Owen Anderson6a903bc2008-06-18 21:41:49 +00002069 for (BasicBlock::iterator BI = CurrentBlock->begin(),
2070 BE = CurrentBlock->end(); BI != BE; ) {
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002071 Instruction *CurInst = BI++;
Duncan Sands1efabaa2009-05-06 06:49:50 +00002072
Victor Hernandez8acf2952009-10-23 21:09:37 +00002073 if (isa<AllocaInst>(CurInst) ||
Victor Hernandez5d034492009-09-18 22:35:49 +00002074 isa<TerminatorInst>(CurInst) || isa<PHINode>(CurInst) ||
Devang Patel92f86192009-10-14 17:29:00 +00002075 CurInst->getType()->isVoidTy() ||
Duncan Sands1efabaa2009-05-06 06:49:50 +00002076 CurInst->mayReadFromMemory() || CurInst->mayHaveSideEffects() ||
John Criswell073e4d12009-03-10 15:04:53 +00002077 isa<DbgInfoIntrinsic>(CurInst))
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002078 continue;
Owen Anderson03986072010-08-07 00:20:35 +00002079
2080 // We don't currently value number ANY inline asm calls.
2081 if (CallInst *CallI = dyn_cast<CallInst>(CurInst))
2082 if (CallI->isInlineAsm())
2083 continue;
Duncan Sands1efabaa2009-05-06 06:49:50 +00002084
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002085 uint32_t ValNo = VN.lookup(CurInst);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002086
Owen Anderson6a903bc2008-06-18 21:41:49 +00002087 // Look for the predecessors for PRE opportunities. We're
2088 // only trying to solve the basic diamond case, where
2089 // a value is computed in the successor and one predecessor,
2090 // but not the other. We also explicitly disallow cases
2091 // where the successor is its own predecessor, because they're
2092 // more complicated to get right.
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002093 unsigned NumWith = 0;
2094 unsigned NumWithout = 0;
2095 BasicBlock *PREPred = 0;
Chris Lattnerf00aae42008-12-01 07:29:03 +00002096 predMap.clear();
2097
Owen Anderson6a903bc2008-06-18 21:41:49 +00002098 for (pred_iterator PI = pred_begin(CurrentBlock),
2099 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
Gabor Greifb0d56ff2010-07-09 14:36:49 +00002100 BasicBlock *P = *PI;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002101 // We're not interested in PRE where the block is its
Bob Wilson76e8c592010-02-03 00:33:21 +00002102 // own predecessor, or in blocks with predecessors
Owen Anderson1b3ea962008-06-20 01:15:47 +00002103 // that are not reachable.
Gabor Greifb0d56ff2010-07-09 14:36:49 +00002104 if (P == CurrentBlock) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002105 NumWithout = 2;
Owen Anderson1b3ea962008-06-20 01:15:47 +00002106 break;
Owen Andersonc21c1002010-11-18 18:32:40 +00002107 } else if (!DT->dominates(&F.getEntryBlock(), P)) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002108 NumWithout = 2;
Owen Anderson1b3ea962008-06-20 01:15:47 +00002109 break;
2110 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002111
Owen Andersonc21c1002010-11-18 18:32:40 +00002112 Value* predV = lookupNumber(P, ValNo);
2113 if (predV == 0) {
Gabor Greifb0d56ff2010-07-09 14:36:49 +00002114 PREPred = P;
Dan Gohmand2d1ae12010-06-22 15:08:57 +00002115 ++NumWithout;
Owen Andersonc21c1002010-11-18 18:32:40 +00002116 } else if (predV == CurInst) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002117 NumWithout = 2;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002118 } else {
Owen Andersonc21c1002010-11-18 18:32:40 +00002119 predMap[P] = predV;
Dan Gohmand2d1ae12010-06-22 15:08:57 +00002120 ++NumWith;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002121 }
2122 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002123
Owen Anderson6a903bc2008-06-18 21:41:49 +00002124 // Don't do PRE when it might increase code size, i.e. when
2125 // we would need to insert instructions in more than one pred.
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002126 if (NumWithout != 1 || NumWith == 0)
Owen Anderson6a903bc2008-06-18 21:41:49 +00002127 continue;
Chris Lattnera546dcf2009-10-31 22:11:15 +00002128
2129 // Don't do PRE across indirect branch.
2130 if (isa<IndirectBrInst>(PREPred->getTerminator()))
2131 continue;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002132
Owen Andersonfdf9f162008-06-19 19:54:19 +00002133 // We can't do PRE safely on a critical edge, so instead we schedule
2134 // the edge to be split and perform the PRE the next time we iterate
2135 // on the function.
Bob Wilsonaff96b22010-02-16 21:06:42 +00002136 unsigned SuccNum = GetSuccessorNumber(PREPred, CurrentBlock);
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002137 if (isCriticalEdge(PREPred->getTerminator(), SuccNum)) {
2138 toSplit.push_back(std::make_pair(PREPred->getTerminator(), SuccNum));
Owen Andersonfdf9f162008-06-19 19:54:19 +00002139 continue;
2140 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002141
Bob Wilson76e8c592010-02-03 00:33:21 +00002142 // Instantiate the expression in the predecessor that lacked it.
Owen Anderson6a903bc2008-06-18 21:41:49 +00002143 // Because we are going top-down through the block, all value numbers
2144 // will be available in the predecessor by the time we need them. Any
Bob Wilson76e8c592010-02-03 00:33:21 +00002145 // that weren't originally present will have been instantiated earlier
Owen Anderson6a903bc2008-06-18 21:41:49 +00002146 // in this loop.
Nick Lewycky42fb7452009-09-27 07:38:41 +00002147 Instruction *PREInstr = CurInst->clone();
Owen Anderson6a903bc2008-06-18 21:41:49 +00002148 bool success = true;
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002149 for (unsigned i = 0, e = CurInst->getNumOperands(); i != e; ++i) {
2150 Value *Op = PREInstr->getOperand(i);
2151 if (isa<Argument>(Op) || isa<Constant>(Op) || isa<GlobalValue>(Op))
2152 continue;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002153
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002154 if (Value *V = lookupNumber(PREPred, VN.lookup(Op))) {
2155 PREInstr->setOperand(i, V);
2156 } else {
2157 success = false;
2158 break;
Owen Anderson8e462e92008-07-11 20:05:13 +00002159 }
Owen Anderson6a903bc2008-06-18 21:41:49 +00002160 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002161
Owen Anderson6a903bc2008-06-18 21:41:49 +00002162 // Fail out if we encounter an operand that is not available in
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002163 // the PRE predecessor. This is typically because of loads which
Owen Anderson6a903bc2008-06-18 21:41:49 +00002164 // are not value numbered precisely.
2165 if (!success) {
2166 delete PREInstr;
Bill Wendling3c793442008-12-22 22:14:07 +00002167 DEBUG(verifyRemoved(PREInstr));
Owen Anderson6a903bc2008-06-18 21:41:49 +00002168 continue;
2169 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002170
Owen Anderson6a903bc2008-06-18 21:41:49 +00002171 PREInstr->insertBefore(PREPred->getTerminator());
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002172 PREInstr->setName(CurInst->getName() + ".pre");
Owen Anderson1b3ea962008-06-20 01:15:47 +00002173 predMap[PREPred] = PREInstr;
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002174 VN.add(PREInstr, ValNo);
Dan Gohmand2d1ae12010-06-22 15:08:57 +00002175 ++NumGVNPRE;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002176
Owen Anderson6a903bc2008-06-18 21:41:49 +00002177 // Update the availability map to include the new instruction.
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00002178 insert_table(ValNo, PREInstr, PREPred);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002179
Owen Anderson6a903bc2008-06-18 21:41:49 +00002180 // Create a PHI to make the value available in this block.
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002181 PHINode* Phi = PHINode::Create(CurInst->getType(),
2182 CurInst->getName() + ".pre-phi",
Owen Anderson6a903bc2008-06-18 21:41:49 +00002183 CurrentBlock->begin());
2184 for (pred_iterator PI = pred_begin(CurrentBlock),
Gabor Greifd323f5e2010-07-09 14:48:08 +00002185 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
2186 BasicBlock *P = *PI;
2187 Phi->addIncoming(predMap[P], P);
2188 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002189
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002190 VN.add(Phi, ValNo);
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00002191 insert_table(ValNo, Phi, CurrentBlock);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002192
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002193 CurInst->replaceAllUsesWith(Phi);
Duncan Sands19d0b472010-02-16 11:11:14 +00002194 if (MD && Phi->getType()->isPointerTy())
Chris Lattnerfa9f99a2008-12-09 22:06:23 +00002195 MD->invalidateCachedPointerInfo(Phi);
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002196 VN.erase(CurInst);
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00002197 erase_table(ValNo, CurInst, CurrentBlock);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002198
David Greene2e6efc42010-01-05 01:27:17 +00002199 DEBUG(dbgs() << "GVN PRE removed: " << *CurInst << '\n');
Dan Gohman81132462009-11-14 02:27:51 +00002200 if (MD) MD->removeInstruction(CurInst);
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002201 CurInst->eraseFromParent();
Bill Wendlingebb6a542008-12-22 21:57:30 +00002202 DEBUG(verifyRemoved(CurInst));
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002203 Changed = true;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002204 }
2205 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002206
Bob Wilson92cdb6e2010-02-16 19:51:59 +00002207 if (splitCriticalEdges())
2208 Changed = true;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002209
Bob Wilson92cdb6e2010-02-16 19:51:59 +00002210 return Changed;
2211}
2212
2213/// splitCriticalEdges - Split critical edges found during the previous
2214/// iteration that may enable further optimization.
2215bool GVN::splitCriticalEdges() {
2216 if (toSplit.empty())
2217 return false;
2218 do {
2219 std::pair<TerminatorInst*, unsigned> Edge = toSplit.pop_back_val();
2220 SplitCriticalEdge(Edge.first, Edge.second, this);
2221 } while (!toSplit.empty());
Evan Cheng7263cf8432010-03-01 22:23:12 +00002222 if (MD) MD->invalidateCachedPredecessors();
Bob Wilson92cdb6e2010-02-16 19:51:59 +00002223 return true;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002224}
2225
Bill Wendling456e8852008-12-22 22:32:22 +00002226/// iterateOnFunction - Executes one iteration of GVN
Owen Anderson676070d2007-08-14 18:04:11 +00002227bool GVN::iterateOnFunction(Function &F) {
Nuno Lopese3127f32008-10-10 16:25:50 +00002228 cleanupGlobalSets();
Owen Andersonc21c1002010-11-18 18:32:40 +00002229
Owen Andersonab6ec2e2007-07-24 17:55:58 +00002230 // Top-down walk of the dominator tree
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002231 bool Changed = false;
Owen Anderson03aacba2008-12-15 03:52:17 +00002232#if 0
2233 // Needed for value numbering with phi construction to work.
Owen Andersonbfe133e2008-12-15 02:03:00 +00002234 ReversePostOrderTraversal<Function*> RPOT(&F);
2235 for (ReversePostOrderTraversal<Function*>::rpo_iterator RI = RPOT.begin(),
2236 RE = RPOT.end(); RI != RE; ++RI)
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002237 Changed |= processBlock(*RI);
Owen Anderson03aacba2008-12-15 03:52:17 +00002238#else
2239 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2240 DE = df_end(DT->getRootNode()); DI != DE; ++DI)
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002241 Changed |= processBlock(DI->getBlock());
Owen Anderson03aacba2008-12-15 03:52:17 +00002242#endif
2243
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002244 return Changed;
Owen Andersonab6ec2e2007-07-24 17:55:58 +00002245}
Nuno Lopese3127f32008-10-10 16:25:50 +00002246
2247void GVN::cleanupGlobalSets() {
2248 VN.clear();
Owen Andersonc21c1002010-11-18 18:32:40 +00002249 NumberTable.clear();
2250 TableAllocator.Reset();
Nuno Lopese3127f32008-10-10 16:25:50 +00002251}
Bill Wendling6b18a392008-12-22 21:36:08 +00002252
2253/// verifyRemoved - Verify that the specified instruction does not occur in our
2254/// internal data structures.
Bill Wendlinge7f08e72008-12-22 22:28:56 +00002255void GVN::verifyRemoved(const Instruction *Inst) const {
2256 VN.verifyRemoved(Inst);
Bill Wendling3c793442008-12-22 22:14:07 +00002257
Bill Wendlinge7f08e72008-12-22 22:28:56 +00002258 // Walk through the value number scope to make sure the instruction isn't
2259 // ferreted away in it.
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00002260 for (DenseMap<uint32_t, NumberTableEntry>::const_iterator
Owen Andersonc21c1002010-11-18 18:32:40 +00002261 I = NumberTable.begin(), E = NumberTable.end(); I != E; ++I) {
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00002262 const NumberTableEntry *Node = &I->second;
2263 assert(Node->Val != Inst && "Inst still in value numbering scope!");
Owen Andersonc21c1002010-11-18 18:32:40 +00002264
Owen Anderson5ab8d4b2010-12-21 23:54:34 +00002265 while (Node->Next) {
2266 Node = Node->Next;
2267 assert(Node->Val != Inst && "Inst still in value numbering scope!");
Bill Wendling3c793442008-12-22 22:14:07 +00002268 }
2269 }
Bill Wendling6b18a392008-12-22 21:36:08 +00002270}