From 3af8a11bc66b93dc3f63ce7192647a27189aeefc Mon Sep 17 00:00:00 2001 From: Philip Reames Date: Wed, 1 Sep 2021 13:29:10 -0700 Subject: [PATCH] [LoopDeletion] Separate logic in breakBackedgeIfNotTaken using symboic max trip count [nfc] As mentioned in D108833, the logic for figuring out if a backedge is dead was somewhat interwoven with the SCEV based logic and the symbolic eval logic. This is my attempt at making the code easier to follow. Note that this is only NFC after the work done in 29fa37ec. Thanks to Nikita for catching that case. Differential Revision: https://reviews.llvm.org/D108848 --- llvm/lib/Transforms/Scalar/LoopDeletion.cpp | 26 +++++++++++++++----------- 1 file changed, 15 insertions(+), 11 deletions(-) diff --git a/llvm/lib/Transforms/Scalar/LoopDeletion.cpp b/llvm/lib/Transforms/Scalar/LoopDeletion.cpp index dd1fa1d..73f97f2 100644 --- a/llvm/lib/Transforms/Scalar/LoopDeletion.cpp +++ b/llvm/lib/Transforms/Scalar/LoopDeletion.cpp @@ -393,19 +393,23 @@ breakBackedgeIfNotTaken(Loop *L, DominatorTree &DT, ScalarEvolution &SE, if (!L->getLoopLatch()) return LoopDeletionResult::Unmodified; - auto *BTCMax = SE.getConstantMaxBackedgeTakenCount(L); - if (!BTCMax->isZero()) { - auto *BTC = SE.getBackedgeTakenCount(L); - if (!BTC->isZero()) { - if (!isa(BTC) && SE.isKnownNonZero(BTC)) - return LoopDeletionResult::Unmodified; - if (!canProveExitOnFirstIteration(L, DT, LI)) - return LoopDeletionResult::Unmodified; - } + auto *BTC = SE.getSymbolicMaxBackedgeTakenCount(L); + if (BTC->isZero()) { + // SCEV knows this backedge isn't taken! + breakLoopBackedge(L, DT, SE, LI, MSSA); + return LoopDeletionResult::Deleted; } - breakLoopBackedge(L, DT, SE, LI, MSSA); - return LoopDeletionResult::Deleted; + // If SCEV leaves open the possibility of a zero trip count, see if + // symbolically evaluating the first iteration lets us prove the backedge + // unreachable. + if (isa(BTC) || !SE.isKnownNonZero(BTC)) + if (canProveExitOnFirstIteration(L, DT, LI)) { + breakLoopBackedge(L, DT, SE, LI, MSSA); + return LoopDeletionResult::Deleted; + } + + return LoopDeletionResult::Unmodified; } /// Remove a loop if it is dead. -- 2.7.4