From a026d2091373de37a213cf50f8a5f3be085dd960 Mon Sep 17 00:00:00 2001 From: Isuru Fernando Date: Sat, 20 May 2023 19:28:22 -0500 Subject: [PATCH] Simplify condition in P2P --- sumpy/p2p.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/sumpy/p2p.py b/sumpy/p2p.py index 4be5916da..2ece64208 100644 --- a/sumpy/p2p.py +++ b/sumpy/p2p.py @@ -491,7 +491,8 @@ def get_kernel(self, max_nsources_in_one_box, max_ntargets_in_one_box, "{[istrength]: 0 <= istrength < nstrengths}", "{[inner]: 0 <= inner < work_items_per_group}", "{[itgt_offset_outer]: 0 <= itgt_offset_outer <= tgt_outer_limit}", - "{[isrc_prefetch]: 0 <= isrc_prefetch < max_nsources_in_one_box}", + "{[isrc_prefetch]: 0 <= isrc_prefetch < max_nsources_in_one_box " + " and isrc_prefetch < isrc_end - isrc_start}", "{[isrc_offset]: 0 <= isrc_offset < max_nsources_in_one_box" " and isrc_offset < isrc_end - isrc_start}", ] @@ -530,14 +531,10 @@ def get_kernel(self, max_nsources_in_one_box, max_ntargets_in_one_box, <> isrc_end = isrc_start + box_source_counts_nonchild[src_ibox] \ {id=src_box_insn_2} for isrc_prefetch - <> cond_isrc_prefetch = isrc_prefetch < isrc_end - isrc_start \ - {id=cond_isrc_prefetch} - if cond_isrc_prefetch - local_isrc[idim, isrc_prefetch] = sources[idim, - isrc_prefetch + isrc_start] {id=prefetch_src, dup=idim} - local_isrc_strength[istrength, isrc_prefetch] = strength[ - istrength, isrc_prefetch + isrc_start] {id=prefetch_charge} - end + local_isrc[idim, isrc_prefetch] = sources[idim, + isrc_prefetch + isrc_start] {id=prefetch_src, dup=idim} + local_isrc_strength[istrength, isrc_prefetch] = strength[ + istrength, isrc_prefetch + isrc_start] {id=prefetch_charge} end for inner if cond_itgt