update topi test (#13)

* update schedule for dcgan * update schedule for dcgan
jroesch · Dec 1, 2018 · 5d0827c · 5d0827c
1 parent cabcc35
commit 5d0827c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 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
@@ -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
@@ -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)