diff options
Diffstat (limited to 'src/graph/backends/BackendRegistry.cpp')
-rw-r--r-- | src/graph/backends/BackendRegistry.cpp | 8 |
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); |