From 98f8f0db9a2241083b86803ad2e2097cd0f3765b Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Tue, 8 Sep 2015 15:37:25 -0700 Subject: [PATCH] Added support for predux_mul for CUDA devices --- Eigen/src/Core/arch/CUDA/PacketMath.h | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/Eigen/src/Core/arch/CUDA/PacketMath.h b/Eigen/src/Core/arch/CUDA/PacketMath.h index 195debd0d..a2d803c06 100644 --- a/Eigen/src/Core/arch/CUDA/PacketMath.h +++ b/Eigen/src/Core/arch/CUDA/PacketMath.h @@ -244,6 +244,13 @@ template<> EIGEN_DEVICE_FUNC inline double predux_min(const double2& a) return fmin(a.x, a.y); } +template<> EIGEN_DEVICE_FUNC inline float predux_mul(const float4& a) { + return a.x * a.y * a.z * a.w; +} +template<> EIGEN_DEVICE_FUNC inline double predux_mul(const double2& a) { + return a.x * a.y; +} + template<> EIGEN_DEVICE_FUNC inline float4 pabs(const float4& a) { return make_float4(fabsf(a.x), fabsf(a.y), fabsf(a.z), fabsf(a.w)); }