test(bench): bench proving with cached / jit trace

triton-vm/Cargo.toml

@@ -55,6 +55,10 @@ workspace = true
 name = "bezout_coeffs"
 harness = false
 
+[[bench]]
+name = "cached_vs_jit_trace"
+harness = false
+
 [[bench]]
 name = "mem_io"
 harness = false

triton-vm/benches/cached_vs_jit_trace.rs

@@ -0,0 +1,34 @@
+use criterion::criterion_group;
+use criterion::criterion_main;
+use criterion::Criterion;
+use twenty_first::prelude::*;
+
+use triton_vm::config::CacheDecision;
+use triton_vm::prelude::*;
+
+criterion_main!(benches);
+criterion_group!(
+    name = benches;
+    config = Criterion::default().sample_size(10);
+    targets = prove_fib<1>, prove_fib<100>, prove_fib<1_000>,
+);
+
+fn prove_fib<const N: u64>(c: &mut Criterion) {
+    let stark = Stark::default();
+    let program = triton_vm::example_programs::FIBONACCI_SEQUENCE.clone();
+    let public_input = PublicInput::from(bfe_array![N]);
+    let non_determinism = NonDeterminism::default();
+    let (aet, output) = program
+        .trace_execution(public_input, non_determinism)
+        .unwrap();
+    let claim = Claim::about_program(&program)
+        .with_input(bfe_vec![N])
+        .with_output(output);
+
+    let mut group = c.benchmark_group(format!("prove_fib_{N}"));
+    triton_vm::config::overwrite_lde_trace_caching_to(CacheDecision::Cache);
+    group.bench_function("cache", |b| b.iter(|| stark.prove(&claim, &aet, &mut None)));
+    triton_vm::config::overwrite_lde_trace_caching_to(CacheDecision::NoCache);
+    group.bench_function("jit", |b| b.iter(|| stark.prove(&claim, &aet, &mut None)));
+    group.finish();
+}