Skip to content

Commit 060e2dc

Browse files
committed
supported_nvidia_libs.py updates: add nvidia-cublasmp-cu12, nvidia-cublasmp-cu13, nvidia-cudss-cu13
1 parent b09d7ed commit 060e2dc

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,6 +237,7 @@
237237
),
238238
}
239239
SUPPORTED_LINUX_SONAMES_OTHER = {
240+
"cublasmp": ("libcublasmp.so.0",),
240241
"cufftMp": ("libcufftMp.so.11",),
241242
"mathdx": ("libmathdx.so.0",),
242243
"cudss": ("libcudss.so.0",),
@@ -443,7 +444,8 @@
443444
"nvvm": ("nvidia/cu13/lib", "nvidia/cuda_nvcc/nvvm/lib64"),
444445
}
445446
SITE_PACKAGES_LIBDIRS_LINUX_OTHER = {
446-
"cudss": ("nvidia/cu12/lib",),
447+
"cublasmp": ("nvidia/cublasmp/cu13/lib", "nvidia/cublasmp/cu12/lib"),
448+
"cudss": ("nvidia/cu13/lib", "nvidia/cu12/lib"),
447449
"cufftMp": ("nvidia/cufftmp/cu12/lib",),
448450
"mathdx": ("nvidia/cu13/lib", "nvidia/cu12/lib"),
449451
"nccl": ("nvidia/nccl/lib",),

cuda_pathfinder/pyproject.toml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ test = [
1717
test_nvidia_wheels_cu12 = [
1818
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl]==12.*",
1919
"cuda-toolkit[cufile]==12.*; sys_platform != 'win32'",
20+
"nvidia-cublasmp-cu12; sys_platform != 'win32'",
2021
"nvidia-cudss-cu12",
2122
"nvidia-cufftmp-cu12; sys_platform != 'win32'",
2223
"nvidia-libmathdx-cu12",
@@ -26,6 +27,8 @@ test_nvidia_wheels_cu12 = [
2627
test_nvidia_wheels_cu13 = [
2728
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg,cccl,nvvm]==13.*",
2829
"cuda-toolkit[cufile]==13.*; sys_platform != 'win32'",
30+
"nvidia-cublasmp-cu13; sys_platform != 'win32'",
31+
"nvidia-cudss-cu13",
2932
"nvidia-nccl-cu13; sys_platform != 'win32'",
3033
"nvidia-nvshmem-cu13; sys_platform != 'win32'",
3134
]

0 commit comments

Comments
 (0)