@@ -15,15 +15,14 @@ jobs:
1515 matrix :
1616 python-version :
1717 - " 3.10"
18- - " 3.11"
19- - " 3.12"
18+ - " 3.14"
2019 runner : ["linux.12xlarge"]
2120 gpu-arch-type : ["cpu"]
2221 include :
2322 - python-version : " 3.10"
2423 runner : linux.g5.4xlarge.nvidia.gpu
2524 gpu-arch-type : cuda
26- gpu-arch-version : " 12.6 "
25+ gpu-arch-version : " 12.8 "
2726 fail-fast : false
2827 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
2928 permissions :
8079 matrix :
8180 python-version :
8281 - " 3.10"
83- # TODO put back 3.11 (See blame)
84- # - "3.11"
85- - " 3.12"
82+ - " 3.14"
8683 runner : ["macos-m1-stable"]
8784 fail-fast : false
8885 uses : pytorch/test-infra/.github/workflows/macos_job.yml@main
@@ -105,16 +102,15 @@ jobs:
105102 matrix :
106103 python-version :
107104 - " 3.10"
108- - " 3.11"
109- - " 3.12"
105+ - " 3.14"
110106 runner : ["windows.4xlarge"]
111107 gpu-arch-type : ["cpu"]
112108 # TODO: put GPU testing back
113109 # include:
114- # - python-version: "3.9 "
110+ # - python-version: "3.10 "
115111 # runner: windows.g5.4xlarge.nvidia.gpu
116112 # gpu-arch-type: cuda
117- # gpu-arch-version: "11 .8"
113+ # gpu-arch-version: "12 .8"
118114 fail-fast : false
119115 uses : pytorch/test-infra/.github/workflows/windows_job.yml@main
120116 permissions :
@@ -138,40 +134,6 @@ jobs:
138134
139135 ./.github/scripts/unittest.sh
140136
141- # onnx:
142- # uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
143- # permissions:
144- # id-token: write
145- # contents: read
146- # with:
147- # repository: pytorch/vision
148- # test-infra-ref: main
149- # script: |
150- # set -euo pipefail
151-
152- # export PYTHON_VERSION=3.10
153- # export GPU_ARCH_TYPE=cpu
154- # export GPU_ARCH_VERSION=''
155-
156- # ./.github/scripts/setup-env.sh
157-
158- # # Prepare conda
159- # CONDA_PATH=$(which conda)
160- # eval "$(${CONDA_PATH} shell.bash hook)"
161- # conda activate ci
162-
163- # echo '::group::Install ONNX'
164- # pip install --progress-bar=off onnx onnxruntime
165- # echo '::endgroup::'
166-
167- # echo '::group::Install testing utilities'
168- # pip install --progress-bar=off pytest "numpy<2"
169- # echo '::endgroup::'
170-
171- # echo '::group::Run ONNX tests'
172- # pytest --junit-xml="${RUNNER_TEST_RESULTS_DIR}/test-results.xml" -v --durations=25 test/test_onnx.py
173- # echo '::endgroup::'
174-
175137 unittests-extended :
176138 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
177139 permissions :
0 commit comments