diff --git a/release/promote.sh b/release/promote.sh index 54e21ce7e..5bf7fe0b5 100644 --- a/release/promote.sh +++ b/release/promote.sh @@ -12,6 +12,7 @@ TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.1.1} TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.16.1} TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.7.1} TORCHREC_VERSION=${TORCHREC_VERSION:-0.6.0} +FBGEMMGPU_VERSION=${FBGEMMGPU_VERSION:-0.6.0} DRY_RUN=${DRY_RUN:-enabled} @@ -104,6 +105,7 @@ promote_pypi() { # 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}" # promote_conda torchtriton conda "2.1.0" diff --git a/release/release_versions.sh b/release/release_versions.sh index 53dbe435b..981a18ea0 100644 --- a/release/release_versions.sh +++ b/release/release_versions.sh @@ -6,4 +6,5 @@ TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.16.1} TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.1.1} TORCHTEXT_VERSION=${TORCHTEXT_VERSION:-0.16.1} TORCHDATA_VERSION=${TORCHDATA_VERSION:-0.7.1} -TORCHREC_VERSION=${TORCHREC_VERSION:-0.6.0} \ No newline at end of file +TORCHREC_VERSION=${TORCHREC_VERSION:-0.6.0} +FBGEMMGPU_VERSION=${FBGEMMGPU_VERSION:-0.6.0}