From ebd7694df81e9f8fcdb4f4d050f3ff15d92c7207 Mon Sep 17 00:00:00 2001 From: Elen Kalda Date: Thu, 24 Mar 2022 14:27:36 +0000 Subject: [PATCH] [microNPU] Remove unused import and command stream printing This is a follow up to https://github.com/apache/tvm/pull/10695. Change-Id: I7f2dc14826cefea81fe5ff69c6255cdb5dc7f5c0 --- tests/python/contrib/test_ethosu/test_codegen.py | 2 +- tests/python/contrib/test_ethosu/test_replace_identity.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/python/contrib/test_ethosu/test_codegen.py b/tests/python/contrib/test_ethosu/test_codegen.py index 999970442108..49349209f92a 100644 --- a/tests/python/contrib/test_ethosu/test_codegen.py +++ b/tests/python/contrib/test_ethosu/test_codegen.py @@ -460,7 +460,7 @@ def pooling(x): op = tf.nn.relu(op) return op - _compare_tvm_with_tflite(pooling, [ifm_shape], accel_type, print_cmm=True) + _compare_tvm_with_tflite(pooling, [ifm_shape], accel_type) @pytest.mark.parametrize("accel_type", ACCEL_TYPES) diff --git a/tests/python/contrib/test_ethosu/test_replace_identity.py b/tests/python/contrib/test_ethosu/test_replace_identity.py index 741065a1610e..2155d33f43c0 100644 --- a/tests/python/contrib/test_ethosu/test_replace_identity.py +++ b/tests/python/contrib/test_ethosu/test_replace_identity.py @@ -14,7 +14,6 @@ # KIND, either express or implied. See the License for the # specific language governing permissions and limitations # under the License. -from numpy.core.shape_base import block import pytest pytest.importorskip("ethosu.vela")