Change __CUDACC_VER_*__ preprocessor directives to CUDA_VERSION because clang doesn't define the former.

[ROCm/rccl commit: d08e9b5279]
This commit is contained in:
Christian Sigg
2018-12-13 15:59:22 +01:00
committed by Sylvain Jeaugey
parent 7fa81a40a1
commit 457c2ab5ae
4 changed files with 6 additions and 6 deletions
+1 -1
View File
@@ -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 __CUDACC_VER_MAJOR__ >= 10
#if CUDA_VERSION >= 10000
if (attr.type == cudaMemoryTypeDevice && attr.device != comm->cudaDev) {
#else
if (attr.memoryType == cudaMemoryTypeDevice && attr.device != comm->cudaDev) {