Skip to content

Commit 369a079

Browse files
[Hardware][Apple-CPU] Disable OneDNN build for Apple Silicon (vllm-project#24200)
Signed-off-by: ignaciosica <[email protected]> Co-authored-by: Li, Jiang <[email protected]>
1 parent 402759d commit 369a079

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

cmake/cpu_extension.cmake

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,7 @@ is_avx512_disabled(AVX512_DISABLED)
8888

8989
if (MACOSX_FOUND AND CMAKE_SYSTEM_PROCESSOR STREQUAL "arm64")
9090
message(STATUS "Apple Silicon Detected")
91+
set(APPLE_SILICON_FOUND TRUE)
9192
set(ENABLE_NUMA OFF)
9293
check_sysctl(hw.optional.neon ASIMD_FOUND)
9394
check_sysctl(hw.optional.arm.FEAT_BF16 ARM_BF16_FOUND)
@@ -189,7 +190,7 @@ else()
189190
set(USE_ACL OFF)
190191
endif()
191192

192-
if ((AVX512_FOUND AND NOT AVX512_DISABLED) OR ASIMD_FOUND OR POWER9_FOUND OR POWER10_FOUND OR POWER11_FOUND)
193+
if ((AVX512_FOUND AND NOT AVX512_DISABLED) OR (ASIMD_FOUND AND NOT APPLE_SILICON_FOUND) OR POWER9_FOUND OR POWER10_FOUND OR POWER11_FOUND)
193194
FetchContent_Declare(
194195
oneDNN
195196
GIT_REPOSITORY https://github.com/oneapi-src/oneDNN.git

0 commit comments

Comments
 (0)