1 files changed, 8 insertions, 2 deletions
diff --git a/src/armnn/backends/NeonWorkloadFactory.hpp b/src/armnn/backends/NeonWorkloadFactory.hpp
index 0e39cfe8b1..66a69f3baf 100644
--- a/src/armnn/backends/NeonWorkloadFactory.hpp
+++ b/src/armnn/backends/NeonWorkloadFactory.hpp
@@ -4,7 +4,7 @@
 //
 #pragma once
 
-#include "WorkloadFactory.hpp"
+#include "AclBaseMemoryManager.hpp"
 #include "OutputHandler.hpp"
 
 #include <boost/core/ignore_unused.hpp>
@@ -16,7 +16,7 @@ namespace armnn
 class NeonWorkloadFactory : public IWorkloadFactory
 {
 public:
-    virtual ~NeonWorkloadFactory() { };
+    NeonWorkloadFactory();
 
     virtual Compute GetCompute() const override { return Compute::CpuAcc; }
 
@@ -95,6 +95,12 @@ public:
 
     virtual std::unique_ptr<IWorkload> CreateFloor(const FloorQueueDescriptor& descriptor,
                                                    const WorkloadInfo& info) const override;
+
+    void Finalize() override;
+
+private:
+
+    mutable AclBaseMemoryManager m_MemoryManager;
 };
 
 } //namespace armnn