add construction of AET (witness generation) to profiling

TritonVM · Apr 19, 2023 · c6e7b1e · c6e7b1e
1 parent ab49df7
commit c6e7b1e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/triton-vm/benches/prove_fib_100.rs b/triton-vm/benches/prove_fib_100.rs
@@ -2,12 +2,12 @@ use criterion::criterion_group;
 use criterion::criterion_main;
 use criterion::BenchmarkId;
 use criterion::Criterion;
+use triton_opcodes::program::Program;
 use triton_profiler::prof_start;
 use triton_profiler::prof_stop;
 use triton_profiler::triton_profiler::Report;
 use triton_profiler::triton_profiler::TritonProfiler;
 
-use triton_opcodes::program::Program;
 use triton_vm::proof::Claim;
 use triton_vm::shared_tests::FIBONACCI_SEQUENCE;
 use triton_vm::stark::Stark;
@@ -25,12 +25,16 @@ fn prove_fib_100(criterion: &mut Criterion) {
     let mut report: Report = Report::placeholder();
 
     // stark object
+    prof_start!(maybe_profiler, "parse program");
     let program = match Program::from_code(FIBONACCI_SEQUENCE) {
         Err(e) => panic!("Cannot compile source code into program: {e}"),
         Ok(p) => p,
     };
+    prof_stop!(maybe_profiler, "parse program");
     let input = vec![100_u64.into()];
+    prof_start!(maybe_profiler, "generate AET");
     let (aet, output, err) = simulate(&program, input.clone(), vec![]);
+    prof_stop!(maybe_profiler, "generate AET");
     if let Some(error) = err {
         panic!("The VM encountered the following problem: {error}");
     }
@@ -44,10 +48,7 @@ fn prove_fib_100(criterion: &mut Criterion) {
         padded_height,
     };
     let stark = Stark::new(claim, Default::default());
-    //start the profiler
-    prof_start!(maybe_profiler, "prove");
     let _proof = stark.prove(aet.clone(), &mut maybe_profiler);
-    prof_stop!(maybe_profiler, "prove");
 
     if let Some(profiler) = maybe_profiler.as_mut() {
         profiler.finish();

diff --git a/triton-vm/benches/prove_halt.rs b/triton-vm/benches/prove_halt.rs
@@ -1,10 +1,12 @@
 use criterion::criterion_group;
 use criterion::criterion_main;
 use criterion::Criterion;
-
 use triton_opcodes::program::Program;
+use triton_profiler::prof_start;
+use triton_profiler::prof_stop;
 use triton_profiler::triton_profiler::Report;
 use triton_profiler::triton_profiler::TritonProfiler;
+
 use triton_vm::proof::Claim;
 use triton_vm::shared_tests::save_proof;
 use triton_vm::stark::Stark;
@@ -18,13 +20,17 @@ fn prove_halt(_criterion: &mut Criterion) {
     let mut report: Report = Report::placeholder();
 
     // stark object
+    prof_start!(maybe_profiler, "parse program");
     let program = match Program::from_code("halt") {
         Err(e) => panic!("Cannot compile source code into program: {e}"),
         Ok(p) => p,
     };
+    prof_stop!(maybe_profiler, "parse program");
 
     // witness
+    prof_start!(maybe_profiler, "generate AET");
     let (aet, output, err) = simulate(&program, vec![], vec![]);
+    prof_stop!(maybe_profiler, "generate AET");
     if let Some(error) = err {
         panic!("The VM encountered the following problem: {error}");
     }