|
|
@@ -489,7 +489,7 @@ Status SingleOpModel::BuildKernelExTask(const domi::KernelExDef &kernel_def, AiC |
|
|
|
return SUCCESS; |
|
|
|
} |
|
|
|
|
|
|
|
Status SingleOpModel::BuildCpuKernelTask(const domi::KernelDef &kernel_def, AicpuCCTask **task, uint64_t kernel_id) { |
|
|
|
Status SingleOpModel::BuildCpuKernelTask(const domi::KernelDef &kernel_def, AiCpuCCTask **task, uint64_t kernel_id) { |
|
|
|
const auto &context = kernel_def.context(); |
|
|
|
auto iter = op_list_.find(context.op_index()); |
|
|
|
if (iter == op_list_.end()) { |
|
|
@@ -611,7 +611,7 @@ Status SingleOpModel::BuildTaskListForDynamicOp(StreamResource *stream_resource, |
|
|
|
} else if (lib_name == kEngineNameAiCpu) { |
|
|
|
const auto &task_def = task_defs[0]; |
|
|
|
GELOGD("Building AICPU_CC task"); |
|
|
|
AicpuCCTask *task = nullptr; |
|
|
|
AiCpuCCTask *task = nullptr; |
|
|
|
uint64_t dynamic_singleop_kernel_id = aicpu_kernel_id++; |
|
|
|
GELOGI("Build dynamic singleOp CCTask, kernel_id = %lu", dynamic_singleop_kernel_id); |
|
|
|
GE_CHK_STATUS_RET_NOLOG(BuildCpuKernelTask(task_def.kernel(), &task, dynamic_singleop_kernel_id)); |
|
|
|