Add further test guards

Change-Id: Ib1633341ed84e83c29c4220dd9bc4a7b6d17a5d0
apache · Jan 5, 2022 · 7a0cfb9 · 7a0cfb9
1 parent 9785c8c
commit 7a0cfb9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/tests/python/contrib/test_ethosu/cascader/conftest.py b/tests/python/contrib/test_ethosu/cascader/conftest.py
@@ -14,6 +14,10 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+import pytest
+
+pytest.importorskip("ethosu.vela")
+
 import tvm
 from tvm import relay
 from tvm.relay.testing import run_opt_pass
@@ -22,8 +26,6 @@
 from .infra import create_te_graph
 from ..infra import make_ethosu_conv2d
 
-import pytest
-
 
 def make_TwoConv2DTE():
     def _get_func():

diff --git a/tests/python/contrib/test_ethosu/cascader/test_graph.py b/tests/python/contrib/test_ethosu/cascader/test_graph.py
@@ -16,6 +16,8 @@
 # under the License.
 import pytest
 
+pytest.importorskip("ethosu.vela")
+
 import tvm.contrib.ethosu.cascader as cs