blob: cbe7add582adba6fe578b7f25252451211714bc8 [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"
Daniel Dunbar005975c2009-07-25 00:23:56 +000043#include "llvm/Support/raw_ostream.h"
Chris Lattner7741aa52009-09-20 19:03:47 +000044#include "llvm/Target/TargetData.h"
Owen Andersonec747c42008-06-19 19:54:19 +000045#include "llvm/Transforms/Utils/BasicBlockUtils.h"
Dale Johannesena19b67f2009-06-17 20:48:23 +000046#include "llvm/Transforms/Utils/Local.h"
Chris Lattner6e5ea272009-10-10 23:50:30 +000047#include "llvm/Transforms/Utils/SSAUpdater.h"
Duncan Sands05f68372008-10-08 07:23:46 +000048#include <cstdio>
Owen Anderson85c40642007-07-24 17:55:58 +000049using namespace llvm;
50
Bill Wendling3858cae2008-12-22 22:14:07 +000051STATISTIC(NumGVNInstr, "Number of instructions deleted");
52STATISTIC(NumGVNLoad, "Number of loads deleted");
53STATISTIC(NumGVNPRE, "Number of instructions PRE'd");
Owen Anderson7558f202008-07-15 16:28:06 +000054STATISTIC(NumGVNBlocks, "Number of blocks merged");
Bill Wendling3858cae2008-12-22 22:14:07 +000055STATISTIC(NumPRELoad, "Number of loads PRE'd");
Chris Lattner1be83222008-03-22 04:13:49 +000056
Evan Cheng019a2e12008-06-20 01:01:07 +000057static cl::opt<bool> EnablePRE("enable-pre",
Owen Anderson3a053612008-07-17 19:41:00 +000058 cl::init(true), cl::Hidden);
Dan Gohman828f89f2009-06-15 18:30:15 +000059static cl::opt<bool> EnableLoadPRE("enable-load-pre", cl::init(true));
Owen Andersona2bf7662008-06-19 19:57:25 +000060
Owen Anderson85c40642007-07-24 17:55:58 +000061//===----------------------------------------------------------------------===//
62// ValueTable Class
63//===----------------------------------------------------------------------===//
64
65/// This class holds the mapping between values and value numbers. It is used
66/// as an efficient mechanism to determine the expression-wise equivalence of
67/// two values.
68namespace {
Chris Lattnerfa2d1ba2009-09-02 06:11:42 +000069 struct Expression {
Dan Gohman7ce405e2009-06-04 22:49:04 +000070 enum ExpressionOpcode { ADD, FADD, SUB, FSUB, MUL, FMUL,
71 UDIV, SDIV, FDIV, UREM, SREM,
Daniel Dunbar3be44e62009-09-20 02:20:51 +000072 FREM, SHL, LSHR, ASHR, AND, OR, XOR, ICMPEQ,
73 ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
74 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
75 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
76 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
Owen Anderson85c40642007-07-24 17:55:58 +000077 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
78 SHUFFLE, SELECT, TRUNC, ZEXT, SEXT, FPTOUI,
Daniel Dunbar3be44e62009-09-20 02:20:51 +000079 FPTOSI, UITOFP, SITOFP, FPTRUNC, FPEXT,
Owen Anderson771d1122008-05-13 08:17:22 +000080 PTRTOINT, INTTOPTR, BITCAST, GEP, CALL, CONSTANT,
Owen Andersona7d4c7c2009-10-19 22:14:22 +000081 INSERTVALUE, EXTRACTVALUE, EMPTY, TOMBSTONE };
Owen Anderson85c40642007-07-24 17:55:58 +000082
83 ExpressionOpcode opcode;
84 const Type* type;
Owen Anderson85c40642007-07-24 17:55:58 +000085 SmallVector<uint32_t, 4> varargs;
Chris Lattnerff36c952009-09-21 02:42:51 +000086 Value *function;
Daniel Dunbar3be44e62009-09-20 02:20:51 +000087
Owen Anderson85c40642007-07-24 17:55:58 +000088 Expression() { }
89 Expression(ExpressionOpcode o) : opcode(o) { }
Daniel Dunbar3be44e62009-09-20 02:20:51 +000090
Owen Anderson85c40642007-07-24 17:55:58 +000091 bool operator==(const Expression &other) const {
92 if (opcode != other.opcode)
93 return false;
94 else if (opcode == EMPTY || opcode == TOMBSTONE)
95 return true;
96 else if (type != other.type)
97 return false;
Owen Anderson5e9366f2007-10-18 19:39:33 +000098 else if (function != other.function)
99 return false;
Owen Anderson85c40642007-07-24 17:55:58 +0000100 else {
101 if (varargs.size() != other.varargs.size())
102 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000103
Owen Anderson85c40642007-07-24 17:55:58 +0000104 for (size_t i = 0; i < varargs.size(); ++i)
105 if (varargs[i] != other.varargs[i])
106 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000107
Owen Anderson85c40642007-07-24 17:55:58 +0000108 return true;
109 }
110 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000111
Owen Anderson85c40642007-07-24 17:55:58 +0000112 bool operator!=(const Expression &other) const {
Bill Wendling9b5d4b72008-12-22 22:16:31 +0000113 return !(*this == other);
Owen Anderson85c40642007-07-24 17:55:58 +0000114 }
115 };
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000116
Chris Lattnerfa2d1ba2009-09-02 06:11:42 +0000117 class ValueTable {
Owen Anderson85c40642007-07-24 17:55:58 +0000118 private:
119 DenseMap<Value*, uint32_t> valueNumbering;
120 DenseMap<Expression, uint32_t> expressionNumbering;
Owen Andersonbcf2bd52008-05-12 20:15:55 +0000121 AliasAnalysis* AA;
122 MemoryDependenceAnalysis* MD;
123 DominatorTree* DT;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000124
Owen Anderson85c40642007-07-24 17:55:58 +0000125 uint32_t nextValueNumber;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000126
Owen Anderson85c40642007-07-24 17:55:58 +0000127 Expression::ExpressionOpcode getOpcode(BinaryOperator* BO);
128 Expression::ExpressionOpcode getOpcode(CmpInst* C);
129 Expression::ExpressionOpcode getOpcode(CastInst* C);
130 Expression create_expression(BinaryOperator* BO);
131 Expression create_expression(CmpInst* C);
132 Expression create_expression(ShuffleVectorInst* V);
133 Expression create_expression(ExtractElementInst* C);
134 Expression create_expression(InsertElementInst* V);
135 Expression create_expression(SelectInst* V);
136 Expression create_expression(CastInst* C);
137 Expression create_expression(GetElementPtrInst* G);
Owen Anderson5e9366f2007-10-18 19:39:33 +0000138 Expression create_expression(CallInst* C);
Owen Anderson771d1122008-05-13 08:17:22 +0000139 Expression create_expression(Constant* C);
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000140 Expression create_expression(ExtractValueInst* C);
141 Expression create_expression(InsertValueInst* C);
142
143 uint32_t lookup_or_add_call(CallInst* C);
Owen Anderson85c40642007-07-24 17:55:58 +0000144 public:
Dan Gohman936a6522009-04-01 16:37:47 +0000145 ValueTable() : nextValueNumber(1) { }
Chris Lattnerff36c952009-09-21 02:42:51 +0000146 uint32_t lookup_or_add(Value *V);
147 uint32_t lookup(Value *V) const;
148 void add(Value *V, uint32_t num);
Owen Anderson85c40642007-07-24 17:55:58 +0000149 void clear();
Chris Lattnerff36c952009-09-21 02:42:51 +0000150 void erase(Value *v);
Owen Anderson85c40642007-07-24 17:55:58 +0000151 unsigned size();
Owen Andersonbcf2bd52008-05-12 20:15:55 +0000152 void setAliasAnalysis(AliasAnalysis* A) { AA = A; }
Chris Lattner02ca4422008-12-01 00:40:32 +0000153 AliasAnalysis *getAliasAnalysis() const { return AA; }
Owen Andersonbcf2bd52008-05-12 20:15:55 +0000154 void setMemDep(MemoryDependenceAnalysis* M) { MD = M; }
155 void setDomTree(DominatorTree* D) { DT = D; }
Owen Anderson8a8d13c2008-07-03 17:44:33 +0000156 uint32_t getNextUnusedValueNumber() { return nextValueNumber; }
Bill Wendling2a023742008-12-22 21:36:08 +0000157 void verifyRemoved(const Value *) const;
Owen Anderson85c40642007-07-24 17:55:58 +0000158 };
159}
160
161namespace llvm {
Chris Lattner92eea072007-09-17 18:34:04 +0000162template <> struct DenseMapInfo<Expression> {
Owen Andersonbf8a3eb2007-08-02 18:16:06 +0000163 static inline Expression getEmptyKey() {
164 return Expression(Expression::EMPTY);
165 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000166
Owen Andersonbf8a3eb2007-08-02 18:16:06 +0000167 static inline Expression getTombstoneKey() {
168 return Expression(Expression::TOMBSTONE);
169 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000170
Owen Anderson85c40642007-07-24 17:55:58 +0000171 static unsigned getHashValue(const Expression e) {
172 unsigned hash = e.opcode;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000173
Anton Korobeynikov8522e1c2008-02-20 11:26:25 +0000174 hash = ((unsigned)((uintptr_t)e.type >> 4) ^
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000175 (unsigned)((uintptr_t)e.type >> 9));
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000176
Owen Andersonbf8a3eb2007-08-02 18:16:06 +0000177 for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
178 E = e.varargs.end(); I != E; ++I)
Owen Anderson85c40642007-07-24 17:55:58 +0000179 hash = *I + hash * 37;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000180
Anton Korobeynikov8522e1c2008-02-20 11:26:25 +0000181 hash = ((unsigned)((uintptr_t)e.function >> 4) ^
182 (unsigned)((uintptr_t)e.function >> 9)) +
183 hash * 37;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000184
Owen Anderson85c40642007-07-24 17:55:58 +0000185 return hash;
186 }
Chris Lattner92eea072007-09-17 18:34:04 +0000187 static bool isEqual(const Expression &LHS, const Expression &RHS) {
188 return LHS == RHS;
189 }
Owen Anderson85c40642007-07-24 17:55:58 +0000190 static bool isPod() { return true; }
191};
192}
193
194//===----------------------------------------------------------------------===//
195// ValueTable Internal Functions
196//===----------------------------------------------------------------------===//
Chris Lattner3d7103e2008-03-21 21:14:38 +0000197Expression::ExpressionOpcode ValueTable::getOpcode(BinaryOperator* BO) {
Owen Anderson85c40642007-07-24 17:55:58 +0000198 switch(BO->getOpcode()) {
Chris Lattner3d7103e2008-03-21 21:14:38 +0000199 default: // THIS SHOULD NEVER HAPPEN
Edwin Törökbd448e32009-07-14 16:55:14 +0000200 llvm_unreachable("Binary operator with unknown opcode?");
Chris Lattner3d7103e2008-03-21 21:14:38 +0000201 case Instruction::Add: return Expression::ADD;
Dan Gohman7ce405e2009-06-04 22:49:04 +0000202 case Instruction::FAdd: return Expression::FADD;
Chris Lattner3d7103e2008-03-21 21:14:38 +0000203 case Instruction::Sub: return Expression::SUB;
Dan Gohman7ce405e2009-06-04 22:49:04 +0000204 case Instruction::FSub: return Expression::FSUB;
Chris Lattner3d7103e2008-03-21 21:14:38 +0000205 case Instruction::Mul: return Expression::MUL;
Dan Gohman7ce405e2009-06-04 22:49:04 +0000206 case Instruction::FMul: return Expression::FMUL;
Chris Lattner3d7103e2008-03-21 21:14:38 +0000207 case Instruction::UDiv: return Expression::UDIV;
208 case Instruction::SDiv: return Expression::SDIV;
209 case Instruction::FDiv: return Expression::FDIV;
210 case Instruction::URem: return Expression::UREM;
211 case Instruction::SRem: return Expression::SREM;
212 case Instruction::FRem: return Expression::FREM;
213 case Instruction::Shl: return Expression::SHL;
214 case Instruction::LShr: return Expression::LSHR;
215 case Instruction::AShr: return Expression::ASHR;
216 case Instruction::And: return Expression::AND;
217 case Instruction::Or: return Expression::OR;
218 case Instruction::Xor: return Expression::XOR;
Owen Anderson85c40642007-07-24 17:55:58 +0000219 }
220}
221
222Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
Nick Lewycky8f5253b2009-07-08 03:04:38 +0000223 if (isa<ICmpInst>(C)) {
Owen Anderson85c40642007-07-24 17:55:58 +0000224 switch (C->getPredicate()) {
Chris Lattner3d7103e2008-03-21 21:14:38 +0000225 default: // THIS SHOULD NEVER HAPPEN
Edwin Törökbd448e32009-07-14 16:55:14 +0000226 llvm_unreachable("Comparison with unknown predicate?");
Chris Lattner3d7103e2008-03-21 21:14:38 +0000227 case ICmpInst::ICMP_EQ: return Expression::ICMPEQ;
228 case ICmpInst::ICMP_NE: return Expression::ICMPNE;
229 case ICmpInst::ICMP_UGT: return Expression::ICMPUGT;
230 case ICmpInst::ICMP_UGE: return Expression::ICMPUGE;
231 case ICmpInst::ICMP_ULT: return Expression::ICMPULT;
232 case ICmpInst::ICMP_ULE: return Expression::ICMPULE;
233 case ICmpInst::ICMP_SGT: return Expression::ICMPSGT;
234 case ICmpInst::ICMP_SGE: return Expression::ICMPSGE;
235 case ICmpInst::ICMP_SLT: return Expression::ICMPSLT;
236 case ICmpInst::ICMP_SLE: return Expression::ICMPSLE;
Owen Anderson85c40642007-07-24 17:55:58 +0000237 }
Nick Lewycky8f5253b2009-07-08 03:04:38 +0000238 } else {
239 switch (C->getPredicate()) {
240 default: // THIS SHOULD NEVER HAPPEN
Edwin Törökbd448e32009-07-14 16:55:14 +0000241 llvm_unreachable("Comparison with unknown predicate?");
Nick Lewycky8f5253b2009-07-08 03:04:38 +0000242 case FCmpInst::FCMP_OEQ: return Expression::FCMPOEQ;
243 case FCmpInst::FCMP_OGT: return Expression::FCMPOGT;
244 case FCmpInst::FCMP_OGE: return Expression::FCMPOGE;
245 case FCmpInst::FCMP_OLT: return Expression::FCMPOLT;
246 case FCmpInst::FCMP_OLE: return Expression::FCMPOLE;
247 case FCmpInst::FCMP_ONE: return Expression::FCMPONE;
248 case FCmpInst::FCMP_ORD: return Expression::FCMPORD;
249 case FCmpInst::FCMP_UNO: return Expression::FCMPUNO;
250 case FCmpInst::FCMP_UEQ: return Expression::FCMPUEQ;
251 case FCmpInst::FCMP_UGT: return Expression::FCMPUGT;
252 case FCmpInst::FCMP_UGE: return Expression::FCMPUGE;
253 case FCmpInst::FCMP_ULT: return Expression::FCMPULT;
254 case FCmpInst::FCMP_ULE: return Expression::FCMPULE;
255 case FCmpInst::FCMP_UNE: return Expression::FCMPUNE;
256 }
Owen Anderson85c40642007-07-24 17:55:58 +0000257 }
258}
259
Chris Lattner3d7103e2008-03-21 21:14:38 +0000260Expression::ExpressionOpcode ValueTable::getOpcode(CastInst* C) {
Owen Anderson85c40642007-07-24 17:55:58 +0000261 switch(C->getOpcode()) {
Chris Lattner3d7103e2008-03-21 21:14:38 +0000262 default: // THIS SHOULD NEVER HAPPEN
Edwin Törökbd448e32009-07-14 16:55:14 +0000263 llvm_unreachable("Cast operator with unknown opcode?");
Chris Lattner3d7103e2008-03-21 21:14:38 +0000264 case Instruction::Trunc: return Expression::TRUNC;
265 case Instruction::ZExt: return Expression::ZEXT;
266 case Instruction::SExt: return Expression::SEXT;
267 case Instruction::FPToUI: return Expression::FPTOUI;
268 case Instruction::FPToSI: return Expression::FPTOSI;
269 case Instruction::UIToFP: return Expression::UITOFP;
270 case Instruction::SIToFP: return Expression::SITOFP;
271 case Instruction::FPTrunc: return Expression::FPTRUNC;
272 case Instruction::FPExt: return Expression::FPEXT;
273 case Instruction::PtrToInt: return Expression::PTRTOINT;
274 case Instruction::IntToPtr: return Expression::INTTOPTR;
275 case Instruction::BitCast: return Expression::BITCAST;
Owen Anderson85c40642007-07-24 17:55:58 +0000276 }
277}
278
Owen Anderson5e9366f2007-10-18 19:39:33 +0000279Expression ValueTable::create_expression(CallInst* C) {
280 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000281
Owen Anderson5e9366f2007-10-18 19:39:33 +0000282 e.type = C->getType();
Owen Anderson5e9366f2007-10-18 19:39:33 +0000283 e.function = C->getCalledFunction();
284 e.opcode = Expression::CALL;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000285
Owen Anderson5e9366f2007-10-18 19:39:33 +0000286 for (CallInst::op_iterator I = C->op_begin()+1, E = C->op_end();
287 I != E; ++I)
Owen Anderson07f478f2008-04-11 05:11:49 +0000288 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000289
Owen Anderson5e9366f2007-10-18 19:39:33 +0000290 return e;
291}
292
Owen Anderson85c40642007-07-24 17:55:58 +0000293Expression ValueTable::create_expression(BinaryOperator* BO) {
294 Expression e;
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000295 e.varargs.push_back(lookup_or_add(BO->getOperand(0)));
296 e.varargs.push_back(lookup_or_add(BO->getOperand(1)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000297 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000298 e.type = BO->getType();
299 e.opcode = getOpcode(BO);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000300
Owen Anderson85c40642007-07-24 17:55:58 +0000301 return e;
302}
303
304Expression ValueTable::create_expression(CmpInst* C) {
305 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000306
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000307 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
308 e.varargs.push_back(lookup_or_add(C->getOperand(1)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000309 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000310 e.type = C->getType();
311 e.opcode = getOpcode(C);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000312
Owen Anderson85c40642007-07-24 17:55:58 +0000313 return e;
314}
315
316Expression ValueTable::create_expression(CastInst* C) {
317 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000318
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000319 e.varargs.push_back(lookup_or_add(C->getOperand(0)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000320 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000321 e.type = C->getType();
322 e.opcode = getOpcode(C);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000323
Owen Anderson85c40642007-07-24 17:55:58 +0000324 return e;
325}
326
327Expression ValueTable::create_expression(ShuffleVectorInst* S) {
328 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000329
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000330 e.varargs.push_back(lookup_or_add(S->getOperand(0)));
331 e.varargs.push_back(lookup_or_add(S->getOperand(1)));
332 e.varargs.push_back(lookup_or_add(S->getOperand(2)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000333 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000334 e.type = S->getType();
335 e.opcode = Expression::SHUFFLE;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000336
Owen Anderson85c40642007-07-24 17:55:58 +0000337 return e;
338}
339
340Expression ValueTable::create_expression(ExtractElementInst* E) {
341 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000342
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000343 e.varargs.push_back(lookup_or_add(E->getOperand(0)));
344 e.varargs.push_back(lookup_or_add(E->getOperand(1)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000345 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000346 e.type = E->getType();
347 e.opcode = Expression::EXTRACT;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000348
Owen Anderson85c40642007-07-24 17:55:58 +0000349 return e;
350}
351
352Expression ValueTable::create_expression(InsertElementInst* I) {
353 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000354
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000355 e.varargs.push_back(lookup_or_add(I->getOperand(0)));
356 e.varargs.push_back(lookup_or_add(I->getOperand(1)));
357 e.varargs.push_back(lookup_or_add(I->getOperand(2)));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000358 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000359 e.type = I->getType();
360 e.opcode = Expression::INSERT;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000361
Owen Anderson85c40642007-07-24 17:55:58 +0000362 return e;
363}
364
365Expression ValueTable::create_expression(SelectInst* I) {
366 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000367
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000368 e.varargs.push_back(lookup_or_add(I->getCondition()));
369 e.varargs.push_back(lookup_or_add(I->getTrueValue()));
370 e.varargs.push_back(lookup_or_add(I->getFalseValue()));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000371 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000372 e.type = I->getType();
373 e.opcode = Expression::SELECT;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000374
Owen Anderson85c40642007-07-24 17:55:58 +0000375 return e;
376}
377
378Expression ValueTable::create_expression(GetElementPtrInst* G) {
379 Expression e;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000380
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000381 e.varargs.push_back(lookup_or_add(G->getPointerOperand()));
Owen Anderson5e9366f2007-10-18 19:39:33 +0000382 e.function = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000383 e.type = G->getType();
384 e.opcode = Expression::GEP;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000385
Owen Anderson85c40642007-07-24 17:55:58 +0000386 for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
387 I != E; ++I)
Owen Anderson07f478f2008-04-11 05:11:49 +0000388 e.varargs.push_back(lookup_or_add(*I));
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000389
Owen Anderson85c40642007-07-24 17:55:58 +0000390 return e;
391}
392
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000393Expression ValueTable::create_expression(ExtractValueInst* E) {
394 Expression e;
395
396 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
397 for (ExtractValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
398 II != IE; ++II)
399 e.varargs.push_back(*II);
400 e.function = 0;
401 e.type = E->getType();
402 e.opcode = Expression::EXTRACTVALUE;
403
404 return e;
405}
406
407Expression ValueTable::create_expression(InsertValueInst* E) {
408 Expression e;
409
410 e.varargs.push_back(lookup_or_add(E->getAggregateOperand()));
411 e.varargs.push_back(lookup_or_add(E->getInsertedValueOperand()));
412 for (InsertValueInst::idx_iterator II = E->idx_begin(), IE = E->idx_end();
413 II != IE; ++II)
414 e.varargs.push_back(*II);
415 e.function = 0;
416 e.type = E->getType();
417 e.opcode = Expression::INSERTVALUE;
418
419 return e;
420}
421
Owen Anderson85c40642007-07-24 17:55:58 +0000422//===----------------------------------------------------------------------===//
423// ValueTable External Functions
424//===----------------------------------------------------------------------===//
425
Owen Andersone6b4ff82008-06-18 21:41:49 +0000426/// add - Insert a value into the table with a specified value number.
Chris Lattnerff36c952009-09-21 02:42:51 +0000427void ValueTable::add(Value *V, uint32_t num) {
Owen Andersone6b4ff82008-06-18 21:41:49 +0000428 valueNumbering.insert(std::make_pair(V, num));
429}
430
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000431uint32_t ValueTable::lookup_or_add_call(CallInst* C) {
432 if (AA->doesNotAccessMemory(C)) {
433 Expression exp = create_expression(C);
434 uint32_t& e = expressionNumbering[exp];
435 if (!e) e = nextValueNumber++;
436 valueNumbering[C] = e;
437 return e;
438 } else if (AA->onlyReadsMemory(C)) {
439 Expression exp = create_expression(C);
440 uint32_t& e = expressionNumbering[exp];
441 if (!e) {
442 e = nextValueNumber++;
443 valueNumbering[C] = e;
444 return e;
445 }
Dan Gohmanc8d26652009-11-14 02:27:51 +0000446 if (!MD) {
447 e = nextValueNumber++;
448 valueNumbering[C] = e;
449 return e;
450 }
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000451
452 MemDepResult local_dep = MD->getDependency(C);
453
454 if (!local_dep.isDef() && !local_dep.isNonLocal()) {
455 valueNumbering[C] = nextValueNumber;
456 return nextValueNumber++;
457 }
458
459 if (local_dep.isDef()) {
460 CallInst* local_cdep = cast<CallInst>(local_dep.getInst());
461
462 if (local_cdep->getNumOperands() != C->getNumOperands()) {
463 valueNumbering[C] = nextValueNumber;
464 return nextValueNumber++;
465 }
466
467 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
468 uint32_t c_vn = lookup_or_add(C->getOperand(i));
469 uint32_t cd_vn = lookup_or_add(local_cdep->getOperand(i));
470 if (c_vn != cd_vn) {
471 valueNumbering[C] = nextValueNumber;
472 return nextValueNumber++;
473 }
474 }
475
476 uint32_t v = lookup_or_add(local_cdep);
477 valueNumbering[C] = v;
478 return v;
479 }
480
481 // Non-local case.
482 const MemoryDependenceAnalysis::NonLocalDepInfo &deps =
483 MD->getNonLocalCallDependency(CallSite(C));
484 // FIXME: call/call dependencies for readonly calls should return def, not
485 // clobber! Move the checking logic to MemDep!
486 CallInst* cdep = 0;
487
488 // Check to see if we have a single dominating call instruction that is
489 // identical to C.
490 for (unsigned i = 0, e = deps.size(); i != e; ++i) {
491 const MemoryDependenceAnalysis::NonLocalDepEntry *I = &deps[i];
492 // Ignore non-local dependencies.
493 if (I->second.isNonLocal())
494 continue;
495
496 // We don't handle non-depedencies. If we already have a call, reject
497 // instruction dependencies.
498 if (I->second.isClobber() || cdep != 0) {
499 cdep = 0;
500 break;
501 }
502
503 CallInst *NonLocalDepCall = dyn_cast<CallInst>(I->second.getInst());
504 // FIXME: All duplicated with non-local case.
505 if (NonLocalDepCall && DT->properlyDominates(I->first, C->getParent())){
506 cdep = NonLocalDepCall;
507 continue;
508 }
509
510 cdep = 0;
511 break;
512 }
513
514 if (!cdep) {
515 valueNumbering[C] = nextValueNumber;
516 return nextValueNumber++;
517 }
518
519 if (cdep->getNumOperands() != C->getNumOperands()) {
520 valueNumbering[C] = nextValueNumber;
521 return nextValueNumber++;
522 }
523 for (unsigned i = 1; i < C->getNumOperands(); ++i) {
524 uint32_t c_vn = lookup_or_add(C->getOperand(i));
525 uint32_t cd_vn = lookup_or_add(cdep->getOperand(i));
526 if (c_vn != cd_vn) {
527 valueNumbering[C] = nextValueNumber;
528 return nextValueNumber++;
529 }
530 }
531
532 uint32_t v = lookup_or_add(cdep);
533 valueNumbering[C] = v;
534 return v;
535
536 } else {
537 valueNumbering[C] = nextValueNumber;
538 return nextValueNumber++;
539 }
540}
541
Owen Anderson85c40642007-07-24 17:55:58 +0000542/// lookup_or_add - Returns the value number for the specified value, assigning
543/// it a new number if it did not have one before.
Chris Lattnerff36c952009-09-21 02:42:51 +0000544uint32_t ValueTable::lookup_or_add(Value *V) {
Owen Anderson85c40642007-07-24 17:55:58 +0000545 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
546 if (VI != valueNumbering.end())
547 return VI->second;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000548
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000549 if (!isa<Instruction>(V)) {
Owen Andersonb472cd82009-10-19 21:14:57 +0000550 valueNumbering[V] = nextValueNumber;
Owen Anderson85c40642007-07-24 17:55:58 +0000551 return nextValueNumber++;
552 }
Owen Andersona7d4c7c2009-10-19 22:14:22 +0000553
554 Instruction* I = cast<Instruction>(V);
555 Expression exp;
556 switch (I->getOpcode()) {
557 case Instruction::Call:
558 return lookup_or_add_call(cast<CallInst>(I));
559 case Instruction::Add:
560 case Instruction::FAdd:
561 case Instruction::Sub:
562 case Instruction::FSub:
563 case Instruction::Mul:
564 case Instruction::FMul:
565 case Instruction::UDiv:
566 case Instruction::SDiv:
567 case Instruction::FDiv:
568 case Instruction::URem:
569 case Instruction::SRem:
570 case Instruction::FRem:
571 case Instruction::Shl:
572 case Instruction::LShr:
573 case Instruction::AShr:
574 case Instruction::And:
575 case Instruction::Or :
576 case Instruction::Xor:
577 exp = create_expression(cast<BinaryOperator>(I));
578 break;
579 case Instruction::ICmp:
580 case Instruction::FCmp:
581 exp = create_expression(cast<CmpInst>(I));
582 break;
583 case Instruction::Trunc:
584 case Instruction::ZExt:
585 case Instruction::SExt:
586 case Instruction::FPToUI:
587 case Instruction::FPToSI:
588 case Instruction::UIToFP:
589 case Instruction::SIToFP:
590 case Instruction::FPTrunc:
591 case Instruction::FPExt:
592 case Instruction::PtrToInt:
593 case Instruction::IntToPtr:
594 case Instruction::BitCast:
595 exp = create_expression(cast<CastInst>(I));
596 break;
597 case Instruction::Select:
598 exp = create_expression(cast<SelectInst>(I));
599 break;
600 case Instruction::ExtractElement:
601 exp = create_expression(cast<ExtractElementInst>(I));
602 break;
603 case Instruction::InsertElement:
604 exp = create_expression(cast<InsertElementInst>(I));
605 break;
606 case Instruction::ShuffleVector:
607 exp = create_expression(cast<ShuffleVectorInst>(I));
608 break;
609 case Instruction::ExtractValue:
610 exp = create_expression(cast<ExtractValueInst>(I));
611 break;
612 case Instruction::InsertValue:
613 exp = create_expression(cast<InsertValueInst>(I));
614 break;
615 case Instruction::GetElementPtr:
616 exp = create_expression(cast<GetElementPtrInst>(I));
617 break;
618 default:
619 valueNumbering[V] = nextValueNumber;
620 return nextValueNumber++;
621 }
622
623 uint32_t& e = expressionNumbering[exp];
624 if (!e) e = nextValueNumber++;
625 valueNumbering[V] = e;
626 return e;
Owen Anderson85c40642007-07-24 17:55:58 +0000627}
628
629/// lookup - Returns the value number of the specified value. Fails if
630/// the value has not yet been numbered.
Chris Lattnerff36c952009-09-21 02:42:51 +0000631uint32_t ValueTable::lookup(Value *V) const {
Jeffrey Yasskin8154d2e2009-11-10 01:02:17 +0000632 DenseMap<Value*, uint32_t>::const_iterator VI = valueNumbering.find(V);
Chris Lattner3d7103e2008-03-21 21:14:38 +0000633 assert(VI != valueNumbering.end() && "Value not numbered?");
634 return VI->second;
Owen Anderson85c40642007-07-24 17:55:58 +0000635}
636
637/// clear - Remove all entries from the ValueTable
638void ValueTable::clear() {
639 valueNumbering.clear();
640 expressionNumbering.clear();
641 nextValueNumber = 1;
642}
643
Owen Anderson5aff8002007-07-31 23:27:13 +0000644/// erase - Remove a value from the value numbering
Chris Lattnerff36c952009-09-21 02:42:51 +0000645void ValueTable::erase(Value *V) {
Owen Anderson5aff8002007-07-31 23:27:13 +0000646 valueNumbering.erase(V);
647}
648
Bill Wendling2a023742008-12-22 21:36:08 +0000649/// verifyRemoved - Verify that the value is removed from all internal data
650/// structures.
651void ValueTable::verifyRemoved(const Value *V) const {
Jeffrey Yasskin8154d2e2009-11-10 01:02:17 +0000652 for (DenseMap<Value*, uint32_t>::const_iterator
Bill Wendling2a023742008-12-22 21:36:08 +0000653 I = valueNumbering.begin(), E = valueNumbering.end(); I != E; ++I) {
654 assert(I->first != V && "Inst still occurs in value numbering map!");
655 }
656}
657
Owen Anderson85c40642007-07-24 17:55:58 +0000658//===----------------------------------------------------------------------===//
Bill Wendling42f17f62008-12-22 22:32:22 +0000659// GVN Pass
Owen Anderson85c40642007-07-24 17:55:58 +0000660//===----------------------------------------------------------------------===//
661
662namespace {
Chris Lattnerfa2d1ba2009-09-02 06:11:42 +0000663 struct ValueNumberScope {
Owen Anderson2a412722008-06-20 01:15:47 +0000664 ValueNumberScope* parent;
665 DenseMap<uint32_t, Value*> table;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000666
Owen Anderson2a412722008-06-20 01:15:47 +0000667 ValueNumberScope(ValueNumberScope* p) : parent(p) { }
668 };
669}
670
671namespace {
Owen Anderson85c40642007-07-24 17:55:58 +0000672
Chris Lattnerfa2d1ba2009-09-02 06:11:42 +0000673 class GVN : public FunctionPass {
Owen Anderson85c40642007-07-24 17:55:58 +0000674 bool runOnFunction(Function &F);
675 public:
676 static char ID; // Pass identification, replacement for typeid
Dan Gohmanc8d26652009-11-14 02:27:51 +0000677 explicit GVN(bool nopre = false, bool noloads = false)
678 : FunctionPass(&ID), NoPRE(nopre), NoLoads(noloads), MD(0) { }
Owen Anderson85c40642007-07-24 17:55:58 +0000679
680 private:
Evan Chengf036e552009-10-30 20:12:24 +0000681 bool NoPRE;
Dan Gohmanc8d26652009-11-14 02:27:51 +0000682 bool NoLoads;
Chris Lattner02ca4422008-12-01 00:40:32 +0000683 MemoryDependenceAnalysis *MD;
684 DominatorTree *DT;
685
Owen Anderson85c40642007-07-24 17:55:58 +0000686 ValueTable VN;
Owen Anderson2a412722008-06-20 01:15:47 +0000687 DenseMap<BasicBlock*, ValueNumberScope*> localAvail;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000688
Owen Anderson85c40642007-07-24 17:55:58 +0000689 // This transformation requires dominator postdominator info
690 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Owen Anderson85c40642007-07-24 17:55:58 +0000691 AU.addRequired<DominatorTree>();
Dan Gohmanc8d26652009-11-14 02:27:51 +0000692 if (!NoLoads)
693 AU.addRequired<MemoryDependenceAnalysis>();
Owen Anderson5e9366f2007-10-18 19:39:33 +0000694 AU.addRequired<AliasAnalysis>();
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000695
Owen Andersonaef6a922008-06-23 17:49:45 +0000696 AU.addPreserved<DominatorTree>();
Owen Anderson5e9366f2007-10-18 19:39:33 +0000697 AU.addPreserved<AliasAnalysis>();
Owen Anderson85c40642007-07-24 17:55:58 +0000698 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000699
Owen Anderson85c40642007-07-24 17:55:58 +0000700 // Helper fuctions
701 // FIXME: eliminate or document these better
Owen Anderson85c40642007-07-24 17:55:58 +0000702 bool processLoad(LoadInst* L,
Chris Lattner7de20452008-03-21 22:01:16 +0000703 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerff36c952009-09-21 02:42:51 +0000704 bool processInstruction(Instruction *I,
Chris Lattner7de20452008-03-21 22:01:16 +0000705 SmallVectorImpl<Instruction*> &toErase);
Owen Andersonbf8a3eb2007-08-02 18:16:06 +0000706 bool processNonLocalLoad(LoadInst* L,
Chris Lattner7de20452008-03-21 22:01:16 +0000707 SmallVectorImpl<Instruction*> &toErase);
Chris Lattnerff36c952009-09-21 02:42:51 +0000708 bool processBlock(BasicBlock *BB);
Owen Andersone6b4ff82008-06-18 21:41:49 +0000709 void dump(DenseMap<uint32_t, Value*>& d);
Owen Andersonbe168b32007-08-14 18:04:11 +0000710 bool iterateOnFunction(Function &F);
Chris Lattnerff36c952009-09-21 02:42:51 +0000711 Value *CollapsePhi(PHINode* p);
Owen Andersone6b4ff82008-06-18 21:41:49 +0000712 bool performPRE(Function& F);
Chris Lattnerff36c952009-09-21 02:42:51 +0000713 Value *lookupNumber(BasicBlock *BB, uint32_t num);
Nuno Lopes274474b2008-10-10 16:25:50 +0000714 void cleanupGlobalSets();
Bill Wendling2a023742008-12-22 21:36:08 +0000715 void verifyRemoved(const Instruction *I) const;
Owen Anderson85c40642007-07-24 17:55:58 +0000716 };
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000717
Owen Anderson85c40642007-07-24 17:55:58 +0000718 char GVN::ID = 0;
Owen Anderson85c40642007-07-24 17:55:58 +0000719}
720
721// createGVNPass - The public interface to this file...
Dan Gohmanc8d26652009-11-14 02:27:51 +0000722FunctionPass *llvm::createGVNPass(bool NoPRE, bool NoLoads) {
723 return new GVN(NoPRE, NoLoads);
724}
Owen Anderson85c40642007-07-24 17:55:58 +0000725
726static RegisterPass<GVN> X("gvn",
727 "Global Value Numbering");
728
Owen Andersone6b4ff82008-06-18 21:41:49 +0000729void GVN::dump(DenseMap<uint32_t, Value*>& d) {
Owen Anderson5d72a422007-07-25 19:57:03 +0000730 printf("{\n");
Owen Andersone6b4ff82008-06-18 21:41:49 +0000731 for (DenseMap<uint32_t, Value*>::iterator I = d.begin(),
Owen Anderson5d72a422007-07-25 19:57:03 +0000732 E = d.end(); I != E; ++I) {
Owen Andersone6b4ff82008-06-18 21:41:49 +0000733 printf("%d\n", I->first);
Owen Anderson5d72a422007-07-25 19:57:03 +0000734 I->second->dump();
735 }
736 printf("}\n");
737}
738
Chris Lattnerff36c952009-09-21 02:42:51 +0000739static bool isSafeReplacement(PHINode* p, Instruction *inst) {
Owen Andersond68b1af2009-08-26 22:55:11 +0000740 if (!isa<PHINode>(inst))
741 return true;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000742
Owen Andersond68b1af2009-08-26 22:55:11 +0000743 for (Instruction::use_iterator UI = p->use_begin(), E = p->use_end();
744 UI != E; ++UI)
745 if (PHINode* use_phi = dyn_cast<PHINode>(UI))
746 if (use_phi->getParent() == inst->getParent())
747 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000748
Owen Andersond68b1af2009-08-26 22:55:11 +0000749 return true;
750}
751
Chris Lattnerff36c952009-09-21 02:42:51 +0000752Value *GVN::CollapsePhi(PHINode *PN) {
753 Value *ConstVal = PN->hasConstantValue(DT);
754 if (!ConstVal) return 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000755
Chris Lattnerff36c952009-09-21 02:42:51 +0000756 Instruction *Inst = dyn_cast<Instruction>(ConstVal);
757 if (!Inst)
758 return ConstVal;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000759
Chris Lattnerff36c952009-09-21 02:42:51 +0000760 if (DT->dominates(Inst, PN))
761 if (isSafeReplacement(PN, Inst))
762 return Inst;
Owen Andersone02ad522007-08-16 22:51:56 +0000763 return 0;
764}
Owen Anderson5d72a422007-07-25 19:57:03 +0000765
Chris Lattnerdcded152008-12-02 08:16:11 +0000766/// IsValueFullyAvailableInBlock - Return true if we can prove that the value
767/// we're analyzing is fully available in the specified block. As we go, keep
Chris Lattner159b98f2008-12-05 07:49:08 +0000768/// track of which blocks we know are fully alive in FullyAvailableBlocks. This
769/// map is actually a tri-state map with the following values:
770/// 0) we know the block *is not* fully available.
771/// 1) we know the block *is* fully available.
772/// 2) we do not know whether the block is fully available or not, but we are
773/// currently speculating that it will be.
774/// 3) we are speculating for this block and have used that to speculate for
775/// other blocks.
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000776static bool IsValueFullyAvailableInBlock(BasicBlock *BB,
Chris Lattner159b98f2008-12-05 07:49:08 +0000777 DenseMap<BasicBlock*, char> &FullyAvailableBlocks) {
Chris Lattnerdcded152008-12-02 08:16:11 +0000778 // Optimistically assume that the block is fully available and check to see
779 // if we already know about this block in one lookup.
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000780 std::pair<DenseMap<BasicBlock*, char>::iterator, char> IV =
Chris Lattner159b98f2008-12-05 07:49:08 +0000781 FullyAvailableBlocks.insert(std::make_pair(BB, 2));
Chris Lattnerdcded152008-12-02 08:16:11 +0000782
783 // If the entry already existed for this block, return the precomputed value.
Chris Lattner159b98f2008-12-05 07:49:08 +0000784 if (!IV.second) {
785 // If this is a speculative "available" value, mark it as being used for
786 // speculation of other blocks.
787 if (IV.first->second == 2)
788 IV.first->second = 3;
789 return IV.first->second != 0;
790 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000791
Chris Lattnerdcded152008-12-02 08:16:11 +0000792 // Otherwise, see if it is fully available in all predecessors.
793 pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000794
Chris Lattnerdcded152008-12-02 08:16:11 +0000795 // If this block has no predecessors, it isn't live-in here.
796 if (PI == PE)
Chris Lattner159b98f2008-12-05 07:49:08 +0000797 goto SpeculationFailure;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000798
Chris Lattnerdcded152008-12-02 08:16:11 +0000799 for (; PI != PE; ++PI)
800 // If the value isn't fully available in one of our predecessors, then it
801 // isn't fully available in this block either. Undo our previous
802 // optimistic assumption and bail out.
803 if (!IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
Chris Lattner159b98f2008-12-05 07:49:08 +0000804 goto SpeculationFailure;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000805
Chris Lattnerdcded152008-12-02 08:16:11 +0000806 return true;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000807
Chris Lattner159b98f2008-12-05 07:49:08 +0000808// SpeculationFailure - If we get here, we found out that this is not, after
809// all, a fully-available block. We have a problem if we speculated on this and
810// used the speculation to mark other blocks as available.
811SpeculationFailure:
812 char &BBVal = FullyAvailableBlocks[BB];
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000813
Chris Lattner159b98f2008-12-05 07:49:08 +0000814 // If we didn't speculate on this, just return with it set to false.
815 if (BBVal == 2) {
816 BBVal = 0;
817 return false;
818 }
819
820 // If we did speculate on this value, we could have blocks set to 1 that are
821 // incorrect. Walk the (transitive) successors of this block and mark them as
822 // 0 if set to one.
823 SmallVector<BasicBlock*, 32> BBWorklist;
824 BBWorklist.push_back(BB);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000825
Chris Lattner159b98f2008-12-05 07:49:08 +0000826 while (!BBWorklist.empty()) {
827 BasicBlock *Entry = BBWorklist.pop_back_val();
828 // Note that this sets blocks to 0 (unavailable) if they happen to not
829 // already be in FullyAvailableBlocks. This is safe.
830 char &EntryVal = FullyAvailableBlocks[Entry];
831 if (EntryVal == 0) continue; // Already unavailable.
832
833 // Mark as unavailable.
834 EntryVal = 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000835
Chris Lattner159b98f2008-12-05 07:49:08 +0000836 for (succ_iterator I = succ_begin(Entry), E = succ_end(Entry); I != E; ++I)
837 BBWorklist.push_back(*I);
838 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000839
Chris Lattner159b98f2008-12-05 07:49:08 +0000840 return false;
Chris Lattnerdcded152008-12-02 08:16:11 +0000841}
842
Chris Lattnerd6b1d052009-09-20 20:09:34 +0000843
Chris Lattner012b3602009-09-21 17:24:04 +0000844/// CanCoerceMustAliasedValueToLoad - Return true if
845/// CoerceAvailableValueToLoadType will succeed.
846static bool CanCoerceMustAliasedValueToLoad(Value *StoredVal,
847 const Type *LoadTy,
848 const TargetData &TD) {
849 // If the loaded or stored value is an first class array or struct, don't try
850 // to transform them. We need to be able to bitcast to integer.
851 if (isa<StructType>(LoadTy) || isa<ArrayType>(LoadTy) ||
852 isa<StructType>(StoredVal->getType()) ||
853 isa<ArrayType>(StoredVal->getType()))
854 return false;
855
856 // The store has to be at least as big as the load.
857 if (TD.getTypeSizeInBits(StoredVal->getType()) <
858 TD.getTypeSizeInBits(LoadTy))
859 return false;
860
861 return true;
862}
863
864
Chris Lattnerd6b1d052009-09-20 20:09:34 +0000865/// CoerceAvailableValueToLoadType - If we saw a store of a value to memory, and
866/// then a load from a must-aliased pointer of a different type, try to coerce
867/// the stored value. LoadedTy is the type of the load we want to replace and
868/// InsertPt is the place to insert new instructions.
869///
870/// If we can't do it, return null.
871static Value *CoerceAvailableValueToLoadType(Value *StoredVal,
872 const Type *LoadedTy,
873 Instruction *InsertPt,
874 const TargetData &TD) {
Chris Lattner012b3602009-09-21 17:24:04 +0000875 if (!CanCoerceMustAliasedValueToLoad(StoredVal, LoadedTy, TD))
876 return 0;
877
Chris Lattnerd6b1d052009-09-20 20:09:34 +0000878 const Type *StoredValTy = StoredVal->getType();
879
880 uint64_t StoreSize = TD.getTypeSizeInBits(StoredValTy);
881 uint64_t LoadSize = TD.getTypeSizeInBits(LoadedTy);
882
883 // If the store and reload are the same size, we can always reuse it.
884 if (StoreSize == LoadSize) {
885 if (isa<PointerType>(StoredValTy) && isa<PointerType>(LoadedTy)) {
886 // Pointer to Pointer -> use bitcast.
887 return new BitCastInst(StoredVal, LoadedTy, "", InsertPt);
888 }
889
890 // Convert source pointers to integers, which can be bitcast.
891 if (isa<PointerType>(StoredValTy)) {
892 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
893 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
894 }
895
896 const Type *TypeToCastTo = LoadedTy;
897 if (isa<PointerType>(TypeToCastTo))
898 TypeToCastTo = TD.getIntPtrType(StoredValTy->getContext());
899
900 if (StoredValTy != TypeToCastTo)
901 StoredVal = new BitCastInst(StoredVal, TypeToCastTo, "", InsertPt);
902
903 // Cast to pointer if the load needs a pointer type.
904 if (isa<PointerType>(LoadedTy))
905 StoredVal = new IntToPtrInst(StoredVal, LoadedTy, "", InsertPt);
906
907 return StoredVal;
908 }
909
910 // If the loaded value is smaller than the available value, then we can
911 // extract out a piece from it. If the available value is too small, then we
912 // can't do anything.
Chris Lattner012b3602009-09-21 17:24:04 +0000913 assert(StoreSize >= LoadSize && "CanCoerceMustAliasedValueToLoad fail");
Chris Lattnerd6b1d052009-09-20 20:09:34 +0000914
915 // Convert source pointers to integers, which can be manipulated.
916 if (isa<PointerType>(StoredValTy)) {
917 StoredValTy = TD.getIntPtrType(StoredValTy->getContext());
918 StoredVal = new PtrToIntInst(StoredVal, StoredValTy, "", InsertPt);
919 }
920
921 // Convert vectors and fp to integer, which can be manipulated.
922 if (!isa<IntegerType>(StoredValTy)) {
923 StoredValTy = IntegerType::get(StoredValTy->getContext(), StoreSize);
924 StoredVal = new BitCastInst(StoredVal, StoredValTy, "", InsertPt);
925 }
926
927 // If this is a big-endian system, we need to shift the value down to the low
928 // bits so that a truncate will work.
929 if (TD.isBigEndian()) {
930 Constant *Val = ConstantInt::get(StoredVal->getType(), StoreSize-LoadSize);
931 StoredVal = BinaryOperator::CreateLShr(StoredVal, Val, "tmp", InsertPt);
932 }
933
934 // Truncate the integer to the right size now.
935 const Type *NewIntTy = IntegerType::get(StoredValTy->getContext(), LoadSize);
936 StoredVal = new TruncInst(StoredVal, NewIntTy, "trunc", InsertPt);
937
938 if (LoadedTy == NewIntTy)
939 return StoredVal;
940
941 // If the result is a pointer, inttoptr.
942 if (isa<PointerType>(LoadedTy))
943 return new IntToPtrInst(StoredVal, LoadedTy, "inttoptr", InsertPt);
944
945 // Otherwise, bitcast.
946 return new BitCastInst(StoredVal, LoadedTy, "bitcast", InsertPt);
947}
948
Chris Lattner8f912082009-09-21 06:24:16 +0000949/// GetBaseWithConstantOffset - Analyze the specified pointer to see if it can
950/// be expressed as a base pointer plus a constant offset. Return the base and
951/// offset to the caller.
952static Value *GetBaseWithConstantOffset(Value *Ptr, int64_t &Offset,
Chris Lattneraae7fcb2009-09-21 06:48:08 +0000953 const TargetData &TD) {
Chris Lattner8f912082009-09-21 06:24:16 +0000954 Operator *PtrOp = dyn_cast<Operator>(Ptr);
955 if (PtrOp == 0) return Ptr;
956
957 // Just look through bitcasts.
958 if (PtrOp->getOpcode() == Instruction::BitCast)
959 return GetBaseWithConstantOffset(PtrOp->getOperand(0), Offset, TD);
960
961 // If this is a GEP with constant indices, we can look through it.
962 GEPOperator *GEP = dyn_cast<GEPOperator>(PtrOp);
963 if (GEP == 0 || !GEP->hasAllConstantIndices()) return Ptr;
964
965 gep_type_iterator GTI = gep_type_begin(GEP);
966 for (User::op_iterator I = GEP->idx_begin(), E = GEP->idx_end(); I != E;
967 ++I, ++GTI) {
968 ConstantInt *OpC = cast<ConstantInt>(*I);
969 if (OpC->isZero()) continue;
970
971 // Handle a struct and array indices which add their offset to the pointer.
972 if (const StructType *STy = dyn_cast<StructType>(*GTI)) {
Chris Lattneraae7fcb2009-09-21 06:48:08 +0000973 Offset += TD.getStructLayout(STy)->getElementOffset(OpC->getZExtValue());
Chris Lattner8f912082009-09-21 06:24:16 +0000974 } else {
Chris Lattneraae7fcb2009-09-21 06:48:08 +0000975 uint64_t Size = TD.getTypeAllocSize(GTI.getIndexedType());
Chris Lattner8f912082009-09-21 06:24:16 +0000976 Offset += OpC->getSExtValue()*Size;
977 }
978 }
979
980 // Re-sign extend from the pointer size if needed to get overflow edge cases
981 // right.
Chris Lattneraae7fcb2009-09-21 06:48:08 +0000982 unsigned PtrSize = TD.getPointerSizeInBits();
Chris Lattner8f912082009-09-21 06:24:16 +0000983 if (PtrSize < 64)
984 Offset = (Offset << (64-PtrSize)) >> (64-PtrSize);
985
986 return GetBaseWithConstantOffset(GEP->getPointerOperand(), Offset, TD);
987}
988
989
990/// AnalyzeLoadFromClobberingStore - This function is called when we have a
991/// memdep query of a load that ends up being a clobbering store. This means
992/// that the store *may* provide bits used by the load but we can't be sure
993/// because the pointers don't mustalias. Check this case to see if there is
994/// anything more we can do before we give up. This returns -1 if we have to
995/// give up, or a byte number in the stored value of the piece that feeds the
996/// load.
997static int AnalyzeLoadFromClobberingStore(LoadInst *L, StoreInst *DepSI,
Chris Lattneraae7fcb2009-09-21 06:48:08 +0000998 const TargetData &TD) {
Chris Lattner012b3602009-09-21 17:24:04 +0000999 // If the loaded or stored value is an first class array or struct, don't try
1000 // to transform them. We need to be able to bitcast to integer.
1001 if (isa<StructType>(L->getType()) || isa<ArrayType>(L->getType()) ||
1002 isa<StructType>(DepSI->getOperand(0)->getType()) ||
1003 isa<ArrayType>(DepSI->getOperand(0)->getType()))
1004 return -1;
1005
Chris Lattner8f912082009-09-21 06:24:16 +00001006 int64_t StoreOffset = 0, LoadOffset = 0;
1007 Value *StoreBase =
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001008 GetBaseWithConstantOffset(DepSI->getPointerOperand(), 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"
1021 << "Store Ptr = " << *DepSI->getPointerOperand() << "\n"
1022 << "Store Offs = " << StoreOffset << " - " << *DepSI << "\n"
1023 << "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 StoreSize = TD.getTypeSizeInBits(DepSI->getOperand(0)->getType());
1037 uint64_t LoadSize = TD.getTypeSizeInBits(L->getType());
Chris Lattner8f912082009-09-21 06:24:16 +00001038
1039 if ((StoreSize & 7) | (LoadSize & 7))
1040 return -1;
1041 StoreSize >>= 3; // Convert to bytes.
1042 LoadSize >>= 3;
1043
1044
1045 bool isAAFailure = false;
1046 if (StoreOffset < LoadOffset) {
1047 isAAFailure = StoreOffset+int64_t(StoreSize) <= LoadOffset;
1048 } else {
1049 isAAFailure = LoadOffset+int64_t(LoadSize) <= StoreOffset;
1050 }
1051 if (isAAFailure) {
1052#if 0
1053 errs() << "STORE LOAD DEP WITH COMMON BASE:\n"
1054 << "Base = " << *StoreBase << "\n"
1055 << "Store Ptr = " << *DepSI->getPointerOperand() << "\n"
1056 << "Store Offs = " << StoreOffset << " - " << *DepSI << "\n"
1057 << "Load Ptr = " << *L->getPointerOperand() << "\n"
1058 << "Load Offs = " << LoadOffset << " - " << *L << "\n\n";
1059 errs() << "'" << L->getParent()->getParent()->getName() << "'"
1060 << *L->getParent();
1061#endif
1062 return -1;
1063 }
1064
1065 // If the Load isn't completely contained within the stored bits, we don't
1066 // have all the bits to feed it. We could do something crazy in the future
1067 // (issue a smaller load then merge the bits in) but this seems unlikely to be
1068 // valuable.
1069 if (StoreOffset > LoadOffset ||
1070 StoreOffset+StoreSize < LoadOffset+LoadSize)
1071 return -1;
1072
1073 // Okay, we can do this transformation. Return the number of bytes into the
1074 // store that the load is.
1075 return LoadOffset-StoreOffset;
1076}
1077
1078
1079/// GetStoreValueForLoad - This function is called when we have a
1080/// memdep query of a load that ends up being a clobbering store. This means
1081/// that the store *may* provide bits used by the load but we can't be sure
1082/// because the pointers don't mustalias. Check this case to see if there is
1083/// anything more we can do before we give up.
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001084static Value *GetStoreValueForLoad(Value *SrcVal, unsigned Offset,
1085 const Type *LoadTy,
1086 Instruction *InsertPt, const TargetData &TD){
Chris Lattner8f912082009-09-21 06:24:16 +00001087 LLVMContext &Ctx = SrcVal->getType()->getContext();
1088
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001089 uint64_t StoreSize = TD.getTypeSizeInBits(SrcVal->getType())/8;
1090 uint64_t LoadSize = TD.getTypeSizeInBits(LoadTy)/8;
Chris Lattner8f912082009-09-21 06:24:16 +00001091
1092
1093 // Compute which bits of the stored value are being used by the load. Convert
1094 // to an integer type to start with.
1095 if (isa<PointerType>(SrcVal->getType()))
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001096 SrcVal = new PtrToIntInst(SrcVal, TD.getIntPtrType(Ctx), "tmp", InsertPt);
Chris Lattner8f912082009-09-21 06:24:16 +00001097 if (!isa<IntegerType>(SrcVal->getType()))
1098 SrcVal = new BitCastInst(SrcVal, IntegerType::get(Ctx, StoreSize*8),
1099 "tmp", InsertPt);
1100
1101 // Shift the bits to the least significant depending on endianness.
1102 unsigned ShiftAmt;
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001103 if (TD.isLittleEndian()) {
Chris Lattner8f912082009-09-21 06:24:16 +00001104 ShiftAmt = Offset*8;
1105 } else {
Chris Lattner1846fa02009-09-21 17:55:47 +00001106 ShiftAmt = (StoreSize-LoadSize-Offset)*8;
Chris Lattner8f912082009-09-21 06:24:16 +00001107 }
1108
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001109 if (ShiftAmt)
1110 SrcVal = BinaryOperator::CreateLShr(SrcVal,
1111 ConstantInt::get(SrcVal->getType(), ShiftAmt), "tmp", InsertPt);
Chris Lattner8f912082009-09-21 06:24:16 +00001112
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001113 if (LoadSize != StoreSize)
1114 SrcVal = new TruncInst(SrcVal, IntegerType::get(Ctx, LoadSize*8),
1115 "tmp", InsertPt);
Chris Lattner8f912082009-09-21 06:24:16 +00001116
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001117 return CoerceAvailableValueToLoadType(SrcVal, LoadTy, InsertPt, TD);
Chris Lattner8f912082009-09-21 06:24:16 +00001118}
1119
Chris Lattner19b84b32009-09-21 06:30:24 +00001120struct AvailableValueInBlock {
1121 /// BB - The basic block in question.
1122 BasicBlock *BB;
1123 /// V - The value that is live out of the block.
1124 Value *V;
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001125 /// Offset - The byte offset in V that is interesting for the load query.
1126 unsigned Offset;
Chris Lattner19b84b32009-09-21 06:30:24 +00001127
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001128 static AvailableValueInBlock get(BasicBlock *BB, Value *V,
1129 unsigned Offset = 0) {
Chris Lattner19b84b32009-09-21 06:30:24 +00001130 AvailableValueInBlock Res;
1131 Res.BB = BB;
1132 Res.V = V;
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001133 Res.Offset = Offset;
Chris Lattner19b84b32009-09-21 06:30:24 +00001134 return Res;
1135 }
1136};
1137
Chris Lattner6e5ea272009-10-10 23:50:30 +00001138/// ConstructSSAForLoadSet - Given a set of loads specified by ValuesPerBlock,
1139/// construct SSA form, allowing us to eliminate LI. This returns the value
1140/// that should be used at LI's definition site.
1141static Value *ConstructSSAForLoadSet(LoadInst *LI,
1142 SmallVectorImpl<AvailableValueInBlock> &ValuesPerBlock,
1143 const TargetData *TD,
1144 AliasAnalysis *AA) {
1145 SmallVector<PHINode*, 8> NewPHIs;
1146 SSAUpdater SSAUpdate(&NewPHIs);
1147 SSAUpdate.Initialize(LI);
1148
1149 const Type *LoadTy = LI->getType();
1150
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001151 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i) {
Chris Lattner19b84b32009-09-21 06:30:24 +00001152 BasicBlock *BB = ValuesPerBlock[i].BB;
1153 Value *AvailableVal = ValuesPerBlock[i].V;
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001154 unsigned Offset = ValuesPerBlock[i].Offset;
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001155
Chris Lattner6e5ea272009-10-10 23:50:30 +00001156 if (SSAUpdate.HasValueForBlock(BB))
1157 continue;
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001158
1159 if (AvailableVal->getType() != LoadTy) {
1160 assert(TD && "Need target data to handle type mismatch case");
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001161 AvailableVal = GetStoreValueForLoad(AvailableVal, Offset, LoadTy,
1162 BB->getTerminator(), *TD);
1163
1164 if (Offset) {
1165 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\n"
Chris Lattner6e5ea272009-10-10 23:50:30 +00001166 << *ValuesPerBlock[i].V << '\n'
1167 << *AvailableVal << '\n' << "\n\n\n");
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001168 }
1169
1170
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001171 DEBUG(errs() << "GVN COERCED NONLOCAL VAL:\n"
Chris Lattner6e5ea272009-10-10 23:50:30 +00001172 << *ValuesPerBlock[i].V << '\n'
1173 << *AvailableVal << '\n' << "\n\n\n");
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001174 }
Chris Lattner6e5ea272009-10-10 23:50:30 +00001175
1176 SSAUpdate.AddAvailableValue(BB, AvailableVal);
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001177 }
Chris Lattner6e5ea272009-10-10 23:50:30 +00001178
1179 // Perform PHI construction.
1180 Value *V = SSAUpdate.GetValueInMiddleOfBlock(LI->getParent());
1181
1182 // If new PHI nodes were created, notify alias analysis.
1183 if (isa<PointerType>(V->getType()))
1184 for (unsigned i = 0, e = NewPHIs.size(); i != e; ++i)
1185 AA->copyValue(LI, NewPHIs[i]);
1186
1187 return V;
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001188}
1189
Owen Andersonf187daf2009-12-02 07:35:19 +00001190static bool isLifetimeStart(Instruction *Inst) {
Chris Lattnerbc6fccc2009-12-02 06:44:58 +00001191 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(Inst))
Owen Andersonf187daf2009-12-02 07:35:19 +00001192 return II->getIntrinsicID() == Intrinsic::lifetime_start;
Chris Lattnerbc6fccc2009-12-02 06:44:58 +00001193 return false;
1194}
1195
Owen Andersone0143452007-08-16 22:02:55 +00001196/// processNonLocalLoad - Attempt to eliminate a load whose dependencies are
1197/// non-local by performing PHI construction.
Chris Lattnerdcded152008-12-02 08:16:11 +00001198bool GVN::processNonLocalLoad(LoadInst *LI,
Chris Lattner7de20452008-03-21 22:01:16 +00001199 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattnerdcded152008-12-02 08:16:11 +00001200 // Find the non-local dependencies of the load.
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001201 SmallVector<MemoryDependenceAnalysis::NonLocalDepEntry, 64> Deps;
Chris Lattneraf713862008-12-09 19:25:07 +00001202 MD->getNonLocalPointerDependency(LI->getOperand(0), true, LI->getParent(),
1203 Deps);
Dan Gohman7e124382009-07-31 20:24:18 +00001204 //DEBUG(errs() << "INVESTIGATING NONLOCAL LOAD: "
1205 // << Deps.size() << *LI << '\n');
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001206
Owen Anderson90e717d2008-08-26 22:07:42 +00001207 // If we had to process more than one hundred blocks to find the
1208 // dependencies, this load isn't worth worrying about. Optimizing
1209 // it will be too expensive.
Chris Lattneraf713862008-12-09 19:25:07 +00001210 if (Deps.size() > 100)
Owen Anderson90e717d2008-08-26 22:07:42 +00001211 return false;
Chris Lattner8d1686f2008-12-18 00:51:32 +00001212
1213 // If we had a phi translation failure, we'll have a single entry which is a
1214 // clobber in the current block. Reject this early.
Edwin Török3ffffac2009-06-17 18:48:18 +00001215 if (Deps.size() == 1 && Deps[0].second.isClobber()) {
1216 DEBUG(
Dan Gohman0be10b02009-07-25 01:43:01 +00001217 errs() << "GVN: non-local load ";
1218 WriteAsOperand(errs(), LI);
Dan Gohman7e124382009-07-31 20:24:18 +00001219 errs() << " is clobbered by " << *Deps[0].second.getInst() << '\n';
Edwin Török3ffffac2009-06-17 18:48:18 +00001220 );
Chris Lattner8d1686f2008-12-18 00:51:32 +00001221 return false;
Edwin Török3ffffac2009-06-17 18:48:18 +00001222 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001223
Chris Lattnerdcded152008-12-02 08:16:11 +00001224 // Filter out useless results (non-locals, etc). Keep track of the blocks
1225 // where we have a value available in repl, also keep track of whether we see
1226 // dependencies that produce an unknown value for the load (such as a call
1227 // that could potentially clobber the load).
Chris Lattner19b84b32009-09-21 06:30:24 +00001228 SmallVector<AvailableValueInBlock, 16> ValuesPerBlock;
Chris Lattnerdcded152008-12-02 08:16:11 +00001229 SmallVector<BasicBlock*, 16> UnavailableBlocks;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001230
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001231 const TargetData *TD = 0;
1232
Chris Lattneraf713862008-12-09 19:25:07 +00001233 for (unsigned i = 0, e = Deps.size(); i != e; ++i) {
1234 BasicBlock *DepBB = Deps[i].first;
1235 MemDepResult DepInfo = Deps[i].second;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001236
Chris Lattner4531da82008-12-05 21:04:20 +00001237 if (DepInfo.isClobber()) {
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001238 // If the dependence is to a store that writes to a superset of the bits
1239 // read by the load, we can extract the bits we need for the load from the
1240 // stored value.
1241 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInfo.getInst())) {
1242 if (TD == 0)
1243 TD = getAnalysisIfAvailable<TargetData>();
1244 if (TD) {
1245 int Offset = AnalyzeLoadFromClobberingStore(LI, DepSI, *TD);
1246 if (Offset != -1) {
1247 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1248 DepSI->getOperand(0),
1249 Offset));
1250 continue;
1251 }
1252 }
1253 }
1254
1255 // FIXME: Handle memset/memcpy.
Chris Lattner4531da82008-12-05 21:04:20 +00001256 UnavailableBlocks.push_back(DepBB);
1257 continue;
1258 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001259
Chris Lattner4531da82008-12-05 21:04:20 +00001260 Instruction *DepInst = DepInfo.getInst();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001261
Chris Lattner4531da82008-12-05 21:04:20 +00001262 // Loading the allocation -> undef.
Chris Lattnerbc6fccc2009-12-02 06:44:58 +00001263 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst) ||
Owen Andersonf187daf2009-12-02 07:35:19 +00001264 // Loading immediately after lifetime begin -> undef.
1265 isLifetimeStart(DepInst)) {
Chris Lattner19b84b32009-09-21 06:30:24 +00001266 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1267 UndefValue::get(LI->getType())));
Chris Lattner46876282008-12-01 01:15:42 +00001268 continue;
1269 }
Owen Andersonc07861a2009-10-28 07:05:35 +00001270
Chris Lattner19b84b32009-09-21 06:30:24 +00001271 if (StoreInst *S = dyn_cast<StoreInst>(DepInst)) {
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001272 // Reject loads and stores that are to the same address but are of
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001273 // different types if we have to.
Chris Lattnerdcded152008-12-02 08:16:11 +00001274 if (S->getOperand(0)->getType() != LI->getType()) {
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001275 if (TD == 0)
1276 TD = getAnalysisIfAvailable<TargetData>();
1277
1278 // If the stored value is larger or equal to the loaded value, we can
1279 // reuse it.
Chris Lattner012b3602009-09-21 17:24:04 +00001280 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(S->getOperand(0),
1281 LI->getType(), *TD)) {
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001282 UnavailableBlocks.push_back(DepBB);
1283 continue;
1284 }
Chris Lattnerdcded152008-12-02 08:16:11 +00001285 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001286
Chris Lattner19b84b32009-09-21 06:30:24 +00001287 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB,
1288 S->getOperand(0)));
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001289 continue;
1290 }
1291
1292 if (LoadInst *LD = dyn_cast<LoadInst>(DepInst)) {
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001293 // If the types mismatch and we can't handle it, reject reuse of the load.
Chris Lattnerdcded152008-12-02 08:16:11 +00001294 if (LD->getType() != LI->getType()) {
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001295 if (TD == 0)
1296 TD = getAnalysisIfAvailable<TargetData>();
1297
1298 // If the stored value is larger or equal to the loaded value, we can
1299 // reuse it.
Chris Lattner012b3602009-09-21 17:24:04 +00001300 if (TD == 0 || !CanCoerceMustAliasedValueToLoad(LD, LI->getType(),*TD)){
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001301 UnavailableBlocks.push_back(DepBB);
1302 continue;
1303 }
Chris Lattnerdcded152008-12-02 08:16:11 +00001304 }
Chris Lattner19b84b32009-09-21 06:30:24 +00001305 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB, LD));
Chris Lattnerdcded152008-12-02 08:16:11 +00001306 continue;
Owen Anderson5d72a422007-07-25 19:57:03 +00001307 }
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001308
1309 UnavailableBlocks.push_back(DepBB);
1310 continue;
Chris Lattner3d7103e2008-03-21 21:14:38 +00001311 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001312
Chris Lattnerdcded152008-12-02 08:16:11 +00001313 // If we have no predecessors that produce a known value for this load, exit
1314 // early.
1315 if (ValuesPerBlock.empty()) return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001316
Chris Lattnerdcded152008-12-02 08:16:11 +00001317 // If all of the instructions we depend on produce a known value for this
1318 // load, then it is fully redundant and we can use PHI insertion to compute
1319 // its value. Insert PHIs and remove the fully redundant value now.
1320 if (UnavailableBlocks.empty()) {
Dan Gohman7e124382009-07-31 20:24:18 +00001321 DEBUG(errs() << "GVN REMOVING NONLOCAL LOAD: " << *LI << '\n');
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001322
Chris Lattnerdcded152008-12-02 08:16:11 +00001323 // Perform PHI construction.
Chris Lattner6e5ea272009-10-10 23:50:30 +00001324 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD,
1325 VN.getAliasAnalysis());
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001326 LI->replaceAllUsesWith(V);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001327
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001328 if (isa<PHINode>(V))
1329 V->takeName(LI);
1330 if (isa<PointerType>(V->getType()))
1331 MD->invalidateCachedPointerInfo(V);
Chris Lattnerdcded152008-12-02 08:16:11 +00001332 toErase.push_back(LI);
1333 NumGVNLoad++;
1334 return true;
1335 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001336
Chris Lattnerdcded152008-12-02 08:16:11 +00001337 if (!EnablePRE || !EnableLoadPRE)
1338 return false;
1339
1340 // Okay, we have *some* definitions of the value. This means that the value
1341 // is available in some of our (transitive) predecessors. Lets think about
1342 // doing PRE of this load. This will involve inserting a new load into the
1343 // predecessor when it's not available. We could do this in general, but
1344 // prefer to not increase code size. As such, we only do this when we know
1345 // that we only have to insert *one* load (which means we're basically moving
1346 // the load, not inserting a new one).
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001347
Owen Andersondd37b182009-05-31 09:03:40 +00001348 SmallPtrSet<BasicBlock *, 4> Blockers;
1349 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1350 Blockers.insert(UnavailableBlocks[i]);
1351
1352 // Lets find first basic block with more than one predecessor. Walk backwards
1353 // through predecessors if needed.
Chris Lattnerdcded152008-12-02 08:16:11 +00001354 BasicBlock *LoadBB = LI->getParent();
Owen Andersondd37b182009-05-31 09:03:40 +00001355 BasicBlock *TmpBB = LoadBB;
1356
1357 bool isSinglePred = false;
Dale Johannesena19b67f2009-06-17 20:48:23 +00001358 bool allSingleSucc = true;
Owen Andersondd37b182009-05-31 09:03:40 +00001359 while (TmpBB->getSinglePredecessor()) {
1360 isSinglePred = true;
1361 TmpBB = TmpBB->getSinglePredecessor();
1362 if (!TmpBB) // If haven't found any, bail now.
1363 return false;
1364 if (TmpBB == LoadBB) // Infinite (unreachable) loop.
1365 return false;
1366 if (Blockers.count(TmpBB))
1367 return false;
Dale Johannesena19b67f2009-06-17 20:48:23 +00001368 if (TmpBB->getTerminator()->getNumSuccessors() != 1)
1369 allSingleSucc = false;
Owen Andersondd37b182009-05-31 09:03:40 +00001370 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001371
Owen Andersondd37b182009-05-31 09:03:40 +00001372 assert(TmpBB);
1373 LoadBB = TmpBB;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001374
Chris Lattnerdcded152008-12-02 08:16:11 +00001375 // If we have a repl set with LI itself in it, this means we have a loop where
1376 // at least one of the values is LI. Since this means that we won't be able
1377 // to eliminate LI even if we insert uses in the other predecessors, we will
1378 // end up increasing code size. Reject this by scanning for LI.
1379 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner19b84b32009-09-21 06:30:24 +00001380 if (ValuesPerBlock[i].V == LI)
Chris Lattnerdcded152008-12-02 08:16:11 +00001381 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001382
Owen Andersondd37b182009-05-31 09:03:40 +00001383 if (isSinglePred) {
1384 bool isHot = false;
1385 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner19b84b32009-09-21 06:30:24 +00001386 if (Instruction *I = dyn_cast<Instruction>(ValuesPerBlock[i].V))
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001387 // "Hot" Instruction is in some loop (because it dominates its dep.
1388 // instruction).
1389 if (DT->dominates(LI, I)) {
1390 isHot = true;
1391 break;
1392 }
Owen Andersondd37b182009-05-31 09:03:40 +00001393
1394 // We are interested only in "hot" instructions. We don't want to do any
1395 // mis-optimizations here.
1396 if (!isHot)
1397 return false;
1398 }
1399
Chris Lattnerdcded152008-12-02 08:16:11 +00001400 // Okay, we have some hope :). Check to see if the loaded value is fully
1401 // available in all but one predecessor.
1402 // FIXME: If we could restructure the CFG, we could make a common pred with
1403 // all the preds that don't have an available LI and insert a new load into
1404 // that one block.
1405 BasicBlock *UnavailablePred = 0;
1406
Chris Lattner159b98f2008-12-05 07:49:08 +00001407 DenseMap<BasicBlock*, char> FullyAvailableBlocks;
Chris Lattnerdcded152008-12-02 08:16:11 +00001408 for (unsigned i = 0, e = ValuesPerBlock.size(); i != e; ++i)
Chris Lattner19b84b32009-09-21 06:30:24 +00001409 FullyAvailableBlocks[ValuesPerBlock[i].BB] = true;
Chris Lattnerdcded152008-12-02 08:16:11 +00001410 for (unsigned i = 0, e = UnavailableBlocks.size(); i != e; ++i)
1411 FullyAvailableBlocks[UnavailableBlocks[i]] = false;
1412
1413 for (pred_iterator PI = pred_begin(LoadBB), E = pred_end(LoadBB);
1414 PI != E; ++PI) {
1415 if (IsValueFullyAvailableInBlock(*PI, FullyAvailableBlocks))
1416 continue;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001417
Chris Lattnerdcded152008-12-02 08:16:11 +00001418 // If this load is not available in multiple predecessors, reject it.
1419 if (UnavailablePred && UnavailablePred != *PI)
1420 return false;
1421 UnavailablePred = *PI;
1422 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001423
Chris Lattnerdcded152008-12-02 08:16:11 +00001424 assert(UnavailablePred != 0 &&
1425 "Fully available value should be eliminated above!");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001426
Chris Lattnerdcded152008-12-02 08:16:11 +00001427 // We don't currently handle critical edges :(
1428 if (UnavailablePred->getTerminator()->getNumSuccessors() != 1) {
Daniel Dunbar005975c2009-07-25 00:23:56 +00001429 DEBUG(errs() << "COULD NOT PRE LOAD BECAUSE OF CRITICAL EDGE '"
Dan Gohman7e124382009-07-31 20:24:18 +00001430 << UnavailablePred->getName() << "': " << *LI << '\n');
Chris Lattnerdcded152008-12-02 08:16:11 +00001431 return false;
Owen Anderson5b299672007-08-07 23:12:31 +00001432 }
Chris Lattner248818e2009-11-27 08:25:10 +00001433
Chris Lattnera5bef152009-11-27 22:05:15 +00001434 // Do PHI translation to get its value in the predecessor if necessary. The
1435 // returned pointer (if non-null) is guaranteed to dominate UnavailablePred.
1436 //
1437 // FIXME: This may insert a computation, but we don't tell scalar GVN
1438 // optimization stuff about it. How do we do this?
Chris Lattner1c2de2b2009-11-28 15:39:14 +00001439 SmallVector<Instruction*, 8> NewInsts;
Chris Lattner80c535b2009-11-28 16:08:18 +00001440 Value *LoadPtr = 0;
Chris Lattnerde0b0302009-11-27 22:50:07 +00001441
Chris Lattner80c535b2009-11-28 16:08:18 +00001442 // If all preds have a single successor, then we know it is safe to insert the
1443 // load on the pred (?!?), so we can insert code to materialize the pointer if
1444 // it is not available.
1445 if (allSingleSucc) {
1446 LoadPtr = MD->InsertPHITranslatedPointer(LI->getOperand(0), LoadBB,
1447 UnavailablePred, TD, *DT,NewInsts);
1448 } else {
1449 LoadPtr = MD->GetAvailablePHITranslatedValue(LI->getOperand(0), LoadBB,
1450 UnavailablePred, TD, *DT);
1451 }
Owen Anderson2c405b92009-12-03 03:43:29 +00001452
1453 // Assign value numbers to these new instructions.
1454 for (SmallVector<Instruction*, 8>::iterator NI = NewInsts.begin(),
1455 NE = NewInsts.end(); NI != NE; ++NI) {
1456 // FIXME: We really _ought_ to insert these value numbers into their
1457 // parent's availability map. However, in doing so, we risk getting into
1458 // ordering issues. If a block hasn't been processed yet, we would be
1459 // marking a value as AVAIL-IN, which isn't what we intend.
1460 VN.lookup_or_add(*NI);
1461 }
Chris Lattner80c535b2009-11-28 16:08:18 +00001462
Chris Lattnera5bef152009-11-27 22:05:15 +00001463 // If we couldn't find or insert a computation of this phi translated value,
1464 // we fail PRE.
Chris Lattner248818e2009-11-27 08:25:10 +00001465 if (LoadPtr == 0) {
Chris Lattnera5bef152009-11-27 22:05:15 +00001466 DEBUG(errs() << "COULDN'T INSERT PHI TRANSLATED VALUE OF: "
1467 << *LI->getOperand(0) << "\n");
1468 return false;
Chris Lattner248818e2009-11-27 08:25:10 +00001469 }
1470
Dale Johannesena19b67f2009-06-17 20:48:23 +00001471 // Make sure it is valid to move this load here. We have to watch out for:
1472 // @1 = getelementptr (i8* p, ...
1473 // test p and branch if == 0
1474 // load @1
1475 // It is valid to have the getelementptr before the test, even if p can be 0,
1476 // as getelementptr only does address arithmetic.
1477 // If we are not pushing the value through any multiple-successor blocks
1478 // we do not have this case. Otherwise, check that the load is safe to
1479 // put anywhere; this can be improved, but should be conservatively safe.
1480 if (!allSingleSucc &&
Chris Lattner1c2de2b2009-11-28 15:39:14 +00001481 // FIXME: REEVALUTE THIS.
Chris Lattner80c535b2009-11-28 16:08:18 +00001482 !isSafeToLoadUnconditionally(LoadPtr, UnavailablePred->getTerminator())) {
1483 assert(NewInsts.empty() && "Should not have inserted instructions");
Dale Johannesena19b67f2009-06-17 20:48:23 +00001484 return false;
Chris Lattner80c535b2009-11-28 16:08:18 +00001485 }
Dale Johannesena19b67f2009-06-17 20:48:23 +00001486
Chris Lattnerdcded152008-12-02 08:16:11 +00001487 // Okay, we can eliminate this load by inserting a reload in the predecessor
1488 // and using PHI construction to get the value in the other predecessors, do
1489 // it.
Dan Gohman7e124382009-07-31 20:24:18 +00001490 DEBUG(errs() << "GVN REMOVING PRE LOAD: " << *LI << '\n');
Chris Lattner80c535b2009-11-28 16:08:18 +00001491 DEBUG(if (!NewInsts.empty())
1492 errs() << "INSERTED " << NewInsts.size() << " INSTS: "
1493 << *NewInsts.back() << '\n');
1494
Chris Lattnerdcded152008-12-02 08:16:11 +00001495 Value *NewLoad = new LoadInst(LoadPtr, LI->getName()+".pre", false,
1496 LI->getAlignment(),
1497 UnavailablePred->getTerminator());
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001498
Chris Lattner6e5ea272009-10-10 23:50:30 +00001499 // Add the newly created load.
1500 ValuesPerBlock.push_back(AvailableValueInBlock::get(UnavailablePred,NewLoad));
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001501
Chris Lattnerdcded152008-12-02 08:16:11 +00001502 // Perform PHI construction.
Chris Lattner6e5ea272009-10-10 23:50:30 +00001503 Value *V = ConstructSSAForLoadSet(LI, ValuesPerBlock, TD,
1504 VN.getAliasAnalysis());
Chris Lattnerd6b1d052009-09-20 20:09:34 +00001505 LI->replaceAllUsesWith(V);
1506 if (isa<PHINode>(V))
1507 V->takeName(LI);
1508 if (isa<PointerType>(V->getType()))
1509 MD->invalidateCachedPointerInfo(V);
Chris Lattnerdcded152008-12-02 08:16:11 +00001510 toErase.push_back(LI);
1511 NumPRELoad++;
Owen Anderson5d72a422007-07-25 19:57:03 +00001512 return true;
1513}
1514
Owen Andersone0143452007-08-16 22:02:55 +00001515/// processLoad - Attempt to eliminate a load, first by eliminating it
1516/// locally, and then attempting non-local elimination if that fails.
Chris Lattner4531da82008-12-05 21:04:20 +00001517bool GVN::processLoad(LoadInst *L, SmallVectorImpl<Instruction*> &toErase) {
Dan Gohmanc8d26652009-11-14 02:27:51 +00001518 if (!MD)
1519 return false;
1520
Chris Lattner4531da82008-12-05 21:04:20 +00001521 if (L->isVolatile())
Owen Anderson85c40642007-07-24 17:55:58 +00001522 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001523
Owen Anderson85c40642007-07-24 17:55:58 +00001524 // ... to a pointer that has been loaded from before...
Chris Lattnerff36c952009-09-21 02:42:51 +00001525 MemDepResult Dep = MD->getDependency(L);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001526
Chris Lattner4531da82008-12-05 21:04:20 +00001527 // If the value isn't available, don't do anything!
Chris Lattnerff36c952009-09-21 02:42:51 +00001528 if (Dep.isClobber()) {
Chris Lattner0907b522009-09-21 05:57:11 +00001529 // FIXME: We should handle memset/memcpy/memmove as dependent instructions
1530 // to forward the value if available.
1531 //if (isa<MemIntrinsic>(Dep.getInst()))
1532 //errs() << "LOAD DEPENDS ON MEM: " << *L << "\n" << *Dep.getInst()<<"\n\n";
1533
1534 // Check to see if we have something like this:
Chris Lattner7741aa52009-09-20 19:03:47 +00001535 // store i32 123, i32* %P
1536 // %A = bitcast i32* %P to i8*
1537 // %B = gep i8* %A, i32 1
1538 // %C = load i8* %B
1539 //
1540 // We could do that by recognizing if the clobber instructions are obviously
1541 // a common base + constant offset, and if the previous store (or memset)
1542 // completely covers this load. This sort of thing can happen in bitfield
1543 // access code.
Chris Lattner0907b522009-09-21 05:57:11 +00001544 if (StoreInst *DepSI = dyn_cast<StoreInst>(Dep.getInst()))
Chris Lattner41eb59c2009-09-21 06:22:46 +00001545 if (const TargetData *TD = getAnalysisIfAvailable<TargetData>()) {
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001546 int Offset = AnalyzeLoadFromClobberingStore(L, DepSI, *TD);
Chris Lattner41eb59c2009-09-21 06:22:46 +00001547 if (Offset != -1) {
1548 Value *AvailVal = GetStoreValueForLoad(DepSI->getOperand(0), Offset,
Chris Lattneraae7fcb2009-09-21 06:48:08 +00001549 L->getType(), L, *TD);
Chris Lattner0907b522009-09-21 05:57:11 +00001550 DEBUG(errs() << "GVN COERCED STORE BITS:\n" << *DepSI << '\n'
1551 << *AvailVal << '\n' << *L << "\n\n\n");
1552
1553 // Replace the load!
1554 L->replaceAllUsesWith(AvailVal);
1555 if (isa<PointerType>(AvailVal->getType()))
1556 MD->invalidateCachedPointerInfo(AvailVal);
1557 toErase.push_back(L);
1558 NumGVNLoad++;
1559 return true;
1560 }
Chris Lattner41eb59c2009-09-21 06:22:46 +00001561 }
Chris Lattner0907b522009-09-21 05:57:11 +00001562
Edwin Török47cf8842009-05-29 09:46:03 +00001563 DEBUG(
1564 // fast print dep, using operator<< on instruction would be too slow
Dan Gohman0be10b02009-07-25 01:43:01 +00001565 errs() << "GVN: load ";
1566 WriteAsOperand(errs(), L);
Chris Lattnerff36c952009-09-21 02:42:51 +00001567 Instruction *I = Dep.getInst();
Dan Gohman7e124382009-07-31 20:24:18 +00001568 errs() << " is clobbered by " << *I << '\n';
Edwin Török47cf8842009-05-29 09:46:03 +00001569 );
Chris Lattner4531da82008-12-05 21:04:20 +00001570 return false;
Edwin Török47cf8842009-05-29 09:46:03 +00001571 }
Chris Lattner4531da82008-12-05 21:04:20 +00001572
1573 // If it is defined in another block, try harder.
Chris Lattnerff36c952009-09-21 02:42:51 +00001574 if (Dep.isNonLocal())
Chris Lattner4531da82008-12-05 21:04:20 +00001575 return processNonLocalLoad(L, toErase);
Eli Friedman350307f2008-02-12 12:08:14 +00001576
Chris Lattnerff36c952009-09-21 02:42:51 +00001577 Instruction *DepInst = Dep.getInst();
Chris Lattner4531da82008-12-05 21:04:20 +00001578 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInst)) {
Chris Lattner7741aa52009-09-20 19:03:47 +00001579 Value *StoredVal = DepSI->getOperand(0);
1580
1581 // The store and load are to a must-aliased pointer, but they may not
1582 // actually have the same type. See if we know how to reuse the stored
1583 // value (depending on its type).
1584 const TargetData *TD = 0;
Chris Lattner10460aa2009-10-21 04:11:19 +00001585 if (StoredVal->getType() != L->getType()) {
1586 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1587 StoredVal = CoerceAvailableValueToLoadType(StoredVal, L->getType(),
1588 L, *TD);
1589 if (StoredVal == 0)
1590 return false;
1591
1592 DEBUG(errs() << "GVN COERCED STORE:\n" << *DepSI << '\n' << *StoredVal
1593 << '\n' << *L << "\n\n\n");
1594 }
1595 else
Chris Lattner7741aa52009-09-20 19:03:47 +00001596 return false;
Chris Lattner7741aa52009-09-20 19:03:47 +00001597 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001598
Chris Lattner4531da82008-12-05 21:04:20 +00001599 // Remove it!
Chris Lattner7741aa52009-09-20 19:03:47 +00001600 L->replaceAllUsesWith(StoredVal);
1601 if (isa<PointerType>(StoredVal->getType()))
1602 MD->invalidateCachedPointerInfo(StoredVal);
Chris Lattner4531da82008-12-05 21:04:20 +00001603 toErase.push_back(L);
1604 NumGVNLoad++;
1605 return true;
1606 }
1607
1608 if (LoadInst *DepLI = dyn_cast<LoadInst>(DepInst)) {
Chris Lattner7741aa52009-09-20 19:03:47 +00001609 Value *AvailableVal = DepLI;
1610
1611 // The loads are of a must-aliased pointer, but they may not actually have
1612 // the same type. See if we know how to reuse the previously loaded value
1613 // (depending on its type).
1614 const TargetData *TD = 0;
Chris Lattner10460aa2009-10-21 04:11:19 +00001615 if (DepLI->getType() != L->getType()) {
1616 if ((TD = getAnalysisIfAvailable<TargetData>())) {
1617 AvailableVal = CoerceAvailableValueToLoadType(DepLI, L->getType(), L,*TD);
1618 if (AvailableVal == 0)
1619 return false;
Chris Lattner7741aa52009-09-20 19:03:47 +00001620
Chris Lattner10460aa2009-10-21 04:11:19 +00001621 DEBUG(errs() << "GVN COERCED LOAD:\n" << *DepLI << "\n" << *AvailableVal
1622 << "\n" << *L << "\n\n\n");
1623 }
1624 else
1625 return false;
Chris Lattner7741aa52009-09-20 19:03:47 +00001626 }
1627
Chris Lattner4531da82008-12-05 21:04:20 +00001628 // Remove it!
Chris Lattner7741aa52009-09-20 19:03:47 +00001629 L->replaceAllUsesWith(AvailableVal);
Chris Lattnerf81b0142008-12-09 22:06:23 +00001630 if (isa<PointerType>(DepLI->getType()))
1631 MD->invalidateCachedPointerInfo(DepLI);
Chris Lattner4531da82008-12-05 21:04:20 +00001632 toErase.push_back(L);
1633 NumGVNLoad++;
1634 return true;
1635 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001636
Chris Lattner8ea60462008-11-30 01:39:32 +00001637 // If this load really doesn't depend on anything, then we must be loading an
1638 // undef value. This can happen when loading for a fresh allocation with no
1639 // intervening stores, for example.
Victor Hernandezb1687302009-10-23 21:09:37 +00001640 if (isa<AllocaInst>(DepInst) || isMalloc(DepInst)) {
Owen Andersonb99ecca2009-07-30 23:03:37 +00001641 L->replaceAllUsesWith(UndefValue::get(L->getType()));
Chris Lattner8ea60462008-11-30 01:39:32 +00001642 toErase.push_back(L);
Chris Lattner8ea60462008-11-30 01:39:32 +00001643 NumGVNLoad++;
Chris Lattner4531da82008-12-05 21:04:20 +00001644 return true;
Eli Friedman350307f2008-02-12 12:08:14 +00001645 }
Owen Andersonc07861a2009-10-28 07:05:35 +00001646
Owen Andersonf187daf2009-12-02 07:35:19 +00001647 // If this load occurs either right after a lifetime begin,
Owen Andersonc07861a2009-10-28 07:05:35 +00001648 // then the loaded value is undefined.
1649 if (IntrinsicInst* II = dyn_cast<IntrinsicInst>(DepInst)) {
Owen Andersonf187daf2009-12-02 07:35:19 +00001650 if (II->getIntrinsicID() == Intrinsic::lifetime_start) {
Owen Andersonc07861a2009-10-28 07:05:35 +00001651 L->replaceAllUsesWith(UndefValue::get(L->getType()));
1652 toErase.push_back(L);
1653 NumGVNLoad++;
1654 return true;
1655 }
1656 }
Eli Friedman350307f2008-02-12 12:08:14 +00001657
Chris Lattner4531da82008-12-05 21:04:20 +00001658 return false;
Owen Anderson85c40642007-07-24 17:55:58 +00001659}
1660
Chris Lattnerff36c952009-09-21 02:42:51 +00001661Value *GVN::lookupNumber(BasicBlock *BB, uint32_t num) {
Owen Andersonaef6a922008-06-23 17:49:45 +00001662 DenseMap<BasicBlock*, ValueNumberScope*>::iterator I = localAvail.find(BB);
1663 if (I == localAvail.end())
1664 return 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001665
Chris Lattnerff36c952009-09-21 02:42:51 +00001666 ValueNumberScope *Locals = I->second;
1667 while (Locals) {
1668 DenseMap<uint32_t, Value*>::iterator I = Locals->table.find(num);
1669 if (I != Locals->table.end())
Owen Anderson2a412722008-06-20 01:15:47 +00001670 return I->second;
Chris Lattnerff36c952009-09-21 02:42:51 +00001671 Locals = Locals->parent;
Owen Anderson2a412722008-06-20 01:15:47 +00001672 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001673
Owen Anderson2a412722008-06-20 01:15:47 +00001674 return 0;
1675}
1676
Owen Andersona03e7862008-12-15 02:03:00 +00001677
Owen Andersonf631bb62007-08-14 18:16:29 +00001678/// processInstruction - When calculating availability, handle an instruction
Owen Anderson85c40642007-07-24 17:55:58 +00001679/// by inserting it into the appropriate sets
Owen Anderson9334fc62008-06-12 19:25:32 +00001680bool GVN::processInstruction(Instruction *I,
Chris Lattner7de20452008-03-21 22:01:16 +00001681 SmallVectorImpl<Instruction*> &toErase) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001682 if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
1683 bool Changed = processLoad(LI, toErase);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001684
Chris Lattnerff36c952009-09-21 02:42:51 +00001685 if (!Changed) {
1686 unsigned Num = VN.lookup_or_add(LI);
1687 localAvail[I->getParent()]->table.insert(std::make_pair(Num, LI));
Owen Andersone6b4ff82008-06-18 21:41:49 +00001688 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001689
Chris Lattnerff36c952009-09-21 02:42:51 +00001690 return Changed;
Owen Andersone6b4ff82008-06-18 21:41:49 +00001691 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001692
Chris Lattnerff36c952009-09-21 02:42:51 +00001693 uint32_t NextNum = VN.getNextUnusedValueNumber();
1694 unsigned Num = VN.lookup_or_add(I);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001695
Chris Lattnerff36c952009-09-21 02:42:51 +00001696 if (BranchInst *BI = dyn_cast<BranchInst>(I)) {
1697 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001698
Owen Andersonef8bf0f2009-04-01 23:53:49 +00001699 if (!BI->isConditional() || isa<Constant>(BI->getCondition()))
1700 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001701
Chris Lattnerff36c952009-09-21 02:42:51 +00001702 Value *BranchCond = BI->getCondition();
1703 uint32_t CondVN = VN.lookup_or_add(BranchCond);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001704
Chris Lattnerff36c952009-09-21 02:42:51 +00001705 BasicBlock *TrueSucc = BI->getSuccessor(0);
1706 BasicBlock *FalseSucc = BI->getSuccessor(1);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001707
Chris Lattnerff36c952009-09-21 02:42:51 +00001708 if (TrueSucc->getSinglePredecessor())
1709 localAvail[TrueSucc]->table[CondVN] =
1710 ConstantInt::getTrue(TrueSucc->getContext());
1711 if (FalseSucc->getSinglePredecessor())
1712 localAvail[FalseSucc]->table[CondVN] =
1713 ConstantInt::getFalse(TrueSucc->getContext());
Owen Andersonef8bf0f2009-04-01 23:53:49 +00001714
1715 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001716
Owen Andersonced50f82008-04-07 09:59:07 +00001717 // Allocations are always uniquely numbered, so we can save time and memory
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001718 // by fast failing them.
Victor Hernandezb1687302009-10-23 21:09:37 +00001719 } else if (isa<AllocaInst>(I) || isa<TerminatorInst>(I)) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001720 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Andersonced50f82008-04-07 09:59:07 +00001721 return false;
Owen Andersone6b4ff82008-06-18 21:41:49 +00001722 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001723
Owen Andersone0143452007-08-16 22:02:55 +00001724 // Collapse PHI nodes
Owen Anderson98f6a6b2007-08-14 18:33:27 +00001725 if (PHINode* p = dyn_cast<PHINode>(I)) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001726 Value *constVal = CollapsePhi(p);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001727
Owen Anderson98f6a6b2007-08-14 18:33:27 +00001728 if (constVal) {
Owen Andersone02ad522007-08-16 22:51:56 +00001729 p->replaceAllUsesWith(constVal);
Dan Gohmanc8d26652009-11-14 02:27:51 +00001730 if (MD && isa<PointerType>(constVal->getType()))
Chris Lattnerf81b0142008-12-09 22:06:23 +00001731 MD->invalidateCachedPointerInfo(constVal);
Owen Anderson575f2812008-12-23 00:49:51 +00001732 VN.erase(p);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001733
Owen Andersone02ad522007-08-16 22:51:56 +00001734 toErase.push_back(p);
Owen Andersone6b4ff82008-06-18 21:41:49 +00001735 } else {
Chris Lattnerff36c952009-09-21 02:42:51 +00001736 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson98f6a6b2007-08-14 18:33:27 +00001737 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001738
Owen Anderson8a8d13c2008-07-03 17:44:33 +00001739 // If the number we were assigned was a brand new VN, then we don't
1740 // need to do a lookup to see if the number already exists
1741 // somewhere in the domtree: it can't!
Chris Lattnerff36c952009-09-21 02:42:51 +00001742 } else if (Num == NextNum) {
1743 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001744
Owen Andersona03e7862008-12-15 02:03:00 +00001745 // Perform fast-path value-number based elimination of values inherited from
1746 // dominators.
Chris Lattnerff36c952009-09-21 02:42:51 +00001747 } else if (Value *repl = lookupNumber(I->getParent(), Num)) {
Owen Andersonc772be72007-12-08 01:37:09 +00001748 // Remove it!
Owen Anderson5aff8002007-07-31 23:27:13 +00001749 VN.erase(I);
Owen Anderson85c40642007-07-24 17:55:58 +00001750 I->replaceAllUsesWith(repl);
Dan Gohmanc8d26652009-11-14 02:27:51 +00001751 if (MD && isa<PointerType>(repl->getType()))
Chris Lattnerf81b0142008-12-09 22:06:23 +00001752 MD->invalidateCachedPointerInfo(repl);
Owen Anderson85c40642007-07-24 17:55:58 +00001753 toErase.push_back(I);
1754 return true;
Owen Andersona03e7862008-12-15 02:03:00 +00001755
Owen Anderson8a8d13c2008-07-03 17:44:33 +00001756 } else {
Chris Lattnerff36c952009-09-21 02:42:51 +00001757 localAvail[I->getParent()]->table.insert(std::make_pair(Num, I));
Owen Anderson85c40642007-07-24 17:55:58 +00001758 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001759
Owen Anderson85c40642007-07-24 17:55:58 +00001760 return false;
1761}
1762
Bill Wendling42f17f62008-12-22 22:32:22 +00001763/// runOnFunction - This is the main transformation entry point for a function.
Owen Andersonbe168b32007-08-14 18:04:11 +00001764bool GVN::runOnFunction(Function& F) {
Dan Gohmanc8d26652009-11-14 02:27:51 +00001765 if (!NoLoads)
1766 MD = &getAnalysis<MemoryDependenceAnalysis>();
Chris Lattner02ca4422008-12-01 00:40:32 +00001767 DT = &getAnalysis<DominatorTree>();
Owen Andersonbcf2bd52008-05-12 20:15:55 +00001768 VN.setAliasAnalysis(&getAnalysis<AliasAnalysis>());
Chris Lattner02ca4422008-12-01 00:40:32 +00001769 VN.setMemDep(MD);
1770 VN.setDomTree(DT);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001771
Chris Lattnerff36c952009-09-21 02:42:51 +00001772 bool Changed = false;
1773 bool ShouldContinue = true;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001774
Owen Anderson26ed2572008-07-16 17:52:31 +00001775 // Merge unconditional branches, allowing PRE to catch more
1776 // optimization opportunities.
1777 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001778 BasicBlock *BB = FI;
Owen Anderson26ed2572008-07-16 17:52:31 +00001779 ++FI;
Owen Andersonf59eef82008-07-17 00:01:40 +00001780 bool removedBlock = MergeBlockIntoPredecessor(BB, this);
1781 if (removedBlock) NumGVNBlocks++;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001782
Chris Lattnerff36c952009-09-21 02:42:51 +00001783 Changed |= removedBlock;
Owen Anderson26ed2572008-07-16 17:52:31 +00001784 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001785
Chris Lattner4bab29b2008-12-09 19:21:47 +00001786 unsigned Iteration = 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001787
Chris Lattnerff36c952009-09-21 02:42:51 +00001788 while (ShouldContinue) {
Dan Gohman0be10b02009-07-25 01:43:01 +00001789 DEBUG(errs() << "GVN iteration: " << Iteration << "\n");
Chris Lattnerff36c952009-09-21 02:42:51 +00001790 ShouldContinue = iterateOnFunction(F);
1791 Changed |= ShouldContinue;
Chris Lattner4bab29b2008-12-09 19:21:47 +00001792 ++Iteration;
Owen Andersonbe168b32007-08-14 18:04:11 +00001793 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001794
Owen Anderson916f4732008-07-18 18:03:38 +00001795 if (EnablePRE) {
Owen Anderson9c935902008-09-03 23:06:07 +00001796 bool PREChanged = true;
1797 while (PREChanged) {
1798 PREChanged = performPRE(F);
Chris Lattnerff36c952009-09-21 02:42:51 +00001799 Changed |= PREChanged;
Owen Anderson9c935902008-09-03 23:06:07 +00001800 }
Owen Anderson916f4732008-07-18 18:03:38 +00001801 }
Chris Lattner4bab29b2008-12-09 19:21:47 +00001802 // FIXME: Should perform GVN again after PRE does something. PRE can move
1803 // computations into blocks where they become fully redundant. Note that
1804 // we can't do this until PRE's critical edge splitting updates memdep.
1805 // Actually, when this happens, we should just fully integrate PRE into GVN.
Nuno Lopes274474b2008-10-10 16:25:50 +00001806
1807 cleanupGlobalSets();
1808
Chris Lattnerff36c952009-09-21 02:42:51 +00001809 return Changed;
Owen Andersonbe168b32007-08-14 18:04:11 +00001810}
1811
1812
Chris Lattnerff36c952009-09-21 02:42:51 +00001813bool GVN::processBlock(BasicBlock *BB) {
Chris Lattner4bab29b2008-12-09 19:21:47 +00001814 // FIXME: Kill off toErase by doing erasing eagerly in a helper function (and
1815 // incrementing BI before processing an instruction).
Owen Anderson9334fc62008-06-12 19:25:32 +00001816 SmallVector<Instruction*, 8> toErase;
Chris Lattnerff36c952009-09-21 02:42:51 +00001817 bool ChangedFunction = false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001818
Owen Anderson9334fc62008-06-12 19:25:32 +00001819 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
1820 BI != BE;) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001821 ChangedFunction |= processInstruction(BI, toErase);
Owen Anderson9334fc62008-06-12 19:25:32 +00001822 if (toErase.empty()) {
1823 ++BI;
1824 continue;
1825 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001826
Owen Anderson9334fc62008-06-12 19:25:32 +00001827 // If we need some instructions deleted, do it now.
1828 NumGVNInstr += toErase.size();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001829
Owen Anderson9334fc62008-06-12 19:25:32 +00001830 // Avoid iterator invalidation.
1831 bool AtStart = BI == BB->begin();
1832 if (!AtStart)
1833 --BI;
1834
1835 for (SmallVector<Instruction*, 4>::iterator I = toErase.begin(),
Chris Lattner02ca4422008-12-01 00:40:32 +00001836 E = toErase.end(); I != E; ++I) {
Dan Gohman7e124382009-07-31 20:24:18 +00001837 DEBUG(errs() << "GVN removed: " << **I << '\n');
Dan Gohmanc8d26652009-11-14 02:27:51 +00001838 if (MD) MD->removeInstruction(*I);
Owen Anderson9334fc62008-06-12 19:25:32 +00001839 (*I)->eraseFromParent();
Bill Wendling84049422008-12-22 21:57:30 +00001840 DEBUG(verifyRemoved(*I));
Chris Lattner02ca4422008-12-01 00:40:32 +00001841 }
Chris Lattner4bab29b2008-12-09 19:21:47 +00001842 toErase.clear();
Owen Anderson9334fc62008-06-12 19:25:32 +00001843
1844 if (AtStart)
1845 BI = BB->begin();
1846 else
1847 ++BI;
Owen Anderson9334fc62008-06-12 19:25:32 +00001848 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001849
Chris Lattnerff36c952009-09-21 02:42:51 +00001850 return ChangedFunction;
Owen Anderson9334fc62008-06-12 19:25:32 +00001851}
1852
Owen Andersone6b4ff82008-06-18 21:41:49 +00001853/// performPRE - Perform a purely local form of PRE that looks for diamond
1854/// control flow patterns and attempts to perform simple PRE at the join point.
Chris Lattner4790cb42009-10-31 22:11:15 +00001855bool GVN::performPRE(Function &F) {
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001856 bool Changed = false;
Owen Andersonec747c42008-06-19 19:54:19 +00001857 SmallVector<std::pair<TerminatorInst*, unsigned>, 4> toSplit;
Chris Lattner3304b562008-12-01 07:29:03 +00001858 DenseMap<BasicBlock*, Value*> predMap;
Owen Andersone6b4ff82008-06-18 21:41:49 +00001859 for (df_iterator<BasicBlock*> DI = df_begin(&F.getEntryBlock()),
1860 DE = df_end(&F.getEntryBlock()); DI != DE; ++DI) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001861 BasicBlock *CurrentBlock = *DI;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001862
Owen Andersone6b4ff82008-06-18 21:41:49 +00001863 // Nothing to PRE in the entry block.
1864 if (CurrentBlock == &F.getEntryBlock()) continue;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001865
Owen Andersone6b4ff82008-06-18 21:41:49 +00001866 for (BasicBlock::iterator BI = CurrentBlock->begin(),
1867 BE = CurrentBlock->end(); BI != BE; ) {
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001868 Instruction *CurInst = BI++;
Duncan Sands2f500832009-05-06 06:49:50 +00001869
Victor Hernandezb1687302009-10-23 21:09:37 +00001870 if (isa<AllocaInst>(CurInst) ||
Victor Hernandez48c3c542009-09-18 22:35:49 +00001871 isa<TerminatorInst>(CurInst) || isa<PHINode>(CurInst) ||
Devang Patele9d08b82009-10-14 17:29:00 +00001872 CurInst->getType()->isVoidTy() ||
Duncan Sands2f500832009-05-06 06:49:50 +00001873 CurInst->mayReadFromMemory() || CurInst->mayHaveSideEffects() ||
John Criswell6e0aa282009-03-10 15:04:53 +00001874 isa<DbgInfoIntrinsic>(CurInst))
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001875 continue;
Duncan Sands2f500832009-05-06 06:49:50 +00001876
Chris Lattnerff36c952009-09-21 02:42:51 +00001877 uint32_t ValNo = VN.lookup(CurInst);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001878
Owen Andersone6b4ff82008-06-18 21:41:49 +00001879 // Look for the predecessors for PRE opportunities. We're
1880 // only trying to solve the basic diamond case, where
1881 // a value is computed in the successor and one predecessor,
1882 // but not the other. We also explicitly disallow cases
1883 // where the successor is its own predecessor, because they're
1884 // more complicated to get right.
Chris Lattnerff36c952009-09-21 02:42:51 +00001885 unsigned NumWith = 0;
1886 unsigned NumWithout = 0;
1887 BasicBlock *PREPred = 0;
Chris Lattner3304b562008-12-01 07:29:03 +00001888 predMap.clear();
1889
Owen Andersone6b4ff82008-06-18 21:41:49 +00001890 for (pred_iterator PI = pred_begin(CurrentBlock),
1891 PE = pred_end(CurrentBlock); PI != PE; ++PI) {
1892 // We're not interested in PRE where the block is its
Owen Anderson2a412722008-06-20 01:15:47 +00001893 // own predecessor, on in blocks with predecessors
1894 // that are not reachable.
1895 if (*PI == CurrentBlock) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001896 NumWithout = 2;
Owen Anderson2a412722008-06-20 01:15:47 +00001897 break;
1898 } else if (!localAvail.count(*PI)) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001899 NumWithout = 2;
Owen Anderson2a412722008-06-20 01:15:47 +00001900 break;
1901 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001902
1903 DenseMap<uint32_t, Value*>::iterator predV =
Chris Lattnerff36c952009-09-21 02:42:51 +00001904 localAvail[*PI]->table.find(ValNo);
Owen Anderson2a412722008-06-20 01:15:47 +00001905 if (predV == localAvail[*PI]->table.end()) {
Owen Andersone6b4ff82008-06-18 21:41:49 +00001906 PREPred = *PI;
Chris Lattnerff36c952009-09-21 02:42:51 +00001907 NumWithout++;
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001908 } else if (predV->second == CurInst) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001909 NumWithout = 2;
Owen Andersone6b4ff82008-06-18 21:41:49 +00001910 } else {
Owen Anderson2a412722008-06-20 01:15:47 +00001911 predMap[*PI] = predV->second;
Chris Lattnerff36c952009-09-21 02:42:51 +00001912 NumWith++;
Owen Andersone6b4ff82008-06-18 21:41:49 +00001913 }
1914 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001915
Owen Andersone6b4ff82008-06-18 21:41:49 +00001916 // Don't do PRE when it might increase code size, i.e. when
1917 // we would need to insert instructions in more than one pred.
Chris Lattnerff36c952009-09-21 02:42:51 +00001918 if (NumWithout != 1 || NumWith == 0)
Owen Andersone6b4ff82008-06-18 21:41:49 +00001919 continue;
Chris Lattner4790cb42009-10-31 22:11:15 +00001920
1921 // Don't do PRE across indirect branch.
1922 if (isa<IndirectBrInst>(PREPred->getTerminator()))
1923 continue;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001924
Owen Andersonec747c42008-06-19 19:54:19 +00001925 // We can't do PRE safely on a critical edge, so instead we schedule
1926 // the edge to be split and perform the PRE the next time we iterate
1927 // on the function.
Chris Lattnerff36c952009-09-21 02:42:51 +00001928 unsigned SuccNum = 0;
Owen Andersonec747c42008-06-19 19:54:19 +00001929 for (unsigned i = 0, e = PREPred->getTerminator()->getNumSuccessors();
1930 i != e; ++i)
Owen Anderson9c935902008-09-03 23:06:07 +00001931 if (PREPred->getTerminator()->getSuccessor(i) == CurrentBlock) {
Chris Lattnerff36c952009-09-21 02:42:51 +00001932 SuccNum = i;
Owen Andersonec747c42008-06-19 19:54:19 +00001933 break;
1934 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001935
Chris Lattnerff36c952009-09-21 02:42:51 +00001936 if (isCriticalEdge(PREPred->getTerminator(), SuccNum)) {
1937 toSplit.push_back(std::make_pair(PREPred->getTerminator(), SuccNum));
Owen Andersonec747c42008-06-19 19:54:19 +00001938 continue;
1939 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001940
Owen Andersone6b4ff82008-06-18 21:41:49 +00001941 // Instantiate the expression the in predecessor that lacked it.
1942 // Because we are going top-down through the block, all value numbers
1943 // will be available in the predecessor by the time we need them. Any
1944 // that weren't original present will have been instantiated earlier
1945 // in this loop.
Nick Lewyckyc94270c2009-09-27 07:38:41 +00001946 Instruction *PREInstr = CurInst->clone();
Owen Andersone6b4ff82008-06-18 21:41:49 +00001947 bool success = true;
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001948 for (unsigned i = 0, e = CurInst->getNumOperands(); i != e; ++i) {
1949 Value *Op = PREInstr->getOperand(i);
1950 if (isa<Argument>(Op) || isa<Constant>(Op) || isa<GlobalValue>(Op))
1951 continue;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001952
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001953 if (Value *V = lookupNumber(PREPred, VN.lookup(Op))) {
1954 PREInstr->setOperand(i, V);
1955 } else {
1956 success = false;
1957 break;
Owen Anderson14c612f2008-07-11 20:05:13 +00001958 }
Owen Andersone6b4ff82008-06-18 21:41:49 +00001959 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001960
Owen Andersone6b4ff82008-06-18 21:41:49 +00001961 // Fail out if we encounter an operand that is not available in
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001962 // the PRE predecessor. This is typically because of loads which
Owen Andersone6b4ff82008-06-18 21:41:49 +00001963 // are not value numbered precisely.
1964 if (!success) {
1965 delete PREInstr;
Bill Wendling3858cae2008-12-22 22:14:07 +00001966 DEBUG(verifyRemoved(PREInstr));
Owen Andersone6b4ff82008-06-18 21:41:49 +00001967 continue;
1968 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001969
Owen Andersone6b4ff82008-06-18 21:41:49 +00001970 PREInstr->insertBefore(PREPred->getTerminator());
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001971 PREInstr->setName(CurInst->getName() + ".pre");
Owen Anderson2a412722008-06-20 01:15:47 +00001972 predMap[PREPred] = PREInstr;
Chris Lattnerff36c952009-09-21 02:42:51 +00001973 VN.add(PREInstr, ValNo);
Owen Andersone6b4ff82008-06-18 21:41:49 +00001974 NumGVNPRE++;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001975
Owen Andersone6b4ff82008-06-18 21:41:49 +00001976 // Update the availability map to include the new instruction.
Chris Lattnerff36c952009-09-21 02:42:51 +00001977 localAvail[PREPred]->table.insert(std::make_pair(ValNo, PREInstr));
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001978
Owen Andersone6b4ff82008-06-18 21:41:49 +00001979 // Create a PHI to make the value available in this block.
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001980 PHINode* Phi = PHINode::Create(CurInst->getType(),
1981 CurInst->getName() + ".pre-phi",
Owen Andersone6b4ff82008-06-18 21:41:49 +00001982 CurrentBlock->begin());
1983 for (pred_iterator PI = pred_begin(CurrentBlock),
1984 PE = pred_end(CurrentBlock); PI != PE; ++PI)
Owen Anderson2a412722008-06-20 01:15:47 +00001985 Phi->addIncoming(predMap[*PI], *PI);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001986
Chris Lattnerff36c952009-09-21 02:42:51 +00001987 VN.add(Phi, ValNo);
1988 localAvail[CurrentBlock]->table[ValNo] = Phi;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001989
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001990 CurInst->replaceAllUsesWith(Phi);
Dan Gohmanc8d26652009-11-14 02:27:51 +00001991 if (MD && isa<PointerType>(Phi->getType()))
Chris Lattnerf81b0142008-12-09 22:06:23 +00001992 MD->invalidateCachedPointerInfo(Phi);
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001993 VN.erase(CurInst);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001994
Dan Gohman7e124382009-07-31 20:24:18 +00001995 DEBUG(errs() << "GVN PRE removed: " << *CurInst << '\n');
Dan Gohmanc8d26652009-11-14 02:27:51 +00001996 if (MD) MD->removeInstruction(CurInst);
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001997 CurInst->eraseFromParent();
Bill Wendling84049422008-12-22 21:57:30 +00001998 DEBUG(verifyRemoved(CurInst));
Chris Lattner66a3a3e2008-12-01 07:35:54 +00001999 Changed = true;
Owen Andersone6b4ff82008-06-18 21:41:49 +00002000 }
2001 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002002
Owen Andersonec747c42008-06-19 19:54:19 +00002003 for (SmallVector<std::pair<TerminatorInst*, unsigned>, 4>::iterator
Anton Korobeynikov2e8710c2008-12-05 19:38:49 +00002004 I = toSplit.begin(), E = toSplit.end(); I != E; ++I)
Owen Andersonec747c42008-06-19 19:54:19 +00002005 SplitCriticalEdge(I->first, I->second, this);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002006
Anton Korobeynikov2e8710c2008-12-05 19:38:49 +00002007 return Changed || toSplit.size();
Owen Andersone6b4ff82008-06-18 21:41:49 +00002008}
2009
Bill Wendling42f17f62008-12-22 22:32:22 +00002010/// iterateOnFunction - Executes one iteration of GVN
Owen Andersonbe168b32007-08-14 18:04:11 +00002011bool GVN::iterateOnFunction(Function &F) {
Nuno Lopes274474b2008-10-10 16:25:50 +00002012 cleanupGlobalSets();
Chris Lattner98054902008-03-21 21:33:23 +00002013
Owen Andersonef8bf0f2009-04-01 23:53:49 +00002014 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2015 DE = df_end(DT->getRootNode()); DI != DE; ++DI) {
2016 if (DI->getIDom())
2017 localAvail[DI->getBlock()] =
2018 new ValueNumberScope(localAvail[DI->getIDom()->getBlock()]);
2019 else
2020 localAvail[DI->getBlock()] = new ValueNumberScope(0);
2021 }
2022
Owen Anderson85c40642007-07-24 17:55:58 +00002023 // Top-down walk of the dominator tree
Chris Lattnerff36c952009-09-21 02:42:51 +00002024 bool Changed = false;
Owen Andersonef136f52008-12-15 03:52:17 +00002025#if 0
2026 // Needed for value numbering with phi construction to work.
Owen Andersona03e7862008-12-15 02:03:00 +00002027 ReversePostOrderTraversal<Function*> RPOT(&F);
2028 for (ReversePostOrderTraversal<Function*>::rpo_iterator RI = RPOT.begin(),
2029 RE = RPOT.end(); RI != RE; ++RI)
Chris Lattnerff36c952009-09-21 02:42:51 +00002030 Changed |= processBlock(*RI);
Owen Andersonef136f52008-12-15 03:52:17 +00002031#else
2032 for (df_iterator<DomTreeNode*> DI = df_begin(DT->getRootNode()),
2033 DE = df_end(DT->getRootNode()); DI != DE; ++DI)
Chris Lattnerff36c952009-09-21 02:42:51 +00002034 Changed |= processBlock(DI->getBlock());
Owen Andersonef136f52008-12-15 03:52:17 +00002035#endif
2036
Chris Lattnerff36c952009-09-21 02:42:51 +00002037 return Changed;
Owen Anderson85c40642007-07-24 17:55:58 +00002038}
Nuno Lopes274474b2008-10-10 16:25:50 +00002039
2040void GVN::cleanupGlobalSets() {
2041 VN.clear();
Nuno Lopes274474b2008-10-10 16:25:50 +00002042
2043 for (DenseMap<BasicBlock*, ValueNumberScope*>::iterator
2044 I = localAvail.begin(), E = localAvail.end(); I != E; ++I)
2045 delete I->second;
2046 localAvail.clear();
2047}
Bill Wendling2a023742008-12-22 21:36:08 +00002048
2049/// verifyRemoved - Verify that the specified instruction does not occur in our
2050/// internal data structures.
Bill Wendlingf9c0e9e2008-12-22 22:28:56 +00002051void GVN::verifyRemoved(const Instruction *Inst) const {
2052 VN.verifyRemoved(Inst);
Bill Wendling3858cae2008-12-22 22:14:07 +00002053
Bill Wendlingf9c0e9e2008-12-22 22:28:56 +00002054 // Walk through the value number scope to make sure the instruction isn't
2055 // ferreted away in it.
Jeffrey Yasskin8154d2e2009-11-10 01:02:17 +00002056 for (DenseMap<BasicBlock*, ValueNumberScope*>::const_iterator
Bill Wendlingf9c0e9e2008-12-22 22:28:56 +00002057 I = localAvail.begin(), E = localAvail.end(); I != E; ++I) {
2058 const ValueNumberScope *VNS = I->second;
2059
2060 while (VNS) {
Jeffrey Yasskin8154d2e2009-11-10 01:02:17 +00002061 for (DenseMap<uint32_t, Value*>::const_iterator
Bill Wendlingf9c0e9e2008-12-22 22:28:56 +00002062 II = VNS->table.begin(), IE = VNS->table.end(); II != IE; ++II) {
2063 assert(II->second != Inst && "Inst still in value numbering scope!");
2064 }
2065
2066 VNS = VNS->parent;
Bill Wendling3858cae2008-12-22 22:14:07 +00002067 }
2068 }
Bill Wendling2a023742008-12-22 21:36:08 +00002069}