mirror of
https://github.com/arnaucube/poulpy.git
synced 2026-02-10 05:06:44 +01:00
Backend refactor (#120)
* remove spqlios, split cpu_ref and cpu_avx into different crates * remove spqlios submodule * update crate naming & add avx tests
This commit is contained in:
committed by
GitHub
parent
84598e42fe
commit
9e007c988f
80
poulpy-cpu-avx/benches/fft.rs
Normal file
80
poulpy-cpu-avx/benches/fft.rs
Normal file
@@ -0,0 +1,80 @@
|
||||
use std::hint::black_box;
|
||||
|
||||
use criterion::{BenchmarkId, Criterion, criterion_group, criterion_main};
|
||||
use poulpy_cpu_avx::{ReimFFTAvx, ReimIFFTAvx};
|
||||
use poulpy_hal::reference::fft64::reim::{ReimDFTExecute, ReimFFTTable, ReimIFFTTable};
|
||||
|
||||
pub fn bench_ifft_avx2_fma(c: &mut Criterion) {
|
||||
let group_name: String = "ifft_avx2_fma".to_string();
|
||||
|
||||
let mut group = c.benchmark_group(group_name);
|
||||
|
||||
if std::is_x86_feature_detected!("avx2") {
|
||||
fn runner(m: usize) -> impl FnMut() {
|
||||
let mut values: Vec<f64> = vec![0f64; m << 1];
|
||||
|
||||
let scale = 1.0f64 / (2 * m) as f64;
|
||||
values
|
||||
.iter_mut()
|
||||
.enumerate()
|
||||
.for_each(|(i, x)| *x = (i + 1) as f64 * scale);
|
||||
|
||||
let table: ReimIFFTTable<f64> = ReimIFFTTable::<f64>::new(m);
|
||||
move || {
|
||||
ReimIFFTAvx::reim_dft_execute(&table, &mut values);
|
||||
black_box(());
|
||||
}
|
||||
}
|
||||
|
||||
for log_m in [9, 10, 11, 12, 13, 14, 15] {
|
||||
let id: BenchmarkId = BenchmarkId::from_parameter(format!("n: {}", 2 << log_m));
|
||||
|
||||
let mut runner = runner(1 << log_m);
|
||||
group.bench_with_input(id, &(), |b, _| b.iter(&mut runner));
|
||||
}
|
||||
} else {
|
||||
eprintln!("skipping: CPU lacks avx2");
|
||||
return;
|
||||
}
|
||||
|
||||
group.finish();
|
||||
}
|
||||
|
||||
pub fn bench_fft_avx2_fma(c: &mut Criterion) {
|
||||
let group_name: String = "fft_avx2_fma".to_string();
|
||||
|
||||
let mut group = c.benchmark_group(group_name);
|
||||
|
||||
if std::is_x86_feature_detected!("avx2") {
|
||||
fn runner(m: usize) -> impl FnMut() {
|
||||
let mut values: Vec<f64> = vec![0f64; m << 1];
|
||||
|
||||
let scale = 1.0f64 / (2 * m) as f64;
|
||||
values
|
||||
.iter_mut()
|
||||
.enumerate()
|
||||
.for_each(|(i, x)| *x = (i + 1) as f64 * scale);
|
||||
|
||||
let table: ReimFFTTable<f64> = ReimFFTTable::<f64>::new(m);
|
||||
move || {
|
||||
ReimFFTAvx::reim_dft_execute(&table, &mut values);
|
||||
black_box(());
|
||||
}
|
||||
}
|
||||
|
||||
for log_m in [9, 10, 11, 12, 13, 14, 15] {
|
||||
let id: BenchmarkId = BenchmarkId::from_parameter(format!("n: {}", 2 << log_m));
|
||||
|
||||
let mut runner = runner(1 << log_m);
|
||||
group.bench_with_input(id, &(), |b, _| b.iter(&mut runner));
|
||||
}
|
||||
} else {
|
||||
eprintln!("skipping: CPU lacks avx2");
|
||||
return;
|
||||
}
|
||||
|
||||
group.finish();
|
||||
}
|
||||
|
||||
criterion_group!(benches_x86, bench_fft_avx2_fma, bench_ifft_avx2_fma,);
|
||||
criterion_main!(benches_x86);
|
||||
26
poulpy-cpu-avx/benches/vec_znx.rs
Normal file
26
poulpy-cpu-avx/benches/vec_znx.rs
Normal file
@@ -0,0 +1,26 @@
|
||||
// poulpy-backend/benches/vec_znx_add.rs
|
||||
use criterion::{Criterion, criterion_group, criterion_main};
|
||||
use poulpy_cpu_avx::FFT64Avx;
|
||||
use poulpy_hal::reference::vec_znx::{bench_vec_znx_add, bench_vec_znx_automorphism, bench_vec_znx_normalize_inplace};
|
||||
|
||||
#[allow(dead_code)]
|
||||
fn bench_vec_znx_add_cpu_avx_fft64(c: &mut Criterion) {
|
||||
bench_vec_znx_add::<FFT64Avx>(c, "FFT64Avx");
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
fn bench_vec_znx_normalize_inplace_cpu_avx_fft64(c: &mut Criterion) {
|
||||
bench_vec_znx_normalize_inplace::<FFT64Avx>(c, "FFT64Avx");
|
||||
}
|
||||
|
||||
fn bench_vec_znx_automorphism_cpu_avx_fft64(c: &mut Criterion) {
|
||||
bench_vec_znx_automorphism::<FFT64Avx>(c, "FFT64Avx");
|
||||
}
|
||||
|
||||
criterion_group!(
|
||||
benches,
|
||||
bench_vec_znx_add_cpu_avx_fft64,
|
||||
bench_vec_znx_normalize_inplace_cpu_avx_fft64,
|
||||
bench_vec_znx_automorphism_cpu_avx_fft64,
|
||||
);
|
||||
criterion_main!(benches);
|
||||
11
poulpy-cpu-avx/benches/vmp.rs
Normal file
11
poulpy-cpu-avx/benches/vmp.rs
Normal file
@@ -0,0 +1,11 @@
|
||||
// poulpy-backend/benches/vec_znx_add.rs
|
||||
use criterion::{Criterion, criterion_group, criterion_main};
|
||||
use poulpy_cpu_avx::FFT64Avx;
|
||||
use poulpy_hal::bench_suite::vmp::bench_vmp_apply_dft_to_dft;
|
||||
|
||||
fn bench_vmp_apply_dft_to_dft_cpu_avx_fft64(c: &mut Criterion) {
|
||||
bench_vmp_apply_dft_to_dft::<FFT64Avx>(c, "FFT64Avx");
|
||||
}
|
||||
|
||||
criterion_group!(benches_x86, bench_vmp_apply_dft_to_dft_cpu_avx_fft64,);
|
||||
criterion_main!(benches_x86);
|
||||
Reference in New Issue
Block a user