forked from OSchip/llvm-project
Actually load profiling information now! Block layout can use real, live,
actual profile info, and works! :) llvm-svn: 11324
This commit is contained in:
parent
fa9a8f83a5
commit
f10f6b10f9
|
@ -15,14 +15,22 @@
|
||||||
#include "llvm/Pass.h"
|
#include "llvm/Pass.h"
|
||||||
#include "llvm/Analysis/ProfileInfo.h"
|
#include "llvm/Analysis/ProfileInfo.h"
|
||||||
#include "llvm/Analysis/ProfileInfoLoader.h"
|
#include "llvm/Analysis/ProfileInfoLoader.h"
|
||||||
|
#include "Support/CommandLine.h"
|
||||||
using namespace llvm;
|
using namespace llvm;
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
cl::opt<std::string>
|
||||||
|
ProfileInfoFilename("profile-info-file", cl::init("llvmprof.out"),
|
||||||
|
cl::desc(""));
|
||||||
|
|
||||||
class LoaderPass : public Pass, public ProfileInfo {
|
class LoaderPass : public Pass, public ProfileInfo {
|
||||||
std::string Filename;
|
std::string Filename;
|
||||||
|
std::map<BasicBlock*, unsigned> ExecutionCounts;
|
||||||
public:
|
public:
|
||||||
LoaderPass(const std::string &filename = "llvmprof.out")
|
LoaderPass(const std::string &filename = "")
|
||||||
: Filename(filename) {}
|
: Filename(filename) {
|
||||||
|
if (filename.empty()) Filename = ProfileInfoFilename;
|
||||||
|
}
|
||||||
|
|
||||||
virtual void getAnalysisUsage(AnalysisUsage &AU) const {
|
virtual void getAnalysisUsage(AnalysisUsage &AU) const {
|
||||||
AU.setPreservesAll();
|
AU.setPreservesAll();
|
||||||
|
@ -35,7 +43,10 @@ namespace {
|
||||||
/// run - Load the profile information from the specified file.
|
/// run - Load the profile information from the specified file.
|
||||||
virtual bool run(Module &M);
|
virtual bool run(Module &M);
|
||||||
|
|
||||||
unsigned getExecutionCount(BasicBlock *BB) { return 0; }
|
virtual unsigned getExecutionCount(BasicBlock *BB) {
|
||||||
|
std::map<BasicBlock*, unsigned>::iterator I = ExecutionCounts.find(BB);
|
||||||
|
return I != ExecutionCounts.end() ? I->second : 0;
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
RegisterOpt<LoaderPass>
|
RegisterOpt<LoaderPass>
|
||||||
|
@ -53,6 +64,11 @@ Pass *llvm::createProfileLoaderPass(const std::string &Filename) {
|
||||||
}
|
}
|
||||||
|
|
||||||
bool LoaderPass::run(Module &M) {
|
bool LoaderPass::run(Module &M) {
|
||||||
|
ProfileInfoLoader PIL("opt", Filename, M);
|
||||||
|
if (PIL.hasAccurateBlockCounts()) {
|
||||||
|
std::vector<std::pair<BasicBlock*, unsigned> > Counts;
|
||||||
|
PIL.getBlockCounts(Counts);
|
||||||
|
ExecutionCounts.insert(Counts.begin(), Counts.end());
|
||||||
|
}
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue