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

fixes to vec_znx_dft

Browse Source
This commit is contained in:
Jean-Philippe Bossuat
2025-02-13 10:58:53 +01:00
parent 7019812b65
commit 73326d49ee
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
base2k/src/vec_znx_dft.rs
View File

@@ -63,7 +63,7 @@ impl Module {
b.limbs(), b.limbs(),
a_limbs a_limbs
); );
unsafe { vec_znx_dft::vec_znx_idft_tmp_a(self.0, b.0, a_limbs as u64, a.0, a_limbs as u64) } unsafe { vec_znx_dft::vec_znx_idft_tmp_a(self.0, b.0, b.limbs() as u64, a.0, a_limbs as u64) }
} }
// Returns the size of the scratch space for [vec_znx_idft]. // Returns the size of the scratch space for [vec_znx_idft].
@@ -86,7 +86,7 @@ impl Module {
vec_znx_dft::vec_znx_dft( vec_znx_dft::vec_znx_dft(
self.0, self.0,
b.0, b.0,
a_limbs as u64, b.limbs() as u64,
a.as_ptr(), a.as_ptr(),
a_limbs as u64, a_limbs as u64,
a.n as u64, a.n as u64,
@@ -97,21 +97,21 @@ impl Module {
// b <- IDFT(a), scratch space size obtained with [vec_znx_idft_tmp_bytes]. // b <- IDFT(a), scratch space size obtained with [vec_znx_idft_tmp_bytes].
pub fn vec_znx_idft( pub fn vec_znx_idft(
&self, &self,
b_vector: &mut VecZnxBig, b: &mut VecZnxBig,
a_vector: &mut VecZnxDft, a: &mut VecZnxDft,
a_limbs: usize, a_limbs: usize,
tmp_bytes: &mut [u8], tmp_bytes: &mut [u8],
) { ) {
assert!( assert!(
b_vector.limbs() >= a_limbs, b.limbs() >= a_limbs,
"invalid c_vector: b_vector.limbs()={} < a_limbs={}", "invalid c_vector: b.limbs()={} < a_limbs={}",
b_vector.limbs(), b.limbs(),
a_limbs a_limbs
); );
assert!( assert!(
a_vector.limbs() >= a_limbs, a.limbs() >= a_limbs,
"invalid c_vector: c_vector.limbs()={} < a_limbs={}", "invalid c_vector: a.limbs()={} < a_limbs={}",
a_vector.limbs(), a.limbs(),
a_limbs a_limbs
); );
assert!( assert!(
@@ -123,9 +123,9 @@ impl Module {
unsafe { unsafe {
vec_znx_dft::vec_znx_idft( vec_znx_dft::vec_znx_idft(
self.0, self.0,
b_vector.0, b.0,
a_limbs as u64, a.limbs() as u64,
a_vector.0, a.0,
a_limbs as u64, a_limbs as u64,
tmp_bytes.as_mut_ptr(), tmp_bytes.as_mut_ptr(),
) )