[bugfix][AMD] Disable aiter allreduce+RMSNorm fusion under DP attention / EP (#27835)

2026-07-02 15:31:25 -07:00
parent caf2e5da2d
commit f3904f0293
2 changed files with 133 additions and 0 deletions
@@ -796,6 +796,8 @@ class LayerCommunicator:
                    _use_aiter
                    and batch_size > 0
                    and get_parallel().tp_size != 6
+                    and not is_dp_attention_enabled()
+                    and get_moe_a2a_backend().is_none()
                    and get_global_server_args().enable_aiter_allreduce_fusion
                )
            )