Skip to content

Commit a98de70

Browse files
committed
Better error message: UNKNOWN libname='unknown-libname'
1 parent 5ed86c5 commit a98de70

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

cuda_bindings/cuda/bindings/_path_finder/find_nvidia_headers.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,9 @@
1111

1212
@functools.cache
1313
def find_nvidia_header_directory(libname: str) -> str:
14-
assert libname == "nvshmem"
14+
if libname != "nvshmem":
15+
raise RuntimeError(f"UNKNOWN {libname=}")
16+
1517
assert not IS_WINDOWS
1618

1719
# Installed from a wheel

cuda_bindings/tests/test_path_finder_find_headers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88

99
def test_find_nvidia_header_directory(info_summary_append):
10-
with pytest.raises(AssertionError):
10+
with pytest.raises(RuntimeError, match="^UNKNOWN libname='unknown-libname'$"):
1111
find_nvidia_headers.find_nvidia_header_directory("unknown-libname")
1212

1313
hdr_dir = find_nvidia_headers.find_nvidia_header_directory("nvshmem")

0 commit comments

Comments
 (0)