Add basic block level interface to change immediate dominator

and create new node.

llvm-svn: 37414

1 files changed, 2 insertions, 2 deletions

diff --git a/llvm/lib/Transforms/Utils/CodeExtractor.cpp b/llvm/lib/Transforms/Utils/CodeExtractor.cpp
index ce00e0e1d8b..f0eb6250f72 100644
--- a/llvm/lib/Transforms/Utils/CodeExtractor.cpp
+++ b/llvm/lib/Transforms/Utils/CodeExtractor.cpp
@@ -143,14 +143,14 @@ void CodeExtractor::severSplitPHINodes(BasicBlock *&Header) {
   // blocks that dominate TIBB plus the new block itself.
   if (EF) {
     BasicBlock* idom = EF->getIDom(OldPred);
-    DT->createNewNode(NewBB, DT->getNode(idom));
+    DT->createNewNode(NewBB, idom);
     EF->addNewBlock(NewBB, idom);
 
     // Additionally, NewBB replaces OldPred as the immediate dominator of blocks
     Function *F = Header->getParent();
     for (Function::iterator I = F->begin(), E = F->end(); I != E; ++I)
       if (EF->getIDom(I) == OldPred) {
-        DT->changeImmediateDominator(DT->getNode(I), DT->getNode(NewBB));
+        DT->changeImmediateDominator(I, NewBB);
         EF->setImmediateDominator(I, NewBB);
       }
   }