We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d8e9630 commit 2f8a63fCopy full SHA for 2f8a63f
1 file changed
pylops_mpi/utils/_mpi.py
@@ -19,10 +19,14 @@ def mpi_allgather(base_comm: MPI.Comm,
19
recv_buf: Optional[NDArray] = None,
20
engine: str = "numpy",
21
) -> List[NDArray]:
22
- """MPI_Allallgather/allallgather
+ """MPI_Allgather/allgather
23
24
- Dispatch allgather routine based on type of input and availability of
25
- CUDA-Aware MPI
+ Dispatch the appropriate allgather routine based on buffer sizes and
+ CUDA-aware MPI availability.
26
+
27
+ If all ranks provide buffers of equal size, the standard `Allgather`
28
+ collective is used. Otherwise, `Allgatherv` is invoked to handle
29
+ variable-sized buffers.
30
31
Parameters
32
----------
0 commit comments