[llvm-commits] CVS: llvm/lib/CodeGen/SelectionDAG/SelectionDAGCSEMap.cpp SelectionDAG.cpp

Chris Lattner lattner at cs.uiuc.edu
Mon Aug 7 16:03:19 PDT 2006



Changes in directory llvm/lib/CodeGen/SelectionDAG:

SelectionDAGCSEMap.cpp added (r1.1)
SelectionDAG.cpp updated: 1.318 -> 1.319
---
Log message:

Revamp the "CSEMap" datastructure used in the SelectionDAG class.  This 
eliminates a bunch of std::map's in the SelectionDAG, replacing them with a
home-grown hashtable.

This is still a work in progress: not all the maps have been moved over and the
hashtable never resizes.  That said, this still speeds up llc 20% on kimwitu++
with -fast -regalloc=local using a release build.


---
Diffs of the changes:  (+517 -312)

 SelectionDAG.cpp       |  590 +++++++++++++++++++++++--------------------------
 SelectionDAGCSEMap.cpp |  239 +++++++++++++++++++
 2 files changed, 517 insertions(+), 312 deletions(-)


Index: llvm/lib/CodeGen/SelectionDAG/SelectionDAGCSEMap.cpp
diff -c /dev/null llvm/lib/CodeGen/SelectionDAG/SelectionDAGCSEMap.cpp:1.1
*** /dev/null	Mon Aug  7 18:03:13 2006
--- llvm/lib/CodeGen/SelectionDAG/SelectionDAGCSEMap.cpp	Mon Aug  7 18:03:03 2006
***************
*** 0 ****
--- 1,239 ----
+ //===-- SelectionDAGCSEMap.cpp - Implement the SelectionDAG CSE Map -------===//
+ //
+ //                     The LLVM Compiler Infrastructure
+ //
+ // This file was developed by Chris Lattner and is distributed under
+ // the University of Illinois Open Source License. See LICENSE.TXT for details.
+ //
+ //===----------------------------------------------------------------------===//
+ //
+ // This implements the SelectionDAGCSEMap class.
+ //
+ //===----------------------------------------------------------------------===//
+ 
+ #include "llvm/CodeGen/SelectionDAG.h"
+ using namespace llvm;
+ 
+ //===----------------------------------------------------------------------===//
+ // SelectionDAGCSEMap::NodeID Implementation
+ 
+ SelectionDAGCSEMap::NodeID::NodeID(SDNode *N) {
+   SetOpcode(N->getOpcode());
+   // Add the return value info.
+   SetValueTypes(N->value_begin());
+   // Add the operand info.
+   SetOperands(N->op_begin(), N->getNumOperands());
+ }
+ 
+ SelectionDAGCSEMap::NodeID::NodeID(unsigned short ID, const void *VTList) {
+   SetOpcode(ID);
+   SetValueTypes(VTList);
+   SetOperands();
+ }
+ SelectionDAGCSEMap::NodeID::NodeID(unsigned short ID, const void *VTList,
+                                    SDOperand Op) {
+   SetOpcode(ID);
+   SetValueTypes(VTList);
+   SetOperands(Op);
+ }
+ SelectionDAGCSEMap::NodeID::NodeID(unsigned short ID, const void *VTList, 
+                                    SDOperand Op1, SDOperand Op2) {
+   SetOpcode(ID);
+   SetValueTypes(VTList);
+   SetOperands(Op1, Op2);
+ }
+ SelectionDAGCSEMap::NodeID::NodeID(unsigned short ID, const void *VTList, 
+                                    SDOperand Op1, SDOperand Op2,
+                                    SDOperand Op3) {
+   SetOpcode(ID);
+   SetValueTypes(VTList);
+   SetOperands(Op1, Op2, Op3);
+ }
+ SelectionDAGCSEMap::NodeID::NodeID(unsigned short ID, const void *VTList, 
+                                    const SDOperand *OpList, unsigned N) {
+   SetOpcode(ID);
+   SetValueTypes(VTList);
+   SetOperands(OpList, N);
+ }
+ 
+ void SelectionDAGCSEMap::NodeID::AddPointer(const void *Ptr) {
+   // Note: this adds pointers to the hash using sizes and endianness that depend
+   // on the host.  It doesn't matter however, because hashing on pointer values
+   // in inherently unstable.  Nothing in the SelectionDAG should depend on the
+   // ordering of nodes in the CSEMap.
+   union {
+     intptr_t PtrI;
+     unsigned char PtrA[sizeof(intptr_t)];
+   };
+   PtrI = (intptr_t)Ptr;
+   Bits.append(PtrA, PtrA+sizeof(intptr_t));
+ }
+ 
+ void SelectionDAGCSEMap::NodeID::AddOperand(SDOperand Op) {
+   AddPointer(Op.Val);
+   // 2 bytes of resno might be too small, three should certainly be enough. :)
+   assert(Op.ResNo < (1 << 24) && "ResNo too large for CSE Map to handle!");
+   Bits.push_back((Op.ResNo >>  0) & 0xFF);
+   Bits.push_back((Op.ResNo >>  8) & 0xFF);
+   Bits.push_back((Op.ResNo >> 16) & 0xFF);
+ }
+ 
+ void SelectionDAGCSEMap::NodeID::SetOperands(const SDOperand *Ops, 
+                                              unsigned NumOps) {
+   for (; NumOps; --NumOps, ++Ops)
+     AddOperand(*Ops);
+ }
+ 
+ 
+ /// ComputeHash - Compute a strong hash value for this NodeID, for lookup in
+ /// the SelectionDAGCSEMap.
+ unsigned SelectionDAGCSEMap::NodeID::ComputeHash() const {
+   // FIXME: this hash function sucks.
+   unsigned Hash = 0;
+   for (unsigned i = 0, e = Bits.size(); i != e; ++i)
+     Hash += Bits[i];
+   return Hash;
+ }
+ 
+ bool SelectionDAGCSEMap::NodeID::operator==(const NodeID &RHS) const {
+   if (Bits.size() != RHS.Bits.size()) return false;
+   return memcmp(&Bits[0], &RHS.Bits[0], Bits.size()) == 0;
+ }
+ 
+ 
+ //===----------------------------------------------------------------------===//
+ // SelectionDAGCSEMap Implementation
+ 
+ SelectionDAGCSEMap::SelectionDAGCSEMap() {
+   NumBuckets = 256;
+   Buckets = new void*[NumBuckets];
+   memset(Buckets, 0, NumBuckets*sizeof(void*));
+ }
+ SelectionDAGCSEMap::~SelectionDAGCSEMap() {
+   delete [] Buckets;
+ }
+ 
+ /// GetNextPtr - In order to save space, each bucket is a singly-linked-list. In
+ /// order to make deletion more efficient, we make the list circular, so we can
+ /// delete a node without computing its hash.  The problem with this is that the
+ /// start of the hash buckets are not SDNodes.  If NextInBucketPtr is a bucket
+ /// pointer, this method returns null: use GetBucketPtr when this happens.
+ SDNode *SelectionDAGCSEMap::GetNextPtr(void *NextInBucketPtr) {
+   if (NextInBucketPtr >= Buckets && NextInBucketPtr < Buckets+NumBuckets)
+     return 0;
+   return static_cast<SDNode*>(NextInBucketPtr);
+ }
+ 
+ void **SelectionDAGCSEMap::GetBucketPtr(void *NextInBucketPtr) {
+   assert(NextInBucketPtr >= Buckets && NextInBucketPtr < Buckets+NumBuckets &&
+          "NextInBucketPtr is not a bucket ptr");
+   return static_cast<void**>(NextInBucketPtr);
+ }
+ 
+ /// GetBucketFor - Hash the specified node ID and return the hash bucket for the
+ /// specified ID.
+ void **SelectionDAGCSEMap::GetBucketFor(const NodeID &ID) const {
+   // TODO: if load is high, resize hash table.
+   
+   // NumBuckets is always a power of 2.
+   unsigned BucketNum = ID.ComputeHash() & (NumBuckets-1);
+   return Buckets+BucketNum;
+ }
+ 
+ /// FindNodeOrInsertPos - Look up the node specified by ID.  If it exists,
+ /// return it.  If not, return the insertion token that will make insertion
+ /// faster.
+ SDNode *SelectionDAGCSEMap::FindNodeOrInsertPos(const NodeID &ID,
+                                                 void *&InsertPos) {
+   void **Bucket = GetBucketFor(ID);
+   void *Probe = *Bucket;
+   
+   InsertPos = 0;
+   
+   unsigned Opc = ID.getOpcode();
+   while (SDNode *NodeInBucket = GetNextPtr(Probe)) {
+     // If we found a node with the same opcode, it might be a matching node.
+     // Because it is in the same bucket as this one, we know the hash values
+     // match.  Compute the NodeID for the possible match and do a final compare.
+     if (NodeInBucket->getOpcode() == Opc) {
+       NodeID OtherID(NodeInBucket);
+       if (OtherID == ID)
+         return NodeInBucket;
+     }
+ 
+     Probe = NodeInBucket->getNextInBucket();
+   }
+   
+   // Didn't find the node, return null with the bucket as the InsertPos.
+   InsertPos = Bucket;
+   return 0;
+ }
+ 
+ /// InsertNode - Insert the specified node into the CSE Map, knowing that it
+ /// is not already in the map.  InsertPos must be obtained from 
+ /// FindNodeOrInsertPos.
+ void SelectionDAGCSEMap::InsertNode(SDNode *N, void *InsertPos) {
+   /// The insert position is actually a bucket pointer.
+   void **Bucket = static_cast<void**>(InsertPos);
+   
+   void *Next = *Bucket;
+   
+   // If this is the first insertion into this bucket, its next pointer will be
+   // null.  Pretend as if it pointed to itself.
+   if (Next == 0)
+     Next = Bucket;
+ 
+   // Set the nodes next pointer, and make the bucket point to the node.
+   N->SetNextInBucket(Next);
+   *Bucket = N;
+ }
+ 
+ 
+ /// RemoveNode - Remove a node from the CSE map, returning true if one was
+ /// removed or false if the node was not in the CSE map.
+ bool SelectionDAGCSEMap::RemoveNode(SDNode *N) {
+   // Because each bucket is a circular list, we don't need to compute N's hash
+   // to remove it.  Chase around the list until we find the node (or bucket)
+   // which points to N.
+   void *Ptr = N->getNextInBucket();
+   if (Ptr == 0) return false;  // Not in CSEMap.
+   
+   void *NodeNextPtr = Ptr;
+   N->SetNextInBucket(0);
+   while (1) {
+     if (SDNode *NodeInBucket = GetNextPtr(Ptr)) {
+       // Advance pointer.
+       Ptr = NodeInBucket->getNextInBucket();
+       
+       // We found a node that points to N, change it to point to N's next node,
+       // removing N from the list.
+       if (Ptr == N) {
+         NodeInBucket->SetNextInBucket(NodeNextPtr);
+         return true;
+       }
+     } else {
+       void **Bucket = GetBucketPtr(Ptr);
+       Ptr = *Bucket;
+       
+       // If we found that the bucket points to N, update the bucket to point to
+       // whatever is next.
+       if (Ptr == N) {
+         *Bucket = NodeNextPtr;
+         return true;
+       }
+     }
+   }
+ }
+ 
+ /// GetOrInsertSimpleNode - If there is an existing simple SDNode exactly
+ /// equal to the specified node, return it.  Otherwise, insert 'N' and it
+ /// instead.  This only works on *simple* SDNodes, not ConstantSDNode or any
+ /// other classes derived from SDNode.
+ SDNode *SelectionDAGCSEMap::GetOrInsertNode(SDNode *N) {
+   SelectionDAGCSEMap::NodeID ID(N);
+   void *IP;
+   if (SDNode *E = FindNodeOrInsertPos(ID, IP))
+     return E;
+   InsertNode(N, IP);
+   return N;
+ }


Index: llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp
diff -u llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp:1.318 llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp:1.319
--- llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp:1.318	Mon Aug  7 17:13:29 2006
+++ llvm/lib/CodeGen/SelectionDAG/SelectionDAG.cpp	Mon Aug  7 18:03:03 2006
@@ -29,7 +29,6 @@
 #include <set>
 #include <cmath>
 #include <algorithm>
-#include <queue>
 using namespace llvm;
 
 static bool isCommutativeBinOp(unsigned Opcode) {
@@ -430,40 +429,18 @@
     Erased =ValueNodes.erase(std::make_pair(SVN->getValue(), SVN->getOffset()));
     break;
   }    
-  case ISD::LOAD:
-    Erased = Loads.erase(std::make_pair(N->getOperand(1),
-                                        std::make_pair(N->getOperand(0),
-                                                       N->getValueType(0))));
-    break;
   default:
     if (N->getNumValues() == 1) {
       if (N->getNumOperands() == 0) {
         Erased = NullaryOps.erase(std::make_pair(N->getOpcode(),
                                                  N->getValueType(0)));
-      } else if (N->getNumOperands() == 1) {
-        Erased = 
-          UnaryOps.erase(std::make_pair(N->getOpcode(),
-                                        std::make_pair(N->getOperand(0),
-                                                       N->getValueType(0))));
-      } else if (N->getNumOperands() == 2) {
-        Erased = 
-          BinaryOps.erase(std::make_pair(N->getOpcode(),
-                                         std::make_pair(N->getOperand(0),
-                                                        N->getOperand(1))));
       } else { 
-        std::vector<SDOperand> Ops(N->op_begin(), N->op_end());
-        Erased = 
-          OneResultNodes.erase(std::make_pair(N->getOpcode(),
-                                              std::make_pair(N->getValueType(0),
-                                                             Ops)));
+        // Remove it from the CSE Map.
+        Erased = CSEMap.RemoveNode(N);
       }
     } else {
-      // Remove the node from the ArbitraryNodes map.
-      std::vector<MVT::ValueType> RV(N->value_begin(), N->value_end());
-      std::vector<SDOperand>     Ops(N->op_begin(), N->op_end());
-      Erased =
-        ArbitraryNodes.erase(std::make_pair(N->getOpcode(),
-                                            std::make_pair(RV, Ops)));
+      // Remove it from the CSE Map.
+      Erased = CSEMap.RemoveNode(N);
     }
     break;
   }
@@ -495,43 +472,8 @@
     if (N->getValueType(i) == MVT::Flag)
       return 0;   // Never CSE anything that produces a flag.
   
-  if (N->getNumValues() == 1) {
-    if (N->getNumOperands() == 1) {
-      SDNode *&U = UnaryOps[std::make_pair(N->getOpcode(),
-                                           std::make_pair(N->getOperand(0),
-                                                          N->getValueType(0)))];
-      if (U) return U;
-      U = N;
-    } else if (N->getNumOperands() == 2) {
-      SDNode *&B = BinaryOps[std::make_pair(N->getOpcode(),
-                                            std::make_pair(N->getOperand(0),
-                                                           N->getOperand(1)))];
-      if (B) return B;
-      B = N;
-    } else {
-      std::vector<SDOperand> Ops(N->op_begin(), N->op_end());
-      SDNode *&ORN = OneResultNodes[std::make_pair(N->getOpcode(),
-                                      std::make_pair(N->getValueType(0), Ops))];
-      if (ORN) return ORN;
-      ORN = N;
-    }
-  } else {  
-    if (N->getOpcode() == ISD::LOAD) {
-      SDNode *&L = Loads[std::make_pair(N->getOperand(1),
-                                        std::make_pair(N->getOperand(0),
-                                                       N->getValueType(0)))];
-      if (L) return L;
-      L = N;
-    } else {
-      // Remove the node from the ArbitraryNodes map.
-      std::vector<MVT::ValueType> RV(N->value_begin(), N->value_end());
-      std::vector<SDOperand>     Ops(N->op_begin(), N->op_end());
-      SDNode *&AN = ArbitraryNodes[std::make_pair(N->getOpcode(),
-                                                  std::make_pair(RV, Ops))];
-      if (AN) return AN;
-      AN = N;
-    }
-  }
+  SDNode *New = CSEMap.GetOrInsertNode(N);
+  if (New != N) return New;  // Node already existed.
   return 0;
 }
 
@@ -539,7 +481,8 @@
 /// were replaced with those specified.  If this node is never memoized, 
 /// return null, otherwise return a pointer to the slot it would take.  If a
 /// node already exists with these operands, the slot will be non-null.
-SDNode **SelectionDAG::FindModifiedNodeSlot(SDNode *N, SDOperand Op) {
+SDNode *SelectionDAG::FindModifiedNodeSlot(SDNode *N, SDOperand Op,
+                                           void *&InsertPos) {
   if (N->getOpcode() == ISD::HANDLENODE || N->getValueType(0) == MVT::Flag)
     return 0;    // Never add these nodes.
   
@@ -548,26 +491,20 @@
     if (N->getValueType(i) == MVT::Flag)
       return 0;   // Never CSE anything that produces a flag.
   
-  if (N->getNumValues() == 1) {
-    return &UnaryOps[std::make_pair(N->getOpcode(),
-                                    std::make_pair(Op, N->getValueType(0)))];
-  } else {  
-    // Remove the node from the ArbitraryNodes map.
-    std::vector<MVT::ValueType> RV(N->value_begin(), N->value_end());
-    std::vector<SDOperand> Ops;
-    Ops.push_back(Op);
-    return &ArbitraryNodes[std::make_pair(N->getOpcode(),
-                                          std::make_pair(RV, Ops))];
-  }
-  return 0;
+  SelectionDAGCSEMap::NodeID ID;
+  ID.SetOpcode(N->getOpcode());
+  ID.SetValueTypes(N->value_begin());
+  ID.SetOperands(Op);
+  return CSEMap.FindNodeOrInsertPos(ID, InsertPos);
 }
 
 /// FindModifiedNodeSlot - Find a slot for the specified node if its operands
 /// were replaced with those specified.  If this node is never memoized, 
 /// return null, otherwise return a pointer to the slot it would take.  If a
 /// node already exists with these operands, the slot will be non-null.
-SDNode **SelectionDAG::FindModifiedNodeSlot(SDNode *N, 
-                                            SDOperand Op1, SDOperand Op2) {
+SDNode *SelectionDAG::FindModifiedNodeSlot(SDNode *N, 
+                                           SDOperand Op1, SDOperand Op2,
+                                           void *&InsertPos) {
   if (N->getOpcode() == ISD::HANDLENODE || N->getValueType(0) == MVT::Flag)
     return 0;    // Never add these nodes.
   
@@ -575,19 +512,12 @@
   for (unsigned i = 1, e = N->getNumValues(); i != e; ++i)
     if (N->getValueType(i) == MVT::Flag)
       return 0;   // Never CSE anything that produces a flag.
-  
-  if (N->getNumValues() == 1) {
-    return &BinaryOps[std::make_pair(N->getOpcode(),
-                                     std::make_pair(Op1, Op2))];
-  } else {  
-    std::vector<MVT::ValueType> RV(N->value_begin(), N->value_end());
-    std::vector<SDOperand> Ops;
-    Ops.push_back(Op1);
-    Ops.push_back(Op2);
-    return &ArbitraryNodes[std::make_pair(N->getOpcode(),
-                                          std::make_pair(RV, Ops))];
-  }
-  return 0;
+                                              
+  SelectionDAGCSEMap::NodeID ID;
+  ID.SetOpcode(N->getOpcode());
+  ID.SetValueTypes(N->value_begin());
+  ID.SetOperands(Op1, Op2);
+  return CSEMap.FindNodeOrInsertPos(ID, InsertPos);
 }
 
 
@@ -595,8 +525,9 @@
 /// were replaced with those specified.  If this node is never memoized, 
 /// return null, otherwise return a pointer to the slot it would take.  If a
 /// node already exists with these operands, the slot will be non-null.
-SDNode **SelectionDAG::FindModifiedNodeSlot(SDNode *N, 
-                                            const std::vector<SDOperand> &Ops) {
+SDNode *SelectionDAG::FindModifiedNodeSlot(SDNode *N, 
+                                           const std::vector<SDOperand> &Ops,
+                                           void *&InsertPos) {
   if (N->getOpcode() == ISD::HANDLENODE || N->getValueType(0) == MVT::Flag)
     return 0;    // Never add these nodes.
   
@@ -605,30 +536,11 @@
     if (N->getValueType(i) == MVT::Flag)
       return 0;   // Never CSE anything that produces a flag.
   
-  if (N->getNumValues() == 1) {
-    if (N->getNumOperands() == 1) {
-      return &UnaryOps[std::make_pair(N->getOpcode(),
-                                      std::make_pair(Ops[0],
-                                                     N->getValueType(0)))];
-    } else if (N->getNumOperands() == 2) {
-      return &BinaryOps[std::make_pair(N->getOpcode(),
-                                       std::make_pair(Ops[0], Ops[1]))];
-    } else {
-      return &OneResultNodes[std::make_pair(N->getOpcode(),
-                                            std::make_pair(N->getValueType(0),
-                                                           Ops))];
-    }
-  } else {  
-    if (N->getOpcode() == ISD::LOAD) {
-      return &Loads[std::make_pair(Ops[1],
-                                   std::make_pair(Ops[0], N->getValueType(0)))];
-    } else {
-      std::vector<MVT::ValueType> RV(N->value_begin(), N->value_end());
-      return &ArbitraryNodes[std::make_pair(N->getOpcode(),
-                                            std::make_pair(RV, Ops))];
-    }
-  }
-  return 0;
+  SelectionDAGCSEMap::NodeID ID;
+  ID.SetOpcode(N->getOpcode());
+  ID.SetValueTypes(N->value_begin());
+  ID.SetOperands(&Ops[0], Ops.size());
+  return CSEMap.FindNodeOrInsertPos(ID, InsertPos);
 }
 
 
@@ -1235,14 +1147,19 @@
   }
 
   SDNode *N;
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
   if (VT != MVT::Flag) { // Don't CSE flag producing nodes
-    SDNode *&E = UnaryOps[std::make_pair(Opcode, std::make_pair(Operand, VT))];
-    if (E) return SDOperand(E, 0);
-    E = N = new SDNode(Opcode, Operand);
+    SelectionDAGCSEMap::NodeID ID(Opcode, VTs, Operand);
+    void *IP = 0;
+    if (SDNode *E = CSEMap.FindNodeOrInsertPos(ID, IP))
+      return SDOperand(E, 0);
+    N = new SDNode(Opcode, Operand);
+    N->setValueTypes(VTs, 1);
+    CSEMap.InsertNode(N, IP);
   } else {
     N = new SDNode(Opcode, Operand);
+    N->setValueTypes(VTs, 1);
   }
-  N->setValueTypes(VT);
   AllNodes.push_back(N);
   return SDOperand(N, 0);
 }
@@ -1502,16 +1419,20 @@
 
   // Memoize this node if possible.
   SDNode *N;
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
   if (VT != MVT::Flag) {
-    SDNode *&BON = BinaryOps[std::make_pair(Opcode, std::make_pair(N1, N2))];
-    if (BON) return SDOperand(BON, 0);
-
-    BON = N = new SDNode(Opcode, N1, N2);
+    SelectionDAGCSEMap::NodeID ID(Opcode, VTs, N1, N2);
+    void *IP = 0;
+    if (SDNode *E = CSEMap.FindNodeOrInsertPos(ID, IP))
+      return SDOperand(E, 0);
+    N = new SDNode(Opcode, N1, N2);
+    N->setValueTypes(VTs, 1);
+    CSEMap.InsertNode(N, IP);
   } else {
     N = new SDNode(Opcode, N1, N2);
+    N->setValueTypes(VTs, 1);
   }
 
-  N->setValueTypes(VT);
   AllNodes.push_back(N);
   return SDOperand(N, 0);
 }
@@ -1554,22 +1475,22 @@
     break;
   }
 
-  std::vector<SDOperand> Ops;
-  Ops.reserve(3);
-  Ops.push_back(N1);
-  Ops.push_back(N2);
-  Ops.push_back(N3);
-
   // Memoize node if it doesn't produce a flag.
   SDNode *N;
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
+
   if (VT != MVT::Flag) {
-    SDNode *&E = OneResultNodes[std::make_pair(Opcode,std::make_pair(VT, Ops))];
-    if (E) return SDOperand(E, 0);
-    E = N = new SDNode(Opcode, N1, N2, N3);
+    SelectionDAGCSEMap::NodeID ID(Opcode, VTs, N1, N2, N3);
+    void *IP = 0;
+    if (SDNode *E = CSEMap.FindNodeOrInsertPos(ID, IP))
+      return SDOperand(E, 0);
+    N = new SDNode(Opcode, N1, N2, N3);
+    N->setValueTypes(VTs, 1);
+    CSEMap.InsertNode(N, IP);
   } else {
     N = new SDNode(Opcode, N1, N2, N3);
+    N->setValueTypes(VTs, 1);
   }
-  N->setValueTypes(VT);
   AllNodes.push_back(N);
   return SDOperand(N, 0);
 }
@@ -1602,12 +1523,15 @@
 SDOperand SelectionDAG::getLoad(MVT::ValueType VT,
                                 SDOperand Chain, SDOperand Ptr,
                                 SDOperand SV) {
-  SDNode *&N = Loads[std::make_pair(Ptr, std::make_pair(Chain, VT))];
-  if (N) return SDOperand(N, 0);
-  N = new SDNode(ISD::LOAD, Chain, Ptr, SV);
-
-  // Loads have a token chain.
-  setNodeValueTypes(N, VT, MVT::Other);
+  MVT::ValueType *VTs = getNodeValueTypes(VT, MVT::Other);
+  
+  SelectionDAGCSEMap::NodeID ID(ISD::LOAD, VTs, Chain, Ptr, SV);
+  void *IP = 0;
+  if (SDNode *E = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(E, 0);
+  SDNode *N = new SDNode(ISD::LOAD, Chain, Ptr, SV);
+  N->setValueTypes(VTs, 2);
+  CSEMap.InsertNode(N, IP);
   AllNodes.push_back(N);
   return SDOperand(N, 0);
 }
@@ -1720,15 +1644,19 @@
 
   // Memoize nodes.
   SDNode *N;
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
   if (VT != MVT::Flag) {
-    SDNode *&E =
-      OneResultNodes[std::make_pair(Opcode, std::make_pair(VT, Ops))];
-    if (E) return SDOperand(E, 0);
-    E = N = new SDNode(Opcode, Ops);
+    SelectionDAGCSEMap::NodeID ID(Opcode, VTs, &Ops[0], Ops.size());
+    void *IP = 0;
+    if (SDNode *E = CSEMap.FindNodeOrInsertPos(ID, IP))
+      return SDOperand(E, 0);
+    N = new SDNode(Opcode, Ops);
+    N->setValueTypes(VTs, 1);
+    CSEMap.InsertNode(N, IP);
   } else {
     N = new SDNode(Opcode, Ops);
+    N->setValueTypes(VTs, 1);
   }
-  N->setValueTypes(VT);
   AllNodes.push_back(N);
   return SDOperand(N, 0);
 }
@@ -1787,25 +1715,36 @@
 
   // Memoize the node unless it returns a flag.
   SDNode *N;
+  MVT::ValueType *VTs = getNodeValueTypes(ResultTys);
   if (ResultTys.back() != MVT::Flag) {
-    SDNode *&E =
-      ArbitraryNodes[std::make_pair(Opcode, std::make_pair(ResultTys, Ops))];
-    if (E) return SDOperand(E, 0);
-    E = N = new SDNode(Opcode, Ops);
+    SelectionDAGCSEMap::NodeID ID;
+    ID.SetOpcode(Opcode);
+    ID.SetValueTypes(VTs);
+    ID.SetOperands(&Ops[0], Ops.size());
+    void *IP = 0;
+    if (SDNode *E = CSEMap.FindNodeOrInsertPos(ID, IP))
+      return SDOperand(E, 0);
+    N = new SDNode(Opcode, Ops);
+    N->setValueTypes(VTs, ResultTys.size());
+    CSEMap.InsertNode(N, IP);
   } else {
     N = new SDNode(Opcode, Ops);
+    N->setValueTypes(VTs, ResultTys.size());
   }
-  setNodeValueTypes(N, ResultTys);
   AllNodes.push_back(N);
   return SDOperand(N, 0);
 }
 
-void SelectionDAG::setNodeValueTypes(SDNode *N, 
-                                     std::vector<MVT::ValueType> &RetVals) {
+MVT::ValueType *SelectionDAG::getNodeValueTypes(MVT::ValueType VT) {
+  return SDNode::getValueTypeList(VT);
+}
+
+MVT::ValueType *SelectionDAG::getNodeValueTypes(
+                                        std::vector<MVT::ValueType> &RetVals) {
   switch (RetVals.size()) {
-  case 0: return;
-  case 1: N->setValueTypes(RetVals[0]); return;
-  case 2: setNodeValueTypes(N, RetVals[0], RetVals[1]); return;
+  case 0: assert(0 && "Cannot have nodes without results!");
+  case 1: return SDNode::getValueTypeList(RetVals[0]);
+  case 2: return getNodeValueTypes(RetVals[0], RetVals[1]);
   default: break;
   }
   
@@ -1816,23 +1755,21 @@
     I = VTList.begin();
   }
 
-  N->setValueTypes(&(*I)[0], I->size());
+  return &(*I)[0];
 }
 
-void SelectionDAG::setNodeValueTypes(SDNode *N, MVT::ValueType VT1, 
-                                     MVT::ValueType VT2) {
+MVT::ValueType *SelectionDAG::getNodeValueTypes(MVT::ValueType VT1, 
+                                                MVT::ValueType VT2) {
   for (std::list<std::vector<MVT::ValueType> >::iterator I = VTList.begin(),
        E = VTList.end(); I != E; ++I) {
-    if (I->size() == 2 && (*I)[0] == VT1 && (*I)[1] == VT2) {
-      N->setValueTypes(&(*I)[0], 2);
-      return;
-    }
+    if (I->size() == 2 && (*I)[0] == VT1 && (*I)[1] == VT2)
+      return &(*I)[0];
   }
   std::vector<MVT::ValueType> V;
   V.push_back(VT1);
   V.push_back(VT2);
   VTList.push_front(V);
-  N->setValueTypes(&(*VTList.begin())[0], 2);
+  return &(*VTList.begin())[0];
 }
 
 /// UpdateNodeOperands - *Mutate* the specified node in-place to have the
@@ -1850,12 +1787,12 @@
   if (Op == N->getOperand(0)) return InN;
   
   // See if the modified node already exists.
-  SDNode **NewSlot = FindModifiedNodeSlot(N, Op);
-  if (NewSlot && *NewSlot)
-    return SDOperand(*NewSlot, InN.ResNo);
+  void *InsertPos = 0;
+  if (SDNode *Existing = FindModifiedNodeSlot(N, Op, InsertPos))
+    return SDOperand(Existing, InN.ResNo);
   
   // Nope it doesn't.  Remove the node from it's current place in the maps.
-  if (NewSlot)
+  if (InsertPos)
     RemoveNodeFromCSEMaps(N);
   
   // Now we update the operands.
@@ -1864,7 +1801,7 @@
   N->OperandList[0] = Op;
   
   // If this gets put into a CSE map, add it.
-  if (NewSlot) *NewSlot = N;
+  if (InsertPos) CSEMap.InsertNode(N, InsertPos);
   return InN;
 }
 
@@ -1879,12 +1816,12 @@
     return InN;   // No operands changed, just return the input node.
   
   // See if the modified node already exists.
-  SDNode **NewSlot = FindModifiedNodeSlot(N, Op1, Op2);
-  if (NewSlot && *NewSlot)
-    return SDOperand(*NewSlot, InN.ResNo);
+  void *InsertPos = 0;
+  if (SDNode *Existing = FindModifiedNodeSlot(N, Op1, Op2, InsertPos))
+    return SDOperand(Existing, InN.ResNo);
   
   // Nope it doesn't.  Remove the node from it's current place in the maps.
-  if (NewSlot)
+  if (InsertPos)
     RemoveNodeFromCSEMaps(N);
   
   // Now we update the operands.
@@ -1900,7 +1837,7 @@
   }
   
   // If this gets put into a CSE map, add it.
-  if (NewSlot) *NewSlot = N;
+  if (InsertPos) CSEMap.InsertNode(N, InsertPos);
   return InN;
 }
 
@@ -1957,12 +1894,12 @@
   if (!AnyChange) return InN;
   
   // See if the modified node already exists.
-  SDNode **NewSlot = FindModifiedNodeSlot(N, Ops);
-  if (NewSlot && *NewSlot)
-    return SDOperand(*NewSlot, InN.ResNo);
+  void *InsertPos = 0;
+  if (SDNode *Existing = FindModifiedNodeSlot(N, Ops, InsertPos))
+    return SDOperand(Existing, InN.ResNo);
   
   // Nope it doesn't.  Remove the node from it's current place in the maps.
-  if (NewSlot)
+  if (InsertPos)
     RemoveNodeFromCSEMaps(N);
   
   // Now we update the operands.
@@ -1975,7 +1912,7 @@
   }
 
   // If this gets put into a CSE map, add it.
-  if (NewSlot) *NewSlot = N;
+  if (InsertPos) CSEMap.InsertNode(N, InsertPos);
   return InN;
 }
 
@@ -1999,7 +1936,7 @@
   RemoveNodeFromCSEMaps(N);
   
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  N->setValueTypes(VT);
+  N->setValueTypes(getNodeValueTypes(VT), 1);
 
   ON = N;   // Memoize the new node.
   return SDOperand(N, 0);
@@ -2008,16 +1945,17 @@
 SDOperand SelectionDAG::SelectNodeTo(SDNode *N, unsigned TargetOpc,
                                      MVT::ValueType VT, SDOperand Op1) {
   // If an identical node already exists, use it.
-  SDNode *&ON = UnaryOps[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                        std::make_pair(Op1, VT))];
-  if (ON) return SDOperand(ON, 0);
-  
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs, Op1);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
+                                       
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  N->setValueTypes(VT);
+  N->setValueTypes(getNodeValueTypes(VT), 1);
   N->setOperands(Op1);
-  
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);
   return SDOperand(N, 0);
 }
 
@@ -2025,16 +1963,18 @@
                                      MVT::ValueType VT, SDOperand Op1,
                                      SDOperand Op2) {
   // If an identical node already exists, use it.
-  SDNode *&ON = BinaryOps[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                         std::make_pair(Op1, Op2))];
-  if (ON) return SDOperand(ON, 0);
-  
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs, Op1, Op2);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
+                                       
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  N->setValueTypes(VT);
+  N->setValueTypes(VTs, 1);
   N->setOperands(Op1, Op2);
   
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 
@@ -2042,18 +1982,18 @@
                                      MVT::ValueType VT, SDOperand Op1,
                                      SDOperand Op2, SDOperand Op3) {
   // If an identical node already exists, use it.
-  std::vector<SDOperand> OpList;
-  OpList.push_back(Op1); OpList.push_back(Op2); OpList.push_back(Op3);
-  SDNode *&ON = OneResultNodes[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                              std::make_pair(VT, OpList))];
-  if (ON) return SDOperand(ON, 0);
-  
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs, Op1, Op2, Op3);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
+                                       
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  N->setValueTypes(VT);
+  N->setValueTypes(VTs, 1);
   N->setOperands(Op1, Op2, Op3);
 
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 
@@ -2062,40 +2002,46 @@
                                      SDOperand Op2, SDOperand Op3,
                                      SDOperand Op4) {
   // If an identical node already exists, use it.
-  std::vector<SDOperand> OpList;
-  OpList.push_back(Op1); OpList.push_back(Op2); OpList.push_back(Op3);
-  OpList.push_back(Op4);
-  SDNode *&ON = OneResultNodes[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                              std::make_pair(VT, OpList))];
-  if (ON) return SDOperand(ON, 0);
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs);
+  ID.AddOperand(Op1);
+  ID.AddOperand(Op2);
+  ID.AddOperand(Op3);
+  ID.AddOperand(Op4);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
   
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  N->setValueTypes(VT);
+  N->setValueTypes(VTs, 1);
   N->setOperands(Op1, Op2, Op3, Op4);
 
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 
 SDOperand SelectionDAG::SelectNodeTo(SDNode *N, unsigned TargetOpc,
                                      MVT::ValueType VT, SDOperand Op1,
-                                     SDOperand Op2, SDOperand Op3,SDOperand Op4,
-                                     SDOperand Op5) {
-  // If an identical node already exists, use it.
-  std::vector<SDOperand> OpList;
-  OpList.push_back(Op1); OpList.push_back(Op2); OpList.push_back(Op3);
-  OpList.push_back(Op4); OpList.push_back(Op5);
-  SDNode *&ON = OneResultNodes[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                              std::make_pair(VT, OpList))];
-  if (ON) return SDOperand(ON, 0);
-  
+                                     SDOperand Op2, SDOperand Op3,
+                                     SDOperand Op4, SDOperand Op5) {
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs);
+  ID.AddOperand(Op1);
+  ID.AddOperand(Op2);
+  ID.AddOperand(Op3);
+  ID.AddOperand(Op4);
+  ID.AddOperand(Op5);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
+                                       
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  N->setValueTypes(VT);
+  N->setValueTypes(VTs, 1);
   N->setOperands(Op1, Op2, Op3, Op4, Op5);
   
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 
@@ -2103,20 +2049,24 @@
                                      MVT::ValueType VT, SDOperand Op1,
                                      SDOperand Op2, SDOperand Op3,SDOperand Op4,
                                      SDOperand Op5, SDOperand Op6) {
-  // If an identical node already exists, use it.
-  std::vector<SDOperand> OpList;
-  OpList.push_back(Op1); OpList.push_back(Op2); OpList.push_back(Op3);
-  OpList.push_back(Op4); OpList.push_back(Op5); OpList.push_back(Op6);
-  SDNode *&ON = OneResultNodes[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                              std::make_pair(VT, OpList))];
-  if (ON) return SDOperand(ON, 0);
-
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs);
+  ID.AddOperand(Op1);
+  ID.AddOperand(Op2);
+  ID.AddOperand(Op3);
+  ID.AddOperand(Op4);
+  ID.AddOperand(Op5);
+  ID.AddOperand(Op6);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
+                                       
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  N->setValueTypes(VT);
+  N->setValueTypes(VTs, 1);
   N->setOperands(Op1, Op2, Op3, Op4, Op5, Op6);
   
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 
@@ -2125,21 +2075,26 @@
                                      SDOperand Op2, SDOperand Op3,SDOperand Op4,
                                      SDOperand Op5, SDOperand Op6,
 				     SDOperand Op7) {
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
   // If an identical node already exists, use it.
-  std::vector<SDOperand> OpList;
-  OpList.push_back(Op1); OpList.push_back(Op2); OpList.push_back(Op3);
-  OpList.push_back(Op4); OpList.push_back(Op5); OpList.push_back(Op6);
-  OpList.push_back(Op7);
-  SDNode *&ON = OneResultNodes[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                              std::make_pair(VT, OpList))];
-  if (ON) return SDOperand(ON, 0);
-
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs);
+  ID.AddOperand(Op1);
+  ID.AddOperand(Op2);
+  ID.AddOperand(Op3);
+  ID.AddOperand(Op4);
+  ID.AddOperand(Op5);
+  ID.AddOperand(Op6);
+  ID.AddOperand(Op7);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
+                                       
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  N->setValueTypes(VT);
+  N->setValueTypes(VTs, 1);
   N->setOperands(Op1, Op2, Op3, Op4, Op5, Op6, Op7);
   
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 SDOperand SelectionDAG::SelectNodeTo(SDNode *N, unsigned TargetOpc,
@@ -2148,41 +2103,44 @@
                                      SDOperand Op5, SDOperand Op6,
 				     SDOperand Op7, SDOperand Op8) {
   // If an identical node already exists, use it.
-  std::vector<SDOperand> OpList;
-  OpList.push_back(Op1); OpList.push_back(Op2); OpList.push_back(Op3);
-  OpList.push_back(Op4); OpList.push_back(Op5); OpList.push_back(Op6);
-  OpList.push_back(Op7); OpList.push_back(Op8);
-  SDNode *&ON = OneResultNodes[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                              std::make_pair(VT, OpList))];
-  if (ON) return SDOperand(ON, 0);
-
+  MVT::ValueType *VTs = getNodeValueTypes(VT);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs);
+  ID.AddOperand(Op1);
+  ID.AddOperand(Op2);
+  ID.AddOperand(Op3);
+  ID.AddOperand(Op4);
+  ID.AddOperand(Op5);
+  ID.AddOperand(Op6);
+  ID.AddOperand(Op7);
+  ID.AddOperand(Op8);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
+                                       
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  N->setValueTypes(VT);
+  N->setValueTypes(VTs, 1);
   N->setOperands(Op1, Op2, Op3, Op4, Op5, Op6, Op7, Op8);
   
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 
 SDOperand SelectionDAG::SelectNodeTo(SDNode *N, unsigned TargetOpc, 
                                      MVT::ValueType VT1, MVT::ValueType VT2,
                                      SDOperand Op1, SDOperand Op2) {
-  // If an identical node already exists, use it.
-  std::vector<SDOperand> OpList;
-  OpList.push_back(Op1); OpList.push_back(Op2); 
-  std::vector<MVT::ValueType> VTList;
-  VTList.push_back(VT1); VTList.push_back(VT2);
-  SDNode *&ON = ArbitraryNodes[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                              std::make_pair(VTList, OpList))];
-  if (ON) return SDOperand(ON, 0);
+  MVT::ValueType *VTs = getNodeValueTypes(VT1, VT2);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs, Op1, Op2);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
 
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  setNodeValueTypes(N, VT1, VT2);
+  N->setValueTypes(VTs, 2);
   N->setOperands(Op1, Op2);
   
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 
@@ -2191,20 +2149,19 @@
                                      SDOperand Op1, SDOperand Op2, 
                                      SDOperand Op3) {
   // If an identical node already exists, use it.
-  std::vector<SDOperand> OpList;
-  OpList.push_back(Op1); OpList.push_back(Op2); OpList.push_back(Op3);
-  std::vector<MVT::ValueType> VTList;
-  VTList.push_back(VT1); VTList.push_back(VT2);
-  SDNode *&ON = ArbitraryNodes[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                              std::make_pair(VTList, OpList))];
-  if (ON) return SDOperand(ON, 0);
+  MVT::ValueType *VTs = getNodeValueTypes(VT1, VT2);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs,
+                                Op1, Op2, Op3);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
 
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  setNodeValueTypes(N, VT1, VT2);
+  N->setValueTypes(VTs, 2);
   N->setOperands(Op1, Op2, Op3);
   
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 
@@ -2213,21 +2170,22 @@
                                      SDOperand Op1, SDOperand Op2,
                                      SDOperand Op3, SDOperand Op4) {
   // If an identical node already exists, use it.
-  std::vector<SDOperand> OpList;
-  OpList.push_back(Op1); OpList.push_back(Op2); OpList.push_back(Op3);
-  OpList.push_back(Op4);
-  std::vector<MVT::ValueType> VTList;
-  VTList.push_back(VT1); VTList.push_back(VT2);
-  SDNode *&ON = ArbitraryNodes[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                              std::make_pair(VTList, OpList))];
-  if (ON) return SDOperand(ON, 0);
-
+  MVT::ValueType *VTs = getNodeValueTypes(VT1, VT2);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs);
+  ID.AddOperand(Op1);
+  ID.AddOperand(Op2);
+  ID.AddOperand(Op3);
+  ID.AddOperand(Op4);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
+                                       
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  setNodeValueTypes(N, VT1, VT2);
+  N->setValueTypes(VTs, 2);
   N->setOperands(Op1, Op2, Op3, Op4);
 
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 
@@ -2237,21 +2195,23 @@
                                      SDOperand Op3, SDOperand Op4, 
                                      SDOperand Op5) {
   // If an identical node already exists, use it.
-  std::vector<SDOperand> OpList;
-  OpList.push_back(Op1); OpList.push_back(Op2); OpList.push_back(Op3);
-  OpList.push_back(Op4); OpList.push_back(Op5);
-  std::vector<MVT::ValueType> VTList;
-  VTList.push_back(VT1); VTList.push_back(VT2);
-  SDNode *&ON = ArbitraryNodes[std::make_pair(ISD::BUILTIN_OP_END+TargetOpc,
-                                              std::make_pair(VTList, OpList))];
-  if (ON) return SDOperand(ON, 0);
-
+  MVT::ValueType *VTs = getNodeValueTypes(VT1, VT2);
+  SelectionDAGCSEMap::NodeID ID(ISD::BUILTIN_OP_END+TargetOpc, VTs);
+  ID.AddOperand(Op1);
+  ID.AddOperand(Op2);
+  ID.AddOperand(Op3);
+  ID.AddOperand(Op4);
+  ID.AddOperand(Op5);
+  void *IP = 0;
+  if (SDNode *ON = CSEMap.FindNodeOrInsertPos(ID, IP))
+    return SDOperand(ON, 0);
+                                       
   RemoveNodeFromCSEMaps(N);
   N->MorphNodeTo(ISD::BUILTIN_OP_END+TargetOpc);
-  setNodeValueTypes(N, VT1, VT2);
+  N->setValueTypes(VTs, 2);
   N->setOperands(Op1, Op2, Op3, Op4, Op5);
   
-  ON = N;   // Memoize the new node.
+  CSEMap.InsertNode(N, IP);   // Memoize the new node.
   return SDOperand(N, 0);
 }
 
@@ -2288,7 +2248,8 @@
 }
 SDNode *SelectionDAG::getTargetNode(unsigned Opcode, MVT::ValueType VT,
                                     SDOperand Op1, SDOperand Op2, SDOperand Op3,
-                                    SDOperand Op4, SDOperand Op5, SDOperand Op6) {
+                                    SDOperand Op4, SDOperand Op5,
+                                    SDOperand Op6) {
   std::vector<SDOperand> Ops;
   Ops.reserve(6);
   Ops.push_back(Op1);
@@ -2344,7 +2305,8 @@
   return getNode(ISD::BUILTIN_OP_END+Opcode, ResultTys, Ops).Val;
 }
 SDNode *SelectionDAG::getTargetNode(unsigned Opcode, MVT::ValueType VT1,
-                                    MVT::ValueType VT2, SDOperand Op1, SDOperand Op2) {
+                                    MVT::ValueType VT2, SDOperand Op1,
+                                    SDOperand Op2) {
   std::vector<MVT::ValueType> ResultTys;
   ResultTys.push_back(VT1);
   ResultTys.push_back(VT2);
@@ -2354,8 +2316,8 @@
   return getNode(ISD::BUILTIN_OP_END+Opcode, ResultTys, Ops).Val;
 }
 SDNode *SelectionDAG::getTargetNode(unsigned Opcode, MVT::ValueType VT1,
-                                    MVT::ValueType VT2, SDOperand Op1, SDOperand Op2,
-                                    SDOperand Op3) {
+                                    MVT::ValueType VT2, SDOperand Op1,
+                                    SDOperand Op2, SDOperand Op3) {
   std::vector<MVT::ValueType> ResultTys;
   ResultTys.push_back(VT1);
   ResultTys.push_back(VT2);
@@ -2366,8 +2328,9 @@
   return getNode(ISD::BUILTIN_OP_END+Opcode, ResultTys, Ops).Val;
 }
 SDNode *SelectionDAG::getTargetNode(unsigned Opcode, MVT::ValueType VT1,
-                                    MVT::ValueType VT2, SDOperand Op1, SDOperand Op2,
-                                    SDOperand Op3, SDOperand Op4) {
+                                    MVT::ValueType VT2, SDOperand Op1,
+                                    SDOperand Op2, SDOperand Op3, 
+                                    SDOperand Op4) {
   std::vector<MVT::ValueType> ResultTys;
   ResultTys.push_back(VT1);
   ResultTys.push_back(VT2);
@@ -2379,8 +2342,9 @@
   return getNode(ISD::BUILTIN_OP_END+Opcode, ResultTys, Ops).Val;
 }
 SDNode *SelectionDAG::getTargetNode(unsigned Opcode, MVT::ValueType VT1,
-                                    MVT::ValueType VT2, SDOperand Op1, SDOperand Op2,
-                                    SDOperand Op3, SDOperand Op4, SDOperand Op5) {
+                                    MVT::ValueType VT2, SDOperand Op1,
+                                    SDOperand Op2, SDOperand Op3, SDOperand Op4,
+                                    SDOperand Op5) {
   std::vector<MVT::ValueType> ResultTys;
   ResultTys.push_back(VT1);
   ResultTys.push_back(VT2);
@@ -2393,9 +2357,9 @@
   return getNode(ISD::BUILTIN_OP_END+Opcode, ResultTys, Ops).Val;
 }
 SDNode *SelectionDAG::getTargetNode(unsigned Opcode, MVT::ValueType VT1,
-                                    MVT::ValueType VT2, SDOperand Op1, SDOperand Op2,
-                                    SDOperand Op3, SDOperand Op4, SDOperand Op5,
-                                    SDOperand Op6) {
+                                    MVT::ValueType VT2, SDOperand Op1,
+                                    SDOperand Op2, SDOperand Op3, SDOperand Op4,
+                                    SDOperand Op5, SDOperand Op6) {
   std::vector<MVT::ValueType> ResultTys;
   ResultTys.push_back(VT1);
   ResultTys.push_back(VT2);
@@ -2409,9 +2373,10 @@
   return getNode(ISD::BUILTIN_OP_END+Opcode, ResultTys, Ops).Val;
 }
 SDNode *SelectionDAG::getTargetNode(unsigned Opcode, MVT::ValueType VT1,
-                                    MVT::ValueType VT2, SDOperand Op1, SDOperand Op2,
-                                    SDOperand Op3, SDOperand Op4, SDOperand Op5,
-                                    SDOperand Op6, SDOperand Op7) {
+                                    MVT::ValueType VT2, SDOperand Op1,
+                                    SDOperand Op2, SDOperand Op3, SDOperand Op4,
+                                    SDOperand Op5, SDOperand Op6,
+                                    SDOperand Op7) {
   std::vector<MVT::ValueType> ResultTys;
   ResultTys.push_back(VT1);
   ResultTys.push_back(VT2);
@@ -2440,7 +2405,8 @@
 SDNode *SelectionDAG::getTargetNode(unsigned Opcode, MVT::ValueType VT1,
                                     MVT::ValueType VT2, MVT::ValueType VT3,
                                     SDOperand Op1, SDOperand Op2,
-                                    SDOperand Op3, SDOperand Op4, SDOperand Op5) {
+                                    SDOperand Op3, SDOperand Op4,
+                                    SDOperand Op5) {
   std::vector<MVT::ValueType> ResultTys;
   ResultTys.push_back(VT1);
   ResultTys.push_back(VT2);
@@ -2491,7 +2457,8 @@
   return getNode(ISD::BUILTIN_OP_END+Opcode, ResultTys, Ops).Val;
 }
 SDNode *SelectionDAG::getTargetNode(unsigned Opcode, MVT::ValueType VT1, 
-                                    MVT::ValueType VT2, std::vector<SDOperand> &Ops) {
+                                    MVT::ValueType VT2,
+                                    std::vector<SDOperand> &Ops) {
   std::vector<MVT::ValueType> ResultTys;
   ResultTys.push_back(VT1);
   ResultTys.push_back(VT2);
@@ -2744,7 +2711,6 @@
 void SDNode::ANCHOR() {
 }
 
-
 /// getValueTypeList - Return a pointer to the specified value type.
 ///
 MVT::ValueType *SDNode::getValueTypeList(MVT::ValueType VT) {
@@ -2752,7 +2718,7 @@
   VTs[VT] = VT;
   return &VTs[VT];
 }
-
+  
 /// hasNUsesOfValue - Return true if there are exactly NUSES uses of the
 /// indicated value.  This method ignores uses of other values defined by this
 /// operation.






More information about the llvm-commits mailing list