Skip to content

Commit 3ecaba1

Browse files
committed
Wrap init_process_group method since it is too long in built doc
1 parent ba05137 commit 3ecaba1

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

intermediate_source/ddp_tutorial.rst

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,12 @@ be found in
8787
init_method="file:///{your local file path}"
8888
8989
# initialize the process group
90-
dist.init_process_group("gloo", init_method=init_method, rank=rank, world_size=world_size)
90+
dist.init_process_group(
91+
"gloo",
92+
init_method=init_method,
93+
rank=rank,
94+
world_size=world_size
95+
)
9196
else:
9297
os.environ['MASTER_ADDR'] = 'localhost'
9398
os.environ['MASTER_PORT'] = '12355'

0 commit comments

Comments
 (0)