diff options
| author |  Rasmus Munk Larsen <rmlarsen@google.com> | 2019-11-27 10:17:09 -0800 |
|---|---|---|
| committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2019-11-27 10:17:09 -0800 |
| commit | ea51a9eace7e4f0ea839e61eb2df85ccfb94aee8 (patch) | |
| tree | 16bd8229c448330ae39aab05fb35fa49f38703fb | |
| parent | 5a3ebda36b17f8de0efd0459515f4fa1050184ec (diff) | |
Add missing EIGEN_DEVICE_FUNC attribute to template specializations for pexp to fix GPU build.
| -rw-r--r-- | Eigen/src/Core/arch/Default/GenericPacketMathFunctions.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/Eigen/src/Core/arch/Default/GenericPacketMathFunctions.h b/Eigen/src/Core/arch/Default/GenericPacketMathFunctions.h index 8054e537a..2b69512e3 100644 --- a/Eigen/src/Core/arch/Default/GenericPacketMathFunctions.h +++ b/Eigen/src/Core/arch/Default/GenericPacketMathFunctions.h @@ -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 |