From efb7f4f35c0b20dc145008b0d50708be54b5d97a Mon Sep 17 00:00:00 2001 From: wangxiaotian22 Date: Mon, 28 Dec 2020 17:50:15 +0800 Subject: [PATCH] block comment modify --- ge/graph/build/memory/block_mem_assigner.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/graph/build/memory/block_mem_assigner.cc b/ge/graph/build/memory/block_mem_assigner.cc index aab2ad41..493f08b5 100755 --- a/ge/graph/build/memory/block_mem_assigner.cc +++ b/ge/graph/build/memory/block_mem_assigner.cc @@ -562,7 +562,7 @@ void BlockMemAssigner::MarkContinuousAllocedForOneInputFromVariable(const NodePt GE_IF_BOOL_EXEC(peer_out_anchor == nullptr, return); auto in_node = peer_out_anchor->GetOwnerNode(); GE_IF_BOOL_EXEC(in_node == nullptr, return); - if (in_node->GetType() == VARIABLE) { + if (in_node->GetType() == VARIABLE || in_node->GetType() == CONSTANT) { GELOGI("node only one input and from variable, set continuous alloced. node_name:%s", node->GetName().c_str()); (void)ge::AttrUtils::SetBool(node_op_desc, ATTR_NAME_CONTINUOUS_INPUT_ALLOC, true); }