Skip to content

Commit c9936da

Browse files
authored
Skip flaky test_autocast_cpu test (#8580)
1 parent c449221 commit c9936da

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

Diff for: test/test_ops.py

+2
Original file line numberDiff line numberDiff line change
@@ -507,11 +507,13 @@ def test_autocast(self, aligned, deterministic, x_dtype, rois_dtype):
507507
rois_dtype=rois_dtype,
508508
)
509509

510+
@pytest.mark.skip(reason="1/5000 flaky failure")
510511
@pytest.mark.parametrize("aligned", (True, False))
511512
@pytest.mark.parametrize("deterministic", (True, False))
512513
@pytest.mark.parametrize("x_dtype", (torch.float, torch.bfloat16))
513514
@pytest.mark.parametrize("rois_dtype", (torch.float, torch.bfloat16))
514515
def test_autocast_cpu(self, aligned, deterministic, x_dtype, rois_dtype):
516+
515517
with torch.cpu.amp.autocast():
516518
self.test_forward(
517519
torch.device("cpu"),

0 commit comments

Comments
 (0)