Skip to content

pathfinder: support cusolverMp dynamic loading and header searching #1845

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
rwgk merged 9 commits into from
Apr 1, 2026
+19 −0
Merged

pathfinder: support cusolverMp dynamic loading and header searching #1845

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
d471a77
add lib & header descriptors for cusolverMp
yangcal Apr 1, 2026
25f54ae
update find_nvidia_headers tester
yangcal Apr 1, 2026
066ebc0
update dependency group
yangcal Apr 1, 2026
5899014
sonames typo fix
yangcal Apr 1, 2026
41e5cda
ruff pre-commit hook
yangcal Apr 1, 2026
3e45960
update path directories to favor cu13 assets
yangcal Apr 1, 2026
89d6a58
Merge branch 'NVIDIA:main' into pathfinder-cusolverMp
yangcal Apr 1, 2026
4087a47
mark cusolverMp header unavailable on windows
yangcal Apr 1, 2026
b5bb1e9
Merge branch 'pathfinder-cusolverMp' of github.com:yangcal/cuda-pytho…
yangcal Apr 1, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
9 changes: 9 additions & 0 deletions cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
2 changes: 2 additions & 0 deletions cuda_pathfinder/pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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'",
Expand All @@ -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'",
Expand Down
1 change: 1 addition & 0 deletions cuda_pathfinder/tests/test_find_nvidia_headers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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-.*$",
Expand Down
Loading