test: abort signal

Cargo.toml

@@ -70,7 +70,7 @@ num-traits = "0.2.19"
 nunny = "0.2.1"
 once_cell = "1.19.0"
 # TODO: update when paladin is released to 0.4.4 with abort signal support
-paladin-core = { git = "https://github.com/0xPolygonZero/paladin.git", branch = "feat/abort-signal" }
+paladin-core = { git = "https://github.com/0xPolygonZero/paladin.git", branch = "test_abort_signal" }
 parking_lot = "0.12.3"
 pest = "2.7.10"
 pest_derive = "2.7.10"

evm_arithmetization/src/fixed_recursive_verifier.rs

@@ -2063,6 +2063,7 @@ where
                 )?;
             }
 
+            tracing::log::info!(">>>>>>>>>>>>>>>>>>> calling check_abort_signal from `prove_segment_with_all_proofs`: {abort_signal:?}");
             check_abort_signal(abort_signal.clone())?;
         }
 
@@ -2184,6 +2185,7 @@ where
                 )?;
             }
 
+            tracing::log::info!(">>>>>>>>>>>>>>>>>>> calling check_abort_signal from `prove_segment_after_initial_stark`: {abort_signal:?}");
             check_abort_signal(abort_signal.clone())?;
         }
 

evm_arithmetization/src/prover.rs

@@ -42,6 +42,8 @@ where
 {
     features_check(&inputs);
 
+    tracing::log::info!(">>>>>>>>>>>>>>>>>>> prove {abort_signal:?}");
+
     // Sanity check on the provided config
     assert_eq!(DEFAULT_CAP_LEN, 1 << config.fri_config.cap_height);
 
@@ -53,6 +55,7 @@ where
         generate_traces(all_stark, &inputs, config, segment_data, timing)?
     );
 
+    tracing::log::info!(">>>>>>>>>>>>>>>>>>> calling check_abort_signal from `prove`: {abort_signal:?}");
     check_abort_signal(abort_signal.clone())?;
 
     let proof = prove_with_traces(
@@ -85,6 +88,8 @@ where
     let rate_bits = config.fri_config.rate_bits;
     let cap_height = config.fri_config.cap_height;
 
+    tracing::info!(">>>>>>>>>>>>>>>>>>> PROVE WITH TRACES abort signal {abort_signal:?}");
+
     // For each STARK, we compute the polynomial commitments for the polynomials
     // interpolating its trace.
     let trace_commitments = timed!(
@@ -314,6 +319,7 @@ where
     C: GenericConfig<D, F = F>,
     S: Stark<F, D>,
 {
+    tracing::info!(">>>>>>> PROVING SINGLE TABLE abort signal {abort_signal:?}");
     check_abort_signal(abort_signal.clone())?;
 
     // Clear buffered outputs.
@@ -342,8 +348,10 @@ where
 /// the workers, which will result in an early abort for all the other processes
 /// involved in the same set of transactions.
 pub fn check_abort_signal(abort_signal: Option<Arc<AtomicBool>>) -> Result<()> {
+    tracing::info!(">>>>>>>>>>>>>>>>>>> check abort signal {abort_signal:?}");
     if let Some(signal) = abort_signal {
         if signal.load(Ordering::Relaxed) {
+            tracing::info!(">>>>>>>>>>>>>>>>>>> STOPING JOB WITH ABORT SIGNAL!!!");
             return Err(anyhow!("Stopping job from abort signal."));
         }
     }
@@ -399,6 +407,7 @@ pub mod testing {
         F: RichField + Extendable<D>,
         C: GenericConfig<D, F = F>,
     {
+        tracing::log::info!(">>>>>>>>>>>>>>>>>>> prove_all_segments {abort_signal:?}");
         let segment_data_iterator = SegmentDataIterator::<F>::new(&inputs, Some(max_cpu_len_log));
         let inputs = inputs.trim();
         let mut proofs = vec![];

zero/src/prover_state/mod.rs

@@ -220,15 +220,15 @@ 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)?;
+                .prove_segment_after_initial_stark(all_proof, &table_circuits, abort_signal)?;
 
         Ok(proof_with_pvs)
     }