Add the Numba monkey patches

Author: Diptorup Deb

numba_dpex/__init__.py

@@ -17,14 +17,19 @@
 from numba import __version__ as numba_version
 from numba.np import arrayobj
 from numba.np.ufunc.decorators import Vectorize
+from numba.parfors import parfor
 
 from numba_dpex._patches import _empty_nd_impl
 from numba_dpex.vectorizers import Vectorize as DpexVectorize
 
-from .numba_patches import patch_mk_alloc
+from .numba_patches import patch_is_ufunc, patch_mk_alloc
+from .numba_patches.patch_parfor import (
+    _arrayexpr_tree_to_ir as _patched_arrayexpr_tree_to_ir,
+)
 
 # Monkey patches
-
+patch_is_ufunc.patch()
+parfor._arrayexpr_tree_to_ir = _patched_arrayexpr_tree_to_ir
 arrayobj._empty_nd_impl = _empty_nd_impl
 patch_mk_alloc.patch()