diff --git a/grudge/execution.py b/grudge/execution.py index 0104584ef28a8707f893b1c9d0ffd576c3453ade..3d5f430df4e5c01cd0c1e970ff654218f54de272 100644 --- a/grudge/execution.py +++ b/grudge/execution.py @@ -118,7 +118,8 @@ class ExecutionMapper(mappers.Evaluator, # {{{ elementwise reductions def _map_elementwise_reduction(self, op_name, field_expr, dd): - @memoize_in(self, "elementwise_%s_prg" % op_name) + @memoize_in(self.array_context, + (ExecutionMapper, "elementwise_%s_prg" % op_name)) def prg(): return make_loopy_program( "{[iel, idof, jdof]: 0<=iel