mirror of
https://github.com/ml-explore/mlx.git
synced 2025-11-06 03:58:12 +08:00
compile changes if stream changes (#1644)
This commit is contained in:
@@ -184,10 +184,6 @@ void init_array(nb::module_& m) {
|
||||
R"pbdoc(
|
||||
A helper object to apply updates at specific indices.
|
||||
)pbdoc")
|
||||
.def(
|
||||
nb::init<const array&>(),
|
||||
"x"_a,
|
||||
nb::sig("def __init__(self, x: array)"))
|
||||
.def("__getitem__", &ArrayAt::set_indices, "indices"_a.none())
|
||||
.def("add", &ArrayAt::add, "value"_a)
|
||||
.def("subtract", &ArrayAt::subtract, "value"_a)
|
||||
@@ -202,10 +198,6 @@ void init_array(nb::module_& m) {
|
||||
R"pbdoc(
|
||||
A helper object to iterate over the 1st dimension of an array.
|
||||
)pbdoc")
|
||||
.def(
|
||||
nb::init<const array&>(),
|
||||
"x"_a,
|
||||
nb::sig("def __init__(self, x: array)"))
|
||||
.def("__next__", &ArrayPythonIterator::next)
|
||||
.def("__iter__", [](const ArrayPythonIterator& it) { return it; });
|
||||
|
||||
|
||||
@@ -48,7 +48,6 @@ void init_stream(nb::module_& m) {
|
||||
R"pbdoc(
|
||||
A stream for running operations on a given device.
|
||||
)pbdoc")
|
||||
.def(nb::init<int, Device>(), "index"_a, "device"_a)
|
||||
.def_ro("device", &Stream::device)
|
||||
.def(
|
||||
"__repr__",
|
||||
|
||||
Reference in New Issue
Block a user