Use faster field implementation in wasm target

Author: sebastiencs

Cargo.lock

@@ -45,13 +45,16 @@ indexing_slicing = "warn"
 mina-p2p-messages = { path = "mina-p2p-messages" }
 poseidon = { path = "poseidon" }
 ledger = { path = "ledger", package = "mina-tree" }
-mina-hasher = { git = "https://github.com/openmina/proof-systems", rev = "d0cd63b" }
-mina-signer = { git = "https://github.com/openmina/proof-systems", rev = "d0cd63b" }
-mina-curves = { git = "https://github.com/openmina/proof-systems", rev = "d0cd63b" }
-o1-utils = { git = "https://github.com/openmina/proof-systems", rev = "d0cd63b" }
-kimchi = { git = "https://github.com/openmina/proof-systems", rev = "d0cd63b" }
-mina-poseidon = {git = "https://github.com/openmina/proof-systems", rev = "d0cd63b"}
-poly-commitment = {git = "https://github.com/openmina/proof-systems", rev = "d0cd63b"}
+
+mina-hasher = { git = "https://github.com/openmina/proof-systems", rev = "c478b19" }
+mina-signer = { git = "https://github.com/openmina/proof-systems", rev = "c478b19" }
+mina-curves = { git = "https://github.com/openmina/proof-systems", rev = "c478b19" }
+# mina-curves = { git = "https://github.com/openmina/proof-systems", rev = "c478b19", features = [ "32x9" ] }
+o1-utils = { git = "https://github.com/openmina/proof-systems", rev = "c478b19" }
+kimchi = { git = "https://github.com/openmina/proof-systems", rev = "c478b19" }
+mina-poseidon = {git = "https://github.com/openmina/proof-systems", rev = "c478b19" }
+poly-commitment = {git = "https://github.com/openmina/proof-systems", rev = "c478b19" }
+
 libp2p = { git = "https://github.com/openmina/rust-libp2p", rev = "5c44c7d9", default-features = false }
 vrf = { path = "vrf" }
 openmina-node-account = { path = "node/account" }
@@ -63,6 +66,9 @@ linkme = "0.3.22"
 static_assertions = "1.1.0"
 juniper = { version = "0.16" }
 
+ark-ff = { version = "0.3.0", features = [ "parallel", "asm", "std" ] }
+# ark-ff = { version = "0.3.0", features = [ "parallel", "asm", "std", "32x9" ] }
+
 
 [profile.fuzz]
 inherits = "release"
@@ -76,10 +82,10 @@ incremental = false
 codegen-units = 1
 
 [patch.crates-io]
-ark-ff = { git = "https://github.com/openmina/algebra", rev = "33a1de2" } # branch: fix-openmina
-ark-ec = { git = "https://github.com/openmina/algebra", rev = "33a1de2" } # branch: fix-openmina
-ark-poly = { git = "https://github.com/openmina/algebra", rev = "33a1de2" } # branch: fix-openmina
-ark-serialize = { git = "https://github.com/openmina/algebra", rev = "33a1de2" } # branch: fix-openmina
+ark-ff = { git = "https://github.com/openmina/algebra", rev = "f42a8ec" } # branch: fix-openmina-webnode
+ark-ec = { git = "https://github.com/openmina/algebra", rev = "f42a8ec" } # branch: fix-openmina-webnode
+ark-poly = { git = "https://github.com/openmina/algebra", rev = "f42a8ec" } # branch: fix-openmina-webnode
+ark-serialize = { git = "https://github.com/openmina/algebra", rev = "f42a8ec" } # branch: fix-openmina-webnode
 
 [profile.test.package."*"]
 opt-level = 3

Cargo.toml

@@ -31,7 +31,7 @@ mina-hasher = { workspace = true }
 mina-p2p-messages = { workspace = true }
 poseidon = { workspace = true }
 hex = "0.4.3"
-ark-ff = { version = "0.3.0", features = [ "parallel", "asm", "std" ] }
+ark-ff = { workspace = true }
 
 [target.'cfg(not(target_family = "wasm"))'.dependencies]
 redux = { workspace = true, features=["serializable_callbacks"] }

core/Cargo.toml

@@ -46,7 +46,7 @@ libc = "0.2"
 
 itertools = "0.10"
 
-ark-ff = { version = "0.3.0", features = [ "parallel", "asm", "std" ] }
+ark-ff = { workspace = true }
 ark-ec = { version = "0.3.0", features = [ "std" ] }
 ark-serialize = { version = "0.3.0", features = [ "std" ] }
 ark-poly = { version = "0.3.0", features = [ "std" ] }

ledger/Cargo.toml

@@ -1760,14 +1760,20 @@ mod tests {
 
     #[test]
     fn test_size_account() {
+        const SIZE_WITH_9LIMBS: usize = 296;
+
         #[cfg(not(target_family = "wasm"))]
         const SIZE: usize = 280;
 
         // FIXME: was 2496bytes before zkapp got boxed, what should be the size now?
         #[cfg(target_family = "wasm")]
         const SIZE: usize = 280;
 
-        assert_eq!(std::mem::size_of::<Account>(), SIZE);
+        if std::mem::size_of::<BigInteger256>() == 9 * 4 {
+            assert_eq!(std::mem::size_of::<Account>(), SIZE_WITH_9LIMBS);
+        } else {
+            assert_eq!(std::mem::size_of::<Account>(), SIZE);
+        }
     }
 
     #[test]

ledger/src/account/account.rs

@@ -90,14 +90,15 @@ fn hash_field(f: &Fp) -> u32 {
     let mut acc = 0;
 
     let bigint: BigInteger256 = (*f).into();
-    let nignore: usize = bigint.0.iter().rev().take_while(|&b| *b == 0).count();
+    let bigint = bigint.to_64x4();
+    let nignore: usize = bigint.iter().rev().take_while(|&b| *b == 0).count();
 
-    for bigint in bigint.0.iter().take(BigInteger256::NUM_LIMBS - nignore) {
+    for bigint in bigint.iter().take(BigInteger256::NUM_LIMBS - nignore) {
         acc = mix(acc, (bigint & 0xFFFF_FFFF) as u32);
         acc = mix(acc, (bigint >> 32) as u32);
     }
 
-    if bigint.0.last().unwrap() & 0x8000_0000_0000_0000 != 0 {
+    if bigint.last().unwrap() & 0x8000_0000_0000_0000 != 0 {
         // TODO: Not sure if that condition is correct
         acc += 1;
     }

ledger/src/port_ocaml/hash.rs

@@ -316,17 +316,17 @@ mod floating_point {
     }
 
     const COEFFICIENTS: [(Sgn, BigInteger256); 11] = [
-        (Sgn::Pos, BigInteger256::new([405058, 0, 0, 0])),
-        (Sgn::Neg, BigInteger256::new([1007582, 0, 0, 0])),
-        (Sgn::Pos, BigInteger256::new([465602, 0, 0, 0])),
-        (Sgn::Neg, BigInteger256::new([161365, 0, 0, 0])),
-        (Sgn::Pos, BigInteger256::new([44739, 0, 0, 0])),
-        (Sgn::Neg, BigInteger256::new([10337, 0, 0, 0])),
-        (Sgn::Pos, BigInteger256::new([2047, 0, 0, 0])),
-        (Sgn::Neg, BigInteger256::new([354, 0, 0, 0])),
-        (Sgn::Pos, BigInteger256::new([54, 0, 0, 0])),
-        (Sgn::Neg, BigInteger256::new([7, 0, 0, 0])),
-        (Sgn::Pos, BigInteger256::new([0, 0, 0, 0])),
+        (Sgn::Pos, BigInteger256::from_64x4([405058, 0, 0, 0])),
+        (Sgn::Neg, BigInteger256::from_64x4([1007582, 0, 0, 0])),
+        (Sgn::Pos, BigInteger256::from_64x4([465602, 0, 0, 0])),
+        (Sgn::Neg, BigInteger256::from_64x4([161365, 0, 0, 0])),
+        (Sgn::Pos, BigInteger256::from_64x4([44739, 0, 0, 0])),
+        (Sgn::Neg, BigInteger256::from_64x4([10337, 0, 0, 0])),
+        (Sgn::Pos, BigInteger256::from_64x4([2047, 0, 0, 0])),
+        (Sgn::Neg, BigInteger256::from_64x4([354, 0, 0, 0])),
+        (Sgn::Pos, BigInteger256::from_64x4([54, 0, 0, 0])),
+        (Sgn::Neg, BigInteger256::from_64x4([7, 0, 0, 0])),
+        (Sgn::Pos, BigInteger256::from_64x4([0, 0, 0, 0])),
     ];
 
     pub struct Params {

ledger/src/proofs/block.rs

@@ -109,7 +109,10 @@ fn sqrt_exn<F: FieldWitness>(x: F, w: &mut Witness<F>) -> F {
 }
 
 fn is_square<F: FieldWitness>(x: F) -> bool {
-    let s = x.pow(F::Params::MODULUS_MINUS_ONE_DIV_TWO);
+    use ark_ff::BigInteger;
+
+    let modulus_minus_one_div_two = F::Params::MODULUS_MINUS_ONE_DIV_TWO.to_64x4();
+    let s = x.pow(modulus_minus_one_div_two);
     s.is_zero() || s.is_one()
 }
 

ledger/src/proofs/group_map.rs

@@ -14,6 +14,7 @@ use crate::ToInputs;
 
 use super::common::{range_check, ForZkappCheck};
 
+#[derive(Debug)]
 pub enum RangeCheckFlaggedKind {
     Add,
     Sub,

ledger/src/proofs/numbers/currency.rs

@@ -58,15 +58,15 @@ impl MessagesForNextWrapProof {
         let field: Fq = hash_fields(&fields);
 
         let bigint: BigInteger256 = field.into_repr();
-        bigint.0
+        bigint.to_64x4()
     }
 
     pub fn hash_checked(&self, w: &mut Witness<Fq>) -> [u64; 4] {
         let fields: Vec<Fq> = self.to_fields();
         let field: Fq = checked_hash2(&fields, w);
 
         let bigint: BigInteger256 = field.into_repr();
-        bigint.0
+        bigint.to_64x4()
     }
 
     // TODO: De-duplicate with above
@@ -75,7 +75,7 @@ impl MessagesForNextWrapProof {
         let field: Fq = crate::proofs::transaction::checked_hash3(&fields, w);
 
         let bigint: BigInteger256 = field.into_repr();
-        bigint.0
+        bigint.to_64x4()
     }
 
     /// Implementation of `to_field_elements`
@@ -152,7 +152,7 @@ where
         let field: Fp = hash_fields(&fields);
 
         let bigint: BigInteger256 = field.into_repr();
-        bigint.0
+        bigint.to_64x4()
     }
 
     /// Implementation of `to_field_elements`