github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-08 22:28:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(core): blind rotate binding

Browse Source
This commit is contained in:
J-B Orfila
2022-12-02 15:34:23 +01:00
committed by jborfila
parent c933f6d900
commit ba984c2537
1 changed files with 29 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
tfhe/src/core_crypto/algorithms/lwe_programmable_bootstrapping.rs
View File

@@ -3,11 +3,40 @@ use crate::core_crypto::commons::numeric::CastInto;
use crate::core_crypto::commons::traits::*;
use crate::core_crypto::entities::*;
use crate::core_crypto::fft_impl::crypto::bootstrap::{bootstrap_scratch, FourierLweBootstrapKey};
use crate::core_crypto::fft_impl::crypto::ggsw::cmux_scratch;
use crate::core_crypto::fft_impl::math::fft::FftView;
use crate::core_crypto::specification::parameters::*;
use aligned_vec::CACHELINE_ALIGN;
use concrete_fft::c64;
use dyn_stack::{DynStack, SizeOverflow, StackReq};
pub fn blind_rotate<Scalar, InputCont, OutputCont, KeyCont>(
input: &LweCiphertext<InputCont>,
lut: &mut GlweCiphertext<OutputCont>,
bsk: &FourierLweBootstrapKey<KeyCont>,
fft: FftView<'_>,
stack: DynStack<'_>,
) where
// CastInto required for PBS modulus switch which returns a usize
Scalar: UnsignedTorus + CastInto<usize>,
InputCont: Container<Element = Scalar>,
OutputCont: ContainerMut<Element = Scalar>,
KeyCont: Container<Element = c64>,
{
bsk.as_view()
.blind_rotate(lut.as_mut_view(), input.as_ref(), fft, stack);
}
/// Returns the required memory for [`blind_rotate`].
pub fn blind_rotate_scratch<Scalar>(
glwe_size: GlweSize,
polynomial_size: PolynomialSize,
fft: FftView<'_>,
) -> Result<StackReq, SizeOverflow> {
StackReq::try_new_aligned::<Scalar>(glwe_size.0 * polynomial_size.0, CACHELINE_ALIGN)?
.try_and(cmux_scratch::<Scalar>(glwe_size, polynomial_size, fft)?)
}
pub fn programmable_bootstrap_lwe_ciphertext<Scalar, InputCont, OutputCont, AccCont, KeyCont>(
input: &LweCiphertext<InputCont>,
output: &mut LweCiphertext<OutputCont>,