fix the artifacts name issue (#3041)

pytorch · Jul 26, 2024 · 77278fe · 77278fe
1 parent d73341c
commit 77278fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/release-wheel-linux.yml b/.github/workflows/release-wheel-linux.yml
@@ -201,12 +201,12 @@ jobs:
           if [[ "${{ inputs.cxx11-tarball-release }}" == "true" ]]; then
               bazel build //:libtorchtrt --compilation_mode opt --config=default
               cp bazel-bin/libtorchtrt.tar.gz \
-              release/tarball/libtorchtrt-${TORCHTRT_VERSION}-tensorrt${TENSORRT_VERSION}-cuda${CU_VERSION}-libtorch${PYTORCH_VERSION}-x86_64-linux.tar.gz
+              release/tarball/libtorchtrt-${BUILD_VERSION}-tensorrt${TRT_VERSION}-cuda${CU_VERSION:2}-libtorch${PYTORCH_VERSION}-x86_64-linux.tar.gz
           else
               ${CONDA_RUN} python setup.py bdist_wheel --release
 
               cp bazel-bin/libtorchtrt.tar.gz \
-              release/tarball/libtorchtrt-${BUILD_VERSION}-pre-cxx11-abi-tensorrt${TRT_VERSION}-cuda${CU_VERSION}-libtorch${BUILD_VERSION}-x86_64-linux.tar.gz
+              release/tarball/libtorchtrt-${BUILD_VERSION}-pre-cxx11-abi-tensorrt${TRT_VERSION}-cuda${CU_VERSION:2}-libtorch${PYTORCH_VERSION}-x86_64-linux.tar.gz
 
               ${CONDA_RUN} python -m pip install auditwheel
               ${CONDA_RUN} python -m auditwheel repair \