diff options
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 |