Skip to content

Commit 55583d9

Browse files
committed
run_cuda_bindings_path_finder.py: optionally use args as libnames (to aid debugging)
1 parent ddea021 commit 55583d9

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

toolshed/run_cuda_bindings_path_finder.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,12 @@
1414

1515

1616
def run(args):
17-
assert len(args) == 0
17+
if args:
18+
libnames = args
19+
else:
20+
libnames = ALL_LIBNAMES
1821

19-
for libname in ALL_LIBNAMES:
22+
for libname in libnames:
2023
print(f"{libname=}")
2124
try:
2225
loaded_dl = path_finder._load_nvidia_dynamic_library(libname)

0 commit comments

Comments
 (0)