blob: d485e07e0769ce4c31eb6c3b9311b679a4ab0a7c [file] [log] [blame]
Chris Lattner159b98f2008-12-05 07:49:08 +00001//===- GVN.cpp - Eliminate redundant values and loads ---------------------===//
Owen Anderson85c40642007-07-24 17:55:58 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner081ce942007-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 Anderson85c40642007-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 Criswell6e0aa282009-03-10 15:04:53 +000013// Note that this pass does the value numbering itself; it does not use the
Matthijs Kooijman9aac1db2008-06-05 07:55:49 +000014// ValueNumbering analysis passes.
15//
Owen Anderson85c40642007-07-24 17:55:58 +000016//===----------------------------------------------------------------------===//
17
18#define DEBUG_TYPE "gvn"
Owen Anderson85c40642007-07-24 17:55:58 +000019#include "llvm/Transforms/Scalar.h"
Owen Anderson5d72a422007-07-25 19:57:03 +000020#include "llvm/BasicBlock.h"
Owen Andersonacfa3ad2007-07-26 18:26:51 +000021#include "llvm/Constants.h"
Owen Anderson85c40642007-07-24 17:55:58 +000022#include "llvm/DerivedTypes.h"
Owen Andersonacfa3ad2007-07-26 18:26:51 +000023#include "llvm/Function.h"
Devang Patela7379552009-03-06 02:59:27 +000024#include "llvm/IntrinsicInst.h"
Owen Anderson24be4c12009-07-03 00:17:18 +000025#include "llvm/LLVMContext.h"
Chris Lattner0907b522009-09-21 05:57:11 +000026#include "llvm/Operator.h"
Owen Andersonacfa3ad2007-07-26 18:26:51 +000027#include "llvm/Value.h"
Owen Anderson85c40642007-07-24 17:55:58 +000028#include "llvm/ADT/DenseMap.h"
29#include "llvm/ADT/DepthFirstIterator.h"
Owen Andersona03e7862008-12-15 02:03:00 +000030#include "llvm/ADT/PostOrderIterator.h"
Owen Anderson85c40642007-07-24 17:55:58 +000031#include "llvm/ADT/SmallPtrSet.h"
32#include "llvm/ADT/SmallVector.h"
33#include "llvm/ADT/Statistic.h"
Owen Anderson5e9366f2007-10-18 19:39:33 +000034#include "llvm/Analysis/Dominators.h"
35#include "llvm/Analysis/AliasAnalysis.h"
Victor Hernandez28f4d2f2009-10-27 20:05:49 +000036#include "llvm/Analysis/MemoryBuiltins.h"
Owen Anderson85c40642007-07-24 17:55:58 +000037#include "llvm/Analysis/MemoryDependenceAnalysis.h"
38#include "llvm/Support/CFG.h"
Owen Andersona2bf7662008-06-19 19:57:25 +000039#include "llvm/Support/CommandLine.h"
Chris Lattner9c5be3c2008-03-29 04:36:18 +000040#include "llvm/Support/Debug.h"
Edwin Török675d5622009-07-11 20:10:48 +000041#include "llvm/Support/ErrorHandling.h"
Chris Lattner0907b522009-09-21 05:57:11 +000042#include "llvm/Support/GetElementPtrTypeIterator.h"
Chris Lattnercb00f732009-12-06 01:57:02 +000043#include "llvm/Support/IRBuilder.h"
Daniel Dunbar005975c2009-07-25 00:23:56 +000044#include "llvm/Support/raw_ostream.h"
Chris Lattner7741aa52009-09-20 19:03:47 +000045#include "llvm/Target/TargetData.h"
Owen Andersonec747c42008-06-19 19:54:19 +000046#include "llvm/Transforms/Utils/BasicBlockUtils.h"
Dale Johannesena19b67f2009-06-17 20:48:23 +000047#include "llvm/Transforms/Utils/Local.h"
Chris Lattner6e5ea272009-10-10 23:50:30 +000048#include "llvm/Transforms/Utils/SSAUpdater.h"
Duncan Sands05f68372008-10-08 07:23:46 +000049#include <cstdio>
Owen Anderson85c40642007-07-24 17:55:58 +000050using namespace llvm;
51
Bill Wendling3858cae2008-12-22 22:14:07 +000052STATISTIC(NumGVNInstr, "Number of instructions deleted");
53STATISTIC(NumGVNLoad, "Number of loads deleted");
54STATISTIC(NumGVNPRE, "Number of instructions PRE'd");
Owen Anderson7558f202008-07-15 16:28:06 +000055STATISTIC(NumGVNBlocks, "Number of blocks merged");
Bill Wendling3858cae2008-12-22 22:14:07 +000056STATISTIC(NumPRELoad, "Number of loads PRE'd");
Chris Lattner1be83222008-03-22 04:13:49 +000057
Evan Cheng019a2e12008-06-20 01:01:07 +000058static cl::opt<bool> EnablePRE("enable-pre",
Owen Anderson3a053612008-07-17 19:41:00 +000059 cl::init(true), cl::Hidden);
Dan Gohman828f89f2009-06-15 18:30:15 +000060static cl::opt<bool> EnableLoadPRE("enable-load-pre", cl::init(true));
Owen Andersona2bf7662008-06-19 19:57:25 +000061
Owen Anderson85c40642007-07-24 17:55:58 +000062//===----------------------------------------------------------------------===//
63// ValueTable Class
64//===----------------------------------------------------------------------===//
65
66/// This class holds the mapping between values and value numbers. It is used
67/// as an efficient mechanism to determine the expression-wise equivalence of
68/// two values.
69namespace {
Chris Lattnerfa2d1ba2009-09-02 06:11:42 +000070 struct Expression {
Dan Gohman7ce405e2009-06-04 22:49:04 +000071 enum ExpressionOpcode { ADD, FADD, SUB, FSUB, MUL, FMUL,
72 UDIV, SDIV, FDIV, UREM, SREM,
Daniel Dunbar3be44e62009-09-20 02:20:51 +000073 FREM, SHL, LSHR, ASHR, AND, OR, XOR, ICMPEQ,
74 ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
75 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
76 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
77 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
Owen Anderson85c40642007-07-24 17:55:58 +000078 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
79 SHUFFLE, SELECT, TRUNC, ZEXT, SEXT, FPTOUI,
Daniel Dunbar3be44e62009-09-20 02:20:51 +000080 FPTOSI, UITOFP, SITOFP, FPTRUNC, FPEXT,
Owen Anderson771d1122008-05-13 08:17:22 +000081 PTRTOINT, INTTOPTR, BITCAST, GEP, CALL, CONSTANT,
Owen Andersona7d4c7c2009-10-19 22:14:22 +000082 INSERTVALUE, EXTRACTVALUE, EMPTY, TOMBSTONE };
Owen Anderson85c40642007-07-24 17:55:58 +000083
84 ExpressionOpcode opcode;
85 const Type* type;
Owen Anderson85c40642007-07-24 17:55:58 +000086 SmallVector<uint32_t, 4> varargs;
Chris Lattnerff36c952009-09-21 02:42:51 +000087 Value *function;
Daniel Dunbar3be44e62009-09-20 02:20:51 +000088
Owen Anderson85c40642007-07-24 17:55:58 +000089 Expression() { }
90 Expression(ExpressionOpcode o) : opcode(o) { }
Daniel Dunbar3be44e62009-09-20 02:20:51 +000091
Owen Anderson85c40642007-07-24 17:55:58 +000092 bool operator==(const Expression &other) const {
93 if (opcode != other.opcode)
94 return false;
95 else if (opcode == EMPTY || opcode == TOMBSTONE)
96 return true;
97 else if (type != other.type)
98 return false;
Owen Anderson5e9366f2007-10-18 19:39:33 +000099 else if (function != other.function)
100 return false;
Owen Anderson85c40642007-07-24 17:55:58 +0000101 else {
102 if (varargs.size() != other.varargs.size())
103 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000104
Owen Anderson85c40642007-07-24 17:55:58 +0000105 for (size_t i = 0; i < varargs.size(); ++i)
106 if (varargs[i] != other.varargs[i])
107 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000108
Owen Anderson85c40642007-07-24 17:55:58 +0000109 return true;
110 }
111 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000112
Owen Anderson85c40642007-07-24 17:55:58 +0000113 bool operator!=(const Expression &other) const {
Bill Wendling9b5d4b72008-12-22 22:16:31 +0000114 return !(*this == other);
Owen Anderson85c40642007-07-24 17:55:58 +0000115 }
116 };
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000117
Chris Lattnerfa2d1ba2009-09-02 06:11:42 +0000118 class ValueTable {
Owen Anderson85c40642007-07-24 17:55:58 +0000119 private:
120 DenseMap<Value*, uint32_t> valueNumbering;
121 DenseMap<Expression, uint32_t> expressionNumbering;
Owen Andersonbcf2bd52008-05-12 20:15:55 +0000122 AliasAnalysis* AA;
123 MemoryDependenceAnalysis* MD;
124 DominatorTree* DT;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000125
Owen Anderson85c40642007-07-24 17:55:58 +0000126 uint32_t nextValueNumber;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000127
Owen Anderson85c40642007-07-24 17:55:58 +0000128 Expression::ExpressionOpcode getOpcode(BinaryOperator* BO);
129 Expression::ExpressionOpcode getOpcode(CmpInst* C);
130 Expression::ExpressionOpcode getOpcode(CastInst* C);
131 Expression create_expression(BinaryOperator* BO);
132 Expression create_expression(CmpInst* C);
133 Expression create_expression(ShuffleVectorInst* V);
134 Expression create_expression(ExtractElementInst* C);
135 Expression create_expression(InsertElementInst* V);
136 Expression create_expression(SelectInst* V);
137 Expression create_expression(CastInst* C);
138 Expression create_expression(GetElementPtrInst* G);
Owen Anderson5e9366f2007-10-18 19:39:33 +0000139 Expression create_expression(CallInst* C);
Owen Anderson771d1122008-05-13 08:17:22 +0000140 Expression create_expression(Constant* C);
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000141 Expression create_expression(ExtractValueInst* C);
142 Expression create_expression(InsertValueInst* C);
143
144 uint32_t lookup_or_add_call(CallInst* C);
Owen Anderson85c40642007-07-24 17:55:58 +0000145 public:
Dan Gohman936a6522009-04-01 16:37:47 +0000146 ValueTable() : nextValueNumber(1) { }
Chris Lattnerff36c952009-09-21 02:42:51 +0000147 uint32_t lookup_or_add(Value *V);
148 uint32_t lookup(Value *V) const;
149 void add(Value *V, uint32_t num);
Owen Anderson85c40642007-07-24 17:55:58 +0000150 void clear();
Chris Lattnerff36c952009-09-21 02:42:51 +0000151 void erase(Value *v);
Owen Anderson85c40642007-07-24 17:55:58 +0000152 unsigned size();
Owen Andersonbcf2bd52008-05-12 20:15:55 +0000153 void setAliasAnalysis(AliasAnalysis* A) { AA = A; }
Chris Lattner02ca4422008-12-01 00:40:32 +0000154 AliasAnalysis *getAliasAnalysis() const { return AA; }
Owen Andersonbcf2bd52008-05-12 20:15:55 +0000155 void setMemDep(MemoryDependenceAnalysis* M) { MD = M; }
156 void setDomTree(DominatorTree* D) { DT = D; }
Owen Anderson8a8d13c2008-07-03 17:44:33 +0000157 uint32_t getNextUnusedValueNumber() { return nextValueNumber; }
Bill Wendling2a023742008-12-22 21:36:08 +0000158 void verifyRemoved(const Value *) const;
Owen Anderson85c40642007-07-24 17:55:58 +0000159 };
160}
161
162namespace llvm {
Chris Lattner92eea072007-09-17 18:34:04 +0000163template <> struct DenseMapInfo<Expression> {
Owen Andersonbf8a3eb2007-08-02 18:16:06 +0000164 static inline Expression getEmptyKey() {
165 return Expression(Expression::EMPTY);
166 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000167
Owen Andersonbf8a3eb2007-08-02 18:16:06 +0000168 static inline Expression getTombstoneKey() {
169 return Expression(Expression::TOMBSTONE);
170 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000171
Owen Anderson85c40642007-07-24 17:55:58 +0000172 static unsigned getHashValue(const Expression e) {
173 unsigned hash = e.opcode;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000174
Anton Korobeynikov8522e1c2008-02-20 11:26:25 +0000175 hash = ((unsigned)((uintptr_t)e.type >> 4) ^
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000176 (unsigned)((uintptr_t)e.type >> 9));
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000177
Owen Andersonbf8a3eb2007-08-02 18:16:06 +0000178 for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
179 E = e.varargs.end(); I != E; ++I)
Owen Anderson85c40642007-07-24 17:55:58 +0000180 hash = *I + hash * 37;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000181
Anton Korobeynikov8522e1c2008-02-20 11:26:25 +0000182 hash = ((unsigned)((uintptr_t)e.function >> 4) ^
183 (unsigned)((uintptr_t)e.function >> 9)) +
184 hash * 37;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000185
Owen Anderson85c40642007-07-24 17:55:58 +0000186 return hash;
187 }
Chris Lattner92eea072007-09-17 18:34:04 +0000188 static bool isEqual(const Expression &LHS, const Expression &RHS) {
189 return LHS == RHS;
190 }
Owen Anderson85c40642007-07-24 17:55:58 +0000191 static bool isPod() { return true; }
192};
193}
194
195//===----------------------------------------------------------------------===//
196// ValueTable Internal Functions
197//===----------------------------------------------------------------------===//
Chris Lattner3d7103e2008-03-21 21:14:38 +0000198Expression::ExpressionOpcode ValueTable::getOpcode(BinaryOperator* BO) {
Owen Anderson85c40642007-07-24 17:55:58 +0000199 switch(BO->getOpcode()) {
Chris Lattner3d7103e2008-03-21 21:14:38 +0000200 default: // THIS SHOULD NEVER HAPPEN
Edwin Törökbd448e32009-07-14 16:55:14 +0000201 llvm_unreachable("Binary operator with unknown opcode?");
Chris Lattner3d7103e2008-03-21 21:14:38 +0000202 case Instruction::Add: return Expression::ADD;
Dan Gohman7ce405e2009-06-04 22:49:04 +0000203 case Instruction::FAdd: return Expression::FADD;
Chris Lattner3d7103e2008-03-21 21:14:38 +0000204 case Instruction::Sub: return Expression::SUB;
Dan Gohman7ce405e2009-06-04 22:49:04 +0000205 case Instruction::FSub: return Expression::FSUB;
Chris Lattner3d7103e2008-03-21 21:14:38 +0000206 case Instruction::Mul: return Expression::MUL;
Dan Gohman7ce405e2009-06-04 22:49:04 +0000207 case Instruction::FMul: return Expression::FMUL;
Chris Lattner3d7103e2008-03-21 21:14:38 +0000208 case Instruction::UDiv: return Expression::UDIV;
209 case Instruction::SDiv: return Expression::SDIV;
210 case Instruction::FDiv: return Expression::FDIV;
211 case Instruction::URem: return Expression::UREM;
212 case Instruction::SRem: return Expression::SREM;
213 case Instruction::FRem: return Expression::FREM;
214 case Instruction::Shl: return Expression::SHL;
215 case Instruction::LShr: return Expression::LSHR;
216 case Instruction::AShr: return Expression::ASHR;
217 case Instruction::And: return Expression::AND;
218 case Instruction::Or: return Expression::OR;
219 case Instruction::Xor: return Expression::XOR;
Owen Anderson85c40642007-07-24 17:55:58 +0000220 }
221}
222
223Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
Nick Lewycky8f5253b2009-07-08 03:04:38 +0000224 if (isa<ICmpInst>(C)) {
Owen Anderson85c40642007-07-24 17:55:58 +0000225 switch (C->getPredicate()) {
Chris Lattner3d7103e2008-03-21 21:14:38 +0000226 default: // THIS SHOULD NEVER HAPPEN
Edwin Törökbd448e32009-07-14 16:55:14 +0000227 llvm_unreachable("Comparison with unknown predicate?");
Chris Lattner3d7103e2008-03-21 21:14:38 +0000228 case ICmpInst::ICMP_EQ: return Expression::ICMPEQ;
229 case ICmpInst::ICMP_NE: return Expression::ICMPNE;
230 case ICmpInst::ICMP_UGT: return Expression::ICMPUGT;
231 case ICmpInst::ICMP_UGE: return Expression::ICMPUGE;
232 case ICmpInst::ICMP_ULT: return Expression::ICMPULT;
233 case ICmpInst::ICMP_ULE: return Expression::ICMPULE;
234 case ICmpInst::ICMP_SGT: return Expression::ICMPSGT;
235 case ICmpInst::ICMP_SGE: return Expression::ICMPSGE;
236 case ICmpInst::ICMP_SLT: return Expression::ICMPSLT;
237 case ICmpInst::ICMP_SLE: return Expression::ICMPSLE;
Owen Anderson85c40642007-07-24 17:55:58 +0000238 }
Nick Lewycky8f5253b2009-07-08 03:04:38 +0000239 } else {
240 switch (C->getPredicate()) {
241 default: // THIS SHOULD NEVER HAPPEN
Edwin Törökbd448e32009-07-14 16:55:14 +0000242 llvm_unreachable("Comparison with unknown predicate?");
Nick Lewycky8f5253b2009-07-08 03:04:38 +0000243 case FCmpInst::FCMP_OEQ: return Expression::FCMPOEQ;
244 case FCmpInst::FCMP_OGT: return Expression::FCMPOGT;
245 case FCmpInst::FCMP_OGE: return Expression::FCMPOGE;
246 case FCmpInst::FCMP_OLT: return Expression::FCMPOLT;
247 case FCmpInst::FCMP_OLE: return Expression::FCMPOLE;
248 case FCmpInst::FCMP_ONE: return Expression::FCMPONE;
249 case FCmpInst::FCMP_ORD: return Expression::FCMPORD;
250 case FCmpInst::FCMP_UNO: return Expression::FCMPUNO;
251 case FCmpInst::FCMP_UEQ: return Expression::FCMPUEQ;
252 case FCmpInst::FCMP_UGT: return Expression::FCMPUGT;
253 case FCmpInst::FCMP_UGE: return Expression::FCMPUGE;
254 case FCmpInst::FCMP_ULT: return Expression::FCMPULT;
255 case FCmpInst::FCMP_ULE: return Expression::FCMPULE;
256 case FCmpInst::FCMP_UNE: return Expression::FCMPUNE;
257 }
Owen Anderson85c40642007-07-24 17:55:58 +0000258 }
259}
260
Chris Lattner3d7103e2008-03-21 21:14:38 +0000261Expression::ExpressionOpcode ValueTable::getOpcode(CastInst* C) {
Owen Anderson85c40642007-07-24 17:55:58 +0000262 switch(C->getOpcode()) {
Chris Lattner3d7103e2008-03-21 21:14:38 +0000263 default: // THIS SHOULD NEVER HAPPEN
Edwin Törökbd448e32009-07-14 16:55:14 +0000264 llvm_unreachable("Cast operator with unknown opcode?");
Chris Lattner3d7103e2008-03-21 21:14:38 +0000265 case Instruction::Trunc: return Expression::TRUNC;
266 case Instruction::ZExt: return Expression::ZEXT;
267 case Instruction::SExt: return Expression::SEXT;
268 case Instruction::FPToUI: return Expression::FPTOUI;
269 case Instruction::FPToSI: return Expression::FPTOSI;
270 case Instruction::UIToFP: return Expression::UITOFP;
271 case Instruction::SIToFP: return Expression::SITOFP;
272 case Instruction::FPTrunc: return Expression::FPTRUNC;
273 case Instruction::FPExt: return Expression::FPEXT;
274 case Instruction::PtrToInt: return Expression::PTRTOINT;
275 case Instruction::IntToPtr: return Expression::INTTOPTR;
276 case Instruction::BitCast: return Expression::BITCAST;
Owen Anderson85c40642007-07-24 17:55:58 +0000277 }
278}
279
Owen Anderson5e9366f2007-10-18 19:39:33 +0000280Expression ValueTable::create_expression(CallInst* C) {
281 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000282
Owen Anderson5e9366f2007-10-18 19:39:33 +0000283 e.type = C->getType();
Owen Anderson5e9366f2007-10-18 19:39:33 +0000284 e.function = C->getCalledFunction();
285 e.opcode = Expression::CALL;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000286
Owen Anderson5e9366f2007-10-18 19:39:33 +0000287 for (CallInst::op_iterator I = C->op_begin()+1, E = C->op_end();
288 I != E; ++I)
Owen Anderson07f478f2008-04-11 05:11:49 +0000289 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000290
Owen Anderson5e9366f2007-10-18 19:39:33 +0000291 return e;
292}
293
Owen Anderson85c40642007-07-24 17:55:58 +0000294Expression ValueTable::create_expression(BinaryOperator* BO) {
295 Expression e;
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000296 e.varargs.push_back(lookup_or_add(BO->getOperand(0)));
297 e.varargs.push_back(lookup_or_add(BO->getOperand(1)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000298 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000299 e.type = BO->getType();
300 e.opcode = getOpcode(BO);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000301
Owen Anderson85c40642007-07-24 17:55:58 +0000302 return e;
303}
304
305Expression ValueTable::create_expression(CmpInst* C) {
306 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000307
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000308 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
309 e.varargs.push_back(lookup_or_add(C->getOperand(1)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000310 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000311 e.type = C->getType();
312 e.opcode = getOpcode(C);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000313
Owen Anderson85c40642007-07-24 17:55:58 +0000314 return e;
315}
316
317Expression ValueTable::create_expression(CastInst* C) {
318 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000319
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000320 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000321 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000322 e.type = C->getType();
323 e.opcode = getOpcode(C);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000324
Owen Anderson85c40642007-07-24 17:55:58 +0000325 return e;
326}
327
328Expression ValueTable::create_expression(ShuffleVectorInst* S) {
329 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000330
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000331 e.varargs.push_back(lookup_or_add(S->getOperand(0)));
332 e.varargs.push_back(lookup_or_add(S->getOperand(1)));
333 e.varargs.push_back(lookup_or_add(S->getOperand(2)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000334 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000335 e.type = S->getType();
336 e.opcode = Expression::SHUFFLE;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000337
Owen Anderson85c40642007-07-24 17:55:58 +0000338 return e;
339}
340
341Expression ValueTable::create_expression(ExtractElementInst* E) {
342 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000343
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000344 e.varargs.push_back(lookup_or_add(E->getOperand(0)));
345 e.varargs.push_back(lookup_or_add(E->getOperand(1)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000346 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000347 e.type = E->getType();
348 e.opcode = Expression::EXTRACT;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000349
Owen Anderson85c40642007-07-24 17:55:58 +0000350 return e;
351}
352
353Expression ValueTable::create_expression(InsertElementInst* I) {
354 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000355
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000356 e.varargs.push_back(lookup_or_add(I->getOperand(0)));
357 e.varargs.push_back(lookup_or_add(I->getOperand(1)));
358 e.varargs.push_back(lookup_or_add(I->getOperand(2)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000359 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000360 e.type = I->getType();
361 e.opcode = Expression::INSERT;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000362
Owen Anderson85c40642007-07-24 17:55:58 +0000363 return e;
364}
365
366Expression ValueTable::create_expression(SelectInst* I) {
367 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000368
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000369 e.varargs.push_back(lookup_or_add(I->getCondition()));
370 e.varargs.push_back(lookup_or_add(I->getTrueValue()));
371 e.varargs.push_back(lookup_or_add(I->getFalseValue()));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000372 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000373 e.type = I->getType();
374 e.opcode = Expression::SELECT;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000375
Owen Anderson85c40642007-07-24 17:55:58 +0000376 return e;
377}
378
379Expression ValueTable::create_expression(GetElementPtrInst* G) {
380 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000381
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000382 e.varargs.push_back(lookup_or_add(G->getPointerOperand()));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000383 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000384 e.type = G->getType();
385 e.opcode = Expression::GEP;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000386
Owen Anderson85c40642007-07-24 17:55:58 +0000387 for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
388 I != E; ++I)
Owen Anderson07f478f2008-04-11 05:11:49 +0000389 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000390
Owen Anderson85c40642007-07-24 17:55:58 +0000391 return e;
392}
393
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000394Expression ValueTable::create_expression(ExtractValueInst* E) {
395 Expression e;
396
397 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
398 for (ExtractValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
399 II != IE; ++II)
400 e.varargs.push_back(*II);
401 e.function = 0;
402 e.type = E->getType();
403 e.opcode = Expression::EXTRACTVALUE;
404
405 return e;
406}
407
408Expression ValueTable::create_expression(InsertValueInst* E) {
409 Expression e;
410
411 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
412 e.varargs.push_back(lookup_or_add(E->getInsertedValueOperand()));
413 for (InsertValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
414 II != IE; ++II)
415 e.varargs.push_back(*II);
416 e.function = 0;
417 e.type = E->getType();
418 e.opcode = Expression::INSERTVALUE;
419
420 return e;
421}
422
Owen Anderson85c40642007-07-24 17:55:58 +0000423//===----------------------------------------------------------------------===//
424// ValueTable External Functions
425//===----------------------------------------------------------------------===//
426
Owen Andersone6b4ff82008-06-18 21:41:49 +0000427/// add - Insert a value into the table with a specified value number.
Chris Lattnerff36c952009-09-21 02:42:51 +0000428void ValueTable::add(Value *V, uint32_t num) {
Owen Andersone6b4ff82008-06-18 21:41:49 +0000429 valueNumbering.insert(std::make_pair(V, num));
430}
431
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000432uint32_t ValueTable::lookup_or_add_call(CallInst* C) {
433 if (AA->doesNotAccessMemory(C)) {
434 Expression exp = create_expression(C);
435 uint32_t& e = expressionNumbering[exp];
436 if (!e) e = nextValueNumber++;
437 valueNumbering[C] = e;
438 return e;
439 } else if (AA->onlyReadsMemory(C)) {
440 Expression exp = create_expression(C);
441 uint32_t& e = expressionNumbering[exp];
442 if (!e) {
443 e = nextValueNumber++;
444 valueNumbering[C] = e;
445 return e;
446 }
Dan Gohmanc8d26652009-11-14 02:27:51 +0000447 if (!MD) {
448 e = nextValueNumber++;
449 valueNumbering[C] = e;
450 return e;
451 }
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000452
453 MemDepResult local_dep = MD->getDependency(C);
454
455 if (!local_dep.isDef() && !local_dep.isNonLocal()) {
456 valueNumbering[C] = nextValueNumber;
457 return nextValueNumber++;
458 }
459
460 if (local_dep.isDef()) {
461 CallInst* local_cdep = cast<CallInst>(local_dep.getInst());
462
463 if (local_cdep->getNumOperands() != C->getNumOperands()) {
464 valueNumbering[C] = nextValueNumber;
465 return nextValueNumber++;
466 }
467
468 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
469 uint32_t c_vn = lookup_or_add(C->getOperand(i));
470 uint32_t cd_vn = lookup_or_add(local_cdep->getOperand(i));
471 if (c_vn != cd_vn) {
472 valueNumbering[C] = nextValueNumber;
473 return nextValueNumber++;
474 }
475 }
476
477 uint32_t v = lookup_or_add(local_cdep);
478 valueNumbering[C] = v;
479 return v;
480 }
481
482 // Non-local case.
483 const MemoryDependenceAnalysis::NonLocalDepInfo &deps =
484 MD->getNonLocalCallDependency(CallSite(C));
485 // FIXME: call/call dependencies for readonly calls should return def, not
486 // clobber! Move the checking logic to MemDep!
487 CallInst* cdep = 0;
488
489 // Check to see if we have a single dominating call instruction that is
490 // identical to C.
491 for (unsigned i = 0, e = deps.size(); i != e; ++i) {
492 const MemoryDependenceAnalysis::NonLocalDepEntry *I = &deps[i];
493 // Ignore non-local dependencies.
494 if (I->second.isNonLocal())
495 continue;
496
497 // We don't handle non-depedencies. If we already have a call, reject
498 // instruction dependencies.
499 if (I->second.isClobber() || cdep != 0) {
500 cdep = 0;
501 break;
502 }
503
504 CallInst *NonLocalDepCall = dyn_cast<CallInst>(I->second.getInst());
505 // FIXME: All duplicated with non-local case.
506 if (NonLocalDepCall && DT->properlyDominates(I->first, C->getParent())){
507 cdep = NonLocalDepCall;
508 continue;
509 }
510
511 cdep = 0;
512 break;
513 }
514
515 if (!cdep) {
516 valueNumbering[C] = nextValueNumber;
517 return nextValueNumber++;
518 }
519
520 if (cdep->getNumOperands() != C->getNumOperands()) {
521 valueNumbering[C] = nextValueNumber;
522 return nextValueNumber++;
523 }
524 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
525 uint32_t c_vn = lookup_or_add(C->getOperand(i));
526 uint32_t cd_vn = lookup_or_add(cdep->getOperand(i));
527 if (c_vn != cd_vn) {
528 valueNumbering[C] = nextValueNumber;
529 return nextValueNumber++;
530 }
531 }
532
533 uint32_t v = lookup_or_add(cdep);
534 valueNumbering[C] = v;
535 return v;
536
537 } else {
538 valueNumbering[C] = nextValueNumber;
539 return nextValueNumber++;
540 }
541}
542
Owen Anderson85c40642007-07-24 17:55:58 +0000543/// lookup_or_add - Returns the value number for the specified value, assigning
544/// it a new number if it did not have one before.
Chris Lattnerff36c952009-09-21 02:42:51 +0000545uint32_t ValueTable::lookup_or_add(Value *V) {
Owen Anderson85c40642007-07-24 17:55:58 +0000546 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
547 if (VI != valueNumbering.end())
548 return VI->second;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000549
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000550 if (!isa<Instruction>(V)) {
Owen Andersonb472cd82009-10-19 21:14:57 +0000551 valueNumbering[V] = nextValueNumber;
Owen Anderson85c40642007-07-24 17:55:58 +0000552 return nextValueNumber++;
553 }
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000554
555 Instruction* I = cast<Instruction>(V);
556 Expression exp;
557 switch (I->getOpcode()) {
558 case Instruction::Call:
559 return lookup_or_add_call(cast<CallInst>(I));
560 case Instruction::Add:
561 case Instruction::FAdd:
562 case Instruction::Sub:
563 case Instruction::FSub:
564 case Instruction::Mul:
565 case Instruction::FMul:
566 case Instruction::UDiv:
567 case Instruction::SDiv:
568 case Instruction::FDiv:
569 case Instruction::URem:
570 case Instruction::SRem:
571 case Instruction::FRem:
572 case Instruction::Shl:
573 case Instruction::LShr:
574 case Instruction::AShr:
575 case Instruction::And:
576 case Instruction::Or :
577 case Instruction::Xor:
578 exp = create_expression(cast<BinaryOperator>(I));
579 break;
580 case Instruction::ICmp:
581 case Instruction::FCmp:
582 exp = create_expression(cast<CmpInst>(I));
583 break;
584 case Instruction::Trunc:
585 case Instruction::ZExt:
586 case Instruction::SExt:
587 case Instruction::FPToUI:
588 case Instruction::FPToSI:
589 case Instruction::UIToFP:
590 case Instruction::SIToFP:
591 case Instruction::FPTrunc:
592 case Instruction::FPExt:
593 case Instruction::PtrToInt:
594 case Instruction::IntToPtr:
595 case Instruction::BitCast:
596 exp = create_expression(cast<CastInst>(I));
597 break;
598 case Instruction::Select:
599 exp = create_expression(cast<SelectInst>(I));
600 break;
601 case Instruction::ExtractElement:
602 exp = create_expression(cast<ExtractElementInst>(I));
603 break;
604 case Instruction::InsertElement:
605 exp = create_expression(cast<InsertElementInst>(I));
606 break;
607 case Instruction::ShuffleVector:
608 exp = create_expression(cast<ShuffleVectorInst>(I));
609 break;
610 case Instruction::ExtractValue:
611 exp = create_expression(cast<ExtractValueInst>(I));
612 break;
613 case Instruction::InsertValue:
614 exp = create_expression(cast<InsertValueInst>(I));
615 break;
616 case Instruction::GetElementPtr:
617 exp = create_expression(cast<GetElementPtrInst>(I));
618 break;
619 default:
620 valueNumbering[V] = nextValueNumber;
621 return nextValueNumber++;
622 }
623
624 uint32_t& e = expressionNumbering[exp];
625 if (!e) e = nextValueNumber++;
626 valueNumbering[V] = e;
627 return e;
Owen Anderson85c40642007-07-24 17:55:58 +0000628}
629
630/// lookup - Returns the value number of the specified value. Fails if
631/// the value has not yet been numbered.
Chris Lattnerff36c952009-09-21 02:42:51 +0000632uint32_t ValueTable::lookup(Value *V) const {
Jeffrey Yasskin8154d2e2009-11-10 01:02:17 +0000633 DenseMap<Value*, uint32_t>::const_iterator VI = valueNumbering.find(V);
Chris Lattner3d7103e2008-03-21 21:14:38 +0000634 assert(VI != valueNumbering.end() && "Value not numbered?");
635 return VI->second;
Owen Anderson85c40642007-07-24 17:55:58 +0000636}
637
638/// clear - Remove all entries from the ValueTable
639void ValueTable::clear() {
640 valueNumbering.clear();
641 expressionNumbering.clear();
642 nextValueNumber = 1;
643}
644
Owen Anderson5aff8002007-07-31 23:27:13 +0000645/// erase - Remove a value from the value numbering
Chris Lattnerff36c952009-09-21 02:42:51 +0000646void ValueTable::erase(Value *V) {
Owen Anderson5aff8002007-07-31 23:27:13 +0000647 valueNumbering.erase(V);
648}
649
Bill Wendling2a023742008-12-22 21:36:08 +0000650/// verifyRemoved - Verify that the value is removed from all internal data
651/// structures.
652void ValueTable::verifyRemoved(const Value *V) const {
Jeffrey Yasskin8154d2e2009-11-10 01:02:17 +0000653 for (DenseMap<Value*, uint32_t>::const_iterator
Bill Wendling2a023742008-12-22 21:36:08 +0000654 I = valueNumbering.begin(), E = valueNumbering.end(); I != E; ++I) {
655 assert(I->first != V && "Inst still occurs in value numbering map!");
656 }
657}
658
Owen Anderson85c40642007-07-24 17:55:58 +0000659//===----------------------------------------------------------------------===//
Bill Wendling42f17f62008-12-22 22:32:22 +0000660// GVN Pass
Owen Anderson85c40642007-07-24 17:55:58 +0000661//===----------------------------------------------------------------------===//
662
663namespace {
Chris Lattnerfa2d1ba2009-09-02 06:11:42 +0000664 struct ValueNumberScope {
Owen Anderson2a412722008-06-20 01:15:47 +0000665 ValueNumberScope* parent;
666 DenseMap<uint32_t, Value*> table;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000667
Owen Anderson2a412722008-06-20 01:15:47 +0000668 ValueNumberScope(ValueNumberScope* p) : parent(p) { }
669 };
670}
671
672namespace {
Owen Anderson85c40642007-07-24 17:55:58 +0000673
Chris Lattnerfa2d1ba2009-09-02 06:11:42 +0000674 class GVN : public FunctionPass {
Owen Anderson85c40642007-07-24 17:55:58 +0000675 bool runOnFunction(Function &F);
676 public:
677 static char ID; // Pass identification, replacement for typeid
Dan Gohmanc8d26652009-11-14 02:27:51 +0000678 explicit GVN(bool nopre = false, bool noloads = false)
679 : FunctionPass(&ID), NoPRE(nopre), NoLoads(noloads), MD(0) { }
Owen Anderson85c40642007-07-24 17:55:58 +0000680
681 private:
Evan Chengf036e552009-10-30 20:12:24 +0000682 bool NoPRE;
Dan Gohmanc8d26652009-11-14 02:27:51 +0000683 bool NoLoads;
Chris Lattner02ca4422008-12-01 00:40:32 +0000684 MemoryDependenceAnalysis *MD;
685 DominatorTree *DT;
686
Owen Anderson85c40642007-07-24 17:55:58 +0000687 ValueTable VN;
Owen Anderson2a412722008-06-20 01:15:47 +0000688 DenseMap<BasicBlock*, ValueNumberScope*> localAvail;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000689
Owen Anderson85c40642007-07-24 17:55:58 +0000690 // This transformation requires dominator postdominator info
691 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Owen Anderson85c40642007-07-24 17:55:58 +0000692 AU.addRequired<DominatorTree>();
Dan Gohmanc8d26652009-11-14 02:27:51 +0000693 if (!NoLoads)
694 AU.addRequired<MemoryDependenceAnalysis>();
Owen Anderson5e9366f2007-10-18 19:39:33 +0000695 AU.addRequired<AliasAnalysis>();
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000696
Owen Andersonaef6a922008-06-23 17:49:45 +0000697 AU.addPreserved<DominatorTree>();
Owen Anderson5e9366f2007-10-18 19:39:33 +0000698 AU.addPreserved<AliasAnalysis>();
Owen Anderson85c40642007-07-24 17:55:58 +0000699 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000700
Owen Anderson85c40642007-07-24 17:55:58 +0000701 // Helper fuctions
702 // FIXME: eliminate or document these better
Owen Anderson85c40642007-07-24 17:55:58 +0000703 bool processLoad(LoadInst* L,
Chris Lattner7de20452008-03-21 22:01:16 +0000704 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerff36c952009-09-21 02:42:51 +0000705 bool processInstruction(Instruction *I,
Chris Lattner7de20452008-03-21 22:01:16 +0000706 SmallVectorImpl<Instruction*> &toErase);
Owen Andersonbf8a3eb2007-08-02 18:16:06 +0000707 bool processNonLocalLoad(LoadInst* L,
Chris Lattner7de20452008-03-21 22:01:16 +0000708 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerff36c952009-09-21 02:42:51 +0000709 bool processBlock(BasicBlock *BB);
Owen Andersone6b4ff82008-06-18 21:41:49 +0000710 void dump(DenseMap<uint32_t, Value*>& d);
Owen Andersonbe168b32007-08-14 18:04:11 +0000711 bool iterateOnFunction(Function &F);
Chris Lattnerff36c952009-09-21 02:42:51 +0000712 Value *CollapsePhi(PHINode* p);
Owen Andersone6b4ff82008-06-18 21:41:49 +0000713 bool performPRE(Function& F);
Chris Lattnerff36c952009-09-21 02:42:51 +0000714 Value *lookupNumber(BasicBlock *BB, uint32_t num);
Nuno Lopes274474b2008-10-10 16:25:50 +0000715 void cleanupGlobalSets();
Bill Wendling2a023742008-12-22 21:36:08 +0000716 void verifyRemoved(const Instruction *I) const;
Owen Anderson85c40642007-07-24 17:55:58 +0000717 };
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000718
Owen Anderson85c40642007-07-24 17:55:58 +0000719 char GVN::ID = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000720}
721
722// createGVNPass - The public interface to this file...
Dan Gohmanc8d26652009-11-14 02:27:51 +0000723FunctionPass *llvm::createGVNPass(bool NoPRE, bool NoLoads) {
724 return new GVN(NoPRE, NoLoads);
725}
Owen Anderson85c40642007-07-24 17:55:58 +0000726
727static RegisterPass<GVN> X("gvn",
728 "Global Value Numbering");
729
Owen Andersone6b4ff82008-06-18 21:41:49 +0000730void GVN::dump(DenseMap<uint32_t, Value*>& d) {
Owen Anderson5d72a422007-07-25 19:57:03 +0000731 printf("{\n");
Owen Andersone6b4ff82008-06-18 21:41:49 +0000732 for (DenseMap<uint32_t, Value*>::iterator I = d.begin(),
Owen Anderson5d72a422007-07-25 19:57:03 +0000733 E = d.end(); I != E; ++I) {
Owen Andersone6b4ff82008-06-18 21:41:49 +0000734 printf("%d\n", I->first);
Owen Anderson5d72a422007-07-25 19:57:03 +0000735 I->second->dump();
736 }
737 printf("}\n");
738}
739
Chris Lattnerff36c952009-09-21 02:42:51 +0000740static bool isSafeReplacement(PHINode* p, Instruction *inst) {
Owen Andersond68b1af2009-08-26 22:55:11 +0000741 if (!isa<PHINode>(inst))
742 return true;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000743
Owen Andersond68b1af2009-08-26 22:55:11 +0000744 for (Instruction::use_iterator UI = p->use_begin(), E = p->use_end();
745 UI != E; ++UI)
746 if (PHINode* use_phi = dyn_cast<PHINode>(UI))
747 if (use_phi->getParent() == inst->getParent())
748 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000749
Owen Andersond68b1af2009-08-26 22:55:11 +0000750 return true;
751}
752
Chris Lattnerff36c952009-09-21 02:42:51 +0000753Value *GVN::CollapsePhi(PHINode *PN) {
754 Value *ConstVal = PN->hasConstantValue(DT);
755 if (!ConstVal) return 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000756
Chris Lattnerff36c952009-09-21 02:42:51 +0000757 Instruction *Inst = dyn_cast<Instruction>(ConstVal);
758 if (!Inst)
759 return ConstVal;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000760
Chris Lattnerff36c952009-09-21 02:42:51 +0000761 if (DT->dominates(Inst, PN))
762 if (isSafeReplacement(PN, Inst))
763 return Inst;
Owen Andersone02ad522007-08-16 22:51:56 +0000764 return 0;
765}
Owen Anderson5d72a422007-07-25 19:57:03 +0000766
Chris Lattnerdcded152008-12-02 08:16:11 +0000767/// IsValueFullyAvailableInBlock - Return true if we can prove that the value
768/// we're analyzing is fully available in the specified block. As we go, keep
Chris Lattner159b98f2008-12-05 07:49:08 +0000769/// track of which blocks we know are fully alive in FullyAvailableBlocks. This
770/// map is actually a tri-state map with the following values:
771/// 0) we know the block *is not* fully available.
772/// 1) we know the block *is* fully available.
773/// 2) we do not know whether the block is fully available or not, but we are
774/// currently speculating that it will be.
775/// 3) we are speculating for this block and have used that to speculate for
776/// other blocks.
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000777static bool IsValueFullyAvailableInBlock(BasicBlock *BB,
Chris Lattner159b98f2008-12-05 07:49:08 +0000778 DenseMap<BasicBlock*, char> &FullyAvailableBlocks) {
Chris Lattnerdcded152008-12-02 08:16:11 +0000779 // Optimistically assume that the block is fully available and check to see
780 // if we already know about this block in one lookup.
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000781 std::pair<DenseMap<BasicBlock*, char>::iterator, char> IV =
Chris Lattner159b98f2008-12-05 07:49:08 +0000782 FullyAvailableBlocks.insert(std::make_pair(BB, 2));
Chris Lattnerdcded152008-12-02 08:16:11 +0000783
784 // If the entry already existed for this block, return the precomputed value.
Chris Lattner159b98f2008-12-05 07:49:08 +0000785 if (!IV.second) {
786 // If this is a speculative "available" value, mark it as being used for
787 // speculation of other blocks.
788 if (IV.first->second == 2)
789 IV.first->second = 3;
790 return IV.first->second != 0;
791 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000792
Chris Lattnerdcded152008-12-02 08:16:11 +0000793 // Otherwise, see if it is fully available in all predecessors.
794 pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000795
Chris Lattnerdcded152008-12-02 08:16:11 +0000796 // If this block has no predecessors, it isn't live-in here.
797 if (PI == PE)
Chris Lattner159b98f2008-12-05 07:49:08 +0000798 goto SpeculationFailure;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000799
Chris Lattnerdcded152008-12-02 08:16:11 +0000800 for (; PI != PE; ++PI)
801 // If the value isn't fully available in one of our predecessors, then it
802 // isn't fully available in this block either. Undo our previous
803 // optimistic assumption and bail out.
804 if (!IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
Chris Lattner159b98f2008-12-05 07:49:08 +0000805 goto SpeculationFailure;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000806
Chris Lattnerdcded152008-12-02 08:16:11 +0000807 return true;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000808
Chris Lattner159b98f2008-12-05 07:49:08 +0000809// SpeculationFailure - If we get here, we found out that this is not, after
810// all, a fully-available block. We have a problem if we speculated on this and
811// used the speculation to mark other blocks as available.
812SpeculationFailure:
813 char &BBVal = FullyAvailableBlocks[BB];
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000814
Chris Lattner159b98f2008-12-05 07:49:08 +0000815 // If we didn't speculate on this, just return with it set to false.
816 if (BBVal == 2) {
817 BBVal = 0;
818 return false;
819 }
820
821 // If we did speculate on this value, we could have blocks set to 1 that are
822 // incorrect. Walk the (transitive) successors of this block and mark them as
823 // 0 if set to one.
824 SmallVector<BasicBlock*, 32> BBWorklist;
825 BBWorklist.push_back(BB);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000826
Chris Lattner159b98f2008-12-05 07:49:08 +0000827 while (!BBWorklist.empty()) {
828 BasicBlock *Entry = BBWorklist.pop_back_val();
829 // Note that this sets blocks to 0 (unavailable) if they happen to not
830 // already be in FullyAvailableBlocks. This is safe.
831 char &EntryVal = FullyAvailableBlocks[Entry];
832 if (EntryVal == 0) continue; // Already unavailable.
833
834 // Mark as unavailable.
835 EntryVal = 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000836
Chris Lattner159b98f2008-12-05 07:49:08 +0000837 for (succ_iterator I = succ_begin(Entry), E = succ_end(Entry); I != E; ++I)
838 BBWorklist.push_back(*I);
839 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000840
Chris Lattner159b98f2008-12-05 07:49:08 +0000841 return false;
Chris Lattnerdcded152008-12-02 08:16:11 +0000842}
843
Chris Lattnerd6b1d052009-09-20 20:09:34 +0000844
Chris Lattner012b3602009-09-21 17:24:04 +0000845/// CanCoerceMustAliasedValueToLoad - Return true if
846/// CoerceAvailableValueToLoadType will succeed.
847static bool CanCoerceMustAliasedValueToLoad(Value *StoredVal,
848 const Type *LoadTy,
849 const TargetData &TD) {
850 // If the loaded or stored value is an first class array or struct, don't try
851 // to transform them. We need to be able to bitcast to integer.
852 if (isa<StructType>(LoadTy) || isa<ArrayType>(LoadTy) ||
853 isa<StructType>(StoredVal->getType()) ||
854 isa<ArrayType>(StoredVal->getType()))
855 return false;
856
857 // The store has to be at least as big as the load.
858 if (TD.getTypeSizeInBits(StoredVal->getType()) <
859 TD.getTypeSizeInBits(LoadTy))
860 return false;
861
862 return true;
863}
864
865
Chris Lattnerd6b1d052009-09-20 20:09:34 +0000866/// CoerceAvailableValueToLoadType - If we saw a store of a value to memory, and
867/// then a load from a must-aliased pointer of a different type, try to coerce
868/// the stored value. LoadedTy is the type of the load we want to replace and
869/// InsertPt is the place to insert new instructions.
870///
871/// If we can't do it, return null.
872static Value *CoerceAvailableValueToLoadType(Value *StoredVal,
873 const Type *LoadedTy,
874 Instruction *InsertPt,
875 const TargetData &TD) {
Chris Lattner012b3602009-09-21 17:24:04 +0000876 if (!CanCoerceMustAliasedValueToLoad(StoredVal, LoadedTy, TD))
877 return 0;
878
Chris Lattnerd6b1d052009-09-20 20:09:34 +0000879 const Type *StoredValTy = StoredVal->getType();
880
881 uint64_t StoreSize = TD.getTypeSizeInBits(StoredValTy);
882 uint64_t LoadSize = TD.getTypeSizeInBits(LoadedTy);
883
884 // If the store and reload are the same size, we can always reuse it.
885 if (StoreSize == LoadSize) {
886 if (isa<PointerType>(StoredValTy) && isa<PointerType>(LoadedTy)) {
887 // Pointer to Pointer -> use bitcast.
888 return new BitCastInst(StoredVal, LoadedTy, "", InsertPt);
889 }
890
891 // Convert source pointers to integers, which can be bitcast.
892 if (isa<PointerType>(StoredValTy)) {
893 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
894 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
895 }
896
897 const Type *TypeToCastTo = LoadedTy;
898 if (isa<PointerType>(TypeToCastTo))
899 TypeToCastTo = TD.getIntPtrType(StoredValTy->getContext());
900
901 if (StoredValTy != TypeToCastTo)
902 StoredVal = new BitCastInst(StoredVal, TypeToCastTo, "", InsertPt);
903
904 // Cast to pointer if the load needs a pointer type.
905 if (isa<PointerType>(LoadedTy))
906 StoredVal = new IntToPtrInst(StoredVal, LoadedTy, "", InsertPt);
907
908 return StoredVal;
909 }
910
911 // If the loaded value is smaller than the available value, then we can
912 // extract out a piece from it. If the available value is too small, then we
913 // can't do anything.
Chris Lattner012b3602009-09-21 17:24:04 +0000914 assert(StoreSize >= LoadSize && "CanCoerceMustAliasedValueToLoad fail");
Chris Lattnerd6b1d052009-09-20 20:09:34 +0000915
916 // Convert source pointers to integers, which can be manipulated.
917 if (isa<PointerType>(StoredValTy)) {
918 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
919 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
920 }
921
922 // Convert vectors and fp to integer, which can be manipulated.
923 if (!isa<IntegerType>(StoredValTy)) {
924 StoredValTy = IntegerType::get(StoredValTy->getContext(), StoreSize);
925 StoredVal = new BitCastInst(StoredVal, StoredValTy, "", InsertPt);
926 }
927
928 // If this is a big-endian system, we need to shift the value down to the low
929 // bits so that a truncate will work.
930 if (TD.isBigEndian()) {
931 Constant *Val = ConstantInt::get(StoredVal->getType(), StoreSize-LoadSize);
932 StoredVal = BinaryOperator::CreateLShr(StoredVal, Val, "tmp", InsertPt);
933 }
934
935 // Truncate the integer to the right size now.
936 const Type *NewIntTy = IntegerType::get(StoredValTy->getContext(), LoadSize);
937 StoredVal = new TruncInst(StoredVal, NewIntTy, "trunc", InsertPt);
938
939 if (LoadedTy == NewIntTy)
940 return StoredVal;
941
942 // If the result is a pointer, inttoptr.
943 if (isa<PointerType>(LoadedTy))
944 return new IntToPtrInst(StoredVal, LoadedTy, "inttoptr", InsertPt);
945
946 // Otherwise, bitcast.
947 return new BitCastInst(StoredVal, LoadedTy, "bitcast", InsertPt);
948}
949
Chris Lattner8f912082009-09-21 06:24:16 +0000950/// GetBaseWithConstantOffset - Analyze the specified pointer to see if it can
951/// be expressed as a base pointer plus a constant offset. Return the base and
952/// offset to the caller.
953static Value *GetBaseWithConstantOffset(Value *Ptr, int64_t &Offset,
Chris Lattneraae7fcb2009-09-21 06:48:08 +0000954 const TargetData &TD) {
Chris Lattner8f912082009-09-21 06:24:16 +0000955 Operator *PtrOp = dyn_cast<Operator>(Ptr);
956 if (PtrOp == 0) return Ptr;
957
958 // Just look through bitcasts.
959 if (PtrOp->getOpcode() == Instruction::BitCast)
960 return GetBaseWithConstantOffset(PtrOp->getOperand(0), Offset, TD);
961
962 // If this is a GEP with constant indices, we can look through it.
963 GEPOperator *GEP = dyn_cast<GEPOperator>(PtrOp);
964 if (GEP == 0 || !GEP->hasAllConstantIndices()) return Ptr;
965
966 gep_type_iterator GTI = gep_type_begin(GEP);
967 for (User::op_iterator I = GEP->idx_begin(), E = GEP->idx_end(); I != E;
968 ++I, ++GTI) {
969 ConstantInt *OpC = cast<ConstantInt>(*I);
970 if (OpC->isZero()) continue;
971
972 // Handle a struct and array indices which add their offset to the pointer.
973 if (const StructType *STy = dyn_cast<StructType>(*GTI)) {
Chris Lattneraae7fcb2009-09-21 06:48:08 +0000974 Offset += TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue());
Chris Lattner8f912082009-09-21 06:24:16 +0000975 } else {
Chris Lattneraae7fcb2009-09-21 06:48:08 +0000976 uint64_t Size = TD.getTypeAllocSize(GTI.getIndexedType());
Chris Lattner8f912082009-09-21 06:24:16 +0000977 Offset += OpC->getSExtValue()*Size;
978 }
979 }
980
981 // Re-sign extend from the pointer size if needed to get overflow edge cases
982 // right.
Chris Lattneraae7fcb2009-09-21 06:48:08 +0000983 unsigned PtrSize = TD.getPointerSizeInBits();
Chris Lattner8f912082009-09-21 06:24:16 +0000984 if (PtrSize < 64)
985 Offset = (Offset << (64-PtrSize)) >> (64-PtrSize);
986
987 return GetBaseWithConstantOffset(GEP->getPointerOperand(), Offset, TD);
988}
989
990
Chris Lattnercb00f732009-12-06 01:57:02 +0000991/// AnalyzeLoadFromClobberingWrite - This function is called when we have a
992/// memdep query of a load that ends up being a clobbering memory write (store,
993/// memset, memcpy, memmove). This means that the write *may* provide bits used
994/// by the load but we can't be sure because the pointers don't mustalias.
995///
996/// Check this case to see if there is anything more we can do before we give
997/// up. This returns -1 if we have to give up, or a byte number in the stored
998/// value of the piece that feeds the load.
999static int AnalyzeLoadFromClobberingWrite(LoadInst *L, Value *WritePtr,
1000 uint64_t WriteSizeInBits,
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001001 const TargetData &TD) {
Chris Lattner012b3602009-09-21 17:24:04 +00001002 // If the loaded or stored value is an first class array or struct, don't try
1003 // to transform them. We need to be able to bitcast to integer.
Chris Lattnercb00f732009-12-06 01:57:02 +00001004 if (isa<StructType>(L->getType()) || isa<ArrayType>(L->getType()))
Chris Lattner012b3602009-09-21 17:24:04 +00001005 return -1;
1006
Chris Lattner8f912082009-09-21 06:24:16 +00001007 int64_t StoreOffset = 0, LoadOffset = 0;
Chris Lattnercb00f732009-12-06 01:57:02 +00001008 Value *StoreBase = GetBaseWithConstantOffset(WritePtr, StoreOffset, TD);
Chris Lattner8f912082009-09-21 06:24:16 +00001009 Value *LoadBase =
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001010 GetBaseWithConstantOffset(L->getPointerOperand(), LoadOffset, TD);
Chris Lattner8f912082009-09-21 06:24:16 +00001011 if (StoreBase != LoadBase)
1012 return -1;
1013
1014 // If the load and store are to the exact same address, they should have been
1015 // a must alias. AA must have gotten confused.
1016 // FIXME: Study to see if/when this happens.
1017 if (LoadOffset == StoreOffset) {
1018#if 0
1019 errs() << "STORE/LOAD DEP WITH COMMON POINTER MISSED:\n"
1020 << "Base = " << *StoreBase << "\n"
Chris Lattnercb00f732009-12-06 01:57:02 +00001021 << "Store Ptr = " << *WritePtr << "\n"
1022 << "Store Offs = " << StoreOffset << "\n"
Chris Lattner8f912082009-09-21 06:24:16 +00001023 << "Load Ptr = " << *L->getPointerOperand() << "\n"
1024 << "Load Offs = " << LoadOffset << " - " << *L << "\n\n";
1025 errs() << "'" << L->getParent()->getParent()->getName() << "'"
1026 << *L->getParent();
1027#endif
1028 return -1;
1029 }
1030
1031 // If the load and store don't overlap at all, the store doesn't provide
1032 // anything to the load. In this case, they really don't alias at all, AA
1033 // must have gotten confused.
1034 // FIXME: Investigate cases where this bails out, e.g. rdar://7238614. Then
1035 // remove this check, as it is duplicated with what we have below.
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001036 uint64_t LoadSize = TD.getTypeSizeInBits(L->getType());
Chris Lattner8f912082009-09-21 06:24:16 +00001037
Chris Lattnercb00f732009-12-06 01:57:02 +00001038 if ((WriteSizeInBits & 7) | (LoadSize & 7))
Chris Lattner8f912082009-09-21 06:24:16 +00001039 return -1;
Chris Lattnercb00f732009-12-06 01:57:02 +00001040 uint64_t StoreSize = WriteSizeInBits >> 3; // Convert to bytes.
Chris Lattner8f912082009-09-21 06:24:16 +00001041 LoadSize >>= 3;
1042
1043
1044 bool isAAFailure = false;
1045 if (StoreOffset < LoadOffset) {
1046 isAAFailure = StoreOffset+int64_t(StoreSize) <= LoadOffset;
1047 } else {
1048 isAAFailure = LoadOffset+int64_t(LoadSize) <= StoreOffset;
1049 }
1050 if (isAAFailure) {
1051#if 0
1052 errs() << "STORE LOAD DEP WITH COMMON BASE:\n"
1053 << "Base = " << *StoreBase << "\n"
Chris Lattnercb00f732009-12-06 01:57:02 +00001054 << "Store Ptr = " << *WritePtr << "\n"
1055 << "Store Offs = " << StoreOffset << "\n"
Chris Lattner8f912082009-09-21 06:24:16 +00001056 << "Load Ptr = " << *L->getPointerOperand() << "\n"
1057 << "Load Offs = " << LoadOffset << " - " << *L << "\n\n";
1058 errs() << "'" << L->getParent()->getParent()->getName() << "'"
1059 << *L->getParent();
1060#endif
1061 return -1;
1062 }
1063
1064 // If the Load isn't completely contained within the stored bits, we don't
1065 // have all the bits to feed it. We could do something crazy in the future
1066 // (issue a smaller load then merge the bits in) but this seems unlikely to be
1067 // valuable.
1068 if (StoreOffset > LoadOffset ||
1069 StoreOffset+StoreSize < LoadOffset+LoadSize)
1070 return -1;
1071
1072 // Okay, we can do this transformation. Return the number of bytes into the
1073 // store that the load is.
1074 return LoadOffset-StoreOffset;
1075}
1076
Chris Lattnercb00f732009-12-06 01:57:02 +00001077/// AnalyzeLoadFromClobberingStore - This function is called when we have a
1078/// memdep query of a load that ends up being a clobbering store.
1079static int AnalyzeLoadFromClobberingStore(LoadInst *L, StoreInst *DepSI,
1080 const TargetData &TD) {
1081 // Cannot handle reading from store of first-class aggregate yet.
1082 if (isa<StructType>(DepSI->getOperand(0)->getType()) ||
1083 isa<ArrayType>(DepSI->getOperand(0)->getType()))
1084 return -1;
1085
1086 Value *StorePtr = DepSI->getPointerOperand();
1087 uint64_t StoreSize = TD.getTypeSizeInBits(StorePtr->getType());
1088 return AnalyzeLoadFromClobberingWrite(L, StorePtr, StoreSize, TD);
1089}
1090
1091static int AnalyzeLoadFromClobberingMemInst(LoadInst *L, MemIntrinsic *MI,
1092 const TargetData &TD) {
1093 // If the mem operation is a non-constant size, we can't handle it.
1094 ConstantInt *SizeCst = dyn_cast<ConstantInt>(MI->getLength());
1095 if (SizeCst == 0) return -1;
1096 uint64_t MemSizeInBits = SizeCst->getZExtValue()*8;
1097
1098 if (MI->getIntrinsicID() == Intrinsic::memset)
1099 return AnalyzeLoadFromClobberingWrite(L, MI->getDest(), MemSizeInBits, TD);
1100
1101 // Unhandled memcpy/memmove.
1102 return -1;
1103}
1104
Chris Lattner8f912082009-09-21 06:24:16 +00001105
1106/// GetStoreValueForLoad - This function is called when we have a
1107/// memdep query of a load that ends up being a clobbering store. This means
1108/// that the store *may* provide bits used by the load but we can't be sure
1109/// because the pointers don't mustalias. Check this case to see if there is
1110/// anything more we can do before we give up.
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001111static Value *GetStoreValueForLoad(Value *SrcVal, unsigned Offset,
1112 const Type *LoadTy,
1113 Instruction *InsertPt, const TargetData &TD){
Chris Lattner8f912082009-09-21 06:24:16 +00001114 LLVMContext &Ctx = SrcVal->getType()->getContext();
1115
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001116 uint64_t StoreSize = TD.getTypeSizeInBits(SrcVal->getType())/8;
1117 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
Chris Lattner8f912082009-09-21 06:24:16 +00001118
1119
1120 // Compute which bits of the stored value are being used by the load. Convert
1121 // to an integer type to start with.
1122 if (isa<PointerType>(SrcVal->getType()))
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001123 SrcVal = new PtrToIntInst(SrcVal, TD.getIntPtrType(Ctx), "tmp", InsertPt);
Chris Lattner8f912082009-09-21 06:24:16 +00001124 if (!isa<IntegerType>(SrcVal->getType()))
1125 SrcVal = new BitCastInst(SrcVal, IntegerType::get(Ctx, StoreSize*8),
1126 "tmp", InsertPt);
1127
1128 // Shift the bits to the least significant depending on endianness.
1129 unsigned ShiftAmt;
Chris Lattnercb00f732009-12-06 01:57:02 +00001130 if (TD.isLittleEndian())
Chris Lattner8f912082009-09-21 06:24:16 +00001131 ShiftAmt = Offset*8;
Chris Lattnercb00f732009-12-06 01:57:02 +00001132 else
Chris Lattner1846fa02009-09-21 17:55:47 +00001133 ShiftAmt = (StoreSize-LoadSize-Offset)*8;
Chris Lattner8f912082009-09-21 06:24:16 +00001134
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001135 if (ShiftAmt)
1136 SrcVal = BinaryOperator::CreateLShr(SrcVal,
1137 ConstantInt::get(SrcVal->getType(), ShiftAmt), "tmp", InsertPt);
Chris Lattner8f912082009-09-21 06:24:16 +00001138
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001139 if (LoadSize != StoreSize)
1140 SrcVal = new TruncInst(SrcVal, IntegerType::get(Ctx, LoadSize*8),
1141 "tmp", InsertPt);
Chris Lattner8f912082009-09-21 06:24:16 +00001142
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001143 return CoerceAvailableValueToLoadType(SrcVal, LoadTy, InsertPt, TD);
Chris Lattner8f912082009-09-21 06:24:16 +00001144}
1145
Chris Lattnercb00f732009-12-06 01:57:02 +00001146/// GetMemInstValueForLoad - This function is called when we have a
1147/// memdep query of a load that ends up being a clobbering mem intrinsic.
1148static Value *GetMemInstValueForLoad(MemIntrinsic *SrcInst, unsigned Offset,
1149 const Type *LoadTy, Instruction *InsertPt,
1150 const TargetData &TD){
1151 LLVMContext &Ctx = LoadTy->getContext();
1152 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
1153
1154 IRBuilder<> Builder(InsertPt->getParent(), InsertPt);
1155
1156 // We know that this method is only called when the mem transfer fully
1157 // provides the bits for the load.
1158 if (MemSetInst *MSI = dyn_cast<MemSetInst>(SrcInst)) {
1159 // memset(P, 'x', 1234) -> splat('x'), even if x is a variable, and
1160 // independently of what the offset is.
1161 Value *Val = MSI->getValue();
1162 if (LoadSize != 1)
1163 Val = Builder.CreateZExt(Val, IntegerType::get(Ctx, LoadSize*8));
1164
1165 Value *OneElt = Val;
1166
1167 // Splat the value out to the right number of bits.
1168 for (unsigned NumBytesSet = 1; NumBytesSet != LoadSize; ) {
1169 // If we can double the number of bytes set, do it.
1170 if (NumBytesSet*2 <= LoadSize) {
1171 Value *ShVal = Builder.CreateShl(Val, NumBytesSet*8);
1172 Val = Builder.CreateOr(Val, ShVal);
1173 NumBytesSet <<= 1;
1174 continue;
1175 }
1176
1177 // Otherwise insert one byte at a time.
1178 Value *ShVal = Builder.CreateShl(Val, 1*8);
1179 Val = Builder.CreateOr(OneElt, ShVal);
1180 ++NumBytesSet;
1181 }
1182
1183 return CoerceAvailableValueToLoadType(Val, LoadTy, InsertPt, TD);
1184 }
1185
1186 // ABORT;
1187 return 0;
1188}
1189
1190
1191
Chris Lattner19b84b32009-09-21 06:30:24 +00001192struct AvailableValueInBlock {
1193 /// BB - The basic block in question.
1194 BasicBlock *BB;
1195 /// V - The value that is live out of the block.
1196 Value *V;
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001197 /// Offset - The byte offset in V that is interesting for the load query.
1198 unsigned Offset;
Chris Lattner19b84b32009-09-21 06:30:24 +00001199
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001200 static AvailableValueInBlock get(BasicBlock *BB, Value *V,
1201 unsigned Offset = 0) {
Chris Lattner19b84b32009-09-21 06:30:24 +00001202 AvailableValueInBlock Res;
1203 Res.BB = BB;
1204 Res.V = V;
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001205 Res.Offset = Offset;
Chris Lattner19b84b32009-09-21 06:30:24 +00001206 return Res;
1207 }
1208};
1209
Chris Lattner6e5ea272009-10-10 23:50:30 +00001210/// ConstructSSAForLoadSet - Given a set of loads specified by ValuesPerBlock,
1211/// construct SSA form, allowing us to eliminate LI. This returns the value
1212/// that should be used at LI's definition site.
1213static Value *ConstructSSAForLoadSet(LoadInst *LI,
1214 SmallVectorImpl<AvailableValueInBlock> &ValuesPerBlock,
1215 const TargetData *TD,
1216 AliasAnalysis *AA) {
1217 SmallVector<PHINode*, 8> NewPHIs;
1218 SSAUpdater SSAUpdate(&NewPHIs);
1219 SSAUpdate.Initialize(LI);
1220
1221 const Type *LoadTy = LI->getType();
1222
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001223 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
Chris Lattner19b84b32009-09-21 06:30:24 +00001224 BasicBlock *BB = ValuesPerBlock[i].BB;
1225 Value *AvailableVal = ValuesPerBlock[i].V;
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001226 unsigned Offset = ValuesPerBlock[i].Offset;
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001227
Chris Lattner6e5ea272009-10-10 23:50:30 +00001228 if (SSAUpdate.HasValueForBlock(BB))
1229 continue;
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001230
1231 if (AvailableVal->getType() != LoadTy) {
1232 assert(TD && "Need target data to handle type mismatch case");
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001233 AvailableVal = GetStoreValueForLoad(AvailableVal, Offset, LoadTy,
1234 BB->getTerminator(), *TD);
1235
1236 if (Offset) {
1237 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\n"
Chris Lattner6e5ea272009-10-10 23:50:30 +00001238 << *ValuesPerBlock[i].V << '\n'
1239 << *AvailableVal << '\n' << "\n\n\n");
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001240 }
1241
1242
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001243 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\n"
Chris Lattner6e5ea272009-10-10 23:50:30 +00001244 << *ValuesPerBlock[i].V << '\n'
1245 << *AvailableVal << '\n' << "\n\n\n");
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001246 }
Chris Lattner6e5ea272009-10-10 23:50:30 +00001247
1248 SSAUpdate.AddAvailableValue(BB, AvailableVal);
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001249 }
Chris Lattner6e5ea272009-10-10 23:50:30 +00001250
1251 // Perform PHI construction.
1252 Value *V = SSAUpdate.GetValueInMiddleOfBlock(LI->getParent());
1253
1254 // If new PHI nodes were created, notify alias analysis.
1255 if (isa<PointerType>(V->getType()))
1256 for (unsigned i = 0, e = NewPHIs.size(); i != e; ++i)
1257 AA->copyValue(LI, NewPHIs[i]);
1258
1259 return V;
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001260}
1261
Owen Andersonf187daf2009-12-02 07:35:19 +00001262static bool isLifetimeStart(Instruction *Inst) {
Chris Lattnerbc6fccc2009-12-02 06:44:58 +00001263 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(Inst))
Owen Andersonf187daf2009-12-02 07:35:19 +00001264 return II->getIntrinsicID() == Intrinsic::lifetime_start;
Chris Lattnerbc6fccc2009-12-02 06:44:58 +00001265 return false;
1266}
1267
Owen Andersone0143452007-08-16 22:02:55 +00001268/// processNonLocalLoad - Attempt to eliminate a load whose dependencies are
1269/// non-local by performing PHI construction.
Chris Lattnerdcded152008-12-02 08:16:11 +00001270bool GVN::processNonLocalLoad(LoadInst *LI,
Chris Lattner7de20452008-03-21 22:01:16 +00001271 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattnerdcded152008-12-02 08:16:11 +00001272 // Find the non-local dependencies of the load.
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001273 SmallVector<MemoryDependenceAnalysis::NonLocalDepEntry, 64> Deps;
Chris Lattneraf713862008-12-09 19:25:07 +00001274 MD->getNonLocalPointerDependency(LI->getOperand(0), true, LI->getParent(),
1275 Deps);
Dan Gohman7e124382009-07-31 20:24:18 +00001276 //DEBUG(errs() << "INVESTIGATING NONLOCAL LOAD: "
1277 // << Deps.size() << *LI << '\n');
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001278
Owen Anderson90e717d2008-08-26 22:07:42 +00001279 // If we had to process more than one hundred blocks to find the
1280 // dependencies, this load isn't worth worrying about. Optimizing
1281 // it will be too expensive.
Chris Lattneraf713862008-12-09 19:25:07 +00001282 if (Deps.size() > 100)
Owen Anderson90e717d2008-08-26 22:07:42 +00001283 return false;
Chris Lattner8d1686f2008-12-18 00:51:32 +00001284
1285 // If we had a phi translation failure, we'll have a single entry which is a
1286 // clobber in the current block. Reject this early.
Edwin Török3ffffac2009-06-17 18:48:18 +00001287 if (Deps.size() == 1 && Deps[0].second.isClobber()) {
1288 DEBUG(
Dan Gohman0be10b02009-07-25 01:43:01 +00001289 errs() << "GVN: non-local load ";
1290 WriteAsOperand(errs(), LI);
Dan Gohman7e124382009-07-31 20:24:18 +00001291 errs() << " is clobbered by " << *Deps[0].second.getInst() << '\n';
Edwin Török3ffffac2009-06-17 18:48:18 +00001292 );
Chris Lattner8d1686f2008-12-18 00:51:32 +00001293 return false;
Edwin Török3ffffac2009-06-17 18:48:18 +00001294 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001295
Chris Lattnerdcded152008-12-02 08:16:11 +00001296 // Filter out useless results (non-locals, etc). Keep track of the blocks
1297 // where we have a value available in repl, also keep track of whether we see
1298 // dependencies that produce an unknown value for the load (such as a call
1299 // that could potentially clobber the load).
Chris Lattner19b84b32009-09-21 06:30:24 +00001300 SmallVector<AvailableValueInBlock, 16> ValuesPerBlock;
Chris Lattnerdcded152008-12-02 08:16:11 +00001301 SmallVector<BasicBlock*, 16> UnavailableBlocks;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001302
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001303 const TargetData *TD = 0;
1304
Chris Lattneraf713862008-12-09 19:25:07 +00001305 for (unsigned i = 0, e = Deps.size(); i != e; ++i) {
1306 BasicBlock *DepBB = Deps[i].first;
1307 MemDepResult DepInfo = Deps[i].second;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001308
Chris Lattner4531da82008-12-05 21:04:20 +00001309 if (DepInfo.isClobber()) {
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001310 // If the dependence is to a store that writes to a superset of the bits
1311 // read by the load, we can extract the bits we need for the load from the
1312 // stored value.
1313 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInfo.getInst())) {
1314 if (TD == 0)
1315 TD = getAnalysisIfAvailable<TargetData>();
1316 if (TD) {
1317 int Offset = AnalyzeLoadFromClobberingStore(LI, DepSI, *TD);
1318 if (Offset != -1) {
1319 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1320 DepSI->getOperand(0),
1321 Offset));
1322 continue;
1323 }
1324 }
1325 }
Chris Lattnercb00f732009-12-06 01:57:02 +00001326
1327#if 0
1328 // If the clobbering value is a memset/memcpy/memmove, see if we can
1329 // forward a value on from it.
1330 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(Dep.getInst())) {
1331 if (TD == 0)
1332 TD = getAnalysisIfAvailable<TargetData>();
1333 if (TD) {
1334 int Offset = AnalyzeLoadFromClobberingMemInst(L, DepMI, *TD);
1335 if (Offset != -1)
1336 AvailVal = GetMemInstValueForLoad(DepMI, Offset, L->getType(), L,*TD);
1337 }
1338 }
1339#endif
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001340
Chris Lattner4531da82008-12-05 21:04:20 +00001341 UnavailableBlocks.push_back(DepBB);
1342 continue;
1343 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001344
Chris Lattner4531da82008-12-05 21:04:20 +00001345 Instruction *DepInst = DepInfo.getInst();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001346
Chris Lattner4531da82008-12-05 21:04:20 +00001347 // Loading the allocation -> undef.
Chris Lattnerbc6fccc2009-12-02 06:44:58 +00001348 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst) ||
Owen Andersonf187daf2009-12-02 07:35:19 +00001349 // Loading immediately after lifetime begin -> undef.
1350 isLifetimeStart(DepInst)) {
Chris Lattner19b84b32009-09-21 06:30:24 +00001351 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1352 UndefValue::get(LI->getType())));
Chris Lattner46876282008-12-01 01:15:42 +00001353 continue;
1354 }
Owen Andersonc07861a2009-10-28 07:05:35 +00001355
Chris Lattner19b84b32009-09-21 06:30:24 +00001356 if (StoreInst *S = dyn_cast<StoreInst>(DepInst)) {
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001357 // Reject loads and stores that are to the same address but are of
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001358 // different types if we have to.
Chris Lattnerdcded152008-12-02 08:16:11 +00001359 if (S->getOperand(0)->getType() != LI->getType()) {
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001360 if (TD == 0)
1361 TD = getAnalysisIfAvailable<TargetData>();
1362
1363 // If the stored value is larger or equal to the loaded value, we can
1364 // reuse it.
Chris Lattner012b3602009-09-21 17:24:04 +00001365 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(S->getOperand(0),
1366 LI->getType(), *TD)) {
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001367 UnavailableBlocks.push_back(DepBB);
1368 continue;
1369 }
Chris Lattnerdcded152008-12-02 08:16:11 +00001370 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001371
Chris Lattner19b84b32009-09-21 06:30:24 +00001372 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1373 S->getOperand(0)));
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001374 continue;
1375 }
1376
1377 if (LoadInst *LD = dyn_cast<LoadInst>(DepInst)) {
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001378 // If the types mismatch and we can't handle it, reject reuse of the load.
Chris Lattnerdcded152008-12-02 08:16:11 +00001379 if (LD->getType() != LI->getType()) {
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001380 if (TD == 0)
1381 TD = getAnalysisIfAvailable<TargetData>();
1382
1383 // If the stored value is larger or equal to the loaded value, we can
1384 // reuse it.
Chris Lattner012b3602009-09-21 17:24:04 +00001385 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(LD, LI->getType(),*TD)){
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001386 UnavailableBlocks.push_back(DepBB);
1387 continue;
1388 }
Chris Lattnerdcded152008-12-02 08:16:11 +00001389 }
Chris Lattner19b84b32009-09-21 06:30:24 +00001390 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB, LD));
Chris Lattnerdcded152008-12-02 08:16:11 +00001391 continue;
Owen Anderson5d72a422007-07-25 19:57:03 +00001392 }
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001393
1394 UnavailableBlocks.push_back(DepBB);
1395 continue;
Chris Lattner3d7103e2008-03-21 21:14:38 +00001396 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001397
Chris Lattnerdcded152008-12-02 08:16:11 +00001398 // If we have no predecessors that produce a known value for this load, exit
1399 // early.
1400 if (ValuesPerBlock.empty()) return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001401
Chris Lattnerdcded152008-12-02 08:16:11 +00001402 // If all of the instructions we depend on produce a known value for this
1403 // load, then it is fully redundant and we can use PHI insertion to compute
1404 // its value. Insert PHIs and remove the fully redundant value now.
1405 if (UnavailableBlocks.empty()) {
Dan Gohman7e124382009-07-31 20:24:18 +00001406 DEBUG(errs() << "GVN REMOVING NONLOCAL LOAD: " << *LI << '\n');
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001407
Chris Lattnerdcded152008-12-02 08:16:11 +00001408 // Perform PHI construction.
Chris Lattner6e5ea272009-10-10 23:50:30 +00001409 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD,
1410 VN.getAliasAnalysis());
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001411 LI->replaceAllUsesWith(V);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001412
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001413 if (isa<PHINode>(V))
1414 V->takeName(LI);
1415 if (isa<PointerType>(V->getType()))
1416 MD->invalidateCachedPointerInfo(V);
Chris Lattnerdcded152008-12-02 08:16:11 +00001417 toErase.push_back(LI);
1418 NumGVNLoad++;
1419 return true;
1420 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001421
Chris Lattnerdcded152008-12-02 08:16:11 +00001422 if (!EnablePRE || !EnableLoadPRE)
1423 return false;
1424
1425 // Okay, we have *some* definitions of the value. This means that the value
1426 // is available in some of our (transitive) predecessors. Lets think about
1427 // doing PRE of this load. This will involve inserting a new load into the
1428 // predecessor when it's not available. We could do this in general, but
1429 // prefer to not increase code size. As such, we only do this when we know
1430 // that we only have to insert *one* load (which means we're basically moving
1431 // the load, not inserting a new one).
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001432
Owen Andersondd37b182009-05-31 09:03:40 +00001433 SmallPtrSet<BasicBlock *, 4> Blockers;
1434 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1435 Blockers.insert(UnavailableBlocks[i]);
1436
1437 // Lets find first basic block with more than one predecessor. Walk backwards
1438 // through predecessors if needed.
Chris Lattnerdcded152008-12-02 08:16:11 +00001439 BasicBlock *LoadBB = LI->getParent();
Owen Andersondd37b182009-05-31 09:03:40 +00001440 BasicBlock *TmpBB = LoadBB;
1441
1442 bool isSinglePred = false;
Dale Johannesena19b67f2009-06-17 20:48:23 +00001443 bool allSingleSucc = true;
Owen Andersondd37b182009-05-31 09:03:40 +00001444 while (TmpBB->getSinglePredecessor()) {
1445 isSinglePred = true;
1446 TmpBB = TmpBB->getSinglePredecessor();
1447 if (!TmpBB) // If haven't found any, bail now.
1448 return false;
1449 if (TmpBB == LoadBB) // Infinite (unreachable) loop.
1450 return false;
1451 if (Blockers.count(TmpBB))
1452 return false;
Dale Johannesena19b67f2009-06-17 20:48:23 +00001453 if (TmpBB->getTerminator()->getNumSuccessors() != 1)
1454 allSingleSucc = false;
Owen Andersondd37b182009-05-31 09:03:40 +00001455 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001456
Owen Andersondd37b182009-05-31 09:03:40 +00001457 assert(TmpBB);
1458 LoadBB = TmpBB;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001459
Chris Lattnerdcded152008-12-02 08:16:11 +00001460 // If we have a repl set with LI itself in it, this means we have a loop where
1461 // at least one of the values is LI. Since this means that we won't be able
1462 // to eliminate LI even if we insert uses in the other predecessors, we will
1463 // end up increasing code size. Reject this by scanning for LI.
1464 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner19b84b32009-09-21 06:30:24 +00001465 if (ValuesPerBlock[i].V == LI)
Chris Lattnerdcded152008-12-02 08:16:11 +00001466 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001467
Owen Andersondd37b182009-05-31 09:03:40 +00001468 if (isSinglePred) {
1469 bool isHot = false;
1470 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner19b84b32009-09-21 06:30:24 +00001471 if (Instruction *I = dyn_cast<Instruction>(ValuesPerBlock[i].V))
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001472 // "Hot" Instruction is in some loop (because it dominates its dep.
1473 // instruction).
1474 if (DT->dominates(LI, I)) {
1475 isHot = true;
1476 break;
1477 }
Owen Andersondd37b182009-05-31 09:03:40 +00001478
1479 // We are interested only in "hot" instructions. We don't want to do any
1480 // mis-optimizations here.
1481 if (!isHot)
1482 return false;
1483 }
1484
Chris Lattnerdcded152008-12-02 08:16:11 +00001485 // Okay, we have some hope :). Check to see if the loaded value is fully
1486 // available in all but one predecessor.
1487 // FIXME: If we could restructure the CFG, we could make a common pred with
1488 // all the preds that don't have an available LI and insert a new load into
1489 // that one block.
1490 BasicBlock *UnavailablePred = 0;
1491
Chris Lattner159b98f2008-12-05 07:49:08 +00001492 DenseMap<BasicBlock*, char> FullyAvailableBlocks;
Chris Lattnerdcded152008-12-02 08:16:11 +00001493 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner19b84b32009-09-21 06:30:24 +00001494 FullyAvailableBlocks[ValuesPerBlock[i].BB] = true;
Chris Lattnerdcded152008-12-02 08:16:11 +00001495 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1496 FullyAvailableBlocks[UnavailableBlocks[i]] = false;
1497
1498 for (pred_iterator PI = pred_begin(LoadBB), E = pred_end(LoadBB);
1499 PI != E; ++PI) {
1500 if (IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
1501 continue;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001502
Chris Lattnerdcded152008-12-02 08:16:11 +00001503 // If this load is not available in multiple predecessors, reject it.
1504 if (UnavailablePred && UnavailablePred != *PI)
1505 return false;
1506 UnavailablePred = *PI;
1507 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001508
Chris Lattnerdcded152008-12-02 08:16:11 +00001509 assert(UnavailablePred != 0 &&
1510 "Fully available value should be eliminated above!");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001511
Chris Lattnerdcded152008-12-02 08:16:11 +00001512 // We don't currently handle critical edges :(
1513 if (UnavailablePred->getTerminator()->getNumSuccessors() != 1) {
Daniel Dunbar005975c2009-07-25 00:23:56 +00001514 DEBUG(errs() << "COULD NOT PRE LOAD BECAUSE OF CRITICAL EDGE '"
Dan Gohman7e124382009-07-31 20:24:18 +00001515 << UnavailablePred->getName() << "': " << *LI << '\n');
Chris Lattnerdcded152008-12-02 08:16:11 +00001516 return false;
Owen Anderson5b299672007-08-07 23:12:31 +00001517 }
Chris Lattner248818e2009-11-27 08:25:10 +00001518
Chris Lattnera5bef152009-11-27 22:05:15 +00001519 // Do PHI translation to get its value in the predecessor if necessary. The
1520 // returned pointer (if non-null) is guaranteed to dominate UnavailablePred.
1521 //
1522 // FIXME: This may insert a computation, but we don't tell scalar GVN
1523 // optimization stuff about it. How do we do this?
Chris Lattner1c2de2b2009-11-28 15:39:14 +00001524 SmallVector<Instruction*, 8> NewInsts;
Chris Lattner80c535b2009-11-28 16:08:18 +00001525 Value *LoadPtr = 0;
Chris Lattnerde0b0302009-11-27 22:50:07 +00001526
Chris Lattner80c535b2009-11-28 16:08:18 +00001527 // If all preds have a single successor, then we know it is safe to insert the
1528 // load on the pred (?!?), so we can insert code to materialize the pointer if
1529 // it is not available.
1530 if (allSingleSucc) {
1531 LoadPtr = MD->InsertPHITranslatedPointer(LI->getOperand(0), LoadBB,
1532 UnavailablePred, TD, *DT,NewInsts);
1533 } else {
1534 LoadPtr = MD->GetAvailablePHITranslatedValue(LI->getOperand(0), LoadBB,
1535 UnavailablePred, TD, *DT);
1536 }
Owen Anderson2c405b92009-12-03 03:43:29 +00001537
1538 // Assign value numbers to these new instructions.
1539 for (SmallVector<Instruction*, 8>::iterator NI = NewInsts.begin(),
1540 NE = NewInsts.end(); NI != NE; ++NI) {
1541 // FIXME: We really _ought_ to insert these value numbers into their
1542 // parent's availability map. However, in doing so, we risk getting into
1543 // ordering issues. If a block hasn't been processed yet, we would be
1544 // marking a value as AVAIL-IN, which isn't what we intend.
1545 VN.lookup_or_add(*NI);
1546 }
Chris Lattner80c535b2009-11-28 16:08:18 +00001547
Chris Lattnera5bef152009-11-27 22:05:15 +00001548 // If we couldn't find or insert a computation of this phi translated value,
1549 // we fail PRE.
Chris Lattner248818e2009-11-27 08:25:10 +00001550 if (LoadPtr == 0) {
Chris Lattnera5bef152009-11-27 22:05:15 +00001551 DEBUG(errs() << "COULDN'T INSERT PHI TRANSLATED VALUE OF: "
1552 << *LI->getOperand(0) << "\n");
1553 return false;
Chris Lattner248818e2009-11-27 08:25:10 +00001554 }
1555
Dale Johannesena19b67f2009-06-17 20:48:23 +00001556 // Make sure it is valid to move this load here. We have to watch out for:
1557 // @1 = getelementptr (i8* p, ...
1558 // test p and branch if == 0
1559 // load @1
1560 // It is valid to have the getelementptr before the test, even if p can be 0,
1561 // as getelementptr only does address arithmetic.
1562 // If we are not pushing the value through any multiple-successor blocks
1563 // we do not have this case. Otherwise, check that the load is safe to
1564 // put anywhere; this can be improved, but should be conservatively safe.
1565 if (!allSingleSucc &&
Chris Lattner1c2de2b2009-11-28 15:39:14 +00001566 // FIXME: REEVALUTE THIS.
Chris Lattner80c535b2009-11-28 16:08:18 +00001567 !isSafeToLoadUnconditionally(LoadPtr, UnavailablePred->getTerminator())) {
1568 assert(NewInsts.empty() && "Should not have inserted instructions");
Dale Johannesena19b67f2009-06-17 20:48:23 +00001569 return false;
Chris Lattner80c535b2009-11-28 16:08:18 +00001570 }
Dale Johannesena19b67f2009-06-17 20:48:23 +00001571
Chris Lattnerdcded152008-12-02 08:16:11 +00001572 // Okay, we can eliminate this load by inserting a reload in the predecessor
1573 // and using PHI construction to get the value in the other predecessors, do
1574 // it.
Dan Gohman7e124382009-07-31 20:24:18 +00001575 DEBUG(errs() << "GVN REMOVING PRE LOAD: " << *LI << '\n');
Chris Lattner80c535b2009-11-28 16:08:18 +00001576 DEBUG(if (!NewInsts.empty())
1577 errs() << "INSERTED " << NewInsts.size() << " INSTS: "
1578 << *NewInsts.back() << '\n');
1579
Chris Lattnerdcded152008-12-02 08:16:11 +00001580 Value *NewLoad = new LoadInst(LoadPtr, LI->getName()+".pre", false,
1581 LI->getAlignment(),
1582 UnavailablePred->getTerminator());
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001583
Chris Lattner6e5ea272009-10-10 23:50:30 +00001584 // Add the newly created load.
1585 ValuesPerBlock.push_back(AvailableValueInBlock::get(UnavailablePred,NewLoad));
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001586
Chris Lattnerdcded152008-12-02 08:16:11 +00001587 // Perform PHI construction.
Chris Lattner6e5ea272009-10-10 23:50:30 +00001588 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD,
1589 VN.getAliasAnalysis());
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001590 LI->replaceAllUsesWith(V);
1591 if (isa<PHINode>(V))
1592 V->takeName(LI);
1593 if (isa<PointerType>(V->getType()))
1594 MD->invalidateCachedPointerInfo(V);
Chris Lattnerdcded152008-12-02 08:16:11 +00001595 toErase.push_back(LI);
1596 NumPRELoad++;
Owen Anderson5d72a422007-07-25 19:57:03 +00001597 return true;
1598}
1599
Owen Andersone0143452007-08-16 22:02:55 +00001600/// processLoad - Attempt to eliminate a load, first by eliminating it
1601/// locally, and then attempting non-local elimination if that fails.
Chris Lattner4531da82008-12-05 21:04:20 +00001602bool GVN::processLoad(LoadInst *L, SmallVectorImpl<Instruction*> &toErase) {
Dan Gohmanc8d26652009-11-14 02:27:51 +00001603 if (!MD)
1604 return false;
1605
Chris Lattner4531da82008-12-05 21:04:20 +00001606 if (L->isVolatile())
Owen Anderson85c40642007-07-24 17:55:58 +00001607 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001608
Owen Anderson85c40642007-07-24 17:55:58 +00001609 // ... to a pointer that has been loaded from before...
Chris Lattnerff36c952009-09-21 02:42:51 +00001610 MemDepResult Dep = MD->getDependency(L);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001611
Chris Lattner4531da82008-12-05 21:04:20 +00001612 // If the value isn't available, don't do anything!
Chris Lattnerff36c952009-09-21 02:42:51 +00001613 if (Dep.isClobber()) {
Chris Lattner0907b522009-09-21 05:57:11 +00001614 // Check to see if we have something like this:
Chris Lattner7741aa52009-09-20 19:03:47 +00001615 // store i32 123, i32* %P
1616 // %A = bitcast i32* %P to i8*
1617 // %B = gep i8* %A, i32 1
1618 // %C = load i8* %B
1619 //
1620 // We could do that by recognizing if the clobber instructions are obviously
1621 // a common base + constant offset, and if the previous store (or memset)
1622 // completely covers this load. This sort of thing can happen in bitfield
1623 // access code.
Chris Lattnercb00f732009-12-06 01:57:02 +00001624 Value *AvailVal = 0;
Chris Lattner0907b522009-09-21 05:57:11 +00001625 if (StoreInst *DepSI = dyn_cast<StoreInst>(Dep.getInst()))
Chris Lattner41eb59c2009-09-21 06:22:46 +00001626 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001627 int Offset = AnalyzeLoadFromClobberingStore(L, DepSI, *TD);
Chris Lattnercb00f732009-12-06 01:57:02 +00001628 if (Offset != -1)
1629 AvailVal = GetStoreValueForLoad(DepSI->getOperand(0), Offset,
1630 L->getType(), L, *TD);
Chris Lattner41eb59c2009-09-21 06:22:46 +00001631 }
Chris Lattner0907b522009-09-21 05:57:11 +00001632
Chris Lattnercb00f732009-12-06 01:57:02 +00001633 // If the clobbering value is a memset/memcpy/memmove, see if we can forward
1634 // a value on from it.
1635 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(Dep.getInst())) {
1636 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
1637 int Offset = AnalyzeLoadFromClobberingMemInst(L, DepMI, *TD);
1638 if (Offset != -1)
1639 AvailVal = GetMemInstValueForLoad(DepMI, Offset, L->getType(), L,*TD);
1640 }
1641 }
1642
1643 if (AvailVal) {
1644 DEBUG(errs() << "GVN COERCED INST:\n" << *Dep.getInst() << '\n'
1645 << *AvailVal << '\n' << *L << "\n\n\n");
1646
1647 // Replace the load!
1648 L->replaceAllUsesWith(AvailVal);
1649 if (isa<PointerType>(AvailVal->getType()))
1650 MD->invalidateCachedPointerInfo(AvailVal);
1651 toErase.push_back(L);
1652 NumGVNLoad++;
1653 return true;
1654 }
1655
Edwin Török47cf8842009-05-29 09:46:03 +00001656 DEBUG(
1657 // fast print dep, using operator<< on instruction would be too slow
Dan Gohman0be10b02009-07-25 01:43:01 +00001658 errs() << "GVN: load ";
1659 WriteAsOperand(errs(), L);
Chris Lattnerff36c952009-09-21 02:42:51 +00001660 Instruction *I = Dep.getInst();
Dan Gohman7e124382009-07-31 20:24:18 +00001661 errs() << " is clobbered by " << *I << '\n';
Edwin Török47cf8842009-05-29 09:46:03 +00001662 );
Chris Lattner4531da82008-12-05 21:04:20 +00001663 return false;
Edwin Török47cf8842009-05-29 09:46:03 +00001664 }
Chris Lattner4531da82008-12-05 21:04:20 +00001665
1666 // If it is defined in another block, try harder.
Chris Lattnerff36c952009-09-21 02:42:51 +00001667 if (Dep.isNonLocal())
Chris Lattner4531da82008-12-05 21:04:20 +00001668 return processNonLocalLoad(L, toErase);
Eli Friedman350307f2008-02-12 12:08:14 +00001669
Chris Lattnerff36c952009-09-21 02:42:51 +00001670 Instruction *DepInst = Dep.getInst();
Chris Lattner4531da82008-12-05 21:04:20 +00001671 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInst)) {
Chris Lattner7741aa52009-09-20 19:03:47 +00001672 Value *StoredVal = DepSI->getOperand(0);
1673
1674 // The store and load are to a must-aliased pointer, but they may not
1675 // actually have the same type. See if we know how to reuse the stored
1676 // value (depending on its type).
1677 const TargetData *TD = 0;
Chris Lattner10460aa2009-10-21 04:11:19 +00001678 if (StoredVal->getType() != L->getType()) {
1679 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1680 StoredVal = CoerceAvailableValueToLoadType(StoredVal, L->getType(),
1681 L, *TD);
1682 if (StoredVal == 0)
1683 return false;
1684
1685 DEBUG(errs() << "GVN COERCED STORE:\n" << *DepSI << '\n' << *StoredVal
1686 << '\n' << *L << "\n\n\n");
1687 }
1688 else
Chris Lattner7741aa52009-09-20 19:03:47 +00001689 return false;
Chris Lattner7741aa52009-09-20 19:03:47 +00001690 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001691
Chris Lattner4531da82008-12-05 21:04:20 +00001692 // Remove it!
Chris Lattner7741aa52009-09-20 19:03:47 +00001693 L->replaceAllUsesWith(StoredVal);
1694 if (isa<PointerType>(StoredVal->getType()))
1695 MD->invalidateCachedPointerInfo(StoredVal);
Chris Lattner4531da82008-12-05 21:04:20 +00001696 toErase.push_back(L);
1697 NumGVNLoad++;
1698 return true;
1699 }
1700
1701 if (LoadInst *DepLI = dyn_cast<LoadInst>(DepInst)) {
Chris Lattner7741aa52009-09-20 19:03:47 +00001702 Value *AvailableVal = DepLI;
1703
1704 // The loads are of a must-aliased pointer, but they may not actually have
1705 // the same type. See if we know how to reuse the previously loaded value
1706 // (depending on its type).
1707 const TargetData *TD = 0;
Chris Lattner10460aa2009-10-21 04:11:19 +00001708 if (DepLI->getType() != L->getType()) {
1709 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1710 AvailableVal = CoerceAvailableValueToLoadType(DepLI, L->getType(), L,*TD);
1711 if (AvailableVal == 0)
1712 return false;
Chris Lattner7741aa52009-09-20 19:03:47 +00001713
Chris Lattner10460aa2009-10-21 04:11:19 +00001714 DEBUG(errs() << "GVN COERCED LOAD:\n" << *DepLI << "\n" << *AvailableVal
1715 << "\n" << *L << "\n\n\n");
1716 }
1717 else
1718 return false;
Chris Lattner7741aa52009-09-20 19:03:47 +00001719 }
1720
Chris Lattner4531da82008-12-05 21:04:20 +00001721 // Remove it!
Chris Lattner7741aa52009-09-20 19:03:47 +00001722 L->replaceAllUsesWith(AvailableVal);
Chris Lattnerf81b0142008-12-09 22:06:23 +00001723 if (isa<PointerType>(DepLI->getType()))
1724 MD->invalidateCachedPointerInfo(DepLI);
Chris Lattner4531da82008-12-05 21:04:20 +00001725 toErase.push_back(L);
1726 NumGVNLoad++;
1727 return true;
1728 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001729
Chris Lattner8ea60462008-11-30 01:39:32 +00001730 // If this load really doesn't depend on anything, then we must be loading an
1731 // undef value. This can happen when loading for a fresh allocation with no
1732 // intervening stores, for example.
Victor Hernandezb1687302009-10-23 21:09:37 +00001733 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst)) {
Owen Andersonb99ecca2009-07-30 23:03:37 +00001734 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Chris Lattner8ea60462008-11-30 01:39:32 +00001735 toErase.push_back(L);
Chris Lattner8ea60462008-11-30 01:39:32 +00001736 NumGVNLoad++;
Chris Lattner4531da82008-12-05 21:04:20 +00001737 return true;
Eli Friedman350307f2008-02-12 12:08:14 +00001738 }
Owen Andersonc07861a2009-10-28 07:05:35 +00001739
Owen Andersonf187daf2009-12-02 07:35:19 +00001740 // If this load occurs either right after a lifetime begin,
Owen Andersonc07861a2009-10-28 07:05:35 +00001741 // then the loaded value is undefined.
1742 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(DepInst)) {
Owen Andersonf187daf2009-12-02 07:35:19 +00001743 if (II->getIntrinsicID() == Intrinsic::lifetime_start) {
Owen Andersonc07861a2009-10-28 07:05:35 +00001744 L->replaceAllUsesWith(UndefValue::get(L->getType()));
1745 toErase.push_back(L);
1746 NumGVNLoad++;
1747 return true;
1748 }
1749 }
Eli Friedman350307f2008-02-12 12:08:14 +00001750
Chris Lattner4531da82008-12-05 21:04:20 +00001751 return false;
Owen Anderson85c40642007-07-24 17:55:58 +00001752}
1753
Chris Lattnerff36c952009-09-21 02:42:51 +00001754Value *GVN::lookupNumber(BasicBlock *BB, uint32_t num) {
Owen Andersonaef6a922008-06-23 17:49:45 +00001755 DenseMap<BasicBlock*, ValueNumberScope*>::iterator I = localAvail.find(BB);
1756 if (I == localAvail.end())
1757 return 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001758
Chris Lattnerff36c952009-09-21 02:42:51 +00001759 ValueNumberScope *Locals = I->second;
1760 while (Locals) {
1761 DenseMap<uint32_t, Value*>::iterator I = Locals->table.find(num);
1762 if (I != Locals->table.end())
Owen Anderson2a412722008-06-20 01:15:47 +00001763 return I->second;
Chris Lattnerff36c952009-09-21 02:42:51 +00001764 Locals = Locals->parent;
Owen Anderson2a412722008-06-20 01:15:47 +00001765 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001766
Owen Anderson2a412722008-06-20 01:15:47 +00001767 return 0;
1768}
1769
Owen Andersona03e7862008-12-15 02:03:00 +00001770
Owen Andersonf631bb62007-08-14 18:16:29 +00001771/// processInstruction - When calculating availability, handle an instruction
Owen Anderson85c40642007-07-24 17:55:58 +00001772/// by inserting it into the appropriate sets
Owen Anderson9334fc62008-06-12 19:25:32 +00001773bool GVN::processInstruction(Instruction *I,
Chris Lattner7de20452008-03-21 22:01:16 +00001774 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001775 if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
1776 bool Changed = processLoad(LI, toErase);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001777
Chris Lattnerff36c952009-09-21 02:42:51 +00001778 if (!Changed) {
1779 unsigned Num = VN.lookup_or_add(LI);
1780 localAvail[I->getParent()]->table.insert(std::make_pair(Num, LI));
Owen Andersone6b4ff82008-06-18 21:41:49 +00001781 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001782
Chris Lattnerff36c952009-09-21 02:42:51 +00001783 return Changed;
Owen Andersone6b4ff82008-06-18 21:41:49 +00001784 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001785
Chris Lattnerff36c952009-09-21 02:42:51 +00001786 uint32_t NextNum = VN.getNextUnusedValueNumber();
1787 unsigned Num = VN.lookup_or_add(I);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001788
Chris Lattnerff36c952009-09-21 02:42:51 +00001789 if (BranchInst *BI = dyn_cast<BranchInst>(I)) {
1790 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001791
Owen Andersonef8bf0f2009-04-01 23:53:49 +00001792 if (!BI->isConditional() || isa<Constant>(BI->getCondition()))
1793 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001794
Chris Lattnerff36c952009-09-21 02:42:51 +00001795 Value *BranchCond = BI->getCondition();
1796 uint32_t CondVN = VN.lookup_or_add(BranchCond);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001797
Chris Lattnerff36c952009-09-21 02:42:51 +00001798 BasicBlock *TrueSucc = BI->getSuccessor(0);
1799 BasicBlock *FalseSucc = BI->getSuccessor(1);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001800
Chris Lattnerff36c952009-09-21 02:42:51 +00001801 if (TrueSucc->getSinglePredecessor())
1802 localAvail[TrueSucc]->table[CondVN] =
1803 ConstantInt::getTrue(TrueSucc->getContext());
1804 if (FalseSucc->getSinglePredecessor())
1805 localAvail[FalseSucc]->table[CondVN] =
1806 ConstantInt::getFalse(TrueSucc->getContext());
Owen Andersonef8bf0f2009-04-01 23:53:49 +00001807
1808 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001809
Owen Andersonced50f82008-04-07 09:59:07 +00001810 // Allocations are always uniquely numbered, so we can save time and memory
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001811 // by fast failing them.
Victor Hernandezb1687302009-10-23 21:09:37 +00001812 } else if (isa<AllocaInst>(I) || isa<TerminatorInst>(I)) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001813 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Andersonced50f82008-04-07 09:59:07 +00001814 return false;
Owen Andersone6b4ff82008-06-18 21:41:49 +00001815 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001816
Owen Andersone0143452007-08-16 22:02:55 +00001817 // Collapse PHI nodes
Owen Anderson98f6a6b2007-08-14 18:33:27 +00001818 if (PHINode* p = dyn_cast<PHINode>(I)) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001819 Value *constVal = CollapsePhi(p);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001820
Owen Anderson98f6a6b2007-08-14 18:33:27 +00001821 if (constVal) {
Owen Andersone02ad522007-08-16 22:51:56 +00001822 p->replaceAllUsesWith(constVal);
Dan Gohmanc8d26652009-11-14 02:27:51 +00001823 if (MD && isa<PointerType>(constVal->getType()))
Chris Lattnerf81b0142008-12-09 22:06:23 +00001824 MD->invalidateCachedPointerInfo(constVal);
Owen Anderson575f2812008-12-23 00:49:51 +00001825 VN.erase(p);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001826
Owen Andersone02ad522007-08-16 22:51:56 +00001827 toErase.push_back(p);
Owen Andersone6b4ff82008-06-18 21:41:49 +00001828 } else {
Chris Lattnerff36c952009-09-21 02:42:51 +00001829 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson98f6a6b2007-08-14 18:33:27 +00001830 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001831
Owen Anderson8a8d13c2008-07-03 17:44:33 +00001832 // If the number we were assigned was a brand new VN, then we don't
1833 // need to do a lookup to see if the number already exists
1834 // somewhere in the domtree: it can't!
Chris Lattnerff36c952009-09-21 02:42:51 +00001835 } else if (Num == NextNum) {
1836 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001837
Owen Andersona03e7862008-12-15 02:03:00 +00001838 // Perform fast-path value-number based elimination of values inherited from
1839 // dominators.
Chris Lattnerff36c952009-09-21 02:42:51 +00001840 } else if (Value *repl = lookupNumber(I->getParent(), Num)) {
Owen Andersonc772be72007-12-08 01:37:09 +00001841 // Remove it!
Owen Anderson5aff8002007-07-31 23:27:13 +00001842 VN.erase(I);
Owen Anderson85c40642007-07-24 17:55:58 +00001843 I->replaceAllUsesWith(repl);
Dan Gohmanc8d26652009-11-14 02:27:51 +00001844 if (MD && isa<PointerType>(repl->getType()))
Chris Lattnerf81b0142008-12-09 22:06:23 +00001845 MD->invalidateCachedPointerInfo(repl);
Owen Anderson85c40642007-07-24 17:55:58 +00001846 toErase.push_back(I);
1847 return true;
Owen Andersona03e7862008-12-15 02:03:00 +00001848
Owen Anderson8a8d13c2008-07-03 17:44:33 +00001849 } else {
Chris Lattnerff36c952009-09-21 02:42:51 +00001850 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson85c40642007-07-24 17:55:58 +00001851 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001852
Owen Anderson85c40642007-07-24 17:55:58 +00001853 return false;
1854}
1855
Bill Wendling42f17f62008-12-22 22:32:22 +00001856/// runOnFunction - This is the main transformation entry point for a function.
Owen Andersonbe168b32007-08-14 18:04:11 +00001857bool GVN::runOnFunction(Function& F) {
Dan Gohmanc8d26652009-11-14 02:27:51 +00001858 if (!NoLoads)
1859 MD = &getAnalysis<MemoryDependenceAnalysis>();
Chris Lattner02ca4422008-12-01 00:40:32 +00001860 DT = &getAnalysis<DominatorTree>();
Owen Andersonbcf2bd52008-05-12 20:15:55 +00001861 VN.setAliasAnalysis(&getAnalysis<AliasAnalysis>());
Chris Lattner02ca4422008-12-01 00:40:32 +00001862 VN.setMemDep(MD);
1863 VN.setDomTree(DT);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001864
Chris Lattnerff36c952009-09-21 02:42:51 +00001865 bool Changed = false;
1866 bool ShouldContinue = true;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001867
Owen Anderson26ed2572008-07-16 17:52:31 +00001868 // Merge unconditional branches, allowing PRE to catch more
1869 // optimization opportunities.
1870 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001871 BasicBlock *BB = FI;
Owen Anderson26ed2572008-07-16 17:52:31 +00001872 ++FI;
Owen Andersonf59eef82008-07-17 00:01:40 +00001873 bool removedBlock = MergeBlockIntoPredecessor(BB, this);
1874 if (removedBlock) NumGVNBlocks++;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001875
Chris Lattnerff36c952009-09-21 02:42:51 +00001876 Changed |= removedBlock;
Owen Anderson26ed2572008-07-16 17:52:31 +00001877 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001878
Chris Lattner4bab29b2008-12-09 19:21:47 +00001879 unsigned Iteration = 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001880
Chris Lattnerff36c952009-09-21 02:42:51 +00001881 while (ShouldContinue) {
Dan Gohman0be10b02009-07-25 01:43:01 +00001882 DEBUG(errs() << "GVN iteration: " << Iteration << "\n");
Chris Lattnerff36c952009-09-21 02:42:51 +00001883 ShouldContinue = iterateOnFunction(F);
1884 Changed |= ShouldContinue;
Chris Lattner4bab29b2008-12-09 19:21:47 +00001885 ++Iteration;
Owen Andersonbe168b32007-08-14 18:04:11 +00001886 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001887
Owen Anderson916f4732008-07-18 18:03:38 +00001888 if (EnablePRE) {
Owen Anderson9c935902008-09-03 23:06:07 +00001889 bool PREChanged = true;
1890 while (PREChanged) {
1891 PREChanged = performPRE(F);
Chris Lattnerff36c952009-09-21 02:42:51 +00001892 Changed |= PREChanged;
Owen Anderson9c935902008-09-03 23:06:07 +00001893 }
Owen Anderson916f4732008-07-18 18:03:38 +00001894 }
Chris Lattner4bab29b2008-12-09 19:21:47 +00001895 // FIXME: Should perform GVN again after PRE does something. PRE can move
1896 // computations into blocks where they become fully redundant. Note that
1897 // we can't do this until PRE's critical edge splitting updates memdep.
1898 // Actually, when this happens, we should just fully integrate PRE into GVN.
Nuno Lopes274474b2008-10-10 16:25:50 +00001899
1900 cleanupGlobalSets();
1901
Chris Lattnerff36c952009-09-21 02:42:51 +00001902 return Changed;
Owen Andersonbe168b32007-08-14 18:04:11 +00001903}
1904
1905
Chris Lattnerff36c952009-09-21 02:42:51 +00001906bool GVN::processBlock(BasicBlock *BB) {
Chris Lattner4bab29b2008-12-09 19:21:47 +00001907 // FIXME: Kill off toErase by doing erasing eagerly in a helper function (and
1908 // incrementing BI before processing an instruction).
Owen Anderson9334fc62008-06-12 19:25:32 +00001909 SmallVector<Instruction*, 8> toErase;
Chris Lattnerff36c952009-09-21 02:42:51 +00001910 bool ChangedFunction = false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001911
Owen Anderson9334fc62008-06-12 19:25:32 +00001912 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
1913 BI != BE;) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001914 ChangedFunction |= processInstruction(BI, toErase);
Owen Anderson9334fc62008-06-12 19:25:32 +00001915 if (toErase.empty()) {
1916 ++BI;
1917 continue;
1918 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001919
Owen Anderson9334fc62008-06-12 19:25:32 +00001920 // If we need some instructions deleted, do it now.
1921 NumGVNInstr += toErase.size();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001922
Owen Anderson9334fc62008-06-12 19:25:32 +00001923 // Avoid iterator invalidation.
1924 bool AtStart = BI == BB->begin();
1925 if (!AtStart)
1926 --BI;
1927
1928 for (SmallVector<Instruction*, 4>::iterator I = toErase.begin(),
Chris Lattner02ca4422008-12-01 00:40:32 +00001929 E = toErase.end(); I != E; ++I) {
Dan Gohman7e124382009-07-31 20:24:18 +00001930 DEBUG(errs() << "GVN removed: " << **I << '\n');
Dan Gohmanc8d26652009-11-14 02:27:51 +00001931 if (MD) MD->removeInstruction(*I);
Owen Anderson9334fc62008-06-12 19:25:32 +00001932 (*I)->eraseFromParent();
Bill Wendling84049422008-12-22 21:57:30 +00001933 DEBUG(verifyRemoved(*I));
Chris Lattner02ca4422008-12-01 00:40:32 +00001934 }
Chris Lattner4bab29b2008-12-09 19:21:47 +00001935 toErase.clear();
Owen Anderson9334fc62008-06-12 19:25:32 +00001936
1937 if (AtStart)
1938 BI = BB->begin();
1939 else
1940 ++BI;
Owen Anderson9334fc62008-06-12 19:25:32 +00001941 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001942
Chris Lattnerff36c952009-09-21 02:42:51 +00001943 return ChangedFunction;
Owen Anderson9334fc62008-06-12 19:25:32 +00001944}
1945
Owen Andersone6b4ff82008-06-18 21:41:49 +00001946/// performPRE - Perform a purely local form of PRE that looks for diamond
1947/// control flow patterns and attempts to perform simple PRE at the join point.
Chris Lattner4790cb42009-10-31 22:11:15 +00001948bool GVN::performPRE(Function &F) {
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001949 bool Changed = false;
Owen Andersonec747c42008-06-19 19:54:19 +00001950 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> toSplit;
Chris Lattner3304b562008-12-01 07:29:03 +00001951 DenseMap<BasicBlock*, Value*> predMap;
Owen Andersone6b4ff82008-06-18 21:41:49 +00001952 for (df_iterator<BasicBlock*> DI = df_begin(&F.getEntryBlock()),
1953 DE = df_end(&F.getEntryBlock()); DI != DE; ++DI) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001954 BasicBlock *CurrentBlock = *DI;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001955
Owen Andersone6b4ff82008-06-18 21:41:49 +00001956 // Nothing to PRE in the entry block.
1957 if (CurrentBlock == &F.getEntryBlock()) continue;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001958
Owen Andersone6b4ff82008-06-18 21:41:49 +00001959 for (BasicBlock::iterator BI = CurrentBlock->begin(),
1960 BE = CurrentBlock->end(); BI != BE; ) {
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001961 Instruction *CurInst = BI++;
Duncan Sands2f500832009-05-06 06:49:50 +00001962
Victor Hernandezb1687302009-10-23 21:09:37 +00001963 if (isa<AllocaInst>(CurInst) ||
Victor Hernandez48c3c542009-09-18 22:35:49 +00001964 isa<TerminatorInst>(CurInst) || isa<PHINode>(CurInst) ||
Devang Patele9d08b82009-10-14 17:29:00 +00001965 CurInst->getType()->isVoidTy() ||
Duncan Sands2f500832009-05-06 06:49:50 +00001966 CurInst->mayReadFromMemory() || CurInst->mayHaveSideEffects() ||
John Criswell6e0aa282009-03-10 15:04:53 +00001967 isa<DbgInfoIntrinsic>(CurInst))
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001968 continue;
Duncan Sands2f500832009-05-06 06:49:50 +00001969
Chris Lattnerff36c952009-09-21 02:42:51 +00001970 uint32_t ValNo = VN.lookup(CurInst);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001971
Owen Andersone6b4ff82008-06-18 21:41:49 +00001972 // Look for the predecessors for PRE opportunities. We're
1973 // only trying to solve the basic diamond case, where
1974 // a value is computed in the successor and one predecessor,
1975 // but not the other. We also explicitly disallow cases
1976 // where the successor is its own predecessor, because they're
1977 // more complicated to get right.
Chris Lattnerff36c952009-09-21 02:42:51 +00001978 unsigned NumWith = 0;
1979 unsigned NumWithout = 0;
1980 BasicBlock *PREPred = 0;
Chris Lattner3304b562008-12-01 07:29:03 +00001981 predMap.clear();
1982
Owen Andersone6b4ff82008-06-18 21:41:49 +00001983 for (pred_iterator PI = pred_begin(CurrentBlock),
1984 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
1985 // We're not interested in PRE where the block is its
Owen Anderson2a412722008-06-20 01:15:47 +00001986 // own predecessor, on in blocks with predecessors
1987 // that are not reachable.
1988 if (*PI == CurrentBlock) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001989 NumWithout = 2;
Owen Anderson2a412722008-06-20 01:15:47 +00001990 break;
1991 } else if (!localAvail.count(*PI)) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001992 NumWithout = 2;
Owen Anderson2a412722008-06-20 01:15:47 +00001993 break;
1994 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001995
1996 DenseMap<uint32_t, Value*>::iterator predV =
Chris Lattnerff36c952009-09-21 02:42:51 +00001997 localAvail[*PI]->table.find(ValNo);
Owen Anderson2a412722008-06-20 01:15:47 +00001998 if (predV == localAvail[*PI]->table.end()) {
Owen Andersone6b4ff82008-06-18 21:41:49 +00001999 PREPred = *PI;
Chris Lattnerff36c952009-09-21 02:42:51 +00002000 NumWithout++;
Chris Lattner66a3a3e2008-12-01 07:35:54 +00002001 } else if (predV->second == CurInst) {
Chris Lattnerff36c952009-09-21 02:42:51 +00002002 NumWithout = 2;
Owen Andersone6b4ff82008-06-18 21:41:49 +00002003 } else {
Owen Anderson2a412722008-06-20 01:15:47 +00002004 predMap[*PI] = predV->second;
Chris Lattnerff36c952009-09-21 02:42:51 +00002005 NumWith++;
Owen Andersone6b4ff82008-06-18 21:41:49 +00002006 }
2007 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002008
Owen Andersone6b4ff82008-06-18 21:41:49 +00002009 // Don't do PRE when it might increase code size, i.e. when
2010 // we would need to insert instructions in more than one pred.
Chris Lattnerff36c952009-09-21 02:42:51 +00002011 if (NumWithout != 1 || NumWith == 0)
Owen Andersone6b4ff82008-06-18 21:41:49 +00002012 continue;
Chris Lattner4790cb42009-10-31 22:11:15 +00002013
2014 // Don't do PRE across indirect branch.
2015 if (isa<IndirectBrInst>(PREPred->getTerminator()))
2016 continue;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002017
Owen Andersonec747c42008-06-19 19:54:19 +00002018 // We can't do PRE safely on a critical edge, so instead we schedule
2019 // the edge to be split and perform the PRE the next time we iterate
2020 // on the function.
Chris Lattnerff36c952009-09-21 02:42:51 +00002021 unsigned SuccNum = 0;
Owen Andersonec747c42008-06-19 19:54:19 +00002022 for (unsigned i = 0, e = PREPred->getTerminator()->getNumSuccessors();
2023 i != e; ++i)
Owen Anderson9c935902008-09-03 23:06:07 +00002024 if (PREPred->getTerminator()->getSuccessor(i) == CurrentBlock) {
Chris Lattnerff36c952009-09-21 02:42:51 +00002025 SuccNum = i;
Owen Andersonec747c42008-06-19 19:54:19 +00002026 break;
2027 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002028
Chris Lattnerff36c952009-09-21 02:42:51 +00002029 if (isCriticalEdge(PREPred->getTerminator(), SuccNum)) {
2030 toSplit.push_back(std::make_pair(PREPred->getTerminator(), SuccNum));
Owen Andersonec747c42008-06-19 19:54:19 +00002031 continue;
2032 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002033
Owen Andersone6b4ff82008-06-18 21:41:49 +00002034 // Instantiate the expression the in predecessor that lacked it.
2035 // Because we are going top-down through the block, all value numbers
2036 // will be available in the predecessor by the time we need them. Any
2037 // that weren't original present will have been instantiated earlier
2038 // in this loop.
Nick Lewyckyc94270c2009-09-27 07:38:41 +00002039 Instruction *PREInstr = CurInst->clone();
Owen Andersone6b4ff82008-06-18 21:41:49 +00002040 bool success = true;
Chris Lattner66a3a3e2008-12-01 07:35:54 +00002041 for (unsigned i = 0, e = CurInst->getNumOperands(); i != e; ++i) {
2042 Value *Op = PREInstr->getOperand(i);
2043 if (isa<Argument>(Op) || isa<Constant>(Op) || isa<GlobalValue>(Op))
2044 continue;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002045
Chris Lattner66a3a3e2008-12-01 07:35:54 +00002046 if (Value *V = lookupNumber(PREPred, VN.lookup(Op))) {
2047 PREInstr->setOperand(i, V);
2048 } else {
2049 success = false;
2050 break;
Owen Anderson14c612f2008-07-11 20:05:13 +00002051 }
Owen Andersone6b4ff82008-06-18 21:41:49 +00002052 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002053
Owen Andersone6b4ff82008-06-18 21:41:49 +00002054 // Fail out if we encounter an operand that is not available in
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002055 // the PRE predecessor. This is typically because of loads which
Owen Andersone6b4ff82008-06-18 21:41:49 +00002056 // are not value numbered precisely.
2057 if (!success) {
2058 delete PREInstr;
Bill Wendling3858cae2008-12-22 22:14:07 +00002059 DEBUG(verifyRemoved(PREInstr));
Owen Andersone6b4ff82008-06-18 21:41:49 +00002060 continue;
2061 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002062
Owen Andersone6b4ff82008-06-18 21:41:49 +00002063 PREInstr->insertBefore(PREPred->getTerminator());
Chris Lattner66a3a3e2008-12-01 07:35:54 +00002064 PREInstr->setName(CurInst->getName() + ".pre");
Owen Anderson2a412722008-06-20 01:15:47 +00002065 predMap[PREPred] = PREInstr;
Chris Lattnerff36c952009-09-21 02:42:51 +00002066 VN.add(PREInstr, ValNo);
Owen Andersone6b4ff82008-06-18 21:41:49 +00002067 NumGVNPRE++;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002068
Owen Andersone6b4ff82008-06-18 21:41:49 +00002069 // Update the availability map to include the new instruction.
Chris Lattnerff36c952009-09-21 02:42:51 +00002070 localAvail[PREPred]->table.insert(std::make_pair(ValNo, PREInstr));
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002071
Owen Andersone6b4ff82008-06-18 21:41:49 +00002072 // Create a PHI to make the value available in this block.
Chris Lattner66a3a3e2008-12-01 07:35:54 +00002073 PHINode* Phi = PHINode::Create(CurInst->getType(),
2074 CurInst->getName() + ".pre-phi",
Owen Andersone6b4ff82008-06-18 21:41:49 +00002075 CurrentBlock->begin());
2076 for (pred_iterator PI = pred_begin(CurrentBlock),
2077 PE = pred_end(CurrentBlock); PI != PE; ++PI)
Owen Anderson2a412722008-06-20 01:15:47 +00002078 Phi->addIncoming(predMap[*PI], *PI);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002079
Chris Lattnerff36c952009-09-21 02:42:51 +00002080 VN.add(Phi, ValNo);
2081 localAvail[CurrentBlock]->table[ValNo] = Phi;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002082
Chris Lattner66a3a3e2008-12-01 07:35:54 +00002083 CurInst->replaceAllUsesWith(Phi);
Dan Gohmanc8d26652009-11-14 02:27:51 +00002084 if (MD && isa<PointerType>(Phi->getType()))
Chris Lattnerf81b0142008-12-09 22:06:23 +00002085 MD->invalidateCachedPointerInfo(Phi);
Chris Lattner66a3a3e2008-12-01 07:35:54 +00002086 VN.erase(CurInst);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002087
Dan Gohman7e124382009-07-31 20:24:18 +00002088 DEBUG(errs() << "GVN PRE removed: " << *CurInst << '\n');
Dan Gohmanc8d26652009-11-14 02:27:51 +00002089 if (MD) MD->removeInstruction(CurInst);
Chris Lattner66a3a3e2008-12-01 07:35:54 +00002090 CurInst->eraseFromParent();
Bill Wendling84049422008-12-22 21:57:30 +00002091 DEBUG(verifyRemoved(CurInst));
Chris Lattner66a3a3e2008-12-01 07:35:54 +00002092 Changed = true;
Owen Andersone6b4ff82008-06-18 21:41:49 +00002093 }
2094 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002095
Owen Andersonec747c42008-06-19 19:54:19 +00002096 for (SmallVector<std::pair<TerminatorInst*, unsigned>, 4>::iterator
Anton Korobeynikov2e8710c2008-12-05 19:38:49 +00002097 I = toSplit.begin(), E = toSplit.end(); I != E; ++I)
Owen Andersonec747c42008-06-19 19:54:19 +00002098 SplitCriticalEdge(I->first, I->second, this);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002099
Anton Korobeynikov2e8710c2008-12-05 19:38:49 +00002100 return Changed || toSplit.size();
Owen Andersone6b4ff82008-06-18 21:41:49 +00002101}
2102
Bill Wendling42f17f62008-12-22 22:32:22 +00002103/// iterateOnFunction - Executes one iteration of GVN
Owen Andersonbe168b32007-08-14 18:04:11 +00002104bool GVN::iterateOnFunction(Function &F) {
Nuno Lopes274474b2008-10-10 16:25:50 +00002105 cleanupGlobalSets();
Chris Lattner98054902008-03-21 21:33:23 +00002106
Owen Andersonef8bf0f2009-04-01 23:53:49 +00002107 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2108 DE = df_end(DT->getRootNode()); DI != DE; ++DI) {
2109 if (DI->getIDom())
2110 localAvail[DI->getBlock()] =
2111 new ValueNumberScope(localAvail[DI->getIDom()->getBlock()]);
2112 else
2113 localAvail[DI->getBlock()] = new ValueNumberScope(0);
2114 }
2115
Owen Anderson85c40642007-07-24 17:55:58 +00002116 // Top-down walk of the dominator tree
Chris Lattnerff36c952009-09-21 02:42:51 +00002117 bool Changed = false;
Owen Andersonef136f52008-12-15 03:52:17 +00002118#if 0
2119 // Needed for value numbering with phi construction to work.
Owen Andersona03e7862008-12-15 02:03:00 +00002120 ReversePostOrderTraversal<Function*> RPOT(&F);
2121 for (ReversePostOrderTraversal<Function*>::rpo_iterator RI = RPOT.begin(),
2122 RE = RPOT.end(); RI != RE; ++RI)
Chris Lattnerff36c952009-09-21 02:42:51 +00002123 Changed |= processBlock(*RI);
Owen Andersonef136f52008-12-15 03:52:17 +00002124#else
2125 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2126 DE = df_end(DT->getRootNode()); DI != DE; ++DI)
Chris Lattnerff36c952009-09-21 02:42:51 +00002127 Changed |= processBlock(DI->getBlock());
Owen Andersonef136f52008-12-15 03:52:17 +00002128#endif
2129
Chris Lattnerff36c952009-09-21 02:42:51 +00002130 return Changed;
Owen Anderson85c40642007-07-24 17:55:58 +00002131}
Nuno Lopes274474b2008-10-10 16:25:50 +00002132
2133void GVN::cleanupGlobalSets() {
2134 VN.clear();
Nuno Lopes274474b2008-10-10 16:25:50 +00002135
2136 for (DenseMap<BasicBlock*, ValueNumberScope*>::iterator
2137 I = localAvail.begin(), E = localAvail.end(); I != E; ++I)
2138 delete I->second;
2139 localAvail.clear();
2140}
Bill Wendling2a023742008-12-22 21:36:08 +00002141
2142/// verifyRemoved - Verify that the specified instruction does not occur in our
2143/// internal data structures.
Bill Wendlingf9c0e9e2008-12-22 22:28:56 +00002144void GVN::verifyRemoved(const Instruction *Inst) const {
2145 VN.verifyRemoved(Inst);
Bill Wendling3858cae2008-12-22 22:14:07 +00002146
Bill Wendlingf9c0e9e2008-12-22 22:28:56 +00002147 // Walk through the value number scope to make sure the instruction isn't
2148 // ferreted away in it.
Jeffrey Yasskin8154d2e2009-11-10 01:02:17 +00002149 for (DenseMap<BasicBlock*, ValueNumberScope*>::const_iterator
Bill Wendlingf9c0e9e2008-12-22 22:28:56 +00002150 I = localAvail.begin(), E = localAvail.end(); I != E; ++I) {
2151 const ValueNumberScope *VNS = I->second;
2152
2153 while (VNS) {
Jeffrey Yasskin8154d2e2009-11-10 01:02:17 +00002154 for (DenseMap<uint32_t, Value*>::const_iterator
Bill Wendlingf9c0e9e2008-12-22 22:28:56 +00002155 II = VNS->table.begin(), IE = VNS->table.end(); II != IE; ++II) {
2156 assert(II->second != Inst && "Inst still in value numbering scope!");
2157 }
2158
2159 VNS = VNS->parent;
Bill Wendling3858cae2008-12-22 22:14:07 +00002160 }
2161 }
Bill Wendling2a023742008-12-22 21:36:08 +00002162}