try moving smax to separate file

LoopModels · Feb 16, 2024 · faa6dab · faa6dab
1 parent 5f94615
commit faa6dab
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/include/Math/Exp.hpp b/include/Math/Exp.hpp
@@ -428,11 +428,6 @@ template <int l = 8> constexpr auto smax(auto x, auto y) {
   return o + softplus(l * d) / l;
 }
 
-template <int l = 8> constexpr auto smax(auto x, auto y, auto z) {
-  double m = std::max(std::max(value(x), value(y)), value(z));
-  constexpr double f = l, i = 1 / f;
-  return m + log(exp(f * (x - m)) + exp(f * (y - m)) + exp(f * (z - m))) * i;
-}
 template <int l = 8> constexpr auto smin(auto x, auto y) {
   return smax<-l>(x, y);
 }

diff --git a/include/Math/ExpDual.hpp b/include/Math/ExpDual.hpp
@@ -0,0 +1,12 @@
+#pragma once
+#include "Math/Dual.hpp"
+#include "Math/Exp.hpp"
+namespace poly::math {
+
+template <int l = 8> constexpr auto smax(auto x, auto y, auto z) {
+  double m = std::max(std::max(value(x), value(y)), value(z));
+  constexpr double f = l, i = 1 / f;
+  return m + log(exp(f * (x - m)) + exp(f * (y - m)) + exp(f * (z - m))) * i;
+}
+
+} // namespace poly::math