diff options
| author | Madeleine Dunn <madeleine.dunn@arm.com> | 2023-10-05 18:19:23 +0100 |
|---|---|---|
| committer | Madeleine Dunn <madeleine.dunn@arm.com> | 2024-04-03 16:06:42 +0100 |
| commit | a67bcd6eccaa4de00a08a3def8323debecff84d7 (patch) | |
| tree | 2fa3d79a02de4de1e937f21dabe5766073a4a067 /tests | |
| parent | 879138ebe9dd4b9387bddbdf4af45ba390f14bf6 (diff) | |
| download | mlia-a67bcd6eccaa4de00a08a3def8323debecff84d7.tar.gz | |
feat: Implement sparsity
- Add a placeholder file and registry option for sparsity
Resolves: MLIA-999
Signed-off-by: Madeleine Dunn <madeleine.dunn@arm.com>
Change-Id: I273192ba6813309f5226e3d8e0b686ce87ee6b79
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_cli_commands.py | 3 | ||||
| -rw-r--r-- | tests/test_nn_rewrite_core_rewrite.py | 5 |
2 files changed, 6 insertions, 2 deletions
diff --git a/tests/test_cli_commands.py b/tests/test_cli_commands.py index 9cda27c..5b2b947 100644 --- a/tests/test_cli_commands.py +++ b/tests/test_cli_commands.py @@ -126,7 +126,8 @@ def test_performance_unknown_target( Exception, match=re.escape( "Invalid rewrite target: 'random'. " - "Supported rewrites: ['fully-connected']" + "Supported rewrites: ['fully-connected'," + " 'fully_connected_sparsity24']" ), ), ], diff --git a/tests/test_nn_rewrite_core_rewrite.py b/tests/test_nn_rewrite_core_rewrite.py index b32fafd..d9c74ea 100644 --- a/tests/test_nn_rewrite_core_rewrite.py +++ b/tests/test_nn_rewrite_core_rewrite.py @@ -100,7 +100,10 @@ def test_register_rewrite_function() -> None: def test_builtin_rewrite_names() -> None: """Test if all builtin rewrites are properly registered and returned.""" - assert RewritingOptimizer.builtin_rewrite_names() == ["fully-connected"] + assert RewritingOptimizer.builtin_rewrite_names() == [ + "fully-connected", + "fully_connected_sparsity24", + ] def test_rewrite_function_autoload() -> None: |