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