fix fri oracle index

brevis-network · Aug 8, 2024 · 5e37f69 · 5e37f69
1 parent 2f7fa7b
commit 5e37f69
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/starky/src/stark.rs b/starky/src/stark.rs
@@ -185,8 +185,9 @@ pub trait Stark<F: RichField + Extendable<D>, const D: usize>: Sync {
         let mut batches = vec![zeta_batch, zeta_next_batch];
 
         if self.requires_ctls() {
+            let oracle_index = if self.use_phase2() {2} else {1};
             let ctl_zs_info = FriPolynomialInfo::from_range(
-                2, // auxiliary oracle index
+                oracle_index, // auxiliary oracle index
                 num_lookup_columns + num_ctl_helpers..num_auxiliary_polys,
             );
             let ctl_first_batch = FriBatchInfo {
@@ -272,8 +273,9 @@ pub trait Stark<F: RichField + Extendable<D>, const D: usize>: Sync {
         let mut batches = vec![zeta_batch, zeta_next_batch];
 
         if self.requires_ctls() {
+            let oracle_index = if self.use_phase2() {2} else {1};
             let ctl_zs_info = FriPolynomialInfo::from_range(
-                2, // auxiliary oracle index
+                oracle_index, // auxiliary oracle index
                 num_lookup_columns + num_ctl_helper_polys..num_auxiliary_polys,
             );
             let ctl_first_batch = FriBatchInfoTarget {