Add functionality to value number GEP instructions. This also provides the infrastructure that will

be used for function calls.  NOTE: This does not yet do any transformation of GEPs or function calls.

llvm-svn: 37860

1 files changed, 32 insertions, 2 deletions

diff --git a/llvm/lib/Transforms/Scalar/GVNPRE.cpp b/llvm/lib/Transforms/Scalar/GVNPRE.cpp
index 25cd8165fd0..d290b9c4abe 100644
--- a/llvm/lib/Transforms/Scalar/GVNPRE.cpp
+++ b/llvm/lib/Transforms/Scalar/GVNPRE.cpp
@@ -62,13 +62,14 @@ namespace {
                               FCMPULT, FCMPULE, FCMPUNE, EXTRACT, INSERT,
                               SHUFFLE, SELECT, TRUNC, ZEXT, SEXT, FPTOUI,
                               FPTOSI, UITOFP, SITOFP, FPTRUNC, FPEXT, 
-                              PTRTOINT, INTTOPTR, BITCAST};
+                              PTRTOINT, INTTOPTR, BITCAST, GEP};
     
         ExpressionOpcode opcode;
         const Type* type;
         uint32_t firstVN;
         uint32_t secondVN;
         uint32_t thirdVN;
+        std::vector<uint32_t> varargs;
       
         bool operator< (const Expression& other) const {
           if (opcode < other.opcode)
@@ -91,8 +92,20 @@ namespace {
             return true;
           else if (thirdVN > other.thirdVN)
             return false;
-          else
+          else {
+            if (varargs.size() < other.varargs.size())
+              return true;
+            else if (varargs.size() > other.varargs.size())
+              return false;
+            
+            for (size_t i = 0; i < varargs.size(); ++i)
+              if (varargs[i] < other.varargs[i])
+                return true;
+              else if (varargs[i] > other.varargs[i])
+                return false;
+          
             return false;
+          }
         }
       };
     
@@ -112,6 +125,7 @@ namespace {
       Expression create_expression(InsertElementInst* V);
       Expression create_expression(SelectInst* V);
       Expression create_expression(CastInst* C);
+      Expression create_expression(GetElementPtrInst* G);
     public:
       ValueTable() { nextValueNumber = 1; }
       uint32_t lookup_or_add(Value* V);
@@ -354,6 +368,22 @@ ValueTable::Expression ValueTable::create_expression(SelectInst* I) {
   return e;
 }
 
+ValueTable::Expression ValueTable::create_expression(GetElementPtrInst* G) {
+  Expression e;
+    
+  e.firstVN = lookup_or_add(G->getPointerOperand());
+  e.secondVN = 0;
+  e.thirdVN = 0;
+  e.type = G->getType();
+  e.opcode = Expression::SELECT;
+  
+  for (GetElementPtrInst::op_iterator I = G->idx_begin(), E = G->idx_end();
+       I != E; ++I)
+    e.varargs.push_back(lookup_or_add(*I));
+  
+  return e;
+}
+
 //===----------------------------------------------------------------------===//
 //                     ValueTable External Functions
 //===----------------------------------------------------------------------===//