pull/606/head
chuxing 4 years ago
parent b373acb0e2
commit 3101d4d4b2

@ -381,7 +381,7 @@ Status SingleOpModel::BuildCpuKernelTask(const domi::KernelDef &kernel_def, OpTa
} }
auto builder = AiCpuCCTaskBuilder(iter->second->GetOpDesc(), kernel_def); auto builder = AiCpuCCTaskBuilder(iter->second->GetOpDesc(), kernel_def);
auto ret = builder.BuildTask(*aicpucc_task, kernel_id); auto ret = builder.BuildTask(*aicpucc_task, kernel_id, model_params_);
if (ret != SUCCESS) { if (ret != SUCCESS) {
GELOGE(ret, "build aicpu_CC op task failed"); GELOGE(ret, "build aicpu_CC op task failed");
return ret; return ret;

Loading…
Cancel
Save