From 6de2e90e6f1672f169f12d3535a020c2904acec6 Mon Sep 17 00:00:00 2001 From: zhujingjing Date: Tue, 27 Jul 2021 11:30:03 +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 ff10362d..3ed8aa64 100644 --- a/ge/hybrid/node_executor/aicore/aicore_op_task.cc +++ b/ge/hybrid/node_executor/aicore/aicore_op_task.cc @@ -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(shape_buffer_->GetData()); GELOGD("Node:%s add shape buffer addr to args.", task_context.GetNodeName()); }