This is an automated email from the ASF dual-hosted git repository.
masahi 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 52039c9 [TESTS] Fix running tests without MICRO (#9867)
52039c9 is described below
commit 52039c910aca2d54772b8960d7d868fb75f8936e
Author: Tristan Konolige <[email protected]>
AuthorDate: Sun Jan 9 15:04:20 2022 -0800
[TESTS] Fix running tests without MICRO (#9867)
---
tests/python/relay/aot/aot_test_utils.py | 2 ++
tests/python/relay/utils/external_codegen.py | 1 -
2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/python/relay/aot/aot_test_utils.py
b/tests/python/relay/aot/aot_test_utils.py
index e7ca5c6..3dfbd76 100644
--- a/tests/python/relay/aot/aot_test_utils.py
+++ b/tests/python/relay/aot/aot_test_utils.py
@@ -31,6 +31,8 @@ from typing import Any, NamedTuple, Union, Optional, List,
Dict
import pytest
import numpy as np
+pytest.importorskip("tvm.micro")
+
import tvm
from tvm import relay
from tvm import te
diff --git a/tests/python/relay/utils/external_codegen.py
b/tests/python/relay/utils/external_codegen.py
index 2d73ef8..680373c 100644
--- a/tests/python/relay/utils/external_codegen.py
+++ b/tests/python/relay/utils/external_codegen.py
@@ -24,7 +24,6 @@ import pytest
import tvm
from tvm import relay, runtime
from tvm.contrib import utils
-from tests.python.relay.aot.aot_test_utils import AOTTestModel, compile_and_run
skip_windows = pytest.mark.skipif(sys.platform == "win32", reason="Skip test
on Windows for now")