add binary tree

In addition, introduce the ability to have 2 trees at the same time.
Only for allreduce at the moment.


[ROCm/rccl commit: 65e2ae20e5]
This commit is contained in:
Edgar Gabriel committed 2022-09-13 20:52:32 +00:00
1 parent 35872115f8
commit 7148c0aa7b
6 files changed
+122 -4

No files matched your search

+7 -1
View File
@@ -567,7 +567,13 @@ comp_next:
// Set nstepsPerLoop and nchunksPerLoop
NCCLCHECK(getPatternInfo(info));
NCCLCHECK(getLoopInfo(info));
if (info->comm->topo->pivotA2ANumBiRings == 3 ) work->pad_0 = 1;
if (info->comm->topo->pivotA2ANumBiRings == 3 ) {
if (ncclTypeSize(info->datatype)*info->count > 65536) {
work->pad_0 = 1;
} else {
work->pad_0 = 2;
}
}
work->opCount = info->opCount;
work->header.type = ncclWorkTypeColl;
work->sendbuff = info->sendbuff;