gpu process sumcheck in dag form

Author: hero78119

Cargo.lock

@@ -1,15 +1,15 @@
 [workspace]
 members = [
-  "ceno_cli",
-  "ceno_emul",
-  "ceno_host",
-  "ceno_rt",
-  "ceno_syscall",
-  "ceno_zkvm",
-  "derive",
-  "examples-builder",
-  "examples",
-  "guest_libs/*",
+    "ceno_cli",
+    "ceno_emul",
+    "ceno_host",
+    "ceno_rt",
+    "ceno_syscall",
+    "ceno_zkvm",
+    "derive",
+    "examples-builder",
+    "examples",
+    "guest_libs/*",
 ]
 resolver = "2"
 
@@ -62,11 +62,11 @@ secp = "0.4.1"
 serde = { version = "1.0", features = ["derive", "rc"] }
 serde_json = "1.0"
 smallvec = { version = "1.13.2", features = [
-  "const_generics",
-  "const_new",
-  "serde",
-  "union",
-  "write",
+    "const_generics",
+    "const_new",
+    "serde",
+    "union",
+    "write",
 ] }
 strum = "0.26"
 strum_macros = "0.26"
@@ -75,13 +75,13 @@ thiserror = "2"
 thread_local = "1.1"
 tiny-keccak = { version = "2.0.2", features = ["keccak"] }
 tracing = { version = "0.1", features = [
-  "attributes",
+    "attributes",
 ] }
 tracing-forest = { version = "0.1.6" }
 tracing-subscriber = { version = "0.3", features = ["env-filter"] }
 uint = "0.8"
 
-ceno_gpu = { path = "utils/cuda_hal", package = "cuda_hal" }
+ceno_gpu = { git = "ssh://[email protected]/scroll-tech/ceno-gpu.git", package = "cuda_hal", branch = "main", default-features = false, features = ["bb31"] }
 
 [profile.dev]
 lto = "thin"
@@ -96,17 +96,17 @@ opt-level = 3
 [profile.release]
 lto = "thin"
 
-# [patch."ssh://[email protected]/scroll-tech/ceno-gpu.git"]
-# ceno_gpu = { path = "../ceno-gpu/cuda_hal", package = "cuda_hal", default-features = false, features=["bb31"] }
+[patch."ssh://git@github.com/scroll-tech/ceno-gpu.git"]
+ceno_gpu = { path = "../ceno-gpu/cuda_hal", package = "cuda_hal", default-features = false, features = ["bb31"] }
 
-# [patch."https://github.com/scroll-tech/gkr-backend"]
-# ff_ext = { path = "../gkr-backend/crates/ff_ext", package = "ff_ext" }
-# mpcs = { path = "../gkr-backend/crates/mpcs", package = "mpcs" }
-# multilinear_extensions = { path = "../gkr-backend/crates/multilinear_extensions", package = "multilinear_extensions" }
-# p3 = { path = "../gkr-backend/crates/p3", package = "p3" }
-# poseidon = { path = "../gkr-backend/crates/poseidon", package = "poseidon" }
-# sp1-curves = { path = "../gkr-backend/crates/curves", package = "sp1-curves" }
-# sumcheck = { path = "../gkr-backend/crates/sumcheck", package = "sumcheck" }
-# transcript = { path = "../gkr-backend/crates/transcript", package = "transcript" }
-# whir = { path = "../gkr-backend/crates/whir", package = "whir" }
-# witness = { path = "../gkr-backend/crates/witness", package = "witness" }
+[patch."https://github.com/scroll-tech/gkr-backend"]
+ff_ext = { path = "../gkr-backend/crates/ff_ext", package = "ff_ext" }
+mpcs = { path = "../gkr-backend/crates/mpcs", package = "mpcs" }
+multilinear_extensions = { path = "../gkr-backend/crates/multilinear_extensions", package = "multilinear_extensions" }
+p3 = { path = "../gkr-backend/crates/p3", package = "p3" }
+poseidon = { path = "../gkr-backend/crates/poseidon", package = "poseidon" }
+sp1-curves = { path = "../gkr-backend/crates/curves", package = "sp1-curves" }
+sumcheck = { path = "../gkr-backend/crates/sumcheck", package = "sumcheck" }
+transcript = { path = "../gkr-backend/crates/transcript", package = "transcript" }
+whir = { path = "../gkr-backend/crates/whir", package = "whir" }
+witness = { path = "../gkr-backend/crates/witness", package = "witness" }

Cargo.toml

@@ -103,6 +103,15 @@ pub struct Layer<E: ExtensionField> {
     pub main_sumcheck_expression_monomial_terms: Option<Vec<Term<Expression<E>, Expression<E>>>>,
     pub main_sumcheck_expression: Option<Expression<E>>,
 
+    // flatten computation dag
+    pub main_sumcheck_expression_dag: Option<(
+        Vec<u32>,
+        Vec<Instance>,
+        Vec<Expression<E>>,
+        Vec<Either<E::BaseField, E>>,
+        (usize, usize),
+    )>,
+
     // rotation sumcheck expression, only optionally valid for zerocheck
     // store in 2 forms: expression & monomial
     pub rotation_sumcheck_expression_monomial_terms:
@@ -175,6 +184,7 @@ impl<E: ExtensionField> Layer<E> {
                     expr_names,
                     main_sumcheck_expression_monomial_terms: None,
                     main_sumcheck_expression: None,
+                    main_sumcheck_expression_dag: None,
                     rotation_sumcheck_expression_monomial_terms: None,
                     rotation_sumcheck_expression: None,
                 };