aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/devices
diff options
context:
space:
mode:
authorRaul Farkas <raul.farkas@arm.com>2022-11-04 16:12:52 +0000
committerRaul Farkas <raul.farkas@arm.com>2022-11-07 11:56:49 +0000
commit27668f6837c4f0a7197d13baff7bffc11f652fba (patch)
treefc363c7538d8f62ea841a55ec7d912bf9f41d446 /src/mlia/devices
parente356ab410ff9c1d36186cac57ca7457964ef97f2 (diff)
downloadmlia-27668f6837c4f0a7197d13baff7bffc11f652fba.tar.gz
MLIA-688 Fix --supported-ops-report flag
* Add `api.generate_supported_operators_report` that calls a report generator based on given target profile. * Update `--supported-ops-report` cli flag help section and README.md info by specifying that it is currently only supported for Ethos-U. * Add tests for `generate_supported_operators_report` method. * Add stub methods `report` for Cortex-A and Tosa operators.py module that throw a "Not supported" exception. Change-Id: Iaf373fe48299d807e48fd6a0e6c029ce32203f4f
Diffstat (limited to 'src/mlia/devices')
-rw-r--r--src/mlia/devices/cortexa/operators.py8
-rw-r--r--src/mlia/devices/ethosu/operators.py2
-rw-r--r--src/mlia/devices/tosa/operators.py8
3 files changed, 17 insertions, 1 deletions
diff --git a/src/mlia/devices/cortexa/operators.py b/src/mlia/devices/cortexa/operators.py
index 8fd2571..d46b107 100644
--- a/src/mlia/devices/cortexa/operators.py
+++ b/src/mlia/devices/cortexa/operators.py
@@ -29,3 +29,11 @@ def get_cortex_a_compatibility_info(
) -> CortexACompatibilityInfo | None:
"""Return list of model's operators."""
return None
+
+
+def report() -> None:
+ """Generate supported operators report."""
+ raise Exception(
+ "Generating a supported operators report is not "
+ "currently supported with Cortex-A target profile."
+ )
diff --git a/src/mlia/devices/ethosu/operators.py b/src/mlia/devices/ethosu/operators.py
index ff0d99f..1a4ce8d 100644
--- a/src/mlia/devices/ethosu/operators.py
+++ b/src/mlia/devices/ethosu/operators.py
@@ -9,6 +9,6 @@ from mlia.tools import vela_wrapper
logger = logging.getLogger(__name__)
-def generate_supported_operators_report() -> None:
+def report() -> None:
"""Generate supported operators report."""
vela_wrapper.generate_supported_operators_report()
diff --git a/src/mlia/devices/tosa/operators.py b/src/mlia/devices/tosa/operators.py
index 6cfb87f..03f6fb8 100644
--- a/src/mlia/devices/tosa/operators.py
+++ b/src/mlia/devices/tosa/operators.py
@@ -68,3 +68,11 @@ def get_tosa_checker(tflite_model_path: PathOrFileLike) -> TOSAChecker | None:
checker = tc.TOSAChecker(str(tflite_model_path))
return cast(TOSAChecker, checker)
+
+
+def report() -> None:
+ """Generate supported operators report."""
+ raise Exception(
+ "Generating a supported operators report is not "
+ "currently supported with TOSA target profile."
+ )