poulpy/poulpy-cpu-avx/src/tests.rs
poulpy-cpu-avx/src/tests.rsRaw
1use poulpy_hal::{api::ModuleNew, layouts::Module, test_suite::convolution::test_bivariate_tensoring};
2
3use crate::FFT64Avx;
4
5#[cfg(test)]
6mod poulpy_cpu_avx {
7 use poulpy_hal::{backend_test_suite, cross_backend_test_suite};
8
9 cross_backend_test_suite! {
10 mod vec_znx,
11 backend_ref = poulpy_cpu_ref::FFT64Ref,
12 backend_test = crate::FFT64Avx,
13 size = 1<<8,
14 base2k = 12,
15 tests = {
16 test_vec_znx_add => poulpy_hal::test_suite::vec_znx::test_vec_znx_add,
17 test_vec_znx_add_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_add_inplace,
18 test_vec_znx_add_scalar => poulpy_hal::test_suite::vec_znx::test_vec_znx_add_scalar,
19 test_vec_znx_add_scalar_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_add_scalar_inplace,
20 test_vec_znx_sub => poulpy_hal::test_suite::vec_znx::test_vec_znx_sub,
21 test_vec_znx_sub_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_sub_inplace,
22 test_vec_znx_sub_negate_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_sub_negate_inplace,
23 test_vec_znx_sub_scalar => poulpy_hal::test_suite::vec_znx::test_vec_znx_sub_scalar,
24 test_vec_znx_sub_scalar_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_sub_scalar_inplace,
25 test_vec_znx_rsh => poulpy_hal::test_suite::vec_znx::test_vec_znx_rsh,
26 test_vec_znx_rsh_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_rsh_inplace,
27 test_vec_znx_lsh => poulpy_hal::test_suite::vec_znx::test_vec_znx_lsh,
28 test_vec_znx_lsh_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_lsh_inplace,
29 test_vec_znx_negate => poulpy_hal::test_suite::vec_znx::test_vec_znx_negate,
30 test_vec_znx_negate_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_negate_inplace,
31 test_vec_znx_rotate => poulpy_hal::test_suite::vec_znx::test_vec_znx_rotate,
32 test_vec_znx_rotate_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_rotate_inplace,
33 test_vec_znx_automorphism => poulpy_hal::test_suite::vec_znx::test_vec_znx_automorphism,
34 test_vec_znx_automorphism_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_automorphism_inplace,
35 test_vec_znx_mul_xp_minus_one => poulpy_hal::test_suite::vec_znx::test_vec_znx_mul_xp_minus_one,
36 test_vec_znx_mul_xp_minus_one_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_mul_xp_minus_one_inplace,
37 test_vec_znx_normalize => poulpy_hal::test_suite::vec_znx::test_vec_znx_normalize,
38 test_vec_znx_normalize_inplace => poulpy_hal::test_suite::vec_znx::test_vec_znx_normalize_inplace,
39 test_vec_znx_switch_ring => poulpy_hal::test_suite::vec_znx::test_vec_znx_switch_ring,
40 test_vec_znx_split_ring => poulpy_hal::test_suite::vec_znx::test_vec_znx_split_ring,
41 test_vec_znx_copy => poulpy_hal::test_suite::vec_znx::test_vec_znx_copy,
42 }
43 }
44 cross_backend_test_suite! {
45 mod svp,
46 backend_ref = poulpy_cpu_ref::FFT64Ref,
47 backend_test = crate::FFT64Avx,
48 size = 1<<8,
49 base2k = 12,
50 tests = {
51 test_svp_apply_dft_to_dft => poulpy_hal::test_suite::svp::test_svp_apply_dft_to_dft,
52 test_svp_apply_dft_to_dft_inplace => poulpy_hal::test_suite::svp::test_svp_apply_dft_to_dft_inplace,
53 }
54 }
55 cross_backend_test_suite! {
56 mod vec_znx_big,
57 backend_ref = poulpy_cpu_ref::FFT64Ref,
58 backend_test = crate::FFT64Avx,
59 size = 1<<8,
60 base2k = 12,
61 tests = {
62 test_vec_znx_big_add => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_add,
63 test_vec_znx_big_add_inplace => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_add_inplace,
64 test_vec_znx_big_add_small => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_add_small,
65 test_vec_znx_big_add_small_inplace => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_add_small_inplace,
66 test_vec_znx_big_sub => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_sub,
67 test_vec_znx_big_sub_inplace => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_sub_inplace,
68 test_vec_znx_big_automorphism => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_automorphism,
69 test_vec_znx_big_automorphism_inplace => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_automorphism_inplace,
70 test_vec_znx_big_negate => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_negate,
71 test_vec_znx_big_negate_inplace => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_negate_inplace,
72 test_vec_znx_big_normalize => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_normalize,
73 test_vec_znx_big_sub_negate_inplace => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_sub_negate_inplace,
74 test_vec_znx_big_sub_small_a => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_sub_small_a,
75 test_vec_znx_big_sub_small_a_inplace => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_sub_small_a_inplace,
76 test_vec_znx_big_sub_small_b => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_sub_small_b,
77 test_vec_znx_big_sub_small_b_inplace => poulpy_hal::test_suite::vec_znx_big::test_vec_znx_big_sub_small_b_inplace,
78 }
79 }
80 cross_backend_test_suite! {
81 mod vec_znx_dft,
82 backend_ref = poulpy_cpu_ref::FFT64Ref,
83 backend_test = crate::FFT64Avx,
84 size = 1<<8,
85 base2k = 12,
86 tests = {
87 test_vec_znx_dft_add => poulpy_hal::test_suite::vec_znx_dft::test_vec_znx_dft_add,
88 test_vec_znx_dft_add_inplace => poulpy_hal::test_suite::vec_znx_dft::test_vec_znx_dft_add_inplace,
89 test_vec_znx_dft_sub => poulpy_hal::test_suite::vec_znx_dft::test_vec_znx_dft_sub,
90 test_vec_znx_dft_sub_inplace => poulpy_hal::test_suite::vec_znx_dft::test_vec_znx_dft_sub_inplace,
91 test_vec_znx_dft_sub_negate_inplace => poulpy_hal::test_suite::vec_znx_dft::test_vec_znx_dft_sub_negate_inplace,
92 test_vec_znx_idft_apply => poulpy_hal::test_suite::vec_znx_dft::test_vec_znx_idft_apply,
93 test_vec_znx_idft_apply_consume => poulpy_hal::test_suite::vec_znx_dft::test_vec_znx_idft_apply_consume,
94 test_vec_znx_idft_apply_tmpa => poulpy_hal::test_suite::vec_znx_dft::test_vec_znx_idft_apply_tmpa,
95 }
96 }
97 cross_backend_test_suite! {
98 mod vmp,
99 backend_ref = poulpy_cpu_ref::FFT64Ref,
100 backend_test = crate::FFT64Avx,
101 size = 1<<8,
102 base2k = 12,
103 tests = {
104 test_vmp_apply_dft_to_dft => poulpy_hal::test_suite::vmp::test_vmp_apply_dft_to_dft,
105 test_vmp_apply_dft_to_dft_add => poulpy_hal::test_suite::vmp::test_vmp_apply_dft_to_dft_add,
106 }
107 }
108
109 backend_test_suite! {
110 mod sampling,
111 backend = crate::FFT64Avx,
112 size = 1<<12,
113 tests = {
114 test_vec_znx_fill_uniform => poulpy_hal::test_suite::vec_znx::test_vec_znx_fill_uniform,
115 test_vec_znx_fill_normal => poulpy_hal::test_suite::vec_znx::test_vec_znx_fill_normal,
116 test_vec_znx_add_normal => poulpy_hal::test_suite::vec_znx::test_vec_znx_add_normal,
117 }
118 }
119}
120
121#[test]
122fn test_convolution_fft64_avx() {
123 let module: Module<FFT64Avx> = Module::<FFT64Avx>::new(64);
124 test_bivariate_tensoring(&module);
125}