AI/rocm-systems
2
0
Forka 0
Codice Problemi Pull Request Actions Pacchetti Progetti Rilasci Wiki Attività

topo_expl: fix build after switching to rocm-smi-lib (#405)

Sfoglia il codice sorgente 
* topo_expl: fix build after switching to rocm-smi-lib

* Use minimal of 4 channels for gfx908
This commit is contained in:
Wenkai Du
2021-07-27 08:30:08 -07:00
committato da GitHub
parent 56801656f3
commit 135d47d125
4 ha cambiato i file con 20 aggiunte e 3 eliminazioni
Scarica il file Patch Scarica il file Diff Expand all files Collapse all files
tools/topo_expl/topo_expl.cpp
+2 -2
Vedi File
@@ -184,8 +184,8 @@ int main(int argc,char* argv[])
for (int i = 0; i < nranks; i++) {
comm[i].rank = i;
comm[i].nRanks = nranks;
NCCLCHECK(ncclCalloc(&comm[i].connectSend, comm->nRanks));
NCCLCHECK(ncclCalloc(&comm[i].connectRecv, comm->nRanks));
NCCLCHECK(ncclCalloc(&comm[i].connectSend, NCCL_MAX_CONNS*comm->nRanks));
NCCLCHECK(ncclCalloc(&comm[i].connectRecv, NCCL_MAX_CONNS*comm->nRanks));
comm[i].p2pSendCount = comm[i].p2pRecvCount = 0;
NCCLCHECK(ncclCalloc(&comm[i].p2pSends, comm->nRanks));
NCCLCHECK(ncclCalloc(&comm[i].p2pRecvs, comm->nRanks));