From 16544c46491b36a753bad2ccef3ff48e0778c88a Mon Sep 17 00:00:00 2001 From: Hao Gao <gaohao95@gmail.com> Date: Sun, 4 Mar 2018 23:00:47 -0600 Subject: [PATCH] Fix bug --- sumpy/fmm.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/sumpy/fmm.py b/sumpy/fmm.py index d21ee06a..bc9a79d6 100644 --- a/sumpy/fmm.py +++ b/sumpy/fmm.py @@ -458,6 +458,7 @@ class SumpyExpansionWrangler(object): wait_for = mpole_exps.events + has_evt = False for isrc_level, ssn in enumerate(source_boxes_by_level): if len(target_boxes_by_source_level[isrc_level]) == 0: continue @@ -485,14 +486,16 @@ class SumpyExpansionWrangler(object): **kwargs) + has_evt = True wait_for = [evt] for pot_i, pot_res_i in zip(pot, pot_res): assert pot_i is pot_res_i - for pot_i in pot: - # Intentionally only adding the last event. - pot_i.add_event(evt) + if has_evt: + for pot_i in pot: + # Intentionally only adding the last event. + pot_i.add_event(evt) return pot -- GitLab