arnaucube/poulpy
1
0
Fork 0
mirror of https://github.com/arnaucube/poulpy.git synced 2026-02-10 05:06:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add take_ggsw_prepared_slice

Browse Source
This commit is contained in:
Pro7ech
2025-10-10 12:52:48 +02:00
parent 8d3fed3ae7
commit c49db0688f
1 changed files with 25 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
poulpy-core/src/scratch.rs
View File

@@ -57,6 +57,12 @@ pub trait TakeGGSWPrepared<B: Backend> {
A: GGSWInfos; A: GGSWInfos;
} }
pub trait TakeGGSWPreparedSlice<B: Backend> {
fn take_ggsw_prepared_slice<A>(&mut self, size: usize, infos: &A) -> (Vec<GGSWCiphertextPrepared<&mut [u8], B>>, &mut Self)
where
A: GGSWInfos;
}
pub trait TakeGLWESecret { pub trait TakeGLWESecret {
fn take_glwe_secret(&mut self, n: Degree, rank: Rank) -> (GLWESecret<&mut [u8]>, &mut Self); fn take_glwe_secret(&mut self, n: Degree, rank: Rank) -> (GLWESecret<&mut [u8]>, &mut Self);
} }
@@ -286,6 +292,25 @@ where
} }
} }
impl<B: Backend> TakeGGSWPreparedSlice<B> for Scratch<B>
where
Scratch<B>: TakeGGSWPrepared<B>,
{
fn take_ggsw_prepared_slice<A>(&mut self, size: usize, infos: &A) -> (Vec<GGSWCiphertextPrepared<&mut [u8], B>>, &mut Self)
where
A: GGSWInfos,
{
let mut scratch: &mut Scratch<B> = self;
let mut cts: Vec<GGSWCiphertextPrepared<&mut [u8], B>> = Vec::with_capacity(size);
for _ in 0..size {
let (ct, new_scratch) = scratch.take_ggsw_prepared(infos);
scratch = new_scratch;
cts.push(ct)
}
(cts, scratch)
}
}
impl<B: Backend> TakeGLWEPk for Scratch<B> impl<B: Backend> TakeGLWEPk for Scratch<B>
where where
Scratch<B>: TakeVecZnx, Scratch<B>: TakeVecZnx,