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

Uses nullptr as send buffer for non-root ranks during scatter

Browse Source 
[ROCm/rccl-tests commit: bf4a866109]
This commit is contained in:
Wenkai Du committed 2020-10-19 14:06:23 -04:00
1 parent 17ed1e67fa
commit 625bb2bc98
1 file changed
+4 -1
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl-tests/src/scatter.cu
+4 -1
View File
@@ -86,7 +86,10 @@ testResult_t ScatterRunColl(void* sendbuff, void* recvbuff, size_t count, ncclDa
return testNcclError;
#else
#if defined(RCCL_GATHER_SCATTER) && defined(USE_RCCL_GATHER_SCATTER)
NCCLCHECK(ncclScatter(sendbuff, recvbuff, count, type, root, comm, stream));
if (rank == root)
NCCLCHECK(ncclScatter(sendbuff, recvbuff, count, type, root, comm, stream));
else
NCCLCHECK(ncclScatter(0, recvbuff, count, type, root, comm, stream));
#else
NCCLCHECK(ncclGroupStart());
if (rank == root) {