Skip to content

[AMD][MI35X] Serve Qwen3.5 MXFP4-AttnFP8-V2 on the MI355X SGLang #2793

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
yichiche wants to merge 1 commit into
base: main
Choose a base branch
from
+12 −4
Open

[AMD][MI35X] Serve Qwen3.5 MXFP4-AttnFP8-V2 on the MI355X SGLang #2793

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 2 additions & 4 deletions configs/amd-master.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -314,7 +314,7 @@ qwen3.5-fp8-mi355x-sglang-disagg:

qwen3.5-fp4-mi355x-sglang:
image: lmsysorg/sglang-rocm:v0.5.18-rocm720-mi35x-20260829
model: amd/Qwen3.5-397B-A17B-MXFP4
model: amd/Qwen3.5-397B-A17B-MXFP4-AttnFP8-V2
model-prefix: qwen3.5
runner: mi355x
precision: fp4
Expand All @@ -326,7 +326,6 @@ qwen3.5-fp4-mi355x-sglang:
osl: 1024
search-space:
- { tp: 2, conc-start: 4, conc-end: 256 }
- { tp: 4, conc-start: 4, conc-end: 16 }

qwen3.5-fp4-mi355x-atom:
image: rocm/atom:rocm7.2.2_ubuntu24.04_py3.12_pytorch_release_2.10.0_atom0.1.2.post
Expand All @@ -346,7 +345,7 @@ qwen3.5-fp4-mi355x-atom:

qwen3.5-fp4-mi355x-sglang-mtp:
image: lmsysorg/sglang-rocm:v0.5.18-rocm720-mi35x-20260829
model: amd/Qwen3.5-397B-A17B-MXFP4
model: amd/Qwen3.5-397B-A17B-MXFP4-AttnFP8-V2
model-prefix: qwen3.5
runner: mi355x
precision: fp4
Expand All @@ -358,7 +357,6 @@ qwen3.5-fp4-mi355x-sglang-mtp:
osl: 1024
search-space:
- { tp: 2, conc-start: 4, conc-end: 128, spec-decoding: mtp }
- { tp: 4, conc-start: 4, conc-end: 16, spec-decoding: mtp }

qwen3.5-fp4-mi355x-sglang-agentic-mtp:
image: lmsysorg/sglang-rocm:v0.5.18-rocm720-mi35x-20260829
Expand Down
10 changes: 10 additions & 0 deletions perf-changelog.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6821,3 +6821,13 @@
description:
- "Refresh to collect TensorRT-LLM server metrics."
pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/2774

- config-keys:
- qwen3.5-fp4-mi355x-sglang
- qwen3.5-fp4-mi355x-sglang-mtp
scenario-type:
- fixed-seq-len
description:
- "Serve amd/Qwen3.5-397B-A17B-MXFP4-AttnFP8-V2 instead of amd/Qwen3.5-397B-A17B-MXFP4, the newer AMD checkpoint that keeps MXFP4 MoE weights, quantizes the shared expert into the fused MoE kernel, and runs attention in FP8"
- "Drop the TP4 band from both arms: TP4 shards the FP8 linear_attn in_proj_ba to N=32, for which aiter's CK gemm_a8w8_bpreshuffle has no instance (NPerBlock=64, no MN padding), so the server crashes during prefill"
pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/2793
Loading