From 0b235b136aea2040194638de1e050fb6be9985cb Mon Sep 17 00:00:00 2001 From: Wenkai Du Date: Wed, 10 Jun 2020 23:24:08 +0000 Subject: [PATCH] Correct szie when init sendbuff [ROCm/rccl-tests commit: 83b846cf4fb8e16f6e07f5577168c40db7173e65] --- projects/rccl-tests/src/common.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/projects/rccl-tests/src/common.cu b/projects/rccl-tests/src/common.cu index 908a69b9c5..ed88f51905 100644 --- a/projects/rccl-tests/src/common.cu +++ b/projects/rccl-tests/src/common.cu @@ -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()); }