M core/src/automorphism.rs → core/src/automorphism.rs

@@ -142,6 +142,10 @@ let keyswitch: usize = GLWECiphertext::keyswitch_inplace_scratch_space(module, out_size, rank, ksk_size);
         tmp_dft + tmp_idft + idft + keyswitch
     }
 
+    pub fn automorphism_inplace_scratch_space(module: &Module<FFT64>, out_size: usize, ksk_size: usize, rank: usize) -> usize {
+        AutomorphismKey::automorphism_scratch_space(module, out_size, out_size, ksk_size, rank)
+    }
+
     pub fn external_product_scratch_space(
         module: &Module<FFT64>,
         out_size: usize,
@@ -316,6 +320,20 @@ });
         });
 
         self.p = (lhs.p * rhs.p) % (module.cyclotomic_order() as i64);
+    }
+
+    pub fn automorphism_inplace<DataRhs>(
+        &mut self,
+        module: &Module<FFT64>,
+        rhs: &AutomorphismKey<DataRhs, FFT64>,
+        scratch: &mut Scratch,
+    ) where
+        MatZnxDft<DataRhs, FFT64>: MatZnxDftToRef<FFT64>,
+    {
+        unsafe {
+            let self_ptr: *mut AutomorphismKey<DataSelf, FFT64> = self as *mut AutomorphismKey<DataSelf, FFT64>;
+            self.automorphism(&module, &*self_ptr, rhs, scratch);
+        }
     }
 
     pub fn keyswitch<DataLhs, DataRhs>(

M core/src/glwe_ciphertext.rs → core/src/glwe_ciphertext.rs

@@ -294,9 +294,9 @@ VecZnx<DataLhs>: VecZnxToRef,
         MatZnxDft<DataRhs, FFT64>: MatZnxDftToRef<FFT64>,
     {
         self.keyswitch(module, lhs, &rhs.key, scratch);
-        //(0..self.rank() + 1).for_each(|i| {
-        //    module.vec_znx_automorphism_inplace(rhs.p(), self, i);
-        //})
+        (0..self.rank() + 1).for_each(|i| {
+            module.vec_znx_automorphism_inplace(rhs.p(), self, i);
+        })
     }
 
     pub fn automorphism_inplace<DataRhs>(

M core/src/test_fft64/automorphism_key.rs → core/src/test_fft64/automorphism_key.rs

@@ -1,4 +1,4 @@
-use base2k::{FFT64, Module, ScalarZnxOps, ScalarZnxToRef, ScratchOwned, Stats, VecZnxOps, ZnxView};
+use base2k::{FFT64, Module, ScalarZnxOps, ScratchOwned, Stats, VecZnxOps};
 use sampling::source::Source;
 
 use crate::{

M core/src/test_fft64/glwe.rs → core/src/test_fft64/glwe.rs

@@ -1,6 +1,6 @@
 use base2k::{
     Decoding, Encoding, FFT64, FillUniform, Module, ScalarZnx, ScalarZnxAlloc, ScratchOwned, Stats, VecZnxOps, VecZnxToMut,
-    ZnxViewMut, ZnxZero,
+    ZnxView, ZnxViewMut, ZnxZero,
 };
 use itertools::izip;
 use sampling::source::Source;
@@ -420,7 +420,7 @@ #[test]
 fn automorphism() {
     (1..4).for_each(|rank| {
         println!("test automorphism rank: {}", rank);
-        test_automorphism(12, 12, 1, 60, 45, 60, rank, 3.2);
+        test_automorphism(12, 12, -5, 60, 45, 60, rank, 3.2);
     });
 }
 
@@ -447,7 +447,6 @@ let mut source_xs: Source = Source::new([0u8; 32]);
     let mut source_xe: Source = Source::new([0u8; 32]);
     let mut source_xa: Source = Source::new([0u8; 32]);
 
-    // Random input plaintext
     pt_want
         .data
         .fill_uniform(basek, 0, pt_want.size(), &mut source_xa);
@@ -486,14 +485,15 @@ scratch.borrow(),
     );
 
     ct_out.automorphism(&module, &ct_in, &autokey, scratch.borrow());
-
     ct_out.decrypt(&module, &mut pt_have, &sk_dft, scratch.borrow());
-
     module.vec_znx_automorphism_inplace(p, &mut pt_want, 0);
-
     module.vec_znx_sub_ab_inplace(&mut pt_have, 0, &pt_want, 0);
+    module.vec_znx_normalize_inplace(basek, &mut pt_have, 0, scratch.borrow());
 
     let noise_have: f64 = pt_have.data.std(0, basek).log2();
+
+    println!("{}", noise_have);
+
     let noise_want: f64 = noise_gglwe_product(
         module.n() as f64,
         basek,
@@ -519,7 +519,7 @@ #[test]
 fn automorphism_inplace() {
     (1..4).for_each(|rank| {
         println!("test automorphism_inplace rank: {}", rank);
-        test_automorphism_inplace(12, 12, 1, 60, 60, rank, 3.2);
+        test_automorphism_inplace(12, 12, -5, 60, 60, rank, 3.2);
     });
 }
 
@@ -575,12 +575,10 @@ scratch.borrow(),
     );
 
     ct.automorphism_inplace(&module, &autokey, scratch.borrow());
-
     ct.decrypt(&module, &mut pt_have, &sk_dft, scratch.borrow());
-
     module.vec_znx_automorphism_inplace(p, &mut pt_want, 0);
-
     module.vec_znx_sub_ab_inplace(&mut pt_have, 0, &pt_want, 0);
+    module.vec_znx_normalize_inplace(basek, &mut pt_have, 0, scratch.borrow());
 
     let noise_have: f64 = pt_have.data.std(0, basek).log2();
     let noise_want: f64 = noise_gglwe_product(