Browse Source

For const input in dynamic aicpu single_op.

tags/v1.2.0
unknown 3 years ago
parent
commit
df480ba44f
1 changed files with 6 additions and 6 deletions
  1. +6
    -6
      ge/single_op/task/op_task.cc

+ 6
- 6
ge/single_op/task/op_task.cc View File

@@ -369,10 +369,10 @@ Status AiCpuBaseTask::SetExtInfoAndType(const std::string &kernel_ext_info, uint
return SUCCESS; return SUCCESS;
} }


Status SetInputConst() {
Status AiCpuTask::SetInputConst() {
input_is_const_.clear(); input_is_const_.clear();
const vector<bool> v_is_input_const = op_desc_->GetIsInputConst(); const vector<bool> 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<uint32_t>(i)); const GeTensorDescPtr tensor_desc = op_desc_->MutableInputDesc(static_cast<uint32_t>(i));
if (tensor_desc == nullptr) { if (tensor_desc == nullptr) {
GELOGD("SingleOp: %s, Index: %zu, has no input", op_desc_->GetName().c_str(), i); 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]) { 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); GELOGD("SingleOp: %s, Index: %zu, input is const", op_desc_->GetName().c_str(), i);
input_is_const_.push_back(true); input_is_const_.push_back(true);
continue;
continue;
} }
input_is_const_.push_back(false); input_is_const_.push_back(false);
} }
@@ -405,7 +405,7 @@ Status AiCpuBaseTask::UpdateExtInfo(const std::vector<GeTensorDesc> &input_desc,
// get input_desc from op_desc if const input, num_inputs_ is op_desc_ input_size // 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<uint32_t>(input_index)); auto const_input_desc = op_desc_->MutableInputDesc(static_cast<uint32_t>(input_index));
GE_CHECK_NOTNULL(const_input_desc); 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); "Input[%zu] update input shape failed.", i);
continue; continue;
} }
@@ -498,8 +498,8 @@ Status AiCpuBaseTask::UpdateIoAddr(const vector<DataBuffer> &inputs, const vecto
if (input_index < input_is_const_.size() && input_is_const_[input_index]) { if (input_index < input_is_const_.size() && input_is_const_[input_index]) {
// const input no need update addr // const input no need update addr
GE_CHECK_NOTNULL(arg_base); 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; continue;
} }
GE_CHK_BOOL_RET_STATUS(non_const_index < inputs.size(), PARAM_INVALID, GE_CHK_BOOL_RET_STATUS(non_const_index < inputs.size(), PARAM_INVALID,


Loading…
Cancel
Save