pathfinder: support cusolverMp dynamic loading and header searching (#1845)

* add lib & header descriptors for cusolverMp

* update find_nvidia_headers tester

* update dependency group

* sonames typo fix

* ruff pre-commit hook

* update path directories to favor cu13 assets

* mark cusolverMp header unavailable on windows

Author: yangcal

cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py

@@ -308,6 +308,13 @@ class DescriptorSpec:
         dependencies=("nvshmem_host",),
         requires_rtld_deepbind=True,
     ),
+    DescriptorSpec(
+        name="cusolverMp",
+        packaged_with="other",
+        linux_sonames=("libcusolverMp.so.0",),
+        site_packages_linux=("nvidia/cu13/lib", "nvidia/cu12/lib"),
+        dependencies=("cublas", "cudart", "cusolver", "nccl"),
+    ),
     DescriptorSpec(
         name="mathdx",
         packaged_with="other",

cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py

@@ -137,6 +137,15 @@ class HeaderDescriptorSpec:
     # -----------------------------------------------------------------------
     # Third-party / separately packaged headers
     # -----------------------------------------------------------------------
+    HeaderDescriptorSpec(
+        name="cusolverMp",
+        packaged_with="other",
+        header_basename="cusolverMp.h",
+        site_packages_dirs=("nvidia/cu13/include", "nvidia/cu12/include"),
+        available_on_windows=False,
+        conda_targets_layout=False,
+        use_ctk_root_canary=False,
+    ),
     HeaderDescriptorSpec(
         name="cusparseLt",
         packaged_with="other",

cuda_pathfinder/pyproject.toml

@@ -25,6 +25,7 @@ cu12 = [
     "nvidia-cublasmp-cu12; sys_platform != 'win32'",
     "nvidia-cudss-cu12",
     "nvidia-cufftmp-cu12; sys_platform != 'win32'",
+    "nvidia-cusolvermp-cu12; sys_platform != 'win32'",
     "nvidia-cusparselt-cu12",
     "nvidia-libmathdx-cu12",
     "nvidia-nccl-cu12; sys_platform != 'win32'",
@@ -37,6 +38,7 @@ cu13 = [
     "nvidia-cublasmp-cu13; sys_platform != 'win32'",
     "nvidia-cudss-cu13",
     "nvidia-cufftmp-cu13; sys_platform != 'win32'",
+    "nvidia-cusolvermp-cu13; sys_platform != 'win32'",
     "nvidia-cusparselt-cu13",
     "nvidia-libmathdx-cu13",
     "nvidia-nccl-cu13; sys_platform != 'win32'",

cuda_pathfinder/tests/test_find_nvidia_headers.py

@@ -40,6 +40,7 @@
 assert STRICTNESS in ("see_what_works", "all_must_work")
 
 NON_CTK_IMPORTLIB_METADATA_DISTRIBUTIONS_NAMES = {
+    "cusolverMp": r"^nvidia-cusolvermp-.*$",
     "cusparseLt": r"^nvidia-cusparselt-.*$",
     "cute": r"^nvidia-cutlass$",
     "cutensor": r"^cutensor-.*$",