Change ncclCommRegister size to maxBytes in serial comm init

[ROCm/rccl-tests commit: d028efcf35]
Αυτή η υποβολή περιλαμβάνεται σε:
Kaiming Ouyang authored and Sylvain Jeaugey committed 2024-06-06 06:54:48 -07:00
1 γονέας 411a7912f8
υποβολή 1922bd71cb
1 file changed
+2 -2
@@ -1000,8 +1000,8 @@ testResult_t run() {
sendRegHandles = (local_register) ? (void **)malloc(sizeof(*sendRegHandles)*nThreads*nGpus) : NULL;
recvRegHandles = (local_register) ? (void **)malloc(sizeof(*recvRegHandles)*nThreads*nGpus) : NULL;
for (int i=0; i<nGpus*nThreads; i++) {
if (local_register) NCCLCHECK(ncclCommRegister(comms[i], sendbuffs[i], sendBytes, &sendRegHandles[i]));
if (local_register) NCCLCHECK(ncclCommRegister(comms[i], recvbuffs[i], recvBytes, &recvRegHandles[i]));
if (local_register) NCCLCHECK(ncclCommRegister(comms[i], sendbuffs[i], maxBytes, &sendRegHandles[i]));
if (local_register) NCCLCHECK(ncclCommRegister(comms[i], recvbuffs[i], maxBytes, &recvRegHandles[i]));
}
#endif
}