-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
64a5bbb
commit 7d5ed9e
Showing
16 changed files
with
5,312 additions
and
4 deletions.
There are no files selected for viewing
344 changes: 341 additions & 3 deletions
344
stwo_cairo_prover/crates/prover/src/cairo_air/opcodes_air.rs
Large diffs are not rendered by default.
Oops, something went wrong.
640 changes: 640 additions & 0 deletions
640
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_f_is_imm_f/component.rs
Large diffs are not rendered by default.
Oops, something went wrong.
5 changes: 5 additions & 0 deletions
5
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_f_is_imm_f/mod.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
pub mod component; | ||
pub mod prover; | ||
|
||
pub use component::{Claim, Component, Eval, InteractionClaim}; | ||
pub use prover::{ClaimGenerator, InputType, InteractionClaimGenerator}; |
814 changes: 814 additions & 0 deletions
814
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_f_is_imm_f/prover.rs
Large diffs are not rendered by default.
Oops, something went wrong.
632 changes: 632 additions & 0 deletions
632
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_f_is_imm_t/component.rs
Large diffs are not rendered by default.
Oops, something went wrong.
5 changes: 5 additions & 0 deletions
5
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_f_is_imm_t/mod.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
pub mod component; | ||
pub mod prover; | ||
|
||
pub use component::{Claim, Component, Eval, InteractionClaim}; | ||
pub use prover::{ClaimGenerator, InputType, InteractionClaimGenerator}; |
776 changes: 776 additions & 0 deletions
776
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_f_is_imm_t/prover.rs
Large diffs are not rendered by default.
Oops, something went wrong.
376 changes: 376 additions & 0 deletions
376
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_t_is_imm_f/component.rs
Large diffs are not rendered by default.
Oops, something went wrong.
5 changes: 5 additions & 0 deletions
5
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_t_is_imm_f/mod.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
pub mod component; | ||
pub mod prover; | ||
|
||
pub use component::{Claim, Component, Eval, InteractionClaim}; | ||
pub use prover::{ClaimGenerator, InputType, InteractionClaimGenerator}; |
688 changes: 688 additions & 0 deletions
688
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_t_is_imm_f/prover.rs
Large diffs are not rendered by default.
Oops, something went wrong.
368 changes: 368 additions & 0 deletions
368
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_t_is_imm_t/component.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,368 @@ | ||
#![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; 26]; | ||
let interaction_log_sizes = vec![log_size; SECURE_EXTENSION_DEGREE * 9]; | ||
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_134217728 = E::F::from(M31::from(134217728)); | ||
let M31_136 = E::F::from(M31::from(136)); | ||
let M31_2 = E::F::from(M31::from(2)); | ||
let M31_256 = E::F::from(M31::from(256)); | ||
let M31_262144 = E::F::from(M31::from(262144)); | ||
let M31_32768 = E::F::from(M31::from(32768)); | ||
let M31_32769 = E::F::from(M31::from(32769)); | ||
let M31_511 = E::F::from(M31::from(511)); | ||
let M31_512 = E::F::from(M31::from(512)); | ||
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 offset1_col4 = eval.next_trace_mask(); | ||
let dst_base_fp_col5 = eval.next_trace_mask(); | ||
let op0_base_fp_col6 = eval.next_trace_mask(); | ||
let ap_update_add_1_col7 = eval.next_trace_mask(); | ||
let dst_id_col8 = eval.next_trace_mask(); | ||
let msb_col9 = eval.next_trace_mask(); | ||
let mid_limbs_set_col10 = eval.next_trace_mask(); | ||
let dst_limb_0_col11 = eval.next_trace_mask(); | ||
let dst_limb_1_col12 = eval.next_trace_mask(); | ||
let dst_limb_2_col13 = eval.next_trace_mask(); | ||
let op0_id_col14 = eval.next_trace_mask(); | ||
let msb_col15 = eval.next_trace_mask(); | ||
let mid_limbs_set_col16 = eval.next_trace_mask(); | ||
let op0_limb_0_col17 = eval.next_trace_mask(); | ||
let op0_limb_1_col18 = eval.next_trace_mask(); | ||
let op0_limb_2_col19 = eval.next_trace_mask(); | ||
let op1_id_col20 = eval.next_trace_mask(); | ||
let msb_col21 = eval.next_trace_mask(); | ||
let mid_limbs_set_col22 = eval.next_trace_mask(); | ||
let op1_limb_0_col23 = eval.next_trace_mask(); | ||
let op1_limb_1_col24 = eval.next_trace_mask(); | ||
let op1_limb_2_col25 = eval.next_trace_mask(); | ||
|
||
// decode_instruction_9aed6a790187299c. | ||
|
||
eval.add_to_relation(&[RelationEntry::new( | ||
&self.verifyinstruction_lookup_elements, | ||
E::EF::one(), | ||
&[ | ||
input_pc_col0.clone(), | ||
offset0_col3.clone(), | ||
offset1_col4.clone(), | ||
M31_32769.clone(), | ||
dst_base_fp_col5.clone(), | ||
op0_base_fp_col6.clone(), | ||
M31_1.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_1.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
ap_update_add_1_col7.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_1.clone(), | ||
], | ||
)]); | ||
|
||
// read_small. | ||
|
||
eval.add_to_relation(&[RelationEntry::new( | ||
&self.memoryaddresstoid_lookup_elements, | ||
E::EF::one(), | ||
&[ | ||
(((dst_base_fp_col5.clone() * input_fp_col2.clone()) | ||
+ ((M31_1.clone() - dst_base_fp_col5.clone()) * input_ap_col1.clone())) | ||
+ (offset0_col3.clone() - M31_32768.clone())), | ||
dst_id_col8.clone(), | ||
], | ||
)]); | ||
|
||
// cond_decode_small_sign. | ||
|
||
// msb is a bit. | ||
eval.add_constraint((msb_col9.clone() * (msb_col9.clone() - M31_1.clone()))); | ||
// mid_limbs_set is a bit. | ||
eval.add_constraint( | ||
(mid_limbs_set_col10.clone() * (mid_limbs_set_col10.clone() - M31_1.clone())), | ||
); | ||
// Cannot have msb equals 0 and mid_limbs_set equals 1. | ||
eval.add_constraint( | ||
((M31_1.clone() * mid_limbs_set_col10.clone()) * (msb_col9.clone() - M31_1.clone())), | ||
); | ||
|
||
eval.add_to_relation(&[RelationEntry::new( | ||
&self.memoryidtobig_lookup_elements, | ||
E::EF::one(), | ||
&[ | ||
dst_id_col8.clone(), | ||
dst_limb_0_col11.clone(), | ||
dst_limb_1_col12.clone(), | ||
dst_limb_2_col13.clone(), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
(mid_limbs_set_col10.clone() * M31_511.clone()), | ||
((M31_136.clone() * msb_col9.clone()) - mid_limbs_set_col10.clone()), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
(msb_col9.clone() * M31_256.clone()), | ||
], | ||
)]); | ||
|
||
// read_small. | ||
|
||
eval.add_to_relation(&[RelationEntry::new( | ||
&self.memoryaddresstoid_lookup_elements, | ||
E::EF::one(), | ||
&[ | ||
(((op0_base_fp_col6.clone() * input_fp_col2.clone()) | ||
+ ((M31_1.clone() - op0_base_fp_col6.clone()) * input_ap_col1.clone())) | ||
+ (offset1_col4.clone() - M31_32768.clone())), | ||
op0_id_col14.clone(), | ||
], | ||
)]); | ||
|
||
// cond_decode_small_sign. | ||
|
||
// msb is a bit. | ||
eval.add_constraint((msb_col15.clone() * (msb_col15.clone() - M31_1.clone()))); | ||
// mid_limbs_set is a bit. | ||
eval.add_constraint( | ||
(mid_limbs_set_col16.clone() * (mid_limbs_set_col16.clone() - M31_1.clone())), | ||
); | ||
// Cannot have msb equals 0 and mid_limbs_set equals 1. | ||
eval.add_constraint( | ||
((M31_1.clone() * mid_limbs_set_col16.clone()) * (msb_col15.clone() - M31_1.clone())), | ||
); | ||
|
||
eval.add_to_relation(&[RelationEntry::new( | ||
&self.memoryidtobig_lookup_elements, | ||
E::EF::one(), | ||
&[ | ||
op0_id_col14.clone(), | ||
op0_limb_0_col17.clone(), | ||
op0_limb_1_col18.clone(), | ||
op0_limb_2_col19.clone(), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
(mid_limbs_set_col16.clone() * M31_511.clone()), | ||
((M31_136.clone() * msb_col15.clone()) - mid_limbs_set_col16.clone()), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
(msb_col15.clone() * M31_256.clone()), | ||
], | ||
)]); | ||
|
||
// read_small. | ||
|
||
eval.add_to_relation(&[RelationEntry::new( | ||
&self.memoryaddresstoid_lookup_elements, | ||
E::EF::one(), | ||
&[ | ||
(input_pc_col0.clone() + M31_1.clone()), | ||
op1_id_col20.clone(), | ||
], | ||
)]); | ||
|
||
// cond_decode_small_sign. | ||
|
||
// msb is a bit. | ||
eval.add_constraint((msb_col21.clone() * (msb_col21.clone() - M31_1.clone()))); | ||
// mid_limbs_set is a bit. | ||
eval.add_constraint( | ||
(mid_limbs_set_col22.clone() * (mid_limbs_set_col22.clone() - M31_1.clone())), | ||
); | ||
// Cannot have msb equals 0 and mid_limbs_set equals 1. | ||
eval.add_constraint( | ||
((M31_1.clone() * mid_limbs_set_col22.clone()) * (msb_col21.clone() - M31_1.clone())), | ||
); | ||
|
||
eval.add_to_relation(&[RelationEntry::new( | ||
&self.memoryidtobig_lookup_elements, | ||
E::EF::one(), | ||
&[ | ||
op1_id_col20.clone(), | ||
op1_limb_0_col23.clone(), | ||
op1_limb_1_col24.clone(), | ||
op1_limb_2_col25.clone(), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
(mid_limbs_set_col22.clone() * M31_511.clone()), | ||
((M31_136.clone() * msb_col21.clone()) - mid_limbs_set_col22.clone()), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
M31_0.clone(), | ||
(msb_col21.clone() * M31_256.clone()), | ||
], | ||
)]); | ||
|
||
// dst equals op0 + op1. | ||
eval.add_constraint( | ||
(((((dst_limb_0_col11.clone() + (dst_limb_1_col12.clone() * M31_512.clone())) | ||
+ (dst_limb_2_col13.clone() * M31_262144.clone())) | ||
- msb_col9.clone()) | ||
- (M31_134217728.clone() * mid_limbs_set_col10.clone())) | ||
- (((((op0_limb_0_col17.clone() | ||
+ (op0_limb_1_col18.clone() * M31_512.clone())) | ||
+ (op0_limb_2_col19.clone() * M31_262144.clone())) | ||
- msb_col15.clone()) | ||
- (M31_134217728.clone() * mid_limbs_set_col16.clone())) | ||
+ ((((op1_limb_0_col23.clone() | ||
+ (op1_limb_1_col24.clone() * M31_512.clone())) | ||
+ (op1_limb_2_col25.clone() * M31_262144.clone())) | ||
- msb_col21.clone()) | ||
- (M31_134217728.clone() * mid_limbs_set_col22.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_col7.clone()), | ||
input_fp_col2.clone(), | ||
], | ||
)]); | ||
|
||
eval.finalize_logup(); | ||
eval | ||
} | ||
} |
5 changes: 5 additions & 0 deletions
5
stwo_cairo_prover/crates/prover/src/components/add_opcode_is_small_t_is_imm_t/mod.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
pub mod component; | ||
pub mod prover; | ||
|
||
pub use component::{Claim, Component, Eval, InteractionClaim}; | ||
pub use prover::{ClaimGenerator, InputType, InteractionClaimGenerator}; |
Oops, something went wrong.