Skip to content

[Fix] Fix extra uninstall of cutlass packages #25756

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Fridge003 merged 5 commits into from
May 19, 2026
Merged

[Fix] Fix extra uninstall of cutlass packages #25756

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
e7f9ffd
upd
Fridge003 May 19, 2026
910ae3b
upd
Fridge003 May 19, 2026
56ff529
Revert "[Fix] Try to fix error caused by latest cutedsl packages (#2…
Fridge003 May 19, 2026
f040d0b
upd
Fridge003 May 19, 2026
70683f5
add
Fridge003 May 19, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 0 additions & 17 deletions scripts/ci/cuda/ci_install_dependency.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -335,22 +335,6 @@ download_flashinfer_cache() {
mark_step_done "${FUNCNAME[0]}"
}

purge_cutlass_libs_base() {
# nvidia-cutlass-dsl[cu13] extras are additive on PyPI: requires_dist always
# pulls -libs-base AND -libs-cu13 when [cu13] is requested. Both wheels write
# to the same site-packages paths with different content, leaving the wrapper
# (cutlass.py, cu13 style) mismatched with the binding (_gpu_ops_gen.py, base
# style) -> GPUModuleOp signature TypeError. See vllm-project/vllm#40082.
# Uninstall -libs-base, then force-reinstall -libs-cu13 so its files win.
$PIP_UNINSTALL_CMD nvidia-cutlass-dsl-libs-base $PIP_UNINSTALL_SUFFIX || true
CUTLASS_DSL_VERSION=$(grep -Po -m1 'nvidia-cutlass-dsl(\[[^]]+\])?==\K[0-9A-Za-z\.\-]+' python/pyproject.toml || echo "")
if [ -n "$CUTLASS_DSL_VERSION" ]; then
$PIP_CMD install --force-reinstall --no-deps "nvidia-cutlass-dsl-libs-cu13==${CUTLASS_DSL_VERSION}" $PIP_INSTALL_SUFFIX
fi

mark_step_done "${FUNCNAME[0]}"
}

stabilize_flashinfer_jit_paths() {
# In venv mode, FlashInfer JIT writes build.ninja with hardcoded -isystem
# paths. Per-job venvs get unique paths, but the JIT cache is shared on the
Expand Down Expand Up @@ -504,7 +488,6 @@ main() {
install_sglang_kernel
install_sglang_router
download_flashinfer_cache
purge_cutlass_libs_base
stabilize_flashinfer_jit_paths
install_extra_deps
install_test_tools
Expand Down
Loading