diff --git a/tests/third_party/cupy/fft_tests/test_fft.py b/tests/third_party/cupy/fft_tests/test_fft.py index 59f37b57bd0..a6a528f817d 100644 --- a/tests/third_party/cupy/fft_tests/test_fft.py +++ b/tests/third_party/cupy/fft_tests/test_fft.py @@ -125,7 +125,7 @@ def test_ifft(self, xp, dtype): {"shape": (3, 4), "s": (1, 5), "axes": None}, {"shape": (3, 4), "s": None, "axes": (-2, -1)}, {"shape": (3, 4), "s": None, "axes": (-1, -2)}, - # {"shape": (3, 4), "s": None, "axes": (0,)}, # mkl_fft gh-109 + {"shape": (3, 4), "s": None, "axes": (0,)}, {"shape": (3, 4), "s": None, "axes": None}, # {"shape": (3, 4), "s": None, "axes": ()}, # mkl_fft gh-108 {"shape": (2, 3, 4), "s": None, "axes": None}, @@ -133,10 +133,10 @@ def test_ifft(self, xp, dtype): {"shape": (2, 3, 4), "s": (1, 4, 10), "axes": None}, {"shape": (2, 3, 4), "s": None, "axes": (-3, -2, -1)}, {"shape": (2, 3, 4), "s": None, "axes": (-1, -2, -3)}, - # {"shape": (2, 3, 4), "s": None, "axes": (0, 1)}, # mkl_fft gh-109 + {"shape": (2, 3, 4), "s": None, "axes": (0, 1)}, {"shape": (2, 3, 4), "s": None, "axes": None}, # {"shape": (2, 3, 4), "s": None, "axes": ()}, # mkl_fft gh-108 - # {"shape": (2, 3, 4), "s": (2, 3), "axes": (0, 1, 2)}, # mkl_fft gh-109 + {"shape": (2, 3, 4), "s": (2, 3), "axes": (0, 1, 2)}, {"shape": (2, 3, 4, 5), "s": None, "axes": None}, # {"shape": (0, 5), "s": None, "axes": None}, # mkl_fft gh-110 # {"shape": (2, 0, 5), "s": None, "axes": None}, # mkl_fft gh-110 @@ -211,7 +211,7 @@ def test_ifft2(self, xp, dtype, order): {"shape": (3, 4), "s": None, "axes": (-2, -1)}, {"shape": (3, 4), "s": None, "axes": (-1, -2)}, {"shape": (3, 4), "s": None, "axes": [-1, -2]}, - # {"shape": (3, 4), "s": None, "axes": (0,)}, # mkl_fft gh-109 + {"shape": (3, 4), "s": None, "axes": (0,)}, # {"shape": (3, 4), "s": None, "axes": ()}, # mkl_fft gh-108 {"shape": (3, 4), "s": None, "axes": None}, {"shape": (2, 3, 4), "s": None, "axes": None}, @@ -219,11 +219,11 @@ def test_ifft2(self, xp, dtype, order): {"shape": (2, 3, 4), "s": (1, 4, 10), "axes": None}, {"shape": (2, 3, 4), "s": None, "axes": (-3, -2, -1)}, {"shape": (2, 3, 4), "s": None, "axes": (-1, -2, -3)}, - # {"shape": (2, 3, 4), "s": None, "axes": (-1, -3)}, # mkl_fft gh-109 - # {"shape": (2, 3, 4), "s": None, "axes": (0, 1)}, # mkl_fft gh-109 + {"shape": (2, 3, 4), "s": None, "axes": (-1, -3)}, + {"shape": (2, 3, 4), "s": None, "axes": (0, 1)}, {"shape": (2, 3, 4), "s": None, "axes": None}, # {"shape": (2, 3, 4), "s": None, "axes": ()}, # mkl_fft gh-108 - # {"shape": (2, 3, 4), "s": (2, 3), "axes": (0, 1, 2)}, # mkl_fft gh-109 + {"shape": (2, 3, 4), "s": (2, 3), "axes": (0, 1, 2)}, {"shape": (2, 3, 4), "s": (4, 3, 2), "axes": (2, 0, 1)}, {"shape": (2, 3, 4, 5), "s": None, "axes": None}, # {"shape": (0, 5), "s": None, "axes": None}, # mkl_fft gh-110