tmp

PolyhedraZK · Sep 3, 2024 · 4bde21d · 4bde21d
1 parent c055738
commit 4bde21d
Show file tree

Hide file tree

Showing 2 changed files with 251 additions and 69 deletions.
diff --git a/src/prover/scratchpad.rs b/src/prover/scratchpad.rs
@@ -6,10 +6,15 @@ pub struct GkrScratchpad<C: GKRConfig> {
     pub(crate) hg_evals_5: Vec<C::ChallengeField>,
     pub(crate) hg_evals_1: Vec<C::ChallengeField>,
     pub(crate) hg_evals: Vec<C::Field>,
+    pub(crate) simd_var_v_evals: Vec<C::ChallengeField>,
+    pub(crate) simd_var_hg_evals: Vec<C::ChallengeField>,
 
     pub(crate) eq_evals_at_rx: Vec<C::ChallengeField>,
     pub(crate) eq_evals_at_rz0: Vec<C::ChallengeField>,
     pub(crate) eq_evals_at_rz1: Vec<C::ChallengeField>,
+    pub(crate) eq_evals_at_rz_combined: Vec<C::Field>,
+    pub(crate) eq_evals_at_r_simd0: Vec<C::ChallengeField>,
+    pub(crate) eq_evals_at_r_simd1: Vec<C::ChallengeField>,
     pub(crate) eq_evals_first_half: Vec<C::ChallengeField>,
     pub(crate) eq_evals_second_half: Vec<C::ChallengeField>,
 
@@ -26,10 +31,15 @@ impl<C: GKRConfig> GkrScratchpad<C> {
             hg_evals_5: vec![C::ChallengeField::default(); max_input_num],
             hg_evals_1: vec![C::ChallengeField::default(); max_input_num],
             hg_evals: vec![C::Field::default(); max_input_num],
+            simd_var_v_evals: vec![C::ChallengeField::default(); C::get_field_pack_size()],
+            simd_var_hg_evals: vec![C::ChallengeField::default(); C::get_field_pack_size()],
 
             eq_evals_at_rx: vec![C::ChallengeField::default(); max_input_num],
             eq_evals_at_rz0: vec![C::ChallengeField::default(); max_output_num],
             eq_evals_at_rz1: vec![C::ChallengeField::default(); max_output_num],
+            eq_evals_at_rz_combined: vec![C::Field::default(); max_output_num],
+            eq_evals_at_r_simd0: vec![C::ChallengeField::default(); C::get_field_pack_size()],
+            eq_evals_at_r_simd1: vec![C::ChallengeField::default(); C::get_field_pack_size()],
             eq_evals_first_half: vec![C::ChallengeField::default(); max_output_num],
             eq_evals_second_half: vec![C::ChallengeField::default(); max_output_num],