Skip to content

Commit ab2ffb7

Browse files
committed
Actually run backward criterion tests.
These were (apparently) accidentally turned off two years ago in #9287. [ghstack-poisoned]
1 parent 1e3f728 commit ab2ffb7

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

torch/testing/_internal/common_nn.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
TEST_WITH_ROCM, _assertGradAndGradgradChecks
2020
from torch.testing._internal.common_cuda import TEST_CUDA
2121
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
2323
from torch.autograd import Variable
2424
import torch.backends.cudnn
2525

@@ -4249,7 +4249,6 @@ def padding3d_circular(input, pad):
42494249
target_fn=lambda: torch.zeros(5, 68, 64, 64, dtype=torch.float),
42504250
reference_fn=lambda i, t, m: ((i - t).abs().pow(2).sum() /
42514251
(i.numel() if get_reduction(m) == 'mean' else 1)),
4252-
check_forward_only=True,
42534252
desc='prec',
42544253
check_bfloat16=TEST_WITH_ROCM,
42554254
),
@@ -4964,7 +4963,7 @@ class CriterionTest(InputVariableMixin, TestBase):
49644963
def __init__(self, *args, **kwargs):
49654964
super().__init__(*args, **kwargs)
49664965
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)
49684967
self.check_gradgrad = kwargs.get('check_gradgrad', True)
49694968
self.check_half = kwargs.get('check_half', True)
49704969
self.check_bfloat16 = kwargs.get('check_bfloat16', False)

0 commit comments

Comments
 (0)