L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Switch to nanobind (#839)

Browse Source 
* mostly builds

* most tests pass

* fix circle build

* add back buffer protocol

* includes

* fix for py38

* limit to cpu device

* include

* fix stubs

* move signatures for docs

* stubgen + docs fix

* doc for compiled function, comments
This commit is contained in:
Awni Hannun
2024-03-18 20:12:25 -07:00
committed by GitHub
parent d39ed54f8e
commit 9a8ee00246
34 changed files with 2343 additions and 2344 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/src/constants.cpp
+6 -6
View File
@@ -1,11 +1,11 @@
// init_constants.cpp
// Copyright © 2023-2024 Apple Inc.
#include <pybind11/pybind11.h>
#include <nanobind/nanobind.h>
#include <limits>
namespace py = pybind11;
namespace nb = nanobind;
void init_constants(py::module_& m) {
void init_constants(nb::module_& m) {
m.attr("Inf") = std::numeric_limits<double>::infinity();
m.attr("Infinity") = std::numeric_limits<double>::infinity();
m.attr("NAN") = NAN;
@@ -19,6 +19,6 @@ void init_constants(py::module_& m) {
m.attr("inf") = std::numeric_limits<double>::infinity();
m.attr("infty") = std::numeric_limits<double>::infinity();
m.attr("nan") = NAN;
m.attr("newaxis") = pybind11::none();
m.attr("newaxis") = nb::none();
m.attr("pi") = 3.1415926535897932384626433;
}
}