Browse Source

GE supports aicore DEPEND_SHAPE_RANGE op update outputs shape

pull/2008/head
zhujingjing 3 years ago
parent
commit
6de2e90e6f
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/node_executor/aicore/aicore_op_task.cc

+ 1
- 1
ge/hybrid/node_executor/aicore/aicore_op_task.cc View File

@@ -568,7 +568,7 @@ Status AiCoreOpTask::UpdateArgs(TaskContext &task_context) {
}

if (shape_buffer_ != nullptr) {
GE_CHK_RT_RET(rtMemset(shape_buffer_->GetData(), shape_buffer_->GetSize(), 0, size));
GE_CHK_RT_RET(rtMemset(shape_buffer_->GetData(), shape_buffer_->GetSize(), 0, shape_buffer_->GetSize()));
arg_base_[index++] = reinterpret_cast<uintptr_t>(shape_buffer_->GetData());
GELOGD("Node:%s add shape buffer addr to args.", task_context.GetNodeName());
}


Loading…
Cancel
Save