refine graph draw

NHZlX · NHZlX · commit 943950c19084 · 2018-08-08T14:32:04.000Z
diff --git a/paddle/fluid/inference/analysis/dfg_graphviz_draw_pass.cc b/paddle/fluid/inference/analysis/dfg_graphviz_draw_pass.cc
@@ -46,9 +46,9 @@ std::string DFG_GraphvizDrawPass::Draw(DataFlowGraph *graph) {
   for (size_t i = 0; i < graph->nodes.size(); i++) {
     const Node &node = graph->nodes.Get(i);
     if (!config_.display_deleted_node && node.deleted()) continue;
-    for (auto &in : node.inlinks) {
-      if (!config_.display_deleted_node && in->deleted()) continue;
-      dot.AddEdge(in->repr(), node.repr(), {});
+    for (auto &out : node.outlinks) {
+      if (!config_.display_deleted_node && out->deleted()) continue;
+      dot.AddEdge(node.repr(), out->repr(), {});
     }
   }
   return dot.Build();

Original file line number	Diff line number	Diff line change
`@@ -46,9 +46,9 @@ std::string DFG_GraphvizDrawPass::Draw(DataFlowGraph *graph) {`
`46`	`46`	`for (size_t i = 0; i < graph->nodes.size(); i++) {`
`47`	`47`	`const Node &node = graph->nodes.Get(i);`
`48`	`48`	`if (!config_.display_deleted_node && node.deleted()) continue;`
`49`		`- for (auto &in : node.inlinks) {`
`50`		`- if (!config_.display_deleted_node && in->deleted()) continue;`
`51`		`- dot.AddEdge(in->repr(), node.repr(), {});`
	`49`	`+ for (auto &out : node.outlinks) {`
	`50`	`+ if (!config_.display_deleted_node && out->deleted()) continue;`
	`51`	`+ dot.AddEdge(node.repr(), out->repr(), {});`
`52`	`52`	`}`
`53`	`53`	`}`
`54`	`54`	`return dot.Build();`