From 62d277af483ba03cc7852d5ad77c0e673226cfc5 Mon Sep 17 00:00:00 2001 From: chenyemeng Date: Wed, 2 Dec 2020 12:02:07 +0800 Subject: [PATCH] modify for security check --- ge/graph/passes/cond_remove_pass.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/ge/graph/passes/cond_remove_pass.cc b/ge/graph/passes/cond_remove_pass.cc index bf2e1170..62272809 100644 --- a/ge/graph/passes/cond_remove_pass.cc +++ b/ge/graph/passes/cond_remove_pass.cc @@ -82,6 +82,7 @@ Status CondRemovePass::Run(NodePtr &node) { Status CondRemovePass::RemoveDeadCondLink(const int32_t index, const NodePtr &node) { const auto &in_anchor = node->GetInDataAnchor(index); + GE_CHECK_NOTNULL(in_anchor); const auto &peerout_anchor = in_anchor->GetPeerOutAnchor(); if (GraphUtils::RemoveEdge(peerout_anchor, in_anchor) != SUCCESS) { GELOGE(FAILED, "Remove edge from node %s index %d to node %s index %d.",