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