aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/backend/registry.py
diff options
context:
space:
mode:
authorAnnie Tallund <annie.tallund@arm.com>2023-01-12 07:49:06 +0100
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2023-02-08 15:23:29 +0000
commit836efd40317a397761ec8b66e3f4398faac43ad0 (patch)
tree5133ffd51d8d6772551333a4b337d36a501a8a91 /src/mlia/backend/registry.py
parenta4fb8c72f15146c95df16c25e75f03344e9814fd (diff)
downloadmlia-836efd40317a397761ec8b66e3f4398faac43ad0.tar.gz
MLIA-770 List all available backends
- Rely on target and backend registry for support information - Make above information less Ethos(TM)-U specific Change-Id: I8dbfb84401016412a3d719a84eb592f21d79c46b
Diffstat (limited to 'src/mlia/backend/registry.py')
-rw-r--r--src/mlia/backend/registry.py14
1 files changed, 13 insertions, 1 deletions
diff --git a/src/mlia/backend/registry.py b/src/mlia/backend/registry.py
index 6a0da74..988c8c3 100644
--- a/src/mlia/backend/registry.py
+++ b/src/mlia/backend/registry.py
@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Backend module."""
from mlia.backend.config import BackendConfiguration
@@ -6,3 +6,15 @@ from mlia.utils.registry import Registry
# All supported targets are required to be registered here.
registry = Registry[BackendConfiguration]()
+
+
+def get_supported_backends() -> list:
+ """Get a list of all backends supported by the backend manager."""
+ return sorted(list(registry.items.keys()))
+
+
+def get_supported_systems() -> dict:
+ """Get a list of all systems supported by the backend manager."""
+ return {
+ backend: config.supported_systems for backend, config in registry.items.items()
+ }