[CI] Revert ModelOpt NVFP4 threshold relax (#29844)

This commit is contained in:
Xiaoyu Zhang
2026-07-04 21:17:32 +08:00
committed by GitHub
parent 36fc0093d6
commit 6dd0cefb2a
3 changed files with 29 additions and 12 deletions
+27
View File
@@ -196,6 +196,27 @@ setup_pip_toolchain() {
mark_step_done "${FUNCNAME[0]}"
}
remove_stale_cuda12_nvidia_wheels() {
if [ "$CU_MAJOR" != "13" ]; then
mark_step_done "${FUNCNAME[0]}"
return
fi
mapfile -t STALE_CUDA12_NVIDIA_WHEELS < <(
python3 -m pip list --format=freeze | sed -n 's/^\(nvidia-.*-cu12\)==.*/\1/p'
)
if [ ${#STALE_CUDA12_NVIDIA_WHEELS[@]} -eq 0 ]; then
echo "No stale CUDA 12 NVIDIA wheels found for ${CU_VERSION} job"
mark_step_done "${FUNCNAME[0]}"
return
fi
echo "Removing stale CUDA 12 NVIDIA wheels from ${CU_VERSION} job: ${STALE_CUDA12_NVIDIA_WHEELS[*]}"
$PIP_UNINSTALL_CMD "${STALE_CUDA12_NVIDIA_WHEELS[@]}" $PIP_UNINSTALL_SUFFIX
mark_step_done "${FUNCNAME[0]}"
}
uninstall_stale_flashinfer() {
# Keep flashinfer packages if version matches to avoid re-downloading:
# - flashinfer-cubin: 150+ MB
@@ -552,8 +573,14 @@ main() {
install_apt_packages
clean_site_packages
setup_pip_toolchain
remove_stale_cuda12_nvidia_wheels
uninstall_stale_flashinfer
install_sglang
# Diffusion B200 CI imports torch inside install_sglang_kernel after removing
# stale CUDA 12 NVIDIA wheels, so opt into one early LD_LIBRARY_PATH refresh.
if [ "${SGLANG_CI_EARLY_LD_LIBRARY_PATH:-0}" = "1" ]; then
setup_ld_library_path
fi
install_sglang_kernel
install_sglang_router
download_flashinfer_cache