Skip to content

Commit 1465ab7

Browse files
jeffdailyjeffdaily
authored
[ROCm] update binary build matrix to 6.4 and 7.0 (#7237)
1 parent 60e16ae commit 1465ab7

3 files changed

Lines changed: 143 additions & 143 deletions

File tree

‎tools/scripts/generate_binary_build_matrix.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,9 @@
3333
}
3434

3535
ROCM_ARCHES_DICT = {
36-
"nightly": ["6.3", "6.4"],
37-
"test": ["6.3", "6.4"],
38-
"release": ["6.3", "6.4"],
36+
"nightly": ["6.4", "7.0"],
37+
"test": ["6.4", "7.0"],
38+
"release": ["6.4", "7.0"],
3939
}
4040

4141
CUDA_CUDNN_VERSIONS = {

‎tools/tests/assets/build_matrix_linux_wheel_cuda.json‎

Lines changed: 70 additions & 70 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": "6.3",
63-
"desired_cuda": "rocm6.3",
64-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
62+
"gpu_arch_version": "6.4",
63+
"desired_cuda": "rocm6.4",
64+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
6565
"package_type": "manywheel",
66-
"build_name": "manywheel-py3_10-rocm6_3",
66+
"build_name": "manywheel-py3_10-rocm6_4",
6767
"validation_runner": "linux.2xlarge",
68-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
68+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
6969
"channel": "nightly",
7070
"upload_to_base_bucket": "no",
7171
"stable_version": "2.8.0"
7272
},
7373
{
7474
"python_version": "3.10",
7575
"gpu_arch_type": "rocm",
76-
"gpu_arch_version": "6.4",
77-
"desired_cuda": "rocm6.4",
78-
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
76+
"gpu_arch_version": "7.0",
77+
"desired_cuda": "rocm7.0",
78+
"container_image": "pytorch/manylinux2_28-builder:rocm7.0",
7979
"package_type": "manywheel",
80-
"build_name": "manywheel-py3_10-rocm6_4",
80+
"build_name": "manywheel-py3_10-rocm7_0",
8181
"validation_runner": "linux.2xlarge",
82-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
82+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm7.0",
8383
"channel": "nightly",
8484
"upload_to_base_bucket": "no",
8585
"stable_version": "2.8.0"
@@ -143,27 +143,27 @@
143143
{
144144
"python_version": "3.11",
145145
"gpu_arch_type": "rocm",
146-
"gpu_arch_version": "6.3",
147-
"desired_cuda": "rocm6.3",
148-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
146+
"gpu_arch_version": "6.4",
147+
"desired_cuda": "rocm6.4",
148+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
149149
"package_type": "manywheel",
150-
"build_name": "manywheel-py3_11-rocm6_3",
150+
"build_name": "manywheel-py3_11-rocm6_4",
151151
"validation_runner": "linux.2xlarge",
152-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
152+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
153153
"channel": "nightly",
154154
"upload_to_base_bucket": "no",
155155
"stable_version": "2.8.0"
156156
},
157157
{
158158
"python_version": "3.11",
159159
"gpu_arch_type": "rocm",
160-
"gpu_arch_version": "6.4",
161-
"desired_cuda": "rocm6.4",
162-
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
160+
"gpu_arch_version": "7.0",
161+
"desired_cuda": "rocm7.0",
162+
"container_image": "pytorch/manylinux2_28-builder:rocm7.0",
163163
"package_type": "manywheel",
164-
"build_name": "manywheel-py3_11-rocm6_4",
164+
"build_name": "manywheel-py3_11-rocm7_0",
165165
"validation_runner": "linux.2xlarge",
166-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
166+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm7.0",
167167
"channel": "nightly",
168168
"upload_to_base_bucket": "no",
169169
"stable_version": "2.8.0"
@@ -227,27 +227,27 @@
227227
{
228228
"python_version": "3.12",
229229
"gpu_arch_type": "rocm",
230-
"gpu_arch_version": "6.3",
231-
"desired_cuda": "rocm6.3",
232-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
230+
"gpu_arch_version": "6.4",
231+
"desired_cuda": "rocm6.4",
232+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
233233
"package_type": "manywheel",
234-
"build_name": "manywheel-py3_12-rocm6_3",
234+
"build_name": "manywheel-py3_12-rocm6_4",
235235
"validation_runner": "linux.2xlarge",
236-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
236+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
237237
"channel": "nightly",
238238
"upload_to_base_bucket": "no",
239239
"stable_version": "2.8.0"
240240
},
241241
{
242242
"python_version": "3.12",
243243
"gpu_arch_type": "rocm",
244-
"gpu_arch_version": "6.4",
245-
"desired_cuda": "rocm6.4",
246-
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
244+
"gpu_arch_version": "7.0",
245+
"desired_cuda": "rocm7.0",
246+
"container_image": "pytorch/manylinux2_28-builder:rocm7.0",
247247
"package_type": "manywheel",
248-
"build_name": "manywheel-py3_12-rocm6_4",
248+
"build_name": "manywheel-py3_12-rocm7_0",
249249
"validation_runner": "linux.2xlarge",
250-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
250+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm7.0",
251251
"channel": "nightly",
252252
"upload_to_base_bucket": "no",
253253
"stable_version": "2.8.0"
@@ -311,27 +311,27 @@
311311
{
312312
"python_version": "3.13",
313313
"gpu_arch_type": "rocm",
314-
"gpu_arch_version": "6.3",
315-
"desired_cuda": "rocm6.3",
316-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
314+
"gpu_arch_version": "6.4",
315+
"desired_cuda": "rocm6.4",
316+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
317317
"package_type": "manywheel",
318-
"build_name": "manywheel-py3_13-rocm6_3",
318+
"build_name": "manywheel-py3_13-rocm6_4",
319319
"validation_runner": "linux.2xlarge",
320-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
320+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
321321
"channel": "nightly",
322322
"upload_to_base_bucket": "no",
323323
"stable_version": "2.8.0"
324324
},
325325
{
326326
"python_version": "3.13",
327327
"gpu_arch_type": "rocm",
328-
"gpu_arch_version": "6.4",
329-
"desired_cuda": "rocm6.4",
330-
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
328+
"gpu_arch_version": "7.0",
329+
"desired_cuda": "rocm7.0",
330+
"container_image": "pytorch/manylinux2_28-builder:rocm7.0",
331331
"package_type": "manywheel",
332-
"build_name": "manywheel-py3_13-rocm6_4",
332+
"build_name": "manywheel-py3_13-rocm7_0",
333333
"validation_runner": "linux.2xlarge",
334-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
334+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm7.0",
335335
"channel": "nightly",
336336
"upload_to_base_bucket": "no",
337337
"stable_version": "2.8.0"
@@ -395,27 +395,27 @@
395395
{
396396
"python_version": "3.13t",
397397
"gpu_arch_type": "rocm",
398-
"gpu_arch_version": "6.3",
399-
"desired_cuda": "rocm6.3",
400-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
398+
"gpu_arch_version": "6.4",
399+
"desired_cuda": "rocm6.4",
400+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
401401
"package_type": "manywheel",
402-
"build_name": "manywheel-py3_13t-rocm6_3",
402+
"build_name": "manywheel-py3_13t-rocm6_4",
403403
"validation_runner": "linux.2xlarge",
404-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
404+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
405405
"channel": "nightly",
406406
"upload_to_base_bucket": "no",
407407
"stable_version": "2.8.0"
408408
},
409409
{
410410
"python_version": "3.13t",
411411
"gpu_arch_type": "rocm",
412-
"gpu_arch_version": "6.4",
413-
"desired_cuda": "rocm6.4",
414-
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
412+
"gpu_arch_version": "7.0",
413+
"desired_cuda": "rocm7.0",
414+
"container_image": "pytorch/manylinux2_28-builder:rocm7.0",
415415
"package_type": "manywheel",
416-
"build_name": "manywheel-py3_13t-rocm6_4",
416+
"build_name": "manywheel-py3_13t-rocm7_0",
417417
"validation_runner": "linux.2xlarge",
418-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
418+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm7.0",
419419
"channel": "nightly",
420420
"upload_to_base_bucket": "no",
421421
"stable_version": "2.8.0"
@@ -479,27 +479,27 @@
479479
{
480480
"python_version": "3.14",
481481
"gpu_arch_type": "rocm",
482-
"gpu_arch_version": "6.3",
483-
"desired_cuda": "rocm6.3",
484-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
482+
"gpu_arch_version": "6.4",
483+
"desired_cuda": "rocm6.4",
484+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
485485
"package_type": "manywheel",
486-
"build_name": "manywheel-py3_14-rocm6_3",
486+
"build_name": "manywheel-py3_14-rocm6_4",
487487
"validation_runner": "linux.2xlarge",
488-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
488+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
489489
"channel": "nightly",
490490
"upload_to_base_bucket": "no",
491491
"stable_version": "2.8.0"
492492
},
493493
{
494494
"python_version": "3.14",
495495
"gpu_arch_type": "rocm",
496-
"gpu_arch_version": "6.4",
497-
"desired_cuda": "rocm6.4",
498-
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
496+
"gpu_arch_version": "7.0",
497+
"desired_cuda": "rocm7.0",
498+
"container_image": "pytorch/manylinux2_28-builder:rocm7.0",
499499
"package_type": "manywheel",
500-
"build_name": "manywheel-py3_14-rocm6_4",
500+
"build_name": "manywheel-py3_14-rocm7_0",
501501
"validation_runner": "linux.2xlarge",
502-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
502+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm7.0",
503503
"channel": "nightly",
504504
"upload_to_base_bucket": "no",
505505
"stable_version": "2.8.0"
@@ -563,27 +563,27 @@
563563
{
564564
"python_version": "3.14t",
565565
"gpu_arch_type": "rocm",
566-
"gpu_arch_version": "6.3",
567-
"desired_cuda": "rocm6.3",
568-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
566+
"gpu_arch_version": "6.4",
567+
"desired_cuda": "rocm6.4",
568+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
569569
"package_type": "manywheel",
570-
"build_name": "manywheel-py3_14t-rocm6_3",
570+
"build_name": "manywheel-py3_14t-rocm6_4",
571571
"validation_runner": "linux.2xlarge",
572-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
572+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
573573
"channel": "nightly",
574574
"upload_to_base_bucket": "no",
575575
"stable_version": "2.8.0"
576576
},
577577
{
578578
"python_version": "3.14t",
579579
"gpu_arch_type": "rocm",
580-
"gpu_arch_version": "6.4",
581-
"desired_cuda": "rocm6.4",
582-
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
580+
"gpu_arch_version": "7.0",
581+
"desired_cuda": "rocm7.0",
582+
"container_image": "pytorch/manylinux2_28-builder:rocm7.0",
583583
"package_type": "manywheel",
584-
"build_name": "manywheel-py3_14t-rocm6_4",
584+
"build_name": "manywheel-py3_14t-rocm7_0",
585585
"validation_runner": "linux.2xlarge",
586-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
586+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm7.0",
587587
"channel": "nightly",
588588
"upload_to_base_bucket": "no",
589589
"stable_version": "2.8.0"

0 commit comments

Comments
 (0)