Skip to content

Commit

Permalink
Release scripts for 2.3 release (pytorch#1793)
Browse files Browse the repository at this point in the history
  • Loading branch information
atalman authored Apr 22, 2024
1 parent 94d6e5c commit 2a8481b
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 31 deletions.
9 changes: 4 additions & 5 deletions analytics/validate_pypi_staging.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,16 +13,15 @@
"manylinux1_x86_64",
"manylinux2014_aarch64",
"win_amd64",
"macosx_10_9_x86_64",
"macosx_11_0_arm64",
]
PYTHON_VERSIONS = ["cp38", "cp39", "cp310", "cp311", "cp312"]
S3_PYPI_STAGING = "pytorch-backup"
PACKAGE_RELEASES = {
"torch": "2.2.0",
"torchvision": "0.17.0",
"torchaudio": "2.2.0",
"torchtext": "0.17.0",
"torch": "2.3.0",
"torchvision": "0.18.0",
"torchaudio": "2.3.0",
"torchtext": "0.18.0",
}

PATTERN_V = "Version:"
Expand Down
14 changes: 4 additions & 10 deletions release/promote.sh
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,10 @@ DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
source "${DIR}/release_versions.sh"

# Make sure to update these versions when doing a release first
PYTORCH_VERSION=${PYTORCH_VERSION:-2.2.1}
TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.17.1}
TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.2.1}
TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.17.1}
TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.7.1}
PYTORCH_VERSION=${PYTORCH_VERSION:-2.3.0}
TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.18.0}
TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.3.0}
TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.18.0}
TORCHREC_VERSION=${TORCHREC_VERSION:-0.6.0}
TENSORRT_VERSION=${TENSORRT_VERSION:-2.2.0}

Expand Down Expand Up @@ -110,7 +109,6 @@ promote_pypi() {
# promote_s3 torchvision whl "${TORCHVISION_VERSION}"
# promote_s3 torchaudio whl "${TORCHAUDIO_VERSION}"
# promote_s3 torchtext whl "${TORCHTEXT_VERSION}"
# promote_s3 torchdata whl "${TORCHDATA_VERSION}"
# promote_s3 torchrec whl "${TORCHREC_VERSION}"
# promote_s3 fbgemm-gpu whl "${FBGEMMGPU_VERSION}"
# promote_s3 "libtorch-*" libtorch "${PYTORCH_VERSION}"
Expand All @@ -124,25 +122,21 @@ promote_pypi() {
# promote_conda torchvision conda "${TORCHVISION_VERSION}"
# promote_conda torchaudio conda "${TORCHAUDIO_VERSION}"
# promote_conda torchtext conda "${TORCHTEXT_VERSION}"
# promote_conda torchdata conda "${TORCHDATA_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="manylinux2014_aarch64" 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}" # intel mac
# PLATFORM="macosx_11_0" VERSION_SUFFIX="" promote_pypi torch "${PYTORCH_VERSION}" # m1 mac

# PLATFORM="linux_x86_64" VERSION_SUFFIX="${LINUX_VERSION_SUFFIX}" promote_pypi torchvision "${TORCHVISION_VERSION}"
# PLATFORM="manylinux2014_aarch64" 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}"
# PLATFORM="macosx_11_0" VERSION_SUFFIX="" promote_pypi torchvision "${TORCHVISION_VERSION}"

# PLATFORM="linux_x86_64" VERSION_SUFFIX="${LINUX_VERSION_SUFFIX}" promote_pypi torchaudio "${TORCHAUDIO_VERSION}"
# PLATFORM="manylinux2014_aarch64" VERSION_SUFFIX="" promote_pypi torchaudio "${TORCHAUDIO_VERSION}"
# PLATFORM="win_amd64" VERSION_SUFFIX="${WIN_VERSION_SUFFIX}" promote_pypi torchaudio "${TORCHAUDIO_VERSION}"
# PLATFORM="macosx_10_15" VERSION_SUFFIX="" promote_pypi torchaudio "${TORCHAUDIO_VERSION}"
# PLATFORM="macosx_11_0" VERSION_SUFFIX="" promote_pypi torchaudio "${TORCHAUDIO_VERSION}"
12 changes: 1 addition & 11 deletions release/pypi/promote_pypi_to_staging.sh
Original file line number Diff line number Diff line change
Expand Up @@ -29,28 +29,18 @@ MACOS_ARM64="macosx_.*_arm64"
PLATFORM="linux_x86_64" VERSION_SUFFIX="${LINUX_VERSION_SUFFIX}" upload_pypi_to_staging torch "${PYTORCH_VERSION}"
PLATFORM="manylinux2014_aarch64" VERSION_SUFFIX="" upload_pypi_to_staging torch "${PYTORCH_VERSION}"
PLATFORM="win_amd64" VERSION_SUFFIX="${CPU_VERSION_SUFFIX}" upload_pypi_to_staging torch "${PYTORCH_VERSION}"
PLATFORM="${MACOS_X86_64}" VERSION_SUFFIX="" upload_pypi_to_staging torch "${PYTORCH_VERSION}" # intel mac
PLATFORM="${MACOS_ARM64}" VERSION_SUFFIX="" upload_pypi_to_staging torch "${PYTORCH_VERSION}" # m1 mac
PLATFORM="${MACOS_ARM64}" VERSION_SUFFIX="" upload_pypi_to_staging torch "${PYTORCH_VERSION}"

PLATFORM="linux_x86_64" VERSION_SUFFIX="${LINUX_VERSION_SUFFIX}" upload_pypi_to_staging torchvision "${TORCHVISION_VERSION}"
PLATFORM="linux_aarch64" VERSION_SUFFIX="" upload_pypi_to_staging torchvision "${TORCHVISION_VERSION}"
PLATFORM="win_amd64" VERSION_SUFFIX="${CPU_VERSION_SUFFIX}" upload_pypi_to_staging torchvision "${TORCHVISION_VERSION}"
PLATFORM="${MACOS_X86_64}" VERSION_SUFFIX="" upload_pypi_to_staging torchvision "${TORCHVISION_VERSION}"
PLATFORM="${MACOS_ARM64}" VERSION_SUFFIX="" upload_pypi_to_staging torchvision "${TORCHVISION_VERSION}"

PLATFORM="linux_x86_64" VERSION_SUFFIX="${LINUX_VERSION_SUFFIX}" upload_pypi_to_staging torchaudio "${TORCHAUDIO_VERSION}"
PLATFORM="linux_aarch64" VERSION_SUFFIX="" upload_pypi_to_staging torchaudio "${TORCHAUDIO_VERSION}"
PLATFORM="win_amd64" VERSION_SUFFIX="${CPU_VERSION_SUFFIX}" upload_pypi_to_staging torchaudio "${TORCHAUDIO_VERSION}"
PLATFORM="${MACOS_X86_64}" VERSION_SUFFIX="" upload_pypi_to_staging torchaudio "${TORCHAUDIO_VERSION}"
PLATFORM="${MACOS_ARM64}" VERSION_SUFFIX="" upload_pypi_to_staging torchaudio "${TORCHAUDIO_VERSION}"

PLATFORM="linux_x86" VERSION_SUFFIX="${CPU_VERSION_SUFFIX}" upload_pypi_to_staging torchtext "${TORCHTEXT_VERSION}"
PLATFORM="win_amd64" VERSION_SUFFIX="${CPU_VERSION_SUFFIX}" upload_pypi_to_staging torchtext "${TORCHTEXT_VERSION}"
PLATFORM="${MACOS_X86_64}" VERSION_SUFFIX="" upload_pypi_to_staging torchtext "${TORCHTEXT_VERSION}"
PLATFORM="${MACOS_ARM64}" VERSION_SUFFIX="" upload_pypi_to_staging torchtext "${TORCHTEXT_VERSION}"

# Please note torchdata is not released currently hence turning it off
#PLATFORM="manylinux2014_x86_64" VERSION_SUFFIX="" upload_pypi_to_staging torchdata "${TORCHDATA_VERSION}"
#PLATFORM="win_amd64" VERSION_SUFFIX="" upload_pypi_to_staging torchdata "${TORCHDATA_VERSION}"
#PLATFORM="${MACOS_X86_64}" VERSION_SUFFIX="" upload_pypi_to_staging torchdata "${TORCHDATA_VERSION}"
#PLATFORM="${MACOS_ARM64}" VERSION_SUFFIX="" upload_pypi_to_staging torchdata "${TORCHDATA_VERSION}"
9 changes: 4 additions & 5 deletions release/release_versions.sh
Original file line number Diff line number Diff line change
@@ -1,11 +1,10 @@
#!/usr/bin/env bash

# Make sure to update these versions when doing a release first
PYTORCH_VERSION=${PYTORCH_VERSION:-2.2.1}
TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.17.1}
TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.2.1}
TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.17.1}
TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.7.1}
PYTORCH_VERSION=${PYTORCH_VERSION:-2.3.0}
TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.18.0}
TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.3.0}
TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.18.0}
TORCHREC_VERSION=${TORCHREC_VERSION:-0.6.0}
TENSORRT_VERSION=${TENSORRT_VERSION:-2.2.0}

Expand Down

0 comments on commit 2a8481b

Please sign in to comment.