From f787c8caf5455e20f9bbf279f8f8d9527a232211 Mon Sep 17 00:00:00 2001
From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com>
Date: Tue, 1 Jul 2025 12:11:10 -0400
Subject: [PATCH 1/3] Add torch 2.8 rc / 2.9 nightly to tests

---
 .github/workflows/tests.yml | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index a5299195b..b6291a6f7 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -101,8 +101,8 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-22.04, ubuntu-22.04-arm, windows-2025, macos-15]
-        # Test with the oldest supported torch version and the two newest.
-        torch_version: ["2.2.2", "2.6.0", "2.7.1"]
+        # Test with the oldest supported torch version, the newest stable, current RC, and nightly.
+        torch_version: ["2.2.2", "2.7.1", "2.8.0", "2.9.0"]
         include:
           - os: ubuntu-22.04
             arch: x86_64
@@ -144,7 +144,7 @@ jobs:
 
       - name: Install dependencies
         run: |
-          pip install torch==${{ matrix.torch_version }} --index-url https://download.pytorch.org/whl/cpu
+          pip install torch~=${{ matrix.torch_version }}.dev0 --index-url https://download.pytorch.org/whl/${{ (matrix.torch_version == '2.8.0' && 'test/cpu') || (matrix.torch_version == '2.9.0' && 'nightly/cpu') || '/cpu' }}
           pip install -e ".[test]"
           pip install pytest-cov
 
@@ -372,6 +372,9 @@ jobs:
             pypi_index: "https://download.pytorch.org/whl/cu128"
           - cuda_version: "12.9.1"
             torch_version: "2.8.0"
+            pypi_index: "https://download.pytorch.org/whl/test/cu129"
+          - cuda_version: "12.9.1"
+            torch_version: "2.9.0"
             pypi_index: "https://download.pytorch.org/whl/nightly/cu129"
 
 

From 40171b5dfbaa3870ea4c2c242230b71cea991931 Mon Sep 17 00:00:00 2001
From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com>
Date: Tue, 1 Jul 2025 12:20:27 -0400
Subject: [PATCH 2/3] Update tests.yml

---
 .github/workflows/tests.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index b6291a6f7..40082f381 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -144,7 +144,7 @@ jobs:
 
       - name: Install dependencies
         run: |
-          pip install torch~=${{ matrix.torch_version }}.dev0 --index-url https://download.pytorch.org/whl/${{ (matrix.torch_version == '2.8.0' && 'test/cpu') || (matrix.torch_version == '2.9.0' && 'nightly/cpu') || '/cpu' }}
+          pip install torch~=${{ matrix.torch_version }}.dev0 --index-url https://download.pytorch.org/whl/${{ (matrix.torch_version == '2.8.0' && 'test/cpu') || (matrix.torch_version == '2.9.0' && 'nightly/cpu') || 'cpu' }}
           pip install -e ".[test]"
           pip install pytest-cov
 

From 7ae59170e2fc67b1e1188c24fb22fd31b810ad23 Mon Sep 17 00:00:00 2001
From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com>
Date: Tue, 1 Jul 2025 12:57:27 -0400
Subject: [PATCH 3/3] Update tests.yml

---
 .github/workflows/tests.yml | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 40082f381..847c7ef7a 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -101,8 +101,8 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-22.04, ubuntu-22.04-arm, windows-2025, macos-15]
-        # Test with the oldest supported torch version, the newest stable, current RC, and nightly.
-        torch_version: ["2.2.2", "2.7.1", "2.8.0", "2.9.0"]
+        # Test with the oldest supported torch version, the newest two stable/RC.
+        torch_version: ["2.2.2", "2.7.1", "2.8.0"]
         include:
           - os: ubuntu-22.04
             arch: x86_64
@@ -144,7 +144,7 @@ jobs:
 
       - name: Install dependencies
         run: |
-          pip install torch~=${{ matrix.torch_version }}.dev0 --index-url https://download.pytorch.org/whl/${{ (matrix.torch_version == '2.8.0' && 'test/cpu') || (matrix.torch_version == '2.9.0' && 'nightly/cpu') || 'cpu' }}
+          pip install torch==${{ matrix.torch_version }} --index-url https://download.pytorch.org/whl/${{ (matrix.torch_version == '2.8.0' && 'test/cpu') || 'cpu' }}
           pip install -e ".[test]"
           pip install pytest-cov
 
@@ -373,9 +373,6 @@ jobs:
           - cuda_version: "12.9.1"
             torch_version: "2.8.0"
             pypi_index: "https://download.pytorch.org/whl/test/cu129"
-          - cuda_version: "12.9.1"
-            torch_version: "2.9.0"
-            pypi_index: "https://download.pytorch.org/whl/nightly/cu129"
 
 
           # Linux L40S runners