Add basic benchmarks for the prover

This commit is contained in:
George Kadianakis
2023-12-20 18:18:03 +02:00
parent f87a71f6a6
commit 89d9528a01
3 changed files with 42 additions and 2 deletions

36
benches/bench_prove.rs Normal file
View File

@@ -0,0 +1,36 @@
use ark_std::UniformRand;
use sigmabus_poc::transcript::{tests::poseidon_test_config, PoseidonTranscript};
use criterion::{criterion_group, criterion_main, Criterion};
use ark_bn254::{Bn254, Fr, G1Projective};
use ark_std::rand::{RngCore, SeedableRng};
use ark_std::test_rng;
use sigmabus_poc::sigmabus::Sigmabus;
fn bench_prove(c: &mut Criterion) {
let mut rng = ark_std::rand::rngs::StdRng::seed_from_u64(test_rng().next_u64());
let poseidon_config = poseidon_test_config::<Fr>();
// generate the trusted setup
let params = Sigmabus::<Bn254>::setup(&mut rng, &poseidon_config);
// compute the witness x
let x = Fr::rand(&mut rng);
let mut transcript_p = PoseidonTranscript::<G1Projective>::new(&poseidon_config);
// generate Sigmabus proof for X==x*G
c.bench_function("prove", |b| {
b.iter(|| {
let _proof = Sigmabus::<Bn254>::prove(&mut rng, &params, &mut transcript_p, x).unwrap();
});
});
}
criterion_group! {
name=prover_benches;
config=Criterion::default();
targets=
bench_prove,
}
criterion_main!(prover_benches);