diff --git a/vta/tests/python/integration/test_benchmark_topi_conv2d_transpose.py b/vta/tests/python/integration/test_benchmark_topi_conv2d_transpose.py index 2f4e6c4935b49..399c468355a37 100644 --- a/vta/tests/python/integration/test_benchmark_topi_conv2d_transpose.py +++ b/vta/tests/python/integration/test_benchmark_topi_conv2d_transpose.py @@ -157,12 +157,13 @@ def _run(env, remote): ('DCGAN.CT3', Workload(1, 16, 16, 256, 128, 4, 4, 1, 1, 2, 2)), ] - # for tsk in tasks: - # print(tsk) - # name, wkl = tsk - # run_vta_conv2d_transpose(env, remote, name, wkl) - # exit() + for tsk in tasks: + print(tsk) + name, wkl = tsk + run_vta_conv2d_transpose(env, remote, name, wkl) + return + # TUNER map_list = {} for i, tsk in enumerate(tasks): print(tsk) diff --git a/vta/tests/python/integration/test_benchmark_topi_group_conv2d.py b/vta/tests/python/integration/test_benchmark_topi_group_conv2d.py index 53758245eb51e..65d861f0e8511 100644 --- a/vta/tests/python/integration/test_benchmark_topi_group_conv2d.py +++ b/vta/tests/python/integration/test_benchmark_topi_group_conv2d.py @@ -161,12 +161,13 @@ def _run(env, remote): ('mobilenet.D9', Workload(1, 7, 7, 1024, 1024, 64, 3, 3, 1, 1, 1, 1)), ] - # for tsk in tasks: - # print(tsk) - # name, wkl = tsk - # run_vta_group_conv2d(env, remote, name, wkl) - # return + for tsk in tasks: + print(tsk) + name, wkl = tsk + run_vta_group_conv2d(env, remote, name, wkl) + return + # TUNER map_list = {} for i, tsk in enumerate(tasks): print(tsk)