diff --git a/python/cudf/cudf/_lib/aggregation.pyx b/python/cudf/cudf/_lib/aggregation.pyx index 3c96b90f0a1..c2f01183689 100644 --- a/python/cudf/cudf/_lib/aggregation.pyx +++ b/python/cudf/cudf/_lib/aggregation.pyx @@ -7,7 +7,6 @@ import pylibcudf import cudf from cudf._lib.types import SUPPORTED_NUMPY_TO_PYLIBCUDF_TYPES -from cudf.utils import cudautils _agg_name_map = { "COUNT_VALID": "COUNT", @@ -196,7 +195,7 @@ class Aggregation: # Handling UDF type nb_type = numpy_support.from_dtype(kwargs['dtype']) type_signature = (nb_type[:],) - ptx_code, output_dtype = cudautils.compile_udf(op, type_signature) + ptx_code, output_dtype = cudf.utils.cudautils.compile_udf(op, type_signature) output_np_dtype = cudf.dtype(output_dtype) if output_np_dtype not in SUPPORTED_NUMPY_TO_PYLIBCUDF_TYPES: raise TypeError(f"Result of window function has unsupported dtype {op[1]}") diff --git a/python/cudf/cudf/utils/_numba.py b/python/cudf/cudf/utils/_numba.py index d9dde58d998..d6dedc1783a 100644 --- a/python/cudf/cudf/utils/_numba.py +++ b/python/cudf/cudf/utils/_numba.py @@ -130,9 +130,7 @@ def _setup_numba(): if driver_version < (12, 0): patch_numba_linker_cuda_11() else: - from pynvjitlink.patch import patch_numba_linker - - patch_numba_linker() + numba_config.CUDA_ENABLE_PYNVJITLINK = True class _CUDFNumbaConfig: