Merge remote-tracking branch 'nccl/master' into develop

This commit is contained in:
BertanDogancay
2024-10-02 09:29:22 -05:00
bovenliggende e453f1ced9 ab2b89c4c3
commit 84081064a0
54 gewijzigde bestanden met toevoegingen van 2158 en 964 verwijderingen
+2 -2
Bestand weergeven
@@ -342,8 +342,8 @@ static ncclResult_t selectTransport(struct ncclComm* comm, struct ncclTopoGraph*
// handle intra-node network connections
int n1 = -1, n2 = -1;
if (connIndex == NCCL_CONN_IDX_P2P_NET) {
NCCLCHECK(ncclTopoGetIntraNetDev(comm->topo, comm->rank, graph, channelId, (type == 1) ? 1 : 0, &n1));
NCCLCHECK(ncclTopoGetIntraNetDev(comm->topo, peer, graph, channelId, (type == 1) ? 0 : 1, &n2));
NCCLCHECK(ncclTopoGetIntraNetDev(comm->topo, comm->rank, graph, channelId, (type == 1) ? 1 : 0, nullptr, &n1));
NCCLCHECK(ncclTopoGetIntraNetDev(comm->topo, peer, graph, channelId, (type == 1) ? 0 : 1, nullptr, &n2));
}
bool xgmi;
NCCLCHECK(ncclTopoGetLinkType(comm->topo, myInfo->cudaDev, peerInfo->cudaDev, &xgmi));