diff --git a/evm/src/prover.rs b/evm/src/prover.rs index 0ebebf4922..c607035f9e 100644 --- a/evm/src/prover.rs +++ b/evm/src/prover.rs @@ -499,25 +499,28 @@ where // When opening the `Z`s polys at the "next" point, need to look at the point `next_step` steps away. let next_step = 1 << quotient_degree_bits; + let powers_vec = SmallPowers::::new(F::coset_shift().to_noncanonical_u64() as u32) + .take(degree << quotient_degree_bits) + .collect_vec(); // Evaluation of the first Lagrange polynomial on the LDE domain, with custom lde_onto_coset. let lagrange_first = PolynomialValues::::selector(degree, 0); let coeffs = ifft(lagrange_first).lde(quotient_degree_bits); - let modified_poly: PolynomialCoeffs = - SmallPowers::::new(F::coset_shift().to_noncanonical_u64() as u32) - .zip(&coeffs.coeffs) - .map(|(r, &c)| r * c) - .collect::>() - .into(); + let modified_poly: PolynomialCoeffs = powers_vec + .iter() + .zip(&coeffs.coeffs) + .map(|(&r, &c)| r * c) + .collect::>() + .into(); let lagrange_first = modified_poly.fft_with_options(None, None); // Evaluation of the last Lagrange polynomial on the LDE domain, with custom lde_onto_coset. let lagrange_last = PolynomialValues::::selector(degree, degree - 1); let coeffs = ifft(lagrange_last).lde(quotient_degree_bits); - let modified_poly: PolynomialCoeffs = - SmallPowers::::new(F::coset_shift().to_noncanonical_u64() as u32) - .zip(&coeffs.coeffs) - .map(|(r, &c)| r * c) - .collect::>() - .into(); + let modified_poly: PolynomialCoeffs = powers_vec + .into_iter() + .zip(&coeffs.coeffs) + .map(|(r, &c)| r * c) + .collect::>() + .into(); let lagrange_last = modified_poly.fft_with_options(None, None); let z_h_on_coset = ZeroPolyOnCoset::::new(degree_bits, quotient_degree_bits); diff --git a/starky/src/prover.rs b/starky/src/prover.rs index 3603780ce6..d814b44b62 100644 --- a/starky/src/prover.rs +++ b/starky/src/prover.rs @@ -228,25 +228,28 @@ where // When opening the `Z`s polys at the "next" point, need to look at the point `next_step` steps away. let next_step = 1 << quotient_degree_bits; + let powers_vec = SmallPowers::::new(F::coset_shift().to_noncanonical_u64() as u32) + .take(degree << quotient_degree_bits) + .collect_vec(); // Evaluation of the first Lagrange polynomial on the LDE domain, with custom lde_onto_coset. let lagrange_first = PolynomialValues::::selector(degree, 0); let coeffs = ifft(lagrange_first).lde(quotient_degree_bits); - let modified_poly: PolynomialCoeffs = - SmallPowers::::new(F::coset_shift().to_noncanonical_u64() as u32) - .zip(&coeffs.coeffs) - .map(|(r, &c)| r * c) - .collect::>() - .into(); + let modified_poly: PolynomialCoeffs = powers_vec + .iter() + .zip(&coeffs.coeffs) + .map(|(&r, &c)| r * c) + .collect::>() + .into(); let lagrange_first = modified_poly.fft_with_options(None, None); // Evaluation of the last Lagrange polynomial on the LDE domain, with custom lde_onto_coset. let lagrange_last = PolynomialValues::::selector(degree, degree - 1); let coeffs = ifft(lagrange_last).lde(quotient_degree_bits); - let modified_poly: PolynomialCoeffs = - SmallPowers::::new(F::coset_shift().to_noncanonical_u64() as u32) - .zip(&coeffs.coeffs) - .map(|(r, &c)| r * c) - .collect::>() - .into(); + let modified_poly: PolynomialCoeffs = powers_vec + .into_iter() + .zip(&coeffs.coeffs) + .map(|(r, &c)| r * c) + .collect::>() + .into(); let lagrange_last = modified_poly.fft_with_options(None, None); let z_h_on_coset = ZeroPolyOnCoset::::new(degree_bits, quotient_degree_bits);