aboutsummaryrefslogtreecommitdiffhomepage
path: root/Eigen
diff options
context:
space:
mode:
authorGravatar Benoit Steiner <benoit.steiner.goog@gmail.com>2014-10-03 19:55:35 -0700
committerGravatar Benoit Steiner <benoit.steiner.goog@gmail.com>2014-10-03 19:55:35 -0700
commitbbce6fa65d8a196f05e0428d014e0e3865e202f3 (patch)
tree56d1d2130ab250b9497a5d2465047420e4356508 /Eigen
parent95a430a2ca8489a85d0a12ffa66d260011c11745 (diff)
define EIGEN_VECTORIZE_CUDA when compiling with nvcc
Diffstat (limited to 'Eigen')
-rw-r--r--Eigen/Core5
1 files changed, 5 insertions, 0 deletions
diff --git a/Eigen/Core b/Eigen/Core
index 537ac16b2..acdeca5f4 100644
--- a/Eigen/Core
+++ b/Eigen/Core
@@ -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