Fix Triton MoE GEMM shared memory exhaustion by reducing stage count#2723

Open

nidal567 wants to merge 2 commits intoROCm:mainfrom

nidal567:golden-tot-testing

Commits on Apr 13, 2026

Fix Triton MoE GEMM shared memory exhaustion
nidal567
committed

Commits on Apr 14, 2026

Fix: set num_stages=1 on gfx950 using get_arch() conditionally for gfx950 to ensure no bottlenecks for gfx942
nidal567
committed