enable other tests

apache · Jan 19, 2020 · 41ffef7 · 41ffef7
1 parent efd9d1e
commit 41ffef7
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/tests/python/relay/test_pass_partition_graph.py b/tests/python/relay/test_pass_partition_graph.py
@@ -179,7 +179,6 @@ def __init__(self, backend):
 
     def annotate_call(self, call):
         new_args = []
-        has_arg = "nn.conv2d"
         for arg in call.args:
             new_arg = super().visit(arg)
             if call.op.name == "nn.conv2d" or isinstance(new_arg, (relay.expr.Var, relay.expr.Constant)):
@@ -571,9 +570,9 @@ def test_exec(mod, params, ref_mod, ref_params, out_shape):
     test_partition()
     test_partition_mobilenet()
 
-    # net = get_net()
-    # mod, params = tvm.relay.testing.create_workload(net)
-    # ref_mod, ref_params = tvm.relay.testing.create_workload(net)
+    net = get_net()
+    mod, params = tvm.relay.testing.create_workload(net)
+    ref_mod, ref_params = tvm.relay.testing.create_workload(net)
     # test_exec(mod, params, ref_mod, ref_params, (1, 16, 224, 224))
 
     mod, params = relay.testing.mobilenet.get_workload()
@@ -582,10 +581,10 @@ def test_exec(mod, params, ref_mod, ref_params, out_shape):
 
 
 if __name__ == "__main__":
-    # test_multi_node_compiler()
-    # test_extern_ccompiler_single_op()
-    # test_extern_ccompiler_default_ops()
-    # test_extern_ccompiler()
-    # test_extern_dnnl()
-    # test_extern_dnnl_mobilenet()
+    test_multi_node_compiler()
+    test_extern_ccompiler_single_op()
+    test_extern_ccompiler_default_ops()
+    test_extern_ccompiler()
+    test_extern_dnnl()
+    test_extern_dnnl_mobilenet()
     test_partition_conv_bias_relu()