File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
paddle/fluid/framework/ir/xpu Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -100,7 +100,7 @@ void DeleteIsolatedNodePass::CollectReservedPersistableNodeNames(
100100 std::unordered_set<std::string>* reserved_persistable_node_names) const {
101101 for (auto * node : graph->Nodes ()) {
102102 if (!node || node->Name () == " fetch" || node->Name () == " feed" ) continue ;
103- if (!node->IsVar () || !node->Var ()->Persistable ()) continue ;
103+ if (!node->IsVar () || !node->Var () || !node-> Var () ->Persistable ()) continue ;
104104 for (auto * out_node : node->outputs ) {
105105 auto op_type = out_node->Op ()->Type ();
106106 if (control_flow_op_input_map_.count (op_type) == 0 ) {
@@ -135,7 +135,7 @@ int DeleteIsolatedNodePass::RemoveIsolatedNodes(
135135 const std::unordered_set<ir::Node*> nodes = graph->Nodes ();
136136 for (auto * node : nodes) {
137137 if (!node || node->Name () == " fetch" || node->Name () == " feed" ) continue ;
138- if (!node->IsVar () || !node->Var ()->Persistable ()) continue ;
138+ if (!node->IsVar () || !node->Var () || !node-> Var () ->Persistable ()) continue ;
139139 auto name = node->Var ()->Name ();
140140 if (reserved_persistable_node_names.count (name) > 0 ) continue ;
141141 delete_nodes.insert (node);
Original file line number Diff line number Diff line change @@ -64,7 +64,7 @@ int ConvertActivationType(std::string act_type) {
6464
6565Node* FindNodeWithName (Graph* graph, std::string name) {
6666 for (auto * node : graph->Nodes ()) {
67- if (node->IsVar () && node->Var ()->Name () == name) {
67+ if (node->IsVar () && node->Var () && node-> Var () ->Name () == name) {
6868 return node;
6969 }
7070 }
You can’t perform that action at this time.
0 commit comments