diff options
Diffstat (limited to 'test/kernels/relu_test.cc')
-rw-r--r-- | test/kernels/relu_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/kernels/relu_test.cc b/test/kernels/relu_test.cc index 25a212a..c291dea 100644 --- a/test/kernels/relu_test.cc +++ b/test/kernels/relu_test.cc @@ -46,7 +46,7 @@ KERNEL_TEST_CASE("relu/int AVX2") { return kernel_relu_test<CPUType::AVX2, int>( KERNEL_TEST_CASE("relu/float AVX2") { return kernel_relu_test<CPUType::AVX2, float>(); } KERNEL_TEST_CASE("relu/double AVX2") { return kernel_relu_test<CPUType::AVX2, double>(); } -#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512 +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW template INTGEMM_AVX512BW void kernel_relu_test<CPUType::AVX512BW, int8_t>(); template INTGEMM_AVX512BW void kernel_relu_test<CPUType::AVX512BW, int16_t>(); template INTGEMM_AVX512BW void kernel_relu_test<CPUType::AVX512BW, int>(); |