From d9881cd840d890ddb76a717a6ee90a0117efb04b Mon Sep 17 00:00:00 2001
From: Isuru Fernando <isuruf@gmail.com>
Date: Thu, 27 Jul 2023 16:54:03 -0500
Subject: [PATCH] Use work-items for unoptimized E2Es
---
sumpy/e2e.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/sumpy/e2e.py b/sumpy/e2e.py
index 9a732378..3dcf2727 100644
--- a/sumpy/e2e.py
+++ b/sumpy/e2e.py
@@ -144,7 +144,7 @@ class E2EBase(KernelCacheMixin, ABC):
def get_optimized_kernel(self):
# FIXME
knl = self.get_kernel()
- knl = lp.split_iname(knl, "itgt_box", 16, outer_tag="g.0")
+ knl = lp.split_iname(knl, "itgt_box", 64, outer_tag="g.0", inner_tag="l.0")
return knl
@@ -278,7 +278,7 @@ class E2EFromCSR(E2EBase):
def get_optimized_kernel(self):
# FIXME
knl = self.get_kernel()
- knl = lp.split_iname(knl, "itgt_box", 16, outer_tag="g.0")
+ knl = lp.split_iname(knl, "itgt_box", 64, outer_tag="g.0", inner_tag="l.0")
return knl
--
GitLab