[tvm] branch main updated: [CI] Add onnx dependency to test_auto_tensorize.py::test_vnni_bert_int8 (#14102)

[ekalda]

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

ekalda pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new a8c5804c49 [CI] Add onnx dependency to 
test_auto_tensorize.py::test_vnni_bert_int8 (#14102)
a8c5804c49 is described below

commit a8c5804c49c45505a611fb1e501ad049403450ad
Author: neildhickey <neil.hic...@arm.com>
AuthorDate: Thu Feb 23 19:37:19 2023 +0000

    [CI] Add onnx dependency to test_auto_tensorize.py::test_vnni_bert_int8 
(#14102)
    
    The test modified by this PR relies on onnx conversion to run successfully, 
skipping if onnx isn't available, which is the case for ci_cpu docker image
---
 tests/python/integration/test_auto_tensorize.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/python/integration/test_auto_tensorize.py 
b/tests/python/integration/test_auto_tensorize.py
index 15e6bbe24c..8900a1ff6c 100644
--- a/tests/python/integration/test_auto_tensorize.py
+++ b/tests/python/integration/test_auto_tensorize.py
@@ -336,6 +336,7 @@ def test_dp4a_conv2d():
 @tvm.testing.requires_cascadelake
 @pytest.mark.skipif(tvm.testing.IS_IN_CI, reason="Slow on CI")
 def test_vnni_bert_int8():
+    pytest.importorskip("onnx")
     relay_mod, params, input_info = load_quantized_bert_base()
     _test_bert_int8(
         relay_mod,