irecv¶

paddle.distributed. irecv ( tensor, src=None, group=None ) [source]

Receive a tensor to the sender.

Parameters

• tensor (Tensor) – The Tensor to receive. Its data type should be float16, float32, float64, int32, int64, int8, uint8, bool or bfloat16.

• src (int) – The source rank id.

• group (Group, optional) – The group instance return by new_group or None for global default group. Default: None.

Returns

Return a task object.

Warning

This API only supports the dygraph mode.

Examples

>>> 
>>> import paddle
>>> import paddle.distributed as dist

>>> dist.init_parallel_env()
>>> if dist.get_rank() == 0:
...     data = paddle.to_tensor([7, 8, 9])
...     task = dist.isend(data, dst=1)
>>> else:
...     data = paddle.to_tensor([1, 2, 3])
...     task = dist.irecv(data, src=0)
>>> task.wait()
>>> print(data)
>>> # [7, 8, 9] (2 GPUs)