From ffe91236cc1ba5a51088283475e9b589a7e32860 Mon Sep 17 00:00:00 2001 From: vient Date: Tue, 23 Jul 2024 02:25:25 +0200 Subject: [PATCH] Fix BF16 detection in x86_64 ukernel cmake `_mm512_cvtneps_pbh` was used incorrectly in IREE_UK_BUILD_X86_64_AVX512_BF16 cmake test Signed-off-by: vient --- runtime/src/iree/builtins/ukernel/arch/x86_64/CMakeLists.txt | 4 ++-- 1 file changed, 2 insertions(+), 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 index 9557c8caea9f..63d137636146 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 @@ if(IREE_UK_TRY_X86_64_AVX512_BF16) string(JOIN "\n" IREE_UK_BUILD_X86_64_AVX512_BF16_TEST "#include " "int main() {" - " __m256bh a;" - " _mm512_cvtneps_pbh(a);" + " __m512 a;" + " __m256bh b = _mm512_cvtneps_pbh(a);" " return 0;" "}" )