Allow zero byte sendrecv in alltoallv (#1349)
* Allow zero byte sendrecv in alltoallv
* Fix previous merge error
[ROCm/rccl commit: 821d2e1f30]
Tá an tiomantas seo le fáil i:
1 tuismitheoir
4cd1b3a9f5
tiomantas
09acdb6b49
2 comhad athraithe
+4
-4
Níor tháinig aon chomhaid le do chuardach
@@ -609,7 +609,7 @@ static ncclResult_t addP2pToPlan(
|
||||
|
||||
struct ncclWorkElemP2p elem = {0};
|
||||
elem.proto = info.protocol;
|
||||
elem.peer = addr == nullptr ? -1 : peer;
|
||||
elem.peer = peer;
|
||||
elem.nWarps = NCCL_MAX_NTHREADS/comm->WarpSize;
|
||||
elem.reg = proxyOp.reg;
|
||||
elem.p2pType = isSendNotRecv ? ncclWorkP2pTypeSend : ncclWorkP2pTypeRecv;
|
||||
@@ -628,7 +628,7 @@ static ncclResult_t addP2pToPlan(
|
||||
// Calculate the opCount after appendWorkElemP2p since it will always return
|
||||
// with channel->nWork equal to one plus the work index this p2p settled in.
|
||||
proxyOp.opCount = uint64_t(plan->channels[channelId].nWork)<<1 | 1;
|
||||
if (addr != nullptr) NCCLCHECK(addProxyOpIfNeeded(comm, plan, &proxyOp));
|
||||
NCCLCHECK(addProxyOpIfNeeded(comm, plan, &proxyOp));
|
||||
return ncclSuccess;
|
||||
}
|
||||
|
||||
|
||||
Tagairt in Eagrán Nua
Cuir bac ar úsáideoir