From 8841df9d9403b4446e535cca126e141a688ef61b Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Wed, 8 Aug 2018 15:22:11 -0700 Subject: [PATCH] update topi test (#13) * update schedule for dcgan * update schedule for dcgan --- .../test_benchmark_topi_conv2d_transpose.py | 11 ++++++----- .../integration/test_benchmark_topi_group_conv2d.py | 11 ++++++----- 2 files changed, 12 insertions(+), 10 deletions(-) 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)