Rename Python bindings for integer indexing

Aligns with most non-constructors

Author: ndgrigorian

dpctl/tensor/libtensor/source/integer_advanced_indexing.cpp

@@ -244,13 +244,13 @@ std::vector<dpctl::tensor::usm_ndarray> parse_py_ind(const sycl::queue &q,
 }
 
 std::pair<sycl::event, sycl::event>
-usm_ndarray_take(const dpctl::tensor::usm_ndarray &src,
-                 const py::object &py_ind,
-                 const dpctl::tensor::usm_ndarray &dst,
-                 int axis_start,
-                 std::uint8_t mode,
-                 sycl::queue &exec_q,
-                 const std::vector<sycl::event> &depends)
+py_take(const dpctl::tensor::usm_ndarray &src,
+        const py::object &py_ind,
+        const dpctl::tensor::usm_ndarray &dst,
+        int axis_start,
+        std::uint8_t mode,
+        sycl::queue &exec_q,
+        const std::vector<sycl::event> &depends)
 {
     std::vector<dpctl::tensor::usm_ndarray> ind = parse_py_ind(exec_q, py_ind);
 
@@ -515,13 +515,13 @@ usm_ndarray_take(const dpctl::tensor::usm_ndarray &src,
 }
 
 std::pair<sycl::event, sycl::event>
-usm_ndarray_put(const dpctl::tensor::usm_ndarray &dst,
-                const py::object &py_ind,
-                const dpctl::tensor::usm_ndarray &val,
-                int axis_start,
-                std::uint8_t mode,
-                sycl::queue &exec_q,
-                const std::vector<sycl::event> &depends)
+py_put(const dpctl::tensor::usm_ndarray &dst,
+       const py::object &py_ind,
+       const dpctl::tensor::usm_ndarray &val,
+       int axis_start,
+       std::uint8_t mode,
+       sycl::queue &exec_q,
+       const std::vector<sycl::event> &depends)
 {
     std::vector<dpctl::tensor::usm_ndarray> ind = parse_py_ind(exec_q, py_ind);
     int k = ind.size();

dpctl/tensor/libtensor/source/integer_advanced_indexing.hpp

@@ -39,22 +39,22 @@ namespace py_internal
 {
 
 extern std::pair<sycl::event, sycl::event>
-usm_ndarray_take(const dpctl::tensor::usm_ndarray &,
-                 const py::object &,
-                 const dpctl::tensor::usm_ndarray &,
-                 int,
-                 std::uint8_t,
-                 sycl::queue &,
-                 const std::vector<sycl::event> & = {});
+py_take(const dpctl::tensor::usm_ndarray &,
+        const py::object &,
+        const dpctl::tensor::usm_ndarray &,
+        int,
+        std::uint8_t,
+        sycl::queue &,
+        const std::vector<sycl::event> & = {});
 
 extern std::pair<sycl::event, sycl::event>
-usm_ndarray_put(const dpctl::tensor::usm_ndarray &,
-                const py::object &,
-                const dpctl::tensor::usm_ndarray &,
-                int,
-                std::uint8_t,
-                sycl::queue &,
-                const std::vector<sycl::event> & = {});
+py_put(const dpctl::tensor::usm_ndarray &,
+       const py::object &,
+       const dpctl::tensor::usm_ndarray &,
+       int,
+       std::uint8_t,
+       sycl::queue &,
+       const std::vector<sycl::event> & = {});
 
 extern void init_advanced_indexing_dispatch_tables(void);
 

dpctl/tensor/libtensor/source/tensor_ctors.cpp

@@ -101,8 +101,8 @@ using dpctl::tensor::py_internal::usm_ndarray_full;
 using dpctl::tensor::py_internal::usm_ndarray_zeros;
 
 /* ============== Advanced Indexing ============= */
-using dpctl::tensor::py_internal::usm_ndarray_put;
-using dpctl::tensor::py_internal::usm_ndarray_take;
+using dpctl::tensor::py_internal::py_put;
+using dpctl::tensor::py_internal::py_take;
 
 using dpctl::tensor::py_internal::py_extract;
 using dpctl::tensor::py_internal::py_mask_positions;
@@ -324,7 +324,7 @@ PYBIND11_MODULE(_tensor_impl, m)
           py::arg("fill_value"), py::arg("dst"), py::arg("sycl_queue"),
           py::arg("depends") = py::list());
 
-    m.def("_take", &usm_ndarray_take,
+    m.def("_take", &py_take,
           "Takes elements at usm_ndarray indices `ind` and axes starting "
           "at axis `axis_start` from array `src` and copies them "
           "into usm_ndarray `dst` synchronously."
@@ -333,7 +333,7 @@ PYBIND11_MODULE(_tensor_impl, m)
           py::arg("mode"), py::arg("sycl_queue"),
           py::arg("depends") = py::list());
 
-    m.def("_put", &usm_ndarray_put,
+    m.def("_put", &py_put,
           "Puts elements at usm_ndarray indices `ind` and axes starting "
           "at axis `axis_start` into array `dst` from "
           "usm_ndarray `val` synchronously."