Change ncclCommRegister size to maxBytes in serial comm init

[ROCm/rccl-tests commit: d028efcf35]
이 커밋은 다음에 포함됨:
Kaiming Ouyang 작성하고 Sylvain Jeaugey가 2024-06-06 06:54:48 -07:00를 커밋함
1 부모 411a7912f8
커밋 1922bd71cb
1 파일이 변경됨
+2 -2
+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
}