1use criterion::{criterion_group, criterion_main, Criterion};
 2use rns::poly::PolyRNS;
 3use rns::ring::RingRNS;
 4
 5fn div_floor_by_last_modulus_ntt_true(c: &mut Criterion) {
 6    fn runner(r: RingRNS<u64>) -> Box<dyn FnMut()> {
 7        let a: PolyRNS<u64> = r.new_polyrns();
 8        let mut b: [rns::poly::Poly<u64>; 2] = [r.new_poly(), r.new_poly()];
 9        let mut c: PolyRNS<u64> = r.new_polyrns();
10
11        Box::new(move || r.div_by_last_modulus::<false, true>(&a, &mut b, &mut c))
12    }
13
14    let mut b: criterion::BenchmarkGroup<'_, criterion::measurement::WallTime> =
15        c.benchmark_group("div_floor_by_last_modulus_ntt_true");
16    for log_n in 11..18 {
17        let n = 1 << log_n;
18        let moduli: Vec<u64> = vec![
19            0x1fffffffffe00001u64,
20            0x1fffffffffc80001u64,
21            0x1fffffffffb40001,
22            0x1fffffffff500001,
23        ];
24
25        let ring_rns: RingRNS<u64> = RingRNS::new(n, moduli);
26
27        let runners = [(format!("prime/n={}/level={}", n, ring_rns.level()), {
28            runner(ring_rns)
29        })];
30
31        for (name, mut runner) in runners {
32            b.bench_with_input(name, &(), |b, _| b.iter(&mut runner));
33        }
34    }
35}
36
37criterion_group!(benches, div_floor_by_last_modulus_ntt_true);
38criterion_main!(benches);