diff options
author | Raul Farkas <raul.farkas@arm.com> | 2022-11-04 16:12:52 +0000 |
---|---|---|
committer | Raul Farkas <raul.farkas@arm.com> | 2022-11-07 11:56:49 +0000 |
commit | 27668f6837c4f0a7197d13baff7bffc11f652fba (patch) | |
tree | fc363c7538d8f62ea841a55ec7d912bf9f41d446 /src/mlia/devices | |
parent | e356ab410ff9c1d36186cac57ca7457964ef97f2 (diff) | |
download | mlia-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.py | 8 | ||||
-rw-r--r-- | src/mlia/devices/ethosu/operators.py | 2 | ||||
-rw-r--r-- | src/mlia/devices/tosa/operators.py | 8 |
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." + ) |