[tvm] 07/13: jenkinsfile and one test

[an...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

andrewzhaoluo pushed a commit to branch ci-docker-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git

commit a5ee1bf68e5812d55ba3972d59df41fa8f57faa7
Author: Andrew Luo <an...@gmail.com>
AuthorDate: Thu Jan 13 15:13:26 2022 -0800

    jenkinsfile and one test
---
 Jenkinsfile                                | 4 ++--
 tests/python/frontend/onnx/test_forward.py | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/Jenkinsfile b/Jenkinsfile
index f547e0d..e788951 100755
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -46,11 +46,11 @@ import org.jenkinsci.plugins.pipeline.modeldefinition.Utils
 
 // NOTE: these lines are scanned by docker/dev_common.sh. Please update the regex as needed. -->
 ci_lint = "tlcpack/ci-lint:v0.67"
-ci_gpu = "tlcpackstaging/ci_gpu:20220113-230019-74854135a"
+ci_gpu = "tlcpack/ci-gpu:20220112-000102-d1ee20191"
 ci_cpu = "tlcpack/ci-cpu:v0.80"
 ci_wasm = "tlcpack/ci-wasm:v0.71"
 ci_i386 = "tlcpack/ci-i386:v0.74"
-ci_qemu = "tlcpackstaging/ci_qemu:20220113-230019-74854135a"
+ci_qemu = "tlcpack/ci-qemu:20220112-000102-d1ee20191"
 ci_arm = "tlcpack/ci-arm:v0.06"
 // <--- End of regex-scanned config.
 
diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
index 7485faa..1d69312 100644
--- a/tests/python/frontend/onnx/test_forward.py
+++ b/tests/python/frontend/onnx/test_forward.py
@@ -234,7 +234,8 @@ def verify_with_ort(
 
 
 def quantize_and_verify_with_ort(onnx_model, input_names, input_shapes, target, dev):
-    from onnxruntime.quantization import CalibrationDataReader, QuantType, quantize_static
+    from onnxruntime.quantization import (CalibrationDataReader, QuantType,
+                                          quantize_static)
 
     input_arrays = [np.random.random(shape).astype("float32") for shape in input_shapes]