Add infinity() support to numext::numeric_limits, use it in lgamma.

This makes the infinity access a __device__ function, removing
nvcc warnings.
This commit is contained in:
Eugene Brevdo
2016-03-02 21:35:48 -08:00
parent 832380c455
commit 6afea46838
3 changed files with 11 additions and 1 deletions

View File

@@ -90,6 +90,11 @@ template<typename T> struct GenericNumTraits
static inline T lowest() {
return IsInteger ? (numext::numeric_limits<T>::min)() : (-(numext::numeric_limits<T>::max)());
}
EIGEN_DEVICE_FUNC
static inline T infinity() {
return numext::numeric_limits<T>::infinity();
}
};
template<typename T> struct NumTraits : GenericNumTraits<T>