From c72139134cbd5be74e443ca632209d39df45cb78 Mon Sep 17 00:00:00 2001
From: Devang Patel <dpatel@apple.com>
Date: Wed, 3 Sep 2008 00:02:42 +0000
Subject: [PATCH] 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
---
 lib/Transforms/Scalar/LoopStrengthReduce.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/lib/Transforms/Scalar/LoopStrengthReduce.cpp b/lib/Transforms/Scalar/LoopStrengthReduce.cpp
index 83231c1ae81..b2c5977a1b6 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();
   }
-- 
2.34.1