Skip to content

Commit c7ecb1e

Browse files
committed
Add torch2.9 in regression tests
1 parent 0c081fd commit c7ecb1e

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

.github/workflows/regression_test.yml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,12 @@ jobs:
7777
gpu-arch-type: "cuda"
7878
gpu-arch-version: "12.6"
7979
dev-requirements-overrides: ""
80+
- name: CUDA 2.9
81+
runs-on: linux.g5.12xlarge.nvidia.gpu
82+
torch-spec: 'torch==2.9.0'
83+
gpu-arch-type: "cuda"
84+
gpu-arch-version: "12.6"
85+
dev-requirements-overrides: ""
8086

8187
- name: CPU 2.6
8288
runs-on: linux.4xlarge
@@ -96,6 +102,12 @@ jobs:
96102
gpu-arch-type: "cpu"
97103
gpu-arch-version: ""
98104
dev-requirements-overrides: ""
105+
- name: CPU 2.9
106+
runs-on: linux.4xlarge
107+
torch-spec: 'torch==2.9.0 --index-url https://download.pytorch.org/whl/cpu'
108+
gpu-arch-type: "cpu"
109+
gpu-arch-version: ""
110+
dev-requirements-overrides: ""
99111

100112
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
101113
with:

0 commit comments

Comments
 (0)