Added creation of new tree and added switch for using treesplit for specific cases (#551)

[ROCm/rccl commit: 98f0809a39]
这个提交包含在:
akolliasAMD 编写, GitHub 于 2022-05-25 18:55:14 -04:00 提交
1 父节点 a2a4888497
当前提交 22dc8bd246
8 个文件已更改
+229 -6

没有与搜索条件相匹配的文件

+2
查看文件
@@ -789,6 +789,8 @@ ncclResult_t initTransportsRank_3(struct ncclComm* comm, struct allGather3Data_t
NCCLCHECK(ncclCalloc(&rings, nranks*MAXCHANNELS));
NCCLCHECK(ncclTopoPostset(comm, nodesFirstRank, nodesTreePatterns, allTopoRanks, rings, &collNetGraph, nc));
if (comm->topo->pivotA2ANumBiRings == 3) NCCLCHECK(ncclTreeBasePostset(comm, &treeGraph));
free(allTopoRanks);
free(nodesTreePatterns);
free(nodesFirstRank);