shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

[SandboxIR][Bench] SandboxIR creation (#108278)

Browse Source 
Adds a benchmark for the overhead of SandboxIR creation.
This commit is contained in:
vporpo 2024-09-11 15:28:37 -07:00 committed by GitHub
parent 07a7bdc806
commit 31d4837273
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 35 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
llvm/benchmarks/SandboxIRBench.cpp
View File

@ -89,6 +89,31 @@ static std::string generateBBWalkIR(unsigned Size) {
return SS.str();
}
template <IR IRTy> static void SBoxIRCreation(benchmark::State &State) {
static_assert(IRTy != IR::LLVM, "Expected SBoxTracking or SBoxNoTracking");
LLVMContext LLVMCtx;
unsigned NumInstrs = State.range(0);
std::unique_ptr<llvm::Module> LLVMM;
std::string IRStr = generateBBWalkIR(NumInstrs);
LLVMM = parseIR(LLVMCtx, IRStr.c_str());
llvm::Function *LLVMF = &*LLVMM->getFunction("foo");
for (auto _ : State) {
State.PauseTiming();
sandboxir::Context Ctx(LLVMCtx);
if constexpr (IRTy == IR::SBoxTracking)
Ctx.save();
State.ResumeTiming();
sandboxir::Function *F = Ctx.createFunction(LLVMF);
benchmark::DoNotOptimize(F);
State.PauseTiming();
if constexpr (IRTy == IR::SBoxTracking)
Ctx.accept();
State.ResumeTiming();
}
}
template <IR IRTy> static void BBWalk(benchmark::State &State) {
LLVMContext LLVMCtx;
sandboxir::Context Ctx(LLVMCtx);
@ -189,6 +214,16 @@ template <IR IRTy> static void RUOW(benchmark::State &State) {
finalize<IRTy>(Ctx);
}
// Measure the time it takes to create Sandbox IR without/with tracking.
BENCHMARK(SBoxIRCreation<IR::SBoxNoTracking>)
->Args({10})
->Args({100})
->Args({1000});
BENCHMARK(SBoxIRCreation<IR::SBoxTracking>)
->Args({10})
->Args({100})
->Args({1000});
BENCHMARK(GetType<IR::LLVM>);
BENCHMARK(GetType<IR::SBoxNoTracking>);