1 files changed, 7 insertions, 6 deletions
diff --git a/src/backends/BackendRegistry.hpp b/src/backends/BackendRegistry.hpp
index ff01d21715..e2c526d293 100644
--- a/src/backends/BackendRegistry.hpp
+++ b/src/backends/BackendRegistry.hpp
@@ -5,7 +5,6 @@
 #pragma once
 
 #include <armnn/Types.hpp>
-#include <string>
 #include <functional>
 #include <memory>
 #include <unordered_map>
@@ -21,21 +20,23 @@ public:
     using FactoryFunction = std::function<IBackendUniquePtr()>;
 
     static BackendRegistry& Instance();
-    void Register(const std::string& name, FactoryFunction factory);
-    FactoryFunction GetFactory(const std::string& name) const;
+
+    void Register(const BackendId& id, FactoryFunction factory);
+    FactoryFunction GetFactory(const BackendId& id) const;
 
     struct Helper
     {
-        Helper(const std::string& name, FactoryFunction factory)
+        Helper(const BackendId& id, FactoryFunction factory)
         {
-            BackendRegistry::Instance().Register(name, factory);
+            BackendRegistry::Instance().Register(id, factory);
         }
     };
 
     size_t Size() const { return m_BackendFactories.size(); }
+    BackendIdSet GetBackendIds() const;
 
 protected:
-    using FactoryStorage = std::unordered_map<std::string, FactoryFunction>;
+    using FactoryStorage = std::unordered_map<BackendId, FactoryFunction>;
 
     // For testing only
     static void Swap(FactoryStorage& other);