File tree Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -12,6 +12,7 @@ TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:-2.1.1}
12
12
TORCHTEXT_VERSION=${TORCHTEXT_VERSION:- 0.16.1}
13
13
TORCHDATA_VERSION=${TORCHDATA_VERSION:- 0.7.1}
14
14
TORCHREC_VERSION=${TORCHREC_VERSION:- 0.6.0}
15
+ FBGEMMGPU_VERSION=${FBGEMMGPU_VERSION:- 0.6.0}
15
16
16
17
DRY_RUN=${DRY_RUN:- enabled}
17
18
@@ -104,6 +105,7 @@ promote_pypi() {
104
105
# promote_s3 torchtext whl "${TORCHTEXT_VERSION}"
105
106
# promote_s3 torchdata whl "${TORCHDATA_VERSION}"
106
107
# promote_s3 torchrec whl "${TORCHREC_VERSION}"
108
+ # promote_s3 fbgemm-gpu whl "${FBGEMMGPU_VERSION}"
107
109
# promote_s3 "libtorch-*" libtorch "${PYTORCH_VERSION}"
108
110
109
111
# promote_conda torchtriton conda "2.1.0"
Original file line number Diff line number Diff line change @@ -6,4 +6,5 @@ TORCHVISION_VERSION=${TORCHVISION_VERSION:-0.16.1}
6
6
TORCHAUDIO_VERSION=${TORCHAUDIO_VERSION:- 2.1.1}
7
7
TORCHTEXT_VERSION=${TORCHTEXT_VERSION:- 0.16.1}
8
8
TORCHDATA_VERSION=${TORCHDATA_VERSION:- 0.7.1}
9
- TORCHREC_VERSION=${TORCHREC_VERSION:- 0.6.0}
9
+ TORCHREC_VERSION=${TORCHREC_VERSION:- 0.6.0}
10
+ FBGEMMGPU_VERSION=${FBGEMMGPU_VERSION:- 0.6.0}
You can’t perform that action at this time.
0 commit comments