pytorch/torch/testing/_internal
Nikita Shulga 172ed51a17 Mark parts of spectral tests as slow (#46509)
Summary:
According to https://app.circleci.com/pipelines/github/pytorch/pytorch/228154/workflows/31951076-b633-4391-bd0d-b2953c940876/jobs/8290059
TestFFTCUDA.test_fftn_backward_cuda_complex128 takes 242 seconds to finish, where most of the time spent checking 2nd gradient

Refactor common part of test_fft_backward and test_fftn_backward into _fft_grad_check_helper
Introduce `slowAwareTest` decorator
Split test into fast and slow parts by checking 2nd degree gradient only during the slow part

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

Reviewed By: walterddr

Differential Revision: D24378901

Pulled By: malfet

fbshipit-source-id: 606670c2078480219905f63b9b278b835e760a66
2020-10-19 10:11:46 -07:00
..
codegen Enable CUDA Fuser for ROCm (#45965) 2020-10-08 10:41:56 -07:00
data remediation of S205607 2020-07-17 17:19:47 -07:00
distributed [Flaky tests] Fix test_all_gather_timeout test (#45989) 2020-10-16 09:02:46 -07:00
generated remediation of S205607 2020-07-17 17:19:47 -07:00
test_module remediation of S205607 2020-07-17 17:19:47 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
autocast_test_lists.py [PyTorch] Enable type check for autocast_test_lists (#45107) 2020-09-22 16:54:26 -07:00
common_cuda.py Adjust TF32 tests (#44240) 2020-09-24 10:25:58 -07:00
common_device_type.py Enable type check for torch.testing._internal.common_device_type. (#44911) 2020-09-18 13:42:11 -07:00
common_distributed.py Disallow creation of ProcessGroupNCCL without GPUs. (#45642) 2020-10-05 12:05:48 -07:00
common_methods_invocations.py Fix deprecated warnings for nan_to_num (#46309) 2020-10-16 06:07:14 -07:00
common_nn.py Fix perfornance issue of GroupNorm on CUDA when feature map is small. (#46170) 2020-10-14 23:34:33 -07:00
common_quantization.py Revert D24290811: [quant][eagermode] Move custom_module registration to prepare/convert_custom_config_dict 2020-10-14 16:42:55 -07:00
common_quantized.py minor style edits to torch/testing/_internal/common_quantized.py (#44807) 2020-09-17 08:02:43 -07:00
common_utils.py Mark parts of spectral tests as slow (#46509) 2020-10-19 10:11:46 -07:00
dist_utils.py [RPC profiling] Extend RPC profiling to support async function execution over RPC. (#44664) 2020-09-25 13:19:26 -07:00
expecttest.py Add some more docs to expecttest. (#46263) 2020-10-13 15:17:11 -07:00
hypothesis_utils.py [quant] conv_transpose1d_prepack / conv_transpose1d_unpack (#40360) 2020-09-11 04:12:28 -07:00
jit_metaprogramming_utils.py enable mypy check for jit_metaprogramming_utils (#44752) 2020-09-16 15:44:37 -07:00
jit_utils.py [JIT] Improve class type annotation inference (#45940) 2020-10-14 23:28:47 -07:00
te_utils.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00