From 16d2a7a34355a654af9519bf34849df1137826ff Mon Sep 17 00:00:00 2001 From: inkcherry Date: Fri, 21 Nov 2025 15:03:06 +0000 Subject: [PATCH] updata finished request collection Signed-off-by: inkcherry --- .../kv_transfer/kv_connector/v1/moriio_connector.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py b/vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py index eb381c392d3ef..d737cf704b368 100644 --- a/vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py +++ b/vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py @@ -518,6 +518,8 @@ class MoRIIOWriter: self.worker.moriio_wrapper.send_notify( task.request_id, task.remote_ip, remote_port ) + # mark request as done, then we can free the blocks + self.worker.moriio_wrapper.done_req_ids.append(task.request_id) del self.worker.moriio_wrapper.done_remote_allocate_req_dict[ task.request_id ] @@ -1936,15 +1938,13 @@ class MoRIIOConnectorWorker: if self.is_producer: done_sending = self.moriio_wrapper.pop_finished_req_ids() + + else: if self.mode == MoRIIOMode.WRITE: - done_recving = set() + done_recving = self.moriio_wrapper.pop_finished_write_req_ids() else: done_recving = self._pop_done_transfers() - else: - done_sending, done_recving = ( - set(), - self.moriio_wrapper.pop_finished_write_req_ids(), - ) + return done_sending, done_recving