Add packed_at() to PreProcessed Column Enum

starkware-libs · Jan 16, 2025 · 1b92ae4 · 1b92ae4
1 parent e9f0546
commit 1b92ae4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/stwo_cairo_prover/crates/prover/src/cairo_air/preprocessed.rs b/stwo_cairo_prover/crates/prover/src/cairo_air/preprocessed.rs
@@ -53,6 +53,13 @@ impl PreProcessedColumn {
             PreProcessedColumn::Seq(column) => column.gen_column_simd(),
         }
     }
+
+    pub fn packed_at(&self, vec_row: usize) -> PackedM31 {
+        match self {
+            PreProcessedColumn::IsFirst(column) => column.packed_at(vec_row),
+            PreProcessedColumn::Seq(column) => column.packed_at(vec_row),
+        }
+    }
 }
 
 /// Returns column info for the preprocessed trace.