Skip to content

[AMD] MiMo-V2.6 recipes: use ROCM_AITER_UNIFIED_ATTN - #1210

Closed
vllmellm wants to merge 2 commits into
vllm-project:mainfrom
EmbeddedLLM:mimo-v26-unified-attn-encoder-dp
Closed

vllmellm wants to merge 2 commits into
vllm-project:mainfrom
EmbeddedLLM:mimo-v26-unified-attn-encoder-dp

Conversation

@vllmellm

@vllmellm vllmellm commented Nov 3, 2026

Copy link
Copy Markdown
Contributor
  • Add --attention-backend ROCM_AITER_UNIFIED_ATTN --block-size 16 to the AMD blocks and guide commands of MiMo-V2.6-Flash-RL (mi300x/mi325x/mi355x, both variants) and MiMo-V2.6-Pro-RL (mi355x); faster than the DiffKV default per vllm-project/vllm#64064.
  • Add the encoder_parallel opt-in feature (--mm-encoder-tp-mode data) to both, per vllm-project/vllm#64179.
  • Pro-RL: DiffKV is no longer the only backend covering the asymmetric head dims, so drop that claim.
  • node scripts/validate-source.mjs --source vllm --dir <vllm>/recipes passes.

@vercel

vercel Bot commented Nov 3, 2026 •

Copy link
Copy Markdown
Contributor

The latest updates on your projects. Learn more about Vercel for GitHub.

Project Deployment Actions Updated
vllm-recipes Ready Ready Preview Nov 5, 2026 2:32pm UTC

Request Review

@gemini-code-assist gemini-code-assist 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.

Code Review

This pull request updates the configuration and documentation for the Xiaomi MiMo V2.6 Flash and Pro RL models. It introduces the encoder_parallel feature and configures the --attention-backend ROCM_AITER_UNIFIED_ATTN and --block-size 16 options for AMD ROCm environments to improve performance. The review feedback highlights that the AMD launch commands in the MiMo-V2.6-Flash-RL.yaml guide are missing the --gpu-memory-utilization 0.95 flag, which should be added to ensure consistency with the actual recipe configurations.

@vllmellm
vllmellm force-pushed the mimo-v26-unified-attn-encoder-dp branch 2 times, most recently from 49eb05a to 20b99eb Compare November 3, 2026 14:09
Signed-off-by: vllmellm <vllm.ellm@embeddedllm.com>
@hongxiayang

Copy link
Copy Markdown
Contributor

please do it in vllm-project/vllm repo

This branch was successfully deployed

1 active deployment
Preview — 9c6a8f17 Deployed Nov 5, 2026 by vercel[bot]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants