frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm] enable test_cufft_plan_cache test (#57520)

Browse Source 
Summary:
This pr enables the test_cufft_plan_cache in test_spectral suite.

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

Reviewed By: ejguan

Differential Revision: D28936128

Pulled By: ngimel

fbshipit-source-id: c843ab31c50855b624a986155c17c8d24e89a2ac
This commit is contained in:
Michael Melesse
2021-06-08 14:38:04 -07:00
committed by Facebook GitHub Bot
parent 43274ca145
commit 1bb1a9e22b
2 changed files with 0 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/test_spectral_ops.py
View File

@ -602,7 +602,6 @@ class TestFFT(TestCase):
self._test_fft_ifft_rfft_irfft(device, dtype)
@deviceCountAtLeast(1)
@skipCUDAIfRocm
@onlyCUDA
@dtypes(torch.double)
def test_cufft_plan_cache(self, devices, dtype):