From 10e0bcd6227bb84a9b824598ed6ee0a9387ac623 Mon Sep 17 00:00:00 2001 From: zhaozx-cn <59479021+zhaozx-cn@users.noreply.github.com> Date: Wed, 24 Jun 2026 17:33:02 +0800 Subject: [PATCH] [NPU] fix dcp break (#29140) --- python/sglang/srt/hardware_backend/npu/memory_pool_npu.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/sglang/srt/hardware_backend/npu/memory_pool_npu.py b/python/sglang/srt/hardware_backend/npu/memory_pool_npu.py index c81d0f819..7e527c6f0 100644 --- a/python/sglang/srt/hardware_backend/npu/memory_pool_npu.py +++ b/python/sglang/srt/hardware_backend/npu/memory_pool_npu.py @@ -177,6 +177,7 @@ class NPUMHATokenToKVPool(MHATokenToKVPool): k_scale: Optional[float] = None, v_scale: Optional[float] = None, layer_id_override: Optional[int] = None, + dcp_kv_mask: Optional[torch.Tensor] = None, ): loc, _ = unwrap_write_loc(loc_info) if layer_id_override is not None: