Add profiling flamegraphs with pprof

This commit is contained in:
2025-12-06 15:58:11 -05:00
parent 74c28c9344
commit 1311cb830b
5 changed files with 444 additions and 36 deletions

View File

@@ -1,21 +1,34 @@
use criterion::{Criterion, criterion_group, criterion_main};
use pprof::criterion::{Output, PProfProfiler};
const PPROF_SAMPLING_FREQ_HZ: i32 = 997;
macro_rules! bench_days {
($($day_num:literal => $day_mod:ident),* $(,)?) => {
$(
use aoc::$day_mod;
mod $day_mod {
use super::*;
use aoc::$day_mod;
fn $day_mod(c: &mut Criterion) {
c.bench_function(concat!(stringify!($day_mod), " part1"), |b| {
b.iter(|| $day_mod::part1($day_mod::INPUT))
});
c.bench_function(concat!(stringify!($day_mod), " part2"), |b| {
b.iter(|| $day_mod::part2($day_mod::INPUT))
});
pub fn part1(c: &mut Criterion) {
c.bench_function(concat!(stringify!($day_mod), " part1"), |b| {
b.iter(|| $day_mod::part1($day_mod::INPUT))
});
}
pub fn part2(c: &mut Criterion) {
c.bench_function(concat!(stringify!($day_mod), " part2"), |b| {
b.iter(|| $day_mod::part2($day_mod::INPUT))
});
}
}
)*
criterion_group!(benches, $($day_mod),*);
criterion_group! {
name = benches;
config = Criterion::default().with_profiler(PProfProfiler::new(PPROF_SAMPLING_FREQ_HZ, Output::Flamegraph(None)));
targets = $($day_mod::part1, $day_mod::part2),*
}
criterion_main!(benches);
};
}