[AMD][TBO] Fix mori ep dual stream accuracy (#19888)

2026-03-18 17:00:55 +08:00
parent 8b46f1f4ec
commit f0d7a3f427
2 changed files with 59 additions and 22 deletions
--- a/python/sglang/srt/batch_overlap/two_batch_overlap.py
+++ b/python/sglang/srt/batch_overlap/two_batch_overlap.py
@@ -1035,7 +1035,8 @@ class MaybeTboDeepEPDispatcher(BaseDispatcher):
            ]
        elif get_moe_a2a_backend().is_mori():
            self._inners = [
-                MoriEPDispatcher(**kwargs) for _ in range(num_inner_dispatchers)
+                MoriEPDispatcher(instance_id=i, **kwargs)
+                for i in range(num_inner_dispatchers)
            ]
        elif get_moe_a2a_backend().is_nixl():
            self._inners = [