zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-28 00:58:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Bump nanobind to 2.4 + fix (#1710)

Browse Source 
* bump nanobind to 2.4 + fix

* fix
This commit is contained in:
Awni Hannun
2024-12-17 10:57:54 -08:00
committed by GitHub
parent a6b426422e
commit f110357aaa
11 changed files with 36 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/src/convert.h
View File

@@ -12,6 +12,11 @@
namespace mx = mlx::core;
namespace nb = nanobind;
struct ArrayLike {
ArrayLike(nb::object obj) : obj(obj) {};
nb::object obj;
};
using ArrayInitType = std::variant<
nb::bool_,
nb::int_,
@@ -23,7 +28,7 @@ using ArrayInitType = std::variant<
std::complex<float>,
nb::list,
nb::tuple,
nb::object>;
ArrayLike>;
mx::array nd_array_to_mlx(
nb::ndarray<nb::ro, nb::c_contig, nb::device::cpu> nd_array,