diff --git a/.circleci/config.yml b/.circleci/config.yml index 94ac3e3fd23..c1176abb97b 100644 --- a/.circleci/config.yml +++ b/.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 diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index 95490c2155d..a7a042cc75a 100755 --- a/.circleci/regenerate.py +++ b/.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"], }