Testing is adapted for cuda devices

Author: npolina4

dpnp/dpnp_iface.py

@@ -72,6 +72,7 @@
     "get_usm_ndarray_or_scalar",
     "is_supported_array_or_scalar",
     "is_supported_array_type",
+    "not_implemented_for_cuda_backend",
     "synchronize_array_data",
 ]
 
@@ -800,6 +801,21 @@ def is_supported_array_type(a):
     return isinstance(a, (dpnp_array, dpt.usm_ndarray))
 
 
+def not_implemented_for_cuda_backend(obj):
+    """
+    Raise NotImplementedError for cuda devices.
+
+    Parameters
+    ----------
+    obj : {SyclDevice, SyclQueue, dpnp.ndarray, usm_ndarray}
+        An input object with sycl_device property to check device backend.
+
+    """
+    sycl_device = getattr(obj, "sycl_device", None)
+    if sycl_device is not None and "cuda" in sycl_device.backend.name:
+        raise NotImplementedError("function not implemented for cuda backend")
+
+
 def synchronize_array_data(a):
     """
     The dpctl interface was reworked to make asynchronous execution.

dpnp/dpnp_iface_indexing.py

@@ -173,6 +173,9 @@ def choose(x1, choices, out=None, mode="raise"):
     :obj:`dpnp.take_along_axis` : Preferable if choices is an array.
 
     """
+
+    dpnp.not_implemented_for_cuda_backend(x1)
+
     x1_desc = dpnp.get_dpnp_descriptor(x1, copy_when_nondefault_queue=False)
 
     choices_list = []

dpnp/dpnp_iface_libmath.py

@@ -78,6 +78,8 @@ def erf(in_array1):
 
     """
 
+    dpnp.not_implemented_for_cuda_backend(in_array1)
+
     x1_desc = dpnp.get_dpnp_descriptor(
         in_array1, copy_when_strides=False, copy_when_nondefault_queue=False
     )

dpnp/dpnp_iface_mathematical.py

@@ -2477,6 +2477,8 @@ def modf(x1, **kwargs):
 
     """
 
+    dpnp.not_implemented_for_cuda_backend(x1)
+
     x1_desc = dpnp.get_dpnp_descriptor(x1, copy_when_nondefault_queue=False)
     if x1_desc and not kwargs:
         return dpnp_modf(x1_desc)

dpnp/dpnp_iface_sorting.py

@@ -174,6 +174,8 @@ def partition(x1, kth, axis=-1, kind="introselect", order=None):
 
     """
 
+    dpnp.not_implemented_for_cuda_backend(x1)
+
     x1_desc = dpnp.get_dpnp_descriptor(x1, copy_when_nondefault_queue=False)
     if x1_desc:
         if not isinstance(kth, int):

dpnp/dpnp_iface_statistics.py

@@ -371,6 +371,8 @@ def correlate(x1, x2, mode="valid"):
 
     """
 
+    dpnp.not_implemented_for_cuda_backend(x1)
+
     x1_desc = dpnp.get_dpnp_descriptor(x1, copy_when_nondefault_queue=False)
     x2_desc = dpnp.get_dpnp_descriptor(x2, copy_when_nondefault_queue=False)
     if x1_desc and x2_desc:
@@ -655,6 +657,8 @@ def median(x1, axis=None, out=None, overwrite_input=False, keepdims=False):
 
     """
 
+    dpnp.not_implemented_for_cuda_backend(x1)
+
     x1_desc = dpnp.get_dpnp_descriptor(x1, copy_when_nondefault_queue=False)
     if x1_desc:
         if axis is not None:

tests/conftest.py

@@ -64,13 +64,20 @@ def pytest_collection_modifyitems(config, items):
         test_path, "skipped_tests_gpu_no_fp64.tbl"
     )
 
+    # global skip file for cuda backend
+    test_exclude_file_cuda = os.path.join(
+        test_path, "skipped_tests_cuda.tbl"
+    )
+
     dev = dpctl.select_default_device()
     is_cpu = dev.is_cpu
     is_gpu_no_fp64 = not dev.has_aspect_fp64
+    is_cuda = "cuda" in str(dev.backend.name)
 
     print("")
     print(f"DPNP current device is CPU: {is_cpu}")
     print(f"DPNP current device is GPU without fp64 support: {is_gpu_no_fp64}")
+    print(f"DPNP current device is GPU with cuda backend: {is_cuda}")
     print(f"DPNP version: {dpnp.__version__}, location: {dpnp}")
     print(f"NumPy version: {numpy.__version__}, location: {numpy}")
     print(f"Python version: {sys.version}")
@@ -81,6 +88,10 @@ def pytest_collection_modifyitems(config, items):
             excluded_tests.extend(
                 get_excluded_tests(test_exclude_file_gpu_no_fp64)
             )
+        if is_cuda:
+            excluded_tests.extend(
+                get_excluded_tests(test_exclude_file_cuda)
+            )
     else:
         excluded_tests.extend(get_excluded_tests(test_exclude_file))
 

tests/skipped_tests_cuda.tbl

@@ -39,6 +39,9 @@
 for device in available_devices:
     if device.default_selector_score < 0:
         pass
+    elif device.backend.name in "cuda":
+        valid_devices = [device]
+        break
     elif device.backend.name not in list_of_backend_str:
         pass
     elif device.device_type.name not in list_of_device_type_str: