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 5ed86c5 commit a98de70Copy full SHA for a98de70
cuda_bindings/cuda/bindings/_path_finder/find_nvidia_headers.py
@@ -11,7 +11,9 @@
11
12
@functools.cache
13
def find_nvidia_header_directory(libname: str) -> str:
14
- assert libname == "nvshmem"
+ if libname != "nvshmem":
15
+ raise RuntimeError(f"UNKNOWN {libname=}")
16
+
17
assert not IS_WINDOWS
18
19
# Installed from a wheel
cuda_bindings/tests/test_path_finder_find_headers.py
@@ -7,7 +7,7 @@
7
8
9
def test_find_nvidia_header_directory(info_summary_append):
10
- with pytest.raises(AssertionError):
+ with pytest.raises(RuntimeError, match="^UNKNOWN libname='unknown-libname'$"):
find_nvidia_headers.find_nvidia_header_directory("unknown-libname")
hdr_dir = find_nvidia_headers.find_nvidia_header_directory("nvshmem")
0 commit comments