blob: 6b6777d489aa65442f0207c11cb352e42664de5c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Ethos-U target module."""
from mlia.backend.corstone import CORSTONE_PRIORITY
from mlia.target.ethos_u.advisor import configure_and_get_ethosu_advisor
from mlia.target.ethos_u.config import EthosUConfiguration
from mlia.target.ethos_u.config import get_default_ethos_u_backends
from mlia.target.registry import registry
from mlia.target.registry import TargetInfo
SUPPORTED_BACKENDS_PRIORITY = ["Vela", *CORSTONE_PRIORITY]
for ethos_u in ("ethos-u55", "ethos-u65"):
registry.register(
ethos_u,
TargetInfo(
supported_backends=SUPPORTED_BACKENDS_PRIORITY,
default_backends=get_default_ethos_u_backends(SUPPORTED_BACKENDS_PRIORITY),
advisor_factory_func=configure_and_get_ethosu_advisor,
target_profile_cls=EthosUConfiguration,
),
)
|
