introduce a hw topology aware bintree

for hayabusa architecture.


[ROCm/rccl commit: e645b02cd8]
This commit is contained in:
Edgar Gabriel committed 2022-10-03 15:26:21 +00:00
1 parent dbbdf65020
commit f2736a4fb3
4 files changed
+204 -2

No files matched your search

+1 -1
View File
@@ -1224,7 +1224,7 @@ comp_next:
NCCLCHECK(getPatternInfo(info));
NCCLCHECK(getLoopInfo(info));
if (info->comm->topo->pivotA2ANumBiRings == 3 ) {
if (ncclTypeSize(info->datatype)*info->count > 65536) {
if (ncclTypeSize(info->datatype)*info->count > 131072) {
work->pad_0 = 1;
} else {
work->pad_0 = 2;