Feat: Native Poseidon2 chip

crates/vm/src/arch/testing/mod.rs

@@ -144,6 +144,16 @@ impl<F: PrimeField32> VmChipTestBuilder<F> {
         self.memory.write(address_space, pointer, value);
     }
 
+    pub fn write_usize<const N: usize>(
+        &mut self,
+        address_space: usize,
+        pointer: usize,
+        value: [usize; N],
+    ) {
+        self.memory
+            .write(address_space, pointer, value.map(F::from_canonical_usize));
+    }
+
     pub fn write_heap<const NUM_LIMBS: usize>(
         &mut self,
         register: usize,

docs/specs/ISA.md

@@ -490,6 +490,15 @@ and only support `CHUNK = 8` and `WIDTH = 16` in BabyBear Poseidon2 above. For t
 is read in two batches of size `CHUNK`, and, similarly, the output is written in either one or two batches of
 size `CHUNK`, depending on the output size of the corresponding opcode.
 
+### Verification
+
+We have the following special opcode tailored to optimize verification. Due to already having a large number of operands,
+the address space is fixed to be `AS::Native = 5`.
+
+| Name                                                                                                                                                                                                                         | Operands        | Description                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                 |
+|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|-----------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| **VERIFY_BATCH** `[CHUNK, PID]` <br/><br/> Here `CHUNK` and `PID` are **constants** that determine different opcodes. `PID` is an internal identifier for particular Poseidon2 constants dependent on the field (see below). | `a,b,c,d,e,f,g` | Computes `mmcs::verify_batch`. In the native address space, `[a], [b], [d], [e], [f]` should be the array start pointers for the dimensions array, the opened values array (which contains more arrays), the proof (which contains arrays of length `CHUNK`) and the commitment (which is an array of length `CHUNK`). `[c]` should be the length of the opened values array (and so should be equal to the length of the dimensions array as well). `g` should be the reciprocal of the size (in field elements) of the values contained in the opened values array: if the opened values array contains field elements, `g` should be 1; if the opened values array contains extension field elements, `g` should be 1/4. |
+
 ## Phantom Sub-Instructions
 
 As mentioned in [System](#system), the **PHANTOM** instruction has different behavior based on the operand `c`.

extensions/native/circuit/Cargo.toml

@@ -19,6 +19,7 @@ openvm-instructions = { workspace = true }
 openvm-rv32im-circuit = { workspace = true }
 openvm-native-compiler = { workspace = true }
 
+
 strum.workspace = true
 itertools.workspace = true
 tracing.workspace = true

extensions/native/circuit/src/extension.rs

@@ -18,7 +18,7 @@ use openvm_instructions::{
 use openvm_native_compiler::{
     CastfOpcode, FieldArithmeticOpcode, FieldExtensionOpcode, FriOpcode, NativeBranchEqualOpcode,
     NativeJalOpcode, NativeLoadStore4Opcode, NativeLoadStoreOpcode, NativePhantom,
-    BLOCK_LOAD_STORE_SIZE,
+    VerifyBatchOpcode, BLOCK_LOAD_STORE_SIZE,
 };
 use openvm_poseidon2_air::Poseidon2Config;
 use openvm_rv32im_circuit::{
@@ -31,6 +31,7 @@ use strum::IntoEnumIterator;
 
 use crate::{
     adapters::{convert_adapter::ConvertAdapterChip, *},
+    chip::NativePoseidon2Chip,
     phantom::*,
     *,
 };
@@ -85,8 +86,8 @@ pub enum NativeExecutor<F: PrimeField32> {
     Jal(NativeJalChip<F>),
     FieldArithmetic(FieldArithmeticChip<F>),
     FieldExtension(FieldExtensionChip<F>),
-    Poseidon2(NativePoseidon2Chip<F>),
     FriReducedOpening(FriReducedOpeningChip<F>),
+    VerifyBatch(NativePoseidon2Chip<F, 1>),
 }
 
 #[derive(From, ChipUsageGetter, Chip, AnyEnum, BytesStateful)]
@@ -198,18 +199,22 @@ impl<F: PrimeField32> VmExtension<F> for Native {
             FriOpcode::iter().map(VmOpcode::with_default_offset),
         )?;
 
-        let poseidon2_chip = NativePoseidon2Chip::new(
+        let verify_batch_chip = NativePoseidon2Chip::new(
             execution_bus,
             program_bus,
             memory_bridge,
-            Poseidon2Config::default(),
+            VerifyBatchOpcode::default_offset(),
             Poseidon2Opcode::default_offset(),
-            builder.system_config().max_constraint_degree,
             offline_memory.clone(),
+            Poseidon2Config::default(),
         );
         inventory.add_executor(
-            poseidon2_chip,
-            Poseidon2Opcode::iter().map(VmOpcode::with_default_offset),
+            verify_batch_chip,
+            [
+                VmOpcode::with_default_offset(VerifyBatchOpcode::VERIFY_BATCH),
+                VmOpcode::with_default_offset(Poseidon2Opcode::PERM_POS2),
+                VmOpcode::with_default_offset(Poseidon2Opcode::COMP_POS2),
+            ],
         )?;
 
         builder.add_phantom_sub_executor(