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
..
blas	Update GEMM (#424)	2024-01-17 12:42:39 -08:00
comparative	Update GEMM (#424)	2024-01-17 12:42:39 -08:00
batch_matmul_bench.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
llama_jax_bench.py	copyright + ack	2023-11-30 11:12:53 -08:00
llama_mlx_bench.py	copyright + ack	2023-11-30 11:12:53 -08:00
llama_torch_bench.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
single_ops.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
time_utils.py	copyright + ack	2023-11-30 11:12:53 -08:00