From f2e2925f95a271505f3ba103bac71b3b6d066c57 Mon Sep 17 00:00:00 2001 From: Chris Lattner Date: Thu, 27 Feb 2003 22:37:44 +0000 Subject: Change behavior of changeExitBlock function to replace all instances of exit block git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@5661 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/Analysis/LoopInfo.cpp | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'lib/Analysis/LoopInfo.cpp') diff --git a/lib/Analysis/LoopInfo.cpp b/lib/Analysis/LoopInfo.cpp index 035f92c33c..a54d659dc8 100644 --- a/lib/Analysis/LoopInfo.cpp +++ b/lib/Analysis/LoopInfo.cpp @@ -235,8 +235,8 @@ void Loop::addBasicBlockToLoop(BasicBlock *NewBB, LoopInfo &LI) { } } -/// changeExitBlock - This method is used to update loop information. One -/// instance of the specified Old basic block is removed from the exit list +/// changeExitBlock - This method is used to update loop information. All +/// instances of the specified Old basic block are removed from the exit list /// and replaced with New. /// void Loop::changeExitBlock(BasicBlock *Old, BasicBlock *New) { @@ -246,4 +246,10 @@ void Loop::changeExitBlock(BasicBlock *Old, BasicBlock *New) { std::find(ExitBlocks.begin(), ExitBlocks.end(), Old); assert(I != ExitBlocks.end() && "Old exit block not found!"); *I = New; + + I = std::find(I+1, ExitBlocks.end(), Old); + while (I != ExitBlocks.end()) { + *I = New; + I = std::find(I+1, ExitBlocks.end(), Old); + } } -- cgit v1.2.3