enable LTO linker items only for cuda 12

NVIDIA · Oct 24, 2024 · b42c67d · b42c67d
1 parent 27b304d
commit b42c67d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/numba_cuda/numba/cuda/cudadrv/nvrtc.py b/numba_cuda/numba/cuda/cudadrv/nvrtc.py
@@ -3,7 +3,6 @@
 from numba.core import config
 from numba.cuda.cudadrv.error import (NvrtcError, NvrtcCompilationError,
                                       NvrtcSupportError)
-from cuda.cuda.cudadrv.driver import get_version
 
 import functools
 import os
@@ -106,9 +105,6 @@ class NVRTC:
         'nvrtcGetProgramLog': (nvrtc_result, nvrtc_program, c_char_p),
     }
 
-    if get_version() >= (12, 0):
-        _PROTOTYPES |= _CU12ONLY_PROTOTYPES
-
     # Singleton reference
     __INSTANCE = None
 
@@ -123,6 +119,10 @@ def __new__(cls):
                     cls.__INSTANCE = None
                     raise NvrtcSupportError("NVRTC cannot be loaded") from e
 
+                from numba.cuda.cudadrv.runtime import get_version
+                if get_version() >= (12, 0):
+                    inst._PROTOTYPES |= inst._CU12ONLY_PROTOTYPES
+
                 # Find & populate functions
                 for name, proto in inst._PROTOTYPES.items():
                     func = getattr(lib, name)