Skip to content

Commit 0fe1899

Browse files
ethanwee1jeffdaily
andauthored
Move ROCm 7.1,7.2->7.2,7.14 (#8451)
Updates nightly, test, and release binary matrices from ROCm 7.1/7.2 to 7.2/7.14, enabling torchvision validation and ROCm 7.14 support for the PyTorch 2.14 release. --------- Co-authored-by: Jeff Daily <jeff.daily@amd.com>
1 parent e89b6ee commit 0fe1899

3 files changed

Lines changed: 161 additions & 161 deletions

File tree

tools/scripts/generate_binary_build_matrix.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@
5050
}
5151

5252
ROCM_ARCHES_DICT = {
53-
"nightly": ["7.1", "7.2"],
53+
"nightly": ["7.2", "7.14"],
5454
"test": ["7.1", "7.2"],
5555
"release": ["7.1", "7.2"],
5656
}

tools/tests/assets/build_matrix_linux_wheel_cuda.json

Lines changed: 80 additions & 80 deletions
Original file line numberDiff line numberDiff line change
@@ -59,27 +59,27 @@
5959
{
6060
"python_version": "3.10",
6161
"gpu_arch_type": "rocm",
62-
"gpu_arch_version": "7.1",
63-
"desired_cuda": "rocm7.1",
64-
"container_image": "pytorch/manylinux2_28-builder:rocm7.1",
62+
"gpu_arch_version": "7.2",
63+
"desired_cuda": "rocm7.2",
64+
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
6565
"package_type": "manywheel",
66-
"build_name": "manywheel-py3_10-rocm7_1",
66+
"build_name": "manywheel-py3_10-rocm7_2",
6767
"validation_runner": "linux.2xlarge",
68-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.1",
68+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
6969
"channel": "nightly",
7070
"upload_to_base_bucket": "no",
7171
"stable_version": "2.13.0"
7272
},
7373
{
7474
"python_version": "3.10",
7575
"gpu_arch_type": "rocm",
76-
"gpu_arch_version": "7.2",
77-
"desired_cuda": "rocm7.2",
78-
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
76+
"gpu_arch_version": "7.14",
77+
"desired_cuda": "rocm7.14",
78+
"container_image": "pytorch/manylinux2_28-builder:rocm7.14",
7979
"package_type": "manywheel",
80-
"build_name": "manywheel-py3_10-rocm7_2",
80+
"build_name": "manywheel-py3_10-rocm7_14",
8181
"validation_runner": "linux.2xlarge",
82-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
82+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.14",
8383
"channel": "nightly",
8484
"upload_to_base_bucket": "no",
8585
"stable_version": "2.13.0"
@@ -143,27 +143,27 @@
143143
{
144144
"python_version": "3.11",
145145
"gpu_arch_type": "rocm",
146-
"gpu_arch_version": "7.1",
147-
"desired_cuda": "rocm7.1",
148-
"container_image": "pytorch/manylinux2_28-builder:rocm7.1",
146+
"gpu_arch_version": "7.2",
147+
"desired_cuda": "rocm7.2",
148+
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
149149
"package_type": "manywheel",
150-
"build_name": "manywheel-py3_11-rocm7_1",
150+
"build_name": "manywheel-py3_11-rocm7_2",
151151
"validation_runner": "linux.2xlarge",
152-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.1",
152+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
153153
"channel": "nightly",
154154
"upload_to_base_bucket": "no",
155155
"stable_version": "2.13.0"
156156
},
157157
{
158158
"python_version": "3.11",
159159
"gpu_arch_type": "rocm",
160-
"gpu_arch_version": "7.2",
161-
"desired_cuda": "rocm7.2",
162-
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
160+
"gpu_arch_version": "7.14",
161+
"desired_cuda": "rocm7.14",
162+
"container_image": "pytorch/manylinux2_28-builder:rocm7.14",
163163
"package_type": "manywheel",
164-
"build_name": "manywheel-py3_11-rocm7_2",
164+
"build_name": "manywheel-py3_11-rocm7_14",
165165
"validation_runner": "linux.2xlarge",
166-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
166+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.14",
167167
"channel": "nightly",
168168
"upload_to_base_bucket": "no",
169169
"stable_version": "2.13.0"
@@ -227,27 +227,27 @@
227227
{
228228
"python_version": "3.12",
229229
"gpu_arch_type": "rocm",
230-
"gpu_arch_version": "7.1",
231-
"desired_cuda": "rocm7.1",
232-
"container_image": "pytorch/manylinux2_28-builder:rocm7.1",
230+
"gpu_arch_version": "7.2",
231+
"desired_cuda": "rocm7.2",
232+
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
233233
"package_type": "manywheel",
234-
"build_name": "manywheel-py3_12-rocm7_1",
234+
"build_name": "manywheel-py3_12-rocm7_2",
235235
"validation_runner": "linux.2xlarge",
236-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.1",
236+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
237237
"channel": "nightly",
238238
"upload_to_base_bucket": "no",
239239
"stable_version": "2.13.0"
240240
},
241241
{
242242
"python_version": "3.12",
243243
"gpu_arch_type": "rocm",
244-
"gpu_arch_version": "7.2",
245-
"desired_cuda": "rocm7.2",
246-
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
244+
"gpu_arch_version": "7.14",
245+
"desired_cuda": "rocm7.14",
246+
"container_image": "pytorch/manylinux2_28-builder:rocm7.14",
247247
"package_type": "manywheel",
248-
"build_name": "manywheel-py3_12-rocm7_2",
248+
"build_name": "manywheel-py3_12-rocm7_14",
249249
"validation_runner": "linux.2xlarge",
250-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
250+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.14",
251251
"channel": "nightly",
252252
"upload_to_base_bucket": "no",
253253
"stable_version": "2.13.0"
@@ -311,27 +311,27 @@
311311
{
312312
"python_version": "3.13",
313313
"gpu_arch_type": "rocm",
314-
"gpu_arch_version": "7.1",
315-
"desired_cuda": "rocm7.1",
316-
"container_image": "pytorch/manylinux2_28-builder:rocm7.1",
314+
"gpu_arch_version": "7.2",
315+
"desired_cuda": "rocm7.2",
316+
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
317317
"package_type": "manywheel",
318-
"build_name": "manywheel-py3_13-rocm7_1",
318+
"build_name": "manywheel-py3_13-rocm7_2",
319319
"validation_runner": "linux.2xlarge",
320-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.1",
320+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
321321
"channel": "nightly",
322322
"upload_to_base_bucket": "no",
323323
"stable_version": "2.13.0"
324324
},
325325
{
326326
"python_version": "3.13",
327327
"gpu_arch_type": "rocm",
328-
"gpu_arch_version": "7.2",
329-
"desired_cuda": "rocm7.2",
330-
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
328+
"gpu_arch_version": "7.14",
329+
"desired_cuda": "rocm7.14",
330+
"container_image": "pytorch/manylinux2_28-builder:rocm7.14",
331331
"package_type": "manywheel",
332-
"build_name": "manywheel-py3_13-rocm7_2",
332+
"build_name": "manywheel-py3_13-rocm7_14",
333333
"validation_runner": "linux.2xlarge",
334-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
334+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.14",
335335
"channel": "nightly",
336336
"upload_to_base_bucket": "no",
337337
"stable_version": "2.13.0"
@@ -395,27 +395,27 @@
395395
{
396396
"python_version": "3.14",
397397
"gpu_arch_type": "rocm",
398-
"gpu_arch_version": "7.1",
399-
"desired_cuda": "rocm7.1",
400-
"container_image": "pytorch/manylinux2_28-builder:rocm7.1",
398+
"gpu_arch_version": "7.2",
399+
"desired_cuda": "rocm7.2",
400+
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
401401
"package_type": "manywheel",
402-
"build_name": "manywheel-py3_14-rocm7_1",
402+
"build_name": "manywheel-py3_14-rocm7_2",
403403
"validation_runner": "linux.2xlarge",
404-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.1",
404+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
405405
"channel": "nightly",
406406
"upload_to_base_bucket": "no",
407407
"stable_version": "2.13.0"
408408
},
409409
{
410410
"python_version": "3.14",
411411
"gpu_arch_type": "rocm",
412-
"gpu_arch_version": "7.2",
413-
"desired_cuda": "rocm7.2",
414-
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
412+
"gpu_arch_version": "7.14",
413+
"desired_cuda": "rocm7.14",
414+
"container_image": "pytorch/manylinux2_28-builder:rocm7.14",
415415
"package_type": "manywheel",
416-
"build_name": "manywheel-py3_14-rocm7_2",
416+
"build_name": "manywheel-py3_14-rocm7_14",
417417
"validation_runner": "linux.2xlarge",
418-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
418+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.14",
419419
"channel": "nightly",
420420
"upload_to_base_bucket": "no",
421421
"stable_version": "2.13.0"
@@ -479,27 +479,27 @@
479479
{
480480
"python_version": "3.14t",
481481
"gpu_arch_type": "rocm",
482-
"gpu_arch_version": "7.1",
483-
"desired_cuda": "rocm7.1",
484-
"container_image": "pytorch/manylinux2_28-builder:rocm7.1",
482+
"gpu_arch_version": "7.2",
483+
"desired_cuda": "rocm7.2",
484+
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
485485
"package_type": "manywheel",
486-
"build_name": "manywheel-py3_14t-rocm7_1",
486+
"build_name": "manywheel-py3_14t-rocm7_2",
487487
"validation_runner": "linux.2xlarge",
488-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.1",
488+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
489489
"channel": "nightly",
490490
"upload_to_base_bucket": "no",
491491
"stable_version": "2.13.0"
492492
},
493493
{
494494
"python_version": "3.14t",
495495
"gpu_arch_type": "rocm",
496-
"gpu_arch_version": "7.2",
497-
"desired_cuda": "rocm7.2",
498-
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
496+
"gpu_arch_version": "7.14",
497+
"desired_cuda": "rocm7.14",
498+
"container_image": "pytorch/manylinux2_28-builder:rocm7.14",
499499
"package_type": "manywheel",
500-
"build_name": "manywheel-py3_14t-rocm7_2",
500+
"build_name": "manywheel-py3_14t-rocm7_14",
501501
"validation_runner": "linux.2xlarge",
502-
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
502+
"installation": "pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.14",
503503
"channel": "nightly",
504504
"upload_to_base_bucket": "no",
505505
"stable_version": "2.13.0"
@@ -563,27 +563,27 @@
563563
{
564564
"python_version": "3.15",
565565
"gpu_arch_type": "rocm",
566-
"gpu_arch_version": "7.1",
567-
"desired_cuda": "rocm7.1",
568-
"container_image": "pytorch/manylinux2_28-builder:rocm7.1",
566+
"gpu_arch_version": "7.2",
567+
"desired_cuda": "rocm7.2",
568+
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
569569
"package_type": "manywheel",
570-
"build_name": "manywheel-py3_15-rocm7_1",
570+
"build_name": "manywheel-py3_15-rocm7_2",
571571
"validation_runner": "linux.2xlarge",
572-
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.1",
572+
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
573573
"channel": "nightly",
574574
"upload_to_base_bucket": "no",
575575
"stable_version": "2.13.0"
576576
},
577577
{
578578
"python_version": "3.15",
579579
"gpu_arch_type": "rocm",
580-
"gpu_arch_version": "7.2",
581-
"desired_cuda": "rocm7.2",
582-
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
580+
"gpu_arch_version": "7.14",
581+
"desired_cuda": "rocm7.14",
582+
"container_image": "pytorch/manylinux2_28-builder:rocm7.14",
583583
"package_type": "manywheel",
584-
"build_name": "manywheel-py3_15-rocm7_2",
584+
"build_name": "manywheel-py3_15-rocm7_14",
585585
"validation_runner": "linux.2xlarge",
586-
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
586+
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.14",
587587
"channel": "nightly",
588588
"upload_to_base_bucket": "no",
589589
"stable_version": "2.13.0"
@@ -647,27 +647,27 @@
647647
{
648648
"python_version": "3.15t",
649649
"gpu_arch_type": "rocm",
650-
"gpu_arch_version": "7.1",
651-
"desired_cuda": "rocm7.1",
652-
"container_image": "pytorch/manylinux2_28-builder:rocm7.1",
650+
"gpu_arch_version": "7.2",
651+
"desired_cuda": "rocm7.2",
652+
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
653653
"package_type": "manywheel",
654-
"build_name": "manywheel-py3_15t-rocm7_1",
654+
"build_name": "manywheel-py3_15t-rocm7_2",
655655
"validation_runner": "linux.2xlarge",
656-
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.1",
656+
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
657657
"channel": "nightly",
658658
"upload_to_base_bucket": "no",
659659
"stable_version": "2.13.0"
660660
},
661661
{
662662
"python_version": "3.15t",
663663
"gpu_arch_type": "rocm",
664-
"gpu_arch_version": "7.2",
665-
"desired_cuda": "rocm7.2",
666-
"container_image": "pytorch/manylinux2_28-builder:rocm7.2",
664+
"gpu_arch_version": "7.14",
665+
"desired_cuda": "rocm7.14",
666+
"container_image": "pytorch/manylinux2_28-builder:rocm7.14",
667667
"package_type": "manywheel",
668-
"build_name": "manywheel-py3_15t-rocm7_2",
668+
"build_name": "manywheel-py3_15t-rocm7_14",
669669
"validation_runner": "linux.2xlarge",
670-
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.2",
670+
"installation": "pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.14",
671671
"channel": "nightly",
672672
"upload_to_base_bucket": "no",
673673
"stable_version": "2.13.0"

0 commit comments

Comments
 (0)