From e75b5411c7491e7d76988a52a8c730fc35722285 Mon Sep 17 00:00:00 2001 From: Ishaan Desai Date: Tue, 13 Feb 2024 14:42:58 +0100 Subject: [PATCH] Use Allgatherv instead of allgather when collecting number of micro simulations on each rank in initialization (#81) --- micro_manager/micro_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/micro_manager/micro_manager.py b/micro_manager/micro_manager.py index 1a11a4c6..75e8cf0b 100644 --- a/micro_manager/micro_manager.py +++ b/micro_manager/micro_manager.py @@ -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)