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 bfb6e24b..924aa354 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 @@ -148,7 +148,7 @@ Status KernelTaskInfo::Init(const domi::TaskDef &task_def, DavinciModel *davinci ret = InitAICPUCustomTask(context.op_index(), kernel_def); } else if (kernel_type_ == ccKernelType::AI_CPU || kernel_type_ == ccKernelType::CUST_AI_CPU) { ret = InitAicpuTask(context.op_index(), kernel_def); - } else { + } else if (kernel_type_ != ccKernelType::HOST_CPU) { // Skip host cpu task in davinci model. if (kernel_def.args().empty() || args_size_ == 0) { REPORT_INNER_ERROR("E19999", "kernel_def.args() is empty, op:%s(%s), check invalid", op_desc_->GetName().c_str(), op_desc_->GetType().c_str()); @@ -444,7 +444,7 @@ Status KernelTaskInfo::Distribute() { reinterpret_cast(kernel_name_.c_str()), 1, args_, args_size_, nullptr, stream_, dump_flag_); call_save_dump_ = true; - } else { + } else if (kernel_type_ != ccKernelType::HOST_CPU) { // Skip host cpu task in davinci model. /* default: not skt launch */ const SuperKernelTaskInfo &skt_info = davinci_model_->GetSuperKernelTaskInfo(); GELOGD(