File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
cuda_core/cuda/core/experimental Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change 2929_nvjitlink_input_types = None # populated if nvJitLink cannot be used
3030
3131
32+ def _nvjitlink_has_version_symbol (inner_nvjitlink ) -> bool :
33+ # This condition is equivalent to testing for version >= 12.3
34+ return bool (inner_nvjitlink ._inspect_function_pointer ("__nvJitLinkVersion" ))
35+
36+
3237# Note: this function is reused in the tests
3338def _decide_nvjitlink_or_driver () -> bool :
3439 """Returns True if falling back to the cuLink* driver APIs."""
@@ -53,7 +58,7 @@ def _decide_nvjitlink_or_driver() -> bool:
5358 from cuda .bindings ._internal import nvjitlink as inner_nvjitlink
5459
5560 try :
56- if inner_nvjitlink . _inspect_function_pointer ( "__nvJitLinkVersion" ):
61+ if _nvjitlink_has_version_symbol ( inner_nvjitlink ):
5762 return False # Use nvjitlink
5863 except RuntimeError :
5964 warn_detail = "not available"
You can’t perform that action at this time.
0 commit comments