Range Check Felt252Packed27

stwo_cairo_prover/crates/prover/src/components/mod.rs

@@ -28,6 +28,7 @@ pub mod mul_opcode_small;
 pub mod mul_opcode_small_imm;
 pub mod range_check_builtin_bits_128;
 pub mod range_check_builtin_bits_96;
+pub mod range_check_felt_252_packed_27;
 pub mod range_check_vector;
 pub mod ret_opcode;
 pub mod utils;

stwo_cairo_prover/crates/prover/src/components/range_check_felt_252_packed_27/component.rs

@@ -0,0 +1,212 @@
+#![allow(non_camel_case_types)]
+#![allow(unused_imports)]
+use num_traits::{One, Zero};
+use serde::{Deserialize, Serialize};
+use stwo_cairo_serialize::CairoSerialize;
+use stwo_prover::constraint_framework::logup::{LogupAtRow, LookupElements};
+use stwo_prover::constraint_framework::{
+    EvalAtRow, FrameworkComponent, FrameworkEval, RelationEntry,
+};
+use stwo_prover::core::backend::simd::m31::LOG_N_LANES;
+use stwo_prover::core::channel::Channel;
+use stwo_prover::core::fields::m31::M31;
+use stwo_prover::core::fields::qm31::SecureField;
+use stwo_prover::core::fields::secure_column::SECURE_EXTENSION_DEGREE;
+use stwo_prover::core::pcs::TreeVec;
+
+use crate::cairo_air::preprocessed::PreProcessedColumn;
+use crate::relations;
+
+pub struct Eval {
+    pub claim: Claim,
+    pub range_check_felt_252_packed_27_lookup_elements: relations::RangeCheckFelt252Packed27,
+    pub range_check_18_lookup_elements: relations::RangeCheck_18,
+    pub range_check_9_9_lookup_elements: relations::RangeCheck_9_9,
+}
+
+#[derive(Copy, Clone, Serialize, Deserialize, CairoSerialize)]
+pub struct Claim {
+    pub log_size: u32,
+}
+impl Claim {
+    pub fn log_sizes(&self) -> TreeVec<Vec<u32>> {
+        let trace_log_sizes = vec![self.log_size; 19];
+        let interaction_log_sizes = vec![self.log_size; SECURE_EXTENSION_DEGREE * 8];
+        let preprocessed_log_sizes = vec![self.log_size];
+        TreeVec::new(vec![
+            preprocessed_log_sizes,
+            trace_log_sizes,
+            interaction_log_sizes,
+        ])
+    }
+
+    pub fn mix_into(&self, channel: &mut impl Channel) {
+        channel.mix_u64(self.log_size as u64);
+    }
+}
+
+#[derive(Copy, Clone, Serialize, Deserialize, CairoSerialize)]
+pub struct InteractionClaim {
+    pub claimed_sum: SecureField,
+}
+impl InteractionClaim {
+    pub fn mix_into(&self, channel: &mut impl Channel) {
+        channel.mix_felts(&[self.claimed_sum]);
+    }
+}
+
+pub type Component = FrameworkComponent<Eval>;
+
+impl FrameworkEval for Eval {
+    fn log_size(&self) -> u32 {
+        self.claim.log_size
+    }
+
+    fn max_constraint_log_degree_bound(&self) -> u32 {
+        self.log_size() + 1
+    }
+
+    #[allow(unused_parens)]
+    #[allow(clippy::double_parens)]
+    #[allow(non_snake_case)]
+    fn evaluate<E: EvalAtRow>(&self, mut eval: E) -> E {
+        let M31_262144 = E::F::from(M31::from(262144));
+        let M31_4194304 = E::F::from(M31::from(4194304));
+        let input_limb_0_col0 = eval.next_trace_mask();
+        let input_limb_1_col1 = eval.next_trace_mask();
+        let input_limb_2_col2 = eval.next_trace_mask();
+        let input_limb_3_col3 = eval.next_trace_mask();
+        let input_limb_4_col4 = eval.next_trace_mask();
+        let input_limb_5_col5 = eval.next_trace_mask();
+        let input_limb_6_col6 = eval.next_trace_mask();
+        let input_limb_7_col7 = eval.next_trace_mask();
+        let input_limb_8_col8 = eval.next_trace_mask();
+        let input_limb_9_col9 = eval.next_trace_mask();
+        let limb_0_high_part_col10 = eval.next_trace_mask();
+        let limb_1_low_part_col11 = eval.next_trace_mask();
+        let limb_2_high_part_col12 = eval.next_trace_mask();
+        let limb_3_low_part_col13 = eval.next_trace_mask();
+        let limb_4_high_part_col14 = eval.next_trace_mask();
+        let limb_5_low_part_col15 = eval.next_trace_mask();
+        let limb_6_high_part_col16 = eval.next_trace_mask();
+        let limb_7_low_part_col17 = eval.next_trace_mask();
+        let limb_8_high_part_col18 = eval.next_trace_mask();
+        let multiplicity = eval.next_trace_mask();
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_9_9_lookup_elements,
+            E::EF::one(),
+            &[
+                limb_0_high_part_col10.clone(),
+                limb_1_low_part_col11.clone(),
+            ],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_18_lookup_elements,
+            E::EF::one(),
+            &[(input_limb_0_col0.clone() - (limb_0_high_part_col10.clone() * M31_262144.clone()))],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_18_lookup_elements,
+            E::EF::one(),
+            &[((input_limb_1_col1.clone() - limb_1_low_part_col11.clone()) * M31_4194304.clone())],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_9_9_lookup_elements,
+            E::EF::one(),
+            &[
+                limb_2_high_part_col12.clone(),
+                limb_3_low_part_col13.clone(),
+            ],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_18_lookup_elements,
+            E::EF::one(),
+            &[(input_limb_2_col2.clone() - (limb_2_high_part_col12.clone() * M31_262144.clone()))],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_18_lookup_elements,
+            E::EF::one(),
+            &[((input_limb_3_col3.clone() - limb_3_low_part_col13.clone()) * M31_4194304.clone())],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_9_9_lookup_elements,
+            E::EF::one(),
+            &[
+                limb_4_high_part_col14.clone(),
+                limb_5_low_part_col15.clone(),
+            ],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_18_lookup_elements,
+            E::EF::one(),
+            &[(input_limb_4_col4.clone() - (limb_4_high_part_col14.clone() * M31_262144.clone()))],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_18_lookup_elements,
+            E::EF::one(),
+            &[((input_limb_5_col5.clone() - limb_5_low_part_col15.clone()) * M31_4194304.clone())],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_9_9_lookup_elements,
+            E::EF::one(),
+            &[
+                limb_6_high_part_col16.clone(),
+                limb_7_low_part_col17.clone(),
+            ],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_18_lookup_elements,
+            E::EF::one(),
+            &[(input_limb_6_col6.clone() - (limb_6_high_part_col16.clone() * M31_262144.clone()))],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_18_lookup_elements,
+            E::EF::one(),
+            &[((input_limb_7_col7.clone() - limb_7_low_part_col17.clone()) * M31_4194304.clone())],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_9_9_lookup_elements,
+            E::EF::one(),
+            &[limb_8_high_part_col18.clone(), input_limb_9_col9.clone()],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_18_lookup_elements,
+            E::EF::one(),
+            &[(input_limb_8_col8.clone() - (limb_8_high_part_col18.clone() * M31_262144.clone()))],
+        ));
+
+        eval.add_to_relation(RelationEntry::new(
+            &self.range_check_felt_252_packed_27_lookup_elements,
+            -E::EF::from(multiplicity),
+            &[
+                input_limb_0_col0.clone(),
+                input_limb_1_col1.clone(),
+                input_limb_2_col2.clone(),
+                input_limb_3_col3.clone(),
+                input_limb_4_col4.clone(),
+                input_limb_5_col5.clone(),
+                input_limb_6_col6.clone(),
+                input_limb_7_col7.clone(),
+                input_limb_8_col8.clone(),
+                input_limb_9_col9.clone(),
+            ],
+        ));
+
+        eval.finalize_logup_in_pairs();
+        eval
+    }
+}

stwo_cairo_prover/crates/prover/src/components/range_check_felt_252_packed_27/mod.rs

@@ -0,0 +1,5 @@
+pub mod component;
+pub mod prover;
+
+pub use component::{Claim, Component, Eval, InteractionClaim};
+pub use prover::{ClaimGenerator, InputType, InteractionClaimGenerator};