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

std::numeric_limits doesn't work reliably on CUDA devices. Use our own definition of numeric_limit<T>::max() and numeric_limit<T>::min() instead of the stl ones.

Browse Source
This commit is contained in:
Benoit Steiner
2015-08-21 16:01:40 -07:00
parent e5c78d85c8
commit a09cfe650f
2 changed files with 55 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
Eigen/src/Core/NumTraits.h
View File

@@ -83,8 +83,24 @@ template<typename T> struct GenericNumTraits
// make sure to override this for floating-point types
return Real(0);
}
static inline T highest() { return (std::numeric_limits<T>::max)(); }
static inline T lowest() { return IsInteger ? (std::numeric_limits<T>::min)() : (-(std::numeric_limits<T>::max)()); }
EIGEN_DEVICE_FUNC
static inline T highest() {
#if defined(__CUDA_ARCH__)
return internal::device::numeric_limits<T>::max();
#else
return (std::numeric_limits<T>::max)();
#endif
}
EIGEN_DEVICE_FUNC
static inline T lowest() {
#if defined(__CUDA_ARCH__)
return IsInteger ? (internal::device::numeric_limits<T>::min)() : (-(internal::device::numeric_limits<T>::max)());
#else
return IsInteger ? (std::numeric_limits<T>::min)() : (-(std::numeric_limits<T>::max)());
#endif
};
template<typename T> struct NumTraits : GenericNumTraits<T>