Skip to content

Commit a6f24d7

Browse files
committed
Add support for find_bitcode_lib("nvshmem_device")
(site-packages, Conda) x (cu12, cu13)
1 parent 4cea0a1 commit a6f24d7

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

cuda_pathfinder/cuda/pathfinder/_static_libs/find_bitcode_lib.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,11 @@ class _BitcodeLibInfo(TypedDict):
4040
"nvidia/cuda_nvcc/nvvm/libdevice",
4141
),
4242
},
43+
"nvshmem_device": {
44+
"filename": "libnvshmem_device.bc",
45+
"rel_path": "lib",
46+
"site_packages_dirs": ("nvidia/nvshmem/lib",),
47+
},
4348
}
4449

4550
# Public API: just the supported library names

0 commit comments

Comments
 (0)