Change to "nppc" as ANCHOR_LIBNAME

Author: rwgk

cuda_bindings/cuda/bindings/_path_finder/find_nvidia_dynamic_library.py

@@ -108,14 +108,14 @@ def _find_dll_using_lib_dir(lib_dir, libname, error_messages, attachments):
     return None
 
 
-def _find_nvvm_lib_dir_from_other_abs_path(other_abs_path):
+def _find_nvvm_lib_dir_from_anchor_abs_path(anchor_abs_path):
     nvvm_subdir = "bin" if IS_WINDOWS else "lib64"
-    while other_abs_path:
-        if os.path.isdir(other_abs_path):
-            nvvm_lib_dir = os.path.join(other_abs_path, "nvvm", nvvm_subdir)
+    while anchor_abs_path:
+        if os.path.isdir(anchor_abs_path):
+            nvvm_lib_dir = os.path.join(anchor_abs_path, "nvvm", nvvm_subdir)
             if os.path.isdir(nvvm_lib_dir):
                 return nvvm_lib_dir
-        other_abs_path = os.path.dirname(other_abs_path)
+        anchor_abs_path = os.path.dirname(anchor_abs_path)
     return None
 
 
@@ -148,9 +148,9 @@ def __init__(self, libname: str):
                     libname, self.lib_searched_for, self.error_messages, self.attachments
                 )
 
-    def retry_with_other_abs_path(self, other_abs_path):
+    def retry_with_anchor_abs_path(self, anchor_abs_path):
         assert self.libname == "nvvm"
-        nvvm_lib_dir = _find_nvvm_lib_dir_from_other_abs_path(other_abs_path)
+        nvvm_lib_dir = _find_nvvm_lib_dir_from_anchor_abs_path(anchor_abs_path)
         if nvvm_lib_dir is None:
             return
         if IS_WINDOWS:

cuda_bindings/cuda/bindings/_path_finder/load_nvidia_dynamic_library.py

@@ -26,8 +26,11 @@
         load_with_system_search,
     )
 
+# "nvvm" is found from this anchor
+ANCHOR_LIBNAME = "nppc"  # libnppc.so.12 1.6M, nppc64_12.dll 288K
 
-def _load_other_in_subprocess(libname, error_messages):
+
+def _load_anchor_in_subprocess(libname, error_messages):
     code = f"""\
 from cuda.bindings._path_finder.load_nvidia_dynamic_library import load_nvidia_dynamic_library
 import json
@@ -60,14 +63,13 @@ def _load_nvidia_dynamic_library_no_cache(libname: str) -> LoadedDL:
         if loaded is not None:
             return loaded
         if libname == "nvvm":
-            # Use cudart as anchor point (libcudart.so.12 is only ~720K, cudart64_12.dll ~560K).
-            loaded_cudart = check_if_already_loaded_from_elsewhere("cudart")
-            if loaded_cudart is not None:
-                found.retry_with_other_abs_path(loaded_cudart.abs_path)
+            loaded_anchor = check_if_already_loaded_from_elsewhere(ANCHOR_LIBNAME)
+            if loaded_anchor is not None:
+                found.retry_with_anchor_abs_path(loaded_anchor.abs_path)
             else:
-                cudart_abs_path = _load_other_in_subprocess("cudart", found.error_messages)
-                if cudart_abs_path is not None:
-                    found.retry_with_other_abs_path(cudart_abs_path)
+                anchor_abs_path = _load_anchor_in_subprocess(ANCHOR_LIBNAME, found.error_messages)
+                if anchor_abs_path is not None:
+                    found.retry_with_anchor_abs_path(anchor_abs_path)
         found.raise_if_abs_path_is_None()
 
     # Load the library from the found path