openvm_circuit/system/program/
air.rs

Help
1use openvm_circuit_primitives_derive::AlignedBorrow;
2use openvm_stark_backend::{
3    air_builders::PartitionedAirBuilder,
4    interaction::InteractionBuilder,
5    p3_air::{Air, BaseAir},
6    p3_field::Field,
7    p3_matrix::Matrix,
8    rap::{BaseAirWithPublicValues, PartitionedBaseAir},
9};
10
11use super::ProgramBus;
12
13#[derive(Copy, Clone, Debug, AlignedBorrow, PartialEq, Eq)]
14#[repr(C)]
15pub struct ProgramCols<T> {
16    pub exec: ProgramExecutionCols<T>,
17    pub exec_freq: T,
18}
19
20#[derive(Copy, Clone, Debug, AlignedBorrow, PartialEq, Eq)]
21#[repr(C)]
22pub struct ProgramExecutionCols<T> {
23    pub pc: T,
24
25    pub opcode: T,
26    pub a: T,
27    pub b: T,
28    pub c: T,
29    pub d: T,
30    pub e: T,
31    pub f: T,
32    pub g: T,
33}
34
35#[derive(Clone, Copy, Debug)]
36pub struct ProgramAir {
37    pub bus: ProgramBus,
38}
39
40impl<F: Field> BaseAirWithPublicValues<F> for ProgramAir {}
41impl<F: Field> PartitionedBaseAir<F> for ProgramAir {
42    fn cached_main_widths(&self) -> Vec<usize> {
43        vec![ProgramExecutionCols::<F>::width()]
44    }
45    fn common_main_width(&self) -> usize {
46        1
47    }
48}
49impl<F: Field> BaseAir<F> for ProgramAir {
50    fn width(&self) -> usize {
51        ProgramCols::<F>::width()
52    }
53}
54
55impl<AB: PartitionedAirBuilder + InteractionBuilder> Air<AB> for ProgramAir {
56    fn eval(&self, builder: &mut AB) {
57        let common_trace = builder.common_main();
58        let cached_trace = &builder.cached_mains()[0];
59
60        let exec_freq = common_trace.row_slice(0)[0];
61        let exec_cols = cached_trace.row_slice(0).to_vec();
62
63        self.bus
64            .inner
65            .add_key_with_lookups(builder, exec_cols, exec_freq);
66    }
67}
openvm_circuit/system/program/air.rs

openvm_circuit/system/program/
air.rs
