Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Remove/njit offload #1041

Merged
merged 7 commits into from
May 15, 2023
Merged

Remove/njit offload #1041

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
0b793d7
Remove unit tests related to njit auto-offload.
May 15, 2023
ed48543
Remove OffloadCompiler pipeline.
May 15, 2023
eddf1a9
Remove all dpnp overloads using stubs and dpnp backend iface.
May 15, 2023
de3442e
Remove rename_numpy_functions_pass
May 15, 2023
f736747
Remove retarget and offload_dispatcher modules.
May 15, 2023
52aa2a3
Remove kernel compute-follows-data unit test case.
May 15, 2023
dd063f3
Remove deprecated example.
May 15, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 1 addition & 7 deletions numba_dpex/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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,
Expand Down Expand Up @@ -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"]
5 changes: 0 additions & 5 deletions numba_dpex/core/compiler.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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,
Expand Down
60 changes: 0 additions & 60 deletions numba_dpex/core/offload_dispatcher.py
View file
Open in desktop

This file was deleted.

Loading