From 3110dc94c1b399a25c9fc856c8dc3e2382dcc801 Mon Sep 17 00:00:00 2001 From: lianghuikang <505519763@qq.com> Date: Sat, 3 Jul 2021 11:29:02 +0800 Subject: [PATCH] fix core dump --- tf_adapter/optimizers/control_flow_conversion_pass.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/tf_adapter/optimizers/control_flow_conversion_pass.cc b/tf_adapter/optimizers/control_flow_conversion_pass.cc index 0f1e33c2a..c6d7c9afe 100644 --- a/tf_adapter/optimizers/control_flow_conversion_pass.cc +++ b/tf_adapter/optimizers/control_flow_conversion_pass.cc @@ -69,6 +69,7 @@ Status ControlFlowConversionPass::Run(const GraphOptimizationPassOptions &option // Delete _lower_using_switch_merge before LowerFunctionalOpsPass for (int i = 2; i < graph->num_node_ids(); ++i) { Node *n = graph->FindNodeId(i); + if (n == nullptr) { continue; } if (n->IsIfNode() || n->type_string() == "Case" || n->IsWhileNode()) { n->ClearAttr(kLowerUsingSwitchMergeAttr); } } -- Gitee