From 836efd40317a397761ec8b66e3f4398faac43ad0 Mon Sep 17 00:00:00 2001 From: Annie Tallund Date: Thu, 12 Jan 2023 07:49:06 +0100 Subject: 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 --- src/mlia/backend/registry.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) (limited to 'src/mlia/backend/registry.py') 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() + } -- cgit v1.2.1