IVGCVSW-4077 Fix issue when NEON import disabled
* Removes workaround which handled null dstFactory
when NEON import was disabled, and now handles
this in the correct way.
Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: Ief42b3c52d018f0fa71be4d4d37516f2caad1e0d
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 1797baf..d164759 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -708,9 +708,9 @@
{
ITensorHandleFactory* dstFactory = registry.GetFactory(pref);
- // Handles some cases where dstFactory is null when Neon memory import is disabled
+ // Handles cases when a destPref is not listed in TensorHandleFactoryRegistry
if (!dstFactory) {
- return EdgeStrategy::CopyToTarget;
+ continue;
}
if ((dstFactory->GetImportFlags() & srcFactory->GetExportFlags()) != 0)
@@ -726,7 +726,7 @@
for (auto&& pref : dstPrefs)
{
ITensorHandleFactory* dstFactory = registry.GetFactory(pref);
- if (dstFactory->SupportsMapUnmap())
+ if (dstFactory && dstFactory->SupportsMapUnmap())
{
return EdgeStrategy::CopyToTarget;
}