Merge pull request #471 from worldfnd/perf/first-sumcheck-round-in-base

perf(prover): run the first sumcheck round in the base field

Author: shreyas-londhe

Cargo.lock

@@ -123,6 +123,7 @@ chrono = "0.4.41"
 # This is a workaround because different package selection based on target does not mix well with workspace dependencies.
 divan = "0.1.21"
 hex = "0.4.3"
+inferno = { version = "0.12", default-features = false }
 itertools = "0.14.0"
 num-bigint = "0.4"
 paste = "1.0.15"
@@ -153,6 +154,7 @@ tokio-util = "0.7.13"
 tower = "0.5.2"
 tower-http = { version = "0.6.6", features = ["cors", "timeout", "trace"] }
 tracing = "0.1.41"
+tracing-flame = "0.2"
 tracing-subscriber = { version = "0.3.18", features = ["env-filter", "ansi"] }
 tracing-tracy = "=0.11.4"
 tracy-client = "=0.18.0"

Cargo.toml

@@ -18,7 +18,7 @@ use {
     },
     provekit_prover::{prove_from_alphas, run_zk_sumcheck_prover, WhirR1CSCommitment},
     tracing::instrument,
-    whir::transcript::ProverState,
+    whir::{algebra::embedding::Identity, transcript::ProverState},
 };
 
 pub trait MavrosR1CSProver {
@@ -53,13 +53,11 @@ impl MavrosR1CSProver for WhirR1CSScheme<Bn254Field> {
             .as_ref()
             .ok_or_else(|| anyhow::anyhow!("c1 must carry blinding state"))?;
 
-        let [a, b, c] = [witgen.out_a, witgen.out_b, witgen.out_c];
         // `g` is committed separately in the extension field; open `blinding_eval`
         // against the blinding vector (g flattened at offset 0).
         let (alpha, blinding_eval) = run_zk_sumcheck_prover(
-            a,
-            b,
-            c,
+            &Identity::new(),
+            [witgen.out_a, witgen.out_b, witgen.out_c],
             &mut merlin,
             self.m_0,
             &blinding.polynomial,