Do not use patch_arrayexpr_tree_to_ir

IntelPython · Mar 29, 2024 · a8ec606 · a8ec606
1 parent 1617753
commit a8ec606
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 11 deletions.
diff --git a/numba_dpex/__init__.py b/numba_dpex/__init__.py
@@ -17,7 +17,7 @@
 from numba import __version__ as numba_version
 
 from .kernel_api_impl.spirv import target as spirv_kernel_target
-from .numba_patches import patch_arrayexpr_tree_to_ir, patch_is_ufunc
+from .numba_patches import patch_is_ufunc
 from .register_kernel_api_overloads import init_kernel_api_spirv_overloads
 
 
@@ -71,7 +71,6 @@ def parse_sem_version(version_string: str) -> Tuple[int, int, int]:
 
 # Monkey patches
 patch_is_ufunc.patch()
-patch_arrayexpr_tree_to_ir.patch()
 
 from numba import prange  # noqa E402
 

diff --git a/numba_dpex/numba_patches/patch_arrayexpr_tree_to_ir.py b/numba_dpex/numba_patches/patch_arrayexpr_tree_to_ir.py
@@ -192,12 +192,3 @@ def _arrayexpr_tree_to_ir(
     typemap.pop(expr_out_var.name, None)
     typemap[expr_out_var.name] = el_typ
     return out_ir
-
-
-def patch():
-    """
-    Patches the _arrayexpr_tree_to_ir function in numba.parfor.parfor.py to
-    support array expression nodes that were generated from dpnp expressions.
-    """
-
-    parfor._arrayexpr_tree_to_ir = _arrayexpr_tree_to_ir