Skip to content

Commit 6762d61

Browse files
jithunnair-amdjithunnair-amd
authored
Upgrade nightly wheels to ROCm6.4 (#6592)
Generated assets using `cp -a tools/tests/test_generate_binary_build_matrix.py . && python3 test_generate_binary_build_matrix.py --update-reference-files`
1 parent ac6c4f2 commit 6762d61

File tree

3 files changed

+121
-121
lines changed

3 files changed

+121
-121
lines changed

‎tools/scripts/generate_binary_build_matrix.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
"release": ["11.8", "12.6", "12.8"],
3333
}
3434
ROCM_ARCHES_DICT = {
35-
"nightly": ["6.2.4", "6.3"],
35+
"nightly": ["6.3", "6.4"],
3636
"test": ["6.2.4", "6.3"],
3737
"release": ["6.2.4", "6.3"],
3838
}

‎tools/tests/assets/build_matrix_linux_wheel_cuda.json

Lines changed: 60 additions & 60 deletions
Original file line numberDiff line numberDiff line change
@@ -63,13 +63,13 @@
6363
{
6464
"python_version": "3.9",
6565
"gpu_arch_type": "rocm",
66-
"gpu_arch_version": "6.2.4",
67-
"desired_cuda": "rocm6.2.4",
68-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
66+
"gpu_arch_version": "6.3",
67+
"desired_cuda": "rocm6.3",
68+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
6969
"package_type": "manywheel",
70-
"build_name": "manywheel-py3_9-rocm6_2_4",
70+
"build_name": "manywheel-py3_9-rocm6_3",
7171
"validation_runner": "linux.2xlarge",
72-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
72+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
7373
"channel": "nightly",
7474
"upload_to_base_bucket": "no",
7575
"stable_version": "2.7.0",
@@ -78,13 +78,13 @@
7878
{
7979
"python_version": "3.9",
8080
"gpu_arch_type": "rocm",
81-
"gpu_arch_version": "6.3",
82-
"desired_cuda": "rocm6.3",
83-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
81+
"gpu_arch_version": "6.4",
82+
"desired_cuda": "rocm6.4",
83+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
8484
"package_type": "manywheel",
85-
"build_name": "manywheel-py3_9-rocm6_3",
85+
"build_name": "manywheel-py3_9-rocm6_4",
8686
"validation_runner": "linux.2xlarge",
87-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
87+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
8888
"channel": "nightly",
8989
"upload_to_base_bucket": "no",
9090
"stable_version": "2.7.0",
@@ -153,13 +153,13 @@
153153
{
154154
"python_version": "3.10",
155155
"gpu_arch_type": "rocm",
156-
"gpu_arch_version": "6.2.4",
157-
"desired_cuda": "rocm6.2.4",
158-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
156+
"gpu_arch_version": "6.3",
157+
"desired_cuda": "rocm6.3",
158+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
159159
"package_type": "manywheel",
160-
"build_name": "manywheel-py3_10-rocm6_2_4",
160+
"build_name": "manywheel-py3_10-rocm6_3",
161161
"validation_runner": "linux.2xlarge",
162-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
162+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
163163
"channel": "nightly",
164164
"upload_to_base_bucket": "no",
165165
"stable_version": "2.7.0",
@@ -168,13 +168,13 @@
168168
{
169169
"python_version": "3.10",
170170
"gpu_arch_type": "rocm",
171-
"gpu_arch_version": "6.3",
172-
"desired_cuda": "rocm6.3",
173-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
171+
"gpu_arch_version": "6.4",
172+
"desired_cuda": "rocm6.4",
173+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
174174
"package_type": "manywheel",
175-
"build_name": "manywheel-py3_10-rocm6_3",
175+
"build_name": "manywheel-py3_10-rocm6_4",
176176
"validation_runner": "linux.2xlarge",
177-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
177+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
178178
"channel": "nightly",
179179
"upload_to_base_bucket": "no",
180180
"stable_version": "2.7.0",
@@ -243,13 +243,13 @@
243243
{
244244
"python_version": "3.11",
245245
"gpu_arch_type": "rocm",
246-
"gpu_arch_version": "6.2.4",
247-
"desired_cuda": "rocm6.2.4",
248-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
246+
"gpu_arch_version": "6.3",
247+
"desired_cuda": "rocm6.3",
248+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
249249
"package_type": "manywheel",
250-
"build_name": "manywheel-py3_11-rocm6_2_4",
250+
"build_name": "manywheel-py3_11-rocm6_3",
251251
"validation_runner": "linux.2xlarge",
252-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
252+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
253253
"channel": "nightly",
254254
"upload_to_base_bucket": "no",
255255
"stable_version": "2.7.0",
@@ -258,13 +258,13 @@
258258
{
259259
"python_version": "3.11",
260260
"gpu_arch_type": "rocm",
261-
"gpu_arch_version": "6.3",
262-
"desired_cuda": "rocm6.3",
263-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
261+
"gpu_arch_version": "6.4",
262+
"desired_cuda": "rocm6.4",
263+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
264264
"package_type": "manywheel",
265-
"build_name": "manywheel-py3_11-rocm6_3",
265+
"build_name": "manywheel-py3_11-rocm6_4",
266266
"validation_runner": "linux.2xlarge",
267-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
267+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
268268
"channel": "nightly",
269269
"upload_to_base_bucket": "no",
270270
"stable_version": "2.7.0",
@@ -333,13 +333,13 @@
333333
{
334334
"python_version": "3.12",
335335
"gpu_arch_type": "rocm",
336-
"gpu_arch_version": "6.2.4",
337-
"desired_cuda": "rocm6.2.4",
338-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
336+
"gpu_arch_version": "6.3",
337+
"desired_cuda": "rocm6.3",
338+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
339339
"package_type": "manywheel",
340-
"build_name": "manywheel-py3_12-rocm6_2_4",
340+
"build_name": "manywheel-py3_12-rocm6_3",
341341
"validation_runner": "linux.2xlarge",
342-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
342+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
343343
"channel": "nightly",
344344
"upload_to_base_bucket": "no",
345345
"stable_version": "2.7.0",
@@ -348,13 +348,13 @@
348348
{
349349
"python_version": "3.12",
350350
"gpu_arch_type": "rocm",
351-
"gpu_arch_version": "6.3",
352-
"desired_cuda": "rocm6.3",
353-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
351+
"gpu_arch_version": "6.4",
352+
"desired_cuda": "rocm6.4",
353+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
354354
"package_type": "manywheel",
355-
"build_name": "manywheel-py3_12-rocm6_3",
355+
"build_name": "manywheel-py3_12-rocm6_4",
356356
"validation_runner": "linux.2xlarge",
357-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
357+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
358358
"channel": "nightly",
359359
"upload_to_base_bucket": "no",
360360
"stable_version": "2.7.0",
@@ -423,13 +423,13 @@
423423
{
424424
"python_version": "3.13",
425425
"gpu_arch_type": "rocm",
426-
"gpu_arch_version": "6.2.4",
427-
"desired_cuda": "rocm6.2.4",
428-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
426+
"gpu_arch_version": "6.3",
427+
"desired_cuda": "rocm6.3",
428+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
429429
"package_type": "manywheel",
430-
"build_name": "manywheel-py3_13-rocm6_2_4",
430+
"build_name": "manywheel-py3_13-rocm6_3",
431431
"validation_runner": "linux.2xlarge",
432-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
432+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
433433
"channel": "nightly",
434434
"upload_to_base_bucket": "no",
435435
"stable_version": "2.7.0",
@@ -438,13 +438,13 @@
438438
{
439439
"python_version": "3.13",
440440
"gpu_arch_type": "rocm",
441-
"gpu_arch_version": "6.3",
442-
"desired_cuda": "rocm6.3",
443-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
441+
"gpu_arch_version": "6.4",
442+
"desired_cuda": "rocm6.4",
443+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
444444
"package_type": "manywheel",
445-
"build_name": "manywheel-py3_13-rocm6_3",
445+
"build_name": "manywheel-py3_13-rocm6_4",
446446
"validation_runner": "linux.2xlarge",
447-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
447+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
448448
"channel": "nightly",
449449
"upload_to_base_bucket": "no",
450450
"stable_version": "2.7.0",
@@ -513,13 +513,13 @@
513513
{
514514
"python_version": "3.13t",
515515
"gpu_arch_type": "rocm",
516-
"gpu_arch_version": "6.2.4",
517-
"desired_cuda": "rocm6.2.4",
518-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
516+
"gpu_arch_version": "6.3",
517+
"desired_cuda": "rocm6.3",
518+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
519519
"package_type": "manywheel",
520-
"build_name": "manywheel-py3_13t-rocm6_2_4",
520+
"build_name": "manywheel-py3_13t-rocm6_3",
521521
"validation_runner": "linux.2xlarge",
522-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
522+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
523523
"channel": "nightly",
524524
"upload_to_base_bucket": "no",
525525
"stable_version": "2.7.0",
@@ -528,13 +528,13 @@
528528
{
529529
"python_version": "3.13t",
530530
"gpu_arch_type": "rocm",
531-
"gpu_arch_version": "6.3",
532-
"desired_cuda": "rocm6.3",
533-
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
531+
"gpu_arch_version": "6.4",
532+
"desired_cuda": "rocm6.4",
533+
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
534534
"package_type": "manywheel",
535-
"build_name": "manywheel-py3_13t-rocm6_3",
535+
"build_name": "manywheel-py3_13t-rocm6_4",
536536
"validation_runner": "linux.2xlarge",
537-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
537+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
538538
"channel": "nightly",
539539
"upload_to_base_bucket": "no",
540540
"stable_version": "2.7.0",

0 commit comments

Comments
 (0)