2024-08-11 nightly release (0033ddc)

pytorch · Aug 11, 2024 · fedf101 · fedf101
1 parent b841460
commit fedf101
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git 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

diff --git 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"),