more parallelization

hyperplonk/src/snark.rs

@@ -76,7 +76,7 @@ where
             .collect();
 
         let selector_commitments = selector_oracles
-            .iter()
+            .par_iter()
             .map(|poly| PCS::commit(&pcs_prover_param, poly))
             .collect::<Result<Vec<_>, _>>()?;
 

subroutines/src/poly_iop/sum_check/prover.rs

@@ -9,7 +9,7 @@ use arithmetic::{fix_variables, VirtualPolynomial};
 use ark_ff::PrimeField;
 use ark_poly::DenseMultilinearExtension;
 use ark_std::{end_timer, start_timer, vec::Vec};
-use rayon::prelude::IntoParallelIterator;
+use rayon::prelude::{IntoParallelIterator, IntoParallelRefIterator};
 use std::sync::Arc;
 
 #[cfg(feature = "parallel")]
@@ -71,7 +71,7 @@ impl SumCheckProver for IOPProverState {
         let mut flattened_ml_extensions: Vec<DenseMultilinearExtension<F>> = self
             .poly
             .flattened_ml_extensions
-            .iter()
+            .par_iter()
             .map(|x| x.as_ref().clone())
             .collect();
 
@@ -132,9 +132,7 @@ impl SumCheckProver for IOPProverState {
                             }
                         })
                         .collect::<Vec<F>>();
-                    for val in evals.iter() {
-                        *e += val
-                    }
+                    *e += evals.par_iter().sum::<F>();
                 }
             } else {
                 for (t, e) in products_sum.iter_mut().enumerate() {
@@ -161,10 +159,7 @@ impl SumCheckProver for IOPProverState {
                             tmp
                         })
                         .collect::<Vec<F>>();
-
-                    for i in products.iter() {
-                        *e += i
-                    }
+                    *e += products.par_iter().sum::<F>();
                 }
             }
         }
@@ -190,7 +185,7 @@ impl SumCheckProver for IOPProverState {
 
         // update prover's state to the partial evaluated polynomial
         self.poly.flattened_ml_extensions = flattened_ml_extensions
-            .iter()
+            .par_iter()
             .map(|x| Arc::new(x.clone()))
             .collect();