You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2020/09/11 16:55:47 UTC
[incubator-tvm] branch master updated: [TEST] Fix Some Failed Test
Cases and Tutorials of The Issue #6453 (#6454)
This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new d956c2f [TEST] Fix Some Failed Test Cases and Tutorials of The Issue #6453 (#6454)
d956c2f is described below
commit d956c2f4174702babde9a21f7c7cafedcf5f3a7c
Author: Johnson9009 <jo...@163.com>
AuthorDate: Sat Sep 12 00:55:36 2020 +0800
[TEST] Fix Some Failed Test Cases and Tutorials of The Issue #6453 (#6454)
---
tests/python/contrib/test_miopen.py | 1 +
tests/python/contrib/test_mps.py | 1 +
tests/python/contrib/test_nnpack.py | 1 +
tests/python/contrib/test_rocblas.py | 1 +
tests/python/contrib/test_sort.py | 1 +
tests/python/contrib/test_sparse.py | 1 +
tests/python/frontend/tensorflow/test_bn_dynamic.py | 1 +
tests/python/integration/test_ewise_fpga.py | 1 +
tests/python/relay/test_adt.py | 1 +
tests/python/relay/test_op_fast_math.py | 1 +
tests/python/topi/python/test_topi_bitserial_conv2d.py | 1 +
tests/python/topi/python/test_topi_bitserial_dense.py | 1 +
tests/python/topi/python/test_topi_bnn.py | 1 +
tests/python/topi/python/test_topi_broadcast.py | 1 +
tests/python/topi/python/test_topi_conv1d.py | 1 +
tests/python/topi/python/test_topi_conv2d_hwnc_tensorcore.py | 1 +
tests/python/topi/python/test_topi_conv3d_ndhwc.py | 1 +
tests/python/topi/python/test_topi_correlation.py | 1 +
tests/python/topi/python/test_topi_dilate.py | 1 +
tests/python/topi/python/test_topi_group_conv2d_NCHWc_int8.py | 1 +
tests/python/topi/python/test_topi_image.py | 1 +
tests/python/topi/python/test_topi_math.py | 1 +
tests/python/topi/python/test_topi_matmul.py | 1 +
tests/python/topi/python/test_topi_reduce.py | 1 +
tests/python/topi/python/test_topi_space_to_depth.py | 1 +
tests/python/topi/python/test_topi_upsampling.py | 1 +
tests/python/unittest/test_arith_deduce_bound.py | 1 +
tests/python/unittest/test_arith_detect_clip_bound.py | 1 +
tests/python/unittest/test_arith_detect_linear_equation.py | 1 +
tests/python/unittest/test_auto_scheduler_search_policy.py | 1 +
tests/python/unittest/test_runtime_container.py | 1 +
tests/python/unittest/test_runtime_graph.py | 1 +
tests/python/unittest/test_runtime_graph_debug.py | 1 +
tests/python/unittest/test_target_codegen_c_host.py | 1 +
tests/python/unittest/test_target_codegen_cross_llvm.py | 1 +
tests/python/unittest/test_target_codegen_llvm.py | 1 +
tests/python/unittest/test_target_codegen_rocm.py | 1 +
tests/python/unittest/test_target_codegen_vulkan.py | 1 +
tests/python/unittest/test_te_schedule_bound_inference.py | 1 +
tests/python/unittest/test_te_tensor.py | 1 +
tests/python/unittest/test_tir_buffer.py | 1 +
tests/python/unittest/test_tir_intrin.py | 1 +
tests/python/unittest/test_tir_transform_inject_copy_intrin.py | 1 +
tests/python/unittest/test_tir_transform_instrument_bound_checkers.py | 1 +
tests/python/unittest/test_tir_transform_loop_partition.py | 1 +
tests/python/unittest/test_tir_transform_prim_func_pass.py | 1 +
tutorials/optimize/opt_gemm.py | 1 +
vta/tests/python/integration/test_benchmark_gemm.py | 1 +
vta/tutorials/optimize/convolution_opt.py | 1 +
49 files changed, 49 insertions(+)
diff --git a/tests/python/contrib/test_miopen.py b/tests/python/contrib/test_miopen.py
index c80d465..3177367 100644
--- a/tests/python/contrib/test_miopen.py
+++ b/tests/python/contrib/test_miopen.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
from tvm.contrib import miopen
import numpy as np
diff --git a/tests/python/contrib/test_mps.py b/tests/python/contrib/test_mps.py
index 2e2c143..75a76f0 100644
--- a/tests/python/contrib/test_mps.py
+++ b/tests/python/contrib/test_mps.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import numpy as np
from tvm.contrib import mps
diff --git a/tests/python/contrib/test_nnpack.py b/tests/python/contrib/test_nnpack.py
index 5c4d87b..a92b930 100644
--- a/tests/python/contrib/test_nnpack.py
+++ b/tests/python/contrib/test_nnpack.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import numpy as np
import scipy.signal
diff --git a/tests/python/contrib/test_rocblas.py b/tests/python/contrib/test_rocblas.py
index 670268f..9b8bacb 100644
--- a/tests/python/contrib/test_rocblas.py
+++ b/tests/python/contrib/test_rocblas.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import numpy as np
from tvm.contrib import rocblas
diff --git a/tests/python/contrib/test_sort.py b/tests/python/contrib/test_sort.py
index 5660efa..7bd3a9c 100644
--- a/tests/python/contrib/test_sort.py
+++ b/tests/python/contrib/test_sort.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import numpy as np
diff --git a/tests/python/contrib/test_sparse.py b/tests/python/contrib/test_sparse.py
index 9cea49b..39cff18 100644
--- a/tests/python/contrib/test_sparse.py
+++ b/tests/python/contrib/test_sparse.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import tvm.contrib.sparse as tvmsp
import tvm.runtime.ndarray as _nd
diff --git a/tests/python/frontend/tensorflow/test_bn_dynamic.py b/tests/python/frontend/tensorflow/test_bn_dynamic.py
index d7a00b2..ac02ef1 100644
--- a/tests/python/frontend/tensorflow/test_bn_dynamic.py
+++ b/tests/python/frontend/tensorflow/test_bn_dynamic.py
@@ -21,6 +21,7 @@ This is a test script to test fused_batch_norm operators
in TensorFlow frontend when mean and variance are not given.
"""
import tvm
+import tvm.testing
import numpy as np
try:
diff --git a/tests/python/integration/test_ewise_fpga.py b/tests/python/integration/test_ewise_fpga.py
index ab05f7f..2e5364b 100644
--- a/tests/python/integration/test_ewise_fpga.py
+++ b/tests/python/integration/test_ewise_fpga.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import numpy as np
import os
diff --git a/tests/python/relay/test_adt.py b/tests/python/relay/test_adt.py
index 784abcb..c443fcd 100644
--- a/tests/python/relay/test_adt.py
+++ b/tests/python/relay/test_adt.py
@@ -14,6 +14,7 @@
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
+import pytest
import tvm
from tvm import relay
from tvm.relay import testing
diff --git a/tests/python/relay/test_op_fast_math.py b/tests/python/relay/test_op_fast_math.py
index d067ab5..1658658 100644
--- a/tests/python/relay/test_op_fast_math.py
+++ b/tests/python/relay/test_op_fast_math.py
@@ -18,6 +18,7 @@ import numpy as np
import scipy
from scipy import special
import tvm
+import tvm.testing
import tvm.relay as relay
from tvm import topi
from tvm import te
diff --git a/tests/python/topi/python/test_topi_bitserial_conv2d.py b/tests/python/topi/python/test_topi_bitserial_conv2d.py
index b276aad..8e55c7b 100644
--- a/tests/python/topi/python/test_topi_bitserial_conv2d.py
+++ b/tests/python/topi/python/test_topi_bitserial_conv2d.py
@@ -18,6 +18,7 @@ import numpy as np
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
from tvm.topi.util import get_const_tuple
from tvm.contrib.pickle_memoize import memoize
diff --git a/tests/python/topi/python/test_topi_bitserial_dense.py b/tests/python/topi/python/test_topi_bitserial_dense.py
index db6ad5b..8b565ac 100644
--- a/tests/python/topi/python/test_topi_bitserial_dense.py
+++ b/tests/python/topi/python/test_topi_bitserial_dense.py
@@ -20,6 +20,7 @@ import numpy as np
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
from tvm.topi.util import get_const_tuple
from tvm.contrib.pickle_memoize import memoize
diff --git a/tests/python/topi/python/test_topi_bnn.py b/tests/python/topi/python/test_topi_bnn.py
index 753fb17..1bec2d0 100644
--- a/tests/python/topi/python/test_topi_bnn.py
+++ b/tests/python/topi/python/test_topi_bnn.py
@@ -17,6 +17,7 @@
"""Test code for binary neural network operators."""
import numpy as np
import tvm
+import tvm.testing
from tvm import te
from tvm import topi
from tvm.topi.util import get_const_tuple
diff --git a/tests/python/topi/python/test_topi_broadcast.py b/tests/python/topi/python/test_topi_broadcast.py
index 6f3e91b..44be28c 100644
--- a/tests/python/topi/python/test_topi_broadcast.py
+++ b/tests/python/topi/python/test_topi_broadcast.py
@@ -19,6 +19,7 @@ import numpy as np
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
diff --git a/tests/python/topi/python/test_topi_conv1d.py b/tests/python/topi/python/test_topi_conv1d.py
index 3d98120..5339103 100644
--- a/tests/python/topi/python/test_topi_conv1d.py
+++ b/tests/python/topi/python/test_topi_conv1d.py
@@ -20,6 +20,7 @@ import itertools
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
from tvm.contrib.pickle_memoize import memoize
from tvm.topi.util import get_const_tuple
diff --git a/tests/python/topi/python/test_topi_conv2d_hwnc_tensorcore.py b/tests/python/topi/python/test_topi_conv2d_hwnc_tensorcore.py
index 81563ba..7589764 100644
--- a/tests/python/topi/python/test_topi_conv2d_hwnc_tensorcore.py
+++ b/tests/python/topi/python/test_topi_conv2d_hwnc_tensorcore.py
@@ -20,6 +20,7 @@
import numpy as np
import tvm
import os
+import tvm.testing
import tvm.topi.testing
from tvm import te, autotvm, topi
from tvm.contrib.pickle_memoize import memoize
diff --git a/tests/python/topi/python/test_topi_conv3d_ndhwc.py b/tests/python/topi/python/test_topi_conv3d_ndhwc.py
index a9d54ea..2baacc0 100644
--- a/tests/python/topi/python/test_topi_conv3d_ndhwc.py
+++ b/tests/python/topi/python/test_topi_conv3d_ndhwc.py
@@ -20,6 +20,7 @@ import numpy as np
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
from tvm.contrib.pickle_memoize import memoize
from tvm.topi.util import get_const_tuple
diff --git a/tests/python/topi/python/test_topi_correlation.py b/tests/python/topi/python/test_topi_correlation.py
index acff50d..17c21d6 100644
--- a/tests/python/topi/python/test_topi_correlation.py
+++ b/tests/python/topi/python/test_topi_correlation.py
@@ -20,6 +20,7 @@ import tvm
from tvm import te
from tvm import autotvm
from tvm import topi
+import tvm.testing
import tvm.topi.testing
from tvm.contrib.pickle_memoize import memoize
from tvm.topi.util import get_const_tuple
diff --git a/tests/python/topi/python/test_topi_dilate.py b/tests/python/topi/python/test_topi_dilate.py
index 872ee05..ab5c61d 100644
--- a/tests/python/topi/python/test_topi_dilate.py
+++ b/tests/python/topi/python/test_topi_dilate.py
@@ -17,6 +17,7 @@
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
import numpy as np
diff --git a/tests/python/topi/python/test_topi_group_conv2d_NCHWc_int8.py b/tests/python/topi/python/test_topi_group_conv2d_NCHWc_int8.py
index 36b7f29..89e0e04 100644
--- a/tests/python/topi/python/test_topi_group_conv2d_NCHWc_int8.py
+++ b/tests/python/topi/python/test_topi_group_conv2d_NCHWc_int8.py
@@ -22,6 +22,7 @@ import tvm
from tvm import te
from tvm import autotvm
from tvm import topi
+import tvm.testing
import tvm.topi.testing
from tvm.contrib.pickle_memoize import memoize
from tvm.topi.util import get_const_tuple
diff --git a/tests/python/topi/python/test_topi_image.py b/tests/python/topi/python/test_topi_image.py
index 79e0930..518ee1f 100644
--- a/tests/python/topi/python/test_topi_image.py
+++ b/tests/python/topi/python/test_topi_image.py
@@ -19,6 +19,7 @@ import numpy as np
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
from tvm.contrib.pickle_memoize import memoize
diff --git a/tests/python/topi/python/test_topi_math.py b/tests/python/topi/python/test_topi_math.py
index 48b6d9b..149ed82 100644
--- a/tests/python/topi/python/test_topi_math.py
+++ b/tests/python/topi/python/test_topi_math.py
@@ -20,6 +20,7 @@ from scipy import special
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
from tvm.topi import util
diff --git a/tests/python/topi/python/test_topi_matmul.py b/tests/python/topi/python/test_topi_matmul.py
index f3b3396..f6933d7 100644
--- a/tests/python/topi/python/test_topi_matmul.py
+++ b/tests/python/topi/python/test_topi_matmul.py
@@ -16,6 +16,7 @@
# under the License.
import numpy as np
import tvm
+import tvm.testing
from tvm import te
from tvm import topi
from tvm.topi.util import get_const_tuple
diff --git a/tests/python/topi/python/test_topi_reduce.py b/tests/python/topi/python/test_topi_reduce.py
index 2df26dd..daf380d 100644
--- a/tests/python/topi/python/test_topi_reduce.py
+++ b/tests/python/topi/python/test_topi_reduce.py
@@ -20,6 +20,7 @@ import numpy as np
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
diff --git a/tests/python/topi/python/test_topi_space_to_depth.py b/tests/python/topi/python/test_topi_space_to_depth.py
index 7cdaa68..3970186 100644
--- a/tests/python/topi/python/test_topi_space_to_depth.py
+++ b/tests/python/topi/python/test_topi_space_to_depth.py
@@ -19,6 +19,7 @@ import numpy as np
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
diff --git a/tests/python/topi/python/test_topi_upsampling.py b/tests/python/topi/python/test_topi_upsampling.py
index 4bce660..85042c0 100644
--- a/tests/python/topi/python/test_topi_upsampling.py
+++ b/tests/python/topi/python/test_topi_upsampling.py
@@ -19,6 +19,7 @@ import numpy as np
import tvm
from tvm import te
from tvm import topi
+import tvm.testing
import tvm.topi.testing
import math
from tvm.topi.util import nchw_pack_layout
diff --git a/tests/python/unittest/test_arith_deduce_bound.py b/tests/python/unittest/test_arith_deduce_bound.py
index d72a0e0..5c6976a 100644
--- a/tests/python/unittest/test_arith_deduce_bound.py
+++ b/tests/python/unittest/test_arith_deduce_bound.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
diff --git a/tests/python/unittest/test_arith_detect_clip_bound.py b/tests/python/unittest/test_arith_detect_clip_bound.py
index 4d887d6..e12afa8 100644
--- a/tests/python/unittest/test_arith_detect_clip_bound.py
+++ b/tests/python/unittest/test_arith_detect_clip_bound.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
diff --git a/tests/python/unittest/test_arith_detect_linear_equation.py b/tests/python/unittest/test_arith_detect_linear_equation.py
index 16b75f7..cedb557 100644
--- a/tests/python/unittest/test_arith_detect_linear_equation.py
+++ b/tests/python/unittest/test_arith_detect_linear_equation.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
diff --git a/tests/python/unittest/test_auto_scheduler_search_policy.py b/tests/python/unittest/test_auto_scheduler_search_policy.py
index 3302dd5..6ec96a6 100644
--- a/tests/python/unittest/test_auto_scheduler_search_policy.py
+++ b/tests/python/unittest/test_auto_scheduler_search_policy.py
@@ -22,6 +22,7 @@ import numpy as np
import tempfile
import tvm
+import tvm.testing
from tvm import auto_scheduler
from test_auto_scheduler_common import matmul_auto_scheduler_test, PropagatingThread
diff --git a/tests/python/unittest/test_runtime_container.py b/tests/python/unittest/test_runtime_container.py
index a56a65b..3e89a51 100644
--- a/tests/python/unittest/test_runtime_container.py
+++ b/tests/python/unittest/test_runtime_container.py
@@ -17,6 +17,7 @@
import numpy as np
import tvm
+import tvm.testing
import pickle
from tvm import te
from tvm import nd, relay
diff --git a/tests/python/unittest/test_runtime_graph.py b/tests/python/unittest/test_runtime_graph.py
index 4e6f1e9..5968f0b 100644
--- a/tests/python/unittest/test_runtime_graph.py
+++ b/tests/python/unittest/test_runtime_graph.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import numpy as np
import json
diff --git a/tests/python/unittest/test_runtime_graph_debug.py b/tests/python/unittest/test_runtime_graph_debug.py
index db9b0ce..25dadbd 100644
--- a/tests/python/unittest/test_runtime_graph_debug.py
+++ b/tests/python/unittest/test_runtime_graph_debug.py
@@ -17,6 +17,7 @@
import json
import os
import tvm
+import tvm.testing
from tvm import te
import numpy as np
from tvm import rpc
diff --git a/tests/python/unittest/test_target_codegen_c_host.py b/tests/python/unittest/test_target_codegen_c_host.py
index 6afbfc7..a2ed634 100644
--- a/tests/python/unittest/test_target_codegen_c_host.py
+++ b/tests/python/unittest/test_target_codegen_c_host.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import numpy as np
from tvm.contrib import util
diff --git a/tests/python/unittest/test_target_codegen_cross_llvm.py b/tests/python/unittest/test_target_codegen_cross_llvm.py
index 2c6c519..c0ab651 100644
--- a/tests/python/unittest/test_target_codegen_cross_llvm.py
+++ b/tests/python/unittest/test_target_codegen_cross_llvm.py
@@ -16,6 +16,7 @@
# under the License.
"""Test cross compilation"""
import tvm
+import tvm.testing
from tvm import te
import os
import struct
diff --git a/tests/python/unittest/test_target_codegen_llvm.py b/tests/python/unittest/test_target_codegen_llvm.py
index 98190ac..e9f7a01 100644
--- a/tests/python/unittest/test_target_codegen_llvm.py
+++ b/tests/python/unittest/test_target_codegen_llvm.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
from tvm import topi
from tvm.contrib import util, clang
diff --git a/tests/python/unittest/test_target_codegen_rocm.py b/tests/python/unittest/test_target_codegen_rocm.py
index 1d5cbd8..36a659b 100644
--- a/tests/python/unittest/test_target_codegen_rocm.py
+++ b/tests/python/unittest/test_target_codegen_rocm.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import numpy as np
import unittest
diff --git a/tests/python/unittest/test_target_codegen_vulkan.py b/tests/python/unittest/test_target_codegen_vulkan.py
index 61ac36e..d530456 100644
--- a/tests/python/unittest/test_target_codegen_vulkan.py
+++ b/tests/python/unittest/test_target_codegen_vulkan.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import re
import numpy as np
diff --git a/tests/python/unittest/test_te_schedule_bound_inference.py b/tests/python/unittest/test_te_schedule_bound_inference.py
index de2178c..03e9886 100644
--- a/tests/python/unittest/test_te_schedule_bound_inference.py
+++ b/tests/python/unittest/test_te_schedule_bound_inference.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
diff --git a/tests/python/unittest/test_te_tensor.py b/tests/python/unittest/test_te_tensor.py
index a3936fa..6d8d9db 100644
--- a/tests/python/unittest/test_te_tensor.py
+++ b/tests/python/unittest/test_te_tensor.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
import numpy as np
from tvm import te
from tvm.topi.nn.pooling import pool
diff --git a/tests/python/unittest/test_tir_buffer.py b/tests/python/unittest/test_tir_buffer.py
index 3007509..fcbab04 100644
--- a/tests/python/unittest/test_tir_buffer.py
+++ b/tests/python/unittest/test_tir_buffer.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
from tvm.tir import Buffer
import numpy as np
diff --git a/tests/python/unittest/test_tir_intrin.py b/tests/python/unittest/test_tir_intrin.py
index 0a1fe56..3d54175 100644
--- a/tests/python/unittest/test_tir_intrin.py
+++ b/tests/python/unittest/test_tir_intrin.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
from tvm import topi
from tvm.contrib import util, clang
diff --git a/tests/python/unittest/test_tir_transform_inject_copy_intrin.py b/tests/python/unittest/test_tir_transform_inject_copy_intrin.py
index 49a643a..86bf87d 100644
--- a/tests/python/unittest/test_tir_transform_inject_copy_intrin.py
+++ b/tests/python/unittest/test_tir_transform_inject_copy_intrin.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
diff --git a/tests/python/unittest/test_tir_transform_instrument_bound_checkers.py b/tests/python/unittest/test_tir_transform_instrument_bound_checkers.py
index bb9e5ee..187013e 100644
--- a/tests/python/unittest/test_tir_transform_instrument_bound_checkers.py
+++ b/tests/python/unittest/test_tir_transform_instrument_bound_checkers.py
@@ -16,6 +16,7 @@
# under the License.
import pytest
import tvm
+import tvm.testing
from tvm import te
import numpy as np
diff --git a/tests/python/unittest/test_tir_transform_loop_partition.py b/tests/python/unittest/test_tir_transform_loop_partition.py
index f9beac7..be2f307 100644
--- a/tests/python/unittest/test_tir_transform_loop_partition.py
+++ b/tests/python/unittest/test_tir_transform_loop_partition.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import numpy
diff --git a/tests/python/unittest/test_tir_transform_prim_func_pass.py b/tests/python/unittest/test_tir_transform_prim_func_pass.py
index c513640..647c446 100644
--- a/tests/python/unittest/test_tir_transform_prim_func_pass.py
+++ b/tests/python/unittest/test_tir_transform_prim_func_pass.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
diff --git a/tutorials/optimize/opt_gemm.py b/tutorials/optimize/opt_gemm.py
index 96a75ac..ead6660 100644
--- a/tutorials/optimize/opt_gemm.py
+++ b/tutorials/optimize/opt_gemm.py
@@ -56,6 +56,7 @@ Intel i7-4770HQ CPU. The cache line size should be 64 bytes for all the x86 CPUs
# Then we write a baseline implementation, the simplest way to write a matrix multiplication in TVM.
import tvm
+import tvm.testing
from tvm import te
import numpy
import timeit
diff --git a/vta/tests/python/integration/test_benchmark_gemm.py b/vta/tests/python/integration/test_benchmark_gemm.py
index f37b418..432d6f7 100644
--- a/vta/tests/python/integration/test_benchmark_gemm.py
+++ b/vta/tests/python/integration/test_benchmark_gemm.py
@@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import tvm
+import tvm.testing
from tvm import te
import numpy as np
from tvm.contrib import util
diff --git a/vta/tutorials/optimize/convolution_opt.py b/vta/tutorials/optimize/convolution_opt.py
index 05479c3..3f079e8 100644
--- a/vta/tutorials/optimize/convolution_opt.py
+++ b/vta/tutorials/optimize/convolution_opt.py
@@ -39,6 +39,7 @@ from __future__ import absolute_import, print_function
import os
import tvm
+import tvm.testing
from tvm import te
import vta
import numpy as np