Browse Source

update

tags/v1.2.0
chuxing 3 years ago
parent
commit
3101d4d4b2
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/single_op/single_op_model.cc

+ 1
- 1
ge/single_op/single_op_model.cc View File

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


Loading…
Cancel
Save