Skip to content

Commit 52edffa

Browse files
committed
Add test guards
Change-Id: I1d7633e20daab33642fa5c4a12e474a4def4d8b8
1 parent eea0801 commit 52edffa

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

tests/python/contrib/test_ethosu/cascader/test_propagator.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,12 @@
1414
# KIND, either express or implied. See the License for the
1515
# specific language governing permissions and limitations
1616
# under the License.
17-
from tvm.contrib.ethosu.cascader import StripeConfig, Propagator
18-
1917
import pytest
18+
19+
pytest.importorskip("ethosu.vela")
20+
2021
from math import isclose
22+
from tvm.contrib.ethosu.cascader import StripeConfig, Propagator
2123

2224

2325
def test_propagator():

tests/python/contrib/test_ethosu/cascader/test_stripe_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,12 @@
1414
# KIND, either express or implied. See the License for the
1515
# specific language governing permissions and limitations
1616
# under the License.
17-
from tvm.contrib.ethosu.cascader.stripe_config import StripeConfig, count_stripes
18-
1917
import pytest
2018

19+
pytest.importorskip("ethosu.vela")
20+
21+
from tvm.contrib.ethosu.cascader.stripe_config import StripeConfig, count_stripes
22+
2123

2224
def test_stripe_config():
2325
shape = [1, 2, 3]

0 commit comments

Comments
 (0)