Skip to content

Commit

Permalink
Add test guards
Browse files Browse the repository at this point in the history
Change-Id: I34e48e4ce4fd14e6c92601412f5029aa08efe8fc
  • Loading branch information
mbaret committed Nov 25, 2021
1 parent 9661b33 commit 6e1b3d4
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -14,12 +14,15 @@
# 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")

from tvm import te
import tvm.contrib.ethosu.cascader as cs
from tvm.relay.backend.contrib.ethosu.te.convolution import match_ethosu_conv2d, conv2d_compute

import numpy as np
import pytest


def _make_matrices(kernel, stride, dilation, padding, ifm_channels, ifm_layout, ofm_layout):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,13 +14,15 @@
# 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")

from tvm import te
from tvm.topi.transform import reshape
import tvm.contrib.ethosu.cascader as cs
from tvm.relay.backend.contrib.ethosu.te.inline import match_ethosu_inline

import pytest


def test_ethosu_inline_matcher():
ifm_shape = (2, 5, 6)
Expand Down
6 changes: 4 additions & 2 deletions tests/python/contrib/test_ethosu/cascader/test_ethosu_part.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,11 +14,13 @@
# 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.contrib.ethosu.cascader as pl
from tvm.contrib.ethosu.cascader.parts import EthosuPart

import pytest


def test_ethosu_part():
te_subgraph = pl.TESubgraph([], None)
Expand Down

0 comments on commit 6e1b3d4

Please sign in to comment.