Skip to content

Commit 64d6c2d

Browse files
committed
Merge branch 'main' into graph-node-identity
2 parents 84f0b30 + 1c8f297 commit 64d6c2d

3 files changed

Lines changed: 40 additions & 1 deletion

File tree

‎cuda_pathfinder/cuda/pathfinder/_static_libs/find_bitcode_lib.py‎

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ class _BitcodeLibInfo(TypedDict):
2929
filename: str
3030
rel_path: str
3131
site_packages_dirs: tuple[str, ...]
32+
available_on_windows: bool
3233

3334

3435
_SUPPORTED_BITCODE_LIBS_INFO: dict[str, _BitcodeLibInfo] = {
@@ -39,11 +40,22 @@ class _BitcodeLibInfo(TypedDict):
3940
"nvidia/cu13/nvvm/libdevice",
4041
"nvidia/cuda_nvcc/nvvm/libdevice",
4142
),
43+
"available_on_windows": True,
44+
},
45+
"nvshmem_device": {
46+
"filename": "libnvshmem_device.bc",
47+
"rel_path": "lib",
48+
"site_packages_dirs": ("nvidia/nvshmem/lib",),
49+
"available_on_windows": False,
4250
},
4351
}
4452

4553
# Public API: just the supported library names
46-
SUPPORTED_BITCODE_LIBS: tuple[str, ...] = tuple(sorted(_SUPPORTED_BITCODE_LIBS_INFO.keys()))
54+
SUPPORTED_BITCODE_LIBS: tuple[str, ...] = tuple(
55+
sorted(
56+
name for name, info in _SUPPORTED_BITCODE_LIBS_INFO.items() if not IS_WINDOWS or info["available_on_windows"]
57+
)
58+
)
4759

4860

4961
def _no_such_file_in_dir(dir_path: str, filename: str, error_messages: list[str], attachments: list[str]) -> None:

‎cuda_pathfinder/docs/nv-versions.json‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,10 @@
33
"version": "latest",
44
"url": "https://nvidia.github.io/cuda-python/cuda-pathfinder/latest/"
55
},
6+
{
7+
"version": "1.5.1",
8+
"url": "https://nvidia.github.io/cuda-python/cuda-pathfinder/1.5.1/"
9+
},
610
{
711
"version": "1.5.0",
812
"url": "https://nvidia.github.io/cuda-python/cuda-pathfinder/1.5.0/"
Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
.. SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
.. SPDX-License-Identifier: Apache-2.0
3+
4+
.. py:currentmodule:: cuda.pathfinder
5+
6+
``cuda-pathfinder`` 1.5.1 Release notes
7+
=======================================
8+
9+
Highlights
10+
----------
11+
12+
* Add ``locate_nvidia_header_directory()`` support for ``mathdx``, ``cutlass``,
13+
and ``cute``.
14+
(`PR #1816 <https://github.com/NVIDIA/cuda-python/pull/1816>`_)
15+
16+
* Add ``cusolverMp`` support for dynamic loading and header discovery.
17+
(`PR #1845 <https://github.com/NVIDIA/cuda-python/pull/1845>`_)
18+
19+
* Extend NVIDIA binary search paths with ``nvidia/cu13/bin``.
20+
(`PR #1846 <https://github.com/NVIDIA/cuda-python/pull/1846>`_)
21+
22+
* Add ``find_bitcode_lib("nvshmem_device")`` support.
23+
(`PR #1828 <https://github.com/NVIDIA/cuda-python/pull/1828>`_)

0 commit comments

Comments
 (0)