From b08527b0c1ffdbd44347ca3a7869f10b0cb3cbb6 Mon Sep 17 00:00:00 2001 From: Rasmus Munk Larsen Date: Fri, 31 May 2019 15:26:06 -0700 Subject: Clean up CUDA/NVCC version macros and their use in Eigen, and a few other CUDA build failures. --- test/main.h | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) (limited to 'test/main.h') diff --git a/test/main.h b/test/main.h index 1fe631ca9..93e894460 100644 --- a/test/main.h +++ b/test/main.h @@ -52,15 +52,17 @@ #endif // Same for cuda_fp16.h -#if defined(__CUDACC_VER_MAJOR__) && (__CUDACC_VER_MAJOR__ >= 9) -#define EIGEN_TEST_CUDACC_VER ((__CUDACC_VER_MAJOR__ * 10000) + (__CUDACC_VER_MINOR__ * 100)) -#elif defined(__CUDACC_VER__) -#define EIGEN_TEST_CUDACC_VER __CUDACC_VER__ +#if defined(__CUDACC__) && !defined(EIGEN_NO_CUDA) + // Means the compiler is either nvcc or clang with CUDA enabled + #define EIGEN_CUDACC __CUDACC__ +#endif +#if defined(EIGEN_CUDACC) +#include + #define EIGEN_CUDA_SDK_VER (CUDA_VERSION * 10) #else -#define EIGEN_TEST_CUDACC_VER 0 + #define EIGEN_CUDA_SDK_VER 0 #endif - -#if EIGEN_TEST_CUDACC_VER >= 70500 +#if EIGEN_CUDA_SDK_VER >= 70500 #include #endif -- cgit v1.2.3