Browse Source

For const input in dynamic aicpu single_op.

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

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

@@ -406,7 +406,7 @@ Status AiCpuBaseTask::UpdateExtInfo(const std::vector<GeTensorDesc> &input_desc,
auto const_input_desc = op_desc_->MutableInputDesc(static_cast<uint32_t>(input_index));
GE_CHECK_NOTNULL(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.", input_index);
continue;
}
GE_CHK_BOOL_RET_STATUS(non_const_index < input_desc.size(), PARAM_INVALID,


Loading…
Cancel
Save