TensorPlay
API symbolsdistributed
Copy
View MarkdownDownload .md

P2POp

class tensorplay.distributed.distributed_c10d.P2POp(op, tensor, peer: int | None = None, group=None, tag: int = 0, group_peer: int | None = None)[source]

A class to build point-to-point operations for batch_isend_irecv.

Args mirror torch: op is isend() or irecv(), peer is a global rank (or group_peer a group rank).

Ask DeepWiki