address comments

Change-Id: I67b61a2d2f67de25c47d2ace0e3a22c59ba8ea15
apache · Nov 16, 2021 · e1458c5 · e1458c5
1 parent b548e03
commit e1458c5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/tests/python/contrib/test_ethosu/test_codegen.py b/tests/python/contrib/test_ethosu/test_codegen.py
@@ -477,7 +477,7 @@ def create_relay_graph():
     ethosu_module = imported_modules[0]
 
     # Verify generated C source
-    get_cs = tvm._ffi.get_global_func("runtime.module.ethosu.getcs")
+    get_cs = tvm._ffi.get_global_func("runtime.module.ethos-u.getcs")
     cmms = get_cs(ethosu_module)
     cmms = bytes.fromhex(cmms)
 

diff --git a/tests/python/contrib/test_ethosu/test_encode_constants.py b/tests/python/contrib/test_ethosu/test_encode_constants.py
@@ -302,8 +302,8 @@ def get_graph():
     # Check tile address for the scalar constant input hasn't been
     # overwritten.
     extern_calls = tir_mod["main"].body.body.body.body.body
-    binary_elmtwise = extern_calls[-1].value
-    args = binary_elmtwise.args
+    binary_elementwise = extern_calls[-1].value
+    args = binary_elementwise.args
 
     reason = "Tile address overwritten"
     assert args[26] == 0, reason

diff --git a/tests/python/contrib/test_ethosu/test_legalize.py b/tests/python/contrib/test_ethosu/test_legalize.py
@@ -734,10 +734,10 @@ def verify(ext_func):
     mod = create_graph()
     mod = partition_ethosu_by_table(mod, pattern_table)
 
-    mod["tvmgen_default_ethosu_main_0"] = dataflow_pattern.rewrite(
-        rewriter, mod["tvmgen_default_ethosu_main_0"]
+    mod["tvmgen_default_ethos_u_main_0"] = dataflow_pattern.rewrite(
+        rewriter, mod["tvmgen_default_ethos_u_main_0"]
     )
-    verify(mod["tvmgen_default_ethosu_main_0"])
+    verify(mod["tvmgen_default_ethos_u_main_0"])
 
 
 @pytest.mark.parametrize(