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
65
66
67
68
69
|
# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Ethos-U configuration."""
from __future__ import annotations
import logging
from typing import Any
from mlia.backend.vela.compiler import resolve_compiler_config
from mlia.backend.vela.compiler import VelaCompilerOptions
from mlia.target.config import TargetProfile
from mlia.utils.filesystem import get_vela_config
logger = logging.getLogger(__name__)
class EthosUConfiguration(TargetProfile):
"""Ethos-U configuration."""
def __init__(self, **kwargs: Any) -> None:
"""Init Ethos-U target configuration."""
target = kwargs["target"]
super().__init__(target)
mac = kwargs["mac"]
self.mac = mac
self.compiler_options = VelaCompilerOptions(
system_config=kwargs["system_config"],
memory_mode=kwargs["memory_mode"],
config_files=str(get_vela_config()),
accelerator_config=f"{self.target}-{mac}", # type: ignore
)
def verify(self) -> None:
"""Check the parameters."""
super().verify()
target_mac_ranges = {
"ethos-u55": [32, 64, 128, 256],
"ethos-u65": [256, 512],
}
if self.target not in target_mac_ranges:
raise ValueError(f"Unsupported target: {self.target}")
target_mac_range = target_mac_ranges[self.target]
if self.mac not in target_mac_range:
raise ValueError(
f"Mac value for selected device should be in {target_mac_range}."
)
@property
def resolved_compiler_config(self) -> dict[str, Any]:
"""Resolve compiler configuration."""
return resolve_compiler_config(self.compiler_options)
def __str__(self) -> str:
"""Return string representation."""
return (
f"Ethos-U target={self.target} "
f"mac={self.mac} "
f"compiler_options={self.compiler_options}"
)
def __repr__(self) -> str:
"""Return string representation."""
return f"<Ethos-U configuration target={self.target}>"
|
