Fix Analysis/Dominators/2006-10-02-BreakCritEdges.ll


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@33210 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Transforms/Utils/BreakCriticalEdges.cpp b/lib/Transforms/Utils/BreakCriticalEdges.cpp
index e8cc027..adbf584 100644
--- a/lib/Transforms/Utils/BreakCriticalEdges.cpp
+++ b/lib/Transforms/Utils/BreakCriticalEdges.cpp
@@ -212,7 +212,8 @@
 
   // Should we update ImmediateDominator information?
   if (ImmediateDominators *ID = P->getAnalysisToUpdate<ImmediateDominators>()) {
-    if (ID->get(TIBB)) {  // Only do this if TIBB is reachable.
+    // Only do this if TIBB is reachable.
+    if (ID->get(TIBB) || &TIBB->getParent()->getEntryBlock() == TIBB) {
       // TIBB is the new immediate dominator for NewBB.
       ID->addNewBlock(NewBB, TIBB);