1
0
mirror of https://github.com/paboyle/Grid.git synced 2024-11-10 15:55:37 +00:00

Fix CUDA_ARCH. Need to simplify. See when new eigen release happens

This commit is contained in:
Peter Boyle 2018-07-23 06:09:33 -04:00
parent a0714de8ec
commit b35401b86b

View File

@ -6,25 +6,42 @@
#endif
#ifdef __NVCC__
#pragma push
#pragma diag_suppress code_is_unreachable
#define __NVCC__REDEFINE__
#undef __NVCC__
#undef __CUDACC__
#ifdef __CUDA_ARCH__
#define __CUDA_ARCH__REDEFINE__
#define __CUDA_ARCH__REDEFINE__ 1
#define __CUDA_ARCH_SAVE__ __CUDA_ARCH__
#undef __CUDA_ARCH__
#warning "Hiding CUDA_ARCH from Eigen"
#endif
#endif
#include <Grid/Eigen/Dense>
#ifdef __NVCC__
#ifdef __NVCC__REDEFINE__
#pragma pop
#define __NVCC__
#define __CUDACC__
#ifdef __CUDA_ARCH__REDEFINE__
#define __CUDA_ARCH__
#define __CUDA_ARCH__ __CUDA_ARCH_SAVE__
#warning "redefined CUDA_ARCH outside Eigen"
#endif
#endif