From aede8ca25d3231d122b670353b754142b64e5d69 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Wed, 31 Mar 2021 16:12:13 +0100 Subject: IVGCVSW-5783 'Add AsyncExecution Capability' * Added AsyncExecution to the BackendCapability enum class. * Logged a warning if backends do not support AsyncExecution capability if AsyncNetwork is created. Signed-off-by: Sadik Armagan Change-Id: I49f8467297f4b6b8e414cb6a3638a7d3f1bb886a --- src/backends/neon/NeonBackend.cpp | 10 ++++++++++ src/backends/neon/NeonBackend.hpp | 6 ++++++ 2 files changed, 16 insertions(+) (limited to 'src/backends/neon') diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp index d48b32bb45..a1299fb458 100644 --- a/src/backends/neon/NeonBackend.cpp +++ b/src/backends/neon/NeonBackend.cpp @@ -131,6 +131,16 @@ IBackendInternal::ILayerSupportSharedPtr NeonBackend::GetLayerSupport(const Mode return layerSupport; } +bool NeonBackend::HasCapability(BackendCapability capabilityClass) const +{ + auto search = cpuAccCapabilities.find(capabilityClass); + if (search != cpuAccCapabilities.end()) + { + return true; + } + return false; +} + OptimizationViews NeonBackend::OptimizeSubgraphView(const SubgraphView& subgraph) const { OptimizationViews optimizationViews; diff --git a/src/backends/neon/NeonBackend.hpp b/src/backends/neon/NeonBackend.hpp index 42c6666930..34606f992c 100644 --- a/src/backends/neon/NeonBackend.hpp +++ b/src/backends/neon/NeonBackend.hpp @@ -9,6 +9,10 @@ namespace armnn { +const std::set cpuAccCapabilities { + // add new capabilities here.. +}; + class NeonBackend : public IBackendInternal { public: @@ -47,6 +51,8 @@ public: IBackendInternal::IBackendSpecificModelContextPtr CreateBackendSpecificModelContext( const ModelOptions& modelOptions) const override; + + bool HasCapability(BackendCapability capabilityClass) const override; }; } // namespace armnn -- cgit v1.2.1