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 CGGI BR for extended LUT

Browse Source
This commit is contained in:
Jean-Philippe Bossuat
2025-06-20 11:00:04 +02:00
parent 4c1a84d702
commit 52154d6f8a
3 changed files with 239 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
core/src/blind_rotation/lut.rs
View File

@@ -16,6 +16,10 @@ impl LookUpTable {
Self { data, basek, k }
}
pub fn extension_factor(&self) -> usize {
self.data.len()
}
pub fn set(&mut self, module: &Module<FFT64>, f: fn(i64) -> i64, message_modulus: usize) {
let basek: usize = self.basek;
@@ -29,7 +33,7 @@ impl LookUpTable {
let f_scaled = |x: i64| (f(x) % message_modulus as i64) * scale;
// If LUT size > module.n()
let domain_size: usize = self.data[0].n() * self.data.len();
let domain_size: usize = self.data[0].n() * self.extension_factor();
let size: usize = self.k.div_ceil(self.basek);
@@ -63,7 +67,7 @@ impl LookUpTable {
let mut tmp_bytes: Vec<u8> = alloc_aligned(lut_full.n() * size_of::<i64>());
lut_full.normalize(self.basek, 0, &mut tmp_bytes);
if self.data.len() > 1 {
if self.extension_factor() > 1 {
let mut scratch: ScratchOwned = ScratchOwned::new(module.bytes_of_vec_znx(1, size));
module.vec_znx_split(&mut self.data, 0, &lut_full, 0, scratch.borrow());
} else {