jnz gen

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

@@ -0,0 +1,246 @@
+#![allow(non_camel_case_types)]
+#![allow(unused_imports)]
+use num_traits::{One, Zero};
+use serde::{Deserialize, Serialize};
+use stwo_prover::constraint_framework::logup::{LogupAtRow, LogupSums, 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::relations;
+
+pub struct Eval {
+    pub claim: Claim,
+    pub memoryaddresstoid_lookup_elements: relations::MemoryAddressToId,
+    pub memoryidtobig_lookup_elements: relations::MemoryIdToBig,
+    pub opcodes_lookup_elements: relations::Opcodes,
+    pub verifyinstruction_lookup_elements: relations::VerifyInstruction,
+}
+
+#[derive(Copy, Clone, Serialize, Deserialize)]
+pub struct Claim {
+    pub n_calls: usize,
+}
+impl Claim {
+    pub fn log_sizes(&self) -> TreeVec<Vec<u32>> {
+        let log_size = std::cmp::max(self.n_calls.next_power_of_two().ilog2(), LOG_N_LANES);
+        let trace_log_sizes = vec![log_size; 34];
+        let interaction_log_sizes = vec![log_size; SECURE_EXTENSION_DEGREE * 5];
+        let preprocessed_log_sizes = vec![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.n_calls as u64);
+    }
+}
+
+#[derive(Copy, Clone, Serialize, Deserialize)]
+pub struct InteractionClaim {
+    pub logup_sums: LogupSums,
+}
+impl InteractionClaim {
+    pub fn mix_into(&self, channel: &mut impl Channel) {
+        let (total_sum, claimed_sum) = self.logup_sums;
+        channel.mix_felts(&[total_sum]);
+        if let Some(claimed_sum) = claimed_sum {
+            channel.mix_felts(&[claimed_sum.0]);
+            channel.mix_u64(claimed_sum.1 as u64);
+        }
+    }
+}
+
+pub type Component = FrameworkComponent<Eval>;
+
+impl FrameworkEval for Eval {
+    fn log_size(&self) -> u32 {
+        std::cmp::max(self.claim.n_calls.next_power_of_two().ilog2(), LOG_N_LANES)
+    }
+
+    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_0 = E::F::from(M31::from(0));
+        let M31_1 = E::F::from(M31::from(1));
+        let M31_2 = E::F::from(M31::from(2));
+        let M31_32767 = E::F::from(M31::from(32767));
+        let M31_32768 = E::F::from(M31::from(32768));
+        let M31_32769 = E::F::from(M31::from(32769));
+        let input_pc_col0 = eval.next_trace_mask();
+        let input_ap_col1 = eval.next_trace_mask();
+        let input_fp_col2 = eval.next_trace_mask();
+        let offset0_col3 = eval.next_trace_mask();
+        let ap_update_add_1_col4 = eval.next_trace_mask();
+        let dst_id_col5 = eval.next_trace_mask();
+        let dst_limb_0_col6 = eval.next_trace_mask();
+        let dst_limb_1_col7 = eval.next_trace_mask();
+        let dst_limb_2_col8 = eval.next_trace_mask();
+        let dst_limb_3_col9 = eval.next_trace_mask();
+        let dst_limb_4_col10 = eval.next_trace_mask();
+        let dst_limb_5_col11 = eval.next_trace_mask();
+        let dst_limb_6_col12 = eval.next_trace_mask();
+        let dst_limb_7_col13 = eval.next_trace_mask();
+        let dst_limb_8_col14 = eval.next_trace_mask();
+        let dst_limb_9_col15 = eval.next_trace_mask();
+        let dst_limb_10_col16 = eval.next_trace_mask();
+        let dst_limb_11_col17 = eval.next_trace_mask();
+        let dst_limb_12_col18 = eval.next_trace_mask();
+        let dst_limb_13_col19 = eval.next_trace_mask();
+        let dst_limb_14_col20 = eval.next_trace_mask();
+        let dst_limb_15_col21 = eval.next_trace_mask();
+        let dst_limb_16_col22 = eval.next_trace_mask();
+        let dst_limb_17_col23 = eval.next_trace_mask();
+        let dst_limb_18_col24 = eval.next_trace_mask();
+        let dst_limb_19_col25 = eval.next_trace_mask();
+        let dst_limb_20_col26 = eval.next_trace_mask();
+        let dst_limb_21_col27 = eval.next_trace_mask();
+        let dst_limb_22_col28 = eval.next_trace_mask();
+        let dst_limb_23_col29 = eval.next_trace_mask();
+        let dst_limb_24_col30 = eval.next_trace_mask();
+        let dst_limb_25_col31 = eval.next_trace_mask();
+        let dst_limb_26_col32 = eval.next_trace_mask();
+        let dst_limb_27_col33 = eval.next_trace_mask();
+
+        // decode_instruction_d2b4cd588a3e2a7b.
+
+        eval.add_to_relation(&[RelationEntry::new(
+            &self.verifyinstruction_lookup_elements,
+            E::EF::one(),
+            &[
+                input_pc_col0.clone(),
+                offset0_col3.clone(),
+                M31_32767.clone(),
+                M31_32769.clone(),
+                M31_0.clone(),
+                M31_1.clone(),
+                M31_1.clone(),
+                M31_0.clone(),
+                M31_0.clone(),
+                M31_0.clone(),
+                M31_0.clone(),
+                M31_0.clone(),
+                M31_0.clone(),
+                M31_1.clone(),
+                M31_0.clone(),
+                ap_update_add_1_col4.clone(),
+            ],
+        )]);
+
+        // read_positive_num_bits_252.
+
+        eval.add_to_relation(&[RelationEntry::new(
+            &self.memoryaddresstoid_lookup_elements,
+            E::EF::one(),
+            &[
+                (input_ap_col1.clone() + (offset0_col3.clone() - M31_32768.clone())),
+                dst_id_col5.clone(),
+            ],
+        )]);
+
+        eval.add_to_relation(&[RelationEntry::new(
+            &self.memoryidtobig_lookup_elements,
+            E::EF::one(),
+            &[
+                dst_id_col5.clone(),
+                dst_limb_0_col6.clone(),
+                dst_limb_1_col7.clone(),
+                dst_limb_2_col8.clone(),
+                dst_limb_3_col9.clone(),
+                dst_limb_4_col10.clone(),
+                dst_limb_5_col11.clone(),
+                dst_limb_6_col12.clone(),
+                dst_limb_7_col13.clone(),
+                dst_limb_8_col14.clone(),
+                dst_limb_9_col15.clone(),
+                dst_limb_10_col16.clone(),
+                dst_limb_11_col17.clone(),
+                dst_limb_12_col18.clone(),
+                dst_limb_13_col19.clone(),
+                dst_limb_14_col20.clone(),
+                dst_limb_15_col21.clone(),
+                dst_limb_16_col22.clone(),
+                dst_limb_17_col23.clone(),
+                dst_limb_18_col24.clone(),
+                dst_limb_19_col25.clone(),
+                dst_limb_20_col26.clone(),
+                dst_limb_21_col27.clone(),
+                dst_limb_22_col28.clone(),
+                dst_limb_23_col29.clone(),
+                dst_limb_24_col30.clone(),
+                dst_limb_25_col31.clone(),
+                dst_limb_26_col32.clone(),
+                dst_limb_27_col33.clone(),
+            ],
+        )]);
+
+        // dst equals 0.
+        eval.add_constraint(
+            ((((((((((((((((((((((((((((M31_0.clone()
+                + dst_limb_0_col6.clone())
+                + dst_limb_1_col7.clone())
+                + dst_limb_2_col8.clone())
+                + dst_limb_3_col9.clone())
+                + dst_limb_4_col10.clone())
+                + dst_limb_5_col11.clone())
+                + dst_limb_6_col12.clone())
+                + dst_limb_7_col13.clone())
+                + dst_limb_8_col14.clone())
+                + dst_limb_9_col15.clone())
+                + dst_limb_10_col16.clone())
+                + dst_limb_11_col17.clone())
+                + dst_limb_12_col18.clone())
+                + dst_limb_13_col19.clone())
+                + dst_limb_14_col20.clone())
+                + dst_limb_15_col21.clone())
+                + dst_limb_16_col22.clone())
+                + dst_limb_17_col23.clone())
+                + dst_limb_18_col24.clone())
+                + dst_limb_19_col25.clone())
+                + dst_limb_20_col26.clone())
+                + dst_limb_21_col27.clone())
+                + dst_limb_22_col28.clone())
+                + dst_limb_23_col29.clone())
+                + dst_limb_24_col30.clone())
+                + dst_limb_25_col31.clone())
+                + dst_limb_26_col32.clone())
+                + dst_limb_27_col33.clone()),
+        );
+        eval.add_to_relation(&[RelationEntry::new(
+            &self.opcodes_lookup_elements,
+            E::EF::one(),
+            &[
+                input_pc_col0.clone(),
+                input_ap_col1.clone(),
+                input_fp_col2.clone(),
+            ],
+        )]);
+
+        eval.add_to_relation(&[RelationEntry::new(
+            &self.opcodes_lookup_elements,
+            -E::EF::one(),
+            &[
+                (input_pc_col0.clone() + M31_2.clone()),
+                (input_ap_col1.clone() + ap_update_add_1_col4.clone()),
+                input_fp_col2.clone(),
+            ],
+        )]);
+
+        eval.finalize_logup();
+        eval
+    }
+}

stwo_cairo_prover/crates/prover/src/components/jnz_opcode_is_taken_f_dst_base_fp_f/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};