Replace CUDA_VERSION by CUDART_VERSION

[ROCm/rccl commit: c244b51ae7]
2018-12-13 15:22:17 -08:00
parent 6a93dca51b
commit 4e32c7a19a
4 changed files with 6 additions and 6 deletions
@@ -58,7 +58,7 @@ static void* const ncclKerns[ncclCollCount*ncclNumOps*ncclNumTypes*2] = {
 };

 ncclResult_t ncclLaunchCooperativeKernelMultiDevice(struct cudaLaunchParams *paramsList, int* cudaDevs, int numDevices, int cgMode) {
-#if CUDA_VERSION >= 9000
+#if CUDART_VERSION >= 9000
  if (cgMode & 0x01) {
    CUDACHECK(cudaLaunchCooperativeKernelMultiDevice(paramsList, numDevices,
            // These flags are to reduce the latency of using this API