updated repo for publishing (#74)

2026-02-10 21:26:41 +01:00 · 2025-08-17 14:57:39 +02:00
parent 0be569eca0
commit 62eb87cc07
244 changed files with 374 additions and 539 deletions
--- a/poulpy-core/src/noise/gglwe_ct.rs
+++ b/poulpy-core/src/noise/gglwe_ct.rs
@@ -0,0 +1,73 @@
+use poulpy_backend::hal::{
+    api::{
+        ScratchOwnedAlloc, ScratchOwnedBorrow, SvpApplyInplace, VecZnxBigAddInplace, VecZnxBigAddSmallInplace,
+        VecZnxBigAllocBytes, VecZnxBigNormalize, VecZnxDftAllocBytes, VecZnxDftFromVecZnx, VecZnxDftToVecZnxBigConsume,
+        VecZnxNormalizeTmpBytes, VecZnxSubScalarInplace, ZnxZero,
+    },
+    layouts::{Backend, DataRef, Module, ScalarZnx, ScratchOwned},
+    oep::{ScratchOwnedAllocImpl, ScratchOwnedBorrowImpl, TakeVecZnxBigImpl, TakeVecZnxDftImpl},
+};
+
+use crate::layouts::{GGLWECiphertext, GLWECiphertext, GLWEPlaintext, Infos, prepared::GLWESecretPrepared};
+
+impl<D: DataRef> GGLWECiphertext<D> {
+    pub fn assert_noise<B, DataSk, DataWant>(
+        self,
+        module: &Module<B>,
+        sk: &GLWESecretPrepared<DataSk, B>,
+        pt_want: &ScalarZnx<DataWant>,
+        max_noise: f64,
+    ) where
+        DataSk: DataRef,
+        DataWant: DataRef,
+        Module<B>: VecZnxDftAllocBytes
+            + VecZnxBigAllocBytes
+            + VecZnxDftFromVecZnx<B>
+            + SvpApplyInplace<B>
+            + VecZnxDftToVecZnxBigConsume<B>
+            + VecZnxBigAddInplace<B>
+            + VecZnxBigAddSmallInplace<B>
+            + VecZnxBigNormalize<B>
+            + VecZnxNormalizeTmpBytes
+            + VecZnxSubScalarInplace,
+        B: Backend + TakeVecZnxDftImpl<B> + TakeVecZnxBigImpl<B> + ScratchOwnedAllocImpl<B> + ScratchOwnedBorrowImpl<B>,
+    {
+        let digits: usize = self.digits();
+        let basek: usize = self.basek();
+        let k: usize = self.k();
+
+        let mut scratch: ScratchOwned<B> = ScratchOwned::alloc(GLWECiphertext::decrypt_scratch_space(
+            module,
+            self.n(),
+            basek,
+            k,
+        ));
+        let mut pt: GLWEPlaintext<Vec<u8>> = GLWEPlaintext::alloc(self.n(), basek, k);
+
+        (0..self.rank_in()).for_each(|col_i| {
+            (0..self.rows()).for_each(|row_i| {
+                self.at(row_i, col_i)
+                    .decrypt(module, &mut pt, sk, scratch.borrow());
+
+                module.vec_znx_sub_scalar_inplace(
+                    &mut pt.data,
+                    0,
+                    (digits - 1) + row_i * digits,
+                    pt_want,
+                    col_i,
+                );
+
+                let noise_have: f64 = pt.data.std(basek, 0).log2();
+
+                assert!(
+                    noise_have <= max_noise,
+                    "noise_have: {} > max_noise: {}",
+                    noise_have,
+                    max_noise
+                );
+
+                pt.data.zero();
+            });
+        });
+    }
+}