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