From 5dc4102960501addf22e383ca3ae3bfbda0b7967 Mon Sep 17 00:00:00 2001 From: gjsheu Date: Wed, 5 Aug 2026 09:54:00 +0800 Subject: [PATCH] =?UTF-8?q?[npu]=20[bugfix]=20Fix=20PD=E2=80=91disaggregat?= =?UTF-8?q?ion=20error=20(#33523)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- python/sglang/srt/disaggregation/ascend/conn.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/python/sglang/srt/disaggregation/ascend/conn.py b/python/sglang/srt/disaggregation/ascend/conn.py index 2162a49bd..d323d64d0 100644 --- a/python/sglang/srt/disaggregation/ascend/conn.py +++ b/python/sglang/srt/disaggregation/ascend/conn.py @@ -108,7 +108,14 @@ class AscendKVManager(MooncakeKVManager): dst_kv_indices: npt.NDArray[np.int32], executor: concurrent.futures.ThreadPoolExecutor, dst_layer_ids: Optional[List[int]] = None, + dst_device_kv_indices: Optional[npt.NDArray[np.int32]] = None, ): + if dst_device_kv_indices is not None: + raise NotImplementedError( + "Ascend PD transfer does not support HiSparse " + "destination device KV indices" + ) + # Group by indices prefill_kv_blocks, dst_kv_blocks = group_concurrent_contiguous( prefill_kv_indices, dst_kv_indices