block comment modify

4 years ago · efb7f4f35c
parent d19de55b3c
commit efb7f4f35c
1 changed files with 1 additions and 1 deletions
--- 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);
    }