You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by le...@apache.org on 2022/08/22 10:54:08 UTC
[tvm] 05/05: Apply transformers fix.
This is an automated email from the ASF dual-hosted git repository.
leandron pushed a commit to branch ci-docker-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 52f5c155f17ef459a4a09e79640364ca3f4388dd
Author: Leandro Nunes <le...@arm.com>
AuthorDate: Mon Aug 22 11:53:27 2022 +0100
Apply transformers fix.
Change-Id: Id96272f8bc7a10a8389307ab8247d0ffa595b7c4
---
tests/python/unittest/test_meta_schedule_integration.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/tests/python/unittest/test_meta_schedule_integration.py b/tests/python/unittest/test_meta_schedule_integration.py
index 50456dfd24..afce19a590 100644
--- a/tests/python/unittest/test_meta_schedule_integration.py
+++ b/tests/python/unittest/test_meta_schedule_integration.py
@@ -108,6 +108,10 @@ def test_meta_schedule_integration_extract_from_resnet():
@requires_torch
def test_meta_schedule_integration_extract_from_bert_base():
+ pytest.importorskip(
+ "transformers", reason="transformers package is required to import bert_base"
+ )
+
expected = {
"fused_nn_dense_2": (
12,