saberbud/ceres-solver
1
0
Fork 0
mirror of https://github.com/ceres-solver/ceres-solver.git synced 2026-08-29 08:34:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support fma Jet

Browse Source 
Change-Id: I9b1ad49611e3e6f117190d56512ef2ec4bdbb1c1
This commit is contained in:
Sergiu Deitsch
2022-02-01 20:52:27 +01:00
parent a0d81ad63e
commit 7d6524daff
2 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/ceres/jet.h
+12
View File
@@ -452,6 +452,7 @@ using std::exp;
using std::exp2;
using std::expm1;
using std::floor;
using std::fma;
using std::fmax;
using std::fmin;
using std::hypot;
@@ -713,6 +714,17 @@ inline Jet<T, N> hypot(const Jet<T, N>& x,
}
#endif // defined(CERES_HAS_CPP17)
// Like x * y + z but rounded only once.
template <typename T, int N>
inline Jet<T, N> fma(const Jet<T, N>& x,
const Jet<T, N>& y,
const Jet<T, N>& z) {
// d/dx fma(x, y, z) = y
// d/dy fma(x, y, z) = x
// d/dz fma(x, y, z) = 1
return Jet<T, N>(fma(x.a, y.a, z.a), y.a * x.v + x.a * y.v + z.v);
}
template <typename T, int N>
inline Jet<T, N> fmax(const Jet<T, N>& x, const Jet<T, N>& y) {
using std::isgreater;
internal/ceres/jet_test.cc
+6
View File
@@ -645,6 +645,12 @@ TEST(Jet, Midpoint) {
#endif // defined(CERES_HAS_CPP20)
TEST(Jet, Fma) {
J v = fma(x, y, z);
J w = x * y + z;
EXPECT_THAT(v, IsAlmostEqualTo(w));
}
TEST(Jet, Fmax) {
EXPECT_THAT(fmax(x, y), IsAlmostEqualTo(x));
EXPECT_THAT(fmax(y, x), IsAlmostEqualTo(x));