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 <[email protected]> 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]
