mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-24 09:21:16 +08:00
56 lines
1.6 KiB
CMake
56 lines
1.6 KiB
CMake
|
include(CMakeParseArguments)
|
||
|
|
|
||
|
|
###############################################################################
|
||
|
|
# Build metal library
|
||
|
|
#
|
||
|
|
# Adds a custom target ${TARGET} to build ${OUTPUT_DIRECTORY}/{TITLE}.metallib
|
||
|
|
# from list ${SOURCES}, including list ${INCLUDE_DIRS}, depends on list ${DEPS}
|
||
|
|
#
|
||
|
|
# Args:
|
||
|
|
# TARGET: Custom target to be added for the metal library
|
||
|
|
# TITLE: Name of the .metallib
|
||
|
|
# OUTPUT_DIRECTORY: Where to place ${TITLE}.metallib
|
||
|
|
# SOURCES: List of source files
|
||
|
|
# INCLUDE_DIRS: List of include dirs
|
||
|
|
# DEPS: List of depedency files (like headers)
|
||
|
|
#
|
||
|
|
macro(mlx_build_metallib)
|
||
|
|
# Parse args
|
||
|
|
set(oneValueArgs TARGET TITLE OUTPUT_DIRECTORY)
|
||
|
|
set(multiValueArgs SOURCES INCLUDE_DIRS DEPS)
|
||
|
|
cmake_parse_arguments(
|
||
|
|
MTLLIB
|
||
|
|
""
|
||
|
|
"${oneValueArgs}"
|
||
|
|
"${multiValueArgs}"
|
||
|
|
${ARGN}
|
||
|
|
)
|
||
|
|
|
||
|
|
# Set output
|
||
|
|
set(MTLLIB_BUILD_TARGET "${MTLLIB_OUTPUT_DIRECTORY}/${MTLLIB_TITLE}.metallib")
|
||
|
|
|
||
|
|
# Collect compile options
|
||
|
|
set(MTLLIB_COMPILE_OPTIONS -Wall -Wextra -fno-fast-math)
|
||
|
|
|
||
|
|
# Prepare metllib build command
|
||
|
|
add_custom_command(
|
||
|
|
OUTPUT ${MTLLIB_BUILD_TARGET}
|
||
|
|
COMMAND xcrun -sdk macosx metal
|
||
|
|
"$<LIST:TRANSFORM,${MTLLIB_INCLUDE_DIRS},PREPEND,-I>"
|
||
|
|
${MTLLIB_COMPILE_OPTIONS}
|
||
|
|
${MTLLIB_SOURCES}
|
||
|
|
-o ${MTLLIB_BUILD_TARGET}
|
||
|
|
DEPENDS ${MTLLIB_DEPS} ${MTLLIB_SOURCES}
|
||
|
|
COMMAND_EXPAND_LISTS
|
||
|
|
COMMENT "Building ${MTLLIB_TITLE}.metallib"
|
||
|
|
VERBATIM
|
||
|
|
)
|
||
|
|
|
||
|
|
# Add metallib custom target
|
||
|
|
add_custom_target(
|
||
|
|
${MTLLIB_TARGET}
|
||
|
|
DEPENDS
|
||
|
|
${MTLLIB_BUILD_TARGET}
|
||
|
|
)
|
||
|
|
|
||
|
|
endmacro(mlx_build_metallib)
|