From 6d991efb049061c9cfa052018fbffe96b05b8e7c Mon Sep 17 00:00:00 2001 From: Elen Kalda Date: Mon, 9 May 2022 21:09:29 +0100 Subject: [PATCH] [microNPU] Remove spurious prints and improve documentation (#11247) --- tests/python/contrib/test_ethosu/test_encode_constants.py | 2 -- tests/python/contrib/test_ethosu/test_replace_conv2d.py | 8 ++++++++ 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/tests/python/contrib/test_ethosu/test_encode_constants.py b/tests/python/contrib/test_ethosu/test_encode_constants.py index 457edc861d06..92e6cd3e19cb 100644 --- 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 index b49890a9cf36..63f9fc44c778 100644 --- 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]