Skip to content

Commit ee39d63

Browse files
NicolasHugfacebook-github-bot
authored andcommitted
[fbsync] Skip flaky test_autocast_cpu test (#8580)
Reviewed By: ahmadsharif1 Differential Revision: D61331607 fbshipit-source-id: cb66215023ea77336b9d16180c362cba3dde6201
1 parent 4199772 commit ee39d63

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/test_ops.py

Lines changed: 2 additions & 0 deletions
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)