Skip to content

[ROCm][Bugfix] Flatten 3D attn_out before o_proj in MLA forward pass - #1

Draft
psakhamo wants to merge 1 commit into
mainfrom
fix/rocm-mla-attn-out-3d-flatten
Draft

psakhamo wants to merge 1 commit into
mainfrom
fix/rocm-mla-attn-out-3d-flatten

Commits

Commits on Jun 2, 2026