introduce a hw topology aware bintree
for hayabusa architecture.
[ROCm/rccl commit: e645b02cd8]
This commit is contained in:
1 parent
dbbdf65020
commit
f2736a4fb3
4 files changed
+204
-2
No files matched your search
@@ -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;
|
||||
|
||||
Reference in new issue
Block a user