fix memoization in a couple of spots

inducer · May 23, 2022 · bd79083 · bd79083
1 parent 22d09f4
commit bd79083
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/grudge/dt_utils.py b/grudge/dt_utils.py
@@ -90,7 +90,7 @@ def characteristic_lengthscales(
         methods has been used as a guide. Any concrete time integrator will
         likely require scaling of the values returned by this routine.
     """
-    @memoize_in(dcoll, (characteristic_lengthscales,
+    @memoize_in(dcoll, (characteristic_lengthscales, dd,
                         "compute_characteristic_lengthscales"))
     def _compute_characteristic_lengthscales():
         return freeze(

diff --git a/grudge/reductions.py b/grudge/reductions.py
@@ -342,7 +342,7 @@ def _apply_elementwise_reduction(
             )
         )
     else:
-        @memoize_in(actx, (_apply_elementwise_reduction,
+        @memoize_in(actx, (_apply_elementwise_reduction, dd,
                         "elementwise_%s_prg" % op_name))
         def elementwise_prg():
             # FIXME: This computes the reduction value redundantly for each