Skip to content

Commit 0228db3

Browse files
committed
Do not use patch_arrayexpr_tree_to_ir
1 parent 7804da8 commit 0228db3

File tree

2 files changed

+1
-11
lines changed

2 files changed

+1
-11
lines changed

numba_dpex/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from numba import __version__ as numba_version
1818

1919
from .kernel_api_impl.spirv import target as spirv_kernel_target
20-
from .numba_patches import patch_arrayexpr_tree_to_ir, patch_is_ufunc
20+
from .numba_patches import patch_is_ufunc
2121
from .register_kernel_api_overloads import init_kernel_api_spirv_overloads
2222

2323

@@ -71,7 +71,6 @@ def parse_sem_version(version_string: str) -> Tuple[int, int, int]:
7171

7272
# Monkey patches
7373
patch_is_ufunc.patch()
74-
patch_arrayexpr_tree_to_ir.patch()
7574

7675
from numba import prange # noqa E402
7776

numba_dpex/numba_patches/patch_arrayexpr_tree_to_ir.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -192,12 +192,3 @@ def _arrayexpr_tree_to_ir(
192192
typemap.pop(expr_out_var.name, None)
193193
typemap[expr_out_var.name] = el_typ
194194
return out_ir
195-
196-
197-
def patch():
198-
"""
199-
Patches the _arrayexpr_tree_to_ir function in numba.parfor.parfor.py to
200-
support array expression nodes that were generated from dpnp expressions.
201-
"""
202-
203-
parfor._arrayexpr_tree_to_ir = _arrayexpr_tree_to_ir

0 commit comments

Comments
 (0)