Release 2.2.0 prep script modifications

release/pypi/prep_binary_for_pypi.sh

@@ -52,14 +52,6 @@ for whl_file in "$@"; do
     (
         set -x
 
-        # Special build with pypi cudnn remove it from version
-        if [[ $whl_file == *"with.pypi.cudnn"* ]]; then
-            rm -rf "${whl_dir}/caffe2"
-            rm -rf "${whl_dir}"/torch/lib/libnvrtc*
-
-            sed -i -e "s/-with-pypi-cudnn//g" "${whl_dir}/torch/version.py"
-        fi
-
         find "${dist_info_folder}" -type f -exec sed -i "s!${version_with_suffix}!${version_no_suffix}!" {} \;
         # Moves distinfo from one with a version suffix to one without
         # Example: torch-1.8.0+cpu.dist-info => torch-1.8.0.dist-info

release/pypi/promote_pypi_to_staging.sh

@@ -21,13 +21,12 @@ upload_pypi_to_staging() {
 }
 
 # Uncomment these to promote to pypi
-PYTORCH_LINUX_VERSION_SUFFIX="%2Bcu121.with.pypi.cudnn"
 LINUX_VERSION_SUFFIX="%2Bcu121"
 CPU_VERSION_SUFFIX="%2Bcpu"
 MACOS_X86_64="macosx_.*_x86_64"
 MACOS_ARM64="macosx_.*_arm64"
 
-PLATFORM="linux_x86_64"          VERSION_SUFFIX="${PYTORCH_LINUX_VERSION_SUFFIX}" upload_pypi_to_staging torch "${PYTORCH_VERSION}"
+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

release/pypi/upload_pypi_to_staging.sh

@@ -5,6 +5,9 @@ set -eou pipefail
 DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
 # Allow for users to pass PACKAGE_NAME
 
+# Set TMPDIR
+#export TMPDIR="/home/ec2-user/github/builder/release/pypi/work"
+
 # For use with other packages, i.e. torchvision, etc.
 PACKAGE_NAME=${PACKAGE_NAME:-torch}
 

release/release_versions.sh

@@ -1,10 +1,10 @@
 #!/usr/bin/env bash
 
 # Make sure to update these versions when doing a release first
-PYTORCH_VERSION=${PYTORCH_VERSION:-2.1.1}
-TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.16.1}
-TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.1.1}
-TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.16.1}
+PYTORCH_VERSION=${PYTORCH_VERSION:-2.2.0}
+TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.17.0}
+TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.2.0}
+TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.17.0}
 TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.7.1}
 TORCHREC_VERSION=${TORCHREC_VERSION:-0.6.0}
 FBGEMMGPU_VERSION=${FBGEMMGPU_VERSION:-0.6.0}