native version dpnp.any() (#637)

* native version dpnp.any()

Author: Rubtsowa

dpnp/backend/include/dpnp_iface.hpp

@@ -116,6 +116,17 @@ void dpnp_memory_memcpy_c(void* dst, const void* src, size_t size_in_bytes);
 template <typename _DataType, typename _ResultType>
 INP_DLLEXPORT void dpnp_all_c(void* array, void* result, const size_t size);
 
+/**
+ * @ingroup BACKEND_API
+ * @brief Test whether any array element along a given axis evaluates to True.
+ *
+ * @param [in]  array       Input array.
+ * @param [out] result      Output array.
+ * @param [in]  size        Number of input elements in `array`.
+ */
+template <typename _DataType, typename _ResultType>
+INP_DLLEXPORT void dpnp_any_c(const void* array, void* result, const size_t size);
+
 /**
  * @ingroup BACKEND_API
  * @brief Array initialization

dpnp/backend/include/dpnp_iface_fptr.hpp

@@ -61,6 +61,7 @@ enum class DPNPFuncName : size_t
     DPNP_FN_ABSOLUTE,                 /**< Used in numpy.absolute() implementation  */
     DPNP_FN_ADD,                      /**< Used in numpy.add() implementation  */
     DPNP_FN_ALL,                      /**< Used in numpy.all() implementation  */
+    DPNP_FN_ANY,                      /**< Used in numpy.any() implementation  */
     DPNP_FN_ARANGE,                   /**< Used in numpy.arange() implementation  */
     DPNP_FN_ARCCOS,                   /**< Used in numpy.arccos() implementation  */
     DPNP_FN_ARCCOSH,                  /**< Used in numpy.arccosh() implementation  */

dpnp/backend/kernels/dpnp_krnl_logic.cpp

@@ -52,6 +52,33 @@ void dpnp_all_c(void* array1_in, void* result1, const size_t size)
     return;
 }
 
+template <typename _DataType, typename _ResultType>
+void dpnp_any_c(const void* array1_in, void* result1, const size_t size)
+{
+    if ((array1_in == nullptr) || (result1 == nullptr))
+    {
+        return;
+    }
+
+    const _DataType* array_in = reinterpret_cast<const _DataType*>(array1_in);
+    _ResultType* result = reinterpret_cast<_ResultType*>(result1);
+
+    bool res = false;
+
+    for (size_t i = 0; i < size; ++i)
+    {
+        if (array_in[i])
+        {
+            res = true;
+            break;
+        }
+    }
+
+    result[0] = res;
+
+    return;
+}
+
 void func_map_init_logic(func_map_t& fmap)
 {
     fmap[DPNPFuncName::DPNP_FN_ALL][eft_BLN][eft_BLN] = {eft_BLN, (void*)dpnp_all_c<bool, bool>};
@@ -60,5 +87,11 @@ void func_map_init_logic(func_map_t& fmap)
     fmap[DPNPFuncName::DPNP_FN_ALL][eft_FLT][eft_FLT] = {eft_FLT, (void*)dpnp_all_c<float, bool>};
     fmap[DPNPFuncName::DPNP_FN_ALL][eft_DBL][eft_DBL] = {eft_DBL, (void*)dpnp_all_c<double, bool>};
 
+    fmap[DPNPFuncName::DPNP_FN_ANY][eft_BLN][eft_BLN] = {eft_BLN, (void*)dpnp_any_c<bool, bool>};
+    fmap[DPNPFuncName::DPNP_FN_ANY][eft_INT][eft_INT] = {eft_INT, (void*)dpnp_any_c<int, bool>};
+    fmap[DPNPFuncName::DPNP_FN_ANY][eft_LNG][eft_LNG] = {eft_LNG, (void*)dpnp_any_c<long, bool>};
+    fmap[DPNPFuncName::DPNP_FN_ANY][eft_FLT][eft_FLT] = {eft_FLT, (void*)dpnp_any_c<float, bool>};
+    fmap[DPNPFuncName::DPNP_FN_ANY][eft_DBL][eft_DBL] = {eft_DBL, (void*)dpnp_any_c<double, bool>};
+
     return;
 }

dpnp/dpnp_algo/dpnp_algo.pxd

@@ -34,6 +34,7 @@ cdef extern from "dpnp_iface_fptr.hpp" namespace "DPNPFuncName":  # need this na
         DPNP_FN_ABSOLUTE
         DPNP_FN_ADD
         DPNP_FN_ALL
+        DPNP_FN_ANY
         DPNP_FN_ARANGE
         DPNP_FN_ARCCOS
         DPNP_FN_ARCCOSH

dpnp/dpnp_algo/dpnp_algo_logic.pyx

@@ -80,13 +80,13 @@ cpdef dparray dpnp_all(dparray array1):
 cpdef dparray dpnp_any(dparray array1):
     cdef dparray result = dparray((1,), dtype=numpy.bool)
 
-    res = False
-    for i in range(array1.size):
-        if numpy.bool(array1[i]):
-            res = True
-            break
+    cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(array1.dtype)
+
+    cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_ANY, param1_type, param1_type)
 
-    result[0] = res
+    cdef custom_logic_1in_1out_func_ptr_t func = <custom_logic_1in_1out_func_ptr_t > kernel_data.ptr
+
+    func(array1.get_data(), result.get_data(), array1.size)
 
     return result
 

dpnp/dpnp_iface_logic.py

@@ -107,8 +107,6 @@ def all(in_array1, axis=None, out=None, keepdims=False):
 
     """
 
-    is_dparray1 = isinstance(in_array1, dparray)
-
     if not use_origin_backend(in_array1):
         if not isinstance(in_array1, dparray):
             pass
@@ -164,23 +162,18 @@ def any(in_array1, axis=None, out=None, keepdims=False):
 
     """
 
-    is_dparray1 = isinstance(in_array1, dparray)
-
-    if (not use_origin_backend(in_array1) and is_dparray1):
-        if axis is not None:
-            checker_throw_value_error("any", "axis", type(axis), None)
-        if out is not None:
-            checker_throw_value_error("any", "out", type(out), None)
-        if keepdims is not False:
-            checker_throw_value_error("any", "keepdims", keepdims, False)
-
-        result = dpnp_any(in_array1)
-
-        # scalar returned
-        if result.shape == (1,):
-            return result.dtype.type(result[0])
-
-        return result
+    if (not use_origin_backend(in_array1)):
+        if not isinstance(in_array1, dparray):
+            pass
+        elif axis is not None:
+            pass
+        elif out is not None:
+            pass
+        elif keepdims is not False:
+            pass
+        else:
+            result = dpnp_any(in_array1)
+            return result[0]
 
     return call_origin(numpy.any, axis, out, keepdims)