argmin and argmax to desc (#846)

Author: shssf

dpnp/dpnp_algo/dpnp_algo.pxd

@@ -360,8 +360,8 @@ cpdef dpnp_descriptor dpnp_sort(dpnp_descriptor array1)
 """
 Searching functions
 """
-cpdef dparray dpnp_argmax(dpnp_descriptor array1)
-cpdef dparray dpnp_argmin(dpnp_descriptor array1)
+cpdef dpnp_descriptor dpnp_argmax(dpnp_descriptor array1)
+cpdef dpnp_descriptor dpnp_argmin(dpnp_descriptor array1)
 
 """
 Trigonometric functions

dpnp/dpnp_algo/dpnp_algo_searching.pyx

@@ -44,14 +44,15 @@ __all__ += [
 ctypedef void(*custom_search_1in_1out_func_ptr_t)(void * , void * , size_t)
 
 
-cpdef dparray dpnp_argmax(utils.dpnp_descriptor in_array1):
+cpdef utils.dpnp_descriptor dpnp_argmax(utils.dpnp_descriptor in_array1):
     cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(in_array1.dtype)
     cdef DPNPFuncType output_type = dpnp_dtype_to_DPNPFuncType(dpnp.int64)
 
     cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_ARGMAX, param1_type, output_type)
 
-    result_type = dpnp_DPNPFuncType_to_dtype( < size_t > kernel_data.return_type)
-    cdef dparray result = dparray((1,), dtype=result_type)
+    # ceate result array with type given by FPTR data
+    cdef shape_type_c result_shape = (1,)
+    cdef utils.dpnp_descriptor result = utils.create_output_descriptor(result_shape, kernel_data.return_type, None)
 
     cdef custom_search_1in_1out_func_ptr_t func = <custom_search_1in_1out_func_ptr_t > kernel_data.ptr
 
@@ -60,14 +61,15 @@ cpdef dparray dpnp_argmax(utils.dpnp_descriptor in_array1):
     return result
 
 
-cpdef dparray dpnp_argmin(utils.dpnp_descriptor in_array1):
+cpdef utils.dpnp_descriptor dpnp_argmin(utils.dpnp_descriptor in_array1):
     cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(in_array1.dtype)
     cdef DPNPFuncType output_type = dpnp_dtype_to_DPNPFuncType(dpnp.int64)
 
     cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_ARGMIN, param1_type, output_type)
 
-    result_type = dpnp_DPNPFuncType_to_dtype( < size_t > kernel_data.return_type)
-    cdef dparray result = dparray((1,), dtype=result_type)
+    # ceate result array with type given by FPTR data
+    cdef shape_type_c result_shape = (1,)
+    cdef utils.dpnp_descriptor result = utils.create_output_descriptor(result_shape, kernel_data.return_type, None)
 
     cdef custom_search_1in_1out_func_ptr_t func = <custom_search_1in_1out_func_ptr_t > kernel_data.ptr
 

dpnp/dpnp_iface_searching.py

@@ -102,7 +102,7 @@ def argmax(x1, axis=None, out=None):
         elif out is not None:
             pass
         else:
-            result_obj = dpnp_argmax(x1_desc)
+            result_obj = dpnp_argmax(x1_desc).get_pyobj()
             result = dpnp.convert_single_elem_array_to_scalar(result_obj)
 
             return result
@@ -157,7 +157,7 @@ def argmin(x1, axis=None, out=None):
         elif out is not None:
             pass
         else:
-            result_obj = dpnp_argmin(x1_desc)
+            result_obj = dpnp_argmin(x1_desc).get_pyobj()
             result = dpnp.convert_single_elem_array_to_scalar(result_obj)
 
             return result