awni's commit files

2023-11-29 10:30:41 -08:00
parent e411fcae68
commit 8ca7f9e8e9
130 changed files with 30159 additions and 0 deletions
@@ -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);