blob: b91dffa379e1da6f757581c18506f3a443284568 [file] [log] [blame]
Owen Anderson5fba6c12007-05-29 21:53:49 +00001//===- GVNPRE.cpp - Eliminate redundant values and expressions ------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by the Owen Anderson and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This pass performs a hybrid of global value numbering and partial redundancy
11// elimination, known as GVN-PRE. It performs partial redundancy elimination on
12// values, rather than lexical expressions, allowing a more comprehensive view
13// the optimization. It replaces redundant values with uses of earlier
14// occurences of the same value. While this is beneficial in that it eliminates
15// unneeded computation, it also increases register pressure by creating large
Owen Andersonb232efa2007-06-08 20:57:08 +000016// live ranges, and should be used with caution on platforms that are very
Owen Anderson5fba6c12007-05-29 21:53:49 +000017// sensitive to register pressure.
18//
19//===----------------------------------------------------------------------===//
20
21#define DEBUG_TYPE "gvnpre"
22#include "llvm/Value.h"
23#include "llvm/Transforms/Scalar.h"
24#include "llvm/Instructions.h"
25#include "llvm/Function.h"
26#include "llvm/Analysis/Dominators.h"
Owen Anderson28a2d442007-06-22 00:20:30 +000027#include "llvm/ADT/BitVector.h"
Owen Anderson1ad2c102007-06-19 23:23:54 +000028#include "llvm/ADT/DenseMap.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000029#include "llvm/ADT/DepthFirstIterator.h"
Owen Anderson43ca4b42007-06-25 18:25:31 +000030#include "llvm/ADT/PostOrderIterator.h"
Owen Anderson2ff912b2007-06-21 17:57:53 +000031#include "llvm/ADT/SmallPtrSet.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000032#include "llvm/ADT/Statistic.h"
Owen Andersonbe802402007-06-08 01:03:01 +000033#include "llvm/Support/CFG.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000034#include "llvm/Support/Compiler.h"
Owen Anderson4a6ec8f2007-05-29 23:15:21 +000035#include "llvm/Support/Debug.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000036#include <algorithm>
Owen Anderson331bf6a2007-05-31 22:44:11 +000037#include <deque>
Owen Anderson5fba6c12007-05-29 21:53:49 +000038#include <map>
Owen Anderson331bf6a2007-05-31 22:44:11 +000039#include <vector>
Owen Anderson5fba6c12007-05-29 21:53:49 +000040#include <set>
Owen Anderson5fba6c12007-05-29 21:53:49 +000041using namespace llvm;
42
Owen Andersonb56fba02007-06-19 03:31:41 +000043//===----------------------------------------------------------------------===//
44// ValueTable Class
45//===----------------------------------------------------------------------===//
46
Owen Andersonfd5683a2007-06-21 00:19:05 +000047/// This class holds the mapping between values and value numbers. It is used
48/// as an efficient mechanism to determine the expression-wise equivalence of
49/// two values.
Owen Andersonb56fba02007-06-19 03:31:41 +000050
51namespace {
52 class VISIBILITY_HIDDEN ValueTable {
53 public:
54 struct Expression {
55 enum ExpressionOpcode { ADD, SUB, MUL, UDIV, SDIV, FDIV, UREM, SREM,
56 FREM, SHL, LSHR, ASHR, AND, OR, XOR, ICMPEQ,
57 ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
58 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
59 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
60 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
Owen Andersonb6a39fc2007-06-27 04:10:46 +000061 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
62 SHUFFLE };
Owen Andersonb56fba02007-06-19 03:31:41 +000063
64 ExpressionOpcode opcode;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000065 uint32_t firstVN;
66 uint32_t secondVN;
67 uint32_t thirdVN;
Owen Andersonb56fba02007-06-19 03:31:41 +000068
69 bool operator< (const Expression& other) const {
70 if (opcode < other.opcode)
71 return true;
72 else if (opcode > other.opcode)
73 return false;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000074 else if (firstVN < other.firstVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000075 return true;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000076 else if (firstVN > other.firstVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000077 return false;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000078 else if (secondVN < other.secondVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000079 return true;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000080 else if (secondVN > other.secondVN)
81 return false;
82 else if (thirdVN < other.thirdVN)
83 return true;
84 else if (thirdVN > other.thirdVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000085 return false;
86 else
87 return false;
Owen Andersona75dd4d2007-06-12 00:50:47 +000088 }
Owen Andersonb56fba02007-06-19 03:31:41 +000089 };
90
91 private:
Owen Anderson1ad2c102007-06-19 23:23:54 +000092 DenseMap<Value*, uint32_t> valueNumbering;
Owen Andersonb56fba02007-06-19 03:31:41 +000093 std::map<Expression, uint32_t> expressionNumbering;
94
Owen Andersonb56fba02007-06-19 03:31:41 +000095 uint32_t nextValueNumber;
96
97 Expression::ExpressionOpcode getOpcode(BinaryOperator* BO);
98 Expression::ExpressionOpcode getOpcode(CmpInst* C);
Owen Andersonfd5683a2007-06-21 00:19:05 +000099 Expression create_expression(BinaryOperator* BO);
100 Expression create_expression(CmpInst* C);
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000101 Expression create_expression(ShuffleVectorInst* V);
102 Expression create_expression(ExtractElementInst* C);
103 Expression create_expression(InsertElementInst* V);
Owen Andersonb56fba02007-06-19 03:31:41 +0000104 public:
105 ValueTable() { nextValueNumber = 1; }
106 uint32_t lookup_or_add(Value* V);
107 uint32_t lookup(Value* V);
108 void add(Value* V, uint32_t num);
109 void clear();
Owen Anderson91c54952007-06-19 05:37:32 +0000110 void erase(Value* v);
Owen Anderson28a2d442007-06-22 00:20:30 +0000111 unsigned size();
Owen Andersonb56fba02007-06-19 03:31:41 +0000112 };
113}
114
Owen Andersonfd5683a2007-06-21 00:19:05 +0000115//===----------------------------------------------------------------------===//
116// ValueTable Internal Functions
117//===----------------------------------------------------------------------===//
Owen Andersonb56fba02007-06-19 03:31:41 +0000118ValueTable::Expression::ExpressionOpcode
Owen Andersonfd5683a2007-06-21 00:19:05 +0000119 ValueTable::getOpcode(BinaryOperator* BO) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000120 switch(BO->getOpcode()) {
121 case Instruction::Add:
122 return Expression::ADD;
123 case Instruction::Sub:
124 return Expression::SUB;
125 case Instruction::Mul:
126 return Expression::MUL;
127 case Instruction::UDiv:
128 return Expression::UDIV;
129 case Instruction::SDiv:
130 return Expression::SDIV;
131 case Instruction::FDiv:
132 return Expression::FDIV;
133 case Instruction::URem:
134 return Expression::UREM;
135 case Instruction::SRem:
136 return Expression::SREM;
137 case Instruction::FRem:
138 return Expression::FREM;
139 case Instruction::Shl:
140 return Expression::SHL;
141 case Instruction::LShr:
142 return Expression::LSHR;
143 case Instruction::AShr:
144 return Expression::ASHR;
145 case Instruction::And:
146 return Expression::AND;
147 case Instruction::Or:
148 return Expression::OR;
149 case Instruction::Xor:
150 return Expression::XOR;
151
152 // THIS SHOULD NEVER HAPPEN
153 default:
154 assert(0 && "Binary operator with unknown opcode?");
155 return Expression::ADD;
156 }
157}
158
159ValueTable::Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
160 if (C->getOpcode() == Instruction::ICmp) {
161 switch (C->getPredicate()) {
162 case ICmpInst::ICMP_EQ:
163 return Expression::ICMPEQ;
164 case ICmpInst::ICMP_NE:
165 return Expression::ICMPNE;
166 case ICmpInst::ICMP_UGT:
167 return Expression::ICMPUGT;
168 case ICmpInst::ICMP_UGE:
169 return Expression::ICMPUGE;
170 case ICmpInst::ICMP_ULT:
171 return Expression::ICMPULT;
172 case ICmpInst::ICMP_ULE:
173 return Expression::ICMPULE;
174 case ICmpInst::ICMP_SGT:
175 return Expression::ICMPSGT;
176 case ICmpInst::ICMP_SGE:
177 return Expression::ICMPSGE;
178 case ICmpInst::ICMP_SLT:
179 return Expression::ICMPSLT;
180 case ICmpInst::ICMP_SLE:
181 return Expression::ICMPSLE;
182
183 // THIS SHOULD NEVER HAPPEN
184 default:
185 assert(0 && "Comparison with unknown predicate?");
186 return Expression::ICMPEQ;
187 }
188 } else {
189 switch (C->getPredicate()) {
190 case FCmpInst::FCMP_OEQ:
191 return Expression::FCMPOEQ;
192 case FCmpInst::FCMP_OGT:
193 return Expression::FCMPOGT;
194 case FCmpInst::FCMP_OGE:
195 return Expression::FCMPOGE;
196 case FCmpInst::FCMP_OLT:
197 return Expression::FCMPOLT;
198 case FCmpInst::FCMP_OLE:
199 return Expression::FCMPOLE;
200 case FCmpInst::FCMP_ONE:
201 return Expression::FCMPONE;
202 case FCmpInst::FCMP_ORD:
203 return Expression::FCMPORD;
204 case FCmpInst::FCMP_UNO:
205 return Expression::FCMPUNO;
206 case FCmpInst::FCMP_UEQ:
207 return Expression::FCMPUEQ;
208 case FCmpInst::FCMP_UGT:
209 return Expression::FCMPUGT;
210 case FCmpInst::FCMP_UGE:
211 return Expression::FCMPUGE;
212 case FCmpInst::FCMP_ULT:
213 return Expression::FCMPULT;
214 case FCmpInst::FCMP_ULE:
215 return Expression::FCMPULE;
216 case FCmpInst::FCMP_UNE:
217 return Expression::FCMPUNE;
218
219 // THIS SHOULD NEVER HAPPEN
220 default:
221 assert(0 && "Comparison with unknown predicate?");
222 return Expression::FCMPOEQ;
Owen Anderson223718c2007-06-09 18:35:31 +0000223 }
224 }
Owen Andersonb56fba02007-06-19 03:31:41 +0000225}
226
Owen Andersonfd5683a2007-06-21 00:19:05 +0000227ValueTable::Expression ValueTable::create_expression(BinaryOperator* BO) {
228 Expression e;
229
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000230 e.firstVN = lookup_or_add(BO->getOperand(0));
231 e.secondVN = lookup_or_add(BO->getOperand(1));
232 e.thirdVN = 0;
Owen Andersonfd5683a2007-06-21 00:19:05 +0000233 e.opcode = getOpcode(BO);
234
Owen Andersonfd5683a2007-06-21 00:19:05 +0000235 return e;
236}
237
238ValueTable::Expression ValueTable::create_expression(CmpInst* C) {
239 Expression e;
240
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000241 e.firstVN = lookup_or_add(C->getOperand(0));
242 e.secondVN = lookup_or_add(C->getOperand(1));
243 e.thirdVN = 0;
Owen Andersonfd5683a2007-06-21 00:19:05 +0000244 e.opcode = getOpcode(C);
245
Owen Andersonfd5683a2007-06-21 00:19:05 +0000246 return e;
247}
248
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000249ValueTable::Expression ValueTable::create_expression(ShuffleVectorInst* S) {
250 Expression e;
251
252 e.firstVN = lookup_or_add(S->getOperand(0));
253 e.secondVN = lookup_or_add(S->getOperand(1));
254 e.thirdVN = lookup_or_add(S->getOperand(2));
255 e.opcode = Expression::SHUFFLE;
256
257 return e;
258}
259
260ValueTable::Expression ValueTable::create_expression(ExtractElementInst* E) {
261 Expression e;
262
263 e.firstVN = lookup_or_add(E->getOperand(0));
264 e.secondVN = lookup_or_add(E->getOperand(1));
265 e.thirdVN = 0;
266 e.opcode = Expression::EXTRACT;
267
268 return e;
269}
270
271ValueTable::Expression ValueTable::create_expression(InsertElementInst* I) {
272 Expression e;
273
274 e.firstVN = lookup_or_add(I->getOperand(0));
275 e.secondVN = lookup_or_add(I->getOperand(1));
276 e.thirdVN = lookup_or_add(I->getOperand(2));
277 e.opcode = Expression::INSERT;
278
279 return e;
280}
281
Owen Andersonfd5683a2007-06-21 00:19:05 +0000282//===----------------------------------------------------------------------===//
283// ValueTable External Functions
284//===----------------------------------------------------------------------===//
285
286/// lookup_or_add - Returns the value number for the specified value, assigning
287/// it a new number if it did not have one before.
Owen Andersonb56fba02007-06-19 03:31:41 +0000288uint32_t ValueTable::lookup_or_add(Value* V) {
Owen Anderson1ad2c102007-06-19 23:23:54 +0000289 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
Owen Andersonb56fba02007-06-19 03:31:41 +0000290 if (VI != valueNumbering.end())
291 return VI->second;
Owen Anderson223718c2007-06-09 18:35:31 +0000292
Owen Anderson223718c2007-06-09 18:35:31 +0000293
Owen Andersonb56fba02007-06-19 03:31:41 +0000294 if (BinaryOperator* BO = dyn_cast<BinaryOperator>(V)) {
295 Expression e = create_expression(BO);
296
297 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
298 if (EI != expressionNumbering.end()) {
299 valueNumbering.insert(std::make_pair(V, EI->second));
300 return EI->second;
301 } else {
302 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
303 valueNumbering.insert(std::make_pair(V, nextValueNumber));
304
305 return nextValueNumber++;
306 }
307 } else if (CmpInst* C = dyn_cast<CmpInst>(V)) {
308 Expression e = create_expression(C);
309
310 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
311 if (EI != expressionNumbering.end()) {
312 valueNumbering.insert(std::make_pair(V, EI->second));
313 return EI->second;
314 } else {
315 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
316 valueNumbering.insert(std::make_pair(V, nextValueNumber));
317
318 return nextValueNumber++;
319 }
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000320 } else if (ShuffleVectorInst* U = dyn_cast<ShuffleVectorInst>(V)) {
321 Expression e = create_expression(U);
322
323 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
324 if (EI != expressionNumbering.end()) {
325 valueNumbering.insert(std::make_pair(V, EI->second));
326 return EI->second;
327 } else {
328 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
329 valueNumbering.insert(std::make_pair(V, nextValueNumber));
330
331 return nextValueNumber++;
332 }
333 } else if (ExtractElementInst* U = dyn_cast<ExtractElementInst>(V)) {
334 Expression e = create_expression(U);
335
336 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
337 if (EI != expressionNumbering.end()) {
338 valueNumbering.insert(std::make_pair(V, EI->second));
339 return EI->second;
340 } else {
341 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
342 valueNumbering.insert(std::make_pair(V, nextValueNumber));
343
344 return nextValueNumber++;
345 }
346 } else if (InsertElementInst* U = dyn_cast<InsertElementInst>(V)) {
347 Expression e = create_expression(U);
348
349 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
350 if (EI != expressionNumbering.end()) {
351 valueNumbering.insert(std::make_pair(V, EI->second));
352 return EI->second;
353 } else {
354 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
355 valueNumbering.insert(std::make_pair(V, nextValueNumber));
356
357 return nextValueNumber++;
358 }
Owen Andersonb56fba02007-06-19 03:31:41 +0000359 } else {
360 valueNumbering.insert(std::make_pair(V, nextValueNumber));
361 return nextValueNumber++;
Owen Anderson0b68cda2007-06-03 05:55:58 +0000362 }
Owen Andersonb56fba02007-06-19 03:31:41 +0000363}
364
Owen Andersonfd5683a2007-06-21 00:19:05 +0000365/// lookup - Returns the value number of the specified value. Fails if
366/// the value has not yet been numbered.
Owen Andersonb56fba02007-06-19 03:31:41 +0000367uint32_t ValueTable::lookup(Value* V) {
Owen Anderson1ad2c102007-06-19 23:23:54 +0000368 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
Owen Andersonb56fba02007-06-19 03:31:41 +0000369 if (VI != valueNumbering.end())
370 return VI->second;
371 else
372 assert(0 && "Value not numbered?");
373
374 return 0;
375}
376
Owen Andersonfd5683a2007-06-21 00:19:05 +0000377/// add - Add the specified value with the given value number, removing
378/// its old number, if any
Owen Andersonb56fba02007-06-19 03:31:41 +0000379void ValueTable::add(Value* V, uint32_t num) {
Owen Anderson1ad2c102007-06-19 23:23:54 +0000380 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
Owen Andersonb56fba02007-06-19 03:31:41 +0000381 if (VI != valueNumbering.end())
382 valueNumbering.erase(VI);
383 valueNumbering.insert(std::make_pair(V, num));
384}
385
Owen Anderson191eb062007-06-25 05:41:12 +0000386/// clear - Remove all entries from the ValueTable
Owen Andersonb56fba02007-06-19 03:31:41 +0000387void ValueTable::clear() {
388 valueNumbering.clear();
389 expressionNumbering.clear();
390 nextValueNumber = 1;
391}
Owen Anderson0b68cda2007-06-03 05:55:58 +0000392
Owen Anderson191eb062007-06-25 05:41:12 +0000393/// erase - Remove a value from the value numbering
Owen Anderson91c54952007-06-19 05:37:32 +0000394void ValueTable::erase(Value* V) {
Owen Anderson91c54952007-06-19 05:37:32 +0000395 valueNumbering.erase(V);
Owen Anderson91c54952007-06-19 05:37:32 +0000396}
397
Owen Anderson28a2d442007-06-22 00:20:30 +0000398/// size - Return the number of assigned value numbers
399unsigned ValueTable::size() {
400 // NOTE: zero is never assigned
401 return nextValueNumber;
402}
403
Owen Andersonfd5683a2007-06-21 00:19:05 +0000404//===----------------------------------------------------------------------===//
405// GVNPRE Pass
406//===----------------------------------------------------------------------===//
407
Owen Anderson5fba6c12007-05-29 21:53:49 +0000408namespace {
409
410 class VISIBILITY_HIDDEN GVNPRE : public FunctionPass {
411 bool runOnFunction(Function &F);
412 public:
413 static char ID; // Pass identification, replacement for typeid
Owen Andersonb9cbaed2007-06-19 04:32:55 +0000414 GVNPRE() : FunctionPass((intptr_t)&ID) { }
Owen Anderson5fba6c12007-05-29 21:53:49 +0000415
416 private:
Owen Andersonbe802402007-06-08 01:03:01 +0000417 ValueTable VN;
Owen Andersona75dd4d2007-06-12 00:50:47 +0000418 std::vector<Instruction*> createdExpressions;
Owen Anderson81d156e2007-05-31 00:42:15 +0000419
Owen Anderson2ff912b2007-06-21 17:57:53 +0000420 std::map<BasicBlock*, SmallPtrSet<Value*, 32> > availableOut;
421 std::map<BasicBlock*, SmallPtrSet<Value*, 32> > anticipatedIn;
Owen Anderson4036ad42007-06-12 22:43:57 +0000422
Owen Andersonfd5683a2007-06-21 00:19:05 +0000423 // This transformation requires dominator postdominator info
Owen Anderson5fba6c12007-05-29 21:53:49 +0000424 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
425 AU.setPreservesCFG();
426 AU.addRequired<DominatorTree>();
Owen Anderson5fba6c12007-05-29 21:53:49 +0000427 }
428
429 // Helper fuctions
430 // FIXME: eliminate or document these better
Owen Anderson92c7b222007-06-22 17:04:40 +0000431 void dump(const SmallPtrSet<Value*, 32>& s) const;
432 void clean(SmallPtrSet<Value*, 32>& set);
Owen Anderson2ff912b2007-06-21 17:57:53 +0000433 Value* find_leader(SmallPtrSet<Value*, 32>& vals,
Owen Anderson92c7b222007-06-22 17:04:40 +0000434 uint32_t v);
435 Value* phi_translate(Value* V, BasicBlock* pred, BasicBlock* succ);
Owen Anderson2ff912b2007-06-21 17:57:53 +0000436 void phi_translate_set(SmallPtrSet<Value*, 32>& anticIn, BasicBlock* pred,
Owen Anderson92c7b222007-06-22 17:04:40 +0000437 BasicBlock* succ, SmallPtrSet<Value*, 32>& out);
Owen Anderson5fba6c12007-05-29 21:53:49 +0000438
Owen Anderson2ff912b2007-06-21 17:57:53 +0000439 void topo_sort(SmallPtrSet<Value*, 32>& set,
Owen Anderson92c7b222007-06-22 17:04:40 +0000440 std::vector<Value*>& vec);
Owen Anderson331bf6a2007-05-31 22:44:11 +0000441
Owen Anderson92c7b222007-06-22 17:04:40 +0000442 void cleanup();
443 bool elimination();
Owen Andersondd998e12007-06-18 04:42:29 +0000444
Owen Anderson92c7b222007-06-22 17:04:40 +0000445 void val_insert(SmallPtrSet<Value*, 32>& s, Value* v);
446 void val_replace(SmallPtrSet<Value*, 32>& s, Value* v);
447 bool dependsOnInvoke(Value* V);
Owen Anderson06c1e582007-06-20 22:10:02 +0000448 void buildsets_availout(BasicBlock::iterator I,
Owen Anderson2ff912b2007-06-21 17:57:53 +0000449 SmallPtrSet<Value*, 32>& currAvail,
450 SmallPtrSet<PHINode*, 32>& currPhis,
451 SmallPtrSet<Value*, 32>& currExps,
Owen Andersonf6e21872007-06-22 03:14:03 +0000452 SmallPtrSet<Value*, 32>& currTemps,
453 BitVector& availNumbers,
Owen Anderson92c7b222007-06-22 17:04:40 +0000454 BitVector& expNumbers);
Owen Anderson28a2d442007-06-22 00:20:30 +0000455 bool buildsets_anticout(BasicBlock* BB,
Owen Anderson2ff912b2007-06-21 17:57:53 +0000456 SmallPtrSet<Value*, 32>& anticOut,
Owen Anderson92c7b222007-06-22 17:04:40 +0000457 std::set<BasicBlock*>& visited);
Owen Anderson28a2d442007-06-22 00:20:30 +0000458 unsigned buildsets_anticin(BasicBlock* BB,
Owen Anderson2ff912b2007-06-21 17:57:53 +0000459 SmallPtrSet<Value*, 32>& anticOut,
460 SmallPtrSet<Value*, 32>& currExps,
461 SmallPtrSet<Value*, 32>& currTemps,
Owen Anderson92c7b222007-06-22 17:04:40 +0000462 std::set<BasicBlock*>& visited);
Owen Anderson5477c542007-06-26 23:29:41 +0000463 void buildsets(Function& F);
Owen Anderson06c1e582007-06-20 22:10:02 +0000464
465 void insertion_pre(Value* e, BasicBlock* BB,
466 std::map<BasicBlock*, Value*>& avail,
Owen Anderson92c7b222007-06-22 17:04:40 +0000467 SmallPtrSet<Value*, 32>& new_set);
Owen Anderson06c1e582007-06-20 22:10:02 +0000468 unsigned insertion_mergepoint(std::vector<Value*>& workList,
Owen Anderson28a2d442007-06-22 00:20:30 +0000469 df_iterator<DomTreeNode*>& D,
Owen Anderson92c7b222007-06-22 17:04:40 +0000470 SmallPtrSet<Value*, 32>& new_set);
471 bool insertion(Function& F);
Owen Anderson5fba6c12007-05-29 21:53:49 +0000472
473 };
474
475 char GVNPRE::ID = 0;
476
477}
478
Owen Andersonfd5683a2007-06-21 00:19:05 +0000479// createGVNPREPass - The public interface to this file...
Owen Anderson5fba6c12007-05-29 21:53:49 +0000480FunctionPass *llvm::createGVNPREPass() { return new GVNPRE(); }
481
482RegisterPass<GVNPRE> X("gvnpre",
483 "Global Value Numbering/Partial Redundancy Elimination");
484
Owen Anderson5fba6c12007-05-29 21:53:49 +0000485
Owen Anderson7d76b2a2007-06-08 22:02:36 +0000486STATISTIC(NumInsertedVals, "Number of values inserted");
487STATISTIC(NumInsertedPhis, "Number of PHI nodes inserted");
488STATISTIC(NumEliminated, "Number of redundant instructions eliminated");
489
Owen Andersonfd5683a2007-06-21 00:19:05 +0000490/// find_leader - Given a set and a value number, return the first
491/// element of the set with that value number, or 0 if no such element
492/// is present
Owen Anderson2ff912b2007-06-21 17:57:53 +0000493Value* GVNPRE::find_leader(SmallPtrSet<Value*, 32>& vals, uint32_t v) {
494 for (SmallPtrSet<Value*, 32>::iterator I = vals.begin(), E = vals.end();
Owen Andersonb56fba02007-06-19 03:31:41 +0000495 I != E; ++I)
496 if (v == VN.lookup(*I))
Owen Anderson0b68cda2007-06-03 05:55:58 +0000497 return *I;
Owen Anderson5fba6c12007-05-29 21:53:49 +0000498
Owen Anderson0b68cda2007-06-03 05:55:58 +0000499 return 0;
Owen Anderson5fba6c12007-05-29 21:53:49 +0000500}
501
Owen Andersonfd5683a2007-06-21 00:19:05 +0000502/// val_insert - Insert a value into a set only if there is not a value
503/// with the same value number already in the set
Owen Anderson2ff912b2007-06-21 17:57:53 +0000504void GVNPRE::val_insert(SmallPtrSet<Value*, 32>& s, Value* v) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000505 uint32_t num = VN.lookup(v);
506 Value* leader = find_leader(s, num);
507 if (leader == 0)
508 s.insert(v);
509}
510
Owen Andersonfd5683a2007-06-21 00:19:05 +0000511/// val_replace - Insert a value into a set, replacing any values already in
512/// the set that have the same value number
Owen Anderson2ff912b2007-06-21 17:57:53 +0000513void GVNPRE::val_replace(SmallPtrSet<Value*, 32>& s, Value* v) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000514 uint32_t num = VN.lookup(v);
515 Value* leader = find_leader(s, num);
516 while (leader != 0) {
517 s.erase(leader);
518 leader = find_leader(s, num);
519 }
520 s.insert(v);
521}
522
Owen Andersonfd5683a2007-06-21 00:19:05 +0000523/// phi_translate - Given a value, its parent block, and a predecessor of its
524/// parent, translate the value into legal for the predecessor block. This
525/// means translating its operands (and recursively, their operands) through
526/// any phi nodes in the parent into values available in the predecessor
Owen Anderson4036ad42007-06-12 22:43:57 +0000527Value* GVNPRE::phi_translate(Value* V, BasicBlock* pred, BasicBlock* succ) {
Owen Anderson38b6b222007-06-04 18:05:26 +0000528 if (V == 0)
529 return 0;
530
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000531 // Binary Operations
532 if (isa<BinaryOperator>(V) || isa<CmpInst>(V) ||
533 isa<ExtractElementInst>(V)) {
Owen Anderson191eb062007-06-25 05:41:12 +0000534 User* U = cast<User>(V);
535
Owen Andersonb56fba02007-06-19 03:31:41 +0000536 Value* newOp1 = 0;
Owen Anderson191eb062007-06-25 05:41:12 +0000537 if (isa<Instruction>(U->getOperand(0)))
538 newOp1 = phi_translate(U->getOperand(0), pred, succ);
Owen Andersonb56fba02007-06-19 03:31:41 +0000539 else
Owen Anderson191eb062007-06-25 05:41:12 +0000540 newOp1 = U->getOperand(0);
Owen Andersonb56fba02007-06-19 03:31:41 +0000541
Owen Anderson38b6b222007-06-04 18:05:26 +0000542 if (newOp1 == 0)
543 return 0;
544
Owen Andersonb56fba02007-06-19 03:31:41 +0000545 Value* newOp2 = 0;
Owen Anderson191eb062007-06-25 05:41:12 +0000546 if (isa<Instruction>(U->getOperand(1)))
547 newOp2 = phi_translate(U->getOperand(1), pred, succ);
Owen Andersonb56fba02007-06-19 03:31:41 +0000548 else
Owen Anderson191eb062007-06-25 05:41:12 +0000549 newOp2 = U->getOperand(1);
Owen Andersonb56fba02007-06-19 03:31:41 +0000550
Owen Anderson38b6b222007-06-04 18:05:26 +0000551 if (newOp2 == 0)
552 return 0;
553
Owen Anderson191eb062007-06-25 05:41:12 +0000554 if (newOp1 != U->getOperand(0) || newOp2 != U->getOperand(1)) {
555 Instruction* newVal = 0;
556 if (BinaryOperator* BO = dyn_cast<BinaryOperator>(U))
557 newVal = BinaryOperator::create(BO->getOpcode(),
558 newOp1, newOp2,
559 BO->getName()+".expr");
560 else if (CmpInst* C = dyn_cast<CmpInst>(U))
561 newVal = CmpInst::create(C->getOpcode(),
562 C->getPredicate(),
563 newOp1, newOp2,
564 C->getName()+".expr");
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000565 else if (ExtractElementInst* E = dyn_cast<ExtractElementInst>(U))
566 newVal = new ExtractElementInst(newOp1, newOp2, E->getName()+".expr");
Owen Anderson55994f22007-06-08 20:44:02 +0000567
Owen Andersonb56fba02007-06-19 03:31:41 +0000568 uint32_t v = VN.lookup_or_add(newVal);
Owen Anderson4036ad42007-06-12 22:43:57 +0000569
Owen Andersonb56fba02007-06-19 03:31:41 +0000570 Value* leader = find_leader(availableOut[pred], v);
Owen Anderson4036ad42007-06-12 22:43:57 +0000571 if (leader == 0) {
Owen Andersona75dd4d2007-06-12 00:50:47 +0000572 createdExpressions.push_back(newVal);
Owen Anderson38b6b222007-06-04 18:05:26 +0000573 return newVal;
Owen Andersonc8472092007-06-05 22:11:49 +0000574 } else {
Owen Anderson91c54952007-06-19 05:37:32 +0000575 VN.erase(newVal);
Owen Andersonc8472092007-06-05 22:11:49 +0000576 delete newVal;
Owen Anderson4036ad42007-06-12 22:43:57 +0000577 return leader;
Owen Andersonc8472092007-06-05 22:11:49 +0000578 }
Owen Anderson38b6b222007-06-04 18:05:26 +0000579 }
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000580
581 // Ternary Operations
582 } else if (isa<ShuffleVectorInst>(V) || isa<InsertElementInst>(V)) {
583 User* U = cast<User>(V);
584
585 Value* newOp1 = 0;
586 if (isa<Instruction>(U->getOperand(0)))
587 newOp1 = phi_translate(U->getOperand(0), pred, succ);
588 else
589 newOp1 = U->getOperand(0);
590
591 if (newOp1 == 0)
592 return 0;
593
594 Value* newOp2 = 0;
595 if (isa<Instruction>(U->getOperand(1)))
596 newOp2 = phi_translate(U->getOperand(1), pred, succ);
597 else
598 newOp2 = U->getOperand(1);
599
600 if (newOp2 == 0)
601 return 0;
602
603 Value* newOp3 = 0;
604 if (isa<Instruction>(U->getOperand(2)))
605 newOp3 = phi_translate(U->getOperand(2), pred, succ);
606 else
607 newOp3 = U->getOperand(2);
608
609 if (newOp3 == 0)
610 return 0;
611
612 if (newOp1 != U->getOperand(0) ||
613 newOp2 != U->getOperand(1) ||
614 newOp3 != U->getOperand(2)) {
615 Instruction* newVal = 0;
616 if (ShuffleVectorInst* S = dyn_cast<ShuffleVectorInst>(U))
617 newVal = new ShuffleVectorInst(newOp1, newOp2, newOp3,
618 S->getName()+".expr");
619 else if (InsertElementInst* I = dyn_cast<InsertElementInst>(U))
620 newVal = new InsertElementInst(newOp1, newOp2, newOp3,
621 I->getName()+".expr");
622
623 uint32_t v = VN.lookup_or_add(newVal);
624
625 Value* leader = find_leader(availableOut[pred], v);
626 if (leader == 0) {
627 createdExpressions.push_back(newVal);
628 return newVal;
629 } else {
630 VN.erase(newVal);
631 delete newVal;
632 return leader;
633 }
634 }
635
636 // PHI Nodes
Owen Anderson38b6b222007-06-04 18:05:26 +0000637 } else if (PHINode* P = dyn_cast<PHINode>(V)) {
Owen Andersona75dd4d2007-06-12 00:50:47 +0000638 if (P->getParent() == succ)
Owen Anderson38b6b222007-06-04 18:05:26 +0000639 return P->getIncomingValueForBlock(pred);
640 }
641
642 return V;
643}
644
Owen Andersonfd5683a2007-06-21 00:19:05 +0000645/// phi_translate_set - Perform phi translation on every element of a set
Owen Anderson2ff912b2007-06-21 17:57:53 +0000646void GVNPRE::phi_translate_set(SmallPtrSet<Value*, 32>& anticIn,
Owen Andersona75dd4d2007-06-12 00:50:47 +0000647 BasicBlock* pred, BasicBlock* succ,
Owen Anderson2ff912b2007-06-21 17:57:53 +0000648 SmallPtrSet<Value*, 32>& out) {
649 for (SmallPtrSet<Value*, 32>::iterator I = anticIn.begin(),
Owen Anderson38b6b222007-06-04 18:05:26 +0000650 E = anticIn.end(); I != E; ++I) {
Owen Anderson4036ad42007-06-12 22:43:57 +0000651 Value* V = phi_translate(*I, pred, succ);
Owen Anderson38b6b222007-06-04 18:05:26 +0000652 if (V != 0)
653 out.insert(V);
Owen Anderson5fba6c12007-05-29 21:53:49 +0000654 }
655}
656
Owen Andersonfd5683a2007-06-21 00:19:05 +0000657/// dependsOnInvoke - Test if a value has an phi node as an operand, any of
658/// whose inputs is an invoke instruction. If this is true, we cannot safely
659/// PRE the instruction or anything that depends on it.
Owen Andersondd998e12007-06-18 04:42:29 +0000660bool GVNPRE::dependsOnInvoke(Value* V) {
Owen Anderson2320d432007-06-19 23:07:16 +0000661 if (PHINode* p = dyn_cast<PHINode>(V)) {
662 for (PHINode::op_iterator I = p->op_begin(), E = p->op_end(); I != E; ++I)
663 if (isa<InvokeInst>(*I))
Owen Andersondd998e12007-06-18 04:42:29 +0000664 return true;
Owen Anderson2320d432007-06-19 23:07:16 +0000665 return false;
666 } else {
667 return false;
Owen Anderson658f2c42007-06-16 00:26:54 +0000668 }
Owen Anderson658f2c42007-06-16 00:26:54 +0000669}
670
Owen Andersonfd5683a2007-06-21 00:19:05 +0000671/// clean - Remove all non-opaque values from the set whose operands are not
672/// themselves in the set, as well as all values that depend on invokes (see
673/// above)
Owen Anderson2ff912b2007-06-21 17:57:53 +0000674void GVNPRE::clean(SmallPtrSet<Value*, 32>& set) {
Owen Anderson0b68cda2007-06-03 05:55:58 +0000675 std::vector<Value*> worklist;
Owen Andersond50a29d2007-06-22 00:43:22 +0000676 worklist.reserve(set.size());
Owen Andersonbe802402007-06-08 01:03:01 +0000677 topo_sort(set, worklist);
Owen Anderson5fba6c12007-05-29 21:53:49 +0000678
Owen Andersonbe802402007-06-08 01:03:01 +0000679 for (unsigned i = 0; i < worklist.size(); ++i) {
680 Value* v = worklist[i];
Owen Anderson5fba6c12007-05-29 21:53:49 +0000681
Owen Anderson0eb26572007-06-27 17:03:03 +0000682 // Handle binary ops
683 if (isa<BinaryOperator>(v) || isa<CmpInst>(v) ||
684 isa<ExtractElementInst>(v)) {
685 User* U = cast<User>(v);
686
687 bool lhsValid = !isa<Instruction>(U->getOperand(0));
Owen Andersonbe802402007-06-08 01:03:01 +0000688 if (!lhsValid)
Owen Anderson2ff912b2007-06-21 17:57:53 +0000689 for (SmallPtrSet<Value*, 32>::iterator I = set.begin(), E = set.end();
Owen Andersonbe802402007-06-08 01:03:01 +0000690 I != E; ++I)
Owen Anderson0eb26572007-06-27 17:03:03 +0000691 if (VN.lookup(*I) == VN.lookup(U->getOperand(0))) {
Owen Andersonbe802402007-06-08 01:03:01 +0000692 lhsValid = true;
693 break;
694 }
Owen Andersonb364b412007-06-18 04:30:44 +0000695 if (lhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000696 lhsValid = !dependsOnInvoke(U->getOperand(0));
Owen Anderson0b68cda2007-06-03 05:55:58 +0000697
Owen Anderson0eb26572007-06-27 17:03:03 +0000698 bool rhsValid = !isa<Instruction>(U->getOperand(1));
Owen Andersonbe802402007-06-08 01:03:01 +0000699 if (!rhsValid)
Owen Anderson2ff912b2007-06-21 17:57:53 +0000700 for (SmallPtrSet<Value*, 32>::iterator I = set.begin(), E = set.end();
Owen Andersonf1c04e12007-06-18 04:31:21 +0000701 I != E; ++I)
Owen Anderson0eb26572007-06-27 17:03:03 +0000702 if (VN.lookup(*I) == VN.lookup(U->getOperand(1))) {
Owen Andersonf1c04e12007-06-18 04:31:21 +0000703 rhsValid = true;
704 break;
705 }
Owen Andersonb364b412007-06-18 04:30:44 +0000706 if (rhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000707 rhsValid = !dependsOnInvoke(U->getOperand(1));
Owen Anderson5fba6c12007-05-29 21:53:49 +0000708
Owen Anderson0b68cda2007-06-03 05:55:58 +0000709 if (!lhsValid || !rhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000710 set.erase(U);
Owen Anderson223718c2007-06-09 18:35:31 +0000711
Owen Anderson0eb26572007-06-27 17:03:03 +0000712 // Handle ternary ops
713 } else if (isa<ShuffleVectorInst>(v) || isa<InsertElementInst>(v)) {
714 User* U = cast<User>(v);
715
716 bool lhsValid = !isa<Instruction>(U->getOperand(0));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000717 if (!lhsValid)
718 for (SmallPtrSet<Value*, 32>::iterator I = set.begin(), E = set.end();
719 I != E; ++I)
Owen Anderson0eb26572007-06-27 17:03:03 +0000720 if (VN.lookup(*I) == VN.lookup(U->getOperand(0))) {
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000721 lhsValid = true;
722 break;
723 }
724 if (lhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000725 lhsValid = !dependsOnInvoke(U->getOperand(0));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000726
Owen Anderson0eb26572007-06-27 17:03:03 +0000727 bool rhsValid = !isa<Instruction>(U->getOperand(1));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000728 if (!rhsValid)
729 for (SmallPtrSet<Value*, 32>::iterator I = set.begin(), E = set.end();
730 I != E; ++I)
Owen Anderson0eb26572007-06-27 17:03:03 +0000731 if (VN.lookup(*I) == VN.lookup(U->getOperand(1))) {
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000732 rhsValid = true;
733 break;
734 }
735 if (rhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000736 rhsValid = !dependsOnInvoke(U->getOperand(1));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000737
Owen Anderson0eb26572007-06-27 17:03:03 +0000738 bool thirdValid = !isa<Instruction>(U->getOperand(2));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000739 if (!thirdValid)
740 for (SmallPtrSet<Value*, 32>::iterator I = set.begin(), E = set.end();
741 I != E; ++I)
Owen Anderson0eb26572007-06-27 17:03:03 +0000742 if (VN.lookup(*I) == VN.lookup(U->getOperand(2))) {
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000743 thirdValid = true;
744 break;
745 }
746 if (thirdValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000747 thirdValid = !dependsOnInvoke(U->getOperand(2));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000748
749 if (!lhsValid || !rhsValid || !thirdValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000750 set.erase(U);
Owen Anderson0b68cda2007-06-03 05:55:58 +0000751 }
Owen Anderson5fba6c12007-05-29 21:53:49 +0000752 }
753}
754
Owen Andersonfd5683a2007-06-21 00:19:05 +0000755/// topo_sort - Given a set of values, sort them by topological
756/// order into the provided vector.
Owen Anderson2ff912b2007-06-21 17:57:53 +0000757void GVNPRE::topo_sort(SmallPtrSet<Value*, 32>& set, std::vector<Value*>& vec) {
Owen Anderson2ff912b2007-06-21 17:57:53 +0000758 SmallPtrSet<Value*, 32> visited;
Owen Anderson49409f62007-06-22 21:31:16 +0000759 std::vector<Value*> stack;
760 for (SmallPtrSet<Value*, 32>::iterator I = set.begin(), E = set.end();
761 I != E; ++I) {
762 if (visited.count(*I) == 0)
763 stack.push_back(*I);
764
765 while (!stack.empty()) {
766 Value* e = stack.back();
Owen Anderson0b68cda2007-06-03 05:55:58 +0000767
Owen Anderson0eb26572007-06-27 17:03:03 +0000768 // Handle binary ops
769 if (isa<BinaryOperator>(e) || isa<CmpInst>(e) ||
770 isa<ExtractElementInst>(e)) {
771 User* U = cast<User>(e);
772 Value* l = find_leader(set, VN.lookup(U->getOperand(0)));
773 Value* r = find_leader(set, VN.lookup(U->getOperand(1)));
Owen Anderson49409f62007-06-22 21:31:16 +0000774
775 if (l != 0 && isa<Instruction>(l) &&
776 visited.count(l) == 0)
777 stack.push_back(l);
778 else if (r != 0 && isa<Instruction>(r) &&
779 visited.count(r) == 0)
780 stack.push_back(r);
781 else {
782 vec.push_back(e);
783 visited.insert(e);
784 stack.pop_back();
785 }
Owen Anderson0b68cda2007-06-03 05:55:58 +0000786
Owen Anderson0eb26572007-06-27 17:03:03 +0000787 // Handle ternary ops
788 } else if (isa<InsertElementInst>(e) || isa<ShuffleVectorInst>(e)) {
789 User* U = cast<User>(e);
790 Value* l = find_leader(set, VN.lookup(U->getOperand(0)));
791 Value* r = find_leader(set, VN.lookup(U->getOperand(1)));
792 Value* m = find_leader(set, VN.lookup(U->getOperand(2)));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000793
794 if (l != 0 && isa<Instruction>(l) &&
795 visited.count(l) == 0)
796 stack.push_back(l);
797 else if (r != 0 && isa<Instruction>(r) &&
798 visited.count(r) == 0)
799 stack.push_back(r);
800 else if (m != 0 && isa<Instruction>(m) &&
801 visited.count(m) == 0)
802 stack.push_back(r);
803 else {
804 vec.push_back(e);
805 visited.insert(e);
806 stack.pop_back();
807 }
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000808
Owen Anderson0eb26572007-06-27 17:03:03 +0000809 // Handle opaque ops
Owen Anderson49409f62007-06-22 21:31:16 +0000810 } else {
Owen Anderson0b68cda2007-06-03 05:55:58 +0000811 visited.insert(e);
Owen Anderson223718c2007-06-09 18:35:31 +0000812 vec.push_back(e);
Owen Anderson49409f62007-06-22 21:31:16 +0000813 stack.pop_back();
Owen Anderson223718c2007-06-09 18:35:31 +0000814 }
Owen Anderson331bf6a2007-05-31 22:44:11 +0000815 }
Owen Anderson49409f62007-06-22 21:31:16 +0000816
817 stack.clear();
Owen Anderson331bf6a2007-05-31 22:44:11 +0000818 }
819}
820
Owen Andersonfd5683a2007-06-21 00:19:05 +0000821/// dump - Dump a set of values to standard error
Owen Anderson2ff912b2007-06-21 17:57:53 +0000822void GVNPRE::dump(const SmallPtrSet<Value*, 32>& s) const {
Owen Anderson4a6ec8f2007-05-29 23:15:21 +0000823 DOUT << "{ ";
Owen Anderson2ff912b2007-06-21 17:57:53 +0000824 for (SmallPtrSet<Value*, 32>::iterator I = s.begin(), E = s.end();
Owen Anderson0eca9aa2007-06-03 22:02:14 +0000825 I != E; ++I) {
826 DEBUG((*I)->dump());
827 }
828 DOUT << "}\n\n";
829}
830
Owen Andersonfd5683a2007-06-21 00:19:05 +0000831/// elimination - Phase 3 of the main algorithm. Perform full redundancy
832/// elimination by walking the dominator tree and removing any instruction that
833/// is dominated by another instruction with the same value number.
Owen Anderson06c1e582007-06-20 22:10:02 +0000834bool GVNPRE::elimination() {
Owen Anderson42769842007-06-12 16:57:50 +0000835 DOUT << "\n\nPhase 3: Elimination\n\n";
836
Owen Anderson06c1e582007-06-20 22:10:02 +0000837 bool changed_function = false;
838
Owen Anderson42769842007-06-12 16:57:50 +0000839 std::vector<std::pair<Instruction*, Value*> > replace;
840 std::vector<Instruction*> erase;
841
842 DominatorTree& DT = getAnalysis<DominatorTree>();
843
844 for (df_iterator<DomTreeNode*> DI = df_begin(DT.getRootNode()),
845 E = df_end(DT.getRootNode()); DI != E; ++DI) {
846 BasicBlock* BB = DI->getBlock();
847
Owen Anderson191eb062007-06-25 05:41:12 +0000848 //DOUT << "Block: " << BB->getName() << "\n";
849 //dump(availableOut[BB]);
850 //DOUT << "\n\n";
Owen Anderson42769842007-06-12 16:57:50 +0000851
852 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
853 BI != BE; ++BI) {
854
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000855 if (isa<BinaryOperator>(BI) || isa<CmpInst>(BI) ||
856 isa<ShuffleVectorInst>(BI) || isa<InsertElementInst>(BI) ||
857 isa<ExtractElementInst>(BI)) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000858 Value *leader = find_leader(availableOut[BB], VN.lookup(BI));
Owen Anderson42769842007-06-12 16:57:50 +0000859
860 if (leader != 0)
861 if (Instruction* Instr = dyn_cast<Instruction>(leader))
862 if (Instr->getParent() != 0 && Instr != BI) {
863 replace.push_back(std::make_pair(BI, leader));
864 erase.push_back(BI);
865 ++NumEliminated;
866 }
867 }
868 }
869 }
870
871 while (!replace.empty()) {
872 std::pair<Instruction*, Value*> rep = replace.back();
873 replace.pop_back();
874 rep.first->replaceAllUsesWith(rep.second);
Owen Andersonb0714bb2007-06-20 18:30:20 +0000875 changed_function = true;
Owen Anderson42769842007-06-12 16:57:50 +0000876 }
877
878 for (std::vector<Instruction*>::iterator I = erase.begin(), E = erase.end();
879 I != E; ++I)
880 (*I)->eraseFromParent();
Owen Anderson06c1e582007-06-20 22:10:02 +0000881
882 return changed_function;
Owen Anderson42769842007-06-12 16:57:50 +0000883}
884
Owen Andersonfd5683a2007-06-21 00:19:05 +0000885/// cleanup - Delete any extraneous values that were created to represent
886/// expressions without leaders.
Owen Anderson42769842007-06-12 16:57:50 +0000887void GVNPRE::cleanup() {
888 while (!createdExpressions.empty()) {
889 Instruction* I = createdExpressions.back();
890 createdExpressions.pop_back();
891
892 delete I;
893 }
894}
895
Owen Andersonfd5683a2007-06-21 00:19:05 +0000896/// buildsets_availout - When calculating availability, handle an instruction
897/// by inserting it into the appropriate sets
Owen Anderson06c1e582007-06-20 22:10:02 +0000898void GVNPRE::buildsets_availout(BasicBlock::iterator I,
Owen Anderson2ff912b2007-06-21 17:57:53 +0000899 SmallPtrSet<Value*, 32>& currAvail,
900 SmallPtrSet<PHINode*, 32>& currPhis,
901 SmallPtrSet<Value*, 32>& currExps,
Owen Andersonf6e21872007-06-22 03:14:03 +0000902 SmallPtrSet<Value*, 32>& currTemps,
903 BitVector& availNumbers,
904 BitVector& expNumbers) {
Owen Anderson0eb26572007-06-27 17:03:03 +0000905 // Handle PHI nodes
Owen Anderson06c1e582007-06-20 22:10:02 +0000906 if (PHINode* p = dyn_cast<PHINode>(I)) {
907 VN.lookup_or_add(p);
Owen Andersonf6e21872007-06-22 03:14:03 +0000908 expNumbers.resize(VN.size());
909 availNumbers.resize(VN.size());
910
Owen Anderson06c1e582007-06-20 22:10:02 +0000911 currPhis.insert(p);
912
Owen Anderson0eb26572007-06-27 17:03:03 +0000913 // Handle binary ops
914 } else if (isa<BinaryOperator>(I) || isa<CmpInst>(I) ||
915 isa<ExtractElementInst>(I)) {
916 User* U = cast<User>(I);
917 Value* leftValue = U->getOperand(0);
918 Value* rightValue = U->getOperand(1);
Owen Anderson06c1e582007-06-20 22:10:02 +0000919
Owen Anderson0eb26572007-06-27 17:03:03 +0000920 unsigned num = VN.lookup_or_add(U);
Owen Andersonf6e21872007-06-22 03:14:03 +0000921 expNumbers.resize(VN.size());
922 availNumbers.resize(VN.size());
Owen Anderson06c1e582007-06-20 22:10:02 +0000923
924 if (isa<Instruction>(leftValue))
Owen Anderson21a11312007-06-22 18:27:04 +0000925 if (!expNumbers.test(VN.lookup(leftValue))) {
Owen Andersonf6e21872007-06-22 03:14:03 +0000926 currExps.insert(leftValue);
Owen Anderson21a11312007-06-22 18:27:04 +0000927 expNumbers.set(VN.lookup(leftValue));
Owen Andersonf6e21872007-06-22 03:14:03 +0000928 }
929
Owen Anderson06c1e582007-06-20 22:10:02 +0000930 if (isa<Instruction>(rightValue))
Owen Anderson21a11312007-06-22 18:27:04 +0000931 if (!expNumbers.test(VN.lookup(rightValue))) {
Owen Andersonf6e21872007-06-22 03:14:03 +0000932 currExps.insert(rightValue);
Owen Anderson21a11312007-06-22 18:27:04 +0000933 expNumbers.set(VN.lookup(rightValue));
Owen Andersonf6e21872007-06-22 03:14:03 +0000934 }
935
Owen Anderson0eb26572007-06-27 17:03:03 +0000936 if (!expNumbers.test(VN.lookup(U))) {
937 currExps.insert(U);
Owen Anderson21a11312007-06-22 18:27:04 +0000938 expNumbers.set(num);
Owen Andersonf6e21872007-06-22 03:14:03 +0000939 }
Owen Anderson06c1e582007-06-20 22:10:02 +0000940
Owen Anderson0eb26572007-06-27 17:03:03 +0000941 // Handle ternary ops
942 } else if (isa<InsertElementInst>(I) || isa<ShuffleVectorInst>(I)) {
943 User* U = cast<User>(I);
944 Value* leftValue = U->getOperand(0);
945 Value* rightValue = U->getOperand(1);
946 Value* thirdValue = U->getOperand(2);
Owen Anderson06c1e582007-06-20 22:10:02 +0000947
Owen Anderson0eb26572007-06-27 17:03:03 +0000948 VN.lookup_or_add(U);
Owen Andersonf6e21872007-06-22 03:14:03 +0000949
Owen Anderson0eb26572007-06-27 17:03:03 +0000950 unsigned num = VN.lookup_or_add(U);
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000951 expNumbers.resize(VN.size());
952 availNumbers.resize(VN.size());
953
954 if (isa<Instruction>(leftValue))
955 if (!expNumbers.test(VN.lookup(leftValue))) {
956 currExps.insert(leftValue);
957 expNumbers.set(VN.lookup(leftValue));
958 }
959 if (isa<Instruction>(rightValue))
960 if (!expNumbers.test(VN.lookup(rightValue))) {
961 currExps.insert(rightValue);
962 expNumbers.set(VN.lookup(rightValue));
963 }
964 if (isa<Instruction>(thirdValue))
965 if (!expNumbers.test(VN.lookup(thirdValue))) {
966 currExps.insert(thirdValue);
967 expNumbers.set(VN.lookup(thirdValue));
968 }
969
Owen Anderson0eb26572007-06-27 17:03:03 +0000970 if (!expNumbers.test(VN.lookup(U))) {
971 currExps.insert(U);
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000972 expNumbers.set(num);
973 }
974
Owen Anderson0eb26572007-06-27 17:03:03 +0000975 // Handle opaque ops
976 } else if (!I->isTerminator()){
Owen Anderson06c1e582007-06-20 22:10:02 +0000977 VN.lookup_or_add(I);
Owen Andersonf6e21872007-06-22 03:14:03 +0000978 expNumbers.resize(VN.size());
979 availNumbers.resize(VN.size());
980
Owen Anderson06c1e582007-06-20 22:10:02 +0000981 currTemps.insert(I);
982 }
983
984 if (!I->isTerminator())
Owen Anderson21a11312007-06-22 18:27:04 +0000985 if (!availNumbers.test(VN.lookup(I))) {
Owen Andersonf6e21872007-06-22 03:14:03 +0000986 currAvail.insert(I);
Owen Anderson21a11312007-06-22 18:27:04 +0000987 availNumbers.set(VN.lookup(I));
Owen Andersonf6e21872007-06-22 03:14:03 +0000988 }
Owen Anderson06c1e582007-06-20 22:10:02 +0000989}
Owen Anderson5fba6c12007-05-29 21:53:49 +0000990
Owen Andersonfd5683a2007-06-21 00:19:05 +0000991/// buildsets_anticout - When walking the postdom tree, calculate the ANTIC_OUT
992/// set as a function of the ANTIC_IN set of the block's predecessors
Owen Anderson28a2d442007-06-22 00:20:30 +0000993bool GVNPRE::buildsets_anticout(BasicBlock* BB,
Owen Anderson2ff912b2007-06-21 17:57:53 +0000994 SmallPtrSet<Value*, 32>& anticOut,
Owen Anderson06c1e582007-06-20 22:10:02 +0000995 std::set<BasicBlock*>& visited) {
996 if (BB->getTerminator()->getNumSuccessors() == 1) {
Owen Anderson191eb062007-06-25 05:41:12 +0000997 if (BB->getTerminator()->getSuccessor(0) != BB &&
998 visited.count(BB->getTerminator()->getSuccessor(0)) == 0) {
999 DOUT << "DEFER: " << BB->getName() << "\n";
Owen Anderson28a2d442007-06-22 00:20:30 +00001000 return true;
Owen Anderson191eb062007-06-25 05:41:12 +00001001 }
1002 else {
Owen Anderson06c1e582007-06-20 22:10:02 +00001003 phi_translate_set(anticipatedIn[BB->getTerminator()->getSuccessor(0)],
1004 BB, BB->getTerminator()->getSuccessor(0), anticOut);
Owen Anderson191eb062007-06-25 05:41:12 +00001005 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001006 } else if (BB->getTerminator()->getNumSuccessors() > 1) {
1007 BasicBlock* first = BB->getTerminator()->getSuccessor(0);
1008 anticOut.insert(anticipatedIn[first].begin(), anticipatedIn[first].end());
1009
1010 for (unsigned i = 1; i < BB->getTerminator()->getNumSuccessors(); ++i) {
1011 BasicBlock* currSucc = BB->getTerminator()->getSuccessor(i);
Owen Anderson2ff912b2007-06-21 17:57:53 +00001012 SmallPtrSet<Value*, 32>& succAnticIn = anticipatedIn[currSucc];
Owen Anderson06c1e582007-06-20 22:10:02 +00001013
Owen Anderson2ff912b2007-06-21 17:57:53 +00001014 std::vector<Value*> temp;
1015
1016 for (SmallPtrSet<Value*, 32>::iterator I = anticOut.begin(),
1017 E = anticOut.end(); I != E; ++I)
1018 if (succAnticIn.count(*I) == 0)
1019 temp.push_back(*I);
1020
1021 for (std::vector<Value*>::iterator I = temp.begin(), E = temp.end();
1022 I != E; ++I)
1023 anticOut.erase(*I);
Owen Anderson06c1e582007-06-20 22:10:02 +00001024 }
1025 }
Owen Anderson28a2d442007-06-22 00:20:30 +00001026
1027 return false;
Owen Anderson06c1e582007-06-20 22:10:02 +00001028}
1029
Owen Andersonfd5683a2007-06-21 00:19:05 +00001030/// buildsets_anticin - Walk the postdom tree, calculating ANTIC_OUT for
1031/// each block. ANTIC_IN is then a function of ANTIC_OUT and the GEN
1032/// sets populated in buildsets_availout
Owen Anderson28a2d442007-06-22 00:20:30 +00001033unsigned GVNPRE::buildsets_anticin(BasicBlock* BB,
Owen Anderson2ff912b2007-06-21 17:57:53 +00001034 SmallPtrSet<Value*, 32>& anticOut,
1035 SmallPtrSet<Value*, 32>& currExps,
1036 SmallPtrSet<Value*, 32>& currTemps,
Owen Anderson06c1e582007-06-20 22:10:02 +00001037 std::set<BasicBlock*>& visited) {
Owen Anderson2ff912b2007-06-21 17:57:53 +00001038 SmallPtrSet<Value*, 32>& anticIn = anticipatedIn[BB];
Owen Anderson21a11312007-06-22 18:27:04 +00001039 unsigned old = anticIn.size();
Owen Anderson06c1e582007-06-20 22:10:02 +00001040
Owen Anderson28a2d442007-06-22 00:20:30 +00001041 bool defer = buildsets_anticout(BB, anticOut, visited);
1042 if (defer)
1043 return 0;
Owen Anderson2ff912b2007-06-21 17:57:53 +00001044
Owen Anderson191eb062007-06-25 05:41:12 +00001045
1046
Owen Anderson06c1e582007-06-20 22:10:02 +00001047 anticIn.clear();
Owen Anderson2ff912b2007-06-21 17:57:53 +00001048
Owen Anderson28a2d442007-06-22 00:20:30 +00001049 BitVector numbers(VN.size());
Owen Anderson21a11312007-06-22 18:27:04 +00001050 for (SmallPtrSet<Value*, 32>::iterator I = anticOut.begin(),
1051 E = anticOut.end(); I != E; ++I) {
1052 anticIn.insert(*I);
Owen Anderson191eb062007-06-25 05:41:12 +00001053 unsigned num = VN.lookup_or_add(*I);
1054 numbers.resize(VN.size());
1055 numbers.set(num);
Owen Anderson06c1e582007-06-20 22:10:02 +00001056 }
Owen Anderson21a11312007-06-22 18:27:04 +00001057 for (SmallPtrSet<Value*, 32>::iterator I = currExps.begin(),
1058 E = currExps.end(); I != E; ++I) {
1059 if (!numbers.test(VN.lookup_or_add(*I))) {
1060 anticIn.insert(*I);
1061 numbers.set(VN.lookup(*I));
1062 }
1063 }
1064
1065 for (SmallPtrSet<Value*, 32>::iterator I = currTemps.begin(),
1066 E = currTemps.end(); I != E; ++I)
1067 anticIn.erase(*I);
1068
Owen Anderson06c1e582007-06-20 22:10:02 +00001069 clean(anticIn);
Owen Anderson06c1e582007-06-20 22:10:02 +00001070
Owen Anderson191eb062007-06-25 05:41:12 +00001071 if (old != anticIn.size()) {
1072 DOUT << "OLD: " << old << "\n";
1073 DOUT << "NEW: " << anticIn.size() << "\n";
1074 DOUT << "ANTIC_OUT: " << anticOut.size() << "\n";
1075 anticOut.clear();
Owen Anderson28a2d442007-06-22 00:20:30 +00001076 return 2;
Owen Anderson191eb062007-06-25 05:41:12 +00001077 } else
1078 anticOut.clear();
Owen Anderson28a2d442007-06-22 00:20:30 +00001079 return 1;
Owen Anderson06c1e582007-06-20 22:10:02 +00001080}
1081
Owen Andersonfd5683a2007-06-21 00:19:05 +00001082/// buildsets - Phase 1 of the main algorithm. Construct the AVAIL_OUT
1083/// and the ANTIC_IN sets.
Owen Anderson5477c542007-06-26 23:29:41 +00001084void GVNPRE::buildsets(Function& F) {
Owen Anderson2ff912b2007-06-21 17:57:53 +00001085 std::map<BasicBlock*, SmallPtrSet<Value*, 32> > generatedExpressions;
1086 std::map<BasicBlock*, SmallPtrSet<PHINode*, 32> > generatedPhis;
1087 std::map<BasicBlock*, SmallPtrSet<Value*, 32> > generatedTemporaries;
Owen Anderson06c1e582007-06-20 22:10:02 +00001088
Owen Anderson5fba6c12007-05-29 21:53:49 +00001089 DominatorTree &DT = getAnalysis<DominatorTree>();
1090
Owen Anderson634a0632007-06-06 01:27:49 +00001091 // Phase 1, Part 1: calculate AVAIL_OUT
Owen Anderson5fba6c12007-05-29 21:53:49 +00001092
1093 // Top-down walk of the dominator tree
Devang Patelbdd1aae2007-06-04 00:32:22 +00001094 for (df_iterator<DomTreeNode*> DI = df_begin(DT.getRootNode()),
Owen Anderson5fba6c12007-05-29 21:53:49 +00001095 E = df_end(DT.getRootNode()); DI != E; ++DI) {
1096
1097 // Get the sets to update for this block
Owen Anderson2ff912b2007-06-21 17:57:53 +00001098 SmallPtrSet<Value*, 32>& currExps = generatedExpressions[DI->getBlock()];
1099 SmallPtrSet<PHINode*, 32>& currPhis = generatedPhis[DI->getBlock()];
1100 SmallPtrSet<Value*, 32>& currTemps = generatedTemporaries[DI->getBlock()];
1101 SmallPtrSet<Value*, 32>& currAvail = availableOut[DI->getBlock()];
Owen Anderson5fba6c12007-05-29 21:53:49 +00001102
Owen Andersonb56fba02007-06-19 03:31:41 +00001103 BasicBlock* BB = DI->getBlock();
1104
1105 // A block inherits AVAIL_OUT from its dominator
1106 if (DI->getIDom() != 0)
1107 currAvail.insert(availableOut[DI->getIDom()->getBlock()].begin(),
1108 availableOut[DI->getIDom()->getBlock()].end());
1109
Owen Andersonf6e21872007-06-22 03:14:03 +00001110 BitVector availNumbers(VN.size());
1111 for (SmallPtrSet<Value*, 32>::iterator I = currAvail.begin(),
1112 E = currAvail.end(); I != E; ++I)
1113 availNumbers.set(VN.lookup(*I));
Owen Andersonb56fba02007-06-19 03:31:41 +00001114
Owen Andersonf6e21872007-06-22 03:14:03 +00001115 BitVector expNumbers(VN.size());
Owen Andersonb56fba02007-06-19 03:31:41 +00001116 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
Owen Anderson06c1e582007-06-20 22:10:02 +00001117 BI != BE; ++BI)
Owen Andersonf6e21872007-06-22 03:14:03 +00001118 buildsets_availout(BI, currAvail, currPhis, currExps,
1119 currTemps, availNumbers, expNumbers);
Owen Andersonb56fba02007-06-19 03:31:41 +00001120
Owen Anderson5fba6c12007-05-29 21:53:49 +00001121 }
Owen Anderson191eb062007-06-25 05:41:12 +00001122
1123 // Phase 1, Part 2: calculate ANTIC_IN
Owen Anderson5fba6c12007-05-29 21:53:49 +00001124
Owen Anderson0c423072007-05-29 23:26:30 +00001125 std::set<BasicBlock*> visited;
Owen Anderson5477c542007-06-26 23:29:41 +00001126 SmallPtrSet<BasicBlock*, 4> block_changed;
1127 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ++FI)
1128 block_changed.insert(FI);
Owen Anderson0c423072007-05-29 23:26:30 +00001129
Owen Anderson5fba6c12007-05-29 21:53:49 +00001130 bool changed = true;
1131 unsigned iterations = 0;
Owen Anderson5477c542007-06-26 23:29:41 +00001132
Owen Anderson5fba6c12007-05-29 21:53:49 +00001133 while (changed) {
1134 changed = false;
Owen Anderson2ff912b2007-06-21 17:57:53 +00001135 SmallPtrSet<Value*, 32> anticOut;
Owen Anderson5fba6c12007-05-29 21:53:49 +00001136
Owen Anderson5477c542007-06-26 23:29:41 +00001137 // Postorder walk of the CFG
Owen Anderson43ca4b42007-06-25 18:25:31 +00001138 for (po_iterator<BasicBlock*> BBI = po_begin(&F.getEntryBlock()),
1139 BBE = po_end(&F.getEntryBlock()); BBI != BBE; ++BBI) {
Owen Anderson191eb062007-06-25 05:41:12 +00001140 BasicBlock* BB = *BBI;
Owen Andersond184c182007-06-11 16:25:17 +00001141
Owen Anderson5477c542007-06-26 23:29:41 +00001142 if (block_changed.count(BB) != 0) {
1143 unsigned ret = buildsets_anticin(BB, anticOut,generatedExpressions[BB],
1144 generatedTemporaries[BB], visited);
Owen Anderson28a2d442007-06-22 00:20:30 +00001145
Owen Anderson5477c542007-06-26 23:29:41 +00001146 if (ret == 0) {
1147 changed = true;
1148 continue;
1149 } else {
1150 visited.insert(BB);
1151
1152 if (ret == 2)
1153 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
1154 PI != PE; ++PI) {
1155 block_changed.insert(*PI);
1156 }
1157 else
1158 block_changed.erase(BB);
1159
1160 changed |= (ret == 2);
Owen Anderson191eb062007-06-25 05:41:12 +00001161 }
Owen Anderson28a2d442007-06-22 00:20:30 +00001162 }
Owen Anderson5fba6c12007-05-29 21:53:49 +00001163 }
Owen Anderson38b6b222007-06-04 18:05:26 +00001164
Owen Anderson5fba6c12007-05-29 21:53:49 +00001165 iterations++;
1166 }
1167
Owen Anderson191eb062007-06-25 05:41:12 +00001168 DOUT << "ITERATIONS: " << iterations << "\n";
Owen Anderson06c1e582007-06-20 22:10:02 +00001169}
1170
Owen Andersonfd5683a2007-06-21 00:19:05 +00001171/// insertion_pre - When a partial redundancy has been identified, eliminate it
1172/// by inserting appropriate values into the predecessors and a phi node in
1173/// the main block
Owen Anderson06c1e582007-06-20 22:10:02 +00001174void GVNPRE::insertion_pre(Value* e, BasicBlock* BB,
1175 std::map<BasicBlock*, Value*>& avail,
Owen Anderson2ff912b2007-06-21 17:57:53 +00001176 SmallPtrSet<Value*, 32>& new_set) {
Owen Anderson06c1e582007-06-20 22:10:02 +00001177 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE; ++PI) {
1178 Value* e2 = avail[*PI];
1179 if (!find_leader(availableOut[*PI], VN.lookup(e2))) {
1180 User* U = cast<User>(e2);
1181
1182 Value* s1 = 0;
1183 if (isa<BinaryOperator>(U->getOperand(0)) ||
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001184 isa<CmpInst>(U->getOperand(0)) ||
1185 isa<ShuffleVectorInst>(U->getOperand(0)) ||
1186 isa<ExtractElementInst>(U->getOperand(0)) ||
1187 isa<InsertElementInst>(U->getOperand(0)))
Owen Anderson06c1e582007-06-20 22:10:02 +00001188 s1 = find_leader(availableOut[*PI], VN.lookup(U->getOperand(0)));
1189 else
1190 s1 = U->getOperand(0);
1191
1192 Value* s2 = 0;
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001193 if (isa<BinaryOperator>(U->getOperand(1)) ||
1194 isa<CmpInst>(U->getOperand(1)) ||
1195 isa<ShuffleVectorInst>(U->getOperand(1)) ||
1196 isa<ExtractElementInst>(U->getOperand(1)) ||
1197 isa<InsertElementInst>(U->getOperand(1)))
Owen Anderson06c1e582007-06-20 22:10:02 +00001198 s2 = find_leader(availableOut[*PI], VN.lookup(U->getOperand(1)));
1199 else
1200 s2 = U->getOperand(1);
Owen Anderson0eb26572007-06-27 17:03:03 +00001201
1202 // Ternary Operators
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001203 Value* s3 = 0;
1204 if (isa<ShuffleVectorInst>(U) ||
1205 isa<InsertElementInst>(U))
1206 if (isa<BinaryOperator>(U->getOperand(2)) ||
1207 isa<CmpInst>(U->getOperand(2)) ||
1208 isa<ShuffleVectorInst>(U->getOperand(2)) ||
1209 isa<ExtractElementInst>(U->getOperand(2)) ||
1210 isa<InsertElementInst>(U->getOperand(2)))
1211 s3 = find_leader(availableOut[*PI], VN.lookup(U->getOperand(2)));
1212 else
1213 s3 = U->getOperand(2);
Owen Anderson06c1e582007-06-20 22:10:02 +00001214
1215 Value* newVal = 0;
1216 if (BinaryOperator* BO = dyn_cast<BinaryOperator>(U))
1217 newVal = BinaryOperator::create(BO->getOpcode(), s1, s2,
1218 BO->getName()+".gvnpre",
1219 (*PI)->getTerminator());
1220 else if (CmpInst* C = dyn_cast<CmpInst>(U))
1221 newVal = CmpInst::create(C->getOpcode(), C->getPredicate(), s1, s2,
1222 C->getName()+".gvnpre",
1223 (*PI)->getTerminator());
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001224 else if (ShuffleVectorInst* S = dyn_cast<ShuffleVectorInst>(U))
1225 newVal = new ShuffleVectorInst(s1, s2, s3, S->getName()+".gvnpre",
1226 (*PI)->getTerminator());
1227 else if (InsertElementInst* S = dyn_cast<InsertElementInst>(U))
1228 newVal = new InsertElementInst(s1, s2, s3, S->getName()+".gvnpre",
1229 (*PI)->getTerminator());
1230 else if (ExtractElementInst* S = dyn_cast<ExtractElementInst>(U))
1231 newVal = new ExtractElementInst(s1, s2, S->getName()+".gvnpre",
1232 (*PI)->getTerminator());
Owen Anderson06c1e582007-06-20 22:10:02 +00001233
1234 VN.add(newVal, VN.lookup(U));
1235
Owen Anderson2ff912b2007-06-21 17:57:53 +00001236 SmallPtrSet<Value*, 32>& predAvail = availableOut[*PI];
Owen Anderson06c1e582007-06-20 22:10:02 +00001237 val_replace(predAvail, newVal);
Owen Andersonfd5683a2007-06-21 00:19:05 +00001238
Owen Anderson06c1e582007-06-20 22:10:02 +00001239 std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
1240 if (av != avail.end())
1241 avail.erase(av);
1242 avail.insert(std::make_pair(*PI, newVal));
1243
1244 ++NumInsertedVals;
1245 }
1246 }
1247
1248 PHINode* p = 0;
1249
1250 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE; ++PI) {
1251 if (p == 0)
1252 p = new PHINode(avail[*PI]->getType(), "gvnpre-join", BB->begin());
1253
1254 p->addIncoming(avail[*PI], *PI);
1255 }
1256
1257 VN.add(p, VN.lookup(e));
Owen Anderson06c1e582007-06-20 22:10:02 +00001258 val_replace(availableOut[BB], p);
Owen Anderson06c1e582007-06-20 22:10:02 +00001259 new_set.insert(p);
1260
1261 ++NumInsertedPhis;
1262}
1263
Owen Andersonfd5683a2007-06-21 00:19:05 +00001264/// insertion_mergepoint - When walking the dom tree, check at each merge
1265/// block for the possibility of a partial redundancy. If present, eliminate it
Owen Anderson06c1e582007-06-20 22:10:02 +00001266unsigned GVNPRE::insertion_mergepoint(std::vector<Value*>& workList,
Owen Anderson28a2d442007-06-22 00:20:30 +00001267 df_iterator<DomTreeNode*>& D,
Owen Anderson2ff912b2007-06-21 17:57:53 +00001268 SmallPtrSet<Value*, 32>& new_set) {
Owen Anderson06c1e582007-06-20 22:10:02 +00001269 bool changed_function = false;
1270 bool new_stuff = false;
1271
1272 BasicBlock* BB = D->getBlock();
1273 for (unsigned i = 0; i < workList.size(); ++i) {
1274 Value* e = workList[i];
1275
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001276 if (isa<BinaryOperator>(e) || isa<CmpInst>(e) ||
1277 isa<ExtractElementInst>(e) || isa<InsertElementInst>(e) ||
1278 isa<ShuffleVectorInst>(e)) {
Owen Anderson06c1e582007-06-20 22:10:02 +00001279 if (find_leader(availableOut[D->getIDom()->getBlock()],
1280 VN.lookup(e)) != 0)
1281 continue;
1282
1283 std::map<BasicBlock*, Value*> avail;
1284 bool by_some = false;
1285 int num_avail = 0;
1286
1287 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE;
1288 ++PI) {
1289 Value *e2 = phi_translate(e, *PI, BB);
1290 Value *e3 = find_leader(availableOut[*PI], VN.lookup(e2));
1291
1292 if (e3 == 0) {
1293 std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
1294 if (av != avail.end())
1295 avail.erase(av);
1296 avail.insert(std::make_pair(*PI, e2));
1297 } else {
1298 std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
1299 if (av != avail.end())
1300 avail.erase(av);
1301 avail.insert(std::make_pair(*PI, e3));
1302
1303 by_some = true;
1304 num_avail++;
1305 }
1306 }
1307
1308 if (by_some && num_avail < std::distance(pred_begin(BB), pred_end(BB))) {
1309 insertion_pre(e, BB, avail, new_set);
1310
1311 changed_function = true;
1312 new_stuff = true;
1313 }
1314 }
1315 }
1316
1317 unsigned retval = 0;
1318 if (changed_function)
1319 retval += 1;
1320 if (new_stuff)
1321 retval += 2;
1322
1323 return retval;
1324}
1325
Owen Andersonfd5683a2007-06-21 00:19:05 +00001326/// insert - Phase 2 of the main algorithm. Walk the dominator tree looking for
1327/// merge points. When one is found, check for a partial redundancy. If one is
1328/// present, eliminate it. Repeat this walk until no changes are made.
Owen Anderson06c1e582007-06-20 22:10:02 +00001329bool GVNPRE::insertion(Function& F) {
1330 bool changed_function = false;
1331
1332 DominatorTree &DT = getAnalysis<DominatorTree>();
Owen Andersonbe802402007-06-08 01:03:01 +00001333
Owen Anderson2ff912b2007-06-21 17:57:53 +00001334 std::map<BasicBlock*, SmallPtrSet<Value*, 32> > new_sets;
Owen Andersonbe802402007-06-08 01:03:01 +00001335 bool new_stuff = true;
1336 while (new_stuff) {
1337 new_stuff = false;
Owen Andersonbe802402007-06-08 01:03:01 +00001338 for (df_iterator<DomTreeNode*> DI = df_begin(DT.getRootNode()),
1339 E = df_end(DT.getRootNode()); DI != E; ++DI) {
1340 BasicBlock* BB = DI->getBlock();
1341
Owen Andersond184c182007-06-11 16:25:17 +00001342 if (BB == 0)
1343 continue;
1344
Owen Anderson2ff912b2007-06-21 17:57:53 +00001345 SmallPtrSet<Value*, 32>& new_set = new_sets[BB];
1346 SmallPtrSet<Value*, 32>& availOut = availableOut[BB];
1347 SmallPtrSet<Value*, 32>& anticIn = anticipatedIn[BB];
Owen Andersonbe802402007-06-08 01:03:01 +00001348
Owen Anderson55994f22007-06-08 20:44:02 +00001349 new_set.clear();
1350
Owen Andersonbe802402007-06-08 01:03:01 +00001351 // Replace leaders with leaders inherited from dominator
1352 if (DI->getIDom() != 0) {
Owen Anderson2ff912b2007-06-21 17:57:53 +00001353 SmallPtrSet<Value*, 32>& dom_set = new_sets[DI->getIDom()->getBlock()];
1354 for (SmallPtrSet<Value*, 32>::iterator I = dom_set.begin(),
Owen Andersonbe802402007-06-08 01:03:01 +00001355 E = dom_set.end(); I != E; ++I) {
1356 new_set.insert(*I);
Owen Andersonb56fba02007-06-19 03:31:41 +00001357 val_replace(availOut, *I);
Owen Andersonbe802402007-06-08 01:03:01 +00001358 }
1359 }
1360
1361 // If there is more than one predecessor...
1362 if (pred_begin(BB) != pred_end(BB) && ++pred_begin(BB) != pred_end(BB)) {
1363 std::vector<Value*> workList;
Owen Andersond50a29d2007-06-22 00:43:22 +00001364 workList.reserve(anticIn.size());
Owen Andersonbe802402007-06-08 01:03:01 +00001365 topo_sort(anticIn, workList);
1366
Owen Anderson06c1e582007-06-20 22:10:02 +00001367 unsigned result = insertion_mergepoint(workList, DI, new_set);
1368 if (result & 1)
1369 changed_function = true;
1370 if (result & 2)
1371 new_stuff = true;
Owen Andersonbe802402007-06-08 01:03:01 +00001372 }
1373 }
Owen Andersonbe802402007-06-08 01:03:01 +00001374 }
Owen Anderson634a0632007-06-06 01:27:49 +00001375
Owen Anderson06c1e582007-06-20 22:10:02 +00001376 return changed_function;
1377}
1378
Owen Andersonfd5683a2007-06-21 00:19:05 +00001379// GVNPRE::runOnFunction - This is the main transformation entry point for a
1380// function.
1381//
Owen Anderson06c1e582007-06-20 22:10:02 +00001382bool GVNPRE::runOnFunction(Function &F) {
Owen Andersonfd5683a2007-06-21 00:19:05 +00001383 // Clean out global sets from any previous functions
Owen Anderson06c1e582007-06-20 22:10:02 +00001384 VN.clear();
1385 createdExpressions.clear();
1386 availableOut.clear();
1387 anticipatedIn.clear();
1388
1389 bool changed_function = false;
1390
1391 // Phase 1: BuildSets
Owen Andersonfd5683a2007-06-21 00:19:05 +00001392 // This phase calculates the AVAIL_OUT and ANTIC_IN sets
Owen Anderson5477c542007-06-26 23:29:41 +00001393 buildsets(F);
Owen Anderson06c1e582007-06-20 22:10:02 +00001394
1395 // Phase 2: Insert
Owen Andersonfd5683a2007-06-21 00:19:05 +00001396 // This phase inserts values to make partially redundant values
1397 // fully redundant
Owen Anderson06c1e582007-06-20 22:10:02 +00001398 changed_function |= insertion(F);
1399
Owen Anderson634a0632007-06-06 01:27:49 +00001400 // Phase 3: Eliminate
Owen Andersonfd5683a2007-06-21 00:19:05 +00001401 // This phase performs trivial full redundancy elimination
Owen Anderson06c1e582007-06-20 22:10:02 +00001402 changed_function |= elimination();
Owen Anderson55994f22007-06-08 20:44:02 +00001403
Owen Andersonbe802402007-06-08 01:03:01 +00001404 // Phase 4: Cleanup
Owen Andersonfd5683a2007-06-21 00:19:05 +00001405 // This phase cleans up values that were created solely
1406 // as leaders for expressions
Owen Anderson42769842007-06-12 16:57:50 +00001407 cleanup();
Owen Andersonbe802402007-06-08 01:03:01 +00001408
Owen Andersonb0714bb2007-06-20 18:30:20 +00001409 return changed_function;
Owen Anderson5fba6c12007-05-29 21:53:49 +00001410}