diff --git a/ge/single_op/task/op_task.cc b/ge/single_op/task/op_task.cc
index 0e12b17f..3f42cdf4 100755
--- a/ge/single_op/task/op_task.cc
+++ b/ge/single_op/task/op_task.cc
@@ -464,7 +464,7 @@ Status AiCpuBaseTask::UpdateExtInfo(const std::vector<GeTensorDesc> &input_desc,
     GE_CHK_STATUS_RET(aicpu_ext_handle_->UpdateInputShapeAndType(input_index, input_desc[non_const_index]),
                       "Input[%zu] update input shape failed.", input_index);
     if (DumpManager::GetInstance().GetDumpProperties(kInferSessionId).IsSingleOpNeedDump()) {
-      GE_CHK_STATUS_RET(op_desc_->UpdateInputDesc(input_index, input_desc[non_const_index]),
+      GE_CHK_STATUS_RET(op_desc_->UpdateInputDesc(input_index, input_desc[input_index]),
                         "AicpuTask Update [%zu]th input desc failed", input_index);
     }
     non_const_index++;