[microNPU] Remove spurious prints and improve documentation (apache#…

…11247)
shingjan · May 17, 2022 · 91730be · 91730be
1 parent 898bf9c
commit 91730be
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/tests/python/contrib/test_ethosu/test_encode_constants.py b/tests/python/contrib/test_ethosu/test_encode_constants.py
@@ -359,7 +359,6 @@ def _get_func():
         func = _get_func()
         mod, consts = _lower_to_tir(func)
 
-        print(mod.script())
         script = mod.script(show_meta=True)
         test_mod = tvm.script.from_source(script)
         tvm.ir.assert_structural_equal(test_mod["main"], reference_mod["main"], True)
@@ -513,7 +512,6 @@ def _get_func():
 
         script = mod.script(show_meta=True)
         test_mod = tvm.script.from_source(script)
-        print(mod.script())
         tvm.ir.assert_structural_equal(test_mod["main"], reference_mod["main"], True)
 
         test_const_size = [value.size for value in list(consts.values())]

diff --git a/tests/python/contrib/test_ethosu/test_replace_conv2d.py b/tests/python/contrib/test_ethosu/test_replace_conv2d.py
@@ -138,6 +138,14 @@ def get_conv2d_args(call, include_buffers=False, remove_constants=False):
     conv_args = []
     remove_indices = [0]
 
+    # call.args[41]: BufferLoad for the first half of the weights
+    # call.args[42]: length of the load of the first half of the weights
+    # call.args[43]: BufferLoad for the second half of the weights
+    # call.args[44]: length of the load of the second half of the weights
+    # call.args[46]: BufferLoad for the first half of the bias
+    # call.args[47]: length of the load of the first half of the bias
+    # call.args[48]: BufferLoad for the second half of the bias
+    # call.args[49]: length of the load of the second half of the bias
     if remove_constants:
         remove_indices += [41, 42, 43, 44, 46, 47, 48, 49]