Skip to content

Commit 8dc8316

Browse files
Add torch 2.9+cu130 to tests
1 parent 82d82ef commit 8dc8316

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

.github/workflows/tests.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
name: Unit tests
1+
name: Nightly Tests
22

33
on:
44
workflow_dispatch:
@@ -102,7 +102,7 @@ jobs:
102102
matrix:
103103
os: [ubuntu-22.04, ubuntu-22.04-arm, windows-2025, macos-15]
104104
# Test with the oldest supported torch version, the newest two stable/RC.
105-
torch_version: ["2.3.1", "2.7.1", "2.8.0"]
105+
torch_version: ["2.3.1", "2.8.0", "2.9.0"]
106106
include:
107107
- os: ubuntu-22.04
108108
arch: x86_64
@@ -136,15 +136,15 @@ jobs:
136136
- name: Setup Python
137137
uses: actions/setup-python@v5
138138
with:
139-
python-version: 3.9
139+
python-version: 3.10
140140

141141
- name: Setup MSVC
142142
if: startsWith(matrix.os, 'windows')
143143
uses: ilammy/[email protected] # to use cl for torch.compile
144144

145145
- name: Install dependencies
146146
run: |
147-
pip install torch==${{ matrix.torch_version }} --index-url https://download.pytorch.org/whl/cpu
147+
pip install torch==${{ matrix.torch_version }} --index-url https://download.pytorch.org/whl/${{ (matrix.torch_version == '2.9.0' && 'test/cpu') || 'cpu' }}
148148
pip install -e ".[test]"
149149
pip install pytest-cov
150150
@@ -429,7 +429,7 @@ jobs:
429429
- name: Setup Python
430430
uses: actions/setup-python@v5
431431
with:
432-
python-version: 3.9
432+
python-version: 3.10
433433

434434
- name: Install dependencies
435435
run: |

0 commit comments

Comments
 (0)