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/BasicBlock.h"
  16 #include "llvm/InstrTypes.h"
  17 #include "llvm/Pass.h"
  18 #include "llvm/Analysis/ProfileInfo.h"
  19 #include "llvm/Analysis/ProfileInfoLoader.h"
  20 #include "Support/CommandLine.h"
  21 #include <iostream>
  22
  23 using namespace llvm;
  24
  25 namespace {
  26   cl::opt<std::string>
  27   ProfileInfoFilename("profile-info-file", cl::init("llvmprof.out"),
  28                       cl::value_desc("filename"),
  29                       cl::desc("Profile file loaded by -profile-loader"));
  30
  31   class LoaderPass : public Pass, public ProfileInfo {
  32     std::string Filename;
  33   public:
  34     LoaderPass(const std::string &filename = "")
  35       : Filename(filename) {
  36       if (filename.empty()) Filename = ProfileInfoFilename;
  37     }
  38
  39     virtual void getAnalysisUsage(AnalysisUsage &AU) const {
  40       AU.setPreservesAll();
  41     }
  42
  43     virtual const char *getPassName() const {
  44       return "Profiling information loader";
  45     }
  46
  47     /// run - Load the profile information from the specified file.
  48     virtual bool run(Module &M);
  49   };
  50
  51   RegisterOpt<LoaderPass>
  52   X("profile-loader", "Load profile information from llvmprof.out");
  53
  54   RegisterAnalysisGroup<ProfileInfo, LoaderPass> Y;
  55 }  // End of anonymous namespace
  56
  57
  58 /// createProfileLoaderPass - This function returns a Pass that loads the
  59 /// profiling information for the module from the specified filename, making it
  60 /// available to the optimizers.
  61 Pass *llvm::createProfileLoaderPass(const std::string &Filename) {
  62   return new LoaderPass(Filename);
  63 }
  64
  65 bool LoaderPass::run(Module &M) {
  66   ProfileInfoLoader PIL("profile-loader", Filename, M);
  67   EdgeCounts.clear();
  68   bool PrintedWarning = false;
  69
  70   std::vector<std::pair<ProfileInfoLoader::Edge, unsigned> > ECs;
  71   PIL.getEdgeCounts(ECs);
  72   for (unsigned i = 0, e = ECs.size(); i != e; ++i) {
  73     BasicBlock *BB = ECs[i].first.first;
  74     unsigned SuccNum = ECs[i].first.second;
  75     TerminatorInst *TI = BB->getTerminator();
  76     if (SuccNum >= TI->getNumSuccessors()) {
  77       if (!PrintedWarning) {
  78         std::cerr << "WARNING: profile information is inconsistent with "
  79                   << "the current program!\n";
  80         PrintedWarning = true;
  81       }
  82     } else {
  83       EdgeCounts[std::make_pair(BB, TI->getSuccessor(SuccNum))]+= ECs[i].second;
  84     }
  85   }
  86
  87   return false;
  88 }