experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #31933 from yifeif/cherrypicks_609B5

Browse Source 
[r1.15 cherrypick] Add a note to ignore dlerror for CPU-only pip package
This commit is contained in:
Yifei Feng 2019-08-26 16:53:55 -07:00 committed by GitHub
commit 165b34d99a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/stream_executor/cuda/cudart_stub.cc
View File

@ -25,7 +25,11 @@ void* GetDsoHandle() {
static auto handle = []() -> void* { static auto handle = []() -> void* {
auto handle_or = auto handle_or =
stream_executor::internal::DsoLoader::GetCudaRuntimeDsoHandle(); stream_executor::internal::DsoLoader::GetCudaRuntimeDsoHandle();
if (!handle_or.ok()) return nullptr; if (!handle_or.ok()) {
LOG(INFO) << "Ignore above cudart dlerror if you do not have a GPU set "
"up on your machine.";
return nullptr;
}
return handle_or.ValueOrDie(); return handle_or.ValueOrDie();
}(); }();
return handle; return handle;