@@ -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" ,
0 commit comments