diff --git a/llvm/include/llvm/Analysis/ScalarEvolution.h b/llvm/include/llvm/Analysis/ScalarEvolution.h --- a/llvm/include/llvm/Analysis/ScalarEvolution.h +++ b/llvm/include/llvm/Analysis/ScalarEvolution.h @@ -2049,6 +2049,13 @@ SmallPtrSetImpl &Visited, SmallVectorImpl &ToForget); +#ifndef NDEBUG + /// Iterate over instructions in \p Worlist and their users. Assert all users + /// have been removed from ValueExprMap. + void verifyAllUsersClearedFromMap(SmallVectorImpl &Worklist, + SmallPtrSetImpl &Visited); +#endif + /// Return an existing SCEV for V if there is one, otherwise return nullptr. const SCEV *getExistingSCEV(Value *V); diff --git a/llvm/lib/Analysis/ScalarEvolution.cpp b/llvm/lib/Analysis/ScalarEvolution.cpp --- a/llvm/lib/Analysis/ScalarEvolution.cpp +++ b/llvm/lib/Analysis/ScalarEvolution.cpp @@ -8421,6 +8421,8 @@ SmallVectorImpl &ToForget) { while (!Worklist.empty()) { Instruction *I = Worklist.pop_back_val(); + if (!isSCEVable(I->getType())) + continue; ValueExprMapType::iterator It = ValueExprMap.find_as(static_cast(I)); @@ -8435,12 +8437,42 @@ } } +#ifndef NDEBUG +void ScalarEvolution::verifyAllUsersClearedFromMap( + SmallVectorImpl &Worklist, + SmallPtrSetImpl &Visited) { + while (!Worklist.empty()) { + Instruction *I = Worklist.pop_back_val(); + // If I is an SCEVUnknown and we reached it via a non-SCEVable operand, + // invalidation for the instruction stops here as the SCEV def-use chain is + // broken by the SCEVUnknown. + if (any_of(I->operands(), + [this, &Visited](Value *Op) { + auto *OpI = dyn_cast(Op); + return OpI && Visited.contains(OpI) && + !isSCEVable(Op->getType()); + }) && + isSCEVable(I->getType()) && + cast_if_present(getExistingSCEV(I))) + continue; + assert(ValueExprMap.count(I) == 0 && + "entry for I has not been removed properly"); + PushDefUseChildren(I, Worklist, Visited); + } +} +#endif + void ScalarEvolution::forgetLoop(const Loop *L) { SmallVector LoopWorklist(1, L); SmallVector Worklist; SmallPtrSet Visited; SmallVector ToForget; +#ifndef NDEBUG + SmallVector VerificationWorklist; + SmallPtrSet VerificationVisited; +#endif + // Iterate over all the loops and sub-loops to drop SCEV information. while (!LoopWorklist.empty()) { auto *CurrL = LoopWorklist.pop_back_val(); @@ -8467,6 +8499,9 @@ // Drop information about expressions based on loop-header PHIs. PushLoopPHIs(CurrL, Worklist, Visited); +#ifndef NDEBUG + PushLoopPHIs(CurrL, VerificationWorklist, VerificationVisited); +#endif visitAndClearUsers(Worklist, Visited, ToForget); LoopPropertiesCache.erase(CurrL); @@ -8475,6 +8510,10 @@ LoopWorklist.append(CurrL->begin(), CurrL->end()); } forgetMemoizedResults(ToForget); + +#ifndef NDEBUG + verifyAllUsersClearedFromMap(VerificationWorklist, VerificationVisited); +#endif } void ScalarEvolution::forgetTopmostLoop(const Loop *L) { @@ -8494,6 +8533,14 @@ visitAndClearUsers(Worklist, Visited, ToForget); forgetMemoizedResults(ToForget); + +#ifndef NDEBUG + SmallVector VerificationWorklist; + SmallPtrSet VerificationVisited; + Worklist.push_back(cast(V)); + Visited.insert(cast(V)); + verifyAllUsersClearedFromMap(VerificationWorklist, VerificationVisited); +#endif } void ScalarEvolution::forgetLoopDispositions() { LoopDispositions.clear(); }