Commit 98cf1a9e authored by zhuwenwen's avatar zhuwenwen
Browse files

修改P等待问题

parent dcaa837c
...@@ -286,9 +286,10 @@ class P2pNcclConnector(KVConnectorBase_V1): ...@@ -286,9 +286,10 @@ class P2pNcclConnector(KVConnectorBase_V1):
kv_cache, remote_address) kv_cache, remote_address)
def wait_for_save(self): def wait_for_save(self):
if self.is_producer: pass
assert self.p2p_nccl_engine is not None # if self.is_producer:
self.p2p_nccl_engine.wait_for_sent() # assert self.p2p_nccl_engine is not None
# self.p2p_nccl_engine.wait_for_sent()
def get_finished( def get_finished(
self, finished_req_ids: set[str], self, finished_req_ids: set[str],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment