adding parallel iterator

Layr-Labs · Aug 29, 2024 · 5ba9700 · 5ba9700
1 parent ace9bf4
commit 5ba9700
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/kzg.rs b/src/kzg.rs
@@ -13,6 +13,7 @@ use ark_serialize::Read;
 use ark_std::{ops::Div, str::FromStr, One, Zero};
 use crossbeam_channel::{bounded, Sender};
 use num_traits::ToPrimitive;
+use rayon::iter::{IntoParallelRefIterator, ParallelIterator};
 use std::{fs::File, io, io::BufReader};
 
 #[derive(Debug, PartialEq, Clone)]
@@ -559,15 +560,15 @@ impl Kzg {
         }
 
         let points_projective: Vec<G1Projective> = self.g1[..length]
-            .iter()
+            .par_iter()
             .map(|&p| G1Projective::from(p))
             .collect();
 
         match GeneralEvaluationDomain::<Fr>::new(length) {
             Some(domain) => {
                 let ifft_result = domain.ifft(&points_projective);
                 let ifft_result_affine: Vec<_> =
-                    ifft_result.iter().map(|p| p.into_affine()).collect();
+                    ifft_result.par_iter().map(|p| p.into_affine()).collect();
                 Ok(ifft_result_affine)
             },
             None => Err(KzgError::FftError(