megengine.functional.distributed.remote_send¶

remote_send(inp, dest_rank)[源代码]¶

发送张量到另一个进程。

参数

inp (Tensor) – 需要发送的张量。
dest_rank (int) – 目标进程的编号。

返回

None.

实际案例

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)

megengine.functional.distributed.collective_comm

megengine.functional.distributed.remote_recv