We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2426260 commit b74a84cCopy full SHA for b74a84c
1 file changed
cuda_pathfinder/tests/test_find_nvidia_headers.py
@@ -46,8 +46,15 @@ def test_find_libname_nvshmem(info_summary_append):
46
if IS_WINDOWS:
47
assert hdr_dir is None
48
pytest.skip("nvshmem has no Windows support.")
49
+ if hdr_dir:
50
+ assert os.path.isdir(hdr_dir)
51
+ assert os.path.isfile(os.path.join(hdr_dir, "nvshmem.h"))
52
if STRICTNESS == "all_must_work" or have_nvidia_nvshmem_package():
53
assert hdr_dir is not None
54
if have_nvidia_nvshmem_package():
55
hdr_dir_parts = hdr_dir.split(os.path.sep)
56
assert "site-packages" in hdr_dir_parts
57
+ elif conda_prefix := os.getenv("CONDA_PREFIX"):
58
+ assert hdr_dir.startswith(conda_prefix)
59
+ else:
60
+ assert hdr_dir.startswith("/usr/include/nvshmem_")
0 commit comments