Wrapper around mpi4py.
This module defines a light-weight wrapper around mpi4py. It is the only place where we directly import from mpi4py. This allows to potentially swap mpi4py in the future.
T = TypeVar('T')
module-attribute
MPIComm
Bases: Comm
__init__()
Get_rank()
Get_size()
bcast(value, root=0)
barrier()
Barrier()
Scatter(sendbuf, recvbuf, root=0, **kwargs)
Gather(sendbuf, recvbuf, root=0, **kwargs)
allgather(sendobj)
Send(sendbuf, dest, tag=0, **kwargs)
sendrecv(sendbuf, dest, **kwargs)
Isend(sendbuf, dest, tag=0, **kwargs)
Recv(recvbuf, source, tag=0, **kwargs)
Irecv(recvbuf, source, tag=0, **kwargs)
Split(color, key)
allreduce(sendobj, op=ReductionOperator.NO_OP)
Allreduce(sendobj, recvobj, op)
Allreduce_inplace(recvobj, op)