feat: commit the preprocessed trace for the memory component

kkrt-labs · Nov 27, 2024 · 73f09a6 · 73f09a6
1 parent 4ac232a
commit 73f09a6
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/crates/brainfuck_prover/src/brainfuck_air/mod.rs b/crates/brainfuck_prover/src/brainfuck_air/mod.rs
@@ -7,7 +7,10 @@ use crate::components::{
 };
 use brainfuck_vm::machine::Machine;
 use stwo_prover::{
-    constraint_framework::{INTERACTION_TRACE_IDX, ORIGINAL_TRACE_IDX, PREPROCESSED_TRACE_IDX},
+    constraint_framework::{
+        preprocessed_columns::gen_is_first, INTERACTION_TRACE_IDX, ORIGINAL_TRACE_IDX,
+        PREPROCESSED_TRACE_IDX,
+    },
     core::{
         air::{Component, ComponentProver},
         backend::simd::SimdBackend,
@@ -145,7 +148,10 @@ pub fn prove_brainfuck(
     // └───────────────────────────────────────────────┘
 
     // Generate constant columns (e.g. is_first)
-    let tree_builder = commitment_scheme.tree_builder();
+    let mut tree_builder = commitment_scheme.tree_builder();
+
+    let memory_preprocessed_trace = gen_is_first::<SimdBackend>(LOG_MAX_ROWS);
+    tree_builder.extend_evals([memory_preprocessed_trace]);
     tree_builder.commit(channel);
 
     // ┌───────────────────────────────────────┐