use numba config to turn on pynvjitlink

rapidsai · Dec 19, 2024 · 5468916 · 5468916
1 parent a081a57
commit 5468916
Showing 1 changed file with 1 addition and 3 deletions.
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: