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

Bu işleme şunda yer alıyor:
Wenkai Du
2020-10-19 14:06:23 -04:00
ebeveyn 4c0ec7347c
işleme bf4a866109
+4 -1
Dosyayı Görüntüle
@@ -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) {