diff options
author | Kenneth Heafield <github@kheafield.com> | 2020-03-08 22:22:24 +0300 |
---|---|---|
committer | Kenneth Heafield <github@kheafield.com> | 2020-03-08 22:22:24 +0300 |
commit | 261a5fbcf7558fc3c2ac22b33fe0c2930d440fc3 (patch) | |
tree | f912e6a3cec9691e509d3e63eb3672ae6a357d2c /test/kernels/add_bias_test.cc | |
parent | 1323f245adff16b351b61310625b743b22e90059 (diff) |
Change to INTGEMM_COMPILER_SUPPORTS_AVX512BW and update test
Diffstat (limited to 'test/kernels/add_bias_test.cc')
-rw-r--r-- | test/kernels/add_bias_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/kernels/add_bias_test.cc b/test/kernels/add_bias_test.cc index 7b10b56..2dd4e3d 100644 --- a/test/kernels/add_bias_test.cc +++ b/test/kernels/add_bias_test.cc @@ -48,7 +48,7 @@ KERNEL_TEST_CASE("add_bias/int AVX2") { return kernel_add_bias_test<CPUType::AVX 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>(); } -#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512 +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW template INTGEMM_AVX512BW void kernel_add_bias_test<CPUType::AVX512BW, int8_t>(); template INTGEMM_AVX512BW void kernel_add_bias_test<CPUType::AVX512BW, int16_t>(); template INTGEMM_AVX512BW void kernel_add_bias_test<CPUType::AVX512BW, int>(); |