blob: 6ea9bb486689a6f0ee79afc73156dbe14df11935 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Environment configuration functions."""
from __future__ import annotations
import logging
from functools import lru_cache
import mlia.backend.manager as backend_manager
from mlia.tools.metadata.common import DefaultInstallationManager
from mlia.tools.metadata.common import InstallationManager
from mlia.tools.metadata.corstone import get_corstone_installations
from mlia.tools.metadata.py_package import get_pypackage_backend_installations
logger = logging.getLogger(__name__)
def get_installation_manager(noninteractive: bool = False) -> InstallationManager:
"""Return installation manager."""
backends = get_corstone_installations() + get_pypackage_backend_installations()
return DefaultInstallationManager(backends, noninteractive=noninteractive)
@lru_cache
def get_available_backends() -> list[str]:
"""Return list of the available backends."""
available_backends = ["Vela"]
# Add backends using backend manager
manager = get_installation_manager()
available_backends.extend(
backend
for backend in backend_manager.supported_backends()
if manager.backend_installed(backend)
)
return available_backends
# List of mutually exclusive Corstone backends ordered by priority
_CORSTONE_EXCLUSIVE_PRIORITY = ("Corstone-310", "Corstone-300")
def get_default_backends() -> list[str]:
"""Get default backends for evaluation."""
backends = get_available_backends()
# Filter backends to only include one Corstone backend
for corstone in _CORSTONE_EXCLUSIVE_PRIORITY:
if corstone in backends:
backends = [
backend
for backend in backends
if backend == corstone or backend not in _CORSTONE_EXCLUSIVE_PRIORITY
]
break
return backends
def is_corstone_backend(backend: str) -> bool:
"""Check if the given backend is a Corstone backend."""
return backend in _CORSTONE_EXCLUSIVE_PRIORITY
|