diff options
author | James Conroy <james.conroy@arm.com> | 2019-11-07 14:37:09 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-08 15:56:05 +0000 |
commit | ffab16f9f23868b5b9338a1b63393f3be53211cb (patch) | |
tree | 66a2a3dcd418e08488f8ad2f6b9e96a80fac90ee /src/armnn | |
parent | 55e488a53c98af7dabf9f7b45572d02d850d8866 (diff) | |
download | armnn-ffab16f9f23868b5b9338a1b63393f3be53211cb.tar.gz |
IVGCVSW-4077 Disable NEON memory import
* Temporarily handles cases in CalculateEdgeStrategy
where dstFactory pointer is null when import is
disabled.
* This patch is required for ensuring debug layer
works correctly when executing a model on Neon.
Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I7304723246d362d6d9073c3d0b1224e194a8532c
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/Network.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index f4bfc7a88e..b1cb5a871a 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -699,6 +699,12 @@ EdgeStrategy CalculateEdgeStrategy(BackendsMap& backends, for (auto&& pref : dstPrefs) { ITensorHandleFactory* dstFactory = registry.GetFactory(pref); + + // Handles some cases where dstFactory is null when Neon memory import is disabled + if (!dstFactory) { + return EdgeStrategy::CopyToTarget; + } + if ((dstFactory->GetImportFlags() & srcFactory->GetExportFlags()) != 0) { return EdgeStrategy::ExportToTarget; |