[misc] small qol fixes for release process (#8517)

This commit is contained in:
Simon Mo
2024-09-16 15:11:27 -07:00
committed by GitHub
parent 5478c4b41f
commit 5ce45eb54d
2 changed files with 5 additions and 1 deletions
+3 -1
View File
@@ -371,7 +371,9 @@ def get_vllm_version() -> str:
cuda_version = str(get_nvcc_cuda_version())
if cuda_version != MAIN_CUDA_VERSION:
cuda_version_str = cuda_version.replace(".", "")[:3]
version += f"+cu{cuda_version_str}"
# skip this for source tarball, required for pypi
if "sdist" not in sys.argv:
version += f"+cu{cuda_version_str}"
elif _is_hip():
# Get the HIP version
hipcc_version = get_hipcc_rocm_version()