mirror of
				https://github.com/ml-explore/mlx.git
				synced 2025-11-01 00:28:11 +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
	 Awni Hannun
					Awni Hannun