poulpy_core/noise/
glwe_ct.rs

1use poulpy_hal::{
2    api::{
3        ScratchOwnedAlloc, ScratchOwnedBorrow, SvpApplyDftToDftInplace, VecZnxBigAddInplace, VecZnxBigAddSmallInplace,
4        VecZnxBigAllocBytes, VecZnxBigNormalize, VecZnxDftAllocBytes, VecZnxDftApply, VecZnxIdftApplyConsume,
5        VecZnxNormalizeInplace, VecZnxNormalizeTmpBytes, VecZnxSubInplace,
6    },
7    layouts::{Backend, DataRef, Module, ScratchOwned},
8    oep::{ScratchOwnedAllocImpl, ScratchOwnedBorrowImpl, TakeVecZnxBigImpl, TakeVecZnxDftImpl},
9};
10
11use crate::layouts::{GLWECiphertext, GLWEPlaintext, LWEInfos, prepared::GLWESecretPrepared};
12
13impl<D: DataRef> GLWECiphertext<D> {
14    pub fn assert_noise<B, DataSk, DataPt>(
15        &self,
16        module: &Module<B>,
17        sk_prepared: &GLWESecretPrepared<DataSk, B>,
18        pt_want: &GLWEPlaintext<DataPt>,
19        max_noise: f64,
20    ) where
21        DataSk: DataRef,
22        DataPt: DataRef,
23        Module<B>: VecZnxDftAllocBytes
24            + VecZnxBigAllocBytes
25            + VecZnxDftApply<B>
26            + SvpApplyDftToDftInplace<B>
27            + VecZnxIdftApplyConsume<B>
28            + VecZnxBigAddInplace<B>
29            + VecZnxBigAddSmallInplace<B>
30            + VecZnxBigNormalize<B>
31            + VecZnxNormalizeTmpBytes
32            + VecZnxSubInplace
33            + VecZnxNormalizeInplace<B>,
34        B: Backend + TakeVecZnxDftImpl<B> + TakeVecZnxBigImpl<B> + ScratchOwnedAllocImpl<B> + ScratchOwnedBorrowImpl<B>,
35    {
36        let mut pt_have: GLWEPlaintext<Vec<u8>> = GLWEPlaintext::alloc(self);
37
38        let mut scratch: ScratchOwned<B> = ScratchOwned::alloc(GLWECiphertext::decrypt_scratch_space(module, self));
39
40        self.decrypt(module, &mut pt_have, sk_prepared, scratch.borrow());
41
42        module.vec_znx_sub_inplace(&mut pt_have.data, 0, &pt_want.data, 0);
43        module.vec_znx_normalize_inplace(self.base2k().into(), &mut pt_have.data, 0, scratch.borrow());
44
45        let noise_have: f64 = pt_have.data.std(self.base2k().into(), 0).log2();
46        assert!(noise_have <= max_noise, "{noise_have} {max_noise}");
47    }
48}
poulpy_core/noise/glwe_ct.rs

poulpy_core/noise/
glwe_ct.rs
