From eef9ffe284ac747c73aa325c7554f434ac2f8f69 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Tue, 3 Oct 2023 10:37:51 -0700
Subject: [PATCH] Release 2.1 update promotion scripts

---
 release/promote.sh                      | 14 +++++++-------
 release/pypi/promote_pypi_to_staging.sh |  4 ++--
 release/release_versions.sh             | 10 +++++-----
 3 files changed, 14 insertions(+), 14 deletions(-)

diff --git a/release/promote.sh b/release/promote.sh
index 1147dc0c9..b3656dda6 100644
--- a/release/promote.sh
+++ b/release/promote.sh
@@ -6,11 +6,11 @@ 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.0.0}
-TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.15.0}
-TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.0.0}
-TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.15.0}
-TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.6.0}
+PYTORCH_VERSION=${PYTORCH_VERSION:-2.1.0}
+TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.16.0}
+TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.1.0}
+TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.16.0}
+TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.7.0}
 
 DRY_RUN=${DRY_RUN:-enabled}
 
@@ -104,9 +104,9 @@ promote_pypi() {
 # promote_s3 torchdata whl "${TORCHDATA_VERSION}"
 # promote_s3 "libtorch-*" libtorch "${PYTORCH_VERSION}"
 
-# promote_conda torchtriton conda "2.0.0"
-# promote_conda pytorch-cuda conda "11.7"
+# promote_conda torchtriton conda "2.1.0"
 # promote_conda pytorch-cuda conda "11.8"
+# promote_conda pytorch-cuda conda "12.1"
 
 # promote_conda pytorch conda "${PYTORCH_VERSION}"
 # promote_conda torchvision conda "${TORCHVISION_VERSION}"
diff --git a/release/pypi/promote_pypi_to_staging.sh b/release/pypi/promote_pypi_to_staging.sh
index 74f139680..678d9dd03 100644
--- a/release/pypi/promote_pypi_to_staging.sh
+++ b/release/pypi/promote_pypi_to_staging.sh
@@ -21,8 +21,8 @@ upload_pypi_to_staging() {
 }
 
 # Uncomment these to promote to pypi
-PYTORCH_LINUX_VERSION_SUFFIX="%2Bcu117.with.pypi.cudnn"
-LINUX_VERSION_SUFFIX="%2Bcu117"
+PYTORCH_LINUX_VERSION_SUFFIX="%2Bcu121.with.pypi.cudnn"
+LINUX_VERSION_SUFFIX="%2Bcu121"
 WIN_VERSION_SUFFIX="%2Bcpu"
 MACOS_X86_64="macosx_.*_x86_64"
 MACOS_ARM64="macosx_.*_arm64"
diff --git a/release/release_versions.sh b/release/release_versions.sh
index f0db2a089..ab35075b6 100644
--- a/release/release_versions.sh
+++ b/release/release_versions.sh
@@ -1,8 +1,8 @@
 #!/usr/bin/env bash
 
 # Make sure to update these versions when doing a release first
-PYTORCH_VERSION=${PYTORCH_VERSION:-2.0.0}
-TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.15.0}
-TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.0.0}
-TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.15.0}
-TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.6.0}
+PYTORCH_VERSION=${PYTORCH_VERSION:-2.1.0}
+TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.16.0}
+TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.1.0}
+TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.16.0}
+TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.7.0}