Skip to content

Commit 7ba22b7

Browse files
hyukjleeclaude
andcommitted
[AMD][AgentX] kimik2.7: explicit AITER-MLA=0 + EP1-vs-EP8 900s tune sweep
- Set VLLM_ROCM_USE_AITER_MLA=0 explicitly (64 KV heads incompatible with the ROCm AITER-MLA kernel; was only working via implicit fallback). - Add kimik2.7-fp4-mi355x-vllm-agentic-tune config key: TP8+LMCache, EP1 vs EP8 at conc[16,32], 900s. Probes whether expert parallelism improves perf (current record sweep runs EP=1 on a large MoE). Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
1 parent a8287d5 commit 7ba22b7

2 files changed

Lines changed: 27 additions & 0 deletions

File tree

benchmarks/single_node/agentic/kimik2.7_fp4_mi355x.sh

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,12 @@ if [[ "$version" == "" || ${version:-0} -lt 177 ]]; then
5353
fi
5454

5555
export VLLM_ROCM_USE_AITER=1
56+
# amd/Kimi-K2.7-Code-MXFP4 has 64 KV heads; the ROCm AITER-MLA kernel only
57+
# supports 16 or 128 heads, so it MUST be disabled for this model. Previously
58+
# left unset (inherited master=1) and only worked via vLLM's implicit backend
59+
# fallback -- make it explicit so a fallback/backend-selection change (or the
60+
# --block-size=1 AITER-MLA forcing path) can't turn this into a hard crash.
61+
export VLLM_ROCM_USE_AITER_MLA=0
5662
export VLLM_ROCM_QUICK_REDUCE_QUANTIZATION=INT4
5763
# Avoid intermittent symm_mem all-reduce rendezvous hang at engine init on
5864
# MI35x nodes (see KIMIK27_CONC64_LMCACHE_RUNBOOK error #4).

configs/amd-master.yaml

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -571,6 +571,27 @@ kimik2.7-fp4-mi355x-vllm-agentic:
571571
- { tp: 8, kv-offloading: dram, kv-offload-backend: { name: mooncake, version: "source-main" }, conc-list: [16, 32, 48] }
572572

573573

574+
# Perf-tuning experiment key (same recipe as the record sweep above, via
575+
# model-prefix kimik2.7). Wave 1: does expert parallelism help? Compares EP1
576+
# (current) vs EP8 at TP8+LMCache. Dispatched at 900s via e2e-tests duration
577+
# override to verify the perf delta before committing a full 3600s run.
578+
kimik2.7-fp4-mi355x-vllm-agentic-tune:
579+
image: vllm/vllm-openai-rocm:v0.24.0
580+
model: amd/Kimi-K2.7-Code-MXFP4
581+
model-prefix: kimik2.7
582+
runner: cluster:mi355x-amds
583+
precision: fp4
584+
framework: vllm
585+
multinode: false
586+
scenarios:
587+
agentic-coding:
588+
- dram-utilization: 0.80
589+
duration: 900
590+
search-space:
591+
- { tp: 8, ep: 1, kv-offloading: dram, kv-offload-backend: { name: lmcache, version: "aaf7c0d3" }, conc-list: [16, 32] }
592+
- { tp: 8, ep: 8, kv-offloading: dram, kv-offload-backend: { name: lmcache, version: "aaf7c0d3" }, conc-list: [16, 32] }
593+
594+
574595
kimik2.5-fp4-mi355x-atom:
575596
image: rocm/atom:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0_atom0.1.4_202607091539
576597
model: amd/Kimi-K2.5-MXFP4

0 commit comments

Comments
 (0)