Files
mlx/examples/extensions
Awni Hannun 30bbea2f08 Add gemv masked to JIT plus some fixes (#1310)
* add gemv masked to JIT plus some fixes

* some cleanup

* add utils

* fix

* fix 2

* more cleaning

* fix

* remove unused mps matmul support

* one more nit

* revert
2024-08-07 13:38:07 -07:00
..
2024-04-09 08:50:36 -07:00
2024-04-09 08:50:36 -07:00
2024-04-19 12:24:07 -07:00
2024-05-22 07:48:34 -07:00
2024-05-16 15:36:25 -07:00
2024-04-09 08:50:36 -07:00
2024-05-16 15:36:25 -07:00

Build

pip install -e .

For faster builds during development, you can also pre-install the requirements:

pip install -r requirements.txt

And then run:

python setup.py build_ext -j8 --inplace

Test

python test.py