ci: run setup_ld_library_path before install_sglang_kernel (#24141)

This commit is contained in:
Alison Shao
2026-04-30 10:55:45 -07:00
committed by GitHub
parent 7bb7f6049a
commit dc395bc059
+10
View File
@@ -234,6 +234,16 @@ install_sglang() {
echo "Installing python extras: [${EXTRAS}]"
$PIP_CMD install -e "python[${EXTRAS}]" $PIP_INSTALL_SUFFIX
# Defensive: some runners ended up with nvidia-cusparselt-cu13 metadata
# present but libcusparseLt.so.0 missing on disk, breaking any torch import.
# If the file is missing, force-reinstall the wheel before downstream steps.
SITE_PACKAGES=$(python3 -c "import site; print(site.getsitepackages()[0])")
if [ ! -f "$SITE_PACKAGES/nvidia/cusparselt/lib/libcusparseLt.so.0" ] \
&& pip show nvidia-cusparselt-cu13 >/dev/null 2>&1; then
echo "WARNING: nvidia-cusparselt-cu13 metadata present but libcusparseLt.so.0 missing — reinstalling"
$PIP_CMD install --reinstall nvidia-cusparselt-cu13 $PIP_INSTALL_SUFFIX
fi
mark_step_done "${FUNCNAME[0]}"
}