Move diag_indices from cython to backend

* add c-function for dpnp_diag_indoces

* Change dpnp_diag_indices

* Fix parametrs of dpnp_diag_indices_c

* Fix dpnp_algo.pyx

* Fix dpnp_algo_indexing.pyx

* Update dpnp/dpnp_algo/dpnp_algo_indexing.pyx

Co-authored-by: densmirn <[email protected]>

* Change dpnp_diag_indices

* Update dpnp_krnl_indexing.cpp

* Update dpnp/dpnp_algo/dpnp_algo_indexing.pyx

Co-authored-by: densmirn <[email protected]>

Co-authored-by: densmirn <[email protected]>

Co-authored-by: densmirn <[email protected]>

Author: LukichevaPolina

dpnp/backend/include/dpnp_iface.hpp

@@ -579,6 +579,16 @@ template <typename _DataType>
 INP_DLLEXPORT void dpnp_diag_c(
     void* array, void* result, const int k, size_t* shape, size_t* res_shape, const size_t ndim, const size_t res_ndim);
 
+/**
+ * @ingroup BACKEND_API
+ * @brief math library implementation of diagonal function
+ *
+ * @param [out] result      Output array.
+ * @param [in]  size        Size of array.
+ */
+template <typename _DataType>
+INP_DLLEXPORT void dpnp_diag_indices_c(void* result1, size_t size);
+
 /**
  * @ingroup BACKEND_API
  * @brief math library implementation of diagonal function

dpnp/backend/include/dpnp_iface_fptr.hpp

@@ -96,6 +96,7 @@ enum class DPNPFuncName : size_t
     DPNP_FN_DEGREES,                  /**< Used in numpy.degrees() implementation  */
     DPNP_FN_DET,                      /**< Used in numpy.linalg.det() implementation  */
     DPNP_FN_DIAG,                     /**< Used in numpy.diag() implementation  */
+    DPNP_FN_DIAG_INDICES,             /**< Used in numpy.diag_indices() implementation  */
     DPNP_FN_DIAGONAL,                 /**< Used in numpy.diagonal() implementation  */
     DPNP_FN_DIVIDE,                   /**< Used in numpy.divide() implementation  */
     DPNP_FN_DOT,                      /**< Used in numpy.dot() implementation  */

dpnp/backend/kernels/dpnp_krnl_indexing.cpp

@@ -32,6 +32,15 @@
 #include "dpnpc_memory_adapter.hpp"
 #include "queue_sycl.hpp"
 
+template <typename _DataType>
+class dpnp_diag_indices_c_kernel;
+
+template <typename _DataType>
+void dpnp_diag_indices_c(void* result1, size_t size)
+{
+    dpnp_arange_c<_DataType>(0, 1, result1, size);
+}
+
 template <typename _DataType>
 class dpnp_diagonal_c_kernel;
 
@@ -503,6 +512,11 @@ void dpnp_take_c(void* array1_in, const size_t array1_size, void* indices1, void
 
 void func_map_init_indexing_func(func_map_t& fmap)
 {
+    fmap[DPNPFuncName::DPNP_FN_DIAG_INDICES][eft_INT][eft_INT] = {eft_INT, (void*)dpnp_diag_indices_c<int>};
+    fmap[DPNPFuncName::DPNP_FN_DIAG_INDICES][eft_LNG][eft_LNG] = {eft_LNG, (void*)dpnp_diag_indices_c<long>};
+    fmap[DPNPFuncName::DPNP_FN_DIAG_INDICES][eft_FLT][eft_FLT] = {eft_FLT, (void*)dpnp_diag_indices_c<float>};
+    fmap[DPNPFuncName::DPNP_FN_DIAG_INDICES][eft_DBL][eft_DBL] = {eft_DBL, (void*)dpnp_diag_indices_c<double>};
+
     fmap[DPNPFuncName::DPNP_FN_DIAGONAL][eft_INT][eft_INT] = {eft_INT, (void*)dpnp_diagonal_c<int>};
     fmap[DPNPFuncName::DPNP_FN_DIAGONAL][eft_LNG][eft_LNG] = {eft_LNG, (void*)dpnp_diagonal_c<long>};
     fmap[DPNPFuncName::DPNP_FN_DIAGONAL][eft_FLT][eft_FLT] = {eft_FLT, (void*)dpnp_diagonal_c<float>};

dpnp/dpnp_algo/dpnp_algo.pxd

@@ -74,6 +74,7 @@ cdef extern from "dpnp_iface_fptr.hpp" namespace "DPNPFuncName":  # need this na
         DPNP_FN_DEGREES
         DPNP_FN_DET
         DPNP_FN_DIAG
+        DPNP_FN_DIAG_INDICES
         DPNP_FN_DIAGONAL
         DPNP_FN_DIVIDE
         DPNP_FN_DOT

dpnp/dpnp_algo/dpnp_algo_indexing.pyx

@@ -54,7 +54,7 @@ __all__ += [
     "dpnp_triu_indices_from"
 ]
 
-
+ctypedef void(*fptr_dpnp_diag_indices)(void*, size_t)
 ctypedef void(*custom_indexing_2in_1out_func_ptr_t)(void *, const size_t, void * , void * , size_t)
 ctypedef void(*custom_indexing_2in_1out_func_ptr_t_)(void * , const size_t, void * , const size_t, size_t * , size_t * , const size_t)
 ctypedef void(*custom_indexing_2in_func_ptr_t)(void *, void * , size_t * , const size_t)
@@ -75,15 +75,25 @@ cpdef utils.dpnp_descriptor dpnp_choose(object input, list choices):
 
 
 cpdef tuple dpnp_diag_indices(n, ndim):
-    res_item = dpnp.arange(n, dtype=dpnp.int64)
+    cdef size_t res_size = 0 if n < 0 else n
 
-    # yes, all are the same item
-    result = []
+    cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(dpnp.int64)
+
+    cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_DIAG_INDICES, param1_type, param1_type)
+    
+    cdef fptr_dpnp_diag_indices func = <fptr_dpnp_diag_indices> kernel_data.ptr
+    
+    res_list = []
+    cdef utils.dpnp_descriptor res_arr
+    cdef shape_type_c result_shape = utils._object_to_tuple(res_size)
     for i in range(ndim):
-        result.append(res_item)
+        res_arr = utils.create_output_descriptor(result_shape, kernel_data.return_type, None)
 
-    return tuple(result)
+        func(res_arr.get_data(), res_size)
 
+        res_list.append(res_arr.get_pyobj())
+ 
+    return tuple(res_list)
 
 cpdef utils.dpnp_descriptor dpnp_diagonal(dpnp_descriptor input, offset=0):
     cdef shape_type_c input_shape = input.shape