From 5b001c5684914fc1f4a1caee85cb192318c79ed0 Mon Sep 17 00:00:00 2001 From: unknown Date: Mon, 25 Jan 2021 22:06:18 +0800 Subject: [PATCH] For aicpu all_shape compile. --- ge/graph/load/model_manager/task_info/kernel_task_info.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/graph/load/model_manager/task_info/kernel_task_info.cc b/ge/graph/load/model_manager/task_info/kernel_task_info.cc index 2a3ddaf8..861ef484 100755 --- a/ge/graph/load/model_manager/task_info/kernel_task_info.cc +++ b/ge/graph/load/model_manager/task_info/kernel_task_info.cc @@ -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_->UpdateSessionInfo(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.");