Skip to content

Commit d0a80f8

Browse files
malfetpytorchmergebot
authored andcommitted
Better errors in c10::DriverAPI on dl failure (pytorch#112995)
Pull Request resolved: pytorch#112995 Approved by: https://github.com/Skylion007 ghstack dependencies: pytorch#112994
1 parent 5719117 commit d0a80f8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

c10/cuda/driver_api.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,20 +11,20 @@ namespace {
1111

1212
DriverAPI create_driver_api() {
1313
void* handle_0 = dlopen("libcuda.so", RTLD_LAZY | RTLD_NOLOAD);
14-
TORCH_INTERNAL_ASSERT(handle_0);
14+
TORCH_INTERNAL_ASSERT(handle_0, "Can't open libcuda.so: ", dlerror());
1515
void* handle_1 = DriverAPI::get_nvml_handle();
1616
DriverAPI r{};
1717

1818
#define LOOKUP_LIBCUDA_ENTRY(name) \
1919
r.name##_ = ((decltype(&name))dlsym(handle_0, #name)); \
20-
TORCH_INTERNAL_ASSERT(r.name##_)
20+
TORCH_INTERNAL_ASSERT(r.name##_, "Can't find ", #name, ": ", dlerror())
2121
C10_LIBCUDA_DRIVER_API(LOOKUP_LIBCUDA_ENTRY)
2222
#undef LOOKUP_LIBCUDA_ENTRY
2323

2424
if (handle_1) {
2525
#define LOOKUP_NVML_ENTRY(name) \
2626
r.name##_ = ((decltype(&name))dlsym(handle_1, #name)); \
27-
TORCH_INTERNAL_ASSERT(r.name##_)
27+
TORCH_INTERNAL_ASSERT(r.name##_, "Can't find ", #name, ": ", dlerror())
2828
C10_NVML_DRIVER_API(LOOKUP_NVML_ENTRY)
2929
#undef LOOKUP_NVML_ENTRY
3030
}

0 commit comments

Comments
 (0)