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/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);
 
@@ -85,6 +87,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 +318,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 +347,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 +406,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![];