feat(zkvm): share witness/fixed Jagged inner opening

Cargo.toml

@@ -27,16 +27,16 @@ version = "0.1.0"
 ceno_crypto_primitives = { git = "https://github.com/scroll-tech/ceno-patch.git", package = "ceno_crypto_primitives", branch = "main" }
 ceno_syscall = { git = "https://github.com/scroll-tech/ceno-patch.git", package = "ceno_syscall", branch = "main" }
 
-ff_ext = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "ff_ext", tag = "v1.0.0-alpha.30" }
-mpcs = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "mpcs", tag = "v1.0.0-alpha.30" }
-multilinear_extensions = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "multilinear_extensions", tag = "v1.0.0-alpha.30" }
-p3 = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "p3", tag = "v1.0.0-alpha.30" }
-poseidon = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "poseidon", tag = "v1.0.0-alpha.30" }
-sp1-curves = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "sp1-curves", tag = "v1.0.0-alpha.30" }
-sumcheck = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "sumcheck", tag = "v1.0.0-alpha.30" }
-transcript = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "transcript", tag = "v1.0.0-alpha.30" }
-whir = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "whir", tag = "v1.0.0-alpha.30" }
-witness = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "witness", tag = "v1.0.0-alpha.30" }
+ff_ext = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "ff_ext", tag = "v1.0.0-alpha.31" }
+mpcs = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "mpcs", tag = "v1.0.0-alpha.31" }
+multilinear_extensions = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "multilinear_extensions", tag = "v1.0.0-alpha.31" }
+p3 = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "p3", tag = "v1.0.0-alpha.31" }
+poseidon = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "poseidon", tag = "v1.0.0-alpha.31" }
+sp1-curves = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "sp1-curves", tag = "v1.0.0-alpha.31" }
+sumcheck = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "sumcheck", tag = "v1.0.0-alpha.31" }
+transcript = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "transcript", tag = "v1.0.0-alpha.31" }
+whir = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "whir", tag = "v1.0.0-alpha.31" }
+witness = { git = "https://github.com/scroll-tech/gkr-backend.git", package = "witness", tag = "v1.0.0-alpha.31" }
 
 anyhow = { version = "1.0", default-features = false }
 bincode = "1"

ceno_zkvm/src/scheme.rs

@@ -269,13 +269,19 @@ impl<E: ExtensionField, PCS: PolynomialCommitmentScheme<E> + Serialize> fmt::Dis
     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
         // break down zkvm proof size
 
+        let overall_size = bincode::serialized_size(&self).expect("serialization error");
         // also provide by-circuit stats
         let mut by_circuitname_stats = HashMap::new();
         // opcode circuit mpcs size
         let mpcs_opcode_commitment =
             bincode::serialized_size(&self.witin_commit).expect("serialization error");
         let mpcs_opcode_opening =
             bincode::serialized_size(&self.opening_proof).expect("serialization error");
+        let mpcs_opening_breakdown = format_proof_size_breakdown(
+            "mpcs opening",
+            PCS::proof_size_breakdown(&self.opening_proof),
+            overall_size,
+        );
 
         // tower proof size
         let tower_proof = self
@@ -312,9 +318,6 @@ impl<E: ExtensionField, PCS: PolynomialCommitmentScheme<E> + Serialize> fmt::Dis
             .iter()
             .sum::<u64>();
 
-        // overall size
-        let overall_size = bincode::serialized_size(&self).expect("serialization error");
-
         // break down by circuit name
         let by_circuitname_stats = by_circuitname_stats
             .iter()
@@ -336,6 +339,8 @@ impl<E: ExtensionField, PCS: PolynomialCommitmentScheme<E> + Serialize> fmt::Dis
             "overall_size {:.2}mb. \n\
             mpcs commitment {:?}% \n\
             mpcs opening {:?}% \n\
+            mpcs opening break down: \n\
+            {} \n\
             tower proof {:?}% \n\
             main sumcheck proof {:?}% \n\
             by circuit_name break down: \n\
@@ -344,6 +349,7 @@ impl<E: ExtensionField, PCS: PolynomialCommitmentScheme<E> + Serialize> fmt::Dis
             byte_to_mb(overall_size),
             (mpcs_opcode_commitment * 100).div(overall_size),
             (mpcs_opcode_opening * 100).div(overall_size),
+            mpcs_opening_breakdown,
             (tower_proof * 100).div(overall_size),
             (main_sumcheck * 100).div(overall_size),
             by_circuitname_stats,
@@ -355,6 +361,26 @@ fn byte_to_mb(byte_size: u64) -> f64 {
     byte_size as f64 / (1024.0 * 1024.0)
 }
 
+fn format_proof_size_breakdown(
+    prefix: &str,
+    breakdown: Vec<(String, u64)>,
+    overall_size: u64,
+) -> String {
+    breakdown
+        .into_iter()
+        .map(|(name, size)| {
+            format!(
+                "{}.{}: {:.2}mb({}%, {} bytes)",
+                prefix,
+                name,
+                byte_to_mb(size),
+                (size * 100).div(overall_size),
+                size
+            )
+        })
+        .join("\n")
+}
+
 #[cfg(not(feature = "gpu"))]
 pub fn create_backend<E: ExtensionField, PCS: PolynomialCommitmentScheme<E>>(
     max_num_variables: usize,