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