diff options
author | Nathan Bailey <nathan.bailey@arm.com> | 2024-02-15 14:50:58 +0000 |
---|---|---|
committer | Nathan Bailey <nathan.bailey@arm.com> | 2024-03-14 15:45:40 +0000 |
commit | 0b552d2ae47da4fb9c16d2a59d6ebe12c8307771 (patch) | |
tree | 09b40b939acbe0bcf02dcc77a7ed7ce4aba94322 /tests/test_cli_commands.py | |
parent | 09b272be6e88d84a30cb89fb71f3fc3c64d20d2e (diff) | |
download | mlia-0b552d2ae47da4fb9c16d2a59d6ebe12c8307771.tar.gz |
feat: Enable rewrite parameterisation
Enables user to provide a toml or default profile to change training settings for rewrite optimization
Resolves: MLIA-1004
Signed-off-by: Nathan Bailey <nathan.bailey@arm.com>
Change-Id: I3bf9f44b9a2062fb71ef36eb32c9a69edcc48061
Diffstat (limited to 'tests/test_cli_commands.py')
-rw-r--r-- | tests/test_cli_commands.py | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/tests/test_cli_commands.py b/tests/test_cli_commands.py index 1ce793f..1a9bbb8 100644 --- a/tests/test_cli_commands.py +++ b/tests/test_cli_commands.py @@ -52,13 +52,15 @@ def test_performance_unknown_target( @pytest.mark.parametrize( - "target_profile, pruning, clustering, pruning_target, clustering_target, " - "rewrite, rewrite_target, rewrite_start, rewrite_end, expected_error", + "target_profile, pruning, clustering, optimization_profile, pruning_target, " + "clustering_target, rewrite, rewrite_target, rewrite_start, rewrite_end ," + "expected_error", [ [ "ethos-u55-256", True, False, + None, 0.5, None, False, @@ -73,6 +75,7 @@ def test_performance_unknown_target( False, None, None, + None, True, "fully_connected", "sequential/flatten/Reshape", @@ -83,6 +86,7 @@ def test_performance_unknown_target( "ethos-u55-256", True, False, + None, 0.5, None, True, @@ -98,6 +102,7 @@ def test_performance_unknown_target( "ethos-u65-512", False, True, + None, 0.5, 32, False, @@ -110,6 +115,7 @@ def test_performance_unknown_target( "ethos-u55-256", False, False, + None, 0.5, None, True, @@ -128,6 +134,7 @@ def test_performance_unknown_target( "ethos-u55-256", False, False, + None, 0.5, None, True, @@ -146,6 +153,7 @@ def test_performance_unknown_target( "ethos-u55-256", False, False, + None, "invalid", None, True, @@ -169,6 +177,7 @@ def test_opt_valid_optimization_target( # pylint: disable=too-many-locals,too-m clustering: bool, pruning_target: float | None, clustering_target: int | None, + optimization_profile: str | None, rewrite: bool, rewrite_target: str | None, rewrite_start: str | None, @@ -192,6 +201,7 @@ def test_opt_valid_optimization_target( # pylint: disable=too-many-locals,too-m model=str(model_type), pruning=pruning, clustering=clustering, + optimization_profile=optimization_profile, pruning_target=pruning_target, clustering_target=clustering_target, rewrite=rewrite, |