etha.comm.transfer

Transfer operation types and execution.

Classes

TransferType	Transfer operation types.
Transferable	Base class for transferable objects (chunks and buckets).

Module Contents

class etha.comm.transfer.TransferType(*args, **kwds)

Bases: enum.Enum

Transfer operation types.

BROADCAST = 'broadcast'

P2P = 'p2p'

SELF_COPY = 'self_copy'

SHADOW = 'shadow'

class etha.comm.transfer.Transferable

Base class for transferable objects (chunks and buckets).

execute() → torch.distributed.Work | None

Execute transfer operation.

Returns:

Work handle for async operations, None for SELF_COPY / SHADOW.

buffer: torch.Tensor | None = None

dst_ranks: tuple[int, Ellipsis]

is_source: bool

src_rank: int

transfer_type: TransferType

work: torch.distributed.Work | None = None