Arm(R) Ethos(TM)-U NPU codegen integration

*skipping tests in non ci_cpu containers Change-Id: If331f7a4d241760cadf91ff64e475675ddef634b
apache · Aug 26, 2021 · d6d8bec · d6d8bec
1 parent f858170
commit d6d8bec
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tests/python/contrib/test_ethosu/test_codegen.py b/tests/python/contrib/test_ethosu/test_codegen.py
@@ -15,10 +15,12 @@
 # specific language governing permissions and limitations
 # under the License.
 # pylint: disable=invalid-name, unused-argument
+import pytest
+
+pytest.importorskip("ethosu.vela")
 import os
 import numpy as np
 import pathlib
-import pytest
 
 import tvm
 import tvm.micro as micro

diff --git a/tests/python/contrib/test_ethosu/test_networks.py b/tests/python/contrib/test_ethosu/test_networks.py
@@ -15,6 +15,9 @@
 # specific language governing permissions and limitations
 # under the License.
 # pylint: disable=invalid-name, unused-argument
+import pytest
+
+pytest.importorskip("ethosu.vela")
 from tests.python.relay.aot.aot_test_utils import (
  convert_to_relay,
  generate_ref_data,