lib/Analysis/ProfileInfoLoaderPass.cpp

   1 //===- ProfileInfoLoaderPass.cpp - LLVM Pass to load profile info ---------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements a concrete implementation of profiling information that
  11 // loads the information from a profile dump file.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/Pass.h"
  16 #include "llvm/Analysis/ProfileInfo.h"
  17 #include "llvm/Analysis/ProfileInfoLoader.h"
  18 #include "Support/CommandLine.h"
  19 using namespace llvm;
  20
  21 namespace {
  22   cl::opt<std::string>
  23   ProfileInfoFilename("profile-info-file", cl::init("llvmprof.out"),
  24                       cl::value_desc("filename"),
  25                       cl::desc("Profile file loaded by -profile-loader"));
  26
  27   class LoaderPass : public Pass, public ProfileInfo {
  28     std::string Filename;
  29     std::map<BasicBlock*, unsigned> ExecutionCounts;
  30   public:
  31     LoaderPass(const std::string &filename = "")
  32       : Filename(filename) {
  33       if (filename.empty()) Filename = ProfileInfoFilename;
  34     }
  35
  36     virtual void getAnalysisUsage(AnalysisUsage &AU) const {
  37       AU.setPreservesAll();
  38     }
  39
  40     virtual const char *getPassName() const {
  41       return "Profiling information loader";
  42     }
  43
  44     /// run - Load the profile information from the specified file.
  45     virtual bool run(Module &M);
  46
  47     virtual unsigned getExecutionCount(BasicBlock *BB) {
  48       std::map<BasicBlock*, unsigned>::iterator I = ExecutionCounts.find(BB);
  49       return I != ExecutionCounts.end() ? I->second : 0;
  50     }
  51   };
  52
  53   RegisterOpt<LoaderPass>
  54   X("profile-loader", "Load profile information from llvmprof.out");
  55
  56   RegisterAnalysisGroup<ProfileInfo, LoaderPass> Y;
  57 }  // End of anonymous namespace
  58
  59
  60 /// createProfileLoaderPass - This function returns a Pass that loads the
  61 /// profiling information for the module from the specified filename, making it
  62 /// available to the optimizers.
  63 Pass *llvm::createProfileLoaderPass(const std::string &Filename) {
  64   return new LoaderPass(Filename);
  65 }
  66
  67 bool LoaderPass::run(Module &M) {
  68   ProfileInfoLoader PIL("opt", Filename, M);
  69   if (PIL.hasAccurateBlockCounts()) {
  70     std::vector<std::pair<BasicBlock*, unsigned> > Counts;
  71     PIL.getBlockCounts(Counts);
  72     ExecutionCounts.insert(Counts.begin(), Counts.end());
  73   }
  74   return false;
  75 }