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/BFloat16.h | |
parent | 2dbac2f99f073051b213f41d9dac26e01d94594e (diff) |
Add log2() to Eigen.
Diffstat (limited to 'Eigen/src/Core/arch/Default/BFloat16.h')
-rw-r--r-- | Eigen/src/Core/arch/Default/BFloat16.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/Eigen/src/Core/arch/Default/BFloat16.h b/Eigen/src/Core/arch/Default/BFloat16.h index 351f451a3..616dcf667 100644 --- a/Eigen/src/Core/arch/Default/BFloat16.h +++ b/Eigen/src/Core/arch/Default/BFloat16.h @@ -512,6 +512,9 @@ EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC bfloat16 log1p(const bfloat16& a) { EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC bfloat16 log10(const bfloat16& a) { return bfloat16(::log10f(float(a))); } +EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC bfloat16 log2(const bfloat16& a) { + return bfloat16(static_cast<float>(M_LOG2E) * ::logf(float(a))); +} EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC bfloat16 sqrt(const bfloat16& a) { return bfloat16(::sqrtf(float(a))); } |