From 41ffef7623d2a1b8eac46efdf72c33ce2c24fb38 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Sun, 19 Jan 2020 16:35:26 +0900 Subject: [PATCH] enable other tests --- .../python/relay/test_pass_partition_graph.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/tests/python/relay/test_pass_partition_graph.py b/tests/python/relay/test_pass_partition_graph.py index 10d1c2dbd88d5..a244c50872144 100644 --- 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()