Skip to content

Commit a3c0e67

Browse files
NicolasHugdatumbox
andauthored
increase tol (#3320)
Co-authored-by: Vasilis Vryniotis <[email protected]>
1 parent 9aa0722 commit a3c0e67

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

test/test_ops.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -549,7 +549,7 @@ def _test_forward_with_batchsize(self, device, contiguous, batch_sz, dtype):
549549
out_channels = 2
550550
kernel_size = (3, 2)
551551
groups = 2
552-
tol = 1e-3 if dtype is torch.half else 1e-5
552+
tol = 2e-3 if dtype is torch.half else 1e-5
553553

554554
layer = ops.DeformConv2d(in_channels, out_channels, kernel_size, stride=stride, padding=padding,
555555
dilation=dilation, groups=groups).to(device=x.device, dtype=dtype)
@@ -651,7 +651,6 @@ def test_compare_cpu_cuda_grads(self):
651651

652652
@unittest.skipIf(not torch.cuda.is_available(), "CUDA unavailable")
653653
def test_autocast(self):
654-
set_rng_seed(0)
655654
for dtype in (torch.float, torch.half):
656655
with torch.cuda.amp.autocast():
657656
self._test_forward(torch.device("cuda"), False, dtype=dtype)

0 commit comments

Comments
 (0)