diff --git a/sumpy/e2e.py b/sumpy/e2e.py index 2c309200f0ec83b40688802ba113798503d613d1..724903fd587e45a03a65f5bbd750b85b20615bba 100644 --- a/sumpy/e2e.py +++ b/sumpy/e2e.py @@ -196,7 +196,7 @@ class E2EFromCSR(E2EBase): """] + [""" tgt_expansions[tgt_ibox - tgt_base_ibox, {coeffidx}] = \ simul_reduce(sum, isrc_box, coeff{coeffidx}) \ - {{id_prefix=write_expn}} + {{id_prefix=write_expn,nosync=write_expn*}} """.format(coeffidx=i) for i in range(ncoeff_tgt)] + [""" end """], @@ -314,7 +314,7 @@ class E2EFromChildren(E2EBase): tgt_expansions[tgt_ibox - tgt_base_ibox, {i}] \ + coeff{i} \ {{id_prefix=write_expn,dep=compute_coeff*, - nosync=read_coeff*}} + nosync=read_coeff*:write_expn*}} """.format(i=i) for i in range(ncoeffs)] + [""" end end @@ -422,7 +422,7 @@ class E2EFromParent(E2EBase): tgt_expansions[tgt_ibox - tgt_base_ibox, {i}] = \ tgt_expansions[tgt_ibox - tgt_base_ibox, {i}] + coeff{i} \ - {{id_prefix=write_expn,nosync=read_expn*}} + {{id_prefix=write_expn,nosync=read_expn*:write_expn*}} """.format(i=i) for i in range(ncoeffs)] + [""" end """], diff --git a/sumpy/p2e.py b/sumpy/p2e.py index 1d2e63a2dc9852444f89e1a4df185654d507990b..19db841f9be1e2ad656fbd8b7ff5b552aff00d0b 100644 --- a/sumpy/p2e.py +++ b/sumpy/p2e.py @@ -138,7 +138,7 @@ class P2EFromSingleBox(P2EBase): """] + [""" tgt_expansions[src_ibox-tgt_base_ibox, {coeffidx}] = \ simul_reduce(sum, isrc, strength*coeff{coeffidx}) \ - {{id_prefix=write_expn}} + {{id_prefix=write_expn,nosync=write_expn*}} """.format(coeffidx=i) for i in range(ncoeffs)] + [""" end """], @@ -256,7 +256,8 @@ class P2EFromCSR(P2EBase): """] + [""" tgt_expansions[tgt_ibox - tgt_base_ibox, {coeffidx}] = \ simul_reduce(sum, (isrc_box, isrc), - strength*coeff{coeffidx}) {{id_prefix=write_expn}} + strength*coeff{coeffidx}) \ + {{id_prefix=write_expn,nosync=write_expn*}} """.format(coeffidx=i) for i in range(ncoeffs)] + [""" end """],