aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/backend/install.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/install.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/install.py')
-rw-r--r--src/mlia/backend/install.py24
1 files changed, 5 insertions, 19 deletions
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())