CI: Add memory/peakmemory benchmarks

andgoldschmidt · Aug 1, 2024 · 47c0f36 · 47c0f36
1 parent ac62989
commit 47c0f36
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/asv_bench/benchmarks/finite_difference_1D.py b/asv_bench/benchmarks/finite_difference_1D.py
@@ -15,7 +15,8 @@ def setup(self):
     def time_derivative(self):
         self.differentiator.d(X=self.t, t=self.t, axis=0)
 
+    def mem_derivative(self):
+        self.differentiator.d(X=self.t, t=self.t, axis=0)
 
-class MemSuite:
-    def mem_list(self):
-        return [0] * 256
+    def peakmem_derivative(self):
+        self.differentiator.d(X=self.t, t=self.t, axis=0)