fix: abort signal

0xPolygonZero · Nov 1, 2024 · 7a26c43 · 7a26c43
1 parent 9ba0fa0
commit 7a26c43
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/zero/src/prover_state/mod.rs b/zero/src/prover_state/mod.rs
@@ -220,15 +220,16 @@ impl ProverStateManager {
             input,
             segment_data,
             &mut TimingTree::default(),
-            abort_signal,
+            abort_signal.clone(),
         )?;
 
         let table_circuits = self.load_table_circuits(&config, &all_proof)?;
 
-        let proof_with_pvs =
-            p_state()
-                .state
-                .prove_segment_after_initial_stark(all_proof, &table_circuits, None)?;
+        let proof_with_pvs = p_state().state.prove_segment_after_initial_stark(
+            all_proof,
+            &table_circuits,
+            abort_signal,
+        )?;
 
         Ok(proof_with_pvs)
     }