Load upstream registries if numba is available

Author: ashermancinelli

numba_cuda/numba/cuda/target.py

@@ -6,6 +6,7 @@
 import llvmlite.binding as ll
 from llvmlite import ir
 import warnings
+import importlib.util
 
 from numba.core import types
 
@@ -216,6 +217,13 @@ def load_additional_registries(self):
         self.install_registry(npdatetime.registry)
         self.install_registry(arrayobj.registry)
 
+        if importlib.util.find_spec("numba.core.imputils") is not None:
+            from numba.core.imputils import (
+                builtin_registry as upstream_builtin_registry,
+            )
+
+            self.install_registry(upstream_builtin_registry)
+
     def codegen(self):
         return self._internal_codegen