diff options
Diffstat (limited to 'llvm/lib')
| -rw-r--r-- | llvm/lib/Analysis/Analysis.cpp | 2 | ||||
| -rw-r--r-- | llvm/lib/Analysis/IVUsers.cpp | 96 | ||||
| -rw-r--r-- | llvm/lib/CodeGen/MachineFunctionPass.cpp | 2 | ||||
| -rw-r--r-- | llvm/lib/Passes/PassBuilder.cpp | 1 | ||||
| -rw-r--r-- | llvm/lib/Passes/PassRegistry.def | 2 | ||||
| -rw-r--r-- | llvm/lib/Transforms/Scalar/LoopStrengthReduce.cpp | 8 | 
6 files changed, 68 insertions, 43 deletions
diff --git a/llvm/lib/Analysis/Analysis.cpp b/llvm/lib/Analysis/Analysis.cpp index ffc4b2ef592..105a9fda3e2 100644 --- a/llvm/lib/Analysis/Analysis.cpp +++ b/llvm/lib/Analysis/Analysis.cpp @@ -51,7 +51,7 @@ void llvm::initializeAnalysis(PassRegistry &Registry) {    initializePostDomOnlyPrinterPass(Registry);    initializeAAResultsWrapperPassPass(Registry);    initializeGlobalsAAWrapperPassPass(Registry); -  initializeIVUsersPass(Registry); +  initializeIVUsersWrapperPassPass(Registry);    initializeInstCountPass(Registry);    initializeIntervalPartitionPass(Registry);    initializeLazyBlockFrequencyInfoPassPass(Registry); diff --git a/llvm/lib/Analysis/IVUsers.cpp b/llvm/lib/Analysis/IVUsers.cpp index e974cb9e711..43c0ba17fe4 100644 --- a/llvm/lib/Analysis/IVUsers.cpp +++ b/llvm/lib/Analysis/IVUsers.cpp @@ -12,11 +12,12 @@  //  //===----------------------------------------------------------------------===// +#include "llvm/Analysis/IVUsers.h"  #include "llvm/ADT/STLExtras.h"  #include "llvm/Analysis/AssumptionCache.h"  #include "llvm/Analysis/CodeMetrics.h" -#include "llvm/Analysis/IVUsers.h"  #include "llvm/Analysis/LoopPass.h" +#include "llvm/Analysis/LoopPassManager.h"  #include "llvm/Analysis/ScalarEvolutionExpressions.h"  #include "llvm/Analysis/ValueTracking.h"  #include "llvm/IR/Constants.h" @@ -33,19 +34,35 @@ using namespace llvm;  #define DEBUG_TYPE "iv-users" -char IVUsers::ID = 0; -INITIALIZE_PASS_BEGIN(IVUsers, "iv-users", +char IVUsersAnalysis::PassID; + +IVUsers IVUsersAnalysis::run(Loop &L, AnalysisManager<Loop> &AM) { +  const auto &FAM = +      AM.getResult<FunctionAnalysisManagerLoopProxy>(L).getManager(); +  Function *F = L.getHeader()->getParent(); + +  return IVUsers(&L, FAM.getCachedResult<AssumptionAnalysis>(*F), +                 FAM.getCachedResult<LoopAnalysis>(*F), +                 FAM.getCachedResult<DominatorTreeAnalysis>(*F), +                 FAM.getCachedResult<ScalarEvolutionAnalysis>(*F)); +} + +PreservedAnalyses IVUsersPrinterPass::run(Loop &L, AnalysisManager<Loop> &AM) { +  AM.getResult<IVUsersAnalysis>(L).print(OS); +  return PreservedAnalyses::all(); +} + +char IVUsersWrapperPass::ID = 0; +INITIALIZE_PASS_BEGIN(IVUsersWrapperPass, "iv-users",                        "Induction Variable Users", false, true)  INITIALIZE_PASS_DEPENDENCY(AssumptionCacheTracker)  INITIALIZE_PASS_DEPENDENCY(LoopInfoWrapperPass)  INITIALIZE_PASS_DEPENDENCY(DominatorTreeWrapperPass)  INITIALIZE_PASS_DEPENDENCY(ScalarEvolutionWrapperPass) -INITIALIZE_PASS_END(IVUsers, "iv-users", -                      "Induction Variable Users", false, true) +INITIALIZE_PASS_END(IVUsersWrapperPass, "iv-users", "Induction Variable Users", +                    false, true) -Pass *llvm::createIVUsersPass() { -  return new IVUsers(); -} +Pass *llvm::createIVUsersPass() { return new IVUsersWrapperPass(); }  /// isInteresting - Test whether the given expression is "interesting" when  /// used by the given expression, within the context of analyzing the @@ -246,28 +263,9 @@ IVStrideUse &IVUsers::AddUser(Instruction *User, Value *Operand) {    return IVUses.back();  } -IVUsers::IVUsers() -    : LoopPass(ID) { -  initializeIVUsersPass(*PassRegistry::getPassRegistry()); -} - -void IVUsers::getAnalysisUsage(AnalysisUsage &AU) const { -  AU.addRequired<AssumptionCacheTracker>(); -  AU.addRequired<LoopInfoWrapperPass>(); -  AU.addRequired<DominatorTreeWrapperPass>(); -  AU.addRequired<ScalarEvolutionWrapperPass>(); -  AU.setPreservesAll(); -} - -bool IVUsers::runOnLoop(Loop *l, LPPassManager &LPM) { - -  L = l; -  AC = &getAnalysis<AssumptionCacheTracker>().getAssumptionCache( -      *L->getHeader()->getParent()); -  LI = &getAnalysis<LoopInfoWrapperPass>().getLoopInfo(); -  DT = &getAnalysis<DominatorTreeWrapperPass>().getDomTree(); -  SE = &getAnalysis<ScalarEvolutionWrapperPass>().getSE(); - +IVUsers::IVUsers(Loop *L, AssumptionCache *AC, LoopInfo *LI, DominatorTree *DT, +                 ScalarEvolution *SE) +    : L(L), AC(AC), LI(LI), DT(DT), SE(SE), IVUses() {    // Collect ephemeral values so that AddUsersIfInteresting skips them.    EphValues.clear();    CodeMetrics::collectEphemeralValues(L, AC, EphValues); @@ -277,16 +275,13 @@ bool IVUsers::runOnLoop(Loop *l, LPPassManager &LPM) {    // this loop.  If they are induction variables, inspect their uses.    for (BasicBlock::iterator I = L->getHeader()->begin(); isa<PHINode>(I); ++I)      (void)AddUsersIfInteresting(&*I); - -  return false;  }  void IVUsers::print(raw_ostream &OS, const Module *M) const {    OS << "IV Users for loop ";    L->getHeader()->printAsOperand(OS, false);    if (SE->hasLoopInvariantBackedgeTakenCount(L)) { -    OS << " with backedge-taken count " -       << *SE->getBackedgeTakenCount(L); +    OS << " with backedge-taken count " << *SE->getBackedgeTakenCount(L);    }    OS << ":\n"; @@ -309,9 +304,7 @@ void IVUsers::print(raw_ostream &OS, const Module *M) const {  }  #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) -LLVM_DUMP_METHOD void IVUsers::dump() const { -  print(dbgs()); -} +LLVM_DUMP_METHOD void IVUsers::dump() const { print(dbgs()); }  #endif  void IVUsers::releaseMemory() { @@ -319,6 +312,35 @@ void IVUsers::releaseMemory() {    IVUses.clear();  } +IVUsersWrapperPass::IVUsersWrapperPass() : LoopPass(ID) { +  initializeIVUsersWrapperPassPass(*PassRegistry::getPassRegistry()); +} + +void IVUsersWrapperPass::getAnalysisUsage(AnalysisUsage &AU) const { +  AU.addRequired<AssumptionCacheTracker>(); +  AU.addRequired<LoopInfoWrapperPass>(); +  AU.addRequired<DominatorTreeWrapperPass>(); +  AU.addRequired<ScalarEvolutionWrapperPass>(); +  AU.setPreservesAll(); +} + +bool IVUsersWrapperPass::runOnLoop(Loop *L, LPPassManager &LPM) { +  auto *AC = &getAnalysis<AssumptionCacheTracker>().getAssumptionCache( +      *L->getHeader()->getParent()); +  auto *LI = &getAnalysis<LoopInfoWrapperPass>().getLoopInfo(); +  auto *DT = &getAnalysis<DominatorTreeWrapperPass>().getDomTree(); +  auto *SE = &getAnalysis<ScalarEvolutionWrapperPass>().getSE(); + +  IU.reset(new IVUsers(L, AC, LI, DT, SE)); +  return false; +} + +void IVUsersWrapperPass::print(raw_ostream &OS, const Module *M) const { +  IU->print(OS, M); +} + +void IVUsersWrapperPass::releaseMemory() { IU->releaseMemory(); } +  /// getReplacementExpr - Return a SCEV expression which computes the  /// value of the OperandValToReplace.  const SCEV *IVUsers::getReplacementExpr(const IVStrideUse &IU) const { diff --git a/llvm/lib/CodeGen/MachineFunctionPass.cpp b/llvm/lib/CodeGen/MachineFunctionPass.cpp index 16aecb9dff4..228fe170ab4 100644 --- a/llvm/lib/CodeGen/MachineFunctionPass.cpp +++ b/llvm/lib/CodeGen/MachineFunctionPass.cpp @@ -78,7 +78,7 @@ void MachineFunctionPass::getAnalysisUsage(AnalysisUsage &AU) const {    AU.addPreserved<DominatorTreeWrapperPass>();    AU.addPreserved<AAResultsWrapperPass>();    AU.addPreserved<GlobalsAAWrapperPass>(); -  AU.addPreserved<IVUsers>(); +  AU.addPreserved<IVUsersWrapperPass>();    AU.addPreserved<LoopInfoWrapperPass>();    AU.addPreserved<MemoryDependenceWrapperPass>();    AU.addPreserved<ScalarEvolutionWrapperPass>(); diff --git a/llvm/lib/Passes/PassBuilder.cpp b/llvm/lib/Passes/PassBuilder.cpp index b5d72d4f575..5cbbd733d38 100644 --- a/llvm/lib/Passes/PassBuilder.cpp +++ b/llvm/lib/Passes/PassBuilder.cpp @@ -32,6 +32,7 @@  #include "llvm/Analysis/DependenceAnalysis.h"  #include "llvm/Analysis/DominanceFrontier.h"  #include "llvm/Analysis/GlobalsModRef.h" +#include "llvm/Analysis/IVUsers.h"  #include "llvm/Analysis/LazyCallGraph.h"  #include "llvm/Analysis/LazyValueInfo.h"  #include "llvm/Analysis/LoopAccessAnalysis.h" diff --git a/llvm/lib/Passes/PassRegistry.def b/llvm/lib/Passes/PassRegistry.def index 5ff803f6879..cfe2dc5b6ab 100644 --- a/llvm/lib/Passes/PassRegistry.def +++ b/llvm/lib/Passes/PassRegistry.def @@ -182,6 +182,7 @@ FUNCTION_PASS("verify<regions>", RegionInfoVerifierPass())  #endif  LOOP_ANALYSIS("no-op-loop", NoOpLoopAnalysis())  LOOP_ANALYSIS("access-info", LoopAccessAnalysis()) +LOOP_ANALYSIS("ivusers", IVUsersAnalysis())  #undef LOOP_ANALYSIS  #ifndef LOOP_PASS @@ -198,4 +199,5 @@ LOOP_PASS("loop-deletion", LoopDeletionPass())  LOOP_PASS("simplify-cfg", LoopSimplifyCFGPass())  LOOP_PASS("indvars", IndVarSimplifyPass())  LOOP_PASS("print-access-info", LoopAccessInfoPrinterPass(dbgs())) +LOOP_PASS("print<ivusers>", IVUsersPrinterPass(dbgs()))  #undef LOOP_PASS diff --git a/llvm/lib/Transforms/Scalar/LoopStrengthReduce.cpp b/llvm/lib/Transforms/Scalar/LoopStrengthReduce.cpp index e49bd9ec4a9..77c77eb7d79 100644 --- a/llvm/lib/Transforms/Scalar/LoopStrengthReduce.cpp +++ b/llvm/lib/Transforms/Scalar/LoopStrengthReduce.cpp @@ -4949,7 +4949,7 @@ INITIALIZE_PASS_BEGIN(LoopStrengthReduce, "loop-reduce",  INITIALIZE_PASS_DEPENDENCY(TargetTransformInfoWrapperPass)  INITIALIZE_PASS_DEPENDENCY(DominatorTreeWrapperPass)  INITIALIZE_PASS_DEPENDENCY(ScalarEvolutionWrapperPass) -INITIALIZE_PASS_DEPENDENCY(IVUsers) +INITIALIZE_PASS_DEPENDENCY(IVUsersWrapperPass)  INITIALIZE_PASS_DEPENDENCY(LoopInfoWrapperPass)  INITIALIZE_PASS_DEPENDENCY(LoopSimplify)  INITIALIZE_PASS_END(LoopStrengthReduce, "loop-reduce", @@ -4979,8 +4979,8 @@ void LoopStrengthReduce::getAnalysisUsage(AnalysisUsage &AU) const {    // Requiring LoopSimplify a second time here prevents IVUsers from running    // twice, since LoopSimplify was invalidated by running ScalarEvolution.    AU.addRequiredID(LoopSimplifyID); -  AU.addRequired<IVUsers>(); -  AU.addPreserved<IVUsers>(); +  AU.addRequired<IVUsersWrapperPass>(); +  AU.addPreserved<IVUsersWrapperPass>();    AU.addRequired<TargetTransformInfoWrapperPass>();  } @@ -4988,7 +4988,7 @@ bool LoopStrengthReduce::runOnLoop(Loop *L, LPPassManager & /*LPM*/) {    if (skipLoop(L))      return false; -  auto &IU = getAnalysis<IVUsers>(); +  auto &IU = getAnalysis<IVUsersWrapperPass>().getIU();    auto &SE = getAnalysis<ScalarEvolutionWrapperPass>().getSE();    auto &DT = getAnalysis<DominatorTreeWrapperPass>().getDomTree();    auto &LI = getAnalysis<LoopInfoWrapperPass>().getLoopInfo();  | 

