patch gpu crates

Author: kunxian-xia

Cargo.lock

@@ -88,6 +88,37 @@ alloy-primitives = { git = "https://github.com/scroll-tech/alloy-core", branch =
 ruint = { git = "https://github.com/scroll-tech/uint.git", branch = "v1.12.3" }
 tiny-keccak = { git = "https://github.com/scroll-tech/tiny-keccak", branch = "scroll-patch-v2.0.2-euclid-upgrade" }
 
+[patch."https://github.com/openvm-org/stark-backend.git"]
+openvm-stark-backend = { git = "ssh://[email protected]/scroll-tech/openvm-stark-gpu.git", tag = "v1.0.0-rc.2", features = ["gpu"] }
+openvm-stark-sdk = { git = "ssh://[email protected]/scroll-tech/openvm-stark-gpu.git", tag = "v1.0.0-rc.2", features = ["gpu"] }
+
+[patch."https://github.com/Plonky3/Plonky3.git"]
+p3-air = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-field = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-commit = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-matrix = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-baby-bear = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", features = [
+    "nightly-features",
+], tag = "v0.1.0" }
+p3-util = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-challenger = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-dft = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-fri = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-goldilocks = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-keccak = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-keccak-air = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-blake3 = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-mds = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-merkle-tree = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-monty-31 = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-poseidon = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-poseidon2 = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-poseidon2-air = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-symmetric = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-uni-stark = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+p3-maybe-rayon = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" } # the "parallel" feature is NOT on by default to allow single-threaded benchmarking
+p3-bn254-fr = { git = "ssh://[email protected]/scroll-tech/plonky3-gpu.git", tag = "v0.1.0" }
+
 [profile.maxperf]
 inherits = "release"
 lto = "fat"

Cargo.toml

@@ -20,6 +20,7 @@ openvm-native-compiler = { workspace = true, default-features = false }
 openvm-native-recursion = { workspace = true, default-features = false }
 openvm-native-transpiler = { workspace = true, default-features = false }
 openvm-transpiler = { workspace = true, default-features = false }
+openvm-stark-sdk.workspace = true
 
 alloy-primitives.workspace = true
 eyre.workspace = true

crates/integration/Cargo.toml

@@ -57,7 +57,7 @@ pub trait ProverTester {
     /// Setup directory structure for the test suite.
     fn setup() -> eyre::Result<()> {
         // Setup tracing subscriber.
-        setup_logger()?;
+        // setup_logger()?;
 
         // If user has set an output directory, use it.
         let dir_testrun = if let Ok(env_dir) = std::env::var(ENV_OUTPUT_DIR) {
@@ -159,6 +159,7 @@ impl<T: Clone, P: Clone> ProveVerifyOutcome<T, P> {
 }
 
 /// Setup test environment
+#[allow(dead_code)]
 fn setup_logger() -> eyre::Result<()> {
     let fmt_layer = tracing_subscriber::fmt::layer()
         .pretty()

crates/integration/src/lib.rs

@@ -54,7 +54,7 @@ impl ProverTester for ChunkProverTester {
             }
             Err(_) => {
                 #[cfg(not(feature = "euclidv2"))]
-                let blocks = 12508460usize..=12508463usize;
+                let blocks = 10319966usize..=10319974usize;
                 #[cfg(feature = "euclidv2")]
                 let blocks = 1usize..=4usize;
                 blocks

crates/integration/src/testers/chunk.rs

@@ -1,3 +1,4 @@
+use openvm_stark_sdk::bench::run_with_metric_collection;
 use scroll_zkvm_integration::{
     ProverTester, prove_verify_multi, prove_verify_single,
     testers::chunk::{ChunkProverTester, MultiChunkProverTester, read_block_witness_from_testdata},
@@ -141,9 +142,10 @@ fn guest_profiling() -> eyre::Result<()> {
 fn setup_prove_verify_single() -> eyre::Result<()> {
     ChunkProverTester::setup()?;
 
-    prove_verify_single::<ChunkProverTester>(None)?;
-
-    Ok(())
+    run_with_metric_collection("OUTPUT_PATH", || -> eyre::Result<()> {
+        prove_verify_single::<ChunkProverTester>(None)?;
+        Ok(())
+    })
 }
 
 #[test]