mlx/benchmarks/python
Angelos Katharopoulos c15fe3e61b
Allow arbitrary first dimension in quantization kernels. (#458)
* Allow arbitrary first dim on qmm_t and qmv
* Allow arbitrary first dim on qmm and qvm
* Specialized aligned vs unaligned case
* Add more checks for valid quantizations
2024-01-16 00:46:21 -08:00
..
blas Spelling (#342) 2024-01-01 21:08:17 -08:00
comparative Allow arbitrary first dimension in quantization kernels. (#458) 2024-01-16 00:46:21 -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