Skip to content

Commit

Permalink
black changes
Browse files Browse the repository at this point in the history
Signed-off-by: arunjose696 <[email protected]>
  • Loading branch information
arunjose696 committed Nov 9, 2023
1 parent 12f82ce commit 5dca897
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion unidist/core/backends/mpi/core/communication.py
Original file line number Diff line number Diff line change
Expand Up @@ -563,7 +563,9 @@ def mpi_recv_buffer(comm, source_rank, result_buffer=None):
for i in range(num_partitions):
if i + 1 < num_partitions:
tmp_buffer = bytearray(partitions[i + 1] - partitions[i])
comm.Recv(bigmpi(tmp_buffer), source=source_rank, tag=common.MPITag.BUFFER)
comm.Recv(
bigmpi(tmp_buffer), source=source_rank, tag=common.MPITag.BUFFER
)
result_buffer[partitions[i] : partitions[i + 1]] = tmp_buffer

return result_buffer
Expand Down

0 comments on commit 5dca897

Please sign in to comment.