refresh

Author: abhilash1910

cuda_bindings/cuda/bindings/utils/_nvvm_utils.py

@@ -36,7 +36,7 @@ def check_nvvm_compiler_options(options: Sequence[str]) -> bool:
 
     Parameters
     ----------
-    options : Sequence[bytes]
+    options : Sequence[str]
         List of compiler options as strings (e.g., ["-arch=compute_90", "-g"]).
 
     Returns
@@ -46,7 +46,7 @@ def check_nvvm_compiler_options(options: Sequence[str]) -> bool:
 
     Examples
     --------
-    >>> from cuda.bindings.utils import check_nvvm_options
+    >>> from cuda.bindings.utils import check_nvvm_compiler_options
     >>> check_nvvm_options(["-arch=compute_90", "-g"])
     True
     """

cuda_bindings/tests/test_utils.py

@@ -22,11 +22,11 @@
 
 
 def _is_libnvvm_available() -> bool:
+    from cuda.bindings._internal.nvvm import _inspect_function_pointer
+    from cuda.pathfinder import DynamicLibNotFoundError
     try:
-        from cuda.bindings._internal.nvvm import _inspect_function_pointer
-
         return _inspect_function_pointer("__nvvmCreateProgram") != 0
-    except Exception:
+    except DynamicLibNotFoundError:
         return False
 
 

cuda_core/tests/test_program.py

@@ -72,11 +72,10 @@ def _has_nvrtc_pch_apis_for_tests():
 
 def _has_check_nvvm_compiler_options():
     try:
-        from cuda.bindings.utils import check_nvvm_compiler_options  # noqa: F401
-
-        return True
-    except ImportError:
+        import cuda.bindings.utils as utils
+    except ModuleNotFoundError:
         return False
+    return hasattr(utils, "check_nvvm_compiler_options")
 
 
 has_nvvm_option_checker = pytest.mark.skipif(