diff options
| author |  Rasmus Munk Larsen <rmlarsen@google.com> | 2020-12-04 21:45:09 +0000 |
|---|---|---|
| committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2020-12-04 21:45:09 +0000 |
| commit | f9fac1d5b044afb8104ecb5c1e5183c309bafca4 (patch) | |
| tree | 1323be5978ccf5541530445b7789dcee48143db6 /Eigen/src/Core/arch/Default/Half.h | |
| parent | 2dbac2f99f073051b213f41d9dac26e01d94594e (diff) | |
Add log2() to Eigen.
Diffstat (limited to 'Eigen/src/Core/arch/Default/Half.h')
| -rw-r--r-- | Eigen/src/Core/arch/Default/Half.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/Eigen/src/Core/arch/Default/Half.h b/Eigen/src/Core/arch/Default/Half.h index 76156c416..204076e25 100644 --- a/Eigen/src/Core/arch/Default/Half.h +++ b/Eigen/src/Core/arch/Default/Half.h @@ -622,6 +622,10 @@ EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC half log1p(const half& a) { EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC half log10(const half& a) { return half(::log10f(float(a))); } +EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC half log2(const half& a) { + return half(static_cast<float>(M_LOG2E) * ::logf(float(a))); +} + EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC half sqrt(const half& a) { #if (EIGEN_CUDA_SDK_VER >= 80000 && defined EIGEN_CUDA_ARCH && EIGEN_CUDA_ARCH >= 530) || \ defined(EIGEN_HIP_DEVICE_COMPILE) |