Fix Arithmetic trace length (#454)

0xPolygonZero · Jul 30, 2024 · f24c826 · f24c826
1 parent 8f5c343
commit f24c826
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/evm_arithmetization/src/witness/traces.rs b/evm_arithmetization/src/witness/traces.rs
@@ -59,7 +59,12 @@ impl<T: Copy> Traces<T> {
                 .map(|op| match op {
                     Operation::TernaryOperation { .. } => 2,
                     Operation::BinaryOperation { operator, .. } => match operator {
-                        BinaryOperator::Div | BinaryOperator::Mod => 2,
+                        BinaryOperator::Div
+                        | BinaryOperator::Mod
+                        | BinaryOperator::AddFp254
+                        | BinaryOperator::SubFp254
+                        | BinaryOperator::MulFp254
+                        | BinaryOperator::Shr => 2,
                         _ => 1,
                     },
                     Operation::RangeCheckOperation { .. } => 1,