Fix prover input.

lambdaclass · Jul 1, 2024 · 78468a3 · 78468a3
1 parent b70d141
commit 78468a3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/vm/src/air_private_input.rs b/vm/src/air_private_input.rs
@@ -19,6 +19,8 @@ pub struct AirPrivateInputSerializable {
     #[serde(skip_serializing_if = "Option::is_none")]
     range_check: Option<Vec<PrivateInput>>,
     #[serde(skip_serializing_if = "Option::is_none")]
+    range_check96: Option<Vec<PrivateInput>>,
+    #[serde(skip_serializing_if = "Option::is_none")]
     ecdsa: Option<Vec<PrivateInput>>,
     #[serde(skip_serializing_if = "Option::is_none")]
     bitwise: Option<Vec<PrivateInput>>,
@@ -157,6 +159,7 @@ impl AirPrivateInput {
             memory_path,
             pedersen: self.0.get(&BuiltinName::pedersen).cloned(),
             range_check: self.0.get(&BuiltinName::range_check).cloned(),
+            range_check96: self.0.get(&BuiltinName::range_check96).cloned(),
             ecdsa: self.0.get(&BuiltinName::ecdsa).cloned(),
             bitwise: self.0.get(&BuiltinName::bitwise).cloned(),
             ec_op: self.0.get(&BuiltinName::ec_op).cloned(),

diff --git a/vm/src/vm/runners/builtin_runner/signature.rs b/vm/src/vm/runners/builtin_runner/signature.rs
@@ -220,7 +220,6 @@ impl SignatureBuiltinRunner {
                 private_inputs.push(PrivateInput::Signature(PrivateInputSignature {
                     index: addr
                         .offset
-                        .saturating_sub(self.base)
                         .checked_div(CELLS_PER_SIGNATURE as usize)
                         .unwrap_or_default(),
                     pubkey: *pubkey,