feat(execution_utils): use ExecutionProverConfiguration when instantiating UnrolledProver

Author: robik75

execution_utils/src/lib.rs

@@ -14,6 +14,9 @@ use verifier_common::transcript::Blake2sBufferingTranscript;
 pub use prover_examples;
 pub use setups;
 
+#[cfg(feature = "gpu_prover")]
+pub use gpu_prover;
+
 mod constants;
 mod proofs;
 #[cfg(feature = "verifier_binaries")]

execution_utils/src/unrolled_gpu.rs

@@ -96,13 +96,11 @@ pub const RECURSION_UNIFIED_TXT: &[u8] =
 impl UnrolledProver {
     pub fn new(
         path_without_bin: &String,
-        replay_worker_threads_count: usize,
+        prover_configuration: ExecutionProverConfiguration,
         max_level: UnrolledProverLevel,
     ) -> Self {
+        let mut prover = ExecutionProver::with_configuration(prover_configuration);
         let mut level_data = BTreeMap::new();
-        let mut configuration = ExecutionProverConfiguration::default();
-        configuration.replay_worker_threads_count = replay_worker_threads_count;
-        let mut prover = ExecutionProver::with_configuration(configuration);
 
         {
             let bin_path = format!("{}.bin", path_without_bin);

gpu_prover_test/src/main.rs

@@ -70,7 +70,13 @@ mod tests {
             >(&padded_binary_u32);
         serde_json::to_writer_pretty(File::create("layouts.json").unwrap(), &compiled_layouts)
             .unwrap();
-        let mut prover = ExecutionProver::with_configuration(Default::default());
+        let configuration = ExecutionProverConfiguration {
+            replay_worker_threads_count: 4,
+            host_allocators_per_job_count: 64,
+            host_allocators_per_device_count: 128,
+            ..Default::default()
+        };
+        let mut prover = ExecutionProver::with_configuration(configuration);
         prover.add_binary(
             0,
             ExecutionKind::Unrolled,
@@ -116,9 +122,15 @@ mod tests {
             .map(|el| u32::from_be_bytes(*el))
             .collect();
         let app_path = "../riscv_transpiler/examples/zksync_os/app";
+        let configuration = ExecutionProverConfiguration {
+            replay_worker_threads_count: 4,
+            host_allocators_per_job_count: 64,
+            host_allocators_per_device_count: 128,
+            ..Default::default()
+        };
         let prover = UnrolledProver::new(
             &app_path.to_string(),
-            8,
+            configuration,
             UnrolledProverLevel::RecursionUnified,
         );
         let source = QuasiUARTSource::new_with_reads(witness);