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

Update to non-avx builds

Browse Source
This commit is contained in:
Pro7ech
2025-11-21 15:39:04 +01:00
parent 0fb88c9bd3
commit 3c818d292b
24 changed files with 356 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
poulpy-core/src/scratch.rs
View File

@@ -358,7 +358,7 @@ where
let pairs: u32 = (((infos.rank_out().0 + 1) * infos.rank_out().0) >> 1).max(1);
let mut ksk_infos: GGLWELayout = infos.gglwe_layout();
ksk_infos.rank_in = Rank(pairs);
let (data, scratch) = self.take_gglwe(infos);
let (data, scratch) = self.take_gglwe(&ksk_infos);
(GLWETensorKey(data), scratch)
}
@@ -377,7 +377,7 @@ where
let pairs: u32 = (((infos.rank_out().0 + 1) * infos.rank_out().0) >> 1).max(1);
let mut ksk_infos: GGLWELayout = infos.gglwe_layout();
ksk_infos.rank_in = Rank(pairs);
let (data, scratch) = self.take_gglwe_prepared(module, infos);
let (data, scratch) = self.take_gglwe_prepared(module, &ksk_infos);
(GLWETensorKeyPrepared(data), scratch)
}
}

9
poulpy-core/src/tests/mod.rs
View File

@@ -4,10 +4,10 @@ pub mod test_suite;
mod serialization;
#[cfg(test)]
#[cfg(all(feature = "enable-avx", target_arch = "x86_64"))]
mod poulpy_core {
use poulpy_hal::backend_test_suite;
#[cfg(any(target_arch = "x86_64", target_arch = "x86"))]
backend_test_suite!(
mod cpu_avx,
backend = poulpy_cpu_avx::FFT64Avx,
@@ -69,8 +69,13 @@ mod poulpy_core {
lwe_to_glwe => crate::tests::test_suite::test_lwe_to_glwe,
}
);
}
#[cfg(test)]
#[cfg(not(all(feature = "enable-avx", target_arch = "x86_64")))]
mod poulpy_core {
use poulpy_hal::backend_test_suite;
#[cfg(not(any(target_arch = "x86_64", target_arch = "x86")))]
backend_test_suite!(
mod cpu_ref,
backend = poulpy_cpu_ref::FFT64Ref,