Fix BF16 detection in x86_64 ukernel cmake

`_mm512_cvtneps_pbh` was used incorrectly in IREE_UK_BUILD_X86_64_AVX512_BF16 cmake test
iree-org · Jul 23, 2024 · 2f99e30 · 2f99e30
1 parent 0934526
commit 2f99e30
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/runtime/src/iree/builtins/ukernel/arch/x86_64/CMakeLists.txt b/runtime/src/iree/builtins/ukernel/arch/x86_64/CMakeLists.txt
@@ -316,8 +316,8 @@ if(IREE_UK_TRY_X86_64_AVX512_BF16)
   string(JOIN "\n" IREE_UK_BUILD_X86_64_AVX512_BF16_TEST
     "#include <immintrin.h>"
     "int main() {"
-    "  __m256bh a;"
-    "  _mm512_cvtneps_pbh(a);"
+    "  __m512 a;"
+    "  __m256bh b = _mm512_cvtneps_pbh(a);"
     "  return 0;"
     "}"
   )