From 1c10f47dc65fb32be00a9309bac1f961a64fe2ba Mon Sep 17 00:00:00 2001 From: inkcherry Date: Fri, 21 Nov 2025 12:49:22 +0000 Subject: [PATCH] tp write single pass Signed-off-by: inkcherry --- .../kv_transfer/kv_connector/v1/moriio_connector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 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 461a0c96345c6..eb381c392d3ef 100644 --- a/vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py +++ b/vllm/distributed/kv_transfer/kv_connector/v1/moriio_connector.py @@ -73,7 +73,7 @@ class MoRIIOConstants: COMPLETION_PREFIX = "cmpl" PING_INTERVAL = 5 - MAX_PING_RETRIES = 100000 + MAX_PING_RETRIES = 1000000 DEFAULT_HANDSHAKE_PORT = "6301" DEFAULT_NOTIFY_PORT = "61005" @@ -1932,7 +1932,7 @@ class MoRIIOConnectorWorker: to track which workers are done. """ - done_sending = set() + done_sending, done_recving = set() ,set() if self.is_producer: done_sending = self.moriio_wrapper.pop_finished_req_ids()