You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
2.3 KiB
73 lines
2.3 KiB
/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License. */
|
|
|
|
#include <algorithm>
|
|
#include <unordered_set>
|
|
|
|
#include "paddle/fluid/framework/ir/graph_viz_pass.h"
|
|
|
|
namespace paddle {
|
|
namespace framework {
|
|
namespace ir {
|
|
static const char kGraphVizPath[] = "graph_viz_path";
|
|
|
|
std::unique_ptr<ir::Graph> GraphVizPass::ApplyImpl(
|
|
std::unique_ptr<ir::Graph> graph) const {
|
|
const std::string graph_viz_path = Get<std::string>(kGraphVizPath);
|
|
std::unique_ptr<std::ostream> fout(new std::ofstream(graph_viz_path));
|
|
PADDLE_ENFORCE(fout->good());
|
|
std::ostream& sout = *fout;
|
|
|
|
size_t var_id = 0;
|
|
std::unordered_map<const ir::Node*, size_t> vars;
|
|
|
|
sout << "digraph G {\n";
|
|
|
|
for (const ir::Node* n : graph->Nodes()) {
|
|
if (n->NodeType() != ir::Node::Type::kVariable) continue;
|
|
size_t cur_var_id = var_id++;
|
|
vars[n] = cur_var_id;
|
|
|
|
sout << "var_" << cur_var_id << " [label=\"" << n->Name() << "\"]"
|
|
<< std::endl;
|
|
}
|
|
|
|
size_t op_id = 0;
|
|
for (const ir::Node* n : graph->Nodes()) {
|
|
if (n->NodeType() != ir::Node::Type::kOperation) continue;
|
|
std::string op_name = "op_" + std::to_string(op_id++);
|
|
sout << op_name << " [label=\"" << n->Name() << "\", shape=rect]"
|
|
<< std::endl;
|
|
for (auto in : n->inputs) {
|
|
std::string var_name = "var_" + std::to_string(vars[in]);
|
|
sout << var_name << " -> " << op_name << std::endl;
|
|
}
|
|
|
|
for (auto out : n->outputs) {
|
|
std::string var_name = "var_" + std::to_string(vars[out]);
|
|
sout << op_name << " -> " << var_name << std::endl;
|
|
}
|
|
}
|
|
|
|
sout << "}\n";
|
|
return graph;
|
|
}
|
|
|
|
} // namespace ir
|
|
} // namespace framework
|
|
} // namespace paddle
|
|
|
|
REGISTER_PASS(graph_viz_pass, paddle::framework::ir::GraphVizPass)
|
|
.RequirePassAttr(paddle::framework::ir::kGraphVizPath);
|