From b59d856c4a11525cc1436fe22ca5c3a4c9c2bcd1 Mon Sep 17 00:00:00 2001 From: Isuru Fernando Date: Wed, 6 Apr 2022 20:12:54 -0500 Subject: [PATCH] _rel -> _offset --- sumpy/p2p.py | 25 +++++++++++++------------ 1 file changed, 13 insertions(+), 12 deletions(-) diff --git a/sumpy/p2p.py b/sumpy/p2p.py index 0493a024..9c6c2c27 100644 --- a/sumpy/p2p.py +++ b/sumpy/p2p.py @@ -465,8 +465,8 @@ class P2PFromCSR(P2PBase): loopy_knl = lp.make_kernel([ "{[itgt_box]: 0 <= itgt_box < ntgt_boxes}", - "{[itgt_rel]: 0 <= itgt_rel < max_npoints_in_one_box}", - "{[isrc_rel]: 0 <= isrc_rel < max_npoints_in_one_box}", + "{[itgt_offset]: 0 <= itgt_offset < max_npoints_in_one_box}", + "{[isrc_offset]: 0 <= isrc_offset < max_npoints_in_one_box}", "{[iknl]: 0 <= iknl < noutputs}", "{[isrc_box]: isrc_box_start <= isrc_box < isrc_box_end}", "{[idim]: 0 <= idim < dim}", @@ -482,8 +482,8 @@ class P2PFromCSR(P2PBase): <> isrc_box_start = source_box_starts[itgt_box] <> isrc_box_end = source_box_starts[itgt_box+1] - for itgt_rel - <> itgt = itgt_rel + itgt_start + for itgt_offset + <> itgt = itgt_offset + itgt_start <> cond_itgt = itgt < itgt_end <> acc[iknl] = 0 {id=init_acc, dup=iknl} if cond_itgt @@ -493,13 +493,13 @@ class P2PFromCSR(P2PBase): <> src_ibox = source_box_lists[isrc_box] <> isrc_start = box_source_starts[src_ibox] <> isrc_end = isrc_start + box_source_counts_nonchild[src_ibox] - for isrc_rel - <> cond_isrc = isrc_rel < isrc_end - isrc_start + for isrc_offset + <> cond_isrc = isrc_offset < isrc_end - isrc_start if cond_isrc - <> local_isrc[idim, isrc_rel] = sources[idim, - isrc_rel + isrc_start] {id=prefetch_src, dup=idim} - <> local_isrc_strength[istrength, isrc_rel] = strength[ - istrength, isrc_rel + isrc_start] {id=prefetch_charge} + <> local_isrc[idim, isrc_offset] = sources[idim, + isrc_offset + isrc_start] {id=prefetch_src, dup=idim} + <> local_isrc_strength[istrength, isrc_offset] = strength[ + istrength, isrc_offset + isrc_start] {id=prefetch_charge} end end if cond_itgt @@ -566,8 +566,9 @@ class P2PFromCSR(P2PBase): knl = lp.split_iname(knl, "itgt_box", 4, outer_tag="g.0") else: knl = lp.tag_inames(knl, {"itgt_box": "g.0"}) - knl = lp.rename_inames(knl, ["isrc_rel"], "itgt_rel", existing_ok=True) - knl = lp.split_iname(knl, "itgt_rel", max_npoints_in_one_box, + knl = lp.rename_inames(knl, ["isrc_offset"], "itgt_offset", + existing_ok=True) + knl = lp.split_iname(knl, "itgt_offset", max_npoints_in_one_box, inner_tag="l.0") knl = lp.set_temporary_address_space(knl, ["local_isrc", "local_isrc_strength"], lp.AddressSpace.LOCAL) -- GitLab