fix memoization in a couple of spots

majosm · majosm · commit bd79083fa41e · 2022-05-23T10:43:11.000-05:00
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

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