diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-03-31 16:12:13 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-04-02 10:38:51 +0000 |
commit | aede8ca25d3231d122b670353b754142b64e5d69 (patch) | |
tree | 87ba5bdae701a51b15fa44d550a55932e89966c9 /src/backends/cl | |
parent | 8c7a28ba48a0c7347fd6f940cbcf4b6dbeca932f (diff) | |
download | armnn-aede8ca25d3231d122b670353b754142b64e5d69.tar.gz |
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 <sadik.armagan@arm.com>
Change-Id: I49f8467297f4b6b8e414cb6a3638a7d3f1bb886a
Diffstat (limited to 'src/backends/cl')
-rw-r--r-- | src/backends/cl/ClBackend.cpp | 10 | ||||
-rw-r--r-- | src/backends/cl/ClBackend.hpp | 6 |
2 files changed, 16 insertions, 0 deletions
diff --git a/src/backends/cl/ClBackend.cpp b/src/backends/cl/ClBackend.cpp index 928760ef5f..33aff2731d 100644 --- a/src/backends/cl/ClBackend.cpp +++ b/src/backends/cl/ClBackend.cpp @@ -141,6 +141,16 @@ IBackendInternal::ILayerSupportSharedPtr ClBackend::GetLayerSupport(const ModelO return layerSupport; } +bool ClBackend::HasCapability(BackendCapability capabilityClass) const +{ + auto search = gpuAccCapabilities.find(capabilityClass); + if (search != gpuAccCapabilities.end()) + { + return true; + } + return false; +} + OptimizationViews ClBackend::OptimizeSubgraphView(const SubgraphView& subgraph, const ModelOptions& modelOptions) const { diff --git a/src/backends/cl/ClBackend.hpp b/src/backends/cl/ClBackend.hpp index 2b19fc5b33..f9a5745eb3 100644 --- a/src/backends/cl/ClBackend.hpp +++ b/src/backends/cl/ClBackend.hpp @@ -9,6 +9,10 @@ namespace armnn { +const std::set<armnn::BackendCapability> gpuAccCapabilities { + // add new capabilities here.. +}; + class ClBackend : public IBackendInternal { public: @@ -49,6 +53,8 @@ public: IBackendInternal::IBackendSpecificModelContextPtr CreateBackendSpecificModelContext( const ModelOptions& modelOptions) const override; + + bool HasCapability(BackendCapability capabilityClass) const override; }; } // namespace armnn |