Skip to content

GitLab

Unverified Commit 01018138 authored by fzyzcjy's avatar fzyzcjy Committed by GitHub
Browse files

Tiny fix wrong naming (#10437)

parent 4844fac9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/sglang/srt/layers/moe/utils.py
View file @ 01018138
...@@ -46,7 +46,7 @@ class MoeRunnerBackend(Enum): ...@@ -46,7 +46,7 @@ class MoeRunnerBackend(Enum):
AUTO = "auto" AUTO = "auto"
TRITON = "triton" TRITON = "triton"
TRITON_KERNEL = "triton_kernel" TRITON_KERNEL = "triton_kernel"
FLASHINFER = "flashinfer_trtllm" FLASHINFER_TRTLLM = "flashinfer_trtllm"
FLASHINFER_CUTLASS = "flashinfer_cutlass" FLASHINFER_CUTLASS = "flashinfer_cutlass"
FLASHINFER_MXFP4 = "flashinfer_mxfp4" FLASHINFER_MXFP4 = "flashinfer_mxfp4"
FLASHINFER_CUTEDSL = "flashinfer_cutedsl" FLASHINFER_CUTEDSL = "flashinfer_cutedsl"
...@@ -61,7 +61,7 @@ class MoeRunnerBackend(Enum): ...@@ -61,7 +61,7 @@ class MoeRunnerBackend(Enum):
return self == MoeRunnerBackend.TRITON_KERNEL return self == MoeRunnerBackend.TRITON_KERNEL
def is_flashinfer_trtllm(self): def is_flashinfer_trtllm(self):
return self == MoeRunnerBackend.FLASHINFER return self == MoeRunnerBackend.FLASHINFER_TRTLLM
def is_flashinfer_cutlass(self): def is_flashinfer_cutlass(self):
return self == MoeRunnerBackend.FLASHINFER_CUTLASS return self == MoeRunnerBackend.FLASHINFER_CUTLASS
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment