diff --git a/python/sglang/srt/layers/moe/utils.py b/python/sglang/srt/layers/moe/utils.py index 9fd6e2646..fa136d19c 100644 --- a/python/sglang/srt/layers/moe/utils.py +++ b/python/sglang/srt/layers/moe/utils.py @@ -46,7 +46,7 @@ class MoeRunnerBackend(Enum): AUTO = "auto" TRITON = "triton" TRITON_KERNEL = "triton_kernel" - FLASHINFER = "flashinfer_trtllm" + FLASHINFER_TRTLLM = "flashinfer_trtllm" FLASHINFER_CUTLASS = "flashinfer_cutlass" FLASHINFER_MXFP4 = "flashinfer_mxfp4" FLASHINFER_CUTEDSL = "flashinfer_cutedsl" @@ -61,7 +61,7 @@ class MoeRunnerBackend(Enum): return self == MoeRunnerBackend.TRITON_KERNEL def is_flashinfer_trtllm(self): - return self == MoeRunnerBackend.FLASHINFER + return self == MoeRunnerBackend.FLASHINFER_TRTLLM def is_flashinfer_cutlass(self): return self == MoeRunnerBackend.FLASHINFER_CUTLASS