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: