github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-10 07:08:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(shortint): fix some wopbs function signature

Browse Source
This commit is contained in:
Arthur Meyre
2023-04-28 18:17:43 +02:00
parent 871d4aea17
commit bc6bbe66d9
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tfhe/benches/shortint/bench.rs
View File

@@ -315,7 +315,7 @@ fn _bench_wopbs_param_message_8_norm2_5(c: &mut Criterion) {
bench_group.bench_function(&id, |b| {
b.iter(|| {
let _ = wopbs_key.programmable_bootstrapping_native_crt(&mut ct, vec_lut.as_ref());
let _ = wopbs_key.programmable_bootstrapping_native_crt(&mut ct, &vec_lut);
})
});

4
tfhe/src/shortint/wopbs/mod.rs
View File

@@ -450,11 +450,11 @@ impl WopbsKey {
pub fn programmable_bootstrapping_native_crt<OpOrder: PBSOrderMarker>(
&self,
ct_in: &mut CiphertextBase<OpOrder>,
lut: &WopbsLUTBase,
lut: &ShortintWopbsLUT,
) -> CiphertextBase<OpOrder> {
ShortintEngine::with_thread_local_mut(|engine| {
engine
.programmable_bootstrapping_native_crt(self, ct_in, lut)
.programmable_bootstrapping_native_crt(self, ct_in, lut.as_ref())
.unwrap()
})
}

2
tfhe/src/shortint/wopbs/test.rs
View File

@@ -99,7 +99,7 @@ fn generate_lut_modulus_not_power_of_two(params: (Parameters, Parameters)) {
let mut ct = cks.encrypt_native_crt(m as u64, message_modulus.0 as u8);
let lut = wopbs_key.generate_lut_native_crt(&ct, |x| (x * x) % message_modulus.0 as u64);
let ct_res = wopbs_key.programmable_bootstrapping_native_crt(&mut ct, lut.as_ref());
let ct_res = wopbs_key.programmable_bootstrapping_native_crt(&mut ct, &lut);
let res = cks.decrypt_message_native_crt(&ct_res, message_modulus.0 as u8);
assert_eq!(res as usize, (m * m) % message_modulus.0);
}