Add interaction_element_ids to Component trait.

starkware-libs · May 16, 2024 · a8fa06c · a8fa06c
1 parent c2d33be
commit a8fa06c
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/crates/prover/src/core/air/mod.rs b/crates/prover/src/core/air/mod.rs
@@ -41,6 +41,9 @@ pub trait Component {
         point: CirclePoint<SecureField>,
     ) -> ColumnVec<Vec<CirclePoint<SecureField>>>;
 
+    // Returns the ids of the interaction elements used by the component.
+    fn interaction_element_ids(&self) -> Vec<String>;
+
     /// Evaluates the constraint quotients combination of the component, given the mask values.
     fn evaluate_constraint_quotients_at_point(
         &self,

diff --git a/crates/prover/src/core/prover/mod.rs b/crates/prover/src/core/prover/mod.rs
@@ -321,6 +321,10 @@ mod tests {
             vec![vec![point]]
         }
 
+        fn interaction_element_ids(&self) -> Vec<String> {
+            vec![]
+        }
+
         fn evaluate_constraint_quotients_at_point(
             &self,
             _point: CirclePoint<SecureField>,

diff --git a/crates/prover/src/examples/fibonacci/component.rs b/crates/prover/src/examples/fibonacci/component.rs
@@ -94,6 +94,10 @@ impl Component for FibonacciComponent {
         )
     }
 
+    fn interaction_element_ids(&self) -> Vec<String> {
+        vec![]
+    }
+
     fn evaluate_constraint_quotients_at_point(
         &self,
         point: CirclePoint<SecureField>,

diff --git a/crates/prover/src/examples/wide_fibonacci/component.rs b/crates/prover/src/examples/wide_fibonacci/component.rs
@@ -67,6 +67,13 @@ impl Component for WideFibComponent {
         fixed_mask_points(&vec![vec![0_usize]; self.n_columns()], point)
     }
 
+    fn interaction_element_ids(&self) -> Vec<String> {
+        vec![
+            "wide_fibonacci_alpha".to_string(),
+            "wide_fibonacci_z".to_string(),
+        ]
+    }
+
     fn evaluate_constraint_quotients_at_point(
         &self,
         point: CirclePoint<SecureField>,