mlx/benchmarks/python
Jagrit Digani 776c3d226d
Convolution update (#651)
* Init steel conv and update Conv primitive

* Update slow CPU implementation to support flipping and input dilation winograd conv routing

Co-authored-by: Awni Hannun <awni@apple.com>
2024-02-28 20:11:16 -08:00
..
blas Update GEMM (#424) 2024-01-17 12:42:39 -08:00
comparative feat: Update pre-commit-config.yaml (#667) 2024-02-11 06:08:20 -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