aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMatteo Martincigh <matteo.martincigh@arm.com>2019-07-24 09:15:00 +0100
committerNikhil Raj Arm <nikhil.raj@arm.com>2019-07-24 14:19:58 +0000
commitd73cecba2fb3525453d102603f6f27a9636754b5 (patch)
tree8c7019115be71acc3e79f101a6f47947023c5149
parentee18dc8d1725f472850ab0c398fd7cbc4b850891 (diff)
downloadarmnn-d73cecba2fb3525453d102603f6f27a9636754b5.tar.gz
IVGCVSW-3563 + IVGCVSW-3555 Create new utility functions for dynamic backends
* Created new DynamicBackendUtils class * Added OpenHandle, CloseHandle, GetEntryPoint and GetDlError methods to it Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com> Change-Id: I88c683b2c5d37968a9ebdf335be932ae2d9061e5
-rw-r--r--src/backends/backendsCommon/CMakeLists.txt2
-rw-r--r--src/backends/backendsCommon/DynamicBackendUtils.cpp48
-rw-r--r--src/backends/backendsCommon/DynamicBackendUtils.hpp56
3 files changed, 106 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/CMakeLists.txt b/src/backends/backendsCommon/CMakeLists.txt
index bc1c15beef..bb31ce3813 100644
--- a/src/backends/backendsCommon/CMakeLists.txt
+++ b/src/backends/backendsCommon/CMakeLists.txt
@@ -9,6 +9,8 @@ list(APPEND armnnBackendsCommon_sources
CpuTensorHandle.cpp
CpuTensorHandleFwd.hpp
CpuTensorHandle.hpp
+ DynamicBackendUtils.cpp
+ DynamicBackendUtils.hpp
IBackendInternal.hpp
IBackendContext.hpp
ITensorHandleFactory.cpp
diff --git a/src/backends/backendsCommon/DynamicBackendUtils.cpp b/src/backends/backendsCommon/DynamicBackendUtils.cpp
new file mode 100644
index 0000000000..f5ee12cbfd
--- /dev/null
+++ b/src/backends/backendsCommon/DynamicBackendUtils.cpp
@@ -0,0 +1,48 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "DynamicBackendUtils.hpp"
+
+namespace armnn
+{
+
+void* DynamicBackendUtils::OpenHandle(const std::string& sharedObjectPath)
+{
+ if (sharedObjectPath.empty())
+ {
+ throw RuntimeException("OpenHandle error: shared object path must not be empty");
+ }
+
+ void* sharedObjectHandle = dlopen(sharedObjectPath.c_str(), RTLD_LAZY | RTLD_GLOBAL);
+ if (!sharedObjectHandle)
+ {
+ throw RuntimeException(boost::str(boost::format("OpenHandle error: %1") % GetDlError()));
+ }
+
+ return sharedObjectHandle;
+}
+
+void DynamicBackendUtils::CloseHandle(const void* sharedObjectHandle)
+{
+ if (!sharedObjectHandle)
+ {
+ return;
+ }
+
+ dlclose(const_cast<void*>(sharedObjectHandle));
+}
+
+std::string DynamicBackendUtils::GetDlError()
+{
+ const char* errorMessage = dlerror();
+ if (!errorMessage)
+ {
+ return "";
+ }
+
+ return std::string(errorMessage);
+}
+
+} // namespace armnn
diff --git a/src/backends/backendsCommon/DynamicBackendUtils.hpp b/src/backends/backendsCommon/DynamicBackendUtils.hpp
new file mode 100644
index 0000000000..6bedec4580
--- /dev/null
+++ b/src/backends/backendsCommon/DynamicBackendUtils.hpp
@@ -0,0 +1,56 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <armnn/Exceptions.hpp>
+
+#include <string>
+#include <dlfcn.h>
+
+#include <boost/format.hpp>
+
+namespace armnn
+{
+
+class DynamicBackendUtils
+{
+public:
+ static void* OpenHandle(const std::string& sharedObjectPath);
+ static void CloseHandle(const void* sharedObjectHandle);
+
+ template<typename EntryPointType>
+ static EntryPointType GetEntryPoint(const void* sharedObjectHandle, const char* symbolName);
+
+private:
+ static std::string GetDlError();
+
+ /// This class is to hold utility functions only
+ DynamicBackendUtils() = delete;
+};
+
+template<typename EntryPointType>
+EntryPointType DynamicBackendUtils::GetEntryPoint(const void* sharedObjectHandle, const char* symbolName)
+{
+ if (sharedObjectHandle == nullptr)
+ {
+ throw RuntimeException("GetEntryPoint error: invalid handle");
+ }
+
+ if (symbolName == nullptr)
+ {
+ throw RuntimeException("GetEntryPoint error: invalid symbol");
+ }
+
+ auto entryPoint = reinterpret_cast<EntryPointType>(dlsym(const_cast<void*>(sharedObjectHandle), symbolName));
+ if (!entryPoint)
+ {
+ throw RuntimeException(boost::str(boost::format("GetEntryPoint error: %1") % GetDlError()));
+ }
+
+ return entryPoint;
+}
+
+} // namespace armnn