mirror of
https://github.com/paboyle/Grid.git
synced 2025-04-07 20:50:46 +01:00
42 lines
880 B
C
42 lines
880 B
C
#include <Grid/GridCore.h>
|
|
#pragma once
|
|
// Force Eigen to use MKL if Grid has been configured with --enable-mkl
|
|
#ifdef USE_MKL
|
|
#define EIGEN_USE_MKL_ALL
|
|
#endif
|
|
|
|
|
|
#if defined __GNUC__
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
|
|
#endif
|
|
|
|
/* NVCC save and restore compile environment*/
|
|
#ifdef __NVCC__
|
|
#pragma push
|
|
#pragma diag_suppress code_is_unreachable
|
|
#pragma push_macro("__CUDA_ARCH__")
|
|
#pragma push_macro("__NVCC__")
|
|
#pragma push_macro("__CUDACC__")
|
|
#undef __NVCC__
|
|
#undef __CUDACC__
|
|
#undef __CUDA_ARCH__
|
|
#define __NVCC__REDEFINE__
|
|
#endif
|
|
|
|
#include <Grid/Eigen/Dense>
|
|
#include <Grid/Eigen/unsupported/CXX11/Tensor>
|
|
|
|
/* NVCC restore */
|
|
#ifdef __NVCC__REDEFINE__
|
|
#pragma pop_macro("__CUDACC__")
|
|
#pragma pop_macro("__NVCC__")
|
|
#pragma pop_macro("__CUDA_ARCH__")
|
|
#pragma pop
|
|
#endif
|
|
|
|
#if defined __GNUC__
|
|
#pragma GCC diagnostic pop
|
|
#endif
|
|
|