From fa8d22e665dc5eea26e9dbf85aff730c84e0a30c Mon Sep 17 00:00:00 2001 From: karverma-amd Date: Thu, 17 Sep 2026 01:09:41 -0500 Subject: [PATCH] [AMD][DSV4] Allow moe_a2a_backend='mori' with DSpark + dp attention (#39910) --- python/sglang/srt/arg_groups/speculative_hook.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sglang/srt/arg_groups/speculative_hook.py b/python/sglang/srt/arg_groups/speculative_hook.py index 5f8d5c8c9..390c1f94f 100644 --- a/python/sglang/srt/arg_groups/speculative_hook.py +++ b/python/sglang/srt/arg_groups/speculative_hook.py @@ -559,10 +559,10 @@ def _handle_dspark(server_args: ServerArgs) -> None: if cfg.enable_dp_attention and cfg.dp_size > 1: if not cfg.enable_dp_lm_head: raise ValueError("DSpark with dp attention requires --enable-dp-lm-head.") - if not _is_npu and cfg.moe_a2a_backend not in ("none", "megamoe"): + if not _is_npu and cfg.moe_a2a_backend not in ("none", "megamoe", "mori"): raise ValueError( "DSpark with dp attention supports moe_a2a_backend 'none' " - "(built-in TP MoE) or 'megamoe', got " + "(built-in TP MoE), 'megamoe', or 'mori', got " f"{cfg.moe_a2a_backend!r}." ) if not _is_npu and cfg.moe_a2a_backend != "none":