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

Fix JIT reductions (#1373)

Browse Source
This commit is contained in:
Alex Barron
2024-08-28 16:39:11 -07:00
committed by GitHub
parent a6c3b38fba
commit 28be4de7c2
7 changed files with 63 additions and 296 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -79,6 +79,7 @@ if (MLX_METAL_JIT)
kernels/reduction/reduce_all.h
kernels/reduction/reduce_col.h
kernels/reduction/reduce_row.h
kernels/reduction/reduce_init.h
)
make_jit_source(
steel/gemm/gemm