megengine.functional.distributed.remote_recv¶

remote_recv(src_rank, device=None, inp=None)[源代码]¶

从另一个进程接收张量。

参数

src_rank (int) – 源进程的编号
device (Optional[str]) – 执行此操作的设备。默认为输入张量所在的设备。可以通过指定设备为 ”gpu0:1“ 以在不同的 cuda 流上执行此操作，其中1是 cuda 流的编号，默认 cuda 流编号为0。
inp – 确定接收张量类型的模拟输入。

返回类型

Tensor

返回

被接收的张量。

例如：

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.remote_send

megengine.functional.distributed.reduce_sum