Skip to content

Update GB200 DeepSeek V4.1 Flash vLLM nightly / 更新 GB200 DeepSeek V4.1 Flash vLLM nightly - #3395

Open
xinli-sw wants to merge 1 commit into
mainfrom
config/dsv41flash-gb200-vllm-nightly
Open

xinli-sw wants to merge 1 commit into
mainfrom
config/dsv41flash-gb200-vllm-nightly

Conversation

@xinli-sw

Copy link
Copy Markdown
Collaborator

Pin the GB200 DeepSeek-V4.1-Flash AgentX arm to vLLM nightly 0961bbae. Enable the merged Blackwell sparse indexer settings and select FlashInfer at TP2, MegaAttention at TP4. Preserve the existing topology and memory limits. Also pass the engine readiness timeout from the launcher where needed.

The vLLM tracking issue and upstream recipe document the included changes. Local shell, YAML, and exact-key matrix checks pass; GPU sweep and eval are pending.

中文

将 GB200 DeepSeek-V4.1-Flash AgentX 配置固定到 vLLM nightly 0961bbae。启用已合并的 Blackwell 稀疏索引器参数,TP2 选择 FlashInfer,TP4 选择 MegaAttention;保留现有拓扑与显存限制,并由启动脚本传入所需的引擎就绪超时。

相关改动见 vLLM 跟踪问题 与上游配方。本地 shell、YAML 和指定配置的矩阵检查通过;GPU sweep 与 eval 待运行。

@github-actions

Copy link
Copy Markdown
Contributor

Thanks for the contribution!

  • Review: If this PR changes files owned by someone other than a repository admin or @SemiAnalysisAI/core, ask one eligible CODEOWNER to complete the latest PR_REVIEW_CHECKLIST.md before contacting a core maintainer on Slack. Follow the template exactly, including As a PR reviewer and CODEOWNER, I have reviewed this and have, so sign-off verification triggers.
  • PR verification: Sweeps only run on labeled PRs. Add full-sweep-fail-fast (strongly recommended); use full-sweep-enabled only when matrix jobs should continue after a failure.
  • After merging: PR authors must ensure all GitHub Actions jobs pass. Transient failures often pass on rerun; see how to rerun failed jobs.
中文

感谢你的贡献!

  • **审阅:**如果 PR 修改的文件归属于仓库管理员及 @SemiAnalysisAI/core 之外的 CODEOWNER,请先联系一位有资格的 CODEOWNER 填写最新的 PR_REVIEW_CHECKLIST.md,再通过 Slack 联系核心维护者。必须严格遵循模板,并保留 As a PR reviewer and CODEOWNER, I have reviewed this and have,才能触发签核验证。
  • **PR 验证:**扫描仅在带有标签的 PR 上运行。强烈建议添加 full-sweep-fail-fast;仅当需要矩阵任务在失败后继续运行时才使用 full-sweep-enabled。
  • **合并后:**PR 作者必须确保所有 GitHub Actions 任务通过。临时性失败通常可以通过重新运行恢复;参见重新运行失败任务的说明。

@xinli-sw
xinli-sw marked this pull request as ready for review September 23, 2026 17:53
@xinli-sw
xinli-sw requested a review from a team September 23, 2026 17:53
@claude

claude Bot commented Sep 23, 2026 •

Copy link
Copy Markdown
Contributor

Claude finished @xinli-sw's task in 3m 48s —— View job


Review of PR #3395

  • Gather context (diff, AGENTS.md, touched files)
  • Validate master config, perf-changelog, and image accessibility
  • Review benchmark script and launcher changes
  • Post review

Summary: One blocking issue, posted inline. The new ATTENTION_ARGS block in the shared dsv41flash_fp4_vllm_mtp.sh is gated on GPU memory, so it also activates on the B200 and GB300 vLLM arms, which are symlinked to the same script but stay pinned to older nightlies (cd10ed6f, af1c0149) and are not listed in the perf-changelog entry. That is an undocumented recipe change on two other configs and may fail engine start if those nightlies predate the indexer_sparse_logits option or the DSV41 backend names. Suggested fix is to gate on a launcher-exported variable from runners/launch_gb200-nv.sh instead of GPU memory.

Everything else checks out: the image is a public Docker Hub tag, the changelog entry is appended at the tail, --use-chat-template flows through the shared replay helper, and VLLM_ENGINE_READY_TIMEOUT_S=7200 in the launcher correctly overrides the 3600 s runtime default (the workflow sources runtime_settings.sh before the launcher). Two notes: I could not verify the older vLLM SHAs against upstream because external access and the vLLM source MCP server were unavailable from this runner, and the PR body is missing the AI model disclosure section required by AGENTS.md.

中文

摘要:发现一个阻塞问题,已内联评论。共享脚本 dsv41flash_fp4_vllm_mtp.sh 中新增的 ATTENTION_ARGS 块以显存大小作为条件,因此也会在 B200 和 GB300 的 vLLM 配置上生效;这两个配置通过符号链接使用同一脚本,但仍固定在较旧的 nightly(cd10ed6f、af1c0149),且未列入 perf-changelog 条目。这属于对另外两个配置的未记录改动,若旧 nightly 不支持 indexer_sparse_logits 或 DSV41 后端名称,引擎启动会失败。建议改为由 runners/launch_gb200-nv.sh 导出变量来控制,而非依据显存判断。

其余检查均通过:镜像为公开 Docker Hub 标签,changelog 条目追加在文件末尾,--use-chat-template 由共享回放函数传入,启动脚本中的 VLLM_ENGINE_READY_TIMEOUT_S=7200 能正确覆盖 3600 秒的默认值。另注:本次无法从 runner 访问外部资源核实旧版 vLLM SHA;PR 描述缺少 AGENTS.md 要求的 AI model disclosure 部分。

· [Branch](https://github.com/SemiAnalysisAI/InferenceX/tree/config/dsv41flash-gb200-vllm-nightly)

Comment thread benchmarks/single_node/agentic/dsv41flash_fp4_vllm_mtp.sh Outdated
Comment thread benchmarks/single_node/agentic/dsv41flash_fp4_vllm_mtp.sh
@github-actions

github-actions Bot commented Sep 24, 2026 •

Copy link
Copy Markdown
Contributor

@xinli-sw
xinli-sw marked this pull request as draft September 24, 2026 15:49
@xinli-sw
xinli-sw marked this pull request as ready for review September 24, 2026 22:20

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

Nothing blocking. The comments below are optional suggestions. There is no need to push a fix for them before merging.

Beyond the inline findings, I also checked whether doubling VLLM_ENGINE_READY_TIMEOUT_S to 7200s in runners/launch_gb200-nv.sh:113 could cost a sweep its Slurm wall-clock instead of a clean readiness-timeout error — the default SALLOC_TIME_LIMIT for these single-node sweeps is 480 minutes (8h, from the benchmark workflow), comfortably longer than the new 2h engine-ready timeout, so this isn't a practical risk.

Extended reasoning...

Confirmed via runners/runtime_settings.sh and .github/workflows/.yml that SALLOC_TIME_LIMIT for these sweeps defaults to 480 minutes, well above the new 7200s (120 min) VLLM_ENGINE_READY_TIMEOUT_S, so the candidate wall-clock-loss concern is not realistic under current defaults. Separately verified via git diff and runners/.sh that the new DSV41_BLACKWELL_ATTENTION gate is exported only by launch_gb200-nv.sh (not by launch_b200-nscale-slurm.sh, launch_gb300-nv.sh, or launch_h200-dgxc-slurm.sh), so the attention-config args no longer key off GPU_MEM_MIB alone.

Comment thread perf-changelog.yaml
Comment on lines +8850 to +8852

- config-keys:
- dsv41flash-fp4-b200-vllm-agentic-dspark

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.

🟡 (optional) This entry's config-keys list only dsv41flash-fp4-b200-vllm-agentic-dspark and dsv41flash-fp4-gb200-vllm-agentic-dspark, but the fixed script (benchmarks/single_node/agentic/dsv41flash_fp4_vllm_mtp.sh) is also symlinked from dsv41flash_fp4_gb300_vllm_mtp.sh and dsv41flash_fp4_h200_vllm_mtp.sh, so dsv41flash-fp4-gb300-vllm-agentic-dspark and dsv41flash-fp4-h200-vllm-agentic-dspark get the same SIGPIPE fix but aren't listed. Anyone auditing perf-changelog.yaml by config-key for GB300 or H200 will miss this fix's context even though it changed their behavior too. Fix: include every config that resolves to the shared script (b200, gb200, gb300, h200) in config-keys, not just the two SKUs named in the PR title.

Why this was flagged

configs/nvidia-master.yaml:8210 (dsv41flash-fp4-gb300-vllm-agentic-dspark) and configs/nvidia-master.yaml:8239 (dsv41flash-fp4-h200-vllm-agentic-dspark) both point at benchmarks/single_node/agentic/dsv41flash_fp4_vllm_mtp.sh via symlink, the same script whose GPU_MEM_MIB probe is fixed by this diff. perf-changelog.yaml:8851-8852 lists only b200 and gb200 in config-keys for that fix entry (perf-changelog.yaml:8850-8858). AGENTS.md requires perf-changelog entries to reference the affected config keys as the source of truth for what changed. An engineer or tool filtering perf-changelog.yaml by dsv41flash-fp4-gb300-vllm-agentic-dspark or dsv41flash-fp4-h200-vllm-agentic-dspark would not find this SIGPIPE fix despite it changing their runs' behavior, unlike on base where the entry didn't exist for anyone.

Verification: nit. The SIGPIPE fix lives in the shared script benchmarks/single_node/agentic/dsv41flash_fp4_vllm_mtp.sh (diff replaces nvidia-smi ... | head -n1 | tr -d ' ' with GPU_MEM_MIB=$(nvidia-smi ...) + %%$'\n'* + numeric validation). ls -la confirms four vLLM symlinks point at this one file: dsv41flash_fp4_b200_vllm_mtp.sh, gb200, gb300, and h200 all -> dsv41flash_fp4_vllm_mtp.sh. All…

@xinli-sw
xinli-sw force-pushed the config/dsv41flash-gb200-vllm-nightly branch from a6a32b1 to 67e2f61 Compare September 25, 2026 01:08
更新 GB200 DeepSeek-V4.1-Flash vLLM nightly 配方,并在 TP2 和 TP4 使用 FlashInfer 稀疏注意力。
@xinli-sw
xinli-sw force-pushed the config/dsv41flash-gb200-vllm-nightly branch from e8c965a to e08edaa Compare September 25, 2026 02:03
@functionstackx

Copy link
Copy Markdown
Collaborator

InferenceX has switched away from unmaintainable bash scripts to YAML files that don't repeat the same stuff over and over again. Please merge the latest main into this PR: we have migrated single-node AgentX onto native srt-slurm (#3428), so AgentX configs are now declarative YAML recipes, not per-config 1000+ line bash slop scripts. Please also delete the old benchmarks/single_node/** scripts (see this recipe for the new format).

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

Status: No status

Development

Successfully merging this pull request may close these issues.

2 participants