diff --git a/llvm/lib/Analysis/LoopInfo.cpp b/llvm/lib/Analysis/LoopInfo.cpp index 38b9c6699992..012c3dac7364 100644 --- a/llvm/lib/Analysis/LoopInfo.cpp +++ b/llvm/lib/Analysis/LoopInfo.cpp @@ -326,6 +326,18 @@ Loop *LoopInfo::removeLoop(iterator I) { return L; } +/// removeBlock - This method completely removes BB from all data structures, +/// including all of the Loop objects it is nested in and our mapping from +/// BasicBlocks to loops. +void LoopInfo::removeBlock(BasicBlock *BB) { + std::map::iterator I = BBMap.find(BB); + if (I != BBMap.end()) { + for (Loop *L = I->second; L; L = L->getParentLoop()) + L->removeBlockFromLoop(BB); + + BBMap.erase(I); + } +} //===----------------------------------------------------------------------===//