Sergei Smirnov
76c78a9d40
Co-authored-by: Sergey Smirnov <Sergey.Smirnov@mir.dev> Co-authored-by: Ekaterina Bern <Ekaterina.Bern@mir.dev> Co-authored-by: Mikhail Trubnikov <Mikhail.Trubnikov@mir.dev> Co-authored-by: German Tretiakov <german.tretiakov@mir.dev> Co-authored-by: Ilya Gozman <Ilya.Gozman@mir.dev> Co-authored-by: Alexey.Yazev <Alexey.Yazev@mir.dev> Co-authored-by: Ilya Gozman <92577591+ilyag-grovety@users.noreply.github.com>
61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import sys
|
|
import tvm
|
|
import pytest
|
|
|
|
collect_ignore = []
|
|
if sys.platform.startswith("win"):
|
|
collect_ignore.append("frontend/caffe")
|
|
collect_ignore.append("frontend/caffe2")
|
|
collect_ignore.append("frontend/coreml")
|
|
collect_ignore.append("frontend/darknet")
|
|
collect_ignore.append("frontend/keras")
|
|
collect_ignore.append("frontend/mxnet")
|
|
collect_ignore.append("frontend/pytorch")
|
|
collect_ignore.append("frontend/tensorflow")
|
|
collect_ignore.append("frontend/tflite")
|
|
collect_ignore.append("frontend/onnx")
|
|
collect_ignore.append("driver/tvmc/test_autoscheduler.py")
|
|
collect_ignore.append("unittest/test_auto_scheduler_cost_model.py") # stack overflow
|
|
# collect_ignore.append("unittest/test_auto_scheduler_measure.py") # exception ignored
|
|
collect_ignore.append("unittest/test_auto_scheduler_search_policy.py") # stack overflow
|
|
# collect_ignore.append("unittest/test_auto_scheduler_measure.py") # exception ignored
|
|
|
|
collect_ignore.append("unittest/test_tir_intrin.py")
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
parser.addoption(
|
|
"--enable-corstone300-tests",
|
|
action="store_true",
|
|
default=False,
|
|
help="Run Corstone-300 FVP tests",
|
|
)
|
|
|
|
|
|
def pytest_collection_modifyitems(config, items):
|
|
if not config.getoption("--enable-corstone300-tests"):
|
|
for item in items:
|
|
if "corstone300" in item.keywords:
|
|
item.add_marker(
|
|
pytest.mark.skip(
|
|
reason="Need --enable-corstone300-tests option to run this test"
|
|
)
|
|
)
|