mlx/benchmarks/python
Jagrit Digani 6686e61ca4
Reduce update (#783)
* Split reduction files to reduce compile times

* Add small and medium axis size specializations for row reductions

* Add non-row-reduction options for small and med kernels
2024-03-04 19:09:51 -08:00
..
blas Update GEMM (#424) 2024-01-17 12:42:39 -08:00
comparative Reduce update (#783) 2024-03-04 19:09:51 -08:00
batch_matmul_bench.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
compile_bench.py Shapeless compilation for some graphs (#687) 2024-02-19 21:43:54 -08:00
conv_bench.py Convolution update (#651) 2024-02-28 20:11:16 -08:00
gather_bench.py Scatter optimization : Eliminate 64b integer divide. (#662) 2024-02-10 08:49:51 -08:00
rope_bench.py Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
scatter_bench.py Up to 10x faster scatter. (#709) 2024-02-21 11:09:30 -08:00
single_ops.py Propagate nans in binary ops (#579) 2024-01-29 11:19:38 -08:00
time_utils.py Shapeless compilation for some graphs (#687) 2024-02-19 21:43:54 -08:00