Skip to content

Commit

Permalink
Use Allgatherv instead of allgather when collecting number of micro s…
Browse files Browse the repository at this point in the history
…imulations on each rank in initialization (#81)
  • Loading branch information
IshaanDesai authored Feb 13, 2024
1 parent bed5a4c commit e75b541
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion micro_manager/micro_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -284,7 +284,7 @@ def _initialize(self) -> None:
# Gather number of micro simulations that each rank has, because this rank needs to know how many micro
# simulations have been created by previous ranks, so that it can set
# the correct global IDs
self._comm.Allgather(np.array(self._local_number_of_sims), nms_all_ranks)
self._comm.Allgatherv(np.array(self._local_number_of_sims), nms_all_ranks)

# Get global number of micro simulations
self._global_number_of_sims = np.sum(nms_all_ranks)
Expand Down

0 comments on commit e75b541

Please sign in to comment.