Browse Source

Pre Merge pull request !1875 from 赵之轩/my_dev4

pull/1875/MERGE
赵之轩 Gitee 4 years ago
parent
commit
9f38418de1
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/graph/load/model_manager/task_info/kernel_task_info.cc

+ 2
- 2
ge/graph/load/model_manager/task_info/kernel_task_info.cc View File

@@ -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(


Loading…
Cancel
Save