Skip to content

Commit eef2c3e

Browse files
committed
set stack and heap size for the test
1 parent 94b9a58 commit eef2c3e

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/micro/zephyr/test_ms_tuning.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,12 +60,9 @@ def create_relay_module():
6060
return mod, params, model_info
6161

6262

63-
@tvm.testing.requires_micro
6463
@tvm.testing.requires_micro
6564
def test_ms_tuning_conv2d(workspace_dir, board, microtvm_debug, use_fvp, serial_number):
66-
"""Test AutoTune for microTVM Zephyr"""
67-
if board != "qemu_x86":
68-
pytest.xfail(f"Autotune fails on {board}.")
65+
"""Test meta-schedule tuning for microTVM Zephyr"""
6966

7067
mod, params, model_info = create_relay_module()
7168
input_name = model_info["in_tensor"]
@@ -80,6 +77,8 @@ def test_ms_tuning_conv2d(workspace_dir, board, microtvm_debug, use_fvp, serial_
8077
"project_type": "host_driven",
8178
"use_fvp": bool(use_fvp),
8279
"serial_number": serial_number,
80+
"config_main_stack_size": 8192,
81+
"heap_size_bytes": 512 * 1024,
8382
}
8483

8584
boards_file = pathlib.Path(tvm.micro.get_microtvm_template_projects("zephyr")) / "boards.json"

0 commit comments

Comments
 (0)