diff --git a/sumpy/p2p.py b/sumpy/p2p.py index 85231d1a87c7bd111676a843099e49ae6c9c7b3d..33bc4d70b07aac1371234c5b06a82c854d11afdc 100644 --- a/sumpy/p2p.py +++ b/sumpy/p2p.py @@ -550,6 +550,7 @@ class P2PFromCSR(P2PBase): dict(nsources=np.int32, ntargets=np.int32)) loopy_knl = lp.tag_inames(loopy_knl, "idim*:unr") + loopy_knl = lp.tag_inames(loopy_knl, "istrength*:unr") loopy_knl = lp.tag_array_axes(loopy_knl, "targets", "sep,C") loopy_knl = lp.tag_array_axes(loopy_knl, "sources", "sep,C") for knl in self.target_kernels + self.source_kernels: