blob: 007c6777195f6480b8d45b55f93673d74c3efbeb [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"
Owen Andersonc738f7c2007-06-29 00:40:05 +000026#include "llvm/DerivedTypes.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000027#include "llvm/Analysis/Dominators.h"
Owen Anderson28a2d442007-06-22 00:20:30 +000028#include "llvm/ADT/BitVector.h"
Owen Anderson1ad2c102007-06-19 23:23:54 +000029#include "llvm/ADT/DenseMap.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000030#include "llvm/ADT/DepthFirstIterator.h"
Owen Anderson43ca4b42007-06-25 18:25:31 +000031#include "llvm/ADT/PostOrderIterator.h"
Owen Anderson2ff912b2007-06-21 17:57:53 +000032#include "llvm/ADT/SmallPtrSet.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000033#include "llvm/ADT/Statistic.h"
Owen Anderson02e96982007-07-05 23:11:26 +000034#include "llvm/Transforms/Utils/UnifyFunctionExitNodes.h"
Owen Andersonbe802402007-06-08 01:03:01 +000035#include "llvm/Support/CFG.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000036#include "llvm/Support/Compiler.h"
Owen Anderson4a6ec8f2007-05-29 23:15:21 +000037#include "llvm/Support/Debug.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000038#include <algorithm>
Owen Anderson331bf6a2007-05-31 22:44:11 +000039#include <deque>
Owen Anderson5fba6c12007-05-29 21:53:49 +000040#include <map>
Owen Anderson331bf6a2007-05-31 22:44:11 +000041#include <vector>
Owen Anderson5fba6c12007-05-29 21:53:49 +000042#include <set>
Owen Anderson5fba6c12007-05-29 21:53:49 +000043using namespace llvm;
44
Owen Andersonb56fba02007-06-19 03:31:41 +000045//===----------------------------------------------------------------------===//
46// ValueTable Class
47//===----------------------------------------------------------------------===//
48
Owen Andersonfd5683a2007-06-21 00:19:05 +000049/// This class holds the mapping between values and value numbers. It is used
50/// as an efficient mechanism to determine the expression-wise equivalence of
51/// two values.
Owen Andersonb56fba02007-06-19 03:31:41 +000052
53namespace {
54 class VISIBILITY_HIDDEN ValueTable {
55 public:
56 struct Expression {
57 enum ExpressionOpcode { ADD, SUB, MUL, UDIV, SDIV, FDIV, UREM, SREM,
58 FREM, SHL, LSHR, ASHR, AND, OR, XOR, ICMPEQ,
59 ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
60 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
61 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
62 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
Owen Andersonb6a39fc2007-06-27 04:10:46 +000063 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
Owen Anderson67799d42007-06-29 00:51:03 +000064 SHUFFLE, SELECT, TRUNC, ZEXT, SEXT, FPTOUI,
65 FPTOSI, UITOFP, SITOFP, FPTRUNC, FPEXT,
Owen Andersonb9a494a2007-07-03 22:50:56 +000066 PTRTOINT, INTTOPTR, BITCAST, GEP};
Owen Andersonb56fba02007-06-19 03:31:41 +000067
68 ExpressionOpcode opcode;
Owen Andersonc738f7c2007-06-29 00:40:05 +000069 const Type* type;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000070 uint32_t firstVN;
71 uint32_t secondVN;
72 uint32_t thirdVN;
Owen Andersonb9a494a2007-07-03 22:50:56 +000073 std::vector<uint32_t> varargs;
Owen Andersonb56fba02007-06-19 03:31:41 +000074
75 bool operator< (const Expression& other) const {
76 if (opcode < other.opcode)
77 return true;
78 else if (opcode > other.opcode)
79 return false;
Owen Andersonc738f7c2007-06-29 00:40:05 +000080 else if (type < other.type)
81 return true;
82 else if (type > other.type)
83 return false;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000084 else if (firstVN < other.firstVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000085 return true;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000086 else if (firstVN > other.firstVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000087 return false;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000088 else if (secondVN < other.secondVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000089 return true;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000090 else if (secondVN > other.secondVN)
91 return false;
92 else if (thirdVN < other.thirdVN)
93 return true;
94 else if (thirdVN > other.thirdVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000095 return false;
Owen Andersonb9a494a2007-07-03 22:50:56 +000096 else {
97 if (varargs.size() < other.varargs.size())
98 return true;
99 else if (varargs.size() > other.varargs.size())
100 return false;
101
102 for (size_t i = 0; i < varargs.size(); ++i)
103 if (varargs[i] < other.varargs[i])
104 return true;
105 else if (varargs[i] > other.varargs[i])
106 return false;
107
Owen Andersonb56fba02007-06-19 03:31:41 +0000108 return false;
Owen Andersonb9a494a2007-07-03 22:50:56 +0000109 }
Owen Andersona75dd4d2007-06-12 00:50:47 +0000110 }
Owen Andersonb56fba02007-06-19 03:31:41 +0000111 };
112
113 private:
Owen Anderson1ad2c102007-06-19 23:23:54 +0000114 DenseMap<Value*, uint32_t> valueNumbering;
Owen Andersonb56fba02007-06-19 03:31:41 +0000115 std::map<Expression, uint32_t> expressionNumbering;
116
Owen Andersonb56fba02007-06-19 03:31:41 +0000117 uint32_t nextValueNumber;
118
119 Expression::ExpressionOpcode getOpcode(BinaryOperator* BO);
120 Expression::ExpressionOpcode getOpcode(CmpInst* C);
Owen Anderson67799d42007-06-29 00:51:03 +0000121 Expression::ExpressionOpcode getOpcode(CastInst* C);
Owen Andersonfd5683a2007-06-21 00:19:05 +0000122 Expression create_expression(BinaryOperator* BO);
123 Expression create_expression(CmpInst* C);
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000124 Expression create_expression(ShuffleVectorInst* V);
125 Expression create_expression(ExtractElementInst* C);
126 Expression create_expression(InsertElementInst* V);
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000127 Expression create_expression(SelectInst* V);
Owen Anderson67799d42007-06-29 00:51:03 +0000128 Expression create_expression(CastInst* C);
Owen Andersonb9a494a2007-07-03 22:50:56 +0000129 Expression create_expression(GetElementPtrInst* G);
Owen Andersonb56fba02007-06-19 03:31:41 +0000130 public:
131 ValueTable() { nextValueNumber = 1; }
132 uint32_t lookup_or_add(Value* V);
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000133 uint32_t lookup(Value* V) const;
Owen Andersonb56fba02007-06-19 03:31:41 +0000134 void add(Value* V, uint32_t num);
135 void clear();
Owen Anderson91c54952007-06-19 05:37:32 +0000136 void erase(Value* v);
Owen Anderson28a2d442007-06-22 00:20:30 +0000137 unsigned size();
Owen Andersonb56fba02007-06-19 03:31:41 +0000138 };
139}
140
Owen Andersonfd5683a2007-06-21 00:19:05 +0000141//===----------------------------------------------------------------------===//
142// ValueTable Internal Functions
143//===----------------------------------------------------------------------===//
Owen Andersonb56fba02007-06-19 03:31:41 +0000144ValueTable::Expression::ExpressionOpcode
Owen Andersonfd5683a2007-06-21 00:19:05 +0000145 ValueTable::getOpcode(BinaryOperator* BO) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000146 switch(BO->getOpcode()) {
147 case Instruction::Add:
148 return Expression::ADD;
149 case Instruction::Sub:
150 return Expression::SUB;
151 case Instruction::Mul:
152 return Expression::MUL;
153 case Instruction::UDiv:
154 return Expression::UDIV;
155 case Instruction::SDiv:
156 return Expression::SDIV;
157 case Instruction::FDiv:
158 return Expression::FDIV;
159 case Instruction::URem:
160 return Expression::UREM;
161 case Instruction::SRem:
162 return Expression::SREM;
163 case Instruction::FRem:
164 return Expression::FREM;
165 case Instruction::Shl:
166 return Expression::SHL;
167 case Instruction::LShr:
168 return Expression::LSHR;
169 case Instruction::AShr:
170 return Expression::ASHR;
171 case Instruction::And:
172 return Expression::AND;
173 case Instruction::Or:
174 return Expression::OR;
175 case Instruction::Xor:
176 return Expression::XOR;
177
178 // THIS SHOULD NEVER HAPPEN
179 default:
180 assert(0 && "Binary operator with unknown opcode?");
181 return Expression::ADD;
182 }
183}
184
185ValueTable::Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
186 if (C->getOpcode() == Instruction::ICmp) {
187 switch (C->getPredicate()) {
188 case ICmpInst::ICMP_EQ:
189 return Expression::ICMPEQ;
190 case ICmpInst::ICMP_NE:
191 return Expression::ICMPNE;
192 case ICmpInst::ICMP_UGT:
193 return Expression::ICMPUGT;
194 case ICmpInst::ICMP_UGE:
195 return Expression::ICMPUGE;
196 case ICmpInst::ICMP_ULT:
197 return Expression::ICMPULT;
198 case ICmpInst::ICMP_ULE:
199 return Expression::ICMPULE;
200 case ICmpInst::ICMP_SGT:
201 return Expression::ICMPSGT;
202 case ICmpInst::ICMP_SGE:
203 return Expression::ICMPSGE;
204 case ICmpInst::ICMP_SLT:
205 return Expression::ICMPSLT;
206 case ICmpInst::ICMP_SLE:
207 return Expression::ICMPSLE;
208
209 // THIS SHOULD NEVER HAPPEN
210 default:
211 assert(0 && "Comparison with unknown predicate?");
212 return Expression::ICMPEQ;
213 }
214 } else {
215 switch (C->getPredicate()) {
216 case FCmpInst::FCMP_OEQ:
217 return Expression::FCMPOEQ;
218 case FCmpInst::FCMP_OGT:
219 return Expression::FCMPOGT;
220 case FCmpInst::FCMP_OGE:
221 return Expression::FCMPOGE;
222 case FCmpInst::FCMP_OLT:
223 return Expression::FCMPOLT;
224 case FCmpInst::FCMP_OLE:
225 return Expression::FCMPOLE;
226 case FCmpInst::FCMP_ONE:
227 return Expression::FCMPONE;
228 case FCmpInst::FCMP_ORD:
229 return Expression::FCMPORD;
230 case FCmpInst::FCMP_UNO:
231 return Expression::FCMPUNO;
232 case FCmpInst::FCMP_UEQ:
233 return Expression::FCMPUEQ;
234 case FCmpInst::FCMP_UGT:
235 return Expression::FCMPUGT;
236 case FCmpInst::FCMP_UGE:
237 return Expression::FCMPUGE;
238 case FCmpInst::FCMP_ULT:
239 return Expression::FCMPULT;
240 case FCmpInst::FCMP_ULE:
241 return Expression::FCMPULE;
242 case FCmpInst::FCMP_UNE:
243 return Expression::FCMPUNE;
244
245 // THIS SHOULD NEVER HAPPEN
246 default:
247 assert(0 && "Comparison with unknown predicate?");
248 return Expression::FCMPOEQ;
Owen Anderson223718c2007-06-09 18:35:31 +0000249 }
250 }
Owen Andersonb56fba02007-06-19 03:31:41 +0000251}
252
Owen Anderson67799d42007-06-29 00:51:03 +0000253ValueTable::Expression::ExpressionOpcode
254 ValueTable::getOpcode(CastInst* C) {
255 switch(C->getOpcode()) {
256 case Instruction::Trunc:
257 return Expression::TRUNC;
258 case Instruction::ZExt:
259 return Expression::ZEXT;
260 case Instruction::SExt:
261 return Expression::SEXT;
262 case Instruction::FPToUI:
263 return Expression::FPTOUI;
264 case Instruction::FPToSI:
265 return Expression::FPTOSI;
266 case Instruction::UIToFP:
267 return Expression::UITOFP;
268 case Instruction::SIToFP:
269 return Expression::SITOFP;
270 case Instruction::FPTrunc:
271 return Expression::FPTRUNC;
272 case Instruction::FPExt:
273 return Expression::FPEXT;
274 case Instruction::PtrToInt:
275 return Expression::PTRTOINT;
276 case Instruction::IntToPtr:
277 return Expression::INTTOPTR;
278 case Instruction::BitCast:
279 return Expression::BITCAST;
280
281 // THIS SHOULD NEVER HAPPEN
282 default:
283 assert(0 && "Cast operator with unknown opcode?");
284 return Expression::BITCAST;
285 }
286}
287
Owen Andersonfd5683a2007-06-21 00:19:05 +0000288ValueTable::Expression ValueTable::create_expression(BinaryOperator* BO) {
289 Expression e;
290
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000291 e.firstVN = lookup_or_add(BO->getOperand(0));
292 e.secondVN = lookup_or_add(BO->getOperand(1));
293 e.thirdVN = 0;
Owen Andersonc738f7c2007-06-29 00:40:05 +0000294 e.type = BO->getType();
Owen Andersonfd5683a2007-06-21 00:19:05 +0000295 e.opcode = getOpcode(BO);
296
Owen Andersonfd5683a2007-06-21 00:19:05 +0000297 return e;
298}
299
300ValueTable::Expression ValueTable::create_expression(CmpInst* C) {
301 Expression e;
302
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000303 e.firstVN = lookup_or_add(C->getOperand(0));
304 e.secondVN = lookup_or_add(C->getOperand(1));
305 e.thirdVN = 0;
Owen Andersonc738f7c2007-06-29 00:40:05 +0000306 e.type = C->getType();
Owen Andersonfd5683a2007-06-21 00:19:05 +0000307 e.opcode = getOpcode(C);
308
Owen Andersonfd5683a2007-06-21 00:19:05 +0000309 return e;
310}
311
Owen Anderson67799d42007-06-29 00:51:03 +0000312ValueTable::Expression ValueTable::create_expression(CastInst* C) {
313 Expression e;
314
315 e.firstVN = lookup_or_add(C->getOperand(0));
316 e.secondVN = 0;
317 e.thirdVN = 0;
318 e.type = C->getType();
319 e.opcode = getOpcode(C);
320
321 return e;
322}
323
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000324ValueTable::Expression ValueTable::create_expression(ShuffleVectorInst* S) {
325 Expression e;
326
327 e.firstVN = lookup_or_add(S->getOperand(0));
328 e.secondVN = lookup_or_add(S->getOperand(1));
329 e.thirdVN = lookup_or_add(S->getOperand(2));
Owen Andersonc738f7c2007-06-29 00:40:05 +0000330 e.type = S->getType();
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000331 e.opcode = Expression::SHUFFLE;
332
333 return e;
334}
335
336ValueTable::Expression ValueTable::create_expression(ExtractElementInst* E) {
337 Expression e;
338
339 e.firstVN = lookup_or_add(E->getOperand(0));
340 e.secondVN = lookup_or_add(E->getOperand(1));
341 e.thirdVN = 0;
Owen Andersonc738f7c2007-06-29 00:40:05 +0000342 e.type = E->getType();
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000343 e.opcode = Expression::EXTRACT;
344
345 return e;
346}
347
348ValueTable::Expression ValueTable::create_expression(InsertElementInst* I) {
349 Expression e;
350
351 e.firstVN = lookup_or_add(I->getOperand(0));
352 e.secondVN = lookup_or_add(I->getOperand(1));
353 e.thirdVN = lookup_or_add(I->getOperand(2));
Owen Andersonc738f7c2007-06-29 00:40:05 +0000354 e.type = I->getType();
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000355 e.opcode = Expression::INSERT;
356
357 return e;
358}
359
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000360ValueTable::Expression ValueTable::create_expression(SelectInst* I) {
361 Expression e;
362
363 e.firstVN = lookup_or_add(I->getCondition());
364 e.secondVN = lookup_or_add(I->getTrueValue());
365 e.thirdVN = lookup_or_add(I->getFalseValue());
Owen Andersonc738f7c2007-06-29 00:40:05 +0000366 e.type = I->getType();
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000367 e.opcode = Expression::SELECT;
368
369 return e;
370}
371
Owen Andersonb9a494a2007-07-03 22:50:56 +0000372ValueTable::Expression ValueTable::create_expression(GetElementPtrInst* G) {
373 Expression e;
374
375 e.firstVN = lookup_or_add(G->getPointerOperand());
376 e.secondVN = 0;
377 e.thirdVN = 0;
378 e.type = G->getType();
379 e.opcode = Expression::SELECT;
380
381 for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
382 I != E; ++I)
383 e.varargs.push_back(lookup_or_add(*I));
384
385 return e;
386}
387
Owen Andersonfd5683a2007-06-21 00:19:05 +0000388//===----------------------------------------------------------------------===//
389// ValueTable External Functions
390//===----------------------------------------------------------------------===//
391
392/// lookup_or_add - Returns the value number for the specified value, assigning
393/// it a new number if it did not have one before.
Owen Andersonb56fba02007-06-19 03:31:41 +0000394uint32_t ValueTable::lookup_or_add(Value* V) {
Owen Anderson1ad2c102007-06-19 23:23:54 +0000395 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
Owen Andersonb56fba02007-06-19 03:31:41 +0000396 if (VI != valueNumbering.end())
397 return VI->second;
Owen Anderson223718c2007-06-09 18:35:31 +0000398
Owen Anderson223718c2007-06-09 18:35:31 +0000399
Owen Andersonb56fba02007-06-19 03:31:41 +0000400 if (BinaryOperator* BO = dyn_cast<BinaryOperator>(V)) {
401 Expression e = create_expression(BO);
402
403 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
404 if (EI != expressionNumbering.end()) {
405 valueNumbering.insert(std::make_pair(V, EI->second));
406 return EI->second;
407 } else {
408 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
409 valueNumbering.insert(std::make_pair(V, nextValueNumber));
410
411 return nextValueNumber++;
412 }
413 } else if (CmpInst* C = dyn_cast<CmpInst>(V)) {
414 Expression e = create_expression(C);
415
416 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
417 if (EI != expressionNumbering.end()) {
418 valueNumbering.insert(std::make_pair(V, EI->second));
419 return EI->second;
420 } else {
421 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
422 valueNumbering.insert(std::make_pair(V, nextValueNumber));
423
424 return nextValueNumber++;
425 }
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000426 } else if (ShuffleVectorInst* U = dyn_cast<ShuffleVectorInst>(V)) {
427 Expression e = create_expression(U);
428
429 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
430 if (EI != expressionNumbering.end()) {
431 valueNumbering.insert(std::make_pair(V, EI->second));
432 return EI->second;
433 } else {
434 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
435 valueNumbering.insert(std::make_pair(V, nextValueNumber));
436
437 return nextValueNumber++;
438 }
439 } else if (ExtractElementInst* U = dyn_cast<ExtractElementInst>(V)) {
440 Expression e = create_expression(U);
441
442 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
443 if (EI != expressionNumbering.end()) {
444 valueNumbering.insert(std::make_pair(V, EI->second));
445 return EI->second;
446 } else {
447 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
448 valueNumbering.insert(std::make_pair(V, nextValueNumber));
449
450 return nextValueNumber++;
451 }
452 } else if (InsertElementInst* U = dyn_cast<InsertElementInst>(V)) {
453 Expression e = create_expression(U);
454
455 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
456 if (EI != expressionNumbering.end()) {
457 valueNumbering.insert(std::make_pair(V, EI->second));
458 return EI->second;
459 } else {
460 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
461 valueNumbering.insert(std::make_pair(V, nextValueNumber));
462
463 return nextValueNumber++;
464 }
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000465 } else if (SelectInst* U = dyn_cast<SelectInst>(V)) {
466 Expression e = create_expression(U);
467
468 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
469 if (EI != expressionNumbering.end()) {
470 valueNumbering.insert(std::make_pair(V, EI->second));
471 return EI->second;
472 } else {
473 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
474 valueNumbering.insert(std::make_pair(V, nextValueNumber));
475
476 return nextValueNumber++;
477 }
Owen Anderson67799d42007-06-29 00:51:03 +0000478 } else if (CastInst* U = dyn_cast<CastInst>(V)) {
479 Expression e = create_expression(U);
480
481 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
482 if (EI != expressionNumbering.end()) {
483 valueNumbering.insert(std::make_pair(V, EI->second));
484 return EI->second;
485 } else {
486 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
487 valueNumbering.insert(std::make_pair(V, nextValueNumber));
488
489 return nextValueNumber++;
490 }
Owen Anderson2e4b6fea2007-07-03 23:51:19 +0000491 } else if (GetElementPtrInst* U = dyn_cast<GetElementPtrInst>(V)) {
492 Expression e = create_expression(U);
493
494 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
495 if (EI != expressionNumbering.end()) {
496 valueNumbering.insert(std::make_pair(V, EI->second));
497 return EI->second;
498 } else {
499 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
500 valueNumbering.insert(std::make_pair(V, nextValueNumber));
501
502 return nextValueNumber++;
503 }
Owen Andersonb56fba02007-06-19 03:31:41 +0000504 } else {
505 valueNumbering.insert(std::make_pair(V, nextValueNumber));
506 return nextValueNumber++;
Owen Anderson0b68cda2007-06-03 05:55:58 +0000507 }
Owen Andersonb56fba02007-06-19 03:31:41 +0000508}
509
Owen Andersonfd5683a2007-06-21 00:19:05 +0000510/// lookup - Returns the value number of the specified value. Fails if
511/// the value has not yet been numbered.
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000512uint32_t ValueTable::lookup(Value* V) const {
Owen Anderson1ad2c102007-06-19 23:23:54 +0000513 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
Owen Andersonb56fba02007-06-19 03:31:41 +0000514 if (VI != valueNumbering.end())
515 return VI->second;
516 else
517 assert(0 && "Value not numbered?");
518
519 return 0;
520}
521
Owen Andersonfd5683a2007-06-21 00:19:05 +0000522/// add - Add the specified value with the given value number, removing
523/// its old number, if any
Owen Andersonb56fba02007-06-19 03:31:41 +0000524void ValueTable::add(Value* V, uint32_t num) {
Owen Anderson1ad2c102007-06-19 23:23:54 +0000525 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
Owen Andersonb56fba02007-06-19 03:31:41 +0000526 if (VI != valueNumbering.end())
527 valueNumbering.erase(VI);
528 valueNumbering.insert(std::make_pair(V, num));
529}
530
Owen Anderson191eb062007-06-25 05:41:12 +0000531/// clear - Remove all entries from the ValueTable
Owen Andersonb56fba02007-06-19 03:31:41 +0000532void ValueTable::clear() {
533 valueNumbering.clear();
534 expressionNumbering.clear();
535 nextValueNumber = 1;
536}
Owen Anderson0b68cda2007-06-03 05:55:58 +0000537
Owen Anderson191eb062007-06-25 05:41:12 +0000538/// erase - Remove a value from the value numbering
Owen Anderson91c54952007-06-19 05:37:32 +0000539void ValueTable::erase(Value* V) {
Owen Anderson91c54952007-06-19 05:37:32 +0000540 valueNumbering.erase(V);
Owen Anderson91c54952007-06-19 05:37:32 +0000541}
542
Owen Anderson28a2d442007-06-22 00:20:30 +0000543/// size - Return the number of assigned value numbers
544unsigned ValueTable::size() {
545 // NOTE: zero is never assigned
546 return nextValueNumber;
547}
548
Owen Andersonfd5683a2007-06-21 00:19:05 +0000549//===----------------------------------------------------------------------===//
Owen Anderson56b01eb2007-07-09 16:43:55 +0000550// ValueNumberedSet Class
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000551//===----------------------------------------------------------------------===//
552
553class ValueNumberedSet {
554 private:
555 SmallPtrSet<Value*, 8> contents;
556 BitVector numbers;
557 public:
558 ValueNumberedSet() { numbers.resize(1); }
559
560 typedef SmallPtrSet<Value*, 8>::iterator iterator;
561
562 iterator begin() { return contents.begin(); }
563 iterator end() { return contents.end(); }
564
565 bool insert(Value* v) { return contents.insert(v); }
566 void insert(iterator I, iterator E) { contents.insert(I, E); }
567 void erase(Value* v) { contents.erase(v); }
Owen Anderson41c2cab2007-07-10 00:09:25 +0000568 unsigned count(Value* v) { return contents.count(v); }
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000569 size_t size() { return contents.size(); }
570
Owen Anderson267ba452007-07-09 07:56:55 +0000571 void set(unsigned i) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000572 if (i >= numbers.size())
573 numbers.resize(i+1);
574
575 numbers.set(i);
576 }
577
Owen Andersoneffc7a72007-07-09 22:29:50 +0000578 void operator=(const ValueNumberedSet& other) {
579 contents = other.contents;
Owen Anderson267ba452007-07-09 07:56:55 +0000580 numbers = other.numbers;
581 }
582
583 void reset(unsigned i) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000584 if (i < numbers.size())
585 numbers.reset(i);
586 }
587
Owen Anderson267ba452007-07-09 07:56:55 +0000588 bool test(unsigned i) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000589 if (i >= numbers.size())
590 return false;
591
592 return numbers.test(i);
593 }
594
595 void clear() {
596 contents.clear();
597 numbers.clear();
598 }
599};
600
601//===----------------------------------------------------------------------===//
Owen Andersonfd5683a2007-06-21 00:19:05 +0000602// GVNPRE Pass
603//===----------------------------------------------------------------------===//
604
Owen Anderson5fba6c12007-05-29 21:53:49 +0000605namespace {
606
607 class VISIBILITY_HIDDEN GVNPRE : public FunctionPass {
608 bool runOnFunction(Function &F);
609 public:
610 static char ID; // Pass identification, replacement for typeid
Owen Andersonb9cbaed2007-06-19 04:32:55 +0000611 GVNPRE() : FunctionPass((intptr_t)&ID) { }
Owen Anderson5fba6c12007-05-29 21:53:49 +0000612
613 private:
Owen Andersonbe802402007-06-08 01:03:01 +0000614 ValueTable VN;
Owen Andersona75dd4d2007-06-12 00:50:47 +0000615 std::vector<Instruction*> createdExpressions;
Owen Anderson81d156e2007-05-31 00:42:15 +0000616
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000617 std::map<BasicBlock*, ValueNumberedSet> availableOut;
618 std::map<BasicBlock*, ValueNumberedSet> anticipatedIn;
619 std::map<BasicBlock*, ValueNumberedSet> generatedPhis;
Owen Anderson4036ad42007-06-12 22:43:57 +0000620
Owen Andersonfd5683a2007-06-21 00:19:05 +0000621 // This transformation requires dominator postdominator info
Owen Anderson5fba6c12007-05-29 21:53:49 +0000622 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
623 AU.setPreservesCFG();
Owen Anderson3c3dd902007-07-06 18:12:36 +0000624 AU.addRequiredID(BreakCriticalEdgesID);
Owen Anderson02e96982007-07-05 23:11:26 +0000625 AU.addRequired<UnifyFunctionExitNodes>();
Owen Anderson5fba6c12007-05-29 21:53:49 +0000626 AU.addRequired<DominatorTree>();
Owen Anderson5fba6c12007-05-29 21:53:49 +0000627 }
628
629 // Helper fuctions
630 // FIXME: eliminate or document these better
Owen Anderson267ba452007-07-09 07:56:55 +0000631 void dump(ValueNumberedSet& s) const ;
632 void clean(ValueNumberedSet& set) ;
633 Value* find_leader(ValueNumberedSet& vals, uint32_t v) ;
634 Value* phi_translate(Value* V, BasicBlock* pred, BasicBlock* succ) ;
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000635 void phi_translate_set(ValueNumberedSet& anticIn, BasicBlock* pred,
Owen Anderson267ba452007-07-09 07:56:55 +0000636 BasicBlock* succ, ValueNumberedSet& out) ;
Owen Anderson5fba6c12007-05-29 21:53:49 +0000637
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000638 void topo_sort(ValueNumberedSet& set,
Owen Anderson267ba452007-07-09 07:56:55 +0000639 std::vector<Value*>& vec) ;
Owen Anderson331bf6a2007-05-31 22:44:11 +0000640
Owen Anderson267ba452007-07-09 07:56:55 +0000641 void cleanup() ;
642 bool elimination() ;
Owen Andersondd998e12007-06-18 04:42:29 +0000643
Owen Anderson267ba452007-07-09 07:56:55 +0000644 void val_insert(ValueNumberedSet& s, Value* v) ;
645 void val_replace(ValueNumberedSet& s, Value* v) ;
646 bool dependsOnInvoke(Value* V) ;
Owen Anderson06c1e582007-06-20 22:10:02 +0000647 void buildsets_availout(BasicBlock::iterator I,
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000648 ValueNumberedSet& currAvail,
649 ValueNumberedSet& currPhis,
650 ValueNumberedSet& currExps,
Owen Anderson267ba452007-07-09 07:56:55 +0000651 SmallPtrSet<Value*, 16>& currTemps) ;
Owen Anderson28a2d442007-06-22 00:20:30 +0000652 bool buildsets_anticout(BasicBlock* BB,
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000653 ValueNumberedSet& anticOut,
Owen Anderson267ba452007-07-09 07:56:55 +0000654 std::set<BasicBlock*>& visited) ;
Owen Anderson28a2d442007-06-22 00:20:30 +0000655 unsigned buildsets_anticin(BasicBlock* BB,
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000656 ValueNumberedSet& anticOut,
657 ValueNumberedSet& currExps,
Owen Andersone02da552007-06-28 00:34:34 +0000658 SmallPtrSet<Value*, 16>& currTemps,
Owen Anderson267ba452007-07-09 07:56:55 +0000659 std::set<BasicBlock*>& visited) ;
660 void buildsets(Function& F) ;
Owen Anderson06c1e582007-06-20 22:10:02 +0000661
662 void insertion_pre(Value* e, BasicBlock* BB,
663 std::map<BasicBlock*, Value*>& avail,
Owen Anderson267ba452007-07-09 07:56:55 +0000664 std::map<BasicBlock*,ValueNumberedSet>& new_set) ;
Owen Anderson06c1e582007-06-20 22:10:02 +0000665 unsigned insertion_mergepoint(std::vector<Value*>& workList,
Owen Anderson28a2d442007-06-22 00:20:30 +0000666 df_iterator<DomTreeNode*>& D,
Owen Anderson267ba452007-07-09 07:56:55 +0000667 std::map<BasicBlock*, ValueNumberedSet>& new_set) ;
668 bool insertion(Function& F) ;
Owen Anderson5fba6c12007-05-29 21:53:49 +0000669
670 };
671
672 char GVNPRE::ID = 0;
673
674}
675
Owen Andersonfd5683a2007-06-21 00:19:05 +0000676// createGVNPREPass - The public interface to this file...
Owen Anderson5fba6c12007-05-29 21:53:49 +0000677FunctionPass *llvm::createGVNPREPass() { return new GVNPRE(); }
678
679RegisterPass<GVNPRE> X("gvnpre",
680 "Global Value Numbering/Partial Redundancy Elimination");
681
Owen Anderson5fba6c12007-05-29 21:53:49 +0000682
Owen Anderson7d76b2a2007-06-08 22:02:36 +0000683STATISTIC(NumInsertedVals, "Number of values inserted");
684STATISTIC(NumInsertedPhis, "Number of PHI nodes inserted");
685STATISTIC(NumEliminated, "Number of redundant instructions eliminated");
686
Owen Andersonfd5683a2007-06-21 00:19:05 +0000687/// find_leader - Given a set and a value number, return the first
688/// element of the set with that value number, or 0 if no such element
689/// is present
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000690Value* GVNPRE::find_leader(ValueNumberedSet& vals, uint32_t v) {
691 if (!vals.test(v))
692 return 0;
693
694 for (ValueNumberedSet::iterator I = vals.begin(), E = vals.end();
Owen Andersonb56fba02007-06-19 03:31:41 +0000695 I != E; ++I)
696 if (v == VN.lookup(*I))
Owen Anderson0b68cda2007-06-03 05:55:58 +0000697 return *I;
Owen Anderson5fba6c12007-05-29 21:53:49 +0000698
Owen Anderson7ee197e2007-07-09 23:57:18 +0000699 assert(0 && "No leader found, but present bit is set?");
Owen Anderson0b68cda2007-06-03 05:55:58 +0000700 return 0;
Owen Anderson5fba6c12007-05-29 21:53:49 +0000701}
702
Owen Andersonfd5683a2007-06-21 00:19:05 +0000703/// val_insert - Insert a value into a set only if there is not a value
704/// with the same value number already in the set
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000705void GVNPRE::val_insert(ValueNumberedSet& s, Value* v) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000706 uint32_t num = VN.lookup(v);
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000707 if (!s.test(num))
Owen Andersonb56fba02007-06-19 03:31:41 +0000708 s.insert(v);
709}
710
Owen Andersonfd5683a2007-06-21 00:19:05 +0000711/// val_replace - Insert a value into a set, replacing any values already in
712/// the set that have the same value number
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000713void GVNPRE::val_replace(ValueNumberedSet& s, Value* v) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000714 uint32_t num = VN.lookup(v);
715 Value* leader = find_leader(s, num);
Owen Andersoneffc7a72007-07-09 22:29:50 +0000716 if (leader != 0)
Owen Andersonb56fba02007-06-19 03:31:41 +0000717 s.erase(leader);
Owen Andersonb56fba02007-06-19 03:31:41 +0000718 s.insert(v);
Owen Andersoneffc7a72007-07-09 22:29:50 +0000719 s.set(num);
Owen Andersonb56fba02007-06-19 03:31:41 +0000720}
721
Owen Andersonfd5683a2007-06-21 00:19:05 +0000722/// phi_translate - Given a value, its parent block, and a predecessor of its
723/// parent, translate the value into legal for the predecessor block. This
724/// means translating its operands (and recursively, their operands) through
725/// any phi nodes in the parent into values available in the predecessor
Owen Anderson4036ad42007-06-12 22:43:57 +0000726Value* GVNPRE::phi_translate(Value* V, BasicBlock* pred, BasicBlock* succ) {
Owen Anderson38b6b222007-06-04 18:05:26 +0000727 if (V == 0)
728 return 0;
729
Owen Anderson59bd0532007-07-03 18:37:08 +0000730 // Unary Operations
Owen Anderson6b958c72007-07-03 19:01:42 +0000731 if (CastInst* U = dyn_cast<CastInst>(V)) {
Owen Anderson59bd0532007-07-03 18:37:08 +0000732 Value* newOp1 = 0;
733 if (isa<Instruction>(U->getOperand(0)))
734 newOp1 = phi_translate(U->getOperand(0), pred, succ);
735 else
736 newOp1 = U->getOperand(0);
737
738 if (newOp1 == 0)
739 return 0;
740
741 if (newOp1 != U->getOperand(0)) {
742 Instruction* newVal = 0;
743 if (CastInst* C = dyn_cast<CastInst>(U))
744 newVal = CastInst::create(C->getOpcode(),
745 newOp1, C->getType(),
746 C->getName()+".expr");
747
748 uint32_t v = VN.lookup_or_add(newVal);
749
750 Value* leader = find_leader(availableOut[pred], v);
751 if (leader == 0) {
752 createdExpressions.push_back(newVal);
753 return newVal;
754 } else {
755 VN.erase(newVal);
756 delete newVal;
757 return leader;
758 }
759 }
760
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000761 // Binary Operations
Owen Anderson59bd0532007-07-03 18:37:08 +0000762 } if (isa<BinaryOperator>(V) || isa<CmpInst>(V) ||
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000763 isa<ExtractElementInst>(V)) {
Owen Anderson191eb062007-06-25 05:41:12 +0000764 User* U = cast<User>(V);
765
Owen Andersonb56fba02007-06-19 03:31:41 +0000766 Value* newOp1 = 0;
Owen Anderson191eb062007-06-25 05:41:12 +0000767 if (isa<Instruction>(U->getOperand(0)))
768 newOp1 = phi_translate(U->getOperand(0), pred, succ);
Owen Andersonb56fba02007-06-19 03:31:41 +0000769 else
Owen Anderson191eb062007-06-25 05:41:12 +0000770 newOp1 = U->getOperand(0);
Owen Andersonb56fba02007-06-19 03:31:41 +0000771
Owen Anderson38b6b222007-06-04 18:05:26 +0000772 if (newOp1 == 0)
773 return 0;
774
Owen Andersonb56fba02007-06-19 03:31:41 +0000775 Value* newOp2 = 0;
Owen Anderson191eb062007-06-25 05:41:12 +0000776 if (isa<Instruction>(U->getOperand(1)))
777 newOp2 = phi_translate(U->getOperand(1), pred, succ);
Owen Andersonb56fba02007-06-19 03:31:41 +0000778 else
Owen Anderson191eb062007-06-25 05:41:12 +0000779 newOp2 = U->getOperand(1);
Owen Andersonb56fba02007-06-19 03:31:41 +0000780
Owen Anderson38b6b222007-06-04 18:05:26 +0000781 if (newOp2 == 0)
782 return 0;
783
Owen Anderson191eb062007-06-25 05:41:12 +0000784 if (newOp1 != U->getOperand(0) || newOp2 != U->getOperand(1)) {
785 Instruction* newVal = 0;
786 if (BinaryOperator* BO = dyn_cast<BinaryOperator>(U))
787 newVal = BinaryOperator::create(BO->getOpcode(),
788 newOp1, newOp2,
789 BO->getName()+".expr");
790 else if (CmpInst* C = dyn_cast<CmpInst>(U))
791 newVal = CmpInst::create(C->getOpcode(),
792 C->getPredicate(),
793 newOp1, newOp2,
794 C->getName()+".expr");
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000795 else if (ExtractElementInst* E = dyn_cast<ExtractElementInst>(U))
796 newVal = new ExtractElementInst(newOp1, newOp2, E->getName()+".expr");
Owen Anderson55994f22007-06-08 20:44:02 +0000797
Owen Andersonb56fba02007-06-19 03:31:41 +0000798 uint32_t v = VN.lookup_or_add(newVal);
Owen Anderson4036ad42007-06-12 22:43:57 +0000799
Owen Andersonb56fba02007-06-19 03:31:41 +0000800 Value* leader = find_leader(availableOut[pred], v);
Owen Anderson4036ad42007-06-12 22:43:57 +0000801 if (leader == 0) {
Owen Andersona75dd4d2007-06-12 00:50:47 +0000802 createdExpressions.push_back(newVal);
Owen Anderson38b6b222007-06-04 18:05:26 +0000803 return newVal;
Owen Andersonc8472092007-06-05 22:11:49 +0000804 } else {
Owen Anderson91c54952007-06-19 05:37:32 +0000805 VN.erase(newVal);
Owen Andersonc8472092007-06-05 22:11:49 +0000806 delete newVal;
Owen Anderson4036ad42007-06-12 22:43:57 +0000807 return leader;
Owen Andersonc8472092007-06-05 22:11:49 +0000808 }
Owen Anderson38b6b222007-06-04 18:05:26 +0000809 }
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000810
811 // Ternary Operations
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000812 } else if (isa<ShuffleVectorInst>(V) || isa<InsertElementInst>(V) ||
813 isa<SelectInst>(V)) {
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000814 User* U = cast<User>(V);
815
816 Value* newOp1 = 0;
817 if (isa<Instruction>(U->getOperand(0)))
818 newOp1 = phi_translate(U->getOperand(0), pred, succ);
819 else
820 newOp1 = U->getOperand(0);
821
822 if (newOp1 == 0)
823 return 0;
824
825 Value* newOp2 = 0;
826 if (isa<Instruction>(U->getOperand(1)))
827 newOp2 = phi_translate(U->getOperand(1), pred, succ);
828 else
829 newOp2 = U->getOperand(1);
830
831 if (newOp2 == 0)
832 return 0;
833
834 Value* newOp3 = 0;
835 if (isa<Instruction>(U->getOperand(2)))
836 newOp3 = phi_translate(U->getOperand(2), pred, succ);
837 else
838 newOp3 = U->getOperand(2);
839
840 if (newOp3 == 0)
841 return 0;
842
843 if (newOp1 != U->getOperand(0) ||
844 newOp2 != U->getOperand(1) ||
845 newOp3 != U->getOperand(2)) {
846 Instruction* newVal = 0;
847 if (ShuffleVectorInst* S = dyn_cast<ShuffleVectorInst>(U))
848 newVal = new ShuffleVectorInst(newOp1, newOp2, newOp3,
849 S->getName()+".expr");
850 else if (InsertElementInst* I = dyn_cast<InsertElementInst>(U))
851 newVal = new InsertElementInst(newOp1, newOp2, newOp3,
852 I->getName()+".expr");
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000853 else if (SelectInst* I = dyn_cast<SelectInst>(U))
854 newVal = new SelectInst(newOp1, newOp2, newOp3, I->getName()+".expr");
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000855
856 uint32_t v = VN.lookup_or_add(newVal);
857
858 Value* leader = find_leader(availableOut[pred], v);
859 if (leader == 0) {
860 createdExpressions.push_back(newVal);
861 return newVal;
862 } else {
863 VN.erase(newVal);
864 delete newVal;
865 return leader;
866 }
867 }
868
Owen Anderson2e4b6fea2007-07-03 23:51:19 +0000869 // Varargs operators
870 } else if (GetElementPtrInst* U = dyn_cast<GetElementPtrInst>(V)) {
871 Value* newOp1 = 0;
872 if (isa<Instruction>(U->getPointerOperand()))
873 newOp1 = phi_translate(U->getPointerOperand(), pred, succ);
874 else
875 newOp1 = U->getPointerOperand();
876
877 if (newOp1 == 0)
878 return 0;
879
880 bool changed_idx = false;
881 std::vector<Value*> newIdx;
882 for (GetElementPtrInst::op_iterator I = U->idx_begin(), E = U->idx_end();
883 I != E; ++I)
884 if (isa<Instruction>(*I)) {
885 Value* newVal = phi_translate(*I, pred, succ);
886 newIdx.push_back(newVal);
887 if (newVal != *I)
888 changed_idx = true;
889 } else {
890 newIdx.push_back(*I);
891 }
892
893 if (newOp1 != U->getPointerOperand() || changed_idx) {
Owen Anderson664e2602007-07-04 04:51:16 +0000894 Instruction* newVal = new GetElementPtrInst(newOp1,
Owen Anderson2e4b6fea2007-07-03 23:51:19 +0000895 &newIdx[0], newIdx.size(),
896 U->getName()+".expr");
897
898 uint32_t v = VN.lookup_or_add(newVal);
899
900 Value* leader = find_leader(availableOut[pred], v);
901 if (leader == 0) {
902 createdExpressions.push_back(newVal);
903 return newVal;
904 } else {
905 VN.erase(newVal);
906 delete newVal;
907 return leader;
908 }
909 }
910
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000911 // PHI Nodes
Owen Anderson38b6b222007-06-04 18:05:26 +0000912 } else if (PHINode* P = dyn_cast<PHINode>(V)) {
Owen Andersona75dd4d2007-06-12 00:50:47 +0000913 if (P->getParent() == succ)
Owen Anderson38b6b222007-06-04 18:05:26 +0000914 return P->getIncomingValueForBlock(pred);
915 }
916
917 return V;
918}
919
Owen Andersonfd5683a2007-06-21 00:19:05 +0000920/// phi_translate_set - Perform phi translation on every element of a set
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000921void GVNPRE::phi_translate_set(ValueNumberedSet& anticIn,
Owen Andersona75dd4d2007-06-12 00:50:47 +0000922 BasicBlock* pred, BasicBlock* succ,
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000923 ValueNumberedSet& out) {
924 for (ValueNumberedSet::iterator I = anticIn.begin(),
Owen Anderson38b6b222007-06-04 18:05:26 +0000925 E = anticIn.end(); I != E; ++I) {
Owen Anderson4036ad42007-06-12 22:43:57 +0000926 Value* V = phi_translate(*I, pred, succ);
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000927 if (V != 0 && !out.test(VN.lookup_or_add(V))) {
Owen Anderson38b6b222007-06-04 18:05:26 +0000928 out.insert(V);
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000929 out.set(VN.lookup(V));
930 }
Owen Anderson5fba6c12007-05-29 21:53:49 +0000931 }
932}
933
Owen Andersonfd5683a2007-06-21 00:19:05 +0000934/// dependsOnInvoke - Test if a value has an phi node as an operand, any of
935/// whose inputs is an invoke instruction. If this is true, we cannot safely
936/// PRE the instruction or anything that depends on it.
Owen Andersondd998e12007-06-18 04:42:29 +0000937bool GVNPRE::dependsOnInvoke(Value* V) {
Owen Anderson2320d432007-06-19 23:07:16 +0000938 if (PHINode* p = dyn_cast<PHINode>(V)) {
939 for (PHINode::op_iterator I = p->op_begin(), E = p->op_end(); I != E; ++I)
940 if (isa<InvokeInst>(*I))
Owen Andersondd998e12007-06-18 04:42:29 +0000941 return true;
Owen Anderson2320d432007-06-19 23:07:16 +0000942 return false;
943 } else {
944 return false;
Owen Anderson658f2c42007-06-16 00:26:54 +0000945 }
Owen Anderson658f2c42007-06-16 00:26:54 +0000946}
947
Owen Andersonfd5683a2007-06-21 00:19:05 +0000948/// clean - Remove all non-opaque values from the set whose operands are not
949/// themselves in the set, as well as all values that depend on invokes (see
950/// above)
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000951void GVNPRE::clean(ValueNumberedSet& set) {
Owen Anderson0b68cda2007-06-03 05:55:58 +0000952 std::vector<Value*> worklist;
Owen Andersond50a29d2007-06-22 00:43:22 +0000953 worklist.reserve(set.size());
Owen Andersonbe802402007-06-08 01:03:01 +0000954 topo_sort(set, worklist);
Owen Anderson5fba6c12007-05-29 21:53:49 +0000955
Owen Andersonbe802402007-06-08 01:03:01 +0000956 for (unsigned i = 0; i < worklist.size(); ++i) {
957 Value* v = worklist[i];
Owen Anderson5fba6c12007-05-29 21:53:49 +0000958
Owen Anderson59bd0532007-07-03 18:37:08 +0000959 // Handle unary ops
Owen Anderson6b958c72007-07-03 19:01:42 +0000960 if (CastInst* U = dyn_cast<CastInst>(v)) {
Owen Anderson59bd0532007-07-03 18:37:08 +0000961 bool lhsValid = !isa<Instruction>(U->getOperand(0));
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000962 lhsValid |= set.test(VN.lookup(U->getOperand(0)));
Owen Anderson59bd0532007-07-03 18:37:08 +0000963 if (lhsValid)
964 lhsValid = !dependsOnInvoke(U->getOperand(0));
965
966 if (!lhsValid) {
967 set.erase(U);
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000968 set.reset(VN.lookup(U));
Owen Anderson59bd0532007-07-03 18:37:08 +0000969 }
970
Owen Anderson0eb26572007-06-27 17:03:03 +0000971 // Handle binary ops
Owen Anderson59bd0532007-07-03 18:37:08 +0000972 } else if (isa<BinaryOperator>(v) || isa<CmpInst>(v) ||
Owen Anderson0eb26572007-06-27 17:03:03 +0000973 isa<ExtractElementInst>(v)) {
974 User* U = cast<User>(v);
975
976 bool lhsValid = !isa<Instruction>(U->getOperand(0));
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000977 lhsValid |= set.test(VN.lookup(U->getOperand(0)));
Owen Andersonb364b412007-06-18 04:30:44 +0000978 if (lhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000979 lhsValid = !dependsOnInvoke(U->getOperand(0));
Owen Anderson0b68cda2007-06-03 05:55:58 +0000980
Owen Anderson0eb26572007-06-27 17:03:03 +0000981 bool rhsValid = !isa<Instruction>(U->getOperand(1));
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000982 rhsValid |= set.test(VN.lookup(U->getOperand(1)));
Owen Andersonb364b412007-06-18 04:30:44 +0000983 if (rhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000984 rhsValid = !dependsOnInvoke(U->getOperand(1));
Owen Anderson5fba6c12007-05-29 21:53:49 +0000985
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000986 if (!lhsValid || !rhsValid) {
Owen Anderson0eb26572007-06-27 17:03:03 +0000987 set.erase(U);
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000988 set.reset(VN.lookup(U));
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000989 }
Owen Anderson223718c2007-06-09 18:35:31 +0000990
Owen Anderson0eb26572007-06-27 17:03:03 +0000991 // Handle ternary ops
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000992 } else if (isa<ShuffleVectorInst>(v) || isa<InsertElementInst>(v) ||
993 isa<SelectInst>(v)) {
Owen Anderson0eb26572007-06-27 17:03:03 +0000994 User* U = cast<User>(v);
995
996 bool lhsValid = !isa<Instruction>(U->getOperand(0));
Owen Anderson1c83b5d2007-07-09 06:50:06 +0000997 lhsValid |= set.test(VN.lookup(U->getOperand(0)));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000998 if (lhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000999 lhsValid = !dependsOnInvoke(U->getOperand(0));
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001000
Owen Anderson0eb26572007-06-27 17:03:03 +00001001 bool rhsValid = !isa<Instruction>(U->getOperand(1));
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001002 rhsValid |= set.test(VN.lookup(U->getOperand(1)));
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001003 if (rhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +00001004 rhsValid = !dependsOnInvoke(U->getOperand(1));
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001005
Owen Anderson0eb26572007-06-27 17:03:03 +00001006 bool thirdValid = !isa<Instruction>(U->getOperand(2));
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001007 thirdValid |= set.test(VN.lookup(U->getOperand(2)));
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001008 if (thirdValid)
Owen Anderson0eb26572007-06-27 17:03:03 +00001009 thirdValid = !dependsOnInvoke(U->getOperand(2));
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001010
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001011 if (!lhsValid || !rhsValid || !thirdValid) {
Owen Anderson0eb26572007-06-27 17:03:03 +00001012 set.erase(U);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001013 set.reset(VN.lookup(U));
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001014 }
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001015
1016 // Handle varargs ops
1017 } else if (GetElementPtrInst* U = dyn_cast<GetElementPtrInst>(v)) {
1018 bool ptrValid = !isa<Instruction>(U->getPointerOperand());
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001019 ptrValid |= set.test(VN.lookup(U->getPointerOperand()));
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001020 if (ptrValid)
1021 ptrValid = !dependsOnInvoke(U->getPointerOperand());
1022
1023 bool varValid = true;
1024 for (GetElementPtrInst::op_iterator I = U->idx_begin(), E = U->idx_end();
1025 I != E; ++I)
1026 if (varValid) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001027 varValid &= !isa<Instruction>(*I) || set.test(VN.lookup(*I));
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001028 varValid &= !dependsOnInvoke(*I);
1029 }
1030
1031 if (!ptrValid || !varValid) {
1032 set.erase(U);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001033 set.reset(VN.lookup(U));
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001034 }
Owen Anderson0b68cda2007-06-03 05:55:58 +00001035 }
Owen Anderson5fba6c12007-05-29 21:53:49 +00001036 }
1037}
1038
Owen Andersonfd5683a2007-06-21 00:19:05 +00001039/// topo_sort - Given a set of values, sort them by topological
1040/// order into the provided vector.
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001041void GVNPRE::topo_sort(ValueNumberedSet& set, std::vector<Value*>& vec) {
Owen Andersone02da552007-06-28 00:34:34 +00001042 SmallPtrSet<Value*, 16> visited;
Owen Anderson49409f62007-06-22 21:31:16 +00001043 std::vector<Value*> stack;
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001044 for (ValueNumberedSet::iterator I = set.begin(), E = set.end();
Owen Anderson49409f62007-06-22 21:31:16 +00001045 I != E; ++I) {
1046 if (visited.count(*I) == 0)
1047 stack.push_back(*I);
1048
1049 while (!stack.empty()) {
1050 Value* e = stack.back();
Owen Anderson59bd0532007-07-03 18:37:08 +00001051
1052 // Handle unary ops
Owen Anderson6b958c72007-07-03 19:01:42 +00001053 if (CastInst* U = dyn_cast<CastInst>(e)) {
Owen Anderson59bd0532007-07-03 18:37:08 +00001054 Value* l = find_leader(set, VN.lookup(U->getOperand(0)));
1055
1056 if (l != 0 && isa<Instruction>(l) &&
1057 visited.count(l) == 0)
1058 stack.push_back(l);
1059 else {
1060 vec.push_back(e);
1061 visited.insert(e);
1062 stack.pop_back();
1063 }
1064
Owen Anderson0eb26572007-06-27 17:03:03 +00001065 // Handle binary ops
Owen Anderson59bd0532007-07-03 18:37:08 +00001066 } else if (isa<BinaryOperator>(e) || isa<CmpInst>(e) ||
Owen Anderson0eb26572007-06-27 17:03:03 +00001067 isa<ExtractElementInst>(e)) {
1068 User* U = cast<User>(e);
1069 Value* l = find_leader(set, VN.lookup(U->getOperand(0)));
1070 Value* r = find_leader(set, VN.lookup(U->getOperand(1)));
Owen Anderson49409f62007-06-22 21:31:16 +00001071
1072 if (l != 0 && isa<Instruction>(l) &&
1073 visited.count(l) == 0)
1074 stack.push_back(l);
1075 else if (r != 0 && isa<Instruction>(r) &&
1076 visited.count(r) == 0)
1077 stack.push_back(r);
1078 else {
1079 vec.push_back(e);
1080 visited.insert(e);
1081 stack.pop_back();
1082 }
Owen Anderson0b68cda2007-06-03 05:55:58 +00001083
Owen Anderson0eb26572007-06-27 17:03:03 +00001084 // Handle ternary ops
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001085 } else if (isa<InsertElementInst>(e) || isa<ShuffleVectorInst>(e) ||
1086 isa<SelectInst>(e)) {
Owen Anderson0eb26572007-06-27 17:03:03 +00001087 User* U = cast<User>(e);
1088 Value* l = find_leader(set, VN.lookup(U->getOperand(0)));
1089 Value* r = find_leader(set, VN.lookup(U->getOperand(1)));
1090 Value* m = find_leader(set, VN.lookup(U->getOperand(2)));
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001091
1092 if (l != 0 && isa<Instruction>(l) &&
1093 visited.count(l) == 0)
1094 stack.push_back(l);
1095 else if (r != 0 && isa<Instruction>(r) &&
1096 visited.count(r) == 0)
1097 stack.push_back(r);
1098 else if (m != 0 && isa<Instruction>(m) &&
1099 visited.count(m) == 0)
Owen Andersoncd94fc92007-07-04 18:26:18 +00001100 stack.push_back(m);
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001101 else {
1102 vec.push_back(e);
1103 visited.insert(e);
1104 stack.pop_back();
1105 }
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001106
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001107 // Handle vararg ops
1108 } else if (GetElementPtrInst* U = dyn_cast<GetElementPtrInst>(e)) {
1109 Value* p = find_leader(set, VN.lookup(U->getPointerOperand()));
1110
1111 if (p != 0 && isa<Instruction>(p) &&
1112 visited.count(p) == 0)
1113 stack.push_back(p);
1114 else {
1115 bool push_va = false;
1116 for (GetElementPtrInst::op_iterator I = U->idx_begin(),
1117 E = U->idx_end(); I != E; ++I) {
1118 Value * v = find_leader(set, VN.lookup(*I));
1119 if (v != 0 && isa<Instruction>(v) && visited.count(v) == 0) {
1120 stack.push_back(v);
1121 push_va = true;
1122 }
1123 }
1124
1125 if (!push_va) {
1126 vec.push_back(e);
1127 visited.insert(e);
1128 stack.pop_back();
1129 }
1130 }
1131
Owen Anderson0eb26572007-06-27 17:03:03 +00001132 // Handle opaque ops
Owen Anderson49409f62007-06-22 21:31:16 +00001133 } else {
Owen Anderson0b68cda2007-06-03 05:55:58 +00001134 visited.insert(e);
Owen Anderson223718c2007-06-09 18:35:31 +00001135 vec.push_back(e);
Owen Anderson49409f62007-06-22 21:31:16 +00001136 stack.pop_back();
Owen Anderson223718c2007-06-09 18:35:31 +00001137 }
Owen Anderson331bf6a2007-05-31 22:44:11 +00001138 }
Owen Anderson49409f62007-06-22 21:31:16 +00001139
1140 stack.clear();
Owen Anderson331bf6a2007-05-31 22:44:11 +00001141 }
1142}
1143
Owen Andersonfd5683a2007-06-21 00:19:05 +00001144/// dump - Dump a set of values to standard error
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001145void GVNPRE::dump(ValueNumberedSet& s) const {
Owen Anderson4a6ec8f2007-05-29 23:15:21 +00001146 DOUT << "{ ";
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001147 for (ValueNumberedSet::iterator I = s.begin(), E = s.end();
Owen Anderson0eca9aa2007-06-03 22:02:14 +00001148 I != E; ++I) {
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001149 DOUT << "" << VN.lookup(*I) << ": ";
Owen Anderson0eca9aa2007-06-03 22:02:14 +00001150 DEBUG((*I)->dump());
1151 }
1152 DOUT << "}\n\n";
1153}
1154
Owen Andersonfd5683a2007-06-21 00:19:05 +00001155/// elimination - Phase 3 of the main algorithm. Perform full redundancy
1156/// elimination by walking the dominator tree and removing any instruction that
1157/// is dominated by another instruction with the same value number.
Owen Anderson06c1e582007-06-20 22:10:02 +00001158bool GVNPRE::elimination() {
Owen Anderson06c1e582007-06-20 22:10:02 +00001159 bool changed_function = false;
1160
Owen Anderson42769842007-06-12 16:57:50 +00001161 std::vector<std::pair<Instruction*, Value*> > replace;
1162 std::vector<Instruction*> erase;
1163
1164 DominatorTree& DT = getAnalysis<DominatorTree>();
1165
1166 for (df_iterator<DomTreeNode*> DI = df_begin(DT.getRootNode()),
1167 E = df_end(DT.getRootNode()); DI != E; ++DI) {
1168 BasicBlock* BB = DI->getBlock();
1169
Owen Anderson42769842007-06-12 16:57:50 +00001170 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
1171 BI != BE; ++BI) {
1172
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001173 if (isa<BinaryOperator>(BI) || isa<CmpInst>(BI) ||
1174 isa<ShuffleVectorInst>(BI) || isa<InsertElementInst>(BI) ||
Owen Anderson59bd0532007-07-03 18:37:08 +00001175 isa<ExtractElementInst>(BI) || isa<SelectInst>(BI) ||
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001176 isa<CastInst>(BI) || isa<GetElementPtrInst>(BI)) {
Owen Anderson41c2cab2007-07-10 00:09:25 +00001177
1178 if (availableOut[BB].test(VN.lookup(BI)) && ! availableOut[BB].count(BI)) {
1179 Value *leader = find_leader(availableOut[BB], VN.lookup(BI));
Owen Anderson42769842007-06-12 16:57:50 +00001180 if (Instruction* Instr = dyn_cast<Instruction>(leader))
1181 if (Instr->getParent() != 0 && Instr != BI) {
1182 replace.push_back(std::make_pair(BI, leader));
1183 erase.push_back(BI);
1184 ++NumEliminated;
1185 }
Owen Anderson41c2cab2007-07-10 00:09:25 +00001186 }
Owen Anderson42769842007-06-12 16:57:50 +00001187 }
1188 }
1189 }
1190
1191 while (!replace.empty()) {
1192 std::pair<Instruction*, Value*> rep = replace.back();
1193 replace.pop_back();
1194 rep.first->replaceAllUsesWith(rep.second);
Owen Andersonb0714bb2007-06-20 18:30:20 +00001195 changed_function = true;
Owen Anderson42769842007-06-12 16:57:50 +00001196 }
1197
1198 for (std::vector<Instruction*>::iterator I = erase.begin(), E = erase.end();
1199 I != E; ++I)
1200 (*I)->eraseFromParent();
Owen Anderson06c1e582007-06-20 22:10:02 +00001201
1202 return changed_function;
Owen Anderson42769842007-06-12 16:57:50 +00001203}
1204
Owen Andersonfd5683a2007-06-21 00:19:05 +00001205/// cleanup - Delete any extraneous values that were created to represent
1206/// expressions without leaders.
Owen Anderson42769842007-06-12 16:57:50 +00001207void GVNPRE::cleanup() {
1208 while (!createdExpressions.empty()) {
1209 Instruction* I = createdExpressions.back();
1210 createdExpressions.pop_back();
1211
1212 delete I;
1213 }
1214}
1215
Owen Andersonfd5683a2007-06-21 00:19:05 +00001216/// buildsets_availout - When calculating availability, handle an instruction
1217/// by inserting it into the appropriate sets
Owen Anderson06c1e582007-06-20 22:10:02 +00001218void GVNPRE::buildsets_availout(BasicBlock::iterator I,
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001219 ValueNumberedSet& currAvail,
1220 ValueNumberedSet& currPhis,
1221 ValueNumberedSet& currExps,
1222 SmallPtrSet<Value*, 16>& currTemps) {
Owen Anderson0eb26572007-06-27 17:03:03 +00001223 // Handle PHI nodes
Owen Anderson06c1e582007-06-20 22:10:02 +00001224 if (PHINode* p = dyn_cast<PHINode>(I)) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001225 unsigned num = VN.lookup_or_add(p);
Owen Andersonf6e21872007-06-22 03:14:03 +00001226
Owen Anderson06c1e582007-06-20 22:10:02 +00001227 currPhis.insert(p);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001228 currPhis.set(num);
Owen Anderson59bd0532007-07-03 18:37:08 +00001229
1230 // Handle unary ops
Owen Anderson6b958c72007-07-03 19:01:42 +00001231 } else if (CastInst* U = dyn_cast<CastInst>(I)) {
Owen Anderson59bd0532007-07-03 18:37:08 +00001232 Value* leftValue = U->getOperand(0);
Owen Anderson06c1e582007-06-20 22:10:02 +00001233
Owen Anderson59bd0532007-07-03 18:37:08 +00001234 unsigned num = VN.lookup_or_add(U);
Owen Anderson59bd0532007-07-03 18:37:08 +00001235
1236 if (isa<Instruction>(leftValue))
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001237 if (!currExps.test(VN.lookup(leftValue))) {
Owen Anderson59bd0532007-07-03 18:37:08 +00001238 currExps.insert(leftValue);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001239 currExps.set(VN.lookup(leftValue));
Owen Anderson59bd0532007-07-03 18:37:08 +00001240 }
1241
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001242 if (!currExps.test(num)) {
Owen Anderson59bd0532007-07-03 18:37:08 +00001243 currExps.insert(U);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001244 currExps.set(num);
Owen Anderson59bd0532007-07-03 18:37:08 +00001245 }
1246
Owen Anderson0eb26572007-06-27 17:03:03 +00001247 // Handle binary ops
1248 } else if (isa<BinaryOperator>(I) || isa<CmpInst>(I) ||
1249 isa<ExtractElementInst>(I)) {
1250 User* U = cast<User>(I);
1251 Value* leftValue = U->getOperand(0);
1252 Value* rightValue = U->getOperand(1);
Owen Anderson06c1e582007-06-20 22:10:02 +00001253
Owen Anderson0eb26572007-06-27 17:03:03 +00001254 unsigned num = VN.lookup_or_add(U);
Owen Anderson06c1e582007-06-20 22:10:02 +00001255
1256 if (isa<Instruction>(leftValue))
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001257 if (!currExps.test(VN.lookup(leftValue))) {
Owen Andersonf6e21872007-06-22 03:14:03 +00001258 currExps.insert(leftValue);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001259 currExps.set(VN.lookup(leftValue));
Owen Andersonf6e21872007-06-22 03:14:03 +00001260 }
1261
Owen Anderson06c1e582007-06-20 22:10:02 +00001262 if (isa<Instruction>(rightValue))
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001263 if (!currExps.test(VN.lookup(rightValue))) {
Owen Andersonf6e21872007-06-22 03:14:03 +00001264 currExps.insert(rightValue);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001265 currExps.set(VN.lookup(rightValue));
Owen Andersonf6e21872007-06-22 03:14:03 +00001266 }
1267
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001268 if (!currExps.test(num)) {
Owen Anderson0eb26572007-06-27 17:03:03 +00001269 currExps.insert(U);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001270 currExps.set(num);
Owen Andersonf6e21872007-06-22 03:14:03 +00001271 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001272
Owen Anderson0eb26572007-06-27 17:03:03 +00001273 // Handle ternary ops
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001274 } else if (isa<InsertElementInst>(I) || isa<ShuffleVectorInst>(I) ||
1275 isa<SelectInst>(I)) {
Owen Anderson0eb26572007-06-27 17:03:03 +00001276 User* U = cast<User>(I);
1277 Value* leftValue = U->getOperand(0);
1278 Value* rightValue = U->getOperand(1);
1279 Value* thirdValue = U->getOperand(2);
Owen Anderson06c1e582007-06-20 22:10:02 +00001280
Owen Anderson0eb26572007-06-27 17:03:03 +00001281 VN.lookup_or_add(U);
Owen Andersonf6e21872007-06-22 03:14:03 +00001282
Owen Anderson0eb26572007-06-27 17:03:03 +00001283 unsigned num = VN.lookup_or_add(U);
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001284
1285 if (isa<Instruction>(leftValue))
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001286 if (!currExps.test(VN.lookup(leftValue))) {
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001287 currExps.insert(leftValue);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001288 currExps.set(VN.lookup(leftValue));
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001289 }
1290 if (isa<Instruction>(rightValue))
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001291 if (!currExps.test(VN.lookup(rightValue))) {
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001292 currExps.insert(rightValue);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001293 currExps.set(VN.lookup(rightValue));
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001294 }
1295 if (isa<Instruction>(thirdValue))
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001296 if (!currExps.test(VN.lookup(thirdValue))) {
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001297 currExps.insert(thirdValue);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001298 currExps.set(VN.lookup(thirdValue));
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001299 }
1300
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001301 if (!currExps.test(num)) {
Owen Anderson0eb26572007-06-27 17:03:03 +00001302 currExps.insert(U);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001303 currExps.set(num);
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001304 }
1305
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001306 // Handle vararg ops
1307 } else if (GetElementPtrInst* U = dyn_cast<GetElementPtrInst>(I)) {
1308 Value* ptrValue = U->getPointerOperand();
1309
1310 VN.lookup_or_add(U);
1311
1312 unsigned num = VN.lookup_or_add(U);
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001313
1314 if (isa<Instruction>(ptrValue))
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001315 if (!currExps.test(VN.lookup(ptrValue))) {
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001316 currExps.insert(ptrValue);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001317 currExps.set(VN.lookup(ptrValue));
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001318 }
1319
1320 for (GetElementPtrInst::op_iterator OI = U->idx_begin(), OE = U->idx_end();
1321 OI != OE; ++OI)
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001322 if (isa<Instruction>(*OI) && !currExps.test(VN.lookup(*OI))) {
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001323 currExps.insert(*OI);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001324 currExps.set(VN.lookup(*OI));
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001325 }
1326
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001327 if (!currExps.test(VN.lookup(U))) {
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001328 currExps.insert(U);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001329 currExps.set(num);
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001330 }
1331
Owen Anderson0eb26572007-06-27 17:03:03 +00001332 // Handle opaque ops
1333 } else if (!I->isTerminator()){
Owen Anderson06c1e582007-06-20 22:10:02 +00001334 VN.lookup_or_add(I);
Owen Andersonf6e21872007-06-22 03:14:03 +00001335
Owen Anderson06c1e582007-06-20 22:10:02 +00001336 currTemps.insert(I);
1337 }
1338
1339 if (!I->isTerminator())
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001340 if (!currAvail.test(VN.lookup(I))) {
Owen Andersonf6e21872007-06-22 03:14:03 +00001341 currAvail.insert(I);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001342 currAvail.set(VN.lookup(I));
Owen Andersonf6e21872007-06-22 03:14:03 +00001343 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001344}
Owen Anderson5fba6c12007-05-29 21:53:49 +00001345
Owen Andersonfd5683a2007-06-21 00:19:05 +00001346/// buildsets_anticout - When walking the postdom tree, calculate the ANTIC_OUT
1347/// set as a function of the ANTIC_IN set of the block's predecessors
Owen Anderson28a2d442007-06-22 00:20:30 +00001348bool GVNPRE::buildsets_anticout(BasicBlock* BB,
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001349 ValueNumberedSet& anticOut,
Owen Anderson06c1e582007-06-20 22:10:02 +00001350 std::set<BasicBlock*>& visited) {
1351 if (BB->getTerminator()->getNumSuccessors() == 1) {
Owen Anderson191eb062007-06-25 05:41:12 +00001352 if (BB->getTerminator()->getSuccessor(0) != BB &&
1353 visited.count(BB->getTerminator()->getSuccessor(0)) == 0) {
Owen Anderson28a2d442007-06-22 00:20:30 +00001354 return true;
Owen Anderson191eb062007-06-25 05:41:12 +00001355 }
1356 else {
Owen Anderson06c1e582007-06-20 22:10:02 +00001357 phi_translate_set(anticipatedIn[BB->getTerminator()->getSuccessor(0)],
1358 BB, BB->getTerminator()->getSuccessor(0), anticOut);
Owen Anderson191eb062007-06-25 05:41:12 +00001359 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001360 } else if (BB->getTerminator()->getNumSuccessors() > 1) {
1361 BasicBlock* first = BB->getTerminator()->getSuccessor(0);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001362 for (ValueNumberedSet::iterator I = anticipatedIn[first].begin(),
1363 E = anticipatedIn[first].end(); I != E; ++I) {
1364 anticOut.insert(*I);
1365 anticOut.set(VN.lookup(*I));
1366 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001367
1368 for (unsigned i = 1; i < BB->getTerminator()->getNumSuccessors(); ++i) {
1369 BasicBlock* currSucc = BB->getTerminator()->getSuccessor(i);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001370 ValueNumberedSet& succAnticIn = anticipatedIn[currSucc];
Owen Anderson06c1e582007-06-20 22:10:02 +00001371
Owen Anderson2ff912b2007-06-21 17:57:53 +00001372 std::vector<Value*> temp;
1373
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001374 for (ValueNumberedSet::iterator I = anticOut.begin(),
Owen Anderson2ff912b2007-06-21 17:57:53 +00001375 E = anticOut.end(); I != E; ++I)
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001376 if (!succAnticIn.test(VN.lookup(*I)))
Owen Anderson2ff912b2007-06-21 17:57:53 +00001377 temp.push_back(*I);
1378
1379 for (std::vector<Value*>::iterator I = temp.begin(), E = temp.end();
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001380 I != E; ++I) {
Owen Anderson2ff912b2007-06-21 17:57:53 +00001381 anticOut.erase(*I);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001382 anticOut.reset(VN.lookup(*I));
1383 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001384 }
1385 }
Owen Anderson28a2d442007-06-22 00:20:30 +00001386
1387 return false;
Owen Anderson06c1e582007-06-20 22:10:02 +00001388}
1389
Owen Andersonfd5683a2007-06-21 00:19:05 +00001390/// buildsets_anticin - Walk the postdom tree, calculating ANTIC_OUT for
1391/// each block. ANTIC_IN is then a function of ANTIC_OUT and the GEN
1392/// sets populated in buildsets_availout
Owen Anderson28a2d442007-06-22 00:20:30 +00001393unsigned GVNPRE::buildsets_anticin(BasicBlock* BB,
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001394 ValueNumberedSet& anticOut,
1395 ValueNumberedSet& currExps,
Owen Andersone02da552007-06-28 00:34:34 +00001396 SmallPtrSet<Value*, 16>& currTemps,
Owen Anderson06c1e582007-06-20 22:10:02 +00001397 std::set<BasicBlock*>& visited) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001398 ValueNumberedSet& anticIn = anticipatedIn[BB];
Owen Anderson21a11312007-06-22 18:27:04 +00001399 unsigned old = anticIn.size();
Owen Anderson06c1e582007-06-20 22:10:02 +00001400
Owen Anderson28a2d442007-06-22 00:20:30 +00001401 bool defer = buildsets_anticout(BB, anticOut, visited);
1402 if (defer)
1403 return 0;
Owen Anderson2ff912b2007-06-21 17:57:53 +00001404
Owen Anderson06c1e582007-06-20 22:10:02 +00001405 anticIn.clear();
Owen Anderson2ff912b2007-06-21 17:57:53 +00001406
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001407 for (ValueNumberedSet::iterator I = anticOut.begin(),
Owen Anderson21a11312007-06-22 18:27:04 +00001408 E = anticOut.end(); I != E; ++I) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001409 anticIn.insert(*I);
1410 anticIn.set(VN.lookup(*I));
Owen Anderson06c1e582007-06-20 22:10:02 +00001411 }
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001412 for (ValueNumberedSet::iterator I = currExps.begin(),
Owen Anderson21a11312007-06-22 18:27:04 +00001413 E = currExps.end(); I != E; ++I) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001414 if (!anticIn.test(VN.lookup(*I))) {
Owen Anderson21a11312007-06-22 18:27:04 +00001415 anticIn.insert(*I);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001416 anticIn.set(VN.lookup(*I));
Owen Anderson21a11312007-06-22 18:27:04 +00001417 }
1418 }
1419
Owen Andersone02da552007-06-28 00:34:34 +00001420 for (SmallPtrSet<Value*, 16>::iterator I = currTemps.begin(),
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001421 E = currTemps.end(); I != E; ++I) {
Owen Anderson21a11312007-06-22 18:27:04 +00001422 anticIn.erase(*I);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001423 anticIn.reset(VN.lookup(*I));
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001424 }
Owen Anderson21a11312007-06-22 18:27:04 +00001425
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001426 clean(anticIn);
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001427 anticOut.clear();
Owen Anderson06c1e582007-06-20 22:10:02 +00001428
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001429 if (old != anticIn.size())
Owen Anderson28a2d442007-06-22 00:20:30 +00001430 return 2;
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001431 else
Owen Anderson28a2d442007-06-22 00:20:30 +00001432 return 1;
Owen Anderson06c1e582007-06-20 22:10:02 +00001433}
1434
Owen Andersonfd5683a2007-06-21 00:19:05 +00001435/// buildsets - Phase 1 of the main algorithm. Construct the AVAIL_OUT
1436/// and the ANTIC_IN sets.
Owen Anderson5477c542007-06-26 23:29:41 +00001437void GVNPRE::buildsets(Function& F) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001438 std::map<BasicBlock*, ValueNumberedSet> generatedExpressions;
Owen Andersone02da552007-06-28 00:34:34 +00001439 std::map<BasicBlock*, SmallPtrSet<Value*, 16> > generatedTemporaries;
Owen Anderson06c1e582007-06-20 22:10:02 +00001440
Owen Anderson5fba6c12007-05-29 21:53:49 +00001441 DominatorTree &DT = getAnalysis<DominatorTree>();
1442
Owen Anderson634a0632007-06-06 01:27:49 +00001443 // Phase 1, Part 1: calculate AVAIL_OUT
Owen Anderson5fba6c12007-05-29 21:53:49 +00001444
1445 // Top-down walk of the dominator tree
Devang Patelbdd1aae2007-06-04 00:32:22 +00001446 for (df_iterator<DomTreeNode*> DI = df_begin(DT.getRootNode()),
Owen Anderson5fba6c12007-05-29 21:53:49 +00001447 E = df_end(DT.getRootNode()); DI != E; ++DI) {
1448
1449 // Get the sets to update for this block
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001450 ValueNumberedSet& currExps = generatedExpressions[DI->getBlock()];
1451 ValueNumberedSet& currPhis = generatedPhis[DI->getBlock()];
Owen Andersone02da552007-06-28 00:34:34 +00001452 SmallPtrSet<Value*, 16>& currTemps = generatedTemporaries[DI->getBlock()];
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001453 ValueNumberedSet& currAvail = availableOut[DI->getBlock()];
Owen Anderson5fba6c12007-05-29 21:53:49 +00001454
Owen Andersonb56fba02007-06-19 03:31:41 +00001455 BasicBlock* BB = DI->getBlock();
1456
1457 // A block inherits AVAIL_OUT from its dominator
Owen Andersoneffc7a72007-07-09 22:29:50 +00001458 if (DI->getIDom() != 0)
1459 currAvail = availableOut[DI->getIDom()->getBlock()];
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001460
Owen Andersonb56fba02007-06-19 03:31:41 +00001461 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
Owen Anderson06c1e582007-06-20 22:10:02 +00001462 BI != BE; ++BI)
Owen Andersonf6e21872007-06-22 03:14:03 +00001463 buildsets_availout(BI, currAvail, currPhis, currExps,
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001464 currTemps);
Owen Andersonb56fba02007-06-19 03:31:41 +00001465
Owen Anderson5fba6c12007-05-29 21:53:49 +00001466 }
Owen Anderson191eb062007-06-25 05:41:12 +00001467
1468 // Phase 1, Part 2: calculate ANTIC_IN
Owen Anderson5fba6c12007-05-29 21:53:49 +00001469
Owen Anderson0c423072007-05-29 23:26:30 +00001470 std::set<BasicBlock*> visited;
Owen Anderson5477c542007-06-26 23:29:41 +00001471 SmallPtrSet<BasicBlock*, 4> block_changed;
1472 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ++FI)
1473 block_changed.insert(FI);
Owen Anderson0c423072007-05-29 23:26:30 +00001474
Owen Anderson5fba6c12007-05-29 21:53:49 +00001475 bool changed = true;
1476 unsigned iterations = 0;
Owen Anderson5477c542007-06-26 23:29:41 +00001477
Owen Anderson5fba6c12007-05-29 21:53:49 +00001478 while (changed) {
1479 changed = false;
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001480 ValueNumberedSet anticOut;
Owen Anderson5fba6c12007-05-29 21:53:49 +00001481
Owen Anderson5477c542007-06-26 23:29:41 +00001482 // Postorder walk of the CFG
Owen Anderson43ca4b42007-06-25 18:25:31 +00001483 for (po_iterator<BasicBlock*> BBI = po_begin(&F.getEntryBlock()),
1484 BBE = po_end(&F.getEntryBlock()); BBI != BBE; ++BBI) {
Owen Anderson191eb062007-06-25 05:41:12 +00001485 BasicBlock* BB = *BBI;
Owen Andersond184c182007-06-11 16:25:17 +00001486
Owen Anderson5477c542007-06-26 23:29:41 +00001487 if (block_changed.count(BB) != 0) {
1488 unsigned ret = buildsets_anticin(BB, anticOut,generatedExpressions[BB],
1489 generatedTemporaries[BB], visited);
Owen Anderson28a2d442007-06-22 00:20:30 +00001490
Owen Anderson5477c542007-06-26 23:29:41 +00001491 if (ret == 0) {
1492 changed = true;
1493 continue;
1494 } else {
1495 visited.insert(BB);
1496
1497 if (ret == 2)
1498 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
1499 PI != PE; ++PI) {
1500 block_changed.insert(*PI);
1501 }
1502 else
1503 block_changed.erase(BB);
1504
1505 changed |= (ret == 2);
Owen Anderson191eb062007-06-25 05:41:12 +00001506 }
Owen Anderson28a2d442007-06-22 00:20:30 +00001507 }
Owen Anderson5fba6c12007-05-29 21:53:49 +00001508 }
Owen Anderson38b6b222007-06-04 18:05:26 +00001509
Owen Anderson5fba6c12007-05-29 21:53:49 +00001510 iterations++;
1511 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001512}
1513
Owen Andersonfd5683a2007-06-21 00:19:05 +00001514/// insertion_pre - When a partial redundancy has been identified, eliminate it
1515/// by inserting appropriate values into the predecessors and a phi node in
1516/// the main block
Owen Anderson06c1e582007-06-20 22:10:02 +00001517void GVNPRE::insertion_pre(Value* e, BasicBlock* BB,
1518 std::map<BasicBlock*, Value*>& avail,
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001519 std::map<BasicBlock*, ValueNumberedSet>& new_sets) {
Owen Anderson06c1e582007-06-20 22:10:02 +00001520 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE; ++PI) {
1521 Value* e2 = avail[*PI];
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001522 if (!availableOut[*PI].test(VN.lookup(e2))) {
Owen Anderson06c1e582007-06-20 22:10:02 +00001523 User* U = cast<User>(e2);
1524
1525 Value* s1 = 0;
1526 if (isa<BinaryOperator>(U->getOperand(0)) ||
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001527 isa<CmpInst>(U->getOperand(0)) ||
1528 isa<ShuffleVectorInst>(U->getOperand(0)) ||
1529 isa<ExtractElementInst>(U->getOperand(0)) ||
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001530 isa<InsertElementInst>(U->getOperand(0)) ||
Owen Anderson59bd0532007-07-03 18:37:08 +00001531 isa<SelectInst>(U->getOperand(0)) ||
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001532 isa<CastInst>(U->getOperand(0)) ||
1533 isa<GetElementPtrInst>(U->getOperand(0)))
Owen Anderson06c1e582007-06-20 22:10:02 +00001534 s1 = find_leader(availableOut[*PI], VN.lookup(U->getOperand(0)));
1535 else
1536 s1 = U->getOperand(0);
1537
1538 Value* s2 = 0;
Owen Anderson59bd0532007-07-03 18:37:08 +00001539
1540 if (isa<BinaryOperator>(U) ||
1541 isa<CmpInst>(U) ||
1542 isa<ShuffleVectorInst>(U) ||
1543 isa<ExtractElementInst>(U) ||
1544 isa<InsertElementInst>(U) ||
1545 isa<SelectInst>(U))
1546 if (isa<BinaryOperator>(U->getOperand(1)) ||
1547 isa<CmpInst>(U->getOperand(1)) ||
1548 isa<ShuffleVectorInst>(U->getOperand(1)) ||
1549 isa<ExtractElementInst>(U->getOperand(1)) ||
1550 isa<InsertElementInst>(U->getOperand(1)) ||
1551 isa<SelectInst>(U->getOperand(1)) ||
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001552 isa<CastInst>(U->getOperand(1)) ||
1553 isa<GetElementPtrInst>(U->getOperand(1))) {
Owen Anderson59bd0532007-07-03 18:37:08 +00001554 s2 = find_leader(availableOut[*PI], VN.lookup(U->getOperand(1)));
1555 } else {
1556 s2 = U->getOperand(1);
1557 }
Owen Anderson0eb26572007-06-27 17:03:03 +00001558
1559 // Ternary Operators
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001560 Value* s3 = 0;
1561 if (isa<ShuffleVectorInst>(U) ||
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001562 isa<InsertElementInst>(U) ||
1563 isa<SelectInst>(U))
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001564 if (isa<BinaryOperator>(U->getOperand(2)) ||
1565 isa<CmpInst>(U->getOperand(2)) ||
1566 isa<ShuffleVectorInst>(U->getOperand(2)) ||
1567 isa<ExtractElementInst>(U->getOperand(2)) ||
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001568 isa<InsertElementInst>(U->getOperand(2)) ||
Owen Anderson59bd0532007-07-03 18:37:08 +00001569 isa<SelectInst>(U->getOperand(2)) ||
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001570 isa<CastInst>(U->getOperand(2)) ||
1571 isa<GetElementPtrInst>(U->getOperand(2))) {
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001572 s3 = find_leader(availableOut[*PI], VN.lookup(U->getOperand(2)));
Owen Anderson59bd0532007-07-03 18:37:08 +00001573 } else {
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001574 s3 = U->getOperand(2);
Owen Anderson59bd0532007-07-03 18:37:08 +00001575 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001576
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001577 // Vararg operators
1578 std::vector<Value*> sVarargs;
1579 if (GetElementPtrInst* G = dyn_cast<GetElementPtrInst>(U)) {
1580 for (GetElementPtrInst::op_iterator OI = G->idx_begin(),
1581 OE = G->idx_end(); OI != OE; ++OI) {
1582 if (isa<BinaryOperator>(*OI) ||
1583 isa<CmpInst>(*OI) ||
1584 isa<ShuffleVectorInst>(*OI) ||
1585 isa<ExtractElementInst>(*OI) ||
1586 isa<InsertElementInst>(*OI) ||
1587 isa<SelectInst>(*OI) ||
1588 isa<CastInst>(*OI) ||
1589 isa<GetElementPtrInst>(*OI)) {
1590 sVarargs.push_back(find_leader(availableOut[*PI],
1591 VN.lookup(*OI)));
1592 } else {
1593 sVarargs.push_back(*OI);
1594 }
1595 }
1596 }
1597
Owen Anderson06c1e582007-06-20 22:10:02 +00001598 Value* newVal = 0;
1599 if (BinaryOperator* BO = dyn_cast<BinaryOperator>(U))
1600 newVal = BinaryOperator::create(BO->getOpcode(), s1, s2,
1601 BO->getName()+".gvnpre",
1602 (*PI)->getTerminator());
1603 else if (CmpInst* C = dyn_cast<CmpInst>(U))
1604 newVal = CmpInst::create(C->getOpcode(), C->getPredicate(), s1, s2,
1605 C->getName()+".gvnpre",
1606 (*PI)->getTerminator());
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001607 else if (ShuffleVectorInst* S = dyn_cast<ShuffleVectorInst>(U))
1608 newVal = new ShuffleVectorInst(s1, s2, s3, S->getName()+".gvnpre",
1609 (*PI)->getTerminator());
1610 else if (InsertElementInst* S = dyn_cast<InsertElementInst>(U))
1611 newVal = new InsertElementInst(s1, s2, s3, S->getName()+".gvnpre",
1612 (*PI)->getTerminator());
1613 else if (ExtractElementInst* S = dyn_cast<ExtractElementInst>(U))
1614 newVal = new ExtractElementInst(s1, s2, S->getName()+".gvnpre",
1615 (*PI)->getTerminator());
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001616 else if (SelectInst* S = dyn_cast<SelectInst>(U))
Owen Andersonca1a1842007-07-04 22:33:23 +00001617 newVal = new SelectInst(s1, s2, s3, S->getName()+".gvnpre",
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001618 (*PI)->getTerminator());
Owen Anderson59bd0532007-07-03 18:37:08 +00001619 else if (CastInst* C = dyn_cast<CastInst>(U))
1620 newVal = CastInst::create(C->getOpcode(), s1, C->getType(),
1621 C->getName()+".gvnpre",
1622 (*PI)->getTerminator());
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001623 else if (GetElementPtrInst* G = dyn_cast<GetElementPtrInst>(U))
1624 newVal = new GetElementPtrInst(s1, &sVarargs[0], sVarargs.size(),
1625 G->getName()+".gvnpre",
1626 (*PI)->getTerminator());
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001627
Owen Anderson06c1e582007-06-20 22:10:02 +00001628
1629 VN.add(newVal, VN.lookup(U));
1630
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001631 ValueNumberedSet& predAvail = availableOut[*PI];
Owen Anderson06c1e582007-06-20 22:10:02 +00001632 val_replace(predAvail, newVal);
Owen Anderson02e96982007-07-05 23:11:26 +00001633 val_replace(new_sets[*PI], newVal);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001634 predAvail.set(VN.lookup(newVal));
Owen Andersonfd5683a2007-06-21 00:19:05 +00001635
Owen Anderson06c1e582007-06-20 22:10:02 +00001636 std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
1637 if (av != avail.end())
1638 avail.erase(av);
1639 avail.insert(std::make_pair(*PI, newVal));
1640
1641 ++NumInsertedVals;
1642 }
1643 }
1644
1645 PHINode* p = 0;
1646
1647 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE; ++PI) {
1648 if (p == 0)
1649 p = new PHINode(avail[*PI]->getType(), "gvnpre-join", BB->begin());
Owen Anderson02e96982007-07-05 23:11:26 +00001650
Owen Anderson06c1e582007-06-20 22:10:02 +00001651 p->addIncoming(avail[*PI], *PI);
1652 }
1653
1654 VN.add(p, VN.lookup(e));
Owen Anderson06c1e582007-06-20 22:10:02 +00001655 val_replace(availableOut[BB], p);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001656 availableOut[BB].set(VN.lookup(e));
Owen Anderson7d4bbc12007-07-06 20:29:43 +00001657 generatedPhis[BB].insert(p);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001658 generatedPhis[BB].set(VN.lookup(e));
Owen Anderson02e96982007-07-05 23:11:26 +00001659 new_sets[BB].insert(p);
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001660 new_sets[BB].set(VN.lookup(e));
Owen Anderson06c1e582007-06-20 22:10:02 +00001661
1662 ++NumInsertedPhis;
1663}
1664
Owen Andersonfd5683a2007-06-21 00:19:05 +00001665/// insertion_mergepoint - When walking the dom tree, check at each merge
1666/// block for the possibility of a partial redundancy. If present, eliminate it
Owen Anderson06c1e582007-06-20 22:10:02 +00001667unsigned GVNPRE::insertion_mergepoint(std::vector<Value*>& workList,
Owen Anderson28a2d442007-06-22 00:20:30 +00001668 df_iterator<DomTreeNode*>& D,
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001669 std::map<BasicBlock*, ValueNumberedSet >& new_sets) {
Owen Anderson06c1e582007-06-20 22:10:02 +00001670 bool changed_function = false;
1671 bool new_stuff = false;
1672
1673 BasicBlock* BB = D->getBlock();
1674 for (unsigned i = 0; i < workList.size(); ++i) {
1675 Value* e = workList[i];
1676
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001677 if (isa<BinaryOperator>(e) || isa<CmpInst>(e) ||
1678 isa<ExtractElementInst>(e) || isa<InsertElementInst>(e) ||
Owen Anderson2e4b6fea2007-07-03 23:51:19 +00001679 isa<ShuffleVectorInst>(e) || isa<SelectInst>(e) || isa<CastInst>(e) ||
1680 isa<GetElementPtrInst>(e)) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001681 if (availableOut[D->getIDom()->getBlock()].test(VN.lookup(e)))
Owen Anderson06c1e582007-06-20 22:10:02 +00001682 continue;
1683
1684 std::map<BasicBlock*, Value*> avail;
1685 bool by_some = false;
Owen Anderson7d4bbc12007-07-06 20:29:43 +00001686 bool all_same = true;
1687 Value * first_s = 0;
Owen Anderson06c1e582007-06-20 22:10:02 +00001688
1689 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE;
1690 ++PI) {
1691 Value *e2 = phi_translate(e, *PI, BB);
1692 Value *e3 = find_leader(availableOut[*PI], VN.lookup(e2));
1693
1694 if (e3 == 0) {
1695 std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
1696 if (av != avail.end())
1697 avail.erase(av);
1698 avail.insert(std::make_pair(*PI, e2));
Owen Anderson7d4bbc12007-07-06 20:29:43 +00001699 all_same = false;
Owen Anderson06c1e582007-06-20 22:10:02 +00001700 } else {
1701 std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
1702 if (av != avail.end())
1703 avail.erase(av);
1704 avail.insert(std::make_pair(*PI, e3));
1705
1706 by_some = true;
Owen Anderson7d4bbc12007-07-06 20:29:43 +00001707 if (first_s == 0)
1708 first_s = e3;
1709 else if (first_s != e3)
1710 all_same = false;
Owen Anderson06c1e582007-06-20 22:10:02 +00001711 }
1712 }
1713
Owen Anderson7d4bbc12007-07-06 20:29:43 +00001714 if (by_some && !all_same &&
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001715 !generatedPhis[BB].test(VN.lookup(e))) {
Owen Anderson02e96982007-07-05 23:11:26 +00001716 insertion_pre(e, BB, avail, new_sets);
Owen Anderson06c1e582007-06-20 22:10:02 +00001717
1718 changed_function = true;
1719 new_stuff = true;
1720 }
1721 }
1722 }
1723
1724 unsigned retval = 0;
1725 if (changed_function)
1726 retval += 1;
1727 if (new_stuff)
1728 retval += 2;
1729
1730 return retval;
1731}
1732
Owen Andersonfd5683a2007-06-21 00:19:05 +00001733/// insert - Phase 2 of the main algorithm. Walk the dominator tree looking for
1734/// merge points. When one is found, check for a partial redundancy. If one is
1735/// present, eliminate it. Repeat this walk until no changes are made.
Owen Anderson06c1e582007-06-20 22:10:02 +00001736bool GVNPRE::insertion(Function& F) {
1737 bool changed_function = false;
1738
1739 DominatorTree &DT = getAnalysis<DominatorTree>();
Owen Andersonbe802402007-06-08 01:03:01 +00001740
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001741 std::map<BasicBlock*, ValueNumberedSet> new_sets;
Owen Andersonbe802402007-06-08 01:03:01 +00001742 bool new_stuff = true;
1743 while (new_stuff) {
1744 new_stuff = false;
Owen Andersonbe802402007-06-08 01:03:01 +00001745 for (df_iterator<DomTreeNode*> DI = df_begin(DT.getRootNode()),
1746 E = df_end(DT.getRootNode()); DI != E; ++DI) {
1747 BasicBlock* BB = DI->getBlock();
1748
Owen Andersond184c182007-06-11 16:25:17 +00001749 if (BB == 0)
1750 continue;
1751
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001752 ValueNumberedSet& availOut = availableOut[BB];
1753 ValueNumberedSet& anticIn = anticipatedIn[BB];
Owen Andersonbe802402007-06-08 01:03:01 +00001754
1755 // Replace leaders with leaders inherited from dominator
1756 if (DI->getIDom() != 0) {
Owen Anderson1c83b5d2007-07-09 06:50:06 +00001757 ValueNumberedSet& dom_set = new_sets[DI->getIDom()->getBlock()];
1758 for (ValueNumberedSet::iterator I = dom_set.begin(),
Owen Andersonbe802402007-06-08 01:03:01 +00001759 E = dom_set.end(); I != E; ++I) {
Owen Anderson02e96982007-07-05 23:11:26 +00001760 val_replace(new_sets[BB], *I);
Owen Andersonb56fba02007-06-19 03:31:41 +00001761 val_replace(availOut, *I);
Owen Andersonbe802402007-06-08 01:03:01 +00001762 }
1763 }
1764
1765 // If there is more than one predecessor...
1766 if (pred_begin(BB) != pred_end(BB) && ++pred_begin(BB) != pred_end(BB)) {
1767 std::vector<Value*> workList;
Owen Andersond50a29d2007-06-22 00:43:22 +00001768 workList.reserve(anticIn.size());
Owen Andersonbe802402007-06-08 01:03:01 +00001769 topo_sort(anticIn, workList);
1770
Owen Anderson02e96982007-07-05 23:11:26 +00001771 unsigned result = insertion_mergepoint(workList, DI, new_sets);
Owen Anderson06c1e582007-06-20 22:10:02 +00001772 if (result & 1)
1773 changed_function = true;
1774 if (result & 2)
1775 new_stuff = true;
Owen Andersonbe802402007-06-08 01:03:01 +00001776 }
1777 }
Owen Andersonbe802402007-06-08 01:03:01 +00001778 }
Owen Anderson634a0632007-06-06 01:27:49 +00001779
Owen Anderson06c1e582007-06-20 22:10:02 +00001780 return changed_function;
1781}
1782
Owen Andersonfd5683a2007-06-21 00:19:05 +00001783// GVNPRE::runOnFunction - This is the main transformation entry point for a
1784// function.
1785//
Owen Anderson06c1e582007-06-20 22:10:02 +00001786bool GVNPRE::runOnFunction(Function &F) {
Owen Andersonfd5683a2007-06-21 00:19:05 +00001787 // Clean out global sets from any previous functions
Owen Anderson06c1e582007-06-20 22:10:02 +00001788 VN.clear();
1789 createdExpressions.clear();
1790 availableOut.clear();
1791 anticipatedIn.clear();
Owen Anderson7d4bbc12007-07-06 20:29:43 +00001792 generatedPhis.clear();
1793
Owen Anderson06c1e582007-06-20 22:10:02 +00001794 bool changed_function = false;
1795
1796 // Phase 1: BuildSets
Owen Andersonfd5683a2007-06-21 00:19:05 +00001797 // This phase calculates the AVAIL_OUT and ANTIC_IN sets
Owen Anderson5477c542007-06-26 23:29:41 +00001798 buildsets(F);
Owen Anderson06c1e582007-06-20 22:10:02 +00001799
1800 // Phase 2: Insert
Owen Andersonfd5683a2007-06-21 00:19:05 +00001801 // This phase inserts values to make partially redundant values
1802 // fully redundant
Owen Anderson06c1e582007-06-20 22:10:02 +00001803 changed_function |= insertion(F);
1804
Owen Anderson634a0632007-06-06 01:27:49 +00001805 // Phase 3: Eliminate
Owen Andersonfd5683a2007-06-21 00:19:05 +00001806 // This phase performs trivial full redundancy elimination
Owen Anderson06c1e582007-06-20 22:10:02 +00001807 changed_function |= elimination();
Owen Anderson55994f22007-06-08 20:44:02 +00001808
Owen Andersonbe802402007-06-08 01:03:01 +00001809 // Phase 4: Cleanup
Owen Andersonfd5683a2007-06-21 00:19:05 +00001810 // This phase cleans up values that were created solely
1811 // as leaders for expressions
Owen Anderson42769842007-06-12 16:57:50 +00001812 cleanup();
Owen Andersonbe802402007-06-08 01:03:01 +00001813
Owen Andersonb0714bb2007-06-20 18:30:20 +00001814 return changed_function;
Owen Anderson5fba6c12007-05-29 21:53:49 +00001815}