Upgrade to rocm6.3; Generated assets using 'cp -a tools/tests/test_ge…

…nerate_binary_build_matrix.py ../.. && python3 test_generate_binary_build_matrix.py --update-reference-files'
pytorch · Dec 26, 2024 · fe46d5b · fe46d5b
1 parent 9cef6c2
commit fe46d5b
Show file tree

Hide file tree

Showing 3 changed files with 101 additions and 101 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -33,7 +33,7 @@
     "release": ["11.8", "12.1", "12.4"],
 }
 ROCM_ARCHES_DICT = {
-    "nightly": ["6.1", "6.2.4"],
+    "nightly": ["6.2.4", "6.3"],
     "test": ["6.1", "6.2.4"],
     "release": ["6.1", "6.2"],
 }

diff --git a/tools/tests/assets/build_matrix_linux_wheel_cuda.json b/tools/tests/assets/build_matrix_linux_wheel_cuda.json
@@ -63,13 +63,13 @@
     {
       "python_version": "3.9",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.1",
-      "desired_cuda": "rocm6.1",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_9-rocm6_1",
+      "build_name": "manywheel-py3_9-rocm6_2_4",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -78,13 +78,13 @@
     {
       "python_version": "3.9",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.2.4",
-      "desired_cuda": "rocm6.2.4",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "gpu_arch_version": "6.3",
+      "desired_cuda": "rocm6.3",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.3",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_9-rocm6_2_4",
+      "build_name": "manywheel-py3_9-rocm6_3",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -153,13 +153,13 @@
     {
       "python_version": "3.10",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.1",
-      "desired_cuda": "rocm6.1",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_10-rocm6_1",
+      "build_name": "manywheel-py3_10-rocm6_2_4",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -168,13 +168,13 @@
     {
       "python_version": "3.10",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.2.4",
-      "desired_cuda": "rocm6.2.4",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "gpu_arch_version": "6.3",
+      "desired_cuda": "rocm6.3",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.3",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_10-rocm6_2_4",
+      "build_name": "manywheel-py3_10-rocm6_3",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -243,13 +243,13 @@
     {
       "python_version": "3.11",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.1",
-      "desired_cuda": "rocm6.1",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_11-rocm6_1",
+      "build_name": "manywheel-py3_11-rocm6_2_4",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -258,13 +258,13 @@
     {
       "python_version": "3.11",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.2.4",
-      "desired_cuda": "rocm6.2.4",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "gpu_arch_version": "6.3",
+      "desired_cuda": "rocm6.3",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.3",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_11-rocm6_2_4",
+      "build_name": "manywheel-py3_11-rocm6_3",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -333,13 +333,13 @@
     {
       "python_version": "3.12",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.1",
-      "desired_cuda": "rocm6.1",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_12-rocm6_1",
+      "build_name": "manywheel-py3_12-rocm6_2_4",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -348,13 +348,13 @@
     {
       "python_version": "3.12",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.2.4",
-      "desired_cuda": "rocm6.2.4",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "gpu_arch_version": "6.3",
+      "desired_cuda": "rocm6.3",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.3",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_12-rocm6_2_4",
+      "build_name": "manywheel-py3_12-rocm6_3",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -423,13 +423,13 @@
     {
       "python_version": "3.13",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.1",
-      "desired_cuda": "rocm6.1",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_13-rocm6_1",
+      "build_name": "manywheel-py3_13-rocm6_2_4",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -438,13 +438,13 @@
     {
       "python_version": "3.13",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.2.4",
-      "desired_cuda": "rocm6.2.4",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "gpu_arch_version": "6.3",
+      "desired_cuda": "rocm6.3",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.3",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_13-rocm6_2_4",
+      "build_name": "manywheel-py3_13-rocm6_3",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",

diff --git a/tools/tests/assets/build_matrix_linux_wheel_nocpu.json b/tools/tests/assets/build_matrix_linux_wheel_nocpu.json
@@ -48,13 +48,13 @@
     {
       "python_version": "3.9",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.1",
-      "desired_cuda": "rocm6.1",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_9-rocm6_1",
+      "build_name": "manywheel-py3_9-rocm6_2_4",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -63,13 +63,13 @@
     {
       "python_version": "3.9",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.2.4",
-      "desired_cuda": "rocm6.2.4",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "gpu_arch_version": "6.3",
+      "desired_cuda": "rocm6.3",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.3",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_9-rocm6_2_4",
+      "build_name": "manywheel-py3_9-rocm6_3",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -123,13 +123,13 @@
     {
       "python_version": "3.10",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.1",
-      "desired_cuda": "rocm6.1",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_10-rocm6_1",
+      "build_name": "manywheel-py3_10-rocm6_2_4",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -138,13 +138,13 @@
     {
       "python_version": "3.10",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.2.4",
-      "desired_cuda": "rocm6.2.4",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "gpu_arch_version": "6.3",
+      "desired_cuda": "rocm6.3",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.3",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_10-rocm6_2_4",
+      "build_name": "manywheel-py3_10-rocm6_3",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -198,13 +198,13 @@
     {
       "python_version": "3.11",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.1",
-      "desired_cuda": "rocm6.1",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_11-rocm6_1",
+      "build_name": "manywheel-py3_11-rocm6_2_4",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -213,13 +213,13 @@
     {
       "python_version": "3.11",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.2.4",
-      "desired_cuda": "rocm6.2.4",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "gpu_arch_version": "6.3",
+      "desired_cuda": "rocm6.3",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.3",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_11-rocm6_2_4",
+      "build_name": "manywheel-py3_11-rocm6_3",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -273,13 +273,13 @@
     {
       "python_version": "3.12",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.1",
-      "desired_cuda": "rocm6.1",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_12-rocm6_1",
+      "build_name": "manywheel-py3_12-rocm6_2_4",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -288,13 +288,13 @@
     {
       "python_version": "3.12",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.2.4",
-      "desired_cuda": "rocm6.2.4",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "gpu_arch_version": "6.3",
+      "desired_cuda": "rocm6.3",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.3",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_12-rocm6_2_4",
+      "build_name": "manywheel-py3_12-rocm6_3",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -348,13 +348,13 @@
     {
       "python_version": "3.13",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.1",
-      "desired_cuda": "rocm6.1",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.1",
+      "gpu_arch_version": "6.2.4",
+      "desired_cuda": "rocm6.2.4",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_13-rocm6_1",
+      "build_name": "manywheel-py3_13-rocm6_2_4",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",
@@ -363,13 +363,13 @@
     {
       "python_version": "3.13",
       "gpu_arch_type": "rocm",
-      "gpu_arch_version": "6.2.4",
-      "desired_cuda": "rocm6.2.4",
-      "container_image": "pytorch/manylinux2_28-builder:rocm6.2.4",
+      "gpu_arch_version": "6.3",
+      "desired_cuda": "rocm6.3",
+      "container_image": "pytorch/manylinux2_28-builder:rocm6.3",
       "package_type": "manywheel",
-      "build_name": "manywheel-py3_13-rocm6_2_4",
+      "build_name": "manywheel-py3_13-rocm6_3",
       "validation_runner": "linux.2xlarge",
-      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4",
+      "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3",
       "channel": "nightly",
       "upload_to_base_bucket": "no",
       "stable_version": "2.5.1",