From b8d771ac2e6f847a64b3b203591c5b1c3e198d3a Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Fri, 14 Aug 2020 11:51:12 +0100 Subject: IVGCVSW-5012 Enable zero copy for Neon * Allow memory import if padding is not required in Neon * AddMockImportBackend for fallback tests * Refactor GraphUtils * Memory import unit tests * Fallback unit tests Signed-off-by: Narumol Prangnawarat Change-Id: Ic2e141e12774bf6d915e77745b6f6d2d83d9b82d --- src/armnn/Network.cpp | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'src/armnn/Network.cpp') diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 132924a19a..94a9961a81 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -912,7 +912,15 @@ EdgeStrategy CalculateEdgeStrategy(BackendsMap& backends, if ((dstFactory->GetImportFlags() & srcFactory->GetExportFlags()) != 0) { - return EdgeStrategy::ExportToTarget; + auto srcCapability = srcFactory->GetCapabilities(&layer, &layer, CapabilityClass::PaddingRequired); + auto dstCapability = dstFactory->GetCapabilities(&connectedLayer, + &connectedLayer, + CapabilityClass::PaddingRequired); + // Do not require memory copy if the source and destination do not require padding. + if (srcCapability.empty() && dstCapability.empty()) + { + return EdgeStrategy::ExportToTarget; + } } } } -- cgit v1.2.1