feat: use zip_eq for Polynomial add/sub (privacy-scaling-explorat…

…ions#12)
zemse · May 22, 2023 · e24fa3c · e24fa3c
1 parent b38400a
commit e24fa3c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/snark-verifier/src/util/poly.rs b/snark-verifier/src/util/poly.rs
@@ -1,6 +1,7 @@
 //! Polynomial.
 
 use crate::util::{arithmetic::Field, parallelize};
+use itertools::Itertools;
 use rand::Rng;
 use std::{
     iter::{self, Sum},
@@ -89,7 +90,7 @@ impl<'a, F: Field> Add<&'a Polynomial<F>> for Polynomial<F> {
 
     fn add(mut self, rhs: &'a Polynomial<F>) -> Polynomial<F> {
         parallelize(&mut self.0, |(lhs, start)| {
-            for (lhs, rhs) in lhs.iter_mut().zip(rhs.0[start..].iter()) {
+            for (lhs, rhs) in lhs.iter_mut().zip_eq(rhs.0[start..].iter()) {
                 *lhs += *rhs;
             }
         });
@@ -102,7 +103,7 @@ impl<'a, F: Field> Sub<&'a Polynomial<F>> for Polynomial<F> {
 
     fn sub(mut self, rhs: &'a Polynomial<F>) -> Polynomial<F> {
         parallelize(&mut self.0, |(lhs, start)| {
-            for (lhs, rhs) in lhs.iter_mut().zip(rhs.0[start..].iter()) {
+            for (lhs, rhs) in lhs.iter_mut().zip_eq(rhs.0[start..].iter()) {
                 *lhs -= *rhs;
             }
         });