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

Address comments on Chebyshev evaluation code:

Browse Source 
1. Use pmadd when possible.
2. Add casts to avoid c++03 warnings.
This commit is contained in:
Rasmus Munk Larsen 2019-10-02 12:48:17 -07:00
parent 6e40454a6e
commit fab4e3a753
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -626,17 +626,18 @@ template <typename Packet, int N>
struct pchebevl { struct pchebevl {
EIGEN_DEVICE_FUNC EIGEN_DEVICE_FUNC
static EIGEN_STRONG_INLINE Packet run(Packet x, const typename unpacket_traits<Packet>::type coef[]) { static EIGEN_STRONG_INLINE Packet run(Packet x, const typename unpacket_traits<Packet>::type coef[]) {
typedef typename unpacket_traits<Packet>::type Scalar;
Packet b0 = pset1<Packet>(coef[0]); Packet b0 = pset1<Packet>(coef[0]);
Packet b1 = pset1<Packet>(0.f); Packet b1 = pset1<Packet>(static_cast<Scalar>(0.f));
Packet b2; Packet b2;
for (int i = 1; i < N; i++) { for (int i = 1; i < N; i++) {
b2 = b1; b2 = b1;
b1 = b0; b1 = b0;
b0 = padd(psub(pmul(x, b1), b2), pset1<Packet>(coef[i])); b0 = psub(pmadd(x, b1, pset1<Packet>(coef[i])), b2);
} }
return pmul(pset1<Packet>(0.5f), psub(b0, b2)); return pmul(pset1<Packet>(static_cast<Scalar>(0.5f)), psub(b0, b2));
} }
}; };