Update fft tests to run on CUDA

Author: vlad-perevezentsev

dpnp/tests/test_fft.py

@@ -13,6 +13,7 @@
     get_all_dtypes,
     get_complex_dtypes,
     get_float_dtypes,
+    is_cuda_device,
 )
 
 
@@ -470,6 +471,13 @@ def setup_method(self):
     @pytest.mark.parametrize("norm", ["forward", "backward", "ortho"])
     @pytest.mark.parametrize("order", ["C", "F"])
     def test_fftn(self, dtype, axes, norm, order):
+        if is_cuda_device():
+            if order == "C" and axes == (0, 1, 2):
+                pass
+            elif order == "F" and axes == (-1, -4, -2):
+                pass
+            else:
+                pytest.skip("SAT-7587")
         x1 = numpy.random.uniform(-10, 10, 120)
         x2 = numpy.random.uniform(-10, 10, 120)
         a_np = numpy.array(x1 + 1j * x2, dtype=dtype).reshape(
@@ -512,6 +520,9 @@ def test_fftn_repeated_axes(self, axes):
     @pytest.mark.parametrize("axes", [(2, 3, 3, 2), (0, 0, 3, 3)])
     @pytest.mark.parametrize("s", [(5, 4, 3, 3), (7, 8, 10, 9)])
     def test_fftn_repeated_axes_with_s(self, axes, s):
+        if is_cuda_device():
+            if axes == (0, 0, 3, 3) and s == (7, 8, 10, 9):
+                pytest.skip("SAT-7587")
         x1 = numpy.random.uniform(-10, 10, 120)
         x2 = numpy.random.uniform(-10, 10, 120)
         a_np = numpy.array(x1 + 1j * x2, dtype=numpy.complex64).reshape(
@@ -535,6 +546,11 @@ def test_fftn_repeated_axes_with_s(self, axes, s):
     @pytest.mark.parametrize("axes", [(0, 1, 2, 3), (1, 2, 1, 2), (2, 2, 2, 3)])
     @pytest.mark.parametrize("s", [(2, 3, 4, 5), (5, 4, 7, 8), (2, 5, 1, 2)])
     def test_fftn_out(self, axes, s):
+        if is_cuda_device():
+            if axes == (0, 1, 2, 3):
+                pytest.skip("SAT-7587")
+            elif s == (2, 5, 1, 2) and axes in [(1, 2, 1, 2), (2, 2, 2, 3)]:
+                pytest.skip("SAT-7587")
         x1 = numpy.random.uniform(-10, 10, 120)
         x2 = numpy.random.uniform(-10, 10, 120)
         a_np = numpy.array(x1 + 1j * x2, dtype=numpy.complex64).reshape(
@@ -1141,6 +1157,9 @@ def test_rfftn_repeated_axes_with_s(self, axes, s):
     @pytest.mark.parametrize("axes", [(0, 1, 2, 3), (1, 2, 1, 2), (2, 2, 2, 3)])
     @pytest.mark.parametrize("s", [(2, 3, 4, 5), (5, 6, 7, 9), (2, 5, 1, 2)])
     def test_rfftn_out(self, axes, s):
+        if is_cuda_device():
+            if axes == (0, 1, 2, 3) and s == (2, 5, 1, 2):
+                pytest.skip("SAT-7587")
         x1 = numpy.random.uniform(-10, 10, 120)
         a_np = numpy.array(x1, dtype=numpy.float32).reshape(2, 3, 4, 5)
         a = dpnp.asarray(a_np)

dpnp/tests/third_party/cupy/fft_tests/test_fft.py

@@ -4,7 +4,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
 
 
@@ -162,6 +162,8 @@ class TestFft2:
         type_check=has_support_aspect64(),
     )
     def test_fft2(self, xp, dtype, order):
+        if is_cuda_device() and self.shape == (2, 3, 4, 5):
+            pytest.skip("SAT-7587")
         a = testing.shaped_random(self.shape, xp, dtype)
         if order == "F":
             a = xp.asfortranarray(a)
@@ -186,6 +188,8 @@ def test_fft2(self, xp, dtype, order):
         type_check=has_support_aspect64(),
     )
     def test_ifft2(self, xp, dtype, order):
+        if is_cuda_device() and self.shape == (2, 3, 4, 5):
+            pytest.skip("SAT-7587")
         a = testing.shaped_random(self.shape, xp, dtype)
         if order == "F":
             a = xp.asfortranarray(a)
@@ -249,6 +253,8 @@ class TestFftn:
         type_check=has_support_aspect64(),
     )
     def test_fftn(self, xp, dtype, order):
+        if is_cuda_device() and self.shape == (2, 3, 4, 5):
+            pytest.skip("SAT-7587")
         a = testing.shaped_random(self.shape, xp, dtype)
         if order == "F":
             a = xp.asfortranarray(a)
@@ -273,6 +279,8 @@ def test_fftn(self, xp, dtype, order):
         type_check=has_support_aspect64(),
     )
     def test_ifftn(self, xp, dtype, order):
+        if is_cuda_device() and self.shape == (2, 3, 4, 5):
+            pytest.skip("SAT-7587")
         a = testing.shaped_random(self.shape, xp, dtype)
         if order == "F":
             a = xp.asfortranarray(a)