diff options
author | Madeleine Dunn <madeleine.dunn@arm.com> | 2024-03-26 13:20:42 +0000 |
---|---|---|
committer | Madeleine Dunn <madeleine.dunn@arm.com> | 2024-03-27 10:45:34 +0000 |
commit | 5f063ae1cfbfa2568d2858af0a0ccaf192bb1e8d (patch) | |
tree | 47ff3c52dfca780a69d2eeba037e1602fe50b655 /tests/test_nn_rewrite_core_rewrite.py | |
parent | c7ee5b783f044d7ff641773aa385840f5ff944cc (diff) | |
download | mlia-5f063ae1cfbfa2568d2858af0a0ccaf192bb1e8d.tar.gz |
fix: Update rewrite target name
- Rename "fully_connected" to "fully-connected"
- This will resolve issues with upstreaming rewrite library changes
Signed-off-by: Madeleine Dunn <madeleine.dunn@arm.com>
Change-Id: I2f24ae4917a556fd0bd44f0db6ee4e0f7a68cd24
Diffstat (limited to 'tests/test_nn_rewrite_core_rewrite.py')
-rw-r--r-- | tests/test_nn_rewrite_core_rewrite.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/tests/test_nn_rewrite_core_rewrite.py b/tests/test_nn_rewrite_core_rewrite.py index 363d614..b32fafd 100644 --- a/tests/test_nn_rewrite_core_rewrite.py +++ b/tests/test_nn_rewrite_core_rewrite.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2024, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2023-2024, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Tests for module mlia.nn.rewrite.core.rewrite.""" from __future__ import annotations @@ -41,14 +41,14 @@ def test_rewrite() -> None: @pytest.mark.parametrize( "rewrite_name, expected_error", [ - ("fully_connected", does_not_raise()), + ("fully-connected", does_not_raise()), ("random", does_not_raise()), ], ) def test_rewrite_configuration( test_tflite_model_fp32: Path, rewrite_name: str, expected_error: Any ) -> None: - """Test get_rewrite function only supports rewrite type fully_connected.""" + """Test get_rewrite function only supports rewrite type fully-connected.""" with expected_error: config_obj = RewriteConfiguration( rewrite_name, @@ -67,9 +67,9 @@ def test_rewriting_optimizer( test_tflite_model_fp32: Path, test_tfrecord_fp32: Path, ) -> None: - """Test fc_layer rewrite process with rewrite type fully_connected.""" + """Test fc_layer rewrite process with rewrite type fully-connected.""" config_obj = RewriteConfiguration( - "fully_connected", + "fully-connected", ["sequential/flatten/Reshape", "StatefulPartitionedCall:0"], test_tfrecord_fp32, train_params=MockTrainingParameters(), @@ -100,7 +100,7 @@ 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"] def test_rewrite_function_autoload() -> None: @@ -146,7 +146,7 @@ def test_rewrite_configuration_train_params( ) config_obj = RewriteConfiguration( - "fully_connected", + "fully-connected", ["sequential/flatten/Reshape", "StatefulPartitionedCall:0"], test_tfrecord_fp32, train_params=train_params, |