[DEBUG] flaky gaussian blur

.github/workflows/prototype-tests.yml

@@ -7,13 +7,20 @@ jobs:
   prototype:
     strategy:
       matrix:
-        os:
-          - ubuntu-latest
-          - windows-latest
-          - macos-latest
+        runner:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
       fail-fast: false
 
-    runs-on: ${{ matrix.os }}
+    runs-on: ubuntu-latest
 
     steps:
       - name: Set up python
@@ -43,41 +50,12 @@ jobs:
         id: setup
         run: exit 0
 
-      - name: Run prototype features tests
-        shell: bash
-        run: |
-          pytest \
-            --durations=20 \
-            --cov=torchvision/prototype/features \
-            --cov-report=term-missing \
-            test/test_prototype_features*.py
-
-      - name: Run prototype datasets tests
-        if: success() || ( failure() && steps.setup.conclusion == 'success' )
-        shell: bash
+      - name: Test non-determinism
         run: |
-          pytest \
-            --durations=20 \
-            --cov=torchvision/prototype/datasets \
-            --cov-report=term-missing \
-            test/test_prototype_datasets*.py
+          pip install tqdm
+          python test_gaussian_blur_non_determinism.py
 
       - name: Run prototype transforms tests
         if: success() || ( failure() && steps.setup.conclusion == 'success' )
         shell: bash
-        run: |
-          pytest \
-            --durations=20 \
-            --cov=torchvision/prototype/transforms \
-            --cov-report=term-missing \
-            test/test_prototype_transforms*.py
-
-      - name: Run prototype models tests
-        if: success() || ( failure() && steps.setup.conclusion == 'success' )
-        shell: bash
-        run: |
-          pytest \
-            --durations=20 \
-            --cov=torchvision/prototype/models \
-            --cov-report=term-missing \
-            test/test_prototype_models*.py
+        run: pytest -rP test/test_prototype_transforms_functional.py::TestKernels::test_scripted_vs_eager -k gauss

test/test_prototype_transforms_functional.py

@@ -86,7 +86,7 @@ class TestKernels:
 
     @sample_inputs
     @pytest.mark.parametrize("device", cpu_and_gpu())
-    def test_scripted_vs_eager(self, info, args_kwargs, device):
+    def test_scripted_vs_eager(self, request, info, args_kwargs, device):
         kernel_eager = info.kernel
         kernel_scripted = script(kernel_eager)
 
@@ -95,6 +95,12 @@ def test_scripted_vs_eager(self, info, args_kwargs, device):
         actual = kernel_scripted(*args, **kwargs)
         expected = kernel_eager(*args, **kwargs)
 
+        import pathlib
+
+        artifacts = pathlib.Path(__file__).parent / "artifacts"
+        artifacts.mkdir(exist_ok=True)
+        torch.save((args, kwargs, actual, expected), str(artifacts / f"{request.node.name}.pt"))
+
         assert_close(actual, expected, **info.closeness_kwargs)
 
     def _unbatch(self, batch, *, data_dims):