mirror of
https://github.com/ml-explore/mlx.git
synced 2025-11-01 00:28:11 +08:00
89d327075f
* Share KV smem * Fix bfloat error * Unroll O = S @ V loop * Perf upgrade * Remove commented out function * Add -Wno-c++17-extensions flag to metal flags * Add -Wno-c++17-extensions flag to metal extension flags
46 lines
1.5 KiB
CMake
46 lines
1.5 KiB
CMake
include(CMakeParseArguments)
|
|
|
|
# clang format off
|
|
#
|
|
# ##############################################################################
|
|
# 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 dependency
|
|
# files (like headers)
|
|
#
|
|
# clang format on
|
|
|
|
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 -Wno-c++17-extensions)
|
|
|
|
# Prepare metallib 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)
|