Fix missing StandardCombineInput import in BF16 flashinfer_trtllm MoE (#19400)

Co-authored-by: Alison Shao <alisonshao@MacBook-Pro-D2W773R9CD.local>
This commit is contained in:
Alison Shao
2026-02-26 18:32:08 -08:00
committed by GitHub
parent c2dce06d9f
commit 52c8a3632a

View File

@@ -501,6 +501,8 @@ def fused_experts_none_to_flashinfer_trtllm_bf16(
runner_config: MoeRunnerConfig,
) -> StandardCombineInput:
# lazy import
from sglang.srt.layers.moe.token_dispatcher.standard import StandardCombineInput
try:
from flashinfer.fused_moe import trtllm_bf16_moe
except ImportError as e: