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:
Wenkai Du údaraithe agus GitHub tiomanta 2024-10-11 10:40:32 -07:00
1 tuismitheoir 4cd1b3a9f5
tiomantas 09acdb6b49
2 comhad athraithe
+4 -4

Níor tháinig aon chomhaid le do chuardach

+2 -2
Féach ar an gComhad
@@ -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;
}