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<uintptr_t>(shape_buffer_->GetData());
     GELOGD("Node:%s add shape buffer addr to args.", task_context.GetNodeName());
   }