lib/Transforms/IPO/SCCP.cpp

   1 #include "llvm/Transforms/IPO/SCCP.h"
   2 #include "llvm/Analysis/AssumptionCache.h"
   3 #include "llvm/Analysis/TargetLibraryInfo.h"
   4 #include "llvm/Transforms/IPO.h"
   5 #include "llvm/Transforms/Scalar/SCCP.h"
   6
   7 using namespace llvm;
   8
   9 PreservedAnalyses IPSCCPPass::run(Module &M, ModuleAnalysisManager &AM) {
  10   const DataLayout &DL = M.getDataLayout();
  11   auto &TLI = AM.getResult<TargetLibraryAnalysis>(M);
  12   auto &FAM = AM.getResult<FunctionAnalysisManagerModuleProxy>(M).getManager();
  13   auto getPredicateInfo =
  14       [&FAM](Function &F) -> std::unique_ptr<PredicateInfo> {
  15     return make_unique<PredicateInfo>(F,
  16                                       FAM.getResult<DominatorTreeAnalysis>(F),
  17                                       FAM.getResult<AssumptionAnalysis>(F));
  18   };
  19
  20   if (!runIPSCCP(M, DL, &TLI, getPredicateInfo))
  21     return PreservedAnalyses::all();
  22   return PreservedAnalyses::none();
  23 }
  24
  25 namespace {
  26
  27 //===--------------------------------------------------------------------===//
  28 //
  29 /// IPSCCP Class - This class implements interprocedural Sparse Conditional
  30 /// Constant Propagation.
  31 ///
  32 class IPSCCPLegacyPass : public ModulePass {
  33 public:
  34   static char ID;
  35
  36   IPSCCPLegacyPass() : ModulePass(ID) {
  37     initializeIPSCCPLegacyPassPass(*PassRegistry::getPassRegistry());
  38   }
  39
  40   bool runOnModule(Module &M) override {
  41     if (skipModule(M))
  42       return false;
  43     const DataLayout &DL = M.getDataLayout();
  44     const TargetLibraryInfo *TLI =
  45         &getAnalysis<TargetLibraryInfoWrapperPass>().getTLI();
  46
  47     auto getPredicateInfo =
  48         [this](Function &F) -> std::unique_ptr<PredicateInfo> {
  49       return make_unique<PredicateInfo>(
  50           F, this->getAnalysis<DominatorTreeWrapperPass>(F).getDomTree(),
  51           this->getAnalysis<AssumptionCacheTracker>().getAssumptionCache(F));
  52     };
  53
  54     return runIPSCCP(M, DL, TLI, getPredicateInfo);
  55   }
  56
  57   void getAnalysisUsage(AnalysisUsage &AU) const override {
  58     AU.addRequired<AssumptionCacheTracker>();
  59     AU.addRequired<DominatorTreeWrapperPass>();
  60     AU.addRequired<TargetLibraryInfoWrapperPass>();
  61   }
  62 };
  63
  64 } // end anonymous namespace
  65
  66 char IPSCCPLegacyPass::ID = 0;
  67
  68 INITIALIZE_PASS_BEGIN(IPSCCPLegacyPass, "ipsccp",
  69                       "Interprocedural Sparse Conditional Constant Propagation",
  70                       false, false)
  71 INITIALIZE_PASS_DEPENDENCY(DominatorTreeWrapperPass)
  72 INITIALIZE_PASS_DEPENDENCY(TargetLibraryInfoWrapperPass)
  73 INITIALIZE_PASS_END(IPSCCPLegacyPass, "ipsccp",
  74                     "Interprocedural Sparse Conditional Constant Propagation",
  75                     false, false)
  76
  77 // createIPSCCPPass - This is the public interface to this file.
  78 ModulePass *llvm::createIPSCCPPass() { return new IPSCCPLegacyPass(); }