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/cli | |
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/cli')
-rw-r--r-- | src/mlia/cli/commands.py | 4 | ||||
-rw-r--r-- | src/mlia/cli/options.py | 3 |
2 files changed, 4 insertions, 3 deletions
diff --git a/src/mlia/cli/commands.py b/src/mlia/cli/commands.py index e044e1a..72ae4bb 100644 --- a/src/mlia/cli/commands.py +++ b/src/mlia/cli/commands.py @@ -23,11 +23,11 @@ from pathlib import Path from typing import cast from mlia.api import ExecutionContext +from mlia.api import generate_supported_operators_report from mlia.api import get_advice from mlia.api import PathOrFileLike from mlia.cli.config import get_installation_manager from mlia.cli.options import parse_optimization_parameters -from mlia.devices.ethosu.operators import generate_supported_operators_report from mlia.utils.console import create_section_header from mlia.utils.types import only_one_selected @@ -129,7 +129,7 @@ def operators( "model.tflite") """ if supported_ops_report: - generate_supported_operators_report() + generate_supported_operators_report(target_profile) logger.info("Report saved into SUPPORTED_OPS.md") return diff --git a/src/mlia/cli/options.py b/src/mlia/cli/options.py index e5e85f0..b28fa8f 100644 --- a/src/mlia/cli/options.py +++ b/src/mlia/cli/options.py @@ -124,7 +124,8 @@ def add_custom_supported_operators_options(parser: argparse.ArgumentParser) -> N default=False, help=( "Generate the SUPPORTED_OPS.md file in the " - "current working directory and exit" + "current working directory and exit " + "(Ethos-U target profiles only)" ), ) |