diff options
author | Finn Williams <Finn.Williams@arm.com> | 2021-08-26 11:08:01 +0100 |
---|---|---|
committer | Finn Williams <Finn.Williams@arm.com> | 2021-08-26 11:09:32 +0100 |
commit | dbf5f314f9498afa756e2637251a89a5291abde0 (patch) | |
tree | 2d974a44df4998291e2f2b9dc17c5d1e905f898e /src/armnn/LoadedNetwork.cpp | |
parent | 4e62750a2171a7bccf0c9c37514d07d3d222d1d5 (diff) | |
download | armnn-dbf5f314f9498afa756e2637251a89a5291abde0.tar.gz |
IVGCVSW-6311 Add the pre-import backend capability
* Update acl and ref backend capabilities
* Add check for AsyncExecution capability in LoadedNetwork
* Add new exception for capabilities
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I2f053c845a64f336e96d452793639c6e0d3eaa89
Diffstat (limited to 'src/armnn/LoadedNetwork.cpp')
-rw-r--r-- | src/armnn/LoadedNetwork.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index 531d43dbb5..9b5748bcf2 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -19,6 +19,7 @@ #include <armnn/backends/IMemoryManager.hpp> #include <backendsCommon/MemCopyWorkload.hpp> #include <backendsCommon/MemSyncWorkload.hpp> +#include <armnn/BackendHelper.hpp> #include <fmt/format.h> @@ -144,6 +145,14 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr<IOptimizedNetwork> net, IBackendInternal* backend = it.first->second.get(); + if (networkProperties.m_AsyncEnabled && + HasCapability(BackendOptions::BackendOption{"AsyncExecution", false}, backend->GetCapabilities())) + { + std::string er = backend->GetId(); + er += " does not support AsyncExecution"; + throw BackendCapabilityException(er); + } + if (backend->SupportsTensorAllocatorAPI()) { auto workloadFactory = backend->CreateWorkloadFactory( |