From 4abcff8530ca8e338a427cb561fff934eaeae81c Mon Sep 17 00:00:00 2001 From: Jeff Daily Date: Wed, 24 Mar 2021 16:45:29 -0700 Subject: [PATCH] and ROCm 4.1 nightly wheels --- .circleci/config.yml | 140 ++++++++++++++++++++++++++++++++++++++++ .circleci/regenerate.py | 2 +- 2 files changed, 141 insertions(+), 1 deletion(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 1223abc9129..602fce4cd7d 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -834,6 +834,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 @@ -863,6 +868,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 @@ -892,6 +902,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 @@ -921,6 +936,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 @@ -1657,6 +1677,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 @@ -1811,6 +1861,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 @@ -1965,6 +2045,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 @@ -2119,6 +2229,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 24f663db9cb..5e5db28b985 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]