From 95a430a2ca8489a85d0a12ffa66d260011c11745 Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Fri, 3 Oct 2014 19:45:19 -0700 Subject: Vector primitives for CUDA --- Eigen/Core | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'Eigen/Core') diff --git a/Eigen/Core b/Eigen/Core index 776b7faf3..537ac16b2 100644 --- a/Eigen/Core +++ b/Eigen/Core @@ -294,6 +294,11 @@ using std::ptrdiff_t; #include "src/Core/arch/NEON/Complex.h" #endif +#if defined EIGEN_VECTORIZE_CUDA + #include "src/Core/arch/CUDA/PacketMath.h" + #include "src/Core/arch/CUDA/MathFunctions.h" +#endif + #include "src/Core/arch/Default/Settings.h" #include "src/Core/functors/BinaryFunctors.h" -- cgit v1.2.3 From bbce6fa65d8a196f05e0428d014e0e3865e202f3 Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Fri, 3 Oct 2014 19:55:35 -0700 Subject: define EIGEN_VECTORIZE_CUDA when compiling with nvcc --- Eigen/Core | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'Eigen/Core') 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 +#endif + #if (defined _OPENMP) && (!defined EIGEN_DONT_PARALLELIZE) #define EIGEN_HAS_OPENMP #endif -- cgit v1.2.3