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/install.py | 24 +++++------------------- 1 file changed, 5 insertions(+), 19 deletions(-) (limited to 'src/mlia/backend/install.py') diff --git a/src/mlia/backend/install.py b/src/mlia/backend/install.py index eea3403..37a277b 100644 --- a/src/mlia/backend/install.py +++ b/src/mlia/backend/install.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 """Module for installation process.""" from __future__ import annotations @@ -26,17 +26,20 @@ from mlia.utils.filesystem import temp_directory from mlia.utils.filesystem import working_directory from mlia.utils.py_manager import get_package_manager - logger = logging.getLogger(__name__) # Mapping backend -> device_type -> system_name _SUPPORTED_SYSTEMS = { "Corstone-300": { + "Ethos-U55": "Corstone-300: Cortex-M55+Ethos-U55", + "Ethos-U65": "Corstone-300: Cortex-M55+Ethos-U65", "ethos-u55": "Corstone-300: Cortex-M55+Ethos-U55", "ethos-u65": "Corstone-300: Cortex-M55+Ethos-U65", }, "Corstone-310": { + "Ethos-U55": "Corstone-310: Cortex-M85+Ethos-U55", + "Ethos-U65": "Corstone-310: Cortex-M85+Ethos-U65", "ethos-u55": "Corstone-310: Cortex-M85+Ethos-U55", "ethos-u65": "Corstone-310: Cortex-M85+Ethos-U65", }, @@ -61,23 +64,6 @@ def get_application_name(system_name: str) -> str: return _SYSTEM_TO_APP_MAP[system_name] -def is_supported(backend: str, device_type: str | None = None) -> bool: - """Check if the backend (and optionally device type) is supported.""" - if device_type is None: - return backend in _SUPPORTED_SYSTEMS - - try: - get_system_name(backend, device_type) - return True - except KeyError: - return False - - -def supported_backends() -> list[str]: - """Get a list of all backends supported by the backend manager.""" - return list(_SUPPORTED_SYSTEMS.keys()) - - def get_all_system_names(backend: str) -> list[str]: """Get all systems supported by the backend.""" return list(_SUPPORTED_SYSTEMS.get(backend, {}).values()) -- cgit v1.2.1