diff --git a/ipa-core/benches/dzkp_convert_prover.rs b/ipa-core/benches/dzkp_convert_prover.rs index 8de84ce5d..c8f820bab 100644 --- a/ipa-core/benches/dzkp_convert_prover.rs +++ b/ipa-core/benches/dzkp_convert_prover.rs @@ -1,4 +1,4 @@ -//! Benchmark for the convert_prover function in dzkp_field.rs. +//! Benchmark for the table_indices_prover function in dzkp_field.rs. use criterion::{criterion_group, criterion_main, BatchSize, Criterion}; use ipa_core::protocol::context::dzkp_validator::MultiplicationInputsBlock; @@ -9,7 +9,7 @@ fn convert_prover_benchmark(c: &mut Criterion) { group.bench_function("convert", |b| { b.iter_batched_ref( || thread_rng().gen(), - |input| input.table_indices_prover(), + |input: &mut MultiplicationInputsBlock| input.table_indices_prover(), BatchSize::SmallInput, ) }); diff --git a/ipa-core/src/protocol/context/dzkp_validator.rs b/ipa-core/src/protocol/context/dzkp_validator.rs index 77848c97d..1cfc09784 100644 --- a/ipa-core/src/protocol/context/dzkp_validator.rs +++ b/ipa-core/src/protocol/context/dzkp_validator.rs @@ -166,7 +166,7 @@ impl MultiplicationInputsBlock { } } -#[cfg(test)] +#[cfg(any(test, feature = "enable-benches"))] impl rand::prelude::Distribution for rand::distributions::Standard { fn sample(&self, rng: &mut R) -> MultiplicationInputsBlock { // Generate a random valid block of muliplication intermediates. "Valid" means