Merge pull request #1041 from IntelPython/remove/njit_offload

Remove/njit offload

Author: Diptorup Deb

numba_dpex/__init__.py

@@ -94,7 +94,6 @@ def parse_sem_version(version_string: str) -> Tuple[int, int, int]:
 from numba import prange  # noqa E402
 
 import numba_dpex.core.dpjit_dispatcher  # noqa E402
-import numba_dpex.core.offload_dispatcher  # noqa E402
 
 # Initialize the _dpexrt_python extension
 import numba_dpex.core.runtime  # noqa E402
@@ -111,16 +110,11 @@ def parse_sem_version(version_string: str) -> Tuple[int, int, int]:
 # Re-export all type names
 from numba_dpex.core.types import *  # noqa E402
 from numba_dpex.dpnp_iface import dpnpimpl  # noqa E402
-from numba_dpex.retarget import offload_to_sycl_device  # noqa E402
 
 if config.HAS_NON_HOST_DEVICE:
     # Re export
     from .core.targets import dpjit_target, kernel_target
     from .decorators import dpjit, func, kernel
-
-    # We are importing dpnp stub module to make Numba recognize the
-    # module when we rename Numpy functions.
-    from .dpnp_iface.stubs import dpnp
     from .ocl.stubs import (
         GLOBAL_MEM_FENCE,
         LOCAL_MEM_FENCE,
@@ -150,4 +144,4 @@ def parse_sem_version(version_string: str) -> Tuple[int, int, int]:
 __version__ = get_versions()["version"]
 del get_versions
 
-__all__ = types.__all__ + ["offload_to_sycl_device"] + ["Range", "NdRange"]
+__all__ = types.__all__ + ["Range", "NdRange"]

numba_dpex/core/compiler.py

@@ -14,7 +14,6 @@
     UnreachableError,
 )
 from numba_dpex.core.pipelines.kernel_compiler import KernelCompiler
-from numba_dpex.core.pipelines.offload_compiler import OffloadCompiler
 
 
 @global_compiler_lock
@@ -73,10 +72,6 @@ def compile_with_dpex(
             pipeline_class=KernelCompiler,
         )
     elif isinstance(pyfunc, ir.FunctionIR):
-        # FIXME: Kernels in the form of Numba IR need to be compiled
-        # using the offload compiler due to them retaining parfor
-        # nodes due to the use of gufuncs. Once the kernel builder is
-        # ready we should be able to switch to the KernelCompiler.
         cres = compiler.compile_ir(
             typingctx=typingctx,
             targetctx=targetctx,