sigpy.Communicator

class sigpy.Communicator[source]

Communicator for distributed computing using MPI.

When NCCL is not installed, arrays are moved to CPU, then communicated through MPI, and moved back to original device. When mpi4py is not installed, the communicator errors.

__init__()[source]

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__() Initialize self.
allreduce(input) All reduce operation in-place.
bcast(input[, root, datatype]) Broadcast from root to other nodes.
gatherv(input[, root]) Gather with variable sizes operation.
reduce(input[, root]) Reduce operation in-place.