devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix sqrt/rsqrt for NEON.

Browse Source
This commit is contained in:
Antonio Sánchez
2022-02-15 21:31:51 +00:00
committed by Rasmus Munk Larsen
parent 23755030c9
commit 28e008b99a
2 changed files with 9 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/packetmath.cpp
View File

@@ -954,11 +954,11 @@ void packetmath_real() {
} else {
data1[1] = -((std::numeric_limits<Scalar>::min)());
}
CHECK_CWISE1(numext::sqrt, internal::psqrt);
CHECK_CWISE1_IF(PacketTraits::HasSqrt, numext::sqrt, internal::psqrt);
data1[0] = Scalar(0.0f);
data1[1] = NumTraits<Scalar>::infinity();
CHECK_CWISE1(numext::sqrt, internal::psqrt);
CHECK_CWISE1_IF(PacketTraits::HasSqrt, numext::sqrt, internal::psqrt);
}
if (PacketTraits::HasRsqrt) {
@@ -968,11 +968,11 @@ void packetmath_real() {
} else {
data1[1] = -((std::numeric_limits<Scalar>::min)());
}
CHECK_CWISE1(numext::rsqrt, internal::prsqrt);
CHECK_CWISE1_IF(PacketTraits::HasRsqrt, numext::rsqrt, internal::prsqrt);
data1[0] = Scalar(0.0f);
data1[1] = NumTraits<Scalar>::infinity();
CHECK_CWISE1(numext::rsqrt, internal::prsqrt);
CHECK_CWISE1_IF(PacketTraits::HasRsqrt, numext::rsqrt, internal::prsqrt);
}
// TODO(rmlarsen): Re-enable for half and bfloat16.