Skip to content

Commit 911d02e

Browse files
committed
style(python/distributed/collective): please newer codestyle
1 parent 74d862a commit 911d02e

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

python/paddle/distributed/collective.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)