Fix fused_moe import for non-NPU devices (#25076)

Co-authored-by: Brayden Zhong <b8zhong@uwaterloo.ca>
This commit is contained in:
Polisetty V R K Jyothendra Varma
2026-05-12 23:05:51 +03:00
committed by GitHub
co-authored by Brayden Zhong
parent 00d8730d05
commit 52d4c697bb
+1 -3
View File
@@ -62,9 +62,7 @@ from sglang.srt.utils import add_prefix, is_npu
_is_npu = is_npu()
if not _is_npu:
from sglang.srt.layers.moe.fused_moe_triton import fused_moe
else:
if _is_npu:
from sglang.srt.hardware_backend.npu.quantization.fused_moe_method_npu import (
fused_moe_npu as fused_moe,
)