From 1b92ae453cdc4daff69fc05f9931d6ff97b179c5 Mon Sep 17 00:00:00 2001
From: Gali Michlevich <gali@starkware.co>
Date: Thu, 16 Jan 2025 15:59:52 +0200
Subject: [PATCH] Add packed_at() to PreProcessed Column Enum

---
 .../crates/prover/src/cairo_air/preprocessed.rs            | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/stwo_cairo_prover/crates/prover/src/cairo_air/preprocessed.rs b/stwo_cairo_prover/crates/prover/src/cairo_air/preprocessed.rs
index 049c8dea..fc76d56a 100644
--- 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.