From 13ac37f915a3ec17864338ebdd0da4b5a813d725 Mon Sep 17 00:00:00 2001 From: Deven Desai Date: Thu, 7 Mar 2019 20:12:17 +0000 Subject: [PATCH] fixing a bug in the ROCm versions of the Get*DsoHandle APIs --- .../stream_executor/platform/default/dso_loader.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tensorflow/stream_executor/platform/default/dso_loader.cc b/tensorflow/stream_executor/platform/default/dso_loader.cc index a2a9ad67534..6ed7480ff0c 100644 --- a/tensorflow/stream_executor/platform/default/dso_loader.cc +++ b/tensorflow/stream_executor/platform/default/dso_loader.cc @@ -158,27 +158,27 @@ port::StatusOr GetCudnnDsoHandle() { port::StatusOr GetRocblasDsoHandle() { static auto result = new auto(DsoLoader::GetRocblasDsoHandle()); - return result; + return *result; } port::StatusOr GetMiopenDsoHandle() { static auto result = new auto(DsoLoader::GetMiopenDsoHandle()); - return result; + return *result; } port::StatusOr GetRocfftDsoHandle() { static auto result = new auto(DsoLoader::GetRocfftDsoHandle()); - return result; + return *result; } port::StatusOr GetRocrandDsoHandle() { static auto result = new auto(DsoLoader::GetRocrandDsoHandle()); - return result; + return *result; } port::StatusOr GetHipDsoHandle() { static auto result = new auto(DsoLoader::GetHipDsoHandle()); - return result; + return *result; } } // namespace CachedDsoLoader