add rocm4.2 to nightly wheel builds (#3840)

* add rocm4.2 to nightly wheel builds

* remove rocm4.0.1 build
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>
Co-authored-by: Eli Uriegas <1700823+seemethere@users.noreply.github.com>

.circleci/config.yml

@@ -1002,16 +1002,16 @@ workflows:
           name: binary_linux_wheel_py3.6_cu111
           python_version: '3.6'
           wheel_docker_image: pytorch/manylinux-cuda111
-      - binary_linux_wheel:
-          cu_version: rocm4.0.1
-          name: binary_linux_wheel_py3.6_rocm4.0.1
-          python_version: '3.6'
-          wheel_docker_image: pytorch/manylinux-rocm:4.0.1
       - binary_linux_wheel:
           cu_version: rocm4.1
           name: binary_linux_wheel_py3.6_rocm4.1
           python_version: '3.6'
           wheel_docker_image: pytorch/manylinux-rocm:4.1
+      - binary_linux_wheel:
+          cu_version: rocm4.2
+          name: binary_linux_wheel_py3.6_rocm4.2
+          python_version: '3.6'
+          wheel_docker_image: pytorch/manylinux-rocm:4.2
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1030,16 +1030,16 @@ workflows:
           name: binary_linux_wheel_py3.7_cu111
           python_version: '3.7'
           wheel_docker_image: pytorch/manylinux-cuda111
-      - binary_linux_wheel:
-          cu_version: rocm4.0.1
-          name: binary_linux_wheel_py3.7_rocm4.0.1
-          python_version: '3.7'
-          wheel_docker_image: pytorch/manylinux-rocm:4.0.1
       - binary_linux_wheel:
           cu_version: rocm4.1
           name: binary_linux_wheel_py3.7_rocm4.1
           python_version: '3.7'
           wheel_docker_image: pytorch/manylinux-rocm:4.1
+      - 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:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1058,16 +1058,16 @@ workflows:
           name: binary_linux_wheel_py3.8_cu111
           python_version: '3.8'
           wheel_docker_image: pytorch/manylinux-cuda111
-      - binary_linux_wheel:
-          cu_version: rocm4.0.1
-          name: binary_linux_wheel_py3.8_rocm4.0.1
-          python_version: '3.8'
-          wheel_docker_image: pytorch/manylinux-rocm:4.0.1
       - binary_linux_wheel:
           cu_version: rocm4.1
           name: binary_linux_wheel_py3.8_rocm4.1
           python_version: '3.8'
           wheel_docker_image: pytorch/manylinux-rocm:4.1
+      - 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:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1086,16 +1086,16 @@ workflows:
           name: binary_linux_wheel_py3.9_cu111
           python_version: '3.9'
           wheel_docker_image: pytorch/manylinux-cuda111
-      - binary_linux_wheel:
-          cu_version: rocm4.0.1
-          name: binary_linux_wheel_py3.9_rocm4.0.1
-          python_version: '3.9'
-          wheel_docker_image: pytorch/manylinux-rocm:4.0.1
       - binary_linux_wheel:
           cu_version: rocm4.1
           name: binary_linux_wheel_py3.9_rocm4.1
           python_version: '3.9'
           wheel_docker_image: pytorch/manylinux-rocm:4.1
+      - 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_macos_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1725,15 +1725,15 @@ workflows:
           requires:
           - nightly_binary_linux_wheel_py3.6_cu111_upload
       - binary_linux_wheel:
-          cu_version: rocm4.0.1
+          cu_version: rocm4.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.6_rocm4.0.1
+          name: nightly_binary_linux_wheel_py3.6_rocm4.1
           python_version: '3.6'
-          wheel_docker_image: pytorch/manylinux-rocm:4.0.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -1741,29 +1741,29 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.6_rocm4.0.1_upload
+          name: nightly_binary_linux_wheel_py3.6_rocm4.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.6_rocm4.0.1
-          subfolder: rocm4.0.1/
+          - nightly_binary_linux_wheel_py3.6_rocm4.1
+          subfolder: rocm4.1/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.6_rocm4.0.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.6_rocm4.1_smoke_test_pip
           python_version: '3.6'
           requires:
-          - nightly_binary_linux_wheel_py3.6_rocm4.0.1_upload
+          - nightly_binary_linux_wheel_py3.6_rocm4.1_upload
       - binary_linux_wheel:
-          cu_version: rocm4.1
+          cu_version: rocm4.2
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.6_rocm4.1
+          name: nightly_binary_linux_wheel_py3.6_rocm4.2
           python_version: '3.6'
-          wheel_docker_image: pytorch/manylinux-rocm:4.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.2
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -1771,19 +1771,19 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.6_rocm4.1_upload
+          name: nightly_binary_linux_wheel_py3.6_rocm4.2_upload
           requires:
-          - nightly_binary_linux_wheel_py3.6_rocm4.1
-          subfolder: rocm4.1/
+          - nightly_binary_linux_wheel_py3.6_rocm4.2
+          subfolder: rocm4.2/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.6_rocm4.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.6_rocm4.2_smoke_test_pip
           python_version: '3.6'
           requires:
-          - nightly_binary_linux_wheel_py3.6_rocm4.1_upload
+          - nightly_binary_linux_wheel_py3.6_rocm4.2_upload
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -1878,15 +1878,15 @@ workflows:
           requires:
           - nightly_binary_linux_wheel_py3.7_cu111_upload
       - binary_linux_wheel:
-          cu_version: rocm4.0.1
+          cu_version: rocm4.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm4.0.1
+          name: nightly_binary_linux_wheel_py3.7_rocm4.1
           python_version: '3.7'
-          wheel_docker_image: pytorch/manylinux-rocm:4.0.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -1894,29 +1894,29 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm4.0.1_upload
+          name: nightly_binary_linux_wheel_py3.7_rocm4.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.7_rocm4.0.1
-          subfolder: rocm4.0.1/
+          - nightly_binary_linux_wheel_py3.7_rocm4.1
+          subfolder: rocm4.1/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.7_rocm4.0.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.7_rocm4.1_smoke_test_pip
           python_version: '3.7'
           requires:
-          - nightly_binary_linux_wheel_py3.7_rocm4.0.1_upload
+          - nightly_binary_linux_wheel_py3.7_rocm4.1_upload
       - binary_linux_wheel:
-          cu_version: rocm4.1
+          cu_version: rocm4.2
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm4.1
+          name: nightly_binary_linux_wheel_py3.7_rocm4.2
           python_version: '3.7'
-          wheel_docker_image: pytorch/manylinux-rocm:4.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.2
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -1924,19 +1924,19 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.7_rocm4.1_upload
+          name: nightly_binary_linux_wheel_py3.7_rocm4.2_upload
           requires:
-          - nightly_binary_linux_wheel_py3.7_rocm4.1
-          subfolder: rocm4.1/
+          - nightly_binary_linux_wheel_py3.7_rocm4.2
+          subfolder: rocm4.2/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.7_rocm4.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.7_rocm4.2_smoke_test_pip
           python_version: '3.7'
           requires:
-          - nightly_binary_linux_wheel_py3.7_rocm4.1_upload
+          - nightly_binary_linux_wheel_py3.7_rocm4.2_upload
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -2031,15 +2031,15 @@ workflows:
           requires:
           - nightly_binary_linux_wheel_py3.8_cu111_upload
       - binary_linux_wheel:
-          cu_version: rocm4.0.1
+          cu_version: rocm4.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm4.0.1
+          name: nightly_binary_linux_wheel_py3.8_rocm4.1
           python_version: '3.8'
-          wheel_docker_image: pytorch/manylinux-rocm:4.0.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2047,29 +2047,29 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm4.0.1_upload
+          name: nightly_binary_linux_wheel_py3.8_rocm4.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.8_rocm4.0.1
-          subfolder: rocm4.0.1/
+          - nightly_binary_linux_wheel_py3.8_rocm4.1
+          subfolder: rocm4.1/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.8_rocm4.0.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.8_rocm4.1_smoke_test_pip
           python_version: '3.8'
           requires:
-          - nightly_binary_linux_wheel_py3.8_rocm4.0.1_upload
+          - nightly_binary_linux_wheel_py3.8_rocm4.1_upload
       - binary_linux_wheel:
-          cu_version: rocm4.1
+          cu_version: rocm4.2
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm4.1
+          name: nightly_binary_linux_wheel_py3.8_rocm4.2
           python_version: '3.8'
-          wheel_docker_image: pytorch/manylinux-rocm:4.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.2
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2077,19 +2077,19 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.8_rocm4.1_upload
+          name: nightly_binary_linux_wheel_py3.8_rocm4.2_upload
           requires:
-          - nightly_binary_linux_wheel_py3.8_rocm4.1
-          subfolder: rocm4.1/
+          - nightly_binary_linux_wheel_py3.8_rocm4.2
+          subfolder: rocm4.2/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.8_rocm4.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.8_rocm4.2_smoke_test_pip
           python_version: '3.8'
           requires:
-          - nightly_binary_linux_wheel_py3.8_rocm4.1_upload
+          - nightly_binary_linux_wheel_py3.8_rocm4.2_upload
       - binary_linux_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu
@@ -2184,15 +2184,15 @@ workflows:
           requires:
           - nightly_binary_linux_wheel_py3.9_cu111_upload
       - binary_linux_wheel:
-          cu_version: rocm4.0.1
+          cu_version: rocm4.1
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm4.0.1
+          name: nightly_binary_linux_wheel_py3.9_rocm4.1
           python_version: '3.9'
-          wheel_docker_image: pytorch/manylinux-rocm:4.0.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.1
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2200,29 +2200,29 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm4.0.1_upload
+          name: nightly_binary_linux_wheel_py3.9_rocm4.1_upload
           requires:
-          - nightly_binary_linux_wheel_py3.9_rocm4.0.1
-          subfolder: rocm4.0.1/
+          - nightly_binary_linux_wheel_py3.9_rocm4.1
+          subfolder: rocm4.1/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.9_rocm4.0.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.9_rocm4.1_smoke_test_pip
           python_version: '3.9'
           requires:
-          - nightly_binary_linux_wheel_py3.9_rocm4.0.1_upload
+          - nightly_binary_linux_wheel_py3.9_rocm4.1_upload
       - binary_linux_wheel:
-          cu_version: rocm4.1
+          cu_version: rocm4.2
           filters:
             branches:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm4.1
+          name: nightly_binary_linux_wheel_py3.9_rocm4.2
           python_version: '3.9'
-          wheel_docker_image: pytorch/manylinux-rocm:4.1
+          wheel_docker_image: pytorch/manylinux-rocm:4.2
       - binary_wheel_upload:
           context: org-member
           filters:
@@ -2230,19 +2230,19 @@ workflows:
               only: nightly
             tags:
               only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
-          name: nightly_binary_linux_wheel_py3.9_rocm4.1_upload
+          name: nightly_binary_linux_wheel_py3.9_rocm4.2_upload
           requires:
-          - nightly_binary_linux_wheel_py3.9_rocm4.1
-          subfolder: rocm4.1/
+          - nightly_binary_linux_wheel_py3.9_rocm4.2
+          subfolder: rocm4.2/
       - smoke_test_linux_pip:
           filters:
             branches:
               only:
               - nightly
-          name: nightly_binary_linux_wheel_py3.9_rocm4.1_smoke_test_pip
+          name: nightly_binary_linux_wheel_py3.9_rocm4.2_smoke_test_pip
           python_version: '3.9'
           requires:
-          - nightly_binary_linux_wheel_py3.9_rocm4.1_upload
+          - nightly_binary_linux_wheel_py3.9_rocm4.2_upload
       - binary_macos_wheel:
           conda_docker_image: pytorch/conda-builder:cpu
           cu_version: cpu

.circleci/regenerate.py

@@ -30,7 +30,7 @@ def build_workflows(prefix='', filter_branch=None, upload=False, indentation=6,
     for btype in ["wheel", "conda"]:
         for os_type in ["linux", "macos", "win"]:
             python_versions = PYTHON_VERSIONS
-            cu_versions_dict = {"linux": ["cpu", "cu102", "cu111", "rocm4.0.1", "rocm4.1"],
+            cu_versions_dict = {"linux": ["cpu", "cu102", "cu111", "rocm4.1", "rocm4.2"],
                                 "win": ["cpu", "cu102", "cu111"],
                                 "macos": ["cpu"]}
             cu_versions = cu_versions_dict[os_type]