diff --git a/.circleci/config.yml b/.circleci/config.yml index e7c15dbb8cb..437d613d9ab 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -954,6 +954,11 @@ workflows: 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: conda_docker_image: pytorch/conda-builder:cpu cu_version: cpu @@ -983,6 +988,11 @@ workflows: 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: conda_docker_image: pytorch/conda-builder:cpu cu_version: cpu @@ -1012,6 +1022,11 @@ workflows: 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: conda_docker_image: pytorch/conda-builder:cpu cu_version: cpu @@ -1041,6 +1056,11 @@ workflows: 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_macos_wheel: conda_docker_image: pytorch/conda-builder:cpu cu_version: cpu @@ -1819,6 +1839,36 @@ workflows: python_version: '3.6' requires: - nightly_binary_linux_wheel_py3.6_rocm4.0.1_upload + - binary_linux_wheel: + 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.1 + python_version: '3.6' + wheel_docker_image: pytorch/manylinux-rocm:4.1 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.6_rocm4.1_upload + requires: + - 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.1_smoke_test_pip + python_version: '3.6' + requires: + - nightly_binary_linux_wheel_py3.6_rocm4.1_upload - binary_linux_wheel: conda_docker_image: pytorch/conda-builder:cpu cu_version: cpu @@ -1973,6 +2023,36 @@ workflows: python_version: '3.7' requires: - nightly_binary_linux_wheel_py3.7_rocm4.0.1_upload + - binary_linux_wheel: + 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.1 + python_version: '3.7' + wheel_docker_image: pytorch/manylinux-rocm:4.1 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.7_rocm4.1_upload + requires: + - 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.1_smoke_test_pip + python_version: '3.7' + requires: + - nightly_binary_linux_wheel_py3.7_rocm4.1_upload - binary_linux_wheel: conda_docker_image: pytorch/conda-builder:cpu cu_version: cpu @@ -2127,6 +2207,36 @@ workflows: python_version: '3.8' requires: - nightly_binary_linux_wheel_py3.8_rocm4.0.1_upload + - binary_linux_wheel: + 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.1 + python_version: '3.8' + wheel_docker_image: pytorch/manylinux-rocm:4.1 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.8_rocm4.1_upload + requires: + - 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.1_smoke_test_pip + python_version: '3.8' + requires: + - nightly_binary_linux_wheel_py3.8_rocm4.1_upload - binary_linux_wheel: conda_docker_image: pytorch/conda-builder:cpu cu_version: cpu @@ -2281,6 +2391,36 @@ workflows: python_version: '3.9' requires: - nightly_binary_linux_wheel_py3.9_rocm4.0.1_upload + - binary_linux_wheel: + 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.1 + python_version: '3.9' + wheel_docker_image: pytorch/manylinux-rocm:4.1 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_wheel_py3.9_rocm4.1_upload + requires: + - 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.1_smoke_test_pip + python_version: '3.9' + requires: + - nightly_binary_linux_wheel_py3.9_rocm4.1_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 4206a202814..d7d822db013 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -29,7 +29,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", "cu101", "cu102", "cu111", "rocm4.0.1"], + cu_versions_dict = {"linux": ["cpu", "cu101", "cu102", "cu111", "rocm4.0.1", "rocm4.1"], "win": ["cpu", "cu101", "cu102", "cu111"], "macos": ["cpu"]} cu_versions = cu_versions_dict[os_type]