EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Bugfix] [P/D] Fix connector with pcp dcp (#8538)

Browse Source 
### What this PR does / why we need it?
Fix the issue where a request does not return due to a specific NPU on
node D having no transmission tasks in the scenario where node D is
enabled with DCP.

### Does this PR introduce _any_ user-facing change?

### How was this patch tested?
by nightly

Signed-off-by: liziyu <liziyu16@huawei.com>
This commit is contained in:
liziyu
2026-04-21 22:23:21 +08:00
committed by GitHub
parent 2cb9f76a0f
commit 9e15ce7074
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
vllm_ascend/distributed/kv_transfer/kv_p2p/mooncake_connector.py
View File

@@ -442,8 +442,7 @@ class KVCacheRecvingThread(threading.Thread):
finally:
self._send_done_signal_to_free_remote_port(remote_request_id, remote_host, remote_port_send_num)
if all_task_done:
if len(req_meta["local_block_ids"]) > 0:
self.task_tracker.update_done_task_count(request_id)
self.task_tracker.update_done_task_count(request_id)
if request_id in self.proc_not_transfer_request:
del self.proc_not_transfer_request[request_id]
self.request_queue.task_done()