TensorPlay
API symbolsdistributed
Copy
View MarkdownDownload .md

Work

class tensorplay.distributed.distributed_c10d.Work(event, done=None, tensors=None)[source]

Handle for an async collective (torch.distributed.Work subset).

get_future() mirrors c10d: it returns a tensorplay.futures.Future resolving to the list of output tensors once the collective completes.

Ask DeepWiki