diff --git a/scripts/update_kernel_whl_index.py b/scripts/update_kernel_whl_index.py index 6e2fccbee..093a7d699 100644 --- a/scripts/update_kernel_whl_index.py +++ b/scripts/update_kernel_whl_index.py @@ -11,8 +11,10 @@ DEFAULT_CUDA_VERSION = "130" def check_wheel_cuda_version(path_name, target_cuda_version): - # Pass ROCm wheel - if re.search(f"rocm", path_name): + # Skip non-CUDA backend wheels (rocm, musa, ...). Their + + # local-version tags don't match the CUDA wheel regex below, and they are + # published by the dedicated release-rocm*/release-musa* jobs. + if re.search(r"\+(rocm|musa)", path_name): 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