File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -167,7 +167,7 @@ def mpi_send(base_comm: MPI.Comm,
167167 Tag of the message to be sent.
168168 engine : :obj:`str`, optional
169169 Engine used to store array (``numpy`` or ``cupy``)
170-
170+
171171 """
172172 if deps .cuda_aware_mpi_enabled or engine == "numpy" :
173173 # Determine MPI type based on array dtype
@@ -183,7 +183,7 @@ def mpi_send(base_comm: MPI.Comm,
183183def mpi_recv (base_comm : MPI .Comm ,
184184 recv_buf : Optional [NDArray ] = None ,
185185 source : int = 0 ,
186- count : Optional [int ] = None ,
186+ count : Optional [int ] = None ,
187187 tag : int = 0 ,
188188 engine : Optional [str ] = "numpy" ,
189189 ) -> NDArray :
@@ -205,7 +205,7 @@ def mpi_recv(base_comm: MPI.Comm,
205205 Tag of the message to be sent.
206206 engine : :obj:`str`, optional
207207 Engine used to store array (``numpy`` or ``cupy``)
208-
208+
209209 Returns
210210 -------
211211 recv_buf : :obj:`numpy.ndarray` or :obj:`cupy.ndarray`
You can’t perform that action at this time.
0 commit comments