Fix for HIP breakage - 200115. Adding a missing EIGEN_DEVICE_FUNC attr

This commit is contained in:
Deven Desai 2020-01-16 00:51:43 +00:00
parent f6c6de5d63
commit 6d284bb1b7

View File

@ -714,6 +714,7 @@ struct cephes_helper<double> {
enum IgammaComputationMode { VALUE, DERIVATIVE, SAMPLE_DERIVATIVE }; enum IgammaComputationMode { VALUE, DERIVATIVE, SAMPLE_DERIVATIVE };
template <typename Scalar> template <typename Scalar>
EIGEN_DEVICE_FUNC
static EIGEN_STRONG_INLINE Scalar main_igamma_term(Scalar a, Scalar x) { static EIGEN_STRONG_INLINE Scalar main_igamma_term(Scalar a, Scalar x) {
/* Compute x**a * exp(-x) / gamma(a) */ /* Compute x**a * exp(-x) / gamma(a) */
Scalar logax = a * numext::log(x) - x - lgamma_impl<Scalar>::run(a); Scalar logax = a * numext::log(x) - x - lgamma_impl<Scalar>::run(a);