diff --git a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py index e514b2e088..83f78ef881 100644 --- a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py +++ b/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", diff --git a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py index df1e52eb0f..81f93638c5 100644 --- a/cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py +++ b/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", diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index 2ccc253ac8..2e08bdffae 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/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'", diff --git a/cuda_pathfinder/tests/test_find_nvidia_headers.py b/cuda_pathfinder/tests/test_find_nvidia_headers.py index a7b95e167b..a4ca8df602 100644 --- a/cuda_pathfinder/tests/test_find_nvidia_headers.py +++ b/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-.*$",