diff --git a/numba_dpex/core/targets/kernel_target.py b/numba_dpex/core/targets/kernel_target.py index 1a52d68502..f3e1a703a5 100644 --- a/numba_dpex/core/targets/kernel_target.py +++ b/numba_dpex/core/targets/kernel_target.py @@ -87,7 +87,7 @@ 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 @@ -95,6 +95,7 @@ def load_additional_registries(self): self.install_registry(mathdecl.registry) self.install_registry(cmathdecl.registry) self.install_registry(npydecl.registry) + self.install_registry(enumdecl.registry) class SyclDevice(GPU):