Add cusparseLt in supported_nvidia_headers.py

rwgk · rwgk · commit 40de332dfa80 · 2025-10-29T19:12:40.000-07:00
diff --git 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
@@ -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
@@ -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
@@ -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 \

Original file line number	Diff line number	Diff line change
`@@ -59,6 +59,7 @@`
`59`	`59`	`}`
`60`	`60`
`61`	`61`	`SUPPORTED_HEADERS_NON_CTK_COMMON = {`
	`62`	`+ "cusparseLt": "cusparseLt.h",`
`62`	`63`	`"cutensor": "cutensor.h",`
`63`	`64`	`}`
`64`	`65`	`SUPPORTED_HEADERS_NON_CTK_LINUX_ONLY = {`
`@@ -75,6 +76,7 @@`
`75`	`76`	`)`
`76`	`77`
`77`	`78`	`SUPPORTED_SITE_PACKAGE_HEADER_DIRS_NON_CTK = {`
	`79`	`+ "cusparseLt": ("nvidia/cusparselt/include",),`
`78`	`80`	`"cutensor": ("cutensor/include",),`
`79`	`81`	`"nvshmem": ("nvidia/nvshmem/include",),`
`80`	`82`	`}`
Original file line number	Diff line number	Diff line change
`@@ -33,6 +33,7 @@`
`33`	`33`	`assert STRICTNESS in ("see_what_works", "all_must_work")`
`34`	`34`
`35`	`35`	`NON_CTK_IMPORTLIB_METADATA_DISTRIBUTIONS_NAMES = {`
	`36`	`+ "cusparseLt": r"^nvidia-cusparselt-.*$",`
`36`	`37`	`"cutensor": r"^cutensor-.*$",`
`37`	`38`	`"nvshmem": r"^nvidia-nvshmem-.*$",`
`38`	`39`	`}`