Skip to content

Community post: Frontier MoE sleep/wake at TP=4 on consumer Blackwell — 2s wake, 4.5s swap - #3383

Open
DoradusResearch wants to merge 7 commits into
huggingface:mainfrom
DoradusResearch:vllm-blackwell-sleep-mode
Open

Community post: Frontier MoE sleep/wake at TP=4 on consumer Blackwell — 2s wake, 4.5s swap#3383
DoradusResearch wants to merge 7 commits into
huggingface:mainfrom
DoradusResearch:vllm-blackwell-sleep-mode

Conversation

@DoradusResearch

Copy link
Copy Markdown

Community post submission

Posting a recipe + operator-perspective writeup that I think fits the HF Blog audience well — getting vLLM --enable-sleep-mode model rotation working at TP=4 on 4× NVIDIA RTX PRO 6000 Blackwell (consumer-Blackwell SM_120). DeepSeek-V4-Flash + MiMo-V2.5-Flash hot-swap on the same 4 GPUs in ~4.5 s cross-peer swap (down from ~50 s on the prior image stack).

Why it matters for HF readers

  • DeepSeek-V4-Flash on consumer Blackwell has been a recurring "does anyone have this working yet?" question on r/LocalLLaMA and the vLLM Discord
  • The fix turns out to be a single non-obvious --gpu-memory-utilization knob plus two cumem-allocator cherry-picks plus one upstream PR (#41834)
  • Includes a contribution back: I upstreamed the SM_120-gated workspace-shrink patches as vllm-project/vllm#42856
  • Companion canonical writeup with the full debugging story: https://doradusresearch.ai/blog/vllm-blackwell-sleep-mode/

What ships

  • Prebuilt image: ghcr.io/doradusresearch/vllm-blackwell-sm12x-bundle:v4 (Apache-2.0)
  • Live config + measured numbers from 3 live cycles right before submission

Happy to make any edits the team prefers — style, scope, additional context. Thumbnail is 1200×630 PNG generated from the canonical post's OG image.

Signed-off-by: R. Ogden robert@doradusresearch.ai

R. Ogden added 7 commits May 16, 2026 20:08
DeepSeek-V4-Flash + MiMo-V2.5-Flash rotation at TP=4 on 4x RTX PRO 6000.
~2s wake, ~4.5s cross-peer swap (down from ~50s). Recipe, live config,
known limits. Bundle image on GHCR.

Companion canonical post: https://doradusresearch.ai/blog/vllm-blackwell-sleep-mode/
Companion upstream PR: vllm-project/vllm#42856

Signed-off-by: R. Ogden <robert@doradusresearch.ai>
Surfaces honest expectation: 46 safetensor shards + CUDA graph capture
take ~12 min from docker run to first request. Bundle image is ~29 GB,
so first pull on a fresh host adds image-pull time on top.

Signed-off-by: R. Ogden <robert@doradusresearch.ai>
Initial draft recommended 0.70 based on vLLM's residue assumption
(~4 GiB/GPU). Live measurement after a startup OOM showed MiMo's real
sleeping-peer residue is ~9 GiB/GPU on consumer Blackwell. 0.60 is the
honest correct value; pool boots cleanly with ~7 GiB margin.

Adds decode-TPS sweep table across 7 input sizes (500 - 100K tokens),
~70 - 42 tok/s decode showing the PCIe-sync degradation curve at TP=4.

Signed-off-by: R. Ogden <robert@doradusresearch.ai>
MiMo is currently running on stock cu129-nightly without PR #35489 —
first-cycle rotation is clean but multi-cycle stress hits the same
EINVAL race. Unification onto a single bundle:v5 (cu129-nightly base +
cumem cherry-picks + V-pad overlay) is in build/test.

Signed-off-by: R. Ogden <robert@doradusresearch.ai>
…DSv4 DeepGEMM gap explained, hybrid config working
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