diff options
Diffstat (limited to 'test/kernels/write_test.cc')
| -rw-r--r-- | test/kernels/write_test.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/test/kernels/write_test.cc b/test/kernels/write_test.cc index c263fca..a136a86 100644 --- a/test/kernels/write_test.cc +++ b/test/kernels/write_test.cc @@ -36,6 +36,7 @@ KERNEL_TEST_CASE("write/int SSE2") { return kernel_write_test<CPUType::SSE2, int KERNEL_TEST_CASE("write/float SSE2") { return kernel_write_test<CPUType::SSE2, float>(); } KERNEL_TEST_CASE("write/double SSE2") { return kernel_write_test<CPUType::SSE2, double>(); } +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX2 template INTGEMM_AVX2 void kernel_write_test<CPUType::AVX2, int8_t>(); template INTGEMM_AVX2 void kernel_write_test<CPUType::AVX2, int16_t>(); template INTGEMM_AVX2 void kernel_write_test<CPUType::AVX2, int>(); @@ -46,6 +47,7 @@ KERNEL_TEST_CASE("write/int16 AVX2") { return kernel_write_test<CPUType::AVX2, i KERNEL_TEST_CASE("write/int AVX2") { return kernel_write_test<CPUType::AVX2, int>(); } KERNEL_TEST_CASE("write/float AVX2") { return kernel_write_test<CPUType::AVX2, float>(); } KERNEL_TEST_CASE("write/double AVX2") { return kernel_write_test<CPUType::AVX2, double>(); } +#endif #ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW template INTGEMM_AVX512BW void kernel_write_test<CPUType::AVX512BW, int8_t>(); |