Enable local sendrecv over network if GDR is available on all GPUs (#324)

Tá an tiomantas seo le fáil i:
Wenkai Du údaraithe agus GitHub tiomanta 2021-03-05 19:59:41 -08:00
1 tuismitheoir f4a9b9acba
tiomantas c018edf0f2
14 comhad athraithe
+163 -44

Níor tháinig aon chomhaid le do chuardach

+1
Féach ar an gComhad
@@ -207,6 +207,7 @@ ncclResult_t initTransportsRank_1(struct ncclComm* comm, struct allGather1Data_t
// Topo detection / System graph creation
//NCCLCHECK(ncclTopoGetSystem(comm, &comm->topo));
comm->topo->nRanks = comm->nRanks;
// Compute paths between GPUs and NICs
NCCLCHECK(ncclTopoComputePaths(comm->topo, comm->peerInfo));
// Remove inaccessible GPUs and unused NICs