From c4492213c0138cd19ef97527fb649d1df3c9d02d Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Sat, 10 Aug 2024 16:23:36 +0100 Subject: [PATCH 1/2] Only run 'extended' test if `run-extended` label is set (#8579) --- .github/workflows/tests.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 533a44a9e84..93d6afb00d5 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -139,6 +139,7 @@ jobs: unittests-extended: uses: pytorch/test-infra/.github/workflows/linux_job.yml@main + if: contains(github.event.pull_request.labels.*.name, 'run-extended') with: repository: pytorch/vision test-infra-ref: main From c9936da9fe44ae830360dd8a9eea69103e5f50a9 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Sat, 10 Aug 2024 17:07:41 +0100 Subject: [PATCH 2/2] Skip flaky test_autocast_cpu test (#8580) --- test/test_ops.py | 2 ++ 1 file changed, 2 insertions(+) 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"),