diff --git a/grudge/execution.py b/grudge/execution.py index f8420e78ce1d1d35e89dd53dc63ec42fcd83af2a..a28e0b26fc01bdb29358307bc849ec07c4b1cddd 100644 --- a/grudge/execution.py +++ b/grudge/execution.py @@ -529,7 +529,8 @@ class ExecutionMapper(mappers.Evaluator, continue # Cache operator - cache_key = "diff_batch", in_grp, out_grp, tuple(insn.operators), field.dtype + cache_key = "diff_batch", in_grp, out_grp, tuple(insn.operators),\ + field.dtype try: matrices_ary_dev = self.bound_op.operator_data_cache[cache_key] except KeyError: @@ -540,7 +541,7 @@ class ExecutionMapper(mappers.Evaluator, matrices_ary[i] = matrices[op.rst_axis] matrices_ary_dev = self.array_context.from_numpy(matrices_ary) self.bound_op.operator_data_cache[cache_key] = matrices_ary_dev - + self.array_context.call_loopy( prg(noperators), diff_mat=matrices_ary_dev,