Skip to content

Commit 86ef055

Browse files
committed
Check for double init
1 parent b8a1ce6 commit 86ef055

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

experiments/OGB/main.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,8 @@ def main(
329329
comm = SingleProcessDummyCommunicator()
330330

331331
else:
332-
dist.init_process_group(backend="nccl")
332+
if not dist.is_initialized():
333+
dist.init_process_group(backend="nccl")
333334
comm = Communicator.init_process_group(_communicator)
334335

335336
# Must pass the node rank placement file the first time

0 commit comments

Comments
 (0)