Avoid using dpnp.random in cupy tests on CUDA

Author: vlad-perevezentsev

dpnp/tests/third_party/cupy/creation_tests/test_from_data.py

@@ -6,7 +6,7 @@
 import pytest
 
 import dpnp as cupy
-from dpnp.tests.helper import has_support_aspect64
+from dpnp.tests.helper import has_support_aspect64, is_cuda_device
 from dpnp.tests.third_party.cupy import testing
 
 
@@ -521,7 +521,13 @@ def test_copy_multigpu(self, dtype, order):
         q1 = dpctl.SyclQueue()
         q2 = dpctl.SyclQueue()
 
-        src = cupy.random.uniform(-1, 1, (2, 3), device=q1).astype(dtype)
+        # TODO: remove it once the issue with CUDA support is resolved
+        # for dpnp.random
+        if is_cuda_device():
+            src_np = numpy.random.uniform(-1, 1, (2, 3)).astype(dtype)
+            src = cupy.array(src_np, device=q1)
+        else:
+            src = cupy.random.uniform(-1, 1, (2, 3), device=q1).astype(dtype)
         dst = cupy.copy(src, order, device=q2)
         testing.assert_allclose(src, dst, rtol=0, atol=0)
 

dpnp/tests/third_party/cupy/math_tests/test_rational.py

@@ -1,17 +1,29 @@
 import unittest
 
+import numpy
 import pytest
 
 import dpnp as cupy
+from dpnp.tests.helper import is_cuda_device
 from dpnp.tests.third_party.cupy import testing
 
 
 class TestRational(unittest.TestCase):
     @pytest.mark.usefixtures("allow_fall_back_on_numpy")
     @testing.for_dtypes(["?", "e", "f", "d", "F", "D"])
     def test_gcd_dtype_check(self, dtype):
-        a = cupy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
-        b = cupy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
+        # TODO: remove it once the issue with CUDA support is resolved
+        # for dpnp.random
+        if is_cuda_device():
+            a = cupy.asarray(
+                numpy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
+            )
+            b = cupy.asarray(
+                numpy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
+            )
+        else:
+            a = cupy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
+            b = cupy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
         with pytest.raises(ValueError):
             cupy.gcd(a, b)
 
@@ -25,8 +37,16 @@ def test_gcd_check_boundary_cases(self, xp, dtype):
     @pytest.mark.usefixtures("allow_fall_back_on_numpy")
     @testing.for_dtypes(["?", "e", "f", "d", "F", "D"])
     def test_lcm_dtype_check(self, dtype):
-        a = cupy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
-        b = cupy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
+        if is_cuda_device():
+            a = cupy.asarray(
+                numpy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
+            )
+            b = cupy.asarray(
+                numpy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
+            )
+        else:
+            a = cupy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
+            b = cupy.random.randint(-10, 10, size=(10, 10)).astype(dtype)
         with pytest.raises(ValueError):
             cupy.lcm(a, b)
 

dpnp/tests/third_party/cupy/testing/_random.py

@@ -8,6 +8,7 @@
 import numpy
 
 import dpnp as cupy
+from dpnp.tests.helper import is_cuda_device
 
 _old_python_random_state = None
 _old_numpy_random_state = None
@@ -33,11 +34,15 @@ def do_setup(deterministic=True):
     if not deterministic:
         random.seed()
         numpy.random.seed()
-        cupy.random.seed()
+        # TODO: remove it once the issue with CUDA support is resolved
+        # for dpnp.random
+        if not is_cuda_device():
+            cupy.random.seed()
     else:
         random.seed(99)
         numpy.random.seed(100)
-        cupy.random.seed(101)
+        if not is_cuda_device():
+            cupy.random.seed(101)
 
 
 def do_teardown():