Skip to content

Commit 8988be2

Browse files
Handle new fft tests for CUDA
1 parent f33bcfa commit 8988be2

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -507,6 +507,8 @@ class TestFftn:
507507
type_check=has_support_aspect64(),
508508
)
509509
def test_fftn(self, xp, dtype, order, enable_nd):
510+
if is_cuda_device() and self.shape == (2, 3, 4, 5):
511+
pytest.skip("SAT-7587")
510512
# assert config.enable_nd_planning == enable_nd
511513
a = testing.shaped_random(self.shape, xp, dtype)
512514
if order == "F":
@@ -534,6 +536,8 @@ def test_fftn(self, xp, dtype, order, enable_nd):
534536
type_check=has_support_aspect64(),
535537
)
536538
def test_ifftn(self, xp, dtype, order, enable_nd):
539+
if is_cuda_device() and self.shape == (2, 3, 4, 5):
540+
pytest.skip("SAT-7587")
537541
# assert config.enable_nd_planning == enable_nd
538542
a = testing.shaped_random(self.shape, xp, dtype)
539543
if order == "F":

0 commit comments

Comments
 (0)