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

Merge pull request #39956 from bhack:patch-2 

PiperOrigin-RevId: 314873638
Change-Id: I9fa2b55613c78eb64dec1a40a1f157374498a929
This commit is contained in:
TensorFlower Gardener 2020-06-04 23:22:55 -07:00
parent 2d2500117e d38caa0466
commit e63137a53c
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/stream_executor/cuda/cuda_driver.cc
View File

@ -308,9 +308,12 @@ static port::Status InternalInit() {
if (res == CUDA_SUCCESS) { if (res == CUDA_SUCCESS) {
return port::Status::OK(); return port::Status::OK();
} else if (res == CUDA_ERROR_SHARED_OBJECT_INIT_FAILED) {
LOG(WARNING) << "failed call to cuInit: " << ToString(res);
} else {
LOG(ERROR) << "failed call to cuInit: " << ToString(res);
} }
LOG(ERROR) << "failed call to cuInit: " << ToString(res);
Diagnostician::LogDiagnosticInformation(); Diagnostician::LogDiagnosticInformation();
return port::Status(port::error::ABORTED, return port::Status(port::error::ABORTED,
absl::StrCat("failed call to cuInit: ", ToString(res))); absl::StrCat("failed call to cuInit: ", ToString(res)));