Skip to content

Commit

Permalink
Merge pull request #1254 from IntelPython/feature/target_specific_ove…
Browse files Browse the repository at this point in the history
…rload

Enable target specific overloads in experimental kernel
  • Loading branch information
ZzEeKkAa authored Dec 20, 2023
2 parents 6b76fd3 + 0cb8836 commit ec3ef13
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions numba_dpex/experimental/kernel_dispatcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,11 @@
from numba.core.compiler_lock import global_compiler_lock
from numba.core.dispatcher import Dispatcher, _FunctionCompiler
from numba.core.funcdesc import PythonFunctionDescriptor
from numba.core.target_extension import dispatcher_registry, target_registry
from numba.core.target_extension import (
dispatcher_registry,
target_override,
target_registry,
)
from numba.core.types import void
from numba.core.typing.typeof import Purpose, typeof

Expand Down Expand Up @@ -164,7 +168,8 @@ def _compile_cached(
pass

try:
cres: CompileResult = self._compile_core(args, return_type)
with target_override(DPEX_KERNEL_EXP_TARGET_NAME):
cres: CompileResult = self._compile_core(args, return_type)

if (
self.targetoptions["_compilation_mode"]
Expand Down

0 comments on commit ec3ef13

Please sign in to comment.