From 5e326baf836b591d01fb82eb4b3af54850b46616 Mon Sep 17 00:00:00 2001
From: Isuru Fernando <isuruf@gmail.com>
Date: Sat, 29 Apr 2023 16:24:58 -0500
Subject: [PATCH] Give names to loopy kernels in sumpy.fmm
---
sumpy/fmm.py | 16 ++++++++--------
1 file changed, 8 insertions(+), 8 deletions(-)
diff --git a/sumpy/fmm.py b/sumpy/fmm.py
index 6fe193f9..de7ac6ae 100644
--- a/sumpy/fmm.py
+++ b/sumpy/fmm.py
@@ -110,20 +110,20 @@ class SumpyTreeIndependentDataForWrangler(TreeIndependentDataForWrangler):
return P2EFromSingleBox(self.cl_context,
kernels=self.source_kernels,
expansion=self.multipole_expansion(tgt_order),
- strength_usage=self.strength_usage)
+ strength_usage=self.strength_usage, name="p2m")
@memoize_method
def p2l(self, tgt_order):
return P2EFromCSR(self.cl_context,
kernels=self.source_kernels,
expansion=self.local_expansion(tgt_order),
- strength_usage=self.strength_usage)
+ strength_usage=self.strength_usage, name="p2l")
@memoize_method
def m2m(self, src_order, tgt_order):
return E2EFromChildren(self.cl_context,
self.multipole_expansion(src_order),
- self.multipole_expansion(tgt_order))
+ self.multipole_expansion(tgt_order), name="m2m")
@memoize_method
def m2l(self, src_order, tgt_order,
@@ -134,7 +134,7 @@ class SumpyTreeIndependentDataForWrangler(TreeIndependentDataForWrangler):
m2l_class = E2EFromCSR
return m2l_class(self.cl_context,
self.multipole_expansion(src_order),
- self.local_expansion(tgt_order))
+ self.local_expansion(tgt_order), name="m2l")
@memoize_method
def m2l_translation_class_dependent_data_kernel(self, src_order, tgt_order):
@@ -158,26 +158,26 @@ class SumpyTreeIndependentDataForWrangler(TreeIndependentDataForWrangler):
def l2l(self, src_order, tgt_order):
return E2EFromParent(self.cl_context,
self.local_expansion(src_order),
- self.local_expansion(tgt_order))
+ self.local_expansion(tgt_order), name="l2l")
@memoize_method
def m2p(self, src_order):
return E2PFromCSR(self.cl_context,
self.multipole_expansion(src_order),
- self.target_kernels)
+ self.target_kernels, name="m2p")
@memoize_method
def l2p(self, src_order):
return E2PFromSingleBox(self.cl_context,
self.local_expansion(src_order),
- self.target_kernels)
+ self.target_kernels, name="l2p")
@memoize_method
def p2p(self):
return P2PFromCSR(self.cl_context, target_kernels=self.target_kernels,
source_kernels=self.source_kernels,
exclude_self=self.exclude_self,
- strength_usage=self.strength_usage)
+ strength_usage=self.strength_usage, name="p2p")
@memoize_method
def opencl_fft_app(self, shape, dtype, inverse):
--
GitLab