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() { port::StatusOr<void*> GetRocblasDsoHandle() {
static auto result = new auto(DsoLoader::GetRocblasDsoHandle()); static auto result = new auto(DsoLoader::GetRocblasDsoHandle());
return result; return *result;
} }
port::StatusOr<void*> GetMiopenDsoHandle() { port::StatusOr<void*> GetMiopenDsoHandle() {
static auto result = new auto(DsoLoader::GetMiopenDsoHandle()); static auto result = new auto(DsoLoader::GetMiopenDsoHandle());
return result; return *result;
} }
port::StatusOr<void*> GetRocfftDsoHandle() { port::StatusOr<void*> GetRocfftDsoHandle() {
static auto result = new auto(DsoLoader::GetRocfftDsoHandle()); static auto result = new auto(DsoLoader::GetRocfftDsoHandle());
return result; return *result;
} }
port::StatusOr<void*> GetRocrandDsoHandle() { port::StatusOr<void*> GetRocrandDsoHandle() {
static auto result = new auto(DsoLoader::GetRocrandDsoHandle()); static auto result = new auto(DsoLoader::GetRocrandDsoHandle());
return result; return *result;
} }
port::StatusOr<void*> GetHipDsoHandle() { port::StatusOr<void*> GetHipDsoHandle() {
static auto result = new auto(DsoLoader::GetHipDsoHandle()); static auto result = new auto(DsoLoader::GetHipDsoHandle());
return result; return *result;
} }
} // namespace CachedDsoLoader } // namespace CachedDsoLoader