Add names of all CTK 12.8.1 x86_64-linux libraries (.so) as path_finder.SUPPORTED_LIBNAMES

https://chatgpt.com/share/67f98d0b-148c-8008-9951-9995cf5d860c

Author: rwgk

cuda_bindings/cuda/bindings/path_finder.py

@@ -34,4 +34,43 @@
     "get_nvidia_nvvm_ctk",
     "get_nvidia_static_cudalib_ctk",
     "get_system_ctk",
+    "SUPPORTED_LIBNAMES",
 ]
+
+SUPPORTED_LIBNAMES = (
+    # Core CUDA Runtime and Compiler
+    "cudart",
+    "nvfatbin",
+    "nvJitLink",
+    "nvrtc",
+    "nvvm",
+    # Math Libraries
+    "cublas",
+    "cufft",
+    "curand",
+    "cusolver",
+    "cusparse",
+    "nppc",
+    "nppial",
+    "nppicc",
+    "nppidei",
+    "nppif",
+    "nppig",
+    "nppim",
+    "nppist",
+    "nppisu",
+    "nppitc",
+    "npps",
+    "nvblas",
+    # Profiling and Developer Tools
+    "cupti",
+    "nvperf_host",
+    "nvperf_target",
+    "nvToolsExt",
+    # Debugging and Introspection
+    "accinj64",
+    "cuinj64",
+    "checkpoint",
+    # Miscellaneous
+    "OpenCL",
+)

cuda_bindings/tests/path_finder.py

@@ -6,13 +6,12 @@
     print(f"{k}: {v}", flush=True)
 print()
 
-libnames = ("nvJitLink", "nvrtc", "nvvm")
-
-for libname in libnames:
-    print(path_finder.find_nvidia_dynamic_library(libname))
-    print()
-
-for libname in libnames:
+for libname in path_finder.SUPPORTED_LIBNAMES:
     print(libname)
-    print(path_finder.load_nvidia_dynamic_library(libname))
+    for fun in (path_finder.find_nvidia_dynamic_library, path_finder.load_nvidia_dynamic_library):
+        try:
+            out = fun(libname)
+        except Exception as e:
+            out = f"EXCEPTION: {type(e)} {str(e)}"
+        print(out)
     print()

cuda_bindings/tests/test_path_finder.py

@@ -0,0 +1,15 @@
+import pytest
+
+from cuda.bindings import path_finder
+
+
+@pytest.mark.parametrize("libname", path_finder.SUPPORTED_LIBNAMES)
+def test_find_and_load(libname):
+    print(f"\n{libname}")
+    for fun in (path_finder.find_nvidia_dynamic_library, path_finder.load_nvidia_dynamic_library):
+        try:
+            out = fun(libname)
+        except Exception as e:
+            out = f"EXCEPTION: {type(e)} {str(e)}"
+        print(out)
+    print()