Unverified Commit 40ad5572 authored by Eli Uriegas's avatar Eli Uriegas Committed by GitHub
Browse files

ci: Automate uploading to correct upload channel (#2204)

parent 4e138bd5
...@@ -30,6 +30,18 @@ commands: ...@@ -30,6 +30,18 @@ commands:
# git fetch --force origin ${CIRCLE_BRANCH}/merge:merged/${CIRCLE_BRANCH} # git fetch --force origin ${CIRCLE_BRANCH}/merge:merged/${CIRCLE_BRANCH}
# git checkout "merged/$CIRCLE_BRANCH" # git checkout "merged/$CIRCLE_BRANCH"
# fi # fi
designate_upload_channel:
description: "inserts the correct upload channel into ${BASH_ENV}"
steps:
- run:
name: adding UPLOAD_CHANNEL to BASH_ENV
command: |
our_upload_channel=nightly
# On tags upload to test instead
if [[ -n "${CIRCLE_TAG}" ]]; then
our_upload_channel=test
fi
echo "export UPLOAD_CHANNEL=${our_upload_channel}" >> ${BASH_ENV}
binary_common: &binary_common binary_common: &binary_common
parameters: parameters:
...@@ -59,7 +71,6 @@ binary_common: &binary_common ...@@ -59,7 +71,6 @@ binary_common: &binary_common
default: "pytorch/manylinux-cuda101" default: "pytorch/manylinux-cuda101"
environment: environment:
PYTHON_VERSION: << parameters.python_version >> PYTHON_VERSION: << parameters.python_version >>
BUILD_VERSION: << parameters.build_version >>
PYTORCH_VERSION: << parameters.pytorch_version >> PYTORCH_VERSION: << parameters.pytorch_version >>
UNICODE_ABI: << parameters.unicode_abi >> UNICODE_ABI: << parameters.unicode_abi >>
CU_VERSION: << parameters.cu_version >> CU_VERSION: << parameters.cu_version >>
...@@ -358,12 +369,13 @@ jobs: ...@@ -358,12 +369,13 @@ jobs:
steps: steps:
- attach_workspace: - attach_workspace:
at: ~/workspace at: ~/workspace
- designate_upload_channel
- run: - run:
command: | command: |
# Prevent credential from leaking # Prevent credential from leaking
conda install -yq anaconda-client conda install -yq anaconda-client
set -x set -x
anaconda -t "${CONDA_PYTORCHBOT_TOKEN}" upload ~/workspace/*.tar.bz2 -u pytorch-nightly --label main --no-progress --force anaconda -t "${CONDA_PYTORCHBOT_TOKEN}" upload ~/workspace/*.tar.bz2 -u "pytorch-${UPLOAD_CHANNEL}" --label main --no-progress --force
# Requires org-member context # Requires org-member context
binary_wheel_upload: binary_wheel_upload:
...@@ -376,6 +388,7 @@ jobs: ...@@ -376,6 +388,7 @@ jobs:
steps: steps:
- attach_workspace: - attach_workspace:
at: ~/workspace at: ~/workspace
- designate_upload_channel
- checkout - checkout
- run: - run:
command: | command: |
...@@ -387,7 +400,7 @@ jobs: ...@@ -387,7 +400,7 @@ jobs:
export AWS_SECRET_ACCESS_KEY="${PYTORCH_BINARY_AWS_SECRET_ACCESS_KEY}" export AWS_SECRET_ACCESS_KEY="${PYTORCH_BINARY_AWS_SECRET_ACCESS_KEY}"
set -x set -x
for pkg in ~/workspace/*.whl; do for pkg in ~/workspace/*.whl; do
aws s3 cp "$pkg" "s3://pytorch/whl/nightly/<< parameters.subfolder >>" --acl public-read aws s3 cp "$pkg" "s3://pytorch/whl/${UPLOAD_CHANNEL}/<< parameters.subfolder >>" --acl public-read
done done
...@@ -736,6 +749,8 @@ workflows: ...@@ -736,6 +749,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_cpu_upload name: nightly_binary_linux_wheel_py3.6_cpu_upload
requires: requires:
- nightly_binary_linux_wheel_py3.6_cpu - nightly_binary_linux_wheel_py3.6_cpu
...@@ -753,6 +768,8 @@ workflows: ...@@ -753,6 +768,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_cu92_upload name: nightly_binary_linux_wheel_py3.6_cu92_upload
requires: requires:
- nightly_binary_linux_wheel_py3.6_cu92 - nightly_binary_linux_wheel_py3.6_cu92
...@@ -770,6 +787,8 @@ workflows: ...@@ -770,6 +787,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_cu101_upload name: nightly_binary_linux_wheel_py3.6_cu101_upload
requires: requires:
- nightly_binary_linux_wheel_py3.6_cu101 - nightly_binary_linux_wheel_py3.6_cu101
...@@ -787,6 +806,8 @@ workflows: ...@@ -787,6 +806,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_cu102_upload name: nightly_binary_linux_wheel_py3.6_cu102_upload
requires: requires:
- nightly_binary_linux_wheel_py3.6_cu102 - nightly_binary_linux_wheel_py3.6_cu102
...@@ -804,6 +825,8 @@ workflows: ...@@ -804,6 +825,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cpu_upload name: nightly_binary_linux_wheel_py3.7_cpu_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_cpu - nightly_binary_linux_wheel_py3.7_cpu
...@@ -821,6 +844,8 @@ workflows: ...@@ -821,6 +844,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu92_upload name: nightly_binary_linux_wheel_py3.7_cu92_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu92 - nightly_binary_linux_wheel_py3.7_cu92
...@@ -838,6 +863,8 @@ workflows: ...@@ -838,6 +863,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu101_upload name: nightly_binary_linux_wheel_py3.7_cu101_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu101 - nightly_binary_linux_wheel_py3.7_cu101
...@@ -855,6 +882,8 @@ workflows: ...@@ -855,6 +882,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu102_upload name: nightly_binary_linux_wheel_py3.7_cu102_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu102 - nightly_binary_linux_wheel_py3.7_cu102
...@@ -872,6 +901,8 @@ workflows: ...@@ -872,6 +901,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cpu_upload name: nightly_binary_linux_wheel_py3.8_cpu_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_cpu - nightly_binary_linux_wheel_py3.8_cpu
...@@ -889,6 +920,8 @@ workflows: ...@@ -889,6 +920,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu92_upload name: nightly_binary_linux_wheel_py3.8_cu92_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu92 - nightly_binary_linux_wheel_py3.8_cu92
...@@ -906,6 +939,8 @@ workflows: ...@@ -906,6 +939,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu101_upload name: nightly_binary_linux_wheel_py3.8_cu101_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu101 - nightly_binary_linux_wheel_py3.8_cu101
...@@ -923,6 +958,8 @@ workflows: ...@@ -923,6 +958,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu102_upload name: nightly_binary_linux_wheel_py3.8_cu102_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu102 - nightly_binary_linux_wheel_py3.8_cu102
...@@ -940,6 +977,8 @@ workflows: ...@@ -940,6 +977,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.6_cpu_upload name: nightly_binary_macos_wheel_py3.6_cpu_upload
requires: requires:
- nightly_binary_macos_wheel_py3.6_cpu - nightly_binary_macos_wheel_py3.6_cpu
...@@ -957,6 +996,8 @@ workflows: ...@@ -957,6 +996,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.7_cpu_upload name: nightly_binary_macos_wheel_py3.7_cpu_upload
requires: requires:
- nightly_binary_macos_wheel_py3.7_cpu - nightly_binary_macos_wheel_py3.7_cpu
...@@ -974,6 +1015,8 @@ workflows: ...@@ -974,6 +1015,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.8_cpu_upload name: nightly_binary_macos_wheel_py3.8_cpu_upload
requires: requires:
- nightly_binary_macos_wheel_py3.8_cpu - nightly_binary_macos_wheel_py3.8_cpu
...@@ -990,6 +1033,8 @@ workflows: ...@@ -990,6 +1033,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.6_cpu_upload name: nightly_binary_win_wheel_py3.6_cpu_upload
requires: requires:
- nightly_binary_win_wheel_py3.6_cpu - nightly_binary_win_wheel_py3.6_cpu
...@@ -1006,6 +1051,8 @@ workflows: ...@@ -1006,6 +1051,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.6_cu92_upload name: nightly_binary_win_wheel_py3.6_cu92_upload
requires: requires:
- nightly_binary_win_wheel_py3.6_cu92 - nightly_binary_win_wheel_py3.6_cu92
...@@ -1022,6 +1069,8 @@ workflows: ...@@ -1022,6 +1069,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.6_cu101_upload name: nightly_binary_win_wheel_py3.6_cu101_upload
requires: requires:
- nightly_binary_win_wheel_py3.6_cu101 - nightly_binary_win_wheel_py3.6_cu101
...@@ -1038,6 +1087,8 @@ workflows: ...@@ -1038,6 +1087,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.6_cu102_upload name: nightly_binary_win_wheel_py3.6_cu102_upload
requires: requires:
- nightly_binary_win_wheel_py3.6_cu102 - nightly_binary_win_wheel_py3.6_cu102
...@@ -1054,6 +1105,8 @@ workflows: ...@@ -1054,6 +1105,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.7_cpu_upload name: nightly_binary_win_wheel_py3.7_cpu_upload
requires: requires:
- nightly_binary_win_wheel_py3.7_cpu - nightly_binary_win_wheel_py3.7_cpu
...@@ -1070,6 +1123,8 @@ workflows: ...@@ -1070,6 +1123,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.7_cu92_upload name: nightly_binary_win_wheel_py3.7_cu92_upload
requires: requires:
- nightly_binary_win_wheel_py3.7_cu92 - nightly_binary_win_wheel_py3.7_cu92
...@@ -1086,6 +1141,8 @@ workflows: ...@@ -1086,6 +1141,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.7_cu101_upload name: nightly_binary_win_wheel_py3.7_cu101_upload
requires: requires:
- nightly_binary_win_wheel_py3.7_cu101 - nightly_binary_win_wheel_py3.7_cu101
...@@ -1102,6 +1159,8 @@ workflows: ...@@ -1102,6 +1159,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.7_cu102_upload name: nightly_binary_win_wheel_py3.7_cu102_upload
requires: requires:
- nightly_binary_win_wheel_py3.7_cu102 - nightly_binary_win_wheel_py3.7_cu102
...@@ -1118,6 +1177,8 @@ workflows: ...@@ -1118,6 +1177,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cpu_upload name: nightly_binary_win_wheel_py3.8_cpu_upload
requires: requires:
- nightly_binary_win_wheel_py3.8_cpu - nightly_binary_win_wheel_py3.8_cpu
...@@ -1134,6 +1195,8 @@ workflows: ...@@ -1134,6 +1195,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cu92_upload name: nightly_binary_win_wheel_py3.8_cu92_upload
requires: requires:
- nightly_binary_win_wheel_py3.8_cu92 - nightly_binary_win_wheel_py3.8_cu92
...@@ -1150,6 +1213,8 @@ workflows: ...@@ -1150,6 +1213,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cu101_upload name: nightly_binary_win_wheel_py3.8_cu101_upload
requires: requires:
- nightly_binary_win_wheel_py3.8_cu101 - nightly_binary_win_wheel_py3.8_cu101
...@@ -1166,6 +1231,8 @@ workflows: ...@@ -1166,6 +1231,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cu102_upload name: nightly_binary_win_wheel_py3.8_cu102_upload
requires: requires:
- nightly_binary_win_wheel_py3.8_cu102 - nightly_binary_win_wheel_py3.8_cu102
...@@ -1183,6 +1250,8 @@ workflows: ...@@ -1183,6 +1250,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6_cpu_upload name: nightly_binary_linux_conda_py3.6_cpu_upload
requires: requires:
- nightly_binary_linux_conda_py3.6_cpu - nightly_binary_linux_conda_py3.6_cpu
...@@ -1199,6 +1268,8 @@ workflows: ...@@ -1199,6 +1268,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6_cu92_upload name: nightly_binary_linux_conda_py3.6_cu92_upload
requires: requires:
- nightly_binary_linux_conda_py3.6_cu92 - nightly_binary_linux_conda_py3.6_cu92
...@@ -1215,6 +1286,8 @@ workflows: ...@@ -1215,6 +1286,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6_cu101_upload name: nightly_binary_linux_conda_py3.6_cu101_upload
requires: requires:
- nightly_binary_linux_conda_py3.6_cu101 - nightly_binary_linux_conda_py3.6_cu101
...@@ -1231,6 +1304,8 @@ workflows: ...@@ -1231,6 +1304,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6_cu102_upload name: nightly_binary_linux_conda_py3.6_cu102_upload
requires: requires:
- nightly_binary_linux_conda_py3.6_cu102 - nightly_binary_linux_conda_py3.6_cu102
...@@ -1247,6 +1322,8 @@ workflows: ...@@ -1247,6 +1322,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cpu_upload name: nightly_binary_linux_conda_py3.7_cpu_upload
requires: requires:
- nightly_binary_linux_conda_py3.7_cpu - nightly_binary_linux_conda_py3.7_cpu
...@@ -1263,6 +1340,8 @@ workflows: ...@@ -1263,6 +1340,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu92_upload name: nightly_binary_linux_conda_py3.7_cu92_upload
requires: requires:
- nightly_binary_linux_conda_py3.7_cu92 - nightly_binary_linux_conda_py3.7_cu92
...@@ -1279,6 +1358,8 @@ workflows: ...@@ -1279,6 +1358,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu101_upload name: nightly_binary_linux_conda_py3.7_cu101_upload
requires: requires:
- nightly_binary_linux_conda_py3.7_cu101 - nightly_binary_linux_conda_py3.7_cu101
...@@ -1295,6 +1376,8 @@ workflows: ...@@ -1295,6 +1376,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu102_upload name: nightly_binary_linux_conda_py3.7_cu102_upload
requires: requires:
- nightly_binary_linux_conda_py3.7_cu102 - nightly_binary_linux_conda_py3.7_cu102
...@@ -1311,6 +1394,8 @@ workflows: ...@@ -1311,6 +1394,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cpu_upload name: nightly_binary_linux_conda_py3.8_cpu_upload
requires: requires:
- nightly_binary_linux_conda_py3.8_cpu - nightly_binary_linux_conda_py3.8_cpu
...@@ -1327,6 +1412,8 @@ workflows: ...@@ -1327,6 +1412,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu92_upload name: nightly_binary_linux_conda_py3.8_cu92_upload
requires: requires:
- nightly_binary_linux_conda_py3.8_cu92 - nightly_binary_linux_conda_py3.8_cu92
...@@ -1343,6 +1430,8 @@ workflows: ...@@ -1343,6 +1430,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu101_upload name: nightly_binary_linux_conda_py3.8_cu101_upload
requires: requires:
- nightly_binary_linux_conda_py3.8_cu101 - nightly_binary_linux_conda_py3.8_cu101
...@@ -1359,6 +1448,8 @@ workflows: ...@@ -1359,6 +1448,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu102_upload name: nightly_binary_linux_conda_py3.8_cu102_upload
requires: requires:
- nightly_binary_linux_conda_py3.8_cu102 - nightly_binary_linux_conda_py3.8_cu102
...@@ -1375,6 +1466,8 @@ workflows: ...@@ -1375,6 +1466,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_conda_py3.6_cpu_upload name: nightly_binary_macos_conda_py3.6_cpu_upload
requires: requires:
- nightly_binary_macos_conda_py3.6_cpu - nightly_binary_macos_conda_py3.6_cpu
...@@ -1391,6 +1484,8 @@ workflows: ...@@ -1391,6 +1484,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_conda_py3.7_cpu_upload name: nightly_binary_macos_conda_py3.7_cpu_upload
requires: requires:
- nightly_binary_macos_conda_py3.7_cpu - nightly_binary_macos_conda_py3.7_cpu
...@@ -1407,6 +1502,8 @@ workflows: ...@@ -1407,6 +1502,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_conda_py3.8_cpu_upload name: nightly_binary_macos_conda_py3.8_cpu_upload
requires: requires:
- nightly_binary_macos_conda_py3.8_cpu - nightly_binary_macos_conda_py3.8_cpu
...@@ -1422,6 +1519,8 @@ workflows: ...@@ -1422,6 +1519,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.6_cpu_upload name: nightly_binary_win_conda_py3.6_cpu_upload
requires: requires:
- nightly_binary_win_conda_py3.6_cpu - nightly_binary_win_conda_py3.6_cpu
...@@ -1437,6 +1536,8 @@ workflows: ...@@ -1437,6 +1536,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.6_cu92_upload name: nightly_binary_win_conda_py3.6_cu92_upload
requires: requires:
- nightly_binary_win_conda_py3.6_cu92 - nightly_binary_win_conda_py3.6_cu92
...@@ -1452,6 +1553,8 @@ workflows: ...@@ -1452,6 +1553,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.6_cu101_upload name: nightly_binary_win_conda_py3.6_cu101_upload
requires: requires:
- nightly_binary_win_conda_py3.6_cu101 - nightly_binary_win_conda_py3.6_cu101
...@@ -1467,6 +1570,8 @@ workflows: ...@@ -1467,6 +1570,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.6_cu102_upload name: nightly_binary_win_conda_py3.6_cu102_upload
requires: requires:
- nightly_binary_win_conda_py3.6_cu102 - nightly_binary_win_conda_py3.6_cu102
...@@ -1482,6 +1587,8 @@ workflows: ...@@ -1482,6 +1587,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.7_cpu_upload name: nightly_binary_win_conda_py3.7_cpu_upload
requires: requires:
- nightly_binary_win_conda_py3.7_cpu - nightly_binary_win_conda_py3.7_cpu
...@@ -1497,6 +1604,8 @@ workflows: ...@@ -1497,6 +1604,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.7_cu92_upload name: nightly_binary_win_conda_py3.7_cu92_upload
requires: requires:
- nightly_binary_win_conda_py3.7_cu92 - nightly_binary_win_conda_py3.7_cu92
...@@ -1512,6 +1621,8 @@ workflows: ...@@ -1512,6 +1621,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.7_cu101_upload name: nightly_binary_win_conda_py3.7_cu101_upload
requires: requires:
- nightly_binary_win_conda_py3.7_cu101 - nightly_binary_win_conda_py3.7_cu101
...@@ -1527,6 +1638,8 @@ workflows: ...@@ -1527,6 +1638,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.7_cu102_upload name: nightly_binary_win_conda_py3.7_cu102_upload
requires: requires:
- nightly_binary_win_conda_py3.7_cu102 - nightly_binary_win_conda_py3.7_cu102
...@@ -1542,6 +1655,8 @@ workflows: ...@@ -1542,6 +1655,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cpu_upload name: nightly_binary_win_conda_py3.8_cpu_upload
requires: requires:
- nightly_binary_win_conda_py3.8_cpu - nightly_binary_win_conda_py3.8_cpu
...@@ -1557,6 +1672,8 @@ workflows: ...@@ -1557,6 +1672,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cu92_upload name: nightly_binary_win_conda_py3.8_cu92_upload
requires: requires:
- nightly_binary_win_conda_py3.8_cu92 - nightly_binary_win_conda_py3.8_cu92
...@@ -1572,6 +1689,8 @@ workflows: ...@@ -1572,6 +1689,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cu101_upload name: nightly_binary_win_conda_py3.8_cu101_upload
requires: requires:
- nightly_binary_win_conda_py3.8_cu101 - nightly_binary_win_conda_py3.8_cu101
...@@ -1587,6 +1706,8 @@ workflows: ...@@ -1587,6 +1706,8 @@ workflows:
filters: filters:
branches: branches:
only: nightly only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cu102_upload name: nightly_binary_win_conda_py3.8_cu102_upload
requires: requires:
- nightly_binary_win_conda_py3.8_cu102 - nightly_binary_win_conda_py3.8_cu102
\ No newline at end of file
...@@ -30,6 +30,18 @@ commands: ...@@ -30,6 +30,18 @@ commands:
# git fetch --force origin ${CIRCLE_BRANCH}/merge:merged/${CIRCLE_BRANCH} # git fetch --force origin ${CIRCLE_BRANCH}/merge:merged/${CIRCLE_BRANCH}
# git checkout "merged/$CIRCLE_BRANCH" # git checkout "merged/$CIRCLE_BRANCH"
# fi # fi
designate_upload_channel:
description: "inserts the correct upload channel into ${BASH_ENV}"
steps:
- run:
name: adding UPLOAD_CHANNEL to BASH_ENV
command: |
our_upload_channel=nightly
# On tags upload to test instead
if [[ -n "${CIRCLE_TAG}" ]]; then
our_upload_channel=test
fi
echo "export UPLOAD_CHANNEL=${our_upload_channel}" >> ${BASH_ENV}
binary_common: &binary_common binary_common: &binary_common
parameters: parameters:
...@@ -59,7 +71,6 @@ binary_common: &binary_common ...@@ -59,7 +71,6 @@ binary_common: &binary_common
default: "pytorch/manylinux-cuda101" default: "pytorch/manylinux-cuda101"
environment: environment:
PYTHON_VERSION: << parameters.python_version >> PYTHON_VERSION: << parameters.python_version >>
BUILD_VERSION: << parameters.build_version >>
PYTORCH_VERSION: << parameters.pytorch_version >> PYTORCH_VERSION: << parameters.pytorch_version >>
UNICODE_ABI: << parameters.unicode_abi >> UNICODE_ABI: << parameters.unicode_abi >>
CU_VERSION: << parameters.cu_version >> CU_VERSION: << parameters.cu_version >>
...@@ -358,12 +369,13 @@ jobs: ...@@ -358,12 +369,13 @@ jobs:
steps: steps:
- attach_workspace: - attach_workspace:
at: ~/workspace at: ~/workspace
- designate_upload_channel
- run: - run:
command: | command: |
# Prevent credential from leaking # Prevent credential from leaking
conda install -yq anaconda-client conda install -yq anaconda-client
set -x set -x
anaconda -t "${CONDA_PYTORCHBOT_TOKEN}" upload ~/workspace/*.tar.bz2 -u pytorch-nightly --label main --no-progress --force anaconda -t "${CONDA_PYTORCHBOT_TOKEN}" upload ~/workspace/*.tar.bz2 -u "pytorch-${UPLOAD_CHANNEL}" --label main --no-progress --force
# Requires org-member context # Requires org-member context
binary_wheel_upload: binary_wheel_upload:
...@@ -376,6 +388,7 @@ jobs: ...@@ -376,6 +388,7 @@ jobs:
steps: steps:
- attach_workspace: - attach_workspace:
at: ~/workspace at: ~/workspace
- designate_upload_channel
- checkout - checkout
- run: - run:
command: | command: |
...@@ -387,7 +400,7 @@ jobs: ...@@ -387,7 +400,7 @@ jobs:
export AWS_SECRET_ACCESS_KEY="${PYTORCH_BINARY_AWS_SECRET_ACCESS_KEY}" export AWS_SECRET_ACCESS_KEY="${PYTORCH_BINARY_AWS_SECRET_ACCESS_KEY}"
set -x set -x
for pkg in ~/workspace/*.whl; do for pkg in ~/workspace/*.whl; do
aws s3 cp "$pkg" "s3://pytorch/whl/nightly/<< parameters.subfolder >>" --acl public-read aws s3 cp "$pkg" "s3://pytorch/whl/${UPLOAD_CHANNEL}/<< parameters.subfolder >>" --acl public-read
done done
......
...@@ -103,7 +103,16 @@ def generate_upload_workflow(base_workflow_name, os_type, btype, cu_version, *, ...@@ -103,7 +103,16 @@ def generate_upload_workflow(base_workflow_name, os_type, btype, cu_version, *,
d["subfolder"] = "" if os_type == 'macos' else cu_version + "/" d["subfolder"] = "" if os_type == 'macos' else cu_version + "/"
if filter_branch is not None: if filter_branch is not None:
d["filters"] = {"branches": {"only": filter_branch}} d["filters"] = {
"branches": {
"only": filter_branch
},
"tags": {
# Using a raw string here to avoid having to escape
# anything
"only": r"/v[0-9]+(\.[0-9]+)*-rc[0-9]+/"
}
}
return {f"binary_{btype}_upload": d} return {f"binary_{btype}_upload": d}
......
...@@ -117,6 +117,12 @@ setup_build_version() { ...@@ -117,6 +117,12 @@ setup_build_version() {
else else
export BUILD_VERSION="$BUILD_VERSION$VERSION_SUFFIX" export BUILD_VERSION="$BUILD_VERSION$VERSION_SUFFIX"
fi fi
# Set build version based on tag if on tag
if [[ -n "${CIRCLE_TAG}" ]]; then
# Strip tag
export BUILD_VERSION="$(echo "${CIRCLE_TAG}" | sed -e 's/^v//' -e 's/-.*$//')"
fi
} }
# Set some useful variables for OS X, if applicable # Set some useful variables for OS X, if applicable
......
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