Add explicit panic message

0xPolygonZero · Jul 27, 2024 · 40f594b · 40f594b
1 parent c84b723
commit 40f594b
Showing 1 changed file with 21 additions and 4 deletions.
diff --git a/evm_arithmetization/src/prover.rs b/evm_arithmetization/src/prover.rs
@@ -564,9 +564,8 @@ impl<F: RichField> SegmentDataIterator<F> {
 
         // Run the interpreter to get `registers_after` and the partial data for the
         // next segment.
-        if let Ok((updated_registers, mem_after)) =
-            set_registers_and_run(segment_data.registers_after, &mut self.interpreter)
-        {
+        let run = set_registers_and_run(segment_data.registers_after, &mut self.interpreter);
+        if let Ok((updated_registers, mem_after)) = run {
             let partial_segment_data = Some(build_segment_data(
                 segment_index + 1,
                 Some(updated_registers),
@@ -578,9 +577,27 @@ impl<F: RichField> SegmentDataIterator<F> {
             segment_data.registers_after = updated_registers;
             Some((segment_data, partial_segment_data))
         } else {
-            panic!("Segment generation failed");
+            let inputs = &self.interpreter.get_generation_state().inputs;
+            let block = inputs.block_metadata.block_number;
+            let txn_range = match inputs.txn_hashes.len() {
+                0 => "Dummy".to_string(),
+                1 => format!("{:?}", inputs.txn_number_before),
+                _ => format!(
+                    "{:?}_{:?}",
+                    inputs.txn_number_before,
+                    inputs.txn_number_before + inputs.txn_hashes.len()
+                ),
+            };
+            panic!(
+                "Segment generation {:?} for block {:?} ({}) failed with error {:?}",
+                segment_index,
+                block,
+                txn_range,
+                run.unwrap_err()
+            );
         }
     }
+    }
 }
 
 impl<F: RichField> Iterator for SegmentDataIterator<F> {