Skip to content

Commit d9881cd

Browse files
isurufinducer
authored andcommitted
Use work-items for unoptimized E2Es
1 parent bcb020d commit d9881cd

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

sumpy/e2e.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def get_kernel(self):
144144
def get_optimized_kernel(self):
145145
# FIXME
146146
knl = self.get_kernel()
147-
knl = lp.split_iname(knl, "itgt_box", 16, outer_tag="g.0")
147+
knl = lp.split_iname(knl, "itgt_box", 64, outer_tag="g.0", inner_tag="l.0")
148148

149149
return knl
150150

@@ -278,7 +278,7 @@ def get_kernel(self):
278278
def get_optimized_kernel(self):
279279
# FIXME
280280
knl = self.get_kernel()
281-
knl = lp.split_iname(knl, "itgt_box", 16, outer_tag="g.0")
281+
knl = lp.split_iname(knl, "itgt_box", 64, outer_tag="g.0", inner_tag="l.0")
282282

283283
return knl
284284

0 commit comments

Comments
 (0)