1#![allow(non_snake_case)]
 2use criterion::{criterion_group, criterion_main, Criterion};
 3use shockwave_plus::ShockwavePlus;
 4use shockwave_plus::R1CS;
 5use tensor_pcs::Transcript;
 6
 7fn shockwave_plus_bench(c: &mut Criterion) {
 8    type F = halo2curves::secp256k1::Fp;
 9
10    for exp in [12, 15, 18] {
11        let num_vars = 2usize.pow(exp);
12        let num_input = 3;
13
14        let (r1cs, witness) = R1CS::<F>::produce_synthetic_r1cs(num_vars, num_input);
15
16        let mut group = c.benchmark_group(format!("ShockwavePlus num_cons: {}", r1cs.num_cons));
17        let l = 319;
18        let num_rows = (((2f64 / l as f64).sqrt() * (num_vars as f64).sqrt()) as usize)
19            .next_power_of_two()
20            / 2;
21        let ShockwavePlus = ShockwavePlus::new(r1cs.clone(), l, num_rows);
22        group.bench_function("prove", |b| {
23            b.iter(|| {
24                let mut transcript = Transcript::new(b"bench");
25                ShockwavePlus.prove(&witness, &r1cs.public_input, &mut transcript);
26            })
27        });
28    }
29}
30
31fn set_duration() -> Criterion {
32    Criterion::default().sample_size(10)
33}
34
35criterion_group! {
36    name = benches;
37    config = set_duration();
38    targets = shockwave_plus_bench
39}
40criterion_main!(benches);