guberti commented on code in PR #12207:
URL: https://github.com/apache/tvm/pull/12207#discussion_r933801203
##########
tests/micro/arduino/conftest.py:
##########
@@ -15,35 +15,19 @@
# specific language governing permissions and limitations
# under the License.
-import pytest
+pytest_plugins = [
+ "tvm.micro.testing.pytest_plugin",
+]
-from test_utils import ARDUINO_BOARDS
+import pytest
def pytest_addoption(parser):
- parser.addoption(
- "--arduino-board",
- nargs="+",
- required=True,
- choices=ARDUINO_BOARDS.keys(),
- help="Arduino board for tests.",
- )
parser.addoption(
"--arduino-cli-cmd",
Review Comment:
yea you’re right, build tool is just too abstract.
##########
python/tvm/micro/testing/pytest_plugin.py:
##########
@@ -0,0 +1,108 @@
+# 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.
+
+# pylint: disable=invalid-name,redefined-outer-name
+""" microTVM testing fixtures used to deduce testing argument
+ values from testing parameters """
+
+import pathlib
+import os
+import datetime
+import pytest
+
+from tvm.contrib.utils import tempdir
+
+from .utils import get_supported_boards
+
+
+def pytest_addoption(parser):
+ """Adds more pytest arguments"""
+ parser.addoption(
+ "--board",
+ required=True,
+ choices=list(get_supported_boards("zephyr").keys())
+ + list(get_supported_boards("arduino").keys()),
+ help=(
+ "microTVM boards for tests. Board refers to instances"
+ "of microcontrollers/emulators defined in a platform."
+ ),
+ )
+ parser.addoption(
+ "--test-build-only",
+ action="store_true",
+ help="Only run tests that don't require physical hardware.",
+ )
+ parser.addoption(
+ "--tvm-debug",
+ action="store_true",
+ default=False,
+ help="If set true, it will keep the project directory for debugging.",
+ )
+
+
+@pytest.fixture(scope="session")
+def board(request):
+ return request.config.getoption("--board")
+
+
+@pytest.fixture(scope="session")
+def tvm_debug(request):
Review Comment:
nit: I don't love the name `tvm_debug` if all this flag does is keep the
project directory - IMO `--keep-project-dir` or `--preserve-project` makes more
sense. If it does things besides this, we should document them in the `help`
string.
##########
tests/micro/zephyr/test_zephyr.py:
##########
@@ -89,7 +89,7 @@ def _make_add_sess(temp_dir, model, zephyr_board, west_cmd,
build_config, dtype=
# The same test code can be executed on both the QEMU simulation and on real
hardware.
@tvm.testing.requires_micro
@pytest.mark.skip_boards(["mps2_an521"])
-def test_add_uint(temp_dir, board, west_cmd, tvm_debug):
+def test_add_uint(workspace_dir, board, west_cmd, tvm_debug):
Review Comment:
Huh, I’d forgotten that workspace_dir is kept when debug is passed. That
sounds super useful - good call with this!
##########
python/tvm/micro/testing/evaluation.py:
##########
@@ -153,4 +163,4 @@ def evaluate_model_accuracy(session, aot_executor,
input_data, true_labels, runs
num_correct = sum(u == v for u, v in zip(true_labels, predicted_labels))
average_time = sum(aot_runtimes) / len(aot_runtimes)
accuracy = num_correct / len(predicted_labels)
- return average_time, accuracy
+ return average_time, accuracy, predicted_labels
Review Comment:
Ah, I'd forgotten we need the labels for some hardware in the loop tests.
That seems fine - for anomaly detection and other AOC metrics using the same
"template", we could use confidence values (or even just `None`) in place of
`predicted_labels`. This LGTM now.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscr...@tvm.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
