|
|
@@ -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()); |
|
|
|
} |
|
|
|