Skip to content

Commit 62e2650

Browse files
authored
Add nccl in supported_nvidia_libs.py (#945)
* Add nvidia-nccl-cu12 to cuda_pathfinder/pyproject.toml nvidia_wheels_cu12 * Add nccl in supported_nvidia_libs.py * Bump pathfinder version to 1.2.2a0
1 parent 559f749 commit 62e2650

3 files changed

Lines changed: 5 additions & 2 deletions

File tree

cuda_pathfinder/cuda/pathfinder/_dynamic_libs/supported_nvidia_libs.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -242,6 +242,7 @@
242242
"cufftMp": ("libcufftMp.so.11",),
243243
"mathdx": ("libmathdx.so.0",),
244244
"cudss": ("libcudss.so.0",),
245+
"nccl": ("libnccl.so.2",),
245246
"nvpl_fftw": ("libnvpl_fftw.so.0",),
246247
"nvshmem_host": ("libnvshmem_host.so.3",),
247248
}
@@ -447,6 +448,7 @@
447448
"cudss": ("nvidia/cu12/lib",),
448449
"cufftMp": ("nvidia/cufftmp/cu12/lib",),
449450
"mathdx": ("nvidia/cu13/lib", "nvidia/cu12/lib"),
451+
"nccl": ("nvidia/nccl/lib",),
450452
"nvpl_fftw": ("nvpl/lib",),
451453
"nvshmem_host": ("nvidia/nvshmem/lib",),
452454
}
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.2.1"
4+
__version__ = "1.2.2a0"

cuda_pathfinder/pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,10 @@ test = [
1717
nvidia_wheels_cu12 = [
1818
"cuda-toolkit[nvcc,cublas,nvrtc,cudart,cufft,curand,cusolver,cusparse,npp,nvfatbin,nvjitlink,nvjpeg]==12.*",
1919
"cuda-toolkit[cufile]==12.*; sys_platform != 'win32'",
20-
"nvidia-libmathdx-cu12",
2120
"nvidia-cudss-cu12",
2221
"nvidia-cufftmp-cu12; sys_platform != 'win32'",
22+
"nvidia-libmathdx-cu12",
23+
"nvidia-nccl-cu12; sys_platform != 'win32'",
2324
"nvidia-nvshmem-cu12; sys_platform != 'win32'",
2425
]
2526
nvidia_wheels_cu13 = [

0 commit comments

Comments
 (0)