Also print code generated so far.

powdr-labs · Jan 14, 2025 · faf14f5 · faf14f5
1 parent 524ae00
commit faf14f5
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/executor/src/witgen/jit/witgen_inference.rs b/executor/src/witgen/jit/witgen_inference.rs
@@ -15,7 +15,7 @@ use powdr_number::FieldElement;
 
 use crate::witgen::{
     data_structures::mutable_state::MutableState, global_constraints::RangeConstraintSet,
-    range_constraints::RangeConstraint, FixedData, QueryCallback,
+    jit::effect::format_code, range_constraints::RangeConstraint, FixedData, QueryCallback,
 };
 
 use super::{
@@ -230,7 +230,11 @@ impl<'a, T: FieldElement, FixedEval: FixedEvaluator<T>> WitgenInference<'a, T, F
         let result = (lhs_evaluated - rhs_evaluated)
             .solve()
             .map_err(|e| {
-                panic!("Error solving equality constraint: {lhs} = {rhs} on row {offset}: {e}");
+                panic!(
+                    "Error solving equality constraint: {lhs} = {rhs} on row {offset}: {e}\n\
+                    Code generated since the previous branching point:\n{}",
+                    format_code(&self.code)
+                );
             })
             .unwrap();
         if result.complete && result.effects.is_empty() {
@@ -266,7 +270,11 @@ impl<'a, T: FieldElement, FixedEval: FixedEvaluator<T>> WitgenInference<'a, T, F
         (lhs_evaluated - rhs_evaluated)
             .solve()
             .map_err(|e| {
-                panic!("Error solving equality constraint: {lhs} = {rhs} on row {offset}: {e}");
+                panic!(
+                    "Error solving equality constraint: {lhs} = {rhs} on row {offset}: {e}\n\
+                    Code generated since the previous branching point:\n{}",
+                    format_code(&self.code)
+                );
             })
             .unwrap()
     }