Skip to content

[ROCm][CI] Pin Inductor pointwise configs in the MoRIIO dummy-weight P/D test - #62280

Merged
AndreasKaratzas merged 2 commits into
vllm-project:mainfrom
AndreasKaratzas:akaratza_moriio_pin_pointwise
Oct 19, 2026
Merged

AndreasKaratzas merged 2 commits into
vllm-project:mainfrom
AndreasKaratzas:akaratza_moriio_pin_pointwise

Conversation

@AndreasKaratzas

Copy link
Copy Markdown
Member
  • Turn off Inductor pointwise autotuning (triton.autotune_pointwise) next to deterministic mode in test_write_dummy_e2e_rocm.py, and update the comment and docstring that explain why

Since the ROCm 10 image became the default (#58761), the attention-only cases fail with first-token log-prob distances of one or two bf16 steps (0.03-0.14), e.g. https://buildkite.com/vllm/ci/builds/95663. P/D is not involved: two plain direct serves of the same model on the same GPU disagree by the same amount, while each server is bitwise stable across repeats. Inductor's deterministic mode only pins reduction configs and still picks the XBLOCK of pointwise kernels by timing. On ROCm the XBLOCK changes which multiply-adds of the fused QK-norm + RoPE kernel get contracted into FMAs (e.g. 4 unfused packed adds at XBLOCK 1024, none at 4096), and the candidates are within timing noise of each other, so each server ends up with a random config. Pinning every pointwise kernel's config in two fresh processes makes them agree bitwise; flipping only the QK-norm + RoPE kernel's XBLOCK reproduces the CI distances.

With this change the CI-failing qwen3_dense-TRITON_ATTN-P_FULL_AND_PIECEWISE-D_FULL_AND_PIECEWISE case passes 10/10 on MI355 with ROCm 10 (distances exactly 0.0, full external prefix-cache hits).

AI assistance (Claude Code) was used to draft this change. Every line was reviewed by the submitter.

…P/D test

Inductor's deterministic mode pins only reduction configs and still
picks pointwise XBLOCK by timing. On ROCm the XBLOCK changes which
multiply-adds of the fused QK-norm + RoPE kernel are contracted into
FMAs, and the candidates are within timing noise, so separately
compiled servers differ by one or two bf16 logit steps and the
attention-only cases fail on the ROCm 10 image. Turn pointwise
autotuning off as well.

Co-authored-by: Claude <noreply@anthropic.com>
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
@AndreasKaratzas

Copy link
Copy Markdown
Member Author

/ci run

@mergify mergify Bot added rocm Related to AMD ROCm kv-connector labels Oct 18, 2026
@github-project-automation github-project-automation Bot moved this to Todo in AMD Oct 18, 2026
@github-actions

Copy link
Copy Markdown

✅ Triggered Buildkite CI #95692 for commit 5ea6711ee1a0.

ℹ️ This PR is 2 commits behind upstream main. Merged upstream main into this branch as 5ea6711ee1a0 before starting CI. Pull the branch before pushing more changes.

@AndreasKaratzas
AndreasKaratzas marked this pull request as ready for review October 18, 2026 10:32

@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.

@simondanielsson simondanielsson left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM

@AndreasKaratzas
AndreasKaratzas merged commit 8e5cae0 into vllm-project:main Oct 19, 2026
27 checks passed
@github-project-automation github-project-automation Bot moved this from Todo to Done in AMD Oct 19, 2026
@AndreasKaratzas
AndreasKaratzas deleted the akaratza_moriio_pin_pointwise branch October 19, 2026 00:31
jackLei0901 added a commit to jackLei0901/vllm_dev that referenced this pull request Oct 19, 2026
Include vllm-project#62274, vllm-project#62280, vllm-project#62286 and vllm-project#62241. The CUTLASS PR diff is unchanged; updated-base runtime validation is left to CI.

Assisted-by: OpenAI Codex
Signed-off-by: jackLei0901 <42642542+jackLei0901@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

kv-connector rocm Related to AMD ROCm

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

3 participants