fix memoization in a couple of spots

majosm · majosm · commit 5fffa0b56d40 · 2022-07-01T13:31:23.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
@@ -344,7 +344,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
`@@ -344,7 +344,7 @@ def _apply_elementwise_reduction(`
`344`	`344`	`)`
`345`	`345`	`)`
`346`	`346`	`else:`
`347`		`- @memoize_in(actx, (_apply_elementwise_reduction,`
	`347`	`+ @memoize_in(actx, (_apply_elementwise_reduction, dd,`
`348`	`348`	`"elementwise_%s_prg" % op_name))`
`349`	`349`	`def elementwise_prg():`
`350`	`350`	`# FIXME: This computes the reduction value redundantly for each`