Replace CUDA_VERSION by CUDART_VERSION

[ROCm/rccl commit: c244b51ae7]

projects/rccl/src/include/common_coll.h

@@ -18,7 +18,7 @@ static ncclResult_t PointerCheck(const void* pointer, struct ncclComm* comm, con
     WARN("%s : %s is not a valid pointer", opname, ptrname);
     return ncclInvalidArgument;
   }
-#if CUDA_VERSION >= 10000
+#if CUDART_VERSION >= 10000
   if (attr.type == cudaMemoryTypeDevice && attr.device != comm->cudaDev) {
 #else
   if (attr.memoryType == cudaMemoryTypeDevice && attr.device != comm->cudaDev) {

projects/rccl/src/include/core.h

@@ -18,7 +18,7 @@
 #include <stdlib.h>
 #include <cuda_runtime.h>
 
-#if CUDA_VERSION < 9000
+#if CUDART_VERSION < 9000
 struct cudaLaunchParams {
   void *func;
   dim3 gridDim;