megengine.functional.distributed.remote_send

remote_send(inp, dest_rank)

Send tensor to another process.

Parameters:

• inp (Tensor) – Tensor to send.

• dest_rank (int) – Rank of destination process.

Returns:

None.

Examples

if rank == 0:
    data = mge.tensor(1)
    # Tensor(1)
    F.distributed.remote_send(data, 1) # return None
else:
    data = F.distributed.remote_recv(0)
    # Tensor(1)