Provide dpctl implemenation registry

Author: ZzEeKkAa

numba_dpex/core/targets/dpjit_target.py

@@ -14,6 +14,7 @@
 from numba.core.target_extension import CPU, target_registry
 
 from numba_dpex.core.datamodel.models import _init_dpjit_data_model_manager
+from numba_dpex.dpctl_iface import dpctlimpl
 from numba_dpex.dpnp_iface import dpnp_ufunc_db
 
 
@@ -69,6 +70,7 @@ def load_additional_registries(self):
         Load dpjit-specific registries.
         """
         self.install_registry(dpex_function_registry)
+        self.install_registry(dpctlimpl.registry)
 
         # loading CPU specific registries
         super().load_additional_registries()

numba_dpex/dpctl_iface/dpctlimpl.py

@@ -0,0 +1,15 @@
+# SPDX-FileCopyrightText: 2024 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from numba.core.imputils import Registry
+
+registry = Registry("dpctlimpl")
+
+lower_builtin = registry.lower
+lower_getattr = registry.lower_getattr
+lower_getattr_generic = registry.lower_getattr_generic
+lower_setattr = registry.lower_setattr
+lower_setattr_generic = registry.lower_setattr_generic
+lower_cast = registry.lower_cast
+lower_constant = registry.lower_constant

numba_dpex/kernel_api_impl/spirv/target.py

@@ -293,13 +293,15 @@ def load_additional_registries(self):
         """
         # pylint: disable=import-outside-toplevel
         from numba_dpex import printimpl
+        from numba_dpex.dpctl_iface import dpctlimpl
         from numba_dpex.dpnp_iface import dpnpimpl
         from numba_dpex.ocl import mathimpl, oclimpl
 
         self.insert_func_defn(oclimpl.registry.functions)
         self.insert_func_defn(mathimpl.registry.functions)
         self.insert_func_defn(dpnpimpl.registry.functions)
         self.install_registry(printimpl.registry)
+        self.install_registry(dpctlimpl.registry)
         self.install_registry(spirv_registry)
         # Replace dpnp math functions with their OpenCL versions.
         self.replace_dpnp_ufunc_with_ocl_intrinsics()