EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

chore(gb200): update to CUDA 12.9 and improve build process (#8772)

Browse Source
This commit is contained in:
ishandhanani
2025-08-08 13:42:47 -07:00
committed by GitHub
parent 36bfddecb9
commit 4e7f025219
7 changed files with 86 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
sgl-kernel/rename_wheels.sh
View File

@@ -7,8 +7,19 @@ wheel_files=($WHEEL_DIR/*.whl)
for wheel in "${wheel_files[@]}"; do
intermediate_wheel="${wheel/linux/manylinux2014}"
if ls /usr/local/ | grep -q "12.8"; then
new_wheel="${intermediate_wheel/-cp39/+cu128-cp39}"
# Extract the current python version from the wheel name
if [[ $intermediate_wheel =~ -cp([0-9]+)- ]]; then
cp_version="${BASH_REMATCH[1]}"
else
echo "Could not extract Python version from wheel name: $intermediate_wheel"
continue
fi
# Detect CUDA version and add appropriate suffix
if ls /usr/local/ | grep -q "12.9"; then
new_wheel="${intermediate_wheel/-cp${cp_version}/+cu129-cp${cp_version}}"
elif ls /usr/local/ | grep -q "12.8"; then
new_wheel="${intermediate_wheel/-cp${cp_version}/+cu128-cp${cp_version}}"
else
new_wheel="$intermediate_wheel"
fi