minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[Deps] Retire the CUDA 12 lane (#38404)

Browse Source 
Co-authored-by: Mohammad Angkad <mohammad.angkad@radixark.ai>
This commit is contained in:
Mohammad Miadh Angkad
2026-09-10 16:58:09 -07:00
committed by GitHub
co-authored by Mohammad Angkad
parent fae8cd84cb
commit 52c191da52
38 changed files with 170 additions and 737 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/update_kernel_whl_index.py
+12 -11
View File
@@ -5,9 +5,9 @@ import hashlib
import pathlib
import re
# All the CUDA versions that the wheels will cover
SUPPORTED_CUDA_VERSIONS = ["129", "130"]
DEFAULT_CUDA_VERSION = "130"
# Local version a CUDA wheel carries, e.g. sglang_kernel-0.4.6.post1+cu130-...whl
CUDA_LOCAL_VERSION_PATTERN = re.compile(r"\+cu(\d+)")
def check_wheel_cuda_version(path_name, target_cuda_version):
@@ -19,16 +19,17 @@ def check_wheel_cuda_version(path_name, target_cuda_version):
):
return False
# For other CUDA versions, the wheel path name will contain the cuda version suffix, e.g. sglang_kernel-0.4.0+cu130-cp310-abi3-manylinux2014_x86_64.whl
if target_cuda_version != DEFAULT_CUDA_VERSION:
return target_cuda_version in path_name
# Match on the wheel's own +cuNNN tag rather than a list of known versions,
# so a wheel built for a CUDA version this script has never heard of is
# rejected instead of landing in the target index.
match = CUDA_LOCAL_VERSION_PATTERN.search(path_name)
if match is not None:
return match.group(1) == target_cuda_version
# For the default CUDA version, the wheel path name will not contain any cuda version suffix, e.g. sglang_kernel-0.4.0-cp310-abi3-manylinux2014_x86_64.whl
# So we need to check if the wheel path name contains any other cuda version suffix
for cuda_version in SUPPORTED_CUDA_VERSIONS:
if cuda_version != DEFAULT_CUDA_VERSION and cuda_version in path_name:
return False
return True
# An untagged wheel is the default-CUDA build, e.g.
# sglang_kernel-0.4.0-cp310-abi3-manylinux2014_x86_64.whl (PyPI rejects
# local versions, so that upload strips the tag).
return target_cuda_version == DEFAULT_CUDA_VERSION
def update_wheel_index(cuda_version=DEFAULT_CUDA_VERSION, rocm_version=None):