Skip to content

Commit 262ed52

Browse files
author
Diptorup Deb
committed
Install enumdecl registry in DpexKernelTargetContext
1 parent aa787db commit 262ed52

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

numba_dpex/core/targets/kernel_target.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,14 +87,15 @@ def resolve_argument_type(self, val):
8787

8888
def load_additional_registries(self):
8989
"""Register the OpenCL API and math and other functions."""
90-
from numba.core.typing import cmathdecl, npydecl
90+
from numba.core.typing import cmathdecl, enumdecl, npydecl
9191

9292
from ...ocl import mathdecl, ocldecl
9393

9494
self.install_registry(ocldecl.registry)
9595
self.install_registry(mathdecl.registry)
9696
self.install_registry(cmathdecl.registry)
9797
self.install_registry(npydecl.registry)
98+
self.install_registry(enumdecl.registry)
9899

99100

100101
class SyclDevice(GPU):

0 commit comments

Comments
 (0)