Add new dominator tree node into dominator tree node map.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@37475 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 6 insertions, 2 deletions

diff --git a/lib/VMCore/Dominators.cpp b/lib/VMCore/Dominators.cpp
index b9cbde44ad..c75143e5b6 100644
--- a/lib/VMCore/Dominators.cpp
+++ b/lib/VMCore/Dominators.cpp
@@ -289,7 +289,9 @@ void DominatorTree::calculate(Function& F) {
 
         // Add a new tree node for this BasicBlock, and link it as a child of
         // IDomNode
-        BBNode = IDomNode->addChild(new DomTreeNode(I, IDomNode));
+        DomTreeNode *C = new DomTreeNode(I, IDomNode);
+        DomTreeNodes[I] = C;
+        BBNode = IDomNode->addChild(C);
       }
     }
 
@@ -338,7 +340,9 @@ DomTreeNode *DominatorTree::getNodeForBlock(BasicBlock *BB) {
 
   // Add a new tree node for this BasicBlock, and link it as a child of
   // IDomNode
-  return BBNode = IDomNode->addChild(new DomTreeNode(BB, IDomNode));
+  DomTreeNode *C = new DomTreeNode(BB, IDomNode);
+  DomTreeNodes[BB] = C;
+  return BBNode = IDomNode->addChild(C);
 }
 
 static std::ostream &operator<<(std::ostream &o,