fix

paddle/fluid/framework/ir/graph_viz_pass.cc

@@ -27,14 +27,14 @@ using inference::analysis::Dot;
 namespace {
 const char kGraphVizPath[] = "graph_viz_path";
 
-std::string FormatName(const Node* op) {
+std::string FormatName(const Node* node) {
-  if (!op->Op() ||
+  if (!node->IsOp() || !node->Op() ||
-      !op->Op()->HasAttr(OpProtoAndCheckerMaker::OpNamescopeAttrName())) {
+      !node->Op()->HasAttr(OpProtoAndCheckerMaker::OpNamescopeAttrName())) {
-    return op->Name();
+    return node->Name();
   }
   const std::string full_scope = boost::get<std::string>(
-      op->Op()->GetAttr(OpProtoAndCheckerMaker::OpNamescopeAttrName()));
+      node->Op()->GetAttr(OpProtoAndCheckerMaker::OpNamescopeAttrName()));
-  return string::Sprintf("%s%s", full_scope.c_str(), op->Name().c_str());
+  return string::Sprintf("%s%s", full_scope.c_str(), node->Name().c_str());
 }
 }  // namespace
 

paddle/fluid/framework/ir/node.h

@@ -55,7 +55,7 @@ class Node {
   std::string Name() const { return name_; }
 
   VarDesc* Var() {
-    PADDLE_ENFORCE(type_ == Type::kVariable);
+    PADDLE_ENFORCE(IsVar());
     return var_desc_.get();
   }