|
|
|
@@ -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<const void *>(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( |
|
|
|
|