diff options
author | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-08-07 15:14:56 +0100 |
---|---|---|
committer | Ferran Balaguer Arm <ferran.balaguer@arm.com> | 2019-08-20 14:31:07 +0000 |
commit | bfeb2711da172b26931c58af7b15d434ef49e24e (patch) | |
tree | 3516b24857effe4c98737f48026dabf75d1f360a /src/armnn/Graph.cpp | |
parent | aec942c3f258db46e0fb8054d39c0e7c3c411728 (diff) | |
download | armnn-bfeb2711da172b26931c58af7b15d434ef49e24e.tar.gz |
IVGCVSW-3606 Support memory import for Reference backend
Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
Change-Id: I94bd191f88e0911ad4e4727610e81cd7afa95512
Diffstat (limited to 'src/armnn/Graph.cpp')
-rw-r--r-- | src/armnn/Graph.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/Graph.cpp b/src/armnn/Graph.cpp index f0ef0e18f7..5e2acd55a3 100644 --- a/src/armnn/Graph.cpp +++ b/src/armnn/Graph.cpp @@ -335,7 +335,6 @@ void Graph::AddCompatibilityLayers(std::map<BackendId, std::unique_ptr<IBackendI auto backend = backendIt->second.get(); auto tensorHandleFactoryIds = backend->GetHandleFactoryPreferences(); bool found = false; - boost::ignore_unused(found); for (auto preference : tensorHandleFactoryIds) { @@ -344,10 +343,12 @@ void Graph::AddCompatibilityLayers(std::map<BackendId, std::unique_ptr<IBackendI { auto srcPref = srcOutputSlot.GetTensorHandleFactoryId(); auto srcFactory = registry.GetFactory(srcPref); + if (srcFactory) { bool canExportImport = - (factory->GetImportFlags() & srcFactory->GetExportFlags()) != 0; + (factory->GetImportFlags() & srcFactory->GetExportFlags()) != 0; + if (factory->SupportsMapUnmap() || canExportImport) { compOutputSlot.SetTensorHandleFactory(preference); |