blob: 164730c3ca06d0f61bef011cc8ea6ade7e28d92c [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 Anderson5e5599b2007-07-25 19:57:03 +000020#include "llvm/BasicBlock.h"
Owen Andersondbf23cc2007-07-26 18:26:51 +000021#include "llvm/Constants.h"
Owen Andersonab6ec2e2007-07-24 17:55:58 +000022#include "llvm/DerivedTypes.h"
Chris Lattner17079fc2009-12-28 21:28:46 +000023#include "llvm/GlobalVariable.h"
Owen Andersondbf23cc2007-07-26 18:26:51 +000024#include "llvm/Function.h"
Devang Patele8c6d312009-03-06 02:59:27 +000025#include "llvm/IntrinsicInst.h"
Owen Andersonb5618da2009-07-03 00:17:18 +000026#include "llvm/LLVMContext.h"
Chris Lattner0a9616d2009-09-21 05:57:11 +000027#include "llvm/Operator.h"
Owen Andersondbf23cc2007-07-26 18:26:51 +000028#include "llvm/Value.h"
Owen Andersonab6ec2e2007-07-24 17:55:58 +000029#include "llvm/ADT/DenseMap.h"
30#include "llvm/ADT/DepthFirstIterator.h"
Owen Andersonbfe133e2008-12-15 02:03:00 +000031#include "llvm/ADT/PostOrderIterator.h"
Owen Andersonab6ec2e2007-07-24 17:55:58 +000032#include "llvm/ADT/SmallPtrSet.h"
33#include "llvm/ADT/SmallVector.h"
34#include "llvm/ADT/Statistic.h"
Owen Anderson09b83ba2007-10-18 19:39:33 +000035#include "llvm/Analysis/AliasAnalysis.h"
Chris Lattner778cb922009-12-06 05:29:56 +000036#include "llvm/Analysis/ConstantFolding.h"
37#include "llvm/Analysis/Dominators.h"
Victor Hernandezf390e042009-10-27 20:05:49 +000038#include "llvm/Analysis/MemoryBuiltins.h"
Owen Andersonab6ec2e2007-07-24 17:55:58 +000039#include "llvm/Analysis/MemoryDependenceAnalysis.h"
Chris Lattner972e6d82009-12-09 01:59:31 +000040#include "llvm/Analysis/PHITransAddr.h"
Owen Andersonab6ec2e2007-07-24 17:55:58 +000041#include "llvm/Support/CFG.h"
Owen Andersone780d662008-06-19 19:57:25 +000042#include "llvm/Support/CommandLine.h"
Chris Lattnerd528b212008-03-29 04:36:18 +000043#include "llvm/Support/Debug.h"
Torok Edwin56d06592009-07-11 20:10:48 +000044#include "llvm/Support/ErrorHandling.h"
Chris Lattner0a9616d2009-09-21 05:57:11 +000045#include "llvm/Support/GetElementPtrTypeIterator.h"
Chris Lattner42376062009-12-06 01:57:02 +000046#include "llvm/Support/IRBuilder.h"
Daniel Dunbar0dd5e1e2009-07-25 00:23:56 +000047#include "llvm/Support/raw_ostream.h"
Chris Lattner1dd48c32009-09-20 19:03:47 +000048#include "llvm/Target/TargetData.h"
Owen Andersonfdf9f162008-06-19 19:54:19 +000049#include "llvm/Transforms/Utils/BasicBlockUtils.h"
Dale Johannesen81b64632009-06-17 20:48:23 +000050#include "llvm/Transforms/Utils/Local.h"
Chris Lattnerb6c65fa2009-10-10 23:50:30 +000051#include "llvm/Transforms/Utils/SSAUpdater.h"
Owen Andersonab6ec2e2007-07-24 17:55:58 +000052using namespace llvm;
53
Bill Wendling3c793442008-12-22 22:14:07 +000054STATISTIC(NumGVNInstr, "Number of instructions deleted");
55STATISTIC(NumGVNLoad, "Number of loads deleted");
56STATISTIC(NumGVNPRE, "Number of instructions PRE'd");
Owen Anderson53d546e2008-07-15 16:28:06 +000057STATISTIC(NumGVNBlocks, "Number of blocks merged");
Bill Wendling3c793442008-12-22 22:14:07 +000058STATISTIC(NumPRELoad, "Number of loads PRE'd");
Chris Lattner168be762008-03-22 04:13:49 +000059
Evan Cheng9598f932008-06-20 01:01:07 +000060static cl::opt<bool> EnablePRE("enable-pre",
Owen Andersonaddbe3e2008-07-17 19:41:00 +000061 cl::init(true), cl::Hidden);
Dan Gohmana8f8a852009-06-15 18:30:15 +000062static cl::opt<bool> EnableLoadPRE("enable-load-pre", cl::init(true));
Bob Wilsond517b522010-02-01 21:17:14 +000063static cl::opt<bool> EnableFullLoadPRE("enable-full-load-pre", cl::init(false));
Owen Andersone780d662008-06-19 19:57:25 +000064
Owen Andersonab6ec2e2007-07-24 17:55:58 +000065//===----------------------------------------------------------------------===//
66// ValueTable Class
67//===----------------------------------------------------------------------===//
68
69/// This class holds the mapping between values and value numbers. It is used
70/// as an efficient mechanism to determine the expression-wise equivalence of
71/// two values.
72namespace {
Chris Lattner2dd09db2009-09-02 06:11:42 +000073 struct Expression {
Owen Andersoncdea3572010-01-17 19:33:27 +000074 enum ExpressionOpcode {
75 ADD = Instruction::Add,
76 FADD = Instruction::FAdd,
77 SUB = Instruction::Sub,
78 FSUB = Instruction::FSub,
79 MUL = Instruction::Mul,
80 FMUL = Instruction::FMul,
81 UDIV = Instruction::UDiv,
82 SDIV = Instruction::SDiv,
83 FDIV = Instruction::FDiv,
84 UREM = Instruction::URem,
85 SREM = Instruction::SRem,
86 FREM = Instruction::FRem,
87 SHL = Instruction::Shl,
88 LSHR = Instruction::LShr,
89 ASHR = Instruction::AShr,
90 AND = Instruction::And,
91 OR = Instruction::Or,
92 XOR = Instruction::Xor,
93 TRUNC = Instruction::Trunc,
94 ZEXT = Instruction::ZExt,
95 SEXT = Instruction::SExt,
96 FPTOUI = Instruction::FPToUI,
97 FPTOSI = Instruction::FPToSI,
98 UITOFP = Instruction::UIToFP,
99 SITOFP = Instruction::SIToFP,
100 FPTRUNC = Instruction::FPTrunc,
101 FPEXT = Instruction::FPExt,
102 PTRTOINT = Instruction::PtrToInt,
103 INTTOPTR = Instruction::IntToPtr,
104 BITCAST = Instruction::BitCast,
105 ICMPEQ, ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
106 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
107 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
108 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
109 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
110 SHUFFLE, SELECT, GEP, CALL, CONSTANT,
111 INSERTVALUE, EXTRACTVALUE, EMPTY, TOMBSTONE };
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000112
113 ExpressionOpcode opcode;
114 const Type* type;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000115 SmallVector<uint32_t, 4> varargs;
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000116 Value *function;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000117
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000118 Expression() { }
119 Expression(ExpressionOpcode o) : opcode(o) { }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000120
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000121 bool operator==(const Expression &other) const {
122 if (opcode != other.opcode)
123 return false;
124 else if (opcode == EMPTY || opcode == TOMBSTONE)
125 return true;
126 else if (type != other.type)
127 return false;
Owen Anderson09b83ba2007-10-18 19:39:33 +0000128 else if (function != other.function)
129 return false;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000130 else {
131 if (varargs.size() != other.varargs.size())
132 return false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000133
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000134 for (size_t i = 0; i < varargs.size(); ++i)
135 if (varargs[i] != other.varargs[i])
136 return false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000137
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000138 return true;
139 }
140 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000141
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000142 bool operator!=(const Expression &other) const {
Bill Wendling86f01cb2008-12-22 22:16:31 +0000143 return !(*this == other);
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000144 }
145 };
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000146
Chris Lattner2dd09db2009-09-02 06:11:42 +0000147 class ValueTable {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000148 private:
149 DenseMap<Value*, uint32_t> valueNumbering;
150 DenseMap<Expression, uint32_t> expressionNumbering;
Owen Andersonf7928602008-05-12 20:15:55 +0000151 AliasAnalysis* AA;
152 MemoryDependenceAnalysis* MD;
153 DominatorTree* DT;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000154
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000155 uint32_t nextValueNumber;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000156
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000157 Expression::ExpressionOpcode getOpcode(CmpInst* C);
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000158 Expression create_expression(BinaryOperator* BO);
159 Expression create_expression(CmpInst* C);
160 Expression create_expression(ShuffleVectorInst* V);
161 Expression create_expression(ExtractElementInst* C);
162 Expression create_expression(InsertElementInst* V);
163 Expression create_expression(SelectInst* V);
164 Expression create_expression(CastInst* C);
165 Expression create_expression(GetElementPtrInst* G);
Owen Anderson09b83ba2007-10-18 19:39:33 +0000166 Expression create_expression(CallInst* C);
Owen Anderson69057b82008-05-13 08:17:22 +0000167 Expression create_expression(Constant* C);
Owen Anderson168ad692009-10-19 22:14:22 +0000168 Expression create_expression(ExtractValueInst* C);
169 Expression create_expression(InsertValueInst* C);
170
171 uint32_t lookup_or_add_call(CallInst* C);
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000172 public:
Dan Gohmanc4971722009-04-01 16:37:47 +0000173 ValueTable() : nextValueNumber(1) { }
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000174 uint32_t lookup_or_add(Value *V);
175 uint32_t lookup(Value *V) const;
176 void add(Value *V, uint32_t num);
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000177 void clear();
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000178 void erase(Value *v);
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000179 unsigned size();
Owen Andersonf7928602008-05-12 20:15:55 +0000180 void setAliasAnalysis(AliasAnalysis* A) { AA = A; }
Chris Lattner8541ede2008-12-01 00:40:32 +0000181 AliasAnalysis *getAliasAnalysis() const { return AA; }
Owen Andersonf7928602008-05-12 20:15:55 +0000182 void setMemDep(MemoryDependenceAnalysis* M) { MD = M; }
183 void setDomTree(DominatorTree* D) { DT = D; }
Owen Anderson3ea90a72008-07-03 17:44:33 +0000184 uint32_t getNextUnusedValueNumber() { return nextValueNumber; }
Bill Wendling6b18a392008-12-22 21:36:08 +0000185 void verifyRemoved(const Value *) const;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000186 };
187}
188
189namespace llvm {
Chris Lattner0625bd62007-09-17 18:34:04 +0000190template <> struct DenseMapInfo<Expression> {
Owen Anderson9699a6e2007-08-02 18:16:06 +0000191 static inline Expression getEmptyKey() {
192 return Expression(Expression::EMPTY);
193 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000194
Owen Anderson9699a6e2007-08-02 18:16:06 +0000195 static inline Expression getTombstoneKey() {
196 return Expression(Expression::TOMBSTONE);
197 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000198
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000199 static unsigned getHashValue(const Expression e) {
200 unsigned hash = e.opcode;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000201
Anton Korobeynikov1bfd1212008-02-20 11:26:25 +0000202 hash = ((unsigned)((uintptr_t)e.type >> 4) ^
Owen Anderson168ad692009-10-19 22:14:22 +0000203 (unsigned)((uintptr_t)e.type >> 9));
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000204
Owen Anderson9699a6e2007-08-02 18:16:06 +0000205 for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
206 E = e.varargs.end(); I != E; ++I)
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000207 hash = *I + hash * 37;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000208
Anton Korobeynikov1bfd1212008-02-20 11:26:25 +0000209 hash = ((unsigned)((uintptr_t)e.function >> 4) ^
210 (unsigned)((uintptr_t)e.function >> 9)) +
211 hash * 37;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000212
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000213 return hash;
214 }
Chris Lattner0625bd62007-09-17 18:34:04 +0000215 static bool isEqual(const Expression &LHS, const Expression &RHS) {
216 return LHS == RHS;
217 }
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000218};
Chris Lattner45d040b2009-12-15 07:26:43 +0000219
220template <>
221struct isPodLike<Expression> { static const bool value = true; };
222
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000223}
224
225//===----------------------------------------------------------------------===//
226// ValueTable Internal Functions
227//===----------------------------------------------------------------------===//
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000228
229Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
Nick Lewyckya21d3da2009-07-08 03:04:38 +0000230 if (isa<ICmpInst>(C)) {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000231 switch (C->getPredicate()) {
Chris Lattner2876a642008-03-21 21:14:38 +0000232 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinfbcc6632009-07-14 16:55:14 +0000233 llvm_unreachable("Comparison with unknown predicate?");
Chris Lattner2876a642008-03-21 21:14:38 +0000234 case ICmpInst::ICMP_EQ: return Expression::ICMPEQ;
235 case ICmpInst::ICMP_NE: return Expression::ICMPNE;
236 case ICmpInst::ICMP_UGT: return Expression::ICMPUGT;
237 case ICmpInst::ICMP_UGE: return Expression::ICMPUGE;
238 case ICmpInst::ICMP_ULT: return Expression::ICMPULT;
239 case ICmpInst::ICMP_ULE: return Expression::ICMPULE;
240 case ICmpInst::ICMP_SGT: return Expression::ICMPSGT;
241 case ICmpInst::ICMP_SGE: return Expression::ICMPSGE;
242 case ICmpInst::ICMP_SLT: return Expression::ICMPSLT;
243 case ICmpInst::ICMP_SLE: return Expression::ICMPSLE;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000244 }
Nick Lewyckya21d3da2009-07-08 03:04:38 +0000245 } else {
246 switch (C->getPredicate()) {
247 default: // THIS SHOULD NEVER HAPPEN
Torok Edwinfbcc6632009-07-14 16:55:14 +0000248 llvm_unreachable("Comparison with unknown predicate?");
Nick Lewyckya21d3da2009-07-08 03:04:38 +0000249 case FCmpInst::FCMP_OEQ: return Expression::FCMPOEQ;
250 case FCmpInst::FCMP_OGT: return Expression::FCMPOGT;
251 case FCmpInst::FCMP_OGE: return Expression::FCMPOGE;
252 case FCmpInst::FCMP_OLT: return Expression::FCMPOLT;
253 case FCmpInst::FCMP_OLE: return Expression::FCMPOLE;
254 case FCmpInst::FCMP_ONE: return Expression::FCMPONE;
255 case FCmpInst::FCMP_ORD: return Expression::FCMPORD;
256 case FCmpInst::FCMP_UNO: return Expression::FCMPUNO;
257 case FCmpInst::FCMP_UEQ: return Expression::FCMPUEQ;
258 case FCmpInst::FCMP_UGT: return Expression::FCMPUGT;
259 case FCmpInst::FCMP_UGE: return Expression::FCMPUGE;
260 case FCmpInst::FCMP_ULT: return Expression::FCMPULT;
261 case FCmpInst::FCMP_ULE: return Expression::FCMPULE;
262 case FCmpInst::FCMP_UNE: return Expression::FCMPUNE;
263 }
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000264 }
265}
266
Owen Anderson09b83ba2007-10-18 19:39:33 +0000267Expression ValueTable::create_expression(CallInst* C) {
268 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000269
Owen Anderson09b83ba2007-10-18 19:39:33 +0000270 e.type = C->getType();
Owen Anderson09b83ba2007-10-18 19:39:33 +0000271 e.function = C->getCalledFunction();
272 e.opcode = Expression::CALL;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000273
Owen Anderson09b83ba2007-10-18 19:39:33 +0000274 for (CallInst::op_iterator I = C->op_begin()+1, E = C->op_end();
275 I != E; ++I)
Owen Anderson1e73f292008-04-11 05:11:49 +0000276 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000277
Owen Anderson09b83ba2007-10-18 19:39:33 +0000278 return e;
279}
280
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000281Expression ValueTable::create_expression(BinaryOperator* BO) {
282 Expression e;
Owen Anderson168ad692009-10-19 22:14:22 +0000283 e.varargs.push_back(lookup_or_add(BO->getOperand(0)));
284 e.varargs.push_back(lookup_or_add(BO->getOperand(1)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000285 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000286 e.type = BO->getType();
Owen Andersoncdea3572010-01-17 19:33:27 +0000287 e.opcode = static_cast<Expression::ExpressionOpcode>(BO->getOpcode());
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000288
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000289 return e;
290}
291
292Expression ValueTable::create_expression(CmpInst* C) {
293 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000294
Owen Anderson168ad692009-10-19 22:14:22 +0000295 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
296 e.varargs.push_back(lookup_or_add(C->getOperand(1)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000297 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000298 e.type = C->getType();
299 e.opcode = getOpcode(C);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000300
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000301 return e;
302}
303
304Expression ValueTable::create_expression(CastInst* C) {
305 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000306
Owen Anderson168ad692009-10-19 22:14:22 +0000307 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000308 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000309 e.type = C->getType();
Owen Andersoncdea3572010-01-17 19:33:27 +0000310 e.opcode = static_cast<Expression::ExpressionOpcode>(C->getOpcode());
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000311
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000312 return e;
313}
314
315Expression ValueTable::create_expression(ShuffleVectorInst* S) {
316 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000317
Owen Anderson168ad692009-10-19 22:14:22 +0000318 e.varargs.push_back(lookup_or_add(S->getOperand(0)));
319 e.varargs.push_back(lookup_or_add(S->getOperand(1)));
320 e.varargs.push_back(lookup_or_add(S->getOperand(2)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000321 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000322 e.type = S->getType();
323 e.opcode = Expression::SHUFFLE;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000324
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000325 return e;
326}
327
328Expression ValueTable::create_expression(ExtractElementInst* E) {
329 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000330
Owen Anderson168ad692009-10-19 22:14:22 +0000331 e.varargs.push_back(lookup_or_add(E->getOperand(0)));
332 e.varargs.push_back(lookup_or_add(E->getOperand(1)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000333 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000334 e.type = E->getType();
335 e.opcode = Expression::EXTRACT;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000336
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000337 return e;
338}
339
340Expression ValueTable::create_expression(InsertElementInst* I) {
341 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000342
Owen Anderson168ad692009-10-19 22:14:22 +0000343 e.varargs.push_back(lookup_or_add(I->getOperand(0)));
344 e.varargs.push_back(lookup_or_add(I->getOperand(1)));
345 e.varargs.push_back(lookup_or_add(I->getOperand(2)));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000346 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000347 e.type = I->getType();
348 e.opcode = Expression::INSERT;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000349
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000350 return e;
351}
352
353Expression ValueTable::create_expression(SelectInst* I) {
354 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000355
Owen Anderson168ad692009-10-19 22:14:22 +0000356 e.varargs.push_back(lookup_or_add(I->getCondition()));
357 e.varargs.push_back(lookup_or_add(I->getTrueValue()));
358 e.varargs.push_back(lookup_or_add(I->getFalseValue()));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000359 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000360 e.type = I->getType();
361 e.opcode = Expression::SELECT;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000362
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000363 return e;
364}
365
366Expression ValueTable::create_expression(GetElementPtrInst* G) {
367 Expression e;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000368
Owen Anderson168ad692009-10-19 22:14:22 +0000369 e.varargs.push_back(lookup_or_add(G->getPointerOperand()));
Owen Anderson09b83ba2007-10-18 19:39:33 +0000370 e.function = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000371 e.type = G->getType();
372 e.opcode = Expression::GEP;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000373
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000374 for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
375 I != E; ++I)
Owen Anderson1e73f292008-04-11 05:11:49 +0000376 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000377
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000378 return e;
379}
380
Owen Anderson168ad692009-10-19 22:14:22 +0000381Expression ValueTable::create_expression(ExtractValueInst* E) {
382 Expression e;
383
384 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
385 for (ExtractValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
386 II != IE; ++II)
387 e.varargs.push_back(*II);
388 e.function = 0;
389 e.type = E->getType();
390 e.opcode = Expression::EXTRACTVALUE;
391
392 return e;
393}
394
395Expression ValueTable::create_expression(InsertValueInst* E) {
396 Expression e;
397
398 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
399 e.varargs.push_back(lookup_or_add(E->getInsertedValueOperand()));
400 for (InsertValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
401 II != IE; ++II)
402 e.varargs.push_back(*II);
403 e.function = 0;
404 e.type = E->getType();
405 e.opcode = Expression::INSERTVALUE;
406
407 return e;
408}
409
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000410//===----------------------------------------------------------------------===//
411// ValueTable External Functions
412//===----------------------------------------------------------------------===//
413
Owen Anderson6a903bc2008-06-18 21:41:49 +0000414/// add - Insert a value into the table with a specified value number.
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000415void ValueTable::add(Value *V, uint32_t num) {
Owen Anderson6a903bc2008-06-18 21:41:49 +0000416 valueNumbering.insert(std::make_pair(V, num));
417}
418
Owen Anderson168ad692009-10-19 22:14:22 +0000419uint32_t ValueTable::lookup_or_add_call(CallInst* C) {
420 if (AA->doesNotAccessMemory(C)) {
421 Expression exp = create_expression(C);
422 uint32_t& e = expressionNumbering[exp];
423 if (!e) e = nextValueNumber++;
424 valueNumbering[C] = e;
425 return e;
426 } else if (AA->onlyReadsMemory(C)) {
427 Expression exp = create_expression(C);
428 uint32_t& e = expressionNumbering[exp];
429 if (!e) {
430 e = nextValueNumber++;
431 valueNumbering[C] = e;
432 return e;
433 }
Dan Gohman81132462009-11-14 02:27:51 +0000434 if (!MD) {
435 e = nextValueNumber++;
436 valueNumbering[C] = e;
437 return e;
438 }
Owen Anderson168ad692009-10-19 22:14:22 +0000439
440 MemDepResult local_dep = MD->getDependency(C);
441
442 if (!local_dep.isDef() && !local_dep.isNonLocal()) {
443 valueNumbering[C] = nextValueNumber;
444 return nextValueNumber++;
445 }
446
447 if (local_dep.isDef()) {
448 CallInst* local_cdep = cast<CallInst>(local_dep.getInst());
449
450 if (local_cdep->getNumOperands() != C->getNumOperands()) {
451 valueNumbering[C] = nextValueNumber;
452 return nextValueNumber++;
453 }
454
455 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
456 uint32_t c_vn = lookup_or_add(C->getOperand(i));
457 uint32_t cd_vn = lookup_or_add(local_cdep->getOperand(i));
458 if (c_vn != cd_vn) {
459 valueNumbering[C] = nextValueNumber;
460 return nextValueNumber++;
461 }
462 }
463
464 uint32_t v = lookup_or_add(local_cdep);
465 valueNumbering[C] = v;
466 return v;
467 }
468
469 // Non-local case.
470 const MemoryDependenceAnalysis::NonLocalDepInfo &deps =
471 MD->getNonLocalCallDependency(CallSite(C));
472 // FIXME: call/call dependencies for readonly calls should return def, not
473 // clobber! Move the checking logic to MemDep!
474 CallInst* cdep = 0;
475
476 // Check to see if we have a single dominating call instruction that is
477 // identical to C.
478 for (unsigned i = 0, e = deps.size(); i != e; ++i) {
Chris Lattner0c315472009-12-09 07:08:01 +0000479 const NonLocalDepEntry *I = &deps[i];
Owen Anderson168ad692009-10-19 22:14:22 +0000480 // Ignore non-local dependencies.
Chris Lattner0c315472009-12-09 07:08:01 +0000481 if (I->getResult().isNonLocal())
Owen Anderson168ad692009-10-19 22:14:22 +0000482 continue;
483
484 // We don't handle non-depedencies. If we already have a call, reject
485 // instruction dependencies.
Chris Lattner0c315472009-12-09 07:08:01 +0000486 if (I->getResult().isClobber() || cdep != 0) {
Owen Anderson168ad692009-10-19 22:14:22 +0000487 cdep = 0;
488 break;
489 }
490
Chris Lattner0c315472009-12-09 07:08:01 +0000491 CallInst *NonLocalDepCall = dyn_cast<CallInst>(I->getResult().getInst());
Owen Anderson168ad692009-10-19 22:14:22 +0000492 // FIXME: All duplicated with non-local case.
Chris Lattner0c315472009-12-09 07:08:01 +0000493 if (NonLocalDepCall && DT->properlyDominates(I->getBB(), C->getParent())){
Owen Anderson168ad692009-10-19 22:14:22 +0000494 cdep = NonLocalDepCall;
495 continue;
496 }
497
498 cdep = 0;
499 break;
500 }
501
502 if (!cdep) {
503 valueNumbering[C] = nextValueNumber;
504 return nextValueNumber++;
505 }
506
507 if (cdep->getNumOperands() != C->getNumOperands()) {
508 valueNumbering[C] = nextValueNumber;
509 return nextValueNumber++;
510 }
511 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
512 uint32_t c_vn = lookup_or_add(C->getOperand(i));
513 uint32_t cd_vn = lookup_or_add(cdep->getOperand(i));
514 if (c_vn != cd_vn) {
515 valueNumbering[C] = nextValueNumber;
516 return nextValueNumber++;
517 }
518 }
519
520 uint32_t v = lookup_or_add(cdep);
521 valueNumbering[C] = v;
522 return v;
523
524 } else {
525 valueNumbering[C] = nextValueNumber;
526 return nextValueNumber++;
527 }
528}
529
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000530/// lookup_or_add - Returns the value number for the specified value, assigning
531/// it a new number if it did not have one before.
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000532uint32_t ValueTable::lookup_or_add(Value *V) {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000533 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
534 if (VI != valueNumbering.end())
535 return VI->second;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000536
Owen Anderson168ad692009-10-19 22:14:22 +0000537 if (!isa<Instruction>(V)) {
Owen Anderson1059b5b2009-10-19 21:14:57 +0000538 valueNumbering[V] = nextValueNumber;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000539 return nextValueNumber++;
540 }
Owen Anderson168ad692009-10-19 22:14:22 +0000541
542 Instruction* I = cast<Instruction>(V);
543 Expression exp;
544 switch (I->getOpcode()) {
545 case Instruction::Call:
546 return lookup_or_add_call(cast<CallInst>(I));
547 case Instruction::Add:
548 case Instruction::FAdd:
549 case Instruction::Sub:
550 case Instruction::FSub:
551 case Instruction::Mul:
552 case Instruction::FMul:
553 case Instruction::UDiv:
554 case Instruction::SDiv:
555 case Instruction::FDiv:
556 case Instruction::URem:
557 case Instruction::SRem:
558 case Instruction::FRem:
559 case Instruction::Shl:
560 case Instruction::LShr:
561 case Instruction::AShr:
562 case Instruction::And:
563 case Instruction::Or :
564 case Instruction::Xor:
565 exp = create_expression(cast<BinaryOperator>(I));
566 break;
567 case Instruction::ICmp:
568 case Instruction::FCmp:
569 exp = create_expression(cast<CmpInst>(I));
570 break;
571 case Instruction::Trunc:
572 case Instruction::ZExt:
573 case Instruction::SExt:
574 case Instruction::FPToUI:
575 case Instruction::FPToSI:
576 case Instruction::UIToFP:
577 case Instruction::SIToFP:
578 case Instruction::FPTrunc:
579 case Instruction::FPExt:
580 case Instruction::PtrToInt:
581 case Instruction::IntToPtr:
582 case Instruction::BitCast:
583 exp = create_expression(cast<CastInst>(I));
584 break;
585 case Instruction::Select:
586 exp = create_expression(cast<SelectInst>(I));
587 break;
588 case Instruction::ExtractElement:
589 exp = create_expression(cast<ExtractElementInst>(I));
590 break;
591 case Instruction::InsertElement:
592 exp = create_expression(cast<InsertElementInst>(I));
593 break;
594 case Instruction::ShuffleVector:
595 exp = create_expression(cast<ShuffleVectorInst>(I));
596 break;
597 case Instruction::ExtractValue:
598 exp = create_expression(cast<ExtractValueInst>(I));
599 break;
600 case Instruction::InsertValue:
601 exp = create_expression(cast<InsertValueInst>(I));
602 break;
603 case Instruction::GetElementPtr:
604 exp = create_expression(cast<GetElementPtrInst>(I));
605 break;
606 default:
607 valueNumbering[V] = nextValueNumber;
608 return nextValueNumber++;
609 }
610
611 uint32_t& e = expressionNumbering[exp];
612 if (!e) e = nextValueNumber++;
613 valueNumbering[V] = e;
614 return e;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000615}
616
617/// lookup - Returns the value number of the specified value. Fails if
618/// the value has not yet been numbered.
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000619uint32_t ValueTable::lookup(Value *V) const {
Jeffrey Yasskinb40d3f72009-11-10 01:02:17 +0000620 DenseMap<Value*, uint32_t>::const_iterator VI = valueNumbering.find(V);
Chris Lattner2876a642008-03-21 21:14:38 +0000621 assert(VI != valueNumbering.end() && "Value not numbered?");
622 return VI->second;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000623}
624
625/// clear - Remove all entries from the ValueTable
626void ValueTable::clear() {
627 valueNumbering.clear();
628 expressionNumbering.clear();
629 nextValueNumber = 1;
630}
631
Owen Anderson10ffa862007-07-31 23:27:13 +0000632/// erase - Remove a value from the value numbering
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000633void ValueTable::erase(Value *V) {
Owen Anderson10ffa862007-07-31 23:27:13 +0000634 valueNumbering.erase(V);
635}
636
Bill Wendling6b18a392008-12-22 21:36:08 +0000637/// verifyRemoved - Verify that the value is removed from all internal data
638/// structures.
639void ValueTable::verifyRemoved(const Value *V) const {
Jeffrey Yasskinb40d3f72009-11-10 01:02:17 +0000640 for (DenseMap<Value*, uint32_t>::const_iterator
Bill Wendling6b18a392008-12-22 21:36:08 +0000641 I = valueNumbering.begin(), E = valueNumbering.end(); I != E; ++I) {
642 assert(I->first != V && "Inst still occurs in value numbering map!");
643 }
644}
645
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000646//===----------------------------------------------------------------------===//
Bill Wendling456e8852008-12-22 22:32:22 +0000647// GVN Pass
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000648//===----------------------------------------------------------------------===//
649
650namespace {
Chris Lattner2dd09db2009-09-02 06:11:42 +0000651 struct ValueNumberScope {
Owen Anderson1b3ea962008-06-20 01:15:47 +0000652 ValueNumberScope* parent;
653 DenseMap<uint32_t, Value*> table;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000654
Owen Anderson1b3ea962008-06-20 01:15:47 +0000655 ValueNumberScope(ValueNumberScope* p) : parent(p) { }
656 };
657}
658
659namespace {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000660
Chris Lattner2dd09db2009-09-02 06:11:42 +0000661 class GVN : public FunctionPass {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000662 bool runOnFunction(Function &F);
663 public:
664 static char ID; // Pass identification, replacement for typeid
Dan Gohman81132462009-11-14 02:27:51 +0000665 explicit GVN(bool nopre = false, bool noloads = false)
666 : FunctionPass(&ID), NoPRE(nopre), NoLoads(noloads), MD(0) { }
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000667
668 private:
Evan Cheng5a6b9c42009-10-30 20:12:24 +0000669 bool NoPRE;
Dan Gohman81132462009-11-14 02:27:51 +0000670 bool NoLoads;
Chris Lattner8541ede2008-12-01 00:40:32 +0000671 MemoryDependenceAnalysis *MD;
672 DominatorTree *DT;
673
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000674 ValueTable VN;
Owen Anderson1b3ea962008-06-20 01:15:47 +0000675 DenseMap<BasicBlock*, ValueNumberScope*> localAvail;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000676
Bob Wilson92cdb6e2010-02-16 19:51:59 +0000677 // List of critical edges to be split between iterations.
678 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> toSplit;
679
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000680 // This transformation requires dominator postdominator info
681 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000682 AU.addRequired<DominatorTree>();
Dan Gohman81132462009-11-14 02:27:51 +0000683 if (!NoLoads)
684 AU.addRequired<MemoryDependenceAnalysis>();
Owen Anderson09b83ba2007-10-18 19:39:33 +0000685 AU.addRequired<AliasAnalysis>();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000686
Owen Anderson54e02192008-06-23 17:49:45 +0000687 AU.addPreserved<DominatorTree>();
Owen Anderson09b83ba2007-10-18 19:39:33 +0000688 AU.addPreserved<AliasAnalysis>();
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000689 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000690
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000691 // Helper fuctions
692 // FIXME: eliminate or document these better
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000693 bool processLoad(LoadInst* L,
Chris Lattner804209d2008-03-21 22:01:16 +0000694 SmallVectorImpl<Instruction*> &toErase);
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000695 bool processInstruction(Instruction *I,
Chris Lattner804209d2008-03-21 22:01:16 +0000696 SmallVectorImpl<Instruction*> &toErase);
Owen Anderson9699a6e2007-08-02 18:16:06 +0000697 bool processNonLocalLoad(LoadInst* L,
Chris Lattner804209d2008-03-21 22:01:16 +0000698 SmallVectorImpl<Instruction*> &toErase);
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000699 bool processBlock(BasicBlock *BB);
Owen Anderson6a903bc2008-06-18 21:41:49 +0000700 void dump(DenseMap<uint32_t, Value*>& d);
Owen Anderson676070d2007-08-14 18:04:11 +0000701 bool iterateOnFunction(Function &F);
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000702 Value *CollapsePhi(PHINode* p);
Owen Anderson6a903bc2008-06-18 21:41:49 +0000703 bool performPRE(Function& F);
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000704 Value *lookupNumber(BasicBlock *BB, uint32_t num);
Nuno Lopese3127f32008-10-10 16:25:50 +0000705 void cleanupGlobalSets();
Bill Wendling6b18a392008-12-22 21:36:08 +0000706 void verifyRemoved(const Instruction *I) const;
Bob Wilson92cdb6e2010-02-16 19:51:59 +0000707 bool splitCriticalEdges();
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000708 };
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000709
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000710 char GVN::ID = 0;
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000711}
712
713// createGVNPass - The public interface to this file...
Dan Gohman81132462009-11-14 02:27:51 +0000714FunctionPass *llvm::createGVNPass(bool NoPRE, bool NoLoads) {
715 return new GVN(NoPRE, NoLoads);
716}
Owen Andersonab6ec2e2007-07-24 17:55:58 +0000717
718static RegisterPass<GVN> X("gvn",
719 "Global Value Numbering");
720
Owen Anderson6a903bc2008-06-18 21:41:49 +0000721void GVN::dump(DenseMap<uint32_t, Value*>& d) {
Dan Gohman57e80862009-12-18 03:25:51 +0000722 errs() << "{\n";
Owen Anderson6a903bc2008-06-18 21:41:49 +0000723 for (DenseMap<uint32_t, Value*>::iterator I = d.begin(),
Owen Anderson5e5599b2007-07-25 19:57:03 +0000724 E = d.end(); I != E; ++I) {
Dan Gohman57e80862009-12-18 03:25:51 +0000725 errs() << I->first << "\n";
Owen Anderson5e5599b2007-07-25 19:57:03 +0000726 I->second->dump();
727 }
Dan Gohman57e80862009-12-18 03:25:51 +0000728 errs() << "}\n";
Owen Anderson5e5599b2007-07-25 19:57:03 +0000729}
730
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000731static bool isSafeReplacement(PHINode* p, Instruction *inst) {
Owen Anderson109ca5a2009-08-26 22:55:11 +0000732 if (!isa<PHINode>(inst))
733 return true;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000734
Owen Anderson109ca5a2009-08-26 22:55:11 +0000735 for (Instruction::use_iterator UI = p->use_begin(), E = p->use_end();
736 UI != E; ++UI)
737 if (PHINode* use_phi = dyn_cast<PHINode>(UI))
738 if (use_phi->getParent() == inst->getParent())
739 return false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000740
Owen Anderson109ca5a2009-08-26 22:55:11 +0000741 return true;
742}
743
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000744Value *GVN::CollapsePhi(PHINode *PN) {
745 Value *ConstVal = PN->hasConstantValue(DT);
746 if (!ConstVal) return 0;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000747
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000748 Instruction *Inst = dyn_cast<Instruction>(ConstVal);
749 if (!Inst)
750 return ConstVal;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000751
Chris Lattner1eefa9c2009-09-21 02:42:51 +0000752 if (DT->dominates(Inst, PN))
753 if (isSafeReplacement(PN, Inst))
754 return Inst;
Owen Andersonf5023a72007-08-16 22:51:56 +0000755 return 0;
756}
Owen Anderson5e5599b2007-07-25 19:57:03 +0000757
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000758/// IsValueFullyAvailableInBlock - Return true if we can prove that the value
759/// we're analyzing is fully available in the specified block. As we go, keep
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000760/// track of which blocks we know are fully alive in FullyAvailableBlocks. This
761/// map is actually a tri-state map with the following values:
762/// 0) we know the block *is not* fully available.
763/// 1) we know the block *is* fully available.
764/// 2) we do not know whether the block is fully available or not, but we are
765/// currently speculating that it will be.
766/// 3) we are speculating for this block and have used that to speculate for
767/// other blocks.
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000768static bool IsValueFullyAvailableInBlock(BasicBlock *BB,
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000769 DenseMap<BasicBlock*, char> &FullyAvailableBlocks) {
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000770 // Optimistically assume that the block is fully available and check to see
771 // if we already know about this block in one lookup.
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000772 std::pair<DenseMap<BasicBlock*, char>::iterator, char> IV =
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000773 FullyAvailableBlocks.insert(std::make_pair(BB, 2));
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000774
775 // If the entry already existed for this block, return the precomputed value.
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000776 if (!IV.second) {
777 // If this is a speculative "available" value, mark it as being used for
778 // speculation of other blocks.
779 if (IV.first->second == 2)
780 IV.first->second = 3;
781 return IV.first->second != 0;
782 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000783
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000784 // Otherwise, see if it is fully available in all predecessors.
785 pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000786
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000787 // If this block has no predecessors, it isn't live-in here.
788 if (PI == PE)
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000789 goto SpeculationFailure;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000790
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000791 for (; PI != PE; ++PI)
792 // If the value isn't fully available in one of our predecessors, then it
793 // isn't fully available in this block either. Undo our previous
794 // optimistic assumption and bail out.
795 if (!IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000796 goto SpeculationFailure;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000797
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000798 return true;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000799
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000800// SpeculationFailure - If we get here, we found out that this is not, after
801// all, a fully-available block. We have a problem if we speculated on this and
802// used the speculation to mark other blocks as available.
803SpeculationFailure:
804 char &BBVal = FullyAvailableBlocks[BB];
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000805
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000806 // If we didn't speculate on this, just return with it set to false.
807 if (BBVal == 2) {
808 BBVal = 0;
809 return false;
810 }
811
812 // If we did speculate on this value, we could have blocks set to 1 that are
813 // incorrect. Walk the (transitive) successors of this block and mark them as
814 // 0 if set to one.
815 SmallVector<BasicBlock*, 32> BBWorklist;
816 BBWorklist.push_back(BB);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000817
Dan Gohman28943872010-01-05 16:27:25 +0000818 do {
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000819 BasicBlock *Entry = BBWorklist.pop_back_val();
820 // Note that this sets blocks to 0 (unavailable) if they happen to not
821 // already be in FullyAvailableBlocks. This is safe.
822 char &EntryVal = FullyAvailableBlocks[Entry];
823 if (EntryVal == 0) continue; // Already unavailable.
824
825 // Mark as unavailable.
826 EntryVal = 0;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000827
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000828 for (succ_iterator I = succ_begin(Entry), E = succ_end(Entry); I != E; ++I)
829 BBWorklist.push_back(*I);
Dan Gohman28943872010-01-05 16:27:25 +0000830 } while (!BBWorklist.empty());
Daniel Dunbar7d6781b2009-09-20 02:20:51 +0000831
Chris Lattnerd2a653a2008-12-05 07:49:08 +0000832 return false;
Chris Lattner1db9bbe2008-12-02 08:16:11 +0000833}
834
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000835
Chris Lattner9045f232009-09-21 17:24:04 +0000836/// CanCoerceMustAliasedValueToLoad - Return true if
837/// CoerceAvailableValueToLoadType will succeed.
838static bool CanCoerceMustAliasedValueToLoad(Value *StoredVal,
839 const Type *LoadTy,
840 const TargetData &TD) {
841 // If the loaded or stored value is an first class array or struct, don't try
842 // to transform them. We need to be able to bitcast to integer.
Duncan Sands19d0b472010-02-16 11:11:14 +0000843 if (LoadTy->isStructTy() || LoadTy->isArrayTy() ||
844 StoredVal->getType()->isStructTy() ||
845 StoredVal->getType()->isArrayTy())
Chris Lattner9045f232009-09-21 17:24:04 +0000846 return false;
847
848 // The store has to be at least as big as the load.
849 if (TD.getTypeSizeInBits(StoredVal->getType()) <
850 TD.getTypeSizeInBits(LoadTy))
851 return false;
852
853 return true;
854}
855
856
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000857/// CoerceAvailableValueToLoadType - If we saw a store of a value to memory, and
858/// then a load from a must-aliased pointer of a different type, try to coerce
859/// the stored value. LoadedTy is the type of the load we want to replace and
860/// InsertPt is the place to insert new instructions.
861///
862/// If we can't do it, return null.
863static Value *CoerceAvailableValueToLoadType(Value *StoredVal,
864 const Type *LoadedTy,
865 Instruction *InsertPt,
866 const TargetData &TD) {
Chris Lattner9045f232009-09-21 17:24:04 +0000867 if (!CanCoerceMustAliasedValueToLoad(StoredVal, LoadedTy, TD))
868 return 0;
869
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000870 const Type *StoredValTy = StoredVal->getType();
871
872 uint64_t StoreSize = TD.getTypeSizeInBits(StoredValTy);
873 uint64_t LoadSize = TD.getTypeSizeInBits(LoadedTy);
874
875 // If the store and reload are the same size, we can always reuse it.
876 if (StoreSize == LoadSize) {
Duncan Sands19d0b472010-02-16 11:11:14 +0000877 if (StoredValTy->isPointerTy() && LoadedTy->isPointerTy()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000878 // Pointer to Pointer -> use bitcast.
879 return new BitCastInst(StoredVal, LoadedTy, "", InsertPt);
880 }
881
882 // Convert source pointers to integers, which can be bitcast.
Duncan Sands19d0b472010-02-16 11:11:14 +0000883 if (StoredValTy->isPointerTy()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000884 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
885 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
886 }
887
888 const Type *TypeToCastTo = LoadedTy;
Duncan Sands19d0b472010-02-16 11:11:14 +0000889 if (TypeToCastTo->isPointerTy())
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000890 TypeToCastTo = TD.getIntPtrType(StoredValTy->getContext());
891
892 if (StoredValTy != TypeToCastTo)
893 StoredVal = new BitCastInst(StoredVal, TypeToCastTo, "", InsertPt);
894
895 // Cast to pointer if the load needs a pointer type.
Duncan Sands19d0b472010-02-16 11:11:14 +0000896 if (LoadedTy->isPointerTy())
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000897 StoredVal = new IntToPtrInst(StoredVal, LoadedTy, "", InsertPt);
898
899 return StoredVal;
900 }
901
902 // If the loaded value is smaller than the available value, then we can
903 // extract out a piece from it. If the available value is too small, then we
904 // can't do anything.
Chris Lattner9045f232009-09-21 17:24:04 +0000905 assert(StoreSize >= LoadSize && "CanCoerceMustAliasedValueToLoad fail");
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000906
907 // Convert source pointers to integers, which can be manipulated.
Duncan Sands19d0b472010-02-16 11:11:14 +0000908 if (StoredValTy->isPointerTy()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000909 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
910 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
911 }
912
913 // Convert vectors and fp to integer, which can be manipulated.
Duncan Sands19d0b472010-02-16 11:11:14 +0000914 if (!StoredValTy->isIntegerTy()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000915 StoredValTy = IntegerType::get(StoredValTy->getContext(), StoreSize);
916 StoredVal = new BitCastInst(StoredVal, StoredValTy, "", InsertPt);
917 }
918
919 // If this is a big-endian system, we need to shift the value down to the low
920 // bits so that a truncate will work.
921 if (TD.isBigEndian()) {
922 Constant *Val = ConstantInt::get(StoredVal->getType(), StoreSize-LoadSize);
923 StoredVal = BinaryOperator::CreateLShr(StoredVal, Val, "tmp", InsertPt);
924 }
925
926 // Truncate the integer to the right size now.
927 const Type *NewIntTy = IntegerType::get(StoredValTy->getContext(), LoadSize);
928 StoredVal = new TruncInst(StoredVal, NewIntTy, "trunc", InsertPt);
929
930 if (LoadedTy == NewIntTy)
931 return StoredVal;
932
933 // If the result is a pointer, inttoptr.
Duncan Sands19d0b472010-02-16 11:11:14 +0000934 if (LoadedTy->isPointerTy())
Chris Lattnera0aa8fb2009-09-20 20:09:34 +0000935 return new IntToPtrInst(StoredVal, LoadedTy, "inttoptr", InsertPt);
936
937 // Otherwise, bitcast.
938 return new BitCastInst(StoredVal, LoadedTy, "bitcast", InsertPt);
939}
940
Chris Lattnerd28f9082009-09-21 06:24:16 +0000941/// GetBaseWithConstantOffset - Analyze the specified pointer to see if it can
942/// be expressed as a base pointer plus a constant offset. Return the base and
943/// offset to the caller.
944static Value *GetBaseWithConstantOffset(Value *Ptr, int64_t &Offset,
Chris Lattner4d8af2f2009-09-21 06:48:08 +0000945 const TargetData &TD) {
Chris Lattnerd28f9082009-09-21 06:24:16 +0000946 Operator *PtrOp = dyn_cast<Operator>(Ptr);
947 if (PtrOp == 0) return Ptr;
948
949 // Just look through bitcasts.
950 if (PtrOp->getOpcode() == Instruction::BitCast)
951 return GetBaseWithConstantOffset(PtrOp->getOperand(0), Offset, TD);
952
953 // If this is a GEP with constant indices, we can look through it.
954 GEPOperator *GEP = dyn_cast<GEPOperator>(PtrOp);
955 if (GEP == 0 || !GEP->hasAllConstantIndices()) return Ptr;
956
957 gep_type_iterator GTI = gep_type_begin(GEP);
958 for (User::op_iterator I = GEP->idx_begin(), E = GEP->idx_end(); I != E;
959 ++I, ++GTI) {
960 ConstantInt *OpC = cast<ConstantInt>(*I);
961 if (OpC->isZero()) continue;
962
963 // Handle a struct and array indices which add their offset to the pointer.
964 if (const StructType *STy = dyn_cast<StructType>(*GTI)) {
Chris Lattner4d8af2f2009-09-21 06:48:08 +0000965 Offset += TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue());
Chris Lattnerd28f9082009-09-21 06:24:16 +0000966 } else {
Chris Lattner4d8af2f2009-09-21 06:48:08 +0000967 uint64_t Size = TD.getTypeAllocSize(GTI.getIndexedType());
Chris Lattnerd28f9082009-09-21 06:24:16 +0000968 Offset += OpC->getSExtValue()*Size;
969 }
970 }
971
972 // Re-sign extend from the pointer size if needed to get overflow edge cases
973 // right.
Chris Lattner4d8af2f2009-09-21 06:48:08 +0000974 unsigned PtrSize = TD.getPointerSizeInBits();
Chris Lattnerd28f9082009-09-21 06:24:16 +0000975 if (PtrSize < 64)
976 Offset = (Offset << (64-PtrSize)) >> (64-PtrSize);
977
978 return GetBaseWithConstantOffset(GEP->getPointerOperand(), Offset, TD);
979}
980
981
Chris Lattner42376062009-12-06 01:57:02 +0000982/// AnalyzeLoadFromClobberingWrite - This function is called when we have a
983/// memdep query of a load that ends up being a clobbering memory write (store,
984/// memset, memcpy, memmove). This means that the write *may* provide bits used
985/// by the load but we can't be sure because the pointers don't mustalias.
986///
987/// Check this case to see if there is anything more we can do before we give
988/// up. This returns -1 if we have to give up, or a byte number in the stored
989/// value of the piece that feeds the load.
Chris Lattner0def8612009-12-09 07:34:10 +0000990static int AnalyzeLoadFromClobberingWrite(const Type *LoadTy, Value *LoadPtr,
991 Value *WritePtr,
Chris Lattner42376062009-12-06 01:57:02 +0000992 uint64_t WriteSizeInBits,
Chris Lattner4d8af2f2009-09-21 06:48:08 +0000993 const TargetData &TD) {
Chris Lattner9045f232009-09-21 17:24:04 +0000994 // If the loaded or stored value is an first class array or struct, don't try
995 // to transform them. We need to be able to bitcast to integer.
Duncan Sands19d0b472010-02-16 11:11:14 +0000996 if (LoadTy->isStructTy() || LoadTy->isArrayTy())
Chris Lattner9045f232009-09-21 17:24:04 +0000997 return -1;
998
Chris Lattnerd28f9082009-09-21 06:24:16 +0000999 int64_t StoreOffset = 0, LoadOffset = 0;
Chris Lattner42376062009-12-06 01:57:02 +00001000 Value *StoreBase = GetBaseWithConstantOffset(WritePtr, StoreOffset, TD);
Chris Lattnerd28f9082009-09-21 06:24:16 +00001001 Value *LoadBase =
Chris Lattner0def8612009-12-09 07:34:10 +00001002 GetBaseWithConstantOffset(LoadPtr, LoadOffset, TD);
Chris Lattnerd28f9082009-09-21 06:24:16 +00001003 if (StoreBase != LoadBase)
1004 return -1;
1005
1006 // If the load and store are to the exact same address, they should have been
1007 // a must alias. AA must have gotten confused.
1008 // FIXME: Study to see if/when this happens.
1009 if (LoadOffset == StoreOffset) {
1010#if 0
David Greene2e6efc42010-01-05 01:27:17 +00001011 dbgs() << "STORE/LOAD DEP WITH COMMON POINTER MISSED:\n"
Chris Lattnerd28f9082009-09-21 06:24:16 +00001012 << "Base = " << *StoreBase << "\n"
Chris Lattner42376062009-12-06 01:57:02 +00001013 << "Store Ptr = " << *WritePtr << "\n"
1014 << "Store Offs = " << StoreOffset << "\n"
Chris Lattner3ddf8042009-12-10 00:04:46 +00001015 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattner946b58d2009-12-09 02:41:54 +00001016 abort();
Chris Lattnerd28f9082009-09-21 06:24:16 +00001017#endif
1018 return -1;
1019 }
1020
1021 // If the load and store don't overlap at all, the store doesn't provide
1022 // anything to the load. In this case, they really don't alias at all, AA
1023 // must have gotten confused.
1024 // FIXME: Investigate cases where this bails out, e.g. rdar://7238614. Then
1025 // remove this check, as it is duplicated with what we have below.
Chris Lattner0def8612009-12-09 07:34:10 +00001026 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy);
Chris Lattnerd28f9082009-09-21 06:24:16 +00001027
Chris Lattner42376062009-12-06 01:57:02 +00001028 if ((WriteSizeInBits & 7) | (LoadSize & 7))
Chris Lattnerd28f9082009-09-21 06:24:16 +00001029 return -1;
Chris Lattner42376062009-12-06 01:57:02 +00001030 uint64_t StoreSize = WriteSizeInBits >> 3; // Convert to bytes.
Chris Lattnerd28f9082009-09-21 06:24:16 +00001031 LoadSize >>= 3;
1032
1033
1034 bool isAAFailure = false;
1035 if (StoreOffset < LoadOffset) {
1036 isAAFailure = StoreOffset+int64_t(StoreSize) <= LoadOffset;
1037 } else {
1038 isAAFailure = LoadOffset+int64_t(LoadSize) <= StoreOffset;
1039 }
1040 if (isAAFailure) {
1041#if 0
David Greene2e6efc42010-01-05 01:27:17 +00001042 dbgs() << "STORE LOAD DEP WITH COMMON BASE:\n"
Chris Lattnerd28f9082009-09-21 06:24:16 +00001043 << "Base = " << *StoreBase << "\n"
Chris Lattner42376062009-12-06 01:57:02 +00001044 << "Store Ptr = " << *WritePtr << "\n"
1045 << "Store Offs = " << StoreOffset << "\n"
Chris Lattner3ddf8042009-12-10 00:04:46 +00001046 << "Load Ptr = " << *LoadPtr << "\n";
Chris Lattner946b58d2009-12-09 02:41:54 +00001047 abort();
Chris Lattnerd28f9082009-09-21 06:24:16 +00001048#endif
1049 return -1;
1050 }
1051
1052 // If the Load isn't completely contained within the stored bits, we don't
1053 // have all the bits to feed it. We could do something crazy in the future
1054 // (issue a smaller load then merge the bits in) but this seems unlikely to be
1055 // valuable.
1056 if (StoreOffset > LoadOffset ||
1057 StoreOffset+StoreSize < LoadOffset+LoadSize)
1058 return -1;
1059
1060 // Okay, we can do this transformation. Return the number of bytes into the
1061 // store that the load is.
1062 return LoadOffset-StoreOffset;
1063}
1064
Chris Lattner42376062009-12-06 01:57:02 +00001065/// AnalyzeLoadFromClobberingStore - This function is called when we have a
1066/// memdep query of a load that ends up being a clobbering store.
Chris Lattner07df9ef2009-12-09 07:37:07 +00001067static int AnalyzeLoadFromClobberingStore(const Type *LoadTy, Value *LoadPtr,
1068 StoreInst *DepSI,
Chris Lattner42376062009-12-06 01:57:02 +00001069 const TargetData &TD) {
1070 // Cannot handle reading from store of first-class aggregate yet.
Duncan Sands19d0b472010-02-16 11:11:14 +00001071 if (DepSI->getOperand(0)->getType()->isStructTy() ||
1072 DepSI->getOperand(0)->getType()->isArrayTy())
Chris Lattner42376062009-12-06 01:57:02 +00001073 return -1;
1074
1075 Value *StorePtr = DepSI->getPointerOperand();
Chris Lattner9ccc8792009-12-10 00:11:45 +00001076 uint64_t StoreSize = TD.getTypeSizeInBits(DepSI->getOperand(0)->getType());
Chris Lattner07df9ef2009-12-09 07:37:07 +00001077 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
Chris Lattner0def8612009-12-09 07:34:10 +00001078 StorePtr, StoreSize, TD);
Chris Lattner42376062009-12-06 01:57:02 +00001079}
1080
Chris Lattner07df9ef2009-12-09 07:37:07 +00001081static int AnalyzeLoadFromClobberingMemInst(const Type *LoadTy, Value *LoadPtr,
1082 MemIntrinsic *MI,
Chris Lattner42376062009-12-06 01:57:02 +00001083 const TargetData &TD) {
1084 // If the mem operation is a non-constant size, we can't handle it.
1085 ConstantInt *SizeCst = dyn_cast<ConstantInt>(MI->getLength());
1086 if (SizeCst == 0) return -1;
1087 uint64_t MemSizeInBits = SizeCst->getZExtValue()*8;
Chris Lattner778cb922009-12-06 05:29:56 +00001088
1089 // If this is memset, we just need to see if the offset is valid in the size
1090 // of the memset..
Chris Lattner42376062009-12-06 01:57:02 +00001091 if (MI->getIntrinsicID() == Intrinsic::memset)
Chris Lattner07df9ef2009-12-09 07:37:07 +00001092 return AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr, MI->getDest(),
1093 MemSizeInBits, TD);
Chris Lattner42376062009-12-06 01:57:02 +00001094
Chris Lattner778cb922009-12-06 05:29:56 +00001095 // If we have a memcpy/memmove, the only case we can handle is if this is a
1096 // copy from constant memory. In that case, we can read directly from the
1097 // constant memory.
1098 MemTransferInst *MTI = cast<MemTransferInst>(MI);
1099
1100 Constant *Src = dyn_cast<Constant>(MTI->getSource());
1101 if (Src == 0) return -1;
1102
1103 GlobalVariable *GV = dyn_cast<GlobalVariable>(Src->getUnderlyingObject());
1104 if (GV == 0 || !GV->isConstant()) return -1;
1105
1106 // See if the access is within the bounds of the transfer.
Chris Lattner07df9ef2009-12-09 07:37:07 +00001107 int Offset = AnalyzeLoadFromClobberingWrite(LoadTy, LoadPtr,
1108 MI->getDest(), MemSizeInBits, TD);
Chris Lattner778cb922009-12-06 05:29:56 +00001109 if (Offset == -1)
1110 return Offset;
1111
1112 // Otherwise, see if we can constant fold a load from the constant with the
1113 // offset applied as appropriate.
1114 Src = ConstantExpr::getBitCast(Src,
1115 llvm::Type::getInt8PtrTy(Src->getContext()));
1116 Constant *OffsetCst =
1117 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1118 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
Chris Lattner07df9ef2009-12-09 07:37:07 +00001119 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
Chris Lattner778cb922009-12-06 05:29:56 +00001120 if (ConstantFoldLoadFromConstPtr(Src, &TD))
1121 return Offset;
Chris Lattner42376062009-12-06 01:57:02 +00001122 return -1;
1123}
1124
Chris Lattnerd28f9082009-09-21 06:24:16 +00001125
1126/// GetStoreValueForLoad - This function is called when we have a
1127/// memdep query of a load that ends up being a clobbering store. This means
1128/// that the store *may* provide bits used by the load but we can't be sure
1129/// because the pointers don't mustalias. Check this case to see if there is
1130/// anything more we can do before we give up.
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001131static Value *GetStoreValueForLoad(Value *SrcVal, unsigned Offset,
1132 const Type *LoadTy,
1133 Instruction *InsertPt, const TargetData &TD){
Chris Lattnerd28f9082009-09-21 06:24:16 +00001134 LLVMContext &Ctx = SrcVal->getType()->getContext();
1135
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001136 uint64_t StoreSize = TD.getTypeSizeInBits(SrcVal->getType())/8;
1137 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
Chris Lattnerd28f9082009-09-21 06:24:16 +00001138
Chris Lattnerf8ba1252009-12-09 18:13:28 +00001139 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
Chris Lattnerd28f9082009-09-21 06:24:16 +00001140
1141 // Compute which bits of the stored value are being used by the load. Convert
1142 // to an integer type to start with.
Duncan Sands19d0b472010-02-16 11:11:14 +00001143 if (SrcVal->getType()->isPointerTy())
Chris Lattnerf8ba1252009-12-09 18:13:28 +00001144 SrcVal = Builder.CreatePtrToInt(SrcVal, TD.getIntPtrType(Ctx), "tmp");
Duncan Sands19d0b472010-02-16 11:11:14 +00001145 if (!SrcVal->getType()->isIntegerTy())
Chris Lattnerf8ba1252009-12-09 18:13:28 +00001146 SrcVal = Builder.CreateBitCast(SrcVal, IntegerType::get(Ctx, StoreSize*8),
1147 "tmp");
Chris Lattnerd28f9082009-09-21 06:24:16 +00001148
1149 // Shift the bits to the least significant depending on endianness.
1150 unsigned ShiftAmt;
Chris Lattner42376062009-12-06 01:57:02 +00001151 if (TD.isLittleEndian())
Chris Lattnerd28f9082009-09-21 06:24:16 +00001152 ShiftAmt = Offset*8;
Chris Lattner42376062009-12-06 01:57:02 +00001153 else
Chris Lattner24705382009-09-21 17:55:47 +00001154 ShiftAmt = (StoreSize-LoadSize-Offset)*8;
Chris Lattnerd28f9082009-09-21 06:24:16 +00001155
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001156 if (ShiftAmt)
Chris Lattnerf8ba1252009-12-09 18:13:28 +00001157 SrcVal = Builder.CreateLShr(SrcVal, ShiftAmt, "tmp");
Chris Lattnerd28f9082009-09-21 06:24:16 +00001158
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001159 if (LoadSize != StoreSize)
Chris Lattnerf8ba1252009-12-09 18:13:28 +00001160 SrcVal = Builder.CreateTrunc(SrcVal, IntegerType::get(Ctx, LoadSize*8),
1161 "tmp");
Chris Lattnerd28f9082009-09-21 06:24:16 +00001162
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001163 return CoerceAvailableValueToLoadType(SrcVal, LoadTy, InsertPt, TD);
Chris Lattnerd28f9082009-09-21 06:24:16 +00001164}
1165
Chris Lattner42376062009-12-06 01:57:02 +00001166/// GetMemInstValueForLoad - This function is called when we have a
1167/// memdep query of a load that ends up being a clobbering mem intrinsic.
1168static Value *GetMemInstValueForLoad(MemIntrinsic *SrcInst, unsigned Offset,
1169 const Type *LoadTy, Instruction *InsertPt,
1170 const TargetData &TD){
1171 LLVMContext &Ctx = LoadTy->getContext();
1172 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
1173
1174 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
1175
1176 // We know that this method is only called when the mem transfer fully
1177 // provides the bits for the load.
1178 if (MemSetInst *MSI = dyn_cast<MemSetInst>(SrcInst)) {
1179 // memset(P, 'x', 1234) -> splat('x'), even if x is a variable, and
1180 // independently of what the offset is.
1181 Value *Val = MSI->getValue();
1182 if (LoadSize != 1)
1183 Val = Builder.CreateZExt(Val, IntegerType::get(Ctx, LoadSize*8));
1184
1185 Value *OneElt = Val;
1186
1187 // Splat the value out to the right number of bits.
1188 for (unsigned NumBytesSet = 1; NumBytesSet != LoadSize; ) {
1189 // If we can double the number of bytes set, do it.
1190 if (NumBytesSet*2 <= LoadSize) {
1191 Value *ShVal = Builder.CreateShl(Val, NumBytesSet*8);
1192 Val = Builder.CreateOr(Val, ShVal);
1193 NumBytesSet <<= 1;
1194 continue;
1195 }
1196
1197 // Otherwise insert one byte at a time.
1198 Value *ShVal = Builder.CreateShl(Val, 1*8);
1199 Val = Builder.CreateOr(OneElt, ShVal);
1200 ++NumBytesSet;
1201 }
1202
1203 return CoerceAvailableValueToLoadType(Val, LoadTy, InsertPt, TD);
1204 }
Chris Lattner778cb922009-12-06 05:29:56 +00001205
1206 // Otherwise, this is a memcpy/memmove from a constant global.
1207 MemTransferInst *MTI = cast<MemTransferInst>(SrcInst);
1208 Constant *Src = cast<Constant>(MTI->getSource());
1209
1210 // Otherwise, see if we can constant fold a load from the constant with the
1211 // offset applied as appropriate.
1212 Src = ConstantExpr::getBitCast(Src,
1213 llvm::Type::getInt8PtrTy(Src->getContext()));
1214 Constant *OffsetCst =
1215 ConstantInt::get(Type::getInt64Ty(Src->getContext()), (unsigned)Offset);
1216 Src = ConstantExpr::getGetElementPtr(Src, &OffsetCst, 1);
1217 Src = ConstantExpr::getBitCast(Src, PointerType::getUnqual(LoadTy));
1218 return ConstantFoldLoadFromConstPtr(Src, &TD);
Chris Lattner42376062009-12-06 01:57:02 +00001219}
1220
1221
1222
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001223struct AvailableValueInBlock {
1224 /// BB - The basic block in question.
1225 BasicBlock *BB;
Chris Lattner93236ba2009-12-06 04:54:31 +00001226 enum ValType {
1227 SimpleVal, // A simple offsetted value that is accessed.
1228 MemIntrin // A memory intrinsic which is loaded from.
1229 };
1230
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001231 /// V - The value that is live out of the block.
Chris Lattner93236ba2009-12-06 04:54:31 +00001232 PointerIntPair<Value *, 1, ValType> Val;
1233
1234 /// Offset - The byte offset in Val that is interesting for the load query.
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001235 unsigned Offset;
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001236
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001237 static AvailableValueInBlock get(BasicBlock *BB, Value *V,
1238 unsigned Offset = 0) {
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001239 AvailableValueInBlock Res;
1240 Res.BB = BB;
Chris Lattner93236ba2009-12-06 04:54:31 +00001241 Res.Val.setPointer(V);
1242 Res.Val.setInt(SimpleVal);
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001243 Res.Offset = Offset;
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001244 return Res;
1245 }
Chris Lattner93236ba2009-12-06 04:54:31 +00001246
1247 static AvailableValueInBlock getMI(BasicBlock *BB, MemIntrinsic *MI,
1248 unsigned Offset = 0) {
1249 AvailableValueInBlock Res;
1250 Res.BB = BB;
1251 Res.Val.setPointer(MI);
1252 Res.Val.setInt(MemIntrin);
1253 Res.Offset = Offset;
1254 return Res;
1255 }
1256
1257 bool isSimpleValue() const { return Val.getInt() == SimpleVal; }
1258 Value *getSimpleValue() const {
1259 assert(isSimpleValue() && "Wrong accessor");
1260 return Val.getPointer();
1261 }
1262
1263 MemIntrinsic *getMemIntrinValue() const {
1264 assert(!isSimpleValue() && "Wrong accessor");
1265 return cast<MemIntrinsic>(Val.getPointer());
1266 }
Chris Lattner927b0ac2009-12-21 23:04:33 +00001267
1268 /// MaterializeAdjustedValue - Emit code into this block to adjust the value
1269 /// defined here to the specified type. This handles various coercion cases.
1270 Value *MaterializeAdjustedValue(const Type *LoadTy,
1271 const TargetData *TD) const {
1272 Value *Res;
1273 if (isSimpleValue()) {
1274 Res = getSimpleValue();
1275 if (Res->getType() != LoadTy) {
1276 assert(TD && "Need target data to handle type mismatch case");
1277 Res = GetStoreValueForLoad(Res, Offset, LoadTy, BB->getTerminator(),
1278 *TD);
1279
1280 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\nOffset: " << Offset << " "
1281 << *getSimpleValue() << '\n'
1282 << *Res << '\n' << "\n\n\n");
1283 }
1284 } else {
1285 Res = GetMemInstValueForLoad(getMemIntrinValue(), Offset,
1286 LoadTy, BB->getTerminator(), *TD);
1287 DEBUG(errs() << "GVN COERCED NONLOCAL MEM INTRIN:\nOffset: " << Offset
1288 << " " << *getMemIntrinValue() << '\n'
1289 << *Res << '\n' << "\n\n\n");
1290 }
1291 return Res;
1292 }
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001293};
1294
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001295/// ConstructSSAForLoadSet - Given a set of loads specified by ValuesPerBlock,
1296/// construct SSA form, allowing us to eliminate LI. This returns the value
1297/// that should be used at LI's definition site.
1298static Value *ConstructSSAForLoadSet(LoadInst *LI,
1299 SmallVectorImpl<AvailableValueInBlock> &ValuesPerBlock,
1300 const TargetData *TD,
Chris Lattnerbf200182009-12-21 23:15:48 +00001301 const DominatorTree &DT,
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001302 AliasAnalysis *AA) {
Chris Lattnerbf200182009-12-21 23:15:48 +00001303 // Check for the fully redundant, dominating load case. In this case, we can
1304 // just use the dominating value directly.
1305 if (ValuesPerBlock.size() == 1 &&
1306 DT.properlyDominates(ValuesPerBlock[0].BB, LI->getParent()))
1307 return ValuesPerBlock[0].MaterializeAdjustedValue(LI->getType(), TD);
1308
1309 // Otherwise, we have to construct SSA form.
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001310 SmallVector<PHINode*, 8> NewPHIs;
1311 SSAUpdater SSAUpdate(&NewPHIs);
1312 SSAUpdate.Initialize(LI);
1313
1314 const Type *LoadTy = LI->getType();
1315
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001316 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
Chris Lattner93236ba2009-12-06 04:54:31 +00001317 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1318 BasicBlock *BB = AV.BB;
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001319
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001320 if (SSAUpdate.HasValueForBlock(BB))
1321 continue;
Chris Lattner93236ba2009-12-06 04:54:31 +00001322
Chris Lattner927b0ac2009-12-21 23:04:33 +00001323 SSAUpdate.AddAvailableValue(BB, AV.MaterializeAdjustedValue(LoadTy, TD));
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001324 }
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001325
1326 // Perform PHI construction.
1327 Value *V = SSAUpdate.GetValueInMiddleOfBlock(LI->getParent());
1328
1329 // If new PHI nodes were created, notify alias analysis.
Duncan Sands19d0b472010-02-16 11:11:14 +00001330 if (V->getType()->isPointerTy())
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001331 for (unsigned i = 0, e = NewPHIs.size(); i != e; ++i)
1332 AA->copyValue(LI, NewPHIs[i]);
1333
1334 return V;
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001335}
1336
Owen Andersonb9878ee2009-12-02 07:35:19 +00001337static bool isLifetimeStart(Instruction *Inst) {
Chris Lattnerc4680252009-12-02 06:44:58 +00001338 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(Inst))
Owen Andersonb9878ee2009-12-02 07:35:19 +00001339 return II->getIntrinsicID() == Intrinsic::lifetime_start;
Chris Lattnerc4680252009-12-02 06:44:58 +00001340 return false;
1341}
1342
Owen Anderson221a4362007-08-16 22:02:55 +00001343/// processNonLocalLoad - Attempt to eliminate a load whose dependencies are
1344/// non-local by performing PHI construction.
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001345bool GVN::processNonLocalLoad(LoadInst *LI,
Chris Lattner804209d2008-03-21 22:01:16 +00001346 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001347 // Find the non-local dependencies of the load.
Chris Lattner9b7d99e2009-12-22 04:25:02 +00001348 SmallVector<NonLocalDepResult, 64> Deps;
Chris Lattnerb6fc4b82008-12-09 19:25:07 +00001349 MD->getNonLocalPointerDependency(LI->getOperand(0), true, LI->getParent(),
1350 Deps);
David Greene2e6efc42010-01-05 01:27:17 +00001351 //DEBUG(dbgs() << "INVESTIGATING NONLOCAL LOAD: "
Dan Gohmanef3ef7f2009-07-31 20:24:18 +00001352 // << Deps.size() << *LI << '\n');
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001353
Owen Andersonb39e0de2008-08-26 22:07:42 +00001354 // If we had to process more than one hundred blocks to find the
1355 // dependencies, this load isn't worth worrying about. Optimizing
1356 // it will be too expensive.
Chris Lattnerb6fc4b82008-12-09 19:25:07 +00001357 if (Deps.size() > 100)
Owen Andersonb39e0de2008-08-26 22:07:42 +00001358 return false;
Chris Lattnerb6372932008-12-18 00:51:32 +00001359
1360 // If we had a phi translation failure, we'll have a single entry which is a
1361 // clobber in the current block. Reject this early.
Chris Lattner0c315472009-12-09 07:08:01 +00001362 if (Deps.size() == 1 && Deps[0].getResult().isClobber()) {
Torok Edwinba93ea72009-06-17 18:48:18 +00001363 DEBUG(
David Greene2e6efc42010-01-05 01:27:17 +00001364 dbgs() << "GVN: non-local load ";
1365 WriteAsOperand(dbgs(), LI);
1366 dbgs() << " is clobbered by " << *Deps[0].getResult().getInst() << '\n';
Torok Edwinba93ea72009-06-17 18:48:18 +00001367 );
Chris Lattnerb6372932008-12-18 00:51:32 +00001368 return false;
Torok Edwinba93ea72009-06-17 18:48:18 +00001369 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001370
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001371 // Filter out useless results (non-locals, etc). Keep track of the blocks
1372 // where we have a value available in repl, also keep track of whether we see
1373 // dependencies that produce an unknown value for the load (such as a call
1374 // that could potentially clobber the load).
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001375 SmallVector<AvailableValueInBlock, 16> ValuesPerBlock;
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001376 SmallVector<BasicBlock*, 16> UnavailableBlocks;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001377
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001378 const TargetData *TD = 0;
1379
Chris Lattnerb6fc4b82008-12-09 19:25:07 +00001380 for (unsigned i = 0, e = Deps.size(); i != e; ++i) {
Chris Lattner0c315472009-12-09 07:08:01 +00001381 BasicBlock *DepBB = Deps[i].getBB();
1382 MemDepResult DepInfo = Deps[i].getResult();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001383
Chris Lattner0e3d6332008-12-05 21:04:20 +00001384 if (DepInfo.isClobber()) {
Chris Lattnerca5f9cb2009-12-09 18:21:46 +00001385 // The address being loaded in this non-local block may not be the same as
1386 // the pointer operand of the load if PHI translation occurs. Make sure
1387 // to consider the right address.
1388 Value *Address = Deps[i].getAddress();
1389
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001390 // If the dependence is to a store that writes to a superset of the bits
1391 // read by the load, we can extract the bits we need for the load from the
1392 // stored value.
1393 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInfo.getInst())) {
1394 if (TD == 0)
1395 TD = getAnalysisIfAvailable<TargetData>();
Chris Lattnerca5f9cb2009-12-09 18:21:46 +00001396 if (TD && Address) {
1397 int Offset = AnalyzeLoadFromClobberingStore(LI->getType(), Address,
Chris Lattner07df9ef2009-12-09 07:37:07 +00001398 DepSI, *TD);
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001399 if (Offset != -1) {
1400 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1401 DepSI->getOperand(0),
1402 Offset));
1403 continue;
1404 }
1405 }
1406 }
Chris Lattner42376062009-12-06 01:57:02 +00001407
Chris Lattner42376062009-12-06 01:57:02 +00001408 // If the clobbering value is a memset/memcpy/memmove, see if we can
1409 // forward a value on from it.
Chris Lattner93236ba2009-12-06 04:54:31 +00001410 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(DepInfo.getInst())) {
Chris Lattner42376062009-12-06 01:57:02 +00001411 if (TD == 0)
1412 TD = getAnalysisIfAvailable<TargetData>();
Chris Lattnerca5f9cb2009-12-09 18:21:46 +00001413 if (TD && Address) {
1414 int Offset = AnalyzeLoadFromClobberingMemInst(LI->getType(), Address,
Chris Lattner07df9ef2009-12-09 07:37:07 +00001415 DepMI, *TD);
Chris Lattner93236ba2009-12-06 04:54:31 +00001416 if (Offset != -1) {
1417 ValuesPerBlock.push_back(AvailableValueInBlock::getMI(DepBB, DepMI,
1418 Offset));
1419 continue;
1420 }
Chris Lattner42376062009-12-06 01:57:02 +00001421 }
1422 }
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001423
Chris Lattner0e3d6332008-12-05 21:04:20 +00001424 UnavailableBlocks.push_back(DepBB);
1425 continue;
1426 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001427
Chris Lattner0e3d6332008-12-05 21:04:20 +00001428 Instruction *DepInst = DepInfo.getInst();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001429
Chris Lattner0e3d6332008-12-05 21:04:20 +00001430 // Loading the allocation -> undef.
Chris Lattnerc4680252009-12-02 06:44:58 +00001431 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst) ||
Owen Andersonb9878ee2009-12-02 07:35:19 +00001432 // Loading immediately after lifetime begin -> undef.
1433 isLifetimeStart(DepInst)) {
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001434 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1435 UndefValue::get(LI->getType())));
Chris Lattner7e61daf2008-12-01 01:15:42 +00001436 continue;
1437 }
Owen Anderson2b2bd282009-10-28 07:05:35 +00001438
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001439 if (StoreInst *S = dyn_cast<StoreInst>(DepInst)) {
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001440 // Reject loads and stores that are to the same address but are of
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001441 // different types if we have to.
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001442 if (S->getOperand(0)->getType() != LI->getType()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001443 if (TD == 0)
1444 TD = getAnalysisIfAvailable<TargetData>();
1445
1446 // If the stored value is larger or equal to the loaded value, we can
1447 // reuse it.
Chris Lattner9045f232009-09-21 17:24:04 +00001448 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(S->getOperand(0),
1449 LI->getType(), *TD)) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001450 UnavailableBlocks.push_back(DepBB);
1451 continue;
1452 }
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001453 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001454
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001455 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1456 S->getOperand(0)));
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001457 continue;
1458 }
1459
1460 if (LoadInst *LD = dyn_cast<LoadInst>(DepInst)) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001461 // If the types mismatch and we can't handle it, reject reuse of the load.
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001462 if (LD->getType() != LI->getType()) {
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001463 if (TD == 0)
1464 TD = getAnalysisIfAvailable<TargetData>();
1465
1466 // If the stored value is larger or equal to the loaded value, we can
1467 // reuse it.
Chris Lattner9045f232009-09-21 17:24:04 +00001468 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(LD, LI->getType(),*TD)){
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001469 UnavailableBlocks.push_back(DepBB);
1470 continue;
1471 }
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001472 }
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001473 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB, LD));
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001474 continue;
Owen Anderson5e5599b2007-07-25 19:57:03 +00001475 }
Chris Lattner4d8af2f2009-09-21 06:48:08 +00001476
1477 UnavailableBlocks.push_back(DepBB);
1478 continue;
Chris Lattner2876a642008-03-21 21:14:38 +00001479 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001480
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001481 // If we have no predecessors that produce a known value for this load, exit
1482 // early.
1483 if (ValuesPerBlock.empty()) return false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001484
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001485 // If all of the instructions we depend on produce a known value for this
1486 // load, then it is fully redundant and we can use PHI insertion to compute
1487 // its value. Insert PHIs and remove the fully redundant value now.
1488 if (UnavailableBlocks.empty()) {
David Greene2e6efc42010-01-05 01:27:17 +00001489 DEBUG(dbgs() << "GVN REMOVING NONLOCAL LOAD: " << *LI << '\n');
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001490
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001491 // Perform PHI construction.
Chris Lattnerbf200182009-12-21 23:15:48 +00001492 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001493 VN.getAliasAnalysis());
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001494 LI->replaceAllUsesWith(V);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001495
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001496 if (isa<PHINode>(V))
1497 V->takeName(LI);
Duncan Sands19d0b472010-02-16 11:11:14 +00001498 if (V->getType()->isPointerTy())
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001499 MD->invalidateCachedPointerInfo(V);
Bob Wilson1da90412010-02-22 21:39:41 +00001500 VN.erase(LI);
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001501 toErase.push_back(LI);
1502 NumGVNLoad++;
1503 return true;
1504 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001505
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001506 if (!EnablePRE || !EnableLoadPRE)
1507 return false;
1508
1509 // Okay, we have *some* definitions of the value. This means that the value
1510 // is available in some of our (transitive) predecessors. Lets think about
1511 // doing PRE of this load. This will involve inserting a new load into the
1512 // predecessor when it's not available. We could do this in general, but
1513 // prefer to not increase code size. As such, we only do this when we know
1514 // that we only have to insert *one* load (which means we're basically moving
1515 // the load, not inserting a new one).
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001516
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001517 SmallPtrSet<BasicBlock *, 4> Blockers;
1518 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1519 Blockers.insert(UnavailableBlocks[i]);
1520
1521 // Lets find first basic block with more than one predecessor. Walk backwards
1522 // through predecessors if needed.
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001523 BasicBlock *LoadBB = LI->getParent();
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001524 BasicBlock *TmpBB = LoadBB;
1525
1526 bool isSinglePred = false;
Dale Johannesen81b64632009-06-17 20:48:23 +00001527 bool allSingleSucc = true;
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001528 while (TmpBB->getSinglePredecessor()) {
1529 isSinglePred = true;
1530 TmpBB = TmpBB->getSinglePredecessor();
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001531 if (TmpBB == LoadBB) // Infinite (unreachable) loop.
1532 return false;
1533 if (Blockers.count(TmpBB))
1534 return false;
Dale Johannesen81b64632009-06-17 20:48:23 +00001535 if (TmpBB->getTerminator()->getNumSuccessors() != 1)
1536 allSingleSucc = false;
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001537 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001538
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001539 assert(TmpBB);
1540 LoadBB = TmpBB;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001541
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001542 // If we have a repl set with LI itself in it, this means we have a loop where
1543 // at least one of the values is LI. Since this means that we won't be able
1544 // to eliminate LI even if we insert uses in the other predecessors, we will
1545 // end up increasing code size. Reject this by scanning for LI.
Bob Wilsond517b522010-02-01 21:17:14 +00001546 if (!EnableFullLoadPRE) {
1547 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
1548 if (ValuesPerBlock[i].isSimpleValue() &&
1549 ValuesPerBlock[i].getSimpleValue() == LI)
1550 return false;
1551 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001552
Chris Lattner93236ba2009-12-06 04:54:31 +00001553 // FIXME: It is extremely unclear what this loop is doing, other than
1554 // artificially restricting loadpre.
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001555 if (isSinglePred) {
1556 bool isHot = false;
Chris Lattner93236ba2009-12-06 04:54:31 +00001557 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
1558 const AvailableValueInBlock &AV = ValuesPerBlock[i];
1559 if (AV.isSimpleValue())
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001560 // "Hot" Instruction is in some loop (because it dominates its dep.
1561 // instruction).
Chris Lattner93236ba2009-12-06 04:54:31 +00001562 if (Instruction *I = dyn_cast<Instruction>(AV.getSimpleValue()))
1563 if (DT->dominates(LI, I)) {
1564 isHot = true;
1565 break;
1566 }
1567 }
Owen Andersoncc0c75c2009-05-31 09:03:40 +00001568
1569 // We are interested only in "hot" instructions. We don't want to do any
1570 // mis-optimizations here.
1571 if (!isHot)
1572 return false;
1573 }
1574
Bob Wilsond517b522010-02-01 21:17:14 +00001575 // Check to see how many predecessors have the loaded value fully
1576 // available.
1577 DenseMap<BasicBlock*, Value*> PredLoads;
Chris Lattnerd2a653a2008-12-05 07:49:08 +00001578 DenseMap<BasicBlock*, char> FullyAvailableBlocks;
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001579 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner0cdc17e2009-09-21 06:30:24 +00001580 FullyAvailableBlocks[ValuesPerBlock[i].BB] = true;
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001581 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1582 FullyAvailableBlocks[UnavailableBlocks[i]] = false;
1583
1584 for (pred_iterator PI = pred_begin(LoadBB), E = pred_end(LoadBB);
1585 PI != E; ++PI) {
Bob Wilsond517b522010-02-01 21:17:14 +00001586 BasicBlock *Pred = *PI;
1587 if (IsValueFullyAvailableInBlock(Pred, FullyAvailableBlocks)) {
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001588 continue;
Bob Wilsond517b522010-02-01 21:17:14 +00001589 }
1590 PredLoads[Pred] = 0;
Bob Wilson92cdb6e2010-02-16 19:51:59 +00001591
Bob Wilsond517b522010-02-01 21:17:14 +00001592 if (Pred->getTerminator()->getNumSuccessors() != 1) {
Bob Wilson92cdb6e2010-02-16 19:51:59 +00001593 if (isa<IndirectBrInst>(Pred->getTerminator())) {
1594 DEBUG(dbgs() << "COULD NOT PRE LOAD BECAUSE OF INDBR CRITICAL EDGE '"
1595 << Pred->getName() << "': " << *LI << '\n');
1596 return false;
1597 }
Bob Wilsonaff96b22010-02-16 21:06:42 +00001598 unsigned SuccNum = GetSuccessorNumber(Pred, LoadBB);
Bob Wilson92cdb6e2010-02-16 19:51:59 +00001599 toSplit.push_back(std::make_pair(Pred->getTerminator(), SuccNum));
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001600 return false;
Bob Wilsond517b522010-02-01 21:17:14 +00001601 }
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001602 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001603
Bob Wilsond517b522010-02-01 21:17:14 +00001604 // Decide whether PRE is profitable for this load.
1605 unsigned NumUnavailablePreds = PredLoads.size();
1606 assert(NumUnavailablePreds != 0 &&
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001607 "Fully available value should be eliminated above!");
Bob Wilsond517b522010-02-01 21:17:14 +00001608 if (!EnableFullLoadPRE) {
1609 // If this load is unavailable in multiple predecessors, reject it.
1610 // FIXME: If we could restructure the CFG, we could make a common pred with
1611 // all the preds that don't have an available LI and insert a new load into
1612 // that one block.
1613 if (NumUnavailablePreds != 1)
1614 return false;
Owen Anderson0cc1a762007-08-07 23:12:31 +00001615 }
Bob Wilsond517b522010-02-01 21:17:14 +00001616
1617 // Check if the load can safely be moved to all the unavailable predecessors.
1618 bool CanDoPRE = true;
Chris Lattner44da5bd2009-11-28 15:39:14 +00001619 SmallVector<Instruction*, 8> NewInsts;
Bob Wilsond517b522010-02-01 21:17:14 +00001620 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1621 E = PredLoads.end(); I != E; ++I) {
1622 BasicBlock *UnavailablePred = I->first;
1623
1624 // Do PHI translation to get its value in the predecessor if necessary. The
1625 // returned pointer (if non-null) is guaranteed to dominate UnavailablePred.
1626
1627 // If all preds have a single successor, then we know it is safe to insert
1628 // the load on the pred (?!?), so we can insert code to materialize the
1629 // pointer if it is not available.
1630 PHITransAddr Address(LI->getOperand(0), TD);
1631 Value *LoadPtr = 0;
1632 if (allSingleSucc) {
1633 LoadPtr = Address.PHITranslateWithInsertion(LoadBB, UnavailablePred,
1634 *DT, NewInsts);
1635 } else {
1636 Address.PHITranslateValue(LoadBB, UnavailablePred);
1637 LoadPtr = Address.getAddr();
Chris Lattner972e6d82009-12-09 01:59:31 +00001638
Bob Wilsond517b522010-02-01 21:17:14 +00001639 // Make sure the value is live in the predecessor.
1640 if (Instruction *Inst = dyn_cast_or_null<Instruction>(LoadPtr))
1641 if (!DT->dominates(Inst->getParent(), UnavailablePred))
1642 LoadPtr = 0;
1643 }
1644
1645 // If we couldn't find or insert a computation of this phi translated value,
1646 // we fail PRE.
1647 if (LoadPtr == 0) {
1648 DEBUG(dbgs() << "COULDN'T INSERT PHI TRANSLATED VALUE OF: "
1649 << *LI->getOperand(0) << "\n");
1650 CanDoPRE = false;
1651 break;
1652 }
1653
1654 // Make sure it is valid to move this load here. We have to watch out for:
1655 // @1 = getelementptr (i8* p, ...
1656 // test p and branch if == 0
1657 // load @1
1658 // It is valid to have the getelementptr before the test, even if p can be 0,
1659 // as getelementptr only does address arithmetic.
1660 // If we are not pushing the value through any multiple-successor blocks
1661 // we do not have this case. Otherwise, check that the load is safe to
1662 // put anywhere; this can be improved, but should be conservatively safe.
1663 if (!allSingleSucc &&
1664 // FIXME: REEVALUTE THIS.
1665 !isSafeToLoadUnconditionally(LoadPtr,
1666 UnavailablePred->getTerminator(),
1667 LI->getAlignment(), TD)) {
1668 CanDoPRE = false;
1669 break;
1670 }
1671
1672 I->second = LoadPtr;
Chris Lattner972e6d82009-12-09 01:59:31 +00001673 }
1674
Bob Wilsond517b522010-02-01 21:17:14 +00001675 if (!CanDoPRE) {
1676 while (!NewInsts.empty())
1677 NewInsts.pop_back_val()->eraseFromParent();
Dale Johannesen81b64632009-06-17 20:48:23 +00001678 return false;
Chris Lattner32140312009-11-28 16:08:18 +00001679 }
Dale Johannesen81b64632009-06-17 20:48:23 +00001680
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001681 // Okay, we can eliminate this load by inserting a reload in the predecessor
1682 // and using PHI construction to get the value in the other predecessors, do
1683 // it.
David Greene2e6efc42010-01-05 01:27:17 +00001684 DEBUG(dbgs() << "GVN REMOVING PRE LOAD: " << *LI << '\n');
Chris Lattner32140312009-11-28 16:08:18 +00001685 DEBUG(if (!NewInsts.empty())
David Greene2e6efc42010-01-05 01:27:17 +00001686 dbgs() << "INSERTED " << NewInsts.size() << " INSTS: "
Chris Lattner32140312009-11-28 16:08:18 +00001687 << *NewInsts.back() << '\n');
1688
Bob Wilsond517b522010-02-01 21:17:14 +00001689 // Assign value numbers to the new instructions.
1690 for (unsigned i = 0, e = NewInsts.size(); i != e; ++i) {
1691 // FIXME: We really _ought_ to insert these value numbers into their
1692 // parent's availability map. However, in doing so, we risk getting into
1693 // ordering issues. If a block hasn't been processed yet, we would be
1694 // marking a value as AVAIL-IN, which isn't what we intend.
1695 VN.lookup_or_add(NewInsts[i]);
1696 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001697
Bob Wilsond517b522010-02-01 21:17:14 +00001698 for (DenseMap<BasicBlock*, Value*>::iterator I = PredLoads.begin(),
1699 E = PredLoads.end(); I != E; ++I) {
1700 BasicBlock *UnavailablePred = I->first;
1701 Value *LoadPtr = I->second;
1702
1703 Value *NewLoad = new LoadInst(LoadPtr, LI->getName()+".pre", false,
1704 LI->getAlignment(),
1705 UnavailablePred->getTerminator());
1706
1707 // Add the newly created load.
1708 ValuesPerBlock.push_back(AvailableValueInBlock::get(UnavailablePred,
1709 NewLoad));
Bob Wilson923261b2010-02-23 05:55:00 +00001710 MD->invalidateCachedPointerInfo(LoadPtr);
1711 DEBUG(dbgs() << "GVN INSERTED " << *NewLoad << '\n');
Bob Wilsond517b522010-02-01 21:17:14 +00001712 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001713
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001714 // Perform PHI construction.
Chris Lattnerbf200182009-12-21 23:15:48 +00001715 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD, *DT,
Chris Lattnerb6c65fa2009-10-10 23:50:30 +00001716 VN.getAliasAnalysis());
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001717 LI->replaceAllUsesWith(V);
1718 if (isa<PHINode>(V))
1719 V->takeName(LI);
Duncan Sands19d0b472010-02-16 11:11:14 +00001720 if (V->getType()->isPointerTy())
Chris Lattnera0aa8fb2009-09-20 20:09:34 +00001721 MD->invalidateCachedPointerInfo(V);
Bob Wilson1da90412010-02-22 21:39:41 +00001722 VN.erase(LI);
Chris Lattner1db9bbe2008-12-02 08:16:11 +00001723 toErase.push_back(LI);
1724 NumPRELoad++;
Owen Anderson5e5599b2007-07-25 19:57:03 +00001725 return true;
1726}
1727
Owen Anderson221a4362007-08-16 22:02:55 +00001728/// processLoad - Attempt to eliminate a load, first by eliminating it
1729/// locally, and then attempting non-local elimination if that fails.
Chris Lattner0e3d6332008-12-05 21:04:20 +00001730bool GVN::processLoad(LoadInst *L, SmallVectorImpl<Instruction*> &toErase) {
Dan Gohman81132462009-11-14 02:27:51 +00001731 if (!MD)
1732 return false;
1733
Chris Lattner0e3d6332008-12-05 21:04:20 +00001734 if (L->isVolatile())
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001735 return false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001736
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001737 // ... to a pointer that has been loaded from before...
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001738 MemDepResult Dep = MD->getDependency(L);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001739
Chris Lattner0e3d6332008-12-05 21:04:20 +00001740 // If the value isn't available, don't do anything!
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001741 if (Dep.isClobber()) {
Chris Lattner0a9616d2009-09-21 05:57:11 +00001742 // Check to see if we have something like this:
Chris Lattner1dd48c32009-09-20 19:03:47 +00001743 // store i32 123, i32* %P
1744 // %A = bitcast i32* %P to i8*
1745 // %B = gep i8* %A, i32 1
1746 // %C = load i8* %B
1747 //
1748 // We could do that by recognizing if the clobber instructions are obviously
1749 // a common base + constant offset, and if the previous store (or memset)
1750 // completely covers this load. This sort of thing can happen in bitfield
1751 // access code.
Chris Lattner42376062009-12-06 01:57:02 +00001752 Value *AvailVal = 0;
Chris Lattner0a9616d2009-09-21 05:57:11 +00001753 if (StoreInst *DepSI = dyn_cast<StoreInst>(Dep.getInst()))
Chris Lattner9d7fb292009-09-21 06:22:46 +00001754 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattner07df9ef2009-12-09 07:37:07 +00001755 int Offset = AnalyzeLoadFromClobberingStore(L->getType(),
1756 L->getPointerOperand(),
1757 DepSI, *TD);
Chris Lattner42376062009-12-06 01:57:02 +00001758 if (Offset != -1)
1759 AvailVal = GetStoreValueForLoad(DepSI->getOperand(0), Offset,
1760 L->getType(), L, *TD);
Chris Lattner9d7fb292009-09-21 06:22:46 +00001761 }
Chris Lattner0a9616d2009-09-21 05:57:11 +00001762
Chris Lattner42376062009-12-06 01:57:02 +00001763 // If the clobbering value is a memset/memcpy/memmove, see if we can forward
1764 // a value on from it.
1765 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(Dep.getInst())) {
1766 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattner07df9ef2009-12-09 07:37:07 +00001767 int Offset = AnalyzeLoadFromClobberingMemInst(L->getType(),
1768 L->getPointerOperand(),
1769 DepMI, *TD);
Chris Lattner42376062009-12-06 01:57:02 +00001770 if (Offset != -1)
1771 AvailVal = GetMemInstValueForLoad(DepMI, Offset, L->getType(), L,*TD);
1772 }
1773 }
1774
1775 if (AvailVal) {
David Greene2e6efc42010-01-05 01:27:17 +00001776 DEBUG(dbgs() << "GVN COERCED INST:\n" << *Dep.getInst() << '\n'
Chris Lattner42376062009-12-06 01:57:02 +00001777 << *AvailVal << '\n' << *L << "\n\n\n");
1778
1779 // Replace the load!
1780 L->replaceAllUsesWith(AvailVal);
Duncan Sands19d0b472010-02-16 11:11:14 +00001781 if (AvailVal->getType()->isPointerTy())
Chris Lattner42376062009-12-06 01:57:02 +00001782 MD->invalidateCachedPointerInfo(AvailVal);
Bob Wilson1da90412010-02-22 21:39:41 +00001783 VN.erase(L);
Chris Lattner42376062009-12-06 01:57:02 +00001784 toErase.push_back(L);
1785 NumGVNLoad++;
1786 return true;
1787 }
1788
Torok Edwin72070282009-05-29 09:46:03 +00001789 DEBUG(
1790 // fast print dep, using operator<< on instruction would be too slow
David Greene2e6efc42010-01-05 01:27:17 +00001791 dbgs() << "GVN: load ";
1792 WriteAsOperand(dbgs(), L);
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001793 Instruction *I = Dep.getInst();
David Greene2e6efc42010-01-05 01:27:17 +00001794 dbgs() << " is clobbered by " << *I << '\n';
Torok Edwin72070282009-05-29 09:46:03 +00001795 );
Chris Lattner0e3d6332008-12-05 21:04:20 +00001796 return false;
Torok Edwin72070282009-05-29 09:46:03 +00001797 }
Chris Lattner0e3d6332008-12-05 21:04:20 +00001798
1799 // If it is defined in another block, try harder.
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001800 if (Dep.isNonLocal())
Chris Lattner0e3d6332008-12-05 21:04:20 +00001801 return processNonLocalLoad(L, toErase);
Eli Friedman716c10c2008-02-12 12:08:14 +00001802
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001803 Instruction *DepInst = Dep.getInst();
Chris Lattner0e3d6332008-12-05 21:04:20 +00001804 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInst)) {
Chris Lattner1dd48c32009-09-20 19:03:47 +00001805 Value *StoredVal = DepSI->getOperand(0);
1806
1807 // The store and load are to a must-aliased pointer, but they may not
1808 // actually have the same type. See if we know how to reuse the stored
1809 // value (depending on its type).
1810 const TargetData *TD = 0;
Chris Lattner8ed7bef2009-10-21 04:11:19 +00001811 if (StoredVal->getType() != L->getType()) {
1812 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1813 StoredVal = CoerceAvailableValueToLoadType(StoredVal, L->getType(),
1814 L, *TD);
1815 if (StoredVal == 0)
1816 return false;
1817
David Greene2e6efc42010-01-05 01:27:17 +00001818 DEBUG(dbgs() << "GVN COERCED STORE:\n" << *DepSI << '\n' << *StoredVal
Chris Lattner8ed7bef2009-10-21 04:11:19 +00001819 << '\n' << *L << "\n\n\n");
1820 }
1821 else
Chris Lattner1dd48c32009-09-20 19:03:47 +00001822 return false;
Chris Lattner1dd48c32009-09-20 19:03:47 +00001823 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001824
Chris Lattner0e3d6332008-12-05 21:04:20 +00001825 // Remove it!
Chris Lattner1dd48c32009-09-20 19:03:47 +00001826 L->replaceAllUsesWith(StoredVal);
Duncan Sands19d0b472010-02-16 11:11:14 +00001827 if (StoredVal->getType()->isPointerTy())
Chris Lattner1dd48c32009-09-20 19:03:47 +00001828 MD->invalidateCachedPointerInfo(StoredVal);
Bob Wilson1da90412010-02-22 21:39:41 +00001829 VN.erase(L);
Chris Lattner0e3d6332008-12-05 21:04:20 +00001830 toErase.push_back(L);
1831 NumGVNLoad++;
1832 return true;
1833 }
1834
1835 if (LoadInst *DepLI = dyn_cast<LoadInst>(DepInst)) {
Chris Lattner1dd48c32009-09-20 19:03:47 +00001836 Value *AvailableVal = DepLI;
1837
1838 // The loads are of a must-aliased pointer, but they may not actually have
1839 // the same type. See if we know how to reuse the previously loaded value
1840 // (depending on its type).
1841 const TargetData *TD = 0;
Chris Lattner8ed7bef2009-10-21 04:11:19 +00001842 if (DepLI->getType() != L->getType()) {
1843 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1844 AvailableVal = CoerceAvailableValueToLoadType(DepLI, L->getType(), L,*TD);
1845 if (AvailableVal == 0)
1846 return false;
Chris Lattner1dd48c32009-09-20 19:03:47 +00001847
David Greene2e6efc42010-01-05 01:27:17 +00001848 DEBUG(dbgs() << "GVN COERCED LOAD:\n" << *DepLI << "\n" << *AvailableVal
Chris Lattner8ed7bef2009-10-21 04:11:19 +00001849 << "\n" << *L << "\n\n\n");
1850 }
1851 else
1852 return false;
Chris Lattner1dd48c32009-09-20 19:03:47 +00001853 }
1854
Chris Lattner0e3d6332008-12-05 21:04:20 +00001855 // Remove it!
Chris Lattner1dd48c32009-09-20 19:03:47 +00001856 L->replaceAllUsesWith(AvailableVal);
Duncan Sands19d0b472010-02-16 11:11:14 +00001857 if (DepLI->getType()->isPointerTy())
Chris Lattnerfa9f99a2008-12-09 22:06:23 +00001858 MD->invalidateCachedPointerInfo(DepLI);
Bob Wilson1da90412010-02-22 21:39:41 +00001859 VN.erase(L);
Chris Lattner0e3d6332008-12-05 21:04:20 +00001860 toErase.push_back(L);
1861 NumGVNLoad++;
1862 return true;
1863 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001864
Chris Lattner3ff6d012008-11-30 01:39:32 +00001865 // If this load really doesn't depend on anything, then we must be loading an
1866 // undef value. This can happen when loading for a fresh allocation with no
1867 // intervening stores, for example.
Victor Hernandez8acf2952009-10-23 21:09:37 +00001868 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst)) {
Owen Andersonb292b8c2009-07-30 23:03:37 +00001869 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson1da90412010-02-22 21:39:41 +00001870 VN.erase(L);
Chris Lattner3ff6d012008-11-30 01:39:32 +00001871 toErase.push_back(L);
Chris Lattner3ff6d012008-11-30 01:39:32 +00001872 NumGVNLoad++;
Chris Lattner0e3d6332008-12-05 21:04:20 +00001873 return true;
Eli Friedman716c10c2008-02-12 12:08:14 +00001874 }
Owen Anderson2b2bd282009-10-28 07:05:35 +00001875
Owen Andersonb9878ee2009-12-02 07:35:19 +00001876 // If this load occurs either right after a lifetime begin,
Owen Anderson2b2bd282009-10-28 07:05:35 +00001877 // then the loaded value is undefined.
1878 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(DepInst)) {
Owen Andersonb9878ee2009-12-02 07:35:19 +00001879 if (II->getIntrinsicID() == Intrinsic::lifetime_start) {
Owen Anderson2b2bd282009-10-28 07:05:35 +00001880 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Bob Wilson1da90412010-02-22 21:39:41 +00001881 VN.erase(L);
Owen Anderson2b2bd282009-10-28 07:05:35 +00001882 toErase.push_back(L);
1883 NumGVNLoad++;
1884 return true;
1885 }
1886 }
Eli Friedman716c10c2008-02-12 12:08:14 +00001887
Chris Lattner0e3d6332008-12-05 21:04:20 +00001888 return false;
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001889}
1890
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001891Value *GVN::lookupNumber(BasicBlock *BB, uint32_t num) {
Owen Anderson54e02192008-06-23 17:49:45 +00001892 DenseMap<BasicBlock*, ValueNumberScope*>::iterator I = localAvail.find(BB);
1893 if (I == localAvail.end())
1894 return 0;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001895
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001896 ValueNumberScope *Locals = I->second;
1897 while (Locals) {
1898 DenseMap<uint32_t, Value*>::iterator I = Locals->table.find(num);
1899 if (I != Locals->table.end())
Owen Anderson1b3ea962008-06-20 01:15:47 +00001900 return I->second;
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001901 Locals = Locals->parent;
Owen Anderson1b3ea962008-06-20 01:15:47 +00001902 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001903
Owen Anderson1b3ea962008-06-20 01:15:47 +00001904 return 0;
1905}
1906
Owen Andersonbfe133e2008-12-15 02:03:00 +00001907
Owen Anderson398602a2007-08-14 18:16:29 +00001908/// processInstruction - When calculating availability, handle an instruction
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001909/// by inserting it into the appropriate sets
Owen Andersonaccdca12008-06-12 19:25:32 +00001910bool GVN::processInstruction(Instruction *I,
Chris Lattner804209d2008-03-21 22:01:16 +00001911 SmallVectorImpl<Instruction*> &toErase) {
Devang Patel03936a12010-02-11 00:20:49 +00001912 // Ignore dbg info intrinsics.
1913 if (isa<DbgInfoIntrinsic>(I))
1914 return false;
1915
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001916 if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
1917 bool Changed = processLoad(LI, toErase);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001918
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001919 if (!Changed) {
1920 unsigned Num = VN.lookup_or_add(LI);
1921 localAvail[I->getParent()]->table.insert(std::make_pair(Num, LI));
Owen Anderson6a903bc2008-06-18 21:41:49 +00001922 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001923
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001924 return Changed;
Owen Anderson6a903bc2008-06-18 21:41:49 +00001925 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001926
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001927 uint32_t NextNum = VN.getNextUnusedValueNumber();
1928 unsigned Num = VN.lookup_or_add(I);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001929
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001930 if (BranchInst *BI = dyn_cast<BranchInst>(I)) {
1931 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001932
Owen Anderson98f912b2009-04-01 23:53:49 +00001933 if (!BI->isConditional() || isa<Constant>(BI->getCondition()))
1934 return false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001935
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001936 Value *BranchCond = BI->getCondition();
1937 uint32_t CondVN = VN.lookup_or_add(BranchCond);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001938
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001939 BasicBlock *TrueSucc = BI->getSuccessor(0);
1940 BasicBlock *FalseSucc = BI->getSuccessor(1);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001941
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001942 if (TrueSucc->getSinglePredecessor())
1943 localAvail[TrueSucc]->table[CondVN] =
1944 ConstantInt::getTrue(TrueSucc->getContext());
1945 if (FalseSucc->getSinglePredecessor())
1946 localAvail[FalseSucc]->table[CondVN] =
1947 ConstantInt::getFalse(TrueSucc->getContext());
Owen Anderson98f912b2009-04-01 23:53:49 +00001948
1949 return false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001950
Owen Anderson0c1e6342008-04-07 09:59:07 +00001951 // Allocations are always uniquely numbered, so we can save time and memory
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001952 // by fast failing them.
Victor Hernandez8acf2952009-10-23 21:09:37 +00001953 } else if (isa<AllocaInst>(I) || isa<TerminatorInst>(I)) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001954 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson0c1e6342008-04-07 09:59:07 +00001955 return false;
Owen Anderson6a903bc2008-06-18 21:41:49 +00001956 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001957
Owen Anderson221a4362007-08-16 22:02:55 +00001958 // Collapse PHI nodes
Owen Andersonbc271a02007-08-14 18:33:27 +00001959 if (PHINode* p = dyn_cast<PHINode>(I)) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001960 Value *constVal = CollapsePhi(p);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001961
Owen Andersonbc271a02007-08-14 18:33:27 +00001962 if (constVal) {
Owen Andersonf5023a72007-08-16 22:51:56 +00001963 p->replaceAllUsesWith(constVal);
Duncan Sands19d0b472010-02-16 11:11:14 +00001964 if (MD && constVal->getType()->isPointerTy())
Chris Lattnerfa9f99a2008-12-09 22:06:23 +00001965 MD->invalidateCachedPointerInfo(constVal);
Owen Anderson164274e2008-12-23 00:49:51 +00001966 VN.erase(p);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001967
Owen Andersonf5023a72007-08-16 22:51:56 +00001968 toErase.push_back(p);
Owen Anderson6a903bc2008-06-18 21:41:49 +00001969 } else {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001970 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Andersonbc271a02007-08-14 18:33:27 +00001971 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001972
Owen Anderson3ea90a72008-07-03 17:44:33 +00001973 // If the number we were assigned was a brand new VN, then we don't
1974 // need to do a lookup to see if the number already exists
1975 // somewhere in the domtree: it can't!
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001976 } else if (Num == NextNum) {
1977 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001978
Owen Andersonbfe133e2008-12-15 02:03:00 +00001979 // Perform fast-path value-number based elimination of values inherited from
1980 // dominators.
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001981 } else if (Value *repl = lookupNumber(I->getParent(), Num)) {
Owen Anderson086b2c42007-12-08 01:37:09 +00001982 // Remove it!
Owen Anderson10ffa862007-07-31 23:27:13 +00001983 VN.erase(I);
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001984 I->replaceAllUsesWith(repl);
Duncan Sands19d0b472010-02-16 11:11:14 +00001985 if (MD && repl->getType()->isPointerTy())
Chris Lattnerfa9f99a2008-12-09 22:06:23 +00001986 MD->invalidateCachedPointerInfo(repl);
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001987 toErase.push_back(I);
1988 return true;
Owen Andersonbfe133e2008-12-15 02:03:00 +00001989
Owen Anderson3ea90a72008-07-03 17:44:33 +00001990 } else {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00001991 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001992 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00001993
Owen Andersonab6ec2e2007-07-24 17:55:58 +00001994 return false;
1995}
1996
Bill Wendling456e8852008-12-22 22:32:22 +00001997/// runOnFunction - This is the main transformation entry point for a function.
Owen Anderson676070d2007-08-14 18:04:11 +00001998bool GVN::runOnFunction(Function& F) {
Dan Gohman81132462009-11-14 02:27:51 +00001999 if (!NoLoads)
2000 MD = &getAnalysis<MemoryDependenceAnalysis>();
Chris Lattner8541ede2008-12-01 00:40:32 +00002001 DT = &getAnalysis<DominatorTree>();
Owen Andersonf7928602008-05-12 20:15:55 +00002002 VN.setAliasAnalysis(&getAnalysis<AliasAnalysis>());
Chris Lattner8541ede2008-12-01 00:40:32 +00002003 VN.setMemDep(MD);
2004 VN.setDomTree(DT);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002005
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002006 bool Changed = false;
2007 bool ShouldContinue = true;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002008
Owen Andersonac310962008-07-16 17:52:31 +00002009 // Merge unconditional branches, allowing PRE to catch more
2010 // optimization opportunities.
2011 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002012 BasicBlock *BB = FI;
Owen Andersonac310962008-07-16 17:52:31 +00002013 ++FI;
Owen Andersonc0623812008-07-17 00:01:40 +00002014 bool removedBlock = MergeBlockIntoPredecessor(BB, this);
2015 if (removedBlock) NumGVNBlocks++;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002016
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002017 Changed |= removedBlock;
Owen Andersonac310962008-07-16 17:52:31 +00002018 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002019
Chris Lattner0a5a8d52008-12-09 19:21:47 +00002020 unsigned Iteration = 0;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002021
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002022 while (ShouldContinue) {
David Greene2e6efc42010-01-05 01:27:17 +00002023 DEBUG(dbgs() << "GVN iteration: " << Iteration << "\n");
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002024 ShouldContinue = iterateOnFunction(F);
Bob Wilson92cdb6e2010-02-16 19:51:59 +00002025 if (splitCriticalEdges())
2026 ShouldContinue = true;
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002027 Changed |= ShouldContinue;
Chris Lattner0a5a8d52008-12-09 19:21:47 +00002028 ++Iteration;
Owen Anderson676070d2007-08-14 18:04:11 +00002029 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002030
Owen Anderson04a6e0b2008-07-18 18:03:38 +00002031 if (EnablePRE) {
Owen Anderson2fbfb702008-09-03 23:06:07 +00002032 bool PREChanged = true;
2033 while (PREChanged) {
2034 PREChanged = performPRE(F);
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002035 Changed |= PREChanged;
Owen Anderson2fbfb702008-09-03 23:06:07 +00002036 }
Owen Anderson04a6e0b2008-07-18 18:03:38 +00002037 }
Chris Lattner0a5a8d52008-12-09 19:21:47 +00002038 // FIXME: Should perform GVN again after PRE does something. PRE can move
2039 // computations into blocks where they become fully redundant. Note that
2040 // we can't do this until PRE's critical edge splitting updates memdep.
2041 // Actually, when this happens, we should just fully integrate PRE into GVN.
Nuno Lopese3127f32008-10-10 16:25:50 +00002042
2043 cleanupGlobalSets();
2044
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002045 return Changed;
Owen Anderson676070d2007-08-14 18:04:11 +00002046}
2047
2048
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002049bool GVN::processBlock(BasicBlock *BB) {
Chris Lattner0a5a8d52008-12-09 19:21:47 +00002050 // FIXME: Kill off toErase by doing erasing eagerly in a helper function (and
2051 // incrementing BI before processing an instruction).
Owen Andersonaccdca12008-06-12 19:25:32 +00002052 SmallVector<Instruction*, 8> toErase;
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002053 bool ChangedFunction = false;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002054
Owen Andersonaccdca12008-06-12 19:25:32 +00002055 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
2056 BI != BE;) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002057 ChangedFunction |= processInstruction(BI, toErase);
Owen Andersonaccdca12008-06-12 19:25:32 +00002058 if (toErase.empty()) {
2059 ++BI;
2060 continue;
2061 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002062
Owen Andersonaccdca12008-06-12 19:25:32 +00002063 // If we need some instructions deleted, do it now.
2064 NumGVNInstr += toErase.size();
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002065
Owen Andersonaccdca12008-06-12 19:25:32 +00002066 // Avoid iterator invalidation.
2067 bool AtStart = BI == BB->begin();
2068 if (!AtStart)
2069 --BI;
2070
2071 for (SmallVector<Instruction*, 4>::iterator I = toErase.begin(),
Chris Lattner8541ede2008-12-01 00:40:32 +00002072 E = toErase.end(); I != E; ++I) {
David Greene2e6efc42010-01-05 01:27:17 +00002073 DEBUG(dbgs() << "GVN removed: " << **I << '\n');
Dan Gohman81132462009-11-14 02:27:51 +00002074 if (MD) MD->removeInstruction(*I);
Owen Andersonaccdca12008-06-12 19:25:32 +00002075 (*I)->eraseFromParent();
Bill Wendlingebb6a542008-12-22 21:57:30 +00002076 DEBUG(verifyRemoved(*I));
Chris Lattner8541ede2008-12-01 00:40:32 +00002077 }
Chris Lattner0a5a8d52008-12-09 19:21:47 +00002078 toErase.clear();
Owen Andersonaccdca12008-06-12 19:25:32 +00002079
2080 if (AtStart)
2081 BI = BB->begin();
2082 else
2083 ++BI;
Owen Andersonaccdca12008-06-12 19:25:32 +00002084 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002085
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002086 return ChangedFunction;
Owen Andersonaccdca12008-06-12 19:25:32 +00002087}
2088
Owen Anderson6a903bc2008-06-18 21:41:49 +00002089/// performPRE - Perform a purely local form of PRE that looks for diamond
2090/// control flow patterns and attempts to perform simple PRE at the join point.
Chris Lattnera546dcf2009-10-31 22:11:15 +00002091bool GVN::performPRE(Function &F) {
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002092 bool Changed = false;
Chris Lattnerf00aae42008-12-01 07:29:03 +00002093 DenseMap<BasicBlock*, Value*> predMap;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002094 for (df_iterator<BasicBlock*> DI = df_begin(&F.getEntryBlock()),
2095 DE = df_end(&F.getEntryBlock()); DI != DE; ++DI) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002096 BasicBlock *CurrentBlock = *DI;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002097
Owen Anderson6a903bc2008-06-18 21:41:49 +00002098 // Nothing to PRE in the entry block.
2099 if (CurrentBlock == &F.getEntryBlock()) continue;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002100
Owen Anderson6a903bc2008-06-18 21:41:49 +00002101 for (BasicBlock::iterator BI = CurrentBlock->begin(),
2102 BE = CurrentBlock->end(); BI != BE; ) {
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002103 Instruction *CurInst = BI++;
Duncan Sands1efabaa2009-05-06 06:49:50 +00002104
Victor Hernandez8acf2952009-10-23 21:09:37 +00002105 if (isa<AllocaInst>(CurInst) ||
Victor Hernandez5d034492009-09-18 22:35:49 +00002106 isa<TerminatorInst>(CurInst) || isa<PHINode>(CurInst) ||
Devang Patel92f86192009-10-14 17:29:00 +00002107 CurInst->getType()->isVoidTy() ||
Duncan Sands1efabaa2009-05-06 06:49:50 +00002108 CurInst->mayReadFromMemory() || CurInst->mayHaveSideEffects() ||
John Criswell073e4d12009-03-10 15:04:53 +00002109 isa<DbgInfoIntrinsic>(CurInst))
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002110 continue;
Duncan Sands1efabaa2009-05-06 06:49:50 +00002111
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002112 uint32_t ValNo = VN.lookup(CurInst);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002113
Owen Anderson6a903bc2008-06-18 21:41:49 +00002114 // Look for the predecessors for PRE opportunities. We're
2115 // only trying to solve the basic diamond case, where
2116 // a value is computed in the successor and one predecessor,
2117 // but not the other. We also explicitly disallow cases
2118 // where the successor is its own predecessor, because they're
2119 // more complicated to get right.
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002120 unsigned NumWith = 0;
2121 unsigned NumWithout = 0;
2122 BasicBlock *PREPred = 0;
Chris Lattnerf00aae42008-12-01 07:29:03 +00002123 predMap.clear();
2124
Owen Anderson6a903bc2008-06-18 21:41:49 +00002125 for (pred_iterator PI = pred_begin(CurrentBlock),
2126 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
2127 // We're not interested in PRE where the block is its
Bob Wilson76e8c592010-02-03 00:33:21 +00002128 // own predecessor, or in blocks with predecessors
Owen Anderson1b3ea962008-06-20 01:15:47 +00002129 // that are not reachable.
2130 if (*PI == CurrentBlock) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002131 NumWithout = 2;
Owen Anderson1b3ea962008-06-20 01:15:47 +00002132 break;
2133 } else if (!localAvail.count(*PI)) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002134 NumWithout = 2;
Owen Anderson1b3ea962008-06-20 01:15:47 +00002135 break;
2136 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002137
2138 DenseMap<uint32_t, Value*>::iterator predV =
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002139 localAvail[*PI]->table.find(ValNo);
Owen Anderson1b3ea962008-06-20 01:15:47 +00002140 if (predV == localAvail[*PI]->table.end()) {
Owen Anderson6a903bc2008-06-18 21:41:49 +00002141 PREPred = *PI;
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002142 NumWithout++;
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002143 } else if (predV->second == CurInst) {
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002144 NumWithout = 2;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002145 } else {
Owen Anderson1b3ea962008-06-20 01:15:47 +00002146 predMap[*PI] = predV->second;
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002147 NumWith++;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002148 }
2149 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002150
Owen Anderson6a903bc2008-06-18 21:41:49 +00002151 // Don't do PRE when it might increase code size, i.e. when
2152 // we would need to insert instructions in more than one pred.
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002153 if (NumWithout != 1 || NumWith == 0)
Owen Anderson6a903bc2008-06-18 21:41:49 +00002154 continue;
Chris Lattnera546dcf2009-10-31 22:11:15 +00002155
2156 // Don't do PRE across indirect branch.
2157 if (isa<IndirectBrInst>(PREPred->getTerminator()))
2158 continue;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002159
Owen Andersonfdf9f162008-06-19 19:54:19 +00002160 // We can't do PRE safely on a critical edge, so instead we schedule
2161 // the edge to be split and perform the PRE the next time we iterate
2162 // on the function.
Bob Wilsonaff96b22010-02-16 21:06:42 +00002163 unsigned SuccNum = GetSuccessorNumber(PREPred, CurrentBlock);
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002164 if (isCriticalEdge(PREPred->getTerminator(), SuccNum)) {
2165 toSplit.push_back(std::make_pair(PREPred->getTerminator(), SuccNum));
Owen Andersonfdf9f162008-06-19 19:54:19 +00002166 continue;
2167 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002168
Bob Wilson76e8c592010-02-03 00:33:21 +00002169 // Instantiate the expression in the predecessor that lacked it.
Owen Anderson6a903bc2008-06-18 21:41:49 +00002170 // Because we are going top-down through the block, all value numbers
2171 // will be available in the predecessor by the time we need them. Any
Bob Wilson76e8c592010-02-03 00:33:21 +00002172 // that weren't originally present will have been instantiated earlier
Owen Anderson6a903bc2008-06-18 21:41:49 +00002173 // in this loop.
Nick Lewycky42fb7452009-09-27 07:38:41 +00002174 Instruction *PREInstr = CurInst->clone();
Owen Anderson6a903bc2008-06-18 21:41:49 +00002175 bool success = true;
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002176 for (unsigned i = 0, e = CurInst->getNumOperands(); i != e; ++i) {
2177 Value *Op = PREInstr->getOperand(i);
2178 if (isa<Argument>(Op) || isa<Constant>(Op) || isa<GlobalValue>(Op))
2179 continue;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002180
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002181 if (Value *V = lookupNumber(PREPred, VN.lookup(Op))) {
2182 PREInstr->setOperand(i, V);
2183 } else {
2184 success = false;
2185 break;
Owen Anderson8e462e92008-07-11 20:05:13 +00002186 }
Owen Anderson6a903bc2008-06-18 21:41:49 +00002187 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002188
Owen Anderson6a903bc2008-06-18 21:41:49 +00002189 // Fail out if we encounter an operand that is not available in
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002190 // the PRE predecessor. This is typically because of loads which
Owen Anderson6a903bc2008-06-18 21:41:49 +00002191 // are not value numbered precisely.
2192 if (!success) {
2193 delete PREInstr;
Bill Wendling3c793442008-12-22 22:14:07 +00002194 DEBUG(verifyRemoved(PREInstr));
Owen Anderson6a903bc2008-06-18 21:41:49 +00002195 continue;
2196 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002197
Owen Anderson6a903bc2008-06-18 21:41:49 +00002198 PREInstr->insertBefore(PREPred->getTerminator());
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002199 PREInstr->setName(CurInst->getName() + ".pre");
Owen Anderson1b3ea962008-06-20 01:15:47 +00002200 predMap[PREPred] = PREInstr;
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002201 VN.add(PREInstr, ValNo);
Owen Anderson6a903bc2008-06-18 21:41:49 +00002202 NumGVNPRE++;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002203
Owen Anderson6a903bc2008-06-18 21:41:49 +00002204 // Update the availability map to include the new instruction.
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002205 localAvail[PREPred]->table.insert(std::make_pair(ValNo, PREInstr));
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002206
Owen Anderson6a903bc2008-06-18 21:41:49 +00002207 // Create a PHI to make the value available in this block.
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002208 PHINode* Phi = PHINode::Create(CurInst->getType(),
2209 CurInst->getName() + ".pre-phi",
Owen Anderson6a903bc2008-06-18 21:41:49 +00002210 CurrentBlock->begin());
2211 for (pred_iterator PI = pred_begin(CurrentBlock),
2212 PE = pred_end(CurrentBlock); PI != PE; ++PI)
Owen Anderson1b3ea962008-06-20 01:15:47 +00002213 Phi->addIncoming(predMap[*PI], *PI);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002214
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002215 VN.add(Phi, ValNo);
2216 localAvail[CurrentBlock]->table[ValNo] = Phi;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002217
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002218 CurInst->replaceAllUsesWith(Phi);
Duncan Sands19d0b472010-02-16 11:11:14 +00002219 if (MD && Phi->getType()->isPointerTy())
Chris Lattnerfa9f99a2008-12-09 22:06:23 +00002220 MD->invalidateCachedPointerInfo(Phi);
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002221 VN.erase(CurInst);
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002222
David Greene2e6efc42010-01-05 01:27:17 +00002223 DEBUG(dbgs() << "GVN PRE removed: " << *CurInst << '\n');
Dan Gohman81132462009-11-14 02:27:51 +00002224 if (MD) MD->removeInstruction(CurInst);
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002225 CurInst->eraseFromParent();
Bill Wendlingebb6a542008-12-22 21:57:30 +00002226 DEBUG(verifyRemoved(CurInst));
Chris Lattner6f5bf6a2008-12-01 07:35:54 +00002227 Changed = true;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002228 }
2229 }
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002230
Bob Wilson92cdb6e2010-02-16 19:51:59 +00002231 if (splitCriticalEdges())
2232 Changed = true;
Daniel Dunbar7d6781b2009-09-20 02:20:51 +00002233
Bob Wilson92cdb6e2010-02-16 19:51:59 +00002234 return Changed;
2235}
2236
2237/// splitCriticalEdges - Split critical edges found during the previous
2238/// iteration that may enable further optimization.
2239bool GVN::splitCriticalEdges() {
2240 if (toSplit.empty())
2241 return false;
2242 do {
2243 std::pair<TerminatorInst*, unsigned> Edge = toSplit.pop_back_val();
2244 SplitCriticalEdge(Edge.first, Edge.second, this);
2245 } while (!toSplit.empty());
2246 MD->invalidateCachedPredecessors();
2247 return true;
Owen Anderson6a903bc2008-06-18 21:41:49 +00002248}
2249
Bill Wendling456e8852008-12-22 22:32:22 +00002250/// iterateOnFunction - Executes one iteration of GVN
Owen Anderson676070d2007-08-14 18:04:11 +00002251bool GVN::iterateOnFunction(Function &F) {
Nuno Lopese3127f32008-10-10 16:25:50 +00002252 cleanupGlobalSets();
Chris Lattnerbeb216d2008-03-21 21:33:23 +00002253
Owen Anderson98f912b2009-04-01 23:53:49 +00002254 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2255 DE = df_end(DT->getRootNode()); DI != DE; ++DI) {
2256 if (DI->getIDom())
2257 localAvail[DI->getBlock()] =
2258 new ValueNumberScope(localAvail[DI->getIDom()->getBlock()]);
2259 else
2260 localAvail[DI->getBlock()] = new ValueNumberScope(0);
2261 }
2262
Owen Andersonab6ec2e2007-07-24 17:55:58 +00002263 // Top-down walk of the dominator tree
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002264 bool Changed = false;
Owen Anderson03aacba2008-12-15 03:52:17 +00002265#if 0
2266 // Needed for value numbering with phi construction to work.
Owen Andersonbfe133e2008-12-15 02:03:00 +00002267 ReversePostOrderTraversal<Function*> RPOT(&F);
2268 for (ReversePostOrderTraversal<Function*>::rpo_iterator RI = RPOT.begin(),
2269 RE = RPOT.end(); RI != RE; ++RI)
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002270 Changed |= processBlock(*RI);
Owen Anderson03aacba2008-12-15 03:52:17 +00002271#else
2272 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2273 DE = df_end(DT->getRootNode()); DI != DE; ++DI)
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002274 Changed |= processBlock(DI->getBlock());
Owen Anderson03aacba2008-12-15 03:52:17 +00002275#endif
2276
Chris Lattner1eefa9c2009-09-21 02:42:51 +00002277 return Changed;
Owen Andersonab6ec2e2007-07-24 17:55:58 +00002278}
Nuno Lopese3127f32008-10-10 16:25:50 +00002279
2280void GVN::cleanupGlobalSets() {
2281 VN.clear();
Nuno Lopese3127f32008-10-10 16:25:50 +00002282
2283 for (DenseMap<BasicBlock*, ValueNumberScope*>::iterator
2284 I = localAvail.begin(), E = localAvail.end(); I != E; ++I)
2285 delete I->second;
2286 localAvail.clear();
2287}
Bill Wendling6b18a392008-12-22 21:36:08 +00002288
2289/// verifyRemoved - Verify that the specified instruction does not occur in our
2290/// internal data structures.
Bill Wendlinge7f08e72008-12-22 22:28:56 +00002291void GVN::verifyRemoved(const Instruction *Inst) const {
2292 VN.verifyRemoved(Inst);
Bill Wendling3c793442008-12-22 22:14:07 +00002293
Bill Wendlinge7f08e72008-12-22 22:28:56 +00002294 // Walk through the value number scope to make sure the instruction isn't
2295 // ferreted away in it.
Jeffrey Yasskinb40d3f72009-11-10 01:02:17 +00002296 for (DenseMap<BasicBlock*, ValueNumberScope*>::const_iterator
Bill Wendlinge7f08e72008-12-22 22:28:56 +00002297 I = localAvail.begin(), E = localAvail.end(); I != E; ++I) {
2298 const ValueNumberScope *VNS = I->second;
2299
2300 while (VNS) {
Jeffrey Yasskinb40d3f72009-11-10 01:02:17 +00002301 for (DenseMap<uint32_t, Value*>::const_iterator
Bill Wendlinge7f08e72008-12-22 22:28:56 +00002302 II = VNS->table.begin(), IE = VNS->table.end(); II != IE; ++II) {
2303 assert(II->second != Inst && "Inst still in value numbering scope!");
2304 }
2305
2306 VNS = VNS->parent;
Bill Wendling3c793442008-12-22 22:14:07 +00002307 }
2308 }
Bill Wendling6b18a392008-12-22 21:36:08 +00002309}