aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/ClWorkloadFactory.hpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2020-11-26 10:38:11 +0000
committerSadik Armagan <sadik.armagan@arm.com>2020-11-26 10:38:11 +0000
commitdea8fb6b96663de5d3df2f9fceb9bd09432fd7aa (patch)
treed041bf8e9d406c80ce089fc5b8d84b44381332e1 /src/backends/cl/ClWorkloadFactory.hpp
parentf4f150c30d3c34e9f26757ca43e4a2694b882bce (diff)
downloadarmnn-dea8fb6b96663de5d3df2f9fceb9bd09432fd7aa.tar.gz
IVGCVSW-5481 'Add ClCompileContext to ClWorkloadFactory'
* Introduced CLCompileContext to ClWorkloadFactory Signed-off-by: Sadik Armagan <sadik.armagan@arm.com> Change-Id: Ied38f4336210502e5f518b9955ae6a5ba3d242b3
Diffstat (limited to 'src/backends/cl/ClWorkloadFactory.hpp')
-rw-r--r--src/backends/cl/ClWorkloadFactory.hpp7
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/cl/ClWorkloadFactory.hpp b/src/backends/cl/ClWorkloadFactory.hpp
index 84eae5076a..c8812cfe1b 100644
--- a/src/backends/cl/ClWorkloadFactory.hpp
+++ b/src/backends/cl/ClWorkloadFactory.hpp
@@ -12,6 +12,8 @@
#include <backendsCommon/WorkloadFactoryBase.hpp>
#include <aclCommon/BaseMemoryManager.hpp>
+#include <arm_compute/core/CL/CLCompileContext.h>
+
namespace armnn
{
@@ -24,6 +26,8 @@ public:
ClWorkloadFactory(const std::shared_ptr<ClMemoryManager>& memoryManager,
const IBackendInternal::IBackendSpecificModelContextPtr& modelContextPtr);
+ void AfterWorkloadsCreated() override;
+
const BackendId& GetBackendId() const override;
static bool IsLayerSupported(const Layer& layer,
@@ -254,8 +258,11 @@ private:
const WorkloadInfo& info,
Args&&... args);
+ void InitializeCLCompileContext();
+
mutable std::shared_ptr<ClMemoryManager> m_MemoryManager;
const IBackendInternal::IBackendSpecificModelContextPtr m_ModelContextPtr;
+ arm_compute::CLCompileContext m_CLCompileContext;
};
} // namespace armnn