[MoE Refactor] Separate Router into OO Classes (#30623)

Signed-off-by: Bill Nell <bnell@redhat.com>
2026-01-18 11:40:49 -05:00
parent 2f03035a61
commit 327a02d8db
45 changed files with 1750 additions and 684 deletions
--- a/tests/model_executor/test_enabled_custom_ops.py
+++ b/tests/model_executor/test_enabled_custom_ops.py
@@ -17,7 +17,7 @@ from vllm.model_executor.layers.activation import (
    ReLUSquaredActivation,
    SiluAndMul,
 )
-from vllm.model_executor.layers.fused_moe.fused_moe import (
+from vllm.model_executor.layers.fused_moe.router.fused_topk_router import (
    dispatch_topk_func,
    vllm_topk_softmax,
 )