Add dpjit specific typing context

Author: ZzEeKkAa

numba_dpex/core/descriptor.py

@@ -10,7 +10,11 @@
 
 from numba_dpex.core import config
 
-from .targets.dpjit_target import DPEX_TARGET_NAME, DpexTargetContext
+from .targets.dpjit_target import (
+    DPEX_TARGET_NAME,
+    DpexTargetContext,
+    DpexTypingContext,
+)
 from .targets.kernel_target import (
     DPEX_KERNEL_TARGET_NAME,
     CompilationMode,
@@ -110,7 +114,7 @@ def _toplevel_target_context(self):
     @cached_property
     def _toplevel_typing_context(self):
         # Lazily-initialized top-level typing context, for all threads
-        return typing.Context()
+        return DpexTypingContext()
 
     @property
     def target_context(self):

numba_dpex/core/targets/dpjit_target.py

@@ -7,6 +7,7 @@
 
 from functools import cached_property
 
+from numba.core import typing
 from numba.core.compiler_lock import global_compiler_lock
 from numba.core.cpu import CPUContext
 from numba.core.imputils import Registry
@@ -28,6 +29,10 @@ class Dpex(CPU):
 dpex_function_registry = Registry()
 
 
+class DpexTypingContext(typing.Context):
+    """Custom typing context to support dpjit compilation."""
+
+
 class DpexTargetContext(CPUContext):
     def __init__(self, typingctx, target=DPEX_TARGET_NAME):
         super().__init__(typingctx, target)