Unverified Commit ed398d28 authored by Matthew Douglas's avatar Matthew Douglas Committed by GitHub
Browse files

CI: Test with PyTorch 2.8.0 RC (#1693)

* Add torch 2.8 rc / 2.9 nightly to tests

* Update tests.yml

* Update tests.yml
parent e28d4d91
...@@ -101,8 +101,8 @@ jobs: ...@@ -101,8 +101,8 @@ jobs:
fail-fast: false fail-fast: false
matrix: matrix:
os: [ubuntu-22.04, ubuntu-22.04-arm, windows-2025, macos-15] os: [ubuntu-22.04, ubuntu-22.04-arm, windows-2025, macos-15]
# Test with the oldest supported torch version and the two newest. # Test with the oldest supported torch version, the newest two stable/RC.
torch_version: ["2.2.2", "2.6.0", "2.7.1"] torch_version: ["2.2.2", "2.7.1", "2.8.0"]
include: include:
- os: ubuntu-22.04 - os: ubuntu-22.04
arch: x86_64 arch: x86_64
...@@ -144,7 +144,7 @@ jobs: ...@@ -144,7 +144,7 @@ jobs:
- name: Install dependencies - name: Install dependencies
run: | run: |
pip install torch==${{ matrix.torch_version }} --index-url https://download.pytorch.org/whl/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 -e ".[test]"
pip install pytest-cov pip install pytest-cov
...@@ -372,7 +372,7 @@ jobs: ...@@ -372,7 +372,7 @@ jobs:
pypi_index: "https://download.pytorch.org/whl/cu128" pypi_index: "https://download.pytorch.org/whl/cu128"
- cuda_version: "12.9.1" - cuda_version: "12.9.1"
torch_version: "2.8.0" torch_version: "2.8.0"
pypi_index: "https://download.pytorch.org/whl/nightly/cu129" pypi_index: "https://download.pytorch.org/whl/test/cu129"
# Linux L40S runners # Linux L40S runners
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment