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
commit e63137a53c
1 changed files with 4 additions and 1 deletions

View File

@ -308,9 +308,12 @@ static port::Status InternalInit() {
if (res == CUDA_SUCCESS) {
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();
return port::Status(port::error::ABORTED,
absl::StrCat("failed call to cuInit: ", ToString(res)));