aboutsummaryrefslogtreecommitdiff
path: root/src/graph/backends/BackendRegistry.cpp
diff options
context:
space:
mode:
authorAnthony Barbier <anthony.barbier@arm.com>2018-08-22 13:44:36 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:54:54 +0000
commit890ad1be4eff1a805fe07667f4044b1d38a07b39 (patch)
tree98d2e9e4ef76d1e015d867dd6945bee3cb7ba5f0 /src/graph/backends/BackendRegistry.cpp
parentaaba4c626bcc6365e0108130633ce43fafe9da45 (diff)
downloadComputeLibrary-890ad1be4eff1a805fe07667f4044b1d38a07b39.tar.gz
COMPMID-1246: Fix bug in handling backends that can't be loaded in the Graph API
Change-Id: Iefd175af2f472179d86df5358a1527a79c5666ed Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/145182 Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com> Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/graph/backends/BackendRegistry.cpp')
-rw-r--r--src/graph/backends/BackendRegistry.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/src/graph/backends/BackendRegistry.cpp b/src/graph/backends/BackendRegistry.cpp
index 2803322e64..dccfefca52 100644
--- a/src/graph/backends/BackendRegistry.cpp
+++ b/src/graph/backends/BackendRegistry.cpp
@@ -48,6 +48,14 @@ IDeviceBackend *BackendRegistry::find_backend(Target target)
return _registered_backends[target].get();
}
+IDeviceBackend &BackendRegistry::get_backend(Target target)
+{
+ IDeviceBackend *backend = find_backend(target);
+ ARM_COMPUTE_ERROR_ON_MSG(!backend, "Requested backend doesn't exist!");
+ ARM_COMPUTE_ERROR_ON_MSG(!backend->is_backend_supported(), "Requested backend isn't supported");
+ return *backend;
+}
+
bool BackendRegistry::contains(Target target) const
{
auto it = _registered_backends.find(target);