poulpy-cpu-avx 0.8.1

A crate providing concrete AVX accelerated CPU implementations of poulpy-hal through its open extension points
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use crate::{FFT64Avx, NTT4x30Avx};
use poulpy_ckks::{
    impl_ckks_add_defaults, impl_ckks_conjugate_defaults, impl_ckks_copy_defaults, impl_ckks_dft_defaults,
    impl_ckks_encapsulated_mod_up_default, impl_ckks_encryption_defaults, impl_ckks_imag_defaults, impl_ckks_mul_defaults,
    impl_ckks_neg_defaults, impl_ckks_plaintext_defaults, impl_ckks_pow2_defaults, impl_ckks_rotate_defaults,
    impl_ckks_sub_defaults,
};

impl_ckks_encapsulated_mod_up_default!(FFT64Avx);
impl_ckks_encapsulated_mod_up_default!(NTT4x30Avx);
impl_ckks_conjugate_defaults!(FFT64Avx);
impl_ckks_conjugate_defaults!(NTT4x30Avx);
impl_ckks_copy_defaults!(FFT64Avx);
impl_ckks_copy_defaults!(NTT4x30Avx);
impl_ckks_encryption_defaults!(FFT64Avx);
impl_ckks_encryption_defaults!(NTT4x30Avx);
impl_ckks_imag_defaults!(FFT64Avx);
impl_ckks_imag_defaults!(NTT4x30Avx);
impl_ckks_mul_defaults!(FFT64Avx);
impl_ckks_mul_defaults!(NTT4x30Avx);
impl_ckks_neg_defaults!(FFT64Avx);
impl_ckks_neg_defaults!(NTT4x30Avx);
impl_ckks_pow2_defaults!(FFT64Avx);
impl_ckks_pow2_defaults!(NTT4x30Avx);
impl_ckks_rotate_defaults!(FFT64Avx);
impl_ckks_rotate_defaults!(NTT4x30Avx);
// `f64` encodes through the AVX2/FMA kernels; `Quad` has no accelerated
// transform and falls back to the generic scalar table. Rust has no
// specialization, so accelerated backends list their precisions explicitly.
macro_rules! select_avx_encoding_transform {
    ($be:ty) => {
        impl ::poulpy_cpu_ref::ckks_encoding::CKKSEncodingTransform<f64> for $be {
            type Fft = crate::FFT64AvxReimTable;
        }

        impl ::poulpy_cpu_ref::ckks_encoding::CKKSEncodingTransform<poulpy_ckks::Quad> for $be {
            type Fft = ::poulpy_cpu_ref::FFT64ReimTable<poulpy_ckks::Quad>;
        }
    };
}

select_avx_encoding_transform!(FFT64Avx);
select_avx_encoding_transform!(NTT4x30Avx);

::poulpy_cpu_ref::impl_ckks_encoding!(FFT64Avx);
::poulpy_cpu_ref::impl_ckks_paco_coeff_encoding!(FFT64Avx);
::poulpy_cpu_ref::impl_ckks_ship_coeff_encoding!(FFT64Avx);
::poulpy_cpu_ref::impl_ckks_encoding!(NTT4x30Avx);
::poulpy_cpu_ref::impl_ckks_paco_coeff_encoding!(NTT4x30Avx);
::poulpy_cpu_ref::impl_ckks_ship_coeff_encoding!(NTT4x30Avx);
impl_ckks_add_defaults!(FFT64Avx);
impl_ckks_add_defaults!(NTT4x30Avx);
impl_ckks_sub_defaults!(FFT64Avx);
impl_ckks_sub_defaults!(NTT4x30Avx);
impl_ckks_plaintext_defaults!(FFT64Avx);
impl_ckks_plaintext_defaults!(NTT4x30Avx);
impl_ckks_dft_defaults!(FFT64Avx);
impl_ckks_dft_defaults!(NTT4x30Avx);