poulpy_core/noise/
gglwe_ct.rs

1use poulpy_hal::{
2    api::{
3        ScratchOwnedAlloc, ScratchOwnedBorrow, SvpApplyDftToDftInplace, VecZnxBigAddInplace, VecZnxBigAddSmallInplace,
4        VecZnxBigAllocBytes, VecZnxBigNormalize, VecZnxDftAllocBytes, VecZnxDftApply, VecZnxIdftApplyConsume,
5        VecZnxNormalizeTmpBytes, VecZnxSubScalarInplace,
6    },
7    layouts::{Backend, DataRef, Module, ScalarZnx, ScratchOwned, ZnxZero},
8    oep::{ScratchOwnedAllocImpl, ScratchOwnedBorrowImpl, TakeVecZnxBigImpl, TakeVecZnxDftImpl},
9};
10
11use crate::layouts::{GGLWECiphertext, GGLWELayoutInfos, GLWECiphertext, GLWEPlaintext, LWEInfos, prepared::GLWESecretPrepared};
12
13impl<D: DataRef> GGLWECiphertext<D> {
14    pub fn assert_noise<B, DataSk, DataWant>(
15        &self,
16        module: &Module<B>,
17        sk: &GLWESecretPrepared<DataSk, B>,
18        pt_want: &ScalarZnx<DataWant>,
19        max_noise: f64,
20    ) where
21        DataSk: DataRef,
22        DataWant: 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            + VecZnxSubScalarInplace,
33        B: Backend + TakeVecZnxDftImpl<B> + TakeVecZnxBigImpl<B> + ScratchOwnedAllocImpl<B> + ScratchOwnedBorrowImpl<B>,
34    {
35        let digits: usize = self.digits().into();
36        let base2k: usize = self.base2k().into();
37
38        let mut scratch: ScratchOwned<B> = ScratchOwned::alloc(GLWECiphertext::decrypt_scratch_space(module, self));
39        let mut pt: GLWEPlaintext<Vec<u8>> = GLWEPlaintext::alloc(self);
40
41        (0..self.rank_in().into()).for_each(|col_i| {
42            (0..self.rows().into()).for_each(|row_i| {
43                self.at(row_i, col_i)
44                    .decrypt(module, &mut pt, sk, scratch.borrow());
45
46                module.vec_znx_sub_scalar_inplace(
47                    &mut pt.data,
48                    0,
49                    (digits - 1) + row_i * digits,
50                    pt_want,
51                    col_i,
52                );
53
54                let noise_have: f64 = pt.data.std(base2k, 0).log2();
55
56                println!("noise_have: {noise_have}");
57
58                assert!(
59                    noise_have <= max_noise,
60                    "noise_have: {noise_have} > max_noise: {max_noise}"
61                );
62
63                pt.data.zero();
64            });
65        });
66    }
67}
poulpy_core/noise/gglwe_ct.rs

poulpy_core/noise/
gglwe_ct.rs
