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

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

Browse Source 
* topo_expl: fix build after switching to rocm-smi-lib

* Use minimal of 4 channels for gfx908

[ROCm/rccl commit: 135d47d125]
This commit is contained in:
Wenkai Du
2021-07-27 08:30:08 -07:00
committed by GitHub
parent 3d382b5ba3
commit 8fbeb14175
4 changed files with 20 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/tools/topo_expl/topo_expl.cpp
+2 -2
View 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));