feat: add context to mock mode

succinctlabs · Jan 10, 2025 · 8f8eb34 · 8f8eb34
1 parent fd6609b
commit 8f8eb34
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/crates/sdk/src/cpu/mod.rs b/crates/sdk/src/cpu/mod.rs
@@ -114,7 +114,7 @@ impl CpuProver {
 
         // If we're in mock mode, return a mock proof.
         if self.mock {
-            return self.mock_prove_impl(pk, stdin, mode);
+            return self.mock_prove_impl(pk, stdin, context, mode);
         }
 
         // Generate the core proof.
@@ -186,13 +186,13 @@ impl CpuProver {
         }
     }
 
-    pub(crate) fn mock_prove_impl(
-        &self,
+    pub(crate) fn mock_prove_impl<'a>(
+        &'a self,
         pk: &SP1ProvingKey,
         stdin: &SP1Stdin,
+        context: SP1Context<'a>,
         mode: SP1ProofMode,
     ) -> Result<SP1ProofWithPublicValues> {
-        let context = SP1Context::default();
         let (public_values, _) = self.prover.execute(&pk.elf, stdin, context)?;
         Ok(SP1ProofWithPublicValues::create_mock_proof(pk, public_values, mode, self.version()))
     }

diff --git a/crates/sdk/src/cpu/prove.rs b/crates/sdk/src/cpu/prove.rs
@@ -288,7 +288,7 @@ impl<'a> CpuProveBuilder<'a> {
 
         // Run the prover.
         if mock {
-            prover.mock_prove_impl(pk, &stdin, mode)
+            prover.mock_prove_impl(pk, &stdin, context, mode)
         } else {
             prover.prove_impl(pk, &stdin, opts, context, mode)
         }