From ed2415f9ab801f3ebd154639dedfb6a8c4c1b22f Mon Sep 17 00:00:00 2001 From: Manupa Karunaratne Date: Mon, 30 Mar 2020 17:42:46 +0100 Subject: [PATCH] [RELAY] Re-wrote the Graph Partitioner to support multiple outputs *rebased --- tests/python/relay/test_pass_partition_graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/relay/test_pass_partition_graph.py b/tests/python/relay/test_pass_partition_graph.py index 594faf7ae462..0dfc89d469ca 100644 --- a/tests/python/relay/test_pass_partition_graph.py +++ b/tests/python/relay/test_pass_partition_graph.py @@ -782,7 +782,7 @@ def expected(): mod["main"] = create_graph() ref_mod = expected() partitioned = transform.PartitionGraph()(mod) - assert relay.analysis.alpha_equal(partitioned, ref_mod) + assert tvm.ir.structural_equal(partitioned, ref_mod, map_free_vars=True) def test_mixed_single_multiple_outputs(): @@ -857,7 +857,7 @@ def expected(): ref_mod = expected() partitioned = transform.PartitionGraph()(mod) - assert relay.analysis.alpha_equal(partitioned, ref_mod) + assert tvm.ir.structural_equal(partitioned, ref_mod, map_free_vars=True) if __name__ == "__main__": test_multi_node_compiler()