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 Packages Projects Releases Wiki Activity

Gemm update (#1518)

Browse Source
This commit is contained in:
Jagrit Digani
2024-10-30 19:30:28 -07:00
committed by GitHub
parent 884af42da2
commit 960e3f0f05
9 changed files with 701 additions and 196 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -50,7 +50,9 @@ set(STEEL_HEADERS
steel/gemm/transforms.h
steel/gemm/kernels/steel_gemm_fused.h
steel/gemm/kernels/steel_gemm_masked.h
steel/gemm/kernels/steel_gemm_splitk.h)
steel/gemm/kernels/steel_gemm_splitk.h
steel/utils/type_traits.h
steel/utils/integral_constant.h)
if(NOT MLX_METAL_JIT)
build_kernel(arange arange.h)