diff options
Diffstat (limited to 'test/kernels/quantize_test.cc')
-rw-r--r-- | test/kernels/quantize_test.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/test/kernels/quantize_test.cc b/test/kernels/quantize_test.cc index ae3c068..93280f7 100644 --- a/test/kernels/quantize_test.cc +++ b/test/kernels/quantize_test.cc @@ -28,8 +28,10 @@ void kernel_quantize_test() { template INTGEMM_SSE2 void kernel_quantize_test<CPUType::SSE2>(); KERNEL_TEST_CASE("quantize SSE2") { return kernel_quantize_test<CPUType::SSE2>(); } +#ifdef INTGEMM_COMPILER_SUPPORTS_AVX2 template INTGEMM_AVX2 void kernel_quantize_test<CPUType::AVX2>(); KERNEL_TEST_CASE("quantize AVX2") { return kernel_quantize_test<CPUType::AVX2>(); } +#endif #ifdef INTGEMM_COMPILER_SUPPORTS_AVX512BW template INTGEMM_AVX512BW void kernel_quantize_test<CPUType::AVX512BW>(); |