From 669450c4f1899bb0fcb91115a61aaf0f866928e2 Mon Sep 17 00:00:00 2001
From: Jean-Philippe Bossuat <jeanphilippe.bossuat@gmail.com>
Date: Tue, 6 May 2025 14:46:26 +0200
Subject: [PATCH] added encrypt_tmp_bytes

---
 rlwe/src/encryption.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/rlwe/src/encryption.rs b/rlwe/src/encryption.rs
index f8b11c9..3b291f9 100644
--- a/rlwe/src/encryption.rs
+++ b/rlwe/src/encryption.rs
@@ -1,5 +1,6 @@
 use base2k::{
-    AddNormal, Backend, FillUniform, Module, VecZnxDftOps, ScalarZnxDftOps, ScalarZnxDftToRef, VecZnxBigOps, Scratch, VecZnx, VecZnxToMut, VecZnxToRef, ZnxInfos, FFT64
+    AddNormal, Backend, FFT64, FillUniform, Module, ScalarZnxDftOps, ScalarZnxDftToRef, Scratch, VecZnx, VecZnxBigAlloc,
+    VecZnxBigOps, VecZnxBigScratch, VecZnxDftAlloc, VecZnxDftOps, VecZnxToMut, VecZnxToRef, ZnxInfos,
 };
 
 use sampling::source::Source;
@@ -23,6 +24,10 @@ pub trait EncryptSk<B: Backend, D> {
     ) where
         P: VecZnxToRef,
         S: ScalarZnxDftToRef<B>;
+
+    fn encrypt_tmp_bytes(module: &Module<B>, size: usize) -> usize {
+        (module.vec_znx_big_normalize_tmp_bytes() | module.bytes_of_vec_znx_dft(1, size)) + module.bytes_of_vec_znx_big(1, size)
+    }
 }
 
 impl<C> EncryptSk<FFT64, CipherVecZnx<C>> for CipherVecZnx<C>