diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-07-21 14:07:12 +0100 |
---|---|---|
committer | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2022-08-19 10:23:23 +0100 |
commit | e51c8ce569f7d0cd28606572f96260b93997b02c (patch) | |
tree | 23c336cfa51ffdb9382897c11fe681d84a00caac /src/mlia/cli/main.py | |
parent | 664d8c55609253e68d153a91514c8fefa00557b1 (diff) | |
download | mlia-e51c8ce569f7d0cd28606572f96260b93997b02c.tar.gz |
MLIA-550 Enable TOSA checker in the CLI
Change-Id: I9dfa9f17a93ac24be0aa9ad68e4a72b79aecfb9d
Diffstat (limited to 'src/mlia/cli/main.py')
-rw-r--r-- | src/mlia/cli/main.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/mlia/cli/main.py b/src/mlia/cli/main.py index 33fcdeb..f8fc00c 100644 --- a/src/mlia/cli/main.py +++ b/src/mlia/cli/main.py @@ -4,6 +4,7 @@ import argparse import logging import sys +from functools import partial from inspect import signature from pathlib import Path from typing import Dict @@ -44,6 +45,7 @@ Supported targets: - Ethos-U55 <op compatibility, perf estimation, model opt> - Ethos-U65 <op compatibility, perf estimation, model opt> + - TOSA <op compatibility> """.strip() @@ -79,7 +81,7 @@ def get_commands() -> List[CommandInfo]: performance, ["perf"], [ - add_target_options, + partial(add_target_options, profiles_to_skip=["tosa"]), add_tflite_model_options, add_output_options, add_debug_options, @@ -90,7 +92,7 @@ def get_commands() -> List[CommandInfo]: optimization, ["opt"], [ - add_target_options, + partial(add_target_options, profiles_to_skip=["tosa"]), add_keras_model_options, add_multi_optimization_options, add_output_options, |