mirror of
https://github.com/Gericom/teak-llvm.git
synced 2025-06-20 03:55:48 -04:00
fc93dd8e18
This patch adds support for generating a call graph profile from Branch Frequency Info.
The CGProfile module pass simply gets the block profile count for each BB and scans for call instructions. For each call instruction it adds an edge from the current function to the called function with the current BB block profile count as the weight.
After scanning all the functions, it generates an appending module flag containing the data. The format looks like:
!llvm.module.flags = !{!0}
!0 = !{i32 5, !"CG Profile", !1}
!1 = !{!2, !3, !4} ; List of edges
!2 = !{void ()* @a, void ()* @b, i64 32} ; Edge from a to b with a weight of 32
!3 = !{void (i1)* @freq, void ()* @a, i64 11}
!4 = !{void (i1)* @freq, void ()* @b, i64 20}
Differential Revision: https://reviews.llvm.org/D48105
llvm-svn: 335306
82 lines
3.0 KiB
C++
82 lines
3.0 KiB
C++
//===-- Instrumentation.cpp - TransformUtils Infrastructure ---------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines the common initialization infrastructure for the
|
|
// Instrumentation library.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Transforms/Instrumentation.h"
|
|
#include "llvm-c/Initialization.h"
|
|
#include "llvm/IR/IntrinsicInst.h"
|
|
#include "llvm/InitializePasses.h"
|
|
#include "llvm/PassRegistry.h"
|
|
|
|
using namespace llvm;
|
|
|
|
/// Moves I before IP. Returns new insert point.
|
|
static BasicBlock::iterator moveBeforeInsertPoint(BasicBlock::iterator I, BasicBlock::iterator IP) {
|
|
// If I is IP, move the insert point down.
|
|
if (I == IP)
|
|
return ++IP;
|
|
// Otherwise, move I before IP and return IP.
|
|
I->moveBefore(&*IP);
|
|
return IP;
|
|
}
|
|
|
|
/// Instrumentation passes often insert conditional checks into entry blocks.
|
|
/// Call this function before splitting the entry block to move instructions
|
|
/// that must remain in the entry block up before the split point. Static
|
|
/// allocas and llvm.localescape calls, for example, must remain in the entry
|
|
/// block.
|
|
BasicBlock::iterator llvm::PrepareToSplitEntryBlock(BasicBlock &BB,
|
|
BasicBlock::iterator IP) {
|
|
assert(&BB.getParent()->getEntryBlock() == &BB);
|
|
for (auto I = IP, E = BB.end(); I != E; ++I) {
|
|
bool KeepInEntry = false;
|
|
if (auto *AI = dyn_cast<AllocaInst>(I)) {
|
|
if (AI->isStaticAlloca())
|
|
KeepInEntry = true;
|
|
} else if (auto *II = dyn_cast<IntrinsicInst>(I)) {
|
|
if (II->getIntrinsicID() == llvm::Intrinsic::localescape)
|
|
KeepInEntry = true;
|
|
}
|
|
if (KeepInEntry)
|
|
IP = moveBeforeInsertPoint(I, IP);
|
|
}
|
|
return IP;
|
|
}
|
|
|
|
/// initializeInstrumentation - Initialize all passes in the TransformUtils
|
|
/// library.
|
|
void llvm::initializeInstrumentation(PassRegistry &Registry) {
|
|
initializeAddressSanitizerPass(Registry);
|
|
initializeAddressSanitizerModulePass(Registry);
|
|
initializeBoundsCheckingLegacyPassPass(Registry);
|
|
initializeGCOVProfilerLegacyPassPass(Registry);
|
|
initializeCGProfilePassPass(Registry);
|
|
initializePGOInstrumentationGenLegacyPassPass(Registry);
|
|
initializePGOInstrumentationUseLegacyPassPass(Registry);
|
|
initializePGOIndirectCallPromotionLegacyPassPass(Registry);
|
|
initializePGOMemOPSizeOptLegacyPassPass(Registry);
|
|
initializeInstrProfilingLegacyPassPass(Registry);
|
|
initializeMemorySanitizerPass(Registry);
|
|
initializeHWAddressSanitizerPass(Registry);
|
|
initializeThreadSanitizerPass(Registry);
|
|
initializeSanitizerCoverageModulePass(Registry);
|
|
initializeDataFlowSanitizerPass(Registry);
|
|
initializeEfficiencySanitizerPass(Registry);
|
|
}
|
|
|
|
/// LLVMInitializeInstrumentation - C binding for
|
|
/// initializeInstrumentation.
|
|
void LLVMInitializeInstrumentation(LLVMPassRegistryRef R) {
|
|
initializeInstrumentation(*unwrap(R));
|
|
}
|