WIP: update vm chip to support sha512

Author: Avaneesh-axiom

Cargo.lock

@@ -230,6 +230,7 @@ rrs-lib = "0.1.0"
 rand = { version = "0.8.5", default-features = false }
 hex = { version = "0.4.3", default-features = false }
 serde-big-array = "0.5.1"
+ndarray = "0.16"
 
 # default-features = false for no_std for use in guest programs
 itertools = { version = "0.13.0", default-features = false }

Cargo.toml

@@ -294,7 +294,7 @@ fn get_const_cols_ref_fields(
     let slice_var = format_ident!("{}_slice", f.ident.clone().unwrap());
 
     if f.attrs.iter().any(|attr| attr.path().is_ident("plain")) {
-        // treat the field as a struct that derives AlignedBorrow
+        // treat the field as a struct that derives AlignedBorrow (and doesn't depend on the config)
         let f_ty = &f.ty;
         return Ok(FieldInfo {
             ty: parse_quote! {

crates/circuits/sha-macros/src/lib.rs

@@ -16,6 +16,7 @@ openvm-instructions = { workspace = true }
 openvm-sha256-transpiler = { workspace = true }
 openvm-rv32im-circuit = { workspace = true }
 openvm-sha-air = { workspace = true }
+openvm-sha-macros = { workspace = true }
 
 derive-new.workspace = true
 derive_more = { workspace = true, features = ["from"] }
@@ -24,6 +25,7 @@ serde.workspace = true
 sha2 = { version = "0.10", default-features = false }
 strum = { workspace = true }
 bitcode.workspace = true
+ndarray.workspace = true
 
 [dev-dependencies]
 openvm-stark-sdk = { workspace = true }

extensions/sha256/circuit/Cargo.toml

@@ -17,14 +17,15 @@ use openvm_rv32im_circuit::{
     Rv32MExecutor, Rv32MPeriphery,
 };
 use openvm_sha256_transpiler::Rv32Sha256Opcode;
+use openvm_sha_air::Sha256Config;
 use openvm_stark_backend::p3_field::PrimeField32;
 use serde::{Deserialize, Serialize};
 use strum::IntoEnumIterator;
 
 use crate::*;
 
 #[derive(Clone, Debug, VmConfig, derive_new::new, Serialize, Deserialize)]
-pub struct Sha256Rv32Config {
+pub struct Sha2Rv32Config {
     #[system]
     pub system: SystemConfig,
     #[extension]
@@ -34,38 +35,39 @@ pub struct Sha256Rv32Config {
     #[extension]
     pub io: Rv32Io,
     #[extension]
-    pub sha256: Sha256,
+    pub sha2: Sha2,
 }
 
-impl Default for Sha256Rv32Config {
+impl Default for Sha2Rv32Config {
     fn default() -> Self {
         Self {
             system: SystemConfig::default().with_continuations(),
             rv32i: Rv32I,
             rv32m: Rv32M::default(),
             io: Rv32Io,
-            sha256: Sha256,
+            sha2: Sha2,
         }
     }
 }
 
 #[derive(Clone, Copy, Debug, Default, Serialize, Deserialize)]
-pub struct Sha256;
+pub struct Sha2;
 
 #[derive(ChipUsageGetter, Chip, InstructionExecutor, From, AnyEnum, BytesStateful)]
-pub enum Sha256Executor<F: PrimeField32> {
-    Sha256(Sha256VmChip<F>),
+pub enum Sha2Executor<F: PrimeField32> {
+    Sha256(ShaVmChip<F, Sha256Config>),
+    Sha512(ShaVmChip<F, Sha512Config>),
 }
 
 #[derive(From, ChipUsageGetter, Chip, AnyEnum, BytesStateful)]
-pub enum Sha256Periphery<F: PrimeField32> {
+pub enum ShaPeriphery<F: PrimeField32> {
     BitwiseOperationLookup(SharedBitwiseOperationLookupChip<8>),
     Phantom(PhantomChip<F>),
 }
 
-impl<F: PrimeField32> VmExtension<F> for Sha256 {
-    type Executor = Sha256Executor<F>;
-    type Periphery = Sha256Periphery<F>;
+impl<F: PrimeField32> VmExtension<F> for Sha2 {
+    type Executor = ShaExecutor<F>;
+    type Periphery = ShaPeriphery<F>;
 
     fn build(
         &self,
@@ -84,7 +86,7 @@ impl<F: PrimeField32> VmExtension<F> for Sha256 {
             chip
         };
 
-        let sha256_chip = Sha256VmChip::new(
+        let sha256_chip = ShaVmChip::new(
             builder.system_port(),
             builder.system_config().memory_config.pointer_max_bits,
             bitwise_lu_chip,