|
|
@@ -381,7 +381,7 @@ Status SingleOpModel::BuildCpuKernelTask(const domi::KernelDef &kernel_def, OpTa |
|
|
|
} |
|
|
|
|
|
|
|
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) { |
|
|
|
GELOGE(ret, "build aicpu_CC op task failed"); |
|
|
|
return ret; |
|
|
|