diff --git a/python/sglang/srt/layers/moe/utils.py b/python/sglang/srt/layers/moe/utils.py index 65da83c0ec44..0d5fa7ddbce4 100644 --- a/python/sglang/srt/layers/moe/utils.py +++ b/python/sglang/srt/layers/moe/utils.py @@ -26,6 +26,7 @@ class MoeA2ABackend(Enum): MORI = "mori" ASCEND_FUSEEP = "ascend_fuseep" FLASHINFER = "flashinfer" + CUSTOMIZED = "customized" @classmethod def _missing_(cls, value): @@ -57,6 +58,9 @@ def is_ascend_fuseep(self): def is_mori(self): return self == MoeA2ABackend.MORI + def is_customized(self): + return self == MoeA2ABackend.CUSTOMIZED + class MoeRunnerBackend(Enum):