zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Detect metal version and propagate correctly for JIT (#1109)

Browse Source 
* detect metal version and propagate correctly for JIT

* remove softmax

* fix versions
This commit is contained in:
Awni Hannun
2024-05-15 17:42:09 -07:00
committed by GitHub
parent c417e42116
commit 1873ffda01
7 changed files with 33 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mlx/backend/metal/CMakeLists.txt
View File

@@ -5,10 +5,16 @@ add_custom_command(
${CMAKE_CURRENT_BINARY_DIR}/compiled_preamble.cpp
${CMAKE_C_COMPILER}
${PROJECT_SOURCE_DIR}
"-D${MLX_METAL_VERSION}"
DEPENDS make_compiled_preamble.sh
kernels/compiled_preamble.h
kernels/unary.h
kernels/binary.h
kernels/bf16.h
kernels/erf.h
kernels/expm1f.h
kernels/utils.h
kernels/bf16_math.h
)
add_custom_target(