Handle dpnp functions and tests to run on CUDA devices

dpnp/dpnp_iface.py

@@ -68,6 +68,7 @@
     "get_normalized_queue_device",
     "get_result_array",
     "get_usm_ndarray",
+    "is_cuda_backend",
     "get_usm_ndarray_or_scalar",
     "is_supported_array_or_scalar",
     "is_supported_array_type",
@@ -733,6 +734,40 @@ def get_usm_ndarray_or_scalar(a):
     return a if dpnp.isscalar(a) else get_usm_ndarray(a)
 
 
+def is_cuda_backend(obj=None):
+    """
+    Checks that object has a CUDA backend.
+
+    Parameters
+    ----------
+    obj : {Device, SyclDevice, SyclQueue, dpnp.ndarray, usm_ndarray, None},
+          optional
+        An input object with sycl_device property to check device backend.
+        If `obj` is ``None``, device backend will be checked for the default
+        queue.
+        Default: ``None``.
+
+    Returns
+    -------
+    out : bool
+        Return ``True`` if object has a cuda backend, otherwise``False``.
+
+    """
+
+    if obj is None:
+        sycl_device = dpctl.SyclQueue().sycl_device
+    elif isinstance(obj, dpctl.SyclDevice):
+        sycl_device = obj
+    else:
+        sycl_device = getattr(obj, "sycl_device", None)
+    if (
+        sycl_device is not None
+        and sycl_device.backend == dpctl.backend_type.cuda
+    ):
+        return True
+    return False
+
+
 def is_supported_array_or_scalar(a):
     """
     Return ``True`` if `a` is a scalar or an array of either

dpnp/dpnp_iface_indexing.py

@@ -173,6 +173,7 @@ def choose(x1, choices, out=None, mode="raise"):
     :obj:`dpnp.take_along_axis` : Preferable if choices is an array.
 
     """
+
     x1_desc = dpnp.get_dpnp_descriptor(x1, copy_when_nondefault_queue=False)
 
     choices_list = []
@@ -192,6 +193,8 @@ def choose(x1, choices, out=None, mode="raise"):
             pass
         elif not choices_list:
             pass
+        elif dpnp.is_cuda_backend(x1):
+            pass
         else:
             size = x1_desc.size
             choices_size = choices_list[0].size

dpnp/dpnp_iface_libmath.py

@@ -81,7 +81,7 @@ def erf(in_array1):
     x1_desc = dpnp.get_dpnp_descriptor(
         in_array1, copy_when_strides=False, copy_when_nondefault_queue=False
     )
-    if x1_desc:
+    if x1_desc and not dpnp.is_cuda_backend(in_array1):
         return dpnp_erf(x1_desc).get_pyobj()
 
     result = create_output_descriptor_py(

dpnp/dpnp_iface_mathematical.py

@@ -2691,8 +2691,13 @@ def modf(x1, **kwargs):
     """
 
     x1_desc = dpnp.get_dpnp_descriptor(x1, copy_when_nondefault_queue=False)
-    if x1_desc and not kwargs:
-        return dpnp_modf(x1_desc)
+    if x1_desc:
+        if not kwargs:
+            pass
+        elif dpnp.is_cuda_backend(x1):
+            pass
+        else:
+            return dpnp_modf(x1_desc)
 
     return call_origin(numpy.modf, x1, **kwargs)
 

dpnp/dpnp_iface_sorting.py

@@ -197,6 +197,8 @@ def partition(x1, kth, axis=-1, kind="introselect", order=None):
             pass
         elif order is not None:
             pass
+        elif dpnp.is_cuda_backend(x1):
+            pass
         else:
             return dpnp_partition(x1_desc, kth, axis, kind, order).get_pyobj()
 

dpnp/dpnp_iface_statistics.py

@@ -397,6 +397,8 @@ def correlate(x1, x2, mode="valid"):
             pass
         elif mode != "valid":
             pass
+        elif dpnp.is_cuda_backend(x1) or dpnp.is_cuda_backend(x2):
+            pass
         else:
             return dpnp_correlate(x1_desc, x2_desc).get_pyobj()
 

dpnp/random/dpnp_iface_random.py

@@ -150,6 +150,8 @@ def beta(a, b, size=None):
             pass
         elif b <= 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_beta(a, b, size).get_pyobj()
 
@@ -196,6 +198,8 @@ def binomial(n, p, size=None):
             pass
         elif n < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_binomial(int(n), p, size).get_pyobj()
 
@@ -244,6 +248,8 @@ def chisquare(df, size=None):
             pass
         elif df <= 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             # TODO:
             # float to int, safe
@@ -312,6 +318,8 @@ def exponential(scale=1.0, size=None):
             pass
         elif scale < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_exponential(scale, size).get_pyobj()
 
@@ -348,6 +356,8 @@ def f(dfnum, dfden, size=None):
             pass
         elif dfden <= 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_f(dfnum, dfden, size).get_pyobj()
 
@@ -386,6 +396,8 @@ def gamma(shape, scale=1.0, size=None):
             pass
         elif shape < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_gamma(shape, scale, size).get_pyobj()
 
@@ -420,6 +432,8 @@ def geometric(p, size=None):
             pass
         elif p > 1 or p <= 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_geometric(p, size).get_pyobj()
 
@@ -456,6 +470,8 @@ def gumbel(loc=0.0, scale=1.0, size=None):
             pass
         elif scale < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_gumbel(loc, scale, size).get_pyobj()
 
@@ -504,6 +520,8 @@ def hypergeometric(ngood, nbad, nsample, size=None):
             pass
         elif nsample < 1:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             _m = int(ngood)
             _l = int(ngood) + int(nbad)
@@ -542,6 +560,8 @@ def laplace(loc=0.0, scale=1.0, size=None):
             pass
         elif scale < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_laplace(loc, scale, size).get_pyobj()
 
@@ -576,6 +596,8 @@ def logistic(loc=0.0, scale=1.0, size=None):
             pass
         elif scale < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             result = dpnp_rng_logistic(loc, scale, size).get_pyobj()
             if size is None or size == 1:
@@ -617,6 +639,8 @@ def lognormal(mean=0.0, sigma=1.0, size=None):
             pass
         elif sigma < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_lognormal(mean, sigma, size).get_pyobj()
 
@@ -674,6 +698,8 @@ def multinomial(n, pvals, size=None):
             pass
         elif pvals_sum < 0.0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             if size is None:
                 shape = (d,)
@@ -725,6 +751,8 @@ def multivariate_normal(mean, cov, size=None, check_valid="warn", tol=1e-8):
             pass
         elif mean_.shape[0] != cov_.shape[0]:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             final_shape = list(shape[:])
             final_shape.append(mean_.shape[0])
@@ -777,6 +805,8 @@ def negative_binomial(n, p, size=None):
             pass
         elif n <= 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_negative_binomial(n, p, size).get_pyobj()
 
@@ -862,6 +892,8 @@ def noncentral_chisquare(df, nonc, size=None):
             pass
         elif nonc < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_noncentral_chisquare(df, nonc, size).get_pyobj()
 
@@ -912,6 +944,8 @@ def pareto(a, size=None):
             pass
         elif a <= 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_pareto(a, size).get_pyobj()
 
@@ -981,6 +1015,8 @@ def poisson(lam=1.0, size=None):
             pass
         elif lam < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_poisson(lam, size).get_pyobj()
 
@@ -1016,6 +1052,8 @@ def power(a, size=None):
             pass
         elif a <= 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_power(a, size).get_pyobj()
 
@@ -1423,6 +1461,8 @@ def rayleigh(scale=1.0, size=None):
             pass
         elif scale < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_rayleigh(scale, size).get_pyobj()
 
@@ -1495,6 +1535,8 @@ def shuffle(x1):
     if x1_desc:
         if not dpnp.is_type_supported(x1_desc.dtype):
             pass
+        elif dpnp.is_cuda_backend(x1):
+            pass
         else:
             dpnp_rng_shuffle(x1_desc).get_pyobj()
             return
@@ -1545,6 +1587,8 @@ def seed(seed=None, device=None, sycl_queue=None):
             pass
         elif seed < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             # TODO:
             # migrate to a single approach with RandomState class
@@ -1577,7 +1621,10 @@ def standard_cauchy(size=None):
     """
 
     if not use_origin_backend(size):
-        return dpnp_rng_standard_cauchy(size).get_pyobj()
+        if dpnp.is_cuda_backend():
+            pass
+        else:
+            return dpnp_rng_standard_cauchy(size).get_pyobj()
 
     return call_origin(numpy.random.standard_cauchy, size)
 
@@ -1602,7 +1649,10 @@ def standard_exponential(size=None):
     """
 
     if not use_origin_backend(size):
-        return dpnp_rng_standard_exponential(size).get_pyobj()
+        if dpnp.is_cuda_backend():
+            pass
+        else:
+            return dpnp_rng_standard_exponential(size).get_pyobj()
 
     return call_origin(numpy.random.standard_exponential, size)
 
@@ -1636,6 +1686,8 @@ def standard_gamma(shape, size=None):
             pass
         elif shape < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_standard_gamma(shape, size).get_pyobj()
 
@@ -1714,6 +1766,8 @@ def standard_t(df, size=None):
             pass
         elif df <= 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_standard_t(df, size).get_pyobj()
 
@@ -1758,6 +1812,8 @@ def triangular(left, mode, right, size=None):
             pass
         elif left == right:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_triangular(left, mode, right, size).get_pyobj()
 
@@ -1862,6 +1918,8 @@ def vonmises(mu, kappa, size=None):
             return dpnp.nan
         elif kappa < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_vonmises(mu, kappa, size).get_pyobj()
 
@@ -1898,6 +1956,8 @@ def wald(mean, scale, size=None):
             pass
         elif scale <= 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_wald(mean, scale, size).get_pyobj()
 
@@ -1930,6 +1990,8 @@ def weibull(a, size=None):
             pass
         elif a < 0:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_weibull(a, size).get_pyobj()
 
@@ -1962,6 +2024,8 @@ def zipf(a, size=None):
             pass
         elif a <= 1:
             pass
+        elif dpnp.is_cuda_backend():
+            pass
         else:
             return dpnp_rng_zipf(a, size).get_pyobj()