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);