From 010181388cd1001b21c6bca731860ff9d275620e Mon Sep 17 00:00:00 2001 From: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com> Date: Mon, 15 Sep 2025 10:24:41 +0800 Subject: [PATCH] Tiny fix wrong naming (#10437) --- python/sglang/srt/layers/moe/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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