poulpy_ckks/default/
conjugate.rs

1use anyhow::Result;
2use poulpy_core::{
3    GLWEAutomorphism, GLWEShift,
4    layouts::{GGLWEInfos, GGLWEPreparedToBackendRef, GLWEInfos, GLWEToBackendMut, GLWEToBackendRef, GetGaloisElement, LWEInfos},
5};
6use poulpy_hal::layouts::{Backend, ScratchArena};
7
8use crate::{CKKSInfos, SetCKKSInfos, checked_log_budget_sub, ckks_offset_unary};
9
10pub trait CKKSConjugateDefault<BE: Backend> {
11    fn ckks_conjugate_tmp_bytes_default<C, K>(&self, ct_infos: &C, key_infos: &K) -> usize
12    where
13        C: GLWEInfos,
14        K: GGLWEInfos,
15        Self: GLWEAutomorphism<BE>,
16    {
17        self.glwe_automorphism_tmp_bytes(ct_infos, ct_infos, key_infos)
18    }
19
20    fn ckks_conjugate_into_default<Dst, Src, K>(
21        &self,
22        dst: &mut Dst,
23        src: &Src,
24        key: &K,
25        scratch: &mut ScratchArena<'_, BE>,
26    ) -> Result<()>
27    where
28        Self: GLWEAutomorphism<BE> + GLWEShift<BE>,
29        Dst: GLWEToBackendMut<BE> + GLWEInfos + LWEInfos + CKKSInfos + SetCKKSInfos,
30        Src: GLWEToBackendRef<BE> + GLWEInfos + LWEInfos + CKKSInfos,
31        K: GetGaloisElement + GGLWEPreparedToBackendRef<BE> + GGLWEInfos,
32    {
33        let offset = ckks_offset_unary(dst, src);
34        if offset != 0 {
35            self.glwe_lsh(dst, src, offset, scratch);
36            self.glwe_automorphism_assign(dst, key, dst.size() + key.dsize().as_usize(), scratch);
37        } else {
38            self.glwe_automorphism(dst, src, key, src.size() + key.dsize().as_usize(), scratch);
39        }
40
41        dst.set_meta(src.meta());
42        dst.set_log_budget(checked_log_budget_sub("conjugate", dst.log_budget(), offset)?);
43        Ok(())
44    }
45
46    fn ckks_conjugate_assign_default<Dst, K>(&self, dst: &mut Dst, key: &K, scratch: &mut ScratchArena<'_, BE>) -> Result<()>
47    where
48        Self: GLWEAutomorphism<BE>,
49        Dst: GLWEToBackendMut<BE> + GLWEInfos,
50        K: GetGaloisElement + GGLWEPreparedToBackendRef<BE> + GGLWEInfos,
51    {
52        self.glwe_automorphism_assign(dst, key, dst.size() + key.dsize().as_usize(), scratch);
53        Ok(())
54    }
55}
poulpy_ckks/default/conjugate.rs

poulpy_ckks/default/
conjugate.rs
