mlx/examples/extensions
Awni Hannun d568c7ee36
Rename block sparse (#1149)
* block_sparse_mm to gather_mm

* rename

* nit

* nit
2024-05-22 07:48:34 -07:00
..
axpby Fix extensions (#1126) 2024-05-16 15:36:25 -07:00
mlx_sample_extensions Fix extensions (#1126) 2024-05-16 15:36:25 -07:00
bindings.cpp Extensions (#962) 2024-04-09 08:50:36 -07:00
CMakeLists.txt Extensions (#962) 2024-04-09 08:50:36 -07:00
pyproject.toml fix gguf loading quants (#1014) 2024-04-19 12:24:07 -07:00
README.md Rename block sparse (#1149) 2024-05-22 07:48:34 -07:00
requirements.txt Fix extensions (#1126) 2024-05-16 15:36:25 -07:00
setup.py Extensions (#962) 2024-04-09 08:50:36 -07:00
test.py Fix extensions (#1126) 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