aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-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