Skip to content

fix: stabilize GB200 16n4g issue-2579 repro#3013

Open
jinglinglingling wants to merge 3 commits into
mainfrom
linglinj/issue2579-16n4g-stable
Open

fix: stabilize GB200 16n4g issue-2579 repro#3013
jinglinglingling wants to merge 3 commits into
mainfrom
linglinj/issue2579-16n4g-stable

Conversation

@jinglinglingling

@jinglinglingling jinglinglingling commented Jun 30, 2026

Copy link
Copy Markdown
Contributor

Issue

related to #2579.
closes #2812

Summary

Validated that the GB200 grpo-qwen3-235b-16n4g repro is now stable with CUDA13 + vLLM0.20 compatibility fixes.

In the latest long run (max_num_steps=30), training advanced through Step 9 and entered Step 10.
The run stopped due to Slurm time limit, not due to runtime crash.

Changes

  1. Force Triton MoE backend and disable FlashInfer MoE path for Qwen perf recipes used in issue-2579 repro.
  2. Add optional combined wakeup tags (weights + kv_cache) path to reduce refit/wakeup instability.
  3. Make data-plane imports more robust via lazy/optional dependency handling (e.g., tensordict) for worker environments.
  4. Add fused-MoE shard-dim compatibility handling in vLLM backend with retry fallback for known shape-mismatch path.
  5. Harden policy shutdown/destructor behavior during Python/Ray finalization to avoid teardown-time crashes.

Validation

  • grpo-qwen3-235b-16n4g run progressed stably past historical Step-4 failure point.
  • Reached Step 10。
  • No ActorDiedError, RayChannelTimeoutError, or NCCL collective timeout observed in this run.

Capture the CUDA13/vLLM0.20 compatibility fixes that let the 16n4g Qwen235 repro advance stably, including Triton MoE routing, safer policy/refit lifecycle handling, and optional lazy/combined weight-sync paths needed for this environment.

Signed-off-by: Linglin Jing <linglinj@nvidia.com>
@jinglinglingling jinglinglingling requested review from a team as code owners June 30, 2026 13:54
@copy-pr-bot

copy-pr-bot Bot commented Jun 30, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@jinglinglingling jinglinglingling added the CI:Lfast Runs a fast test suite and re-use nightly `main` container (but sync dependencies to PRs version) label Jun 30, 2026
@jinglinglingling

Copy link
Copy Markdown
Contributor Author

/ok to test 0946a24

Run ruff formatting adjustments on the GB200 issue-2579 Python changes so lint and CI quality checks can pass without altering runtime behavior.

Signed-off-by: Linglin Jing <linglinj@nvidia.com>
Merge latest main to satisfy branch freshness checks and keep the GB200 issue-2579 fixes aligned with current upstream behavior.

Signed-off-by: Linglin Jing <linglinj@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CI:Lfast Runs a fast test suite and re-use nightly `main` container (but sync dependencies to PRs version)

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Performance Test failing at vllm async generation worker

1 participant