GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-07-26 06:44:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

nvcc doesn't support std::min or std::max on GPU. Use our own custom implementation instead

Browse Source
This commit is contained in:
Benoit Steiner 2015-08-27 16:59:55 -07:00
parent d4c24eb016
commit 2ed1495eec
1 changed files with 31 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
Eigen/src/Core/MathFunctions.h
View File

@ -674,10 +674,11 @@ inline EIGEN_MATHFUNC_RETVAL(random, Scalar) random()
****************************************************************************/ ****************************************************************************/
namespace numext { namespace numext {
#ifndef __CUDA_ARCH__
template<typename T> template<typename T>
EIGEN_DEVICE_FUNC EIGEN_DEVICE_FUNC
inline T mini(const T& x, const T& y) EIGEN_ALWAYS_INLINE T mini(const T& x, const T& y)
{ {
EIGEN_USING_STD_MATH(min); EIGEN_USING_STD_MATH(min);
return min EIGEN_NOT_A_MACRO (x,y); return min EIGEN_NOT_A_MACRO (x,y);
@ -685,11 +686,38 @@ inline T mini(const T& x, const T& y)
template<typename T> template<typename T>
EIGEN_DEVICE_FUNC EIGEN_DEVICE_FUNC
inline T maxi(const T& x, const T& y) EIGEN_ALWAYS_INLINE T maxi(const T& x, const T& y)
{ {
EIGEN_USING_STD_MATH(max); EIGEN_USING_STD_MATH(max);
return max EIGEN_NOT_A_MACRO (x,y); return max EIGEN_NOT_A_MACRO (x,y);
} }
#else
template<typename T>
EIGEN_DEVICE_FUNC
EIGEN_ALWAYS_INLINE T mini(const T& x, const T& y)
{
return y < x ? y : x;
}
template<>
EIGEN_DEVICE_FUNC
EIGEN_ALWAYS_INLINE float mini(const float& x, const float& y)
{
return fmin(x, y);
}
template<typename T>
EIGEN_DEVICE_FUNC
EIGEN_ALWAYS_INLINE T maxi(const T& x, const T& y)
{
return x < y ? y : x;
}
template<>
EIGEN_DEVICE_FUNC
EIGEN_ALWAYS_INLINE float maxi(const float& x, const float& y)
{
return fmax(x, y);
}
#endif
template<typename Scalar> template<typename Scalar>
EIGEN_DEVICE_FUNC EIGEN_DEVICE_FUNC