From 5704adad9553667872cc1d71ddc1f05645c94e2b Mon Sep 17 00:00:00 2001 From: Wang Yao Date: Fri, 19 Jul 2019 11:30:01 -0700 Subject: [PATCH] Fix test --- tests/python/unittest/test_graph_tuner_core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/unittest/test_graph_tuner_core.py b/tests/python/unittest/test_graph_tuner_core.py index f49497fe2710..30b037e1598e 100644 --- a/tests/python/unittest/test_graph_tuner_core.py +++ b/tests/python/unittest/test_graph_tuner_core.py @@ -274,7 +274,7 @@ def test_many_sub_graphs(): net = relay.Function(relay.analysis.free_vars(out), out) net, params = relay.testing.create_workload(net) - tasks = autotvm.task.extract_from_program(net, + tasks = autotvm.task.extract_from_program(net["main"], target=target, params=params, ops=(relay.op.nn.conv2d,))