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

wip

Browse Source
This commit is contained in:
Jean-Philippe Bossuat
2025-10-26 16:32:22 +01:00
parent 98208d5e67
commit 881483d1bb
11 changed files with 173 additions and 140 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
poulpy-backend/src/cpu_fft64_avx/vec_znx_dft.rs
View File

@@ -194,10 +194,10 @@ unsafe impl VecZnxDftCopyImpl<Self> for FFT64Avx {
}
unsafe impl VecZnxDftZeroImpl<Self> for FFT64Avx {
fn vec_znx_dft_zero_impl<R>(_module: &Module<Self>, res: &mut R)
fn vec_znx_dft_zero_impl<R>(_module: &Module<Self>, res: &mut R, res_col: usize)
where
R: VecZnxDftToMut<Self>,
{
vec_znx_dft_zero(res);
vec_znx_dft_zero(res, res_col);
}
}

2
poulpy-backend/src/cpu_fft64_ref/tests.rs
View File

@@ -4,6 +4,6 @@ use crate::FFT64Ref;
#[test]
fn test_convolution_fft64_ref() {
let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(64);
let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(8);
test_convolution(&module);
}

4
poulpy-backend/src/cpu_fft64_ref/vec_znx_dft.rs
View File

@@ -194,10 +194,10 @@ unsafe impl VecZnxDftCopyImpl<Self> for FFT64Ref {
}
unsafe impl VecZnxDftZeroImpl<Self> for FFT64Ref {
fn vec_znx_dft_zero_impl<R>(_module: &Module<Self>, res: &mut R)
fn vec_znx_dft_zero_impl<R>(_module: &Module<Self>, res: &mut R, res_col: usize)
where
R: VecZnxDftToMut<Self>,
{
vec_znx_dft_zero(res);
vec_znx_dft_zero(res, res_col);
}
}

6
poulpy-backend/src/cpu_spqlios/fft64/vec_znx_dft.rs
View File

@@ -12,7 +12,7 @@ use poulpy_hal::{
reference::{
fft64::{
reim::{ReimCopy, ReimZero, reim_copy_ref, reim_negate_inplace_ref, reim_negate_ref, reim_zero_ref},
vec_znx_dft::vec_znx_dft_copy,
vec_znx_dft::{vec_znx_dft_copy, vec_znx_dft_zero},
},
znx::znx_zero_ref,
},
@@ -426,10 +426,10 @@ impl ReimZero for FFT64Spqlios {
}
unsafe impl VecZnxDftZeroImpl<Self> for FFT64Spqlios {
fn vec_znx_dft_zero_impl<R>(_module: &Module<Self>, res: &mut R)
fn vec_znx_dft_zero_impl<R>(_module: &Module<Self>, res: &mut R, res_col: usize)
where
R: VecZnxDftToMut<Self>,
{
res.to_mut().data.fill(0);
vec_znx_dft_zero(res, res_col);
}
}