Skip to content

Commit 1f11622

Browse files
jithunnair-amdjithunnair-amd
authored
Upgrade nightly wheels to ROCm6.3 (#6114)
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 9cef6c2 commit 1f11622

File tree

3 files changed

+101
-101
lines changed

3 files changed

+101
-101
lines changed

‎tools/scripts/generate_binary_build_matrix.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
"release": ["11.8", "12.1", "12.4"],
3434
}
3535
ROCM_ARCHES_DICT = {
36-
"nightly": ["6.1", "6.2.4"],
36+
"nightly": ["6.2.4", "6.3"],
3737
"test": ["6.1", "6.2.4"],
3838
"release": ["6.1", "6.2"],
3939
}

‎tools/tests/assets/build_matrix_linux_wheel_cuda.json

Lines changed: 50 additions & 50 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.1",
67-
"desired_cuda": "rocm6.1",
68-
"container_image": "pytorch/manylinux2_28-builder:rocm6.1",
66+
"gpu_arch_version": "6.2.4",
67+
"desired_cuda": "rocm6.2.4",
68+
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
6969
"package_type": "manywheel",
70-
"build_name": "manywheel-py3_9-rocm6_1",
70+
"build_name": "manywheel-py3_9-rocm6_2_4",
7171
"validation_runner": "linux.2xlarge",
72-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
72+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
7373
"channel": "nightly",
7474
"upload_to_base_bucket": "no",
7575
"stable_version": "2.5.1",
@@ -78,13 +78,13 @@
7878
{
7979
"python_version": "3.9",
8080
"gpu_arch_type": "rocm",
81-
"gpu_arch_version": "6.2.4",
82-
"desired_cuda": "rocm6.2.4",
83-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
81+
"gpu_arch_version": "6.3",
82+
"desired_cuda": "rocm6.3",
83+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
8484
"package_type": "manywheel",
85-
"build_name": "manywheel-py3_9-rocm6_2_4",
85+
"build_name": "manywheel-py3_9-rocm6_3",
8686
"validation_runner": "linux.2xlarge",
87-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
87+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
8888
"channel": "nightly",
8989
"upload_to_base_bucket": "no",
9090
"stable_version": "2.5.1",
@@ -153,13 +153,13 @@
153153
{
154154
"python_version": "3.10",
155155
"gpu_arch_type": "rocm",
156-
"gpu_arch_version": "6.1",
157-
"desired_cuda": "rocm6.1",
158-
"container_image": "pytorch/manylinux2_28-builder:rocm6.1",
156+
"gpu_arch_version": "6.2.4",
157+
"desired_cuda": "rocm6.2.4",
158+
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
159159
"package_type": "manywheel",
160-
"build_name": "manywheel-py3_10-rocm6_1",
160+
"build_name": "manywheel-py3_10-rocm6_2_4",
161161
"validation_runner": "linux.2xlarge",
162-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
162+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
163163
"channel": "nightly",
164164
"upload_to_base_bucket": "no",
165165
"stable_version": "2.5.1",
@@ -168,13 +168,13 @@
168168
{
169169
"python_version": "3.10",
170170
"gpu_arch_type": "rocm",
171-
"gpu_arch_version": "6.2.4",
172-
"desired_cuda": "rocm6.2.4",
173-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
171+
"gpu_arch_version": "6.3",
172+
"desired_cuda": "rocm6.3",
173+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
174174
"package_type": "manywheel",
175-
"build_name": "manywheel-py3_10-rocm6_2_4",
175+
"build_name": "manywheel-py3_10-rocm6_3",
176176
"validation_runner": "linux.2xlarge",
177-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
177+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
178178
"channel": "nightly",
179179
"upload_to_base_bucket": "no",
180180
"stable_version": "2.5.1",
@@ -243,13 +243,13 @@
243243
{
244244
"python_version": "3.11",
245245
"gpu_arch_type": "rocm",
246-
"gpu_arch_version": "6.1",
247-
"desired_cuda": "rocm6.1",
248-
"container_image": "pytorch/manylinux2_28-builder:rocm6.1",
246+
"gpu_arch_version": "6.2.4",
247+
"desired_cuda": "rocm6.2.4",
248+
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
249249
"package_type": "manywheel",
250-
"build_name": "manywheel-py3_11-rocm6_1",
250+
"build_name": "manywheel-py3_11-rocm6_2_4",
251251
"validation_runner": "linux.2xlarge",
252-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
252+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
253253
"channel": "nightly",
254254
"upload_to_base_bucket": "no",
255255
"stable_version": "2.5.1",
@@ -258,13 +258,13 @@
258258
{
259259
"python_version": "3.11",
260260
"gpu_arch_type": "rocm",
261-
"gpu_arch_version": "6.2.4",
262-
"desired_cuda": "rocm6.2.4",
263-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
261+
"gpu_arch_version": "6.3",
262+
"desired_cuda": "rocm6.3",
263+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
264264
"package_type": "manywheel",
265-
"build_name": "manywheel-py3_11-rocm6_2_4",
265+
"build_name": "manywheel-py3_11-rocm6_3",
266266
"validation_runner": "linux.2xlarge",
267-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
267+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
268268
"channel": "nightly",
269269
"upload_to_base_bucket": "no",
270270
"stable_version": "2.5.1",
@@ -333,13 +333,13 @@
333333
{
334334
"python_version": "3.12",
335335
"gpu_arch_type": "rocm",
336-
"gpu_arch_version": "6.1",
337-
"desired_cuda": "rocm6.1",
338-
"container_image": "pytorch/manylinux2_28-builder:rocm6.1",
336+
"gpu_arch_version": "6.2.4",
337+
"desired_cuda": "rocm6.2.4",
338+
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
339339
"package_type": "manywheel",
340-
"build_name": "manywheel-py3_12-rocm6_1",
340+
"build_name": "manywheel-py3_12-rocm6_2_4",
341341
"validation_runner": "linux.2xlarge",
342-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
342+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
343343
"channel": "nightly",
344344
"upload_to_base_bucket": "no",
345345
"stable_version": "2.5.1",
@@ -348,13 +348,13 @@
348348
{
349349
"python_version": "3.12",
350350
"gpu_arch_type": "rocm",
351-
"gpu_arch_version": "6.2.4",
352-
"desired_cuda": "rocm6.2.4",
353-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
351+
"gpu_arch_version": "6.3",
352+
"desired_cuda": "rocm6.3",
353+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
354354
"package_type": "manywheel",
355-
"build_name": "manywheel-py3_12-rocm6_2_4",
355+
"build_name": "manywheel-py3_12-rocm6_3",
356356
"validation_runner": "linux.2xlarge",
357-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
357+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
358358
"channel": "nightly",
359359
"upload_to_base_bucket": "no",
360360
"stable_version": "2.5.1",
@@ -423,13 +423,13 @@
423423
{
424424
"python_version": "3.13",
425425
"gpu_arch_type": "rocm",
426-
"gpu_arch_version": "6.1",
427-
"desired_cuda": "rocm6.1",
428-
"container_image": "pytorch/manylinux2_28-builder:rocm6.1",
426+
"gpu_arch_version": "6.2.4",
427+
"desired_cuda": "rocm6.2.4",
428+
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
429429
"package_type": "manywheel",
430-
"build_name": "manywheel-py3_13-rocm6_1",
430+
"build_name": "manywheel-py3_13-rocm6_2_4",
431431
"validation_runner": "linux.2xlarge",
432-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
432+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
433433
"channel": "nightly",
434434
"upload_to_base_bucket": "no",
435435
"stable_version": "2.5.1",
@@ -438,13 +438,13 @@
438438
{
439439
"python_version": "3.13",
440440
"gpu_arch_type": "rocm",
441-
"gpu_arch_version": "6.2.4",
442-
"desired_cuda": "rocm6.2.4",
443-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
441+
"gpu_arch_version": "6.3",
442+
"desired_cuda": "rocm6.3",
443+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
444444
"package_type": "manywheel",
445-
"build_name": "manywheel-py3_13-rocm6_2_4",
445+
"build_name": "manywheel-py3_13-rocm6_3",
446446
"validation_runner": "linux.2xlarge",
447-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
447+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
448448
"channel": "nightly",
449449
"upload_to_base_bucket": "no",
450450
"stable_version": "2.5.1",

‎tools/tests/assets/build_matrix_linux_wheel_nocpu.json

Lines changed: 50 additions & 50 deletions
Original file line numberDiff line numberDiff line change
@@ -48,13 +48,13 @@
4848
{
4949
"python_version": "3.9",
5050
"gpu_arch_type": "rocm",
51-
"gpu_arch_version": "6.1",
52-
"desired_cuda": "rocm6.1",
53-
"container_image": "pytorch/manylinux2_28-builder:rocm6.1",
51+
"gpu_arch_version": "6.2.4",
52+
"desired_cuda": "rocm6.2.4",
53+
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
5454
"package_type": "manywheel",
55-
"build_name": "manywheel-py3_9-rocm6_1",
55+
"build_name": "manywheel-py3_9-rocm6_2_4",
5656
"validation_runner": "linux.2xlarge",
57-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
57+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
5858
"channel": "nightly",
5959
"upload_to_base_bucket": "no",
6060
"stable_version": "2.5.1",
@@ -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.5.1",
@@ -123,13 +123,13 @@
123123
{
124124
"python_version": "3.10",
125125
"gpu_arch_type": "rocm",
126-
"gpu_arch_version": "6.1",
127-
"desired_cuda": "rocm6.1",
128-
"container_image": "pytorch/manylinux2_28-builder:rocm6.1",
126+
"gpu_arch_version": "6.2.4",
127+
"desired_cuda": "rocm6.2.4",
128+
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
129129
"package_type": "manywheel",
130-
"build_name": "manywheel-py3_10-rocm6_1",
130+
"build_name": "manywheel-py3_10-rocm6_2_4",
131131
"validation_runner": "linux.2xlarge",
132-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
132+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
133133
"channel": "nightly",
134134
"upload_to_base_bucket": "no",
135135
"stable_version": "2.5.1",
@@ -138,13 +138,13 @@
138138
{
139139
"python_version": "3.10",
140140
"gpu_arch_type": "rocm",
141-
"gpu_arch_version": "6.2.4",
142-
"desired_cuda": "rocm6.2.4",
143-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
141+
"gpu_arch_version": "6.3",
142+
"desired_cuda": "rocm6.3",
143+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
144144
"package_type": "manywheel",
145-
"build_name": "manywheel-py3_10-rocm6_2_4",
145+
"build_name": "manywheel-py3_10-rocm6_3",
146146
"validation_runner": "linux.2xlarge",
147-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
147+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
148148
"channel": "nightly",
149149
"upload_to_base_bucket": "no",
150150
"stable_version": "2.5.1",
@@ -198,13 +198,13 @@
198198
{
199199
"python_version": "3.11",
200200
"gpu_arch_type": "rocm",
201-
"gpu_arch_version": "6.1",
202-
"desired_cuda": "rocm6.1",
203-
"container_image": "pytorch/manylinux2_28-builder:rocm6.1",
201+
"gpu_arch_version": "6.2.4",
202+
"desired_cuda": "rocm6.2.4",
203+
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
204204
"package_type": "manywheel",
205-
"build_name": "manywheel-py3_11-rocm6_1",
205+
"build_name": "manywheel-py3_11-rocm6_2_4",
206206
"validation_runner": "linux.2xlarge",
207-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
207+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
208208
"channel": "nightly",
209209
"upload_to_base_bucket": "no",
210210
"stable_version": "2.5.1",
@@ -213,13 +213,13 @@
213213
{
214214
"python_version": "3.11",
215215
"gpu_arch_type": "rocm",
216-
"gpu_arch_version": "6.2.4",
217-
"desired_cuda": "rocm6.2.4",
218-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
216+
"gpu_arch_version": "6.3",
217+
"desired_cuda": "rocm6.3",
218+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
219219
"package_type": "manywheel",
220-
"build_name": "manywheel-py3_11-rocm6_2_4",
220+
"build_name": "manywheel-py3_11-rocm6_3",
221221
"validation_runner": "linux.2xlarge",
222-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
222+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
223223
"channel": "nightly",
224224
"upload_to_base_bucket": "no",
225225
"stable_version": "2.5.1",
@@ -273,13 +273,13 @@
273273
{
274274
"python_version": "3.12",
275275
"gpu_arch_type": "rocm",
276-
"gpu_arch_version": "6.1",
277-
"desired_cuda": "rocm6.1",
278-
"container_image": "pytorch/manylinux2_28-builder:rocm6.1",
276+
"gpu_arch_version": "6.2.4",
277+
"desired_cuda": "rocm6.2.4",
278+
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
279279
"package_type": "manywheel",
280-
"build_name": "manywheel-py3_12-rocm6_1",
280+
"build_name": "manywheel-py3_12-rocm6_2_4",
281281
"validation_runner": "linux.2xlarge",
282-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
282+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
283283
"channel": "nightly",
284284
"upload_to_base_bucket": "no",
285285
"stable_version": "2.5.1",
@@ -288,13 +288,13 @@
288288
{
289289
"python_version": "3.12",
290290
"gpu_arch_type": "rocm",
291-
"gpu_arch_version": "6.2.4",
292-
"desired_cuda": "rocm6.2.4",
293-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
291+
"gpu_arch_version": "6.3",
292+
"desired_cuda": "rocm6.3",
293+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
294294
"package_type": "manywheel",
295-
"build_name": "manywheel-py3_12-rocm6_2_4",
295+
"build_name": "manywheel-py3_12-rocm6_3",
296296
"validation_runner": "linux.2xlarge",
297-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
297+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
298298
"channel": "nightly",
299299
"upload_to_base_bucket": "no",
300300
"stable_version": "2.5.1",
@@ -348,13 +348,13 @@
348348
{
349349
"python_version": "3.13",
350350
"gpu_arch_type": "rocm",
351-
"gpu_arch_version": "6.1",
352-
"desired_cuda": "rocm6.1",
353-
"container_image": "pytorch/manylinux2_28-builder:rocm6.1",
351+
"gpu_arch_version": "6.2.4",
352+
"desired_cuda": "rocm6.2.4",
353+
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
354354
"package_type": "manywheel",
355-
"build_name": "manywheel-py3_13-rocm6_1",
355+
"build_name": "manywheel-py3_13-rocm6_2_4",
356356
"validation_runner": "linux.2xlarge",
357-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
357+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
358358
"channel": "nightly",
359359
"upload_to_base_bucket": "no",
360360
"stable_version": "2.5.1",
@@ -363,13 +363,13 @@
363363
{
364364
"python_version": "3.13",
365365
"gpu_arch_type": "rocm",
366-
"gpu_arch_version": "6.2.4",
367-
"desired_cuda": "rocm6.2.4",
368-
"container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
366+
"gpu_arch_version": "6.3",
367+
"desired_cuda": "rocm6.3",
368+
"container_image": "pytorch/manylinux2_28-builder:rocm6.3",
369369
"package_type": "manywheel",
370-
"build_name": "manywheel-py3_13-rocm6_2_4",
370+
"build_name": "manywheel-py3_13-rocm6_3",
371371
"validation_runner": "linux.2xlarge",
372-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
372+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
373373
"channel": "nightly",
374374
"upload_to_base_bucket": "no",
375375
"stable_version": "2.5.1",

0 commit comments

Comments
 (0)