/
promote.sh
101 lines (89 loc) · 3.72 KB
/
promote.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
#!/usr/bin/env bash
set -eou pipefail
# Make sure to update these versions when doing a release first
PYTORCH_VERSION=${PYTORCH_VERSION:-1.8.0}
TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.9.0}
TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-0.8.0}
TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.9.0}
TORCHSERVE_VERSION=${TORCHSERVE_VERSION:-0.2.1}
TORCHCSPRNG_VERSION=${TORCHCSPRNG_VERSION:-0.2.0}
PYTORCH_DIR=${PYTORCH_DIR:-~/pytorch}
DRY_RUN=${DRY_RUN:-enabled}
promote_s3() {
local package_name
package_name=$1
local package_type
package_type=$2
local promote_version
promote_version=$3
echo "=-=-=-= Promoting ${package_name}'s v${promote_version} ${package_type} packages' =-=-=-="
(
set -x
TEST_PYTORCH_PROMOTE_VERSION="${promote_version}" \
PACKAGE_NAME="${package_name}" \
PACKAGE_TYPE="${package_type}" \
TEST_WITHOUT_GIT_TAG=1 \
DRY_RUN="${DRY_RUN}" ${PYTORCH_DIR}/scripts/release/promote/s3_to_s3.sh
)
echo
}
promote_conda() {
local package_name
package_name=$1
local package_type
package_type=$2
local promote_version
promote_version=$3
echo "=-=-=-= Promoting ${package_name}'s v${promote_version} ${package_type} packages' =-=-=-="
(
set -x
TEST_PYTORCH_PROMOTE_VERSION="${promote_version}" \
PACKAGE_NAME="${package_name}" \
PACKAGE_TYPE="${package_type}" \
TEST_WITHOUT_GIT_TAG=1 \
DRY_RUN="${DRY_RUN}" ${PYTORCH_DIR}/scripts/release/promote/conda_to_conda.sh
)
echo
}
promote_pypi() {
local package_name
package_name=$1
local promote_version
promote_version=$2
echo "=-=-=-= Promoting ${package_name}'s v${promote_version} to pypi' =-=-=-="
(
set -x
TEST_PYTORCH_PROMOTE_VERSION="${promote_version}" \
PACKAGE_NAME="${package_name}" \
TEST_WITHOUT_GIT_TAG=1 \
DRY_RUN="${DRY_RUN}" ${PYTORCH_DIR}/scripts/release/promote/wheel_to_pypi.sh
)
echo
}
# promote_s3 torch whl "${PYTORCH_VERSION}"
# promote_s3 torchvision whl "${TORCHVISION_VERSION}"
# promote_s3 torchaudio whl "${TORCHAUDIO_VERSION}"
# promote_s3 torchtext whl "${TORCHTEXT_VERSION}"
# promote_s3 torchserve whl "${TORCHSERVE_VERSION}"
# promote_s3 torch_model_archiver whl "${TORCHSERVE_VERSION}"
# promote_s3 torchcsprng whl "${TORCHCSPRNG_VERSION}"
# promote_s3 "libtorch-*" libtorch "${PYTORCH_VERSION}"
# promote_conda pytorch conda "${PYTORCH_VERSION}"
# promote_conda torchvision conda "${TORCHVISION_VERSION}"
# promote_conda torchaudio conda "${TORCHAUDIO_VERSION}"
# promote_conda torchtext conda "${TORCHTEXT_VERSION}"
# promote_conda torchserve conda "${TORCHSERVE_VERSION}"
# promote_conda torch-model-archiver conda "${TORCHSERVE_VERSION}"
# promote_conda torchcsprng conda "${TORCHCSPRNG_VERSION}"
# Uncomment these to promote to pypi
# LINUX_VERSION_SUFFIX="%2Bcu102"
# WIN_VERSION_SUFFIX="%2Bcpu"
# PLATFORM="linux_x86_64" VERSION_SUFFIX="${LINUX_VERSION_SUFFIX}" promote_pypi torch "${PYTORCH_VERSION}"
# PLATFORM="win_amd64" VERSION_SUFFIX="${WIN_VERSION_SUFFIX}" promote_pypi torch "${PYTORCH_VERSION}"
# PLATFORM="macosx_10_9" VERSION_SUFFIX="" promote_pypi torch "${PYTORCH_VERSION}"
# PLATFORM="linux_x86_64" VERSION_SUFFIX="${LINUX_VERSION_SUFFIX}" promote_pypi torchvision "${TORCHVISION_VERSION}"
# PLATFORM="win_amd64" VERSION_SUFFIX="${WIN_VERSION_SUFFIX}" promote_pypi torchvision "${TORCHVISION_VERSION}"
# PLATFORM="macosx_10_9" VERSION_SUFFIX="" promote_pypi torchvision "${TORCHVISION_VERSION}"
# promote_pypi torchaudio "${TORCHAUDIO_VERSION}"
# promote_pypi torchtext "${TORCHTEXT_VERSION}"
# promote_pypi torchcsprng "${TORCHCSPRNG_VERSION}"