diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-11-26 10:38:11 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-11-26 10:38:11 +0000 |
commit | dea8fb6b96663de5d3df2f9fceb9bd09432fd7aa (patch) | |
tree | d041bf8e9d406c80ce089fc5b8d84b44381332e1 /src/backends/cl/ClWorkloadFactory.hpp | |
parent | f4f150c30d3c34e9f26757ca43e4a2694b882bce (diff) | |
download | armnn-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.hpp | 7 |
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 |