dont import numba cuda indirectly via cython

rapidsai · Nov 18, 2024 · d508aa0 · d508aa0
1 parent e29d055
commit d508aa0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git 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
@@ -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: