Correct szie when init sendbuff

[ROCm/rccl-tests commit: 83b846cf4f]
Этот коммит содержится в:
Wenkai Du
2020-06-10 23:24:08 +00:00
родитель 21e35fe7f4
Коммит 0b235b136a
+2 -2
Просмотреть файл
@@ -885,9 +885,9 @@ testResult_t run() {
HIPCHECK(hipStreamCreateWithFlags(streams+i, hipStreamNonBlocking));
// initialize data buffer to avoid all zero data
#if NCCL_MAJOR >= 2
TESTCHECK(InitData(sendbuffs[i], maxBytes, ncclUint8, 0, i));
TESTCHECK(InitData(sendbuffs[i], sendBytes, ncclUint8, 0, i));
#else
TESTCHECK(InitData(sendbuffs[i], maxBytes, ncclChar, 0, i));
TESTCHECK(InitData(sendbuffs[i], sendBytes, ncclChar, 0, i));
#endif
HIPCHECK(hipDeviceSynchronize());
}