Add missing EIGEN_DEVICE_FUNC attribute to template specializations for pexp to fix GPU build.

This commit is contained in:
Rasmus Munk Larsen 2019-11-27 10:17:09 -08:00
parent 5a3ebda36b
commit ea51a9eace

View File

@ -235,7 +235,7 @@ Packet pexp_float(const Packet _x)
// make it the default path for scalar float
template<>
inline float pexp(const float& a) { return pexp_float(a); }
EIGEN_DEVICE_FUNC inline float pexp(const float& a) { return pexp_float(a); }
template <typename Packet>
EIGEN_DEFINE_FUNCTION_ALLOWING_MULTIPLE_DEFINITIONS
@ -307,7 +307,7 @@ Packet pexp_double(const Packet _x)
// make it the default path for scalar double
template<>
inline double pexp(const double& a) { return pexp_double(a); }
EIGEN_DEVICE_FUNC inline double pexp(const double& a) { return pexp_double(a); }
// The following code is inspired by the following stack-overflow answer:
// https://stackoverflow.com/questions/30463616/payne-hanek-algorithm-implementation-in-c/30465751#30465751