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

1 files changed, 3 insertions, 2 deletions

diff --git a/lib/Transforms/Scalar/LoopStrengthReduce.cpp b/lib/Transforms/Scalar/LoopStrengthReduce.cpp
index 83231c1ae8..b2c5977a1b 100644
--- a/lib/Transforms/Scalar/LoopStrengthReduce.cpp
+++ b/lib/Transforms/Scalar/LoopStrengthReduce.cpp
@@ -1909,6 +1909,7 @@ BinaryOperator *getNewPHIIncrement(BinaryOperator *Incr, PHINode *PHI,
 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 @@ void LoopStrengthReduce::OptimizeIVType(Loop *L) {
 
     // 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();
   }