poulpy_core/noise/
gglwe_ct.rs

1use poulpy_hal::{
2    api::{
3        ScratchOwnedAlloc, ScratchOwnedBorrow, SvpApplyInplace, VecZnxBigAddInplace, VecZnxBigAddSmallInplace,
4        VecZnxBigAllocBytes, VecZnxBigNormalize, VecZnxDftAllocBytes, VecZnxDftFromVecZnx, VecZnxDftToVecZnxBigConsume,
5        VecZnxNormalizeTmpBytes, VecZnxSubScalarInplace, ZnxZero,
6    },
7    layouts::{Backend, DataRef, Module, ScalarZnx, ScratchOwned},
8    oep::{ScratchOwnedAllocImpl, ScratchOwnedBorrowImpl, TakeVecZnxBigImpl, TakeVecZnxDftImpl},
9};
10
11use crate::layouts::{GGLWECiphertext, GLWECiphertext, GLWEPlaintext, Infos, 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            + VecZnxDftFromVecZnx<B>
26            + SvpApplyInplace<B>
27            + VecZnxDftToVecZnxBigConsume<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();
36        let basek: usize = self.basek();
37        let k: usize = self.k();
38
39        let mut scratch: ScratchOwned<B> = ScratchOwned::alloc(GLWECiphertext::decrypt_scratch_space(
40            module,
41            self.n(),
42            basek,
43            k,
44        ));
45        let mut pt: GLWEPlaintext<Vec<u8>> = GLWEPlaintext::alloc(self.n(), basek, k);
46
47        (0..self.rank_in()).for_each(|col_i| {
48            (0..self.rows()).for_each(|row_i| {
49                self.at(row_i, col_i)
50                    .decrypt(module, &mut pt, sk, scratch.borrow());
51
52                module.vec_znx_sub_scalar_inplace(
53                    &mut pt.data,
54                    0,
55                    (digits - 1) + row_i * digits,
56                    pt_want,
57                    col_i,
58                );
59
60                let noise_have: f64 = pt.data.std(basek, 0).log2();
61
62                assert!(
63                    noise_have <= max_noise,
64                    "noise_have: {} > max_noise: {}",
65                    noise_have,
66                    max_noise
67                );
68
69                pt.data.zero();
70            });
71        });
72    }
73}
poulpy_core/noise/gglwe_ct.rs

poulpy_core/noise/
gglwe_ct.rs
