AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Transport] Fix IntraNet (#1582)

Browse Source
This commit is contained in:
Bertan Dogancay
2025-03-04 13:30:36 -05:00
committed by GitHub
parent f957c4fe22
commit d88cca3098
2 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/init.cc
+11
View File
@@ -1631,6 +1631,17 @@ static ncclResult_t initTransportsRank(struct ncclComm* comm, struct ncclComm* p
}
NCCLCHECKGOTO(ncclTransportRingConnect(comm), ret, fail);
// Connect NET for intranode use
if (comm->graphs[NCCL_ALGO_RING].nIntraChannels && rcclParamP2pNetDisable() == 0) {
comm->useIntraNet = 1;
for (int c = 0; c < comm->nChannels; c++) {
struct ncclChannel* channel = comm->channels+c;
if (comm->nRanks == 1) continue;
NCCLCHECKGOTO(ncclTransportP2pConnect(comm, c, 1, &channel->ring.prev, 1, &channel->ring.next, NCCL_CONN_IDX_P2P_NET), ret, fail);
}
NCCLCHECKGOTO(ncclTransportP2pSetup(comm, &comm->graphs[NCCL_ALGO_RING], NCCL_CONN_IDX_P2P_NET), ret, fail);
}
// Connect Trees
NCCLCHECKGOTO(ncclTransportTreeConnect(comm), ret, fail);