From 5cc7ff0931ab91e166329ae367fbf32ecdb17a5f Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Tue, 9 Aug 2022 11:04:38 -0700
Subject: [PATCH] Fix missing error checking for AllocateBuffs due to merge
 (#17)

[ROCm/rccl-tests commit: 9025051bbb62fecc15429876d5d7543fad370ec0]
---
 projects/rccl-tests/src/common.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/rccl-tests/src/common.cu b/projects/rccl-tests/src/common.cu
index c31cff308e..a4577550ba 100644
--- a/projects/rccl-tests/src/common.cu
+++ b/projects/rccl-tests/src/common.cu
@@ -1250,7 +1250,7 @@ testResult_t run() {
     HIPCHECK(hipSetDevice(gpuid));
     for (int j=0; j<ranksPerGpu; j++) {
       int i = ii*ranksPerGpu+j;
-      AllocateBuffs(sendbuffs+i, sendBytes, recvbuffs+i, recvBytes, expected+i, (size_t)maxBytes, nProcs*nThreads*nGpus*ranksPerGpu);
+      TESTCHECK(AllocateBuffs(sendbuffs+i, sendBytes, recvbuffs+i, recvBytes, expected+i, (size_t)maxBytes, nProcs*nThreads*nGpus*ranksPerGpu));
       //PRINT("sendbuffs[%d]=%p(size=%lu) recvbuffs[%d]=%p(size=%lu)\n", i, sendbuffs[i], sendBytes, i, recvbuffs[i], recvBytes);
       if (cumask[0] || cumask[1] || cumask[2] || cumask[3]) {
 	PRINT("cumask: ");