[DCP][Kimi-k3] Decode on the gathered head width - #2060

[DCP][Kimi-k3] Decode on the gathered head width#2060

zejunchen-zejun merged 2 commits into

mainROCm/ATOM:mainfrom

hexwang/mla_nhead96ROCm/ATOM:hexwang/mla_nhead96Copy head branch name to clipboard