From 114ea845ad928f16cacb89149cbccbcb0122d886 Mon Sep 17 00:00:00 2001 From: wjm Date: Sat, 12 Jun 2021 10:10:21 +0800 Subject: [PATCH] fix --- ge/graph/load/model_manager/cpu_queue_schedule.cc | 3 ++- ge/graph/load/model_manager/davinci_model.cc | 4 ++-- ge/graph/load/model_manager/davinci_model.h | 8 +------- 3 files changed, 5 insertions(+), 10 deletions(-) diff --git a/ge/graph/load/model_manager/cpu_queue_schedule.cc b/ge/graph/load/model_manager/cpu_queue_schedule.cc index 63338dff..0fa13139 100644 --- a/ge/graph/load/model_manager/cpu_queue_schedule.cc +++ b/ge/graph/load/model_manager/cpu_queue_schedule.cc @@ -238,6 +238,7 @@ Status CpuTaskModelPostpare::Init(uint32_t model_id, const vector &out } AicpuPareInfo aicpu_info; + aicpu_info.aicpu_info_size = sizeof(AicpuPareInfo); aicpu_info.model_id = model_id; aicpu_info.out_queue_num = output_queue_ids.size(); aicpu_info.out_queueid_list = static_cast(reinterpret_cast(queue_id_list_addr_)); @@ -288,7 +289,7 @@ CpuTaskModelPostpare::~CpuTaskModelPostpare() { /// @ingroup ge /// @brief definiteness queue schedule, active entry stream. /// @param [in] stream: stream to be active. -/// @return: 0 for success / others for failCpuTaskModelPostpareed +/// @return: 0 for success / others for failed /// Status CpuTaskActiveEntry::Init(rtStream_t stream) { if (stream == nullptr) { diff --git a/ge/graph/load/model_manager/davinci_model.cc b/ge/graph/load/model_manager/davinci_model.cc index 7c35ff1d..1f76a1a2 100755 --- a/ge/graph/load/model_manager/davinci_model.cc +++ b/ge/graph/load/model_manager/davinci_model.cc @@ -1687,7 +1687,7 @@ Status DavinciModel::CpuPostpare() { std::shared_ptr postpare_task = MakeShared(rt_entry_stream_); if (postpare_task == nullptr) { REPORT_CALL_ERROR("E19999", "New CpuTaskModelPostpare failed, model_id:%u", model_id_); - GELOGE(MEMALLOC_FAILED, "[New][CpuTaskModelPostpare] failed, model_id:%u", model_id_); + GELOGE(MEMALLOC_FAILED, "[New][CpuTaskModelPostpare] task failed, model_id:%u", model_id_); return MEMALLOC_FAILED; } @@ -1697,7 +1697,7 @@ Status DavinciModel::CpuPostpare() { } cpu_task_list_.push_back(postpare_task); - GELOGI("Set CpuKernel model postpare task enter."); + GELOGI("Set CpuKernel model postpare task success."); return SUCCESS; } diff --git a/ge/graph/load/model_manager/davinci_model.h b/ge/graph/load/model_manager/davinci_model.h index 32db1b8d..31791c47 100755 --- a/ge/graph/load/model_manager/davinci_model.h +++ b/ge/graph/load/model_manager/davinci_model.h @@ -803,11 +803,6 @@ class DavinciModel { /// Status LoadWithQueue(); - /// - /// @ingroup ge - /// @brief ACL, Bind Data Op addr to input queue. - /// @return: 0 for success / others for fail - /// Status CpuPrepare(); Status CpuPostpare(); @@ -966,8 +961,7 @@ class DavinciModel { vector cpu_task_list_; vector input_queue_ids_; // input queue ids created by caller. vector output_queue_ids_; // output queue ids created by caller. - vector input_mbuf_list_; // input mbuf created by dequeue task. - vector output_mbuf_list_; // output mbuf created by dequeue task. + vector output_mbuf_list_; uint64_t session_id_; struct error_message::Context error_context_;