[NPU] bugfix for import sgl-kernel error (#21200)

2026-03-23 19:52:36 +08:00
parent 80d4a0753a
commit 8662ba7db4
1 changed files with 5 additions and 1 deletions
@@ -4,7 +4,11 @@ import weakref
 from typing import Optional

 import torch
-from sgl_kernel.kvcacheio import transfer_kv_all_layer_mla
+
+from sglang.srt.utils import is_npu
+
+if not is_npu():
+    from sgl_kernel.kvcacheio import transfer_kv_all_layer_mla

 from sglang.srt.layers.radix_attention import RadixAttention
 from sglang.srt.mem_cache.allocator import (