Remove unused parfor kernel IR methods

IntelPython · Apr 29, 2024 · 5712ef4 · 5712ef4
1 parent 8f8b98c
commit 5712ef4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 62 deletions.
diff --git a/numba_dpex/core/parfors/kernel_templates/kernel_template_iface.py b/numba_dpex/core/parfors/kernel_templates/kernel_template_iface.py
@@ -30,15 +30,6 @@ def _generate_kernel_ir(self):
     def dump_kernel_string(self):
         raise NotImplementedError
 
-    @abc.abstractmethod
-    def dump_kernel_ir(self):
-        raise NotImplementedError
-
-    @property
-    @abc.abstractmethod
-    def kernel_ir(self):
-        raise NotImplementedError
-
     @property
     @abc.abstractmethod
     def kernel_string(self):

diff --git a/numba_dpex/core/parfors/kernel_templates/range_kernel_template.py b/numba_dpex/core/parfors/kernel_templates/range_kernel_template.py
@@ -50,7 +50,7 @@ def __init__(
         self._param_dict = param_dict
 
         self._kernel_txt = self._generate_kernel_stub_as_string()
-        self._py_func, self._kernel_ir = self._generate_kernel_ir()
+        self._py_func = self._generate_kernel_ir()
 
     def _generate_kernel_stub_as_string(self):
         """Generates a stub dpex kernel for the parfor as a string.
@@ -108,17 +108,7 @@ def _generate_kernel_ir(self):
         globls = {"dpnp": dpnp, "dpex": dpex}
         locls = {}
         exec(self._kernel_txt, globls, locls)
-        kernel_fn = locls[self._kernel_name]
-
-        return kernel_fn, None
-
-    @property
-    def kernel_ir(self):
-        """Returns the Numba IR generated for a RangeKernelTemplate.
-
-        Returns: The Numba functionIR object for the compiled kernel_txt string.
-        """
-        return self._kernel_ir
+        return locls[self._kernel_name]
 
     @property
     def kernel_string(self):
@@ -133,7 +123,3 @@ def dump_kernel_string(self):
         """Helper to print the kernel function string."""
         print(self._kernel_txt)
         sys.stdout.flush()
-
-    def dump_kernel_ir(self):
-        """Helper to dump the Numba IR for the RangeKernelTemplate."""
-        self._kernel_ir.dump()
diff --git a/numba_dpex/core/parfors/kernel_templates/reduction_template.py b/numba_dpex/core/parfors/kernel_templates/reduction_template.py
@@ -47,7 +47,7 @@ def __init__(
         self._typemap = typemap
 
         self._kernel_txt = self._generate_kernel_stub_as_string()
-        self._py_func, self._kernel_ir = self._generate_kernel_ir()
+        self._py_func = self._generate_kernel_ir()
 
     def _generate_kernel_stub_as_string(self):
         """Generate reduction main kernel template"""
@@ -160,18 +160,7 @@ def _generate_kernel_ir(self):
         globls = {"dpnp": dpnp, "kapi": kapi}
         locls = {}
         exec(self._kernel_txt, globls, locls)
-        kernel_fn = locls[self._kernel_name]
-
-        return kernel_fn, None
-
-    @property
-    def kernel_ir(self):
-        """Returns the Numba IR generated for a
-            TreeReduceIntermediateKernelTemplate.
-
-        Returns: The Numba functionIR object for the compiled kernel_txt string.
-        """
-        return self._kernel_ir
+        return locls[self._kernel_name]
 
     @property
     def kernel_string(self):
@@ -189,11 +178,6 @@ def dump_kernel_string(self):
         print(self._kernel_txt)
         sys.stdout.flush()
 
-    def dump_kernel_ir(self):
-        """Helper to dump the Numba IR for a
-        TreeReduceIntermediateKernelTemplate."""
-        self._kernel_ir.dump()
-
 
 class RemainderReduceIntermediateKernelTemplate(KernelTemplateInterface):
     """The class to build reduction remainder kernel_txt template and
@@ -233,7 +217,7 @@ def __init__(
         self._reductionKernelVar = reductionKernelVar
 
         self._kernel_txt = self._generate_kernel_stub_as_string()
-        self._py_func, self._kernel_ir = self._generate_kernel_ir()
+        self._py_func = self._generate_kernel_ir()
 
     def _generate_kernel_stub_as_string(self):
         """Generate reduction remainder kernel template"""
@@ -319,18 +303,7 @@ def _generate_kernel_ir(self):
         globls = {"dpnp": dpnp, "kapi": kapi}
         locls = {}
         exec(self._kernel_txt, globls, locls)
-        kernel_fn = locls[self._kernel_name]
-
-        return kernel_fn, None
-
-    @property
-    def kernel_ir(self):
-        """Returns the Numba IR generated for a
-            RemainderReduceIntermediateKernelTemplate.
-
-        Returns: The Numba functionIR object for the compiled kernel_txt string.
-        """
-        return self._kernel_ir
+        return locls[self._kernel_name]
 
     @property
     def kernel_string(self):
@@ -348,9 +321,3 @@ def dump_kernel_string(self):
 
         print(self._kernel_txt)
         sys.stdout.flush()
-
-    def dump_kernel_ir(self):
-        """Helper to dump the Numba IR for the
-        RemainderReduceIntermediateKernelTemplate."""
-
-        self._kernel_ir.dump()