Skip to content

Revert "[Perf] Narrow DeepSeek V4 eager CUDA graph region" (#51430) - #52093

Closed
Fangzhou-Ai wants to merge 1 commit into
vllm-project:mainfrom
Fangzhou-Ai:afz/revert-dsv4-eager-region
Closed

Revert "[Perf] Narrow DeepSeek V4 eager CUDA graph region" (#51430)#52093
Fangzhou-Ai wants to merge 1 commit into
vllm-project:mainfrom
Fangzhou-Ai:afz/revert-dsv4-eager-region

Conversation

@Fangzhou-Ai

@Fangzhou-Ai Fangzhou-Ai commented Aug 13, 2026

Copy link
Copy Markdown
Contributor

Summary

  • Revert #51430 and restore the wider DeepSeek-V4 eager attention region.
  • Preserve the later ROCm _fused_wqa_wkv_gemm override point while resolving the revert against current main.
  • This is the prerequisite for safely restoring MRV1 piecewise graphs in the stacked follow-up.

Why

On gfx950 TP8, the narrowed eager region corrupts output under the MRV1 piecewise graph path. MRV2 avoids that corruption but regresses 8k/1k concurrency-4 throughput. Restoring the wider eager region makes the legacy path correct and recovers the attention performance needed by the follow-up runner revert.

This revisits closed PR #51750 with new ROCm correctness and performance measurements. No open PR currently reverts #51430.

Test plan

Configuration: DeepSeek-V4-Pro, gfx950, TP=8, DP=1, prefix caching disabled, max_num_batched_tokens=8192, AITER MoE, FP8 KV cache, FULL_AND_PIECEWISE graphs.

Combined with the stacked #51768 revert:

  • Correctness smoke: 1 + 1 returned 2.
  • InferenceX 8k/1k, concurrency 4: 40/40 successful requests.
  • Output throughput: 185.66 tok/s.
  • Mean TPOT: 20.53 ms.
  • Median ITL: 19.47 ms.
  • Mean TTFT: 466.34 ms.
  • Historical reference: 186.99 tok/s, 20.36 ms TPOT, 19.23 ms median ITL.

Pre-commit hooks run during commit passed, including Ruff, formatting, mypy, SPDX, and configuration validation.

Contribution notes

  • Duplicate check: #51750 is closed; no open PR reverting #51430 was found.
  • AI assistance was used for regression isolation, implementation, testing, and PR preparation.
  • The human submitter must review every changed line and independently validate the results before merge.

…ct#51430)"

Restore the wider eager region because the narrowed capture corrupts ROCm output and regresses long-context serving performance.

Co-authored-by: Cursor <cursoragent@cursor.com>
Signed-off-by: Fangzhou Ai <fangzhou.ai@amd.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.

@AndreasKaratzas

Copy link
Copy Markdown
Member

/ci run

@github-actions

Copy link
Copy Markdown

✅ Triggered Buildkite CI #83668 for commit 46b5fb6093aa.

@njhill

njhill commented Aug 14, 2026

Copy link
Copy Markdown
Member

We are planning to address this via #52401 instead.

@Fangzhou-Ai

Copy link
Copy Markdown
Contributor Author

Superseded for the ROCm path by #52488 (MRV1 default + wide eager region on ROCm only; CUDA keeps MRV2 / narrow region). Same approach as #52401.

This global revert can close once #52488 or #52401 lands.

@github-project-automation github-project-automation Bot moved this to Done in NVIDIA Aug 16, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

deepseek Related to DeepSeek models nvidia

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

3 participants