Skip to content

[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

Merged
adibarra merged 35 commits into
mainfrom
amd/agentx_dsv4_atom_pd_0915
Sep 25, 2026
Merged

adibarra merged 35 commits into
mainfrom
amd/agentx_dsv4_atom_pd_0915

Conversation

@seungrokj

@seungrokj seungrokj commented Sep 15, 2026 •

Copy link
Copy Markdown
Collaborator

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 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 = 2× 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 benchmarks/multi_node/amd_utils/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) defines 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 below 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.

Notes

  • 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).
中文

在 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,mooncake RDMA KV 传输,atomesh PD 路由,镜像 rocm/atom-dev:nightly_202609221542。参考 ATOM recipes/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-sticky idx2idx 路由)。

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 均不受其影响。

…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>
seungrokj and others added 5 commits September 15, 2026 15:31
…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>

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

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 plain DeepSeek-V4-Pro throughput recipe runs instead. server_sglang.sh does this correctly: its own python heredoc computes is_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…

Comment thread perf-changelog.yaml Outdated
- "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

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.

🔴 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…

@seungrokj seungrokj added AMD agentx AgentX benchmarks, recipes, and infrastructure labels Sep 15, 2026
seungrokj and others added 5 commits September 15, 2026 16:05
…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>
seungrokj and others added 2 commits September 16, 2026 18:21
… PD agentic 扫描扩展为完整并发网格

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@github-actions

github-actions Bot commented Sep 17, 2026 •

Copy link
Copy Markdown
Contributor

seungrokj and others added 5 commits September 17, 2026 09:56
…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>
seungrokj and others added 8 commits September 22, 2026 17:27
…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>
@seungrokj

Copy link
Copy Markdown
Collaborator Author

seungrokj and others added 2 commits September 23, 2026 09:36
…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>
@seungrokj

Copy link
Copy Markdown
Collaborator Author

@billishyahao @chunfangamd plz review this

@chunfangamd chunfangamd left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

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

@github-actions

Copy link
Copy Markdown
Contributor

❌❌❌ REJECTED ❌❌❌

⚠️ Pareto coverage needs additional review: @functionstackx @cquil11 @Oseltamivir @adibarra. At least 5 points per affected throughput-versus-E2EL frontier are highly recommended. Below 5, or when coverage cannot be verified, merge only with an explicit, recorded admin bypass for the assessed commit; this advisory comment does not grant or enforce a bypass.

@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 AITER_BF16_FP8_MOE_BOUND=0), then re-verify.

❌ 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 (mtp.0..2, target layers 58-60) of deepseek-ai/DeepSeek-V4-Pro-0813 (revision 72e1d32), stored as FP8 E4M3 + E8M0 scales for attention/dense/shared-expert weights and packed FP4 (expert_dtype: fp4) routed experts; the served command runs --method dspark --num-speculative-tokens 3 on that same checkpoint with online_quant_config: None, hf_overrides: None, and FP8 KV shared by target and draft (allowed). Unresolved lead the sign-off must address: the model env in benchmarks/multi_node/amd_utils/models_atom.yaml sets AITER_BF16_FP8_MOE_BOUND=0; in the pinned aiter fused_moe.py this bound defaults to 256 and, on gfx950 with interleaved MXFP4 experts (ATOM_MOE_GU_ITLV=1), selects bf16 MoE activations below the bound and fp8 at or above it, so 0 forces fp8 MoE activations for every batch, and the draft's mtp.*.ffn.experts share that MoE path. Whether this lowers effective draft MoE activation precision below the pinned image's default is not excluded by the diff and must be documented (the ATOM PD-Max recipe setting it does not override the rule). Evidence: run 35855544139 c1 server log, checkpoint config.

⚠️ Check 14 (Pareto coverage): WARN — curve dsv4 / agentic-coding / cluster:mi355x-amds / fp4 / atom-disagg / rocm/atom-dev:nightly_202609221542 / p90_e2el / run 35855544139 (source SHA 1ed4ab45): 4/5 frontier points from 5 measured bmk_agentic_* artifacts (c16, c64, c128, c256 on the frontier; c1 is dominated by c16 at higher p90 E2EL and lower tput_per_gpu). Helper infx.workflows.pareto_coverage reproduces 4/5, no canonical-flag restriction stamped. No admin-exception comment exists on the PR; admin bypass not requested and not verified. Request an explicit admin bypass for SHA 5499d1cc and this curve, or add a measured point (for example one more concurrency between c16 and c256).

Passed and not applicable checks

✅ Check 0 (CODEOWNER): PASS — @chunfangamd is a named owner of configs/amd-master.yaml; the other four changed paths fall under the * catch-all, which any CODEOWNER satisfies.

✅ Check 1 (Passing sweep on in-PR commit): PASS — commit 1ed4ab45 is in the PR and carries success for all five multi-node agentic / cells (c1, c16, c64, c128, c256) and all three multi-node agentic eval / cells in run 35855544139; head 5499d1cc unchanged since the sign-off.

✅ 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 dsv4 bar of 0.91 in infx/evals/thresholds.yaml, on image rocm/atom-dev:nightly_202609221542 matching the PR config. Note the c256 cell sits about 5 points below the other two.

➖ Check 3 (Recipe linked/merged): N/A — disaggregated/multi-node submission (benchmarks/multi_node/**, multinode: true, disagg: true, framework: atom-disagg); the recipe-link requirement applies to single-node recipes only.

✅ Check 4 (Reuse command): PASS — /reuse-sweep-run 35855544139 posted as a whole-line comment by seungrokj (COLLABORATOR) on 2026-09-23.

✅ Check 5 (Latest checklist template): PASS — every item of the current docs/PR_REVIEW_CHECKLIST.md template has a corresponding checked item in the sign-off.

✅ Check 6 (Upstream images / engine-first): PASS — entry dsv4-fp4-mi355x-atom-disagg-agentic-lmcache-dspark uses framework: atom-disagg, so the vLLM/SGLang upstream-image rule does not apply to it; engine-first ordering is satisfied by existing dsv4-fp4-mi355x-vllm-agentic-mtp (vLLM) and dsv4-fp4-mi355x-sglang-agentic-mtp (SGLang) entries for dsv4 on cluster:mi355x-amds.

✅ Check 7 (No deprecated models/scenarios): PASS — dsv4 agentic-coding is active in MODELS.md on 2026-09-23, and the DSpark arm on DeepSeek-V4-Pro-0813 is the agreed plan of record.

✅ Check 8 (No architecture hacks): PASS — engine kwargs show hf_overrides: None; no layer/expert/indexer trimming; --index-cache-dtype fp4, FP8 KV, --level 3, and the prefill-chunk/state-checkpoint knobs change precision or scheduling, not model FLOPs, and match the merged single-node ATOM dsv4 recipe.

✅ Check 9 (Spec-decode via chat template): PASS — trace replay runs aiperf with --endpoint /v1/chat/completions --endpoint-type chat (benchmark_lib build_replay_cmd) against the atomesh router; evals also hit /v1/chat/completions.

✅ Check 10 (No engine patches): PASS — no .patch, git apply, sed -i, heredoc rewrites, or engine wheel installs; setup_deps.sh installs nothing for ENGINE=atom-disagg, and only client-side agentic deps are installed.

✅ Check 11 (Agentic spec-decode golden AL): PASS — throughput cells run --spec-decode-acceptance-length 3.01 with --method dspark --num-speculative-tokens 3, equal to golden_al_distribution/dsv4-pro-0813-dspark.yaml thinking_on level 3 (3.01); server logs show 3.01 toks/fwd achieved, and eval-only cells omit the flag (real acceptance 3.25-3.62). Informational: the master-config labels the c1-c128 tiers spec-decoding: "mtp" although the served method is DSpark, and the launcher's SPEC_DECODE_AL=2.49 default is overridden by the YAML value.

➖ Check 12 (Append-only): N/A — the new perf-changelog.yaml entry does not set append-only: true.

Assessed commit: 5499d1cc4a5c428c9489493382ef4f493423f468.

@chunfangamd chunfangamd left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

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

@github-actions

Copy link
Copy Markdown
Contributor

❌❌❌ REJECTED ❌❌❌

⚠️ Pareto coverage needs additional review: @functionstackx @cquil11 @Oseltamivir @adibarra. At least 5 points per affected throughput-versus-E2EL frontier are highly recommended. Below 5, or when coverage cannot be verified, merge only with an explicit, recorded admin bypass for the assessed commit; this advisory comment does not grant or enforce a bypass.

@chunfangamd — blocking: Check 13 (draft runs as shipped) fails. The updated additional detail section only argues that online_quant is off; it still does not record the draft head's stored precision, the pinned image's default handling, or the effective serving precision, and it does not address AITER_BF16_FP8_MOE_BOUND=0, which the previous verdict flagged. Independent tracing shows that env var lowers the DSpark draft's MoE activation precision below the pinned image's default. Please either drop AITER_BF16_FP8_MOE_BOUND=0 from the DeepSeek-V4-Pro-AgentX env in benchmarks/multi_node/amd_utils/models_atom.yaml (re-sweep) or document, with pinned-implementation evidence, why it does not reach the draft, then edit the existing review and re-verify.

❌ 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 mtp.0..2 of deepseek-ai/DeepSeek-V4-Pro-0813 stores routed experts as packed MXFP4 (expert_dtype: fp4, E8M0 scales) and attention/shared-expert/main_proj weights as FP8 E4M3; in ATOM DSparkLayer inherits the V4 Block, so those experts run through the same FusedMoE → aiter fused_moe path (per_1x32, gate_mode=INTERLEAVE under ATOM_MOE_GU_ITLV=1, no explicit activation dtype), where aiter's resolve_activation_dtype picks BF16 activations for M < AITER_BF16_FP8_MOE_BOUND (default 256) and FP8 at or above it on gfx950. The recipe env AITER_BF16_FP8_MOE_BOUND=0 (models_atom.yaml DeepSeek-V4-Pro-AgentX.env, exported in the run 35855544139 c1 server log) forces FP8 activations at every batch size, including the draft's small decode batches where the default is BF16. That is submission-side online quantization of draft activations via an environment variable (CONTRIBUTING.md "Draft-model precision", forbidden list). The ATOM PD-Max recipe setting the same value does not override the rule, and ATOM source does not set it by default. Note: verified against aiter fused_moe.py at commit 12bb8056 (2026-09-22T15:31Z, immediately before the nightly_202609221542 build); the image contents themselves were not inspected. Other draft settings are compliant: online_quant_config: None, hf_overrides: None, --method dspark --num-speculative-tokens 3 on the same checkpoint, FP8 KV shared by target and draft.

⚠️ Check 14 (Pareto coverage): WARN — curve dsv4 / agentic-coding / cluster:mi355x-amds / fp4 / atom-disagg / rocm/atom-dev:nightly_202609221542 / p90_e2el / run 35855544139 attempt 1 (source SHA 1ed4ab45): 4/5 frontier points from 5 measured bmk_agentic_* artifacts (tput_per_gpu vs p90_e2el: c16, c64, c128, c256 on the frontier; c1 is dominated by c16 at higher p90 E2EL and lower throughput). infx.workflows.pareto_coverage reproduces 4/5 with no canonical-flag restriction. No admin-exception comment exists on the PR; admin bypass not requested and not verified (@chunfangamd and @seungrokj hold write, not admin). Request an explicit admin bypass for SHA 5499d1cc and this curve, or add a measured point.

Passed and not applicable checks

✅ Check 0 (CODEOWNER): PASS — @chunfangamd is a named owner of configs/amd-master.yaml; the other four changed paths fall under the * catch-all, which any CODEOWNER satisfies.

✅ Check 1 (Passing sweep on in-PR commit): PASS — commit 1ed4ab45 is in the PR and carries success for all five multi-node agentic / cells (c1, c16, c64, c128, c256) and all three multi-node agentic eval / cells in run 35855544139; head 5499d1cc is unchanged since the sign-off.

✅ Check 2 (Evals pass): PASS — gsm8k 5-shot, n=1319, on image rocm/atom-dev:nightly_202609221542 matching the PR config: c16 em_strict 0.978, c128 0.976, c256 (DSpark + lmcache tier) 0.922; all above the dsv4 bar of 0.91 in infx/evals/thresholds.yaml. Eval-only cells run real acceptance (no forced AL).

➖ Check 3 (Recipe linked/merged): N/A — disaggregated/multi-node submission (benchmarks/multi_node/**, multinode: true, disagg: true, framework: atom-disagg); the recipe-link requirement applies to single-node recipes only.

✅ Check 4 (Reuse command): PASS — /reuse-sweep-run 35855544139 posted as a whole-line comment by seungrokj (COLLABORATOR).

✅ Check 5 (Latest checklist template): PASS — every item of the current docs/PR_REVIEW_CHECKLIST.md template has a corresponding checked item in the sign-off.

✅ Check 6 (Upstream images / engine-first): PASS — the only entry, dsv4-fp4-mi355x-atom-disagg-agentic-lmcache-dspark, uses framework: atom-disagg, so the vLLM/SGLang upstream-image rule does not apply to it; engine-first ordering is satisfied by the existing dsv4-fp4-mi355x-vllm-agentic-mtp and dsv4-fp4-mi355x-sglang-agentic-mtp entries on cluster:mi355x-amds.

✅ Check 7 (No deprecated models/scenarios): PASS — dsv4 agentic coding is active in MODELS.md on 2026-09-23, and the DSpark arm on DeepSeek-V4-Pro-0813 is the agreed plan of record.

✅ Check 8 (No architecture hacks): PASS — server logs show hf_overrides: None and online_quant_config: None; no layer/expert/indexer trimming. --index-cache-dtype fp4, FP8 KV, --level 3 and the prefill-chunk/state-checkpoint knobs change precision or scheduling, not model FLOPs.

✅ Check 9 (Spec-decode via chat template): PASS — trace replay runs aiperf with --endpoint /v1/chat/completions --endpoint-type chat (build_replay_cmd in benchmarks/benchmark_lib.sh) against the atomesh router.

✅ Check 10 (No engine patches): PASS — no .patch, git apply, sed -i, heredoc rewrites of engine sources, or engine wheel installs anywhere in the diff.

✅ Check 11 (Agentic spec-decode golden AL): PASS — throughput cells run --spec-decode-acceptance-length 3.01 with --method dspark --num-speculative-tokens 3, equal to golden_al_distribution/dsv4-pro-0813-dspark.yaml thinking_on level 3 (3.01); the server log confirms forced acceptance at 3.01, and eval-only cells omit the flag. Informational: the master config labels the c1–c128 tiers spec-decoding: "mtp" although the served method is DSpark.

➖ Check 12 (Append-only): N/A — the new perf-changelog.yaml entry does not set append-only: true.

Assessed commit: 5499d1cc4a5c428c9489493382ef4f493423f468.

seungrokj and others added 2 commits September 24, 2026 11:20
…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
@SemiAnalysisAI SemiAnalysisAI deleted a comment from seungrokj Sep 25, 2026
@adibarra

Copy link
Copy Markdown
Collaborator

/reuse-sweep-run 35855544139

@adibarra
adibarra merged commit 3a4d35d into main Sep 25, 2026
18 of 30 checks passed
@adibarra
adibarra deleted the amd/agentx_dsv4_atom_pd_0915 branch September 25, 2026 16:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

agentx AgentX benchmarks, recipes, and infrastructure AMD full-sweep-enabled

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

4 participants