|
@@ -349,9 +349,17 @@ Status DynamicSingleOp::ExecuteAsync(const vector<GeTensorDesc> &input_desc, |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
if (op_task_->GetOpTaskType() == OP_TASK_TBE) { |
|
|
if (op_task_->GetOpTaskType() == OP_TASK_TBE) { |
|
|
return ExecuteTbeTask(input_desc, inputs, output_desc, outputs); |
|
|
|
|
|
|
|
|
auto ret = ExecuteTbeTask(input_desc, inputs, output_desc, outputs); |
|
|
|
|
|
if (ret == SUCCESS) { |
|
|
|
|
|
GE_CHK_STATUS_RET_NOLOG(ProfilingTaskInfo()); |
|
|
|
|
|
} |
|
|
|
|
|
return ret; |
|
|
} else if (op_task_->GetOpTaskType() == OP_TASK_AICPU || op_task_->GetOpTaskType() == OP_TASK_AICPUCC) { |
|
|
} else if (op_task_->GetOpTaskType() == OP_TASK_AICPU || op_task_->GetOpTaskType() == OP_TASK_AICPUCC) { |
|
|
return op_task_->LaunchKernel(input_desc, input_buffers, output_desc, output_buffers, stream_); |
|
|
|
|
|
|
|
|
auto aicpu_ret = op_task_->LaunchKernel(input_desc, input_buffers, output_desc, output_buffers, stream_); |
|
|
|
|
|
if (aicpu_ret == SUCCESS) { |
|
|
|
|
|
GE_CHK_STATUS_RET_NOLOG(ProfilingTaskInfo()); |
|
|
|
|
|
} |
|
|
|
|
|
return aicpu_ret; |
|
|
} else { |
|
|
} else { |
|
|
GELOGE(ACL_ERROR_GE_OP_TASK_TYPE_INVALID, |
|
|
GELOGE(ACL_ERROR_GE_OP_TASK_TYPE_INVALID, |
|
|
"Only TBE_Task, AI_CPU_Task and AI_CPUCC_Task are supported, but got %u", |
|
|
"Only TBE_Task, AI_CPU_Task and AI_CPUCC_Task are supported, but got %u", |
|
|