[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 (