Install enumdecl registry in DpexKernelTargetContext

IntelPython · Oct 12, 2023 · 5d90597 · 5d90597
1 parent de15cd5
commit 5d90597
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/numba_dpex/core/targets/kernel_target.py b/numba_dpex/core/targets/kernel_target.py
@@ -87,14 +87,15 @@ def resolve_argument_type(self, val):
 
     def load_additional_registries(self):
         """Register the OpenCL API and math and other functions."""
-        from numba.core.typing import cmathdecl, npydecl
+        from numba.core.typing import cmathdecl, enumdecl, npydecl
 
         from ...ocl import mathdecl, ocldecl
 
         self.install_registry(ocldecl.registry)
         self.install_registry(mathdecl.registry)
         self.install_registry(cmathdecl.registry)
         self.install_registry(npydecl.registry)
+        self.install_registry(enumdecl.registry)
 
 
 class SyclDevice(GPU):