[RELAY] Re-wrote the Graph Partitioner to support multiple outputs

*rebased
apache · Mar 31, 2020 · ed2415f · ed2415f
1 parent cf2d501
commit ed2415f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()