diff --git a/test/test_ops.py b/test/test_ops.py index 99b259f73f5..22ba0ce2d5b 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -507,11 +507,13 @@ def test_autocast(self, aligned, deterministic, x_dtype, rois_dtype): rois_dtype=rois_dtype, ) + @pytest.mark.skip(reason="1/5000 flaky failure") @pytest.mark.parametrize("aligned", (True, False)) @pytest.mark.parametrize("deterministic", (True, False)) @pytest.mark.parametrize("x_dtype", (torch.float, torch.bfloat16)) @pytest.mark.parametrize("rois_dtype", (torch.float, torch.bfloat16)) def test_autocast_cpu(self, aligned, deterministic, x_dtype, rois_dtype): + with torch.cpu.amp.autocast(): self.test_forward( torch.device("cpu"),