diff --git a/lib/Grid_Eigen_Dense.h b/lib/Grid_Eigen_Dense.h index 05a3d791..dcbfa40b 100644 --- a/lib/Grid_Eigen_Dense.h +++ b/lib/Grid_Eigen_Dense.h @@ -10,8 +10,11 @@ #pragma diag_suppress code_is_unreachable #undef __NVCC__ #undef __CUDACC__ +#ifdef __CUDA_ARCH__ +#define __CUDA_ARCH__REDEFINE__ #undef __CUDA_ARCH__ #endif +#endif #include @@ -19,8 +22,11 @@ #pragma pop #define __NVCC__ #define __CUDACC__ + +#ifdef __CUDA_ARCH__REDEFINE__ #define __CUDA_ARCH__ #endif +#endif #if defined __GNUC__ #pragma GCC diagnostic pop