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)

previous

megengine.functional.distributed.collective_comm

next

megengine.functional.distributed.remote_recv