From 392dcb081f32215c23dd1f8b5c7059650a0a594d Mon Sep 17 00:00:00 2001 From: "Ralf W. Grosse-Kunstleve" Date: Wed, 29 Oct 2025 17:24:28 -0700 Subject: [PATCH 1/5] Add support for Linux site-packages --- .../cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py | 4 ++++ cuda_pathfinder/pyproject.toml | 2 ++ 2 files changed, 6 insertions(+) diff --git a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py index c205b5eda4..96c1394623 100644 --- a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py +++ b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py @@ -244,6 +244,7 @@ "cufftMp": ("libcufftMp.so.12", "libcufftMp.so.11"), "mathdx": ("libmathdx.so.0",), "cudss": ("libcudss.so.0",), + "cusparseLt": ("libcusparseLt.so.0",), "cutensor": ("libcutensor.so.2",), "cutensorMg": ("libcutensorMg.so.2",), "nccl": ("libnccl.so.2",), @@ -407,6 +408,7 @@ SUPPORTED_WINDOWS_DLLS_OTHER = { "mathdx": ("mathdx64_0.dll",), "cudss": ("cudss64_0.dll",), + "cusparseLt": ("TODO",), "cutensor": ("cutensor.dll",), "cutensorMg": ("cutensorMg.dll",), } @@ -454,6 +456,7 @@ "cublasmp": ("nvidia/cublasmp/cu13/lib", "nvidia/cublasmp/cu12/lib"), "cudss": ("nvidia/cu13/lib", "nvidia/cu12/lib"), "cufftMp": ("nvidia/cufftmp/cu13/lib", "nvidia/cufftmp/cu12/lib"), + "cusparseLt": ("nvidia/cusparselt/lib",), "cutensor": ("cutensor/lib",), "cutensorMg": ("cutensor/lib",), "mathdx": ("nvidia/cu13/lib", "nvidia/cu12/lib"), @@ -495,6 +498,7 @@ SITE_PACKAGES_LIBDIRS_WINDOWS_OTHER = { "cudss": ("nvidia/cu13/bin", "nvidia/cu12/bin"), "mathdx": ("nvidia/cu13/bin/x86_64", "nvidia/cu12/bin"), + "cusparseLt": ("TODO"), "cutensor": ("cutensor/bin",), "cutensorMg": ("cutensor/bin",), } diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index 3db1aecbc8..8eb0af1383 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/cuda_pathfinder/pyproject.toml @@ -22,6 +22,7 @@ cu12 = [ "nvidia-cublasmp-cu12; sys_platform != 'win32'", "nvidia-cudss-cu12", "nvidia-cufftmp-cu12; sys_platform != 'win32'", + "nvidia-cusparselt-cu12", "nvidia-libmathdx-cu12", "nvidia-nccl-cu12; sys_platform != 'win32'", "nvidia-nvshmem-cu12; sys_platform != 'win32'", @@ -32,6 +33,7 @@ cu13 = [ "cutensor-cu13", "nvidia-cublasmp-cu13; sys_platform != 'win32'", "nvidia-cudss-cu13", + "nvidia-cusparselt-cu12", "nvidia-nccl-cu13; sys_platform != 'win32'", "nvidia-nvshmem-cu13; sys_platform != 'win32'", ] From 26460151498645f8e66afa8e259afd1f3d162df9 Mon Sep 17 00:00:00 2001 From: "Ralf W. Grosse-Kunstleve" Date: Wed, 29 Oct 2025 17:41:51 -0700 Subject: [PATCH 2/5] Add support for Windows site-packages --- .../cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py | 4 ++-- cuda_pathfinder/pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py index 96c1394623..0997cff4cf 100644 --- a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py +++ b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py @@ -408,7 +408,7 @@ SUPPORTED_WINDOWS_DLLS_OTHER = { "mathdx": ("mathdx64_0.dll",), "cudss": ("cudss64_0.dll",), - "cusparseLt": ("TODO",), + "cusparseLt": ("cusparseLt.dll",), "cutensor": ("cutensor.dll",), "cutensorMg": ("cutensorMg.dll",), } @@ -498,7 +498,7 @@ SITE_PACKAGES_LIBDIRS_WINDOWS_OTHER = { "cudss": ("nvidia/cu13/bin", "nvidia/cu12/bin"), "mathdx": ("nvidia/cu13/bin/x86_64", "nvidia/cu12/bin"), - "cusparseLt": ("TODO"), + "cusparseLt": ("nvidia/cusparselt/bin",), "cutensor": ("cutensor/bin",), "cutensorMg": ("cutensor/bin",), } diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index 8eb0af1383..956cab381c 100644 --- a/cuda_pathfinder/pyproject.toml +++ b/cuda_pathfinder/pyproject.toml @@ -33,7 +33,7 @@ cu13 = [ "cutensor-cu13", "nvidia-cublasmp-cu13; sys_platform != 'win32'", "nvidia-cudss-cu13", - "nvidia-cusparselt-cu12", + "nvidia-cusparselt-cu13", "nvidia-nccl-cu13; sys_platform != 'win32'", "nvidia-nvshmem-cu13; sys_platform != 'win32'", ] From 3ee040a824fb094a62423133cbfaccee9cbbf77f Mon Sep 17 00:00:00 2001 From: "Ralf W. Grosse-Kunstleve" Date: Wed, 29 Oct 2025 17:52:07 -0700 Subject: [PATCH 3/5] Add cusparselt to toolshed/conda_create_for_pathfinder_testing.* commands --- toolshed/conda_create_for_pathfinder_testing.ps1 | 1 + toolshed/conda_create_for_pathfinder_testing.sh | 1 + 2 files changed, 2 insertions(+) diff --git a/toolshed/conda_create_for_pathfinder_testing.ps1 b/toolshed/conda_create_for_pathfinder_testing.ps1 index a2c6c31ef0..8aa1690f68 100644 --- a/toolshed/conda_create_for_pathfinder_testing.ps1 +++ b/toolshed/conda_create_for_pathfinder_testing.ps1 @@ -14,6 +14,7 @@ conda create --yes -n "pathfinder_testing_cu$CudaVersion" python=3.13 "cuda-tool conda activate "pathfinder_testing_cu$CudaVersion" $cpkgs = @( + "cusparselt", "cutensor", "libcublasmp-dev", "libcudss-dev", diff --git a/toolshed/conda_create_for_pathfinder_testing.sh b/toolshed/conda_create_for_pathfinder_testing.sh index 2433787581..cc4cda1fde 100755 --- a/toolshed/conda_create_for_pathfinder_testing.sh +++ b/toolshed/conda_create_for_pathfinder_testing.sh @@ -14,6 +14,7 @@ conda create --yes -n "pathfinder_testing_cu$1" python=3.13 cuda-toolkit="$1" conda activate "pathfinder_testing_cu$1" for cpkg in \ + cusparselt \ cutensor \ libcublasmp-dev \ libcudss-dev \ From 40de332dfa8019494fe2e76df8041440f405c1f7 Mon Sep 17 00:00:00 2001 From: "Ralf W. Grosse-Kunstleve" Date: Wed, 29 Oct 2025 19:12:40 -0700 Subject: [PATCH 4/5] Add cusparseLt in supported_nvidia_headers.py --- .../cuda/pathfinder/_headers/supported_nvidia_headers.py | 2 ++ cuda_pathfinder/tests/test_find_nvidia_headers.py | 1 + toolshed/conda_create_for_pathfinder_testing.ps1 | 2 +- toolshed/conda_create_for_pathfinder_testing.sh | 2 +- 4 files changed, 5 insertions(+), 2 deletions(-) diff --git a/cuda_pathfinder/cuda/pathfinder/_headers/supported_nvidia_headers.py b/cuda_pathfinder/cuda/pathfinder/_headers/supported_nvidia_headers.py index 7fdf4159ff..af29551dd8 100644 --- a/cuda_pathfinder/cuda/pathfinder/_headers/supported_nvidia_headers.py +++ b/cuda_pathfinder/cuda/pathfinder/_headers/supported_nvidia_headers.py @@ -59,6 +59,7 @@ } SUPPORTED_HEADERS_NON_CTK_COMMON = { + "cusparseLt": "cusparseLt.h", "cutensor": "cutensor.h", } SUPPORTED_HEADERS_NON_CTK_LINUX_ONLY = { @@ -75,6 +76,7 @@ ) SUPPORTED_SITE_PACKAGE_HEADER_DIRS_NON_CTK = { + "cusparseLt": ("nvidia/cusparselt/include",), "cutensor": ("cutensor/include",), "nvshmem": ("nvidia/nvshmem/include",), } diff --git a/cuda_pathfinder/tests/test_find_nvidia_headers.py b/cuda_pathfinder/tests/test_find_nvidia_headers.py index bbc920d765..494d7c0ae9 100644 --- a/cuda_pathfinder/tests/test_find_nvidia_headers.py +++ b/cuda_pathfinder/tests/test_find_nvidia_headers.py @@ -33,6 +33,7 @@ assert STRICTNESS in ("see_what_works", "all_must_work") NON_CTK_IMPORTLIB_METADATA_DISTRIBUTIONS_NAMES = { + "cusparseLt": r"^nvidia-cusparselt-.*$", "cutensor": r"^cutensor-.*$", "nvshmem": r"^nvidia-nvshmem-.*$", } diff --git a/toolshed/conda_create_for_pathfinder_testing.ps1 b/toolshed/conda_create_for_pathfinder_testing.ps1 index 8aa1690f68..115720f6e5 100644 --- a/toolshed/conda_create_for_pathfinder_testing.ps1 +++ b/toolshed/conda_create_for_pathfinder_testing.ps1 @@ -14,7 +14,7 @@ conda create --yes -n "pathfinder_testing_cu$CudaVersion" python=3.13 "cuda-tool conda activate "pathfinder_testing_cu$CudaVersion" $cpkgs = @( - "cusparselt", + "cusparselt-dev", "cutensor", "libcublasmp-dev", "libcudss-dev", diff --git a/toolshed/conda_create_for_pathfinder_testing.sh b/toolshed/conda_create_for_pathfinder_testing.sh index cc4cda1fde..1ed57e6765 100755 --- a/toolshed/conda_create_for_pathfinder_testing.sh +++ b/toolshed/conda_create_for_pathfinder_testing.sh @@ -14,7 +14,7 @@ conda create --yes -n "pathfinder_testing_cu$1" python=3.13 cuda-toolkit="$1" conda activate "pathfinder_testing_cu$1" for cpkg in \ - cusparselt \ + cusparselt-dev \ cutensor \ libcublasmp-dev \ libcudss-dev \ From ba1e17f98dad61a09584f9607a019f608789a68f Mon Sep 17 00:00:00 2001 From: "Ralf W. Grosse-Kunstleve" Date: Wed, 29 Oct 2025 21:42:50 -0700 Subject: [PATCH 5/5] Bump pathfinder version to 1.3.3a0 --- cuda_pathfinder/cuda/pathfinder/_version.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cuda_pathfinder/cuda/pathfinder/_version.py b/cuda_pathfinder/cuda/pathfinder/_version.py index b5fa5ff85c..7c72217355 100644 --- a/cuda_pathfinder/cuda/pathfinder/_version.py +++ b/cuda_pathfinder/cuda/pathfinder/_version.py @@ -1,4 +1,4 @@ # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 -__version__ = "1.3.2" +__version__ = "1.3.3a0"