[AMD][AgentX] Add DeepSeek-V4-Pro FP4 ATOM PD-disagg AgentX hicache MTP on MI355X / 在 MI355X 上添加 DeepSeek-V4-Pro FP4 ATOM PD 分离式 AgentX hicache MTP recipe - #3158
Conversation
…355X / 在 MI355X 上添加 DeepSeek-V4-Pro FP4 ATOM PD 分离式 AgentX hicache MTP recipe Add the ATOM PD-disaggregated agentic arm for DeepSeek-V4-Pro FP4 on MI355X (dsv4-fp4-mi355x-atom-disagg-agentic-hicache-mtp), the ATOM counterpart to the existing sglang-disagg agentic arm, modeled on ATOM's DeepSeek-V4-Agentic-PD-Max recipe (TP / DP-attention / DP+CPU-offload tiers). - configs/amd-master.yaml: new config key (1P1D TP8, atomesh router, mooncake KV transfer); TP EP1 conc 1/2/8/16, DP EP8 conc 64/128 (no offload), DP EP8 conc 256 with lmcache_offload DRAM tier. - benchmarks/multi_node/agentic/dsv4_fp4_mi355x_atom-disagg.sh: agentic launcher. - amd_utils/models_atom.yaml: DeepSeek-V4-Pro-AgentX entry (prefix-cache env, prefill-only TBO, DP session-affinity env, block-256, agentic server knobs). - amd_utils/server_atom.sh: IS_AGENTIC branch (prefix caching, max-num-seqs=2x conc, lmcache_offload multi-connector, dp-sticky/idx2idx router, synthetic AL, trace_replay.sh client). - perf-changelog.yaml: changelog entry. 在 MI355X 上新增 DeepSeek-V4-Pro FP4 的 ATOM PD 分离式 agentic 分支 (dsv4-fp4-mi355x-atom-disagg-agentic-hicache-mtp),作为现有 sglang-disagg agentic 分支的 ATOM 对照,参考 ATOM 的 DeepSeek-V4-Agentic-PD-Max recipe (TP / DP-attention / DP+CPU 卸载三档)。搜索空间:TP EP1 并发 1/2/8/16, DP EP8 并发 64/128(无卸载),DP EP8 并发 256(lmcache_offload DRAM 卸载档)。 server_atom.sh 新增 IS_AGENTIC 分支:开启前缀缓存、max-num-seqs=2x 并发、 lmcache_offload multi 连接器、dp-sticky/idx2idx 路由、合成接受长度、trace_replay 客户端。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…nk 设为 #3158 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…ic runs / 修复 server_atom.sh 在 agentic 运行时解析 DeepSeek-V4-Pro-AgentX recipe server_atom.sh looked up models_atom.yaml by the bare MODEL_NAME, so agentic runs silently loaded the throughput DeepSeek-V4-Pro entry instead of the new DeepSeek-V4-Pro-AgentX entry (block-256, mem_frac 0.9, batched-tokens, the attn/state/level/AL knobs, DP env, prefill-only TBO were all ignored). Mirror server_sglang.sh: derive the '<model>-AgentX' key when IS_AGENTIC, falling back to the base entry so the non-agentic atom-disagg arm is unchanged. server_atom.sh 之前按裸 MODEL_NAME 查 models_atom.yaml,导致 agentic 运行 静默加载吞吐版 DeepSeek-V4-Pro 条目,而非新的 DeepSeek-V4-Pro-AgentX (block-256、mem_frac 0.9、batched-tokens、attn/state/level/AL 等旋钮、DP 环境变量、仅 prefill 的 TBO 全部被忽略)。改为与 server_sglang.sh 一致: IS_AGENTIC 时解析 '<model>-AgentX',缺失时回退到基础条目,保证非 agentic 的 atom-disagg 分支不变。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…25x-vllm 及其 mtp 配置 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…erence / 使 atom PD agentic recipe 与 DeepSeek-V4-Agentic-PD-Max 参考对齐 Three deviations from the ATOM reference recipe: - DP tiers (conc 64/128/256) used ep=8, adding --enable-expert-parallel that the recipe omits. Switch to ep=1 so the TP+DPA path emits exactly --enable-dp-attention (--enable-tbo on prefill only), no expert-parallel. - OFFLOAD_COPY_WORKERS=1 / OFFLOAD_MIN_LOAD_TOKENS=8192 were exported by the launcher outside the SLURM/Docker boundary and never reached the server. Export them in server_atom.sh's in-container offload branch instead. - ATOM_HOST_IP was never set; the recipe exports it per node. Default it to the node's resolved host_ip (matches the mooncake proxy_ip). 与 ATOM 参考 recipe 的三处偏差: - DP 档(并发 64/128/256)使用 ep=8,引入了 recipe 未使用的 --enable-expert-parallel。改为 ep=1,使 TP+DPA 路径仅输出 --enable-dp-attention(--enable-tbo 仅 prefill),不含 expert-parallel。 - OFFLOAD_COPY_WORKERS=1 / OFFLOAD_MIN_LOAD_TOKENS=8192 之前由启动脚本在 SLURM/Docker 边界外导出,未传入服务器进程。改在 server_atom.sh 容器内的 卸载分支导出。 - ATOM_HOST_IP 从未设置;recipe 每个节点都会导出它。默认设为本节点解析出的 host_ip(与 mooncake proxy_ip 一致)。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
There was a problem hiding this comment.
Additional findings (outside the current diff — GitHub can't attach inline comments there):
-
🔴
benchmarks/multi_node/amd_utils/server_atom.sh— The new DeepSeek-V4-Pro-AgentX model entry is silently never applied on ATOM: server_atom.sh looks up models_atom.yaml with the bare${MODEL_NAME}and never appends-AgentX, so every agentic-only knob (attn-prefill-chunk-size, state-checkpoint-interval-tokens, level, spec-decode-acceptance-length, FP8 kv-cache, prefill-only TBO, block-size 256, DP session-affinity env) is dropped and the plainDeepSeek-V4-Prothroughput recipe runs instead. server_sglang.sh does this correctly: its own python heredoc computesis_agentic/model_key = f'{model_name}-AgentX'from IS_AGENTIC (server_sglang.sh:88-89); server_atom.sh has no equivalent logic. …Extended reasoning...
…Fix: mirror that suffix resolution (with the same DI/AgentX/bare-name fallback) in server_atom.sh's yaml lookup before
m = yaml.safe_load(f).get(...).job.slurm:50 computes MODEL_YAML_KEY="${MODEL_NAME}-AgentX" only to validate the key exists in models_atom.yaml (job.slurm:54-63); it never exports that suffixed value. MODEL_NAME itself stays bare and is exported unchanged at job.slurm:307. server_atom.sh:87 does
yaml.safe_load(f).get('${MODEL_NAME}', {}), i.e. get('DeepSeek-V4-Pro', {}) regardless of IS_AGENTIC. Since 'DeepSeek-V4-Pro' also exists as a top-level key (the throughput entry), .get() returns that dict instead of {} or erroring, so the failure is silent: no crash, just wrong config. Every field the PR added under DeepSeek-V4-Pro-AgentX (models_atom.yaml lines ~48-75) becomes dead code for this recipe: MODEL_ENVS lacks MC_GID_INDEX/ATOM_DISABLE_MMAP/ATOM_PREFIX_CACHE_POLICY/ATOM_DP_SESSION_AFFINITY, prefill_tp_dp_flags lacks --enable-tbo, kv_cache_flags lacks fp8, block_size/mem_frac_static/max_num_batched_tokens fall back to throughput defaults, and…Verification: normal. The new ATOM agentic recipe silently never loads its DeepSeek-V4-Pro-AgentX YAML entry; it falls back to the plain throughput config. Reachable chain: - server.sh:18-20 dispatches ENGINE=atom-disagg -> sources server_atom.sh. - job.slurm:50 computes MODEL_YAML_KEY="${MODEL_NAME}-AgentX" only for existence validation (54-63); it is never exported. MODEL_NAME is exported bare at…
| - "Add DeepSeek-V4-Pro MXFP4 ATOM PD-disaggregated AgentX MTP on MI355X (cluster:mi355x-amds, image rocm/atom-dev:nightly_202609140614), the ATOM counterpart to dsv4-fp4-mi355x-sglang-disagg-agentic-hicache-mtp; mooncake RDMA KV transfer, atomesh PD router." | ||
| - "Search space (1P1D TP8): TP EP1 concurrency 1, 2, 8, 16; TP8/EP8 DP-attention concurrency 64, 128 with no offload; TP8/EP8 DP-attention concurrency 256 with an lmcache_offload CPU KV tier (multi connector, max_pending_saves 8, slot_sidecar_staging_slots 4, LRU, 256-token chunks)." | ||
| - "Per ATOM DeepSeek-V4-Agentic-PD-Max recipe: prefix caching on, FP8 KV and index cache, block-size 256, TBO on prefill only, MTP (--method mtp --num-speculative-tokens 3) with synthetic acceptance length 2.49 on throughput runs, dp-sticky idx2idx routing on the DP tiers, max-num-seqs = 2x concurrency." | ||
| pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/TBD |
There was a problem hiding this comment.
🔴 The new changelog entry's pr-link is https://github.com/SemiAnalysisAI/InferenceX/pull/TBD, which fails the required changelog gate and blocks this PR from merging. validate_perf_changelog.py's validate_added_pr_link() only accepts the exact PR-number URL or the literal placeholders "XXX"/".../pull/XXX" (PR_LINK_PLACEHOLDERS, CANONICAL_PR_LINK regex requires \d+); "TBD" matches neither. Fix: set pr-link to the real PR number URL, or use the canonical "XXX" placeholder if the number isn't known yet.
Extended reasoning...
infx/workflows/validate_perf_changelog.py:144 validate_added_pr_link(link, pr_number) is run by the changelog gate CI check on every PR touching perf-changelog.yaml. For an open PR, pr_number is not None, so it computes expected = '.../pull/<pr_number>' and raises ChangelogValidationError unless link is in PR_LINK_PLACEHOLDERS ({'XXX', '.../pull/XXX'}) or equals expected. The added entry (perf-changelog.yaml:7819) has link '.../pull/TBD', which is neither, so the check raises 'new PR entry must use or an XXX placeholder; found .../pull/TBD'. No code in the repo rewrites 'TBD' before that check runs (grep for TBD outside perf-changelog.yaml finds nothing). Same failure occurs on the main-branch path (pr_number is None) which requires CANONICAL_PR_LINK.fullmatch(link) — also fails on the literal string 'TBD'. Net effect: CI blocks merge until the author replaces the placeholder.
Verification: normal. The appended changelog entry ends with pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/TBD (perf-changelog.yaml, appended block in the diff). The changelog gate rejects this value. In infx/workflows/validate_perf_changelog.py, CANONICAL_PR_LINK (lines 21-23) is .../pull/\d+, and TBD is not digits; PR_LINK_PLACEHOLDERS (lines 24-27) = {"XXX", ".../pull/XXX"}, which…
…tom PD agentic 扫描精简为每档一个并发点 Trim dsv4-fp4-mi355x-atom-disagg-agentic-hicache-mtp to a single representative concurrency per tier (1 TP, 64 DP, 256 DP+offload) for bring-up; the full conc lists are retained as comments. 将 dsv4-fp4-mi355x-atom-disagg-agentic-hicache-mtp 精简为每档一个代表性并发点 (1 TP、64 DP、256 DP+卸载)用于 bring-up;完整并发列表以注释保留。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…lay / 为容器内 ATOM agentic 回放设置 INFMAX_CONTAINER_WORKSPACE The in-container trace_replay.sh path aborted immediately because INFMAX_CONTAINER_WORKSPACE was unset (benchmark_lib.sh uses it to locate utils/aiperf + utils/agentic-benchmark). The SGLang client-image path sets it in its env-file; the ATOM path never did. Derive it from ATOM_WS_PATH (-> the container repo root /workspace) in server_atom.sh's agentic branch. Also narrow the sweep to the conc-1 TP tier (c64/c256 commented out) to validate this fix cheaply before committing GPU time to the DP/offload tiers. 容器内 trace_replay.sh 因 INFMAX_CONTAINER_WORKSPACE 未设置而立即中止 (benchmark_lib.sh 用它定位 utils/aiperf 与 utils/agentic-benchmark)。SGLang 客户端镜像路径在其 env-file 中设置了它;ATOM 路径没有。改为在 server_atom.sh 的 agentic 分支中由 ATOM_WS_PATH 推导(即容器仓库根 /workspace)。 同时将扫描收敛到 conc-1 TP 档(注释掉 c64/c256),以低成本验证此修复, 再对 DP/卸载档投入 GPU 时间。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
… AIPERF_EXPERIMENTAL_FAST 传入 ATOM agentic 容器 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…WER 传入 ATOM agentic 容器 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
… PD agentic 扫描扩展为完整并发网格 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
|
View unofficial run (performance): https://inferencex.semianalysis.com/inference?unofficialRun=36159874837 View unofficial run (accuracy): https://inferencex.semianalysis.com/evaluation?unofficialRun=36159874837 |
…cross all concurrencies Add MC_ENABLE_DEST_DEVICE_AFFINITY=1 to the DeepSeek-V4-Pro-AgentX base env so it applies to every concurrency tier on both prefill and decode nodes, matching ATOM recipes/DeepSeek-V4-Agentic-PD-Max.md. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…/c16 Apply the updated ATOM recipe to the dsv4 agentic PD-disagg arm and test the TP tier on rocm/atom-dev:nightly_202609210314: - drop MC_ENABLE_DEST_DEVICE_AFFINITY, add ATOM_MOONCAKE_MATCHED_RAILS=auto - add --index-cache-dtype fp4 - atomesh: cache_aware policies + balance thresholds on the DP tiers, round_robin on the TP tier, --atom-pd-rank-mapping-policy none on both - dense cudagraph capture ladder 1..min(64, 2*conc) and --cudagraph-mode FULL - ATOM_ENABLE_PREFILL_DELAYER=0 on the DP-attention tiers - spec decode --method dspark with synthetic acceptance 3.01 Throwaway test branch: the sweep is trimmed to conc 1 and 16 for an A/B against run 35566949648, which ran the pre-recipe config on the same image. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Run 35576547691 died in drafter.arm_aux_capture on both cells: ValueError: DSpark requires dspark_target_layer_ids on the draft config. --method dspark reads dspark_block_size / dspark_markov_rank / dspark_target_layer_ids out of the served checkpoint's config.json. Only DeepSeek-V4-Pro-0813 bundles that draft head; the base Pro checkpoint does not, so the drafter had nothing to arm and every ModelRunner aborted after weight load. models.yaml already records this on the SGLang side. Point the arm at DeepSeek-V4-Pro-0813 -- the recipe's stated model, and the same checkpoint the sglang-disagg DSpark arm serves -- and alias DeepSeek-V4-Pro-0813-AgentX to the existing entry so server_atom.sh resolves the recipe under the new name. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
…che-dspark Rename the config key dsv4-fp4-mi355x-atom-disagg-agentic-hicache-mtp to dsv4-fp4-mi355x-atom-disagg-agentic-lmcache-dspark to reflect the LMCache KV offload backend and DSpark speculative decoding used by this recipe. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
…tion to 0.50 Update dsv4-fp4-mi355x-atom-disagg-agentic-lmcache-dspark: bump the ATOM image from nightly_202609210314 to nightly_202609221542 and reduce the agentic-coding dram-utilization from 0.80 to 0.50 (smaller LMCache host-offload budget). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Comment out the conc [1,16] and [64,128] tiers of dsv4-fp4-mi355x-atom-disagg-agentic-lmcache-dspark so the sweep exercises only the conc-256 DP-attention CPU-offload (LMCache) tier. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Switch the active conc-256 tier of dsv4-fp4-mi355x-atom-disagg-agentic-lmcache-dspark from dram/LMCache offload to kv-offloading none (the dram/LMCache variant is kept commented out) to A/B the conc-256 point without the CPU-offload tier. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Enable the conc-256 CPU-offload tier (dram + lmcache, dspark) in amd-master.yaml, and make server_atom.sh/models_atom.yaml reproduce the DeepSeek-V4-Agentic-PD-Max recipe exactly: NUMA binding and the ATOM_DP_* ports are now unconditional (the recipe sets them in the TP tier too, where DP attention is off), and PYTHONHASHSEED=0 is exported on the prefill offload node for consistent LMCache prefix hashes. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
At concurrency 256 the DP-attention router now pins each session to a fixed DP rank (dp_sticky prefill/decode policy) and exports AIPERF_HTTP_X_SESSION_ID_FROM_CORRELATION_ID=1 so aiperf derives the session id from the request correlation id. Conc 64/128 keep cache_aware and the TP tier keeps round_robin. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Restore the full concurrency sweep for the DSV4-Pro ATOM PD-disagg AgentX arm. conc 1/16/64 keep the run-35643358891 shape with only the image bumped; conc 128 matches run 35810485934 (MTP, DP attention, GPU-resident KV) and conc 256 matches run 35825955863 (DSpark draft model, lmcache DRAM tier, dram-utilization 0.50), so the two measured cells reproduce exactly. The MTP tiers keep dram-utilization 0.80 in their own scenario block. Carry the server_atom.sh PORT export so the lm-eval cells can run: without it run_lm_eval's check_env_vars guard aborts the eval and hangs the allocation (run 35643358891). Record the measured conc-128 / conc-256 throughput and the conc-128 gsm8k accuracy in perf-changelog.yaml, and fix the entry's config key, which still named the pre-rename hicache-mtp key. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
…OM PD-disagg AgentX Add the ATOM PD-disaggregated agentic arm for DeepSeek-V4-Pro-0813 MXFP4 on MI355X -- dsv4-fp4-mi355x-atom-disagg-agentic-lmcache-dspark -- the ATOM counterpart to dsv4-fp4-mi355x-sglang-disagg-agentic-hicache-mtp, so the two engines are directly comparable on the AgentX trace workload. 1P1D TP8, mooncake RDMA KV transfer, atomesh PD router, cluster:mi355x-amds. Modeled on ATOM recipes/DeepSeek-V4-Agentic-PD-Max.md. Docker image: rocm/atom-dev:nightly_202609221542 Recipe / search space: - TP8 (EP1): conc 1, 16 -- MTP, GPU-resident KV, dram-utilization 0.80. - DP-attention: conc 64, 128 -- MTP, GPU-resident KV, dram-utilization 0.80. - DP-attention + CPU offload: conc 256 -- DSpark draft model + lmcache host KV tier (multi connector, max_pending_saves=8, slot_sidecar_staging_slots=4, LRU, 256-token chunks), dram-utilization 0.50, dp-sticky idx2idx routing. - conc 1/16/64 keep the shape run 35643358891 used and change only the image. Per ATOM DeepSeek-V4-Agentic-PD-Max recipe: prefix caching on, FP8 KV and index cache, block-size 256, TBO on prefill only, max-num-seqs = 2x concurrency. online_quant: this config does not use online_quant. server_atom.sh only emits --online_quant_config (ONLINE_QUANT_ARG) when the model entry sets online_quant_config / online_quant_dpa_config in models_atom.yaml (read at server_atom.sh:65-66, assembled at :156-172, injected into the prefill/decode server commands at :317/561/652). The DeepSeek-V4-Pro-AgentX entry and its -0813-AgentX alias define neither field, so ONLINE_QUANT_ARG stays empty and --online_quant_config is never passed. Online (runtime) quantization is unnecessary here because the served DeepSeek-V4-Pro-0813 checkpoint is already offline MXFP4-quantized; online_quant (e.g. the ptpc_fp8 configs in deprecated/models_atom.yaml) is for BF16 / unquantized checkpoints. DSpark accuracy is not touched by online_quant. Because online_quant is off, no target layer is re-quantized at runtime, so it cannot perturb draft/target agreement (acceptance length) or gsm8k. The DSpark draft head is read from the -0813 config.json and verifies against the unchanged MXFP4 target; the gsm8k numbers reflect the offline MXFP4 checkpoint + DSpark only. Measured results: - conc-128 (run 35810485934, 3627 s, 13917/15339 profiled, 0 errors): 464105 total tok/s (29007 tok/s/GPU over 16 GPUs), 3934 output tok/s, TTFT mean 9.75 s / p90 15.89, TPOT mean 12.98 ms / p90 14.43, interactivity mean 77.0 / p90 69.3, e2e mean 23.01 s, 3.84 QPS, 96.6% theoretical prefix-cache hit. - conc-256 (run 35825955863, 3630 s, 21317/24162 profiled, 0 errors): 776348 total tok/s (48522 tok/s/GPU), 5794 output tok/s, TTFT mean 18.10 s / p90 32.78, TPOT mean 15.95 ms / p90 17.57, interactivity mean 62.7 / p90 56.9, e2e mean 33.89 s, 5.88 QPS, 96.8% theoretical prefix-cache hit. Not a clean scaling comparison vs conc-128 -- this tier also swaps MTP for the DSpark draft model and adds the lmcache host KV tier. Accuracy: conc-128 DP-attention cell (run 35851118405, lm-eval gsm8k 5-shot, eval concurrency 128, n=1319): em_strict 0.9719 +/- 0.0045, em_flexible 0.9712 +/- 0.0046. conc 1, 16, 64 are unmeasured on this image. Note: server_atom.sh exports PORT before sourcing benchmark_lib.sh on the eval path: run_lm_eval's check_env_vars guard runs before it parses --port, and job.slurm's docker -e allowlist forwards ROUTER_PORT but never PORT, so every multi-node lm-eval cell aborted ~8 s in and hung the decode node in "Waiting until router closes..." until cancelled (run 35643358891). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
|
@billishyahao @chunfangamd plz review this |
chunfangamd
left a comment
There was a problem hiding this comment.
As a PR reviewer and CODEOWNER, I have reviewed this and have:
- Verified that as of the moment of typing this, this is the latest version of PR_REVIEW_CHECKLIST.md
- Verified that the general code quality meets the InferenceX standard and does not make the code quality any worse.
- Verified that this PR has passed PR validation. Please link to GitHub Action workflow that shows this. https://github.com/SemiAnalysisAI/InferenceX/actions/runs/35855544139
- Verified that this PR passes evals. Please link to GitHub Action workflow that shows this. https://github.com/SemiAnalysisAI/InferenceX/actions/runs/35855544139
- Verified that speculative decoding PRs uses chat templates to align the AL distribution to real world
- Verified that every draft model and draft head is served as it ships: the draft that ships with the served checkpoint, at its stored precision, through the pinned upstream image's default handling, with the shipped and effective draft precision recorded in the additional detail section. No submission-side quantization, dtype override, checkpoint substitution, or patch may lower draft precision below that default, regardless of eval results or AL. See Draft-model precision for what counts as the default and the MLPerf comparison.
- For agentic workloads: verified that speculative-decoding configs (EAGLE / MTP / draft models) run with simulated synthetic acceptance, with the acceptance-length value taken from the committed golden AL curve in golden_al_distribution/ for that model, thinking mode, and draft length. A submission may choose any supported draft length, but it may not substitute a different acceptance target.
- Verified against the current MODELS.md that this PR does not submit a deprecated model, scenario, or model-scenario combination.
- Verified that the model architecture isn't changed with benchmark hacks like using --hf-overrides to skipping indexer for every x layers on models that don't natively support this. As a general rule, we won't accept optimizations that reduces the number of model architecture FLOPs. Anything that makes that same computation run faster is fair game; target/verifier FLOPs at lower precisions is fine, given that the config passes private evals, but this does not permit lowering draft-model or draft-head precision below what ships. As an general north star princple, we should only use optimizations which is used in production by customers that care about accuracy
- If an company claims that they support vLLM/SGLang as first class LLM inference engines on their hardware, I have verified that the respective vLLM submission made using upstream https://hub.docker.com/u/vllm docker repo, upstream SGLang https://hub.docker.com/u/lmsysorg docker repo. The only exceptions are for new hardware, such as MI455X UALoE72, Vera Rubin NVL72, Rubin NVL8, etc., and for new model architectures where there is an actual reason why vLLM/SGLang does not fundamentally support them yet as supported by vLLM/SGLang community maintainers
- If an company claims that they support vLLM/SGLang as first class upstream in-tree LLM inference engines on their hardware, I have have verified that the respective vLLM/SGLang submission has been made before additional frameworks (TRT-LLM, ATOM, etc.). The only exceptions are for new hardware, such as MI455X UALoE72, Vera Rubin NVL72, Rubin NVL8, etc., and for new model architectures where there is an actual reason why vLLM/SGLang does not fundamentally support them yet.
- Verified that every single-node vLLM/SGLang recipe in this PR is documented in the official vLLM recipes and/or the SGLang cookbook:
- I linked the corresponding upstream PR in the vLLM recipe repo or SGLang repo and verified that it is MERGED before this InferenceX PR merges. An opened, draft, or closed-without-merge upstream PR does not satisfy this requirement. If the matching recipe was already published, I linked the published recipe/cookbook page in the additional detail section below.
- Verified that this PR does not patch the inference engine or serving stack — the pinned image must run as shipped. This covers .patch files / git apply / patch, inline patches embedded in benchmark scripts (e.g. a python3/sed heredoc that rewrites installed engine sources before serving), in-place edits of site-packages, monkey-patching, overwriting container files, and installing forked/rebuilt engine wheels on top of the pinned image. The only exception is a patch covered by a filled-out waiver at docs/waiver/
<PR_NUMBER>.md— named after the PR that introduces the patch and filed in that same PR, stating what is patched, why the unmodified upstream image cannot run this benchmark, the upstream PR/issue link, and the removal plan — which I have linked below in the additional detail section. - If this PR uses
append-only: true, verified that it only adds generated points or recipe variants inside a selected existing config/scenario and existing same-image visual curve: every previously generated point remains present with the same recipe, no prior point is removed or rerun, and every benchmark-affecting change in the complete diff can affect only the corresponding newly appended points (never an existing point), regardless of which file contains it. - If any of the above criteria cannot reasonably be satisfied, I have provided additional reasoning below.
- Reported measured throughput/E2EL Pareto counts and evidence per affected curve (≥5 points strongly recommended). Below 5 or unverifiable: tag a core maintainer for review; recorded admin bypass required before merge. N/A if no curves are affected. Details.
Additional detail section:
- insert any additional info here
Signed: @chunfangamd
❌❌❌ REJECTED ❌❌❌@chunfangamd — blocking: the sign-off's "Additional detail section" is the untouched template placeholder ("insert any additional info here"), so the draft-precision evidence that the checklist's draft-as-shipped item requires was never recorded, and Check 13 cannot be verified from the sign-off. Please edit the existing review to add it (checkpoint/head, stored precision, pinned-image default handling, effective serving precision, and a statement on ❌ Check 13 (Draft runs as shipped): FAIL — Draft precision could not be verified: the sign-off records no draft evidence (no checkpoint/head identification, stored precision, pinned-image default handling, or effective serving precision). Independent inspection: the draft is the in-checkpoint DSpark head (
Passed and not applicable checks✅ Check 0 (CODEOWNER): PASS — ✅ Check 1 (Passing sweep on in-PR commit): PASS — commit ✅ Check 2 (Evals pass): PASS — gsm8k 5-shot, n=1319, chat-completions, real acceptance (no forced AL on eval-only cells): c16 em_strict 0.978, c128 0.976, c256 (DSpark + lmcache tier) 0.922; all above the ➖ Check 3 (Recipe linked/merged): N/A — disaggregated/multi-node submission ( ✅ Check 4 (Reuse command): PASS — ✅ Check 5 (Latest checklist template): PASS — every item of the current ✅ Check 6 (Upstream images / engine-first): PASS — entry ✅ Check 7 (No deprecated models/scenarios): PASS — ✅ Check 8 (No architecture hacks): PASS — engine kwargs show ✅ Check 9 (Spec-decode via chat template): PASS — trace replay runs aiperf with ✅ Check 10 (No engine patches): PASS — no ✅ Check 11 (Agentic spec-decode golden AL): PASS — throughput cells run ➖ Check 12 (Append-only): N/A — the new Assessed commit: |
chunfangamd
left a comment
There was a problem hiding this comment.
As a PR reviewer and CODEOWNER, I have reviewed this and have:
- Verified that as of the moment of typing this, this is the latest version of PR_REVIEW_CHECKLIST.md
- Verified that the general code quality meets the InferenceX standard and does not make the code quality any worse.
- Verified that this PR has passed PR validation. Please link to GitHub Action workflow that shows this. https://github.com/SemiAnalysisAI/InferenceX/actions/runs/35855544139
- Verified that this PR passes evals. Please link to GitHub Action workflow that shows this. https://github.com/SemiAnalysisAI/InferenceX/actions/runs/35855544139
- Verified that speculative decoding PRs uses chat templates to align the AL distribution to real world
- Verified that every draft model and draft head is served as it ships: the draft that ships with the served checkpoint, at its stored precision, through the pinned upstream image's default handling, with the shipped and effective draft precision recorded in the additional detail section. No submission-side quantization, dtype override, checkpoint substitution, or patch may lower draft precision below that default, regardless of eval results or AL. See Draft-model precision for what counts as the default and the MLPerf comparison.
- For agentic workloads: verified that speculative-decoding configs (EAGLE / MTP / draft models) run with simulated synthetic acceptance, with the acceptance-length value taken from the committed golden AL curve in golden_al_distribution/ for that model, thinking mode, and draft length. A submission may choose any supported draft length, but it may not substitute a different acceptance target.
- Verified against the current MODELS.md that this PR does not submit a deprecated model, scenario, or model-scenario combination.
- Verified that the model architecture isn't changed with benchmark hacks like using --hf-overrides to skipping indexer for every x layers on models that don't natively support this. As a general rule, we won't accept optimizations that reduces the number of model architecture FLOPs. Anything that makes that same computation run faster is fair game; target/verifier FLOPs at lower precisions is fine, given that the config passes private evals, but this does not permit lowering draft-model or draft-head precision below what ships. As an general north star princple, we should only use optimizations which is used in production by customers that care about accuracy
- If an company claims that they support vLLM/SGLang as first class LLM inference engines on their hardware, I have verified that the respective vLLM submission made using upstream https://hub.docker.com/u/vllm docker repo, upstream SGLang https://hub.docker.com/u/lmsysorg docker repo. The only exceptions are for new hardware, such as MI455X UALoE72, Vera Rubin NVL72, Rubin NVL8, etc., and for new model architectures where there is an actual reason why vLLM/SGLang does not fundamentally support them yet as supported by vLLM/SGLang community maintainers
- If an company claims that they support vLLM/SGLang as first class upstream in-tree LLM inference engines on their hardware, I have have verified that the respective vLLM/SGLang submission has been made before additional frameworks (TRT-LLM, ATOM, etc.). The only exceptions are for new hardware, such as MI455X UALoE72, Vera Rubin NVL72, Rubin NVL8, etc., and for new model architectures where there is an actual reason why vLLM/SGLang does not fundamentally support them yet.
- Verified that every single-node vLLM/SGLang recipe in this PR is documented in the official vLLM recipes and/or the SGLang cookbook:
- I linked the corresponding upstream PR in the vLLM recipe repo or SGLang repo and verified that it is MERGED before this InferenceX PR merges. An opened, draft, or closed-without-merge upstream PR does not satisfy this requirement. If the matching recipe was already published, I linked the published recipe/cookbook page in the additional detail section below.
- Verified that this PR does not patch the inference engine or serving stack — the pinned image must run as shipped. This covers .patch files / git apply / patch, inline patches embedded in benchmark scripts (e.g. a python3/sed heredoc that rewrites installed engine sources before serving), in-place edits of site-packages, monkey-patching, overwriting container files, and installing forked/rebuilt engine wheels on top of the pinned image. The only exception is a patch covered by a filled-out waiver at docs/waiver/
<PR_NUMBER>.md— named after the PR that introduces the patch and filed in that same PR, stating what is patched, why the unmodified upstream image cannot run this benchmark, the upstream PR/issue link, and the removal plan — which I have linked below in the additional detail section. - If this PR uses
append-only: true, verified that it only adds generated points or recipe variants inside a selected existing config/scenario and existing same-image visual curve: every previously generated point remains present with the same recipe, no prior point is removed or rerun, and every benchmark-affecting change in the complete diff can affect only the corresponding newly appended points (never an existing point), regardless of which file contains it. - If any of the above criteria cannot reasonably be satisfied, I have provided additional reasoning below.
- Reported measured throughput/E2EL Pareto counts and evidence per affected curve (≥5 points strongly recommended). Below 5 or unverifiable: tag a core maintainer for review; recorded admin bypass required before merge. N/A if no curves are affected. Details.
Additional detail section:
- insert any additional info here
- "online_quant is not used on this arm: server_atom.sh only emits --online_quant_config when models_atom.yaml sets online_quant_config/online_quant_dpa_config (read at server_atom.sh:65-66, assembled at :156-172, injected into the prefill/decode server commands at :317/561/652), and the DeepSeek-V4-Pro-AgentX entry and its -0813-AgentX alias set neither, so ONLINE_QUANT_ARG stays empty and the flag is never passed. The served DeepSeek-V4-Pro-0813 checkpoint is already offline MXFP4, so runtime quantization is unnecessary (online_quant targets BF16/unquantized checkpoints, e.g. the ptpc_fp8 configs in deprecated/models_atom.yaml). DSpark accuracy is therefore not touched by online_quant: no target layer is re-quantized at runtime, so acceptance length and the gsm8k numbers above reflect the offline MXFP4 target plus the DSpark draft head only."
Signed: @chunfangamd
❌❌❌ REJECTED ❌❌❌@chunfangamd — blocking: Check 13 (draft runs as shipped) fails. The updated additional detail section only argues that ❌ Check 13 (Draft runs as shipped): FAIL — Draft served below its shipped baseline, and draft evidence is missing from the sign-off. Baseline: the embedded DSpark head
Passed and not applicable checks✅ Check 0 (CODEOWNER): PASS — ✅ Check 1 (Passing sweep on in-PR commit): PASS — commit ✅ Check 2 (Evals pass): PASS — gsm8k 5-shot, n=1319, on image ➖ Check 3 (Recipe linked/merged): N/A — disaggregated/multi-node submission ( ✅ Check 4 (Reuse command): PASS — ✅ Check 5 (Latest checklist template): PASS — every item of the current ✅ Check 6 (Upstream images / engine-first): PASS — the only entry, ✅ Check 7 (No deprecated models/scenarios): PASS — ✅ Check 8 (No architecture hacks): PASS — server logs show ✅ Check 9 (Spec-decode via chat template): PASS — trace replay runs aiperf with ✅ Check 10 (No engine patches): PASS — no ✅ Check 11 (Agentic spec-decode golden AL): PASS — throughput cells run ➖ Check 12 (Append-only): N/A — the new Assessed commit: |
…OM PD-disagg AgentX Add the ATOM PD-disaggregated agentic arm for DeepSeek-V4-Pro-0813 MXFP4 on MI355X -- dsv4-fp4-mi355x-atom-disagg-agentic-lmcache-dspark -- the ATOM counterpart to dsv4-fp4-mi355x-sglang-disagg-agentic-hicache-mtp, so the two engines are directly comparable on the AgentX trace workload. 1P1D TP8, mooncake RDMA KV transfer, atomesh PD router, cluster:mi355x-amds. Modeled on ATOM recipes/DeepSeek-V4-Agentic-PD-Max.md. Docker image: rocm/atom-dev:nightly_202609221542 Recipe / search space: - TP8 (EP1): conc 1, 16 -- MTP, GPU-resident KV, dram-utilization 0.80. - DP-attention: conc 64, 128 -- MTP, GPU-resident KV, dram-utilization 0.80. - DP-attention + CPU offload: conc 256 -- DSpark draft model + lmcache host KV tier (multi connector, max_pending_saves=8, slot_sidecar_staging_slots=4, LRU, 256-token chunks), dram-utilization 0.50, dp-sticky idx2idx routing. - conc 1/16/64 keep the shape run 35643358891 used and change only the image. Per ATOM DeepSeek-V4-Agentic-PD-Max recipe: prefix caching on, FP8 KV and index cache, block-size 256, TBO on prefill only, max-num-seqs = 2x concurrency. online_quant: this config does not use online_quant. server_atom.sh only emits --online_quant_config (ONLINE_QUANT_ARG) when the model entry sets online_quant_config / online_quant_dpa_config in models_atom.yaml (read at server_atom.sh:65-66, assembled at :156-172, injected into the prefill/decode server commands at :317/561/652). The DeepSeek-V4-Pro-AgentX entry and its -0813-AgentX alias define neither field, so ONLINE_QUANT_ARG stays empty and --online_quant_config is never passed. Online (runtime) quantization is unnecessary here because the served DeepSeek-V4-Pro-0813 checkpoint is already offline MXFP4-quantized; online_quant (e.g. the ptpc_fp8 configs in deprecated/models_atom.yaml) is for BF16 / unquantized checkpoints. DSpark accuracy is not touched by online_quant. Because online_quant is off, no target layer is re-quantized at runtime, so it cannot perturb draft/target agreement (acceptance length) or gsm8k. The DSpark draft head is read from the -0813 config.json and verifies against the unchanged MXFP4 target; the gsm8k numbers reflect the offline MXFP4 checkpoint + DSpark only. Measured results: - conc-128 (run 35810485934, 3627 s, 13917/15339 profiled, 0 errors): 464105 total tok/s (29007 tok/s/GPU over 16 GPUs), 3934 output tok/s, TTFT mean 9.75 s / p90 15.89, TPOT mean 12.98 ms / p90 14.43, interactivity mean 77.0 / p90 69.3, e2e mean 23.01 s, 3.84 QPS, 96.6% theoretical prefix-cache hit. - conc-256 (run 35825955863, 3630 s, 21317/24162 profiled, 0 errors): 776348 total tok/s (48522 tok/s/GPU), 5794 output tok/s, TTFT mean 18.10 s / p90 32.78, TPOT mean 15.95 ms / p90 17.57, interactivity mean 62.7 / p90 56.9, e2e mean 33.89 s, 5.88 QPS, 96.8% theoretical prefix-cache hit. Not a clean scaling comparison vs conc-128 -- this tier also swaps MTP for the DSpark draft model and adds the lmcache host KV tier. Accuracy: conc-128 DP-attention cell (run 35851118405, lm-eval gsm8k 5-shot, eval concurrency 128, n=1319): em_strict 0.9719 +/- 0.0045, em_flexible 0.9712 +/- 0.0046. conc 1, 16, 64 are unmeasured on this image. Note: server_atom.sh exports PORT before sourcing benchmark_lib.sh on the eval path: run_lm_eval's check_env_vars guard runs before it parses --port, and job.slurm's docker -e allowlist forwards ROUTER_PORT but never PORT, so every multi-node lm-eval cell aborted ~8 s in and hung the decode node in "Waiting until router closes..." until cancelled (run 35643358891). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
# Conflicts: # benchmarks/multi_node/amd_utils/models_atom.yaml # benchmarks/multi_node/amd_utils/server_atom.sh # perf-changelog.yaml
|
/reuse-sweep-run 35855544139 |
Summary
Add the ATOM PD-disaggregated agentic arm for DeepSeek-V4-Pro-0813 MXFP4 on MI355X —
dsv4-fp4-mi355x-atom-disagg-agentic-lmcache-dspark— the ATOM counterpart todsv4-fp4-mi355x-sglang-disagg-agentic-hicache-mtp, so the two engines are directly comparable on the AgentX trace workload. 1P1D TP8,mooncakeRDMA KV transfer,atomeshPD router,cluster:mi355x-amds. Modeled on ATOMrecipes/DeepSeek-V4-Agentic-PD-Max.md.Docker image
rocm/atom-dev:nightly_202609221542Recipe / search space
1, 16— MTP, GPU-resident KV, dram-utilization0.80.64, 128— MTP, GPU-resident KV, dram-utilization0.80.256— DSpark draft model +lmcachehost KV tier (multiconnector,max_pending_saves=8,slot_sidecar_staging_slots=4, LRU, 256-token chunks), dram-utilization0.50, dp-stickyidx2idxrouting.1/16/64keep the shape run35643358891used and change only the image.Per ATOM
DeepSeek-V4-Agentic-PD-Maxrecipe: prefix caching on, FP8 KV and index cache, block-size 256, TBO on prefill only,max-num-seqs = 2× concurrency.online_quant
This config does not use online_quant.
server_atom.shonly emits--online_quant_config(ONLINE_QUANT_ARG) when the model entry setsonline_quant_config/online_quant_dpa_configinbenchmarks/multi_node/amd_utils/models_atom.yaml(read atserver_atom.sh:65-66, assembled at:156-172, injected into the prefill/decode server commands at:317/561/652). TheDeepSeek-V4-Pro-AgentXentry (and its-0813-AgentXalias) defines neither field, soONLINE_QUANT_ARGstays empty and--online_quant_configis never passed. Online (runtime) quantization is unnecessary here because the servedDeepSeek-V4-Pro-0813checkpoint is already offline MXFP4-quantized; online_quant (e.g. theptpc_fp8configs indeprecated/models_atom.yaml) is for BF16 / unquantized checkpoints.DSpark accuracy is not touched by online_quant. Because online_quant is off, no target layer is re-quantized at runtime, so it cannot perturb draft/target agreement (acceptance length) or gsm8k. The DSpark draft head is read from the
-0813config.jsonand verifies against the unchanged MXFP4 target; the gsm8k numbers below reflect the offline MXFP4 checkpoint + DSpark only.Measured results
35810485934, 3627 s, 13917/15339 profiled, 0 errors): 464105 total tok/s (29007 tok/s/GPU over 16 GPUs), 3934 output tok/s, TTFT mean 9.75 s / p90 15.89, TPOT mean 12.98 ms / p90 14.43, interactivity mean 77.0 / p90 69.3, e2e mean 23.01 s, 3.84 QPS, 96.6% theoretical prefix-cache hit.35825955863, 3630 s, 21317/24162 profiled, 0 errors): 776348 total tok/s (48522 tok/s/GPU), 5794 output tok/s, TTFT mean 18.10 s / p90 32.78, TPOT mean 15.95 ms / p90 17.57, interactivity mean 62.7 / p90 56.9, e2e mean 33.89 s, 5.88 QPS, 96.8% theoretical prefix-cache hit. Not a clean scaling comparison vs conc-128 — this tier also swaps MTP for the DSpark draft model and adds thelmcachehost KV tier.Accuracy
conc-128 DP-attention cell (run
35851118405, lm-eval gsm8k 5-shot, eval concurrency 128, n=1319):em_strict 0.9719 ± 0.0045,em_flexible 0.9712 ± 0.0046. conc 1, 16, 64 are unmeasured on this image.Notes
server_atom.shexportsPORTbefore sourcingbenchmark_lib.shon the eval path:run_lm_eval'scheck_env_varsguard runs before it parses--port, andjob.slurm's docker-eallowlist forwardsROUTER_PORTbut neverPORT, so every multi-node lm-eval cell aborted ~8 s in and hung the decode node in "Waiting until router closes..." until cancelled (run35643358891).中文
在 MI355X 上新增 DeepSeek-V4-Pro-0813 MXFP4 的 ATOM PD 分离式 agentic 分支
dsv4-fp4-mi355x-atom-disagg-agentic-lmcache-dspark,作为dsv4-fp4-mi355x-sglang-disagg-agentic-hicache-mtp的 ATOM 对照。1P1D TP8,mooncakeRDMA KV 传输,atomeshPD 路由,镜像rocm/atom-dev:nightly_202609221542。参考 ATOMrecipes/DeepSeek-V4-Agentic-PD-Max.md。搜索空间:TP8 并发
1,16与 DP-attention 并发64,128(MTP、GPU 常驻 KV、dram-utilization 0.80);并发256使用 DSpark 草稿模型 +lmcache主机 KV 档(dram-utilization 0.50,dp-stickyidx2idx路由)。online_quant 未启用:
server_atom.sh仅在models_atom.yaml设置了online_quant_config/online_quant_dpa_config时才输出--online_quant_config;DeepSeek-V4-Pro-AgentX(及-0813-AgentX别名)两者都未设置,故ONLINE_QUANT_ARG为空,该标志不会传入。所服务的DeepSeek-V4-Pro-0813已是离线 MXFP4 量化,无需在线量化(在线量化用于 BF16/未量化 checkpoint)。因此 online_quant 不影响 DSpark 精度:不会在运行时重新量化 target 层,草稿/目标一致性(接受长度)与 gsm8k 均不受其影响。