mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 21:49:24 +08:00
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:
committed by
Facebook GitHub Bot
parent
40b6dacb50
commit
85a00c4c92
@ -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)
|
||||
|
Reference in New Issue
Block a user