Skip to content

perf(dsv4-fp4-mi355x-vllm): use AITER a16w4 MoE backend (+21% decode)#1989

Draft
jiacao-amd wants to merge 1 commit into
mainfrom
perf/dsv4-mi355x-aiter-moe
Draft

perf(dsv4-fp4-mi355x-vllm): use AITER a16w4 MoE backend (+21% decode)#1989
jiacao-amd wants to merge 1 commit into
mainfrom
perf/dsv4-mi355x-aiter-moe

Conversation

@jiacao-amd

Copy link
Copy Markdown
Collaborator

Switch the DeepSeek-V4-Pro FP4 MI355X vLLM recipe from --moe-backend triton_unfused to --moe-backend aiter, and export VLLM_ROCM_USE_AITER_MOE=1.

triton_unfused runs a W4A16 Triton dequant path; --moe-backend aiter selects AITER_MXFP4_BF16 (a16w4), a faster AITER kernel for the same numerics. Measured on MI355X TP=8, c64 1k/1k:
output throughput 915 -> 1105 tok/s (+20.7%)
mean TPOT 66.97ms -> 55.36ms

Switch the DeepSeek-V4-Pro FP4 MI355X vLLM recipe from
--moe-backend triton_unfused to --moe-backend aiter, and export
VLLM_ROCM_USE_AITER_MOE=1.

triton_unfused runs a W4A16 Triton dequant path; --moe-backend aiter
selects AITER_MXFP4_BF16 (a16w4), a faster AITER kernel for the same
numerics. Measured on MI355X TP=8, c64 1k/1k:
  output throughput 915 -> 1105 tok/s (+20.7%)
  mean TPOT 66.97ms -> 55.36ms

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

github-actions Bot commented Jul 2, 2026

Copy link
Copy Markdown
Contributor

Thanks for the contribution! For vLLM & SGLang, please ensure that your recipes is similar to the official vLLM recipes and/or the SGLang cookbook

If it is not, please create a PR first before we can merge your single node PR into the master branch. Let's ensure that the documentation is first class such that the entire ML community can benefit from your hard work! Thank you

PR authors are responsible for ensuring that after merging, all GitHub Action jobs fully pass. A lot of the time, failures are just flakes and simply re-running the failed jobs will fix it. If re-running failed jobs is attempted, PR authors are responsible for ensuring it passes. See GitHub's docs on re-running failed jobs: https://docs.github.com/en/actions/how-tos/manage-workflow-runs/re-run-workflows-and-jobs#re-running-failed-jobs-in-a-workflow

As a rule of thumb, generally, PR authors should request a review & get a PR approval from the respective companies' CODEOWNERS before requesting a review from core maintainers.

If additional help is needed, PR authors can reach out to core maintainers over Slack.


感谢你的贡献!对于 vLLM 与 SGLang,请确保你的 recipe 与官方 vLLM recipes 和/或 SGLang cookbook 保持一致

如果不一致,请先创建一个 PR,之后我们才能将你的单节点 PR 合并到 master 分支。让我们确保文档保持一流水准,使整个 ML 社区都能从你的辛勤工作中受益!谢谢

PR 作者有责任确保合并后所有 GitHub Action 任务完全通过。 很多时候失败只是偶发抖动(flake),重新运行失败的任务即可解决。如果选择重新运行失败的任务,PR 作者有责任确保其最终通过。参见 GitHub 关于重新运行失败任务的文档:https://docs.github.com/en/actions/how-tos/manage-workflow-runs/re-run-workflows-and-jobs#re-running-failed-jobs-in-a-workflow

一般而言,PR 作者应先向相应公司的 CODEOWNERS 请求审阅并获得 PR 批准,然后再请求核心维护者审阅。

如需更多帮助,PR 作者可通过 Slack 联系核心维护者。

@jiacao-amd

Copy link
Copy Markdown
Collaborator Author

/sweep test-config --config-files .github/configs/amd-master.yaml --config-keys dsv4-fp4-mi355x-vllm

@github-actions

github-actions Bot commented Jul 2, 2026

Copy link
Copy Markdown
Contributor

@jiacao-amd Kicking off a sweep.

Run: https://github.com/SemiAnalysisAI/InferenceX/actions/runs/28611537783
Command: test-config --config-files .github/configs/amd-master.yaml --config-keys dsv4-fp4-mi355x-vllm
Pinned ref: e493673
Approval: not required (trusted collaborator).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

Status: No status

Development

Successfully merging this pull request may close these issues.

1 participant