Added bytes_of to structs that can be created through the trait extension of Scratch

This commit is contained in:
Jean-Philippe Bossuat
2025-05-28 15:19:52 +02:00
parent 187756a495
commit f58d06ddf5
8 changed files with 41 additions and 0 deletions

View File

@@ -20,6 +20,11 @@ impl TensorKey<Vec<u8>, FFT64> {
});
Self { keys: keys }
}
pub fn bytes_of(module: &Module<FFT64>, basek: usize, k: usize, rows: usize, rank: usize) -> usize {
let pairs: usize = (((rank + 1) * rank) >> 1).max(1);
pairs * GLWESwitchingKey::<Vec<u8>, FFT64>::bytes_of(module, basek, k, rows, 1, rank)
}
}
impl<T, B: Backend> Infos for TensorKey<T, B> {