[ROCm] update to rocm5.1.1 (#5946)

* [ROCm] Upgrade to rocm5.1.1

* retrigger checks
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

.circleci/config.yml

@@ -1124,16 +1124,16 @@ workflows:
           name: binary_linux_wheel_py3.7_cu116
           python_version: '3.7'
           wheel_docker_image: pytorch/manylinux-cuda116
-      - 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:
+          cu_version: rocm5.1.1
+          name: binary_linux_wheel_py3.7_rocm5.1.1
+          python_version: '3.7'
+          wheel_docker_image: pytorch/manylinux-rocm:5.1.1
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1158,16 +1158,16 @@ workflows:
           name: binary_linux_wheel_py3.8_cu116
           python_version: '3.8'
           wheel_docker_image: pytorch/manylinux-cuda116
-      - 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:
+          cu_version: rocm5.1.1
+          name: binary_linux_wheel_py3.8_rocm5.1.1
+          python_version: '3.8'
+          wheel_docker_image: pytorch/manylinux-rocm:5.1.1
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1192,16 +1192,16 @@ workflows:
           name: binary_linux_wheel_py3.9_cu116
           python_version: '3.9'
           wheel_docker_image: pytorch/manylinux-cuda116
-      - 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:
+          cu_version: rocm5.1.1
+          name: binary_linux_wheel_py3.9_rocm5.1.1
+          python_version: '3.9'
+          wheel_docker_image: pytorch/manylinux-rocm:5.1.1
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1226,16 +1226,16 @@ workflows:
           name: binary_linux_wheel_py3.10_cu116
           python_version: '3.10'
           wheel_docker_image: pytorch/manylinux-cuda116
-      - 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_linux_wheel:
+          cu_version: rocm5.1.1
+          name: binary_linux_wheel_py3.10_rocm5.1.1
+          python_version: '3.10'
+          wheel_docker_image: pytorch/manylinux-rocm:5.1.1
       - binary_macos_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1880,15 +1880,15 @@ workflows:
           - nightly_binary_linux_wheel_py3.7_cu116
           subfolder: cu116/
       - 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:
@@ -1896,20 +1896,20 @@ 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:
-          cu_version: rocm5.0
+          cu_version: rocm5.1.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm5.0
+          name: nightly_binary_linux_wheel_py3.7_rocm5.1.1
           python_version: '3.7'
-          wheel_docker_image: pytorch/manylinux-rocm:5.0
+          wheel_docker_image: pytorch/manylinux-rocm:5.1.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -1917,10 +1917,10 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm5.0_upload
+          name: nightly_binary_linux_wheel_py3.7_rocm5.1.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.7_rocm5.0
-          subfolder: rocm5.0/
+          - nightly_binary_linux_wheel_py3.7_rocm5.1.1
+          subfolder: rocm5.1.1/
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -2010,15 +2010,15 @@ workflows:
           - nightly_binary_linux_wheel_py3.8_cu116
           subfolder: cu116/
       - 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:
@@ -2026,20 +2026,20 @@ 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:
-          cu_version: rocm5.0
+          cu_version: rocm5.1.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm5.0
+          name: nightly_binary_linux_wheel_py3.8_rocm5.1.1
           python_version: '3.8'
-          wheel_docker_image: pytorch/manylinux-rocm:5.0
+          wheel_docker_image: pytorch/manylinux-rocm:5.1.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2047,10 +2047,10 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm5.0_upload
+          name: nightly_binary_linux_wheel_py3.8_rocm5.1.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.8_rocm5.0
-          subfolder: rocm5.0/
+          - nightly_binary_linux_wheel_py3.8_rocm5.1.1
+          subfolder: rocm5.1.1/
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -2140,15 +2140,15 @@ workflows:
           - nightly_binary_linux_wheel_py3.9_cu116
           subfolder: cu116/
       - 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:
@@ -2156,20 +2156,20 @@ 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:
-          cu_version: rocm5.0
+          cu_version: rocm5.1.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm5.0
+          name: nightly_binary_linux_wheel_py3.9_rocm5.1.1
           python_version: '3.9'
-          wheel_docker_image: pytorch/manylinux-rocm:5.0
+          wheel_docker_image: pytorch/manylinux-rocm:5.1.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2177,10 +2177,10 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm5.0_upload
+          name: nightly_binary_linux_wheel_py3.9_rocm5.1.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.9_rocm5.0
-          subfolder: rocm5.0/
+          - nightly_binary_linux_wheel_py3.9_rocm5.1.1
+          subfolder: rocm5.1.1/
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -2270,15 +2270,15 @@ workflows:
           - nightly_binary_linux_wheel_py3.10_cu116
           subfolder: cu116/
       - 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:
@@ -2286,20 +2286,20 @@ 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_linux_wheel:
-          cu_version: rocm5.0
+          cu_version: rocm5.1.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.10_rocm5.0
+          name: nightly_binary_linux_wheel_py3.10_rocm5.1.1
           python_version: '3.10'
-          wheel_docker_image: pytorch/manylinux-rocm:5.0
+          wheel_docker_image: pytorch/manylinux-rocm:5.1.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2307,10 +2307,10 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.10_rocm5.0_upload
+          name: nightly_binary_linux_wheel_py3.10_rocm5.1.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.10_rocm5.0
-          subfolder: rocm5.0/
+          - nightly_binary_linux_wheel_py3.10_rocm5.1.1
+          subfolder: rocm5.1.1/
       - 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", "cu116", "rocm4.5.2", "rocm5.0"],
+                "linux": ["cpu", "cu102", "cu113", "cu116", "rocm5.0", "rocm5.1.1"],
                 "win": ["cpu", "cu113", "cu116"],
                 "macos": ["cpu"],
             }