Unskip fft tests for cuda

Author: vlad-perevezentsev

dpnp/tests/test_fft.py

@@ -14,7 +14,6 @@
     get_all_dtypes,
     get_complex_dtypes,
     get_float_dtypes,
-    is_cuda_device,
 )
 
 
@@ -443,11 +442,6 @@ def setup_method(self):
     @pytest.mark.parametrize("norm", [None, "backward", "forward", "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
-            else:
-                pytest.skip("SAT-7587")
         a_np = generate_random_numpy_array((2, 3, 4, 5), dtype, order)
         a = dpnp.array(a_np)
 
@@ -482,9 +476,6 @@ 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")
         a_np = generate_random_numpy_array((2, 3, 4, 5), dtype=numpy.complex64)
         a = dpnp.array(a_np)
 
@@ -504,11 +495,6 @@ 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")
         a_np = generate_random_numpy_array((2, 3, 4, 5), dtype=numpy.complex64)
         a = dpnp.array(a_np)
 
@@ -1082,9 +1068,6 @@ 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")
         x = numpy.random.uniform(-10, 10, 120)
         a_np = numpy.array(x, dtype=numpy.float32).reshape(2, 3, 4, 5)
         a = dpnp.asarray(a_np)

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

@@ -5,7 +5,7 @@
 import pytest
 
 import dpnp as cupy
-from dpnp.tests.helper import has_support_aspect64, is_cuda_device
+from dpnp.tests.helper import has_support_aspect64
 from dpnp.tests.third_party.cupy import testing
 from dpnp.tests.third_party.cupy.testing._loops import _wraps_partial
 
@@ -413,8 +413,6 @@ class TestFft2:
         type_check=has_support_aspect64(),
     )
     def test_fft2(self, xp, dtype, order, enable_nd):
-        if is_cuda_device() and self.shape == (2, 3, 4, 5):
-            pytest.skip("SAT-7587")
         # assert config.enable_nd_planning == enable_nd
         a = testing.shaped_random(self.shape, xp, dtype)
         if order == "F":
@@ -442,8 +440,6 @@ def test_fft2(self, xp, dtype, order, enable_nd):
         type_check=has_support_aspect64(),
     )
     def test_ifft2(self, xp, dtype, order, enable_nd):
-        if is_cuda_device() and self.shape == (2, 3, 4, 5):
-            pytest.skip("SAT-7587")
         # assert config.enable_nd_planning == enable_nd
         a = testing.shaped_random(self.shape, xp, dtype)
         if order == "F":
@@ -507,8 +503,6 @@ class TestFftn:
         type_check=has_support_aspect64(),
     )
     def test_fftn(self, xp, dtype, order, enable_nd):
-        if is_cuda_device() and self.shape == (2, 3, 4, 5):
-            pytest.skip("SAT-7587")
         # assert config.enable_nd_planning == enable_nd
         a = testing.shaped_random(self.shape, xp, dtype)
         if order == "F":
@@ -536,8 +530,6 @@ def test_fftn(self, xp, dtype, order, enable_nd):
         type_check=has_support_aspect64(),
     )
     def test_ifftn(self, xp, dtype, order, enable_nd):
-        if is_cuda_device() and self.shape == (2, 3, 4, 5):
-            pytest.skip("SAT-7587")
         # assert config.enable_nd_planning == enable_nd
         a = testing.shaped_random(self.shape, xp, dtype)
         if order == "F":