feat(provers): Add optional cycle tracking to proving report (#358)

Author: Andrurachi

crates/prover/airbender/src/prover.rs

@@ -147,13 +147,13 @@ impl zkVMProver for AirbenderProver {
             .map_err(|err| Error::ExecutePanic(panic_msg(err)))??;
 
         let start = Instant::now();
-        let (proof, receipt) = match gpu_prover.prove(&input_words)? {
+        let (proof, receipt, cycles) = match gpu_prover.prove(&input_words)? {
             ProveResult {
                 proof: Proof::Real(proof),
                 receipt,
-                ..
+                cycles,
             } if proof.level() == airbender_host::ProverLevel::RecursionUnified => {
-                (proof.into_inner(), receipt)
+                (proof.into_inner(), receipt, cycles)
             }
             _ => Err(Error::Sdk(airbender_host::HostError::Prover(
                 "Expected Proof::Real in ProverLevel::RecursionUnified".to_string(),
@@ -164,7 +164,10 @@ impl zkVMProver for AirbenderProver {
         Ok((
             words_to_le_bytes(receipt.output).into(),
             AirbenderProof(proof),
-            ProgramProvingReport::new(proving_time),
+            ProgramProvingReport {
+                proving_time,
+                total_num_cycles: Some(cycles),
+            },
         ))
     }
 }

crates/prover/core/src/report.rs

@@ -30,12 +30,19 @@ impl ProgramExecutionReport {
 
 /// ProgramProvingReport produces information about proving a particular
 /// program's instance.
+///
+/// Note: Execution is fused into the proving pipeline.
+/// To get separate execution metrics, call `execute()` before `prove()`.
 #[derive(Debug, Clone, Default, Serialize, Deserialize)]
 pub struct ProgramProvingReport {
     pub proving_time: Duration,
+    pub total_num_cycles: Option<u64>,
 }
 impl ProgramProvingReport {
     pub fn new(proving_time: Duration) -> Self {
-        Self { proving_time }
+        Self {
+            proving_time,
+            total_num_cycles: None,
+        }
     }
 }

crates/prover/risc0/src/prover.rs

@@ -163,7 +163,10 @@ impl zkVMProver for Risc0Prover {
         Ok((
             public_values,
             proof,
-            ProgramProvingReport::new(proving_time),
+            ProgramProvingReport {
+                proving_time,
+                total_num_cycles: Some(prove_info.stats.total_cycles),
+            },
         ))
     }
 }