RPCS3/llvm
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm.git synced 2024-11-26 13:10:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Annotate functions with edge counts as well, if they are available.

Browse Source 
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@12233 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Chris Lattner 2004-03-08 20:04:32 +00:00
parent dbbbfef165
commit 8c2730e347
1 changed files with 39 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
tools/llvm-prof/llvm-prof.cpp
View File

@ -13,6 +13,7 @@
// //
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
#include "llvm/InstrTypes.h"
#include "llvm/Module.h" #include "llvm/Module.h"
#include "llvm/Assembly/AsmAnnotationWriter.h" #include "llvm/Assembly/AsmAnnotationWriter.h"
#include "llvm/Analysis/ProfileInfoLoader.h" #include "llvm/Analysis/ProfileInfoLoader.h"
@ -59,21 +60,46 @@ namespace {
class ProfileAnnotator : public AssemblyAnnotationWriter { class ProfileAnnotator : public AssemblyAnnotationWriter {
std::map<const Function *, unsigned> &FuncFreqs; std::map<const Function *, unsigned> &FuncFreqs;
std::map<const BasicBlock*, unsigned> &BlockFreqs; std::map<const BasicBlock*, unsigned> &BlockFreqs;
std::map<ProfileInfoLoader::Edge, unsigned> &EdgeFreqs;
public: public:
ProfileAnnotator(std::map<const Function *, unsigned> &FF, ProfileAnnotator(std::map<const Function *, unsigned> &FF,
std::map<const BasicBlock*, unsigned> &BF) std::map<const BasicBlock*, unsigned> &BF,
: FuncFreqs(FF), BlockFreqs(BF) {} std::map<ProfileInfoLoader::Edge, unsigned> &EF)
: FuncFreqs(FF), BlockFreqs(BF), EdgeFreqs(EF) {}
virtual void emitFunctionAnnot(const Function *F, std::ostream &OS) { virtual void emitFunctionAnnot(const Function *F, std::ostream &OS) {
OS << ";;; %" << F->getName() << " called " << FuncFreqs[F] OS << ";;; %" << F->getName() << " called " << FuncFreqs[F]
<< " times.\n;;;\n"; << " times.\n;;;\n";
} }
virtual void emitBasicBlockAnnot(const BasicBlock *BB, std::ostream &OS) { virtual void emitBasicBlockStartAnnot(const BasicBlock *BB,
std::ostream &OS) {
if (BlockFreqs.empty()) return; if (BlockFreqs.empty()) return;
if (unsigned Count = BlockFreqs[BB]) if (unsigned Count = BlockFreqs[BB])
OS << ";;; Executed " << Count << " times.\n"; OS << "\t;;; Basic block executed " << Count << " times.\n";
else else
OS << ";;; Never executed!\n"; OS << "\t;;; Never executed!\n";
}
virtual void emitBasicBlockEndAnnot(const BasicBlock *BB, std::ostream &OS){
if (EdgeFreqs.empty()) return;
// Figure out how many times each successor executed.
std::vector<std::pair<const BasicBlock*, unsigned> > SuccCounts;
const TerminatorInst *TI = BB->getTerminator();
std::map<ProfileInfoLoader::Edge, unsigned>::iterator I =
EdgeFreqs.lower_bound(std::make_pair(const_cast<BasicBlock*>(BB), 0U));
for (; I != EdgeFreqs.end() && I->first.first == BB; ++I)
if (I->second)
SuccCounts.push_back(std::make_pair(TI->getSuccessor(I->first.second),
I->second));
if (!SuccCounts.empty()) {
OS << "\t;;; Out-edge counts:";
for (unsigned i = 0, e = SuccCounts.size(); i != e; ++i)
OS << " [" << SuccCounts[i].second << " -> "
<< SuccCounts[i].first->getName() << "]";
OS << "\n";
}
} }
}; };
} }
@ -97,6 +123,7 @@ int main(int argc, char **argv) {
std::map<const Function *, unsigned> FuncFreqs; std::map<const Function *, unsigned> FuncFreqs;
std::map<const BasicBlock*, unsigned> BlockFreqs; std::map<const BasicBlock*, unsigned> BlockFreqs;
std::map<ProfileInfoLoader::Edge, unsigned> EdgeFreqs;
// Output a report. Eventually, there will be multiple reports selectable on // Output a report. Eventually, there will be multiple reports selectable on
// the command line, for now, just keep things simple. // the command line, for now, just keep things simple.
@ -177,11 +204,17 @@ int main(int argc, char **argv) {
BlockFreqs.insert(Counts.begin(), Counts.end()); BlockFreqs.insert(Counts.begin(), Counts.end());
} }
if (PI.hasAccurateEdgeCounts()) {
std::vector<std::pair<ProfileInfoLoader::Edge, unsigned> > Counts;
PI.getEdgeCounts(Counts);
EdgeFreqs.insert(Counts.begin(), Counts.end());
}
if (PrintAnnotatedLLVM || PrintAllCode) { if (PrintAnnotatedLLVM || PrintAllCode) {
std::cout << "\n===" << std::string(73, '-') << "===\n"; std::cout << "\n===" << std::string(73, '-') << "===\n";
std::cout << "Annotated LLVM code for the module:\n\n"; std::cout << "Annotated LLVM code for the module:\n\n";
ProfileAnnotator PA(FuncFreqs, BlockFreqs); ProfileAnnotator PA(FuncFreqs, BlockFreqs, EdgeFreqs);
if (FunctionsToPrint.empty() || PrintAllCode) if (FunctionsToPrint.empty() || PrintAllCode)
M->print(std::cout, &PA); M->print(std::cout, &PA);