diff --git a/pytential/source.py b/pytential/source.py index 389cc7cfc..383c722c0 100644 --- a/pytential/source.py +++ b/pytential/source.py @@ -188,7 +188,7 @@ def exec_compute_potential_insn(self, actx, insn, bound_expr, evaluate, p2p = self.get_p2p(actx, source_kernels=insn.source_kernels, target_kernels=insn.target_kernels) - evt, output_for_each_kernel = p2p(actx, + output_for_each_kernel = p2p(actx, targets=flatten(target_discr.nodes(), actx, leaf_class=DOFArray), sources=self._nodes, strength=strengths, **kernel_args) diff --git a/pytential/unregularized.py b/pytential/unregularized.py index 3dee03372..71b6e2555 100644 --- a/pytential/unregularized.py +++ b/pytential/unregularized.py @@ -163,7 +163,7 @@ def exec_compute_potential_insn_direct(self, actx: PyOpenCLArrayContext, p2p = self.get_p2p(actx, source_kernels=insn.source_kernels, target_kernels=insn.target_kernels) - evt, output_for_each_kernel = p2p(actx, + output_for_each_kernel = p2p(actx, targets=flatten(target_discr.nodes(), actx, leaf_class=DOFArray), sources=flatten( self.density_discr.nodes(), actx, leaf_class=DOFArray