Synthesizer is restored based on Ale-75, Playground is deleted from this repository. Find Playground from the repo 'tokamak-zk-evm-playground'

Author: JehyukJang

packages/backend/libs/src/group_structures/mod.rs

@@ -1,4 +1,5 @@
-pub mod math;
+pub mod polynomials;
+pub mod vectors;
 pub mod tools;
 pub mod group_structures;
 pub const s_max: usize = 256;

packages/backend/libs/src/lib.rs

@@ -9,7 +9,7 @@ use std::cmp;
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::math::{DensePolynomialExt, BivariatePolynomial};
+    use crate::polynomials::{DensePolynomialExt, BivariatePolynomial};
 
     // Helper function: Create a simple 2D polynomial
     fn create_simple_polynomial() -> DensePolynomialExt {
@@ -460,4 +460,46 @@ mod tests {
     }
 
     // More tests can be added as needed
+}
+
+#[cfg(test)]
+mod tests_vectors {
+    use icicle_bls12_381::curve::{ScalarField, ScalarCfg};
+    use icicle_core::traits::{Arithmetic, FieldConfig, FieldImpl, GenerateRandom};
+    use icicle_runtime::memory::{HostOrDeviceSlice, HostSlice};
+    use std::cmp;
+    
+    use crate::vectors::{outer_product_two_vecs, point_mul_two_vecs};
+
+    macro_rules! scalar_vec {
+        ( $( $x:expr ),* ) => {
+            vec![
+                $( ScalarField::from_u32($x) ),*
+            ].into_boxed_slice()
+        };
+    }
+
+    #[test]
+    fn test_point_mul_two_vecs() {
+        let vec1 = scalar_vec![1, 2, 3];
+        let vec2 = scalar_vec![4, 5];
+        let vec3 = scalar_vec![2, 0, 2, 4];
+
+        let mut res = vec![ScalarField::zero(); 6].into_boxed_slice();
+        outer_product_two_vecs(&vec1, &vec2, &mut res);
+        println!("res : {:?}", res);
+
+        let mut res = vec![ScalarField::zero(); 6].into_boxed_slice();
+        outer_product_two_vecs(&vec2, &vec1, &mut res);
+        println!("res : {:?}", res);
+
+        let mut res = vec![ScalarField::zero(); 12].into_boxed_slice();
+        outer_product_two_vecs(&vec1, &vec3, &mut res);
+        println!("res : {:?}", res);
+
+        let mut res = vec![ScalarField::zero(); 8].into_boxed_slice();
+        outer_product_two_vecs(&vec3, &vec2, &mut res);
+        println!("res : {:?}", res);
+
+    }
 }

packages/backend/libs/src/math/mod.rs renamed to packages/backend/libs/src/polynomials/mod.rs

@@ -1,4 +1,4 @@
-use super::math::{DensePolynomialExt, BivariatePolynomial};
+use super::polynomials::{DensePolynomialExt, BivariatePolynomial};
 
 use icicle_bls12_381::curve::{ScalarField, ScalarCfg};
 use icicle_bls12_381::polynomials::DensePolynomial;

packages/backend/libs/src/tests.rs

@@ -0,0 +1,59 @@
+use icicle_core::vec_ops::{VecOps, VecOpsConfig};
+use icicle_bls12_381::curve::{ScalarCfg, ScalarField};
+use icicle_core::traits::{Arithmetic, FieldConfig, FieldImpl, GenerateRandom};
+use icicle_runtime::memory::{HostOrDeviceSlice, HostSlice, DeviceSlice, DeviceVec};
+
+pub fn point_mul_two_vecs(lhs: &Box<[ScalarField]>, rhs: &Box<[ScalarField]>, res: &mut Box<[ScalarField]>){
+    if lhs.len() != rhs.len() || lhs.len() != res.len() {
+        panic!("Mismatch of sizes of vectors to be pointwise-multiplied");
+    }
+    let vec_ops_cfg = VecOpsConfig::default();
+    let lhs_buff = HostSlice::from_slice(lhs);
+    let rhs_buff = HostSlice::from_slice(rhs);
+    let res_buff = HostSlice::from_mut_slice(res);
+    ScalarCfg::mul(lhs_buff, rhs_buff, res_buff, &vec_ops_cfg).unwrap();
+}
+
+pub fn outer_product_two_vecs(col_vec: &Box<[ScalarField]>, row_vec: &Box<[ScalarField]>, res: &mut Box<[ScalarField]>){
+    if col_vec.len() * row_vec.len() != res.len() {
+        panic!("Mismatch of sizes of vectors to be outer-producted");
+    }
+
+    let col_len = col_vec.len();
+    let row_len = row_vec.len();
+
+    let vec_ops_cfg = VecOpsConfig::default();
+    let min_len = std::cmp::min(row_len, col_len);
+    let max_len = std::cmp::max(row_len, col_len);
+    let max_dir = if max_len == row_len {true } else {false};
+
+    let base_vec = if max_dir { row_vec } else { col_vec };
+     
+    let mut res_untransposed = vec![ScalarField::zero(); res.len()].into_boxed_slice();
+    for ind in 0 .. min_len {
+        let scaler = if max_dir {col_vec[ind]} else {row_vec[ind]};
+        let scaler_vec = vec![scaler; max_len].into_boxed_slice();
+        let mut res_vec = vec![ScalarField::zero(); max_len].into_boxed_slice();
+        ScalarCfg::mul(
+            HostSlice::from_slice(&scaler_vec),
+            HostSlice::from_slice(&base_vec),
+            HostSlice::from_mut_slice(&mut res_vec),
+            &vec_ops_cfg
+        ).unwrap();
+        res_untransposed[ind * max_len .. (ind + 1) * max_len].copy_from_slice(&res_vec);
+    }
+    
+    if !max_dir {
+        let res_untranposed_buf = HostSlice::from_slice(&res_untransposed);
+        let res_buf = HostSlice::from_mut_slice(res);
+        ScalarCfg::transpose(
+            res_untranposed_buf,
+            min_len as u32,
+            max_len as u32,
+            res_buf,
+            &vec_ops_cfg).unwrap();
+    } else {
+        res.clone_from(&res_untransposed);
+    }
+
+}

packages/backend/libs/src/tools/mod.rs

@@ -1,5 +1,5 @@
-#names=("bufferPrvInPubOut" "bufferPubInPrvOut" "bufferPrvInPrvOut" "ADD" "MUL" "SUB" "DIV" "SDIV" "MOD" "SMOD" "ADDMOD" "MULMOD" "EQ" "ISZERO" "SHL" "SHR" "LT" "GT" "NOT" "BYTE" "SAR" "SIGNEXTEND" "SLT" "SGT" "AND" "OR" "XOR" "DecToBit" "SubEXP")
-names=("ADD" "MUL" "SUB")
+names=("bufferPrvInPubOut" "bufferPubInPrvOut" "bufferPrvInPrvOut" "ADD" "MUL" "SUB" "DIV" "SDIV" "MOD" "SMOD" "ADDMOD" "MULMOD" "EQ" "ISZERO" "SHL" "SHR" "LT" "GT" "NOT" "BYTE" "SAR" "SIGNEXTEND" "SLT" "SGT" "AND" "OR" "XOR" "DecToBit" "SubEXP")
+#names=("ADD" "MUL" "SUB")
 CURVE_NAME="bls12381"
 
 # get the directory of the script