From d20efc974d53f513668c0138e945171f6f4d85e8 Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Fri, 4 Dec 2015 09:38:15 -0800 Subject: Made it possible to use the sigmoid functor within a CUDA kernel. --- unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h') diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h b/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h index 7ba0c2817..34ba4e392 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h @@ -40,8 +40,8 @@ struct scalar_sigmoid_op { return one / (one + std::exp(-x)); } - template - inline Packet packetOp(const Packet& x) const { + template EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE + Packet packetOp(const Packet& x) const { const Packet one = pset1(1); return pdiv(one, padd(one, pexp(pnegate(x)))); } -- cgit v1.2.3