feat: v1.06-testnet (succinctlabs#879)

scroll-tech · Jun 3, 2024 · 1049583 · 1049583
2 parents 466fbb9 + bedc606
commit 1049583
Show file tree

Hide file tree

Showing 156 changed files with 3,542 additions and 2,778 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/core/Cargo.toml b/core/Cargo.toml
@@ -58,6 +58,7 @@ web-time = "1.1.0"
 rayon-scan = "0.1.1"
 thiserror = "1.0.60"
 num-bigint = { version = "0.4.3", default-features = false }
+rand = "0.8.5"
 
 [dev-dependencies]
 tiny-keccak = { version = "2.0.2", features = ["keccak"] }

diff --git a/core/src/air/builder.rs b/core/src/air/builder.rs
@@ -308,6 +308,7 @@ pub trait AluAirBuilder: BaseAirBuilder {
         c: Word<impl Into<Self::Expr>>,
         shard: impl Into<Self::Expr>,
         channel: impl Into<Self::Expr>,
+        nonce: impl Into<Self::Expr>,
         multiplicity: impl Into<Self::Expr>,
     ) {
         let values = once(opcode.into())
@@ -316,6 +317,7 @@ pub trait AluAirBuilder: BaseAirBuilder {
             .chain(c.0.into_iter().map(Into::into))
             .chain(once(shard.into()))
             .chain(once(channel.into()))
+            .chain(once(nonce.into()))
             .collect();
 
         self.send(AirInteraction::new(
@@ -335,6 +337,7 @@ pub trait AluAirBuilder: BaseAirBuilder {
         c: Word<impl Into<Self::Expr>>,
         shard: impl Into<Self::Expr>,
         channel: impl Into<Self::Expr>,
+        nonce: impl Into<Self::Expr>,
         multiplicity: impl Into<Self::Expr>,
     ) {
         let values = once(opcode.into())
@@ -343,6 +346,7 @@ pub trait AluAirBuilder: BaseAirBuilder {
             .chain(c.0.into_iter().map(Into::into))
             .chain(once(shard.into()))
             .chain(once(channel.into()))
+            .chain(once(nonce.into()))
             .collect();
 
         self.receive(AirInteraction::new(
@@ -359,6 +363,7 @@ pub trait AluAirBuilder: BaseAirBuilder {
         shard: impl Into<Self::Expr> + Clone,
         channel: impl Into<Self::Expr> + Clone,
         clk: impl Into<Self::Expr> + Clone,
+        nonce: impl Into<Self::Expr> + Clone,
         syscall_id: impl Into<Self::Expr> + Clone,
         arg1: impl Into<Self::Expr> + Clone,
         arg2: impl Into<Self::Expr> + Clone,
@@ -369,6 +374,7 @@ pub trait AluAirBuilder: BaseAirBuilder {
                 shard.clone().into(),
                 channel.clone().into(),
                 clk.clone().into(),
+                nonce.clone().into(),
                 syscall_id.clone().into(),
                 arg1.clone().into(),
                 arg2.clone().into(),
@@ -385,6 +391,7 @@ pub trait AluAirBuilder: BaseAirBuilder {
         shard: impl Into<Self::Expr> + Clone,
         channel: impl Into<Self::Expr> + Clone,
         clk: impl Into<Self::Expr> + Clone,
+        nonce: impl Into<Self::Expr> + Clone,
         syscall_id: impl Into<Self::Expr> + Clone,
         arg1: impl Into<Self::Expr> + Clone,
         arg2: impl Into<Self::Expr> + Clone,
@@ -395,6 +402,7 @@ pub trait AluAirBuilder: BaseAirBuilder {
                 shard.clone().into(),
                 channel.clone().into(),
                 clk.clone().into(),
+                nonce.clone().into(),
                 syscall_id.clone().into(),
                 arg1.clone().into(),
                 arg2.clone().into(),

diff --git a/core/src/alu/add_sub/mod.rs b/core/src/alu/add_sub/mod.rs
@@ -1,7 +1,8 @@
 use core::borrow::{Borrow, BorrowMut};
 use core::mem::size_of;
 
-use p3_air::{Air, BaseAir};
+use p3_air::{Air, AirBuilder, BaseAir};
+use p3_field::AbstractField;
 use p3_field::PrimeField;
 use p3_matrix::dense::RowMajorMatrix;
 use p3_matrix::Matrix;
@@ -38,6 +39,9 @@ pub struct AddSubCols<T> {
     /// The channel number, used for byte lookup table.
     pub channel: T,
 
+    /// The nonce of the operation.
+    pub nonce: T,
+
     /// Instance of `AddOperation` to handle addition logic in `AddSubChip`'s ALU operations.
     /// It's result will be `a` for the add operation and `b` for the sub operation.
     pub add_operation: AddOperation<T>,
@@ -129,6 +133,13 @@ impl<F: PrimeField> MachineAir<F> for AddSubChip {
         // Pad the trace to a power of two.
         pad_to_power_of_two::<NUM_ADD_SUB_COLS, F>(&mut trace.values);
 
+        // Write the nonces to the trace.
+        for i in 0..trace.height() {
+            let cols: &mut AddSubCols<F> =
+                trace.values[i * NUM_ADD_SUB_COLS..(i + 1) * NUM_ADD_SUB_COLS].borrow_mut();
+            cols.nonce = F::from_canonical_usize(i);
+        }
+
         trace
     }
 
@@ -151,6 +162,14 @@ where
         let main = builder.main();
         let local = main.row_slice(0);
         let local: &AddSubCols<AB::Var> = (*local).borrow();
+        let next = main.row_slice(1);
+        let next: &AddSubCols<AB::Var> = (*next).borrow();
+
+        // Constrain the incrementing nonce.
+        builder.when_first_row().assert_zero(local.nonce);
+        builder
+            .when_transition()
+            .assert_eq(local.nonce + AB::Expr::one(), next.nonce);
 
         // Evaluate the addition operation.
         AddOperation::<AB::F>::eval(
@@ -172,6 +191,7 @@ where
             local.operand_2,
             local.shard,
             local.channel,
+            local.nonce,
             local.is_add,
         );
 
@@ -183,6 +203,7 @@ where
             local.operand_2,
             local.shard,
             local.channel,
+            local.nonce,
             local.is_sub,
         );
 

diff --git a/core/src/alu/bitwise/mod.rs b/core/src/alu/bitwise/mod.rs
@@ -1,7 +1,9 @@
 use core::borrow::{Borrow, BorrowMut};
 use core::mem::size_of;
 
+use p3_air::AirBuilder;
 use p3_air::{Air, BaseAir};
+use p3_field::AbstractField;
 use p3_field::PrimeField;
 use p3_matrix::dense::RowMajorMatrix;
 use p3_matrix::Matrix;
@@ -31,6 +33,9 @@ pub struct BitwiseCols<T> {
     /// The channel number, used for byte lookup table.
     pub channel: T,
 
+    /// The nonce of the operation.
+    pub nonce: T,
+
     /// The output operand.
     pub a: Word<T>,
 
@@ -111,6 +116,12 @@ impl<F: PrimeField> MachineAir<F> for BitwiseChip {
         // Pad the trace to a power of two.
         pad_to_power_of_two::<NUM_BITWISE_COLS, F>(&mut trace.values);
 
+        for i in 0..trace.height() {
+            let cols: &mut BitwiseCols<F> =
+                trace.values[i * NUM_BITWISE_COLS..(i + 1) * NUM_BITWISE_COLS].borrow_mut();
+            cols.nonce = F::from_canonical_usize(i);
+        }
+
         trace
     }
 
@@ -133,6 +144,14 @@ where
         let main = builder.main();
         let local = main.row_slice(0);
         let local: &BitwiseCols<AB::Var> = (*local).borrow();
+        let next = main.row_slice(1);
+        let next: &BitwiseCols<AB::Var> = (*next).borrow();
+
+        // Constrain the incrementing nonce.
+        builder.when_first_row().assert_zero(local.nonce);
+        builder
+            .when_transition()
+            .assert_eq(local.nonce + AB::Expr::one(), next.nonce);
 
         // Get the opcode for the operation.
         let opcode = local.is_xor * ByteOpcode::XOR.as_field::<AB::F>()
@@ -166,6 +185,7 @@ where
             local.c,
             local.shard,
             local.channel,
+            local.nonce,
             local.is_xor + local.is_or + local.is_and,
         );