From 8b9294d7f6274381778f1b27860ce27b717460c2 Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Mon, 10 May 2021 09:21:34 -0700 Subject: [PATCH 1/4] Add CUDA binary builds --- .circleci/config.yml.in | 18 +++++++++++++++--- .circleci/regenerate.py | 42 +++++++++++++++++++++-------------------- 2 files changed, 37 insertions(+), 23 deletions(-) diff --git a/.circleci/config.yml.in b/.circleci/config.yml.in index 16b47bde79..400728a88a 100644 --- a/.circleci/config.yml.in +++ b/.circleci/config.yml.in @@ -71,11 +71,23 @@ binary_common: &binary_common python_version: description: "Python version to build against (e.g., 3.8)" type: string + cuda_version: + description: "CUDA version to build against (e.g., cpu, cu101)" + type: string + default: "cpu" + wheel_docker_image: + description: "Wheel only: what docker image to use" + type: string + default: "pytorch/manylinux-cuda102" + conda_docker_image: + description: "Conda only: what docker image to use" + type: string + default: "pytorch/conda-builder:cuda102" environment: &environment PYTHON_VERSION: << parameters.python_version >> BUILD_VERSION: << parameters.build_version >> PYTORCH_VERSION: << parameters.pytorch_version >> - CU_VERSION: cpu + CU_VERSION: << parameters.cuda_version >> smoke_test_common: &smoke_test_common <<: *binary_common @@ -127,7 +139,7 @@ jobs: binary_linux_wheel: <<: *binary_common docker: - - image: "pytorch/manylinux-cuda102" + - image: << parameters.wheel_docker_image >> resource_class: 2xlarge+ steps: - checkout @@ -144,7 +156,7 @@ jobs: binary_linux_conda: <<: *binary_common docker: - - image: "pytorch/conda-cuda" + - image: "<< parameters.conda_docker_image >>" resource_class: 2xlarge+ steps: - checkout diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index 137fed87c4..2a1646d2c7 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -21,6 +21,10 @@ PYTHON_VERSIONS = ["3.6", "3.7", "3.8", "3.9"] +CU_VERSIONS_DICT = {"linux": ["cpu", "cu102", "cu111"], + "windows": ["cpu", "cu102", "cu111"], + "macos": ["cpu"]} + DOC_VERSION = ('linux', '3.8') @@ -31,7 +35,8 @@ def build_workflows(prefix='', upload=False, filter_branch=None, indentation=6): for btype in ["wheel", "conda"]: for os_type in ["linux", "macos", "windows"]: for python_version in PYTHON_VERSIONS: - w += build_workflow_pair(btype, os_type, python_version, filter_branch, prefix, upload) + for cu_version in CU_VERSIONS_DICT[os_type]: + w += build_workflow_pair(btype, os_type, python_version, cu_version, filter_branch, prefix, upload) if not filter_branch: # Build on every pull request, but upload only on nightly and tags @@ -53,27 +58,19 @@ def build_download_job(filter_branch): return [{"download_third_parties_nix": job}] -def build_workflow_pair(btype, os_type, python_version, filter_branch, prefix='', upload=False): +def build_workflow_pair(btype, os_type, python_version, cu_version, filter_branch, prefix='', upload=False): w = [] - base_workflow_name = "{prefix}binary_{os_type}_{btype}_py{python_version}".format( - prefix=prefix, - os_type=os_type, - btype=btype, - python_version=python_version, - ) - - w.append(generate_base_workflow(base_workflow_name, python_version, filter_branch, os_type, btype)) + base_workflow_name = f"{prefix}binary_{os_type}_{btype}_py{python_version}_{cu_version}" + w.append(generate_base_workflow(base_workflow_name, python_version, cu_version, filter_branch, os_type, btype)) if upload: - is_py3_linux = os_type in ['linux', "windows"] and not python_version.startswith("2.") - w.append(generate_upload_workflow(base_workflow_name, filter_branch, btype)) - if filter_branch == 'nightly' and is_py3_linux: + if filter_branch == 'nightly' and os_type != 'macos': pydistro = 'pip' if btype == 'wheel' else 'conda' - w.append(generate_smoketest_workflow(pydistro, base_workflow_name, filter_branch, python_version, os_type)) + w.append(generate_smoketest_workflow(pydistro, base_workflow_name, filter_branch, python_version, cu_version, os_type)) return w @@ -115,15 +112,20 @@ def docstring_parameters_sync_job(filter_branch): return [{"docstring_parameters_sync": job}] -def generate_base_workflow(base_workflow_name, python_version, filter_branch, os_type, btype): +def generate_base_workflow(base_workflow_name, python_version, cu_version, filter_branch, os_type, btype): d = { "name": base_workflow_name, "python_version": python_version, + "cuda_version": cu_version, } if os_type in ['linux', 'macos']: d['requires'] = ['download_third_parties_nix'] + if btype == 'conda': + d['conda_docker_image'] = f'pytorch/conda-builder:{cu_version.replace("cu1","cuda1")}' + elif cu_version != 'cpu': + d['wheel_docker_image'] = f'pytorch/manylinux-{cu_version.replace("cu1","cuda1")}' if filter_branch: d["filters"] = gen_filter_branch_tree(filter_branch) @@ -157,23 +159,23 @@ def generate_upload_workflow(base_workflow_name, filter_branch, btype): return {"binary_{btype}_upload".format(btype=btype): d} -def generate_smoketest_workflow(pydistro, base_workflow_name, filter_branch, python_version, os_type): +def generate_smoketest_workflow(pydistro, base_workflow_name, filter_branch, python_version, cu_version, os_type): required_build_suffix = "_upload" required_build_name = base_workflow_name + required_build_suffix - smoke_suffix = "smoke_test_{pydistro}".format(pydistro=pydistro) + smoke_suffix = f"smoke_test_{pydistro}".format(pydistro=pydistro) d = { - "name": "{base_workflow_name}_{smoke_suffix}".format( - base_workflow_name=base_workflow_name, smoke_suffix=smoke_suffix), + "name": f"{base_workflow_name}_{smoke_suffix}", "requires": [required_build_name], "python_version": python_version, + "cuda_version": cu_version, } if filter_branch: d["filters"] = gen_filter_branch_tree(filter_branch) - return {"smoke_test_{os_type}_{pydistro}".format(os_type=os_type, pydistro=pydistro): d} + return {f"smoke_test_{os_type}_{pydistro}": d} def indent(indentation, data_list): From 5dfdfb86d5a1b5992e67d57f11b74a58d3d90952 Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Mon, 10 May 2021 09:40:31 -0700 Subject: [PATCH 2/4] Add "cuda_version": "cpu" to doc build jobs --- .circleci/regenerate.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index 2a1646d2c7..f1720979fb 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -79,7 +79,7 @@ def build_doc_job(filter_branch): job = { "name": "build_docs", "python_version": "3.8", - "requires": ["binary_linux_wheel_py3.8", ], + "requires": ["binary_linux_wheel_py3.8_cpu", ], } if filter_branch: @@ -104,7 +104,7 @@ def docstring_parameters_sync_job(filter_branch): job = { "name": "docstring_parameters_sync", "python_version": "3.8", - "requires": ["binary_linux_wheel_py3.8", ], + "requires": ["binary_linux_wheel_py3.8_cpu", ], } if filter_branch: @@ -194,6 +194,7 @@ def unittest_workflows(indentation=6): job = { "name": f"unittest_{os_type}_{device_type}_py{python_version}", "python_version": python_version, + "cuda_version": 'cpu' if device_type=="cpu" else "cu102", } if os_type != "windows": @@ -206,6 +207,7 @@ def unittest_workflows(indentation=6): "stylecheck": { "name": f"stylecheck_py{python_version}", "python_version": python_version, + "cuda_version": 'cpu' if device_type=="cpu" else "cu102", } }) return indent(indentation, jobs) From c196ea510f33c78b9dd5e8d28cc607f3dfad8760 Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Mon, 10 May 2021 09:29:17 -0700 Subject: [PATCH 3/4] Add required cu_versions to pkg_helpers --- packaging/pkg_helpers.bash | 51 +++++++++++++++++++++++++++++++++++--- 1 file changed, 47 insertions(+), 4 deletions(-) diff --git a/packaging/pkg_helpers.bash b/packaging/pkg_helpers.bash index 5dc5397042..173aa4ca07 100644 --- a/packaging/pkg_helpers.bash +++ b/packaging/pkg_helpers.bash @@ -52,17 +52,60 @@ setup_cuda() { # Now work out the CUDA settings case "$CU_VERSION" in + cu112) + if [[ "$OSTYPE" == "msys" ]]; then + export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.2" + else + export CUDA_HOME=/usr/local/cuda-11.2/ + fi + export FORCE_CUDA=1 + export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" + ;; + cu111) + if [[ "$OSTYPE" == "msys" ]]; then + export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.1" + else + export CUDA_HOME=/usr/local/cuda-11.1/ + fi + export FORCE_CUDA=1 + export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" + ;; + cu110) + if [[ "$OSTYPE" == "msys" ]]; then + export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0" + else + export CUDA_HOME=/usr/local/cuda-11.0/ + fi + export FORCE_CUDA=1 + export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0" + ;; + cu102) + if [[ "$OSTYPE" == "msys" ]]; then + export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.2" + else + export CUDA_HOME=/usr/local/cuda-10.2/ + fi + export FORCE_CUDA=1 + export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5" + ;; + cu101) + if [[ "$OSTYPE" == "msys" ]]; then + export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1" + else + export CUDA_HOME=/usr/local/cuda-10.1/ + fi + export FORCE_CUDA=1 + export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5" + ;; cu100) export CUDA_HOME=/usr/local/cuda-10.0/ export FORCE_CUDA=1 - # Hard-coding gencode flags is temporary situation until - # https://github.com/pytorch/pytorch/pull/23408 lands - export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_50,code=compute_50" + export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5" ;; cu92) export CUDA_HOME=/usr/local/cuda-9.2/ export FORCE_CUDA=1 - export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_50,code=compute_50" + export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0" ;; cpu) ;; From 014e98136c1de88b1759af78ff7a787a284c8622 Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Mon, 10 May 2021 09:29:40 -0700 Subject: [PATCH 4/4] Regenerate config.yml --- .circleci/config.yml | 1873 +++++++++++++++++++++++++++++++++++++----- 1 file changed, 1653 insertions(+), 220 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 64a6ae9f9f..2d1e9dd028 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -71,11 +71,23 @@ binary_common: &binary_common python_version: description: "Python version to build against (e.g., 3.8)" type: string + cuda_version: + description: "CUDA version to build against (e.g., cpu, cu101)" + type: string + default: "cpu" + wheel_docker_image: + description: "Wheel only: what docker image to use" + type: string + default: "pytorch/manylinux-cuda102" + conda_docker_image: + description: "Conda only: what docker image to use" + type: string + default: "pytorch/conda-builder:cuda102" environment: &environment PYTHON_VERSION: << parameters.python_version >> BUILD_VERSION: << parameters.build_version >> PYTORCH_VERSION: << parameters.pytorch_version >> - CU_VERSION: cpu + CU_VERSION: << parameters.cuda_version >> smoke_test_common: &smoke_test_common <<: *binary_common @@ -127,7 +139,7 @@ jobs: binary_linux_wheel: <<: *binary_common docker: - - image: "pytorch/manylinux-cuda102" + - image: << parameters.wheel_docker_image >> resource_class: 2xlarge+ steps: - checkout @@ -144,7 +156,7 @@ jobs: binary_linux_conda: <<: *binary_common docker: - - image: "pytorch/conda-cuda" + - image: "<< parameters.conda_docker_image >>" resource_class: 2xlarge+ steps: - checkout @@ -619,114 +631,342 @@ workflows: - download_third_parties_nix: name: download_third_parties_nix - binary_linux_wheel: - name: binary_linux_wheel_py3.6 + cuda_version: cpu + name: binary_linux_wheel_py3.6_cpu + python_version: '3.6' + requires: + - download_third_parties_nix + - binary_linux_wheel: + cuda_version: cu102 + name: binary_linux_wheel_py3.6_cu102 + python_version: '3.6' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda102 + - binary_linux_wheel: + cuda_version: cu111 + name: binary_linux_wheel_py3.6_cu111 python_version: '3.6' requires: - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_linux_wheel: + cuda_version: cpu + name: binary_linux_wheel_py3.7_cpu + python_version: '3.7' + requires: + - download_third_parties_nix + - binary_linux_wheel: + cuda_version: cu102 + name: binary_linux_wheel_py3.7_cu102 + python_version: '3.7' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda102 - binary_linux_wheel: - name: binary_linux_wheel_py3.7 + cuda_version: cu111 + name: binary_linux_wheel_py3.7_cu111 python_version: '3.7' requires: - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_linux_wheel: + cuda_version: cpu + name: binary_linux_wheel_py3.8_cpu + python_version: '3.8' + requires: + - download_third_parties_nix + - binary_linux_wheel: + cuda_version: cu102 + name: binary_linux_wheel_py3.8_cu102 + python_version: '3.8' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda102 - binary_linux_wheel: - name: binary_linux_wheel_py3.8 + cuda_version: cu111 + name: binary_linux_wheel_py3.8_cu111 python_version: '3.8' requires: - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_linux_wheel: + cuda_version: cpu + name: binary_linux_wheel_py3.9_cpu + python_version: '3.9' + requires: + - download_third_parties_nix + - binary_linux_wheel: + cuda_version: cu102 + name: binary_linux_wheel_py3.9_cu102 + python_version: '3.9' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda102 - binary_linux_wheel: - name: binary_linux_wheel_py3.9 + cuda_version: cu111 + name: binary_linux_wheel_py3.9_cu111 python_version: '3.9' requires: - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda111 - binary_macos_wheel: - name: binary_macos_wheel_py3.6 + cuda_version: cpu + name: binary_macos_wheel_py3.6_cpu python_version: '3.6' requires: - download_third_parties_nix - binary_macos_wheel: - name: binary_macos_wheel_py3.7 + cuda_version: cpu + name: binary_macos_wheel_py3.7_cpu python_version: '3.7' requires: - download_third_parties_nix - binary_macos_wheel: - name: binary_macos_wheel_py3.8 + cuda_version: cpu + name: binary_macos_wheel_py3.8_cpu python_version: '3.8' requires: - download_third_parties_nix - binary_macos_wheel: - name: binary_macos_wheel_py3.9 + cuda_version: cpu + name: binary_macos_wheel_py3.9_cpu python_version: '3.9' requires: - download_third_parties_nix - binary_windows_wheel: - name: binary_windows_wheel_py3.6 + cuda_version: cpu + name: binary_windows_wheel_py3.6_cpu + python_version: '3.6' + - binary_windows_wheel: + cuda_version: cu102 + name: binary_windows_wheel_py3.6_cu102 + python_version: '3.6' + wheel_docker_image: pytorch/manylinux-cuda102 + - binary_windows_wheel: + cuda_version: cu111 + name: binary_windows_wheel_py3.6_cu111 python_version: '3.6' + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_windows_wheel: + cuda_version: cpu + name: binary_windows_wheel_py3.7_cpu + python_version: '3.7' + - binary_windows_wheel: + cuda_version: cu102 + name: binary_windows_wheel_py3.7_cu102 + python_version: '3.7' + wheel_docker_image: pytorch/manylinux-cuda102 - binary_windows_wheel: - name: binary_windows_wheel_py3.7 + cuda_version: cu111 + name: binary_windows_wheel_py3.7_cu111 python_version: '3.7' + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_windows_wheel: + cuda_version: cpu + name: binary_windows_wheel_py3.8_cpu + python_version: '3.8' + - binary_windows_wheel: + cuda_version: cu102 + name: binary_windows_wheel_py3.8_cu102 + python_version: '3.8' + wheel_docker_image: pytorch/manylinux-cuda102 - binary_windows_wheel: - name: binary_windows_wheel_py3.8 + cuda_version: cu111 + name: binary_windows_wheel_py3.8_cu111 python_version: '3.8' + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_windows_wheel: + cuda_version: cpu + name: binary_windows_wheel_py3.9_cpu + python_version: '3.9' + - binary_windows_wheel: + cuda_version: cu102 + name: binary_windows_wheel_py3.9_cu102 + python_version: '3.9' + wheel_docker_image: pytorch/manylinux-cuda102 - binary_windows_wheel: - name: binary_windows_wheel_py3.9 + cuda_version: cu111 + name: binary_windows_wheel_py3.9_cu111 python_version: '3.9' + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_linux_conda_py3.6_cpu + python_version: '3.6' + requires: + - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + name: binary_linux_conda_py3.6_cu102 + python_version: '3.6' + requires: + - download_third_parties_nix - binary_linux_conda: - name: binary_linux_conda_py3.6 + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + name: binary_linux_conda_py3.6_cu111 python_version: '3.6' requires: - download_third_parties_nix - binary_linux_conda: - name: binary_linux_conda_py3.7 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_linux_conda_py3.7_cpu + python_version: '3.7' + requires: + - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + name: binary_linux_conda_py3.7_cu102 + python_version: '3.7' + requires: + - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + name: binary_linux_conda_py3.7_cu111 python_version: '3.7' requires: - download_third_parties_nix - binary_linux_conda: - name: binary_linux_conda_py3.8 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_linux_conda_py3.8_cpu + python_version: '3.8' + requires: + - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + name: binary_linux_conda_py3.8_cu102 + python_version: '3.8' + requires: + - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + name: binary_linux_conda_py3.8_cu111 python_version: '3.8' requires: - download_third_parties_nix - binary_linux_conda: - name: binary_linux_conda_py3.9 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_linux_conda_py3.9_cpu + python_version: '3.9' + requires: + - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + name: binary_linux_conda_py3.9_cu102 + python_version: '3.9' + requires: + - download_third_parties_nix + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + name: binary_linux_conda_py3.9_cu111 python_version: '3.9' requires: - download_third_parties_nix - binary_macos_conda: - name: binary_macos_conda_py3.6 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_macos_conda_py3.6_cpu python_version: '3.6' requires: - download_third_parties_nix - binary_macos_conda: - name: binary_macos_conda_py3.7 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_macos_conda_py3.7_cpu python_version: '3.7' requires: - download_third_parties_nix - binary_macos_conda: - name: binary_macos_conda_py3.8 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_macos_conda_py3.8_cpu python_version: '3.8' requires: - download_third_parties_nix - binary_macos_conda: - name: binary_macos_conda_py3.9 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_macos_conda_py3.9_cpu python_version: '3.9' requires: - download_third_parties_nix - binary_windows_conda: - name: binary_windows_conda_py3.6 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_windows_conda_py3.6_cpu + python_version: '3.6' + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + name: binary_windows_conda_py3.6_cu102 + python_version: '3.6' + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + name: binary_windows_conda_py3.6_cu111 python_version: '3.6' - binary_windows_conda: - name: binary_windows_conda_py3.7 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_windows_conda_py3.7_cpu + python_version: '3.7' + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + name: binary_windows_conda_py3.7_cu102 + python_version: '3.7' + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + name: binary_windows_conda_py3.7_cu111 python_version: '3.7' - binary_windows_conda: - name: binary_windows_conda_py3.8 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_windows_conda_py3.8_cpu + python_version: '3.8' + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + name: binary_windows_conda_py3.8_cu102 + python_version: '3.8' + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + name: binary_windows_conda_py3.8_cu111 python_version: '3.8' - binary_windows_conda: - name: binary_windows_conda_py3.9 + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + name: binary_windows_conda_py3.9_cpu + python_version: '3.9' + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + name: binary_windows_conda_py3.9_cu102 + python_version: '3.9' + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + name: binary_windows_conda_py3.9_cu111 python_version: '3.9' - build_docs: name: build_docs python_version: '3.8' requires: - - binary_linux_wheel_py3.8 + - binary_linux_wheel_py3.8_cpu - upload_docs: context: org-member filters: @@ -743,94 +983,115 @@ workflows: name: docstring_parameters_sync python_version: '3.8' requires: - - binary_linux_wheel_py3.8 + - binary_linux_wheel_py3.8_cpu unittest: jobs: - download_third_parties_nix: name: download_third_parties_nix - unittest_linux_cpu: + cuda_version: cpu name: unittest_linux_cpu_py3.6 python_version: '3.6' requires: - download_third_parties_nix - stylecheck: + cuda_version: cpu name: stylecheck_py3.6 python_version: '3.6' - unittest_linux_cpu: + cuda_version: cpu name: unittest_linux_cpu_py3.7 python_version: '3.7' requires: - download_third_parties_nix - unittest_linux_cpu: + cuda_version: cpu name: unittest_linux_cpu_py3.8 python_version: '3.8' requires: - download_third_parties_nix - unittest_linux_cpu: + cuda_version: cpu name: unittest_linux_cpu_py3.9 python_version: '3.9' requires: - download_third_parties_nix - unittest_linux_gpu: + cuda_version: cu102 name: unittest_linux_gpu_py3.6 python_version: '3.6' requires: - download_third_parties_nix - unittest_linux_gpu: + cuda_version: cu102 name: unittest_linux_gpu_py3.7 python_version: '3.7' requires: - download_third_parties_nix - unittest_linux_gpu: + cuda_version: cu102 name: unittest_linux_gpu_py3.8 python_version: '3.8' requires: - download_third_parties_nix - unittest_linux_gpu: + cuda_version: cu102 name: unittest_linux_gpu_py3.9 python_version: '3.9' requires: - download_third_parties_nix - unittest_windows_cpu: + cuda_version: cpu name: unittest_windows_cpu_py3.6 python_version: '3.6' - unittest_windows_cpu: + cuda_version: cpu name: unittest_windows_cpu_py3.7 python_version: '3.7' - unittest_windows_cpu: + cuda_version: cpu name: unittest_windows_cpu_py3.8 python_version: '3.8' - unittest_windows_cpu: + cuda_version: cpu name: unittest_windows_cpu_py3.9 python_version: '3.9' - unittest_windows_gpu: + cuda_version: cu102 name: unittest_windows_gpu_py3.6 python_version: '3.6' - unittest_windows_gpu: + cuda_version: cu102 name: unittest_windows_gpu_py3.7 python_version: '3.7' - unittest_windows_gpu: + cuda_version: cu102 name: unittest_windows_gpu_py3.8 python_version: '3.8' - unittest_windows_gpu: + cuda_version: cu102 name: unittest_windows_gpu_py3.9 python_version: '3.9' - unittest_macos_cpu: + cuda_version: cpu name: unittest_macos_cpu_py3.6 python_version: '3.6' requires: - download_third_parties_nix - unittest_macos_cpu: + cuda_version: cpu name: unittest_macos_cpu_py3.7 python_version: '3.7' requires: - download_third_parties_nix - unittest_macos_cpu: + cuda_version: cpu name: unittest_macos_cpu_py3.8 python_version: '3.8' requires: - download_third_parties_nix - unittest_macos_cpu: + cuda_version: cpu name: unittest_macos_cpu_py3.9 python_version: '3.9' requires: @@ -850,13 +1111,14 @@ workflows: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ name: download_third_parties_nix - binary_linux_wheel: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.6 + name: nightly_binary_linux_wheel_py3.6_cpu python_version: '3.6' requires: - download_third_parties_nix @@ -868,31 +1130,34 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.6_upload + name: nightly_binary_linux_wheel_py3.6_cpu_upload requires: - - nightly_binary_linux_wheel_py3.6 + - nightly_binary_linux_wheel_py3.6_cpu - smoke_test_linux_pip: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.6_smoke_test_pip + name: nightly_binary_linux_wheel_py3.6_cpu_smoke_test_pip python_version: '3.6' requires: - - nightly_binary_linux_wheel_py3.6_upload + - nightly_binary_linux_wheel_py3.6_cpu_upload - binary_linux_wheel: + cuda_version: cu102 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.7 - python_version: '3.7' + name: nightly_binary_linux_wheel_py3.6_cu102 + python_version: '3.6' requires: - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda102 - binary_wheel_upload: context: org-member filters: @@ -901,31 +1166,34 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.7_upload + name: nightly_binary_linux_wheel_py3.6_cu102_upload requires: - - nightly_binary_linux_wheel_py3.7 + - nightly_binary_linux_wheel_py3.6_cu102 - smoke_test_linux_pip: + cuda_version: cu102 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.7_smoke_test_pip - python_version: '3.7' + name: nightly_binary_linux_wheel_py3.6_cu102_smoke_test_pip + python_version: '3.6' requires: - - nightly_binary_linux_wheel_py3.7_upload + - nightly_binary_linux_wheel_py3.6_cu102_upload - binary_linux_wheel: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.8 - python_version: '3.8' + name: nightly_binary_linux_wheel_py3.6_cu111 + python_version: '3.6' requires: - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda111 - binary_wheel_upload: context: org-member filters: @@ -934,29 +1202,31 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.8_upload + name: nightly_binary_linux_wheel_py3.6_cu111_upload requires: - - nightly_binary_linux_wheel_py3.8 + - nightly_binary_linux_wheel_py3.6_cu111 - smoke_test_linux_pip: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.8_smoke_test_pip - python_version: '3.8' + name: nightly_binary_linux_wheel_py3.6_cu111_smoke_test_pip + python_version: '3.6' requires: - - nightly_binary_linux_wheel_py3.8_upload + - nightly_binary_linux_wheel_py3.6_cu111_upload - binary_linux_wheel: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.9 - python_version: '3.9' + name: nightly_binary_linux_wheel_py3.7_cpu + python_version: '3.7' requires: - download_third_parties_nix - binary_wheel_upload: @@ -967,31 +1237,34 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.9_upload + name: nightly_binary_linux_wheel_py3.7_cpu_upload requires: - - nightly_binary_linux_wheel_py3.9 + - nightly_binary_linux_wheel_py3.7_cpu - smoke_test_linux_pip: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_wheel_py3.9_smoke_test_pip - python_version: '3.9' + name: nightly_binary_linux_wheel_py3.7_cpu_smoke_test_pip + python_version: '3.7' requires: - - nightly_binary_linux_wheel_py3.9_upload - - binary_macos_wheel: + - nightly_binary_linux_wheel_py3.7_cpu_upload + - binary_linux_wheel: + cuda_version: cu102 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_wheel_py3.6 - python_version: '3.6' + name: nightly_binary_linux_wheel_py3.7_cu102 + python_version: '3.7' requires: - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda102 - binary_wheel_upload: context: org-member filters: @@ -1000,42 +1273,34 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_wheel_py3.6_upload + name: nightly_binary_linux_wheel_py3.7_cu102_upload requires: - - nightly_binary_macos_wheel_py3.6 - - binary_macos_wheel: + - nightly_binary_linux_wheel_py3.7_cu102 + - smoke_test_linux_pip: + cuda_version: cu102 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_wheel_py3.7 + name: nightly_binary_linux_wheel_py3.7_cu102_smoke_test_pip python_version: '3.7' requires: - - download_third_parties_nix - - binary_wheel_upload: - context: org-member - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_wheel_py3.7_upload - requires: - - nightly_binary_macos_wheel_py3.7 - - binary_macos_wheel: + - nightly_binary_linux_wheel_py3.7_cu102_upload + - binary_linux_wheel: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_wheel_py3.8 - python_version: '3.8' + name: nightly_binary_linux_wheel_py3.7_cu111 + python_version: '3.7' requires: - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda111 - binary_wheel_upload: context: org-member filters: @@ -1044,40 +1309,33 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_wheel_py3.8_upload + name: nightly_binary_linux_wheel_py3.7_cu111_upload requires: - - nightly_binary_macos_wheel_py3.8 - - binary_macos_wheel: + - nightly_binary_linux_wheel_py3.7_cu111 + - smoke_test_linux_pip: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_wheel_py3.9 - python_version: '3.9' + name: nightly_binary_linux_wheel_py3.7_cu111_smoke_test_pip + python_version: '3.7' requires: - - download_third_parties_nix - - binary_wheel_upload: - context: org-member + - nightly_binary_linux_wheel_py3.7_cu111_upload + - binary_linux_wheel: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_wheel_py3.9_upload + name: nightly_binary_linux_wheel_py3.8_cpu + python_version: '3.8' requires: - - nightly_binary_macos_wheel_py3.9 - - binary_windows_wheel: - filters: - branches: - only: - - nightly - tags: - only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.6 - python_version: '3.6' + - download_third_parties_nix - binary_wheel_upload: context: org-member filters: @@ -1086,29 +1344,34 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.6_upload + name: nightly_binary_linux_wheel_py3.8_cpu_upload requires: - - nightly_binary_windows_wheel_py3.6 - - smoke_test_windows_pip: + - nightly_binary_linux_wheel_py3.8_cpu + - smoke_test_linux_pip: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.6_smoke_test_pip - python_version: '3.6' + name: nightly_binary_linux_wheel_py3.8_cpu_smoke_test_pip + python_version: '3.8' requires: - - nightly_binary_windows_wheel_py3.6_upload - - binary_windows_wheel: + - nightly_binary_linux_wheel_py3.8_cpu_upload + - binary_linux_wheel: + cuda_version: cu102 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.7 - python_version: '3.7' + name: nightly_binary_linux_wheel_py3.8_cu102 + python_version: '3.8' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda102 - binary_wheel_upload: context: org-member filters: @@ -1117,29 +1380,34 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.7_upload + name: nightly_binary_linux_wheel_py3.8_cu102_upload requires: - - nightly_binary_windows_wheel_py3.7 - - smoke_test_windows_pip: + - nightly_binary_linux_wheel_py3.8_cu102 + - smoke_test_linux_pip: + cuda_version: cu102 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.7_smoke_test_pip - python_version: '3.7' + name: nightly_binary_linux_wheel_py3.8_cu102_smoke_test_pip + python_version: '3.8' requires: - - nightly_binary_windows_wheel_py3.7_upload - - binary_windows_wheel: + - nightly_binary_linux_wheel_py3.8_cu102_upload + - binary_linux_wheel: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.8 + name: nightly_binary_linux_wheel_py3.8_cu111 python_version: '3.8' + requires: + - download_third_parties_nix + wheel_docker_image: pytorch/manylinux-cuda111 - binary_wheel_upload: context: org-member filters: @@ -1148,29 +1416,33 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.8_upload + name: nightly_binary_linux_wheel_py3.8_cu111_upload requires: - - nightly_binary_windows_wheel_py3.8 - - smoke_test_windows_pip: + - nightly_binary_linux_wheel_py3.8_cu111 + - smoke_test_linux_pip: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.8_smoke_test_pip + name: nightly_binary_linux_wheel_py3.8_cu111_smoke_test_pip python_version: '3.8' requires: - - nightly_binary_windows_wheel_py3.8_upload - - binary_windows_wheel: + - nightly_binary_linux_wheel_py3.8_cu111_upload + - binary_linux_wheel: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.9 + name: nightly_binary_linux_wheel_py3.9_cpu python_version: '3.9' + requires: + - download_third_parties_nix - binary_wheel_upload: context: org-member filters: @@ -1179,32 +1451,35 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.9_upload + name: nightly_binary_linux_wheel_py3.9_cpu_upload requires: - - nightly_binary_windows_wheel_py3.9 - - smoke_test_windows_pip: + - nightly_binary_linux_wheel_py3.9_cpu + - smoke_test_linux_pip: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_wheel_py3.9_smoke_test_pip + name: nightly_binary_linux_wheel_py3.9_cpu_smoke_test_pip python_version: '3.9' requires: - - nightly_binary_windows_wheel_py3.9_upload - - binary_linux_conda: + - nightly_binary_linux_wheel_py3.9_cpu_upload + - binary_linux_wheel: + cuda_version: cu102 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.6 - python_version: '3.6' + name: nightly_binary_linux_wheel_py3.9_cu102 + python_version: '3.9' requires: - download_third_parties_nix - - binary_conda_upload: + wheel_docker_image: pytorch/manylinux-cuda102 + - binary_wheel_upload: context: org-member filters: branches: @@ -1212,32 +1487,35 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.6_upload + name: nightly_binary_linux_wheel_py3.9_cu102_upload requires: - - nightly_binary_linux_conda_py3.6 - - smoke_test_linux_conda: + - nightly_binary_linux_wheel_py3.9_cu102 + - smoke_test_linux_pip: + cuda_version: cu102 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.6_smoke_test_conda - python_version: '3.6' + name: nightly_binary_linux_wheel_py3.9_cu102_smoke_test_pip + python_version: '3.9' requires: - - nightly_binary_linux_conda_py3.6_upload - - binary_linux_conda: + - nightly_binary_linux_wheel_py3.9_cu102_upload + - binary_linux_wheel: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.7 - python_version: '3.7' + name: nightly_binary_linux_wheel_py3.9_cu111 + python_version: '3.9' requires: - download_third_parties_nix - - binary_conda_upload: + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_wheel_upload: context: org-member filters: branches: @@ -1245,32 +1523,34 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.7_upload + name: nightly_binary_linux_wheel_py3.9_cu111_upload requires: - - nightly_binary_linux_conda_py3.7 - - smoke_test_linux_conda: + - nightly_binary_linux_wheel_py3.9_cu111 + - smoke_test_linux_pip: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.7_smoke_test_conda - python_version: '3.7' + name: nightly_binary_linux_wheel_py3.9_cu111_smoke_test_pip + python_version: '3.9' requires: - - nightly_binary_linux_conda_py3.7_upload - - binary_linux_conda: + - nightly_binary_linux_wheel_py3.9_cu111_upload + - binary_macos_wheel: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.8 - python_version: '3.8' + name: nightly_binary_macos_wheel_py3.6_cpu + python_version: '3.6' requires: - download_third_parties_nix - - binary_conda_upload: + - binary_wheel_upload: context: org-member filters: branches: @@ -1278,65 +1558,68 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.8_upload + name: nightly_binary_macos_wheel_py3.6_cpu_upload requires: - - nightly_binary_linux_conda_py3.8 - - smoke_test_linux_conda: + - nightly_binary_macos_wheel_py3.6_cpu + - binary_macos_wheel: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.8_smoke_test_conda - python_version: '3.8' + name: nightly_binary_macos_wheel_py3.7_cpu + python_version: '3.7' requires: - - nightly_binary_linux_conda_py3.8_upload - - binary_linux_conda: + - download_third_parties_nix + - binary_wheel_upload: + context: org-member filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.9 - python_version: '3.9' + name: nightly_binary_macos_wheel_py3.7_cpu_upload requires: - - download_third_parties_nix - - binary_conda_upload: - context: org-member + - nightly_binary_macos_wheel_py3.7_cpu + - binary_macos_wheel: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.9_upload + name: nightly_binary_macos_wheel_py3.8_cpu + python_version: '3.8' requires: - - nightly_binary_linux_conda_py3.9 - - smoke_test_linux_conda: + - download_third_parties_nix + - binary_wheel_upload: + context: org-member filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_linux_conda_py3.9_smoke_test_conda - python_version: '3.9' + name: nightly_binary_macos_wheel_py3.8_cpu_upload requires: - - nightly_binary_linux_conda_py3.9_upload - - binary_macos_conda: + - nightly_binary_macos_wheel_py3.8_cpu + - binary_macos_wheel: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_conda_py3.6 - python_version: '3.6' + name: nightly_binary_macos_wheel_py3.9_cpu + python_version: '3.9' requires: - download_third_parties_nix - - binary_conda_upload: + - binary_wheel_upload: context: org-member filters: branches: @@ -1344,21 +1627,20 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_conda_py3.6_upload + name: nightly_binary_macos_wheel_py3.9_cpu_upload requires: - - nightly_binary_macos_conda_py3.6 - - binary_macos_conda: + - nightly_binary_macos_wheel_py3.9_cpu + - binary_windows_wheel: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_conda_py3.7 - python_version: '3.7' - requires: - - download_third_parties_nix - - binary_conda_upload: + name: nightly_binary_windows_wheel_py3.6_cpu + python_version: '3.6' + - binary_wheel_upload: context: org-member filters: branches: @@ -1366,63 +1648,1000 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_conda_py3.7_upload + name: nightly_binary_windows_wheel_py3.6_cpu_upload requires: - - nightly_binary_macos_conda_py3.7 - - binary_macos_conda: + - nightly_binary_windows_wheel_py3.6_cpu + - smoke_test_windows_pip: + cuda_version: cpu filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_conda_py3.8 - python_version: '3.8' + name: nightly_binary_windows_wheel_py3.6_cpu_smoke_test_pip + python_version: '3.6' requires: - - download_third_parties_nix - - binary_conda_upload: - context: org-member + - nightly_binary_windows_wheel_py3.6_cpu_upload + - binary_windows_wheel: + cuda_version: cu102 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_conda_py3.8_upload - requires: - - nightly_binary_macos_conda_py3.8 - - binary_macos_conda: + name: nightly_binary_windows_wheel_py3.6_cu102 + python_version: '3.6' + wheel_docker_image: pytorch/manylinux-cuda102 + - binary_wheel_upload: + context: org-member filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_conda_py3.9 - python_version: '3.9' + name: nightly_binary_windows_wheel_py3.6_cu102_upload requires: - - download_third_parties_nix - - binary_conda_upload: - context: org-member + - nightly_binary_windows_wheel_py3.6_cu102 + - smoke_test_windows_pip: + cuda_version: cu102 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_macos_conda_py3.9_upload + name: nightly_binary_windows_wheel_py3.6_cu102_smoke_test_pip + python_version: '3.6' requires: - - nightly_binary_macos_conda_py3.9 - - binary_windows_conda: + - nightly_binary_windows_wheel_py3.6_cu102_upload + - binary_windows_wheel: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.6 + name: nightly_binary_windows_wheel_py3.6_cu111 python_version: '3.6' - - binary_conda_upload: + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.6_cu111_upload + requires: + - nightly_binary_windows_wheel_py3.6_cu111 + - smoke_test_windows_pip: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.6_cu111_smoke_test_pip + python_version: '3.6' + requires: + - nightly_binary_windows_wheel_py3.6_cu111_upload + - binary_windows_wheel: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.7_cpu + python_version: '3.7' + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.7_cpu_upload + requires: + - nightly_binary_windows_wheel_py3.7_cpu + - smoke_test_windows_pip: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.7_cpu_smoke_test_pip + python_version: '3.7' + requires: + - nightly_binary_windows_wheel_py3.7_cpu_upload + - binary_windows_wheel: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.7_cu102 + python_version: '3.7' + wheel_docker_image: pytorch/manylinux-cuda102 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.7_cu102_upload + requires: + - nightly_binary_windows_wheel_py3.7_cu102 + - smoke_test_windows_pip: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.7_cu102_smoke_test_pip + python_version: '3.7' + requires: + - nightly_binary_windows_wheel_py3.7_cu102_upload + - binary_windows_wheel: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.7_cu111 + python_version: '3.7' + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.7_cu111_upload + requires: + - nightly_binary_windows_wheel_py3.7_cu111 + - smoke_test_windows_pip: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.7_cu111_smoke_test_pip + python_version: '3.7' + requires: + - nightly_binary_windows_wheel_py3.7_cu111_upload + - binary_windows_wheel: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.8_cpu + python_version: '3.8' + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.8_cpu_upload + requires: + - nightly_binary_windows_wheel_py3.8_cpu + - smoke_test_windows_pip: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.8_cpu_smoke_test_pip + python_version: '3.8' + requires: + - nightly_binary_windows_wheel_py3.8_cpu_upload + - binary_windows_wheel: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.8_cu102 + python_version: '3.8' + wheel_docker_image: pytorch/manylinux-cuda102 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.8_cu102_upload + requires: + - nightly_binary_windows_wheel_py3.8_cu102 + - smoke_test_windows_pip: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.8_cu102_smoke_test_pip + python_version: '3.8' + requires: + - nightly_binary_windows_wheel_py3.8_cu102_upload + - binary_windows_wheel: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.8_cu111 + python_version: '3.8' + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.8_cu111_upload + requires: + - nightly_binary_windows_wheel_py3.8_cu111 + - smoke_test_windows_pip: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.8_cu111_smoke_test_pip + python_version: '3.8' + requires: + - nightly_binary_windows_wheel_py3.8_cu111_upload + - binary_windows_wheel: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.9_cpu + python_version: '3.9' + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.9_cpu_upload + requires: + - nightly_binary_windows_wheel_py3.9_cpu + - smoke_test_windows_pip: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.9_cpu_smoke_test_pip + python_version: '3.9' + requires: + - nightly_binary_windows_wheel_py3.9_cpu_upload + - binary_windows_wheel: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.9_cu102 + python_version: '3.9' + wheel_docker_image: pytorch/manylinux-cuda102 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.9_cu102_upload + requires: + - nightly_binary_windows_wheel_py3.9_cu102 + - smoke_test_windows_pip: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.9_cu102_smoke_test_pip + python_version: '3.9' + requires: + - nightly_binary_windows_wheel_py3.9_cu102_upload + - binary_windows_wheel: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.9_cu111 + python_version: '3.9' + wheel_docker_image: pytorch/manylinux-cuda111 + - binary_wheel_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.9_cu111_upload + requires: + - nightly_binary_windows_wheel_py3.9_cu111 + - smoke_test_windows_pip: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_wheel_py3.9_cu111_smoke_test_pip + python_version: '3.9' + requires: + - nightly_binary_windows_wheel_py3.9_cu111_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_cpu + python_version: '3.6' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_cpu_upload + requires: + - nightly_binary_linux_conda_py3.6_cpu + - smoke_test_linux_conda: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_cpu_smoke_test_conda + python_version: '3.6' + requires: + - nightly_binary_linux_conda_py3.6_cpu_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_cu102 + python_version: '3.6' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_cu102_upload + requires: + - nightly_binary_linux_conda_py3.6_cu102 + - smoke_test_linux_conda: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_cu102_smoke_test_conda + python_version: '3.6' + requires: + - nightly_binary_linux_conda_py3.6_cu102_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_cu111 + python_version: '3.6' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_cu111_upload + requires: + - nightly_binary_linux_conda_py3.6_cu111 + - smoke_test_linux_conda: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.6_cu111_smoke_test_conda + python_version: '3.6' + requires: + - nightly_binary_linux_conda_py3.6_cu111_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_cpu + python_version: '3.7' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_cpu_upload + requires: + - nightly_binary_linux_conda_py3.7_cpu + - smoke_test_linux_conda: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_cpu_smoke_test_conda + python_version: '3.7' + requires: + - nightly_binary_linux_conda_py3.7_cpu_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_cu102 + python_version: '3.7' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_cu102_upload + requires: + - nightly_binary_linux_conda_py3.7_cu102 + - smoke_test_linux_conda: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_cu102_smoke_test_conda + python_version: '3.7' + requires: + - nightly_binary_linux_conda_py3.7_cu102_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_cu111 + python_version: '3.7' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_cu111_upload + requires: + - nightly_binary_linux_conda_py3.7_cu111 + - smoke_test_linux_conda: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.7_cu111_smoke_test_conda + python_version: '3.7' + requires: + - nightly_binary_linux_conda_py3.7_cu111_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_cpu + python_version: '3.8' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_cpu_upload + requires: + - nightly_binary_linux_conda_py3.8_cpu + - smoke_test_linux_conda: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_cpu_smoke_test_conda + python_version: '3.8' + requires: + - nightly_binary_linux_conda_py3.8_cpu_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_cu102 + python_version: '3.8' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_cu102_upload + requires: + - nightly_binary_linux_conda_py3.8_cu102 + - smoke_test_linux_conda: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_cu102_smoke_test_conda + python_version: '3.8' + requires: + - nightly_binary_linux_conda_py3.8_cu102_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_cu111 + python_version: '3.8' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_cu111_upload + requires: + - nightly_binary_linux_conda_py3.8_cu111 + - smoke_test_linux_conda: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.8_cu111_smoke_test_conda + python_version: '3.8' + requires: + - nightly_binary_linux_conda_py3.8_cu111_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_cpu + python_version: '3.9' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_cpu_upload + requires: + - nightly_binary_linux_conda_py3.9_cpu + - smoke_test_linux_conda: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_cpu_smoke_test_conda + python_version: '3.9' + requires: + - nightly_binary_linux_conda_py3.9_cpu_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_cu102 + python_version: '3.9' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_cu102_upload + requires: + - nightly_binary_linux_conda_py3.9_cu102 + - smoke_test_linux_conda: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_cu102_smoke_test_conda + python_version: '3.9' + requires: + - nightly_binary_linux_conda_py3.9_cu102_upload + - binary_linux_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_cu111 + python_version: '3.9' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_cu111_upload + requires: + - nightly_binary_linux_conda_py3.9_cu111 + - smoke_test_linux_conda: + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_linux_conda_py3.9_cu111_smoke_test_conda + python_version: '3.9' + requires: + - nightly_binary_linux_conda_py3.9_cu111_upload + - binary_macos_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_macos_conda_py3.6_cpu + python_version: '3.6' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_macos_conda_py3.6_cpu_upload + requires: + - nightly_binary_macos_conda_py3.6_cpu + - binary_macos_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_macos_conda_py3.7_cpu + python_version: '3.7' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_macos_conda_py3.7_cpu_upload + requires: + - nightly_binary_macos_conda_py3.7_cpu + - binary_macos_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_macos_conda_py3.8_cpu + python_version: '3.8' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_macos_conda_py3.8_cpu_upload + requires: + - nightly_binary_macos_conda_py3.8_cpu + - binary_macos_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_macos_conda_py3.9_cpu + python_version: '3.9' + requires: + - download_third_parties_nix + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_macos_conda_py3.9_cpu_upload + requires: + - nightly_binary_macos_conda_py3.9_cpu + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.6_cpu + python_version: '3.6' + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.6_cpu_upload + requires: + - nightly_binary_windows_conda_py3.6_cpu + - smoke_test_windows_conda: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.6_cpu_smoke_test_conda + python_version: '3.6' + requires: + - nightly_binary_windows_conda_py3.6_cpu_upload + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.6_cu102 + python_version: '3.6' + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.6_cu102_upload + requires: + - nightly_binary_windows_conda_py3.6_cu102 + - smoke_test_windows_conda: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.6_cu102_smoke_test_conda + python_version: '3.6' + requires: + - nightly_binary_windows_conda_py3.6_cu102_upload + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.6_cu111 + python_version: '3.6' + - binary_conda_upload: context: org-member filters: branches: @@ -1430,28 +2649,99 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.6_upload + name: nightly_binary_windows_conda_py3.6_cu111_upload requires: - - nightly_binary_windows_conda_py3.6 + - nightly_binary_windows_conda_py3.6_cu111 - smoke_test_windows_conda: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.6_smoke_test_conda + name: nightly_binary_windows_conda_py3.6_cu111_smoke_test_conda python_version: '3.6' requires: - - nightly_binary_windows_conda_py3.6_upload + - nightly_binary_windows_conda_py3.6_cu111_upload + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.7_cpu + python_version: '3.7' + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.7_cpu_upload + requires: + - nightly_binary_windows_conda_py3.7_cpu + - smoke_test_windows_conda: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.7_cpu_smoke_test_conda + python_version: '3.7' + requires: + - nightly_binary_windows_conda_py3.7_cpu_upload + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.7_cu102 + python_version: '3.7' + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.7_cu102_upload + requires: + - nightly_binary_windows_conda_py3.7_cu102 + - smoke_test_windows_conda: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.7_cu102_smoke_test_conda + python_version: '3.7' + requires: + - nightly_binary_windows_conda_py3.7_cu102_upload - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.7 + name: nightly_binary_windows_conda_py3.7_cu111 python_version: '3.7' - binary_conda_upload: context: org-member @@ -1461,28 +2751,99 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.7_upload + name: nightly_binary_windows_conda_py3.7_cu111_upload requires: - - nightly_binary_windows_conda_py3.7 + - nightly_binary_windows_conda_py3.7_cu111 - smoke_test_windows_conda: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.7_smoke_test_conda + name: nightly_binary_windows_conda_py3.7_cu111_smoke_test_conda python_version: '3.7' requires: - - nightly_binary_windows_conda_py3.7_upload + - nightly_binary_windows_conda_py3.7_cu111_upload + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.8_cpu + python_version: '3.8' + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.8_cpu_upload + requires: + - nightly_binary_windows_conda_py3.8_cpu + - smoke_test_windows_conda: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.8_cpu_smoke_test_conda + python_version: '3.8' + requires: + - nightly_binary_windows_conda_py3.8_cpu_upload + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.8_cu102 + python_version: '3.8' + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.8_cu102_upload + requires: + - nightly_binary_windows_conda_py3.8_cu102 + - smoke_test_windows_conda: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.8_cu102_smoke_test_conda + python_version: '3.8' + requires: + - nightly_binary_windows_conda_py3.8_cu102_upload - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.8 + name: nightly_binary_windows_conda_py3.8_cu111 python_version: '3.8' - binary_conda_upload: context: org-member @@ -1492,28 +2853,99 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.8_upload + name: nightly_binary_windows_conda_py3.8_cu111_upload requires: - - nightly_binary_windows_conda_py3.8 + - nightly_binary_windows_conda_py3.8_cu111 - smoke_test_windows_conda: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.8_smoke_test_conda + name: nightly_binary_windows_conda_py3.8_cu111_smoke_test_conda python_version: '3.8' requires: - - nightly_binary_windows_conda_py3.8_upload + - nightly_binary_windows_conda_py3.8_cu111_upload + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cpu + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.9_cpu + python_version: '3.9' + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.9_cpu_upload + requires: + - nightly_binary_windows_conda_py3.9_cpu + - smoke_test_windows_conda: + cuda_version: cpu + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.9_cpu_smoke_test_conda + python_version: '3.9' + requires: + - nightly_binary_windows_conda_py3.9_cpu_upload + - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda102 + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.9_cu102 + python_version: '3.9' + - binary_conda_upload: + context: org-member + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.9_cu102_upload + requires: + - nightly_binary_windows_conda_py3.9_cu102 + - smoke_test_windows_conda: + cuda_version: cu102 + filters: + branches: + only: + - nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_windows_conda_py3.9_cu102_smoke_test_conda + python_version: '3.9' + requires: + - nightly_binary_windows_conda_py3.9_cu102_upload - binary_windows_conda: + conda_docker_image: pytorch/conda-builder:cuda111 + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.9 + name: nightly_binary_windows_conda_py3.9_cu111 python_version: '3.9' - binary_conda_upload: context: org-member @@ -1523,20 +2955,21 @@ workflows: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.9_upload + name: nightly_binary_windows_conda_py3.9_cu111_upload requires: - - nightly_binary_windows_conda_py3.9 + - nightly_binary_windows_conda_py3.9_cu111 - smoke_test_windows_conda: + cuda_version: cu111 filters: branches: only: - nightly tags: only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ - name: nightly_binary_windows_conda_py3.9_smoke_test_conda + name: nightly_binary_windows_conda_py3.9_cu111_smoke_test_conda python_version: '3.9' requires: - - nightly_binary_windows_conda_py3.9_upload + - nightly_binary_windows_conda_py3.9_cu111_upload docker_build: triggers: - schedule: