|
19 | 19 | TEST_WITH_ROCM, _assertGradAndGradgradChecks |
20 | 20 | from torch.testing._internal.common_cuda import TEST_CUDA |
21 | 21 | from torch.testing._internal.common_device_type import expectedAlertNondeterministic |
22 | | -from torch.autograd.gradcheck import get_numerical_jacobian, iter_tensors |
| 22 | +from torch.autograd.gradcheck import get_numerical_jacobian, iter_tensors, gradcheck, gradgradcheck |
23 | 23 | from torch.autograd import Variable |
24 | 24 | import torch.backends.cudnn |
25 | 25 |
|
@@ -4249,7 +4249,6 @@ def padding3d_circular(input, pad): |
4249 | 4249 | target_fn=lambda: torch.zeros(5, 68, 64, 64, dtype=torch.float), |
4250 | 4250 | reference_fn=lambda i, t, m: ((i - t).abs().pow(2).sum() / |
4251 | 4251 | (i.numel() if get_reduction(m) == 'mean' else 1)), |
4252 | | - check_forward_only=True, |
4253 | 4252 | desc='prec', |
4254 | 4253 | check_bfloat16=TEST_WITH_ROCM, |
4255 | 4254 | ), |
@@ -4964,7 +4963,7 @@ class CriterionTest(InputVariableMixin, TestBase): |
4964 | 4963 | def __init__(self, *args, **kwargs): |
4965 | 4964 | super().__init__(*args, **kwargs) |
4966 | 4965 | self.should_test_cuda = kwargs.get('test_cuda', True) |
4967 | | - self.check_forward_only = kwargs.get('check_forward_only', True) |
| 4966 | + self.check_forward_only = kwargs.get('check_forward_only', False) |
4968 | 4967 | self.check_gradgrad = kwargs.get('check_gradgrad', True) |
4969 | 4968 | self.check_half = kwargs.get('check_half', True) |
4970 | 4969 | self.check_bfloat16 = kwargs.get('check_bfloat16', False) |
|
0 commit comments