Skip to content

Commit e57ccaa

Browse files
authored
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>
1 parent 7ab66ff commit e57ccaa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

+2-2
Original file line numberDiff line numberDiff line change
@@ -316,8 +316,8 @@ if(IREE_UK_TRY_X86_64_AVX512_BF16)
316316
string(JOIN "\n" IREE_UK_BUILD_X86_64_AVX512_BF16_TEST
317317
"#include <immintrin.h>"
318318
"int main() {"
319-
" __m256bh a;"
320-
" _mm512_cvtneps_pbh(a);"
319+
" __m512 a;"
320+
" __m256bh b = _mm512_cvtneps_pbh(a);"
321321
" return 0;"
322322
"}"
323323
)

0 commit comments

Comments
 (0)