GVN.cpp revision 830db6a00e446220fb31f97f116002db147b3f89 
1//===- GVN.cpp - Eliminate redundant values and loads ------------===//
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 global value numbering to eliminate fully redundant
11// instructions.  It also performs simple dead load elimination.
12//
13//===----------------------------------------------------------------------===//
14
15#define DEBUG_TYPE "gvn"
16
17#include "llvm/Transforms/Scalar.h"
18#include "llvm/BasicBlock.h"
19#include "llvm/Constants.h"
20#include "llvm/DerivedTypes.h"
21#include "llvm/Function.h"
22#include "llvm/Instructions.h"
23#include "llvm/Value.h"
24#include "llvm/Analysis/Dominators.h"
25#include "llvm/ADT/BitVector.h"
26#include "llvm/ADT/DenseMap.h"
27#include "llvm/ADT/DepthFirstIterator.h"
28#include "llvm/ADT/SmallPtrSet.h"
29#include "llvm/ADT/SmallVector.h"
30#include "llvm/ADT/Statistic.h"
31#include "llvm/Analysis/MemoryDependenceAnalysis.h"
32#include "llvm/Support/CFG.h"
33#include "llvm/Support/Compiler.h"
34using namespace llvm;
35
36//===----------------------------------------------------------------------===//
37//                         ValueTable Class
38//===----------------------------------------------------------------------===//
39
40/// This class holds the mapping between values and value numbers.  It is used
41/// as an efficient mechanism to determine the expression-wise equivalence of
42/// two values.
43namespace {
44  struct VISIBILITY_HIDDEN Expression {
45    enum ExpressionOpcode { ADD, SUB, MUL, UDIV, SDIV, FDIV, UREM, SREM,
46                            FREM, SHL, LSHR, ASHR, AND, OR, XOR, ICMPEQ,
47                            ICMPNE, ICMPUGT, ICMPUGE, ICMPULT, ICMPULE,
48                            ICMPSGT, ICMPSGE, ICMPSLT, ICMPSLE, FCMPOEQ,
49                            FCMPOGT, FCMPOGE, FCMPOLT, FCMPOLE, FCMPONE,
50                            FCMPORD, FCMPUNO, FCMPUEQ, FCMPUGT, FCMPUGE,
51                            FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
52                            SHUFFLE, SELECT, TRUNC, ZEXT, SEXT, FPTOUI,
53                            FPTOSI, UITOFP, SITOFP, FPTRUNC, FPEXT,
54                            PTRTOINT, INTTOPTR, BITCAST, GEP, EMPTY,
55                            TOMBSTONE };
56
57    ExpressionOpcode opcode;
58    const Type* type;
59    uint32_t firstVN;
60    uint32_t secondVN;
61    uint32_t thirdVN;
62    SmallVector<uint32_t, 4> varargs;
63
64    Expression() { }
65    Expression(ExpressionOpcode o) : opcode(o) { }
66
67    bool operator==(const Expression &other) const {
68      if (opcode != other.opcode)
69        return false;
70      else if (opcode == EMPTY || opcode == TOMBSTONE)
71        return true;
72      else if (type != other.type)
73        return false;
74      else if (firstVN != other.firstVN)
75        return false;
76      else if (secondVN != other.secondVN)
77        return false;
78      else if (thirdVN != other.thirdVN)
79        return false;
80      else {
81        if (varargs.size() != other.varargs.size())
82          return false;
83
84        for (size_t i = 0; i < varargs.size(); ++i)
85          if (varargs[i] != other.varargs[i])
86            return false;
87
88        return true;
89      }
90    }
91
92    bool operator!=(const Expression &other) const {
93      if (opcode != other.opcode)
94        return true;
95      else if (opcode == EMPTY || opcode == TOMBSTONE)
96        return false;
97      else if (type != other.type)
98        return true;
99      else if (firstVN != other.firstVN)
100        return true;
101      else if (secondVN != other.secondVN)
102        return true;
103      else if (thirdVN != other.thirdVN)
104        return true;
105      else {
106        if (varargs.size() != other.varargs.size())
107          return true;
108
109        for (size_t i = 0; i < varargs.size(); ++i)
110          if (varargs[i] != other.varargs[i])
111            return true;
112
113          return false;
114      }
115    }
116  };
117
118  class VISIBILITY_HIDDEN ValueTable {
119    private:
120      DenseMap<Value*, uint32_t> valueNumbering;
121      DenseMap<Expression, uint32_t> expressionNumbering;
122
123      uint32_t nextValueNumber;
124
125      Expression::ExpressionOpcode getOpcode(BinaryOperator* BO);
126      Expression::ExpressionOpcode getOpcode(CmpInst* C);
127      Expression::ExpressionOpcode getOpcode(CastInst* C);
128      Expression create_expression(BinaryOperator* BO);
129      Expression create_expression(CmpInst* C);
130      Expression create_expression(ShuffleVectorInst* V);
131      Expression create_expression(ExtractElementInst* C);
132      Expression create_expression(InsertElementInst* V);
133      Expression create_expression(SelectInst* V);
134      Expression create_expression(CastInst* C);
135      Expression create_expression(GetElementPtrInst* G);
136    public:
137      ValueTable() { nextValueNumber = 1; }
138      uint32_t lookup_or_add(Value* V);
139      uint32_t lookup(Value* V) const;
140      void add(Value* V, uint32_t num);
141      void clear();
142      void erase(Value* v);
143      unsigned size();
144  };
145}
146
147namespace llvm {
148template <> struct DenseMapKeyInfo<Expression> {
149  static inline Expression getEmptyKey() {
150    return Expression(Expression::EMPTY);
151  }
152
153  static inline Expression getTombstoneKey() {
154    return Expression(Expression::TOMBSTONE);
155  }
156
157  static unsigned getHashValue(const Expression e) {
158    unsigned hash = e.opcode;
159
160    hash = e.firstVN + hash * 37;
161    hash = e.secondVN + hash * 37;
162    hash = e.thirdVN + hash * 37;
163
164    hash = (unsigned)((uintptr_t)e.type >> 4) ^
165            (unsigned)((uintptr_t)e.type >> 9) +
166            hash * 37;
167
168    for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
169         E = e.varargs.end(); I != E; ++I)
170      hash = *I + hash * 37;
171
172    return hash;
173  }
174  static bool isPod() { return true; }
175};
176}
177
178//===----------------------------------------------------------------------===//
179//                     ValueTable Internal Functions
180//===----------------------------------------------------------------------===//
181Expression::ExpressionOpcode
182                             ValueTable::getOpcode(BinaryOperator* BO) {
183  switch(BO->getOpcode()) {
184    case Instruction::Add:
185      return Expression::ADD;
186    case Instruction::Sub:
187      return Expression::SUB;
188    case Instruction::Mul:
189      return Expression::MUL;
190    case Instruction::UDiv:
191      return Expression::UDIV;
192    case Instruction::SDiv:
193      return Expression::SDIV;
194    case Instruction::FDiv:
195      return Expression::FDIV;
196    case Instruction::URem:
197      return Expression::UREM;
198    case Instruction::SRem:
199      return Expression::SREM;
200    case Instruction::FRem:
201      return Expression::FREM;
202    case Instruction::Shl:
203      return Expression::SHL;
204    case Instruction::LShr:
205      return Expression::LSHR;
206    case Instruction::AShr:
207      return Expression::ASHR;
208    case Instruction::And:
209      return Expression::AND;
210    case Instruction::Or:
211      return Expression::OR;
212    case Instruction::Xor:
213      return Expression::XOR;
214
215    // THIS SHOULD NEVER HAPPEN
216    default:
217      assert(0 && "Binary operator with unknown opcode?");
218      return Expression::ADD;
219  }
220}
221
222Expression::ExpressionOpcode ValueTable::getOpcode(CmpInst* C) {
223  if (C->getOpcode() == Instruction::ICmp) {
224    switch (C->getPredicate()) {
225      case ICmpInst::ICMP_EQ:
226        return Expression::ICMPEQ;
227      case ICmpInst::ICMP_NE:
228        return Expression::ICMPNE;
229      case ICmpInst::ICMP_UGT:
230        return Expression::ICMPUGT;
231      case ICmpInst::ICMP_UGE:
232        return Expression::ICMPUGE;
233      case ICmpInst::ICMP_ULT:
234        return Expression::ICMPULT;
235      case ICmpInst::ICMP_ULE:
236        return Expression::ICMPULE;
237      case ICmpInst::ICMP_SGT:
238        return Expression::ICMPSGT;
239      case ICmpInst::ICMP_SGE:
240        return Expression::ICMPSGE;
241      case ICmpInst::ICMP_SLT:
242        return Expression::ICMPSLT;
243      case ICmpInst::ICMP_SLE:
244        return Expression::ICMPSLE;
245
246      // THIS SHOULD NEVER HAPPEN
247      default:
248        assert(0 && "Comparison with unknown predicate?");
249        return Expression::ICMPEQ;
250    }
251  } else {
252    switch (C->getPredicate()) {
253      case FCmpInst::FCMP_OEQ:
254        return Expression::FCMPOEQ;
255      case FCmpInst::FCMP_OGT:
256        return Expression::FCMPOGT;
257      case FCmpInst::FCMP_OGE:
258        return Expression::FCMPOGE;
259      case FCmpInst::FCMP_OLT:
260        return Expression::FCMPOLT;
261      case FCmpInst::FCMP_OLE:
262        return Expression::FCMPOLE;
263      case FCmpInst::FCMP_ONE:
264        return Expression::FCMPONE;
265      case FCmpInst::FCMP_ORD:
266        return Expression::FCMPORD;
267      case FCmpInst::FCMP_UNO:
268        return Expression::FCMPUNO;
269      case FCmpInst::FCMP_UEQ:
270        return Expression::FCMPUEQ;
271      case FCmpInst::FCMP_UGT:
272        return Expression::FCMPUGT;
273      case FCmpInst::FCMP_UGE:
274        return Expression::FCMPUGE;
275      case FCmpInst::FCMP_ULT:
276        return Expression::FCMPULT;
277      case FCmpInst::FCMP_ULE:
278        return Expression::FCMPULE;
279      case FCmpInst::FCMP_UNE:
280        return Expression::FCMPUNE;
281
282      // THIS SHOULD NEVER HAPPEN
283      default:
284        assert(0 && "Comparison with unknown predicate?");
285        return Expression::FCMPOEQ;
286    }
287  }
288}
289
290Expression::ExpressionOpcode
291                             ValueTable::getOpcode(CastInst* C) {
292  switch(C->getOpcode()) {
293    case Instruction::Trunc:
294      return Expression::TRUNC;
295    case Instruction::ZExt:
296      return Expression::ZEXT;
297    case Instruction::SExt:
298      return Expression::SEXT;
299    case Instruction::FPToUI:
300      return Expression::FPTOUI;
301    case Instruction::FPToSI:
302      return Expression::FPTOSI;
303    case Instruction::UIToFP:
304      return Expression::UITOFP;
305    case Instruction::SIToFP:
306      return Expression::SITOFP;
307    case Instruction::FPTrunc:
308      return Expression::FPTRUNC;
309    case Instruction::FPExt:
310      return Expression::FPEXT;
311    case Instruction::PtrToInt:
312      return Expression::PTRTOINT;
313    case Instruction::IntToPtr:
314      return Expression::INTTOPTR;
315    case Instruction::BitCast:
316      return Expression::BITCAST;
317
318    // THIS SHOULD NEVER HAPPEN
319    default:
320      assert(0 && "Cast operator with unknown opcode?");
321      return Expression::BITCAST;
322  }
323}
324
325Expression ValueTable::create_expression(BinaryOperator* BO) {
326  Expression e;
327
328  e.firstVN = lookup_or_add(BO->getOperand(0));
329  e.secondVN = lookup_or_add(BO->getOperand(1));
330  e.thirdVN = 0;
331  e.type = BO->getType();
332  e.opcode = getOpcode(BO);
333
334  return e;
335}
336
337Expression ValueTable::create_expression(CmpInst* C) {
338  Expression e;
339
340  e.firstVN = lookup_or_add(C->getOperand(0));
341  e.secondVN = lookup_or_add(C->getOperand(1));
342  e.thirdVN = 0;
343  e.type = C->getType();
344  e.opcode = getOpcode(C);
345
346  return e;
347}
348
349Expression ValueTable::create_expression(CastInst* C) {
350  Expression e;
351
352  e.firstVN = lookup_or_add(C->getOperand(0));
353  e.secondVN = 0;
354  e.thirdVN = 0;
355  e.type = C->getType();
356  e.opcode = getOpcode(C);
357
358  return e;
359}
360
361Expression ValueTable::create_expression(ShuffleVectorInst* S) {
362  Expression e;
363
364  e.firstVN = lookup_or_add(S->getOperand(0));
365  e.secondVN = lookup_or_add(S->getOperand(1));
366  e.thirdVN = lookup_or_add(S->getOperand(2));
367  e.type = S->getType();
368  e.opcode = Expression::SHUFFLE;
369
370  return e;
371}
372
373Expression ValueTable::create_expression(ExtractElementInst* E) {
374  Expression e;
375
376  e.firstVN = lookup_or_add(E->getOperand(0));
377  e.secondVN = lookup_or_add(E->getOperand(1));
378  e.thirdVN = 0;
379  e.type = E->getType();
380  e.opcode = Expression::EXTRACT;
381
382  return e;
383}
384
385Expression ValueTable::create_expression(InsertElementInst* I) {
386  Expression e;
387
388  e.firstVN = lookup_or_add(I->getOperand(0));
389  e.secondVN = lookup_or_add(I->getOperand(1));
390  e.thirdVN = lookup_or_add(I->getOperand(2));
391  e.type = I->getType();
392  e.opcode = Expression::INSERT;
393
394  return e;
395}
396
397Expression ValueTable::create_expression(SelectInst* I) {
398  Expression e;
399
400  e.firstVN = lookup_or_add(I->getCondition());
401  e.secondVN = lookup_or_add(I->getTrueValue());
402  e.thirdVN = lookup_or_add(I->getFalseValue());
403  e.type = I->getType();
404  e.opcode = Expression::SELECT;
405
406  return e;
407}
408
409Expression ValueTable::create_expression(GetElementPtrInst* G) {
410  Expression e;
411
412  e.firstVN = lookup_or_add(G->getPointerOperand());
413  e.secondVN = 0;
414  e.thirdVN = 0;
415  e.type = G->getType();
416  e.opcode = Expression::GEP;
417
418  for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
419       I != E; ++I)
420    e.varargs.push_back(lookup_or_add(*I));
421
422  return e;
423}
424
425//===----------------------------------------------------------------------===//
426//                     ValueTable External Functions
427//===----------------------------------------------------------------------===//
428
429/// lookup_or_add - Returns the value number for the specified value, assigning
430/// it a new number if it did not have one before.
431uint32_t ValueTable::lookup_or_add(Value* V) {
432  DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
433  if (VI != valueNumbering.end())
434    return VI->second;
435
436
437  if (BinaryOperator* BO = dyn_cast<BinaryOperator>(V)) {
438    Expression e = create_expression(BO);
439
440    DenseMap<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
441    if (EI != expressionNumbering.end()) {
442      valueNumbering.insert(std::make_pair(V, EI->second));
443      return EI->second;
444    } else {
445      expressionNumbering.insert(std::make_pair(e, nextValueNumber));
446      valueNumbering.insert(std::make_pair(V, nextValueNumber));
447
448      return nextValueNumber++;
449    }
450  } else if (CmpInst* C = dyn_cast<CmpInst>(V)) {
451    Expression e = create_expression(C);
452
453    DenseMap<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
454    if (EI != expressionNumbering.end()) {
455      valueNumbering.insert(std::make_pair(V, EI->second));
456      return EI->second;
457    } else {
458      expressionNumbering.insert(std::make_pair(e, nextValueNumber));
459      valueNumbering.insert(std::make_pair(V, nextValueNumber));
460
461      return nextValueNumber++;
462    }
463  } else if (ShuffleVectorInst* U = dyn_cast<ShuffleVectorInst>(V)) {
464    Expression e = create_expression(U);
465
466    DenseMap<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
467    if (EI != expressionNumbering.end()) {
468      valueNumbering.insert(std::make_pair(V, EI->second));
469      return EI->second;
470    } else {
471      expressionNumbering.insert(std::make_pair(e, nextValueNumber));
472      valueNumbering.insert(std::make_pair(V, nextValueNumber));
473
474      return nextValueNumber++;
475    }
476  } else if (ExtractElementInst* U = dyn_cast<ExtractElementInst>(V)) {
477    Expression e = create_expression(U);
478
479    DenseMap<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
480    if (EI != expressionNumbering.end()) {
481      valueNumbering.insert(std::make_pair(V, EI->second));
482      return EI->second;
483    } else {
484      expressionNumbering.insert(std::make_pair(e, nextValueNumber));
485      valueNumbering.insert(std::make_pair(V, nextValueNumber));
486
487      return nextValueNumber++;
488    }
489  } else if (InsertElementInst* U = dyn_cast<InsertElementInst>(V)) {
490    Expression e = create_expression(U);
491
492    DenseMap<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
493    if (EI != expressionNumbering.end()) {
494      valueNumbering.insert(std::make_pair(V, EI->second));
495      return EI->second;
496    } else {
497      expressionNumbering.insert(std::make_pair(e, nextValueNumber));
498      valueNumbering.insert(std::make_pair(V, nextValueNumber));
499
500      return nextValueNumber++;
501    }
502  } else if (SelectInst* U = dyn_cast<SelectInst>(V)) {
503    Expression e = create_expression(U);
504
505    DenseMap<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
506    if (EI != expressionNumbering.end()) {
507      valueNumbering.insert(std::make_pair(V, EI->second));
508      return EI->second;
509    } else {
510      expressionNumbering.insert(std::make_pair(e, nextValueNumber));
511      valueNumbering.insert(std::make_pair(V, nextValueNumber));
512
513      return nextValueNumber++;
514    }
515  } else if (CastInst* U = dyn_cast<CastInst>(V)) {
516    Expression e = create_expression(U);
517
518    DenseMap<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
519    if (EI != expressionNumbering.end()) {
520      valueNumbering.insert(std::make_pair(V, EI->second));
521      return EI->second;
522    } else {
523      expressionNumbering.insert(std::make_pair(e, nextValueNumber));
524      valueNumbering.insert(std::make_pair(V, nextValueNumber));
525
526      return nextValueNumber++;
527    }
528  } else if (GetElementPtrInst* U = dyn_cast<GetElementPtrInst>(V)) {
529    Expression e = create_expression(U);
530
531    DenseMap<Expression, uint32_t>::iterator EI = expressionNumbering.find(e);
532    if (EI != expressionNumbering.end()) {
533      valueNumbering.insert(std::make_pair(V, EI->second));
534      return EI->second;
535    } else {
536      expressionNumbering.insert(std::make_pair(e, nextValueNumber));
537      valueNumbering.insert(std::make_pair(V, nextValueNumber));
538
539      return nextValueNumber++;
540    }
541  } else {
542    valueNumbering.insert(std::make_pair(V, nextValueNumber));
543    return nextValueNumber++;
544  }
545}
546
547/// lookup - Returns the value number of the specified value. Fails if
548/// the value has not yet been numbered.
549uint32_t ValueTable::lookup(Value* V) const {
550  DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V);
551  if (VI != valueNumbering.end())
552    return VI->second;
553  else
554    assert(0 && "Value not numbered?");
555
556  return 0;
557}
558
559/// clear - Remove all entries from the ValueTable
560void ValueTable::clear() {
561  valueNumbering.clear();
562  expressionNumbering.clear();
563  nextValueNumber = 1;
564}
565
566/// erase - Remove a value from the value numbering
567void ValueTable::erase(Value* V) {
568  valueNumbering.erase(V);
569}
570
571//===----------------------------------------------------------------------===//
572//                       ValueNumberedSet Class
573//===----------------------------------------------------------------------===//
574namespace {
575class ValueNumberedSet {
576  private:
577    SmallPtrSet<Value*, 8> contents;
578    BitVector numbers;
579  public:
580    ValueNumberedSet() { numbers.resize(1); }
581    ValueNumberedSet(const ValueNumberedSet& other) {
582      numbers = other.numbers;
583      contents = other.contents;
584    }
585
586    typedef SmallPtrSet<Value*, 8>::iterator iterator;
587
588    iterator begin() { return contents.begin(); }
589    iterator end() { return contents.end(); }
590
591    bool insert(Value* v) { return contents.insert(v); }
592    void insert(iterator I, iterator E) { contents.insert(I, E); }
593    void erase(Value* v) { contents.erase(v); }
594    unsigned count(Value* v) { return contents.count(v); }
595    size_t size() { return contents.size(); }
596
597    void set(unsigned i)  {
598      if (i >= numbers.size())
599        numbers.resize(i+1);
600
601      numbers.set(i);
602    }
603
604    void operator=(const ValueNumberedSet& other) {
605      contents = other.contents;
606      numbers = other.numbers;
607    }
608
609    void reset(unsigned i)  {
610      if (i < numbers.size())
611        numbers.reset(i);
612    }
613
614    bool test(unsigned i)  {
615      if (i >= numbers.size())
616        return false;
617
618      return numbers.test(i);
619    }
620
621    void clear() {
622      contents.clear();
623      numbers.clear();
624    }
625};
626}
627
628//===----------------------------------------------------------------------===//
629//                         GVN Pass
630//===----------------------------------------------------------------------===//
631
632namespace {
633
634  class VISIBILITY_HIDDEN GVN : public FunctionPass {
635    bool runOnFunction(Function &F);
636  public:
637    static char ID; // Pass identification, replacement for typeid
638    GVN() : FunctionPass((intptr_t)&ID) { }
639
640  private:
641    ValueTable VN;
642
643    DenseMap<BasicBlock*, ValueNumberedSet> availableOut;
644
645    // This transformation requires dominator postdominator info
646    virtual void getAnalysisUsage(AnalysisUsage &AU) const {
647      AU.setPreservesCFG();
648      AU.addRequired<DominatorTree>();
649      AU.addRequired<MemoryDependenceAnalysis>();
650      AU.addPreserved<MemoryDependenceAnalysis>();
651    }
652
653    // Helper fuctions
654    // FIXME: eliminate or document these better
655    Value* find_leader(ValueNumberedSet& vals, uint32_t v) ;
656    void val_insert(ValueNumberedSet& s, Value* v);
657    bool processLoad(LoadInst* L,
658                     DenseMap<Value*, LoadInst*>& lastLoad,
659                     SmallVector<Instruction*, 4>& toErase);
660    bool processInstruction(Instruction* I,
661                            ValueNumberedSet& currAvail,
662                            DenseMap<Value*, LoadInst*>& lastSeenLoad,
663                            SmallVector<Instruction*, 4>& toErase);
664    bool processNonLocalLoad(LoadInst* L,
665                             SmallVector<Instruction*, 4>& toErase);
666    Value *GetValueForBlock(BasicBlock *BB, LoadInst* orig,
667                            DenseMap<BasicBlock*, Value*> &Phis,
668                            bool top_level = false);
669    void dump(DenseMap<BasicBlock*, Value*>& d);
670  };
671
672  char GVN::ID = 0;
673
674}
675
676// createGVNPass - The public interface to this file...
677FunctionPass *llvm::createGVNPass() { return new GVN(); }
678
679static RegisterPass<GVN> X("gvn",
680                           "Global Value Numbering");
681
682STATISTIC(NumGVNInstr, "Number of instructions deleted");
683STATISTIC(NumGVNLoad, "Number of loads deleted");
684
685/// find_leader - Given a set and a value number, return the first
686/// element of the set with that value number, or 0 if no such element
687/// is present
688Value* GVN::find_leader(ValueNumberedSet& vals, uint32_t v) {
689  if (!vals.test(v))
690    return 0;
691
692  for (ValueNumberedSet::iterator I = vals.begin(), E = vals.end();
693       I != E; ++I)
694    if (v == VN.lookup(*I))
695      return *I;
696
697  assert(0 && "No leader found, but present bit is set?");
698  return 0;
699}
700
701/// val_insert - Insert a value into a set only if there is not a value
702/// with the same value number already in the set
703void GVN::val_insert(ValueNumberedSet& s, Value* v) {
704  uint32_t num = VN.lookup(v);
705  if (!s.test(num))
706    s.insert(v);
707}
708
709void GVN::dump(DenseMap<BasicBlock*, Value*>& d) {
710  printf("{\n");
711  for (DenseMap<BasicBlock*, Value*>::iterator I = d.begin(),
712       E = d.end(); I != E; ++I) {
713    if (I->second == MemoryDependenceAnalysis::None)
714      printf("None\n");
715    else
716      I->second->dump();
717  }
718  printf("}\n");
719}
720
721
722/// GetValueForBlock - Get the value to use within the specified basic block.
723/// available values are in Phis.
724Value *GVN::GetValueForBlock(BasicBlock *BB, LoadInst* orig,
725                               DenseMap<BasicBlock*, Value*> &Phis,
726                               bool top_level) {
727
728  // If we have already computed this value, return the previously computed val.
729  Value *&V = Phis[BB];
730  if (V && ! top_level) return V;
731
732  BasicBlock* singlePred = BB->getSinglePredecessor();
733  if (singlePred)
734    return V = GetValueForBlock(singlePred, orig, Phis);
735
736  // Otherwise, the idom is the loop, so we need to insert a PHI node.  Do so
737  // now, then get values to fill in the incoming values for the PHI.
738  PHINode *PN = new PHINode(orig->getType(), orig->getName()+".rle",
739                            BB->begin());
740  PN->reserveOperandSpace(std::distance(pred_begin(BB), pred_end(BB)));
741  V = PN;
742
743  bool all_same = true;
744  Value* first = 0;
745
746  // Fill in the incoming values for the block.
747  for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI) {
748    Value* val = GetValueForBlock(*PI, orig, Phis);
749    if (first == 0)
750      first = val;
751    else if (all_same && first != val)
752      all_same = false;
753
754    PN->addIncoming(val, *PI);
755  }
756
757  if (all_same) {
758    MemoryDependenceAnalysis& MD = getAnalysis<MemoryDependenceAnalysis>();
759
760    MD.removeInstruction(PN);
761    PN->replaceAllUsesWith(first);
762
763    SmallVector<BasicBlock*, 4> toRemove;
764    for (DenseMap<BasicBlock*, Value*>::iterator I = Phis.begin(),
765         E = Phis.end(); I != E; ++I)
766      if (I->second == PN)
767        toRemove.push_back(I->first);
768    for (SmallVector<BasicBlock*, 4>::iterator I = toRemove.begin(),
769         E= toRemove.end(); I != E; ++I)
770      Phis[*I] = first;
771
772    PN->eraseFromParent();
773
774    Phis[BB] = first;
775
776    return first;
777  }
778
779  return PN;
780}
781
782bool GVN::processNonLocalLoad(LoadInst* L,
783                              SmallVector<Instruction*, 4>& toErase) {
784  MemoryDependenceAnalysis& MD = getAnalysis<MemoryDependenceAnalysis>();
785
786  DenseMap<BasicBlock*, Value*> deps;
787  MD.getNonLocalDependency(L, deps);
788
789  DenseMap<BasicBlock*, Value*> repl;
790  for (DenseMap<BasicBlock*, Value*>::iterator I = deps.begin(), E = deps.end();
791       I != E; ++I)
792    if (I->second == MemoryDependenceAnalysis::None) {
793      return false;
794    } else if (I->second == MemoryDependenceAnalysis::NonLocal) {
795      continue;
796    }else if (StoreInst* S = dyn_cast<StoreInst>(I->second)) {
797      if (S->getPointerOperand() == L->getPointerOperand())
798        repl.insert(std::make_pair(I->first, S->getOperand(0)));
799      else
800        return false;
801    } else if (LoadInst* LD = dyn_cast<LoadInst>(I->second)) {
802      if (LD->getPointerOperand() == L->getPointerOperand())
803        repl.insert(std::make_pair(I->first, LD));
804      else
805        return false;
806    } else {
807      return false;
808    }
809
810  SmallPtrSet<BasicBlock*, 4> visited;
811  Value* v = GetValueForBlock(L->getParent(), L, repl, true);
812
813  MD.removeInstruction(L);
814  L->replaceAllUsesWith(v);
815  toErase.push_back(L);
816
817  return true;
818}
819
820bool GVN::processLoad(LoadInst* L,
821                         DenseMap<Value*, LoadInst*>& lastLoad,
822                         SmallVector<Instruction*, 4>& toErase) {
823  if (L->isVolatile()) {
824    lastLoad[L->getPointerOperand()] = L;
825    return false;
826  }
827
828  Value* pointer = L->getPointerOperand();
829  LoadInst*& last = lastLoad[pointer];
830
831  // ... to a pointer that has been loaded from before...
832  MemoryDependenceAnalysis& MD = getAnalysis<MemoryDependenceAnalysis>();
833  Instruction* dep = MD.getDependency(L);
834  if (dep == MemoryDependenceAnalysis::NonLocal &&
835      L->getParent() != &L->getParent()->getParent()->getEntryBlock())
836    processNonLocalLoad(L, toErase);
837  bool deletedLoad = false;
838
839  while (dep != MemoryDependenceAnalysis::None &&
840         dep != MemoryDependenceAnalysis::NonLocal &&
841         (isa<LoadInst>(dep) || isa<StoreInst>(dep))) {
842    // ... that depends on a store ...
843    if (StoreInst* S = dyn_cast<StoreInst>(dep)) {
844      if (S->getPointerOperand() == pointer) {
845        // Remove it!
846        MD.removeInstruction(L);
847
848        L->replaceAllUsesWith(S->getOperand(0));
849        toErase.push_back(L);
850        deletedLoad = true;
851        NumGVNLoad++;
852      }
853
854      // Whether we removed it or not, we can't
855      // go any further
856      break;
857    } else if (!last) {
858      // If we don't depend on a store, and we haven't
859      // been loaded before, bail.
860      break;
861    } else if (dep == last) {
862      // Remove it!
863      MD.removeInstruction(L);
864
865      L->replaceAllUsesWith(last);
866      toErase.push_back(L);
867      deletedLoad = true;
868      NumGVNLoad++;
869
870      break;
871    } else {
872      dep = MD.getDependency(L, dep);
873    }
874  }
875
876  if (!deletedLoad)
877    last = L;
878
879  return deletedLoad;
880}
881
882/// buildsets_availout - When calculating availability, handle an instruction
883/// by inserting it into the appropriate sets
884bool GVN::processInstruction(Instruction* I,
885                                ValueNumberedSet& currAvail,
886                                DenseMap<Value*, LoadInst*>& lastSeenLoad,
887                                SmallVector<Instruction*, 4>& toErase) {
888  if (LoadInst* L = dyn_cast<LoadInst>(I)) {
889    return processLoad(L, lastSeenLoad, toErase);
890  }
891
892  unsigned num = VN.lookup_or_add(I);
893
894  if (currAvail.test(num)) {
895    Value* repl = find_leader(currAvail, num);
896
897    VN.erase(I);
898    I->replaceAllUsesWith(repl);
899    toErase.push_back(I);
900    return true;
901  } else if (!I->isTerminator()) {
902    currAvail.set(num);
903    currAvail.insert(I);
904  }
905
906  return false;
907}
908
909// GVN::runOnFunction - This is the main transformation entry point for a
910// function.
911//
912bool GVN::runOnFunction(Function &F) {
913  // Clean out global sets from any previous functions
914  VN.clear();
915  availableOut.clear();
916
917  bool changed_function = false;
918
919  DominatorTree &DT = getAnalysis<DominatorTree>();
920
921  SmallVector<Instruction*, 4> toErase;
922
923  // Top-down walk of the dominator tree
924  for (df_iterator<DomTreeNode*> DI = df_begin(DT.getRootNode()),
925         E = df_end(DT.getRootNode()); DI != E; ++DI) {
926
927    // Get the set to update for this block
928    ValueNumberedSet& currAvail = availableOut[DI->getBlock()];
929    DenseMap<Value*, LoadInst*> lastSeenLoad;
930
931    BasicBlock* BB = DI->getBlock();
932
933    // A block inherits AVAIL_OUT from its dominator
934    if (DI->getIDom() != 0)
935      currAvail = availableOut[DI->getIDom()->getBlock()];
936
937    for (BasicBlock::iterator BI = BB->begin(), BE = BB->end();
938         BI != BE; ) {
939      changed_function |= processInstruction(BI, currAvail,
940                                             lastSeenLoad, toErase);
941
942      NumGVNInstr += toErase.size();
943
944      // Avoid iterator invalidation
945      ++BI;
946
947      for (SmallVector<Instruction*, 4>::iterator I = toErase.begin(),
948           E = toErase.end(); I != E; ++I)
949        (*I)->eraseFromParent();
950
951      toErase.clear();
952    }
953  }
954
955  return changed_function;
956}
957