comm_abc

T = TypeVar('T') module-attribute

ReductionOperator

Bases: Enum

OP_NULL = enum.auto() class-attribute instance-attribute

MAX = enum.auto() class-attribute instance-attribute

MIN = enum.auto() class-attribute instance-attribute

SUM = enum.auto() class-attribute instance-attribute

PROD = enum.auto() class-attribute instance-attribute

LAND = enum.auto() class-attribute instance-attribute

BAND = enum.auto() class-attribute instance-attribute

LOR = enum.auto() class-attribute instance-attribute

BOR = enum.auto() class-attribute instance-attribute

LXOR = enum.auto() class-attribute instance-attribute

BXOR = enum.auto() class-attribute instance-attribute

MAXLOC = enum.auto() class-attribute instance-attribute

MINLOC = enum.auto() class-attribute instance-attribute

REPLACE = enum.auto() class-attribute instance-attribute

NO_OP = enum.auto() class-attribute instance-attribute

Request

Bases: ABC

wait() abstractmethod

Comm

Bases: ABC, Generic[T]

Get_rank() abstractmethod

Get_size() abstractmethod

bcast(value, root=0) abstractmethod

barrier() abstractmethod

Barrier() abstractmethod

Scatter(sendbuf, recvbuf, root=0, **kwargs) abstractmethod

Gather(sendbuf, recvbuf, root=0, **kwargs) abstractmethod

allgather(sendobj) abstractmethod

Send(sendbuf, dest, tag=0, **kwargs) abstractmethod

sendrecv(sendbuf, dest, **kwargs) abstractmethod

Isend(sendbuf, dest, tag=0, **kwargs) abstractmethod

Recv(recvbuf, source, tag=0, **kwargs) abstractmethod

Irecv(recvbuf, source, tag=0, **kwargs) abstractmethod

Split(color, key) abstractmethod

allreduce(sendobj, op=ReductionOperator.NO_OP) abstractmethod

Allreduce(sendobj, recvobj, op) abstractmethod

Allreduce_inplace(obj, op) abstractmethod