Add a new device and target string for experimental target.

Author: Diptorup Deb

numba_dpex/experimental/decorators.py

@@ -8,9 +8,13 @@
 import inspect
 
 from numba.core import sigutils
-from numba.core.target_extension import jit_registry, target_registry
+from numba.core.target_extension import (
+    jit_registry,
+    resolve_dispatcher_from_str,
+    target_registry,
+)
 
-from .kernel_dispatcher import KernelDispatcher
+from .target import DPEX_KERNEL_EXP_TARGET_NAME
 
 
 def kernel(func_or_sig=None, **options):
@@ -24,11 +28,14 @@ def kernel(func_or_sig=None, **options):
         * All array arguments passed to a kernel should adhere to compute
           follows data programming model.
     """
+
+    dispatcher = resolve_dispatcher_from_str(DPEX_KERNEL_EXP_TARGET_NAME)
+
     # FIXME: The options need to be evaluated and checked here like it is
     # done in numba.core.decorators.jit
 
     def _kernel_dispatcher(pyfunc):
-        return KernelDispatcher(
+        return dispatcher(
             pyfunc=pyfunc,
             targetoptions=options,
         )
@@ -59,9 +66,7 @@ def _kernel_dispatcher(pyfunc):
             func_or_sig = [func_or_sig]
 
         def _specialized_kernel_dispatcher(pyfunc):
-            return KernelDispatcher(
-                pyfunc=pyfunc,
-            )
+            return dispatcher(pyfunc=pyfunc)
 
         return _specialized_kernel_dispatcher
     func = func_or_sig
@@ -75,4 +80,4 @@ def _specialized_kernel_dispatcher(pyfunc):
     return _kernel_dispatcher(func)
 
 
-jit_registry[target_registry["dpex_kernel"]] = kernel
+jit_registry[target_registry[DPEX_KERNEL_EXP_TARGET_NAME]] = kernel

numba_dpex/experimental/kernel_dispatcher.py

@@ -24,7 +24,7 @@
 from numba_dpex.core.pipelines import kernel_compiler
 from numba_dpex.core.types import DpnpNdArray
 
-from .target import dpex_exp_kernel_target
+from .target import DPEX_KERNEL_EXP_TARGET_NAME, dpex_exp_kernel_target
 
 _KernelModule = namedtuple("_KernelModule", ["kernel_name", "kernel_bitcode"])
 
@@ -318,5 +318,5 @@ def __call__(self, *args, **kw_args):
         raise NotImplementedError
 
 
-_dpex_target = target_registry["dpex_kernel"]
+_dpex_target = target_registry[DPEX_KERNEL_EXP_TARGET_NAME]
 dispatcher_registry[_dpex_target] = KernelDispatcher

numba_dpex/experimental/target.py

@@ -9,15 +9,25 @@
 from functools import cached_property
 
 from numba.core.descriptors import TargetDescriptor
+from numba.core.target_extension import GPU, target_registry
 
 from numba_dpex.core.descriptor import DpexTargetOptions
 from numba_dpex.core.targets.kernel_target import (
-    DPEX_KERNEL_TARGET_NAME,
     DpexKernelTargetContext,
     DpexKernelTypingContext,
 )
 
 
+#  pylint: disable=R0903
+class SyclDeviceExp(GPU):
+    """Mark the hardware target as SYCL Device."""
+
+
+DPEX_KERNEL_EXP_TARGET_NAME = "dpex_kernel_exp"
+
+target_registry[DPEX_KERNEL_EXP_TARGET_NAME] = SyclDeviceExp
+
+
 class DpexExpKernelTypingContext(DpexKernelTypingContext):
     """Experimental typing context class extending the DpexKernelTypingContext
     by overriding super class functions for new experimental types.
@@ -77,4 +87,4 @@ def typing_context(self):
 
 
 # A global instance of the DpexKernelTarget with the experimental features
-dpex_exp_kernel_target = DpexExpKernelTarget(DPEX_KERNEL_TARGET_NAME)
+dpex_exp_kernel_target = DpexExpKernelTarget(DPEX_KERNEL_EXP_TARGET_NAME)