poulpy/poulpy-backend/benches/vec_znx.rs
poulpy-backend/benches/vec_znx.rsRaw
1// poulpy-backend/benches/vec_znx_add.rs
2use criterion::{Criterion, criterion_group, criterion_main};
3use poulpy_backend::{cpu_fft64_ref, cpu_spqlios};
4use poulpy_hal::reference::vec_znx::{bench_vec_znx_add, bench_vec_znx_automorphism, bench_vec_znx_normalize_inplace};
5
6#[allow(dead_code)]
7fn bench_vec_znx_add_cpu_spqlios_fft64(c: &mut Criterion) {
8 bench_vec_znx_add::<cpu_spqlios::FFT64Spqlios>(c, "cpu_spqlios::fft64");
9}
10
11#[allow(dead_code)]
12fn bench_vec_znx_add_cpu_ref_fft64(c: &mut Criterion) {
13 bench_vec_znx_add::<cpu_fft64_ref::FFT64Ref>(c, "cpu_spqlios::fft64");
14}
15
16#[allow(dead_code)]
17fn bench_vec_znx_normalize_inplace_cpu_ref_fft64(c: &mut Criterion) {
18 bench_vec_znx_normalize_inplace::<cpu_fft64_ref::FFT64Ref>(c, "cpu_ref::fft64");
19}
20
21#[allow(dead_code)]
22fn bench_vec_znx_normalize_inplace_cpu_spqlios_fft64(c: &mut Criterion) {
23 bench_vec_znx_normalize_inplace::<cpu_spqlios::FFT64Spqlios>(c, "cpu_spqlios::fft64");
24}
25
26fn bench_vec_znx_automorphism_cpu_ref_fft64(c: &mut Criterion) {
27 bench_vec_znx_automorphism::<cpu_fft64_ref::FFT64Ref>(c, "cpu_ref::fft64");
28}
29
30fn bench_vec_znx_automorphism_cpu_spqlios_fft64(c: &mut Criterion) {
31 bench_vec_znx_automorphism::<cpu_spqlios::FFT64Spqlios>(c, "cpu_spqlios::fft64");
32}
33
34criterion_group!(
35 benches,
36 // bench_vec_znx_add_cpu_spqlios_fft64,
37 // bench_vec_znx_add_cpu_ref_fft64,
38 // bench_vec_znx_normalize_inplace_cpu_ref_fft64,
39 // bench_vec_znx_normalize_inplace_cpu_spqlios_fft64,
40 bench_vec_znx_automorphism_cpu_ref_fft64,
41 bench_vec_znx_automorphism_cpu_spqlios_fft64,
42);
43criterion_main!(benches);