From 91e445ff7071d01af45707f3462b945d13716f12 Mon Sep 17 00:00:00 2001 From: Supadchaya Puangpontip Date: Mon, 22 Jan 2024 16:59:15 -0800 Subject: [PATCH] Add fbgemm to promote s3 script --- release/promote.sh | 2 ++ release/release_versions.sh | 1 + 2 files changed, 3 insertions(+) diff --git a/release/promote.sh b/release/promote.sh index a7f273bc1..bc32916e9 100644 --- a/release/promote.sh +++ b/release/promote.sh @@ -11,6 +11,7 @@ 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} +FBGEMMGPU_VERSION=${FBGEMMGPU_VERSION:-0.6.0} DRY_RUN=${DRY_RUN:-enabled} @@ -102,6 +103,7 @@ promote_pypi() { # promote_s3 torchaudio whl "${TORCHAUDIO_VERSION}" # promote_s3 torchtext whl "${TORCHTEXT_VERSION}" # promote_s3 torchdata whl "${TORCHDATA_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 d362cb1ca..d53ede7a7 100644 --- a/release/release_versions.sh +++ b/release/release_versions.sh @@ -6,3 +6,4 @@ 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} +FBGEMMGPU_VERSION=${FBGEMMGPU_VERSION:-0.6.0}