commit dea8fb6b96663de5d3df2f9fceb9bd09432fd7aa
author Sadik Armagan <sadik.armagan@arm.com> Thu Nov 26 10:38:11 2020 +0000
committer Sadik Armagan <sadik.armagan@arm.com> Thu Nov 26 10:38:11 2020 +0000
tree d041bf8e9d406c80ce089fc5b8d84b44381332e1
parent f4f150c30d3c34e9f26757ca43e4a2694b882bce

IVGCVSW-5481 'Add ClCompileContext to ClWorkloadFactory'

* Introduced CLCompileContext to ClWorkloadFactory

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Ied38f4336210502e5f518b9955ae6a5ba3d242b3

src/backends/cl/ClWorkloadFactory.hpp

diff --git a/src/backends/cl/ClWorkloadFactory.hpp b/src/backends/cl/ClWorkloadFactory.hpp
index 84eae50..c8812cf 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 @@
     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 @@
                                                    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