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