blob: 25cd8165fd094ee152acc16fd199a6f20117dd90 [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 Andersonbe802402007-06-08 01:03:01 +000034#include "llvm/Support/CFG.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000035#include "llvm/Support/Compiler.h"
Owen Anderson4a6ec8f2007-05-29 23:15:21 +000036#include "llvm/Support/Debug.h"
Owen Anderson5fba6c12007-05-29 21:53:49 +000037#include <algorithm>
Owen Anderson331bf6a2007-05-31 22:44:11 +000038#include <deque>
Owen Anderson5fba6c12007-05-29 21:53:49 +000039#include <map>
Owen Anderson331bf6a2007-05-31 22:44:11 +000040#include <vector>
Owen Anderson5fba6c12007-05-29 21:53:49 +000041#include <set>
Owen Anderson5fba6c12007-05-29 21:53:49 +000042using namespace llvm;
43
Owen Andersonb56fba02007-06-19 03:31:41 +000044//===----------------------------------------------------------------------===//
45// ValueTable Class
46//===----------------------------------------------------------------------===//
47
Owen Andersonfd5683a2007-06-21 00:19:05 +000048/// This class holds the mapping between values and value numbers. It is used
49/// as an efficient mechanism to determine the expression-wise equivalence of
50/// two values.
Owen Andersonb56fba02007-06-19 03:31:41 +000051
52namespace {
53 class VISIBILITY_HIDDEN ValueTable {
54 public:
55 struct Expression {
56 enum ExpressionOpcode { ADD, SUB, MUL, UDIV, SDIV, FDIV, UREM, SREM,
57 FREM, SHL, LSHR, ASHR, AND, OR, XOR, ICMPEQ,
58 ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
59 ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
60 FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
61 FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
Owen Andersonb6a39fc2007-06-27 04:10:46 +000062 FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
Owen Anderson67799d42007-06-29 00:51:03 +000063 SHUFFLE, SELECT, TRUNC, ZEXT, SEXT, FPTOUI,
64 FPTOSI, UITOFP, SITOFP, FPTRUNC, FPEXT,
65 PTRTOINT, INTTOPTR, BITCAST};
Owen Andersonb56fba02007-06-19 03:31:41 +000066
67 ExpressionOpcode opcode;
Owen Andersonc738f7c2007-06-29 00:40:05 +000068 const Type* type;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000069 uint32_t firstVN;
70 uint32_t secondVN;
71 uint32_t thirdVN;
Owen Andersonb56fba02007-06-19 03:31:41 +000072
73 bool operator< (const Expression& other) const {
74 if (opcode < other.opcode)
75 return true;
76 else if (opcode > other.opcode)
77 return false;
Owen Andersonc738f7c2007-06-29 00:40:05 +000078 else if (type < other.type)
79 return true;
80 else if (type > other.type)
81 return false;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000082 else if (firstVN < other.firstVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000083 return true;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000084 else if (firstVN > other.firstVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000085 return false;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000086 else if (secondVN < other.secondVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000087 return true;
Owen Andersonb6a39fc2007-06-27 04:10:46 +000088 else if (secondVN > other.secondVN)
89 return false;
90 else if (thirdVN < other.thirdVN)
91 return true;
92 else if (thirdVN > other.thirdVN)
Owen Andersonb56fba02007-06-19 03:31:41 +000093 return false;
94 else
95 return false;
Owen Andersona75dd4d2007-06-12 00:50:47 +000096 }
Owen Andersonb56fba02007-06-19 03:31:41 +000097 };
98
99 private:
Owen Anderson1ad2c102007-06-19 23:23:54 +0000100 DenseMap<Value*, uint32_t> valueNumbering;
Owen Andersonb56fba02007-06-19 03:31:41 +0000101 std::map<Expression, uint32_t> expressionNumbering;
102
Owen Andersonb56fba02007-06-19 03:31:41 +0000103 uint32_t nextValueNumber;
104
105 Expression::ExpressionOpcode getOpcode(BinaryOperator* BO);
106 Expression::ExpressionOpcode getOpcode(CmpInst* C);
Owen Anderson67799d42007-06-29 00:51:03 +0000107 Expression::ExpressionOpcode getOpcode(CastInst* C);
Owen Andersonfd5683a2007-06-21 00:19:05 +0000108 Expression create_expression(BinaryOperator* BO);
109 Expression create_expression(CmpInst* C);
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000110 Expression create_expression(ShuffleVectorInst* V);
111 Expression create_expression(ExtractElementInst* C);
112 Expression create_expression(InsertElementInst* V);
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000113 Expression create_expression(SelectInst* V);
Owen Anderson67799d42007-06-29 00:51:03 +0000114 Expression create_expression(CastInst* C);
Owen Andersonb56fba02007-06-19 03:31:41 +0000115 public:
116 ValueTable() { nextValueNumber = 1; }
117 uint32_t lookup_or_add(Value* V);
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000118 uint32_t lookup(Value* V) const;
Owen Andersonb56fba02007-06-19 03:31:41 +0000119 void add(Value* V, uint32_t num);
120 void clear();
Owen Anderson91c54952007-06-19 05:37:32 +0000121 void erase(Value* v);
Owen Anderson28a2d442007-06-22 00:20:30 +0000122 unsigned size();
Owen Andersonb56fba02007-06-19 03:31:41 +0000123 };
124}
125
Owen Andersonfd5683a2007-06-21 00:19:05 +0000126//===----------------------------------------------------------------------===//
127// ValueTable Internal Functions
128//===----------------------------------------------------------------------===//
Owen Andersonb56fba02007-06-19 03:31:41 +0000129ValueTable::Expression::ExpressionOpcode
Owen Andersonfd5683a2007-06-21 00:19:05 +0000130 ValueTable::getOpcode(BinaryOperator* BO) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000131 switch(BO->getOpcode()) {
132 case Instruction::Add:
133 return Expression::ADD;
134 case Instruction::Sub:
135 return Expression::SUB;
136 case Instruction::Mul:
137 return Expression::MUL;
138 case Instruction::UDiv:
139 return Expression::UDIV;
140 case Instruction::SDiv:
141 return Expression::SDIV;
142 case Instruction::FDiv:
143 return Expression::FDIV;
144 case Instruction::URem:
145 return Expression::UREM;
146 case Instruction::SRem:
147 return Expression::SREM;
148 case Instruction::FRem:
149 return Expression::FREM;
150 case Instruction::Shl:
151 return Expression::SHL;
152 case Instruction::LShr:
153 return Expression::LSHR;
154 case Instruction::AShr:
155 return Expression::ASHR;
156 case Instruction::And:
157 return Expression::AND;
158 case Instruction::Or:
159 return Expression::OR;
160 case Instruction::Xor:
161 return Expression::XOR;
162
163 // THIS SHOULD NEVER HAPPEN
164 default:
165 assert(0 && "Binary operator with unknown opcode?");
166 return Expression::ADD;
167 }
168}
169
170ValueTable::Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
171 if (C->getOpcode() == Instruction::ICmp) {
172 switch (C->getPredicate()) {
173 case ICmpInst::ICMP_EQ:
174 return Expression::ICMPEQ;
175 case ICmpInst::ICMP_NE:
176 return Expression::ICMPNE;
177 case ICmpInst::ICMP_UGT:
178 return Expression::ICMPUGT;
179 case ICmpInst::ICMP_UGE:
180 return Expression::ICMPUGE;
181 case ICmpInst::ICMP_ULT:
182 return Expression::ICMPULT;
183 case ICmpInst::ICMP_ULE:
184 return Expression::ICMPULE;
185 case ICmpInst::ICMP_SGT:
186 return Expression::ICMPSGT;
187 case ICmpInst::ICMP_SGE:
188 return Expression::ICMPSGE;
189 case ICmpInst::ICMP_SLT:
190 return Expression::ICMPSLT;
191 case ICmpInst::ICMP_SLE:
192 return Expression::ICMPSLE;
193
194 // THIS SHOULD NEVER HAPPEN
195 default:
196 assert(0 && "Comparison with unknown predicate?");
197 return Expression::ICMPEQ;
198 }
199 } else {
200 switch (C->getPredicate()) {
201 case FCmpInst::FCMP_OEQ:
202 return Expression::FCMPOEQ;
203 case FCmpInst::FCMP_OGT:
204 return Expression::FCMPOGT;
205 case FCmpInst::FCMP_OGE:
206 return Expression::FCMPOGE;
207 case FCmpInst::FCMP_OLT:
208 return Expression::FCMPOLT;
209 case FCmpInst::FCMP_OLE:
210 return Expression::FCMPOLE;
211 case FCmpInst::FCMP_ONE:
212 return Expression::FCMPONE;
213 case FCmpInst::FCMP_ORD:
214 return Expression::FCMPORD;
215 case FCmpInst::FCMP_UNO:
216 return Expression::FCMPUNO;
217 case FCmpInst::FCMP_UEQ:
218 return Expression::FCMPUEQ;
219 case FCmpInst::FCMP_UGT:
220 return Expression::FCMPUGT;
221 case FCmpInst::FCMP_UGE:
222 return Expression::FCMPUGE;
223 case FCmpInst::FCMP_ULT:
224 return Expression::FCMPULT;
225 case FCmpInst::FCMP_ULE:
226 return Expression::FCMPULE;
227 case FCmpInst::FCMP_UNE:
228 return Expression::FCMPUNE;
229
230 // THIS SHOULD NEVER HAPPEN
231 default:
232 assert(0 && "Comparison with unknown predicate?");
233 return Expression::FCMPOEQ;
Owen Anderson223718c2007-06-09 18:35:31 +0000234 }
235 }
Owen Andersonb56fba02007-06-19 03:31:41 +0000236}
237
Owen Anderson67799d42007-06-29 00:51:03 +0000238ValueTable::Expression::ExpressionOpcode
239 ValueTable::getOpcode(CastInst* C) {
240 switch(C->getOpcode()) {
241 case Instruction::Trunc:
242 return Expression::TRUNC;
243 case Instruction::ZExt:
244 return Expression::ZEXT;
245 case Instruction::SExt:
246 return Expression::SEXT;
247 case Instruction::FPToUI:
248 return Expression::FPTOUI;
249 case Instruction::FPToSI:
250 return Expression::FPTOSI;
251 case Instruction::UIToFP:
252 return Expression::UITOFP;
253 case Instruction::SIToFP:
254 return Expression::SITOFP;
255 case Instruction::FPTrunc:
256 return Expression::FPTRUNC;
257 case Instruction::FPExt:
258 return Expression::FPEXT;
259 case Instruction::PtrToInt:
260 return Expression::PTRTOINT;
261 case Instruction::IntToPtr:
262 return Expression::INTTOPTR;
263 case Instruction::BitCast:
264 return Expression::BITCAST;
265
266 // THIS SHOULD NEVER HAPPEN
267 default:
268 assert(0 && "Cast operator with unknown opcode?");
269 return Expression::BITCAST;
270 }
271}
272
Owen Andersonfd5683a2007-06-21 00:19:05 +0000273ValueTable::Expression ValueTable::create_expression(BinaryOperator* BO) {
274 Expression e;
275
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000276 e.firstVN = lookup_or_add(BO->getOperand(0));
277 e.secondVN = lookup_or_add(BO->getOperand(1));
278 e.thirdVN = 0;
Owen Andersonc738f7c2007-06-29 00:40:05 +0000279 e.type = BO->getType();
Owen Andersonfd5683a2007-06-21 00:19:05 +0000280 e.opcode = getOpcode(BO);
281
Owen Andersonfd5683a2007-06-21 00:19:05 +0000282 return e;
283}
284
285ValueTable::Expression ValueTable::create_expression(CmpInst* C) {
286 Expression e;
287
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000288 e.firstVN = lookup_or_add(C->getOperand(0));
289 e.secondVN = lookup_or_add(C->getOperand(1));
290 e.thirdVN = 0;
Owen Andersonc738f7c2007-06-29 00:40:05 +0000291 e.type = C->getType();
Owen Andersonfd5683a2007-06-21 00:19:05 +0000292 e.opcode = getOpcode(C);
293
Owen Andersonfd5683a2007-06-21 00:19:05 +0000294 return e;
295}
296
Owen Anderson67799d42007-06-29 00:51:03 +0000297ValueTable::Expression ValueTable::create_expression(CastInst* C) {
298 Expression e;
299
300 e.firstVN = lookup_or_add(C->getOperand(0));
301 e.secondVN = 0;
302 e.thirdVN = 0;
303 e.type = C->getType();
304 e.opcode = getOpcode(C);
305
306 return e;
307}
308
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000309ValueTable::Expression ValueTable::create_expression(ShuffleVectorInst* S) {
310 Expression e;
311
312 e.firstVN = lookup_or_add(S->getOperand(0));
313 e.secondVN = lookup_or_add(S->getOperand(1));
314 e.thirdVN = lookup_or_add(S->getOperand(2));
Owen Andersonc738f7c2007-06-29 00:40:05 +0000315 e.type = S->getType();
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000316 e.opcode = Expression::SHUFFLE;
317
318 return e;
319}
320
321ValueTable::Expression ValueTable::create_expression(ExtractElementInst* E) {
322 Expression e;
323
324 e.firstVN = lookup_or_add(E->getOperand(0));
325 e.secondVN = lookup_or_add(E->getOperand(1));
326 e.thirdVN = 0;
Owen Andersonc738f7c2007-06-29 00:40:05 +0000327 e.type = E->getType();
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000328 e.opcode = Expression::EXTRACT;
329
330 return e;
331}
332
333ValueTable::Expression ValueTable::create_expression(InsertElementInst* I) {
334 Expression e;
335
336 e.firstVN = lookup_or_add(I->getOperand(0));
337 e.secondVN = lookup_or_add(I->getOperand(1));
338 e.thirdVN = lookup_or_add(I->getOperand(2));
Owen Andersonc738f7c2007-06-29 00:40:05 +0000339 e.type = I->getType();
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000340 e.opcode = Expression::INSERT;
341
342 return e;
343}
344
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000345ValueTable::Expression ValueTable::create_expression(SelectInst* I) {
346 Expression e;
347
348 e.firstVN = lookup_or_add(I->getCondition());
349 e.secondVN = lookup_or_add(I->getTrueValue());
350 e.thirdVN = lookup_or_add(I->getFalseValue());
Owen Andersonc738f7c2007-06-29 00:40:05 +0000351 e.type = I->getType();
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000352 e.opcode = Expression::SELECT;
353
354 return e;
355}
356
Owen Andersonfd5683a2007-06-21 00:19:05 +0000357//===----------------------------------------------------------------------===//
358// ValueTable External Functions
359//===----------------------------------------------------------------------===//
360
361/// lookup_or_add - Returns the value number for the specified value, assigning
362/// it a new number if it did not have one before.
Owen Andersonb56fba02007-06-19 03:31:41 +0000363uint32_t ValueTable::lookup_or_add(Value* V) {
Owen Anderson1ad2c102007-06-19 23:23:54 +0000364 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
Owen Andersonb56fba02007-06-19 03:31:41 +0000365 if (VI != valueNumbering.end())
366 return VI->second;
Owen Anderson223718c2007-06-09 18:35:31 +0000367
Owen Anderson223718c2007-06-09 18:35:31 +0000368
Owen Andersonb56fba02007-06-19 03:31:41 +0000369 if (BinaryOperator* BO = dyn_cast<BinaryOperator>(V)) {
370 Expression e = create_expression(BO);
371
372 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
373 if (EI != expressionNumbering.end()) {
374 valueNumbering.insert(std::make_pair(V, EI->second));
375 return EI->second;
376 } else {
377 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
378 valueNumbering.insert(std::make_pair(V, nextValueNumber));
379
380 return nextValueNumber++;
381 }
382 } else if (CmpInst* C = dyn_cast<CmpInst>(V)) {
383 Expression e = create_expression(C);
384
385 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
386 if (EI != expressionNumbering.end()) {
387 valueNumbering.insert(std::make_pair(V, EI->second));
388 return EI->second;
389 } else {
390 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
391 valueNumbering.insert(std::make_pair(V, nextValueNumber));
392
393 return nextValueNumber++;
394 }
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000395 } else if (ShuffleVectorInst* U = dyn_cast<ShuffleVectorInst>(V)) {
396 Expression e = create_expression(U);
397
398 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
399 if (EI != expressionNumbering.end()) {
400 valueNumbering.insert(std::make_pair(V, EI->second));
401 return EI->second;
402 } else {
403 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
404 valueNumbering.insert(std::make_pair(V, nextValueNumber));
405
406 return nextValueNumber++;
407 }
408 } else if (ExtractElementInst* U = dyn_cast<ExtractElementInst>(V)) {
409 Expression e = create_expression(U);
410
411 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
412 if (EI != expressionNumbering.end()) {
413 valueNumbering.insert(std::make_pair(V, EI->second));
414 return EI->second;
415 } else {
416 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
417 valueNumbering.insert(std::make_pair(V, nextValueNumber));
418
419 return nextValueNumber++;
420 }
421 } else if (InsertElementInst* U = dyn_cast<InsertElementInst>(V)) {
422 Expression e = create_expression(U);
423
424 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
425 if (EI != expressionNumbering.end()) {
426 valueNumbering.insert(std::make_pair(V, EI->second));
427 return EI->second;
428 } else {
429 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
430 valueNumbering.insert(std::make_pair(V, nextValueNumber));
431
432 return nextValueNumber++;
433 }
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000434 } else if (SelectInst* U = dyn_cast<SelectInst>(V)) {
435 Expression e = create_expression(U);
436
437 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
438 if (EI != expressionNumbering.end()) {
439 valueNumbering.insert(std::make_pair(V, EI->second));
440 return EI->second;
441 } else {
442 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
443 valueNumbering.insert(std::make_pair(V, nextValueNumber));
444
445 return nextValueNumber++;
446 }
Owen Anderson67799d42007-06-29 00:51:03 +0000447 } else if (CastInst* U = dyn_cast<CastInst>(V)) {
448 Expression e = create_expression(U);
449
450 std::map<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
451 if (EI != expressionNumbering.end()) {
452 valueNumbering.insert(std::make_pair(V, EI->second));
453 return EI->second;
454 } else {
455 expressionNumbering.insert(std::make_pair(e, nextValueNumber));
456 valueNumbering.insert(std::make_pair(V, nextValueNumber));
457
458 return nextValueNumber++;
459 }
Owen Andersonb56fba02007-06-19 03:31:41 +0000460 } else {
461 valueNumbering.insert(std::make_pair(V, nextValueNumber));
462 return nextValueNumber++;
Owen Anderson0b68cda2007-06-03 05:55:58 +0000463 }
Owen Andersonb56fba02007-06-19 03:31:41 +0000464}
465
Owen Andersonfd5683a2007-06-21 00:19:05 +0000466/// lookup - Returns the value number of the specified value. Fails if
467/// the value has not yet been numbered.
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000468uint32_t ValueTable::lookup(Value* V) const {
Owen Anderson1ad2c102007-06-19 23:23:54 +0000469 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
Owen Andersonb56fba02007-06-19 03:31:41 +0000470 if (VI != valueNumbering.end())
471 return VI->second;
472 else
473 assert(0 && "Value not numbered?");
474
475 return 0;
476}
477
Owen Andersonfd5683a2007-06-21 00:19:05 +0000478/// add - Add the specified value with the given value number, removing
479/// its old number, if any
Owen Andersonb56fba02007-06-19 03:31:41 +0000480void ValueTable::add(Value* V, uint32_t num) {
Owen Anderson1ad2c102007-06-19 23:23:54 +0000481 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
Owen Andersonb56fba02007-06-19 03:31:41 +0000482 if (VI != valueNumbering.end())
483 valueNumbering.erase(VI);
484 valueNumbering.insert(std::make_pair(V, num));
485}
486
Owen Anderson191eb062007-06-25 05:41:12 +0000487/// clear - Remove all entries from the ValueTable
Owen Andersonb56fba02007-06-19 03:31:41 +0000488void ValueTable::clear() {
489 valueNumbering.clear();
490 expressionNumbering.clear();
491 nextValueNumber = 1;
492}
Owen Anderson0b68cda2007-06-03 05:55:58 +0000493
Owen Anderson191eb062007-06-25 05:41:12 +0000494/// erase - Remove a value from the value numbering
Owen Anderson91c54952007-06-19 05:37:32 +0000495void ValueTable::erase(Value* V) {
Owen Anderson91c54952007-06-19 05:37:32 +0000496 valueNumbering.erase(V);
Owen Anderson91c54952007-06-19 05:37:32 +0000497}
498
Owen Anderson28a2d442007-06-22 00:20:30 +0000499/// size - Return the number of assigned value numbers
500unsigned ValueTable::size() {
501 // NOTE: zero is never assigned
502 return nextValueNumber;
503}
504
Owen Andersonfd5683a2007-06-21 00:19:05 +0000505//===----------------------------------------------------------------------===//
506// GVNPRE Pass
507//===----------------------------------------------------------------------===//
508
Owen Anderson5fba6c12007-05-29 21:53:49 +0000509namespace {
510
511 class VISIBILITY_HIDDEN GVNPRE : public FunctionPass {
512 bool runOnFunction(Function &F);
513 public:
514 static char ID; // Pass identification, replacement for typeid
Owen Andersonb9cbaed2007-06-19 04:32:55 +0000515 GVNPRE() : FunctionPass((intptr_t)&ID) { }
Owen Anderson5fba6c12007-05-29 21:53:49 +0000516
517 private:
Owen Andersonbe802402007-06-08 01:03:01 +0000518 ValueTable VN;
Owen Andersona75dd4d2007-06-12 00:50:47 +0000519 std::vector<Instruction*> createdExpressions;
Owen Anderson81d156e2007-05-31 00:42:15 +0000520
Owen Andersone02da552007-06-28 00:34:34 +0000521 std::map<BasicBlock*, SmallPtrSet<Value*, 16> > availableOut;
522 std::map<BasicBlock*, SmallPtrSet<Value*, 16> > anticipatedIn;
Owen Anderson4036ad42007-06-12 22:43:57 +0000523
Owen Andersonfd5683a2007-06-21 00:19:05 +0000524 // This transformation requires dominator postdominator info
Owen Anderson5fba6c12007-05-29 21:53:49 +0000525 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
526 AU.setPreservesCFG();
527 AU.addRequired<DominatorTree>();
Owen Anderson5fba6c12007-05-29 21:53:49 +0000528 }
529
530 // Helper fuctions
531 // FIXME: eliminate or document these better
Owen Andersone02da552007-06-28 00:34:34 +0000532 void dump(const SmallPtrSet<Value*, 16>& s) const;
533 void clean(SmallPtrSet<Value*, 16>& set, BitVector& presentInSet);
534 Value* find_leader(SmallPtrSet<Value*, 16>& vals,
Owen Anderson92c7b222007-06-22 17:04:40 +0000535 uint32_t v);
536 Value* phi_translate(Value* V, BasicBlock* pred, BasicBlock* succ);
Owen Andersone02da552007-06-28 00:34:34 +0000537 void phi_translate_set(SmallPtrSet<Value*, 16>& anticIn, BasicBlock* pred,
538 BasicBlock* succ, SmallPtrSet<Value*, 16>& out);
Owen Anderson5fba6c12007-05-29 21:53:49 +0000539
Owen Andersone02da552007-06-28 00:34:34 +0000540 void topo_sort(SmallPtrSet<Value*, 16>& set,
Owen Anderson92c7b222007-06-22 17:04:40 +0000541 std::vector<Value*>& vec);
Owen Anderson331bf6a2007-05-31 22:44:11 +0000542
Owen Anderson92c7b222007-06-22 17:04:40 +0000543 void cleanup();
544 bool elimination();
Owen Andersondd998e12007-06-18 04:42:29 +0000545
Owen Andersone02da552007-06-28 00:34:34 +0000546 void val_insert(SmallPtrSet<Value*, 16>& s, Value* v);
547 void val_replace(SmallPtrSet<Value*, 16>& s, Value* v);
Owen Anderson92c7b222007-06-22 17:04:40 +0000548 bool dependsOnInvoke(Value* V);
Owen Anderson06c1e582007-06-20 22:10:02 +0000549 void buildsets_availout(BasicBlock::iterator I,
Owen Andersone02da552007-06-28 00:34:34 +0000550 SmallPtrSet<Value*, 16>& currAvail,
551 SmallPtrSet<PHINode*, 16>& currPhis,
552 SmallPtrSet<Value*, 16>& currExps,
553 SmallPtrSet<Value*, 16>& currTemps,
Owen Andersonf6e21872007-06-22 03:14:03 +0000554 BitVector& availNumbers,
Owen Anderson92c7b222007-06-22 17:04:40 +0000555 BitVector& expNumbers);
Owen Anderson28a2d442007-06-22 00:20:30 +0000556 bool buildsets_anticout(BasicBlock* BB,
Owen Andersone02da552007-06-28 00:34:34 +0000557 SmallPtrSet<Value*, 16>& anticOut,
Owen Anderson92c7b222007-06-22 17:04:40 +0000558 std::set<BasicBlock*>& visited);
Owen Anderson28a2d442007-06-22 00:20:30 +0000559 unsigned buildsets_anticin(BasicBlock* BB,
Owen Andersone02da552007-06-28 00:34:34 +0000560 SmallPtrSet<Value*, 16>& anticOut,
561 SmallPtrSet<Value*, 16>& currExps,
562 SmallPtrSet<Value*, 16>& currTemps,
Owen Anderson92c7b222007-06-22 17:04:40 +0000563 std::set<BasicBlock*>& visited);
Owen Anderson5477c542007-06-26 23:29:41 +0000564 void buildsets(Function& F);
Owen Anderson06c1e582007-06-20 22:10:02 +0000565
566 void insertion_pre(Value* e, BasicBlock* BB,
567 std::map<BasicBlock*, Value*>& avail,
Owen Andersone02da552007-06-28 00:34:34 +0000568 SmallPtrSet<Value*, 16>& new_set);
Owen Anderson06c1e582007-06-20 22:10:02 +0000569 unsigned insertion_mergepoint(std::vector<Value*>& workList,
Owen Anderson28a2d442007-06-22 00:20:30 +0000570 df_iterator<DomTreeNode*>& D,
Owen Andersone02da552007-06-28 00:34:34 +0000571 SmallPtrSet<Value*, 16>& new_set);
Owen Anderson92c7b222007-06-22 17:04:40 +0000572 bool insertion(Function& F);
Owen Anderson5fba6c12007-05-29 21:53:49 +0000573
574 };
575
576 char GVNPRE::ID = 0;
577
578}
579
Owen Andersonfd5683a2007-06-21 00:19:05 +0000580// createGVNPREPass - The public interface to this file...
Owen Anderson5fba6c12007-05-29 21:53:49 +0000581FunctionPass *llvm::createGVNPREPass() { return new GVNPRE(); }
582
583RegisterPass<GVNPRE> X("gvnpre",
584 "Global Value Numbering/Partial Redundancy Elimination");
585
Owen Anderson5fba6c12007-05-29 21:53:49 +0000586
Owen Anderson7d76b2a2007-06-08 22:02:36 +0000587STATISTIC(NumInsertedVals, "Number of values inserted");
588STATISTIC(NumInsertedPhis, "Number of PHI nodes inserted");
589STATISTIC(NumEliminated, "Number of redundant instructions eliminated");
590
Owen Andersonfd5683a2007-06-21 00:19:05 +0000591/// find_leader - Given a set and a value number, return the first
592/// element of the set with that value number, or 0 if no such element
593/// is present
Owen Andersone02da552007-06-28 00:34:34 +0000594Value* GVNPRE::find_leader(SmallPtrSet<Value*, 16>& vals, uint32_t v) {
595 for (SmallPtrSet<Value*, 16>::iterator I = vals.begin(), E = vals.end();
Owen Andersonb56fba02007-06-19 03:31:41 +0000596 I != E; ++I)
597 if (v == VN.lookup(*I))
Owen Anderson0b68cda2007-06-03 05:55:58 +0000598 return *I;
Owen Anderson5fba6c12007-05-29 21:53:49 +0000599
Owen Anderson0b68cda2007-06-03 05:55:58 +0000600 return 0;
Owen Anderson5fba6c12007-05-29 21:53:49 +0000601}
602
Owen Andersonfd5683a2007-06-21 00:19:05 +0000603/// val_insert - Insert a value into a set only if there is not a value
604/// with the same value number already in the set
Owen Andersone02da552007-06-28 00:34:34 +0000605void GVNPRE::val_insert(SmallPtrSet<Value*, 16>& s, Value* v) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000606 uint32_t num = VN.lookup(v);
607 Value* leader = find_leader(s, num);
608 if (leader == 0)
609 s.insert(v);
610}
611
Owen Andersonfd5683a2007-06-21 00:19:05 +0000612/// val_replace - Insert a value into a set, replacing any values already in
613/// the set that have the same value number
Owen Andersone02da552007-06-28 00:34:34 +0000614void GVNPRE::val_replace(SmallPtrSet<Value*, 16>& s, Value* v) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000615 uint32_t num = VN.lookup(v);
616 Value* leader = find_leader(s, num);
617 while (leader != 0) {
618 s.erase(leader);
619 leader = find_leader(s, num);
620 }
621 s.insert(v);
622}
623
Owen Andersonfd5683a2007-06-21 00:19:05 +0000624/// phi_translate - Given a value, its parent block, and a predecessor of its
625/// parent, translate the value into legal for the predecessor block. This
626/// means translating its operands (and recursively, their operands) through
627/// any phi nodes in the parent into values available in the predecessor
Owen Anderson4036ad42007-06-12 22:43:57 +0000628Value* GVNPRE::phi_translate(Value* V, BasicBlock* pred, BasicBlock* succ) {
Owen Anderson38b6b222007-06-04 18:05:26 +0000629 if (V == 0)
630 return 0;
631
Owen Anderson59bd0532007-07-03 18:37:08 +0000632 // Unary Operations
Owen Anderson6b958c72007-07-03 19:01:42 +0000633 if (CastInst* U = dyn_cast<CastInst>(V)) {
Owen Anderson59bd0532007-07-03 18:37:08 +0000634 Value* newOp1 = 0;
635 if (isa<Instruction>(U->getOperand(0)))
636 newOp1 = phi_translate(U->getOperand(0), pred, succ);
637 else
638 newOp1 = U->getOperand(0);
639
640 if (newOp1 == 0)
641 return 0;
642
643 if (newOp1 != U->getOperand(0)) {
644 Instruction* newVal = 0;
645 if (CastInst* C = dyn_cast<CastInst>(U))
646 newVal = CastInst::create(C->getOpcode(),
647 newOp1, C->getType(),
648 C->getName()+".expr");
649
650 uint32_t v = VN.lookup_or_add(newVal);
651
652 Value* leader = find_leader(availableOut[pred], v);
653 if (leader == 0) {
654 createdExpressions.push_back(newVal);
655 return newVal;
656 } else {
657 VN.erase(newVal);
658 delete newVal;
659 return leader;
660 }
661 }
662
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000663 // Binary Operations
Owen Anderson59bd0532007-07-03 18:37:08 +0000664 } if (isa<BinaryOperator>(V) || isa<CmpInst>(V) ||
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000665 isa<ExtractElementInst>(V)) {
Owen Anderson191eb062007-06-25 05:41:12 +0000666 User* U = cast<User>(V);
667
Owen Andersonb56fba02007-06-19 03:31:41 +0000668 Value* newOp1 = 0;
Owen Anderson191eb062007-06-25 05:41:12 +0000669 if (isa<Instruction>(U->getOperand(0)))
670 newOp1 = phi_translate(U->getOperand(0), pred, succ);
Owen Andersonb56fba02007-06-19 03:31:41 +0000671 else
Owen Anderson191eb062007-06-25 05:41:12 +0000672 newOp1 = U->getOperand(0);
Owen Andersonb56fba02007-06-19 03:31:41 +0000673
Owen Anderson38b6b222007-06-04 18:05:26 +0000674 if (newOp1 == 0)
675 return 0;
676
Owen Andersonb56fba02007-06-19 03:31:41 +0000677 Value* newOp2 = 0;
Owen Anderson191eb062007-06-25 05:41:12 +0000678 if (isa<Instruction>(U->getOperand(1)))
679 newOp2 = phi_translate(U->getOperand(1), pred, succ);
Owen Andersonb56fba02007-06-19 03:31:41 +0000680 else
Owen Anderson191eb062007-06-25 05:41:12 +0000681 newOp2 = U->getOperand(1);
Owen Andersonb56fba02007-06-19 03:31:41 +0000682
Owen Anderson38b6b222007-06-04 18:05:26 +0000683 if (newOp2 == 0)
684 return 0;
685
Owen Anderson191eb062007-06-25 05:41:12 +0000686 if (newOp1 != U->getOperand(0) || newOp2 != U->getOperand(1)) {
687 Instruction* newVal = 0;
688 if (BinaryOperator* BO = dyn_cast<BinaryOperator>(U))
689 newVal = BinaryOperator::create(BO->getOpcode(),
690 newOp1, newOp2,
691 BO->getName()+".expr");
692 else if (CmpInst* C = dyn_cast<CmpInst>(U))
693 newVal = CmpInst::create(C->getOpcode(),
694 C->getPredicate(),
695 newOp1, newOp2,
696 C->getName()+".expr");
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000697 else if (ExtractElementInst* E = dyn_cast<ExtractElementInst>(U))
698 newVal = new ExtractElementInst(newOp1, newOp2, E->getName()+".expr");
Owen Anderson55994f22007-06-08 20:44:02 +0000699
Owen Andersonb56fba02007-06-19 03:31:41 +0000700 uint32_t v = VN.lookup_or_add(newVal);
Owen Anderson4036ad42007-06-12 22:43:57 +0000701
Owen Andersonb56fba02007-06-19 03:31:41 +0000702 Value* leader = find_leader(availableOut[pred], v);
Owen Anderson4036ad42007-06-12 22:43:57 +0000703 if (leader == 0) {
Owen Andersona75dd4d2007-06-12 00:50:47 +0000704 createdExpressions.push_back(newVal);
Owen Anderson38b6b222007-06-04 18:05:26 +0000705 return newVal;
Owen Andersonc8472092007-06-05 22:11:49 +0000706 } else {
Owen Anderson91c54952007-06-19 05:37:32 +0000707 VN.erase(newVal);
Owen Andersonc8472092007-06-05 22:11:49 +0000708 delete newVal;
Owen Anderson4036ad42007-06-12 22:43:57 +0000709 return leader;
Owen Andersonc8472092007-06-05 22:11:49 +0000710 }
Owen Anderson38b6b222007-06-04 18:05:26 +0000711 }
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000712
713 // Ternary Operations
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000714 } else if (isa<ShuffleVectorInst>(V) || isa<InsertElementInst>(V) ||
715 isa<SelectInst>(V)) {
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000716 User* U = cast<User>(V);
717
718 Value* newOp1 = 0;
719 if (isa<Instruction>(U->getOperand(0)))
720 newOp1 = phi_translate(U->getOperand(0), pred, succ);
721 else
722 newOp1 = U->getOperand(0);
723
724 if (newOp1 == 0)
725 return 0;
726
727 Value* newOp2 = 0;
728 if (isa<Instruction>(U->getOperand(1)))
729 newOp2 = phi_translate(U->getOperand(1), pred, succ);
730 else
731 newOp2 = U->getOperand(1);
732
733 if (newOp2 == 0)
734 return 0;
735
736 Value* newOp3 = 0;
737 if (isa<Instruction>(U->getOperand(2)))
738 newOp3 = phi_translate(U->getOperand(2), pred, succ);
739 else
740 newOp3 = U->getOperand(2);
741
742 if (newOp3 == 0)
743 return 0;
744
745 if (newOp1 != U->getOperand(0) ||
746 newOp2 != U->getOperand(1) ||
747 newOp3 != U->getOperand(2)) {
748 Instruction* newVal = 0;
749 if (ShuffleVectorInst* S = dyn_cast<ShuffleVectorInst>(U))
750 newVal = new ShuffleVectorInst(newOp1, newOp2, newOp3,
751 S->getName()+".expr");
752 else if (InsertElementInst* I = dyn_cast<InsertElementInst>(U))
753 newVal = new InsertElementInst(newOp1, newOp2, newOp3,
754 I->getName()+".expr");
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000755 else if (SelectInst* I = dyn_cast<SelectInst>(U))
756 newVal = new SelectInst(newOp1, newOp2, newOp3, I->getName()+".expr");
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000757
758 uint32_t v = VN.lookup_or_add(newVal);
759
760 Value* leader = find_leader(availableOut[pred], v);
761 if (leader == 0) {
762 createdExpressions.push_back(newVal);
763 return newVal;
764 } else {
765 VN.erase(newVal);
766 delete newVal;
767 return leader;
768 }
769 }
770
771 // PHI Nodes
Owen Anderson38b6b222007-06-04 18:05:26 +0000772 } else if (PHINode* P = dyn_cast<PHINode>(V)) {
Owen Andersona75dd4d2007-06-12 00:50:47 +0000773 if (P->getParent() == succ)
Owen Anderson38b6b222007-06-04 18:05:26 +0000774 return P->getIncomingValueForBlock(pred);
775 }
776
777 return V;
778}
779
Owen Andersonfd5683a2007-06-21 00:19:05 +0000780/// phi_translate_set - Perform phi translation on every element of a set
Owen Andersone02da552007-06-28 00:34:34 +0000781void GVNPRE::phi_translate_set(SmallPtrSet<Value*, 16>& anticIn,
Owen Andersona75dd4d2007-06-12 00:50:47 +0000782 BasicBlock* pred, BasicBlock* succ,
Owen Andersone02da552007-06-28 00:34:34 +0000783 SmallPtrSet<Value*, 16>& out) {
784 for (SmallPtrSet<Value*, 16>::iterator I = anticIn.begin(),
Owen Anderson38b6b222007-06-04 18:05:26 +0000785 E = anticIn.end(); I != E; ++I) {
Owen Anderson4036ad42007-06-12 22:43:57 +0000786 Value* V = phi_translate(*I, pred, succ);
Owen Anderson38b6b222007-06-04 18:05:26 +0000787 if (V != 0)
788 out.insert(V);
Owen Anderson5fba6c12007-05-29 21:53:49 +0000789 }
790}
791
Owen Andersonfd5683a2007-06-21 00:19:05 +0000792/// dependsOnInvoke - Test if a value has an phi node as an operand, any of
793/// whose inputs is an invoke instruction. If this is true, we cannot safely
794/// PRE the instruction or anything that depends on it.
Owen Andersondd998e12007-06-18 04:42:29 +0000795bool GVNPRE::dependsOnInvoke(Value* V) {
Owen Anderson2320d432007-06-19 23:07:16 +0000796 if (PHINode* p = dyn_cast<PHINode>(V)) {
797 for (PHINode::op_iterator I = p->op_begin(), E = p->op_end(); I != E; ++I)
798 if (isa<InvokeInst>(*I))
Owen Andersondd998e12007-06-18 04:42:29 +0000799 return true;
Owen Anderson2320d432007-06-19 23:07:16 +0000800 return false;
801 } else {
802 return false;
Owen Anderson658f2c42007-06-16 00:26:54 +0000803 }
Owen Anderson658f2c42007-06-16 00:26:54 +0000804}
805
Owen Andersonfd5683a2007-06-21 00:19:05 +0000806/// clean - Remove all non-opaque values from the set whose operands are not
807/// themselves in the set, as well as all values that depend on invokes (see
808/// above)
Owen Andersone02da552007-06-28 00:34:34 +0000809void GVNPRE::clean(SmallPtrSet<Value*, 16>& set, BitVector& presentInSet) {
Owen Anderson0b68cda2007-06-03 05:55:58 +0000810 std::vector<Value*> worklist;
Owen Andersond50a29d2007-06-22 00:43:22 +0000811 worklist.reserve(set.size());
Owen Andersonbe802402007-06-08 01:03:01 +0000812 topo_sort(set, worklist);
Owen Anderson5fba6c12007-05-29 21:53:49 +0000813
Owen Andersonbe802402007-06-08 01:03:01 +0000814 for (unsigned i = 0; i < worklist.size(); ++i) {
815 Value* v = worklist[i];
Owen Anderson5fba6c12007-05-29 21:53:49 +0000816
Owen Anderson59bd0532007-07-03 18:37:08 +0000817 // Handle unary ops
Owen Anderson6b958c72007-07-03 19:01:42 +0000818 if (CastInst* U = dyn_cast<CastInst>(v)) {
Owen Anderson59bd0532007-07-03 18:37:08 +0000819 bool lhsValid = !isa<Instruction>(U->getOperand(0));
820 lhsValid |= presentInSet.test(VN.lookup(U->getOperand(0)));
821 if (lhsValid)
822 lhsValid = !dependsOnInvoke(U->getOperand(0));
823
824 if (!lhsValid) {
825 set.erase(U);
826 presentInSet.flip(VN.lookup(U));
827 }
828
Owen Anderson0eb26572007-06-27 17:03:03 +0000829 // Handle binary ops
Owen Anderson59bd0532007-07-03 18:37:08 +0000830 } else if (isa<BinaryOperator>(v) || isa<CmpInst>(v) ||
Owen Anderson0eb26572007-06-27 17:03:03 +0000831 isa<ExtractElementInst>(v)) {
832 User* U = cast<User>(v);
833
834 bool lhsValid = !isa<Instruction>(U->getOperand(0));
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000835 lhsValid |= presentInSet.test(VN.lookup(U->getOperand(0)));
Owen Andersonb364b412007-06-18 04:30:44 +0000836 if (lhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000837 lhsValid = !dependsOnInvoke(U->getOperand(0));
Owen Anderson0b68cda2007-06-03 05:55:58 +0000838
Owen Anderson0eb26572007-06-27 17:03:03 +0000839 bool rhsValid = !isa<Instruction>(U->getOperand(1));
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000840 rhsValid |= presentInSet.test(VN.lookup(U->getOperand(1)));
Owen Andersonb364b412007-06-18 04:30:44 +0000841 if (rhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000842 rhsValid = !dependsOnInvoke(U->getOperand(1));
Owen Anderson5fba6c12007-05-29 21:53:49 +0000843
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000844 if (!lhsValid || !rhsValid) {
Owen Anderson0eb26572007-06-27 17:03:03 +0000845 set.erase(U);
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000846 presentInSet.flip(VN.lookup(U));
847 }
Owen Anderson223718c2007-06-09 18:35:31 +0000848
Owen Anderson0eb26572007-06-27 17:03:03 +0000849 // Handle ternary ops
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000850 } else if (isa<ShuffleVectorInst>(v) || isa<InsertElementInst>(v) ||
851 isa<SelectInst>(v)) {
Owen Anderson0eb26572007-06-27 17:03:03 +0000852 User* U = cast<User>(v);
853
854 bool lhsValid = !isa<Instruction>(U->getOperand(0));
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000855 lhsValid |= presentInSet.test(VN.lookup(U->getOperand(0)));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000856 if (lhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000857 lhsValid = !dependsOnInvoke(U->getOperand(0));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000858
Owen Anderson0eb26572007-06-27 17:03:03 +0000859 bool rhsValid = !isa<Instruction>(U->getOperand(1));
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000860 rhsValid |= presentInSet.test(VN.lookup(U->getOperand(1)));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000861 if (rhsValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000862 rhsValid = !dependsOnInvoke(U->getOperand(1));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000863
Owen Anderson0eb26572007-06-27 17:03:03 +0000864 bool thirdValid = !isa<Instruction>(U->getOperand(2));
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000865 thirdValid |= presentInSet.test(VN.lookup(U->getOperand(2)));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000866 if (thirdValid)
Owen Anderson0eb26572007-06-27 17:03:03 +0000867 thirdValid = !dependsOnInvoke(U->getOperand(2));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000868
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000869 if (!lhsValid || !rhsValid || !thirdValid) {
Owen Anderson0eb26572007-06-27 17:03:03 +0000870 set.erase(U);
Owen Anderson7dae8ef2007-06-27 17:38:29 +0000871 presentInSet.flip(VN.lookup(U));
872 }
Owen Anderson0b68cda2007-06-03 05:55:58 +0000873 }
Owen Anderson5fba6c12007-05-29 21:53:49 +0000874 }
875}
876
Owen Andersonfd5683a2007-06-21 00:19:05 +0000877/// topo_sort - Given a set of values, sort them by topological
878/// order into the provided vector.
Owen Andersone02da552007-06-28 00:34:34 +0000879void GVNPRE::topo_sort(SmallPtrSet<Value*, 16>& set, std::vector<Value*>& vec) {
880 SmallPtrSet<Value*, 16> visited;
Owen Anderson49409f62007-06-22 21:31:16 +0000881 std::vector<Value*> stack;
Owen Andersone02da552007-06-28 00:34:34 +0000882 for (SmallPtrSet<Value*, 16>::iterator I = set.begin(), E = set.end();
Owen Anderson49409f62007-06-22 21:31:16 +0000883 I != E; ++I) {
884 if (visited.count(*I) == 0)
885 stack.push_back(*I);
886
887 while (!stack.empty()) {
888 Value* e = stack.back();
Owen Anderson59bd0532007-07-03 18:37:08 +0000889
890 // Handle unary ops
Owen Anderson6b958c72007-07-03 19:01:42 +0000891 if (CastInst* U = dyn_cast<CastInst>(e)) {
Owen Anderson59bd0532007-07-03 18:37:08 +0000892 Value* l = find_leader(set, VN.lookup(U->getOperand(0)));
893
894 if (l != 0 && isa<Instruction>(l) &&
895 visited.count(l) == 0)
896 stack.push_back(l);
897 else {
898 vec.push_back(e);
899 visited.insert(e);
900 stack.pop_back();
901 }
902
Owen Anderson0eb26572007-06-27 17:03:03 +0000903 // Handle binary ops
Owen Anderson59bd0532007-07-03 18:37:08 +0000904 } else if (isa<BinaryOperator>(e) || isa<CmpInst>(e) ||
Owen Anderson0eb26572007-06-27 17:03:03 +0000905 isa<ExtractElementInst>(e)) {
906 User* U = cast<User>(e);
907 Value* l = find_leader(set, VN.lookup(U->getOperand(0)));
908 Value* r = find_leader(set, VN.lookup(U->getOperand(1)));
Owen Anderson49409f62007-06-22 21:31:16 +0000909
910 if (l != 0 && isa<Instruction>(l) &&
911 visited.count(l) == 0)
912 stack.push_back(l);
913 else if (r != 0 && isa<Instruction>(r) &&
914 visited.count(r) == 0)
915 stack.push_back(r);
916 else {
917 vec.push_back(e);
918 visited.insert(e);
919 stack.pop_back();
920 }
Owen Anderson0b68cda2007-06-03 05:55:58 +0000921
Owen Anderson0eb26572007-06-27 17:03:03 +0000922 // Handle ternary ops
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000923 } else if (isa<InsertElementInst>(e) || isa<ShuffleVectorInst>(e) ||
924 isa<SelectInst>(e)) {
Owen Anderson0eb26572007-06-27 17:03:03 +0000925 User* U = cast<User>(e);
926 Value* l = find_leader(set, VN.lookup(U->getOperand(0)));
927 Value* r = find_leader(set, VN.lookup(U->getOperand(1)));
928 Value* m = find_leader(set, VN.lookup(U->getOperand(2)));
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000929
930 if (l != 0 && isa<Instruction>(l) &&
931 visited.count(l) == 0)
932 stack.push_back(l);
933 else if (r != 0 && isa<Instruction>(r) &&
934 visited.count(r) == 0)
935 stack.push_back(r);
936 else if (m != 0 && isa<Instruction>(m) &&
937 visited.count(m) == 0)
938 stack.push_back(r);
939 else {
940 vec.push_back(e);
941 visited.insert(e);
942 stack.pop_back();
943 }
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000944
Owen Anderson0eb26572007-06-27 17:03:03 +0000945 // Handle opaque ops
Owen Anderson49409f62007-06-22 21:31:16 +0000946 } else {
Owen Anderson0b68cda2007-06-03 05:55:58 +0000947 visited.insert(e);
Owen Anderson223718c2007-06-09 18:35:31 +0000948 vec.push_back(e);
Owen Anderson49409f62007-06-22 21:31:16 +0000949 stack.pop_back();
Owen Anderson223718c2007-06-09 18:35:31 +0000950 }
Owen Anderson331bf6a2007-05-31 22:44:11 +0000951 }
Owen Anderson49409f62007-06-22 21:31:16 +0000952
953 stack.clear();
Owen Anderson331bf6a2007-05-31 22:44:11 +0000954 }
955}
956
Owen Andersonfd5683a2007-06-21 00:19:05 +0000957/// dump - Dump a set of values to standard error
Owen Andersone02da552007-06-28 00:34:34 +0000958void GVNPRE::dump(const SmallPtrSet<Value*, 16>& s) const {
Owen Anderson4a6ec8f2007-05-29 23:15:21 +0000959 DOUT << "{ ";
Owen Andersone02da552007-06-28 00:34:34 +0000960 for (SmallPtrSet<Value*, 16>::iterator I = s.begin(), E = s.end();
Owen Anderson0eca9aa2007-06-03 22:02:14 +0000961 I != E; ++I) {
Owen Anderson8a9fa5d2007-06-28 23:51:21 +0000962 DOUT << "" << VN.lookup(*I) << ": ";
Owen Anderson0eca9aa2007-06-03 22:02:14 +0000963 DEBUG((*I)->dump());
964 }
965 DOUT << "}\n\n";
966}
967
Owen Andersonfd5683a2007-06-21 00:19:05 +0000968/// elimination - Phase 3 of the main algorithm. Perform full redundancy
969/// elimination by walking the dominator tree and removing any instruction that
970/// is dominated by another instruction with the same value number.
Owen Anderson06c1e582007-06-20 22:10:02 +0000971bool GVNPRE::elimination() {
Owen Anderson42769842007-06-12 16:57:50 +0000972 DOUT << "\n\nPhase 3: Elimination\n\n";
973
Owen Anderson06c1e582007-06-20 22:10:02 +0000974 bool changed_function = false;
975
Owen Anderson42769842007-06-12 16:57:50 +0000976 std::vector<std::pair<Instruction*, Value*> > replace;
977 std::vector<Instruction*> erase;
978
979 DominatorTree& DT = getAnalysis<DominatorTree>();
980
981 for (df_iterator<DomTreeNode*> DI = df_begin(DT.getRootNode()),
982 E = df_end(DT.getRootNode()); DI != E; ++DI) {
983 BasicBlock* BB = DI->getBlock();
984
Owen Anderson191eb062007-06-25 05:41:12 +0000985 //DOUT << "Block: " << BB->getName() << "\n";
986 //dump(availableOut[BB]);
987 //DOUT << "\n\n";
Owen Anderson42769842007-06-12 16:57:50 +0000988
989 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
990 BI != BE; ++BI) {
991
Owen Andersonb6a39fc2007-06-27 04:10:46 +0000992 if (isa<BinaryOperator>(BI) || isa<CmpInst>(BI) ||
993 isa<ShuffleVectorInst>(BI) || isa<InsertElementInst>(BI) ||
Owen Anderson59bd0532007-07-03 18:37:08 +0000994 isa<ExtractElementInst>(BI) || isa<SelectInst>(BI) ||
995 isa<CastInst>(BI)) {
Owen Andersonb56fba02007-06-19 03:31:41 +0000996 Value *leader = find_leader(availableOut[BB], VN.lookup(BI));
Owen Anderson42769842007-06-12 16:57:50 +0000997
998 if (leader != 0)
999 if (Instruction* Instr = dyn_cast<Instruction>(leader))
1000 if (Instr->getParent() != 0 && Instr != BI) {
1001 replace.push_back(std::make_pair(BI, leader));
1002 erase.push_back(BI);
1003 ++NumEliminated;
1004 }
1005 }
1006 }
1007 }
1008
1009 while (!replace.empty()) {
1010 std::pair<Instruction*, Value*> rep = replace.back();
1011 replace.pop_back();
1012 rep.first->replaceAllUsesWith(rep.second);
Owen Andersonb0714bb2007-06-20 18:30:20 +00001013 changed_function = true;
Owen Anderson42769842007-06-12 16:57:50 +00001014 }
1015
1016 for (std::vector<Instruction*>::iterator I = erase.begin(), E = erase.end();
1017 I != E; ++I)
1018 (*I)->eraseFromParent();
Owen Anderson06c1e582007-06-20 22:10:02 +00001019
1020 return changed_function;
Owen Anderson42769842007-06-12 16:57:50 +00001021}
1022
Owen Andersonfd5683a2007-06-21 00:19:05 +00001023/// cleanup - Delete any extraneous values that were created to represent
1024/// expressions without leaders.
Owen Anderson42769842007-06-12 16:57:50 +00001025void GVNPRE::cleanup() {
1026 while (!createdExpressions.empty()) {
1027 Instruction* I = createdExpressions.back();
1028 createdExpressions.pop_back();
1029
1030 delete I;
1031 }
1032}
1033
Owen Andersonfd5683a2007-06-21 00:19:05 +00001034/// buildsets_availout - When calculating availability, handle an instruction
1035/// by inserting it into the appropriate sets
Owen Anderson06c1e582007-06-20 22:10:02 +00001036void GVNPRE::buildsets_availout(BasicBlock::iterator I,
Owen Andersone02da552007-06-28 00:34:34 +00001037 SmallPtrSet<Value*, 16>& currAvail,
1038 SmallPtrSet<PHINode*, 16>& currPhis,
1039 SmallPtrSet<Value*, 16>& currExps,
1040 SmallPtrSet<Value*, 16>& currTemps,
Owen Andersonf6e21872007-06-22 03:14:03 +00001041 BitVector& availNumbers,
1042 BitVector& expNumbers) {
Owen Anderson0eb26572007-06-27 17:03:03 +00001043 // Handle PHI nodes
Owen Anderson06c1e582007-06-20 22:10:02 +00001044 if (PHINode* p = dyn_cast<PHINode>(I)) {
1045 VN.lookup_or_add(p);
Owen Andersonf6e21872007-06-22 03:14:03 +00001046 expNumbers.resize(VN.size());
1047 availNumbers.resize(VN.size());
1048
Owen Anderson06c1e582007-06-20 22:10:02 +00001049 currPhis.insert(p);
Owen Anderson59bd0532007-07-03 18:37:08 +00001050
1051 // Handle unary ops
Owen Anderson6b958c72007-07-03 19:01:42 +00001052 } else if (CastInst* U = dyn_cast<CastInst>(I)) {
Owen Anderson59bd0532007-07-03 18:37:08 +00001053 Value* leftValue = U->getOperand(0);
Owen Anderson06c1e582007-06-20 22:10:02 +00001054
Owen Anderson59bd0532007-07-03 18:37:08 +00001055 unsigned num = VN.lookup_or_add(U);
1056 expNumbers.resize(VN.size());
1057 availNumbers.resize(VN.size());
1058
1059 if (isa<Instruction>(leftValue))
1060 if (!expNumbers.test(VN.lookup(leftValue))) {
1061 currExps.insert(leftValue);
1062 expNumbers.set(VN.lookup(leftValue));
1063 }
1064
1065 if (!expNumbers.test(VN.lookup(U))) {
1066 currExps.insert(U);
1067 expNumbers.set(num);
1068 }
1069
Owen Anderson0eb26572007-06-27 17:03:03 +00001070 // Handle binary ops
1071 } else if (isa<BinaryOperator>(I) || isa<CmpInst>(I) ||
1072 isa<ExtractElementInst>(I)) {
1073 User* U = cast<User>(I);
1074 Value* leftValue = U->getOperand(0);
1075 Value* rightValue = U->getOperand(1);
Owen Anderson06c1e582007-06-20 22:10:02 +00001076
Owen Anderson0eb26572007-06-27 17:03:03 +00001077 unsigned num = VN.lookup_or_add(U);
Owen Andersonf6e21872007-06-22 03:14:03 +00001078 expNumbers.resize(VN.size());
1079 availNumbers.resize(VN.size());
Owen Anderson06c1e582007-06-20 22:10:02 +00001080
1081 if (isa<Instruction>(leftValue))
Owen Anderson21a11312007-06-22 18:27:04 +00001082 if (!expNumbers.test(VN.lookup(leftValue))) {
Owen Andersonf6e21872007-06-22 03:14:03 +00001083 currExps.insert(leftValue);
Owen Anderson21a11312007-06-22 18:27:04 +00001084 expNumbers.set(VN.lookup(leftValue));
Owen Andersonf6e21872007-06-22 03:14:03 +00001085 }
1086
Owen Anderson06c1e582007-06-20 22:10:02 +00001087 if (isa<Instruction>(rightValue))
Owen Anderson21a11312007-06-22 18:27:04 +00001088 if (!expNumbers.test(VN.lookup(rightValue))) {
Owen Andersonf6e21872007-06-22 03:14:03 +00001089 currExps.insert(rightValue);
Owen Anderson21a11312007-06-22 18:27:04 +00001090 expNumbers.set(VN.lookup(rightValue));
Owen Andersonf6e21872007-06-22 03:14:03 +00001091 }
1092
Owen Anderson0eb26572007-06-27 17:03:03 +00001093 if (!expNumbers.test(VN.lookup(U))) {
1094 currExps.insert(U);
Owen Anderson21a11312007-06-22 18:27:04 +00001095 expNumbers.set(num);
Owen Andersonf6e21872007-06-22 03:14:03 +00001096 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001097
Owen Anderson0eb26572007-06-27 17:03:03 +00001098 // Handle ternary ops
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001099 } else if (isa<InsertElementInst>(I) || isa<ShuffleVectorInst>(I) ||
1100 isa<SelectInst>(I)) {
Owen Anderson0eb26572007-06-27 17:03:03 +00001101 User* U = cast<User>(I);
1102 Value* leftValue = U->getOperand(0);
1103 Value* rightValue = U->getOperand(1);
1104 Value* thirdValue = U->getOperand(2);
Owen Anderson06c1e582007-06-20 22:10:02 +00001105
Owen Anderson0eb26572007-06-27 17:03:03 +00001106 VN.lookup_or_add(U);
Owen Andersonf6e21872007-06-22 03:14:03 +00001107
Owen Anderson0eb26572007-06-27 17:03:03 +00001108 unsigned num = VN.lookup_or_add(U);
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001109 expNumbers.resize(VN.size());
1110 availNumbers.resize(VN.size());
1111
1112 if (isa<Instruction>(leftValue))
1113 if (!expNumbers.test(VN.lookup(leftValue))) {
1114 currExps.insert(leftValue);
1115 expNumbers.set(VN.lookup(leftValue));
1116 }
1117 if (isa<Instruction>(rightValue))
1118 if (!expNumbers.test(VN.lookup(rightValue))) {
1119 currExps.insert(rightValue);
1120 expNumbers.set(VN.lookup(rightValue));
1121 }
1122 if (isa<Instruction>(thirdValue))
1123 if (!expNumbers.test(VN.lookup(thirdValue))) {
1124 currExps.insert(thirdValue);
1125 expNumbers.set(VN.lookup(thirdValue));
1126 }
1127
Owen Anderson0eb26572007-06-27 17:03:03 +00001128 if (!expNumbers.test(VN.lookup(U))) {
1129 currExps.insert(U);
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001130 expNumbers.set(num);
1131 }
1132
Owen Anderson0eb26572007-06-27 17:03:03 +00001133 // Handle opaque ops
1134 } else if (!I->isTerminator()){
Owen Anderson06c1e582007-06-20 22:10:02 +00001135 VN.lookup_or_add(I);
Owen Andersonf6e21872007-06-22 03:14:03 +00001136 expNumbers.resize(VN.size());
1137 availNumbers.resize(VN.size());
1138
Owen Anderson06c1e582007-06-20 22:10:02 +00001139 currTemps.insert(I);
1140 }
1141
1142 if (!I->isTerminator())
Owen Anderson21a11312007-06-22 18:27:04 +00001143 if (!availNumbers.test(VN.lookup(I))) {
Owen Andersonf6e21872007-06-22 03:14:03 +00001144 currAvail.insert(I);
Owen Anderson21a11312007-06-22 18:27:04 +00001145 availNumbers.set(VN.lookup(I));
Owen Andersonf6e21872007-06-22 03:14:03 +00001146 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001147}
Owen Anderson5fba6c12007-05-29 21:53:49 +00001148
Owen Andersonfd5683a2007-06-21 00:19:05 +00001149/// buildsets_anticout - When walking the postdom tree, calculate the ANTIC_OUT
1150/// set as a function of the ANTIC_IN set of the block's predecessors
Owen Anderson28a2d442007-06-22 00:20:30 +00001151bool GVNPRE::buildsets_anticout(BasicBlock* BB,
Owen Andersone02da552007-06-28 00:34:34 +00001152 SmallPtrSet<Value*, 16>& anticOut,
Owen Anderson06c1e582007-06-20 22:10:02 +00001153 std::set<BasicBlock*>& visited) {
1154 if (BB->getTerminator()->getNumSuccessors() == 1) {
Owen Anderson191eb062007-06-25 05:41:12 +00001155 if (BB->getTerminator()->getSuccessor(0) != BB &&
1156 visited.count(BB->getTerminator()->getSuccessor(0)) == 0) {
1157 DOUT << "DEFER: " << BB->getName() << "\n";
Owen Anderson28a2d442007-06-22 00:20:30 +00001158 return true;
Owen Anderson191eb062007-06-25 05:41:12 +00001159 }
1160 else {
Owen Anderson06c1e582007-06-20 22:10:02 +00001161 phi_translate_set(anticipatedIn[BB->getTerminator()->getSuccessor(0)],
1162 BB, BB->getTerminator()->getSuccessor(0), anticOut);
Owen Anderson191eb062007-06-25 05:41:12 +00001163 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001164 } else if (BB->getTerminator()->getNumSuccessors() > 1) {
1165 BasicBlock* first = BB->getTerminator()->getSuccessor(0);
1166 anticOut.insert(anticipatedIn[first].begin(), anticipatedIn[first].end());
1167
1168 for (unsigned i = 1; i < BB->getTerminator()->getNumSuccessors(); ++i) {
1169 BasicBlock* currSucc = BB->getTerminator()->getSuccessor(i);
Owen Andersone02da552007-06-28 00:34:34 +00001170 SmallPtrSet<Value*, 16>& succAnticIn = anticipatedIn[currSucc];
Owen Anderson06c1e582007-06-20 22:10:02 +00001171
Owen Anderson2ff912b2007-06-21 17:57:53 +00001172 std::vector<Value*> temp;
1173
Owen Andersone02da552007-06-28 00:34:34 +00001174 for (SmallPtrSet<Value*, 16>::iterator I = anticOut.begin(),
Owen Anderson2ff912b2007-06-21 17:57:53 +00001175 E = anticOut.end(); I != E; ++I)
1176 if (succAnticIn.count(*I) == 0)
1177 temp.push_back(*I);
1178
1179 for (std::vector<Value*>::iterator I = temp.begin(), E = temp.end();
1180 I != E; ++I)
1181 anticOut.erase(*I);
Owen Anderson06c1e582007-06-20 22:10:02 +00001182 }
1183 }
Owen Anderson28a2d442007-06-22 00:20:30 +00001184
1185 return false;
Owen Anderson06c1e582007-06-20 22:10:02 +00001186}
1187
Owen Andersonfd5683a2007-06-21 00:19:05 +00001188/// buildsets_anticin - Walk the postdom tree, calculating ANTIC_OUT for
1189/// each block. ANTIC_IN is then a function of ANTIC_OUT and the GEN
1190/// sets populated in buildsets_availout
Owen Anderson28a2d442007-06-22 00:20:30 +00001191unsigned GVNPRE::buildsets_anticin(BasicBlock* BB,
Owen Andersone02da552007-06-28 00:34:34 +00001192 SmallPtrSet<Value*, 16>& anticOut,
1193 SmallPtrSet<Value*, 16>& currExps,
1194 SmallPtrSet<Value*, 16>& currTemps,
Owen Anderson06c1e582007-06-20 22:10:02 +00001195 std::set<BasicBlock*>& visited) {
Owen Andersone02da552007-06-28 00:34:34 +00001196 SmallPtrSet<Value*, 16>& anticIn = anticipatedIn[BB];
Owen Anderson21a11312007-06-22 18:27:04 +00001197 unsigned old = anticIn.size();
Owen Anderson06c1e582007-06-20 22:10:02 +00001198
Owen Anderson28a2d442007-06-22 00:20:30 +00001199 bool defer = buildsets_anticout(BB, anticOut, visited);
1200 if (defer)
1201 return 0;
Owen Anderson2ff912b2007-06-21 17:57:53 +00001202
Owen Anderson06c1e582007-06-20 22:10:02 +00001203 anticIn.clear();
Owen Anderson2ff912b2007-06-21 17:57:53 +00001204
Owen Anderson28a2d442007-06-22 00:20:30 +00001205 BitVector numbers(VN.size());
Owen Andersone02da552007-06-28 00:34:34 +00001206 for (SmallPtrSet<Value*, 16>::iterator I = anticOut.begin(),
Owen Anderson21a11312007-06-22 18:27:04 +00001207 E = anticOut.end(); I != E; ++I) {
Owen Anderson191eb062007-06-25 05:41:12 +00001208 unsigned num = VN.lookup_or_add(*I);
1209 numbers.resize(VN.size());
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001210
1211 if (isa<Instruction>(*I)) {
1212 anticIn.insert(*I);
1213 numbers.set(num);
1214 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001215 }
Owen Andersone02da552007-06-28 00:34:34 +00001216 for (SmallPtrSet<Value*, 16>::iterator I = currExps.begin(),
Owen Anderson21a11312007-06-22 18:27:04 +00001217 E = currExps.end(); I != E; ++I) {
1218 if (!numbers.test(VN.lookup_or_add(*I))) {
1219 anticIn.insert(*I);
1220 numbers.set(VN.lookup(*I));
1221 }
1222 }
1223
Owen Andersone02da552007-06-28 00:34:34 +00001224 for (SmallPtrSet<Value*, 16>::iterator I = currTemps.begin(),
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001225 E = currTemps.end(); I != E; ++I) {
Owen Anderson21a11312007-06-22 18:27:04 +00001226 anticIn.erase(*I);
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001227 numbers.flip(VN.lookup(*I));
1228 }
Owen Anderson21a11312007-06-22 18:27:04 +00001229
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001230 clean(anticIn, numbers);
1231 anticOut.clear();
Owen Anderson06c1e582007-06-20 22:10:02 +00001232
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001233 if (old != anticIn.size())
Owen Anderson28a2d442007-06-22 00:20:30 +00001234 return 2;
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001235 else
Owen Anderson28a2d442007-06-22 00:20:30 +00001236 return 1;
Owen Anderson06c1e582007-06-20 22:10:02 +00001237}
1238
Owen Andersonfd5683a2007-06-21 00:19:05 +00001239/// buildsets - Phase 1 of the main algorithm. Construct the AVAIL_OUT
1240/// and the ANTIC_IN sets.
Owen Anderson5477c542007-06-26 23:29:41 +00001241void GVNPRE::buildsets(Function& F) {
Owen Andersone02da552007-06-28 00:34:34 +00001242 std::map<BasicBlock*, SmallPtrSet<Value*, 16> > generatedExpressions;
1243 std::map<BasicBlock*, SmallPtrSet<PHINode*, 16> > generatedPhis;
1244 std::map<BasicBlock*, SmallPtrSet<Value*, 16> > generatedTemporaries;
Owen Anderson06c1e582007-06-20 22:10:02 +00001245
Owen Anderson5fba6c12007-05-29 21:53:49 +00001246 DominatorTree &DT = getAnalysis<DominatorTree>();
1247
Owen Anderson634a0632007-06-06 01:27:49 +00001248 // Phase 1, Part 1: calculate AVAIL_OUT
Owen Anderson5fba6c12007-05-29 21:53:49 +00001249
1250 // Top-down walk of the dominator tree
Devang Patelbdd1aae2007-06-04 00:32:22 +00001251 for (df_iterator<DomTreeNode*> DI = df_begin(DT.getRootNode()),
Owen Anderson5fba6c12007-05-29 21:53:49 +00001252 E = df_end(DT.getRootNode()); DI != E; ++DI) {
1253
1254 // Get the sets to update for this block
Owen Andersone02da552007-06-28 00:34:34 +00001255 SmallPtrSet<Value*, 16>& currExps = generatedExpressions[DI->getBlock()];
1256 SmallPtrSet<PHINode*, 16>& currPhis = generatedPhis[DI->getBlock()];
1257 SmallPtrSet<Value*, 16>& currTemps = generatedTemporaries[DI->getBlock()];
1258 SmallPtrSet<Value*, 16>& currAvail = availableOut[DI->getBlock()];
Owen Anderson5fba6c12007-05-29 21:53:49 +00001259
Owen Andersonb56fba02007-06-19 03:31:41 +00001260 BasicBlock* BB = DI->getBlock();
1261
1262 // A block inherits AVAIL_OUT from its dominator
1263 if (DI->getIDom() != 0)
1264 currAvail.insert(availableOut[DI->getIDom()->getBlock()].begin(),
1265 availableOut[DI->getIDom()->getBlock()].end());
1266
Owen Andersonf6e21872007-06-22 03:14:03 +00001267 BitVector availNumbers(VN.size());
Owen Andersone02da552007-06-28 00:34:34 +00001268 for (SmallPtrSet<Value*, 16>::iterator I = currAvail.begin(),
Owen Andersonf6e21872007-06-22 03:14:03 +00001269 E = currAvail.end(); I != E; ++I)
1270 availNumbers.set(VN.lookup(*I));
Owen Andersonb56fba02007-06-19 03:31:41 +00001271
Owen Andersonf6e21872007-06-22 03:14:03 +00001272 BitVector expNumbers(VN.size());
Owen Andersonb56fba02007-06-19 03:31:41 +00001273 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
Owen Anderson06c1e582007-06-20 22:10:02 +00001274 BI != BE; ++BI)
Owen Andersonf6e21872007-06-22 03:14:03 +00001275 buildsets_availout(BI, currAvail, currPhis, currExps,
1276 currTemps, availNumbers, expNumbers);
Owen Andersonb56fba02007-06-19 03:31:41 +00001277
Owen Anderson5fba6c12007-05-29 21:53:49 +00001278 }
Owen Anderson191eb062007-06-25 05:41:12 +00001279
1280 // Phase 1, Part 2: calculate ANTIC_IN
Owen Anderson5fba6c12007-05-29 21:53:49 +00001281
Owen Anderson0c423072007-05-29 23:26:30 +00001282 std::set<BasicBlock*> visited;
Owen Anderson5477c542007-06-26 23:29:41 +00001283 SmallPtrSet<BasicBlock*, 4> block_changed;
1284 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ++FI)
1285 block_changed.insert(FI);
Owen Anderson0c423072007-05-29 23:26:30 +00001286
Owen Anderson5fba6c12007-05-29 21:53:49 +00001287 bool changed = true;
1288 unsigned iterations = 0;
Owen Anderson5477c542007-06-26 23:29:41 +00001289
Owen Anderson5fba6c12007-05-29 21:53:49 +00001290 while (changed) {
1291 changed = false;
Owen Andersone02da552007-06-28 00:34:34 +00001292 SmallPtrSet<Value*, 16> anticOut;
Owen Anderson5fba6c12007-05-29 21:53:49 +00001293
Owen Anderson5477c542007-06-26 23:29:41 +00001294 // Postorder walk of the CFG
Owen Anderson43ca4b42007-06-25 18:25:31 +00001295 for (po_iterator<BasicBlock*> BBI = po_begin(&F.getEntryBlock()),
1296 BBE = po_end(&F.getEntryBlock()); BBI != BBE; ++BBI) {
Owen Anderson191eb062007-06-25 05:41:12 +00001297 BasicBlock* BB = *BBI;
Owen Andersond184c182007-06-11 16:25:17 +00001298
Owen Anderson5477c542007-06-26 23:29:41 +00001299 if (block_changed.count(BB) != 0) {
1300 unsigned ret = buildsets_anticin(BB, anticOut,generatedExpressions[BB],
1301 generatedTemporaries[BB], visited);
Owen Anderson28a2d442007-06-22 00:20:30 +00001302
Owen Anderson5477c542007-06-26 23:29:41 +00001303 if (ret == 0) {
1304 changed = true;
1305 continue;
1306 } else {
1307 visited.insert(BB);
1308
1309 if (ret == 2)
1310 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB);
1311 PI != PE; ++PI) {
1312 block_changed.insert(*PI);
1313 }
1314 else
1315 block_changed.erase(BB);
1316
1317 changed |= (ret == 2);
Owen Anderson191eb062007-06-25 05:41:12 +00001318 }
Owen Anderson28a2d442007-06-22 00:20:30 +00001319 }
Owen Anderson5fba6c12007-05-29 21:53:49 +00001320 }
Owen Anderson38b6b222007-06-04 18:05:26 +00001321
Owen Anderson5fba6c12007-05-29 21:53:49 +00001322 iterations++;
1323 }
1324
Owen Anderson191eb062007-06-25 05:41:12 +00001325 DOUT << "ITERATIONS: " << iterations << "\n";
Owen Anderson06c1e582007-06-20 22:10:02 +00001326}
1327
Owen Andersonfd5683a2007-06-21 00:19:05 +00001328/// insertion_pre - When a partial redundancy has been identified, eliminate it
1329/// by inserting appropriate values into the predecessors and a phi node in
1330/// the main block
Owen Anderson06c1e582007-06-20 22:10:02 +00001331void GVNPRE::insertion_pre(Value* e, BasicBlock* BB,
1332 std::map<BasicBlock*, Value*>& avail,
Owen Andersone02da552007-06-28 00:34:34 +00001333 SmallPtrSet<Value*, 16>& new_set) {
Owen Anderson06c1e582007-06-20 22:10:02 +00001334 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE; ++PI) {
1335 Value* e2 = avail[*PI];
1336 if (!find_leader(availableOut[*PI], VN.lookup(e2))) {
1337 User* U = cast<User>(e2);
1338
1339 Value* s1 = 0;
1340 if (isa<BinaryOperator>(U->getOperand(0)) ||
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001341 isa<CmpInst>(U->getOperand(0)) ||
1342 isa<ShuffleVectorInst>(U->getOperand(0)) ||
1343 isa<ExtractElementInst>(U->getOperand(0)) ||
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001344 isa<InsertElementInst>(U->getOperand(0)) ||
Owen Anderson59bd0532007-07-03 18:37:08 +00001345 isa<SelectInst>(U->getOperand(0)) ||
1346 isa<CastInst>(U->getOperand(0)))
Owen Anderson06c1e582007-06-20 22:10:02 +00001347 s1 = find_leader(availableOut[*PI], VN.lookup(U->getOperand(0)));
1348 else
1349 s1 = U->getOperand(0);
1350
1351 Value* s2 = 0;
Owen Anderson59bd0532007-07-03 18:37:08 +00001352
1353 if (isa<BinaryOperator>(U) ||
1354 isa<CmpInst>(U) ||
1355 isa<ShuffleVectorInst>(U) ||
1356 isa<ExtractElementInst>(U) ||
1357 isa<InsertElementInst>(U) ||
1358 isa<SelectInst>(U))
1359 if (isa<BinaryOperator>(U->getOperand(1)) ||
1360 isa<CmpInst>(U->getOperand(1)) ||
1361 isa<ShuffleVectorInst>(U->getOperand(1)) ||
1362 isa<ExtractElementInst>(U->getOperand(1)) ||
1363 isa<InsertElementInst>(U->getOperand(1)) ||
1364 isa<SelectInst>(U->getOperand(1)) ||
1365 isa<CastInst>(U->getOperand(1))) {
1366 s2 = find_leader(availableOut[*PI], VN.lookup(U->getOperand(1)));
1367 } else {
1368 s2 = U->getOperand(1);
1369 }
Owen Anderson0eb26572007-06-27 17:03:03 +00001370
1371 // Ternary Operators
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001372 Value* s3 = 0;
1373 if (isa<ShuffleVectorInst>(U) ||
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001374 isa<InsertElementInst>(U) ||
1375 isa<SelectInst>(U))
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001376 if (isa<BinaryOperator>(U->getOperand(2)) ||
1377 isa<CmpInst>(U->getOperand(2)) ||
1378 isa<ShuffleVectorInst>(U->getOperand(2)) ||
1379 isa<ExtractElementInst>(U->getOperand(2)) ||
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001380 isa<InsertElementInst>(U->getOperand(2)) ||
Owen Anderson59bd0532007-07-03 18:37:08 +00001381 isa<SelectInst>(U->getOperand(2)) ||
1382 isa<CastInst>(U->getOperand(2))) {
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001383 s3 = find_leader(availableOut[*PI], VN.lookup(U->getOperand(2)));
Owen Anderson59bd0532007-07-03 18:37:08 +00001384 } else {
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001385 s3 = U->getOperand(2);
Owen Anderson59bd0532007-07-03 18:37:08 +00001386 }
Owen Anderson06c1e582007-06-20 22:10:02 +00001387
1388 Value* newVal = 0;
1389 if (BinaryOperator* BO = dyn_cast<BinaryOperator>(U))
1390 newVal = BinaryOperator::create(BO->getOpcode(), s1, s2,
1391 BO->getName()+".gvnpre",
1392 (*PI)->getTerminator());
1393 else if (CmpInst* C = dyn_cast<CmpInst>(U))
1394 newVal = CmpInst::create(C->getOpcode(), C->getPredicate(), s1, s2,
1395 C->getName()+".gvnpre",
1396 (*PI)->getTerminator());
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001397 else if (ShuffleVectorInst* S = dyn_cast<ShuffleVectorInst>(U))
1398 newVal = new ShuffleVectorInst(s1, s2, s3, S->getName()+".gvnpre",
1399 (*PI)->getTerminator());
1400 else if (InsertElementInst* S = dyn_cast<InsertElementInst>(U))
1401 newVal = new InsertElementInst(s1, s2, s3, S->getName()+".gvnpre",
1402 (*PI)->getTerminator());
1403 else if (ExtractElementInst* S = dyn_cast<ExtractElementInst>(U))
1404 newVal = new ExtractElementInst(s1, s2, S->getName()+".gvnpre",
1405 (*PI)->getTerminator());
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001406 else if (SelectInst* S = dyn_cast<SelectInst>(U))
1407 newVal = new SelectInst(S->getCondition(), S->getTrueValue(),
1408 S->getFalseValue(), S->getName()+".gvnpre",
1409 (*PI)->getTerminator());
Owen Anderson59bd0532007-07-03 18:37:08 +00001410 else if (CastInst* C = dyn_cast<CastInst>(U))
1411 newVal = CastInst::create(C->getOpcode(), s1, C->getType(),
1412 C->getName()+".gvnpre",
1413 (*PI)->getTerminator());
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001414
Owen Anderson06c1e582007-06-20 22:10:02 +00001415
1416 VN.add(newVal, VN.lookup(U));
1417
Owen Andersone02da552007-06-28 00:34:34 +00001418 SmallPtrSet<Value*, 16>& predAvail = availableOut[*PI];
Owen Anderson06c1e582007-06-20 22:10:02 +00001419 val_replace(predAvail, newVal);
Owen Andersonfd5683a2007-06-21 00:19:05 +00001420
Owen Anderson06c1e582007-06-20 22:10:02 +00001421 std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
1422 if (av != avail.end())
1423 avail.erase(av);
1424 avail.insert(std::make_pair(*PI, newVal));
1425
1426 ++NumInsertedVals;
1427 }
1428 }
1429
1430 PHINode* p = 0;
1431
1432 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE; ++PI) {
1433 if (p == 0)
1434 p = new PHINode(avail[*PI]->getType(), "gvnpre-join", BB->begin());
1435
1436 p->addIncoming(avail[*PI], *PI);
1437 }
1438
1439 VN.add(p, VN.lookup(e));
Owen Anderson06c1e582007-06-20 22:10:02 +00001440 val_replace(availableOut[BB], p);
Owen Anderson06c1e582007-06-20 22:10:02 +00001441 new_set.insert(p);
1442
1443 ++NumInsertedPhis;
1444}
1445
Owen Andersonfd5683a2007-06-21 00:19:05 +00001446/// insertion_mergepoint - When walking the dom tree, check at each merge
1447/// block for the possibility of a partial redundancy. If present, eliminate it
Owen Anderson06c1e582007-06-20 22:10:02 +00001448unsigned GVNPRE::insertion_mergepoint(std::vector<Value*>& workList,
Owen Anderson28a2d442007-06-22 00:20:30 +00001449 df_iterator<DomTreeNode*>& D,
Owen Andersone02da552007-06-28 00:34:34 +00001450 SmallPtrSet<Value*, 16>& new_set) {
Owen Anderson06c1e582007-06-20 22:10:02 +00001451 bool changed_function = false;
1452 bool new_stuff = false;
1453
1454 BasicBlock* BB = D->getBlock();
1455 for (unsigned i = 0; i < workList.size(); ++i) {
1456 Value* e = workList[i];
1457
Owen Andersonb6a39fc2007-06-27 04:10:46 +00001458 if (isa<BinaryOperator>(e) || isa<CmpInst>(e) ||
1459 isa<ExtractElementInst>(e) || isa<InsertElementInst>(e) ||
Owen Anderson59bd0532007-07-03 18:37:08 +00001460 isa<ShuffleVectorInst>(e) || isa<SelectInst>(e) || isa<CastInst>(e)) {
Owen Anderson06c1e582007-06-20 22:10:02 +00001461 if (find_leader(availableOut[D->getIDom()->getBlock()],
1462 VN.lookup(e)) != 0)
1463 continue;
1464
1465 std::map<BasicBlock*, Value*> avail;
1466 bool by_some = false;
1467 int num_avail = 0;
1468
1469 for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE;
1470 ++PI) {
1471 Value *e2 = phi_translate(e, *PI, BB);
1472 Value *e3 = find_leader(availableOut[*PI], VN.lookup(e2));
1473
1474 if (e3 == 0) {
1475 std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
1476 if (av != avail.end())
1477 avail.erase(av);
1478 avail.insert(std::make_pair(*PI, e2));
1479 } else {
1480 std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
1481 if (av != avail.end())
1482 avail.erase(av);
1483 avail.insert(std::make_pair(*PI, e3));
1484
1485 by_some = true;
1486 num_avail++;
1487 }
1488 }
1489
1490 if (by_some && num_avail < std::distance(pred_begin(BB), pred_end(BB))) {
1491 insertion_pre(e, BB, avail, new_set);
1492
1493 changed_function = true;
1494 new_stuff = true;
1495 }
1496 }
1497 }
1498
1499 unsigned retval = 0;
1500 if (changed_function)
1501 retval += 1;
1502 if (new_stuff)
1503 retval += 2;
1504
1505 return retval;
1506}
1507
Owen Andersonfd5683a2007-06-21 00:19:05 +00001508/// insert - Phase 2 of the main algorithm. Walk the dominator tree looking for
1509/// merge points. When one is found, check for a partial redundancy. If one is
1510/// present, eliminate it. Repeat this walk until no changes are made.
Owen Anderson06c1e582007-06-20 22:10:02 +00001511bool GVNPRE::insertion(Function& F) {
1512 bool changed_function = false;
1513
1514 DominatorTree &DT = getAnalysis<DominatorTree>();
Owen Andersonbe802402007-06-08 01:03:01 +00001515
Owen Andersone02da552007-06-28 00:34:34 +00001516 std::map<BasicBlock*, SmallPtrSet<Value*, 16> > new_sets;
Owen Andersonbe802402007-06-08 01:03:01 +00001517 bool new_stuff = true;
1518 while (new_stuff) {
1519 new_stuff = false;
Owen Andersonbe802402007-06-08 01:03:01 +00001520 for (df_iterator<DomTreeNode*> DI = df_begin(DT.getRootNode()),
1521 E = df_end(DT.getRootNode()); DI != E; ++DI) {
1522 BasicBlock* BB = DI->getBlock();
1523
Owen Andersond184c182007-06-11 16:25:17 +00001524 if (BB == 0)
1525 continue;
1526
Owen Andersone02da552007-06-28 00:34:34 +00001527 SmallPtrSet<Value*, 16>& new_set = new_sets[BB];
1528 SmallPtrSet<Value*, 16>& availOut = availableOut[BB];
1529 SmallPtrSet<Value*, 16>& anticIn = anticipatedIn[BB];
Owen Andersonbe802402007-06-08 01:03:01 +00001530
Owen Anderson55994f22007-06-08 20:44:02 +00001531 new_set.clear();
1532
Owen Andersonbe802402007-06-08 01:03:01 +00001533 // Replace leaders with leaders inherited from dominator
1534 if (DI->getIDom() != 0) {
Owen Andersone02da552007-06-28 00:34:34 +00001535 SmallPtrSet<Value*, 16>& dom_set = new_sets[DI->getIDom()->getBlock()];
1536 for (SmallPtrSet<Value*, 16>::iterator I = dom_set.begin(),
Owen Andersonbe802402007-06-08 01:03:01 +00001537 E = dom_set.end(); I != E; ++I) {
1538 new_set.insert(*I);
Owen Andersonb56fba02007-06-19 03:31:41 +00001539 val_replace(availOut, *I);
Owen Andersonbe802402007-06-08 01:03:01 +00001540 }
1541 }
1542
1543 // If there is more than one predecessor...
1544 if (pred_begin(BB) != pred_end(BB) && ++pred_begin(BB) != pred_end(BB)) {
1545 std::vector<Value*> workList;
Owen Andersond50a29d2007-06-22 00:43:22 +00001546 workList.reserve(anticIn.size());
Owen Andersonbe802402007-06-08 01:03:01 +00001547 topo_sort(anticIn, workList);
1548
Owen Anderson06c1e582007-06-20 22:10:02 +00001549 unsigned result = insertion_mergepoint(workList, DI, new_set);
1550 if (result & 1)
1551 changed_function = true;
1552 if (result & 2)
1553 new_stuff = true;
Owen Andersonbe802402007-06-08 01:03:01 +00001554 }
1555 }
Owen Andersonbe802402007-06-08 01:03:01 +00001556 }
Owen Anderson634a0632007-06-06 01:27:49 +00001557
Owen Anderson06c1e582007-06-20 22:10:02 +00001558 return changed_function;
1559}
1560
Owen Andersonfd5683a2007-06-21 00:19:05 +00001561// GVNPRE::runOnFunction - This is the main transformation entry point for a
1562// function.
1563//
Owen Anderson06c1e582007-06-20 22:10:02 +00001564bool GVNPRE::runOnFunction(Function &F) {
Owen Andersonfd5683a2007-06-21 00:19:05 +00001565 // Clean out global sets from any previous functions
Owen Anderson06c1e582007-06-20 22:10:02 +00001566 VN.clear();
1567 createdExpressions.clear();
1568 availableOut.clear();
1569 anticipatedIn.clear();
1570
1571 bool changed_function = false;
1572
1573 // Phase 1: BuildSets
Owen Andersonfd5683a2007-06-21 00:19:05 +00001574 // This phase calculates the AVAIL_OUT and ANTIC_IN sets
Owen Anderson5477c542007-06-26 23:29:41 +00001575 buildsets(F);
Owen Anderson06c1e582007-06-20 22:10:02 +00001576
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001577 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ++FI) {
Owen Anderson8a9fa5d2007-06-28 23:51:21 +00001578 DOUT << "AVAIL_OUT: " << FI->getName() << "\n";
1579 dump(availableOut[FI]);
1580 DOUT << "\n";
Owen Anderson7dae8ef2007-06-27 17:38:29 +00001581 DOUT << "ANTIC_IN: " << FI->getName() << "\n";
1582 dump(anticipatedIn[FI]);
1583 DOUT << "\n\n";
1584 }
1585
Owen Anderson06c1e582007-06-20 22:10:02 +00001586 // Phase 2: Insert
Owen Andersonfd5683a2007-06-21 00:19:05 +00001587 // This phase inserts values to make partially redundant values
1588 // fully redundant
Owen Anderson06c1e582007-06-20 22:10:02 +00001589 changed_function |= insertion(F);
1590
Owen Anderson634a0632007-06-06 01:27:49 +00001591 // Phase 3: Eliminate
Owen Andersonfd5683a2007-06-21 00:19:05 +00001592 // This phase performs trivial full redundancy elimination
Owen Anderson06c1e582007-06-20 22:10:02 +00001593 changed_function |= elimination();
Owen Anderson55994f22007-06-08 20:44:02 +00001594
Owen Andersonbe802402007-06-08 01:03:01 +00001595 // Phase 4: Cleanup
Owen Andersonfd5683a2007-06-21 00:19:05 +00001596 // This phase cleans up values that were created solely
1597 // as leaders for expressions
Owen Anderson42769842007-06-12 16:57:50 +00001598 cleanup();
Owen Andersonbe802402007-06-08 01:03:01 +00001599
Owen Andersonb0714bb2007-06-20 18:30:20 +00001600 return changed_function;
Owen Anderson5fba6c12007-05-29 21:53:49 +00001601}