From 3a53c26c27c8e2b705f2cc99fc81c53afa3921e1 Mon Sep 17 00:00:00 2001 From: Mohammad Miadh Angkad <176301910+mmangkad@users.noreply.github.com> Date: Fri, 31 Jul 2026 06:21:51 +0800 Subject: [PATCH] [CI] Fix MoE compile and DSA indexer regressions (#32937) --- python/sglang/srt/layers/moe/fused_moe_native.py | 3 ++- test/registered/kernels/ops/attention/test_dsa_indexer.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/python/sglang/srt/layers/moe/fused_moe_native.py b/python/sglang/srt/layers/moe/fused_moe_native.py index a478cf12c..ba9a51e82 100644 --- a/python/sglang/srt/layers/moe/fused_moe_native.py +++ b/python/sglang/srt/layers/moe/fused_moe_native.py @@ -23,7 +23,8 @@ def fused_moe_forward_native( dispatch_output: StandardDispatchOutput, ) -> StandardCombineInput: - x, x_scale, topk_output = dispatch_output + x = dispatch_output.hidden_states + topk_output = dispatch_output.topk_output moe_runner_config = layer.moe_runner_config if moe_runner_config.apply_router_weight_on_input: diff --git a/test/registered/kernels/ops/attention/test_dsa_indexer.py b/test/registered/kernels/ops/attention/test_dsa_indexer.py index dfa3119f1..f032c1a76 100644 --- a/test/registered/kernels/ops/attention/test_dsa_indexer.py +++ b/test/registered/kernels/ops/attention/test_dsa_indexer.py @@ -262,6 +262,7 @@ class MockModelRunner: "dsa_topk_backend": "sgl-kernel", "dsa_paged_mqa_logits_backend": "auto", "disaggregation_mode": "null", + "enable_two_batch_overlap": False, }, )() self.hisparse_coordinator = None