Modifications...

Author: Diptorup Deb

numba_dpex/experimental/decorators.py

@@ -80,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

@@ -172,7 +172,7 @@ def _compile_cached(
 
         except errors.TypingError as e:
             self._failed_cache[key] = e
-            return _KernelCompileResult(False, e, None)
+            return False, e
 
         return True, _KernelCompileResult(*kcres_attrs)
 

numba_dpex/experimental/models.py

@@ -6,10 +6,12 @@
 numba_dpex.experimental module.
 """
 
+from numba.core import types
 from numba.core.datamodel import DataModelManager, models
 from numba.core.extending import register_model
 
 import numba_dpex.core.datamodel.models as dpex_core_models
+from numba_dpex.core.types import Array, DpctlSyclQueue, DpnpNdArray, USMNdArray
 
 from .types import KernelDispatcherType
 
@@ -26,7 +28,20 @@ def _init_exp_data_model_manager() -> DataModelManager:
     """
 
     dmm = dpex_core_models.dpex_data_model_manager.copy()
+    dmm.register(types.CPointer, dpex_core_models.GenericPointerModel)
+    dmm.register(Array, dpex_core_models.USMArrayModel)
 
+    # Register the USMNdArray type to USMArrayModel in numba_dpex's data model
+    # manager. The dpex_data_model_manager is used by the DpexKernelTarget
+    dmm.register(USMNdArray, dpex_core_models.USMArrayModel)
+
+    # Register the DpnpNdArray type to USMArrayModel in numba_dpex's data model
+    # manager. The dpex_data_model_manager is used by the DpexKernelTarget
+    dmm.register(DpnpNdArray, dpex_core_models.USMArrayModel)
+
+    # Register the DpctlSyclQueue type to SyclQueueModel in numba_dpex's data
+    # model manager. The dpex_data_model_manager is used by the DpexKernelTarget
+    dmm.register(DpctlSyclQueue, dpex_core_models.SyclQueueModel)
     return dmm