[ROCm] Update to rocm5.0 wheels (#5660)

Signed-off-by: Kyle Chen <kylechen@amd.com>
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

.circleci/config.yml

@@ -1141,16 +1141,16 @@ workflows:
           name: binary_linux_wheel_py3.7_cu115
           python_version: '3.7'
           wheel_docker_image: pytorch/manylinux-cuda115
-      - binary_linux_wheel:
-          cu_version: rocm4.3.1
-          name: binary_linux_wheel_py3.7_rocm4.3.1
-          python_version: '3.7'
-          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
       - binary_linux_wheel:
           cu_version: rocm4.5.2
           name: binary_linux_wheel_py3.7_rocm4.5.2
           python_version: '3.7'
           wheel_docker_image: pytorch/manylinux-rocm:4.5.2
+      - binary_linux_wheel:
+          cu_version: rocm5.0
+          name: binary_linux_wheel_py3.7_rocm5.0
+          python_version: '3.7'
+          wheel_docker_image: pytorch/manylinux-rocm:5.0
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1175,16 +1175,16 @@ workflows:
           name: binary_linux_wheel_py3.8_cu115
           python_version: '3.8'
           wheel_docker_image: pytorch/manylinux-cuda115
-      - binary_linux_wheel:
-          cu_version: rocm4.3.1
-          name: binary_linux_wheel_py3.8_rocm4.3.1
-          python_version: '3.8'
-          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
       - binary_linux_wheel:
           cu_version: rocm4.5.2
           name: binary_linux_wheel_py3.8_rocm4.5.2
           python_version: '3.8'
           wheel_docker_image: pytorch/manylinux-rocm:4.5.2
+      - binary_linux_wheel:
+          cu_version: rocm5.0
+          name: binary_linux_wheel_py3.8_rocm5.0
+          python_version: '3.8'
+          wheel_docker_image: pytorch/manylinux-rocm:5.0
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1209,16 +1209,16 @@ workflows:
           name: binary_linux_wheel_py3.9_cu115
           python_version: '3.9'
           wheel_docker_image: pytorch/manylinux-cuda115
-      - binary_linux_wheel:
-          cu_version: rocm4.3.1
-          name: binary_linux_wheel_py3.9_rocm4.3.1
-          python_version: '3.9'
-          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
       - binary_linux_wheel:
           cu_version: rocm4.5.2
           name: binary_linux_wheel_py3.9_rocm4.5.2
           python_version: '3.9'
           wheel_docker_image: pytorch/manylinux-rocm:4.5.2
+      - binary_linux_wheel:
+          cu_version: rocm5.0
+          name: binary_linux_wheel_py3.9_rocm5.0
+          python_version: '3.9'
+          wheel_docker_image: pytorch/manylinux-rocm:5.0
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1243,16 +1243,16 @@ workflows:
           name: binary_linux_wheel_py3.10_cu115
           python_version: '3.10'
           wheel_docker_image: pytorch/manylinux-cuda115
-      - binary_linux_wheel:
-          cu_version: rocm4.3.1
-          name: binary_linux_wheel_py3.10_rocm4.3.1
-          python_version: '3.10'
-          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
       - binary_linux_wheel:
           cu_version: rocm4.5.2
           name: binary_linux_wheel_py3.10_rocm4.5.2
           python_version: '3.10'
           wheel_docker_image: pytorch/manylinux-rocm:4.5.2
+      - binary_linux_wheel:
+          cu_version: rocm5.0
+          name: binary_linux_wheel_py3.10_rocm5.0
+          python_version: '3.10'
+          wheel_docker_image: pytorch/manylinux-rocm:5.0
       - binary_macos_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1898,15 +1898,15 @@ workflows:
           - nightly_binary_linux_wheel_py3.7_cu115
           subfolder: cu115/
       - binary_linux_wheel:
-          cu_version: rocm4.3.1
+          cu_version: rocm4.5.2
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm4.3.1
+          name: nightly_binary_linux_wheel_py3.7_rocm4.5.2
           python_version: '3.7'
-          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.5.2
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -1914,20 +1914,20 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
+          name: nightly_binary_linux_wheel_py3.7_rocm4.5.2_upload
           requires:
-          - nightly_binary_linux_wheel_py3.7_rocm4.3.1
-          subfolder: rocm4.3.1/
+          - nightly_binary_linux_wheel_py3.7_rocm4.5.2
+          subfolder: rocm4.5.2/
       - binary_linux_wheel:
-          cu_version: rocm4.5.2
+          cu_version: rocm5.0
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm4.5.2
+          name: nightly_binary_linux_wheel_py3.7_rocm5.0
           python_version: '3.7'
-          wheel_docker_image: pytorch/manylinux-rocm:4.5.2
+          wheel_docker_image: pytorch/manylinux-rocm:5.0
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -1935,10 +1935,10 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm4.5.2_upload
+          name: nightly_binary_linux_wheel_py3.7_rocm5.0_upload
           requires:
-          - nightly_binary_linux_wheel_py3.7_rocm4.5.2
-          subfolder: rocm4.5.2/
+          - nightly_binary_linux_wheel_py3.7_rocm5.0
+          subfolder: rocm5.0/
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -2028,15 +2028,15 @@ workflows:
           - nightly_binary_linux_wheel_py3.8_cu115
           subfolder: cu115/
       - binary_linux_wheel:
-          cu_version: rocm4.3.1
+          cu_version: rocm4.5.2
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm4.3.1
+          name: nightly_binary_linux_wheel_py3.8_rocm4.5.2
           python_version: '3.8'
-          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.5.2
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2044,20 +2044,20 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
+          name: nightly_binary_linux_wheel_py3.8_rocm4.5.2_upload
           requires:
-          - nightly_binary_linux_wheel_py3.8_rocm4.3.1
-          subfolder: rocm4.3.1/
+          - nightly_binary_linux_wheel_py3.8_rocm4.5.2
+          subfolder: rocm4.5.2/
       - binary_linux_wheel:
-          cu_version: rocm4.5.2
+          cu_version: rocm5.0
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm4.5.2
+          name: nightly_binary_linux_wheel_py3.8_rocm5.0
           python_version: '3.8'
-          wheel_docker_image: pytorch/manylinux-rocm:4.5.2
+          wheel_docker_image: pytorch/manylinux-rocm:5.0
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2065,10 +2065,10 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm4.5.2_upload
+          name: nightly_binary_linux_wheel_py3.8_rocm5.0_upload
           requires:
-          - nightly_binary_linux_wheel_py3.8_rocm4.5.2
-          subfolder: rocm4.5.2/
+          - nightly_binary_linux_wheel_py3.8_rocm5.0
+          subfolder: rocm5.0/
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -2158,15 +2158,15 @@ workflows:
           - nightly_binary_linux_wheel_py3.9_cu115
           subfolder: cu115/
       - binary_linux_wheel:
-          cu_version: rocm4.3.1
+          cu_version: rocm4.5.2
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm4.3.1
+          name: nightly_binary_linux_wheel_py3.9_rocm4.5.2
           python_version: '3.9'
-          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.5.2
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2174,20 +2174,20 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
+          name: nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
           requires:
-          - nightly_binary_linux_wheel_py3.9_rocm4.3.1
-          subfolder: rocm4.3.1/
+          - nightly_binary_linux_wheel_py3.9_rocm4.5.2
+          subfolder: rocm4.5.2/
       - binary_linux_wheel:
-          cu_version: rocm4.5.2
+          cu_version: rocm5.0
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm4.5.2
+          name: nightly_binary_linux_wheel_py3.9_rocm5.0
           python_version: '3.9'
-          wheel_docker_image: pytorch/manylinux-rocm:4.5.2
+          wheel_docker_image: pytorch/manylinux-rocm:5.0
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2195,10 +2195,10 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
+          name: nightly_binary_linux_wheel_py3.9_rocm5.0_upload
           requires:
-          - nightly_binary_linux_wheel_py3.9_rocm4.5.2
-          subfolder: rocm4.5.2/
+          - nightly_binary_linux_wheel_py3.9_rocm5.0
+          subfolder: rocm5.0/
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -2288,15 +2288,15 @@ workflows:
           - nightly_binary_linux_wheel_py3.10_cu115
           subfolder: cu115/
       - binary_linux_wheel:
-          cu_version: rocm4.3.1
+          cu_version: rocm4.5.2
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.10_rocm4.3.1
+          name: nightly_binary_linux_wheel_py3.10_rocm4.5.2
           python_version: '3.10'
-          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.5.2
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2304,20 +2304,20 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.10_rocm4.3.1_upload
+          name: nightly_binary_linux_wheel_py3.10_rocm4.5.2_upload
           requires:
-          - nightly_binary_linux_wheel_py3.10_rocm4.3.1
-          subfolder: rocm4.3.1/
+          - nightly_binary_linux_wheel_py3.10_rocm4.5.2
+          subfolder: rocm4.5.2/
       - binary_linux_wheel:
-          cu_version: rocm4.5.2
+          cu_version: rocm5.0
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.10_rocm4.5.2
+          name: nightly_binary_linux_wheel_py3.10_rocm5.0
           python_version: '3.10'
-          wheel_docker_image: pytorch/manylinux-rocm:4.5.2
+          wheel_docker_image: pytorch/manylinux-rocm:5.0
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2325,10 +2325,10 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.10_rocm4.5.2_upload
+          name: nightly_binary_linux_wheel_py3.10_rocm5.0_upload
           requires:
-          - nightly_binary_linux_wheel_py3.10_rocm4.5.2
-          subfolder: rocm4.5.2/
+          - nightly_binary_linux_wheel_py3.10_rocm5.0
+          subfolder: rocm5.0/
       - binary_macos_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu

.circleci/regenerate.py

@@ -32,7 +32,7 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
         for os_type in ["linux", "macos", "win"]:
             python_versions = PYTHON_VERSIONS
             cu_versions_dict = {
-                "linux": ["cpu", "cu102", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],
+                "linux": ["cpu", "cu102", "cu113", "cu115", "rocm4.5.2", "rocm5.0"],
                 "win": ["cpu", "cu113", "cu115"],
                 "macos": ["cpu"],
             }