mlx/benchmarks/python/blas
Jagrit Digani 78102a47ad
Update GEMM (#424)
* 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
2024-01-17 12:42:39 -08:00
..
bench_gemm.py Update GEMM (#424) 2024-01-17 12:42:39 -08:00
bench_gemv.py Spelling (#342) 2024-01-01 21:08:17 -08:00