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 10b74f6e39eab27cf145eb12e320e513f75ed7be 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 | 4 +++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/Jenkinsfile b/Jenkinsfile index 64b8465..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 = "tlcpack/ci-gpu:v0.79" +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 = "tlcpack/ci-qemu:v0.08" +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 2e0d927..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] @@ -5090,6 +5091,7 @@ unsupported_onnx_tests = [ "test_reduce_sum_keepdims_random", "test_reduce_sum_negative_axes_keepdims_example", "test_reduce_sum_negative_axes_keepdims_random", + "test_reshape_allowzero_reordered", "test_rnn_seq_length", "test_round", "test_sequence_insert_at_back",
