zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 17:31:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Move to nanobind v2 (#1316)

Browse Source
This commit is contained in:
Awni Hannun 2024-08-08 17:17:46 -07:00 committed by GitHub
parent 32668a7317
commit 9231617eb3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/extensions/requirements.txt
View File

@ -1,4 +1,4 @@
setuptools>=42
cmake>=3.24
mlx>=0.9.0
nanobind@git+https://github.com/wjakob/nanobind.git@2f04eac452a6d9142dedb957701bdb20125561e4
mlx>=0.16.2
nanobind==2.0

22
python/src/array.cpp
View File

@ -126,7 +126,7 @@ void init_array(nb::module_& m) {
m.attr("float32") = nb::cast(float32);
m.attr("bfloat16") = nb::cast(bfloat16);
m.attr("complex64") = nb::cast(complex64);
nb::class_<Dtype::Category>(
nb::enum_<Dtype::Category>(
m,
"DtypeCategory",
R"pbdoc(
@ -165,16 +165,16 @@ void init_array(nb::module_& m) {
* :ref:`complex64 <data_types>`
See also :func:`~mlx.core.issubdtype`.
)pbdoc");
m.attr("complexfloating") = nb::cast(complexfloating);
m.attr("floating") = nb::cast(floating);
m.attr("inexact") = nb::cast(inexact);
m.attr("signedinteger") = nb::cast(signedinteger);
m.attr("unsignedinteger") = nb::cast(unsignedinteger);
m.attr("integer") = nb::cast(integer);
m.attr("number") = nb::cast(number);
m.attr("generic") = nb::cast(generic);
)pbdoc")
.value("complexfloating", complexfloating)
.value("floating", floating)
.value("inexact", inexact)
.value("signedinteger", signedinteger)
.value("unsignedinteger", unsignedinteger)
.value("integer", integer)
.value("number", number)
.value("generic", generic)
.export_values();
nb::class_<ArrayAt>(
m,
"_ArrayAt",

2
setup.py
View File

@ -176,7 +176,7 @@ if __name__ == "__main__":
include_package_data=True,
extras_require={
"dev": [
"nanobind@git+https://github.com/wjakob/nanobind.git@2f04eac452a6d9142dedb957701bdb20125561e4",
"nanobind==2.0",
"numpy",
"pre-commit",
"setuptools>=42",