Skip to content

GitLab

"git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "609f65ba23e655a5e9883803e01118dc734b7909"
Unverified Commit ffe735ba authored by Nikita Shulga's avatar Nikita Shulga Committed by GitHub
Browse files

Add CUDA binary builds (#1497) 

* Add CUDA binary builds

* Add "cuda_version": "cpu" to doc build jobs

* Add required cu_versions to pkg_helpers

* Regenerate config.yml
parent b78bdde0
Show whitespace changes
Inline Side-by-side
Showing with 1741 additions and 249 deletions
.circleci/config.yml
View file @ ffe735ba
......@@ -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:
......
.circleci/config.yml.in
View file @ ffe735ba
......@@ -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
......
.circleci/regenerate.py
View file @ ffe735ba
......@@ -21,6 +21,10 @@ import os.path
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
......@@ -82,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:
......@@ -107,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:
......@@ -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):
......@@ -192,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":
......@@ -204,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)
......
packaging/pkg_helpers.bash
View file @ ffe735ba
......@@ -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)
;;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment