Skip to content

Commit 7e26f1d

Browse files
authored
Add cuSPARSELt support (#1200)
* Add support for Linux site-packages * Add support for Windows site-packages * Add cusparselt to toolshed/conda_create_for_pathfinder_testing.* commands * Add cusparseLt in supported_nvidia_headers.py * Bump pathfinder version to 1.3.3a0
1 parent 0eb838f commit 7e26f1d

7 files changed

Lines changed: 12 additions & 1 deletion

File tree

cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -244,6 +244,7 @@
244244
"cufftMp": ("libcufftMp.so.12", "libcufftMp.so.11"),
245245
"mathdx": ("libmathdx.so.0",),
246246
"cudss": ("libcudss.so.0",),
247+
"cusparseLt": ("libcusparseLt.so.0",),
247248
"cutensor": ("libcutensor.so.2",),
248249
"cutensorMg": ("libcutensorMg.so.2",),
249250
"nccl": ("libnccl.so.2",),
@@ -407,6 +408,7 @@
407408
SUPPORTED_WINDOWS_DLLS_OTHER = {
408409
"mathdx": ("mathdx64_0.dll",),
409410
"cudss": ("cudss64_0.dll",),
411+
"cusparseLt": ("cusparseLt.dll",),
410412
"cutensor": ("cutensor.dll",),
411413
"cutensorMg": ("cutensorMg.dll",),
412414
}
@@ -454,6 +456,7 @@
454456
"cublasmp": ("nvidia/cublasmp/cu13/lib", "nvidia/cublasmp/cu12/lib"),
455457
"cudss": ("nvidia/cu13/lib", "nvidia/cu12/lib"),
456458
"cufftMp": ("nvidia/cufftmp/cu13/lib", "nvidia/cufftmp/cu12/lib"),
459+
"cusparseLt": ("nvidia/cusparselt/lib",),
457460
"cutensor": ("cutensor/lib",),
458461
"cutensorMg": ("cutensor/lib",),
459462
"mathdx": ("nvidia/cu13/lib", "nvidia/cu12/lib"),
@@ -495,6 +498,7 @@
495498
SITE_PACKAGES_LIBDIRS_WINDOWS_OTHER = {
496499
"cudss": ("nvidia/cu13/bin", "nvidia/cu12/bin"),
497500
"mathdx": ("nvidia/cu13/bin/x86_64", "nvidia/cu12/bin"),
501+
"cusparseLt": ("nvidia/cusparselt/bin",),
498502
"cutensor": ("cutensor/bin",),
499503
"cutensorMg": ("cutensor/bin",),
500504
}

cuda_pathfinder/cuda/pathfinder/_headers/supported_nvidia_headers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@
5959
}
6060

6161
SUPPORTED_HEADERS_NON_CTK_COMMON = {
62+
"cusparseLt": "cusparseLt.h",
6263
"cutensor": "cutensor.h",
6364
}
6465
SUPPORTED_HEADERS_NON_CTK_LINUX_ONLY = {
@@ -75,6 +76,7 @@
7576
)
7677

7778
SUPPORTED_SITE_PACKAGE_HEADER_DIRS_NON_CTK = {
79+
"cusparseLt": ("nvidia/cusparselt/include",),
7880
"cutensor": ("cutensor/include",),
7981
"nvshmem": ("nvidia/nvshmem/include",),
8082
}
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
# SPDX-License-Identifier: Apache-2.0
33

4-
__version__ = "1.3.2"
4+
__version__ = "1.3.3a0"

cuda_pathfinder/pyproject.toml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ cu12 = [
2222
"nvidia-cublasmp-cu12; sys_platform != 'win32'",
2323
"nvidia-cudss-cu12",
2424
"nvidia-cufftmp-cu12; sys_platform != 'win32'",
25+
"nvidia-cusparselt-cu12",
2526
"nvidia-libmathdx-cu12",
2627
"nvidia-nccl-cu12; sys_platform != 'win32'",
2728
"nvidia-nvshmem-cu12; sys_platform != 'win32'",
@@ -32,6 +33,7 @@ cu13 = [
3233
"cutensor-cu13",
3334
"nvidia-cublasmp-cu13; sys_platform != 'win32'",
3435
"nvidia-cudss-cu13",
36+
"nvidia-cusparselt-cu13",
3537
"nvidia-nccl-cu13; sys_platform != 'win32'",
3638
"nvidia-nvshmem-cu13; sys_platform != 'win32'",
3739
]

cuda_pathfinder/tests/test_find_nvidia_headers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
assert STRICTNESS in ("see_what_works", "all_must_work")
3434

3535
NON_CTK_IMPORTLIB_METADATA_DISTRIBUTIONS_NAMES = {
36+
"cusparseLt": r"^nvidia-cusparselt-.*$",
3637
"cutensor": r"^cutensor-.*$",
3738
"nvshmem": r"^nvidia-nvshmem-.*$",
3839
}

toolshed/conda_create_for_pathfinder_testing.ps1

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ conda create --yes -n "pathfinder_testing_cu$CudaVersion" python=3.13 "cuda-tool
1414
conda activate "pathfinder_testing_cu$CudaVersion"
1515

1616
$cpkgs = @(
17+
"cusparselt-dev",
1718
"cutensor",
1819
"libcublasmp-dev",
1920
"libcudss-dev",

toolshed/conda_create_for_pathfinder_testing.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ conda create --yes -n "pathfinder_testing_cu$1" python=3.13 cuda-toolkit="$1"
1414
conda activate "pathfinder_testing_cu$1"
1515

1616
for cpkg in \
17+
cusparselt-dev \
1718
cutensor \
1819
libcublasmp-dev \
1920
libcudss-dev \

0 commit comments

Comments
 (0)