From b8e5e9c1ea7440db974021e19c36a78ef998b9d1 Mon Sep 17 00:00:00 2001
From: Isuru Fernando <isuruf@gmail.com>
Date: Wed, 6 Apr 2022 10:26:50 -0500
Subject: [PATCH] unroll istrength
---
sumpy/p2p.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/sumpy/p2p.py b/sumpy/p2p.py
index 85231d1a..33bc4d70 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:
--
GitLab