diff options
| -rw-r--r-- | llvm/lib/Transforms/Utils/BreakCriticalEdges.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/llvm/lib/Transforms/Utils/BreakCriticalEdges.cpp b/llvm/lib/Transforms/Utils/BreakCriticalEdges.cpp index e8cc027180c..adbf5847c4d 100644 --- a/llvm/lib/Transforms/Utils/BreakCriticalEdges.cpp +++ b/llvm/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); |
