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

fixing a bug in the ROCm versions of the Get*DsoHandle APIs

Browse Source
This commit is contained in:
Deven Desai 2019-03-07 20:12:17 +00:00
parent 2f589bfa2a
commit 13ac37f915
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/stream_executor/platform/default/dso_loader.cc
View File

@ -158,27 +158,27 @@ port::StatusOr<void*> GetCudnnDsoHandle() {
port::StatusOr<void*> GetRocblasDsoHandle() {
static auto result = new auto(DsoLoader::GetRocblasDsoHandle());
return result;
return *result;
}
port::StatusOr<void*> GetMiopenDsoHandle() {
static auto result = new auto(DsoLoader::GetMiopenDsoHandle());
return result;
return *result;
}
port::StatusOr<void*> GetRocfftDsoHandle() {
static auto result = new auto(DsoLoader::GetRocfftDsoHandle());
return result;
return *result;
}
port::StatusOr<void*> GetRocrandDsoHandle() {
static auto result = new auto(DsoLoader::GetRocrandDsoHandle());
return result;
return *result;
}
port::StatusOr<void*> GetHipDsoHandle() {
static auto result = new auto(DsoLoader::GetHipDsoHandle());
return result;
return *result;
}
} // namespace CachedDsoLoader