Skip to content

Commit 0d71a85

Browse files
committed
Add nvidia-cufftmp-cu13 data in supported_nvidia_libs.py
Manually tested with: pip install nvidia-cufftmp-cu13==12.1.3.1 That wheel was yanked, therefore not adding to pyproject.toml
1 parent 3a44b6a commit 0d71a85

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@
241241
}
242242
SUPPORTED_LINUX_SONAMES_OTHER = {
243243
"cublasmp": ("libcublasmp.so.0",),
244-
"cufftMp": ("libcufftMp.so.11",),
244+
"cufftMp": ("libcufftMp.so.12", "libcufftMp.so.11"),
245245
"mathdx": ("libmathdx.so.0",),
246246
"cudss": ("libcudss.so.0",),
247247
"cutensor": ("libcutensor.so.2",),
@@ -453,7 +453,7 @@
453453
SITE_PACKAGES_LIBDIRS_LINUX_OTHER = {
454454
"cublasmp": ("nvidia/cublasmp/cu13/lib", "nvidia/cublasmp/cu12/lib"),
455455
"cudss": ("nvidia/cu13/lib", "nvidia/cu12/lib"),
456-
"cufftMp": ("nvidia/cufftmp/cu12/lib",),
456+
"cufftMp": ("nvidia/cufftmp/cu13/lib", "nvidia/cufftmp/cu12/lib"),
457457
"cutensor": ("cutensor/lib",),
458458
"cutensorMg": ("cutensor/lib",),
459459
"mathdx": ("nvidia/cu13/lib", "nvidia/cu12/lib"),

0 commit comments

Comments
 (0)