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

Этот коммит содержится в:
Christian Sigg
2018-12-13 15:59:22 +01:00
коммит произвёл Sylvain Jeaugey
родитель 469b69a5d0
Коммит d08e9b5279
4 изменённых файлов: 6 добавлений и 6 удалений
+1 -1
Просмотреть файл
@@ -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) {
+1 -1
Просмотреть файл
@@ -18,7 +18,7 @@
#include <stdlib.h>
#include <cuda_runtime.h>
#if __CUDACC_VER_MAJOR__ < 9
#if CUDA_VERSION < 9000
struct cudaLaunchParams {
void *func;
dim3 gridDim;