Add comment.
[oota-llvm.git] / lib / Transforms / Scalar / GVNPRE.cpp
index e1e92f8561f10aafc8f63e66a1af25972655ad3b..7d38dc75a1e24037b7da7865efefe8ffb646391d 100644 (file)
@@ -2,8 +2,8 @@
 //
 //                     The LLVM Compiler Infrastructure
 //
-// This file was developed by the Owen Anderson and is distributed under
-// the University of Illinois Open Source License. See LICENSE.TXT for details.
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
 //
 //===----------------------------------------------------------------------===//
 //
@@ -39,7 +39,6 @@
 #include <algorithm>
 #include <deque>
 #include <map>
-#include <vector>
 using namespace llvm;
 
 //===----------------------------------------------------------------------===//
@@ -71,7 +70,7 @@ struct Expression {
   SmallVector<uint32_t, 4> varargs;
   
   Expression() { }
-  Expression(ExpressionOpcode o) : opcode(o) { }
+  explicit Expression(ExpressionOpcode o) : opcode(o) { }
   
   bool operator==(const Expression &other) const {
     if (opcode != other.opcode)
@@ -156,9 +155,14 @@ namespace {
 }
 
 namespace llvm {
-template <> struct DenseMapKeyInfo<Expression> {
-  static inline Expression getEmptyKey() { return Expression(Expression::EMPTY); }
-  static inline Expression getTombstoneKey() { return Expression(Expression::TOMBSTONE); }
+template <> struct DenseMapInfo<Expression> {
+  static inline Expression getEmptyKey() {
+    return Expression(Expression::EMPTY);
+  }
+  
+  static inline Expression getTombstoneKey() {
+    return Expression(Expression::TOMBSTONE);
+  }
   
   static unsigned getHashValue(const Expression e) {
     unsigned hash = e.opcode;
@@ -167,16 +171,19 @@ template <> struct DenseMapKeyInfo<Expression> {
     hash = e.secondVN + hash * 37;
     hash = e.thirdVN + hash * 37;
     
-    hash = (unsigned)((uintptr_t)e.type >> 4) ^
-            (unsigned)((uintptr_t)e.type >> 9) +
-            hash * 37;
+    hash = ((unsigned)((uintptr_t)e.type >> 4) ^
+            (unsigned)((uintptr_t)e.type >> 9)) +
+           hash * 37;
     
-    for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(), E = e.varargs.end();
-         I != E; ++I)
+    for (SmallVector<uint32_t, 4>::const_iterator I = e.varargs.begin(),
+         E = e.varargs.end(); I != E; ++I)
       hash = *I + hash * 37;
     
     return hash;
   }
+  static bool isEqual(const Expression &LHS, const Expression &RHS) {
+    return LHS == RHS;
+  }
   static bool isPod() { return true; }
 };
 }
@@ -419,7 +426,7 @@ Expression ValueTable::create_expression(GetElementPtrInst* G) {
   e.secondVN = 0;
   e.thirdVN = 0;
   e.type = G->getType();
-  e.opcode = Expression::SELECT;
+  e.opcode = Expression::GEP;
   
   for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
        I != E; ++I)
@@ -659,7 +666,7 @@ namespace {
 
   private:
     ValueTable VN;
-    std::vector<Instruction*> createdExpressions;
+    SmallVector<Instruction*, 8> createdExpressions;
     
     DenseMap<BasicBlock*, ValueNumberedSet> availableOut;
     DenseMap<BasicBlock*, ValueNumberedSet> anticipatedIn;
@@ -683,7 +690,7 @@ namespace {
                            BasicBlock* succ, ValueNumberedSet& out) ;
     
     void topo_sort(ValueNumberedSet& set,
-                   std::vector<Value*>& vec) ;
+                   SmallVector<Value*, 8>& vec) ;
     
     void cleanup() ;
     bool elimination() ;
@@ -695,23 +702,23 @@ namespace {
                             ValueNumberedSet& currAvail,
                             ValueNumberedSet& currPhis,
                             ValueNumberedSet& currExps,
-                            SmallPtrSet<Value*, 16>& currTemps) ;
+                            SmallPtrSet<Value*, 16>& currTemps);
     bool buildsets_anticout(BasicBlock* BB,
                             ValueNumberedSet& anticOut,
-                            SmallPtrSet<BasicBlock*, 8>& visited) ;
+                            SmallPtrSet<BasicBlock*, 8>& visited);
     unsigned buildsets_anticin(BasicBlock* BB,
                            ValueNumberedSet& anticOut,
                            ValueNumberedSet& currExps,
                            SmallPtrSet<Value*, 16>& currTemps,
-                           SmallPtrSet<BasicBlock*, 8>& visited) ;
+                           SmallPtrSet<BasicBlock*, 8>& visited);
     void buildsets(Function& F) ;
     
     void insertion_pre(Value* e, BasicBlock* BB,
-                       std::map<BasicBlock*, Value*>& avail,
-                      std::map<BasicBlock*,ValueNumberedSet>& new_set) ;
-    unsigned insertion_mergepoint(std::vector<Value*>& workList,
+                       DenseMap<BasicBlock*, Value*>& avail,
+                       std::map<BasicBlock*,ValueNumberedSet>& new_set);
+    unsigned insertion_mergepoint(SmallVector<Value*, 8>& workList,
                                   df_iterator<DomTreeNode*>& D,
-                      std::map<BasicBlock*, ValueNumberedSet>& new_set) ;
+                      std::map<BasicBlock*, ValueNumberedSet>& new_set);
     bool insertion(Function& F) ;
   
   };
@@ -724,7 +731,7 @@ namespace {
 FunctionPass *llvm::createGVNPREPass() { return new GVNPRE(); }
 
 static RegisterPass<GVNPRE> X("gvnpre",
-                              "Global Value Numbering/Partial Redundancy Elimination");
+                      "Global Value Numbering/Partial Redundancy Elimination");
 
 
 STATISTIC(NumInsertedVals, "Number of values inserted");
@@ -758,6 +765,8 @@ void GVNPRE::val_insert(ValueNumberedSet& s, Value* v) {
 /// val_replace - Insert a value into a set, replacing any values already in
 /// the set that have the same value number
 void GVNPRE::val_replace(ValueNumberedSet& s, Value* v) {
+  if (s.count(v)) return;
+  
   uint32_t num = VN.lookup(v);
   Value* leader = find_leader(s, num);
   if (leader != 0)
@@ -925,7 +934,7 @@ Value* GVNPRE::phi_translate(Value* V, BasicBlock* pred, BasicBlock* succ) {
       return 0;
     
     bool changed_idx = false;
-    std::vector<Value*> newIdx;
+    SmallVector<Value*, 4> newIdx;
     for (GetElementPtrInst::op_iterator I = U->idx_begin(), E = U->idx_end();
          I != E; ++I)
       if (isa<Instruction>(*I)) {
@@ -939,7 +948,7 @@ Value* GVNPRE::phi_translate(Value* V, BasicBlock* pred, BasicBlock* succ) {
     
     if (newOp1 != U->getPointerOperand() || changed_idx) {
       Instruction* newVal = new GetElementPtrInst(newOp1,
-                                       &newIdx[0], newIdx.size(),
+                                       newIdx.begin(), newIdx.end(),
                                        U->getName()+".expr");
       
       uint32_t v = VN.lookup_or_add(newVal);
@@ -996,7 +1005,7 @@ bool GVNPRE::dependsOnInvoke(Value* V) {
 /// themselves in the set, as well as all values that depend on invokes (see 
 /// above)
 void GVNPRE::clean(ValueNumberedSet& set) {
-  std::vector<Value*> worklist;
+  SmallVector<Value*, 8> worklist;
   worklist.reserve(set.size());
   topo_sort(set, worklist);
   
@@ -1085,9 +1094,9 @@ void GVNPRE::clean(ValueNumberedSet& set) {
 
 /// topo_sort - Given a set of values, sort them by topological
 /// order into the provided vector.
-void GVNPRE::topo_sort(ValueNumberedSet& set, std::vector<Value*>& vec) {
+void GVNPRE::topo_sort(ValueNumberedSet& set, SmallVector<Value*, 8>& vec) {
   SmallPtrSet<Value*, 16> visited;
-  std::vector<Value*> stack;
+  SmallVector<Value*, 8> stack;
   for (ValueNumberedSet::iterator I = set.begin(), E = set.end();
        I != E; ++I) {
     if (visited.count(*I) == 0)
@@ -1205,8 +1214,8 @@ void GVNPRE::dump(ValueNumberedSet& s) const {
 bool GVNPRE::elimination() {
   bool changed_function = false;
   
-  std::vector<std::pair<Instruction*, Value*> > replace;
-  std::vector<Instruction*> erase;
+  SmallVector<std::pair<Instruction*, Value*>, 8> replace;
+  SmallVector<Instruction*, 8> erase;
   
   DominatorTree& DT = getAnalysis<DominatorTree>();
   
@@ -1222,7 +1231,8 @@ bool GVNPRE::elimination() {
           isa<ExtractElementInst>(BI) || isa<SelectInst>(BI) ||
           isa<CastInst>(BI) || isa<GetElementPtrInst>(BI)) {
         
-        if (availableOut[BB].test(VN.lookup(BI)) && !availableOut[BB].count(BI)) {
+        if (availableOut[BB].test(VN.lookup(BI)) &&
+            !availableOut[BB].count(BI)) {
           Value *leader = find_leader(availableOut[BB], VN.lookup(BI));
           if (Instruction* Instr = dyn_cast<Instruction>(leader))
             if (Instr->getParent() != 0 && Instr != BI) {
@@ -1242,8 +1252,8 @@ bool GVNPRE::elimination() {
     changed_function = true;
   }
     
-  for (std::vector<Instruction*>::iterator I = erase.begin(), E = erase.end();
-       I != E; ++I)
+  for (SmallVector<Instruction*, 8>::iterator I = erase.begin(),
+       E = erase.end(); I != E; ++I)
      (*I)->eraseFromParent();
   
   return changed_function;
@@ -1416,14 +1426,14 @@ bool GVNPRE::buildsets_anticout(BasicBlock* BB,
       BasicBlock* currSucc = BB->getTerminator()->getSuccessor(i);
       ValueNumberedSet& succAnticIn = anticipatedIn[currSucc];
       
-      std::vector<Value*> temp;
+      SmallVector<Value*, 16> temp;
       
       for (ValueNumberedSet::iterator I = anticOut.begin(),
            E = anticOut.end(); I != E; ++I)
         if (!succAnticIn.test(VN.lookup(*I)))
           temp.push_back(*I);
 
-      for (std::vector<Value*>::iterator I = temp.begin(), E = temp.end();
+      for (SmallVector<Value*, 16>::iterator I = temp.begin(), E = temp.end();
            I != E; ++I) {
         anticOut.erase(*I);
         anticOut.reset(VN.lookup(*I));
@@ -1562,7 +1572,7 @@ void GVNPRE::buildsets(Function& F) {
 /// by inserting appropriate values into the predecessors and a phi node in
 /// the main block
 void GVNPRE::insertion_pre(Value* e, BasicBlock* BB,
-                           std::map<BasicBlock*, Value*>& avail,
+                           DenseMap<BasicBlock*, Value*>& avail,
                     std::map<BasicBlock*, ValueNumberedSet>& new_sets) {
   for (pred_iterator PI = pred_begin(BB), PE = pred_end(BB); PI != PE; ++PI) {
     Value* e2 = avail[*PI];
@@ -1589,7 +1599,7 @@ void GVNPRE::insertion_pre(Value* e, BasicBlock* BB,
           isa<ShuffleVectorInst>(U) ||
           isa<ExtractElementInst>(U) ||
           isa<InsertElementInst>(U) ||
-          isa<SelectInst>(U))
+          isa<SelectInst>(U)) {
         if (isa<BinaryOperator>(U->getOperand(1)) || 
             isa<CmpInst>(U->getOperand(1)) ||
             isa<ShuffleVectorInst>(U->getOperand(1)) ||
@@ -1602,12 +1612,13 @@ void GVNPRE::insertion_pre(Value* e, BasicBlock* BB,
         } else {
           s2 = U->getOperand(1);
         }
+      }
       
       // Ternary Operators
       Value* s3 = 0;
       if (isa<ShuffleVectorInst>(U) ||
           isa<InsertElementInst>(U) ||
-          isa<SelectInst>(U))
+          isa<SelectInst>(U)) {
         if (isa<BinaryOperator>(U->getOperand(2)) || 
             isa<CmpInst>(U->getOperand(2)) ||
             isa<ShuffleVectorInst>(U->getOperand(2)) ||
@@ -1620,9 +1631,10 @@ void GVNPRE::insertion_pre(Value* e, BasicBlock* BB,
         } else {
           s3 = U->getOperand(2);
         }
+      }
       
       // Vararg operators
-      std::vector<Value*> sVarargs;
+      SmallVector<Value*, 4> sVarargs;
       if (GetElementPtrInst* G = dyn_cast<GetElementPtrInst>(U)) {
         for (GetElementPtrInst::op_iterator OI = G->idx_begin(),
              OE = G->idx_end(); OI != OE; ++OI) {
@@ -1668,7 +1680,7 @@ void GVNPRE::insertion_pre(Value* e, BasicBlock* BB,
                                   C->getName()+".gvnpre", 
                                   (*PI)->getTerminator());
       else if (GetElementPtrInst* G = dyn_cast<GetElementPtrInst>(U))
-        newVal = new GetElementPtrInst(s1, &sVarargs[0], sVarargs.size(), 
+        newVal = new GetElementPtrInst(s1, sVarargs.begin(), sVarargs.end(), 
                                        G->getName()+".gvnpre", 
                                        (*PI)->getTerminator());
                                 
@@ -1680,7 +1692,7 @@ void GVNPRE::insertion_pre(Value* e, BasicBlock* BB,
       val_replace(new_sets[*PI], newVal);
       predAvail.set(VN.lookup(newVal));
             
-      std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
+      DenseMap<BasicBlock*, Value*>::iterator av = avail.find(*PI);
       if (av != avail.end())
         avail.erase(av);
       avail.insert(std::make_pair(*PI, newVal));
@@ -1711,7 +1723,7 @@ void GVNPRE::insertion_pre(Value* e, BasicBlock* BB,
 
 /// insertion_mergepoint - When walking the dom tree, check at each merge
 /// block for the possibility of a partial redundancy.  If present, eliminate it
-unsigned GVNPRE::insertion_mergepoint(std::vector<Value*>& workList,
+unsigned GVNPRE::insertion_mergepoint(SmallVector<Value*, 8>& workList,
                                       df_iterator<DomTreeNode*>& D,
                     std::map<BasicBlock*, ValueNumberedSet >& new_sets) {
   bool changed_function = false;
@@ -1728,7 +1740,7 @@ unsigned GVNPRE::insertion_mergepoint(std::vector<Value*>& workList,
       if (availableOut[D->getIDom()->getBlock()].test(VN.lookup(e)))
         continue;
             
-      std::map<BasicBlock*, Value*> avail;
+      DenseMap<BasicBlock*, Value*> avail;
       bool by_some = false;
       bool all_same = true;
       Value * first_s = 0;
@@ -1739,13 +1751,13 @@ unsigned GVNPRE::insertion_mergepoint(std::vector<Value*>& workList,
         Value *e3 = find_leader(availableOut[*PI], VN.lookup(e2));
               
         if (e3 == 0) {
-          std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
+          DenseMap<BasicBlock*, Value*>::iterator av = avail.find(*PI);
           if (av != avail.end())
             avail.erase(av);
           avail.insert(std::make_pair(*PI, e2));
           all_same = false;
         } else {
-          std::map<BasicBlock*, Value*>::iterator av = avail.find(*PI);
+          DenseMap<BasicBlock*, Value*>::iterator av = avail.find(*PI);
           if (av != avail.end())
             avail.erase(av);
           avail.insert(std::make_pair(*PI, e3));
@@ -1811,7 +1823,7 @@ bool GVNPRE::insertion(Function& F) {
       
       // If there is more than one predecessor...
       if (pred_begin(BB) != pred_end(BB) && ++pred_begin(BB) != pred_end(BB)) {
-        std::vector<Value*> workList;
+        SmallVector<Value*, 8> workList;
         workList.reserve(anticIn.size());
         topo_sort(anticIn, workList);