mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-03-07 18:27:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added support for standard mathematical functions and trancendentals(such as exp, log, abs, ...) on fp16

Browse Source
This commit is contained in:
Benoit Steiner 2016-03-29 09:20:36 -07:00
parent c38295f0a0
commit e02b784ec3
1 changed files with 30 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
Eigen/src/Core/arch/CUDA/Half.h
View File

@ -361,9 +361,6 @@ static inline EIGEN_HALF_CUDA_H bool (isnan)(const Eigen::half& a) {
} // end namespace Eigen
// Standard mathematical functions and trancendentals.
namespace std {
static inline EIGEN_DEVICE_FUNC Eigen::half abs(const Eigen::half& a) {
Eigen::half result;
result.x = a.x & 0x7FFF;
@ -375,6 +372,36 @@ static inline EIGEN_DEVICE_FUNC Eigen::half exp(const Eigen::half& a) {
static inline EIGEN_DEVICE_FUNC Eigen::half log(const Eigen::half& a) {
return Eigen::half(::logf(float(a)));
}
static inline EIGEN_DEVICE_FUNC Eigen::half sqrt(const Eigen::half& a) {
return Eigen::half(::sqrtf(float(a)));
}
static inline EIGEN_DEVICE_FUNC Eigen::half floor(const Eigen::half& a) {
return Eigen::half(::floorf(float(a)));
}
static inline EIGEN_DEVICE_FUNC Eigen::half ceil(const Eigen::half& a) {
return Eigen::half(::ceilf(float(a)));
}
static inline EIGEN_DEVICE_FUNC bool (isnan)(const Eigen::half& a) {
return (Eigen::numext::isnan)(a);
}
static inline EIGEN_DEVICE_FUNC bool (isinf)(const Eigen::half& a) {
return (Eigen::numext::isinf)(a);
}
static inline EIGEN_DEVICE_FUNC bool (isfinite)(const Eigen::half& a) {
return !(Eigen::numext::isinf)(a) && !(Eigen::numext::isnan)(a);
}
namespace std {
// Import the standard mathematical functions and trancendentals into the
// into the std namespace.
using ::abs;
using ::exp;
using ::log;
using ::sqrt;
using ::floor;
using ::ceil;
} // end namespace std