Skip to content

[Bugfix][Compilation] Guard empty splitting ops before early return - #53061

Open
ActiveSky wants to merge 1 commit into
vllm-project:mainfrom
ActiveSky:codex/fix-53030-piecewise-cudagraph-guard
Open

[Bugfix][Compilation] Guard empty splitting ops before early return#53061
ActiveSky wants to merge 1 commit into
vllm-project:mainfrom
ActiveSky:codex/fix-53030-piecewise-cudagraph-guard

Conversation

@ActiveSky

@ActiveSky ActiveSky commented Aug 20, 2026

Copy link
Copy Markdown
Contributor

Summary

  • Apply the empty splitting_ops cudagraph fallback before the early return for non-VLLM_COMPILE modes.
  • Preserve the existing VLLM_COMPILE behavior while sharing the downgrade logic.
  • Add regression coverage for PIECEWISE -> NONE and FULL_AND_PIECEWISE -> FULL.

Fixes #53030

Why this is not duplicate work

I checked issue #53030 and searched open pull requests for 53030, empty splitting_ops, and piecewise cudagraph. No open PR addresses this early-return guard. The related results are separate cudagraph/spec-decode or documentation changes, not this configuration fix.

Tests

  • VLLM_TARGET_DEVICE=cpu VLLM_USE_PRECOMPILED=1 .venv/bin/python -m pytest tests/compile/test_config.py -q -k empty_splitting_ops_disable_piecewise_cudagraphs --maxfail=1 — 2 passed.
  • .venv/bin/ruff check vllm/config/compilation.py tests/compile/test_config.py — passed.
  • .venv/bin/ruff format --check vllm/config/compilation.py tests/compile/test_config.py — passed.
  • GOPROXY=https://goproxy.cn,direct .venv/bin/pre-commit run --files vllm/config/compilation.py tests/compile/test_config.py — all applicable hooks passed.

AI assistance

AI assistance was used to investigate the issue, implement the fix, and run the checks above. The human submitter must review every changed line and the test results before merging.

Apply the empty splitting_ops cudagraph fallback before returning for non-VLLM_COMPILE modes, keeping piecewise modes consistent with the available splitting operators.

Fixes vllm-project#53030

Signed-off-by: jikuixie <jikuixie@gmail.com>

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working torch.compile

Projects

Status: To triage

1 participant