Backend refactor (#120)

* remove spqlios, split cpu_ref and cpu_avx into different crates

* remove spqlios submodule

* update crate naming & add avx tests

poulpy-core/Cargo.toml

@@ -12,11 +12,13 @@ documentation = "https://docs.rs/poulpy"
 rug = {workspace = true}
 criterion = {workspace = true}
 poulpy-hal = {workspace = true}
-poulpy-backend = {workspace = true}
+poulpy-cpu-avx = {workspace = true}
+poulpy-cpu-ref = {workspace = true}
 itertools = {workspace = true}
 byteorder = {workspace = true}
 bytemuck = {workspace = true}
 once_cell = {workspace = true}
+paste = {workspace = true}
 
 [[bench]]
 name = "external_product_glwe_fft64"

poulpy-core/benches/external_product_glwe_fft64.rs

@@ -6,7 +6,7 @@ use std::hint::black_box;
 
 use criterion::{BenchmarkId, Criterion, criterion_group, criterion_main};
 
-use poulpy_backend::cpu_spqlios::FFT64Spqlios;
+use poulpy_cpu_ref::FFT64Ref;
 use poulpy_hal::{
     api::{ModuleNew, ScratchOwnedAlloc, ScratchOwnedBorrow},
     layouts::{Module, ScalarZnx, ScratchOwned},
@@ -26,7 +26,7 @@ fn bench_external_product_glwe_fft64(c: &mut Criterion) {
     }
 
     fn runner(p: Params) -> impl FnMut() {
-        let module: Module<FFT64Spqlios> = Module::<FFT64Spqlios>::new(1 << p.log_n);
+        let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(1 << p.log_n);
 
         let n: Degree = Degree(module.n() as u32);
         let base2k: Base2K = p.base2k;
@@ -66,7 +66,7 @@ fn bench_external_product_glwe_fft64(c: &mut Criterion) {
         let mut ct_glwe_out: GLWE<Vec<u8>> = GLWE::alloc_from_infos(&glwe_out_layout);
         let pt_rgsw: ScalarZnx<Vec<u8>> = ScalarZnx::alloc(n.into(), 1);
 
-        let mut scratch: ScratchOwned<FFT64Spqlios> = ScratchOwned::alloc(
+        let mut scratch: ScratchOwned<FFT64Ref> = ScratchOwned::alloc(
             GGSW::encrypt_sk_tmp_bytes(&module, &ggsw_layout)
                 | GLWE::encrypt_sk_tmp_bytes(&module, &glwe_in_layout)
                 | GLWE::external_product_tmp_bytes(&module, &glwe_out_layout, &glwe_in_layout, &ggsw_layout),
@@ -79,7 +79,7 @@ fn bench_external_product_glwe_fft64(c: &mut Criterion) {
         let mut sk: GLWESecret<Vec<u8>> = GLWESecret::alloc_from_infos(&glwe_in_layout);
         sk.fill_ternary_prob(0.5, &mut source_xs);
 
-        let mut sk_dft: GLWESecretPrepared<Vec<u8>, FFT64Spqlios> = GLWESecretPrepared::alloc(&module, rank);
+        let mut sk_dft: GLWESecretPrepared<Vec<u8>, FFT64Ref> = GLWESecretPrepared::alloc(&module, rank);
         sk_dft.prepare(&module, &sk);
 
         ct_ggsw.encrypt_sk(
@@ -99,7 +99,7 @@ fn bench_external_product_glwe_fft64(c: &mut Criterion) {
             scratch.borrow(),
         );
 
-        let mut ggsw_prepared: GGSWPrepared<Vec<u8>, FFT64Spqlios> = GGSWPrepared::alloc_from_infos(&module, &ct_ggsw);
+        let mut ggsw_prepared: GGSWPrepared<Vec<u8>, FFT64Ref> = GGSWPrepared::alloc_from_infos(&module, &ct_ggsw);
         ggsw_prepared.prepare(&module, &ct_ggsw, scratch.borrow());
 
         move || {
@@ -138,7 +138,7 @@ fn bench_external_product_glwe_inplace_fft64(c: &mut Criterion) {
     }
 
     fn runner(p: Params) -> impl FnMut() {
-        let module: Module<FFT64Spqlios> = Module::<FFT64Spqlios>::new(1 << p.log_n);
+        let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(1 << p.log_n);
 
         let n: Degree = Degree(module.n() as u32);
         let base2k: Base2K = p.base2k;
@@ -169,7 +169,7 @@ fn bench_external_product_glwe_inplace_fft64(c: &mut Criterion) {
         let mut ct_glwe: GLWE<Vec<u8>> = GLWE::alloc_from_infos(&glwe_layout);
         let pt_rgsw: ScalarZnx<Vec<u8>> = ScalarZnx::alloc(n.into(), 1);
 
-        let mut scratch: ScratchOwned<FFT64Spqlios> = ScratchOwned::alloc(
+        let mut scratch: ScratchOwned<FFT64Ref> = ScratchOwned::alloc(
             GGSW::encrypt_sk_tmp_bytes(&module, &ggsw_layout)
                 | GLWE::encrypt_sk_tmp_bytes(&module, &glwe_layout)
                 | GLWE::external_product_tmp_bytes(&module, &glwe_layout, &glwe_layout, &ggsw_layout),
@@ -182,7 +182,7 @@ fn bench_external_product_glwe_inplace_fft64(c: &mut Criterion) {
         let mut sk: GLWESecret<Vec<u8>> = GLWESecret::alloc_from_infos(&glwe_layout);
         sk.fill_ternary_prob(0.5, &mut source_xs);
 
-        let mut sk_dft: GLWESecretPrepared<Vec<u8>, FFT64Spqlios> = GLWESecretPrepared::alloc(&module, rank);
+        let mut sk_dft: GLWESecretPrepared<Vec<u8>, FFT64Ref> = GLWESecretPrepared::alloc(&module, rank);
         sk_dft.prepare(&module, &sk);
 
         ct_ggsw.encrypt_sk(
@@ -202,7 +202,7 @@ fn bench_external_product_glwe_inplace_fft64(c: &mut Criterion) {
             scratch.borrow(),
         );
 
-        let mut ggsw_prepared: GGSWPrepared<Vec<u8>, FFT64Spqlios> = GGSWPrepared::alloc_from_infos(&module, &ct_ggsw);
+        let mut ggsw_prepared: GGSWPrepared<Vec<u8>, FFT64Ref> = GGSWPrepared::alloc_from_infos(&module, &ct_ggsw);
         ggsw_prepared.prepare(&module, &ct_ggsw, scratch.borrow());
         move || {
             let scratch_borrow = scratch.borrow();

poulpy-core/benches/keyswitch_glwe_fft64.rs

@@ -6,7 +6,7 @@ use poulpy_core::layouts::{
 use std::{hint::black_box, time::Duration};
 
 use criterion::{BenchmarkId, Criterion, criterion_group, criterion_main};
-use poulpy_backend::cpu_spqlios::FFT64Spqlios;
+use poulpy_cpu_ref::FFT64Ref;
 use poulpy_hal::{
     api::{ModuleNew, ScratchOwnedAlloc, ScratchOwnedBorrow},
     layouts::{Module, ScratchOwned},
@@ -27,7 +27,7 @@ fn bench_keyswitch_glwe_fft64(c: &mut Criterion) {
     }
 
     fn runner(p: Params) -> impl FnMut() {
-        let module: Module<FFT64Spqlios> = Module::<FFT64Spqlios>::new(1 << p.log_n);
+        let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(1 << p.log_n);
 
         let n: Degree = Degree(module.n() as u32);
         let base2k: Base2K = p.base2k;
@@ -66,7 +66,7 @@ fn bench_keyswitch_glwe_fft64(c: &mut Criterion) {
         let mut ct_in: GLWE<Vec<u8>> = GLWE::alloc_from_infos(&glwe_in_layout);
         let mut ct_out: GLWE<Vec<u8>> = GLWE::alloc_from_infos(&glwe_out_layout);
 
-        let mut scratch: ScratchOwned<FFT64Spqlios> = ScratchOwned::alloc(
+        let mut scratch: ScratchOwned<FFT64Ref> = ScratchOwned::alloc(
             GLWESwitchingKey::encrypt_sk_tmp_bytes(&module, &gglwe_atk_layout)
                 | GLWE::encrypt_sk_tmp_bytes(&module, &glwe_in_layout)
                 | GLWE::keyswitch_tmp_bytes(
@@ -84,7 +84,7 @@ fn bench_keyswitch_glwe_fft64(c: &mut Criterion) {
         let mut sk_in: GLWESecret<Vec<u8>> = GLWESecret::alloc_from_infos(&glwe_in_layout);
         sk_in.fill_ternary_prob(0.5, &mut source_xs);
 
-        let mut sk_in_dft: GLWESecretPrepared<Vec<u8>, FFT64Spqlios> = GLWESecretPrepared::alloc(&module, rank);
+        let mut sk_in_dft: GLWESecretPrepared<Vec<u8>, FFT64Ref> = GLWESecretPrepared::alloc(&module, rank);
         sk_in_dft.prepare(&module, &sk_in);
 
         ksk.encrypt_sk(
@@ -150,7 +150,7 @@ fn bench_keyswitch_glwe_inplace_fft64(c: &mut Criterion) {
     }
 
     fn runner(p: Params) -> impl FnMut() {
-        let module: Module<FFT64Spqlios> = Module::<FFT64Spqlios>::new(1 << p.log_n);
+        let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(1 << p.log_n);
 
         let n: Degree = Degree(module.n() as u32);
         let base2k: Base2K = p.base2k;
@@ -181,7 +181,7 @@ fn bench_keyswitch_glwe_inplace_fft64(c: &mut Criterion) {
         let mut ksk: GLWESwitchingKey<Vec<u8>> = GLWESwitchingKey::alloc_from_infos(&gglwe_layout);
         let mut ct: GLWE<Vec<u8>> = GLWE::alloc_from_infos(&glwe_layout);
 
-        let mut scratch: ScratchOwned<FFT64Spqlios> = ScratchOwned::alloc(
+        let mut scratch: ScratchOwned<FFT64Ref> = ScratchOwned::alloc(
             GLWESwitchingKey::encrypt_sk_tmp_bytes(&module, &gglwe_layout)
                 | GLWE::encrypt_sk_tmp_bytes(&module, &glwe_layout)
                 | GLWE::keyswitch_tmp_bytes(&module, &glwe_layout, &glwe_layout, &gglwe_layout),
@@ -194,7 +194,7 @@ fn bench_keyswitch_glwe_inplace_fft64(c: &mut Criterion) {
         let mut sk_in: GLWESecret<Vec<u8>> = GLWESecret::alloc_from_infos(&glwe_layout);
         sk_in.fill_ternary_prob(0.5, &mut source_xs);
 
-        let mut sk_in_dft: GLWESecretPrepared<Vec<u8>, FFT64Spqlios> = GLWESecretPrepared::alloc(&module, rank);
+        let mut sk_in_dft: GLWESecretPrepared<Vec<u8>, FFT64Ref> = GLWESecretPrepared::alloc(&module, rank);
         sk_in_dft.prepare(&module, &sk_in);
 
         let mut sk_out: GLWESecret<Vec<u8>> = GLWESecret::alloc_from_infos(&glwe_layout);

poulpy-core/examples/encryption.rs

@@ -1,4 +1,3 @@
-use poulpy_backend::cpu_spqlios::FFT64Spqlios;
 use poulpy_core::{
     GLWESub, SIGMA,
     layouts::{
@@ -6,6 +5,7 @@ use poulpy_core::{
         prepared::GLWESecretPrepared,
     },
 };
+use poulpy_cpu_ref::FFT64Ref;
 use poulpy_hal::{
     api::{ModuleNew, ScratchOwnedAlloc, ScratchOwnedBorrow, VecZnxFillUniform},
     layouts::{Module, ScratchOwned},
@@ -31,7 +31,7 @@ fn main() {
     let rank: Rank = Rank(1);
 
     // Instantiate Module (DFT Tables)
-    let module: Module<FFT64Spqlios> = Module::<FFT64Spqlios>::new(n.0 as u64);
+    let module: Module<FFT64Ref> = Module::<FFT64Ref>::new(n.0 as u64);
 
     let glwe_ct_infos: GLWELayout = GLWELayout {
         n,
@@ -53,7 +53,7 @@ fn main() {
     let mut source_xa: Source = Source::new([2u8; 32]);
 
     // Scratch space
-    let mut scratch: ScratchOwned<FFT64Spqlios> = ScratchOwned::alloc(
+    let mut scratch: ScratchOwned<FFT64Ref> = ScratchOwned::alloc(
         GLWE::encrypt_sk_tmp_bytes(&module, &glwe_ct_infos) | GLWE::decrypt_tmp_bytes(&module, &glwe_ct_infos),
     );
 
@@ -62,7 +62,7 @@ fn main() {
     sk.fill_ternary_prob(0.5, &mut source_xs);
 
     // Backend-prepared secret
-    let mut sk_prepared: GLWESecretPrepared<Vec<u8>, FFT64Spqlios> = GLWESecretPrepared::alloc(&module, rank);
+    let mut sk_prepared: GLWESecretPrepared<Vec<u8>, FFT64Ref> = GLWESecretPrepared::alloc(&module, rank);
     sk_prepared.prepare(&module, &sk);
 
     // Uniform plaintext

poulpy-core/src/tests/mod.rs

@@ -3,131 +3,133 @@ pub mod test_suite;
 #[cfg(test)]
 mod serialization;
 
-#[allow(unused_imports)]
-use poulpy_hal::backend_test_suite;
-
 #[cfg(test)]
-backend_test_suite!(
- mod cpu_ref,
- backend = poulpy_backend::cpu_fft64_ref::FFT64Ref,
- size = 1<<8,
- tests = {
- //GLWE Encryption
- glwe_encrypt_sk => crate::tests::test_suite::encryption::test_glwe_encrypt_sk,
- glwe_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_glwe_compressed_encrypt_sk,
- glwe_encrypt_zero_sk => crate::tests::test_suite::encryption::test_glwe_encrypt_zero_sk,
- glwe_encrypt_pk => crate::tests::test_suite::encryption::test_glwe_encrypt_pk,
-// GLWE Base2k Conversion
-glwe_base2k_conv => crate::tests::test_suite::test_glwe_base2k_conversion,
-// GLWE Keyswitch
-glwe_keyswitch => crate::tests::test_suite::keyswitch::test_glwe_keyswitch,
-glwe_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_glwe_keyswitch_inplace,
-// GLWE Automorphism
-glwe_automorphism => crate::tests::test_suite::automorphism::test_glwe_automorphism,
-glwe_automorphism_inplace => crate::tests::test_suite::automorphism::test_glwe_automorphism_inplace,
-// GLWE External Product
-glwe_external_product => crate::tests::test_suite::external_product::test_glwe_external_product,
-glwe_external_product_inplace => crate::tests::test_suite::external_product::test_glwe_external_product_inplace,
-// GLWE Trace
-glwe_trace_inplace => crate::tests::test_suite::test_glwe_trace_inplace,
-glwe_packing => crate::tests::test_suite::test_glwe_packing,
-glwe_packer => crate::tests::test_suite::test_glwe_packer,
-// GGLWE Encryption
-gglwe_switching_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_switching_key_encrypt_sk,
-gglwe_switching_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_switching_key_compressed_encrypt_sk,
-gglwe_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_compressed_encrypt_sk,
-gglwe_automorphism_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_automorphism_key_encrypt_sk,
-gglwe_automorphism_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_automorphism_key_compressed_encrypt_sk,
-gglwe_tensor_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_tensor_key_encrypt_sk,
-gglwe_tensor_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_tensor_key_compressed_encrypt_sk,
-gglwe_to_ggsw_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_to_ggsw_key_encrypt_sk,
-// GGLWE Keyswitching
-gglwe_switching_key_keyswitch => crate::tests::test_suite::keyswitch::test_gglwe_switching_key_keyswitch,
-gglwe_switching_key_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_gglwe_switching_key_keyswitch_inplace,
-// GGLWE External Product
-gglwe_switching_key_external_product => crate::tests::test_suite::external_product::test_gglwe_switching_key_external_product,
-gglwe_switching_key_external_product_inplace => crate::tests::test_suite::external_product::test_gglwe_switching_key_external_product_inplace,
-// GGLWE Automorphism
-gglwe_automorphism_key_automorphism => crate::tests::test_suite::automorphism::test_gglwe_automorphism_key_automorphism,
-gglwe_automorphism_key_automorphism_inplace => crate::tests::test_suite::automorphism::test_gglwe_automorphism_key_automorphism_inplace,
-// GGSW Encryption
-ggsw_encrypt_sk => crate::tests::test_suite::encryption::test_ggsw_encrypt_sk,
-ggsw_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_ggsw_compressed_encrypt_sk,
-// GGSW Keyswitching
-ggsw_keyswitch => crate::tests::test_suite::keyswitch::test_ggsw_keyswitch,
-ggsw_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_ggsw_keyswitch_inplace,
-// GGSW External Product
-ggsw_external_product => crate::tests::test_suite::external_product::test_ggsw_external_product,
-ggsw_external_product_inplace => crate::tests::test_suite::external_product::test_ggsw_external_product_inplace,
-// GGSW Automorphism
-ggsw_automorphism => crate::tests::test_suite::automorphism::test_ggsw_automorphism,
-ggsw_automorphism_inplace => crate::tests::test_suite::automorphism::test_ggsw_automorphism_inplace,
-// LWE
-lwe_keyswitch => crate::tests::test_suite::keyswitch::test_lwe_keyswitch,
-glwe_to_lwe => crate::tests::test_suite::test_glwe_to_lwe,
-lwe_to_glwe => crate::tests::test_suite::test_lwe_to_glwe,
-}
- );
+mod poulpy_core {
+    use poulpy_hal::backend_test_suite;
 
-#[cfg(any(target_arch = "x86_64", target_arch = "x86"))]
-#[cfg(test)]
-backend_test_suite!(
- mod cpu_avx,
- backend = poulpy_backend::cpu_fft64_avx::FFT64Avx,
- size = 1<<8,
- tests = {
- //GLWE Encryption
- glwe_encrypt_sk => crate::tests::test_suite::encryption::test_glwe_encrypt_sk,
- glwe_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_glwe_compressed_encrypt_sk,
- glwe_encrypt_zero_sk => crate::tests::test_suite::encryption::test_glwe_encrypt_zero_sk,
- glwe_encrypt_pk => crate::tests::test_suite::encryption::test_glwe_encrypt_pk,
- // GLWE Base2k Conversion
-glwe_base2k_conv => crate::tests::test_suite::test_glwe_base2k_conversion,
-// GLWE Keyswitch
-glwe_keyswitch => crate::tests::test_suite::keyswitch::test_glwe_keyswitch,
-glwe_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_glwe_keyswitch_inplace,
-// GLWE Automorphism
-glwe_automorphism => crate::tests::test_suite::automorphism::test_glwe_automorphism,
-glwe_automorphism_inplace => crate::tests::test_suite::automorphism::test_glwe_automorphism_inplace,
-// GLWE External Product
-glwe_external_product => crate::tests::test_suite::external_product::test_glwe_external_product,
-glwe_external_product_inplace => crate::tests::test_suite::external_product::test_glwe_external_product_inplace,
-// GLWE Trace
-glwe_trace_inplace => crate::tests::test_suite::test_glwe_trace_inplace,
-glwe_packing => crate::tests::test_suite::test_glwe_packing,
-glwe_packer => crate::tests::test_suite::test_glwe_packer,
-// GGLWE Encryption
-gglwe_switching_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_switching_key_encrypt_sk,
-gglwe_switching_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_switching_key_compressed_encrypt_sk,
-gglwe_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_compressed_encrypt_sk,
-gglwe_automorphism_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_automorphism_key_encrypt_sk,
-gglwe_automorphism_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_automorphism_key_compressed_encrypt_sk,
-gglwe_tensor_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_tensor_key_encrypt_sk,
-gglwe_tensor_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_tensor_key_compressed_encrypt_sk,
-// GGLWE Keyswitching
-gglwe_switching_key_keyswitch => crate::tests::test_suite::keyswitch::test_gglwe_switching_key_keyswitch,
-gglwe_switching_key_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_gglwe_switching_key_keyswitch_inplace,
-// GGLWE External Product
-gglwe_switching_key_external_product => crate::tests::test_suite::external_product::test_gglwe_switching_key_external_product,
-gglwe_switching_key_external_product_inplace => crate::tests::test_suite::external_product::test_gglwe_switching_key_external_product_inplace,
-// GGLWE Automorphism
-gglwe_automorphism_key_automorphism => crate::tests::test_suite::automorphism::test_gglwe_automorphism_key_automorphism,
-gglwe_automorphism_key_automorphism_inplace => crate::tests::test_suite::automorphism::test_gglwe_automorphism_key_automorphism_inplace,
-// GGSW Encryption
-ggsw_encrypt_sk => crate::tests::test_suite::encryption::test_ggsw_encrypt_sk,
-ggsw_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_ggsw_compressed_encrypt_sk,
-// GGSW Keyswitching
-ggsw_keyswitch => crate::tests::test_suite::keyswitch::test_ggsw_keyswitch,
-ggsw_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_ggsw_keyswitch_inplace,
-// GGSW External Product
-ggsw_external_product => crate::tests::test_suite::external_product::test_ggsw_external_product,
-ggsw_external_product_inplace => crate::tests::test_suite::external_product::test_ggsw_external_product_inplace,
-// GGSW Automorphism
-ggsw_automorphism => crate::tests::test_suite::automorphism::test_ggsw_automorphism,
-ggsw_automorphism_inplace => crate::tests::test_suite::automorphism::test_ggsw_automorphism_inplace,
-// LWE
-lwe_keyswitch => crate::tests::test_suite::keyswitch::test_lwe_keyswitch,
-glwe_to_lwe => crate::tests::test_suite::test_glwe_to_lwe,
-lwe_to_glwe => crate::tests::test_suite::test_lwe_to_glwe,
+    #[cfg(any(target_arch = "x86_64", target_arch = "x86"))]
+    backend_test_suite!(
+    mod cpu_avx,
+    backend = poulpy_cpu_avx::FFT64Avx,
+    size = 1<<8,
+    tests = {
+        //GLWE Encryption
+        glwe_encrypt_sk => crate::tests::test_suite::encryption::test_glwe_encrypt_sk,
+        glwe_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_glwe_compressed_encrypt_sk,
+        glwe_encrypt_zero_sk => crate::tests::test_suite::encryption::test_glwe_encrypt_zero_sk,
+        glwe_encrypt_pk => crate::tests::test_suite::encryption::test_glwe_encrypt_pk,
+        // GLWE Base2k Conversion
+        glwe_base2k_conv => crate::tests::test_suite::test_glwe_base2k_conversion,
+        // GLWE Keyswitch
+        glwe_keyswitch => crate::tests::test_suite::keyswitch::test_glwe_keyswitch,
+        glwe_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_glwe_keyswitch_inplace,
+        // GLWE Automorphism
+        glwe_automorphism => crate::tests::test_suite::automorphism::test_glwe_automorphism,
+        glwe_automorphism_inplace => crate::tests::test_suite::automorphism::test_glwe_automorphism_inplace,
+        // GLWE External Product
+        glwe_external_product => crate::tests::test_suite::external_product::test_glwe_external_product,
+        glwe_external_product_inplace => crate::tests::test_suite::external_product::test_glwe_external_product_inplace,
+        // GLWE Trace
+        glwe_trace_inplace => crate::tests::test_suite::test_glwe_trace_inplace,
+        glwe_packing => crate::tests::test_suite::test_glwe_packing,
+        glwe_packer => crate::tests::test_suite::test_glwe_packer,
+        // GGLWE Encryption
+        gglwe_switching_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_switching_key_encrypt_sk,
+        gglwe_switching_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_switching_key_compressed_encrypt_sk,
+        gglwe_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_compressed_encrypt_sk,
+        gglwe_automorphism_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_automorphism_key_encrypt_sk,
+        gglwe_automorphism_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_automorphism_key_compressed_encrypt_sk,
+        gglwe_tensor_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_tensor_key_encrypt_sk,
+        gglwe_tensor_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_tensor_key_compressed_encrypt_sk,
+        gglwe_to_ggsw_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_to_ggsw_key_encrypt_sk,
+        // GGLWE Keyswitching
+        gglwe_switching_key_keyswitch => crate::tests::test_suite::keyswitch::test_gglwe_switching_key_keyswitch,
+        gglwe_switching_key_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_gglwe_switching_key_keyswitch_inplace,
+        // GGLWE External Product
+        gglwe_switching_key_external_product => crate::tests::test_suite::external_product::test_gglwe_switching_key_external_product,
+        gglwe_switching_key_external_product_inplace => crate::tests::test_suite::external_product::test_gglwe_switching_key_external_product_inplace,
+        // GGLWE Automorphism
+        gglwe_automorphism_key_automorphism => crate::tests::test_suite::automorphism::test_gglwe_automorphism_key_automorphism,
+        gglwe_automorphism_key_automorphism_inplace => crate::tests::test_suite::automorphism::test_gglwe_automorphism_key_automorphism_inplace,
+        // GGSW Encryption
+        ggsw_encrypt_sk => crate::tests::test_suite::encryption::test_ggsw_encrypt_sk,
+        ggsw_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_ggsw_compressed_encrypt_sk,
+        // GGSW Keyswitching
+        ggsw_keyswitch => crate::tests::test_suite::keyswitch::test_ggsw_keyswitch,
+        ggsw_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_ggsw_keyswitch_inplace,
+        // GGSW External Product
+        ggsw_external_product => crate::tests::test_suite::external_product::test_ggsw_external_product,
+        ggsw_external_product_inplace => crate::tests::test_suite::external_product::test_ggsw_external_product_inplace,
+        // GGSW Automorphism
+        ggsw_automorphism => crate::tests::test_suite::automorphism::test_ggsw_automorphism,
+        ggsw_automorphism_inplace => crate::tests::test_suite::automorphism::test_ggsw_automorphism_inplace,
+        // LWE
+        lwe_keyswitch => crate::tests::test_suite::keyswitch::test_lwe_keyswitch,
+        glwe_to_lwe => crate::tests::test_suite::test_glwe_to_lwe,
+        lwe_to_glwe => crate::tests::test_suite::test_lwe_to_glwe,
+    }
+    );
+
+    #[cfg(not(any(target_arch = "x86_64", target_arch = "x86")))]
+    backend_test_suite!(
+    mod cpu_ref,
+    backend = poulpy_cpu_ref::FFT64Ref,
+    size = 1<<8,
+    tests = {
+        //GLWE Encryption
+        glwe_encrypt_sk => crate::tests::test_suite::encryption::test_glwe_encrypt_sk,
+        glwe_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_glwe_compressed_encrypt_sk,
+        glwe_encrypt_zero_sk => crate::tests::test_suite::encryption::test_glwe_encrypt_zero_sk,
+        glwe_encrypt_pk => crate::tests::test_suite::encryption::test_glwe_encrypt_pk,
+        // GLWE Base2k Conversion
+        glwe_base2k_conv => crate::tests::test_suite::test_glwe_base2k_conversion,
+        // GLWE Keyswitch
+        glwe_keyswitch => crate::tests::test_suite::keyswitch::test_glwe_keyswitch,
+        glwe_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_glwe_keyswitch_inplace,
+        // GLWE Automorphism
+        glwe_automorphism => crate::tests::test_suite::automorphism::test_glwe_automorphism,
+        glwe_automorphism_inplace => crate::tests::test_suite::automorphism::test_glwe_automorphism_inplace,
+        // GLWE External Product
+        glwe_external_product => crate::tests::test_suite::external_product::test_glwe_external_product,
+        glwe_external_product_inplace => crate::tests::test_suite::external_product::test_glwe_external_product_inplace,
+        // GLWE Trace
+        glwe_trace_inplace => crate::tests::test_suite::test_glwe_trace_inplace,
+        glwe_packing => crate::tests::test_suite::test_glwe_packing,
+        glwe_packer => crate::tests::test_suite::test_glwe_packer,
+        // GGLWE Encryption
+        gglwe_switching_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_switching_key_encrypt_sk,
+        gglwe_switching_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_switching_key_compressed_encrypt_sk,
+        gglwe_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_compressed_encrypt_sk,
+        gglwe_automorphism_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_automorphism_key_encrypt_sk,
+        gglwe_automorphism_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_automorphism_key_compressed_encrypt_sk,
+        gglwe_tensor_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_tensor_key_encrypt_sk,
+        gglwe_tensor_key_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_tensor_key_compressed_encrypt_sk,
+        gglwe_to_ggsw_key_encrypt_sk => crate::tests::test_suite::encryption::test_gglwe_to_ggsw_key_encrypt_sk,
+        // GGLWE Keyswitching
+        gglwe_switching_key_keyswitch => crate::tests::test_suite::keyswitch::test_gglwe_switching_key_keyswitch,
+        gglwe_switching_key_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_gglwe_switching_key_keyswitch_inplace,
+        // GGLWE External Product
+        gglwe_switching_key_external_product => crate::tests::test_suite::external_product::test_gglwe_switching_key_external_product,
+        gglwe_switching_key_external_product_inplace => crate::tests::test_suite::external_product::test_gglwe_switching_key_external_product_inplace,
+        // GGLWE Automorphism
+        gglwe_automorphism_key_automorphism => crate::tests::test_suite::automorphism::test_gglwe_automorphism_key_automorphism,
+        gglwe_automorphism_key_automorphism_inplace => crate::tests::test_suite::automorphism::test_gglwe_automorphism_key_automorphism_inplace,
+        // GGSW Encryption
+        ggsw_encrypt_sk => crate::tests::test_suite::encryption::test_ggsw_encrypt_sk,
+        ggsw_compressed_encrypt_sk => crate::tests::test_suite::encryption::test_ggsw_compressed_encrypt_sk,
+        // GGSW Keyswitching
+        ggsw_keyswitch => crate::tests::test_suite::keyswitch::test_ggsw_keyswitch,
+        ggsw_keyswitch_inplace => crate::tests::test_suite::keyswitch::test_ggsw_keyswitch_inplace,
+        // GGSW External Product
+        ggsw_external_product => crate::tests::test_suite::external_product::test_ggsw_external_product,
+        ggsw_external_product_inplace => crate::tests::test_suite::external_product::test_ggsw_external_product_inplace,
+        // GGSW Automorphism
+        ggsw_automorphism => crate::tests::test_suite::automorphism::test_ggsw_automorphism,
+        ggsw_automorphism_inplace => crate::tests::test_suite::automorphism::test_ggsw_automorphism_inplace,
+        // LWE
+        lwe_keyswitch => crate::tests::test_suite::keyswitch::test_lwe_keyswitch,
+        glwe_to_lwe => crate::tests::test_suite::test_glwe_to_lwe,
+        lwe_to_glwe => crate::tests::test_suite::test_lwe_to_glwe,
+    }
+    );
 }
- );