diff --git a/sumpy/p2p.py b/sumpy/p2p.py index dc8368da09d968ecb64b59c9a6822577c3e71a20..eba84537d94419daf323971986203c340a253171 100644 --- a/sumpy/p2p.py +++ b/sumpy/p2p.py @@ -666,7 +666,9 @@ class P2PFromCSR(P2PBase): # optimization led to a 8% speedup in the performance. knl = lp.concatenate_arrays(knl, ["local_isrc", "local_isrc_strength"], "local_isrc") - knl = lp.tag_array_axes(knl, "local_isrc", "vec,C") + count = self.strength_count + self.dim + if count in [2, 3, 4, 8, 16]: + knl = lp.tag_array_axes(knl, "local_isrc", "vec,C") knl = lp.add_inames_for_unused_hw_axes(knl) # knl = lp.set_options(knl, write_code=True)