Skip to content

Commit 839b682

Browse files
authored
[XPU] fc pass and delete pass nodes check (#60314)
1 parent ccf7bd4 commit 839b682

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

paddle/fluid/framework/ir/xpu/delete_isolated_node_pass.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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);

paddle/fluid/framework/ir/xpu/pass_utils.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ int ConvertActivationType(std::string act_type) {
6464

6565
Node* 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
}

0 commit comments

Comments
 (0)