[AMD][Fix] Route MoRI through the Qwen MoE all-to-all path (#32039)

2026-08-24 13:04:08 -07:00
parent 6e2f87d589
commit 0665740953
2 changed files with 40 additions and 0 deletions
@@ -154,6 +154,7 @@ def can_fuse_shared_expert(
        or getattr(config, "shared_expert_intermediate_size", 0) <= 0
        or config.shared_expert_intermediate_size != config.moe_intermediate_size
        or get_moe_a2a_backend().is_deepep()
+        or get_moe_a2a_backend().is_mori()
    ):
        return False