define EIGEN_VECTORIZE_CUDA when compiling with nvcc

This commit is contained in:
Benoit Steiner 2014-10-03 19:55:35 -07:00
parent 95a430a2ca
commit bbce6fa65d

View File

@ -178,6 +178,11 @@
#endif
#endif
#if defined __CUDACC__
#define EIGEN_VECTORIZE_CUDA
#include <vector_types.h>
#endif
#if (defined _OPENMP) && (!defined EIGEN_DONT_PARALLELIZE)
#define EIGEN_HAS_OPENMP
#endif