Skip to content

[Warmup] Show CuTeDSL compilation progress - #48881

Merged
WoosukKwon merged 1 commit into
mainfrom
agent/cutedsl-warmup-progress
Jul 17, 2026
Merged

WoosukKwon merged 1 commit into
mainfrom
agent/cutedsl-warmup-progress

Conversation

@WoosukKwon

Copy link
Copy Markdown
Collaborator

Summary

  • Show progress while registered CuTeDSL kernels compile during startup.
  • Display the progress bar only on global rank zero to avoid duplicate output from distributed workers.

Duplicate-work check

I searched open PRs for CuTeDSL warmup progress tqdm and cutedsl_warmup. PR #48807 changes exception handling around warmup calls and does not overlap with this rank-zero progress reporting.

Validation

  • Focused mocked check of _compile_cutedsl_warmup_units: verified rank zero wraps compile units with the expected progress description and nonzero ranks do not create a progress bar
  • .venv/bin/pre-commit run --files vllm/model_executor/warmup/cutedsl_warmup.py: passed

Model evaluation

Not applicable; this only changes startup progress reporting.

AI assistance

AI assistance was used to extract and validate this change. The human submitter will review every changed line and is responsible for understanding and defending the change end-to-end.

Show a rank-zero progress bar while registered CuTeDSL kernels compile during startup.

Co-authored-by: Bugen Zhao <i@bugenzhao.com>
Co-authored-by: Giancarlo Delfin <32987265+TheEpicDolphin@users.noreply.github.com>
Co-authored-by: Isotr0py <Isotr0py@outlook.com>
Co-authored-by: Isotr0py <mozf@inferact.ai>
Co-authored-by: Jee Jee Li <jeejeelee@inferact.ai>
Co-authored-by: Roger Wang <hey@rogerw.io>
Co-authored-by: Yifan Qiao <yifanqiao@inferact.ai>
Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
Co-authored-by: OpenAI Codex <codex@openai.com>
Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>
@WoosukKwon
WoosukKwon marked this pull request as ready for review July 17, 2026 03:17

@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 repository is configured for manual code reviews. Comment @claude review for a one-time review, or @claude review always to subscribe this PR to a review on every future push.

Tip: disable this comment in your organization's Code Review settings.

@WoosukKwon
WoosukKwon merged commit ee8f36d into main Jul 17, 2026
12 checks passed
@WoosukKwon
WoosukKwon deleted the agent/cutedsl-warmup-progress branch July 17, 2026 03:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant