From 191171ce8eb8648161701ec06d9a5cad0ec83aeb Mon Sep 17 00:00:00 2001 From: zhujingjing Date: Tue, 27 Jul 2021 11:15:59 +0800 Subject: [PATCH] GE supports aicore DEPEND_SHAPE_RANGE op update outputs shape --- ge/hybrid/node_executor/aicore/aicore_op_task.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/hybrid/node_executor/aicore/aicore_op_task.cc b/ge/hybrid/node_executor/aicore/aicore_op_task.cc index 06e9d59d..ff10362d 100644 --- a/ge/hybrid/node_executor/aicore/aicore_op_task.cc +++ b/ge/hybrid/node_executor/aicore/aicore_op_task.cc @@ -74,7 +74,6 @@ Status AiCoreOpTask::Init(const OpDesc &op_desc, const domi::TaskDef &task_def) shape_buffer_ = TensorBuffer::Create(allocator, size); GE_CHECK_NOTNULL(shape_buffer_); GELOGD("Op [%s] allocate memory for outputs shape success, size=%zu", op_desc.GetName().c_str(), size); - GE_CHK_RT_RET(rtMemset(shape_buffer_->GetData(), shape_buffer_->GetSize(), 0, size)); host_shape_buffer_.reset(new (std::nothrow) uint8_t[shape_buffer_->GetSize()]); GE_CHECK_NOTNULL(host_shape_buffer_); } @@ -569,6 +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)); arg_base_[index++] = reinterpret_cast(shape_buffer_->GetData()); GELOGD("Node:%s add shape buffer addr to args.", task_context.GetNodeName()); }