[CI] Revert ModelOpt NVFP4 threshold relax (#29844)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user