From 75eead2fff65d649687b87593c6620717f04b9cd Mon Sep 17 00:00:00 2001 From: "jiangle.jl" Date: Mon, 21 Oct 2024 11:51:19 +0800 Subject: [PATCH] fix --- chatlearn/runtime/parameter_sync.py | 1 + 1 file changed, 1 insertion(+) diff --git a/chatlearn/runtime/parameter_sync.py b/chatlearn/runtime/parameter_sync.py index a320d4e..88141a9 100644 --- a/chatlearn/runtime/parameter_sync.py +++ b/chatlearn/runtime/parameter_sync.py @@ -608,6 +608,7 @@ def sort_send_actors(self, send_recv_actor_mappings, sorted_send_actors): def sync_broadcast_multi_threads(self, sorted_send_actors, send_recv_actor_mappings, max_workers, requires_grad, group_name=None, stage2=False): with ThreadPoolExecutor(max_workers=max_workers) as executor: + futures = [] for send_actor in sorted_send_actors: recv_actors = send_recv_actor_mappings[send_actor] if self._comm_type == PARAM_SYNC_COMM_TYPE.BROADCAST: