Update to rocm4.5.2 wheels (#5179)

Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: Nikita Shulga <nshulga@fb.com>

.circleci/config.yml

@@ -1119,16 +1119,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.2
-          name: binary_linux_wheel_py3.7_rocm4.2
-          python_version: '3.7'
-          wheel_docker_image: pytorch/manylinux-rocm:4.2
       - 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:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1159,16 +1159,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.2
-          name: binary_linux_wheel_py3.8_rocm4.2
-          python_version: '3.8'
-          wheel_docker_image: pytorch/manylinux-rocm:4.2
       - 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:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1199,16 +1199,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.2
-          name: binary_linux_wheel_py3.9_rocm4.2
-          python_version: '3.9'
-          wheel_docker_image: pytorch/manylinux-rocm:4.2
       - 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_macos_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1872,15 +1872,15 @@ workflows:
           requires:
           - nightly_binary_linux_wheel_py3.7_cu115_upload
       - binary_linux_wheel:
-          cu_version: rocm4.2
+          cu_version: rocm4.3.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm4.2
+          name: nightly_binary_linux_wheel_py3.7_rocm4.3.1
           python_version: '3.7'
-          wheel_docker_image: pytorch/manylinux-rocm:4.2
+          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -1888,29 +1888,29 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm4.2_upload
+          name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.7_rocm4.2
-          subfolder: rocm4.2/
+          - nightly_binary_linux_wheel_py3.7_rocm4.3.1
+          subfolder: rocm4.3.1/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.7_rocm4.2_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_smoke_test_pip
           python_version: '3.7'
           requires:
-          - nightly_binary_linux_wheel_py3.7_rocm4.2_upload
+          - nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
       - 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:
@@ -1918,19 +1918,19 @@ 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/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.7_rocm4.5.2_smoke_test_pip
           python_version: '3.7'
           requires:
-          - nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
+          - nightly_binary_linux_wheel_py3.7_rocm4.5.2_upload
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -2087,15 +2087,15 @@ workflows:
           requires:
           - nightly_binary_linux_wheel_py3.8_cu115_upload
       - binary_linux_wheel:
-          cu_version: rocm4.2
+          cu_version: rocm4.3.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm4.2
+          name: nightly_binary_linux_wheel_py3.8_rocm4.3.1
           python_version: '3.8'
-          wheel_docker_image: pytorch/manylinux-rocm:4.2
+          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2103,29 +2103,29 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm4.2_upload
+          name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.8_rocm4.2
-          subfolder: rocm4.2/
+          - nightly_binary_linux_wheel_py3.8_rocm4.3.1
+          subfolder: rocm4.3.1/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.8_rocm4.2_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_smoke_test_pip
           python_version: '3.8'
           requires:
-          - nightly_binary_linux_wheel_py3.8_rocm4.2_upload
+          - nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
       - 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:
@@ -2133,19 +2133,19 @@ 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/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.8_rocm4.5.2_smoke_test_pip
           python_version: '3.8'
           requires:
-          - nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
+          - nightly_binary_linux_wheel_py3.8_rocm4.5.2_upload
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -2302,15 +2302,15 @@ workflows:
           requires:
           - nightly_binary_linux_wheel_py3.9_cu115_upload
       - binary_linux_wheel:
-          cu_version: rocm4.2
+          cu_version: rocm4.3.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm4.2
+          name: nightly_binary_linux_wheel_py3.9_rocm4.3.1
           python_version: '3.9'
-          wheel_docker_image: pytorch/manylinux-rocm:4.2
+          wheel_docker_image: pytorch/manylinux-rocm:4.3.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2318,29 +2318,29 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm4.2_upload
+          name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.9_rocm4.2
-          subfolder: rocm4.2/
+          - nightly_binary_linux_wheel_py3.9_rocm4.3.1
+          subfolder: rocm4.3.1/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.9_rocm4.2_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_smoke_test_pip
           python_version: '3.9'
           requires:
-          - nightly_binary_linux_wheel_py3.9_rocm4.2_upload
+          - nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
       - 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:
@@ -2348,19 +2348,19 @@ 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/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.9_rocm4.5.2_smoke_test_pip
           python_version: '3.9'
           requires:
-          - nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
+          - nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
       - 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", "cu111", "cu113", "cu115", "rocm4.2", "rocm4.3.1"],
+                "linux": ["cpu", "cu102", "cu111", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],
                 "win": ["cpu", "cu111", "cu113", "cu115"],
                 "macos": ["cpu"],
             }