commit e57ccaab19ee99ae8740ec3270c608682f0b4c9b
author vient <lozko.roma@gmail.com> Thu Aug 08 17:00:01 2024 +0200
committer GitHub <noreply@github.com> Thu Aug 08 08:00:01 2024 -0700
tree f29b08499dadc83fee91b62edf17d970b66265c4
parent 7ab66ff2252b85034aa6b2b7e2eef032a97cdad3

Fix BF16 detection in x86_64 ukernel cmake (#17983)

`_mm512_cvtneps_pbh` was used incorrectly in
IREE_UK_BUILD_X86_64_AVX512_BF16 cmake test, so it always failed.

Signed-off-by: vient <lozko.roma@gmail.com>

runtime/src/iree/builtins/ukernel/arch/x86_64/CMakeLists.txt

diff --git a/runtime/src/iree/builtins/ukernel/arch/x86_64/CMakeLists.txt b/runtime/src/iree/builtins/ukernel/arch/x86_64/CMakeLists.txt
index 9557c8c..63d1376 100644
--- 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 @@
   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;"
     "}"
   )