Vendor in imputils module

Author: ashermancinelli

numba_cuda/numba/cuda/_internal/cuda_bf16.py

@@ -25,8 +25,8 @@
     make_attribute_wrapper,
     register_model,
 )
-from numba.core.imputils import Registry as TargetRegistry
-from numba.core.imputils import lower_cast
+from numba.cuda.core.imputils import Registry as TargetRegistry
+from numba.cuda.core.imputils import lower_cast
 from numba.core.typing import signature
 from numba.core.typing.builtins import (
     BinOp,

numba_cuda/numba/cuda/_internal/cuda_fp16.py

@@ -27,8 +27,8 @@
     make_attribute_wrapper,
     register_model,
 )
-from numba.core.imputils import Registry as TargetRegistry
-from numba.core.imputils import lower_cast
+from numba.cuda.core.imputils import Registry as TargetRegistry
+from numba.cuda.core.imputils import lower_cast
 from numba.core.typing import signature
 from numba.cuda import CUSource, declare_device
 from numba.cuda._internal.cuda_bf16 import _type___nv_bfloat16

numba_cuda/numba/cuda/core/base.py

@@ -16,15 +16,14 @@
     types,
     datamodel,
     config,
-    imputils,
 )
 from numba.cuda import cgutils, debuginfo, utils
 from numba.core import errors
-from numba.cuda.core import targetconfig, funcdesc
+from numba.cuda.core import targetconfig, funcdesc, imputils
 from numba import _dynfunc, _helperlib
 from numba.core.compiler_lock import global_compiler_lock
 from numba.cuda.core.pythonapi import PythonAPI
-from numba.core.imputils import (
+from numba.cuda.core.imputils import (
     user_function,
     user_generator,
     builtin_registry,