Run unit tests against PyTorch nightlies as part of our nightly CI (p…

…ytorch#1569)
felipemello1 · Sep 12, 2024 · f729ce1 · f729ce1
1 parent 6b43a1c
commit f729ce1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 55 deletions.
diff --git a/.github/workflows/gpu_test.yaml b/.github/workflows/gpu_test.yaml
@@ -1,6 +1,9 @@
 name: GPU tests
 
 on:
+  schedule:
+    # Runs at midnight every day
+    - cron:  '0 0 * * *'
   push:
     branches: [ main ]
   pull_request:
@@ -24,7 +27,10 @@ jobs:
     strategy:
       matrix:
         python-version: ['3.9', '3.10', '3.11']
-        torch-version: ["stable"]
+        torch-version: ["stable", "nightly"]
+        # Do not run against nightlies on PR
+        exclude:
+          - torch-version: ${{ github.event_name == 'pull_request' && 'nightly' }}
     steps:
       - name: Check out repo
         uses: actions/checkout@v3
@@ -39,7 +45,7 @@ jobs:
         run: python -m pip install --upgrade pip
       - name: Install torch nightly
         if: ${{ matrix.torch-version == 'nightly' }}
-        run: python -m pip install --pre torch torchvision torchao --index-url https://download.pytorch.org/whl/nightly/cu118
+        run: python -m pip install --pre torch torchvision torchao --index-url https://download.pytorch.org/whl/nightly/cu121
       - name: Install torch stable
         if: ${{ matrix.torch-version == 'stable' }}
         run: python -m pip install torch torchvision torchao

diff --git a/.github/workflows/recipe_test_nightly.yaml b/.github/workflows/recipe_test_nightly.yaml