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

NCCL_TREES variable and rome model fixes (#856)

Browse Source
This commit is contained in:
akolliasAMD
2023-08-21 10:35:37 -06:00
committed by GitHub
parent 148e3430f4
commit d33cd5a233
8 changed files with 58 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/topo_expl/utils.cpp
+1 -1
View File
@@ -1019,7 +1019,7 @@ ncclResult_t initTransportsRank_3(struct ncclComm* comm, struct allGatherInfo *a
NCCLCHECKGOTO(ncclCalloc(&rings, nranks*MAXCHANNELS), ret, fail);
NCCLCHECKGOTO(ncclTopoPostset(comm, nodesFirstRank, nodesTreePatterns, allTopoRanks, rings, graphs, nc), ret, fail);
if (comm->topo->pivotA2ANumBiRings == 3) NCCLCHECK(ncclTreeBasePostset(comm, &treeGraph));
if (comm->topo->treeDefined) NCCLCHECK(ncclTreeBasePostset(comm, &treeGraph));
// AllGather3 - end