Skip to content

Commit 4ae63fd

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent c3522e8 commit 4ae63fd

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

pytorch_lightning/plugins/collective/horovod_collective.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,8 @@ def __init__(
3939
def join(self) -> None:
4040
"""Horovod function that indicates that the rank finished processing data.
4141
42-
All ranks that did not call join() continue to process allreduce operations. This function blocks the Python thread
43-
until all ranks join.
42+
All ranks that did not call join() continue to process allreduce operations. This function blocks the Python
43+
thread until all ranks join.
4444
"""
4545
if self.on_gpu:
4646
hvd.join(self.local_rank)

pytorch_lightning/plugins/collective/torch_collective.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ def reduce(
8484
If local_reduce = True (dp and ddp2), reduces tensor from all local processes.
8585
8686
If local_reduce = False (ddp, ddpspawning and extentions), reduces a tensor from several distributed processes
87-
87+
8888
Args:
8989
tensor: the tensor to sync and reduce
9090
group: the process group to gather results from. Defaults to all processes (world)

0 commit comments

Comments
 (0)