diff options
author | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2023-02-02 14:02:05 +0000 |
---|---|---|
committer | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2023-02-10 13:45:18 +0000 |
commit | 7a661257b6adad0c8f53e32b42ced56a1e7d952f (patch) | |
tree | 938ad8578c5b9edc0573e810ce64ce0a5bda3d8c /src/mlia/cli/main.py | |
parent | 50271dee0a84bfc481ce798184f07b5b0b4bc64d (diff) | |
download | mlia-7a661257b6adad0c8f53e32b42ced56a1e7d952f.tar.gz |
MLIA-769 Expand use of target/backend registries
- Use the target/backend registries to avoid hard-coded names.
- Cache target profiles to avoid re-loading them
Change-Id: I474b7c9ef23894e1d8a3ea06d13a37652054c62e
Diffstat (limited to 'src/mlia/cli/main.py')
-rw-r--r-- | src/mlia/cli/main.py | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/src/mlia/cli/main.py b/src/mlia/cli/main.py index 793e155..b3a9d4c 100644 --- a/src/mlia/cli/main.py +++ b/src/mlia/cli/main.py @@ -18,6 +18,7 @@ from mlia.cli.commands import check from mlia.cli.commands import optimize from mlia.cli.common import CommandInfo from mlia.cli.helpers import CLIActionResolver +from mlia.cli.helpers import copy_profile_file_to_output_dir from mlia.cli.options import add_backend_install_options from mlia.cli.options import add_backend_options from mlia.cli.options import add_backend_uninstall_options @@ -30,11 +31,11 @@ from mlia.cli.options import add_output_directory from mlia.cli.options import add_output_options from mlia.cli.options import add_target_options from mlia.cli.options import get_output_format +from mlia.core.common import AdviceCategory from mlia.core.context import ExecutionContext from mlia.core.errors import ConfigurationError from mlia.core.errors import InternalError from mlia.core.logging import setup_logging -from mlia.target.config import copy_profile_file_to_output_dir from mlia.target.registry import table as target_table @@ -59,7 +60,13 @@ def get_commands() -> list[CommandInfo]: [ add_output_directory, add_model_options, - add_target_options, + partial( + add_target_options, + supported_advice=[ + AdviceCategory.COMPATIBILITY, + AdviceCategory.PERFORMANCE, + ], + ), add_backend_options, add_check_category_options, add_output_options, @@ -72,7 +79,9 @@ def get_commands() -> list[CommandInfo]: [ add_output_directory, add_keras_model_options, - partial(add_target_options, profiles_to_skip=["tosa", "cortex-a"]), + partial( + add_target_options, supported_advice=[AdviceCategory.OPTIMIZATION] + ), partial( add_backend_options, backends_to_skip=["tosa-checker", "ArmNNTFLiteDelegate"], |