remove shift argument from compute_lde_batch

valida-xyz · Jun 14, 2024 · f7ffdf5 · f7ffdf5
1 parent 1e4c51c
commit f7ffdf5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 15 deletions.
diff --git a/commit/src/pcs.rs b/commit/src/pcs.rs
@@ -88,14 +88,10 @@ where
         'a: 'b;
 
     // Compute the (shifted) low-degree extensions only without computing the commitment.
-    fn compute_ldes_batches(
-        &self,
-        polynomials: Vec<In>,
-        coset_shifts: &[Val],
-    ) -> Vec<Self::LdeOwned>;
+    fn compute_ldes_batches(&self, polynomials: Vec<In>) -> Vec<Self::LdeOwned>;
 
-    fn compute_lde_batch(&self, polynomials: In, coset_shift: Val) -> Vec<Self::LdeOwned> {
-        self.compute_ldes_batches(vec![polynomials], &[coset_shift])
+    fn compute_lde_batch(&self, polynomials: In) -> Vec<Self::LdeOwned> {
+        self.compute_ldes_batches(vec![polynomials])
     }
 
     // Commit to polys that are already defined over a coset.

diff --git a/fri/src/two_adic_pcs.rs b/fri/src/two_adic_pcs.rs
@@ -163,17 +163,12 @@ where
             .collect()
     }
 
-    fn compute_ldes_batches(
-        &self,
-        polynomials: Vec<In>,
-        coset_shifts: &[C::Val],
-    ) -> Vec<Self::LdeOwned> {
+    fn compute_ldes_batches(&self, polynomials: Vec<In>) -> Vec<Self::LdeOwned> {
         info_span!("compute all coset LDEs").in_scope(|| {
             polynomials
                 .par_iter()
-                .zip_eq(coset_shifts)
-                .map(|(poly, coset_shift)| {
-                    let shift = C::Val::generator() / *coset_shift;
+                .map(|poly| {
+                    let shift = C::Val::generator();
                     let input = ((*poly).clone()).to_row_major_matrix();
                     self.dft
                         .coset_lde_batch(input, self.fri.log_blowup, shift)