Skip to content

Commit c069dbd

Browse files
mehrdadhylc
authored andcommitted
[microTVM][Zephyr] Hot Fix Bad Merge (apache#8980)
* fix bad merge * Fix test to reflect change of simd size for ARM
1 parent 3d1e97e commit c069dbd

3 files changed

Lines changed: 9 additions & 8 deletions

File tree

tests/micro/zephyr/conftest.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ def zephyr_boards() -> dict:
5656
def pytest_addoption(parser):
5757
parser.addoption(
5858
"--zephyr-board",
59+
required=True,
5960
choices=ZEPHYR_BOARDS.keys(),
6061
help=("Zephyr board for test."),
6162
)

tests/micro/zephyr/test_zephyr.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -380,11 +380,11 @@ def test_tensors(sess):
380380

381381

382382
@tvm.testing.requires_micro
383-
def test_autotune_conv2d(temp_dir, platform, west_cmd, tvm_debug):
383+
def test_autotune_conv2d(temp_dir, board, west_cmd, tvm_debug):
384384
"""Test AutoTune for microTVM Zephyr"""
385385
import tvm.relay as relay
386386

387-
model, zephyr_board = PLATFORMS[platform]
387+
model = conftest.ZEPHYR_BOARDS[board]
388388

389389
# Create a Relay model
390390
data_shape = (1, 3, 16, 16)
@@ -424,7 +424,7 @@ def test_autotune_conv2d(temp_dir, platform, west_cmd, tvm_debug):
424424
module_loader = tvm.micro.AutoTvmModuleLoader(
425425
template_project_dir=template_project_dir,
426426
project_options={
427-
"zephyr_board": zephyr_board,
427+
"zephyr_board": board,
428428
"west_cmd": west_cmd,
429429
"verbose": 1,
430430
"project_type": "host_driven",
@@ -467,7 +467,7 @@ def test_autotune_conv2d(temp_dir, platform, west_cmd, tvm_debug):
467467
lowered,
468468
temp_dir / "project",
469469
{
470-
"zephyr_board": zephyr_board,
470+
"zephyr_board": board,
471471
"west_cmd": west_cmd,
472472
"verbose": 1,
473473
"project_type": "host_driven",
@@ -496,7 +496,7 @@ def test_autotune_conv2d(temp_dir, platform, west_cmd, tvm_debug):
496496
lowered_tuned,
497497
temp_dir / "project",
498498
{
499-
"zephyr_board": zephyr_board,
499+
"zephyr_board": board,
500500
"west_cmd": west_cmd,
501501
"verbose": 1,
502502
"project_type": "host_driven",

tests/python/relay/aot/test_crt_aot.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -562,9 +562,9 @@ def test_name_sanitiser_name_clash():
562562
@pytest.mark.parametrize(
563563
"workspace_byte_alignment,main_workspace_size,sum_workspace_size",
564564
[
565-
(8, 10368, 15392),
566-
(16, 10368, 15424),
567-
(256, 10752, 17664),
565+
(8, 10368, 15200),
566+
(16, 10368, 15232),
567+
(256, 10752, 17408),
568568
],
569569
)
570570
def test_memory_planning(workspace_byte_alignment, main_workspace_size, sum_workspace_size):

0 commit comments

Comments
 (0)