Add cached exp to benchmark

Author: ProExpertProg

cpp/bench/exps.cpp

@@ -1,11 +1,12 @@
+#include "CachedExponentials.hpp"
 #include "Exponentials.hpp"
 
 #include "benchmark-util.hpp"
 #include <iostream>
 
 using namespace ahr;
 using namespace ahr::exp;
-template <typename Exp> static void BM_ExpKXKY(benchmark::State &state) {
+template <space_like Exp> static void BM_ExpKXKY(benchmark::State &state) {
   Dim const X = state.range(0);
   Dim const Y = state.range(1);
 
@@ -21,7 +22,7 @@ template <typename Exp> static void BM_ExpKXKY(benchmark::State &state) {
   }
 }
 
-template <typename Exp> static void BM_ExpM(benchmark::State &state) {
+template <moment_like Exp> static void BM_ExpM(benchmark::State &state) {
   Dim const M = state.range(0);
   Dim const X = state.range(1);
   Dim const Y = state.range(2);
@@ -52,3 +53,16 @@ BENCHMARK_WMIN(BM_ExpKXKY<NuG>)
 BENCHMARK_WMIN(BM_ExpM<GM>)
     ->ArgsProduct({{2, 4, 10}, {2048, 4096}, {2048, 4096}})
     ->Unit(benchmark::kMillisecond);
+
+BENCHMARK_WMIN(BM_ExpKXKY<CachedKXKY<Eta>>)
+    ->ArgsProduct({{2048, 4096, 8192}, {2048, 4096, 8192}})
+    ->Unit(benchmark::kMillisecond);
+BENCHMARK_WMIN(BM_ExpKXKY<CachedKXKY<Nu>>)
+    ->ArgsProduct({{2048, 4096, 8192}, {2048, 4096, 8192}})
+    ->Unit(benchmark::kMillisecond);
+BENCHMARK_WMIN(BM_ExpKXKY<CachedKXKY<NuG>>)
+    ->ArgsProduct({{2048, 4096, 8192}, {2048, 4096, 8192}})
+    ->Unit(benchmark::kMillisecond);
+BENCHMARK_WMIN(BM_ExpM<CachedM<GM>>)
+    ->ArgsProduct({{2, 4, 10}, {2048, 4096}, {2048, 4096}})
+    ->Unit(benchmark::kMillisecond);

cpp/lib/CachedExponentials.hpp

@@ -0,0 +1,55 @@
+#pragma once
+
+#include "Exponentials.hpp"
+
+namespace ahr::exp {
+
+template <typename Exp>
+concept space_like = requires(Exp exp, Dim kx, Dim ky) {
+  { exp(kx, ky) } -> std::same_as<Real>;
+};
+
+template <typename Exp>
+concept moment_like = requires(Exp exp, Dim m) {
+  { exp(m) } -> std::same_as<Real>;
+};
+
+template <typename Exp>
+concept updatable = requires(Exp exp, Real dt) {
+  { exp.update(HyperCoefficients{}, dt) };
+};
+
+template <updatable Exp>
+  requires space_like<Exp>
+struct CachedKXKY : Exp {
+  explicit CachedKXKY(Grid const &grid) : Exp(grid), factors(grid.KX, grid.KY) {}
+
+  void update(HyperCoefficients hyper, Real dt) {
+    Exp::update(hyper, dt);
+    this->grid.for_each_kxky([&](Dim kx, Dim ky) { factors(kx, ky) = Exp::operator()(kx, ky); });
+  }
+
+  Real operator()(Dim kx, Dim ky) { return factors(kx, ky); }
+
+protected:
+  Grid::Buf::R_XY factors; ///< Real but of size (KX,KY)
+};
+
+template <updatable Exp>
+  requires moment_like<Exp>
+struct CachedM : Exp {
+  explicit CachedM(Grid const &grid) : Exp(grid), factors(grid.M) {}
+
+  void update(HyperCoefficients hyper, Real dt) {
+    Exp::update(hyper, dt);
+    for (Dim m = 0; m < this->grid.M; ++m) {
+      factors[m] = Exp::operator()(m);
+    }
+  }
+
+  Real operator()(Dim m) { return factors[m]; }
+
+protected:
+  std::vector<Real> factors;
+};
+} // namespace ahr::exp

cpp/lib/Exponentials.hpp

@@ -20,7 +20,7 @@ class Base {
   Real dt{-1};
 };
 
-class Eta : Base {
+class Eta : protected Base {
 public:
   using Base::Base;
   using Base::update;
@@ -32,7 +32,7 @@ class Eta : Base {
   }
 };
 
-class Nu : Base {
+class Nu : protected Base {
 public:
   using Base::Base;
   using Base::update;
@@ -43,7 +43,7 @@ class Nu : Base {
   }
 };
 
-class NuG : Base {
+class NuG : protected Base {
 public:
   using Base::Base;
   using Base::update;
@@ -54,7 +54,7 @@ class NuG : Base {
   }
 };
 
-class GM : Base {
+class GM : protected Base {
 public:
   using Base::Base;
   using Base::update;