mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 09:51:17 +08:00
78102a47ad
* Organize and collect metal subroutine templates and elements in `metal/kernels/steel/` * Update gemm elements for better performance * Add split-K specialization for gemm * Add `addmm` primitive, op and bindings for fused matmul and bias addition * Update tests and benchmarks as needed |
||
|---|---|---|
| .. | ||
| array.cpp | ||
| CMakeLists.txt | ||
| constants.cpp | ||
| device.cpp | ||
| fft.cpp | ||
| indexing.cpp | ||
| indexing.h | ||
| linalg.cpp | ||
| load.cpp | ||
| load.h | ||
| metal.cpp | ||
| mlx.cpp | ||
| ops.cpp | ||
| random.cpp | ||
| stream.cpp | ||
| transforms.cpp | ||
| utils.h | ||