diff options
| author |  Luke Iwanski <luke@codeplay.com> | 2016-09-19 14:03:54 +0100 |
|---|---|---|
| committer | Luke Iwanski <luke@codeplay.com> | 2016-09-19 14:03:54 +0100 |
| commit | b91e0211727b9ea5d7c30908ed86afc4e50d4c6c (patch) | |
| tree | 4dd857d1aabdd4065e8b7dec1369a4bd06bac5e7 /Eigen/src/Core/arch/CUDA/Half.h | |
| parent | cb81975714a96ecb2faf33ca242feeee3543b1db (diff) | |
| parent | ff47717f25aeede4878f65b214cdce264b8314e8 (diff) | |
Merged with default.
Diffstat (limited to 'Eigen/src/Core/arch/CUDA/Half.h')
| -rw-r--r-- | Eigen/src/Core/arch/CUDA/Half.h | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/Eigen/src/Core/arch/CUDA/Half.h b/Eigen/src/Core/arch/CUDA/Half.h index 4d91420d0..52892db38 100644 --- a/Eigen/src/Core/arch/CUDA/Half.h +++ b/Eigen/src/Core/arch/CUDA/Half.h @@ -389,10 +389,14 @@ EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC half exp(const half& a) { return half(::expf(float(a))); } EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC half log(const half& a) { +#if defined(EIGEN_HAS_CUDA_FP16) && defined __CUDACC_VER__ && __CUDACC_VER__ >= 80000 && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530 + return Eigen::half(::hlog(a)); +#else return half(::logf(float(a))); +#endif } EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC half log1p(const half& a) { - return half(::log1pf(float(a))); + return half(numext::log1p(float(a))); } EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC half log10(const half& a) { return half(::log10f(float(a))); @@ -503,7 +507,11 @@ EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half exph(const Eigen::half& a) { return Eigen::half(::expf(float(a))); } EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half logh(const Eigen::half& a) { +#if defined __CUDACC_VER__ && __CUDACC_VER__ >= 80000 && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 530 + return Eigen::half(::hlog(a)); +#else return Eigen::half(::logf(float(a))); +#endif } EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Eigen::half sqrth(const Eigen::half& a) { return Eigen::half(::sqrtf(float(a))); |