diff --git a/crates/prover/src/builder/mod.rs b/crates/prover/src/builder/mod.rs new file mode 100644 index 000000000..19d747fe3 --- /dev/null +++ b/crates/prover/src/builder/mod.rs @@ -0,0 +1,58 @@ +/// ! This module contains helpers to express and use constraints for components. +use std::fmt::Debug; +use std::ops::{Add, AddAssign, Mul, Sub}; + +use num_traits::{One, Zero}; + +use crate::core::fields::m31::BaseField; +use crate::core::fields::qm31::SecureField; +use crate::core::fields::FieldExpOps; + +/// A trait for evaluating expressions at some point or row. +pub trait EvalAtRow { + /// The base field type. + type F: FieldExpOps + + Copy + + Debug + + AddAssign + + AddAssign + + Add + + Sub + + Mul + + Add + + Mul + + From; + + /// The extension field type. + type EF: One + + Copy + + Debug + + Zero + + Add + + Sub + + Mul + + Add + + Mul + + Sub + + Mul; + + /// Returns the next mask. + fn next_mask(&mut self) -> Self::F { + self.next_interaction_mask(0, [0])[0] + } + + /// Returns the next mask for the given interaction. + fn next_interaction_mask( + &mut self, + interaction: usize, + offsets: [isize; N], + ) -> [Self::F; N]; + + /// Adds a constraint to the component. + fn add_constraint(&mut self, constraint: G) + where + Self::EF: Mul; + + /// Combines 4 base field values into a single extension field value. + fn combine_ef(values: [Self::F; 4]) -> Self::EF; +} diff --git a/crates/prover/src/core/backend/simd/m31.rs b/crates/prover/src/core/backend/simd/m31.rs index 6ea2a7d65..efb664d23 100644 --- a/crates/prover/src/core/backend/simd/m31.rs +++ b/crates/prover/src/core/backend/simd/m31.rs @@ -253,6 +253,12 @@ impl From<[BaseField; N_LANES]> for PackedM31 { } } +impl From for PackedM31 { + fn from(v: BaseField) -> Self { + Self::broadcast(v) + } +} + impl Distribution for Standard { fn sample(&self, rng: &mut R) -> PackedM31 { PackedM31::from_array(rng.gen()) diff --git a/crates/prover/src/lib.rs b/crates/prover/src/lib.rs index 79887aae3..01affbf1a 100644 --- a/crates/prover/src/lib.rs +++ b/crates/prover/src/lib.rs @@ -12,6 +12,7 @@ assert_matches, portable_simd )] +pub mod builder; pub mod core; pub mod examples; pub mod hash_functions;