devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Don't attempts to use lgamma_r for CUDA devices.

Browse Source 
Fix type in lgamma_impl<double>.
This commit is contained in:
Rasmus Munk Larsen
2016-11-01 15:34:19 -07:00
parent 2ebb314fa7
commit b730952414
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
unsupported/Eigen/src/SpecialFunctions/SpecialFunctionsImpl.h
View File

@@ -121,7 +121,7 @@ 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) {
#ifdef _BSD_SOURCE || _SVID_SOURCE #if !defined(__CUDA_ARCH__) && (defined(_BSD_SOURCE) || defined(_SVID_SOURCE))
int signgam; int signgam;
return ::lgammaf_r(x, &signgam); return ::lgammaf_r(x, &signgam);
#else #else
@@ -134,9 +134,9 @@ 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) {
#ifdef _BSD_SOURCE || _SVID_SOURCE #if !defined(__CUDA_ARCH__) && (defined(_BSD_SOURCE) || defined(_SVID_SOURCE))
int signgam; int signgam;
return ::lgammaf_r(x, &signgam); return ::lgamma_r(x, &signgam);
#else #else
return ::lgamma(x); return ::lgamma(x);
#endif #endif