Skip to content

perf(b300): update vLLM AgentX to DSpark6 / 更新 B300 vLLM AgentX 至 DSpark6 - #3297

Open
ivanium wants to merge 1 commit into
mainfrom
agentx/b300-vllm-dspark6
Open

ivanium wants to merge 1 commit into
mainfrom
agentx/b300-vllm-dspark6

Conversation

@ivanium

@ivanium ivanium commented Sep 19, 2026

Copy link
Copy Markdown
Collaborator

Description

Update B300 vLLM AgentX to the nightly-dev-x86_64-cu13.0.1-591bb95 image with DSpark6 (DeepSeek-V4-Pro-0813).

AI model disclosure

  • AI assistance was used to prepare this PR; the exact model/version could not be verified from the runtime.

Type of Change

  • Configuration change

Checklist

  • I have completed the AI model disclosure and kept it current
  • I have tested my changes locally
  • I have updated documentation if necessary
  • For every change that can affect benchmark performance and every recipe addition or modification, I have appended a new entry to the physical end of perf-changelog.yaml and have not edited historical entries
  • Before merging via reuse, an authorized maintainer (OWNER/MEMBER/COLLABORATOR) has commented /use <run_id> (or the legacy /reuse-sweep-run) on this PR. Do this only once there is a final full sweep that is all green with evals passing, since after this comment the sweep label will no longer automatically kick off new sweeps. Remove and re-add the label to force one.
中文

将 B300 vLLM AgentX 更新至 nightly-dev-x86_64-cu13.0.1-591bb95 镜像,并切换至 DSpark6(DeepSeek-V4-Pro-0813)。

AI 模型使用说明:本 PR 使用了 AI 辅助;运行环境未提供可核实的精确模型版本。

@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 任务通过。临时性失败通常可以通过重新运行恢复;参见重新运行失败任务的说明

@ivanium
ivanium force-pushed the agentx/b300-vllm-dspark6 branch from 5ae34c5 to 04198ae Compare September 19, 2026 22:45
@ivanium
ivanium marked this pull request as ready for review September 19, 2026 22:52
@ivanium
ivanium requested a review from a team September 19, 2026 22:52
@claude

claude Bot commented Sep 19, 2026

Copy link
Copy Markdown
Contributor

Claude finished @ivanium's task in 2m 34s —— View job


Review of PR #3297

  • Gather context (diff vs origin/main, AGENTS.md rules)
  • Review benchmarks/single_node/agentic/dsv4_fp4_b300_vllm_mtp.sh
  • Review configs/nvidia-master.yaml (image, config keys, changelog pairing)
  • Review perf-changelog.yaml (appended at end, format)
  • Post review summary

LGTM - no blocking issues found.

Checks performed: spec-decoding: draft_model is an accepted schema value; the synthetic AL 3.77 matches golden_al_distribution/dsv4-pro-0813-dspark.yaml for 6 draft tokens and the sibling B300 SGLang recipe; --enable-expert-parallel stays gated on EP_SIZE; the image is a public Docker Hub tag; the changelog entry is appended at the tail with a trailing newline and a valid pr-link.

Not verified: the indexer_kv_dtype: "mxfp4" attention-config value and --kv-cache-memory-bytes could not be cross-checked against the vLLM nightly source because the vLLM source server failed to connect, and the sandbox did not allow running validate_perf_changelog. The sweep will exercise both.

中文

LGTM,未发现阻塞性问题。已核对:spec-decoding: draft_model 为合法取值;合成 AL 3.77 与 golden 文件中 6 个 draft token 的值一致;--enable-expert-parallel 仍受 EP_SIZE 控制;镜像为公开 Docker Hub tag;changelog 条目已追加至文件末尾。未能核对:indexer_kv_dtype: "mxfp4"--kv-cache-memory-bytes 在该 nightly 中的有效性(vLLM 源码服务连接失败),以及本地未能运行 changelog 校验脚本;sweep 会覆盖这两项。

· [Branch](https://github.com/SemiAnalysisAI/InferenceX/tree/agentx/b300-vllm-dspark6)

@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 finding, I also checked the new hard-coded --kv-cache-memory-bytes values (dsv4_fp4_b300_vllm_mtp.sh:225-237) for crash risk if a node's free GPU memory differs from what was tuned — these appear to be deliberate per-mode reproducibility pins matching the perf-changelog's stated intent, not an accidentally-omitted fallback, so I did not flag it as a bug.

Extended reasoning...

This run confirmed the diff description matches the actual code: NUM_SPEC_TOKENS 3→6, speculative-config switched to dspark/probabilistic, attention-config key swapped to indexer_kv_dtype, new hard-coded --kv-cache-memory-bytes per concurrency bucket, image/model bump, and search-space entries changed from spec-decoding: mtp to spec-decoding: draft_model in configs/nvidia-master.yaml, plus a single well-formed append-only perf-changelog.yaml entry. A CONFIRMED finding is already queued as an inline comment (draft_model isn't in configs/ci-priority.yaml's spec-decoding adjustments list, which only covers mtp/eagle/eagle3, so this recipe silently loses its CI priority boost) — that alone is reason enough to keep this PR out of the auto-approve bucket. I independently examined the hard-coded KV-cache-memory-bytes values as a candidate additional concern (risk of a hard crash instead of graceful auto-sizing on a node with slightly different free GPU memory) and concluded it's an intentional reproducibility choice explicitly called out in the new perf-changelog entry ("pin --kv-cache-memory-bytes per mode"), not a missing-fallback bug per the repo's bash conventions, so I'm not raising it as a separate issue.

Comment thread configs/nvidia-master.yaml Outdated
Comment on lines +1499 to +1507
- { tp: 8, kv-offloading: none, spec-decoding: draft_model, conc-list: [1, 4] }
# TP4 GPU-resident + DSpark6
- { tp: 4, kv-offloading: none, spec-decoding: draft_model, conc-list: [8] }
# TP4 SimpleCPU + DSpark6
- { tp: 4, kv-offloading: dram, kv-offload-backend: { name: vllm-simple }, spec-decoding: draft_model, conc-list: [16] }
# DEP4 SimpleCPU + DSpark6
- { tp: 4, ep: 4, dp-attn: true, kv-offloading: dram, kv-offload-backend: { name: vllm-simple }, spec-decoding: draft_model, conc-list: [48, 64], router: { name: vllm-router, version: "0.1.14" } }
# DEP8 SimpleCPU + DSpark6
- { tp: 8, ep: 8, dp-attn: true, kv-offloading: dram, kv-offload-backend: { name: vllm-simple }, spec-decoding: draft_model, conc-list: [128, 256, 384, 512, 576], router: { name: vllm-router, version: "0.1.14" } }

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 recipe's search-space entries switch from spec-decoding: mtp to draft_model, which silently loses their CI priority boost since configs/ci-priority.yaml's adjustments.spec-decoding only lists mtp/eagle/eagle3, not draft_model. infx/workflows/ci_priority.py:147 does adjustments.get(str(entry.get('spec-decoding','')), 0), so unknown values quietly get 0 instead of erroring. All ~11 job rows generated from this recipe (TP8/TP4/DEP4/DEP8 combos) now score 0.1 lower than equivalent mtp/eagle jobs, demoting them in the CI dispatch queue with no warning. …

Extended reasoning...

…Fix: add draft_model to ci-priority.yaml's spec-decoding adjustments (matching the mtp/eagle/eagle3 entries), or make calculate_priority raise/warn when a spec-decoding value isn't a recognized adjustment key, covering every recipe using spec-decoding: draft_model, not just this one.

configs/nvidia-master.yaml:1499-1507 now write spec-decoding: draft_model for every search-space row of dsv4-fp4-b300-vllm-agentic-mtp (previously mtp). infx/matrix/generate.py serializes each matrix row with key 'spec-decoding' set to that literal string. infx/workflows/ci_priority.py:calculate_priority reads entry.get('spec-decoding','') and looks it up in policy['adjustments']['spec-decoding'] (configs/ci-priority.yaml:19-22), which only defines mtp: 0.1, eagle: 0.1, eagle3: 0.1. 'draft_model' is absent, so .get(...,0) returns 0 instead of 0.1 -- no exception, no log. Base score ~1.0 plus agentic 0.1, fp4 0.1, framework vllm 1.5, model-prefix dsv4 0.25 = 2.95 for this job vs 3.05 for an otherwise-identical mtp job. This same gap pre-dates the diff for amd-master.yaml's draft_model rows, but this PR is…

Verification: nit. The mechanism is real but the impact is a soft CI-scheduling nudge, not a functional failure. The diff changes every search-space row of dsv4-fp4-b300-vllm-agentic-mtp from spec-decoding: mtp to spec-decoding: draft_model (configs/nvidia-master.yaml:1499,1501,1503,1505,1507). configs/ci-priority.yaml:19-22 only defines adjustments.spec-decoding for mtp/eagle/eagle3 (0.1 each), and…

@ivanium
ivanium force-pushed the agentx/b300-vllm-dspark6 branch from 04198ae to e4ee92c Compare September 19, 2026 23:40
@github-actions

github-actions Bot commented Sep 19, 2026

Copy link
Copy Markdown
Contributor

@ivanium
ivanium force-pushed the agentx/b300-vllm-dspark6 branch 2 times, most recently from 5cfe526 to 6ede666 Compare September 20, 2026 00:18
@ivanium
ivanium force-pushed the agentx/b300-vllm-dspark6 branch from 6ede666 to ab71b61 Compare September 20, 2026 22:04
将 B300 DeepSeek-V4-Pro-0813 AgentX 更新至 DSpark6 和新版 vLLM 镜像,调整 DEP8 调度、KV cache 与 batch token 预算。
@ivanium
ivanium force-pushed the agentx/b300-vllm-dspark6 branch from ab71b61 to ae58d8e Compare September 21, 2026 08:25
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.

1 participant