Skip to content

[ROCm][INTERNAL][WIP] Cap profiling token count to fix SP all-gather OOM for MLA+DSA at TP>1 - #3

Draft
psakhamo wants to merge 1 commit into
mainfrom
fix/rocm-mla-dsa-sp-allgather-profiling
Draft

[ROCm][INTERNAL][WIP] Cap profiling token count to fix SP all-gather OOM for MLA+DSA at TP>1#3
psakhamo wants to merge 1 commit into
mainfrom
fix/rocm-mla-dsa-sp-allgather-profiling

Commits

Commits on Jun 2, 2026