Skips spectral tests to prevent ROCm build from timing out (#42667)

Summary:
Per title.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/42667

Reviewed By: ailzhang

Differential Revision: D22978531

Pulled By: mruberry

fbshipit-source-id: 0c3ba116836ed6c433e2c6a0e1a0f2e3c94c7803
This commit is contained in:
Mike Ruberry
2020-08-06 12:38:43 -07:00
committed by Facebook GitHub Bot
parent 40b6dacb50
commit 85a00c4c92

View File

@ -25,6 +25,7 @@ import torch.fft
class TestFFT(TestCase):
exact_dtype = True
@skipCUDAIfRocm
def test_fft_function_clobbered(self, device):
t = torch.randn((100, 2), device=device)
eager_result = fft_fn(t, 1)
@ -38,6 +39,7 @@ class TestFFT(TestCase):
with self.assertRaisesRegex(TypeError, "'module' object is not callable"):
torch.fft(t, 1)
@skipCUDAIfRocm
@unittest.skipIf(not TEST_NUMPY, 'NumPy not found')
@precisionOverride({torch.complex64: 1e-4})
@dtypes(torch.complex64, torch.complex128)
@ -65,6 +67,7 @@ class TestFFT(TestCase):
self.assertEqual(actual, expected, exact_dtype=(dtype is torch.complex128))
# Note: NumPy will throw a ValueError for an empty input
@skipCUDAIfRocm
@skipCPUIfNoMkl
@onlyOnCPUAndCUDA
@dtypes(torch.complex64, torch.complex128)