@@ -47,7 +47,7 @@ def __init__(
4747 self ._typemap = typemap
4848
4949 self ._kernel_txt = self ._generate_kernel_stub_as_string ()
50- self ._py_func , self . _kernel_ir = self ._generate_kernel_ir ()
50+ self ._py_func = self ._generate_kernel_ir ()
5151
5252 def _generate_kernel_stub_as_string (self ):
5353 """Generate reduction main kernel template"""
@@ -160,18 +160,7 @@ def _generate_kernel_ir(self):
160160 globls = {"dpnp" : dpnp , "kapi" : kapi }
161161 locls = {}
162162 exec (self ._kernel_txt , globls , locls )
163- kernel_fn = locls [self ._kernel_name ]
164-
165- return kernel_fn , None
166-
167- @property
168- def kernel_ir (self ):
169- """Returns the Numba IR generated for a
170- TreeReduceIntermediateKernelTemplate.
171-
172- Returns: The Numba functionIR object for the compiled kernel_txt string.
173- """
174- return self ._kernel_ir
163+ return locls [self ._kernel_name ]
175164
176165 @property
177166 def kernel_string (self ):
@@ -189,11 +178,6 @@ def dump_kernel_string(self):
189178 print (self ._kernel_txt )
190179 sys .stdout .flush ()
191180
192- def dump_kernel_ir (self ):
193- """Helper to dump the Numba IR for a
194- TreeReduceIntermediateKernelTemplate."""
195- self ._kernel_ir .dump ()
196-
197181
198182class RemainderReduceIntermediateKernelTemplate (KernelTemplateInterface ):
199183 """The class to build reduction remainder kernel_txt template and
@@ -233,7 +217,7 @@ def __init__(
233217 self ._reductionKernelVar = reductionKernelVar
234218
235219 self ._kernel_txt = self ._generate_kernel_stub_as_string ()
236- self ._py_func , self . _kernel_ir = self ._generate_kernel_ir ()
220+ self ._py_func = self ._generate_kernel_ir ()
237221
238222 def _generate_kernel_stub_as_string (self ):
239223 """Generate reduction remainder kernel template"""
@@ -319,18 +303,7 @@ def _generate_kernel_ir(self):
319303 globls = {"dpnp" : dpnp , "kapi" : kapi }
320304 locls = {}
321305 exec (self ._kernel_txt , globls , locls )
322- kernel_fn = locls [self ._kernel_name ]
323-
324- return kernel_fn , None
325-
326- @property
327- def kernel_ir (self ):
328- """Returns the Numba IR generated for a
329- RemainderReduceIntermediateKernelTemplate.
330-
331- Returns: The Numba functionIR object for the compiled kernel_txt string.
332- """
333- return self ._kernel_ir
306+ return locls [self ._kernel_name ]
334307
335308 @property
336309 def kernel_string (self ):
@@ -348,9 +321,3 @@ def dump_kernel_string(self):
348321
349322 print (self ._kernel_txt )
350323 sys .stdout .flush ()
351-
352- def dump_kernel_ir (self ):
353- """Helper to dump the Numba IR for the
354- RemainderReduceIntermediateKernelTemplate."""
355-
356- self ._kernel_ir .dump ()
0 commit comments