Update Rome model matching (#461)

* Update Rome model matching

* Add another Rome model

* Automatically setup NET GDR level from model
Este commit está contenido en:
Wenkai Du
2021-11-05 08:53:47 -07:00
cometido por GitHub
padre 04c10a6025
commit 0331e39f81
Se han modificado 8 ficheros con 276 adiciones y 52 borrados
+2
Ver fichero
@@ -531,6 +531,8 @@ ncclResult_t initTransportsRank_1(struct ncclComm* comm, struct allGather1Data_t
//NCCLCHECK(ncclTopoGetSystem(comm, &comm->topo));
// save nRanks to ncclTopoSystem as indicator of multi-node
comm->topo->nRanks = comm->nRanks;
// init netGdrLevel
comm->topo->netGdrLevel = -2;
// Compute paths between GPUs and NICs
NCCLCHECK(ncclTopoComputePaths(comm->topo, comm->peerInfo));
// Remove inaccessible GPUs and unused NICs