mlx/benchmarks
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
..
cpp Multi output primitives (#330) 2024-01-08 16:39:08 -08:00
numpy Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
python Allow arbitrary first dimension in quantization kernels. (#458) 2024-01-16 00:46:21 -08:00