@@ -49,14 +49,14 @@ jobs:
49
49
curr_branch : ${{ github.head_ref || github.ref_name }}
50
50
curr_ref_type : ${{ github.ref_type }}
51
51
52
- linux-focal-cuda12_4 -py3_10-gcc9-build :
53
- name : linux-focal-cuda12.4 -py3.10-gcc9
52
+ linux-focal-cuda12_6 -py3_10-gcc9-build :
53
+ name : linux-focal-cuda12.6 -py3.10-gcc9
54
54
uses : ./.github/workflows/_linux-build.yml
55
55
needs : get-label-type
56
56
with :
57
57
runner_prefix : " ${{ needs.get-label-type.outputs.label-type }}"
58
- build-environment : linux-focal-cuda12.4 -py3.10-gcc9
59
- docker-image-name : pytorch-linux-focal-cuda12.4 -cudnn9-py3-gcc9
58
+ build-environment : linux-focal-cuda12.6 -py3.10-gcc9
59
+ docker-image-name : pytorch-linux-focal-cuda12.6 -cudnn9-py3-gcc9
60
60
test-matrix : |
61
61
{ include: [
62
62
{ config: "nogpu_AVX512", shard: 1, num_shards: 2, runner: "${{ needs.get-label-type.outputs.label-type }}linux.4xlarge" },
@@ -67,16 +67,16 @@ jobs:
67
67
]}
68
68
secrets : inherit
69
69
70
- linux-focal-cuda12_4 -py3_10-gcc9-test :
71
- name : linux-focal-cuda12.4 -py3.10-gcc9
70
+ linux-focal-cuda12_6 -py3_10-gcc9-test :
71
+ name : linux-focal-cuda12.6 -py3.10-gcc9
72
72
uses : ./.github/workflows/_linux-test.yml
73
73
needs :
74
- - linux-focal-cuda12_4 -py3_10-gcc9-build
74
+ - linux-focal-cuda12_6 -py3_10-gcc9-build
75
75
- target-determination
76
76
with :
77
- build-environment : linux-focal-cuda12.4 -py3.10-gcc9
78
- docker-image : ${{ needs.linux-focal-cuda12_4 -py3_10-gcc9-build.outputs.docker-image }}
79
- test-matrix : ${{ needs.linux-focal-cuda12_4 -py3_10-gcc9-build.outputs.test-matrix }}
77
+ build-environment : linux-focal-cuda12.6 -py3.10-gcc9
78
+ docker-image : ${{ needs.linux-focal-cuda12_6 -py3_10-gcc9-build.outputs.docker-image }}
79
+ test-matrix : ${{ needs.linux-focal-cuda12_6 -py3_10-gcc9-build.outputs.test-matrix }}
80
80
secrets : inherit
81
81
82
82
linux-focal-cuda11_8-py3_9-gcc9-build :
@@ -170,16 +170,16 @@ jobs:
170
170
test-matrix : ${{ needs.linux-focal-rocm6_3-py3_10-build.outputs.test-matrix }}
171
171
secrets : inherit
172
172
173
- linux-focal-cuda12_4 -py3_10-gcc9-experimental-split-build :
174
- name : linux-focal-cuda12.4 -py3.10-gcc9-experimental-split-build
173
+ linux-focal-cuda12_6 -py3_10-gcc9-experimental-split-build :
174
+ name : linux-focal-cuda12.6 -py3.10-gcc9-experimental-split-build
175
175
uses : ./.github/workflows/_linux-build.yml
176
176
needs : get-label-type
177
177
if : false # See https://github.com/pytorch/pytorch/issues/138750
178
178
with :
179
179
runner_prefix : " ${{ needs.get-label-type.outputs.label-type }}"
180
180
use_split_build : true
181
- build-environment : linux-focal-cuda12.4 -py3.10-gcc9
182
- docker-image-name : pytorch-linux-focal-cuda12.4 -cudnn9-py3-gcc9
181
+ build-environment : linux-focal-cuda12.6 -py3.10-gcc9
182
+ docker-image-name : pytorch-linux-focal-cuda12.6 -cudnn9-py3-gcc9
183
183
test-matrix : |
184
184
{ include: [
185
185
{ config: "nogpu_AVX512", shard: 1, num_shards: 2, runner: "${{ needs.get-label-type.outputs.label-type }}linux.4xlarge" },
@@ -190,16 +190,16 @@ jobs:
190
190
]}
191
191
secrets : inherit
192
192
193
- linux-focal-cuda12_4 -py3_10-gcc9-experimental-split-build-test :
194
- name : linux-focal-cuda12.4 -py3.10-gcc9-experimental-split-build
193
+ linux-focal-cuda12_6 -py3_10-gcc9-experimental-split-build-test :
194
+ name : linux-focal-cuda12.6 -py3.10-gcc9-experimental-split-build
195
195
uses : ./.github/workflows/_linux-test.yml
196
196
needs :
197
- - linux-focal-cuda12_4 -py3_10-gcc9-experimental-split-build
197
+ - linux-focal-cuda12_6 -py3_10-gcc9-experimental-split-build
198
198
- target-determination
199
199
with :
200
- build-environment : linux-focal-cuda12.4 -py3.10-gcc9-experimental-split-build
201
- docker-image : ${{ needs.linux-focal-cuda12_4 -py3_10-gcc9-experimental-split-build.outputs.docker-image }}
202
- test-matrix : ${{ needs.linux-focal-cuda12_4 -py3_10-gcc9-experimental-split-build.outputs.test-matrix }}
200
+ build-environment : linux-focal-cuda12.6 -py3.10-gcc9-experimental-split-build
201
+ docker-image : ${{ needs.linux-focal-cuda12_6 -py3_10-gcc9-experimental-split-build.outputs.docker-image }}
202
+ test-matrix : ${{ needs.linux-focal-cuda12_6 -py3_10-gcc9-experimental-split-build.outputs.test-matrix }}
203
203
secrets : inherit
204
204
205
205
@@ -265,14 +265,14 @@ jobs:
265
265
test-matrix : ${{ needs.linux-focal-cuda11_8-py3_10-gcc9-experimental-split-build.outputs.test-matrix }}
266
266
secrets : inherit
267
267
268
- linux-focal-cuda12_4 -py3-gcc9-slow-gradcheck-build :
269
- name : linux-focal-cuda12.4 -py3-gcc9-slow-gradcheck
268
+ linux-focal-cuda12_6 -py3-gcc9-slow-gradcheck-build :
269
+ name : linux-focal-cuda12.6 -py3-gcc9-slow-gradcheck
270
270
uses : ./.github/workflows/_linux-build.yml
271
271
needs : get-label-type
272
272
with :
273
273
runner_prefix : " ${{ needs.get-label-type.outputs.label-type }}"
274
- build-environment : linux-focal-cuda12.4 -py3-gcc9-slow-gradcheck
275
- docker-image-name : pytorch-linux-focal-cuda12.4 -cudnn9-py3-gcc9
274
+ build-environment : linux-focal-cuda12.6 -py3-gcc9-slow-gradcheck
275
+ docker-image-name : pytorch-linux-focal-cuda12.6 -cudnn9-py3-gcc9
276
276
cuda-arch-list : 8.6
277
277
test-matrix : |
278
278
{ include: [
@@ -287,28 +287,28 @@ jobs:
287
287
]}
288
288
secrets : inherit
289
289
290
- linux-focal-cuda12_4 -py3-gcc9-slow-gradcheck-test :
291
- name : linux-focal-cuda12.4 -py3-gcc9-slow-gradcheck
290
+ linux-focal-cuda12_6 -py3-gcc9-slow-gradcheck-test :
291
+ name : linux-focal-cuda12.6 -py3-gcc9-slow-gradcheck
292
292
uses : ./.github/workflows/_linux-test.yml
293
293
needs :
294
- - linux-focal-cuda12_4 -py3-gcc9-slow-gradcheck-build
294
+ - linux-focal-cuda12_6 -py3-gcc9-slow-gradcheck-build
295
295
- target-determination
296
296
with :
297
- build-environment : linux-focal-cuda12.4 -py3-gcc9-slow-gradcheck
298
- docker-image : ${{ needs.linux-focal-cuda12_4 -py3-gcc9-slow-gradcheck-build.outputs.docker-image }}
299
- test-matrix : ${{ needs.linux-focal-cuda12_4 -py3-gcc9-slow-gradcheck-build.outputs.test-matrix }}
297
+ build-environment : linux-focal-cuda12.6 -py3-gcc9-slow-gradcheck
298
+ docker-image : ${{ needs.linux-focal-cuda12_6 -py3-gcc9-slow-gradcheck-build.outputs.docker-image }}
299
+ test-matrix : ${{ needs.linux-focal-cuda12_6 -py3-gcc9-slow-gradcheck-build.outputs.test-matrix }}
300
300
timeout-minutes : 300
301
301
secrets : inherit
302
302
303
- linux-focal-cuda12_4 -py3_10-gcc9-bazel-test :
304
- name : linux-focal-cuda12.4 -py3.10-gcc9-bazel-test
303
+ linux-focal-cuda12_6 -py3_10-gcc9-bazel-test :
304
+ name : linux-focal-cuda12.6 -py3.10-gcc9-bazel-test
305
305
uses : ./.github/workflows/_bazel-build-test.yml
306
306
needs : get-label-type
307
307
with :
308
308
runner : " ${{ needs.get-label-type.outputs.label-type }}linux.large"
309
- build-environment : linux-focal-cuda12.4 -py3.10-gcc9-bazel-test
310
- docker-image-name : pytorch-linux-focal-cuda12.4 -cudnn9-py3-gcc9
311
- cuda-version : " 12.4 "
309
+ build-environment : linux-focal-cuda12.6 -py3.10-gcc9-bazel-test
310
+ docker-image-name : pytorch-linux-focal-cuda12.6 -cudnn9-py3-gcc9
311
+ cuda-version : " 12.6 "
312
312
test-matrix : |
313
313
{ include: [
314
314
{ config: "default", shard: 1, num_shards: 1, runner: "${{ needs.get-label-type.outputs.label-type }}linux.4xlarge.nvidia.gpu" },