Skip to content

Commit 5ed86c5

Browse files
committed
Add tests/test_path_finder_find_headers.py, with hard-coded paths.
1 parent d174f5a commit 5ed86c5

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed
Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
# Copyright 2025 NVIDIA Corporation. All rights reserved.
2+
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
3+
4+
import pytest
5+
6+
from cuda.bindings._path_finder import find_nvidia_headers
7+
8+
9+
def test_find_nvidia_header_directory(info_summary_append):
10+
with pytest.raises(AssertionError):
11+
find_nvidia_headers.find_nvidia_header_directory("unknown-libname")
12+
13+
hdr_dir = find_nvidia_headers.find_nvidia_header_directory("nvshmem")
14+
assert hdr_dir in [
15+
# pip install nvidia-nvshmem-cu12
16+
"/home/rgrossekunst/forked/cuda-python/venvs/scratch/lib/python3.12/site-packages/nvidia/nvshmem/include",
17+
# conda create -y -n nvshmem python=3.12
18+
# conda activate nvshmem
19+
# conda install -y conda-forge::libnvshmem3 conda-forge::libnvshmem-dev
20+
"/home/rgrossekunst/miniforge3/envs/nvshmem/include",
21+
# sudo apt install libnvshmem3-cuda-12 libnvshmem3-dev-cuda-12
22+
"/usr/include/nvshmem_12",
23+
]
24+
info_summary_append(f"{hdr_dir=!r}")

0 commit comments

Comments
 (0)