diff options
author | Kenneth Heafield <github@kheafield.com> | 2020-11-18 00:10:45 +0300 |
---|---|---|
committer | Kenneth Heafield <github@kheafield.com> | 2020-11-18 00:10:45 +0300 |
commit | d85e98ddceb02989551170b993b0f27fc5a109eb (patch) | |
tree | cd120c92bf17a279886baa01bc35cb61ab061577 /test/kernels/add_bias_test.cc | |
parent | 8f28282c3bd854922da638024d2659be52e892e9 (diff) |
Make AVX2 optional if compiler doesn't support it
Mostly for emscripten
Diffstat (limited to 'test/kernels/add_bias_test.cc')
-rw-r--r-- | test/kernels/add_bias_test.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/test/kernels/add_bias_test.cc b/test/kernels/add_bias_test.cc index 492c669..b9e5fd9 100644 --- a/test/kernels/add_bias_test.cc +++ b/test/kernels/add_bias_test.cc @@ -37,6 +37,7 @@ KERNEL_TEST_CASE("add_bias/int SSE2") { return kernel_add_bias_test<CPUType::SSE KERNEL_TEST_CASE("add_bias/float SSE2") { return kernel_add_bias_test<CPUType::SSE2, float>(); } KERNEL_TEST_CASE("add_bias/double SSE2") { return kernel_add_bias_test<CPUType::SSE2, double>(); } +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX2 template INTGEMM_AVX2 void kernel_add_bias_test<CPUType::AVX2, int8_t>(); template INTGEMM_AVX2 void kernel_add_bias_test<CPUType::AVX2, int16_t>(); template INTGEMM_AVX2 void kernel_add_bias_test<CPUType::AVX2, int>(); @@ -47,6 +48,7 @@ KERNEL_TEST_CASE("add_bias/int16 AVX2") { return kernel_add_bias_test<CPUType::A KERNEL_TEST_CASE("add_bias/int AVX2") { return kernel_add_bias_test<CPUType::AVX2, int>(); } KERNEL_TEST_CASE("add_bias/float AVX2") { return kernel_add_bias_test<CPUType::AVX2, float>(); } KERNEL_TEST_CASE("add_bias/double AVX2") { return kernel_add_bias_test<CPUType::AVX2, double>(); } +#endif #ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW template INTGEMM_AVX512BW void kernel_add_bias_test<CPUType::AVX512BW, int8_t>(); |