From 32399955afc00d566d481e56c050bf22dcd11e76 Mon Sep 17 00:00:00 2001
From: Wenkai Du <wenkai.du@amd.com>
Date: Tue, 12 Nov 2019 23:03:59 +0000
Subject: [PATCH] Fix build with RCCL 1.x API

---
 src/common.cu | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/common.cu b/src/common.cu
index af0e2d22dd..5bf78eeebe 100644
--- a/src/common.cu
+++ b/src/common.cu
@@ -853,7 +853,11 @@ testResult_t run() {
     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
+#if NCCL_MAJOR >= 2
     TESTCHECK(InitData(sendbuffs[i], maxBytes, ncclUint8, 0, i));
+#else
+    TESTCHECK(InitData(sendbuffs[i], maxBytes, ncclChar, 0, i));
+#endif
     HIPCHECK(hipDeviceSynchronize());
   }