From f7ffdf5ce2633887182a16d60ab31bdec6484bb2 Mon Sep 17 00:00:00 2001
From: Dan Dore <dorebell@gmail.com>
Date: Fri, 14 Jun 2024 03:07:40 -0700
Subject: [PATCH] remove shift argument from compute_lde_batch

---
 commit/src/pcs.rs       | 10 +++-------
 fri/src/two_adic_pcs.rs | 11 +++--------
 2 files changed, 6 insertions(+), 15 deletions(-)

diff --git a/commit/src/pcs.rs b/commit/src/pcs.rs
index 8d2b95f3c..7daa08b87 100644
--- 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
index 3137898c6..6ae54bc95 100644
--- 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)