Reconstruct dom info, if loop is unswitched.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@53086 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Transforms/Scalar/LoopUnswitch.cpp b/lib/Transforms/Scalar/LoopUnswitch.cpp
index f57b7bb..02dac92 100644
--- a/lib/Transforms/Scalar/LoopUnswitch.cpp
+++ b/lib/Transforms/Scalar/LoopUnswitch.cpp
@@ -113,8 +113,8 @@
AU.addRequiredID(LCSSAID);
AU.addPreservedID(LCSSAID);
// FIXME: Loop Unswitch does not preserve dominator info in all cases.
- // AU.addPreserved<DominatorTree>();
- // AU.addPreserved<DominanceFrontier>();
+ AU.addPreserved<DominatorTree>();
+ AU.addPreserved<DominanceFrontier>();
}
private:
@@ -206,7 +206,6 @@
DT = getAnalysisToUpdate<DominatorTree>();
currentLoop = L;
bool Changed = false;
-
do {
assert(currentLoop->isLCSSAForm());
redoLoop = false;
@@ -459,7 +458,13 @@
} else {
UnswitchNontrivialCondition(LoopCond, Val, currentLoop);
}
-
+
+ // FIXME: Reconstruct dom info, because it is not preserved properly.
+ Function *F = loopHeader->getParent();
+ if (DT)
+ DT->runOnFunction(*F);
+ if (DF)
+ DF->runOnFunction(*F);
return true;
}