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..0997cff4cf 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": ("cusparseLt.dll",), "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": ("nvidia/cusparselt/bin",), "cutensor": ("cutensor/bin",), "cutensorMg": ("cutensor/bin",), } 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/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" diff --git a/cuda_pathfinder/pyproject.toml b/cuda_pathfinder/pyproject.toml index 3db1aecbc8..956cab381c 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-cu13", "nvidia-nccl-cu13; sys_platform != 'win32'", "nvidia-nvshmem-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 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 a2c6c31ef0..115720f6e5 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-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 2433787581..1ed57e6765 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-dev \ cutensor \ libcublasmp-dev \ libcudss-dev \