summaryrefslogtreecommitdiff
path: root/lib/Transforms/Utils/BreakCriticalEdges.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'lib/Transforms/Utils/BreakCriticalEdges.cpp')
-rw-r--r--lib/Transforms/Utils/BreakCriticalEdges.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/lib/Transforms/Utils/BreakCriticalEdges.cpp b/lib/Transforms/Utils/BreakCriticalEdges.cpp
index e8cc027180..adbf5847c4 100644
--- a/lib/Transforms/Utils/BreakCriticalEdges.cpp
+++ b/lib/Transforms/Utils/BreakCriticalEdges.cpp
@@ -212,7 +212,8 @@ bool llvm::SplitCriticalEdge(TerminatorInst *TI, unsigned SuccNum, Pass *P,
// 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);