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

feat: use sgl-kernel cu129 as default (#10188)

Browse Source
This commit is contained in:
Yineng Zhang
2025-09-08 22:01:17 -07:00
committed by GitHub
parent 16ff3d4b05
commit cdc56ef6c1
4 changed files with 19 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sgl-kernel/rename_wheels.sh
View File

@@ -16,8 +16,8 @@ for wheel in "${wheel_files[@]}"; do
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}}"
if ls /usr/local/ | grep -q "12.4"; then
new_wheel="${intermediate_wheel/-cp${cp_version}/+cu124-cp${cp_version}}"
elif ls /usr/local/ | grep -q "12.8"; then
new_wheel="${intermediate_wheel/-cp${cp_version}/+cu128-cp${cp_version}}"
else