Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
charles9304 committed Oct 21, 2024
1 parent 0bcae3f commit 75eead2
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions chatlearn/runtime/parameter_sync.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 75eead2

Please sign in to comment.