mirror of
https://github.com/arnaucube/poulpy.git
synced 2026-02-10 13:16:44 +01:00
wip on gadget product
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
use crate::{
|
||||
ciphertext::Ciphertext, keys::SecretKey, parameters::Parameters, plaintext::Plaintext,
|
||||
ciphertext::{Ciphertext, GadgetCiphertext}, elem::Elem, keys::SecretKey, parameters::Parameters, plaintext::Plaintext
|
||||
};
|
||||
use base2k::{Module, SvpPPol, SvpPPolOps, VecZnxDft};
|
||||
use std::cmp::min;
|
||||
@@ -35,18 +35,18 @@ impl Parameters {
|
||||
sk: &SvpPPol,
|
||||
tmp_bytes: &mut [u8],
|
||||
) {
|
||||
decrypt_rlwe_thread_safe(self.module(), res, ct, sk, tmp_bytes)
|
||||
decrypt_rlwe_thread_safe(self.module(), &mut res.0, &ct.0, sk, tmp_bytes)
|
||||
}
|
||||
}
|
||||
|
||||
pub fn decrypt_rlwe_thread_safe(
|
||||
module: &Module,
|
||||
res: &mut Plaintext,
|
||||
ct: &Ciphertext,
|
||||
res: &mut Elem,
|
||||
a: &Elem,
|
||||
sk: &SvpPPol,
|
||||
tmp_bytes: &mut [u8],
|
||||
) {
|
||||
let limbs: usize = min(res.limbs(), ct.limbs());
|
||||
let limbs: usize = min(res.limbs(), a.limbs());
|
||||
|
||||
assert!(
|
||||
tmp_bytes.len() >= decrypt_rlwe_thread_safe_tmp_byte(module, limbs),
|
||||
@@ -61,20 +61,20 @@ pub fn decrypt_rlwe_thread_safe(
|
||||
let mut res_big: base2k::VecZnxBig = res_dft.as_vec_znx_big();
|
||||
|
||||
// res_dft <- DFT(ct[1]) * DFT(sk)
|
||||
module.svp_apply_dft(&mut res_dft, sk, &ct.0.value[1], limbs);
|
||||
module.svp_apply_dft(&mut res_dft, sk, &a.value[1], limbs);
|
||||
// res_big <- ct[1] x sk
|
||||
module.vec_znx_idft_tmp_a(&mut res_big, &mut res_dft, limbs);
|
||||
// res_big <- ct[1] x sk + ct[0]
|
||||
module.vec_znx_big_add_small_inplace(&mut res_big, &ct.0.value[0], limbs);
|
||||
module.vec_znx_big_add_small_inplace(&mut res_big, &a.value[0], limbs);
|
||||
// res <- normalize(ct[1] x sk + ct[0])
|
||||
module.vec_znx_big_normalize(
|
||||
ct.log_base2k(),
|
||||
a.log_base2k(),
|
||||
res.at_mut(0),
|
||||
&res_big,
|
||||
&mut tmp_bytes[res_dft_bytes..],
|
||||
);
|
||||
|
||||
res.0.log_base2k = ct.log_base2k();
|
||||
res.0.log_q = min(res.log_q(), ct.log_q());
|
||||
res.0.log_scale = ct.log_scale();
|
||||
res.log_base2k = a.log_base2k();
|
||||
res.log_q = min(res.log_q(), a.log_q());
|
||||
res.log_scale = a.log_scale();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user