AI/rocm-systems
2
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'develop' into 2.13.4

Browse Source 
[ROCm/rccl commit: be935d7ce7]
This commit is contained in:
Edgar Gabriel authored and GitHub committed 2022-09-13 17:19:04 -05:00
2 parents 1d24c476f4 3225ee7cd0
commit 4c17f4dcc1
6 files changed
+125 -7
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/rccl/src/enqueue.cc
+7 -2
View File
@@ -1241,8 +1241,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->sendbuff = info->sendbuff;
work->recvbuff = info->recvbuff;
work->root = info->root;