diff --git a/sumpy/e2e.py b/sumpy/e2e.py index 9a732378710ed77637ef0c2e72abe8e6dc570257..3dcf27272cb1d0cc8e51c051f1da47f8776bdfcb 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