Init data for throughput iterations to avoid all zero data

이 커밋은 다음에 포함됨:
Wenkai Du
2019-09-25 14:07:04 -07:00
부모 ab82f1af6f
커밋 13d0ddd12e
+3
파일 보기
@@ -851,6 +851,9 @@ testResult_t run() {
HIPCHECK(hipSetDevice(localRank*nThreads*nGpus+i));
AllocateBuffs(sendbuffs+i, sendBytes, recvbuffs+i, recvBytes, expected+i, (size_t)maxBytes, nProcs*nThreads*nGpus);
HIPCHECK(hipStreamCreateWithFlags(streams+i, hipStreamNonBlocking));
// initialize data buffer to avoid all zero data
TESTCHECK(InitData(sendbuffs[i], maxBytes, ncclUint8, 0, i));
HIPCHECK(hipDeviceSynchronize());
}
//if parallel init is not selected, use main thread to initialize NCCL