GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-11 03:09:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix safe_abs in int_pow

Browse Source
This commit is contained in:
Charles Schlosser 2023-06-23 04:12:41 +00:00
parent c6db610bc7
commit 387175c258
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Eigen/src/Core/arch/Default/GenericPacketMathFunctions.h
View File

@ -1997,9 +1997,9 @@ struct exponent_helper<ScalarExponent, true> {
// consider the (rare) case where `exp` is an int32_t: abs(-2147483648) != 2147483648 // consider the (rare) case where `exp` is an int32_t: abs(-2147483648) != 2147483648
using safe_abs_type = typename numext::get_integer_by_size<sizeof(ScalarExponent)>::unsigned_type; using safe_abs_type = typename numext::get_integer_by_size<sizeof(ScalarExponent)>::unsigned_type;
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE safe_abs_type safe_abs(const ScalarExponent& exp) { static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE safe_abs_type safe_abs(const ScalarExponent& exp) {
ScalarExponent mask = exp ^ numext::abs(exp); ScalarExponent mask = numext::signbit(exp);
safe_abs_type result = static_cast<safe_abs_type>(exp); safe_abs_type result = safe_abs_type(exp ^ mask);
return result ^ mask; return result + safe_abs_type(ScalarExponent(1) & mask);
} }
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool is_odd(const safe_abs_type& exp) { static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool is_odd(const safe_abs_type& exp) {
return exp % safe_abs_type(2) != safe_abs_type(0); return exp % safe_abs_type(2) != safe_abs_type(0);