Fix build with RCCL 1.x API

[ROCm/rccl-tests commit: 32399955af]
2019-11-12 23:03:59 +00:00
parent 0c07116217
commit 123ea97750
1 changed files with 4 additions and 0 deletions
@@ -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());
  }