aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/target/cortex_a
diff options
context:
space:
mode:
authorBenjamin Klimczak <benjamin.klimczak@arm.com>2023-02-10 13:12:57 +0000
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2023-02-10 14:56:43 +0000
commit718277eaece76902c4950f18d428907b39a18ef1 (patch)
tree6fda1b8f8c2c0151f3697a2e9f31c2f5551cc2ff /src/mlia/target/cortex_a
parentfa1fad9332e2912f12a44a1b07716ee434174308 (diff)
downloadmlia-718277eaece76902c4950f18d428907b39a18ef1.tar.gz
MLIA-769 Add "pretty names" for targets / backends
- Provide "pretty names" to print information for targets and backends. - Use 'target_config' instead of 'target' if a target profile is used. - Fix minor issue in output regarding the output directory. Change-Id: Ib38231f30b4d609a0d1e8f9c52b2fb547c69cb6a
Diffstat (limited to 'src/mlia/target/cortex_a')
-rw-r--r--src/mlia/target/cortex_a/__init__.py5
-rw-r--r--src/mlia/target/cortex_a/events.py2
-rw-r--r--src/mlia/target/cortex_a/handlers.py2
-rw-r--r--src/mlia/target/cortex_a/reporters.py4
4 files changed, 7 insertions, 6 deletions
diff --git a/src/mlia/target/cortex_a/__init__.py b/src/mlia/target/cortex_a/__init__.py
index 87f268a..729dd30 100644
--- a/src/mlia/target/cortex_a/__init__.py
+++ b/src/mlia/target/cortex_a/__init__.py
@@ -9,9 +9,10 @@ from mlia.target.registry import TargetInfo
registry.register(
"cortex-a",
TargetInfo(
- supported_backends=["ArmNNTFLiteDelegate"],
- default_backends=["ArmNNTFLiteDelegate"],
+ supported_backends=["armnn-tflite-delegate"],
+ default_backends=["armnn-tflite-delegate"],
advisor_factory_func=configure_and_get_cortexa_advisor,
target_profile_cls=CortexAConfiguration,
),
+ pretty_name="Cortex-A",
)
diff --git a/src/mlia/target/cortex_a/events.py b/src/mlia/target/cortex_a/events.py
index 76f17ba..675335d 100644
--- a/src/mlia/target/cortex_a/events.py
+++ b/src/mlia/target/cortex_a/events.py
@@ -14,7 +14,7 @@ class CortexAAdvisorStartedEvent(Event):
"""Event with Cortex-A advisor parameters."""
model: Path
- target: CortexAConfiguration
+ target_config: CortexAConfiguration
class CortexAAdvisorEventHandler(EventDispatcher):
diff --git a/src/mlia/target/cortex_a/handlers.py b/src/mlia/target/cortex_a/handlers.py
index d46197c..a952c39 100644
--- a/src/mlia/target/cortex_a/handlers.py
+++ b/src/mlia/target/cortex_a/handlers.py
@@ -35,4 +35,4 @@ class CortexAEventHandler(WorkflowEventsHandler, CortexAAdvisorEventHandler):
def on_cortex_a_advisor_started(self, event: CortexAAdvisorStartedEvent) -> None:
"""Handle CortexAAdvisorStarted event."""
- self.reporter.submit(event.target)
+ self.reporter.submit(event.target_config)
diff --git a/src/mlia/target/cortex_a/reporters.py b/src/mlia/target/cortex_a/reporters.py
index e23bf4d..d214b09 100644
--- a/src/mlia/target/cortex_a/reporters.py
+++ b/src/mlia/target/cortex_a/reporters.py
@@ -23,13 +23,13 @@ from mlia.utils.console import style_improvement
from mlia.utils.types import is_list_of
-def report_target(target: CortexAConfiguration) -> Report:
+def report_target(target_config: CortexAConfiguration) -> Report:
"""Generate report for the target."""
return NestedReport(
"Target information",
"target",
[
- ReportItem("Target", alias="target", value=target.target),
+ ReportItem("Target", alias="target", value=target_config.target),
],
)