Files
mlx/examples/extensions
Awni Hannun 610af352d4 Dispatch bf16 at run time when using the JIT (#1584)
* Dispatch bf16 at run time when using the JIT

* fix extension

* fix extension build

* fix extension build

* Update utils.h
2024-11-15 16:54:36 -08:00
..
2024-04-09 08:50:36 -07:00
2024-10-07 16:52:40 -07:00
2024-05-22 07:48:34 -07:00
2024-10-10 13:05:17 -07:00
2024-08-21 16:05:07 -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