From 5d90597078311320e2d7bf4bfcd3b7d1a60f29f6 Mon Sep 17 00:00:00 2001 From: Diptorup Deb Date: Thu, 12 Oct 2023 12:30:42 -0500 Subject: [PATCH] Install enumdecl registry in DpexKernelTargetContext --- numba_dpex/core/targets/kernel_target.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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):