Skip to content

Commit 74d73dc

Browse files
authored
Make updates to fix issues with clang-cl builds while using AVX512 flags (#10314)
1 parent 4047be7 commit 74d73dc

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

ggml/src/ggml-cpu/CMakeLists.txt

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,14 +143,23 @@ elseif (CMAKE_OSX_ARCHITECTURES STREQUAL "x86_64" OR CMAKE_GENERATOR_PLATFORM_LW
143143
if (GGML_AVX512_VBMI)
144144
add_compile_definitions($<$<COMPILE_LANGUAGE:C>:__AVX512VBMI__>)
145145
add_compile_definitions($<$<COMPILE_LANGUAGE:CXX>:__AVX512VBMI__>)
146+
if (CMAKE_C_COMPILER_ID STREQUAL "Clang")
147+
list(APPEND ARCH_FLAGS -mavx512vbmi)
148+
endif()
146149
endif()
147150
if (GGML_AVX512_VNNI)
148151
add_compile_definitions($<$<COMPILE_LANGUAGE:C>:__AVX512VNNI__>)
149152
add_compile_definitions($<$<COMPILE_LANGUAGE:CXX>:__AVX512VNNI__>)
153+
if (CMAKE_C_COMPILER_ID STREQUAL "Clang")
154+
list(APPEND ARCH_FLAGS -mavx512vnni)
155+
endif()
150156
endif()
151157
if (GGML_AVX512_BF16)
152158
add_compile_definitions($<$<COMPILE_LANGUAGE:C>:__AVX512BF16__>)
153159
add_compile_definitions($<$<COMPILE_LANGUAGE:CXX>:__AVX512BF16__>)
160+
if (CMAKE_C_COMPILER_ID STREQUAL "Clang")
161+
list(APPEND ARCH_FLAGS -mavx512bf16)
162+
endif()
154163
endif()
155164
if (GGML_AMX_TILE)
156165
add_compile_definitions($<$<COMPILE_LANGUAGE:C>:__AMX_TILE__>)

0 commit comments

Comments
 (0)