[CI] Fix MoE compile and DSA indexer regressions (#32937)

2026-07-30 15:21:51 -07:00
parent 85f9998524
commit 3a53c26c27
2 changed files with 3 additions and 1 deletions
@@ -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: