diff --git a/ge/single_op/task/op_task.cc b/ge/single_op/task/op_task.cc index 4e1e9863..84f2d8d3 100755 --- a/ge/single_op/task/op_task.cc +++ b/ge/single_op/task/op_task.cc @@ -369,10 +369,10 @@ Status AiCpuBaseTask::SetExtInfoAndType(const std::string &kernel_ext_info, uint return SUCCESS; } -Status SetInputConst() { +Status AiCpuTask::SetInputConst() { input_is_const_.clear(); const vector v_is_input_const = op_desc_->GetIsInputConst(); - for (size_t i = 0; i < op_desc->GetAllInputsSize(); ++i) { + for (size_t i = 0; i < op_desc_->GetAllInputsSize(); ++i) { const GeTensorDescPtr tensor_desc = op_desc_->MutableInputDesc(static_cast(i)); if (tensor_desc == nullptr) { GELOGD("SingleOp: %s, Index: %zu, has no input", op_desc_->GetName().c_str(), i); @@ -381,7 +381,7 @@ Status SetInputConst() { if (i < v_is_input_const.size() && v_is_input_const[i]) { GELOGD("SingleOp: %s, Index: %zu, input is const", op_desc_->GetName().c_str(), i); input_is_const_.push_back(true); - continue; + continue; } input_is_const_.push_back(false); } @@ -405,7 +405,7 @@ Status AiCpuBaseTask::UpdateExtInfo(const std::vector &input_desc, // get input_desc from op_desc if const input, num_inputs_ is op_desc_ input_size auto const_input_desc = op_desc_->MutableInputDesc(static_cast(input_index)); GE_CHECK_NOTNULL(const_input_desc); - GE_CHK_STATUS_RET(aicpu_ext_handle_->UpdateInputShapeAndType(i, *const_input_desc), + GE_CHK_STATUS_RET(aicpu_ext_handle_->UpdateInputShapeAndType(input_index, *const_input_desc), "Input[%zu] update input shape failed.", i); continue; } @@ -498,8 +498,8 @@ Status AiCpuBaseTask::UpdateIoAddr(const vector &inputs, const vecto if (input_index < input_is_const_.size() && input_is_const_[input_index]) { // const input no need update addr GE_CHECK_NOTNULL(arg_base); - GELOGD("AICpuTask input[%zu] addr = %p", input_index, *arg_base); - *arg_base; + GELOGD("AICpuTask input[%zu] addr = %u", input_index, *arg_base); + arg_base++; continue; } GE_CHK_BOOL_RET_STATUS(non_const_index < inputs.size(), PARAM_INVALID,