13#ifndef LLVM_CLANG_LIB_CODEGEN_CODEGENPGO_H
14#define LLVM_CLANG_LIB_CODEGEN_CODEGENPGO_H
19#include "llvm/ProfileData/InstrProfReader.h"
32 llvm::GlobalVariable *FuncNameVar;
34 std::array <unsigned, llvm::IPVK_Last + 1> NumValueSites;
35 unsigned NumRegionCounters;
36 uint64_t FunctionHash;
37 std::unique_ptr<llvm::DenseMap<const Stmt *, unsigned>> RegionCounterMap;
38 std::unique_ptr<llvm::DenseMap<const Stmt *, uint64_t>> StmtCountMap;
39 std::unique_ptr<llvm::InstrProfRecord> ProfRecord;
40 std::vector<uint64_t> RegionCounts;
41 uint64_t CurrentRegionCount;
45 : CGM(CGModule), FuncNameVar(nullptr), NumValueSites({{0}}),
46 NumRegionCounters(0), FunctionHash(0), CurrentRegionCount(0) {}
66 auto I = StmtCountMap->find(S);
67 if (I == StmtCountMap->end())
87 llvm::GlobalValue::LinkageTypes
Linkage);
90 llvm::Instruction *ValueSite, llvm::Value *ValuePtr);
96 void setFuncName(llvm::Function *Fn);
97 void setFuncName(StringRef Name, llvm::GlobalValue::LinkageTypes
Linkage);
98 void mapRegionCounters(
const Decl *D);
99 void computeRegionCounts(
const Decl *D);
100 void applyFunctionAttributes(llvm::IndexedInstrProfReader *PGOReader,
102 void loadRegionCounts(llvm::IndexedInstrProfReader *PGOReader,
104 bool skipRegionMappingForDecl(
const Decl *D);
105 void emitCounterRegionMapping(
const Decl *D);
113 if (!RegionCounterMap)
119 auto Index = (*RegionCounterMap)[S];
120 if (Index >= RegionCounts.size())
122 return RegionCounts[Index];
This class organizes the cross-function state that is used while generating LLVM code.
void assignRegionCounters(GlobalDecl GD, llvm::Function *Fn)
Assign counters to regions and configure them for PGO of a given function.
std::optional< uint64_t > getStmtCount(const Stmt *S) const
Check if an execution count is known for a given statement.
void setCurrentRegionCount(uint64_t Count)
Set the counter value for the current region.
uint64_t getRegionCount(const Stmt *S)
Return the region count for the counter at the given index.
void setValueProfilingFlag(llvm::Module &M)
void valueProfile(CGBuilderTy &Builder, uint32_t ValueKind, llvm::Instruction *ValueSite, llvm::Value *ValuePtr)
uint64_t getCurrentRegionCount() const
Return the counter value of the current region.
void setCurrentStmt(const Stmt *S)
If the execution count for the current statement is known, record that as the current count.
void emitEmptyCounterMapping(const Decl *D, StringRef FuncName, llvm::GlobalValue::LinkageTypes Linkage)
Emit a coverage mapping range with a counter zero for an unused declaration.
void emitCounterIncrement(CGBuilderTy &Builder, const Stmt *S, llvm::Value *StepV)
CodeGenPGO(CodeGenModule &CGModule)
bool haveRegionCounts() const
Whether or not we have PGO region data for the current function.
Decl - This represents one declaration (or definition), e.g.
GlobalDecl - represents a global declaration.
Stmt - This represents one statement.
Linkage
Describes the different kinds of linkage (C++ [basic.link], C99 6.2.2) that an entity may have.