microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-04-19 14:29:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adjusting Communicator in Python API (#752)

Browse Source
This commit is contained in:
Caio Rocha
2026-02-23 16:33:52 -08:00
committed by GitHub
parent b5256032fe
commit 7738603d63
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/mscclpp/_core/comm.py
View File

@@ -38,6 +38,11 @@ class CommGroup:
):
if interfaceIpPortTrio == "" and (mpi_comm is not None or torch_group is not None):
uniq_id = None
rank, size = (
(mpi_comm.Get_rank(), mpi_comm.Get_size())
if mpi_comm is not None
else (torch_group.rank(), torch_group.size())
)
self.bootstrap = CppTcpBootstrap.create(rank, size)
if rank == 0:
uniq_id = self.bootstrap.create_unique_id()