[perf] update op.moe_fused_gate

06e16a27 · zhuwenwen · 263f45a4 · 06e16a27
Commit 06e16a27 authored Feb 04, 2026 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

vllm/model_executor/layers/fused_moe/router/grouped_topk_router.py ...l_executor/layers/fused_moe/router/grouped_topk_router.py +2 -2

No files found.
--- a/vllm/model_executor/layers/fused_moe/router/grouped_topk_router.py
+++ b/vllm/model_executor/layers/fused_moe/router/grouped_topk_router.py
@@ -339,8 +339,8 @@ class GroupedTopKRouter(BaseRouter):
            grouped_topk_impl = grouped_topk
        if self.use_fused_gate:
-            # if envs.VLLM_USE_LIGHTOP:
+            if envs.VLLM_USE_LIGHTOP:
-            if False:
+                from lightop import op as op
                topk_weights, topk_ids = op.moe_fused_gate(
                    router_logits,
                    self.e_score_correction_bias,