From b327ee28eeeef637042736d71822cc4ad3facbfb Mon Sep 17 00:00:00 2001
From: Neil Hickey <neil.hickey@arm.com>
Date: Thu, 23 Feb 2023 14:23:57 +0000
Subject: [PATCH] [CI] Add onnx dependency to
 test_auto_tensorize.py::test_vnni_bert_int8

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 15e6bbe24c74..8900a1ff6c8a 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,