Skip to content

GitLab

"git@developer.sourcefind.cn:orangecat/ollama.git" did not exist on "176d0f7075b7de1097b2f88121a9a3c9434c2d39"
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 ...@@ -71,11 +71,23 @@ binary_common: &binary_common
python_version: python_version:
description: "Python version to build against (e.g., 3.8)" description: "Python version to build against (e.g., 3.8)"
type: string 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 environment: &environment
PYTHON_VERSION: << parameters.python_version >> PYTHON_VERSION: << parameters.python_version >>
BUILD_VERSION: << parameters.build_version >> BUILD_VERSION: << parameters.build_version >>
PYTORCH_VERSION: << parameters.pytorch_version >> PYTORCH_VERSION: << parameters.pytorch_version >>
CU_VERSION: cpu CU_VERSION: << parameters.cuda_version >>
smoke_test_common: &smoke_test_common smoke_test_common: &smoke_test_common
<<: *binary_common <<: *binary_common
...@@ -127,7 +139,7 @@ jobs: ...@@ -127,7 +139,7 @@ jobs:
binary_linux_wheel: binary_linux_wheel:
<<: *binary_common <<: *binary_common
docker: docker:
- image: "pytorch/manylinux-cuda102" - image: << parameters.wheel_docker_image >>
resource_class: 2xlarge+ resource_class: 2xlarge+
steps: steps:
- checkout - checkout
...@@ -144,7 +156,7 @@ jobs: ...@@ -144,7 +156,7 @@ jobs:
binary_linux_conda: binary_linux_conda:
<<: *binary_common <<: *binary_common
docker: docker:
- image: "pytorch/conda-cuda" - image: "<< parameters.conda_docker_image >>"
resource_class: 2xlarge+ resource_class: 2xlarge+
steps: steps:
- checkout - checkout
...@@ -619,114 +631,342 @@ workflows: ...@@ -619,114 +631,342 @@ workflows:
- download_third_parties_nix: - download_third_parties_nix:
name: download_third_parties_nix name: download_third_parties_nix
- binary_linux_wheel: - 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' python_version: '3.6'
requires: requires:
- download_third_parties_nix - 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: - binary_linux_wheel:
name: binary_linux_wheel_py3.7 cuda_version: cu111
name: binary_linux_wheel_py3.7_cu111
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - 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: - binary_linux_wheel:
name: binary_linux_wheel_py3.8 cuda_version: cu111
name: binary_linux_wheel_py3.8_cu111
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - 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: - binary_linux_wheel:
name: binary_linux_wheel_py3.9 cuda_version: cu111
name: binary_linux_wheel_py3.9_cu111
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda111
- binary_macos_wheel: - binary_macos_wheel:
name: binary_macos_wheel_py3.6 cuda_version: cpu
name: binary_macos_wheel_py3.6_cpu
python_version: '3.6' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_macos_wheel: - binary_macos_wheel:
name: binary_macos_wheel_py3.7 cuda_version: cpu
name: binary_macos_wheel_py3.7_cpu
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_macos_wheel: - binary_macos_wheel:
name: binary_macos_wheel_py3.8 cuda_version: cpu
name: binary_macos_wheel_py3.8_cpu
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_macos_wheel: - binary_macos_wheel:
name: binary_macos_wheel_py3.9 cuda_version: cpu
name: binary_macos_wheel_py3.9_cpu
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_windows_wheel: - 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' 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: - binary_windows_wheel:
name: binary_windows_wheel_py3.7 cuda_version: cu111
name: binary_windows_wheel_py3.7_cu111
python_version: '3.7' 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: - binary_windows_wheel:
name: binary_windows_wheel_py3.8 cuda_version: cu111
name: binary_windows_wheel_py3.8_cu111
python_version: '3.8' 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: - binary_windows_wheel:
name: binary_windows_wheel_py3.9 cuda_version: cu111
name: binary_windows_wheel_py3.9_cu111
python_version: '3.9' 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: - 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' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_conda: - 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' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_conda: - 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' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_conda: - 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' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_macos_conda: - 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' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_macos_conda: - 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' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_macos_conda: - 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' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_macos_conda: - 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' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_windows_conda: - 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' python_version: '3.6'
- binary_windows_conda: - 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' python_version: '3.7'
- binary_windows_conda: - 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' python_version: '3.8'
- binary_windows_conda: - 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' python_version: '3.9'
- build_docs: - build_docs:
name: build_docs name: build_docs
python_version: '3.8' python_version: '3.8'
requires: requires:
- binary_linux_wheel_py3.8 - binary_linux_wheel_py3.8_cpu
- upload_docs: - upload_docs:
context: org-member context: org-member
filters: filters:
...@@ -743,94 +983,115 @@ workflows: ...@@ -743,94 +983,115 @@ workflows:
name: docstring_parameters_sync name: docstring_parameters_sync
python_version: '3.8' python_version: '3.8'
requires: requires:
- binary_linux_wheel_py3.8 - binary_linux_wheel_py3.8_cpu
unittest: unittest:
jobs: jobs:
- download_third_parties_nix: - download_third_parties_nix:
name: download_third_parties_nix name: download_third_parties_nix
- unittest_linux_cpu: - unittest_linux_cpu:
cuda_version: cpu
name: unittest_linux_cpu_py3.6 name: unittest_linux_cpu_py3.6
python_version: '3.6' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- stylecheck: - stylecheck:
cuda_version: cpu
name: stylecheck_py3.6 name: stylecheck_py3.6
python_version: '3.6' python_version: '3.6'
- unittest_linux_cpu: - unittest_linux_cpu:
cuda_version: cpu
name: unittest_linux_cpu_py3.7 name: unittest_linux_cpu_py3.7
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_cpu: - unittest_linux_cpu:
cuda_version: cpu
name: unittest_linux_cpu_py3.8 name: unittest_linux_cpu_py3.8
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_cpu: - unittest_linux_cpu:
cuda_version: cpu
name: unittest_linux_cpu_py3.9 name: unittest_linux_cpu_py3.9
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_gpu: - unittest_linux_gpu:
cuda_version: cu102
name: unittest_linux_gpu_py3.6 name: unittest_linux_gpu_py3.6
python_version: '3.6' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_gpu: - unittest_linux_gpu:
cuda_version: cu102
name: unittest_linux_gpu_py3.7 name: unittest_linux_gpu_py3.7
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_gpu: - unittest_linux_gpu:
cuda_version: cu102
name: unittest_linux_gpu_py3.8 name: unittest_linux_gpu_py3.8
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_gpu: - unittest_linux_gpu:
cuda_version: cu102
name: unittest_linux_gpu_py3.9 name: unittest_linux_gpu_py3.9
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_windows_cpu: - unittest_windows_cpu:
cuda_version: cpu
name: unittest_windows_cpu_py3.6 name: unittest_windows_cpu_py3.6
python_version: '3.6' python_version: '3.6'
- unittest_windows_cpu: - unittest_windows_cpu:
cuda_version: cpu
name: unittest_windows_cpu_py3.7 name: unittest_windows_cpu_py3.7
python_version: '3.7' python_version: '3.7'
- unittest_windows_cpu: - unittest_windows_cpu:
cuda_version: cpu
name: unittest_windows_cpu_py3.8 name: unittest_windows_cpu_py3.8
python_version: '3.8' python_version: '3.8'
- unittest_windows_cpu: - unittest_windows_cpu:
cuda_version: cpu
name: unittest_windows_cpu_py3.9 name: unittest_windows_cpu_py3.9
python_version: '3.9' python_version: '3.9'
- unittest_windows_gpu: - unittest_windows_gpu:
cuda_version: cu102
name: unittest_windows_gpu_py3.6 name: unittest_windows_gpu_py3.6
python_version: '3.6' python_version: '3.6'
- unittest_windows_gpu: - unittest_windows_gpu:
cuda_version: cu102
name: unittest_windows_gpu_py3.7 name: unittest_windows_gpu_py3.7
python_version: '3.7' python_version: '3.7'
- unittest_windows_gpu: - unittest_windows_gpu:
cuda_version: cu102
name: unittest_windows_gpu_py3.8 name: unittest_windows_gpu_py3.8
python_version: '3.8' python_version: '3.8'
- unittest_windows_gpu: - unittest_windows_gpu:
cuda_version: cu102
name: unittest_windows_gpu_py3.9 name: unittest_windows_gpu_py3.9
python_version: '3.9' python_version: '3.9'
- unittest_macos_cpu: - unittest_macos_cpu:
cuda_version: cpu
name: unittest_macos_cpu_py3.6 name: unittest_macos_cpu_py3.6
python_version: '3.6' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_macos_cpu: - unittest_macos_cpu:
cuda_version: cpu
name: unittest_macos_cpu_py3.7 name: unittest_macos_cpu_py3.7
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_macos_cpu: - unittest_macos_cpu:
cuda_version: cpu
name: unittest_macos_cpu_py3.8 name: unittest_macos_cpu_py3.8
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_macos_cpu: - unittest_macos_cpu:
cuda_version: cpu
name: unittest_macos_cpu_py3.9 name: unittest_macos_cpu_py3.9
python_version: '3.9' python_version: '3.9'
requires: requires:
...@@ -850,13 +1111,14 @@ workflows: ...@@ -850,13 +1111,14 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: download_third_parties_nix name: download_third_parties_nix
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
...@@ -868,31 +1130,34 @@ workflows: ...@@ -868,31 +1130,34 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_wheel_py3.6 - nightly_binary_linux_wheel_py3.6_cpu
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.6'
requires: requires:
- nightly_binary_linux_wheel_py3.6_upload - nightly_binary_linux_wheel_py3.6_cpu_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu102
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7 name: nightly_binary_linux_wheel_py3.6_cu102
python_version: '3.7' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -901,31 +1166,34 @@ workflows: ...@@ -901,31 +1166,34 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_wheel_py3.7 - nightly_binary_linux_wheel_py3.6_cu102
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: cu102
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_smoke_test_pip name: nightly_binary_linux_wheel_py3.6_cu102_smoke_test_pip
python_version: '3.7' python_version: '3.6'
requires: requires:
- nightly_binary_linux_wheel_py3.7_upload - nightly_binary_linux_wheel_py3.6_cu102_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8 name: nightly_binary_linux_wheel_py3.6_cu111
python_version: '3.8' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -934,29 +1202,31 @@ workflows: ...@@ -934,29 +1202,31 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_wheel_py3.8 - nightly_binary_linux_wheel_py3.6_cu111
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_smoke_test_pip name: nightly_binary_linux_wheel_py3.6_cu111_smoke_test_pip
python_version: '3.8' python_version: '3.6'
requires: requires:
- nightly_binary_linux_wheel_py3.8_upload - nightly_binary_linux_wheel_py3.6_cu111_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9 name: nightly_binary_linux_wheel_py3.7_cpu
python_version: '3.9' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_wheel_upload: - binary_wheel_upload:
...@@ -967,31 +1237,34 @@ workflows: ...@@ -967,31 +1237,34 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_wheel_py3.9 - nightly_binary_linux_wheel_py3.7_cpu
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_smoke_test_pip name: nightly_binary_linux_wheel_py3.7_cpu_smoke_test_pip
python_version: '3.9' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.9_upload - nightly_binary_linux_wheel_py3.7_cpu_upload
- binary_macos_wheel: - binary_linux_wheel:
cuda_version: cu102
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.6 name: nightly_binary_linux_wheel_py3.7_cu102
python_version: '3.6' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1000,42 +1273,34 @@ workflows: ...@@ -1000,42 +1273,34 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_macos_wheel_py3.6 - nightly_binary_linux_wheel_py3.7_cu102
- binary_macos_wheel: - smoke_test_linux_pip:
cuda_version: cu102
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.7'
requires: requires:
- download_third_parties_nix - nightly_binary_linux_wheel_py3.7_cu102_upload
- binary_wheel_upload: - binary_linux_wheel:
context: org-member cuda_version: cu111
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:
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.8 name: nightly_binary_linux_wheel_py3.7_cu111
python_version: '3.8' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1044,40 +1309,33 @@ workflows: ...@@ -1044,40 +1309,33 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_macos_wheel_py3.8 - nightly_binary_linux_wheel_py3.7_cu111
- binary_macos_wheel: - smoke_test_linux_pip:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.9 name: nightly_binary_linux_wheel_py3.7_cu111_smoke_test_pip
python_version: '3.9' python_version: '3.7'
requires: requires:
- download_third_parties_nix - nightly_binary_linux_wheel_py3.7_cu111_upload
- binary_wheel_upload: - binary_linux_wheel:
context: org-member cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_macos_wheel_py3.9 - download_third_parties_nix
- 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'
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1086,29 +1344,34 @@ workflows: ...@@ -1086,29 +1344,34 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_windows_wheel_py3.6 - nightly_binary_linux_wheel_py3.8_cpu
- smoke_test_windows_pip: - smoke_test_linux_pip:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.6_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_cpu_smoke_test_pip
python_version: '3.6' python_version: '3.8'
requires: requires:
- nightly_binary_windows_wheel_py3.6_upload - nightly_binary_linux_wheel_py3.8_cpu_upload
- binary_windows_wheel: - binary_linux_wheel:
cuda_version: cu102
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7 name: nightly_binary_linux_wheel_py3.8_cu102
python_version: '3.7' python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1117,29 +1380,34 @@ workflows: ...@@ -1117,29 +1380,34 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_windows_wheel_py3.7 - nightly_binary_linux_wheel_py3.8_cu102
- smoke_test_windows_pip: - smoke_test_linux_pip:
cuda_version: cu102
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_cu102_smoke_test_pip
python_version: '3.7' python_version: '3.8'
requires: requires:
- nightly_binary_windows_wheel_py3.7_upload - nightly_binary_linux_wheel_py3.8_cu102_upload
- binary_windows_wheel: - binary_linux_wheel:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1148,29 +1416,33 @@ workflows: ...@@ -1148,29 +1416,33 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_windows_wheel_py3.8 - nightly_binary_linux_wheel_py3.8_cu111
- smoke_test_windows_pip: - smoke_test_linux_pip:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.8'
requires: requires:
- nightly_binary_windows_wheel_py3.8_upload - nightly_binary_linux_wheel_py3.8_cu111_upload
- binary_windows_wheel: - binary_linux_wheel:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.9'
requires:
- download_third_parties_nix
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1179,32 +1451,35 @@ workflows: ...@@ -1179,32 +1451,35 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_windows_wheel_py3.9 - nightly_binary_linux_wheel_py3.9_cpu
- smoke_test_windows_pip: - smoke_test_linux_pip:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.9'
requires: requires:
- nightly_binary_windows_wheel_py3.9_upload - nightly_binary_linux_wheel_py3.9_cpu_upload
- binary_linux_conda: - binary_linux_wheel:
cuda_version: cu102
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6 name: nightly_binary_linux_wheel_py3.9_cu102
python_version: '3.6' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_conda_upload: wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload:
context: org-member context: org-member
filters: filters:
branches: branches:
...@@ -1212,32 +1487,35 @@ workflows: ...@@ -1212,32 +1487,35 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_conda_py3.6 - nightly_binary_linux_wheel_py3.9_cu102
- smoke_test_linux_conda: - smoke_test_linux_pip:
cuda_version: cu102
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6_smoke_test_conda name: nightly_binary_linux_wheel_py3.9_cu102_smoke_test_pip
python_version: '3.6' python_version: '3.9'
requires: requires:
- nightly_binary_linux_conda_py3.6_upload - nightly_binary_linux_wheel_py3.9_cu102_upload
- binary_linux_conda: - binary_linux_wheel:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7 name: nightly_binary_linux_wheel_py3.9_cu111
python_version: '3.7' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_conda_upload: wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload:
context: org-member context: org-member
filters: filters:
branches: branches:
...@@ -1245,32 +1523,34 @@ workflows: ...@@ -1245,32 +1523,34 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_conda_py3.7 - nightly_binary_linux_wheel_py3.9_cu111
- smoke_test_linux_conda: - smoke_test_linux_pip:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_smoke_test_conda name: nightly_binary_linux_wheel_py3.9_cu111_smoke_test_pip
python_version: '3.7' python_version: '3.9'
requires: requires:
- nightly_binary_linux_conda_py3.7_upload - nightly_binary_linux_wheel_py3.9_cu111_upload
- binary_linux_conda: - binary_macos_wheel:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8 name: nightly_binary_macos_wheel_py3.6_cpu
python_version: '3.8' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_conda_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
branches: branches:
...@@ -1278,65 +1558,68 @@ workflows: ...@@ -1278,65 +1558,68 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_conda_py3.8 - nightly_binary_macos_wheel_py3.6_cpu
- smoke_test_linux_conda: - binary_macos_wheel:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_smoke_test_conda name: nightly_binary_macos_wheel_py3.7_cpu
python_version: '3.8' python_version: '3.7'
requires: requires:
- nightly_binary_linux_conda_py3.8_upload - download_third_parties_nix
- binary_linux_conda: - binary_wheel_upload:
context: org-member
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9 name: nightly_binary_macos_wheel_py3.7_cpu_upload
python_version: '3.9'
requires: requires:
- download_third_parties_nix - nightly_binary_macos_wheel_py3.7_cpu
- binary_conda_upload: - binary_macos_wheel:
context: org-member cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_conda_py3.9 - download_third_parties_nix
- smoke_test_linux_conda: - binary_wheel_upload:
context: org-member
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_smoke_test_conda name: nightly_binary_macos_wheel_py3.8_cpu_upload
python_version: '3.9'
requires: requires:
- nightly_binary_linux_conda_py3.9_upload - nightly_binary_macos_wheel_py3.8_cpu
- binary_macos_conda: - binary_macos_wheel:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_conda_py3.6 name: nightly_binary_macos_wheel_py3.9_cpu
python_version: '3.6' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_conda_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
branches: branches:
...@@ -1344,21 +1627,20 @@ workflows: ...@@ -1344,21 +1627,20 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_macos_conda_py3.6 - nightly_binary_macos_wheel_py3.9_cpu
- binary_macos_conda: - binary_windows_wheel:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_conda_py3.7 name: nightly_binary_windows_wheel_py3.6_cpu
python_version: '3.7' python_version: '3.6'
requires: - binary_wheel_upload:
- download_third_parties_nix
- binary_conda_upload:
context: org-member context: org-member
filters: filters:
branches: branches:
...@@ -1366,63 +1648,1000 @@ workflows: ...@@ -1366,63 +1648,1000 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_macos_conda_py3.7 - nightly_binary_windows_wheel_py3.6_cpu
- binary_macos_conda: - smoke_test_windows_pip:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_conda_py3.8 name: nightly_binary_windows_wheel_py3.6_cpu_smoke_test_pip
python_version: '3.8' python_version: '3.6'
requires: requires:
- download_third_parties_nix - nightly_binary_windows_wheel_py3.6_cpu_upload
- binary_conda_upload: - binary_windows_wheel:
context: org-member cuda_version: cu102
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_conda_py3.8_upload name: nightly_binary_windows_wheel_py3.6_cu102
requires: python_version: '3.6'
- nightly_binary_macos_conda_py3.8 wheel_docker_image: pytorch/manylinux-cuda102
- binary_macos_conda: - binary_wheel_upload:
context: org-member
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_conda_py3.9 name: nightly_binary_windows_wheel_py3.6_cu102_upload
python_version: '3.9'
requires: requires:
- download_third_parties_nix - nightly_binary_windows_wheel_py3.6_cu102
- binary_conda_upload: - smoke_test_windows_pip:
context: org-member cuda_version: cu102
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_macos_conda_py3.9 - nightly_binary_windows_wheel_py3.6_cu102_upload
- binary_windows_conda: - binary_windows_wheel:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' 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 context: org-member
filters: filters:
branches: branches:
...@@ -1430,28 +2649,99 @@ workflows: ...@@ -1430,28 +2649,99 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_windows_conda_py3.6 - nightly_binary_windows_conda_py3.6_cu111
- smoke_test_windows_conda: - smoke_test_windows_conda:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.6'
requires: 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: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.7'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -1461,28 +2751,99 @@ workflows: ...@@ -1461,28 +2751,99 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_windows_conda_py3.7 - nightly_binary_windows_conda_py3.7_cu111
- smoke_test_windows_conda: - smoke_test_windows_conda:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.7'
requires: 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: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.8'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -1492,28 +2853,99 @@ workflows: ...@@ -1492,28 +2853,99 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_windows_conda_py3.8 - nightly_binary_windows_conda_py3.8_cu111
- smoke_test_windows_conda: - smoke_test_windows_conda:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.8'
requires: 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: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.9'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -1523,20 +2955,21 @@ workflows: ...@@ -1523,20 +2955,21 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_windows_conda_py3.9 - nightly_binary_windows_conda_py3.9_cu111
- smoke_test_windows_conda: - smoke_test_windows_conda:
cuda_version: cu111
filters: filters:
branches: branches:
only: only:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.9'
requires: requires:
- nightly_binary_windows_conda_py3.9_upload - nightly_binary_windows_conda_py3.9_cu111_upload
docker_build: docker_build:
triggers: triggers:
- schedule: - schedule:
......
.circleci/config.yml.in
View file @ ffe735ba
...@@ -71,11 +71,23 @@ binary_common: &binary_common ...@@ -71,11 +71,23 @@ binary_common: &binary_common
python_version: python_version:
description: "Python version to build against (e.g., 3.8)" description: "Python version to build against (e.g., 3.8)"
type: string 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 environment: &environment
PYTHON_VERSION: << parameters.python_version >> PYTHON_VERSION: << parameters.python_version >>
BUILD_VERSION: << parameters.build_version >> BUILD_VERSION: << parameters.build_version >>
PYTORCH_VERSION: << parameters.pytorch_version >> PYTORCH_VERSION: << parameters.pytorch_version >>
CU_VERSION: cpu CU_VERSION: << parameters.cuda_version >>
smoke_test_common: &smoke_test_common smoke_test_common: &smoke_test_common
<<: *binary_common <<: *binary_common
...@@ -127,7 +139,7 @@ jobs: ...@@ -127,7 +139,7 @@ jobs:
binary_linux_wheel: binary_linux_wheel:
<<: *binary_common <<: *binary_common
docker: docker:
- image: "pytorch/manylinux-cuda102" - image: << parameters.wheel_docker_image >>
resource_class: 2xlarge+ resource_class: 2xlarge+
steps: steps:
- checkout - checkout
...@@ -144,7 +156,7 @@ jobs: ...@@ -144,7 +156,7 @@ jobs:
binary_linux_conda: binary_linux_conda:
<<: *binary_common <<: *binary_common
docker: docker:
- image: "pytorch/conda-cuda" - image: "<< parameters.conda_docker_image >>"
resource_class: 2xlarge+ resource_class: 2xlarge+
steps: steps:
- checkout - checkout
......
.circleci/regenerate.py
View file @ ffe735ba
...@@ -21,6 +21,10 @@ import os.path ...@@ -21,6 +21,10 @@ import os.path
PYTHON_VERSIONS = ["3.6", "3.7", "3.8", "3.9"] 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') DOC_VERSION = ('linux', '3.8')
...@@ -31,7 +35,8 @@ def build_workflows(prefix='', upload=False, filter_branch=None, indentation=6): ...@@ -31,7 +35,8 @@ def build_workflows(prefix='', upload=False, filter_branch=None, indentation=6):
for btype in ["wheel", "conda"]: for btype in ["wheel", "conda"]:
for os_type in ["linux", "macos", "windows"]: for os_type in ["linux", "macos", "windows"]:
for python_version in PYTHON_VERSIONS: 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: if not filter_branch:
# Build on every pull request, but upload only on nightly and tags # Build on every pull request, but upload only on nightly and tags
...@@ -53,27 +58,19 @@ def build_download_job(filter_branch): ...@@ -53,27 +58,19 @@ def build_download_job(filter_branch):
return [{"download_third_parties_nix": job}] 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 = [] w = []
base_workflow_name = "{prefix}binary_{os_type}_{btype}_py{python_version}".format( base_workflow_name = f"{prefix}binary_{os_type}_{btype}_py{python_version}_{cu_version}"
prefix=prefix, w.append(generate_base_workflow(base_workflow_name, python_version, cu_version, filter_branch, os_type, btype))
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))
if upload: 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)) 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' 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 return w
...@@ -82,7 +79,7 @@ def build_doc_job(filter_branch): ...@@ -82,7 +79,7 @@ def build_doc_job(filter_branch):
job = { job = {
"name": "build_docs", "name": "build_docs",
"python_version": "3.8", "python_version": "3.8",
"requires": ["binary_linux_wheel_py3.8", ], "requires": ["binary_linux_wheel_py3.8_cpu", ],
} }
if filter_branch: if filter_branch:
...@@ -107,7 +104,7 @@ def docstring_parameters_sync_job(filter_branch): ...@@ -107,7 +104,7 @@ def docstring_parameters_sync_job(filter_branch):
job = { job = {
"name": "docstring_parameters_sync", "name": "docstring_parameters_sync",
"python_version": "3.8", "python_version": "3.8",
"requires": ["binary_linux_wheel_py3.8", ], "requires": ["binary_linux_wheel_py3.8_cpu", ],
} }
if filter_branch: if filter_branch:
...@@ -115,15 +112,20 @@ def docstring_parameters_sync_job(filter_branch): ...@@ -115,15 +112,20 @@ def docstring_parameters_sync_job(filter_branch):
return [{"docstring_parameters_sync": job}] 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 = { d = {
"name": base_workflow_name, "name": base_workflow_name,
"python_version": python_version, "python_version": python_version,
"cuda_version": cu_version,
} }
if os_type in ['linux', 'macos']: if os_type in ['linux', 'macos']:
d['requires'] = ['download_third_parties_nix'] 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: if filter_branch:
d["filters"] = gen_filter_branch_tree(filter_branch) d["filters"] = gen_filter_branch_tree(filter_branch)
...@@ -157,23 +159,23 @@ def generate_upload_workflow(base_workflow_name, filter_branch, btype): ...@@ -157,23 +159,23 @@ def generate_upload_workflow(base_workflow_name, filter_branch, btype):
return {"binary_{btype}_upload".format(btype=btype): d} 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_suffix = "_upload"
required_build_name = base_workflow_name + required_build_suffix 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 = { d = {
"name": "{base_workflow_name}_{smoke_suffix}".format( "name": f"{base_workflow_name}_{smoke_suffix}",
base_workflow_name=base_workflow_name, smoke_suffix=smoke_suffix),
"requires": [required_build_name], "requires": [required_build_name],
"python_version": python_version, "python_version": python_version,
"cuda_version": cu_version,
} }
if filter_branch: if filter_branch:
d["filters"] = gen_filter_branch_tree(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): def indent(indentation, data_list):
...@@ -192,6 +194,7 @@ def unittest_workflows(indentation=6): ...@@ -192,6 +194,7 @@ def unittest_workflows(indentation=6):
job = { job = {
"name": f"unittest_{os_type}_{device_type}_py{python_version}", "name": f"unittest_{os_type}_{device_type}_py{python_version}",
"python_version": python_version, "python_version": python_version,
"cuda_version": 'cpu' if device_type=="cpu" else "cu102",
} }
if os_type != "windows": if os_type != "windows":
...@@ -204,6 +207,7 @@ def unittest_workflows(indentation=6): ...@@ -204,6 +207,7 @@ def unittest_workflows(indentation=6):
"stylecheck": { "stylecheck": {
"name": f"stylecheck_py{python_version}", "name": f"stylecheck_py{python_version}",
"python_version": python_version, "python_version": python_version,
"cuda_version": 'cpu' if device_type=="cpu" else "cu102",
} }
}) })
return indent(indentation, jobs) return indent(indentation, jobs)
......
packaging/pkg_helpers.bash
View file @ ffe735ba
...@@ -52,17 +52,60 @@ setup_cuda() { ...@@ -52,17 +52,60 @@ setup_cuda() {
# Now work out the CUDA settings # Now work out the CUDA settings
case "$CU_VERSION" in 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) cu100)
export CUDA_HOME=/usr/local/cuda-10.0/ export CUDA_HOME=/usr/local/cuda-10.0/
export FORCE_CUDA=1 export FORCE_CUDA=1
# Hard-coding gencode flags is temporary situation until export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5"
# 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"
;; ;;
cu92) cu92)
export CUDA_HOME=/usr/local/cuda-9.2/ export CUDA_HOME=/usr/local/cuda-9.2/
export FORCE_CUDA=1 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) 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