awni's commit files

python/src/indexing.h

@@ -0,0 +1,12 @@
+#pragma once
+
+#include <pybind11/pybind11.h>
+
+#include "mlx/array.h"
+#include "python/src/utils.h"
+
+namespace py = pybind11;
+using namespace mlx::core;
+
+array mlx_get_item(const array& src, const py::object& obj);
+void mlx_set_item(array& src, const py::object& obj, const ScalarOrArray& v);