diff --git a/sumpy/fmm.py b/sumpy/fmm.py index 86b4412ae4f3fe08d975972b2ac79d51847664fe..6fe193f96ffee54861377fcc81346fb04dade61c 100644 --- a/sumpy/fmm.py +++ b/sumpy/fmm.py @@ -1034,9 +1034,10 @@ class SumpyExpansionWrangler(ExpansionWranglerInterface): assert local_exps_res is target_local_exps_view - local_exps.add_event(evt) + for evt in events: + local_exps.add_event(evt) - return (local_exps, SumpyTimingFuture(queue, [evt])) + return (local_exps, SumpyTimingFuture(queue, events)) def eval_locals(self, level_start_target_box_nrs, target_boxes, local_exps): pot = self.output_zeros(local_exps)