fix memoization in a couple of spots

majosm · majosm · commit 6856380069f5 · 2022-04-09T15:35:30.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
@@ -314,7 +314,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
`@@ -314,7 +314,7 @@ def _apply_elementwise_reduction(`
`314`	`314`	`)`
`315`	`315`	`)`
`316`	`316`	`else:`
`317`		`- @memoize_in(actx, (_apply_elementwise_reduction,`
	`317`	`+ @memoize_in(actx, (_apply_elementwise_reduction, dd,`
`318`	`318`	`"elementwise_%s_prg" % op_name))`
`319`	`319`	`def elementwise_prg():`
`320`	`320`	`# FIXME: This computes the reduction value redundantly for each`