aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonWorkloadFactory.hpp
diff options
context:
space:
mode:
authorAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2018-11-12 18:10:43 +0000
committerAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2018-11-13 14:41:52 +0000
commit56055193e82471a70b82e4eb11a8884c5904af75 (patch)
treebf66d0ba0088d963def8485c7e894b12d7a65b82 /src/backends/neon/NeonWorkloadFactory.hpp
parent95807cef855738ca481ace30f32ed9f245a098dd (diff)
downloadarmnn-56055193e82471a70b82e4eb11a8884c5904af75.tar.gz
IVGCVSW-2066: Add IMemoryManager and integrate into the backends framework
Change-Id: I93223c8678165cbc3d39f461c36bb8610dc81c05
Diffstat (limited to 'src/backends/neon/NeonWorkloadFactory.hpp')
-rw-r--r--src/backends/neon/NeonWorkloadFactory.hpp9
1 files changed, 3 insertions, 6 deletions
diff --git a/src/backends/neon/NeonWorkloadFactory.hpp b/src/backends/neon/NeonWorkloadFactory.hpp
index 98f323afdf..d5444f5828 100644
--- a/src/backends/neon/NeonWorkloadFactory.hpp
+++ b/src/backends/neon/NeonWorkloadFactory.hpp
@@ -11,7 +11,6 @@
#include <boost/core/ignore_unused.hpp>
-
namespace armnn
{
@@ -19,7 +18,7 @@ namespace armnn
class NeonWorkloadFactory : public IWorkloadFactory
{
public:
- NeonWorkloadFactory();
+ NeonWorkloadFactory(const std::shared_ptr<NeonMemoryManager>& memoryManager);
const BackendId& GetBackendId() const override;
@@ -136,9 +135,7 @@ public:
virtual void Acquire() override;
private:
-#ifdef ARMCOMPUTENEON_ENABLED
- mutable NeonMemoryManager m_MemoryManager;
-#endif
+ mutable std::shared_ptr<NeonMemoryManager> m_MemoryManager;
};
-} //namespace armnn
+} // namespace armnn