fix memoization in a couple of spots

majosm · majosm · commit 76f0f1f877ce · 2022-06-21T14:42:55.000-05:00
diff --git a/grudge/dt_utils.py b/grudge/dt_utils.py
@@ -94,7 +94,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 actx.freeze(
diff --git a/grudge/reductions.py b/grudge/reductions.py
@@ -343,7 +343,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

Original file line number	Diff line number	Diff line change
`@@ -343,7 +343,7 @@ def _apply_elementwise_reduction(`
`343`	`343`	`)`
`344`	`344`	`)`
`345`	`345`	`else:`
`346`		`- @memoize_in(actx, (_apply_elementwise_reduction,`
	`346`	`+ @memoize_in(actx, (_apply_elementwise_reduction, dd,`
`347`	`347`	`"elementwise_%s_prg" % op_name))`
`348`	`348`	`def elementwise_prg():`
`349`	`349`	`# FIXME: This computes the reduction value redundantly for each`