zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 01:41:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix cmake version (#1601)

Browse Source
This commit is contained in:
Awni Hannun 2024-11-19 08:45:05 -08:00 committed by GitHub
parent 2419edd5b2
commit 2af7e8a9a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
CMakeLists.txt
View File

@ -101,20 +101,15 @@ elseif(MLX_BUILD_METAL)
set(METAL_CPP_URL
https://developer.apple.com/metal/cpp/files/metal-cpp_macOS15_iOS18-beta.zip
)
# Get the metal version. The mapping is maintained in section 1.6.10 of the
# Metal Shading Language Specification.
# https://developer.apple.com/metal/Metal-Shading-Language-Specification.pdf
if(${CMAKE_OSX_DEPLOYMENT_TARGET} GREATER_EQUAL 15.0)
set(MLX_METAL_VERSION "3.2")
elseif(${CMAKE_OSX_DEPLOYMENT_TARGET} GREATER_EQUAL 14.0)
set(MLX_METAL_VERSION "3.1")
elseif(${CMAKE_OSX_DEPLOYMENT_TARGET} GREATER_EQUAL 13.5)
set(MLX_METAL_VERSION "3.0")
else()
message(FATAL_ERROR "MLX requires macOS 13.5 or higher to build.")
endif()
message(STATUS "METAL VERSION" ${MLX_METAL_VERSION})
if(NOT CMAKE_OSX_DEPLOYMENT_TARGET STREQUAL "")
set(XCRUN_FLAGS "-mmacosx-version-min=${CMAKE_OSX_DEPLOYMENT_TARGET}")
endif()
execute_process(
COMMAND
zsh "-c"
"echo \"__METAL_VERSION__\" | xcrun -sdk macosx metal ${XCRUN_FLAGS} -E -x metal -P - | tail -1 | tr -d '\n'"
OUTPUT_VARIABLE MLX_METAL_VERSION COMMAND_ERROR_IS_FATAL ANY)
FetchContent_Declare(metal_cpp URL ${METAL_CPP_URL})
FetchContent_MakeAvailable(metal_cpp)