[Kernels] Isolate modular kernel code from FusedMoEMethodBase subclasses. (#27123)

2025-11-04 08:59:45 -05:00
parent e4ee658672
commit 938772af03
16 changed files with 271 additions and 311 deletions
--- a/vllm/model_executor/layers/fused_moe/modular_kernel.py
+++ b/vllm/model_executor/layers/fused_moe/modular_kernel.py
@@ -707,6 +707,12 @@ class FusedMoEModularKernel(torch.nn.Module):
            f"{fused_experts.activation_formats[0]}"
        )

+    def supports_expert_map(self) -> bool:
+        """
+        A flag indicating whether or not this class supports expert maps.
+        """
+        return self.fused_experts.supports_expert_map()
+
    def output_is_reduced(self) -> bool:
        """
        Indicates whether or not the output of fused MoE kernel