mlx/examples/extensions
Awni Hannun 40c62c1321
Use int64 stride everywhere (#1671)
* use int64 stride everywhere

* fix ext

* fix ext

* more shape + cleanup

* one more

* few more
2024-12-09 11:09:02 -08:00
..
axpby Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08: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 Chore: add pre-commit hook for cmake (#1362) 2024-09-16 12:53:01 -07:00
pyproject.toml Bump nanobind to 2.2 (#1461) 2024-10-07 16:52:40 -07:00
README.md Rename block sparse (#1149) 2024-05-22 07:48:34 -07:00
requirements.txt version (#1617) 2024-11-22 12:00:03 -08:00
setup.py bump nanobind + fix extension (#1344) 2024-08-21 16:05:07 -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