ΑiSSKJr SSKJrJr SSKrSSKJr SSKJ r J r SSK J r \(aSSKJ r SSKJr SS KJr S SS jjrSSS jjrSSS jjrg)) annotations) TYPE_CHECKINGAnyN)stream)_get_global_group_warn_cur_rank_not_in_group)convert_tensor_to_object)Tensor)task)Groupc.[R"XX#SS9$)a Receive a tensor to the sender. Args: 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. sync_op (bool, optional): Whether this op is a sync op. The default value is True. Returns: Return a task object. Examples: .. code-block:: python >>> # doctest: +REQUIRES(env: DISTRIBUTED) >>> import paddle >>> import paddle.distributed as dist >>> dist.init_parallel_env() >>> if dist.get_rank() == 0: ... data = paddle.to_tensor([7, 8, 9]) ... dist.send(data, dst=1) >>> else: ... data = paddle.to_tensor([1, 2, 3]) ... dist.recv(data, src=0) >>> print(data) >>> # [7, 8, 9] (2 GPUs) F)srcgroupsync_opuse_calc_stream)rrecv)tensorrrrs e/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/distributed/communication/recv.pyrr#sH ;;uu c[XUSS9$)a Receive a tensor to the sender. Args: 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: .. code-block:: python >>> # doctest: +REQUIRES(env: DISTRIBUTED) >>> 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() # type: ignore[union-attr] >>> print(data) >>> # [7, 8, 9] (2 GPUs) F)r)r)rrrs rirecvrLsH UE 22rcBUb[U5S:Xa [S5eUc [5OUn[U5(agUb Ub [S5eUR U5nOUcSOUn[ R "[U54SS9n[XAUS9 [ R"U5R5n[ R "U4[ RS9n[XaUS9 Sn[U5H,upU R5n XgXy-n [X5X'Xy- nM. g)a! Receive a list of Python objects from the sender. Args: object_list (list): The list to store received objects. Must be pre-allocated with correct size. src (int, optional): The source rank id. Default: 0. group (Group, optional): The group instance return by new_group or None for global default group. Default: None. src_in_group (int, optional): The source rank within the group. Cannot be specified together with src. Default: None. Returns: This function does not return any value. Examples: .. code-block:: python >>> # doctest: +REQUIRES(env: DISTRIBUTED) >>> import paddle >>> import paddle.distributed as dist >>> dist.init_parallel_env() >>> if dist.get_rank() == 0: ... data = ["hello", {"key": 100}, [1, 2, 3]] ... dist.send_object_list(data, dst=1) >>> else: ... data = [None] * 3 # type: ignore ... dist.recv_object_list(data, src=0) >>> print(data) >>> # ["hello", {"key": 100}, [1, 2, 3]] (2 GPUs) Nrz#object_list cannot be None or emptyz7Cannot specify both 'src' and 'src_in_group' arguments.int64)dtype)rr) len ValueErrorrrget_global_rankpaddleemptyrsumitemuint8 enumerater ) object_listrr src_in_groupobject_sizes_tensor total_size object_tensoroffsetiobj_sizeobj_views rrecv_object_listr-ssFc+.!3>??#(=  eE"5)) ?I ##L1;aC ,,K(8':'J U3/0557JLL*fllCMu- F !45 ==? &*;<1(E  6r)rNT) rr rintr Group | Nonerboolreturnr )NN)rr r int | Nonerr/r1r )NNN)r$z list[Any]rr2rr/r%r2) __future__rtypingrrr paddle.distributed.communicationr&paddle.distributed.communication.grouprr4paddle.distributed.communication.serialization_utilsr r paddle.base.corer r rrr-rrr:s#% 3%<  & & & & &  &TCG$3 $3#$33?$3 $3R# ?? ? ? ?r