Skip to content

Commit d33759b

Browse files
committed
tower proof new api
1 parent 1817715 commit d33759b

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

ceno_zkvm/src/scheme/gpu/mod.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ use gkr_iop::gpu::gpu_prover::*;
4545

4646
pub struct GpuTowerProver;
4747

48+
use crate::scheme::constants::NUM_FANIN;
4849
use gkr_iop::gpu::{ArcMultilinearExtensionGpu, MultilinearExtensionGpu};
4950

5051
// Extract out_evals from GPU-built tower witnesses
@@ -481,7 +482,7 @@ impl<E: ExtensionField, PCS: PolynomialCommitmentScheme<E>> TowerProver<GpuBacke
481482
let span = entered_span!("prove_tower_relation", profiling_2 = true);
482483
let (point_gl, proof_gpu) = cuda_hal
483484
.tower
484-
.create_proof(&input, basic_tr)
485+
.create_proof(&cuda_hal, &input, NUM_FANIN, basic_tr)
485486
.expect("gpu tower create_proof failed");
486487
exit_span!(span);
487488

gkr_iop/src/gpu/mod.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -266,7 +266,8 @@ impl<E: ExtensionField, PCS: PolynomialCommitmentScheme<E>> ProverBackend for Gp
266266
type MultilinearPoly<'a> = MultilinearExtensionGpu<'a, E>;
267267
type Matrix = RowMajorMatrix<E::BaseField>;
268268
#[cfg(feature = "gpu")]
269-
type PcsData = BasefoldCommitmentWithWitnessGpu<E::BaseField, BufferImpl<'static, E::BaseField>>;
269+
type PcsData =
270+
BasefoldCommitmentWithWitnessGpu<E::BaseField, BufferImpl<'static, E::BaseField>>;
270271
#[cfg(not(feature = "gpu"))]
271272
type PcsData = <PCS as PolynomialCommitmentScheme<E>>::CommitmentWithWitness;
272273

0 commit comments

Comments
 (0)