aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/devices/tosa/reporters.py
diff options
context:
space:
mode:
authorDmitrii Agibov <dmitrii.agibov@arm.com>2022-11-18 17:21:09 +0000
committerDmitrii Agibov <dmitrii.agibov@arm.com>2022-11-29 14:44:13 +0000
commit6a88ee5315b4ce5b023370c1e55e48bf9f2b6f67 (patch)
tree88edabf90228724f4fe2944b0ab23859d824a880 /src/mlia/devices/tosa/reporters.py
parenta34163c9d9a5cc0416bcaea2ebf8383bda9d505c (diff)
downloadmlia-6a88ee5315b4ce5b023370c1e55e48bf9f2b6f67.tar.gz
Rename modules
- Rename module "mlia.devices" into "mlia.target" - Rename module "mlia.target.ethosu" into "mlia.target.ethos_u" - Rename module "mlia.target.cortexa" into "mlia.target.cortex_a" - Rename and update tests Change-Id: I6dca7c8646d881f739fb6b5914d1cc7e45e63dc2
Diffstat (limited to 'src/mlia/devices/tosa/reporters.py')
-rw-r--r--src/mlia/devices/tosa/reporters.py83
1 files changed, 0 insertions, 83 deletions
diff --git a/src/mlia/devices/tosa/reporters.py b/src/mlia/devices/tosa/reporters.py
deleted file mode 100644
index e5559ee..0000000
--- a/src/mlia/devices/tosa/reporters.py
+++ /dev/null
@@ -1,83 +0,0 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
-# SPDX-License-Identifier: Apache-2.0
-"""Reports module."""
-from __future__ import annotations
-
-from typing import Any
-from typing import Callable
-
-from mlia.backend.tosa_checker.compat import Operator
-from mlia.core.advice_generation import Advice
-from mlia.core.reporters import report_advice
-from mlia.core.reporting import Cell
-from mlia.core.reporting import Column
-from mlia.core.reporting import Format
-from mlia.core.reporting import NestedReport
-from mlia.core.reporting import Report
-from mlia.core.reporting import ReportItem
-from mlia.core.reporting import Table
-from mlia.devices.tosa.config import TOSAConfiguration
-from mlia.utils.console import style_improvement
-from mlia.utils.types import is_list_of
-
-
-def report_device(device: TOSAConfiguration) -> Report:
- """Generate report for the device."""
- return NestedReport(
- "Device information",
- "device",
- [
- ReportItem("Target", alias="target", value=device.target),
- ],
- )
-
-
-def report_tosa_operators(ops: list[Operator]) -> Report:
- """Generate report for the operators."""
- return Table(
- [
- Column("#", only_for=["plain_text"]),
- Column(
- "Operator location",
- alias="operator_location",
- fmt=Format(wrap_width=30),
- ),
- Column("Operator name", alias="operator_name", fmt=Format(wrap_width=20)),
- Column(
- "TOSA compatibility",
- alias="is_tosa_compatible",
- fmt=Format(wrap_width=25),
- ),
- ],
- [
- (
- index + 1,
- op.location,
- op.name,
- Cell(
- op.is_tosa_compatible,
- Format(
- style=style_improvement(op.is_tosa_compatible),
- str_fmt=lambda v: "Compatible" if v else "Not compatible",
- ),
- ),
- )
- for index, op in enumerate(ops)
- ],
- name="Operators",
- alias="operators",
- )
-
-
-def tosa_formatters(data: Any) -> Callable[[Any], Report]:
- """Find appropriate formatter for the provided data."""
- if is_list_of(data, Advice):
- return report_advice
-
- if isinstance(data, TOSAConfiguration):
- return report_device
-
- if is_list_of(data, Operator):
- return report_tosa_operators
-
- raise Exception(f"Unable to find appropriate formatter for {data}")