修改P等待问题

98cf1a9e · zhuwenwen · dcaa837c · 98cf1a9e
Commit 98cf1a9e authored Sep 23, 2025 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

vllm/distributed/kv_transfer/kv_connector/v1/p2p/p2p_nccl_connector.py ...ted/kv_transfer/kv_connector/v1/p2p/p2p_nccl_connector.py +4 -3

No files found.
--- a/vllm/distributed/kv_transfer/kv_connector/v1/p2p/p2p_nccl_connector.py
+++ b/vllm/distributed/kv_transfer/kv_connector/v1/p2p/p2p_nccl_connector.py
@@ -286,9 +286,10 @@ class P2pNcclConnector(KVConnectorBase_V1):
                                             kv_cache, remote_address)

    def wait_for_save(self):
-        if self.is_producer:
-            assert self.p2p_nccl_engine is not None
-            self.p2p_nccl_engine.wait_for_sent()
+        pass
+        # if self.is_producer:
+        #     assert self.p2p_nccl_engine is not None
+        #     self.p2p_nccl_engine.wait_for_sent()

    def get_finished(
            self, finished_req_ids: set[str],