arnaucube/poulpy
1
0
Fork 0
mirror of https://github.com/arnaucube/poulpy.git synced 2026-02-10 05:06:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

Non-avx build (wip)

Browse Source
This commit is contained in:
Rasoul Akhavan Mahdavi
2025-11-20 10:22:20 -05:00
parent 8a039e1c3a
commit 0fb88c9bd3
8 changed files with 43 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
poulpy-cpu-avx/benches/fft.rs
View File

@@ -1,3 +1,4 @@
#![cfg(target_arch = "x86_64")]
use std::hint::black_box;
use criterion::{BenchmarkId, Criterion, criterion_group, criterion_main};

1
poulpy-cpu-avx/benches/vec_znx.rs
View File

@@ -1,3 +1,4 @@
#![cfg(target_arch = "x86_64")]
// poulpy-backend/benches/vec_znx_add.rs
use criterion::{Criterion, criterion_group, criterion_main};
use poulpy_cpu_avx::FFT64Avx;

16
poulpy-cpu-avx/benches/vmp.rs
View File

@@ -1,11 +1,27 @@
// poulpy-backend/benches/vec_znx_add.rs
use criterion::{Criterion, criterion_group, criterion_main};
#[cfg(target_arch = "x86_64")]
use poulpy_cpu_avx::FFT64Avx;
#[cfg(not(target_arch = "x86_64"))]
use poulpy_cpu_ref::FFT64Ref;
use poulpy_hal::bench_suite::vmp::bench_vmp_apply_dft_to_dft;
#[cfg(target_arch = "x86_64")]
fn bench_vmp_apply_dft_to_dft_cpu_avx_fft64(c: &mut Criterion) {
bench_vmp_apply_dft_to_dft::<FFT64Avx>(c, "FFT64Avx");
}
#[cfg(not(target_arch = "x86_64"))]
fn bench_vmp_apply_dft_to_dft_cpu_ref_fft64(c: &mut Criterion) {
bench_vmp_apply_dft_to_dft::<FFT64Ref>(c, "FFT64Ref");
}
#[cfg(target_arch = "x86_64")]
criterion_group!(benches_x86, bench_vmp_apply_dft_to_dft_cpu_avx_fft64,);
#[cfg(not(target_arch = "x86_64"))]
criterion_group!(benches_ref, bench_vmp_apply_dft_to_dft_cpu_ref_fft64,);
#[cfg(target_arch = "x86_64")]
criterion_main!(benches_x86);
#[cfg(not(target_arch = "x86_64"))]
criterion_main!(benches_ref);