While removing PHI, use basicblock to identify incoming value.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@55678 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Transforms/Scalar/LoopStrengthReduce.cpp b/lib/Transforms/Scalar/LoopStrengthReduce.cpp
index 83231c1..b2c5977 100644
--- a/lib/Transforms/Scalar/LoopStrengthReduce.cpp
+++ b/lib/Transforms/Scalar/LoopStrengthReduce.cpp
@@ -1909,6 +1909,7 @@
void LoopStrengthReduce::OptimizeIVType(Loop *L) {
BasicBlock *LPH = L->getLoopPreheader();
+ BasicBlock *LatchBB = L->getLoopLatch();
SmallVector<PHINode *, 4> PHIs;
for (BasicBlock::iterator BI = L->getHeader()->begin(),
BE = L->getHeader()->end(); BI != BE; ++BI) {
@@ -2000,8 +2001,8 @@
// Remove old PHI and increment instruction.
SE->deleteValueFromRecords(PHI);
- PHI->removeIncomingValue(Entry);
- PHI->removeIncomingValue(Latch);
+ PHI->removeIncomingValue(LatchBB);
+ PHI->removeIncomingValue(LPH);
SE->deleteValueFromRecords(Incr);
Incr->eraseFromParent();
}