fix: Fix shard_batch_size = 0 support and re-add the cycle count pr…

…int for execution (#163)

* refactor: reinstate shard_batch_size

Restore the ability to run the entire process in one chunk by setting `SHARD_BATCH_SIZE` to `0`.

Fixes #160

* feat: Print summary at end of execution

---------

Co-authored-by: François Garillot <francois@garillot.net>
Co-authored-by: wwared <541936+wwared@users.noreply.github.com>

core/src/runtime/mod.rs

@@ -1037,6 +1037,10 @@ impl<'a> Runtime<'a> {
         self.emit_events = false;
         self.print_report = true;
         while !self.execute()? {}
+
+        // Print the summary.
+        tracing::info!("summary: cycles={}", self.state.global_clk,);
+
         Ok(())
     }
 

prover/src/lib.rs

@@ -423,7 +423,12 @@ impl SphinxProver {
         );
 
         let mut reduce_proofs = Vec::new();
-        let shard_batch_size = opts.recursion_opts.shard_batch_size;
+        // We want the ability to set SHARD_BATCH_SIZE to 0 to run everything in one chunk
+        let shard_batch_size = if opts.recursion_opts.shard_batch_size > 0 {
+            opts.recursion_opts.shard_batch_size
+        } else {
+            usize::MAX
+        };
         for inputs in core_inputs.chunks(shard_batch_size) {
             let proofs = inputs
                 .into_par_iter()