zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-11 19:56:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Enable attn build outside of jit

Browse Source
This commit is contained in:
Jagrit Digani 2024-11-21 13:11:31 -08:00
parent ed4fb26cb9
commit 716277d246
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/backend/metal/kernels/CMakeLists.txt
View File

@ -82,6 +82,8 @@ set(STEEL_ATTN_HEADERS
steel/attn/transforms.h
steel/attn/kernels/steel_attention.h)
build_kernel(steel/attn/kernels/steel_attention ${STEEL_ATTN_HEADERS})
if(NOT MLX_METAL_JIT)
build_kernel(arange arange.h)
build_kernel(binary binary.h binary_ops.h)
@ -107,7 +109,6 @@ if(NOT MLX_METAL_JIT)
build_kernel(steel/gemm/kernels/steel_gemm_fused ${STEEL_HEADERS})
build_kernel(steel/gemm/kernels/steel_gemm_masked ${STEEL_HEADERS})
build_kernel(steel/gemm/kernels/steel_gemm_splitk ${STEEL_HEADERS})
build_kernel(steel/attn/kernels/steel_attention ${STEEL_ATTN_HEADERS})
build_kernel(gemv_masked steel/utils.h)
endif()