Browse Source

For aicpu all_shape compile.

tags/v1.2.0
unknown 3 years ago
parent
commit
df604f679b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/load/model_manager/task_info/kernel_task_info.cc

+ 1
- 1
ge/graph/load/model_manager/task_info/kernel_task_info.cc View File

@@ -979,7 +979,7 @@ Status KernelTaskInfo::InitAicpuTaskExtInfo(const std::string &ext_info) {
GE_CHK_BOOL_RET_STATUS(ext_handle != nullptr, FAILED, "Malloc aicpu_ext_handle mem failed!");
GE_CHK_STATUS_RET(ext_handle->Parse(ext_info),
"Parse kernel ext info failed, kernel_ext_info_size=%zu.", ext_info.size());
GE_CHK_STATUS_RET(ext_handle_->UpdateSessionInfoSessionId(davinci_model_->GetSessionId()),
GE_CHK_STATUS_RET(ext_handle->UpdateSessionInfoSessionId(davinci_model_->GetSessionId()),
"Update session info session id failed.");
GELOGD("Update aicpu_task ext_info session_info session_id is %lu", davinci_model_->GetSessionId());
GE_CHK_STATUS_RET(ext_handle->UpdateExecuteMode(true), "UpdateExecuteMode failed.");


Loading…
Cancel
Save