[AMD CI] Update CI Version Logic. (#13029)

This commit is contained in:
Sai Enduri
2025-11-10 20:42:00 -08:00
committed by GitHub
parent 3594815a8b
commit c751cb38b0

View File

@@ -3,27 +3,25 @@ set -euo pipefail
# Get version from SGLang version.py file
SGLANG_VERSION_FILE="$(dirname "$0")/../../python/sglang/version.py"
SGLANG_VERSION="v0.5.0rc0" # Default version, will be overridden if version.py is found
SGLANG_VERSION="v0.5.5" # Default version, will be overridden if version.py is found
if [ -f "$SGLANG_VERSION_FILE" ]; then
VERSION_FROM_FILE=$(python3 -c '
import re, sys
with open(sys.argv[1], "r") as f:
content = f.read()
match = re.search(r"__version__\s*=\s*[\"'"'"'](.*?)[\"'"'"']", content)
if match:
print("v" + match.group(1))
' "$SGLANG_VERSION_FILE" 2>/dev/null || echo "")
if [ -n "$VERSION_FROM_FILE" ]; then
TMP_VERSION_FILE=$(mktemp)
if git fetch origin main --quiet; then
if git show origin/main:python/sglang/version.py >"$TMP_VERSION_FILE" 2>/dev/null; then
VERSION_FROM_FILE="v$(cat "$SGLANG_VERSION_FILE" | cut -d'"' -f2)"
if [ -n "$VERSION_FROM_FILE" ]; then
SGLANG_VERSION="$VERSION_FROM_FILE"
echo "Using SGLang version from version.py: $SGLANG_VERSION"
echo "Using SGLang version from origin/main: $SGLANG_VERSION"
else
echo "Warning: Could not parse version from origin/main; using default $SGLANG_VERSION" >&2
fi
else
echo "Warning: Could not parse version from $SGLANG_VERSION_FILE, using default: $SGLANG_VERSION" >&2
echo "Warning: version.py not found on origin/main; using default $SGLANG_VERSION" >&2
fi
else
echo "Warning: version.py not found, using default version: $SGLANG_VERSION" >&2
echo "Warning: failed to fetch origin/main; using default $SGLANG_VERSION" >&2
fi
rm -f "$TMP_VERSION_FILE"
# Default base tags (can be overridden by command line arguments)
@@ -121,9 +119,9 @@ find_latest_image() {
echo "Error: no ${gpu_arch} image found in the last 7 days for base ${base_tag}" >&2
echo "Using hard-coded fallback…" >&2
if [[ "${gpu_arch}" == "mi35x" ]]; then
echo "rocm/sgl-dev:v0.5.0rc0-rocm700-mi35x-20250812"
echo "rocm/sgl-dev:v0.5.5-rocm700-mi35x-20251110"
else
echo "rocm/sgl-dev:v0.5.0rc0-rocm630-mi30x-20250812"
echo "rocm/sgl-dev:v0.5.5-rocm630-mi30x-20251110"
fi
}