File tree Expand file tree Collapse file tree 1 file changed +24
-0
lines changed Expand file tree Collapse file tree 1 file changed +24
-0
lines changed Original file line number Diff line number Diff line change 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} " )
You can’t perform that action at this time.
0 commit comments