Add EIGEN prefix for HAS_LGAMMA_R

This commit is contained in:
David Tellenbach 2020-10-08 18:32:19 +02:00
parent 2279f2c62f
commit 8f8d77b516

View File

@ -60,20 +60,20 @@ struct lgamma_retval {
// Since glibc 2.19 // Since glibc 2.19
#if defined(__GLIBC__) && ((__GLIBC__>=2 && __GLIBC_MINOR__ >= 19) || __GLIBC__>2) \ #if defined(__GLIBC__) && ((__GLIBC__>=2 && __GLIBC_MINOR__ >= 19) || __GLIBC__>2) \
&& (defined(_DEFAULT_SOURCE) || defined(_BSD_SOURCE) || defined(_SVID_SOURCE)) && (defined(_DEFAULT_SOURCE) || defined(_BSD_SOURCE) || defined(_SVID_SOURCE))
#define HAS_LGAMMA_R #define EIGEN_HAS_LGAMMA_R
#endif #endif
// Glibc versions before 2.19 // Glibc versions before 2.19
#if defined(__GLIBC__) && ((__GLIBC__==2 && __GLIBC_MINOR__ < 19) || __GLIBC__<2) \ #if defined(__GLIBC__) && ((__GLIBC__==2 && __GLIBC_MINOR__ < 19) || __GLIBC__<2) \
&& (defined(_BSD_SOURCE) || defined(_SVID_SOURCE)) && (defined(_BSD_SOURCE) || defined(_SVID_SOURCE))
#define HAS_LGAMMA_R #define EIGEN_HAS_LGAMMA_R
#endif #endif
template <> template <>
struct lgamma_impl<float> { struct lgamma_impl<float> {
EIGEN_DEVICE_FUNC EIGEN_DEVICE_FUNC
static EIGEN_STRONG_INLINE float run(float x) { static EIGEN_STRONG_INLINE float run(float x) {
#if !defined(EIGEN_GPU_COMPILE_PHASE) && defined (HAS_LGAMMA_R) && !defined(__APPLE__) #if !defined(EIGEN_GPU_COMPILE_PHASE) && defined (EIGEN_HAS_LGAMMA_R) && !defined(__APPLE__)
int dummy; int dummy;
return ::lgammaf_r(x, &dummy); return ::lgammaf_r(x, &dummy);
#elif defined(SYCL_DEVICE_ONLY) #elif defined(SYCL_DEVICE_ONLY)
@ -88,7 +88,7 @@ template <>
struct lgamma_impl<double> { struct lgamma_impl<double> {
EIGEN_DEVICE_FUNC EIGEN_DEVICE_FUNC
static EIGEN_STRONG_INLINE double run(double x) { static EIGEN_STRONG_INLINE double run(double x) {
#if !defined(EIGEN_GPU_COMPILE_PHASE) && defined(HAS_LGAMMA_R) && !defined(__APPLE__) #if !defined(EIGEN_GPU_COMPILE_PHASE) && defined(EIGEN_HAS_LGAMMA_R) && !defined(__APPLE__)
int dummy; int dummy;
return ::lgamma_r(x, &dummy); return ::lgamma_r(x, &dummy);
#elif defined(SYCL_DEVICE_ONLY) #elif defined(SYCL_DEVICE_ONLY)
@ -99,7 +99,7 @@ struct lgamma_impl<double> {
} }
}; };
#undef HAS_LGAMMA_R #undef EIGEN_HAS_LGAMMA_R
#endif #endif
/**************************************************************************** /****************************************************************************