From 5468916e935f7323873023896d1a9dee2486e309 Mon Sep 17 00:00:00 2001 From: isVoid Date: Thu, 19 Dec 2024 08:22:57 +0000 Subject: [PATCH] use numba config to turn on pynvjitlink --- python/cudf/cudf/utils/_numba.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) 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: