update tests

apache · Jun 28, 2023 · 70f60ef · 70f60ef
1 parent 7cc3235
commit 70f60ef
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tests/python/contrib/test_ethosu/test_attr_passing.py b/tests/python/contrib/test_ethosu/test_attr_passing.py
@@ -27,6 +27,7 @@ def test_compiler_attr():
         "accelerator_config": "ethos-u55-32",
         "enable_cascader": True,
         "enable_striping": True,
+        "disable_copying_constants": True,
         "dev_force_block_config": "2x4x16",
         "dev_max_open_plans": "256",
         "dev_max_closed_plans": "128",
@@ -42,6 +43,7 @@ def test_compiler_attr():
             assert compiler_attrs.accelerator_config == config["accelerator_config"]
             assert compiler_attrs.enable_cascader == config["enable_cascader"]
             assert compiler_attrs.enable_striping == config["enable_striping"]
+            assert compiler_attrs.disable_copying_constants == config["disable_copying_constants"]
             assert compiler_attrs.dev_force_block_config == config["dev_force_block_config"]
             assert compiler_attrs.dev_max_open_plans == config["dev_max_open_plans"]
             assert compiler_attrs.dev_max_closed_plans == config["dev_max_closed_plans"]
@@ -60,6 +62,7 @@ def test_compiler_attr_default():
         "accelerator_config": "ethos-u55-256",
         "enable_cascader": False,
         "enable_striping": False,
+        "disable_copying_constants": False,
         "dev_force_block_config": "",
         "dev_max_open_plans": "8",
         "dev_max_closed_plans": "32",

diff --git a/tests/python/driver/tvmc/test_target_options.py b/tests/python/driver/tvmc/test_target_options.py
@@ -105,7 +105,6 @@ def test_ethosu_compiler_attrs():
     parsed, _ = parser.parse_known_args(
         ["--target-ethos-u-accelerator_config=ethos-u55-32", "--target-ethos-u-enable_cascader=1"]
     )
-    print(reconstruct_target_args(parsed))
     assert reconstruct_target_args(parsed) == {
         "ethos-u": {"accelerator_config": "ethos-u55-32", "enable_cascader": 1},
     }