@@ -478,7 +478,8 @@ def is_initialized():
478478
479479 Check whether the distributed environment has been initialized
480480
481- Returns (bool): `True` if distributed environment has been initialized, otherwise `False`.
481+ Returns:
482+ `True` if distributed environment has been initialized, otherwise `False`.
482483
483484 Examples:
484485 .. code-block:: python
@@ -1594,7 +1595,7 @@ def batch_isend_irecv(p2p_op_list):
15941595 corresponding tasks. NCCL are currently supported.
15951596
15961597 Args:
1597- p2p_op_list: A list of point-to-point operations(type of each operator is
1598+ p2p_op_list (List[P2POp]) : A list of point-to-point operations(type of each operator is
15981599 ``paddle.distributed.P2POp``). The order of the isend/irecv in the list
15991600 matters and it needs to match with corresponding isend/irecv on the
16001601 remote end.
@@ -1737,7 +1738,7 @@ def _reduce_scatter_base(output,
17371738
17381739 Args:
17391740 output (Tensor): Output tensor. Its data type should be float16, float32, float64, int32, int64, int8, uint8, bool or bfloat16.
1740- input (Tensor): Input tensor that is of size output tensor size times world size. Its data type
1741+ input (Tensor): Input tensor that is of size output tensor size times world size. Its data type
17411742 should be float16, float32, float64, int32, int64, int8, uint8, bool or bfloat16.
17421743 op (ReduceOp.SUM|ReduceOp.MAX|ReduceOp.MIN|ReduceOp.PROD): Optional. The operation used. Default: ReduceOp.SUM.
17431744 group (ProcessGroup, optional): The process group to work on. If None,
0 commit comments