[CI][Installation] Avoid uploading CUDA 11.8 wheel (vllm-project#10535)

Signed-off-by: simon-mo <simon.mo@hey.com> Co-authored-by: simon-mo <simon.mo@hey.com> Signed-off-by: Maxime Fournioux <55544262+mfournioux@users.noreply.github.com>
mfournioux · Nov 28, 2024 · d73dfc0 · d73dfc0
1 parent b0e4887
commit d73dfc0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.buildkite/upload-wheels.sh b/.buildkite/upload-wheels.sh
@@ -25,7 +25,12 @@ echo "Version: $version"
 
 # If the version contains "dev", rename it to v1.0.0.dev for consistency
 if [[ $version == *dev* ]]; then
-    new_version="1.0.0.dev"
+    suffix="${version##*.}"
+    if [[ $suffix == cu* ]]; then
+        new_version="1.0.0.dev+${suffix}"
+    else
+        new_version="1.0.0.dev"
+    fi
     new_wheel="${wheel/$version/$new_version}"
     mv -- "$wheel" "$new_wheel"
     wheel="$new_wheel"